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 "Teuchos_ParameterList.hpp"
00043
00044 #include "LOCA_MultiPredictor_Secant.H"
00045 #include "LOCA_GlobalData.H"
00046 #include "NOX_Utils.H"
00047 #include "LOCA_Factory.H"
00048 #include "LOCA_Parameter_SublistParser.H"
00049 #include "LOCA_MultiContinuation_ExtendedVector.H"
00050 #include "LOCA_MultiContinuation_ExtendedMultiVector.H"
00051
00052 LOCA::MultiPredictor::Secant::Secant(
00053 const Teuchos::RCP<LOCA::GlobalData>& global_data,
00054 const Teuchos::RCP<LOCA::Parameter::SublistParser>& topParams,
00055 const Teuchos::RCP<Teuchos::ParameterList>& predParams) :
00056 globalData(global_data),
00057 firstStepPredictor(),
00058 isFirstStep(true),
00059 isFirstStepComputed(false),
00060 predictor(),
00061 secant(),
00062 initialized(false)
00063 {
00064 Teuchos::RCP<Teuchos::ParameterList> firstStepList =
00065 topParams->getSublist("First Step Predictor");
00066
00067 firstStepList->get("Method", "Constant");
00068 firstStepPredictor = globalData->locaFactory->createPredictorStrategy(
00069 topParams,
00070 firstStepList);
00071 }
00072
00073 LOCA::MultiPredictor::Secant::~Secant()
00074 {
00075 }
00076
00077 LOCA::MultiPredictor::Secant::Secant(
00078 const LOCA::MultiPredictor::Secant& source,
00079 NOX::CopyType type) :
00080 globalData(source.globalData),
00081 firstStepPredictor(source.firstStepPredictor->clone(type)),
00082 isFirstStep(source.isFirstStep),
00083 isFirstStepComputed(source.isFirstStepComputed),
00084 predictor(),
00085 secant(),
00086 initialized(source.initialized)
00087 {
00088 if (source.initialized) {
00089 predictor = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedMultiVector>(source.predictor->clone(type));
00090
00091 secant = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedVector>(source.secant->clone(type));
00092 }
00093 }
00094
00095 LOCA::MultiPredictor::AbstractStrategy&
00096 LOCA::MultiPredictor::Secant::operator=(
00097 const LOCA::MultiPredictor::AbstractStrategy& s)
00098 {
00099 const LOCA::MultiPredictor::Secant& source =
00100 dynamic_cast<const LOCA::MultiPredictor::Secant&>(s);
00101
00102 if (this != &source) {
00103 globalData = source.globalData;
00104 firstStepPredictor = source.firstStepPredictor->clone(NOX::DeepCopy);
00105 isFirstStep = source.isFirstStep;
00106 isFirstStepComputed = source.isFirstStepComputed;
00107 initialized = source.initialized;
00108
00109 if (source.initialized) {
00110 predictor = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedMultiVector>(source.predictor->clone(NOX::DeepCopy));
00111
00112 secant = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedVector>(source.secant->clone(NOX::DeepCopy));
00113 }
00114 }
00115
00116 return *this;
00117 }
00118
00119 Teuchos::RCP<LOCA::MultiPredictor::AbstractStrategy>
00120 LOCA::MultiPredictor::Secant::clone(NOX::CopyType type) const
00121 {
00122 return Teuchos::rcp(new Secant(*this, type));
00123 }
00124
00125 NOX::Abstract::Group::ReturnType
00126 LOCA::MultiPredictor::Secant::compute(
00127 bool baseOnSecant, const vector<double>& stepSize,
00128 LOCA::MultiContinuation::ExtendedGroup& grp,
00129 const LOCA::MultiContinuation::ExtendedVector& prevXVec,
00130 const LOCA::MultiContinuation::ExtendedVector& xVec)
00131 {
00132 if (globalData->locaUtils->isPrintType(NOX::Utils::StepperDetails))
00133 globalData->locaUtils->out() <<
00134 "\n\tCalling Predictor with method: Secant" << std::endl;
00135
00136
00137 int numParams = stepSize.size();
00138
00139 if (!initialized) {
00140
00141
00142 predictor = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedMultiVector>(xVec.createMultiVector(numParams, NOX::ShapeCopy));
00143
00144
00145 secant = Teuchos::rcp_dynamic_cast<LOCA::MultiContinuation::ExtendedVector>(xVec.clone(NOX::ShapeCopy));
00146
00147 initialized = true;
00148 }
00149
00150
00151 if (isFirstStep && !isFirstStepComputed) {
00152 isFirstStepComputed = true;
00153 return firstStepPredictor->compute(baseOnSecant, stepSize, grp,
00154 prevXVec, xVec);
00155 }
00156
00157 if (isFirstStep && isFirstStepComputed)
00158 isFirstStep = false;
00159
00160
00161 (*predictor)[0].update(1.0, xVec, -1.0, prevXVec, 0.0);
00162
00163 for (int i=0; i<numParams; i++) {
00164
00165 (*predictor)[i] = (*predictor)[0];
00166
00167
00168 (*predictor)[i].scale(1.0/fabs(predictor->getScalar(i,i)));
00169
00170
00171 for (int j=0; j<numParams; j++)
00172 if (i != j)
00173 predictor->getScalar(i,j) = 0.0;
00174 }
00175
00176
00177 setPredictorOrientation(baseOnSecant, stepSize, grp, prevXVec,
00178 xVec, *secant, *predictor);
00179
00180 return NOX::Abstract::Group::Ok;
00181 }
00182
00183 NOX::Abstract::Group::ReturnType
00184 LOCA::MultiPredictor::Secant::evaluate(
00185 const vector<double>& stepSize,
00186 const LOCA::MultiContinuation::ExtendedVector& xVec,
00187 LOCA::MultiContinuation::ExtendedMultiVector& result) const
00188 {
00189
00190 if (isFirstStep) {
00191 return firstStepPredictor->evaluate(stepSize, xVec, result);
00192 }
00193
00194
00195 int numParams = stepSize.size();
00196
00197 for (int i=0; i<numParams; i++)
00198 result[i].update(1.0, xVec, stepSize[i], (*predictor)[i], 0.0);
00199
00200 return NOX::Abstract::Group::Ok;
00201 }
00202
00203 NOX::Abstract::Group::ReturnType
00204 LOCA::MultiPredictor::Secant::computeTangent(
00205 LOCA::MultiContinuation::ExtendedMultiVector& v)
00206 {
00207
00208 if (isFirstStep) {
00209 return firstStepPredictor->computeTangent(v);
00210 }
00211
00212 v = *predictor;
00213
00214 return NOX::Abstract::Group::Ok;
00215 }
00216
00217 bool
00218 LOCA::MultiPredictor::Secant::isTangentScalable() const
00219 {
00220
00221 if (isFirstStep) {
00222 return firstStepPredictor->isTangentScalable();
00223 }
00224
00225 return true;
00226 }