42#ifndef TWOD_DIFFUSION_PROBLEM_TPETRA_HPP
43#define TWOD_DIFFUSION_PROBLEM_TPETRA_HPP
45#include "Teuchos_RCP.hpp"
46#include "Teuchos_Array.hpp"
47#include "Teuchos_Comm.hpp"
49#include "Tpetra_Map.hpp"
50#include "Tpetra_Vector.hpp"
51#include "Tpetra_Import.hpp"
52#include "Tpetra_CrsGraph.hpp"
53#include "Tpetra_CrsMatrix.hpp"
69 typedef Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node>
Tpetra_Map;
70 typedef Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node>
Tpetra_Vector;
78 const Teuchos::RCP<
const Teuchos::Comm<int> >& comm,
80 BasisScalar s = 0.1, BasisScalar mu = 0.2,
100 Teuchos::RCP<const Teuchos::Array<std::string> >
132 template <
typename FuncT>
146 Teuchos::Array<MeshPoint>
mesh;
152 Teuchos::RCP<const Tpetra_Map>
x_map;
161 Teuchos::RCP<const Tpetra_Map>
p_map;
164 Teuchos::RCP<const Tpetra_Map>
g_map;
170 Teuchos::RCP< Teuchos::Array<std::string> >
p_names;
173 Teuchos::RCP<Tpetra_CrsGraph>
graph;
176 Teuchos::RCP<Tpetra_Vector>
b;
179 Teuchos::RCP<Tpetra_CrsMatrix>
A;
183 mutable Teuchos::Array<MeshScalar>
point;
184 Teuchos::RCP< Stokhos::KL::ExponentialRandomField<MeshScalar> >
rf;
186 BasisScalar mean, BasisScalar std_dev,
189 const Teuchos::Array<Scalar>& rv)
const;
194 template <
typename DiffusionFunc>
199 const Teuchos::Array<Scalar>& rv)
const {
200 return std::exp(
func(x,y,rv));
205 Teuchos::RCP<LogNormal_Diffusion_Func<KL_Diffusion_Func> >
lnFunc;
A linear 2-D diffusion problem.
Teuchos::RCP< Tpetra_CrsGraph > graph
Jacobian graph.
Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > Tpetra_CrsMatrix
Teuchos::RCP< const Teuchos::Array< std::string > > get_p_names(LocalOrdinal l) const
Return array of parameter names.
Teuchos::RCP< Tpetra_Vector > x_init
Initial guess.
Teuchos::RCP< Teuchos::Array< std::string > > p_names
Parameter names.
Teuchos::RCP< const Tpetra_Map > get_g_map(LocalOrdinal j) const
Return response function map.
Teuchos::RCP< const Tpetra_Vector > get_p_init(LocalOrdinal l) const
Return initial parameters.
Teuchos::RCP< Tpetra_Import > importer
Importer to overlapped distribution.
Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > Tpetra_CrsGraph
Teuchos::RCP< Tpetra_Vector > b
RHS.
Teuchos::RCP< const Tpetra_Map > get_x_map() const
Return solution vector map.
Teuchos::RCP< const Tpetra_Map > g_map
Response vector map.
void computeResidual(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Vector &f)
Compute residual.
Teuchos::RCP< Tpetra_CrsMatrix > create_W() const
Create W = alpha*M + beta*J matrix.
Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > Tpetra_Map
Teuchos::RCP< LogNormal_Diffusion_Func< KL_Diffusion_Func > > lnFunc
Tpetra::Operator< Scalar, LocalOrdinal, GlobalOrdinal, Node > Tpetra_Operator
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::Array< GlobalOrdinal > bcIndices
Teuchos::RCP< const Tpetra_Vector > get_x_init() const
Return initial solution.
Teuchos::Array< MeshPoint > mesh
Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > Tpetra_Vector
Teuchos::RCP< const Tpetra_Map > get_f_map() const
Return residual vector map.
void computeJacobian(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Operator &J)
Compute Jacobian.
Teuchos::RCP< const Tpetra_Map > get_p_map(LocalOrdinal l) const
Return parameter vector map.
Teuchos::RCP< const Tpetra_Map > p_map
Parameter vector map.
Teuchos::RCP< KL_Diffusion_Func > klFunc
void computeA(const FuncT &func, const Tpetra_Vector &p, Tpetra_CrsMatrix &jac)
Fill coefficient matrix given function to evaluate diffusion coefficient.
void computeResponse(const Epetra_Vector &x, const Epetra_Vector &p, Epetra_Vector &g)
Compute response.
Teuchos::RCP< const Tpetra_Map > x_map
Solution vector map.
Teuchos::RCP< Tpetra_CrsMatrix > A
Matrix to store operator.
Tpetra::Import< LocalOrdinal, GlobalOrdinal, Node > Tpetra_Import
Teuchos::RCP< Tpetra_Vector > p_init
Initial parameters.
KokkosClassic::DefaultNode::DefaultNodeType Node
ScalarType g(const Teuchos::Array< ScalarType > &x, const ScalarType &y)
ScalarType f(const Teuchos::Array< ScalarType > &x, double a, double b)
Teuchos::Array< MeshScalar > point
Scalar operator()(MeshScalar x, MeshScalar y, const Teuchos::Array< Scalar > &rv) const
Teuchos::RCP< Stokhos::KL::ExponentialRandomField< MeshScalar > > rf
KL_Diffusion_Func(MeshScalar xyLeft, MeshScalar xyRight, BasisScalar mean, BasisScalar std_dev, MeshScalar L, LocalOrdinal num_KL)
LogNormal_Diffusion_Func(const DiffusionFunc &func_)
Scalar operator()(MeshScalar x, MeshScalar y, const Teuchos::Array< Scalar > &rv) const
const DiffusionFunc & func