Menu

[r31]: / trunk / include / zhematrix- / zhematrix-zhematrix.hpp  Maximize  Restore  History

Download this file

156 lines (132 with data), 5.0 kB

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
//=============================================================================
/*! zhematrix=zhematrix operator */
inline zhematrix& zhematrix::operator=(const zhematrix& mat)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] zhematrix::operator=(const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
if(Array!=mat.Array){ // if it is NOT self substitution
copy(mat);
}
return *this;
}
///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////
//=============================================================================
/*! zhematrix+=zhematrix operator */
inline zhematrix& zhematrix::operator+=(const zhematrix& mat)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] zhematrix::operator+=(const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(N!=mat.N){
std::cerr << "[ERROR] zhematrix::operator+=(zhematrix&)" << std::endl
<< "These two matrises can not make a summation." << std::endl
<< "Your input was (" << N << "x" << N << ") += ("
<< mat.N << "x" << mat.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
for(long i=0; i<N; i++){ for(long j=0; j<=i; j++){
operator()(i,j) +=mat(i,j);
}}
return *this;
}
//=============================================================================
/*! zhematrix operator-= */
inline zhematrix& zhematrix::operator-=(const zhematrix& mat)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] zhematrix::operator-=(const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(N!=mat.N){
std::cerr << "[ERROR] zhematrix::operator-=(zhematrix&)" << std::endl
<< "These two matrises can not make a sutraction." << std::endl
<< "Your input was (" << N << "x" << N << ") -= ("
<< mat.N << "x" << mat.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
for(long i=0; i<N; i++){ for(long j=0; j<=i; j++){
operator()(i,j) -=mat(i,j);
}}
return *this;
}
//=============================================================================
/*! zhematrix+zhematrix operator */
inline _zhematrix operator+(const zhematrix& matA, const zhematrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator+(const zhematrix&, const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N){
std::cerr << "[ERROR] operator+(zhematrix&, zhematrix&)" << std::endl
<< "These two matrises can not make a summation." << std::endl
<< "Your input was (" << matA.N << "x" << matA.N << ") + ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
long N = matA.N;
zhematrix newmat(N);
for(long i=0; i<N; i++){ for(long j=0; j<=i; j++){
newmat.Array[i+N*j] = matA.Array[i+N*j] + matB.Array[i+N*j];
}}
return _(newmat);
}
//=============================================================================
/*! zhematrix-zhematrix operator */
inline _zhematrix operator-(const zhematrix& matA, const zhematrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator-(const zhematrix&, const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N){
std::cerr << "[ERROR] operator-(zhematrix&, zhematrix&)" << std::endl
<< "These two matrises can not make a subtraction." << std::endl
<< "Your input was (" << matA.N << "x" << matA.N << ") - ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
long N = matA.N;
zhematrix newmat(N);
for(long i=0; i<N; i++){ for(long j=0; j<=i; j++){
newmat.Array[i+N*j] =matA.Array[i+N*j]-matB.Array[i+N*j];
}}
return _(newmat);
}
//=============================================================================
/*! zhematrix*zhematrix operator */
inline _zgematrix operator*(const zhematrix& matA, const zhematrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator*(const zhematrix&, const zhematrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N){
std::cerr << "[ERROR] operator*(zhematrix&, zhematrix&)" << std::endl
<< "These two matrises can not make a product." << std::endl
<< "Your input was (" << matA.N << "x" << matA.N << ") * ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
matB.complete();
zgematrix newmat( matA.N, matB.N );
zhemm_( 'L', 'L', matA.N, matB.N, std::complex<double>(1.0,0.0),
matA.Array, matA.N, matB.Array, matB.N,
std::complex<double>(0.0,0.0), newmat.array, newmat.m );
return _(newmat);
}
Want the latest updates on software, tech news, and AI?
Get latest updates about software, tech news, and AI from SourceForge directly in your inbox once a month.