43#ifndef __Panzer_ModelEvaluator_impl_hpp__
44#define __Panzer_ModelEvaluator_impl_hpp__
46#include "Teuchos_DefaultComm.hpp"
47#include "Teuchos_ArrayRCP.hpp"
49#include "PanzerDiscFE_config.hpp"
66#include "Thyra_TpetraThyraWrappers.hpp"
67#include "Thyra_SpmdVectorBase.hpp"
68#include "Thyra_DefaultSpmdVector.hpp"
69#include "Thyra_DefaultSpmdVectorSpace.hpp"
70#include "Thyra_DefaultMultiVectorProductVectorSpace.hpp"
71#include "Thyra_DefaultMultiVectorProductVector.hpp"
72#include "Thyra_MultiVectorStdOps.hpp"
73#include "Thyra_VectorStdOps.hpp"
76#include "Thyra_ProductVectorBase.hpp"
77#include "Thyra_BlockedLinearOpBase.hpp"
78#include "Thyra_TpetraVector.hpp"
79#include "Thyra_TpetraLinearOp.hpp"
80#include "Tpetra_CrsMatrix.hpp"
84template<
typename Scalar>
86ModelEvaluator(
const Teuchos::RCP<panzer::FieldManagerBuilder>& fmb,
89 const std::vector<Teuchos::RCP<Teuchos::Array<std::string> > >& p_names,
90 const std::vector<Teuchos::RCP<Teuchos::Array<double> > >& p_values,
92 const Teuchos::RCP<panzer::GlobalData>& global_data,
93 bool build_transient_support,
115 using Teuchos::rcp_dynamic_cast;
116 using Teuchos::tuple;
118 using Thyra::createMember;
120 TEUCHOS_ASSERT(
lof_!=Teuchos::null);
123 ae_tm_.buildObjects(builder);
130 RCP<const ThyraObjFactory<Scalar> > tof = rcp_dynamic_cast<const ThyraObjFactory<Scalar> >(lof,
true);
132 x_space_ = tof->getThyraDomainSpace();
133 f_space_ = tof->getThyraRangeSpace();
138 for(std::size_t i=0;i<p_names.size();i++)
146template<
typename Scalar>
150 const Teuchos::RCP<panzer::GlobalData>& global_data,
151 bool build_transient_support,
double t_init)
170 using Teuchos::rcp_dynamic_cast;
172 TEUCHOS_ASSERT(
lof_!=Teuchos::null);
179 RCP<const ThyraObjFactory<Scalar> > tof = rcp_dynamic_cast<const ThyraObjFactory<Scalar> >(
lof_,
true);
181 x_space_ = tof->getThyraDomainSpace();
182 f_space_ = tof->getThyraRangeSpace();
192template<
typename Scalar>
196 TEUCHOS_ASSERT(
false);
201template<
typename Scalar>
202Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
209template<
typename Scalar>
210Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
216template<
typename Scalar>
217Teuchos::RCP<const Teuchos::Array<std::string> >
221 "panzer::ModelEvaluator::get_p_names: Requested parameter index out of range.");
226 Teuchos::RCP<Teuchos::Array<std::string> > names = rcp(
new Teuchos::Array<std::string>);
228 std::ostringstream ss;
229 ss <<
"TANGENT VECTOR: " << param_index;
230 names->push_back(ss.str());
234 Teuchos::RCP<Teuchos::Array<std::string> > names = rcp(
new Teuchos::Array<std::string>);
236 std::ostringstream ss;
237 ss <<
"TIME DERIVATIVE TANGENT VECTOR: " << param_index;
238 names->push_back(ss.str());
242 return Teuchos::null;
245template<
typename Scalar>
246Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
250 "panzer::ModelEvaluator::get_p_space: Requested parameter index out of range.");
259 return Teuchos::null;
262template<
typename Scalar>
263Teuchos::ArrayView<const std::string>
266 TEUCHOS_TEST_FOR_EXCEPTION(!(i>=0 && i<Teuchos::as<int>(
responses_.size())),std::runtime_error,
267 "panzer::ModelEvaluator::get_g_names: Requested response index out of range.");
269 return Teuchos::ArrayView<const std::string>(&(
responses_[i]->name),1);
272template<
typename Scalar>
276 TEUCHOS_ASSERT(i>=0 &&
277 static_cast<typename std::vector<Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar>
> >
::size_type>(i)<
responses_.size());
282template<
typename Scalar>
283Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
286 TEUCHOS_ASSERT(i>=0 &&
287 static_cast<typename std::vector<Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar>
> >
::size_type>(i)<
responses_.size());
292template<
typename Scalar>
293Thyra::ModelEvaluatorBase::InArgs<Scalar>
299template<
typename Scalar>
300Thyra::ModelEvaluatorBase::InArgs<Scalar>
304 using Teuchos::rcp_dynamic_cast;
307 typedef Thyra::ModelEvaluatorBase MEB;
314 MEB::InArgsSetup<Scalar> nomInArgs;
337template<
typename Scalar>
341 typedef Thyra::ModelEvaluatorBase MEB;
347 MEB::InArgsSetup<Scalar> nomInArgs;
348 nomInArgs.setModelEvalDescription(this->description());
349 nomInArgs.setSupports(MEB::IN_ARG_x);
350 Teuchos::RCP<Thyra::VectorBase<Scalar> > x_nom = Thyra::createMember(
x_space_);
351 Thyra::assign(x_nom.ptr(),0.0);
352 nomInArgs.set_x(x_nom);
354 nomInArgs.setSupports(MEB::IN_ARG_x_dot,
true);
355 nomInArgs.setSupports(MEB::IN_ARG_t,
true);
356 nomInArgs.setSupports(MEB::IN_ARG_alpha,
true);
357 nomInArgs.setSupports(MEB::IN_ARG_beta,
true);
358 nomInArgs.setSupports(MEB::IN_ARG_step_size,
true);
359 nomInArgs.setSupports(MEB::IN_ARG_stage_number,
true);
361 Teuchos::RCP<Thyra::VectorBase<Scalar> > x_dot_nom = Thyra::createMember(
x_space_);
362 Thyra::assign(x_dot_nom.ptr(),0.0);
363 nomInArgs.set_x_dot(x_dot_nom);
365 nomInArgs.set_alpha(0.0);
366 nomInArgs.set_beta(0.0);
368 nomInArgs.set_step_size(0.0);
369 nomInArgs.set_stage_number(1.0);
374 std::size_t v_index = 0;
378 Teuchos::RCP<Thyra::VectorBase<Scalar> > v_nom_x = Thyra::createMember(*
tangent_space_[v_index]);
379 Thyra::assign(v_nom_x.ptr(),0.0);
380 nomInArgs.set_p(v_index+
parameters_.size(),v_nom_x);
382 Teuchos::RCP<Thyra::VectorBase<Scalar> > v_nom_xdot = Thyra::createMember(*
tangent_space_[v_index]);
383 Thyra::assign(v_nom_xdot.ptr(),0.0);
393template <
typename Scalar>
400template <
typename Scalar>
407template <
typename Scalar>
409setupModel(
const Teuchos::RCP<panzer::WorksetContainer> & wc,
410 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
411 const std::vector<panzer::BC> & bcs,
416 const Teuchos::ParameterList& closure_models,
417 const Teuchos::ParameterList& user_data,
418 bool writeGraph,
const std::string & graphPrefix,
419 const Teuchos::ParameterList& me_params)
423 PANZER_FUNC_TIME_MONITOR_DIFF(
"panzer::ModelEvaluator::setupModel()",
setupModel);
429 Teuchos::RCP<panzer::FieldManagerBuilder> fmb;
431 PANZER_FUNC_TIME_MONITOR_DIFF(
"allocate FieldManagerBuilder",allocFMB);
436 PANZER_FUNC_TIME_MONITOR_DIFF(
"fmb->setWorksetContainer()",setupWorksets);
437 fmb->setWorksetContainer(wc);
440 PANZER_FUNC_TIME_MONITOR_DIFF(
"fmb->setupVolumeFieldManagers()",setupVolumeFieldManagers);
441 fmb->setupVolumeFieldManagers(physicsBlocks,volume_cm_factory,closure_models,*
lof_,user_data);
444 PANZER_FUNC_TIME_MONITOR_DIFF(
"fmb->setupBCFieldManagers()",setupBCFieldManagers);
445 fmb->setupBCFieldManagers(bcs,physicsBlocks,eqset_factory,bc_cm_factory,bc_factory,closure_models,*
lof_,user_data);
451 fmb->writeVolumeGraphvizDependencyFiles(graphPrefix, physicsBlocks);
453 fmb->writeBCGraphvizDependencyFiles(graphPrefix+
"BC_");
457 PANZER_FUNC_TIME_MONITOR_DIFF(
"AssemblyEngine_TemplateBuilder::buildObjects()",AETM_BuildObjects);
459 ae_tm_.buildObjects(builder);
467 PANZER_FUNC_TIME_MONITOR_DIFF(
"build response library",
buildResponses);
471 buildResponses(physicsBlocks,eqset_factory,volume_cm_factory,closure_models,user_data,writeGraph,graphPrefix+
"Responses_");
472 buildDistroParamDfDp_RL(wc,physicsBlocks,bcs,eqset_factory,bc_factory,volume_cm_factory,closure_models,user_data,writeGraph,graphPrefix+
"Response_DfDp_");
473 buildDistroParamDgDp_RL(wc,physicsBlocks,bcs,eqset_factory,bc_factory,volume_cm_factory,closure_models,user_data,writeGraph,graphPrefix+
"Response_DgDp_");
477 if (me_params.isParameter(
"FD Forward Sensitivities"))
478 do_fd_dfdp_ = me_params.get<
bool>(
"FD Forward Sensitivities");
479 if (me_params.isParameter(
"FD Perturbation Size"))
484template <
typename Scalar>
491 using Teuchos::rcp_dynamic_cast;
492 using Teuchos::rcp_const_cast;
493 typedef Thyra::ModelEvaluatorBase MEB;
504 bool is_transient =
false;
505 if (inArgs.supports(MEB::IN_ARG_x_dot ))
506 is_transient = !Teuchos::is_null(inArgs.get_x_dot());
513 const RCP<const Thyra::VectorBase<Scalar> > x = inArgs.get_x();
514 RCP<const Thyra::VectorBase<Scalar> > x_dot;
518 "ModelEvaluator was not built with transient support enabled!");
522 ae_inargs.
alpha = 0.0;
523 ae_inargs.
beta = 1.0;
526 x_dot = inArgs.get_x_dot();
527 ae_inargs.
alpha = inArgs.get_alpha();
528 ae_inargs.
beta = inArgs.get_beta();
529 ae_inargs.
time = inArgs.get_t();
531 ae_inargs.
step_size= inArgs.get_step_size();
541 for (
int i=0; i<num_param_vecs; i++) {
543 RCP<const Thyra::VectorBase<Scalar> > paramVec = inArgs.get_p(i);
544 if ( paramVec!=Teuchos::null && !
parameters_[i]->is_distributed) {
547 Teuchos::ArrayRCP<const Scalar> p_data;
548 rcp_dynamic_cast<const Thyra::SpmdVectorBase<Scalar> >(paramVec,
true)->getLocalData(Teuchos::ptrFromRef(p_data));
550 for (
unsigned int j=0; j <
parameters_[i]->scalar_value.size(); j++) {
551 parameters_[i]->scalar_value[j].baseValue = p_data[j];
552 parameters_[i]->scalar_value[j].family->setRealValueForAllTypes(
parameters_[i]->scalar_value[j].baseValue);
555 else if ( paramVec!=Teuchos::null &&
parameters_[i]->is_distributed) {
561 TEUCHOS_ASSERT(ged!=Teuchos::null);
564 RCP<LOCPair_GlobalEvaluationData> loc_pair_ged = rcp_dynamic_cast<LOCPair_GlobalEvaluationData>(ged);
565 RCP<ReadOnlyVector_GlobalEvaluationData> ro_ged = rcp_dynamic_cast<ReadOnlyVector_GlobalEvaluationData>(ged);
566 if(loc_pair_ged!=Teuchos::null) {
568 RCP<ThyraObjContainer<Scalar> > th_ged = rcp_dynamic_cast<ThyraObjContainer<Scalar> >(loc_pair_ged->getGlobalLOC(),
true);
572 TEUCHOS_ASSERT(ro_ged!=Teuchos::null);
573 ro_ged->setOwnedVector(paramVec);
581 const RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
582 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
container_);
584 TEUCHOS_ASSERT(!Teuchos::is_null(thGlobalContainer));
615 for (
int i(0); i < num_param_vecs; ++i)
623 auto dxdp = rcp_const_cast<VectorBase<Scalar>>
624 (inArgs.get_p(vIndex + num_param_vecs));
625 if (not dxdp.is_null())
629 auto dxdpBlock = rcp_dynamic_cast<ProductVectorBase<Scalar>>(dxdp);
630 int numParams(
parameters_[i]->scalar_value.size());
631 for (
int j(0); j < numParams; ++j)
633 RCP<ROVGED> dxdpContainer =
lof_->buildReadOnlyDomainContainer();
634 dxdpContainer->setOwnedVector(dxdpBlock->getNonconstVectorBlock(j));
635 string name(
"X TANGENT GATHER CONTAINER: " +
644 auto dxdotdp = rcp_const_cast<VectorBase<Scalar>>
646 if (not dxdotdp.is_null())
649 rcp_dynamic_cast<ProductVectorBase<Scalar>>(dxdotdp);
650 int numParams(
parameters_[i]->scalar_value.size());
651 for (
int j(0); j < numParams; ++j)
653 RCP<ROVGED> dxdotdpContainer =
lof_->buildReadOnlyDomainContainer();
654 dxdotdpContainer->setOwnedVector(
655 dxdotdpBlock->getNonconstVectorBlock(j));
656 string name(
"DXDT TANGENT GATHER CONTAINER: " +
670template <
typename Scalar>
671Thyra::ModelEvaluatorBase::OutArgs<Scalar>
674 typedef Thyra::ModelEvaluatorBase MEB;
677 MEB::OutArgsSetup<Scalar> outArgs;
678 outArgs.setModelEvalDescription(this->description());
680 outArgs.setSupports(MEB::OUT_ARG_f);
681 outArgs.setSupports(MEB::OUT_ARG_W_op);
684 for(std::size_t i=0;i<
responses_.size();i++) {
688 Teuchos::RCP<panzer::ResponseBase> respJacBase
690 if(respJacBase!=Teuchos::null) {
692 Teuchos::RCP<panzer::ResponseMESupportBase<RespEvalT> > resp
693 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<RespEvalT> >(respJacBase);
696 if(resp->supportsDerivative()) {
697 outArgs.setSupports(MEB::OUT_ARG_DgDx,i,MEB::DerivativeSupport(MEB::DERIV_MV_GRADIENT_FORM));
702 outArgs.setSupports(MEB::OUT_ARG_DgDp,i,p,MEB::DerivativeSupport(MEB::DERIV_MV_GRADIENT_FORM));
704 outArgs.setSupports(MEB::OUT_ARG_DgDp,i,p,MEB::DerivativeSupport(MEB::DERIV_MV_JACOBIAN_FORM));
714 outArgs.setSupports(MEB::OUT_ARG_DfDp,p,MEB::DerivativeSupport(MEB::DERIV_MV_BY_COL));
716 outArgs.setSupports(MEB::OUT_ARG_DfDp,p,MEB::DerivativeSupport(MEB::DERIV_LINEAR_OP));
728template <
typename Scalar>
729Teuchos::RCP<Thyra::LinearOpBase<Scalar> >
733 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::create_W_op");
734 Teuchos::RCP<const ThyraObjFactory<Scalar> > tof
735 = Teuchos::rcp_dynamic_cast<const ThyraObjFactory<Scalar> >(
lof_,
true);
737 return tof->getThyraMatrix();
740template <
typename Scalar>
741Teuchos::RCP<const Thyra::LinearOpWithSolveFactoryBase<Scalar> >
748template <
typename Scalar>
749Teuchos::RCP<Thyra::LinearOpBase<Scalar> >
754 using Teuchos::rcp_dynamic_cast;
756 typedef Thyra::ModelEvaluatorBase MEB;
769 this->createOutArgs();
772 TEUCHOS_ASSERT(0<=p && p<Teuchos::as<int>(
parameters_.size()));
778 TEUCHOS_ASSERT(
prototypeOutArgs_.supports(MEB::OUT_ARG_DfDp,p).supports(MEB::DERIV_LINEAR_OP));
782 RCP<Response_Residual<Traits::Jacobian> > response_jacobian
783 = rcp_dynamic_cast<Response_Residual<Traits::Jacobian> >(po.
dfdp_rl->template getResponse<Traits::Jacobian>(
"RESIDUAL"));
785 return response_jacobian->allocateJacobian();
788 TEUCHOS_ASSERT(
prototypeOutArgs_.supports(MEB::OUT_ARG_DfDp,p).supports(MEB::DERIV_MV_BY_COL));
795 TEUCHOS_ASSERT(
false);
797 return Teuchos::null;
800template <
typename Scalar>
802addParameter(
const std::string & name,
const Scalar & initialValue)
804 Teuchos::Array<std::string> tmp_names;
805 tmp_names.push_back(name);
807 Teuchos::Array<Scalar> tmp_values;
808 tmp_values.push_back(initialValue);
813template <
typename Scalar>
816 const Teuchos::Array<Scalar> & initialValues)
820 using Teuchos::rcp_dynamic_cast;
821 using Teuchos::ptrFromRef;
823 TEUCHOS_ASSERT(names.size()==initialValues.size());
832 RCP< Thyra::VectorSpaceBase<double> > tan_space =
833 Thyra::multiVectorProductVectorSpace(
x_space_, param->names->size());
844 this->resetDefaultBase();
846 return parameter_index;
849template <
typename Scalar>
853 const Teuchos::RCP<GlobalEvaluationData> & ged,
855 const Teuchos::RCP<const GlobalIndexer> & ugi)
865 this->resetDefaultBase();
867 return parameter_index;
870template <
typename Scalar>
873 const Teuchos::RCP<GlobalEvaluationData> & ged)
878template <
typename Scalar>
881 const std::vector<WorksetDescriptor> & wkst_desc,
882 const Teuchos::RCP<ResponseMESupportBuilderBase> & builder)
885 builder->setDerivativeInformation(lof_);
887 int respIndex = addResponse(responseName,wkst_desc,*builder);
890 responses_[respIndex]->builder = builder;
896template <
typename Scalar>
902 using Teuchos::ArrayRCP;
903 using Teuchos::Array;
904 using Teuchos::tuple;
905 using Teuchos::rcp_dynamic_cast;
917 ae_inargs.
alpha = 0.0;
918 ae_inargs.
beta = 1.0;
930 const RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
931 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
container_);
933 TEUCHOS_ASSERT(!Teuchos::is_null(thGlobalContainer));
938 const RCP<panzer::ThyraObjContainer<Scalar> > thGhostedContainer =
939 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
ghostedContainer_);
940 Thyra::assign(thGhostedContainer->get_f_th().ptr(),0.0);
949 thGlobalContainer->set_x_th(x);
952 RCP<panzer::LinearObjContainer> counter
953 =
ae_tm_.template getAsObject<panzer::Traits::Residual>()->evaluateOnlyDirichletBCs(ae_inargs);
956 RCP<panzer::LinearObjContainer> result =
lof_->buildLinearObjContainer();
959 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(counter)->set_f_th(
960 thGlobalContainer->get_f_th());
963 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(result)->set_f_th(f);
966 lof_->applyDirichletBCs(*counter,*result);
969template <
typename Scalar>
972 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
976#ifdef Panzer_BUILD_HESSIAN_SUPPORT
982 std::string responseName =
responses_[respIndex]->name;
983 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Hessian> > resp
984 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Hessian> >(
986 resp->setDerivative(D2gDx2);
995 ae_inargs.
beta = 1.0;
997 auto deltaXContainer =
lof_->buildReadOnlyDomainContainer();
998 deltaXContainer->setOwnedVector(delta_x);
1012 TEUCHOS_ASSERT(
false);
1016template <
typename Scalar>
1020 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1024#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1030 std::string responseName =
responses_[respIndex]->name;
1031 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Hessian> > resp
1032 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Hessian> >(
1034 resp->setDerivative(D2gDxDp);
1043 ae_inargs.
beta = 1.0;
1046 auto deltaPContainer =
parameters_[pIndex]->dfdp_rl->getLinearObjFactory()->buildReadOnlyDomainContainer();
1047 deltaPContainer->setOwnedVector(delta_p);
1062 TEUCHOS_ASSERT(
false);
1066template <
typename Scalar>
1070 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1074#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1082 std::string responseName =
responses_[respIndex]->name;
1083 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Hessian> > resp
1084 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Hessian> >(
1086 resp->setDerivative(D2gDp2);
1100 auto deltaPContainer =
parameters_[pIndex]->dfdp_rl->getLinearObjFactory()->buildReadOnlyDomainContainer();
1101 deltaPContainer->setOwnedVector(delta_p);
1116 TEUCHOS_ASSERT(
false);
1120template <
typename Scalar>
1124 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1128#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1136 std::string responseName =
responses_[respIndex]->name;
1137 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Hessian> > resp
1138 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Hessian> >(
1140 resp->setDerivative(D2gDpDx);
1154 auto deltaXContainer =
lof_->buildReadOnlyDomainContainer();
1155 deltaXContainer->setOwnedVector(delta_x);
1170 TEUCHOS_ASSERT(
false);
1174template <
typename Scalar>
1180#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1183 using Teuchos::ArrayRCP;
1184 using Teuchos::Array;
1185 using Teuchos::tuple;
1186 using Teuchos::rcp_dynamic_cast;
1188 typedef Thyra::ModelEvaluatorBase MEB;
1193 bool is_transient =
false;
1194 if (inArgs.supports(MEB::IN_ARG_x_dot ))
1195 is_transient = !Teuchos::is_null(inArgs.get_x_dot());
1199 "ModelEvaluator was not built with transient support enabled!");
1204 const RCP<Thyra::LinearOpBase<Scalar> > W_out = D2fDx2;
1212 auto deltaXContainer =
lof_->buildReadOnlyDomainContainer();
1213 deltaXContainer->setOwnedVector(delta_x);
1231 const RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
1232 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
container_);
1233 const RCP<panzer::ThyraObjContainer<Scalar> > thGhostedContainer =
1234 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
ghostedContainer_);
1237 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(D2fDx2)");
1240 RCP<Thyra::VectorBase<Scalar> > dummy_f = Thyra::createMember(
f_space_);
1241 thGlobalContainer->set_f_th(dummy_f);
1242 thGlobalContainer->set_A_th(W_out);
1245 thGhostedContainer->initializeMatrix(0.0);
1247 ae_tm_.template getAsObject<panzer::Traits::Hessian>()->evaluate(ae_inargs);
1253 thGlobalContainer->set_A_th(Teuchos::null);
1257 thGlobalContainer->set_x_th(Teuchos::null);
1258 thGlobalContainer->set_dxdt_th(Teuchos::null);
1259 thGlobalContainer->set_f_th(Teuchos::null);
1260 thGlobalContainer->set_A_th(Teuchos::null);
1268 TEUCHOS_ASSERT(
false);
1272template <
typename Scalar>
1275 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1279#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1282 using Teuchos::ArrayRCP;
1283 using Teuchos::Array;
1284 using Teuchos::tuple;
1285 using Teuchos::rcp_dynamic_cast;
1287 typedef Thyra::ModelEvaluatorBase MEB;
1292 bool is_transient =
false;
1293 if (inArgs.supports(MEB::IN_ARG_x_dot ))
1294 is_transient = !Teuchos::is_null(inArgs.get_x_dot());
1298 "ModelEvaluator was not built with transient support enabled!");
1303 const RCP<Thyra::LinearOpBase<Scalar> > W_out = D2fDxDp;
1313 auto deltaPContainer =
parameters_[pIndex]->dfdp_rl->getLinearObjFactory()->buildReadOnlyDomainContainer();
1314 deltaPContainer->setOwnedVector(delta_p);
1332 const RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
1333 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
container_);
1334 const RCP<panzer::ThyraObjContainer<Scalar> > thGhostedContainer =
1335 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
ghostedContainer_);
1338 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(D2fDxDp)");
1341 RCP<Thyra::VectorBase<Scalar> > dummy_f = Thyra::createMember(
f_space_);
1342 thGlobalContainer->set_f_th(dummy_f);
1343 thGlobalContainer->set_A_th(W_out);
1346 thGhostedContainer->initializeMatrix(0.0);
1348 ae_tm_.template getAsObject<panzer::Traits::Hessian>()->evaluate(ae_inargs);
1354 thGlobalContainer->set_A_th(Teuchos::null);
1358 thGlobalContainer->set_x_th(Teuchos::null);
1359 thGlobalContainer->set_dxdt_th(Teuchos::null);
1360 thGlobalContainer->set_f_th(Teuchos::null);
1361 thGlobalContainer->set_A_th(Teuchos::null);
1370 TEUCHOS_ASSERT(
false);
1374template <
typename Scalar>
1377 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1381#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1383 using Teuchos::rcp_dynamic_cast;
1384 using Teuchos::null;
1387 TEUCHOS_ASSERT(
parameters_[pIndex]->is_distributed);
1391 TEUCHOS_ASSERT(
parameters_[pIndex]->dfdp_rl!=null);
1396 RCP<Response_Residual<Traits::Hessian> > response_hessian =
1398 response_hessian->setHessian(D2fDpDx);
1405 auto deltaXContainer =
lof_->buildReadOnlyDomainContainer();
1406 deltaXContainer->setOwnedVector(delta_x);
1421 TEUCHOS_ASSERT(
false);
1425template <
typename Scalar>
1428 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & inArgs,
1432#ifdef Panzer_BUILD_HESSIAN_SUPPORT
1434 using Teuchos::rcp_dynamic_cast;
1435 using Teuchos::null;
1438 TEUCHOS_ASSERT(
parameters_[pIndex]->is_distributed);
1442 TEUCHOS_ASSERT(
parameters_[pIndex]->dfdp_rl!=null);
1447 RCP<Response_Residual<Traits::Hessian> > response_hessian =
1449 response_hessian->setHessian(D2fDp2);
1456 auto deltaPContainer =
parameters_[pIndex]->dfdp_rl->getLinearObjFactory()->buildReadOnlyDomainContainer();
1457 deltaPContainer->setOwnedVector(delta_p);
1472 TEUCHOS_ASSERT(
false);
1476template <
typename Scalar>
1478evalModelImpl(
const Thyra::ModelEvaluatorBase::InArgs<Scalar> &inArgs,
1479 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1510template <
typename Scalar>
1513 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1516 using Teuchos::ArrayRCP;
1517 using Teuchos::Array;
1518 using Teuchos::tuple;
1519 using Teuchos::rcp_dynamic_cast;
1521 typedef Thyra::ModelEvaluatorBase MEB;
1526 bool is_transient =
false;
1527 if (inArgs.supports(MEB::IN_ARG_x_dot ))
1528 is_transient = !Teuchos::is_null(inArgs.get_x_dot());
1532 "ModelEvaluator was not built with transient support enabled!");
1537 const RCP<Thyra::VectorBase<Scalar> > f_out = outArgs.get_f();
1538 const RCP<Thyra::LinearOpBase<Scalar> > W_out = outArgs.get_W_op();
1541 if(Teuchos::is_null(f_out) && Teuchos::is_null(W_out) ) {
1566 const RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
1567 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
container_);
1568 const RCP<panzer::ThyraObjContainer<Scalar> > thGhostedContainer =
1569 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(ae_inargs.
ghostedContainer_);
1571 if (!Teuchos::is_null(f_out) && !Teuchos::is_null(W_out)) {
1572 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(f and J)");
1578 thGlobalContainer->set_f_th(f_out);
1579 thGlobalContainer->set_A_th(W_out);
1582 Thyra::assign(thGhostedContainer->get_f_th().ptr(),0.0);
1583 thGhostedContainer->initializeMatrix(0.0);
1585 ae_tm_.template getAsObject<panzer::Traits::Jacobian>()->evaluate(ae_inargs);
1587 else if(!Teuchos::is_null(f_out) && Teuchos::is_null(W_out)) {
1589 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(f)");
1594 thGlobalContainer->set_f_th(f_out);
1597 Thyra::assign(thGhostedContainer->get_f_th().ptr(),0.0);
1599 ae_tm_.template getAsObject<panzer::Traits::Residual>()->evaluate(ae_inargs);
1601 else if(Teuchos::is_null(f_out) && !Teuchos::is_null(W_out)) {
1603 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(J)");
1609 RCP<Thyra::VectorBase<Scalar> > dummy_f = Thyra::createMember(
f_space_);
1610 thGlobalContainer->set_f_th(dummy_f);
1611 thGlobalContainer->set_A_th(W_out);
1614 thGhostedContainer->initializeMatrix(0.0);
1616 ae_tm_.template getAsObject<panzer::Traits::Jacobian>()->evaluate(ae_inargs);
1622 thGlobalContainer->set_A_th(Teuchos::null);
1626 thGlobalContainer->set_x_th(Teuchos::null);
1627 thGlobalContainer->set_dxdt_th(Teuchos::null);
1628 thGlobalContainer->set_f_th(Teuchos::null);
1629 thGlobalContainer->set_A_th(Teuchos::null);
1634 const bool writeToFile =
false;
1635 if (writeToFile && nonnull(W_out)) {
1636 const auto check_blocked = Teuchos::rcp_dynamic_cast<::Thyra::BlockedLinearOpBase<double> >(W_out,
false);
1637 if (check_blocked) {
1638 const int numBlocks = check_blocked->productDomain()->numBlocks();
1639 const int rangeBlocks = check_blocked->productRange()->numBlocks();
1640 TEUCHOS_ASSERT(numBlocks == rangeBlocks);
1641 for (
int row=0; row < numBlocks; ++row) {
1642 for (
int col=0; col < numBlocks; ++col) {
1643 using LO = panzer::LocalOrdinal;
1644 using GO = panzer::GlobalOrdinal;
1646 const auto thyraTpetraOperator = Teuchos::rcp_dynamic_cast<::Thyra::TpetraLinearOp<double,LO,GO,NodeT>>(check_blocked->getNonconstBlock(row,col),
true);
1647 const auto tpetraCrsMatrix = Teuchos::rcp_dynamic_cast<Tpetra::CrsMatrix<double,LO,GO,NodeT>>(thyraTpetraOperator->getTpetraOperator(),
true);
1648 tpetraCrsMatrix->print(std::cout);
1649 std::stringstream ss;
1650 ss <<
"W_out_" <<
write_matrix_count_ <<
".rank_" << tpetraCrsMatrix->getMap()->getComm()->getRank() <<
".block_" << row <<
"_" << col <<
".txt";
1651 std::fstream fs(ss.str().c_str(),std::fstream::out|std::fstream::trunc);
1652 Teuchos::FancyOStream fos(Teuchos::rcpFromRef(fs));
1653 tpetraCrsMatrix->describe(fos,Teuchos::VERB_EXTREME);
1659 using LO = panzer::LocalOrdinal;
1660 using GO = panzer::GlobalOrdinal;
1662 const auto thyraTpetraOperator = Teuchos::rcp_dynamic_cast<::Thyra::TpetraLinearOp<double,LO,GO,NodeT>>(W_out,
true);
1663 const auto tpetraCrsMatrix = Teuchos::rcp_dynamic_cast<Tpetra::CrsMatrix<double,LO,GO,NodeT>>(thyraTpetraOperator->getTpetraOperator(),
true);
1664 tpetraCrsMatrix->print(std::cout);
1665 std::stringstream ss;
1666 ss <<
"W_out_" <<
write_matrix_count_ <<
".rank_" << tpetraCrsMatrix->getMap()->getComm()->getRank() <<
".txt";
1667 std::fstream fs(ss.str().c_str(),std::fstream::out|std::fstream::trunc);
1668 Teuchos::FancyOStream fos(Teuchos::rcpFromRef(fs));
1669 tpetraCrsMatrix->describe(fos,Teuchos::VERB_EXTREME);
1677template <
typename Scalar>
1680 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1694 for(std::size_t i=0;i<
responses_.size();i++) {
1695 Teuchos::RCP<Thyra::VectorBase<Scalar> > vec = outArgs.get_g(i);
1696 if(vec!=Teuchos::null) {
1697 std::string responseName =
responses_[i]->name;
1698 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Residual> > resp
1699 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Residual> >(
1701 resp->setVector(vec);
1713template <
typename Scalar>
1717 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1719 typedef Thyra::ModelEvaluatorBase MEB;
1727 for(std::size_t i=0;i<
responses_.size();i++) {
1729 MEB::Derivative<Scalar> deriv = outArgs.get_DgDx(i);
1733 Teuchos::RCP<Thyra::MultiVectorBase<Scalar> > vec = deriv.getMultiVector();
1735 if(vec!=Teuchos::null) {
1737 std::string responseName =
responses_[i]->name;
1738 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Jacobian> > resp
1739 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Jacobian> >(
1741 resp->setDerivative(vec);
1759template <
typename Scalar>
1763 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1767 using Teuchos::rcp_dynamic_cast;
1769 typedef Thyra::ModelEvaluatorBase MEB;
1775 std::vector<std::string> activeParameterNames;
1776 std::vector<int> activeParameters;
1777 int totalParameterCount = 0;
1784 bool is_active =
false;
1785 for(std::size_t i=0;i<
responses_.size(); i++) {
1787 MEB::Derivative<Scalar> deriv = outArgs.get_DgDp(i,j);
1791 Teuchos::RCP<Thyra::MultiVectorBase<Scalar> > vec = deriv.getMultiVector();
1792 if(vec!=Teuchos::null) {
1794 std::string responseName =
responses_[i]->name;
1795 RCP<panzer::ResponseMESupportBase<panzer::Traits::Tangent> > resp =
1796 rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Tangent> >(
1798 resp->setVector(vec);
1804 for (std::size_t k=0; k<
parameters_[j]->scalar_value.size(); k++) {
1805 std::string name =
"PARAMETER_SENSITIVIES: "+(*
parameters_[j]->names)[k];
1806 activeParameterNames.push_back(name);
1807 totalParameterCount++;
1809 activeParameters.push_back(j);
1818 RCP<panzer::GlobalEvaluationData> ged_activeParameters =
1824 for (std::size_t ap=0; ap<activeParameters.size(); ++ap) {
1825 const int j = activeParameters[ap];
1826 for (
unsigned int k=0; k <
parameters_[j]->scalar_value.size(); k++) {
1828 p.fastAccessDx(paramIndex) = 1.0;
1829 parameters_[j]->scalar_value[k].family->template setValue<panzer::Traits::Tangent>(p);
1835 TEUCHOS_ASSERT(paramIndex==totalParameterCount);
1838 if(totalParameterCount>0) {
1844template <
typename Scalar>
1848 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1850 typedef Thyra::ModelEvaluatorBase MEB;
1868 for(std::size_t r=0;r<
responses_.size();r++) {
1870 MEB::Derivative<Scalar> deriv = outArgs.get_DgDp(r,p);
1874 Teuchos::RCP<Thyra::MultiVectorBase<Scalar> > vec = deriv.getMultiVector();
1876 if(vec!=Teuchos::null) {
1879 std::string responseName =
responses_[r]->name;
1880 Teuchos::RCP<panzer::ResponseMESupportBase<panzer::Traits::Jacobian> > resp
1881 = Teuchos::rcp_dynamic_cast<panzer::ResponseMESupportBase<panzer::Traits::Jacobian> >(
1884 resp->setDerivative(vec);
1903template <
typename Scalar>
1907 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
1910 using Teuchos::rcp_dynamic_cast;
1912 typedef Thyra::ModelEvaluatorBase MEB;
1926 std::vector<std::string> activeParameters;
1928 int totalParameterCount = 0;
1929 for(std::size_t i=0; i <
parameters_.size(); i++) {
1935 MEB::Derivative<Scalar> deriv = outArgs.get_DfDp(i);
1940 Teuchos::RCP<Thyra::MultiVectorBase<Scalar> > mVec = deriv.getMultiVector();
1941 TEUCHOS_ASSERT(mVec->domain()->dim()==Teuchos::as<int>(
parameters_[i]->scalar_value.size()));
1943 for (std::size_t j=0; j <
parameters_[i]->scalar_value.size(); j++) {
1946 RCP<LOCPair_GlobalEvaluationData> loc_pair
1948 RCP<LinearObjContainer> globalContainer = loc_pair->getGlobalLOC();
1951 RCP<Thyra::VectorBase<Scalar> > vec = mVec->col(j);
1952 RCP<panzer::ThyraObjContainer<Scalar> > thGlobalContainer =
1953 Teuchos::rcp_dynamic_cast<panzer::ThyraObjContainer<Scalar> >(globalContainer);
1954 thGlobalContainer->set_f_th(vec);
1957 std::string name =
"PARAMETER_SENSITIVIES: "+(*
parameters_[i]->names)[j];
1961 activeParameters.push_back(name);
1962 totalParameterCount++;
1970 RCP<GlobalEvaluationData> ged_activeParameters
1979 for(std::size_t i=0; i <
parameters_.size(); i++) {
1985 MEB::Derivative<Scalar> deriv = outArgs.get_DfDp(i);
1986 if(deriv.isEmpty()) {
1988 for (
unsigned int j=0; j <
parameters_[i]->scalar_value.size(); j++) {
1991 parameters_[i]->scalar_value[j].family->template setValue<panzer::Traits::Tangent>(p);
1997 for (
unsigned int j=0; j <
parameters_[i]->scalar_value.size(); j++) {
2000 p.fastAccessDx(paramIndex) = 1.0;
2001 parameters_[i]->scalar_value[j].family->template setValue<panzer::Traits::Tangent>(p);
2008 TEUCHOS_ASSERT(paramIndex==totalParameterCount);
2013 if(totalParameterCount>0) {
2014 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(df/dp)");
2019template <
typename Scalar>
2023 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
2025 PANZER_FUNC_TIME_MONITOR(
"panzer::ModelEvaluator::evalModel(df/dp)");
2028 using Teuchos::rcp_dynamic_cast;
2030 typedef Thyra::ModelEvaluatorBase MEB;
2037 MEB::OutArgs<Scalar> outArgs_base = this->createOutArgs();
2038 if (outArgs.get_f() == Teuchos::null)
2039 outArgs_base.set_f(Thyra::createMember(this->
get_f_space()));
2041 outArgs_base.set_f(outArgs.get_f());
2042 outArgs_base.set_W_op(outArgs.get_W_op());
2043 this->evalModel(inArgs, outArgs_base);
2044 RCP<const Thyra::VectorBase<Scalar> > f = outArgs_base.get_f();
2045 RCP<const Thyra::VectorBase<Scalar> > x = inArgs.get_x();
2046 RCP<const Thyra::VectorBase<Scalar> > x_dot;
2047 if (inArgs.supports(MEB::IN_ARG_x_dot))
2048 x_dot = inArgs.get_x_dot();
2051 RCP<Thyra::VectorBase<Scalar> > fd = Thyra::createMember(this->
get_f_space());
2052 MEB::OutArgs<Scalar> outArgs_fd = this->createOutArgs();
2053 outArgs_fd.set_f(fd);
2055 RCP<Thyra::VectorBase<Scalar> > xd = Thyra::createMember(this->
get_x_space());
2056 RCP<Thyra::VectorBase<Scalar> > xd_dot;
2057 if (x_dot != Teuchos::null)
2058 xd_dot = Thyra::createMember(this->
get_x_space());
2060 inArgs_fd.setArgs(inArgs);
2061 inArgs_fd.set_x(xd);
2062 if (x_dot != Teuchos::null)
2063 inArgs_fd.set_x_dot(xd_dot);
2066 for(std::size_t i=0; i <
parameters_.size(); i++) {
2073 MEB::Derivative<Scalar> deriv = outArgs.get_DfDp(i);
2078 RCP<Thyra::MultiVectorBase<Scalar> > dfdp = deriv.getMultiVector();
2079 TEUCHOS_ASSERT(dfdp->domain()->dim()==Teuchos::as<int>(
parameters_[i]->scalar_value.size()));
2082 RCP<const Thyra::VectorBase<Scalar> > p = inArgs.get_p(i);
2083 RCP<const Thyra::VectorBase<Scalar> > dx_v = inArgs.get_p(i+
parameters_.size());
2084 RCP<const Thyra::MultiVectorBase<Scalar> > dx =
2085 rcp_dynamic_cast<const Thyra::DefaultMultiVectorProductVector<Scalar> >(dx_v,
true)->getMultiVector();
2086 RCP<const Thyra::VectorBase<Scalar> > dx_dot_v;
2087 RCP<const Thyra::MultiVectorBase<Scalar> > dx_dot;
2088 if (x_dot != Teuchos::null) {
2091 rcp_dynamic_cast<const Thyra::DefaultMultiVectorProductVector<Scalar> >(dx_dot_v,
true)->getMultiVector();
2095 RCP<Thyra::VectorBase<Scalar> > pd = Thyra::createMember(this->
get_p_space(i));
2096 inArgs_fd.set_p(i,pd);
2098 for (std::size_t j=0; j <
parameters_[i]->scalar_value.size(); j++) {
2101 Thyra::copy(*p, pd.ptr());
2102 Thyra::set_ele(j, Thyra::get_ele(*p,j)+h, pd.ptr());
2105 Thyra::V_VpStV(xd.ptr(), *x, h, *(dx)->col(j));
2106 if (x_dot != Teuchos::null)
2107 Thyra::V_VpStV(xd_dot.ptr(), *x_dot, h, *(dx_dot)->col(j));
2110 Thyra::assign(fd.ptr(), 0.0);
2111 this->evalModel(inArgs_fd, outArgs_fd);
2114 Thyra::V_StVpStV(dfdp->col(j).ptr(), 1.0/h, *fd, -1.0/h, *f);
2117 parameters_[i]->scalar_value[j].family->setRealValueForAllTypes(Thyra::get_ele(*p,j));
2123template <
typename Scalar>
2127 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
2130 using Teuchos::rcp_dynamic_cast;
2131 using Teuchos::null;
2133 typedef Thyra::ModelEvaluatorBase MEB;
2154 MEB::Derivative<Scalar> deriv = outArgs.get_DfDp(p);
2161 RCP<Response_Residual<Traits::Jacobian> > response_jacobian =
2163 response_jacobian->setJacobian(deriv.getLinearOp());
2179template <
typename Scalar>
2181required_basic_g(
const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs)
const
2184 bool activeGArgs =
false;
2185 for(
int i=0;i<outArgs.Ng();i++)
2186 activeGArgs |= (outArgs.get_g(i)!=Teuchos::null);
2191template <
typename Scalar>
2195 typedef Thyra::ModelEvaluatorBase MEB;
2198 bool activeGArgs =
false;
2199 for(
int i=0;i<outArgs.Ng();i++) {
2201 if(outArgs.supports(MEB::OUT_ARG_DgDx,i).none())
2205 activeGArgs |= (!outArgs.get_DgDx(i).isEmpty());
2211template <
typename Scalar>
2215 typedef Thyra::ModelEvaluatorBase MEB;
2218 bool activeGArgs =
false;
2219 for(
int i=0;i<outArgs.Ng();i++) {
2220 for(
int p=0;p<Teuchos::as<int>(
parameters_.size());p++) {
2227 if(outArgs.supports(MEB::OUT_ARG_DgDp,i,p).none())
2230 activeGArgs |= (!outArgs.get_DgDp(i,p).isEmpty());
2237template <
typename Scalar>
2241 typedef Thyra::ModelEvaluatorBase MEB;
2244 bool activeGArgs =
false;
2245 for(
int i=0;i<outArgs.Ng();i++) {
2246 for(
int p=0;p<Teuchos::as<int>(
parameters_.size());p++) {
2253 if(outArgs.supports(MEB::OUT_ARG_DgDp,i,p).none())
2256 activeGArgs |= (!outArgs.get_DgDp(i,p).isEmpty());
2263template <
typename Scalar>
2267 typedef Thyra::ModelEvaluatorBase MEB;
2270 bool activeFPArgs =
false;
2271 for(
int i=0;i<Teuchos::as<int>(
parameters_.size());i++) {
2278 if(outArgs.supports(MEB::OUT_ARG_DfDp,i).none())
2282 activeFPArgs |= (!outArgs.get_DfDp(i).isEmpty());
2285 return activeFPArgs;
2288template <
typename Scalar>
2292 typedef Thyra::ModelEvaluatorBase MEB;
2295 bool activeFPArgs =
false;
2296 for(
int i=0;i<Teuchos::as<int>(
parameters_.size());i++) {
2303 if(outArgs.supports(MEB::OUT_ARG_DfDp,i).none())
2307 activeFPArgs |= (!outArgs.get_DfDp(i).isEmpty());
2310 return activeFPArgs;
2313template <
typename Scalar>
2316 const Teuchos::RCP<panzer::WorksetContainer> & wc,
2317 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
2318 const std::vector<panzer::BC> & bcs,
2322 const Teuchos::ParameterList& closure_models,
2323 const Teuchos::ParameterList& user_data,
2324 const bool write_graphviz_file,
2325 const std::string& graphviz_file_prefix)
2329 using Teuchos::null;
2352 RCP<ResponseLibrary<Traits> > rLibrary
2355 rLibrary->buildResidualResponseEvaluators(physicsBlocks,eqset_factory,bcs,bc_factory,
2356 cm_factory,closure_models,user_data,
2357 write_graphviz_file,graphviz_file_prefix);
2364template <
typename Scalar>
2367 const Teuchos::RCP<panzer::WorksetContainer> & wc,
2368 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
2369 const std::vector<panzer::BC>& ,
2373 const Teuchos::ParameterList& closure_models,
2374 const Teuchos::ParameterList& user_data,
2375 const bool write_graphviz_file,
2376 const std::string& graphviz_file_prefix)
2380 using Teuchos::null;
2399 RCP<const LinearObjFactory<Traits> > param_lof =
parameters_[p]->dfdp_rl->getLinearObjFactory();
2400 RCP<const GlobalIndexer > param_ugi =
parameters_[p]->global_indexer;
2403 RCP<ResponseLibrary<Traits> > rLibrary
2408 for(std::size_t r=0;r<
responses_.size();r++) {
2415 responses_[r]->builder->setDerivativeInformation(param_lof);
2423 rLibrary->buildResponseEvaluators(physicsBlocks,eqset_factory,
2424 cm_factory,closure_models,user_data,
2425 write_graphviz_file,graphviz_file_prefix);
2432template <
typename Scalar>
2440template <
typename Scalar>
2441Teuchos::RCP<typename panzer::ModelEvaluator<Scalar>::ParameterObject>
2444 const Teuchos::Array<Scalar> & in_values)
const
2448 using Teuchos::rcp_dynamic_cast;
2449 using Teuchos::ptrFromRef;
2451 TEUCHOS_ASSERT(in_names.size()==in_values.size());
2462 paramObj->names = rcp(
new Teuchos::Array<std::string>(in_names));
2463 paramObj->is_distributed =
false;
2466 for(
int i=0;i<in_names.size();i++)
2474 Thyra::locallyReplicatedDefaultSpmdVectorSpace<Scalar>(
2478 Teuchos::ArrayRCP<Scalar> data;
2479 RCP<Thyra::VectorBase<Scalar> > initial_value = Thyra::createMember(paramObj->space);
2480 RCP<Thyra::SpmdVectorBase<Scalar> > vec = rcp_dynamic_cast<Thyra::SpmdVectorBase<Scalar> >(initial_value);
2481 vec->getNonconstLocalData(ptrFromRef(data));
2482 for (
unsigned int i=0; i < paramObj->scalar_value.size(); i++)
2483 data[i] = in_values[i];
2485 paramObj->initial_value = initial_value;
2490template <
typename Scalar>
2491Teuchos::RCP<typename panzer::ModelEvaluator<Scalar>::ParameterObject>
2496 const Teuchos::RCP<const GlobalIndexer> & ugi)
const
2503 paramObj->is_distributed =
true;
2504 paramObj->names = rcp(
new Teuchos::Array<std::string>());
2505 paramObj->names->push_back(key);
2506 paramObj->space = vs;
2507 paramObj->initial_value = initial;
2509 paramObj->global_indexer = ugi;
2514template <
typename Scalar>
2517setParameters(
const Thyra::ModelEvaluatorBase::InArgs<Scalar> &inArgs)
const
2519 for(std::size_t i=0; i <
parameters_.size(); i++) {
2526 Teuchos::RCP<const Thyra::VectorBase<Scalar> > p = inArgs.get_p(i);
2527 if (p != Teuchos::null) {
2528 for (
unsigned int j=0; j <
parameters_[i]->scalar_value.size(); j++) {
2529 parameters_[i]->scalar_value[j].family->setRealValueForAllTypes(Thyra::get_ele(*p,j));
2536template <
typename Scalar>
2541 for(std::size_t i=0; i <
parameters_.size(); i++) {
2548 for (
unsigned int j=0; j <
parameters_[i]->scalar_value.size(); j++) {
2549 parameters_[i]->scalar_value[j].family->setRealValueForAllTypes(Thyra::get_ele(*(
parameters_[i]->initial_value),j));
std::vector< double > gather_seeds
void addGlobalEvaluationData(const std::string &key, const Teuchos::RCP< GlobalEvaluationData > &ged)
Teuchos::RCP< panzer::LinearObjContainer > ghostedContainer_
bool apply_dirichlet_beta
std::string second_sensitivities_name
bool evaluate_transient_terms
Teuchos::RCP< panzer::LinearObjContainer > container_
std::string first_sensitivities_name
virtual void evalModelImpl_basic_dfdp_scalar_fd(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
void setOneTimeDirichletBeta(const Scalar &beta) const
virtual void evalModelImpl_basic_dgdp_scalar(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
void setupAssemblyInArgs(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, panzer::AssemblyEngineInArgs &ae_inargs) const
std::vector< bool > active_evaluation_types_
void setupModel(const Teuchos::RCP< panzer::WorksetContainer > &wc, const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const std::vector< panzer::BC > &bcs, const panzer::EquationSetFactory &eqset_factory, const panzer::BCStrategyFactory &bc_factory, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &volume_cm_factory, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &bc_cm_factory, const Teuchos::ParameterList &closure_models, const Teuchos::ParameterList &user_data, bool writeGraph=false, const std::string &graphPrefix="", const Teuchos::ParameterList &me_params=Teuchos::ParameterList())
Teuchos::RCP< panzer::LinearObjContainer > ghostedContainer_
unsigned long long write_matrix_count_
Teuchos::RCP< ReadOnlyVector_GlobalEvaluationData > xContainer_
Thyra::ModelEvaluatorBase::OutArgs< Scalar > createOutArgsImpl() const override
bool required_basic_g(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Does this set of out args require a simple response?
void evalModel_D2fDp2(int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::LinearOpBase< Scalar > > &D2fDp2) const
Teuchos::RCP< Thyra::LinearOpBase< Scalar > > create_DfDp_op(int i) const override
virtual void evalModelImpl_basic_dgdx(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > f_space_
void buildDistroParamDfDp_RL(const Teuchos::RCP< panzer::WorksetContainer > &wc, const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const std::vector< panzer::BC > &bcs, const panzer::EquationSetFactory &eqset_factory, const panzer::BCStrategyFactory &bc_factory, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &cm_factory, const Teuchos::ParameterList &closure_models, const Teuchos::ParameterList &user_data, const bool write_graphviz_file=false, const std::string &graphviz_file_prefix="")
Teuchos::RCP< const Thyra::LinearOpWithSolveFactoryBase< Scalar > > get_W_factory() const override
std::vector< Teuchos::RCP< ParameterObject > > parameters_
bool oneTimeDirichletBeta_on_
void resetParameters() const
std::vector< Teuchos::RCP< ResponseObject > > responses_
void setParameters(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs) const
Teuchos::RCP< ParameterObject > createDistributedParameter(const std::string &key, const Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > &vs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &initial, const Teuchos::RCP< const GlobalIndexer > &ugi) const
std::vector< Teuchos::RCP< Thyra::VectorSpaceBase< double > > > tangent_space_
panzer::AssemblyEngine_TemplateManager< panzer::Traits > ae_tm_
virtual void evalModelImpl_basic_dfdp_scalar(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
bool required_basic_dfdp_scalar(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Are derivatives of the residual with respect to the scalar parameters in the out args?...
bool build_bc_field_managers_
bool require_in_args_refresh_
Teuchos::RCP< const Thyra::LinearOpWithSolveFactoryBase< Scalar > > solverFactory_
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > x_space_
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_p_space(int i) const override
virtual void evalModelImpl_basic_g(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Construct a simple response dicatated by this set of out args.
Thyra::ModelEvaluatorBase::InArgs< Scalar > nominalValues_
void evalModel_D2fDx2(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::LinearOpBase< Scalar > > &D2fDx2) const
bool required_basic_dgdx(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Are their required responses in the out args? DgDx.
bool required_basic_dgdp_distro(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Are their required responses in the out args? DgDp.
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_x_space() const override
virtual void evalModelImpl(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const override
Thyra::ModelEvaluatorBase::OutArgs< Scalar > prototypeOutArgs_
void evalModel_D2fDpDx(int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::LinearOpBase< Scalar > > &D2fDpDx) const
GlobalEvaluationDataContainer nonParamGlobalEvaluationData_
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_f_space() const override
void buildVolumeFieldManagers(const bool value)
virtual void evalModelImpl_basic_dgdp_distro(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
void evalModel_D2gDxDp(int rIndex, int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_p, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &D2gDxDp) const
int addDistributedParameter(const std::string &name, const Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > &vs, const Teuchos::RCP< GlobalEvaluationData > &ged, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &initial, const Teuchos::RCP< const GlobalIndexer > &ugi=Teuchos::null)
bool build_transient_support_
bool required_basic_dgdp_scalar(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Are their required responses in the out args? DgDp.
int addParameter(const std::string &name, const Scalar &initial)
Thyra::ModelEvaluatorBase::InArgs< Scalar > getNominalValues() const override
Scalar oneTimeDirichletBeta_
const std::string & get_g_name(int i) const
void buildDistroParamDgDp_RL(const Teuchos::RCP< panzer::WorksetContainer > &wc, const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const std::vector< panzer::BC > &bcs, const panzer::EquationSetFactory &eqset_factory, const panzer::BCStrategyFactory &bc_factory, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &cm_factory, const Teuchos::ParameterList &closure_models, const Teuchos::ParameterList &user_data, const bool write_graphviz_file=false, const std::string &graphviz_file_prefix="")
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int i) const override
Teuchos::ArrayView< const std::string > get_g_names(int i) const override
void applyDirichletBCs(const Teuchos::RCP< Thyra::VectorBase< Scalar > > &x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &f) const
Thyra::ModelEvaluatorBase::InArgs< Scalar > createInArgs() const override
GlobalEvaluationDataContainer distrParamGlobalEvaluationData_
void evalModel_D2gDpDx(int rIndex, int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &D2gDpDx) const
void buildBCFieldManagers(const bool value)
Teuchos::RCP< const panzer::LinearObjFactory< panzer::Traits > > lof_
void buildResponses(const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const panzer::EquationSetFactory &eqset_factory, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &cm_factory, const Teuchos::ParameterList &closure_models, const Teuchos::ParameterList &user_data, const bool write_graphviz_file=false, const std::string &graphviz_file_prefix="")
virtual void evalModelImpl_basic_dfdp_distro(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
void evalModel_D2fDxDp(int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_p, const Teuchos::RCP< Thyra::LinearOpBase< Scalar > > &D2fDxDp) const
void evalModel_D2gDx2(int rIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &D2gDx2) const
virtual void evalModelImpl_basic(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Evaluate a simple model, meaning a residual and a jacobian, no fancy stochastic galerkin or multipoin...
Teuchos::RCP< ReadOnlyVector_GlobalEvaluationData > xdotContainer_
Teuchos::RCP< Thyra::LinearOpBase< Scalar > > create_W_op() const override
Teuchos::RCP< panzer::ResponseLibrary< panzer::Traits > > responseLibrary_
bool require_out_args_refresh_
void evalModel_D2gDp2(int rIndex, int pIndex, const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &delta_x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &D2gDp2) const
bool build_volume_field_managers_
bool required_basic_dfdp_distro(const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs) const
Are derivatives of the residual with respect to the distributed parameters in the out args?...
Teuchos::RCP< panzer::GlobalData > global_data_
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_g_space(int i) const override
void addNonParameterGlobalEvaluationData(const std::string &name, const Teuchos::RCP< GlobalEvaluationData > &ged)
void initializeNominalValues() const
Initialize the nominal values with good starting conditions.
Teuchos::RCP< ParameterObject > createScalarParameter(const Teuchos::Array< std::string > &names, const Teuchos::Array< Scalar > &in_values) const
Teuchos::RCP< ResponseBase > getResponse(const std::string &responseName) const
void evaluate(const panzer::AssemblyEngineInArgs &input_args)
void addResponsesToInArgs(panzer::AssemblyEngineInArgs &input_args) const
Teuchos::RCP< const LinearObjFactory< panzer::Traits > > cloneWithNewDomain(const LinearObjFactory< panzer::Traits > &lof, const Teuchos::RCP< const GlobalIndexer > &dUgi)
Clone a linear object factory, but using a different domain.
Tpetra::KokkosCompat::KokkosDeviceWrapperNode< PHX::Device > TpetraNodeType
Sacado::ScalarParameterVector< panzer::EvaluationTraits > ParamVec
void registerScalarParameter(const std::string name, panzer::ParamLib &pl, double realValue)
Interface for constructing a BCStrategy_TemplateManager.
Allocates and initializes an equation set template manager.
Teuchos::RCP< panzer::ResponseLibrary< panzer::Traits > > dfdp_rl
Teuchos::RCP< const GlobalIndexer > global_indexer