Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Stokhos_SGQuadMPModelEvaluator.hpp
Go to the documentation of this file.
1// @HEADER
2// ***********************************************************************
3//
4// Stokhos Package
5// Copyright (2009) Sandia Corporation
6//
7// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8// license for use of this work by or on behalf of the U.S. Government.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Eric T. Phipps (etphipp@sandia.gov).
38//
39// ***********************************************************************
40// @HEADER
41
42#ifndef STOKHOS_SGQUADMPMODELEVALUATOR_HPP
43#define STOKHOS_SGQUADMPMODELEVALUATOR_HPP
44
45#include "EpetraExt_ModelEvaluator.h"
46#include "EpetraExt_MultiComm.h"
47
48#include "Teuchos_RCP.hpp"
49#include "Teuchos_Array.hpp"
50
51namespace Stokhos {
52
64 class SGQuadMPModelEvaluator : public EpetraExt::ModelEvaluator {
65 public:
66
67 // Constructor
69 const Teuchos::RCP<EpetraExt::ModelEvaluator>& me,
70 const Teuchos::RCP<const EpetraExt::MultiComm>& mp_comm,
71 const Teuchos::RCP<const Epetra_Map>& mp_block_map);
72
75
77 Teuchos::RCP<const Epetra_Map> get_x_map() const;
78
80 Teuchos::RCP<const Epetra_Map> get_f_map() const;
81
83 Teuchos::RCP<const Epetra_Map> get_p_map(int l) const;
84
86 Teuchos::RCP<const Epetra_Map> get_g_map(int l) const;
87
89 Teuchos::RCP<const Teuchos::Array<std::string> >
90 get_p_names(int l) const;
91
93 Teuchos::RCP<const Epetra_Vector> get_x_init() const;
94
96 Teuchos::RCP<const Epetra_Vector> get_p_init(int l) const;
97
99 Teuchos::RCP<Epetra_Operator> create_W() const;
100
102 InArgs createInArgs() const;
103
105 OutArgs createOutArgs() const;
106
108 void evalModel(const InArgs& inArgs, const OutArgs& outArgs) const;
109
111
112 protected:
113
115 Teuchos::RCP<EpetraExt::ModelEvaluator> me;
116
118 Teuchos::RCP<const EpetraExt::MultiComm> mp_comm;
119
121 Teuchos::RCP<const Epetra_Map> mp_block_map;
122
124 Teuchos::RCP<const Stokhos::Quadrature<int,double> > sg_quad;
125
127 int num_p;
128
130 int num_g;
131
134
137
139 Teuchos::Array<int> mp_p_index_map;
140
142 Teuchos::Array<int> mp_g_index_map;
143
145 mp_vector_t x_dot_mp;
146
148 mp_vector_t x_mp;
149
151 Teuchos::Array<mp_vector_t> p_mp;
152
154 mp_vector_t f_mp;
155
157 mp_operator_t W_mp;
158
160 Teuchos::Array<MPDerivative> dfdp_mp;
161
163 Teuchos::Array<mp_vector_t> g_mp;
164
166 Teuchos::Array<MPDerivative> dgdx_mp;
167
169 Teuchos::Array<MPDerivative> dgdx_dot_mp;
170
172 Teuchos::Array< Teuchos::Array<MPDerivative> > dgdp_mp;
173
174 };
175
176}
177
178#endif // STOKHOS_SGQUADMPMODELEVALUATOR_HPP
ModelEvaluator adaptor that implements the stochastic Galerkin residual and Jacobian computations usi...
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
Return array of parameter names.
SGQuadMPModelEvaluator(const Teuchos::RCP< EpetraExt::ModelEvaluator > &me, const Teuchos::RCP< const EpetraExt::MultiComm > &mp_comm, const Teuchos::RCP< const Epetra_Map > &mp_block_map)
Teuchos::Array< mp_vector_t > p_mp
Parameter vectors.
Teuchos::RCP< const EpetraExt::MultiComm > mp_comm
Parallel MP communicator.
void evalModel(const InArgs &inArgs, const OutArgs &outArgs) const
Evaluate model on InArgs.
Teuchos::Array< mp_vector_t > g_mp
Response vectors.
int num_p_mp
Number of multipoint parameter vectors.
Teuchos::RCP< const Epetra_Map > get_g_map(int l) const
Return observation vector map.
Teuchos::Array< MPDerivative > dgdx_mp
Response derivative.
Teuchos::Array< int > mp_g_index_map
Index map between block-g and g_mp maps.
Teuchos::Array< Teuchos::Array< MPDerivative > > dgdp_mp
Response sensitivities.
Teuchos::RCP< const Epetra_Map > get_p_map(int l) const
Return parameter vector map.
mp_vector_t x_dot_mp
Time derivative vector.
Teuchos::RCP< const Epetra_Vector > get_x_init() const
Return initial solution.
Teuchos::RCP< const Stokhos::Quadrature< int, double > > sg_quad
SG quadrature.
Teuchos::RCP< const Epetra_Map > mp_block_map
Map for layout of parallel MP blocks.
Teuchos::RCP< const Epetra_Vector > get_p_init(int l) const
Return initial parameters.
Teuchos::RCP< const Epetra_Map > get_x_map() const
Return solution vector map.
Teuchos::Array< int > mp_p_index_map
Index map between block-p and p_mp maps.
Teuchos::Array< MPDerivative > dfdp_mp
Residual derivatives.
Teuchos::RCP< const Epetra_Map > get_f_map() const
Return residual vector map.
int num_g_mp
Number of multipoint response vectors.
Teuchos::Array< MPDerivative > dgdx_dot_mp
Response derivative.
Teuchos::RCP< EpetraExt::ModelEvaluator > me
Underlying model evaluator.
Teuchos::RCP< Epetra_Operator > create_W() const
Create W = alpha*M + beta*J matrix.
Top-level namespace for Stokhos classes and functions.