ModelEvaluatorBase::OutArgs<Scalar> DefaultModelEvaluatorWithSolveFactory<Scalar>::createOutArgsImpl() const { typedef ModelEvaluatorBase MEB; const RCP<const ModelEvaluator<Scalar> > thyraModel = this->getUnderlyingModel(); const MEB::OutArgs<Scalar> wrappedOutArgs = thyraModel->createOutArgs(); MEB::OutArgsSetup<Scalar> outArgs; outArgs.setModelEvalDescription(this->description()); outArgs.set_Np_Ng(wrappedOutArgs.Np(),wrappedOutArgs.Ng()); outArgs.setSupports(wrappedOutArgs); outArgs.setSupports(MEB::OUT_ARG_W, wrappedOutArgs.supports(MEB::OUT_ARG_W_op)&&W_factory_.get()!=NULL); return outArgs; }
ModelEvaluatorBase::OutArgs<Scalar> DefaultStateEliminationModelEvaluator<Scalar>::createOutArgsImpl() const { typedef ModelEvaluatorBase MEB; const Teuchos::RCP<const ModelEvaluator<Scalar> > thyraModel = this->getUnderlyingModel(); const MEB::OutArgs<Scalar> wrappedOutArgs = thyraModel->createOutArgs(); const int Np = wrappedOutArgs.Np(), Ng = wrappedOutArgs.Ng(); MEB::OutArgsSetup<Scalar> outArgs; outArgs.setModelEvalDescription(this->description()); outArgs.set_Np_Ng(Np,Ng); outArgs.setSupports(wrappedOutArgs); outArgs.setUnsupportsAndRelated(MEB::IN_ARG_x); // wipe out DgDx ... outArgs.setUnsupportsAndRelated(MEB::OUT_ARG_f); // wipe out f, W, DfDp ... return outArgs; }
void ModelEvaluatorDefaultBase<Scalar>::initializeDefaultBase() { typedef ModelEvaluatorBase MEB; // In case we throw half way thorugh, set to uninitialized isInitialized_ = false; default_W_support_ = false; // // A) Get the InArgs and OutArgs from the subclass // const MEB::InArgs<Scalar> inArgs = this->createInArgs(); const MEB::OutArgs<Scalar> outArgsImpl = this->createOutArgsImpl(); // // B) Validate the subclasses InArgs and OutArgs // #ifdef TEUCHOS_DEBUG assertInArgsOutArgsSetup( this->description(), inArgs, outArgsImpl ); #endif // TEUCHOS_DEBUG // // C) Set up support for default derivative objects and prototype OutArgs // const int l_Ng = outArgsImpl.Ng(); const int l_Np = outArgsImpl.Np(); // Set support for all outputs supported in the underly implementation MEB::OutArgsSetup<Scalar> outArgs; outArgs.setModelEvalDescription(this->description()); outArgs.set_Np_Ng(l_Np,l_Ng); outArgs.setSupports(outArgsImpl); // DfDp DfDp_default_op_support_.clear(); if (outArgs.supports(MEB::OUT_ARG_f)) { for ( int l = 0; l < l_Np; ++l ) { const MEB::DerivativeSupport DfDp_l_impl_support = outArgsImpl.supports(MEB::OUT_ARG_DfDp,l); const DefaultDerivLinearOpSupport DfDp_l_op_support = determineDefaultDerivLinearOpSupport(DfDp_l_impl_support); DfDp_default_op_support_.push_back(DfDp_l_op_support); outArgs.setSupports( MEB::OUT_ARG_DfDp, l, updateDefaultLinearOpSupport( DfDp_l_impl_support, DfDp_l_op_support ) ); } } // DgDx_dot DgDx_dot_default_op_support_.clear(); for ( int j = 0; j < l_Ng; ++j ) { const MEB::DerivativeSupport DgDx_dot_j_impl_support = outArgsImpl.supports(MEB::OUT_ARG_DgDx_dot,j); const DefaultDerivLinearOpSupport DgDx_dot_j_op_support = determineDefaultDerivLinearOpSupport(DgDx_dot_j_impl_support); DgDx_dot_default_op_support_.push_back(DgDx_dot_j_op_support); outArgs.setSupports( MEB::OUT_ARG_DgDx_dot, j, updateDefaultLinearOpSupport( DgDx_dot_j_impl_support, DgDx_dot_j_op_support ) ); } // DgDx DgDx_default_op_support_.clear(); for ( int j = 0; j < l_Ng; ++j ) { const MEB::DerivativeSupport DgDx_j_impl_support = outArgsImpl.supports(MEB::OUT_ARG_DgDx,j); const DefaultDerivLinearOpSupport DgDx_j_op_support = determineDefaultDerivLinearOpSupport(DgDx_j_impl_support); DgDx_default_op_support_.push_back(DgDx_j_op_support); outArgs.setSupports( MEB::OUT_ARG_DgDx, j, updateDefaultLinearOpSupport( DgDx_j_impl_support, DgDx_j_op_support ) ); } // DgDp DgDp_default_op_support_.clear(); DgDp_default_mv_support_.clear(); for ( int j = 0; j < l_Ng; ++j ) { DgDp_default_op_support_.push_back(Array<DefaultDerivLinearOpSupport>()); DgDp_default_mv_support_.push_back(Array<DefaultDerivMvAdjointSupport>()); for ( int l = 0; l < l_Np; ++l ) { const MEB::DerivativeSupport DgDp_j_l_impl_support = outArgsImpl.supports(MEB::OUT_ARG_DgDp,j,l); // LinearOpBase support const DefaultDerivLinearOpSupport DgDp_j_l_op_support = determineDefaultDerivLinearOpSupport(DgDp_j_l_impl_support); DgDp_default_op_support_[j].push_back(DgDp_j_l_op_support); outArgs.setSupports( MEB::OUT_ARG_DgDp, j, l, updateDefaultLinearOpSupport( DgDp_j_l_impl_support, DgDp_j_l_op_support ) ); // MultiVectorBase const DefaultDerivMvAdjointSupport DgDp_j_l_mv_support = determineDefaultDerivMvAdjointSupport( DgDp_j_l_impl_support, *this->get_g_space(j), *this->get_p_space(l) ); DgDp_default_mv_support_[j].push_back(DgDp_j_l_mv_support); outArgs.setSupports( MEB::OUT_ARG_DgDp, j, l, updateDefaultDerivMvAdjointSupport( outArgs.supports(MEB::OUT_ARG_DgDp, j, l), DgDp_j_l_mv_support ) ); } } // 2007/09/09: rabart: ToDo: Move the above code into a private helper // function! // W (given W_op and W_factory) default_W_support_ = false; if ( outArgsImpl.supports(MEB::OUT_ARG_W_op) && !is_null(this->get_W_factory()) && !outArgsImpl.supports(MEB::OUT_ARG_W) ) { default_W_support_ = true; outArgs.setSupports(MEB::OUT_ARG_W); outArgs.set_W_properties(outArgsImpl.get_W_properties()); } // // D) All done! // prototypeOutArgs_ = outArgs; isInitialized_ = true; }