42#ifndef STOKHOS_MPMODELEVALUATOR_HPP
43#define STOKHOS_MPMODELEVALUATOR_HPP
45#include "EpetraExt_ModelEvaluator.h"
46#include "EpetraExt_MultiComm.h"
47#include "EpetraExt_BlockVector.h"
49#include "Teuchos_RCP.hpp"
50#include "Teuchos_Array.hpp"
51#include "Teuchos_ParameterList.hpp"
71 const Teuchos::RCP<EpetraExt::ModelEvaluator>&
me,
72 const Teuchos::RCP<const EpetraExt::MultiComm>&
mp_comm,
74 const Teuchos::RCP<Teuchos::ParameterList>&
params);
80 Teuchos::RCP<const Epetra_Map>
get_x_map()
const;
83 Teuchos::RCP<const Epetra_Map>
get_f_map()
const;
86 Teuchos::RCP<const Epetra_Map>
get_p_map(
int l)
const;
89 Teuchos::RCP<const Epetra_Map>
get_g_map(
int l)
const;
92 Teuchos::RCP<const Teuchos::Array<std::string> >
96 Teuchos::RCP<const Epetra_Vector>
get_x_init()
const;
100 Teuchos::RCP<const Epetra_Vector>
get_p_init(
int l)
const;
103 Teuchos::RCP<Epetra_Operator>
create_W()
const;
106 Teuchos::RCP<EpetraExt::ModelEvaluator::Preconditioner>
create_WPrec()
const;
127 void evalModel(
const InArgs& inArgs,
const OutArgs& outArgs)
const;
136 Teuchos::RCP<const Stokhos::ProductEpetraVector>
get_x_mp_init()
const;
143 Teuchos::RCP<const Stokhos::ProductEpetraVector>
get_p_mp_init(
int l)
const;
161 Teuchos::RCP<Stokhos::ProductEpetraVector>
166 Teuchos::RCP<Stokhos::ProductEpetraMultiVector>
172 Teuchos::RCP<Stokhos::ProductEpetraVector>
177 Teuchos::RCP<Stokhos::ProductEpetraMultiVector>
182 Teuchos::RCP<Stokhos::ProductEpetraVector>
187 Teuchos::RCP<Stokhos::ProductEpetraMultiVector>
192 Teuchos::RCP<Stokhos::ProductEpetraVector>
197 Teuchos::RCP<Stokhos::ProductEpetraMultiVector>
204 Teuchos::RCP<EpetraExt::ModelEvaluator>
me;
210 Teuchos::RCP<const EpetraExt::MultiComm>
mp_comm;
216 Teuchos::RCP<Teuchos::ParameterList>
params;
222 Teuchos::RCP<const Epetra_Map>
x_map;
225 Teuchos::RCP<const Epetra_Map>
f_map;
243 Teuchos::Array< Teuchos::RCP<const Epetra_Map> >
mp_p_map;
246 Teuchos::Array< Teuchos::RCP< Teuchos::Array<std::string> > >
mp_p_names;
258 Teuchos::Array< Teuchos::RCP<const Epetra_Map> >
mp_g_map;
261 mutable Teuchos::RCP< Stokhos::ProductEpetraOperator>
W_mp_blocks;
268 Teuchos::Array< Teuchos::RCP<ProductEpetraVector> >
mp_p_init;
271 mutable Teuchos::RCP<Stokhos::BlockDiagonalOperator>
my_W;
274 mutable Teuchos::RCP<Epetra_Vector>
my_x;
Multi-point model evaluator.
Teuchos::RCP< EpetraExt::ModelEvaluator::Preconditioner > create_WPrec() const
Create preconditioner operator.
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > get_g_mp_base_maps() const
Get base maps of MP responses.
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
Return array of parameter names.
OutArgs createOutArgs() const
Create OutArgs.
Teuchos::RCP< Stokhos::ProductEpetraMultiVector > create_g_mv_mp(int l, int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-point multi-vector using g map.
void set_p_mp_init(int i, const Stokhos::ProductEpetraVector &p_mp_in)
Set initial multi-point parameter.
Teuchos::RCP< const Epetra_Map > x_map
Underlying unknown map.
Teuchos::RCP< const Epetra_Vector > get_x_dot_init() const
Teuchos::RCP< Stokhos::ProductEpetraVector > create_x_mp(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create multi-point vector using x map and owned mp map.
Teuchos::RCP< Stokhos::BlockDiagonalOperator > my_W
W pointer for evaluating preconditioner.
Teuchos::RCP< Epetra_Operator > create_W() const
Create W = alpha*M + beta*J matrix.
Teuchos::RCP< Stokhos::ProductEpetraVector > create_f_mp(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create multi-point vector using f map.
bool supports_x
Whether we support x (and thus f and W)
Teuchos::Array< int > mp_g_index_map
Index map between block-g and g_mp maps.
Teuchos::RCP< Stokhos::ProductEpetraMultiVector > create_x_mv_mp(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-point vector using x map.
Teuchos::RCP< const Epetra_Map > f_map
Underlying residual map.
void set_x_mp_init(const Stokhos::ProductEpetraVector &x_mp_in)
Set initial multi-point solution.
Teuchos::RCP< const Epetra_Map > get_p_map(int l) const
Return parameter vector map.
Teuchos::Array< Teuchos::RCP< Teuchos::Array< std::string > > > mp_p_names
MP coefficient parameter names.
int num_g_mp
Number of multi-point response vectors.
Teuchos::RCP< Stokhos::ProductEpetraOperator > W_mp_blocks
W multi-point components.
Teuchos::RCP< const Epetra_Map > mp_block_map
Map for layout of parallel MP blocks.
Teuchos::RCP< const Epetra_Map > get_g_map(int l) const
Return response map.
Teuchos::RCP< Stokhos::ProductEpetraVector > mp_x_dot_init
Teuchos::RCP< Epetra_Operator > create_DgDp_op(int j, int i) const
Create MP operator representing dg/dp.
Teuchos::RCP< Epetra_Operator > create_DfDp_op(int i) const
Create MP operator representing df/dp.
Teuchos::Array< int > get_p_mp_map_indices() const
Get indices of MP parameters.
Teuchos::RCP< Epetra_Operator > create_DgDx_op(int j) const
Create MP operator representing dg/dx.
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > mp_g_map
Block MP response map.
Teuchos::RCP< EpetraExt::ModelEvaluator > me
Underlying model evaluator.
Teuchos::Array< Teuchos::RCP< ProductEpetraVector > > mp_p_init
MP initial p.
Teuchos::RCP< const Epetra_Vector > get_x_init() const
Return initial solution.
Teuchos::RCP< Stokhos::ProductEpetraMultiVector > create_f_mv_mp(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-point multi-vector using f map.
Teuchos::RCP< const Epetra_Map > get_x_map() const
Return solution vector map.
void set_x_dot_mp_init(const Stokhos::ProductEpetraVector &x_dot_mp_in)
Teuchos::RCP< Epetra_Operator > create_DgDx_dot_op(int j) const
Create MP operator representing dg/dxdot.
Teuchos::RCP< Stokhos::ProductEpetraMultiVector > create_p_mv_mp(int l, int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-point vector using p map.
Teuchos::RCP< const Stokhos::ProductEpetraVector > get_x_dot_mp_init() const
Teuchos::RCP< const Stokhos::ProductEpetraVector > get_x_mp_init() const
Return initial SG x.
unsigned int num_mp_blocks
Number of blocks.
void evalModel(const InArgs &inArgs, const OutArgs &outArgs) const
Evaluate model on InArgs.
Teuchos::RCP< const EpetraExt::MultiComm > mp_comm
Parallel MP communicator.
Teuchos::RCP< const Epetra_Map > mp_f_map
Block MP residual map.
Teuchos::RCP< Stokhos::ProductEpetraVector > create_p_mp(int l, Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create multi-point vector using p map.
Teuchos::RCP< const Epetra_Vector > get_p_init(int l) const
Return initial parameters.
Teuchos::Array< int > get_g_mp_map_indices() const
Get indices of MP responses.
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > mp_p_map
Block MP parameter map.
Teuchos::RCP< const Epetra_Map > get_f_map() const
Return residual vector map.
Teuchos::RCP< const Stokhos::ProductEpetraVector > get_p_mp_init(int l) const
Return initial SG parameters.
int num_p_mp
Number of multi-point parameter vectors.
InArgs createInArgs() const
Create InArgs.
Teuchos::RCP< Epetra_Vector > my_x
x pointer for evaluating preconditioner
int num_g
Number of response vectors of underlying model evaluator.
Teuchos::RCP< Stokhos::ProductEpetraVector > mp_x_init
MP initial x.
Teuchos::RCP< const Epetra_Map > mp_x_map
Block MP unknown map.
Teuchos::Array< int > mp_p_index_map
Index map between block-p and p_mp maps.
Teuchos::RCP< Teuchos::ParameterList > params
Algorithmic parameters.
int num_p
Number of parameter vectors of underlying model evaluator.
Teuchos::RCP< Stokhos::ProductEpetraVector > create_g_mp(int l, Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create multi-point vector using g map.
A container class for products of Epetra_Vector's.
Top-level namespace for Stokhos classes and functions.