9 #ifndef TEMPUS_TEST_VANDERPOL_IMEXPart_IMPLICITMODEL_IMPL_HPP
10 #define TEMPUS_TEST_VANDERPOL_IMEXPart_IMPLICITMODEL_IMPL_HPP
12 #include "Teuchos_StandardParameterEntryValidators.hpp"
14 #include "Thyra_DefaultSpmdVectorSpace.hpp"
15 #include "Thyra_DetachedVectorView.hpp"
16 #include "Thyra_DetachedMultiVectorView.hpp"
17 #include "Thyra_DefaultSerialDenseLinearOpWithSolveFactory.hpp"
18 #include "Thyra_DefaultMultiVectorLinearOpWithSolve.hpp"
19 #include "Thyra_DefaultLinearOpSource.hpp"
20 #include "Thyra_VectorStdOps.hpp"
21 #include "Thyra_MultiVectorStdOps.hpp"
22 #include "Thyra_DefaultMultiVectorProductVector.hpp"
26 namespace Tempus_Test {
28 template <
class Scalar>
32 isInitialized_ =
false;
40 useDfDpAsTangent_ =
false;
48 x_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(dim_);
49 f_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(dim_);
51 y_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(dim_);
52 p_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(np_);
53 dxdp_space_ = Thyra::multiVectorProductVectorSpace(x_space_, np_);
54 dydp_space_ = Thyra::multiVectorProductVectorSpace(y_space_, np_);
58 setParameterList(pList_);
61 template <
class Scalar>
68 template <
class Scalar>
75 template <
class Scalar>
80 "Error, setupInOutArgs_ must be called first!\n");
81 return nominalValues_;
84 template <
class Scalar>
90 this->get_W_factory();
107 V_V(multivec->col(0).
ptr(), *vec);
111 Thyra::linearOpWithSolve<Scalar>(*W_factory, matrix);
116 template <
class Scalar>
121 Thyra::createMembers(x_space_, dim_);
125 template <
class Scalar>
130 Thyra::defaultSerialDenseLinearOpWithSolveFactory<Scalar>();
134 template <
class Scalar>
144 template <
class Scalar>
152 template <
class Scalar>
159 using Teuchos::rcp_dynamic_cast;
161 "Error, setupInOutArgs_ must be called first!\n");
164 inArgs.
get_x().assert_not_null();
166 Scalar x1 = x_in_view[0];
170 Scalar
eps = epsilon_;
173 if (p_in != Teuchos::null) {
179 if (inArgs.
get_p(1) != Teuchos::null) {
181 rcp_dynamic_cast<
const DMVPV>(inArgs.
get_p(1),
true)->getMultiVector();
183 if (inArgs.
get_p(2) != Teuchos::null) {
185 rcp_dynamic_cast<
const DMVPV>(inArgs.
get_p(2),
true)->getMultiVector();
187 if (inArgs.
get_p(3) != Teuchos::null) {
189 rcp_dynamic_cast<
const DMVPV>(inArgs.
get_p(3),
true)->getMultiVector();
193 inArgs.
get_p(4).assert_not_null();
195 Scalar x0 = y_in_view[0];
200 outArgs.
get_DfDp(0).getMultiVector();
202 outArgs.
get_DfDp(4).getMultiVector();
208 f_out_view[0] = (1.0 - x0 * x0) * x1 / eps;
212 DfDp_out_view(0, 0) = -(1.0 - x0 * x0) * x1 / (eps * eps);
215 if (useDfDpAsTangent_ && !
is_null(dxdp_in)) {
217 DfDp_out_view(0, 0) += (1.0 - x0 * x0) / eps * dxdp(0, 0);
219 if (useDfDpAsTangent_ && !
is_null(dydp_in)) {
221 DfDp_out_view(0, 0) += -2.0 * x0 * x1 / eps * dydp(0, 0);
226 DfDx0_out_view(0, 0) = -2.0 * x0 * x1 /
eps;
233 W_view(0, 0) = beta * (1.0 - x0 * x0) / eps;
240 x_dot_in = inArgs.
get_x_dot().assert_not_null();
244 f_out_view[0] = x_dot_in_view[0] - (1.0 - x0 * x0) * x1 / eps;
248 DfDp_out_view(0, 0) = (1.0 - x0 * x0) * x1 / (eps * eps);
252 if (useDfDpAsTangent_ && !
is_null(dxdotdp_in)) {
254 DfDp_out_view(0, 0) += dxdotdp(0, 0);
256 if (useDfDpAsTangent_ && !
is_null(dxdp_in)) {
258 DfDp_out_view(0, 0) += -(1.0 - x0 * x0) / eps * dxdp(0, 0);
260 if (useDfDpAsTangent_ && !
is_null(dydp_in)) {
262 DfDp_out_view(0, 0) += 2.0 * x0 * x1 / eps * dydp(0, 0);
267 DfDx0_out_view(0, 0) = 2.0 * x0 * x1 /
eps;
275 W_view(0, 0) = alpha - beta * (1.0 - x0 * x0) / eps;
281 template <
class Scalar>
288 else if (l == 1 || l == 2)
294 return Teuchos::null;
297 template <
class Scalar>
305 p_strings->push_back(
"Model Coefficient: epsilon");
307 p_strings->push_back(
"DxDp");
309 p_strings->push_back(
"Dx_dotDp");
311 p_strings->push_back(
"DyDp");
313 p_strings->push_back(
"EXPLICIT_ONLY_VECTOR");
317 template <
class Scalar>
325 template <
class Scalar>
328 if (isInitialized_)
return;
334 inArgs.
setSupports(Thyra::ModelEvaluatorBase::IN_ARG_x_dot);
335 inArgs.
setSupports(Thyra::ModelEvaluatorBase::IN_ARG_x);
336 inArgs.
setSupports(Thyra::ModelEvaluatorBase::IN_ARG_t);
337 inArgs.
setSupports(Thyra::ModelEvaluatorBase::IN_ARG_alpha);
338 inArgs.
setSupports(Thyra::ModelEvaluatorBase::IN_ARG_beta);
347 outArgs.
setSupports(Thyra::ModelEvaluatorBase::OUT_ARG_f);
348 outArgs.
setSupports(Thyra::ModelEvaluatorBase::OUT_ARG_W_op);
350 outArgs.
setSupports(Thyra::ModelEvaluatorBase::OUT_ARG_DfDp, 0,
351 Thyra::ModelEvaluatorBase::DERIV_MV_JACOBIAN_FORM);
352 outArgs.
setSupports(Thyra::ModelEvaluatorBase::OUT_ARG_DfDp, 4,
353 Thyra::ModelEvaluatorBase::DERIV_MV_JACOBIAN_FORM);
358 nominalValues_ = inArgs_;
361 nominalValues_.set_t(t0_ic_);
365 x_ic_view[0] = x1_ic_;
367 nominalValues_.set_x(x_ic);
374 p_ic_view[0] = epsilon_;
375 y_ic_view[0] = x0_ic_;
377 nominalValues_.set_p(0, p_ic);
378 nominalValues_.set_p(4, y_ic);
383 x_dot_ic_view[0] = ((1.0 - x0_ic_ * x0_ic_) * x1_ic_ - x0_ic_) / epsilon_;
385 nominalValues_.set_x_dot(x_dot_ic);
388 isInitialized_ =
true;
391 template <
class Scalar>
399 if (paramList != Teuchos::null) tmpPL = paramList;
401 this->setMyParamList(tmpPL);
403 bool haveIC = get<bool>(*pl,
"Provide nominal values");
404 bool useDfDpAsTangent = get<bool>(*pl,
"Use DfDp as Tangent");
405 if (haveIC != haveIC_) isInitialized_ =
false;
407 useDfDpAsTangent_ = useDfDpAsTangent;
408 epsilon_ = get<Scalar>(*pl,
"Coeff epsilon");
409 x0_ic_ = get<Scalar>(*pl,
"IC x0");
410 x1_ic_ = get<Scalar>(*pl,
"IC x1");
411 t0_ic_ = get<Scalar>(*pl,
"IC t0");
415 template <
class Scalar>
422 pl->
set(
"Accept model parameters",
false);
423 pl->
set(
"Provide nominal values",
true);
424 pl->
set(
"Use DfDp as Tangent",
false);
425 Teuchos::setDoubleParameter(
"Coeff epsilon", 1.0e-06,
426 "Coefficient a in model", &*pl);
427 Teuchos::setDoubleParameter(
"IC x0", 2.0,
"Initial Condition for x0", &*pl);
428 Teuchos::setDoubleParameter(
"IC x1", 0.0,
"Initial Condition for x1", &*pl);
429 Teuchos::setDoubleParameter(
"IC t0", 0.0,
"Initial time t0", &*pl);
436 #endif // TEMPUS_TEST_VANDERPOL_IMEXPart_IMPLICITMODEL_IMPL_HPP
Thyra::ModelEvaluatorBase::InArgs< Scalar > getNominalValues() const
bool is_null(const boost::shared_ptr< T > &p)
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
Derivative< Scalar > get_DfDp(int l) const
RCP< const VectorBase< Scalar > > get_x_dot() const
void setModelEvalDescription(const std::string &modelEvalDescription)
Teuchos::RCP< Thyra::LinearOpWithSolveBase< Scalar > > create_W() const
ParameterList & set(std::string const &name, T const &value, std::string const &docString="", RCP< const ParameterEntryValidator > const &validator=null)
void set_Np_Ng(int Np, int Ng)
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
VanDerPol_IMEXPart_ImplicitModel(Teuchos::RCP< Teuchos::ParameterList > pList=Teuchos::null)
Evaluation< VectorBase< Scalar > > get_f() const
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_g_space(int j) const
#define TEUCHOS_ASSERT_IN_RANGE_UPPER_EXCLUSIVE(index, lower_inclusive, upper_exclusive)
void setSupports(EInArgsMembers arg, bool supports=true)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
Thyra::ModelEvaluatorBase::OutArgs< Scalar > createOutArgsImpl() const
void setupInOutArgs_() const
Teuchos::RCP< Thyra::LinearOpBase< Scalar > > create_W_op() const
void validateParametersAndSetDefaults(ParameterList const &validParamList, int const depth=1000)
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_p_space(int l) const
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
void setSupports(EOutArgsMembers arg, bool supports=true)
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_x_space() const
void setParameterList(Teuchos::RCP< Teuchos::ParameterList > const ¶mList)
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_f_space() const
void setModelEvalDescription(const std::string &modelEvalDescription)
Teuchos::RCP< const Thyra::LinearOpWithSolveFactoryBase< Scalar > > get_W_factory() const
RCP< LinearOpBase< Scalar > > get_W_op() const
RCP< const VectorBase< Scalar > > get_x() const
void evalModelImpl(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs_bar, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs_bar) const
Thyra::ModelEvaluatorBase::InArgs< Scalar > createInArgs() const
RCP< const VectorBase< Scalar > > get_p(int l) const