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 "NOX_Epetra_ModelEvaluatorInterface.H"
00043
00044 #include "EpetraExt_ModelEvaluator.h"
00045 #include "Epetra_Vector.h"
00046 #include "Epetra_Operator.h"
00047
00048
00049
00050 NOX::Epetra::ModelEvaluatorInterface::
00051 ModelEvaluatorInterface(const
00052 Teuchos::RCP<EpetraExt::ModelEvaluator>& m) :
00053 model_(m)
00054 {
00055 inargs_ = model_->createInArgs();
00056 outargs_ = model_->createOutArgs();
00057 }
00058
00059
00060
00061 NOX::Epetra::ModelEvaluatorInterface::~ModelEvaluatorInterface()
00062 {
00063
00064 }
00065
00066
00067
00068 bool NOX::Epetra::ModelEvaluatorInterface::
00069 computeF(const Epetra_Vector& x, Epetra_Vector& F, const FillType fillFlag)
00070 {
00071 x_ = Teuchos::rcp(&x, false);
00072 inargs_.set_x(x_);
00073
00074 jacobian_ = Teuchos::null;
00075 outargs_.set_W(jacobian_);
00076
00077 f_ = Teuchos::rcp(&F, false);
00078
00079 if (fillFlag == NOX::Epetra::Interface::Required::Residual)
00080 eval_f_.reset(f_, EpetraExt::ModelEvaluator::EVAL_TYPE_EXACT);
00081 else if (fillFlag == NOX::Epetra::Interface::Required::Jac)
00082 eval_f_.reset(f_, EpetraExt::ModelEvaluator::EVAL_TYPE_APPROX_DERIV);
00083 else
00084 eval_f_.reset(f_, EpetraExt::ModelEvaluator::EVAL_TYPE_VERY_APPROX_DERIV);
00085
00086 outargs_.set_f(eval_f_);
00087
00088 model_->evalModel(inargs_, outargs_);
00089
00090 return true;
00091 }
00092
00093
00094
00095 bool NOX::Epetra::ModelEvaluatorInterface::
00096 computeJacobian(const Epetra_Vector& x, Epetra_Operator& Jac)
00097 {
00098 x_ = Teuchos::rcp(&x, false);
00099 inargs_.set_x(x_);
00100
00101 f_ = Teuchos::null;
00102 eval_f_.reset(f_, EpetraExt::ModelEvaluator::EVAL_TYPE_EXACT);
00103 outargs_.set_f(eval_f_);
00104
00105 jacobian_ = Teuchos::rcp(&Jac, false);
00106 outargs_.set_W(jacobian_);
00107
00108 model_->evalModel(inargs_, outargs_);
00109
00110 return true;
00111 }
00112
00113
00114
00115 bool NOX::Epetra::ModelEvaluatorInterface::
00116 computePreconditioner(const Epetra_Vector& x,
00117 Epetra_Operator& M,
00118 Teuchos::ParameterList* precParams)
00119 {
00120 x_ = Teuchos::rcp(&x, false);
00121 inargs_.set_x(x_);
00122
00123 f_ = Teuchos::null;
00124 eval_f_.reset(f_, EpetraExt::ModelEvaluator::EVAL_TYPE_VERY_APPROX_DERIV);
00125 outargs_.set_f(eval_f_);
00126
00127 jacobian_ = Teuchos::rcp(&M, false);
00128 outargs_.set_W(jacobian_);
00129
00130 model_->evalModel(inargs_, outargs_);
00131
00132 return true;
00133 }
00134
00135
00136