9#ifndef Tempus_ModelEvaluatorPairPartIMEX_Basic_decl_hpp
10#define Tempus_ModelEvaluatorPairPartIMEX_Basic_decl_hpp
12#include "Tempus_config.hpp"
14#include "Thyra_StateFuncModelEvaluatorBase.hpp"
37template <
typename Scalar>
50 int numExplicitOnlyBlocks = 0,
int parameterIndex = -1);
62 virtual Teuchos::RCP<const Thyra::ModelEvaluator<Scalar> >
66 virtual void setInArgs(Thyra::ModelEvaluatorBase::InArgs<Scalar> inArgs)
70 virtual Thyra::ModelEvaluatorBase::InArgs<Scalar>
getInArgs()
74 virtual void setOutArgs(Thyra::ModelEvaluatorBase::OutArgs<Scalar> outArgs)
78 virtual Thyra::ModelEvaluatorBase::OutArgs<Scalar>
getOutArgs()
82 virtual void setForSolve(Teuchos::RCP<TimeDerivative<Scalar> > timeDer,
83 Thyra::ModelEvaluatorBase::InArgs<Scalar> inArgs,
84 Thyra::ModelEvaluatorBase::OutArgs<Scalar> outArgs,
98 virtual Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
102 virtual Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
106 virtual Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
118 virtual Teuchos::RCP<const Thyra::ModelEvaluator<Scalar> >
120 virtual Teuchos::RCP<const Thyra::ModelEvaluator<Scalar> >
126 virtual Teuchos::RCP<Thyra::VectorBase<Scalar> >
getIMEXVector(
130 virtual Teuchos::RCP<const Thyra::VectorBase<Scalar> >
getIMEXVector(
154 virtual Teuchos::RCP<Thyra::LinearOpBase<Scalar> >
create_W_op()
const
157 Teuchos::RCP<const Thyra::LinearOpWithSolveFactoryBase<Scalar> >
160 virtual Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> >
164 virtual Thyra::ModelEvaluatorBase::InArgs<Scalar>
createInArgs()
const;
168 const Thyra::ModelEvaluatorBase::InArgs<Scalar> & in,
169 const Thyra::ModelEvaluatorBase::OutArgs<Scalar> & out)
const;
178 int numExplicitOnlyBlocks = 0,
int parameterIndex = -1);
ModelEvaluator pair for implicit and explicit (IMEX) evaluations.
ModelEvaluator pair for implicit and explicit (IMEX) evaulations.
Teuchos::RCP< TimeDerivative< Scalar > > timeDer_
virtual int getNumExplicitOnlyBlocks() const
virtual Thyra::ModelEvaluatorBase::OutArgs< Scalar > getOutArgs()
Get OutArgs the wrapper ModelEvalutor.
virtual void setNumExplicitOnlyBlocks(int numExp)
int parameterIndex_
implicit parameter index for explicit-only vector
bool useImplicitModel_
if true, use implicitModel_ else explicitModel_
virtual void setInArgs(Thyra::ModelEvaluatorBase::InArgs< Scalar > inArgs)
Set InArgs the wrapper ModelEvalutor.
virtual Teuchos::RCP< Thyra::VectorBase< Scalar > > getIMEXVector(const Teuchos::RCP< Thyra::VectorBase< Scalar > > &full) const
Extract IMEX vector from a full solution vector.
virtual void setParameterIndex(int parameterIndex=-1)
Set the parameter index for explicit-only vector.
int numExplicitOnlyBlocks_
virtual Teuchos::RCP< Thyra::VectorBase< Scalar > > getExplicitOnlyVector(const Teuchos::RCP< Thyra::VectorBase< Scalar > > &full) const
Extract explicit-only vector from a full solution vector.
virtual int getParameterIndex() const
Get the parameter index for explicit-only vector.
void setup(const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &explicitModel, const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &implicitModel, int numExplicitOnlyBlocks=0, int parameterIndex=-1)
Setup ME when using default constructor – for derived classes.
Thyra::ModelEvaluatorBase::OutArgs< Scalar > wrapperImplicitOutArgs_
virtual void setOutArgs(Thyra::ModelEvaluatorBase::OutArgs< Scalar > outArgs)
Set OutArgs the wrapper ModelEvalutor.
virtual void evalModelImpl(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &in, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &out) const
virtual Thyra::ModelEvaluatorBase::InArgs< Scalar > getInArgs()
Get InArgs the wrapper ModelEvalutor.
virtual Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_f_space() const
virtual Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_p_space(int i) const
Get the p space.
virtual void setUseImplicitModel(bool tf)
Set parameter to switch wrapperME base functions between explicit and implicit functions.
virtual Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_g_space(int i) const
Get the g space.
virtual Teuchos::RCP< Thyra::LinearOpBase< Scalar > > create_W_op() const
virtual Thyra::ModelEvaluatorBase::InArgs< Scalar > getNominalValues() const
virtual Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > getExplicitModel() const
virtual Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > getAppModel() const
Get the underlying application ModelEvaluator.
virtual Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_x_space() const
Get the x-solution space.
virtual bool getUseImplicitModel() const
Get parameter to switch wrapperME base functions between explicit and implicit functions.
virtual ~WrapperModelEvaluatorPairPartIMEX_Basic()
Destructor.
Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > implicitModel_
virtual void setForSolve(Teuchos::RCP< TimeDerivative< Scalar > > timeDer, Thyra::ModelEvaluatorBase::InArgs< Scalar > inArgs, Thyra::ModelEvaluatorBase::OutArgs< Scalar > outArgs, EVALUATION_TYPE=SOLVE_FOR_X)
Set parameters for application implicit ModelEvaluator solve.
virtual Thyra::ModelEvaluatorBase::OutArgs< Scalar > createOutArgsImpl() const
virtual void setAppModel(const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &me)
Set the underlying application ModelEvaluator.
virtual Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > getImplicitModel() const
WrapperModelEvaluatorPairPartIMEX_Basic()
Default constructor – Still requires setting the models and running initialize.
Thyra::ModelEvaluatorBase::InArgs< Scalar > wrapperImplicitInArgs_
virtual void setExplicitModel(const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &model)
Teuchos::RCP< const Thyra::LinearOpWithSolveFactoryBase< Scalar > > get_W_factory() const
virtual void setImplicitModel(const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &model)
virtual Thyra::ModelEvaluatorBase::InArgs< Scalar > createInArgs() const
Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > explicitModel_
virtual void initialize()
Initialize after setting member data.
EVALUATION_TYPE
EVALUATION_TYPE indicates the evaluation to apply to the implicit ODE.
@ SOLVE_FOR_X
Solve for x and determine xDot from x.