00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038 #include <BandGenLinLapackSolver.h>
00039 #include <BandGenLinSOE.h>
00040 #include <f2c.h>
00041 #include <math.h>
00042
00043
00044 BandGenLinLapackSolver::BandGenLinLapackSolver()
00045 :BandGenLinSolver(SOLVER_TAGS_BandGenLinLapackSolver),
00046 iPiv(0), iPivSize(0)
00047 {
00048
00049 }
00050
00051 BandGenLinLapackSolver::~BandGenLinLapackSolver()
00052 {
00053 if (iPiv != 0)
00054 delete [] iPiv;
00055 }
00056
00057 #ifdef _WIN32
00058
00059 extern "C" int DGBSV(int *N, int *KL, int *KU, int *NRHS, double *A,
00060 int *LDA, int *iPiv, double *B, int *LDB,
00061 int *INFO);
00062
00063 extern "C" int DGBTRS(char *TRANS,
00064 int *N, int *KL, int *KU, int *NRHS,
00065 double *A, int *LDA, int *iPiv,
00066 double *B, int *LDB, int *INFO);
00067
00068 #else
00069
00070 extern "C" int dgbsv_(int *N, int *KL, int *KU, int *NRHS, double *A,
00071 int *LDA, int *iPiv, double *B, int *LDB, int *INFO);
00072
00073
00074 extern "C" int dgbtrs_(char *TRANS, int *N, int *KL, int *KU, int *NRHS,
00075 double *A, int *LDA, int *iPiv, double *B, int *LDB,
00076 int *INFO);
00077 #endif
00078 int
00079 BandGenLinLapackSolver::solve(void)
00080 {
00081 if (theSOE == 0) {
00082 opserr << "WARNING BandGenLinLapackSolver::solve(void)- ";
00083 opserr << " No LinearSOE object has been set\n";
00084 return -1;
00085 }
00086
00087 int n = theSOE->size;
00088
00089 if (iPivSize < n) {
00090 opserr << "WARNING BandGenLinLapackSolver::solve(void)- ";
00091 opserr << " iPiv not large enough - has setSize() been called?\n";
00092 return -1;
00093 }
00094
00095 int kl = theSOE->numSubD;
00096 int ku = theSOE->numSuperD;
00097 int ldA = 2*kl + ku +1;
00098 int nrhs = 1;
00099 int ldB = n;
00100 int info;
00101 double *Aptr = theSOE->A;
00102 double *Xptr = theSOE->X;
00103 double *Bptr = theSOE->B;
00104 int *iPIV = iPiv;
00105
00106
00107 for (int i=0; i<n; i++) {
00108 *(Xptr++) = *(Bptr++);
00109 }
00110 Xptr = theSOE->X;
00111
00112
00113
00114 #ifdef _WIN32
00115 {if (theSOE->factored == false)
00116
00117 DGBSV(&n,&kl,&ku,&nrhs,Aptr,&ldA,iPIV,Xptr,&ldB,&info);
00118 else {
00119
00120 unsigned int sizeC = 1;
00121
00122 DGBTRS("N", &n,&kl,&ku,&nrhs,Aptr,&ldA,iPIV,Xptr,&ldB,&info);
00123 }}
00124 #else
00125 {if (theSOE->factored == false)
00126
00127 dgbsv_(&n,&kl,&ku,&nrhs,Aptr,&ldA,iPIV,Xptr,&ldB,&info);
00128 else
00129
00130 dgbtrs_("N",&n,&kl,&ku,&nrhs,Aptr,&ldA,iPIV,Xptr,&ldB,&info);
00131 }
00132 #endif
00133
00134 if (info != 0) {
00135 opserr << "WARNING BandGenLinLapackSolver::solve() -";
00136 opserr << "LAPACK routine returned " << info << endln;
00137 return -info;
00138 }
00139
00140 theSOE->factored = true;
00141 return 0;
00142 }
00143
00144
00145
00146 int
00147 BandGenLinLapackSolver::setSize()
00148 {
00149
00150 if (iPivSize < theSOE->size) {
00151 if (iPiv != 0)
00152 delete [] iPiv;
00153
00154 iPiv = new int[theSOE->size];
00155 if (iPiv == 0) {
00156 opserr << "WARNING BandGenLinLapackSolver::setSize() ";
00157 opserr << " - ran out of memory for iPiv of size ";
00158 opserr << theSOE->size << endln;
00159 return -1;
00160 } else
00161 iPivSize = theSOE->size;
00162 }
00163
00164 return 0;
00165 }
00166
00167 int
00168 BandGenLinLapackSolver::sendSelf(int commitTag, Channel &theChannel)
00169 {
00170 return 0;
00171 }
00172
00173 int
00174 BandGenLinLapackSolver::recvSelf(int commitTag,
00175 Channel &theChannel,
00176 FEM_ObjectBroker &theBroker)
00177 {
00178
00179 return 0;
00180 }