Tempus
Version of the Day
Time Integration
|
Partitioned Implicit-Explicit Runge-Kutta (IMEX-RK) time stepper. More...
#include <Tempus_StepperIMEX_RK_Partition_decl.hpp>
Public Member Functions | |
StepperIMEX_RK_Partition () | |
Default constructor. More... | |
StepperIMEX_RK_Partition (const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &appModel, const Teuchos::RCP< StepperObserver< Scalar > > &obs, const Teuchos::RCP< Thyra::NonlinearSolverBase< Scalar > > &solver, bool useFSAL, std::string ICConsistency, bool ICConsistencyCheck, bool zeroInitialGuess, std::string stepperType, Teuchos::RCP< const RKButcherTableau< Scalar > > explicitTableau, Teuchos::RCP< const RKButcherTableau< Scalar > > implicitTableau, Scalar order) | |
Constructor to specialize Stepper parameters. More... | |
virtual Scalar | getAlpha (const Scalar dt) const |
Return alpha = d(xDot)/dx. More... | |
virtual Scalar | getBeta (const Scalar) const |
Return beta = d(x)/dx. More... | |
Teuchos::RCP< const Teuchos::ParameterList > | getValidParameters () const |
void | evalImplicitModelExplicitly (const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &X, const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &Y, Scalar time, Scalar stepSize, Scalar stageNumber, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &G) const |
void | evalExplicitModel (const Teuchos::RCP< const Thyra::VectorBase< Scalar > > &X, Scalar time, Scalar stepSize, Scalar stageNumber, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &F) const |
void | setOrder (Scalar order) |
Basic stepper methods | |
virtual void | setTableaus (std::string stepperType="", Teuchos::RCP< const RKButcherTableau< Scalar > > explicitTableau=Teuchos::null, Teuchos::RCP< const RKButcherTableau< Scalar > > implicitTableau=Teuchos::null) |
Set both the explicit and implicit tableau from ParameterList. More... | |
virtual void | setExplicitTableau (Teuchos::RCP< const RKButcherTableau< Scalar > > explicitTableau) |
Set the explicit tableau from tableau. More... | |
virtual void | setImplicitTableau (Teuchos::RCP< const RKButcherTableau< Scalar > > implicitTableau) |
Set the implicit tableau from tableau. More... | |
virtual void | setModel (const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &appModel) |
virtual Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > | getModel () |
virtual void | setModelPair (const Teuchos::RCP< WrapperModelEvaluatorPairPartIMEX_Basic< Scalar > > &modelPair) |
Create WrapperModelPairIMEX from user-supplied ModelEvaluator pair. More... | |
virtual void | setModelPair (const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &explicitModel, const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &implicitModel) |
Create WrapperModelPairIMEX from explicit/implicit ModelEvaluators. More... | |
virtual void | setObserver (Teuchos::RCP< StepperObserver< Scalar > > obs=Teuchos::null) |
Set Observer. More... | |
virtual Teuchos::RCP < StepperObserver< Scalar > > | getObserver () const |
Get Observer. More... | |
virtual void | initialize () |
Initialize during construction and after changing input parameters. More... | |
virtual void | setInitialConditions (const Teuchos::RCP< SolutionHistory< Scalar > > &solutionHistory) |
Set the initial conditions and make them consistent. More... | |
virtual void | takeStep (const Teuchos::RCP< SolutionHistory< Scalar > > &solutionHistory) |
Take the specified timestep, dt, and return true if successful. More... | |
virtual Teuchos::RCP < Tempus::StepperState< Scalar > > | getDefaultStepperState () |
Provide a StepperState to the SolutionState. This Stepper does not have any special state data, so just provide the base class StepperState with the Stepper description. This can be checked to ensure that the input StepperState can be used by this Stepper. More... | |
virtual Scalar | getOrder () const |
virtual Scalar | getOrderMin () const |
virtual Scalar | getOrderMax () const |
virtual bool | isExplicit () const |
virtual bool | isImplicit () const |
virtual bool | isExplicitImplicit () const |
virtual bool | isOneStepMethod () const |
virtual bool | isMultiStepMethod () const |
virtual OrderODE | getOrderODE () const |
Overridden from Teuchos::Describable | |
virtual void | describe (Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel) const |
Public Member Functions inherited from Tempus::StepperImplicit< Scalar > | |
virtual void | setNonConstModel (const Teuchos::RCP< Thyra::ModelEvaluator< Scalar > > &appModel) |
virtual Teuchos::RCP< const WrapperModelEvaluator< Scalar > > | getWrapperModel () |
virtual void | setSolver (Teuchos::RCP< Thyra::NonlinearSolverBase< Scalar > > solver=Teuchos::null) |
Set solver. More... | |
virtual Teuchos::RCP < Thyra::NonlinearSolverBase < Scalar > > | getSolver () const |
Get solver. More... | |
const Thyra::SolveStatus< Scalar > | solveImplicitODE (const Teuchos::RCP< Thyra::VectorBase< Scalar > > &x) |
Solve problem using x in-place. (Needs to be deprecated!) More... | |
const Thyra::SolveStatus< Scalar > | solveImplicitODE (const Teuchos::RCP< Thyra::VectorBase< Scalar > > &x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &xDot, const Scalar time, const Teuchos::RCP< ImplicitODEParameters< Scalar > > &p) |
Solve implicit ODE, f(x, xDot, t, p) = 0. More... | |
void | evaluateImplicitODE (Teuchos::RCP< Thyra::VectorBase< Scalar > > &f, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &x, const Teuchos::RCP< Thyra::VectorBase< Scalar > > &xDot, const Scalar time, const Teuchos::RCP< ImplicitODEParameters< Scalar > > &p) |
Evaluate implicit ODE residual, f(x, xDot, t, p). More... | |
virtual void | setInitialGuess (Teuchos::RCP< const Thyra::VectorBase< Scalar > > initial_guess) |
Pass initial guess to Newton solver (only relevant for implicit solvers) More... | |
virtual void | setZeroInitialGuess (bool zIG) |
Set parameter so that the initial guess is set to zero (=True) or use last timestep (=False). More... | |
virtual bool | getZeroInitialGuess () const |
virtual Scalar | getInitTimeStep (const Teuchos::RCP< SolutionHistory< Scalar > > &) const |
virtual void | setStepperXDot (Teuchos::RCP< Thyra::VectorBase< Scalar > > xDot) |
Set xDot for Stepper storage. More... | |
virtual Teuchos::RCP < Thyra::VectorBase< Scalar > > | getStepperXDot (Teuchos::RCP< SolutionState< Scalar > > state) |
Get xDot from SolutionState or Stepper storage. More... | |
virtual Teuchos::RCP < Thyra::VectorBase< Scalar > > | getStepperXDotDot (Teuchos::RCP< SolutionState< Scalar > > state) |
Get xDotDot from SolutionState or Stepper storage. More... | |
Public Member Functions inherited from Tempus::Stepper< Scalar > | |
void | setStepperType (std::string s) |
std::string | getStepperType () const |
void | setUseFSAL (bool a) |
bool | getUseFSAL () const |
virtual bool | getUseFSALDefault () const |
void | setICConsistency (std::string s) |
std::string | getICConsistency () const |
virtual std::string | getICConsistencyDefault () const |
void | setICConsistencyCheck (bool c) |
bool | getICConsistencyCheck () const |
virtual bool | getICConsistencyCheckDefault () const |
virtual std::string | description () const |
virtual void | createSubSteppers (std::vector< Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > >) |
Protected Attributes | |
Teuchos::RCP< const RKButcherTableau< Scalar > > | explicitTableau_ |
Teuchos::RCP< const RKButcherTableau< Scalar > > | implicitTableau_ |
Scalar | order_ |
Teuchos::RCP < Thyra::VectorBase< Scalar > > | stageZ_ |
std::vector< Teuchos::RCP < Thyra::VectorBase< Scalar > > > | stageF_ |
std::vector< Teuchos::RCP < Thyra::VectorBase< Scalar > > > | stageGx_ |
Teuchos::RCP < Thyra::VectorBase< Scalar > > | xTilde_ |
Teuchos::RCP < StepperRKObserverComposite < Scalar > > | stepperObserver_ |
Protected Attributes inherited from Tempus::StepperImplicit< Scalar > | |
Teuchos::RCP < WrapperModelEvaluator < Scalar > > | wrapperModel_ |
Teuchos::RCP < Thyra::NonlinearSolverBase < Scalar > > | solver_ |
Teuchos::RCP< const Thyra::VectorBase< Scalar > > | initial_guess_ |
bool | zeroInitialGuess_ |
Teuchos::RCP< StepperObserver < Scalar > > | stepperObserver_ |
Teuchos::RCP < Thyra::VectorBase< Scalar > > | stepperXDot_ |
Teuchos::RCP < Thyra::VectorBase< Scalar > > | stepperXDotDot_ |
Partitioned Implicit-Explicit Runge-Kutta (IMEX-RK) time stepper.
Partitioned IMEX-RK is similar to the IMEX-RK (StepperIMEX_RK), except a portion of the solution only requires explicit integration, and should not be part of the implicit solution to reduce computational costs. Again our ODE can be written as
but now
where is the product vector of and , is still the "slow" physics (and evolved explicitly), and is still the "fast" physics (and evolved implicitly), but a portion of the solution vector, , is "explicit-only" and is only evolved by , while is the Implicit/Explicit (IMEX) solution vector, and is evolved explicitly by evolved implicitly by . Note we can expand this to explicitly show all the terms as
or
where , , and , or
where , and . Using Butcher tableaus for the explicit terms
the basic scheme for this partitioned, -stage, IMEX-RK is
or expanded
where and .
For iterative solvers, it is useful to write the stage solutions as
or expanded as
where
and note that . Rearranging to solve for the implicit term
We additionally need the time derivative at each stage for the implicit solve. Let us define the following time derivative for portion of the solution
where we split arguments into and to emphasize that is the solution for the implicit solve and are parameters in this set of equations. The above time derivative, , is NOT likely the same as the real time derivative, , unless (Reasoning: and unless ). Also note that the explicit term, , is evaluated at the implicit stage time, .
We can form the time derivative
Returning to the governing equation for the IMEX solution vector,
Recall and if we define a pseudo time derivative, which is equivalent to the time derivative for the implicit solve,
we can write
For general DIRK methods, we need to also handle the case when . The IMEX stage values can be simply evaluated similiar to the "explicit-only" stage values, e.g.,
and then we can simply evaluate
We can then form the time derivative as
but again note that the explicit term, , is evaluated at the implicit stage time, .
Partitioned IMEX-RK Algorithm The single-timestep algorithm for the partitioned IMEX-RK is
The First-Step-As-Last (FSAL) principle is not valid for IMEX RK Partition. The default is to set useFSAL=false, and useFSAL=true will result in an error.
Definition at line 227 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
Tempus::StepperIMEX_RK_Partition< Scalar >::StepperIMEX_RK_Partition | ( | ) |
Default constructor.
Requires subsequent setModel(), setSolver() and initialize() calls before calling takeStep().
Definition at line 27 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
Tempus::StepperIMEX_RK_Partition< Scalar >::StepperIMEX_RK_Partition | ( | const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > & | appModel, |
const Teuchos::RCP< StepperObserver< Scalar > > & | obs, | ||
const Teuchos::RCP< Thyra::NonlinearSolverBase< Scalar > > & | solver, | ||
bool | useFSAL, | ||
std::string | ICConsistency, | ||
bool | ICConsistencyCheck, | ||
bool | zeroInitialGuess, | ||
std::string | stepperType, | ||
Teuchos::RCP< const RKButcherTableau< Scalar > > | explicitTableau, | ||
Teuchos::RCP< const RKButcherTableau< Scalar > > | implicitTableau, | ||
Scalar | order | ||
) |
Constructor to specialize Stepper parameters.
Definition at line 41 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Definition at line 680 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
void Tempus::StepperIMEX_RK_Partition< Scalar >::evalExplicitModel | ( | const Teuchos::RCP< const Thyra::VectorBase< Scalar > > & | X, |
Scalar | time, | ||
Scalar | stepSize, | ||
Scalar | stageNumber, | ||
const Teuchos::RCP< Thyra::VectorBase< Scalar > > & | F | ||
) | const |
Definition at line 476 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
void Tempus::StepperIMEX_RK_Partition< Scalar >::evalImplicitModelExplicitly | ( | const Teuchos::RCP< const Thyra::VectorBase< Scalar > > & | X, |
const Teuchos::RCP< const Thyra::VectorBase< Scalar > > & | Y, | ||
Scalar | time, | ||
Scalar | stepSize, | ||
Scalar | stageNumber, | ||
const Teuchos::RCP< Thyra::VectorBase< Scalar > > & | G | ||
) | const |
Definition at line 441 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
inlinevirtual |
Return alpha = d(xDot)/dx.
Implements Tempus::StepperImplicit< Scalar >.
Definition at line 314 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Return beta = d(x)/dx.
Implements Tempus::StepperImplicit< Scalar >.
Definition at line 320 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
virtual |
Provide a StepperState to the SolutionState. This Stepper does not have any special state data, so just provide the base class StepperState with the Stepper description. This can be checked to ensure that the input StepperState can be used by this Stepper.
Implements Tempus::Stepper< Scalar >.
Definition at line 671 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
inlinevirtual |
Reimplemented from Tempus::StepperImplicit< Scalar >.
Definition at line 270 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Get Observer.
Implements Tempus::Stepper< Scalar >.
Definition at line 284 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 299 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 301 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 300 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 310 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
virtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 692 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Initialize during construction and after changing input parameters.
Implements Tempus::Stepper< Scalar >.
Definition at line 356 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 303 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 305 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 304 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 308 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
inlinevirtual |
Implements Tempus::Stepper< Scalar >.
Definition at line 307 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
virtual |
Set the explicit tableau from tableau.
Definition at line 244 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Set the implicit tableau from tableau.
Definition at line 256 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Set the initial conditions and make them consistent.
Reimplemented from Tempus::StepperImplicit< Scalar >.
Definition at line 393 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Reimplemented from Tempus::StepperImplicit< Scalar >.
Definition at line 267 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Create WrapperModelPairIMEX from user-supplied ModelEvaluator pair.
The user-supplied ME pair can contain any user-specific IMEX interactions between explicit and implicit MEs.
Definition at line 293 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Create WrapperModelPairIMEX from explicit/implicit ModelEvaluators.
Use the supplied explicit/implicit MEs to create a WrapperModelPairIMEX with basic IMEX interactions between explicit and implicit MEs.
Definition at line 315 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Set Observer.
Implements Tempus::Stepper< Scalar >.
Definition at line 328 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
inline |
Definition at line 341 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
virtual |
Set both the explicit and implicit tableau from ParameterList.
Definition at line 74 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
virtual |
Take the specified timestep, dt, and return true if successful.
Implements Tempus::Stepper< Scalar >.
Definition at line 512 of file Tempus_StepperIMEX_RK_Partition_impl.hpp.
|
protected |
Definition at line 345 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 346 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 348 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 351 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 352 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 350 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 356 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.
|
protected |
Definition at line 354 of file Tempus_StepperIMEX_RK_Partition_decl.hpp.