44 #ifndef STOKHOS_KLMATRIX_FREE_OPERATOR_HPP 45 #define STOKHOS_KLMATRIX_FREE_OPERATOR_HPP 48 #include "EpetraExt_MultiComm.h" 51 #include "Epetra_Map.h" 52 #include "Epetra_Import.h" 53 #include "Teuchos_ParameterList.hpp" 54 #include "Teuchos_Array.hpp" 68 const Teuchos::RCP<const EpetraExt::MultiComm>&
sg_comm,
70 const Teuchos::RCP<const Stokhos::EpetraSparse3Tensor>&
epetraCijk,
75 const Teuchos::RCP<Teuchos::ParameterList>& params);
85 const Teuchos::RCP<Stokhos::EpetraOperatorOrthogPoly >& poly);
88 virtual Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly >
92 virtual Teuchos::RCP<const Stokhos::EpetraOperatorOrthogPoly >
107 virtual int Apply(
const Epetra_MultiVector& Input,
108 Epetra_MultiVector& Result)
const;
115 Epetra_MultiVector& Y)
const;
118 virtual double NormInf()
const;
121 virtual const char*
Label ()
const;
136 virtual const Epetra_Comm &
Comm()
const;
166 Teuchos::RCP<const EpetraExt::MultiComm>
sg_comm;
169 Teuchos::RCP<const Stokhos::OrthogPolyBasis<int,double> >
sg_basis;
208 Teuchos::RCP<const Cijk_type>
Cijk;
211 Teuchos::RCP<Stokhos::EpetraOperatorOrthogPoly >
block_ops;
238 mutable Teuchos::Array< Teuchos::RCP<const Epetra_MultiVector> >
input_block;
241 mutable Teuchos::Array< Teuchos::RCP<Epetra_MultiVector> >
result_block;
244 mutable Teuchos::RCP<Epetra_MultiVector>
tmp;
259 #endif // STOKHOS_KLMATRIX_FREE_OPERATOR_HPP Teuchos::RCP< const Epetra_Map > range_sg_map
Stores range SG map.
Cijk_type::k_iterator k_end
Ending k iterator.
Teuchos::RCP< Epetra_Map > global_col_map_trans
Stores operator column SG map for transpose.
bool scale_op
Flag indicating whether operator be scaled with <^2>
virtual const char * Label() const
Returns a character std::string describing the operator.
Teuchos::RCP< const Stokhos::EpetraSparse3Tensor > epetraCijk
Stores Epetra Cijk tensor.
bool is_stoch_parallel
Whether we have parallelism over stochastic blocks.
Teuchos::RCP< const EpetraExt::MultiComm > sg_comm
Stores SG parallel communicator.
virtual int SetUseTranspose(bool UseTranspose)
Set to true if the transpose of the operator is requested.
An Epetra operator representing the block stochastic Galerkin operator.
virtual Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly > getSGPolynomial()
Get SG polynomial.
virtual const Epetra_Map & OperatorRangeMap() const
Returns the Epetra_Map object associated with the range of this matrix operator.
int num_blocks
Number of Jacobian blocks (not necessarily equal to expansion_size)
KLMatrixFreeOperator(const Teuchos::RCP< const EpetraExt::MultiComm > &sg_comm, const Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > &sg_basis, const Teuchos::RCP< const Stokhos::EpetraSparse3Tensor > &epetraCijk, const Teuchos::RCP< const Epetra_Map > &domain_base_map, const Teuchos::RCP< const Epetra_Map > &range_base_map, const Teuchos::RCP< const Epetra_Map > &domain_sg_map, const Teuchos::RCP< const Epetra_Map > &range_sg_map, const Teuchos::RCP< Teuchos::ParameterList > ¶ms)
Constructor.
Stokhos::Sparse3Tensor< int, double > Cijk_type
Short-hand for Cijk.
virtual bool UseTranspose() const
Returns the current UseTranspose setting.
int expansion_size
Number of terms in expansion.
Teuchos::RCP< const Epetra_Map > range_base_map
Stores range base map.
Teuchos::RCP< const Stokhos::OrthogPolyBasis< int, double > > sg_basis
Stochastic Galerking basis.
virtual const Epetra_Map & OperatorDomainMap() const
Returns the Epetra_Map object associated with the domain of this matrix operator. ...
Bi-directional iterator for traversing a sparse array.
bool useTranspose
Flag indicating whether transpose was selected.
virtual int Apply(const Epetra_MultiVector &Input, Epetra_MultiVector &Result) const
Returns the result of a Epetra_Operator applied to a Epetra_MultiVector Input in Result as described ...
Teuchos::Array< Teuchos::RCP< const Epetra_MultiVector > > input_block
MultiVectors for each block for Apply() input.
Teuchos::RCP< Epetra_MultiVector > tmp_trans
Temporary multivector used in Apply() for transpose.
virtual double NormInf() const
Returns an approximate infinity norm of the operator matrix.
An abstract class to represent a generic stochastic Galerkin operator as an Epetra_Operator.
Teuchos::RCP< const Epetra_Map > domain_sg_map
Stores domain SG map.
Cijk_type::k_iterator k_begin
Starting k iterator.
int max_num_mat_vec
Maximum number of matvecs in Apply.
Teuchos::RCP< const Cijk_type > Cijk
Stores triple product tensor.
Teuchos::RCP< const Epetra_BlockMap > stoch_col_map
Stores stochastic part of column map.
virtual ~KLMatrixFreeOperator()
Destructor.
Top-level namespace for Stokhos classes and functions.
Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly > block_ops
Stores operators.
virtual bool HasNormInf() const
Returns true if the this object can provide an approximate Inf-norm, false otherwise.
Teuchos::RCP< Epetra_Import > col_importer
Importer from domain map to column map.
std::string label
Label for operator.
Teuchos::RCP< Epetra_MultiVector > input_col_trans
Temporary to store result of importing input into column map (transpose)
virtual int ApplyInverse(const Epetra_MultiVector &X, Epetra_MultiVector &Y) const
Returns the result of the inverse of the operator applied to a Epetra_MultiVector Input in Result as ...
bool include_mean
Flag indicating whether to include mean term.
Teuchos::RCP< Epetra_Import > col_importer_trans
Importer from range map to column map.
KLMatrixFreeOperator & operator=(const KLMatrixFreeOperator &)
Private to prohibit copying.
Teuchos::RCP< Epetra_Map > global_col_map
Stores operator column SG map.
Teuchos::RCP< const Epetra_Map > domain_base_map
Stores domain base map.
virtual void setupOperator(const Teuchos::RCP< Stokhos::EpetraOperatorOrthogPoly > &poly)
Setup operator.
virtual const Epetra_Comm & Comm() const
Returns a reference to the Epetra_Comm communicator associated with this operator.
Teuchos::Array< Teuchos::RCP< Epetra_MultiVector > > result_block
MultiVectors for each block for Apply() result.
Teuchos::RCP< Epetra_MultiVector > input_col
Temporary to store result of importing input into column map.
Teuchos::RCP< Epetra_MultiVector > tmp
Temporary multivector used in Apply()