My Project
_zhematrix-zgbmatrix.hpp
1 //=============================================================================
3 inline _zgematrix operator+(const _zhematrix& matA, const zgbmatrix& matB)
4 {
5 #ifdef CPPL_VERBOSE
6  std::cerr << "# [MARK] operator+(const _zhematrix&, const zgbmatrix&)"
7  << std::endl;
8 #endif//CPPL_VERBOSE
9 
10 #ifdef CPPL_DEBUG
11  if(matA.N!=matB.N || matA.N!=matB.M){
12  std::cerr << "[ERROR] operator+(_zgematrix&, zgbmatrix&)" << std::endl
13  << "These two matrises can not make a summation." << std::endl
14  << "Your input was (" << matA.N << "x" << matA.N << ") + ("
15  << matB.M << "x" << matB.N << ")." << std::endl;
16  exit(1);
17  }
18 #endif//CPPL_DEBUG
19 
20  zgematrix newmat(matA.N, matA.N);
21  for(long i=0; i<matB.M; i++){
22  for(long j=0; j<matA.N; j++){
23  newmat(i,j) =matA(i,j);
24  }
25  for(long j=max(0,i-matB.KL); j<min(matB.N,i+matB.KU+1); j++){
26  newmat(i,j)+=matB(i,j);
27  }
28  }
29 
30  matA.destroy();
31  return _(newmat);
32 }
33 
34 //=============================================================================
36 inline _zgematrix operator-(const _zhematrix& matA, const zgbmatrix& matB)
37 {
38 #ifdef CPPL_VERBOSE
39  std::cerr << "# [MARK] operator-(const _zhematrix&, const zgbmatrix&)"
40  << std::endl;
41 #endif//CPPL_VERBOSE
42 
43 #ifdef CPPL_DEBUG
44  if(matA.N!=matB.N || matA.N!=matB.M){
45  std::cerr << "[ERROR] operator+(_zhematrix&, zgbmatrix&)" << std::endl
46  << "These two matrises can not make a summation." << std::endl
47  << "Your input was (" << matA.N << "x" << matA.N << ") + ("
48  << matB.M << "x" << matB.N << ")." << std::endl;
49  exit(1);
50  }
51 #endif//CPPL_DEBUG
52 
53  zgematrix newmat(matA.N, matA.N);
54  for(long i=0; i<matB.M; i++){
55  for(long j=0; j<matA.N; j++){
56  newmat(i,j) =matA(i,j);
57  }
58  for(long j=max(0,i-matB.KL); j<min(matB.N,i+matB.KU+1); j++){
59  newmat(i,j)-=matB(i,j);
60  }
61  }
62 
63  matA.destroy();
64  return _(newmat);
65 }
66 
67 //=============================================================================
69 inline _zgematrix operator*(const _zhematrix& matA, const zgbmatrix& matB)
70 {
71 #ifdef CPPL_VERBOSE
72  std::cerr << "# [MARK] operator*(const _zhematrix&, const zgbmatrix&)"
73  << std::endl;
74 #endif//CPPL_VERBOSE
75 
76 #ifdef CPPL_DEBUG
77  if(matA.N!=matB.M){
78  std::cerr << "[ERROR] operator*(_zhematrix&, zgbmatrix&)" << std::endl
79  << "These two matrises can not make a product." << std::endl
80  << "Your input was (" << matA.N << "x" << matA.N << ") * ("
81  << matB.M << "x" << matB.N << ")." << std::endl;
82  exit(1);
83  }
84 #endif//CPPL_DEBUG
85 
86  zgematrix newmat( matA.N, matB.N );
87  newmat.zero();
88 
89  long i, j, k;
90 #pragma omp parallel for private(j,k)
91  for(i=0; i<newmat.m; i++){
92  for(j=0; j<newmat.n; j++){
93  for(k=max(0,j-matB.KU); k<min(matB.M,j+matB.KL+1); k++){
94  newmat(i,j)+=matA(i,k)*matB(k,j);
95  }
96  }
97  }
98 
99  matA.destroy();
100  return _(newmat);
101 }
(DO NOT USE) Smart-temporary Complex Double-precision Hermitian Matrix Class
Definition: _zhematrix.hpp:3
friend _zrovector operator-(const _zrovector &)
Definition: _zrovector-unary.hpp:15
Complex Double-precision General Dence Matrix Class.
Definition: zgematrix.hpp:3
(DO NOT USE) Smart-temporary Complex Double-precision General Dence Matrix Class
Definition: _zgematrix.hpp:3
void zero()
Definition: zgematrix-misc.hpp:26
Complex Double-precision General Band Matrix Class.
Definition: zgbmatrix.hpp:3
void destroy() const
Definition: _zhematrix-misc.hpp:3
friend _zrovector operator*(const zrovector &, const zgematrix &)
Definition: zrovector-zgematrix.hpp:3
long const & m
matrix row size (readable)
Definition: zgematrix.hpp:14
long N
matrix column or row size
Definition: _zhematrix.hpp:7
long const & n
matrix column size (readable)
Definition: zgematrix.hpp:15
friend const _zrovector & operator+(const _zrovector &)
Definition: _zrovector-unary.hpp:3