43 #include "EpetraExt_BlockMultiVector.h" 47 const Teuchos::RCP<const EpetraExt::MultiComm>& mp_comm_,
49 const Teuchos::RCP<const Epetra_Map>& base_map_,
50 const Teuchos::RCP<const Epetra_Map>& mp_map_,
51 const Teuchos::RCP<Stokhos::AbstractPreconditionerFactory>& prec_factory_,
52 const Teuchos::RCP<Teuchos::ParameterList>& params_) :
53 label(
"Stokhos MP Block Diagonal Preconditioner"),
55 num_mp_blocks(num_mp_blocks_),
58 prec_factory(prec_factory_),
59 block_precs(num_mp_blocks),
74 TEUCHOS_TEST_FOR_EXCEPTION(prec_factory == Teuchos::null, std::logic_error,
75 "Error! setupPreconditioner() cannot be called when " <<
76 "prec_factory is null!" << std::endl);
78 Teuchos::RCP<Stokhos::ProductContainer<Epetra_Operator> > mp_ops =
80 for (
int i=0; i<num_mp_blocks; i++)
81 block_precs[i] = prec_factory->compute(mp_ops->getCoeffPtr(i));
82 if (num_mp_blocks > 0) {
83 label = std::string(
"Stokhos MP Block Diagonal Preconditioner:\n") +
84 std::string(
" ***** ") +
85 std::string(block_precs[0]->Label());
93 useTranspose = UseTheTranspose;
94 for (
int i=0; i<num_mp_blocks; i++)
95 block_precs[i]->SetUseTranspose(useTranspose);
104 EpetraExt::BlockMultiVector mp_input(
View, *base_map, Input);
105 EpetraExt::BlockMultiVector mp_result(
View, *base_map, Result);
106 for (
int i=0; i<num_mp_blocks; i++) {
107 block_precs[i]->Apply(*(mp_input.GetBlock(i)), *(mp_result.GetBlock(i)));
117 EpetraExt::BlockMultiVector mp_input(
View, *base_map, Input);
118 EpetraExt::BlockMultiVector mp_result(
View, *base_map, Result);
119 for (
int i=0; i<num_mp_blocks; i++) {
120 block_precs[i]->ApplyInverse(*(mp_input.GetBlock(i)),
121 *(mp_result.GetBlock(i)));
131 double product_nrm = 0.0;
132 for (
int i=0; i<num_mp_blocks; i++) {
133 double nrm = block_precs[i]->NormInf();
134 if (nrm > product_nrm)
146 return const_cast<char*
>(label.c_str());
160 if (num_mp_blocks == 0)
162 return block_precs[0]->HasNormInf();
virtual void setupPreconditioner(const Teuchos::RCP< Stokhos::BlockDiagonalOperator > &mp_op, const Epetra_Vector &x)
Setup preconditioner.
virtual int SetUseTranspose(bool UseTranspose)
Set to true if the transpose of the operator is requested.
virtual ~MPBlockDiagonalPreconditioner()
Destructor.
virtual const char * Label() const
Returns a character string describing the operator.
virtual double NormInf() const
Returns an approximate infinity norm of the operator matrix.
const IndexType const IndexType const IndexType const IndexType const ValueType const ValueType * x
virtual const Epetra_Comm & Comm() const
Returns a reference to the Epetra_Comm communicator associated with this operator.
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 ...
virtual const Epetra_Map & OperatorDomainMap() const
Returns the Epetra_Map object associated with the domain of this matrix operator. ...
MPBlockDiagonalPreconditioner(const Teuchos::RCP< const EpetraExt::MultiComm > &mp_comm, int num_mp_blocks, const Teuchos::RCP< const Epetra_Map > &base_map, const Teuchos::RCP< const Epetra_Map > &mp_map, const Teuchos::RCP< Stokhos::AbstractPreconditionerFactory > &prec_factory, const Teuchos::RCP< Teuchos::ParameterList > ¶ms)
Constructor.
virtual const Epetra_Map & OperatorRangeMap() const
Returns the Epetra_Map object associated with the range of this matrix operator.
virtual bool UseTranspose() const
Returns the current UseTranspose setting.
virtual bool HasNormInf() const
Returns true if the this object can provide an approximate Inf-norm, false otherwise.
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 ...