9 #ifndef TEMPUS_TEST_SINCOS_MODEL_IMPL_HPP
10 #define TEMPUS_TEST_SINCOS_MODEL_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"
27 namespace Tempus_Test {
29 template<
class Scalar>
33 isInitialized_ =
false;
39 acceptModelParams_ =
false;
40 useDfDpAsTangent_ =
false;
50 x_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(dim_);
51 f_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(dim_);
53 p_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(np_);
54 g_space_ = Thyra::defaultSpmdVectorSpace<Scalar>(ng_);
56 setParameterList(pList_);
59 DxDp_space_ = Thyra::multiVectorProductVectorSpace(x_space_, np_);
62 template<
class Scalar>
68 "Error, setupInOutArgs_ must be called first!\n");
71 inArgs.
set_t(exact_t);
75 exact_x_view[0] = a_+b_*sin((f_/L_)*t+phi_);
76 exact_x_view[1] = b_*(f_/L_)*cos((f_/L_)*t+phi_);
78 inArgs.
set_x(exact_x);
82 exact_x_dot_view[0] = b_*(f_/L_)*cos((f_/L_)*t+phi_);
83 exact_x_dot_view[1] = -b_*(f_/L_)*(f_/L_)*sin((f_/L_)*t+phi_);
95 template<
class Scalar>
101 "Error, setupInOutArgs_ must be called first!\n");
103 if (!acceptModelParams_) {
112 inArgs.
set_t(exact_t);
117 exact_s_view[0] = 1.0;
118 exact_s_view[1] = 0.0;
120 exact_s_view[0] = (b/L)*t*cos((f/L)*t+phi);
121 exact_s_view[1] = (b/L)*cos((f/L)*t+phi)-(b*f*t/(L*L))*sin((f/L)*t+phi);
123 exact_s_view[0] = -(b*f*t/(L*L))*cos((f/L)*t+phi);
124 exact_s_view[1] = -(b*f/(L*L))*cos((f/L)*t+phi)+(b*f*f*t/(L*L*L))*sin((f/L)*t+phi);
127 inArgs.
set_x(exact_s);
132 exact_s_dot_view[0] = 0.0;
133 exact_s_dot_view[1] = 0.0;
135 exact_s_dot_view[0] = (b/L)*cos((f/L)*t+phi)-(b*f*t/(L*L))*sin((f/L)*t+phi);
136 exact_s_dot_view[1] = -(2.0*b*f/(L*L))*sin((f/L)*t+phi)-(b*f*f*t/(L*L*L))*cos((f/L)*t+phi);
138 exact_s_dot_view[0] = -(b*f/(L*L))*cos((f/L)*t+phi)+(b*f*f*t/(L*L*L))*sin((f/L)*t+phi);
139 exact_s_dot_view[1] = (2.0*b*f*f/(L*L*L))*sin((f/L)*t+phi)+(b*f*f*f*t/(L*L*L*L))*cos((f/L)*t+phi);
146 template<
class Scalar>
155 template<
class Scalar>
164 template<
class Scalar>
170 "Error, setupInOutArgs_ must be called first!\n");
171 return nominalValues_;
175 template<
class Scalar>
197 V_V(multivec->col(0).
ptr(),*vec);
203 V_V(multivec->col(1).
ptr(),*vec);
207 Thyra::linearOpWithSolve<Scalar>(
221 template<
class Scalar>
231 template<
class Scalar>
237 Thyra::defaultSerialDenseLinearOpWithSolveFactory<Scalar>();
253 template<
class Scalar>
266 template<
class Scalar>
276 template<
class Scalar>
288 using Teuchos::rcp_dynamic_cast;
290 "Error, setupInOutArgs_ must be called first!\n");
299 if (acceptModelParams_) {
301 inArgs.
get_p(0).assert_not_null();
309 if (acceptModelParams_) {
310 if (inArgs.
get_p(1) != Teuchos::null)
312 rcp_dynamic_cast<const DMVPV>(inArgs.
get_p(1))->getMultiVector();
313 if (inArgs.
get_p(2) != Teuchos::null)
315 rcp_dynamic_cast<const DMVPV>(inArgs.
get_p(2))->getMultiVector();
323 if (acceptModelParams_) {
333 f_out_view[0] = x_in_view[1];
334 f_out_view[1] = (f/L)*(f/L)*(a-x_in_view[0]);
340 matrix_view(0,0) = 0.0;
341 matrix_view(0,1) = +beta;
342 matrix_view(1,0) = -beta*(f/L)*(f/L);
343 matrix_view(1,1) = 0.0;
348 DfDp_out_view(0,0) = 0.0;
349 DfDp_out_view(0,1) = 0.0;
350 DfDp_out_view(0,2) = 0.0;
351 DfDp_out_view(1,0) = (f/L)*(f/L);
352 DfDp_out_view(1,1) = (2.0*f/(L*L))*(a-x_in_view[0]);
353 DfDp_out_view(1,2) = -(2.0*f*f/(L*L*L))*(a-x_in_view[0]);
356 if (useDfDpAsTangent_ && !
is_null(DxDp_in)) {
358 DfDp_out_view(0,0) += DxDp(1,0);
359 DfDp_out_view(0,1) += DxDp(1,1);
360 DfDp_out_view(0,2) += DxDp(1,2);
361 DfDp_out_view(1,0) += -(f/L)*(f/L) * DxDp(0,0);
362 DfDp_out_view(1,1) += -(f/L)*(f/L) * DxDp(0,1);
363 DfDp_out_view(1,2) += -(f/L)*(f/L) * DxDp(0,2);
370 x_dot_in = inArgs.
get_x_dot().assert_not_null();
375 f_out_view[0] = x_dot_in_view[0] - x_in_view[1];
376 f_out_view[1] = x_dot_in_view[1] - (f/L)*(f/L)*(a-x_in_view[0]);
382 matrix_view(0,0) = alpha;
383 matrix_view(0,1) = -beta;
384 matrix_view(1,0) = +beta*(f/L)*(f/L);
385 matrix_view(1,1) = alpha;
390 DfDp_out_view(0,0) = 0.0;
391 DfDp_out_view(0,1) = 0.0;
392 DfDp_out_view(0,2) = 0.0;
393 DfDp_out_view(1,0) = -(f/L)*(f/L);
394 DfDp_out_view(1,1) = -(2.0*f/(L*L))*(a-x_in_view[0]);
395 DfDp_out_view(1,2) = +(2.0*f*f/(L*L*L))*(a-x_in_view[0]);
398 if (useDfDpAsTangent_ && !
is_null(DxdotDp_in)) {
400 DfDp_out_view(0,0) += DxdotDp(0,0);
401 DfDp_out_view(0,1) += DxdotDp(0,1);
402 DfDp_out_view(0,2) += DxdotDp(0,2);
403 DfDp_out_view(1,0) += DxdotDp(1,0);
404 DfDp_out_view(1,1) += DxdotDp(1,1);
405 DfDp_out_view(1,2) += DxdotDp(1,2);
407 if (useDfDpAsTangent_ && !
is_null(DxDp_in)) {
409 DfDp_out_view(0,0) += -DxDp(1,0);
410 DfDp_out_view(0,1) += -DxDp(1,1);
411 DfDp_out_view(0,2) += -DxDp(1,2);
412 DfDp_out_view(1,0) += (f/L)*(f/L) * DxDp(0,0);
413 DfDp_out_view(1,1) += (f/L)*(f/L) * DxDp(0,1);
414 DfDp_out_view(1,2) += (f/L)*(f/L) * DxDp(0,2);
420 if (acceptModelParams_) {
422 if (g_out != Teuchos::null)
423 Thyra::assign(g_out.
ptr(), *x_in);
426 outArgs.
get_DgDp(0,0).getMultiVector();
427 if (DgDp_out != Teuchos::null)
428 Thyra::assign(DgDp_out.ptr(), Scalar(0.0));
431 outArgs.
get_DgDx(0).getMultiVector();
432 if (DgDx_out != Teuchos::null) {
434 DgDx_out_view(0,0) = 1.0;
435 DgDx_out_view(0,1) = 0.0;
436 DgDx_out_view(1,0) = 0.0;
437 DgDx_out_view(1,1) = 1.0;
442 template<
class Scalar>
447 if (!acceptModelParams_) {
448 return Teuchos::null;
453 else if (l == 1 || l == 2)
455 return Teuchos::null;
458 template<
class Scalar>
463 if (!acceptModelParams_) {
464 return Teuchos::null;
470 p_strings->push_back(
"Model Coefficient: a");
471 p_strings->push_back(
"Model Coefficient: f");
472 p_strings->push_back(
"Model Coefficient: L");
475 p_strings->push_back(
"DxDp");
477 p_strings->push_back(
"Dx_dotDp");
481 template<
class Scalar>
492 template<
class Scalar>
497 if (isInitialized_) {
505 MEB::InArgsSetup<Scalar> inArgs;
506 inArgs.setModelEvalDescription(this->description());
507 inArgs.setSupports( MEB::IN_ARG_t );
508 inArgs.setSupports( MEB::IN_ARG_x );
509 inArgs.setSupports( MEB::IN_ARG_beta );
510 inArgs.setSupports( MEB::IN_ARG_x_dot );
511 inArgs.setSupports( MEB::IN_ARG_alpha );
512 if (acceptModelParams_) {
520 MEB::OutArgsSetup<Scalar> outArgs;
521 outArgs.setModelEvalDescription(this->description());
522 outArgs.setSupports( MEB::OUT_ARG_f );
523 outArgs.setSupports( MEB::OUT_ARG_W_op );
524 if (acceptModelParams_) {
525 outArgs.set_Np_Ng(Np_,Ng_);
526 outArgs.setSupports( MEB::OUT_ARG_DfDp,0,
527 MEB::DERIV_MV_JACOBIAN_FORM );
528 outArgs.setSupports( MEB::OUT_ARG_DgDp,0,0,
529 MEB::DERIV_MV_JACOBIAN_FORM );
530 outArgs.setSupports( MEB::OUT_ARG_DgDx,0,
531 MEB::DERIV_MV_GRADIENT_FORM );
537 nominalValues_ = inArgs_;
540 nominalValues_.set_t(t0_ic_);
544 x_ic_view[0] = a_+b_*sin((f_/L_)*t0_ic_+phi_);
545 x_ic_view[1] = b_*(f_/L_)*cos((f_/L_)*t0_ic_+phi_);
547 nominalValues_.set_x(x_ic);
548 if (acceptModelParams_) {
556 nominalValues_.set_p(0,p_ic);
561 x_dot_ic_view[0] = b_*(f_/L_)*cos((f_/L_)*t0_ic_+phi_);
562 x_dot_ic_view[1] = -b_*(f_/L_)*(f_/L_)*sin((f_/L_)*t0_ic_+phi_);
564 nominalValues_.set_x_dot(x_dot_ic);
567 isInitialized_ =
true;
571 template<
class Scalar>
579 if (paramList != Teuchos::null) tmpPL = paramList;
581 this->setMyParamList(tmpPL);
583 bool acceptModelParams = get<bool>(*pl,
"Accept model parameters");
584 bool haveIC = get<bool>(*pl,
"Provide nominal values");
585 bool useDfDpAsTangent = get<bool>(*pl,
"Use DfDp as Tangent");
586 if ( (acceptModelParams != acceptModelParams_) ||
589 isInitialized_ =
false;
591 acceptModelParams_ = acceptModelParams;
593 useDfDpAsTangent_ = useDfDpAsTangent;
594 a_ = get<Scalar>(*pl,
"Coeff a");
595 f_ = get<Scalar>(*pl,
"Coeff f");
596 L_ = get<Scalar>(*pl,
"Coeff L");
597 x0_ic_ = get<Scalar>(*pl,
"IC x0");
598 x1_ic_ = get<Scalar>(*pl,
"IC x1");
599 t0_ic_ = get<Scalar>(*pl,
"IC t0");
600 calculateCoeffFromIC_();
604 template<
class Scalar>
612 pl->
set(
"Accept model parameters",
false);
613 pl->
set(
"Provide nominal values",
true);
614 pl->
set(
"Use DfDp as Tangent",
false);
615 pl->
set<std::string>(
"Output File Name",
"Tempus_BDF2_SinCos");
616 Teuchos::setDoubleParameter(
617 "Coeff a", 0.0,
"Coefficient a in model", &*pl);
618 Teuchos::setDoubleParameter(
619 "Coeff f", 1.0,
"Coefficient f in model", &*pl);
620 Teuchos::setDoubleParameter(
621 "Coeff L", 1.0,
"Coefficient L in model", &*pl);
622 Teuchos::setDoubleParameter(
623 "IC x0", 0.0,
"Initial Condition for x0", &*pl);
624 Teuchos::setDoubleParameter(
625 "IC x1", 1.0,
"Initial Condition for x1", &*pl);
626 Teuchos::setDoubleParameter(
627 "IC t0", 0.0,
"Initial time t0", &*pl);
628 Teuchos::setIntParameter(
629 "Number of Time Step Sizes", 1,
"Number time step sizes for convergence study", &*pl);
635 template<
class Scalar>
640 phi_ = atan(((f_/L_)/x1_ic_)*(x0_ic_-a_))-(f_/L_)*t0_ic_;
641 b_ = x1_ic_/((f_/L_)*cos((f_/L_)*t0_ic_+phi_));
645 #endif // TEMPUS_TEST_SINCOS_MODEL_IMPL_HPP
bool is_null(const boost::shared_ptr< T > &p)
Derivative< Scalar > get_DfDp(int l) const
RCP< const VectorBase< Scalar > > get_x_dot() const
ParameterList & set(std::string const &name, T const &value, std::string const &docString="", RCP< const ParameterEntryValidator > const &validator=null)
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
Thyra::ModelEvaluatorBase::InArgs< Scalar > createInArgs() const
Evaluation< VectorBase< Scalar > > get_g(int j) const
Evaluation< VectorBase< Scalar > > get_f() const
Teuchos::RCP< Thyra::LinearOpWithSolveBase< Scalar > > create_W() const
Thyra::ModelEvaluatorBase::InArgs< Scalar > getNominalValues() const
Derivative< Scalar > get_DgDp(int j, int l) const
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_f_space() const
Teuchos::RCP< Thyra::LinearOpBase< Scalar > > create_W_op() const
#define TEUCHOS_ASSERT_IN_RANGE_UPPER_EXCLUSIVE(index, lower_inclusive, upper_exclusive)
void set_x(const RCP< const VectorBase< Scalar > > &x)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_x_space() const
Thyra::ModelEvaluatorBase::InArgs< Scalar > getExactSolution(double t) const
SinCosModel(Teuchos::RCP< Teuchos::ParameterList > pList=Teuchos::null)
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::ParameterList > getValidParameters() const
void set_x_dot(const RCP< const VectorBase< Scalar > > &x_dot)
void evalModelImpl(const Thyra::ModelEvaluatorBase::InArgs< Scalar > &inArgs_bar, const Thyra::ModelEvaluatorBase::OutArgs< Scalar > &outArgs_bar) const
Teuchos::RCP< const Thyra::VectorSpaceBase< Scalar > > get_g_space(int j) const
void calculateCoeffFromIC_()
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
Thyra::ModelEvaluatorBase::OutArgs< Scalar > createOutArgsImpl() const
void setupInOutArgs_() const
Thyra::ModelEvaluatorBase::InArgs< Scalar > getExactSensSolution(int j, double t) const
Teuchos::RCP< const Thyra::LinearOpWithSolveFactoryBase< Scalar > > get_W_factory() const
RCP< LinearOpBase< Scalar > > get_W_op() const
RCP< const VectorBase< Scalar > > get_x() const
RCP< const VectorBase< Scalar > > get_p(int l) const
RCP< MultiVectorBase< Scalar > > getMultiVector() const
void setParameterList(Teuchos::RCP< Teuchos::ParameterList > const ¶mList)
Derivative< Scalar > get_DgDx(int j) const