Menu

[r37]: / trunk / include / dgbmatrix- / dgbmatrix-_dsymatrix.hpp  Maximize  Restore  History

Download this file

102 lines (89 with data), 3.1 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
//=============================================================================
/*! dgbmatrix+_dsymatrix operator */
inline _dgematrix operator+(const dgbmatrix& matA, const _dsymatrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator+(const dgbmatrix&, const _dsymatrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N || matA.M!=matB.N){
std::cerr << "[ERROR] operator+(dgbmatrix&, _dgematrix&)" << std::endl
<< "These two matrises can not make a summation." << std::endl
<< "Your input was (" << matA.M << "x" << matA.N << ") + ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
dgematrix newmat(matB.N,matB.N);
for(long i=0; i<matA.M; i++){
for(long j=0; j<matB.N; j++){
newmat(i,j) = matB(i,j);
}
for(long j=max(0,i-matA.KL); j<min(matA.N,i+matA.KU+1); j++){
newmat(i,j)+=matA(i,j);
}
}
matB.destroy();
return _(newmat);
}
//=============================================================================
/*! dgbmatrix-_dsymatrix operator */
inline _dgematrix operator-(const dgbmatrix& matA, const _dsymatrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator-(const dgbmatrix&, const _dsymatrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N || matA.M!=matB.N){
std::cerr << "[ERROR] operator+(dgbmatrix&, _dsymatrix&)" << std::endl
<< "These two matrises can not make a summation." << std::endl
<< "Your input was (" << matA.M << "x" << matA.N << ") + ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
dgematrix newmat(matB.N,matB.N);
for(long i=0; i<matA.M; i++){
for(long j=0; j<matB.N; j++){
newmat(i,j) = -matB(i,j);
}
for(long j=max(0,i-matA.KL); j<min(matA.N,i+matA.KU+1); j++){
newmat(i,j) += matA(i,j);
}
}
matB.destroy();
return _(newmat);
}
//=============================================================================
/*! dgbmatrix*_dsymatrix operator */
inline _dgematrix operator*(const dgbmatrix& matA, const _dsymatrix& matB)
{
#ifdef CPPL_VERBOSE
std::cerr << "# [MARK] operator*(const dgbmatrix&, const _dsymatrix&)"
<< std::endl;
#endif//CPPL_VERBOSE
#ifdef CPPL_DEBUG
if(matA.N!=matB.N){
std::cerr << "[ERROR] operator*(dgbmatrix&, _dsymatrix&)" << std::endl
<< "These two matrises can not make a product." << std::endl
<< "Your input was (" << matA.M << "x" << matA.N << ") * ("
<< matB.N << "x" << matB.N << ")." << std::endl;
exit(1);
}
#endif//CPPL_DEBUG
dgematrix newmat( matA.M, matB.N );
newmat.zero();
long i, j, k;
#pragma omp parallel for private(j,k)
for(i=0; i<newmat.m; i++){
for(j=0; j<newmat.n; j++){
for(k=max(0,i-matA.KL); k<min(matA.N,i+matA.KU+1); k++){
newmat(i,j)+=matA(i,k)*matB(k,j);
}
}
}
matB.destroy();
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.