42 #ifndef TWOD_DIFFUSION_PROBLEM_HPP 43 #define TWOD_DIFFUSION_PROBLEM_HPP 45 #include "Teuchos_RCP.hpp" 46 #include "Teuchos_Array.hpp" 48 #include "Epetra_Map.h" 49 #include "Epetra_LocalMap.h" 50 #include "Epetra_Import.h" 51 #include "Epetra_CrsGraph.h" 52 #include "Epetra_CrsMatrix.h" 63 const Teuchos::RCP<const Epetra_Comm>& comm,
int n,
int d,
64 double s = 0.1,
double mu = 0.2,
74 Teuchos::RCP<const Epetra_Map>
get_x_map()
const;
77 Teuchos::RCP<const Epetra_Map>
get_f_map()
const;
80 Teuchos::RCP<const Epetra_Map>
get_p_map(
int l)
const;
83 Teuchos::RCP<const Epetra_Map>
get_g_map(
int j)
const;
86 Teuchos::RCP<const Teuchos::Array<std::string> >
90 Teuchos::RCP<const Epetra_Vector>
get_x_init()
const;
93 Teuchos::RCP<const Epetra_Vector>
get_p_init(
int l)
const;
96 Teuchos::RCP<Epetra_Operator>
create_W()
const;
100 const Epetra_Vector& p,
105 const Epetra_Vector& p,
110 const Epetra_Vector& p,
132 Teuchos::RCP<Epetra_CrsMatrix>
get_mean()
const {
return A_k[0]; }
137 template <
typename FuncT>
152 Teuchos::Array<MeshPoint>
mesh;
155 Teuchos::RCP<const Stokhos::OrthogPolyBasis<int,double> >
basis;
178 Teuchos::RCP< Teuchos::Array<std::string> >
p_names;
181 Teuchos::RCP<Epetra_CrsGraph>
graph;
184 Teuchos::Array<Teuchos::RCP<Epetra_CrsMatrix> >
A_k;
187 Teuchos::RCP<Epetra_Vector>
b;
193 Teuchos::RCP<Epetra_CrsMatrix>
A;
196 mutable Teuchos::Array<double>
point;
203 #endif // TWOD_DIFFUSION_ME_HPP
A container class storing an orthogonal polynomial whose coefficients are vectors, operators, or in general any type that would have an expensive copy constructor.
void computeSGJacobian(const Stokhos::EpetraVectorOrthogPoly &x_sg, const Stokhos::EpetraVectorOrthogPoly &p_sg, Stokhos::EpetraOperatorOrthogPoly &J_sg)
Compute Jacobian.
Teuchos::Array< Teuchos::RCP< Epetra_CrsMatrix > > A_k
KL coefficients of operator.
Teuchos::RCP< Epetra_Map > x_map
Solution vector map.
Teuchos::RCP< const Epetra_Vector > get_p_init(int l) const
Return initial parameters.
Teuchos::RCP< Epetra_CrsMatrix > A
Matrix to store deterministic operator.
Teuchos::Array< int > bcIndices
void computeResidual(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Vector &f)
Compute residual.
void computeResponse(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Vector &g)
Compute response.
void computeSGResponse(const Stokhos::EpetraVectorOrthogPoly &x_sg, const Stokhos::EpetraVectorOrthogPoly &p_sg, Stokhos::EpetraVectorOrthogPoly &g_sg)
Compute SG response.
Teuchos::RCP< Epetra_Map > g_map
Response vector map.
Teuchos::RCP< Epetra_Map > p_map
Parameter vector map.
void computeJacobian(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Operator &J)
Compute Jacobian.
Teuchos::RCP< const Epetra_Map > get_p_map(int l) const
Return parameter vector map.
Teuchos::RCP< Epetra_Vector > p_init
Initial parameters.
Abstract base class for orthogonal polynomial-based expansions.
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(int l) const
Return array of parameter names.
Teuchos::RCP< const Epetra_Vector > get_x_init() const
Return initial solution.
const IndexType const IndexType const IndexType const IndexType const ValueType const ValueType * x
Teuchos::Array< MeshPoint > mesh
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::Array< double > basis_vals
Array to store values of basis at a point.
Teuchos::RCP< Epetra_CrsMatrix > get_mean() const
Get mean matrix.
Teuchos::RCP< Epetra_Operator > create_W() const
Create W = alpha*M + beta*J matrix.
Teuchos::RCP< Epetra_Vector > x_init
Initial guess.
Teuchos::RCP< Epetra_Import > importer
Importer to overlapped distribution.
Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > basis
Teuchos::Array< Teuchos::RCP< Epetra_Vector > > sg_kx_vec_all
Vectors to store matrix-vector products in SG residual calculation.
twoD_diffusion_problem(const Teuchos::RCP< const Epetra_Comm > &comm, int n, int d, double s=0.1, double mu=0.2, const Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > &basis=Teuchos::null, bool log_normal=false, bool eliminate_bcs=false)
Constructor.
Teuchos::RCP< const Epetra_Map > get_g_map(int j) const
Return response function map.
ScalarType f(const Teuchos::Array< ScalarType > &x, double a, double b)
Teuchos::Array< double > point
Array to store a point for basis evaluation.
Teuchos::RCP< Epetra_Vector > b
Deterministic RHS.
Teuchos::RCP< const Epetra_Map > get_f_map() const
Return residual vector map.
void compute_A(const Epetra_Vector &p)
Compute A matrix.
void fillMatrices(const FuncT &func, int sz)
Fill coefficient matrix given function to evaluate diffusion coefficient.
Teuchos::RCP< Teuchos::Array< std::string > > p_names
Parameter names.
Teuchos::RCP< Epetra_CrsGraph > graph
Jacobian graph.
Teuchos::RCP< const Epetra_Map > get_x_map() const
Return solution vector map.
ScalarType g(const Teuchos::Array< ScalarType > &x, const ScalarType &y)
void computeSGResidual(const Stokhos::EpetraVectorOrthogPoly &x_sg, const Stokhos::EpetraVectorOrthogPoly &p_sg, Stokhos::OrthogPolyExpansion< int, double > &expn, Stokhos::EpetraVectorOrthogPoly &f_sg)
Compute SG residual.
A linear 2-D diffusion problem.