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_MultiContinuation_NaturalConstraint.H"
00043 #include "LOCA_MultiContinuation_NaturalGroup.H"
00044 #include "LOCA_GlobalData.H"
00045 #include "LOCA_ErrorCheck.H"
00046
00047 LOCA::MultiContinuation::NaturalConstraint::NaturalConstraint(
00048 const Teuchos::RCP<LOCA::GlobalData>& global_data,
00049 const Teuchos::RCP<LOCA::MultiContinuation::NaturalGroup>& grp) :
00050 globalData(global_data),
00051 naturalGroup(grp),
00052 constraints(grp->getNumParams(), 1),
00053 isValidConstraints(false),
00054 conParamIDs(grp->getContinuationParameterIDs())
00055 {
00056 }
00057
00058 LOCA::MultiContinuation::NaturalConstraint::NaturalConstraint(
00059 const LOCA::MultiContinuation::NaturalConstraint& source,
00060 NOX::CopyType type) :
00061 globalData(source.globalData),
00062 naturalGroup(),
00063 constraints(source.constraints),
00064 isValidConstraints(source.isValidConstraints),
00065 conParamIDs(source.conParamIDs)
00066 {
00067 if (source.isValidConstraints && type == NOX::DeepCopy)
00068 isValidConstraints = true;
00069 }
00070
00071 LOCA::MultiContinuation::NaturalConstraint::~NaturalConstraint()
00072 {
00073 }
00074
00075 void
00076 LOCA::MultiContinuation::NaturalConstraint::setNaturalGroup(const Teuchos::RCP<LOCA::MultiContinuation::NaturalGroup>& grp)
00077 {
00078 naturalGroup = grp;
00079 }
00080
00081 void
00082 LOCA::MultiContinuation::NaturalConstraint::copy(
00083 const LOCA::MultiContinuation::ConstraintInterface& src)
00084 {
00085 const LOCA::MultiContinuation::NaturalConstraint& source =
00086 dynamic_cast<const LOCA::MultiContinuation::NaturalConstraint&>(src);
00087
00088 if (this != &source) {
00089 globalData = source.globalData;
00090 constraints.assign(source.constraints);
00091 isValidConstraints = source.isValidConstraints;
00092 conParamIDs = source.conParamIDs;
00093 }
00094 }
00095
00096 Teuchos::RCP<LOCA::MultiContinuation::ConstraintInterface>
00097 LOCA::MultiContinuation::NaturalConstraint::clone(NOX::CopyType type) const
00098 {
00099 return Teuchos::rcp(new NaturalConstraint(*this, type));
00100 }
00101
00102 int
00103 LOCA::MultiContinuation::NaturalConstraint::numConstraints() const
00104 {
00105 return constraints.numRows();
00106 }
00107
00108 void
00109 LOCA::MultiContinuation::NaturalConstraint::setX(
00110 const NOX::Abstract::Vector& y)
00111 {
00112 isValidConstraints = false;
00113 }
00114
00115 void
00116 LOCA::MultiContinuation::NaturalConstraint::setParam(int paramID, double val)
00117 {
00118 isValidConstraints = false;
00119 }
00120
00121 void
00122 LOCA::MultiContinuation::NaturalConstraint::setParams(
00123 const vector<int>& paramIDs,
00124 const NOX::Abstract::MultiVector::DenseMatrix& vals)
00125 {
00126 isValidConstraints = false;
00127 }
00128
00129 NOX::Abstract::Group::ReturnType
00130 LOCA::MultiContinuation::NaturalConstraint::computeConstraints()
00131 {
00132 if (isValidConstraints)
00133 return NOX::Abstract::Group::Ok;
00134
00135
00136 const LOCA::MultiContinuation::ExtendedVector& xVec =
00137 dynamic_cast<const LOCA::MultiContinuation::ExtendedVector&>(naturalGroup->
00138 getX());
00139 const LOCA::MultiContinuation::ExtendedVector& prevXVec =
00140 dynamic_cast<const LOCA::MultiContinuation::ExtendedVector&>(naturalGroup->
00141 getPrevX());
00142
00143 for (int i=0; i<naturalGroup->getNumParams(); i++)
00144 constraints(i,0) = xVec.getScalar(i) - prevXVec.getScalar(i) -
00145 naturalGroup->getStepSize(i);
00146
00147 isValidConstraints = true;
00148
00149 return NOX::Abstract::Group::Ok;
00150 }
00151
00152 NOX::Abstract::Group::ReturnType
00153 LOCA::MultiContinuation::NaturalConstraint::computeDX()
00154 {
00155 return NOX::Abstract::Group::Ok;
00156 }
00157
00158 NOX::Abstract::Group::ReturnType
00159 LOCA::MultiContinuation::NaturalConstraint::computeDP(
00160 const vector<int>& paramIDs,
00161 NOX::Abstract::MultiVector::DenseMatrix& dgdp,
00162 bool isValidG)
00163 {
00164 string callingFunction =
00165 "LOCA::MultiContinuation::NaturalConstraint::computeDP()";
00166 NOX::Abstract::Group::ReturnType status;
00167 NOX::Abstract::Group::ReturnType finalStatus = NOX::Abstract::Group::Ok;
00168
00169
00170 if (!isValidG && !isValidConstraints) {
00171 status = computeConstraints();
00172 finalStatus =
00173 globalData->locaErrorCheck->combineAndCheckReturnTypes(status,
00174 finalStatus,
00175 callingFunction);
00176 }
00177 if (!isValidG) {
00178 for (int i=0; i<constraints.numRows(); i++)
00179 dgdp(i,0) = constraints(i,0);
00180 }
00181
00182
00183
00184
00185 vector<int>::const_iterator it;
00186 for (unsigned int i=0; i<paramIDs.size(); i++) {
00187 for (int k=0; k<constraints.numRows(); k++)
00188 dgdp(k,i+1) = 0.0;
00189 it = find(conParamIDs.begin(), conParamIDs.end(), paramIDs[i]);
00190 if (it != conParamIDs.end())
00191 dgdp(it-conParamIDs.begin(),i+1) = 1.0;
00192 }
00193
00194 return finalStatus;
00195 }
00196
00197 bool
00198 LOCA::MultiContinuation::NaturalConstraint::isConstraints() const
00199 {
00200 return isValidConstraints;
00201 }
00202
00203 bool
00204 LOCA::MultiContinuation::NaturalConstraint::isDX() const
00205 {
00206 return true;
00207 }
00208
00209 const NOX::Abstract::MultiVector::DenseMatrix&
00210 LOCA::MultiContinuation::NaturalConstraint::getConstraints() const
00211 {
00212 return constraints;
00213 }
00214
00215 const NOX::Abstract::MultiVector*
00216 LOCA::MultiContinuation::NaturalConstraint::getDX() const
00217 {
00218 return NULL;
00219 }
00220
00221 bool
00222 LOCA::MultiContinuation::NaturalConstraint::isDXZero() const
00223 {
00224 return true;
00225 }
00226