42 #ifndef STOKHOS_SGMODELEVALUATOR_ADAPTIVE_HPP 43 #define STOKHOS_SGMODELEVALUATOR_ADAPTIVE_HPP 47 #include "EpetraExt_ModelEvaluator.h" 48 #include "EpetraExt_MultiComm.h" 49 #include "EpetraExt_BlockVector.h" 52 #include "Teuchos_RCP.hpp" 53 #include "Teuchos_Array.hpp" 54 #include "Teuchos_ParameterList.hpp" 87 const Teuchos::RCP<EpetraExt::ModelEvaluator>& me_,
88 const Teuchos::RCP<Stokhos::AdaptivityManager> & am,
91 const Teuchos::RCP<const Stokhos::ParallelData>& sg_parallel_data_,
92 bool onlyUseLinear_,
int kExpOrder_,
93 const Teuchos::RCP<Teuchos::ParameterList>& params_);
97 const Teuchos::RCP<EpetraExt::ModelEvaluator>&
me,
104 const Teuchos::RCP<Teuchos::ParameterList>&
params,
114 Teuchos::RCP<const Epetra_Map>
get_x_map()
const;
117 Teuchos::RCP<const Epetra_Map>
get_p_map(
int l)
const;
120 Teuchos::RCP<const Teuchos::Array<std::string> >
124 Teuchos::RCP<const Epetra_Vector>
get_x_init()
const;
127 Teuchos::RCP<const Epetra_Vector>
get_p_init(
int l)
const;
133 Teuchos::RCP<const Epetra_Map>
get_f_map()
const;
136 Teuchos::RCP<const Epetra_Map>
get_g_map(
int l)
const;
139 Teuchos::RCP<Epetra_Operator>
create_W()
const;
151 void evalModel(
const InArgs& inArgs,
const OutArgs& outArgs)
const;
162 Teuchos::RCP<const Stokhos::EpetraVectorOrthogPoly>
get_x_sg_init()
const;
168 Teuchos::RCP<const Stokhos::EpetraVectorOrthogPoly>
get_p_sg_init(
int l)
const;
195 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
197 const Epetra_Vector* v=NULL)
const;
200 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
202 const Epetra_Vector* v=NULL)
const;
205 Teuchos::RCP<Stokhos::EpetraMultiVectorOrthogPoly>
207 Epetra_DataAccess CV = Copy,
208 const Epetra_MultiVector* v = NULL)
const;
211 Teuchos::RCP<Stokhos::EpetraMultiVectorOrthogPoly>
213 Epetra_DataAccess CV = Copy,
214 const Epetra_MultiVector* v = NULL)
const;
217 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
219 const Epetra_Vector * v=0)
const;
222 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
224 const Epetra_Vector* v = NULL)
const;
227 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
229 const Epetra_Vector* v = NULL)
const;
232 Teuchos::RCP<Stokhos::EpetraMultiVectorOrthogPoly>
234 const Epetra_MultiVector* v = NULL)
const;
237 Teuchos::RCP<Stokhos::EpetraMultiVectorOrthogPoly>
239 const Epetra_MultiVector* v = NULL)
const;
242 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
244 const Epetra_Vector* v = NULL)
const;
247 Teuchos::RCP<Stokhos::EpetraMultiVectorOrthogPoly>
249 const Epetra_MultiVector* v = NULL)
const;
259 Teuchos::RCP<EpetraExt::ModelEvaluator>
me;
262 Teuchos::RCP<const Stokhos::OrthogPolyBasis<int, double> >
sg_basis;
267 Teuchos::RCP<const Stokhos::Quadrature<int,double> >
sg_quad;
270 Teuchos::RCP<Stokhos::OrthogPolyExpansion<int,double> >
sg_exp;
273 Teuchos::RCP<Teuchos::ParameterList>
params;
288 Teuchos::RCP<const Epetra_Map>
x_map;
291 Teuchos::RCP<const Epetra_Map>
f_map;
297 Teuchos::RCP<const EpetraExt::MultiComm>
sg_comm;
305 Teuchos::RCP<const Stokhos::Sparse3Tensor<int,double> >
Cijk;
356 Teuchos::Array< Teuchos::RCP<const Epetra_Map> >
sg_p_map;
359 Teuchos::Array< Teuchos::RCP< Teuchos::Array<std::string> > >
sg_p_names;
371 Teuchos::Array< Teuchos::RCP<const Epetra_Map> >
sg_g_map;
380 mutable Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly >
f_sg_blocks;
383 mutable Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly >
W_sg_blocks;
385 mutable Teuchos::Array< Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > >
dfdp_sg_blocks;
391 mutable Teuchos::Array< Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > >
dgdx_sg_blocks;
394 Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly>
sg_x_init;
397 Teuchos::Array< Teuchos::RCP<Stokhos::EpetraVectorOrthogPoly> >
sg_p_init;
406 mutable Teuchos::RCP<Epetra_CrsMatrix>
my_W;
409 mutable Teuchos::RCP<Epetra_Vector>
my_x;
413 mutable Teuchos::RCP<Stokhos::AdaptivityManager>
adaptMngr;
void evalModel(const InArgs &inArgs, const OutArgs &outArgs) const
Evaluate model on InArgs.
Teuchos::RCP< const Epetra_Map > get_g_map(int l) const
Return response map.
Teuchos::Array< int > get_p_sg_map_indices() const
Get indices of SG parameters.
bool eval_W_with_f
Whether to always evaluate W with f.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > x_sg_blocks
x stochastic Galerkin components
int num_g_sg
Number of stochastic response vectors.
Teuchos::RCP< Epetra_Operator > create_W() const
Create W = alpha*M + beta*J matrix.
Teuchos::RCP< const Epetra_Map > sg_f_map
Block SG residual map.
Teuchos::Array< int > get_g_sg_map_indices() const
Get indices of SG responses.
Teuchos::RCP< const Epetra_Map > get_x_map() const
Return solution vector map.
Teuchos::RCP< const Epetra_BlockMap > stoch_row_map
Map for stochastic blocks.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_f_sg_overlap(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create vector orthog poly using f map and overlap sg map.
Teuchos::RCP< const Stokhos::EpetraSparse3Tensor > serialCijk
Serial Epetra Cijk for dgdx*.
Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > create_f_mv_sg_overlap(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-vector orthog poly using f map and overlap sg map.
Teuchos::RCP< Stokhos::OrthogPolyExpansion< int, double > > sg_exp
Stochastic Galerkin expansion.
Teuchos::RCP< const Stokhos::Sparse3Tensor< int, double > > Cijk
Teuchos::Array< Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > > dgdx_sg_blocks
dg/dx stochastic Galerkin components
Teuchos::RCP< const Epetra_Map > get_f_map() const
Return residual vector map.
Teuchos::RCP< Epetra_Export > adapted_overlapped_f_exporter
Exporter from SG-overlapped to SG maps.
Teuchos::Array< Teuchos::RCP< Teuchos::Array< std::string > > > sg_p_names
SG coefficient parameter names.
Teuchos::RCP< const Epetra_Map > sg_x_map
Block SG unknown map.
Teuchos::Array< Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > > dgdx_dot_sg_blocks
dg/dxdot stochastic Galerkin components
void set_p_sg_init(int i, const Stokhos::EpetraVectorOrthogPoly &p_sg_in)
Set initial parameter polynomial.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_x_sg(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create vector orthog poly using x map and owned sg map.
Teuchos::RCP< const Epetra_Map > x_map
Underlying unknown map.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_g_sg(int l, Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create vector orthog poly using g map.
Teuchos::RCP< const Stokhos::ParallelData > sg_parallel_data
Parallel SG data.
Teuchos::RCP< const Epetra_BlockMap > get_overlap_stochastic_map() const
Return overlap stochastic map.
Teuchos::RCP< const Epetra_Map > adapted_overlapped_x_map
Adapated block SG overlapped unknown map.
int num_g
Number of response vectors of underlying model evaluator.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > sg_x_init
SG initial x.
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > sg_g_map
Block SG response map.
Nonlinear, stochastic Galerkin ModelEvaluator that constructs an adapted Jacobian.
OutArgs createOutArgs() const
Create OutArgs.
Teuchos::RCP< const Epetra_Map > get_p_map(int l) const
Return parameter vector map.
Teuchos::RCP< const Stokhos::EpetraSparse3Tensor > epetraCijk
Epetra Cijk.
InArgs createInArgs() const
Create InArgs.
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
Return array of parameter names.
Abstract base class for orthogonal polynomial-based expansions.
unsigned int num_W_blocks
Number of W stochastic blocks (may be smaller than num_sg_blocks)
std::vector< Teuchos::RCP< const Stokhos::ProductBasis< int, double > > > sg_row_dof_basis
Teuchos::RCP< EpetraExt::ModelEvaluator > me
Underlying model evaluator.
Teuchos::Array< Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > > dfdp_sg_blocks
Teuchos::RCP< Epetra_CrsMatrix > my_W
W pointer for evaluating W with f.
Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > create_x_mv_sg(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create vector orthog poly using x map and owned sg map.
Teuchos::RCP< const Stokhos::EpetraVectorOrthogPoly > get_p_sg_init(int l) const
Return initial SG parameters.
bool supports_x
Whether we support x (and thus f and W)
Abstract base class for quadrature methods.
int num_p
Number of parameter vectors of underlying model evaluator.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_p_sg(int l, Epetra_DataAccess CV=Copy, const Epetra_Vector *v=0) const
Create vector orthog poly using p map.
A container class storing an orthogonal polynomial whose coefficients are vectors, operators, or in general any type that would have an expensive copy constructor.
Teuchos::RCP< const Epetra_Map > sg_overlapped_x_map
Block SG overlapped unknown map.
Top-level namespace for Stokhos classes and functions.
Teuchos::RCP< const Stokhos::Quadrature< int, double > > sg_quad
Stochastic Galerkin quadrature.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > x_dot_sg_blocks
x_dot stochastic Galerkin components
Teuchos::RCP< const Stokhos::AdaptivityManager > getAdaptivityManager() const
Teuchos::RCP< const Epetra_Map > adapted_overlapped_f_map
Adapted block SG overlapped residual map.
Teuchos::RCP< const Epetra_BlockMap > overlapped_stoch_row_map
Overlapped map for stochastic blocks (local map)
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > get_g_sg_base_maps() const
Get base maps of SG responses.
Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > create_x_mv_sg_overlap(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create vector orthog poly using x map and overlap sg map.
Abstract base class for multivariate orthogonal polynomials generated from tensor products of univari...
Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > create_f_mv_sg(int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-vector orthog poly using f map and owned sg map.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_f_sg(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create vector orthog poly using f map and owned sg map.
Teuchos::RCP< const Epetra_Map > adapted_f_map
Adapted block SG residual map.
Teuchos::RCP< const Epetra_Map > adapted_x_map
Adapted lock SG unknown map.
Teuchos::RCP< const Stokhos::EpetraVectorOrthogPoly > get_x_sg_init() const
Return initial SG x.
Teuchos::Array< Teuchos::RCP< const Epetra_Map > > sg_p_map
Block SG parameter map.
Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly > W_sg_blocks
W stochastic Galerkin components.
Teuchos::RCP< Epetra_Import > adapted_overlapped_x_importer
Importer from SG to SG-overlapped maps.
Teuchos::RCP< const Epetra_BlockMap > get_x_sg_overlap_map() const
Return x sg overlap map.
Teuchos::RCP< const Epetra_Vector > get_p_init(int l) const
Return initial parameters.
Teuchos::RCP< Stokhos::EpetraMultiVectorOrthogPoly > create_g_mv_sg(int l, int num_vecs, Epetra_DataAccess CV=Copy, const Epetra_MultiVector *v=NULL) const
Create multi-vector orthog poly using g map.
Base class for stochastic Galerkin model evaluators.
Teuchos::Array< int > sg_p_index_map
Index map between block-p and p_sg maps.
Teuchos::RCP< const Epetra_Vector > get_x_init() const
Return initial solution.
int num_p_sg
Number of stochastic parameter vectors.
void set_x_sg_init(const Stokhos::EpetraVectorOrthogPoly &x_sg_in)
Set initial solution polynomial.
unsigned int num_sg_blocks
Number of stochastic blocks.
Teuchos::Array< int > sg_g_index_map
Index map between block-g and g_sg maps.
Teuchos::RCP< const EpetraExt::MultiComm > sg_comm
Parallel SG communicator.
Teuchos::RCP< const Epetra_Import > get_x_sg_importer() const
Return x sg importer.
Teuchos::RCP< Epetra_Vector > my_x
x pointer for evaluating preconditioner
Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > sg_basis
Stochastic Galerkin basis.
Teuchos::RCP< Teuchos::ParameterList > params
Algorithmic parameters.
unsigned int num_p_blocks
Number of p stochastic blocks (may be smaller than num_sg_blocks)
Teuchos::RCP< Stokhos::AdaptivityManager > adaptMngr
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > create_x_sg_overlap(Epetra_DataAccess CV=Copy, const Epetra_Vector *v=NULL) const
Create vector orthog poly using x map and overlap sg map.
Teuchos::RCP< const Epetra_Map > sg_overlapped_f_map
Block SG overlapped residual map.
Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > f_sg_blocks
f stochastic Galerkin components
Teuchos::RCP< const Epetra_BlockMap > overlapped_stoch_p_map
Overlapped map for p stochastic blocks (local map)
SGModelEvaluator_Adaptive(const Teuchos::RCP< EpetraExt::ModelEvaluator > &me_, const Teuchos::RCP< Stokhos::AdaptivityManager > &am, const Teuchos::RCP< const Stokhos::Quadrature< int, double > > &sg_quad_, const Teuchos::RCP< Stokhos::OrthogPolyExpansion< int, double > > &sg_exp_, const Teuchos::RCP< const Stokhos::ParallelData > &sg_parallel_data_, bool onlyUseLinear_, int kExpOrder_, const Teuchos::RCP< Teuchos::ParameterList > ¶ms_)
Teuchos::Array< Teuchos::RCP< Stokhos::EpetraVectorOrthogPoly > > sg_p_init
SG initial p.
Teuchos::RCP< const Epetra_Map > f_map
Underlying residual map.