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_Pitchfork_MooreSpence_PhippsBordering.H"
00043 #include "LOCA_Pitchfork_MooreSpence_ExtendedGroup.H"
00044 #include "LOCA_Pitchfork_MooreSpence_AbstractGroup.H"
00045 #include "LOCA_BorderedSolver_AbstractStrategy.H"
00046 #include "LOCA_GlobalData.H"
00047 #include "LOCA_Factory.H"
00048 #include "LOCA_ErrorCheck.H"
00049 #include "Teuchos_LAPACK.hpp"
00050 #include "LOCA_BorderedSolver_JacobianOperator.H"
00051
00052 LOCA::Pitchfork::MooreSpence::PhippsBordering::PhippsBordering(
00053 const Teuchos::RCP<LOCA::GlobalData>& global_data,
00054 const Teuchos::RCP<LOCA::Parameter::SublistParser>& topParams,
00055 const Teuchos::RCP<Teuchos::ParameterList>& slvrParams) :
00056 globalData(global_data),
00057 solverParams(slvrParams),
00058 group(),
00059 pfGroup(),
00060 asymMultiVector(),
00061 asymVector(),
00062 nullVector(),
00063 JnVector(),
00064 dfdp(),
00065 dJndp(),
00066 borderedSolver(),
00067 nullMultiVector(),
00068 JnMultiVector(),
00069 sigma(0.0)
00070 {
00071 borderedSolver =
00072 globalData->locaFactory->createBorderedSolverStrategy(topParams,
00073 solverParams);
00074 }
00075
00076 LOCA::Pitchfork::MooreSpence::PhippsBordering::~PhippsBordering()
00077 {
00078 }
00079
00080 void
00081 LOCA::Pitchfork::MooreSpence::PhippsBordering::setBlocks(
00082 const Teuchos::RCP<LOCA::Pitchfork::MooreSpence::AbstractGroup>& group_,
00083 const Teuchos::RCP<LOCA::Pitchfork::MooreSpence::ExtendedGroup>& pfGroup_,
00084 const Teuchos::RCP<const NOX::Abstract::MultiVector>& asymMultiVector_,
00085 const Teuchos::RCP<const NOX::Abstract::Vector>& nullVector_,
00086 const Teuchos::RCP<const NOX::Abstract::Vector>& JnVector_,
00087 const Teuchos::RCP<const NOX::Abstract::Vector>& dfdp_,
00088 const Teuchos::RCP<const NOX::Abstract::Vector>& dJndp_)
00089 {
00090 group = group_;
00091 pfGroup = pfGroup_;
00092 asymMultiVector = asymMultiVector_;
00093 asymVector = Teuchos::rcp(&(*asymMultiVector)[0], false);
00094 nullVector = nullVector_;
00095 JnVector = JnVector_;
00096 dfdp = dfdp_;
00097 dJndp = dJndp_;
00098
00099
00100 nullMultiVector = nullVector->createMultiVector(1, NOX::DeepCopy);
00101 JnMultiVector = JnVector->createMultiVector(1, NOX::DeepCopy);
00102 sigma = JnVector->norm(NOX::Abstract::Vector::TwoNorm);
00103 JnMultiVector->scale(1.0/sigma);
00104
00105
00106 Teuchos::RCP<const LOCA::BorderedSolver::JacobianOperator> op =
00107 Teuchos::rcp(new LOCA::BorderedSolver::JacobianOperator(group));
00108 borderedSolver->setMatrixBlocksMultiVecConstraint(op,
00109 JnMultiVector,
00110 nullMultiVector,
00111 Teuchos::null);
00112 NOX::Abstract::Group::ReturnType status = borderedSolver->initForSolve();
00113 globalData->locaErrorCheck->checkReturnType(status,
00114 "LOCA::Pitchfork::MooreSpence::PhippsBordering::setBlocks()");
00115 }
00116
00117 NOX::Abstract::Group::ReturnType
00118 LOCA::Pitchfork::MooreSpence::PhippsBordering::solve(
00119 Teuchos::ParameterList& params,
00120 const LOCA::Pitchfork::MooreSpence::ExtendedMultiVector& input,
00121 LOCA::Pitchfork::MooreSpence::ExtendedMultiVector& result) const
00122 {
00123 string callingFunction =
00124 "LOCA::Pitchfork::MooreSpence::PhippsBordering::solve()";
00125 NOX::Abstract::Group::ReturnType status;
00126
00127
00128 Teuchos::RCP<const NOX::Abstract::MultiVector> input_x =
00129 input.getXMultiVec();
00130 Teuchos::RCP<const NOX::Abstract::MultiVector> input_null =
00131 input.getNullMultiVec();
00132 Teuchos::RCP<const NOX::Abstract::MultiVector::DenseMatrix> input_slack = input.getSlacks();
00133 Teuchos::RCP<const NOX::Abstract::MultiVector::DenseMatrix> input_param = input.getBifParams();
00134
00135
00136 Teuchos::RCP<NOX::Abstract::MultiVector> result_x =
00137 result.getXMultiVec();
00138 Teuchos::RCP<NOX::Abstract::MultiVector> result_null =
00139 result.getNullMultiVec();
00140 Teuchos::RCP<NOX::Abstract::MultiVector::DenseMatrix> result_slack =
00141 result.getSlacks();
00142 Teuchos::RCP<NOX::Abstract::MultiVector::DenseMatrix> result_param =
00143 result.getBifParams();
00144
00145 int m = input.numVectors();
00146 vector<int> index_input(m);
00147 for (int i=0; i<m; i++)
00148 index_input[i] = i;
00149
00150
00151
00152
00153
00154
00155 Teuchos::RCP<NOX::Abstract::MultiVector> cont_input_x =
00156 input_x->clone(m+3);
00157 Teuchos::RCP<NOX::Abstract::MultiVector> cont_input_null =
00158 input_null->clone(m+3);
00159
00160 Teuchos::RCP<NOX::Abstract::MultiVector> cont_result_x =
00161 result_x->clone(m+3);
00162 Teuchos::RCP<NOX::Abstract::MultiVector> cont_result_null =
00163 result_null->clone(m+3);
00164
00165
00166 cont_input_x->setBlock(*input_x, index_input);
00167
00168
00169 (*cont_input_x)[m] = *dfdp;
00170
00171
00172 (*cont_input_x)[m+1] = *asymVector;
00173
00174
00175 (*cont_input_x)[m+2].init(0.0);
00176
00177
00178 cont_input_null->setBlock(*input_null, index_input);
00179
00180
00181 (*cont_input_null)[m] = *dJndp;
00182
00183
00184 (*cont_input_null)[m+1].init(0.0);
00185
00186
00187 (*cont_input_null)[m+2].init(0.0);
00188
00189
00190 cont_result_x->init(0.0);
00191 cont_result_null->init(0.0);
00192
00193
00194 status = solveContiguous(params, *cont_input_x, *cont_input_null,
00195 *input_slack, *input_param,
00196 *cont_result_x, *cont_result_null,
00197 *result_slack, *result_param);
00198
00199
00200 Teuchos::RCP<NOX::Abstract::MultiVector> cont_result_x_view =
00201 cont_result_x->subView(index_input);
00202 Teuchos::RCP<NOX::Abstract::MultiVector> cont_result_null_view =
00203 cont_result_null->subView(index_input);
00204
00205
00206 *result_x = *cont_result_x_view;
00207 *result_null = *cont_result_null_view;
00208
00209 return status;
00210 }
00211
00212
00213
00214
00215
00216
00217
00218 NOX::Abstract::Group::ReturnType
00219 LOCA::Pitchfork::MooreSpence::PhippsBordering::solveContiguous(
00220 Teuchos::ParameterList& params,
00221 const NOX::Abstract::MultiVector& input_x,
00222 const NOX::Abstract::MultiVector& input_null,
00223 const NOX::Abstract::MultiVector::DenseMatrix& input_slack,
00224 const NOX::Abstract::MultiVector::DenseMatrix& input_param,
00225 NOX::Abstract::MultiVector& result_x,
00226 NOX::Abstract::MultiVector& result_null,
00227 NOX::Abstract::MultiVector::DenseMatrix& result_slack,
00228 NOX::Abstract::MultiVector::DenseMatrix& result_param) const
00229 {
00230 string callingFunction =
00231 "LOCA::Pitchfork::MooreSpence::PhippsBordering::solveContiguous()";
00232 NOX::Abstract::Group::ReturnType finalStatus = NOX::Abstract::Group::Ok;
00233 NOX::Abstract::Group::ReturnType status;
00234
00235 int m = input_x.numVectors()-3;
00236 vector<int> index_input(m);
00237 vector<int> index_input_dp(m+2);
00238 vector<int> index_null(1);
00239 vector<int> index_dp(1);
00240 vector<int> index_s(1);
00241 for (int i=0; i<m; i++) {
00242 index_input[i] = i;
00243 index_input_dp[i] = i;
00244 }
00245 index_input_dp[m] = m;
00246 index_input_dp[m+1] = m+1;
00247 index_dp[0] = m;
00248 index_s[0] = m+1;
00249 index_null[0] = m+2;
00250
00251 NOX::Abstract::MultiVector::DenseMatrix tmp_mat_1(1, m+2);
00252 NOX::Abstract::MultiVector::DenseMatrix tmp_mat_2(1, m+3);
00253
00254
00255 Teuchos::RCP<NOX::Abstract::MultiVector> input_x_view =
00256 input_x.subView(index_input_dp);
00257 Teuchos::RCP<NOX::Abstract::MultiVector> result_x_view =
00258 result_x.subView(index_input_dp);
00259
00260
00261 if (!group->isJacobian()) {
00262 status = group->computeJacobian();
00263 finalStatus =
00264 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00265 finalStatus,
00266 callingFunction);
00267 }
00268
00269
00270
00271 status = borderedSolver->applyInverse(params, input_x_view.get(), NULL,
00272 *result_x_view, tmp_mat_1);
00273 finalStatus =
00274 globalData->locaErrorCheck->combineAndCheckReturnTypes(status, finalStatus,
00275 callingFunction);
00276 Teuchos::RCP<NOX::Abstract::MultiVector> A =
00277 result_x.subView(index_input);
00278 Teuchos::RCP<NOX::Abstract::MultiVector> B =
00279 result_x.subView(index_dp);
00280 Teuchos::RCP<NOX::Abstract::MultiVector> C =
00281 result_x.subView(index_s);
00282 double b = tmp_mat_1(0,m);
00283 double c = tmp_mat_1(0,m+1);
00284
00285
00286 result_x[m+2] = *nullVector;
00287 Teuchos::RCP<NOX::Abstract::MultiVector> tmp =
00288 result_x.clone(NOX::ShapeCopy);
00289 status = group->computeDJnDxaMulti(*nullVector, *JnVector, result_x,
00290 *tmp);
00291 finalStatus =
00292 globalData->locaErrorCheck->combineAndCheckReturnTypes(status, finalStatus,
00293 callingFunction);
00294
00295
00296 tmp->update(1.0, input_null, -1.0);
00297
00298
00299 if (!group->isJacobian()) {
00300 status = group->computeJacobian();
00301 finalStatus =
00302 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00303 finalStatus,
00304 callingFunction);
00305 }
00306
00307
00308
00309 status = borderedSolver->applyInverse(params, tmp.get(), NULL, result_null,
00310 tmp_mat_2);
00311 finalStatus =
00312 globalData->locaErrorCheck->combineAndCheckReturnTypes(status, finalStatus,
00313 callingFunction);
00314 Teuchos::RCP<NOX::Abstract::MultiVector> D =
00315 result_null.subView(index_input);
00316 Teuchos::RCP<NOX::Abstract::MultiVector> E =
00317 result_null.subView(index_dp);
00318 Teuchos::RCP<NOX::Abstract::MultiVector> K =
00319 result_null.subView(index_s);
00320 Teuchos::RCP<NOX::Abstract::MultiVector> L =
00321 result_null.subView(index_null);
00322 double e = tmp_mat_2(0, m);
00323 double k = tmp_mat_2(0, m+1);
00324 double l = tmp_mat_2(0, m+2);
00325
00326 double ltE = pfGroup->lTransNorm((*E)[0]);
00327 double ltK = pfGroup->lTransNorm((*K)[0]);
00328 double ltL = pfGroup->lTransNorm((*L)[0]);
00329 double ltv = pfGroup->lTransNorm(*nullVector);
00330 double ipv = group->innerProduct(*nullVector, *asymVector);
00331 double ipB = group->innerProduct((*B)[0], *asymVector);
00332 double ipC = group->innerProduct((*C)[0], *asymVector);
00333
00334
00335 double M[16];
00336 M[0] = sigma; M[1] = -l; M[2] = ipv; M[3] = ltL;
00337 M[4] = 0.0; M[5] = sigma; M[6] = 0.0; M[7] = ltv;
00338 M[8] = b; M[9] = e; M[10] = -ipB; M[11] = -ltE;
00339 M[12] = c; M[13] = k; M[14] = -ipC; M[15] = -ltK;
00340
00341
00342 NOX::Abstract::MultiVector::DenseMatrix tmp_mat_3(1, m);
00343 group->innerProduct(*asymMultiVector, *A, tmp_mat_3);
00344 tmp_mat_3 -= input_slack;
00345 tmp_mat_3.scale(-1.0);
00346
00347
00348 NOX::Abstract::MultiVector::DenseMatrix tmp_mat_4(1, m);
00349 pfGroup->lTransNorm(*D, tmp_mat_4);
00350 tmp_mat_4 -= input_param;
00351 tmp_mat_4.scale(-1.0);
00352
00353 double *R = new double[4*m];
00354 for (int i=0; i<m; i++) {
00355 R[4*i] = tmp_mat_1(0,i);
00356 R[4*i+1] = tmp_mat_2(0,i);
00357 R[4*i+2] = tmp_mat_3(0,i);
00358 R[4*i+3] = tmp_mat_4(0,i);
00359 }
00360
00361
00362 int piv[4];
00363 int info;
00364 Teuchos::LAPACK<int,double> dlapack;
00365 dlapack.GESV(4, m, M, 4, piv, R, 4, &info);
00366 if (info != 0) {
00367 globalData->locaErrorCheck->throwError(
00368 callingFunction,
00369 "Solve of 4x4 coefficient matrix failed!");
00370 return NOX::Abstract::Group::Failed;
00371 }
00372
00373 NOX::Abstract::MultiVector::DenseMatrix alpha(1,m);
00374 NOX::Abstract::MultiVector::DenseMatrix beta(1,m);
00375 for (int i=0; i<m; i++) {
00376 alpha(0,i) = R[4*i];
00377 beta(0,i) = R[4*i+1];
00378 result_param(0,i) = R[4*i+2];
00379 result_slack(0,i) = R[4*i+3];
00380 }
00381
00382
00383 A->update(Teuchos::NO_TRANS, -1.0, *B, result_param, 1.0);
00384 A->update(Teuchos::NO_TRANS, -1.0, *C, result_slack, 1.0);
00385 A->update(Teuchos::NO_TRANS, 1.0, *nullMultiVector, alpha, 1.0);
00386
00387
00388
00389 D->update(Teuchos::NO_TRANS, -1.0, *E, result_param, 1.0);
00390 D->update(Teuchos::NO_TRANS, -1.0, *K, result_slack, 1.0);
00391 D->update(Teuchos::NO_TRANS, 1.0, *L, alpha, 1.0);
00392 D->update(Teuchos::NO_TRANS, 1.0, *nullMultiVector, beta, 1.0);
00393
00394 delete [] R;
00395
00396 return finalStatus;
00397 }