NOX::Abstract::Group::ReturnType
LOCA::SingularJacobianSolve::ItRef::computeMulti(
Teuchos::ParameterList& params,
LOCA::Continuation::AbstractGroup& grp,
const NOX::Abstract::Vector*const* inputs,
const NOX::Abstract::Vector& approxNullVec,
const NOX::Abstract::Vector& jacApproxNullVec,
NOX::Abstract::Vector** results,
int nVecs)
{
std::string callingFunction =
"LOCA::SingularJacobianSolve::ItRef::computeMulti()";
NOX::Abstract::Group::ReturnType status, finalStatus;
NOX::Abstract::Vector** remainders = new NOX::Abstract::Vector*[nVecs];
NOX::Abstract::Vector** refinements = new NOX::Abstract::Vector*[nVecs];
finalStatus = grp.applyJacobianInverseMulti(params, inputs, results, nVecs);
LOCA::ErrorCheck::checkReturnType(finalStatus, callingFunction);
for (int i=0; i<nVecs; i++) {
remainders[i] = inputs[i]->clone(NOX::ShapeCopy);
refinements[i] = inputs[i]->clone(NOX::ShapeCopy);
status = grp.applyJacobian(*(results[i]), *(remainders[i]));
finalStatus =
LOCA::ErrorCheck::combineAndCheckReturnTypes(status, finalStatus,
callingFunction);
// r = b-Ax
remainders[i]->update(1.0, *(inputs[i]), -1.0);
}
// Ay=r
status = grp.applyJacobianInverseMulti(params, remainders, refinements,
nVecs);
finalStatus =
LOCA::ErrorCheck::combineAndCheckReturnTypes(status, finalStatus,
callingFunction);
// x+=y
for (int i=0; i<nVecs; i++) {
results[i]->update(1.0, *(refinements[i]), 1.0);
delete remainders[i];
delete refinements[i];
}
delete [] remainders;
delete [] refinements;
return finalStatus;
}
NOX::Abstract::Group::ReturnType
LOCA::SingularJacobianSolve::ItRef::compute(
Teuchos::ParameterList& params,
LOCA::Continuation::AbstractGroup& grp,
const NOX::Abstract::Vector& input,
const NOX::Abstract::Vector& approxNullVec,
const NOX::Abstract::Vector& jacApproxNullVec,
NOX::Abstract::Vector& result)
{
std::string callingFunction =
"LOCA::SingularJacobianSolve::ItRef::compute()";
NOX::Abstract::Group::ReturnType status, finalStatus;
finalStatus = grp.applyJacobianInverse(params, input, result);
LOCA::ErrorCheck::checkReturnType(finalStatus, callingFunction);
NOX::Abstract::Vector* remainder = input.clone(NOX::ShapeCopy);
status = grp.applyJacobian(result, *remainder);
finalStatus =
LOCA::ErrorCheck::combineAndCheckReturnTypes(status, finalStatus,
callingFunction);
// r = b-Ax
remainder->update(1.0, input, -1.0);
NOX::Abstract::Vector* refinement = input.clone(NOX::ShapeCopy);
// Ay=r
status = grp.applyJacobianInverse(params, *remainder, *refinement);
finalStatus =
LOCA::ErrorCheck::combineAndCheckReturnTypes(status, finalStatus,
callingFunction);
// x+=y
result.update(1.0, *refinement, 1.0);
delete remainder;
delete refinement;
return finalStatus;
}