Simple2DTpetraModelEvaluator_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//    Thyra: Interfaces and Support for Abstract Numerical Algorithms
//
// Copyright 2004 NTESS and the Thyra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef SIMPLE_2D_TPETRA_MODEL_EVALUATOR_DECL_HPP
#define SIMPLE_2D_TPETRA_MODEL_EVALUATOR_DECL_HPP


#include "Thyra_StateFuncModelEvaluatorBase.hpp"
#include "Tpetra_CrsGraph.hpp"
#include "Tpetra_Vector.hpp"


template<class Scalar>
class Simple2DTpetraModelEvaluator
  : public Thyra::StateFuncModelEvaluatorBase<Scalar>
{
public:


  Simple2DTpetraModelEvaluator();

  void set_d(const Scalar &d);

  void set_p(const Teuchos::ArrayView<const Scalar> &p);

  void set_x0(const Teuchos::ArrayView<const Scalar> &x0);



  Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> > get_x_space() const;
  Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> > get_f_space() const;
  Thyra::ModelEvaluatorBase::InArgs<Scalar> getNominalValues() const;
  Teuchos::RCP<Thyra::LinearOpBase<Scalar> > create_W_op() const;
  Thyra::ModelEvaluatorBase::InArgs<Scalar> createInArgs() const;


private:


  Thyra::ModelEvaluatorBase::OutArgs<Scalar> createOutArgsImpl() const;
  void evalModelImpl(
    const Thyra::ModelEvaluatorBase::InArgs<Scalar> &inArgs,
    const Thyra::ModelEvaluatorBase::OutArgs<Scalar> &outArgs
    ) const;


private: // data members

  Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> > x_space_;
  Teuchos::RCP<const Thyra::VectorSpaceBase<Scalar> > f_space_;
  Thyra::ModelEvaluatorBase::InArgs<Scalar> nominalValues_;
  Scalar d_;
  Teuchos::RCP<Tpetra::Vector<Scalar> > x0_;
  Teuchos::Array<Scalar> p_;
  Teuchos::RCP<Tpetra::CrsGraph<> > W_op_graph_;
  Thyra::ModelEvaluatorBase::InArgs<Scalar> prototypeInArgs_;
  Thyra::ModelEvaluatorBase::OutArgs<Scalar> prototypeOutArgs_;

};


template<class Scalar>
Teuchos::RCP<Simple2DTpetraModelEvaluator<Scalar> >
simple2DTpetraModelEvaluator()
{
  return Teuchos::rcp(new Simple2DTpetraModelEvaluator<Scalar>);
}


#endif // SIMPLE_2D_TPETRA_MODEL_EVALUATOR_DECL_HPP