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
00039
00040
00041
00042 #include "LOCA_BorderedSolver_UpperTriangularBlockElimination.H"
00043 #include "LOCA_BorderedSolver_AbstractOperator.H"
00044 #include "LOCA_GlobalData.H"
00045 #include "LOCA_ErrorCheck.H"
00046 #include "Teuchos_LAPACK.hpp"
00047 #include "LOCA_Abstract_TransposeSolveGroup.H"
00048
00049 LOCA::BorderedSolver::UpperTriangularBlockElimination::
00050 UpperTriangularBlockElimination(
00051 const Teuchos::RCP<LOCA::GlobalData>& global_data) :
00052 globalData(global_data)
00053 {
00054 }
00055
00056 LOCA::BorderedSolver::UpperTriangularBlockElimination::
00057 ~UpperTriangularBlockElimination()
00058 {
00059 }
00060
00061 NOX::Abstract::Group::ReturnType
00062 LOCA::BorderedSolver::UpperTriangularBlockElimination::
00063 solve(Teuchos::ParameterList& params,
00064 const LOCA::BorderedSolver::AbstractOperator& op,
00065 const NOX::Abstract::MultiVector* A,
00066 const NOX::Abstract::MultiVector::DenseMatrix& C,
00067 const NOX::Abstract::MultiVector* F,
00068 const NOX::Abstract::MultiVector::DenseMatrix* G,
00069 NOX::Abstract::MultiVector& X,
00070 NOX::Abstract::MultiVector::DenseMatrix& Y) const
00071 {
00072 string callingFunction =
00073 "LOCA::BorderedSolver::UpperTriangularBlockElimination::solve()";
00074 NOX::Abstract::Group::ReturnType finalStatus = NOX::Abstract::Group::Ok;
00075 NOX::Abstract::Group::ReturnType status;
00076
00077
00078 bool isZeroF = (F == NULL);
00079 bool isZeroG = (G == NULL);
00080 bool isZeroA = (A == NULL);
00081 bool isZeroY = isZeroG;
00082 bool isZeroX = isZeroF && (isZeroA || isZeroY);
00083
00084
00085 if (isZeroY)
00086 Y.putScalar(0.0);
00087 else {
00088
00089 NOX::Abstract::MultiVector::DenseMatrix M(C);
00090 int *ipiv = new int[M.numRows()];
00091 Teuchos::LAPACK<int,double> L;
00092 int info;
00093
00094 Y.assign(*G);
00095 L.GETRF(M.numRows(), M.numCols(), M.values(), M.stride(), ipiv, &info);
00096 if (info != 0) {
00097 status = NOX::Abstract::Group::Failed;
00098 finalStatus =
00099 globalData->locaErrorCheck->combineAndCheckReturnTypes(
00100 status,
00101 finalStatus,
00102 callingFunction);
00103 }
00104 L.GETRS('N', M.numRows(), Y.numCols(), M.values(), M.stride(), ipiv,
00105 Y.values(), Y.stride(), &info);
00106 delete [] ipiv;
00107 if (info != 0) {
00108 status = NOX::Abstract::Group::Failed;
00109 finalStatus =
00110 globalData->locaErrorCheck->combineAndCheckReturnTypes(
00111 status,
00112 finalStatus,
00113 callingFunction);
00114 }
00115 }
00116
00117
00118 if (isZeroX)
00119 X.init(0.0);
00120 else if (isZeroA || isZeroY) {
00121
00122 status = op.applyInverse(params, *F, X);
00123 finalStatus =
00124 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00125 finalStatus,
00126 callingFunction);
00127 }
00128 else {
00129 Teuchos::RCP<NOX::Abstract::MultiVector> RHS;
00130
00131 if (isZeroF) {
00132 RHS = A->clone(Y.numCols());
00133 RHS->update(Teuchos::NO_TRANS, -1.0, *A, Y, 0.0);
00134 }
00135 else {
00136 RHS = F->clone(NOX::DeepCopy);
00137 RHS->update(Teuchos::NO_TRANS, -1.0, *A, Y, 1.0);
00138 }
00139
00140 status = op.applyInverse(params, *RHS, X);
00141 finalStatus =
00142 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00143 finalStatus,
00144 callingFunction);
00145 }
00146
00147 return finalStatus;
00148 }
00149
00150 NOX::Abstract::Group::ReturnType
00151 LOCA::BorderedSolver::UpperTriangularBlockElimination::
00152 solveTranspose(Teuchos::ParameterList& params,
00153 const LOCA::BorderedSolver::AbstractOperator& op,
00154 const NOX::Abstract::MultiVector* A,
00155 const NOX::Abstract::MultiVector::DenseMatrix& C,
00156 const NOX::Abstract::MultiVector* F,
00157 const NOX::Abstract::MultiVector::DenseMatrix* G,
00158 NOX::Abstract::MultiVector& X,
00159 NOX::Abstract::MultiVector::DenseMatrix& Y) const
00160 {
00161 string callingFunction =
00162 "LOCA::BorderedSolver::UpperTriangularBlockElimination::solveTranspose()";
00163 NOX::Abstract::Group::ReturnType finalStatus = NOX::Abstract::Group::Ok;
00164 NOX::Abstract::Group::ReturnType status;
00165
00166
00167 bool isZeroF = (F == NULL);
00168 bool isZeroG = (G == NULL);
00169 bool isZeroA = (A == NULL);
00170 bool isZeroY = isZeroG;
00171 bool isZeroX = isZeroF && (isZeroA || isZeroY);
00172
00173
00174 if (isZeroY)
00175 Y.putScalar(0.0);
00176 else {
00177
00178 NOX::Abstract::MultiVector::DenseMatrix M(C);
00179 int *ipiv = new int[M.numRows()];
00180 Teuchos::LAPACK<int,double> L;
00181 int info;
00182
00183 Y.assign(*G);
00184 L.GETRF(M.numRows(), M.numCols(), M.values(), M.stride(), ipiv, &info);
00185 if (info != 0) {
00186 status = NOX::Abstract::Group::Failed;
00187 finalStatus =
00188 globalData->locaErrorCheck->combineAndCheckReturnTypes(
00189 status,
00190 finalStatus,
00191 callingFunction);
00192 }
00193 L.GETRS('T', M.numRows(), Y.numCols(), M.values(), M.stride(), ipiv,
00194 Y.values(), Y.stride(), &info);
00195 delete [] ipiv;
00196 if (info != 0) {
00197 status = NOX::Abstract::Group::Failed;
00198 finalStatus =
00199 globalData->locaErrorCheck->combineAndCheckReturnTypes(
00200 status,
00201 finalStatus,
00202 callingFunction);
00203 }
00204 }
00205
00206
00207 if (isZeroX)
00208 X.init(0.0);
00209 else if (isZeroA || isZeroY) {
00210
00211 status = op.applyInverseTranspose(params, *F, X);
00212 finalStatus =
00213 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00214 finalStatus,
00215 callingFunction);
00216 }
00217 else {
00218 Teuchos::RCP<NOX::Abstract::MultiVector> RHS;
00219
00220 if (isZeroF) {
00221 RHS = A->clone(Y.numCols());
00222 RHS->update(Teuchos::NO_TRANS, -1.0, *A, Y, 0.0);
00223 }
00224 else {
00225 RHS = F->clone(NOX::DeepCopy);
00226 RHS->update(Teuchos::NO_TRANS, -1.0, *A, Y, 1.0);
00227 }
00228
00229 status = op.applyInverseTranspose(params, *RHS, X);
00230 finalStatus =
00231 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00232 finalStatus,
00233 callingFunction);
00234 }
00235
00236 return finalStatus;
00237 }