46 #ifndef THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP 47 #define THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP 51 #ifdef HAVE_MUELU_STRATIMIKOS 57 using Teuchos::ParameterList;
62 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
64 paramList_(rcp(new ParameterList()))
69 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
71 const RCP<const LinearOpBase<Scalar> > fwdOp = fwdOpSrc.getOp();
73 #ifdef HAVE_MUELU_TPETRA 74 if (Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::isTpetra(fwdOp))
return true;
77 if (Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::isBlockedOperator(fwdOp))
return true;
83 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
85 return Teuchos::rcp(
new DefaultPreconditioner<Scalar>);
88 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
90 initializePrec(
const RCP<
const LinearOpSourceBase<Scalar> >& fwdOpSrc, PreconditionerBase<Scalar>* prec,
const ESupportSolveUse supportSolveUse)
const {
91 using Teuchos::rcp_dynamic_cast;
94 typedef Xpetra::Map<LocalOrdinal,GlobalOrdinal,Node> XpMap;
95 typedef Xpetra::Operator<Scalar, LocalOrdinal, GlobalOrdinal, Node> XpOp;
96 typedef Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpThyUtils;
97 typedef Xpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpCrsMat;
98 typedef Xpetra::BlockedCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpBlockedCrsMat;
99 typedef Xpetra::Matrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpMat;
100 typedef Xpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpMultVec;
101 typedef Xpetra::MultiVector<double,LocalOrdinal,GlobalOrdinal,Node> XpMultVecDouble;
102 typedef Thyra::LinearOpBase<Scalar> ThyLinOpBase;
103 #ifdef HAVE_MUELU_TPETRA 105 typedef Tpetra::Operator<Scalar,LocalOrdinal,GlobalOrdinal,Node> TpOp;
106 typedef Thyra::TpetraLinearOp<Scalar,LocalOrdinal,GlobalOrdinal,Node> ThyTpLinOp;
110 TEUCHOS_ASSERT(Teuchos::nonnull(fwdOpSrc));
111 TEUCHOS_ASSERT(this->isCompatible(*fwdOpSrc));
112 TEUCHOS_ASSERT(prec);
115 ParameterList paramList = *paramList_;
118 const RCP<const ThyLinOpBase> fwdOp = fwdOpSrc->getOp();
119 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(fwdOp));
122 bool bIsEpetra = XpThyUtils::isEpetra(fwdOp);
123 bool bIsTpetra = XpThyUtils::isTpetra(fwdOp);
124 bool bIsBlocked = XpThyUtils::isBlockedOperator(fwdOp);
125 TEUCHOS_TEST_FOR_EXCEPT((bIsEpetra ==
true && bIsTpetra ==
true));
126 TEUCHOS_TEST_FOR_EXCEPT((bIsEpetra == bIsTpetra) && bIsBlocked ==
false);
127 TEUCHOS_TEST_FOR_EXCEPT((bIsEpetra != bIsTpetra) && bIsBlocked ==
true);
129 RCP<XpMat> A = Teuchos::null;
131 Teuchos::RCP<const Thyra::BlockedLinearOpBase<Scalar> > ThyBlockedOp =
132 Teuchos::rcp_dynamic_cast<
const Thyra::BlockedLinearOpBase<Scalar> >(fwdOp);
133 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(ThyBlockedOp));
135 TEUCHOS_TEST_FOR_EXCEPT(ThyBlockedOp->blockExists(0,0)==
false);
137 Teuchos::RCP<const LinearOpBase<Scalar> > b00 = ThyBlockedOp->getBlock(0,0);
138 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(b00));
140 RCP<const XpCrsMat > xpetraFwdCrsMat00 = XpThyUtils::toXpetra(b00);
141 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMat00));
144 RCP<XpCrsMat> xpetraFwdCrsMatNonConst00 = Teuchos::rcp_const_cast<XpCrsMat>(xpetraFwdCrsMat00);
145 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMatNonConst00));
148 RCP<XpMat> A00 = rcp(
new Xpetra::CrsMatrixWrap<Scalar,LocalOrdinal,GlobalOrdinal,Node>(xpetraFwdCrsMatNonConst00));
149 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(A00));
151 RCP<const XpMap> rowmap00 = A00->getRowMap();
152 RCP< const Teuchos::Comm< int > > comm = rowmap00->getComm();
155 RCP<XpBlockedCrsMat> bMat = Teuchos::rcp(
new XpBlockedCrsMat(ThyBlockedOp, comm));
156 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(bMat));
161 RCP<const XpCrsMat > xpetraFwdCrsMat = XpThyUtils::toXpetra(fwdOp);
162 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMat));
165 RCP<XpCrsMat> xpetraFwdCrsMatNonConst = Teuchos::rcp_const_cast<XpCrsMat>(xpetraFwdCrsMat);
166 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMatNonConst));
169 A = rcp(
new Xpetra::CrsMatrixWrap<Scalar,LocalOrdinal,GlobalOrdinal,Node>(xpetraFwdCrsMatNonConst));
171 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(A));
174 const Teuchos::Ptr<DefaultPreconditioner<Scalar> > defaultPrec = Teuchos::ptr(
dynamic_cast<DefaultPreconditioner<Scalar> *
>(prec));
175 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(defaultPrec));
178 RCP<ThyLinOpBase> thyra_precOp = Teuchos::null;
179 thyra_precOp = rcp_dynamic_cast<Thyra::LinearOpBase<Scalar> >(defaultPrec->getNonconstUnspecifiedPrecOp(),
true);
182 RCP<MueLu::Hierarchy<Scalar,LocalOrdinal,GlobalOrdinal,Node> > H = Teuchos::null;
187 const bool startingOver = (thyra_precOp.is_null() || !paramList.isParameter(
"reuse: type") || paramList.get<std::string>(
"reuse: type") ==
"none");
189 if (startingOver ==
true) {
191 Teuchos::RCP<XpMultVecDouble> coordinates = Teuchos::null;
195 RCP<XpMultVec> nullspace = Teuchos::null;
196 #ifdef HAVE_MUELU_TPETRA 198 typedef Tpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> tMV;
199 RCP<tMV> tpetra_nullspace = Teuchos::null;
200 if (paramList.isType<Teuchos::RCP<tMV> >(
"Nullspace")) {
201 tpetra_nullspace = paramList.get<RCP<tMV> >(
"Nullspace");
202 paramList.remove(
"Nullspace");
203 nullspace = MueLu::TpetraMultiVector_To_XpetraMultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>(tpetra_nullspace);
204 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(nullspace));
215 #if defined(HAVE_MUELU_TPETRA) 218 RCP<ThyTpLinOp> tpetr_precOp = rcp_dynamic_cast<ThyTpLinOp>(thyra_precOp);
219 RCP<MueTpOp> muelu_precOp = rcp_dynamic_cast<MueTpOp>(tpetr_precOp->getTpetraOperator(),
true);
221 H = muelu_precOp->GetHierarchy();
227 "Thyra::MueLuPreconditionerFactory: Hierarchy has no levels in it");
229 "Thyra::MueLuPreconditionerFactory: Hierarchy has no fine level operator");
230 RCP<MueLu::Level> level0 = H->GetLevel(0);
231 RCP<XpOp> O0 = level0->Get<RCP<XpOp> >(
"A");
232 RCP<XpMat> A0 = rcp_dynamic_cast<XpMat>(O0);
238 A->SetFixedBlockSize(A0->GetFixedBlockSize());
248 RCP<ThyLinOpBase > thyraPrecOp = Teuchos::null;
249 #if defined(HAVE_MUELU_TPETRA) 251 RCP<MueTpOp> muelu_tpetraOp = rcp(
new MueTpOp(H));
252 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(muelu_tpetraOp));
253 RCP<TpOp> tpOp = Teuchos::rcp_dynamic_cast<TpOp>(muelu_tpetraOp);
254 thyraPrecOp = Thyra::createLinearOp<Scalar, LocalOrdinal, GlobalOrdinal, Node>(tpOp);
259 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::nonnull(thyraPrecOp));
263 const RCP<MueXpOp> muelu_xpetraOp = rcp(
new MueXpOp(H));
265 RCP<const VectorSpaceBase<Scalar> > thyraRangeSpace = Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::toThyra(muelu_xpetraOp->getRangeMap());
266 RCP<const VectorSpaceBase<Scalar> > thyraDomainSpace = Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::toThyra(muelu_xpetraOp->getDomainMap());
268 RCP <Xpetra::Operator<Scalar, LocalOrdinal, GlobalOrdinal, Node> > xpOp = Teuchos::rcp_dynamic_cast<Xpetra::Operator<Scalar,LocalOrdinal,GlobalOrdinal,Node> >(muelu_xpetraOp);
269 thyraPrecOp = Thyra::xpetraLinearOp<Scalar, LocalOrdinal, GlobalOrdinal, Node>(thyraRangeSpace, thyraDomainSpace,xpOp);
272 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(thyraPrecOp));
274 defaultPrec->initializeUnspecified(thyraPrecOp);
278 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
280 uninitializePrec(PreconditionerBase<Scalar>* prec, RCP<
const LinearOpSourceBase<Scalar> >* fwdOp, ESupportSolveUse* supportSolveUse)
const {
281 TEUCHOS_ASSERT(prec);
284 const Teuchos::Ptr<DefaultPreconditioner<Scalar> > defaultPrec = Teuchos::ptr(
dynamic_cast<DefaultPreconditioner<Scalar> *
>(prec));
285 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(defaultPrec));
289 *fwdOp = Teuchos::null;
292 if (supportSolveUse) {
294 *supportSolveUse = Thyra::SUPPORT_SOLVE_UNSPECIFIED;
297 defaultPrec->uninitialize();
302 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
304 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(paramList));
305 paramList_ = paramList;
308 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
313 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
315 RCP<ParameterList> savedParamList = paramList_;
316 paramList_ = Teuchos::null;
317 return savedParamList;
320 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
325 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
327 static RCP<const ParameterList> validPL;
329 if (Teuchos::is_null(validPL))
330 validPL = rcp(
new ParameterList());
336 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
338 return "Thyra::MueLuPreconditionerFactory";
342 #endif // HAVE_MUELU_STRATIMIKOS 344 #endif // ifdef THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP
void uninitializePrec(PreconditionerBase< Scalar > *prec, Teuchos::RCP< const LinearOpSourceBase< Scalar > > *fwdOp, ESupportSolveUse *supportSolveUse) const
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
Teuchos::RCP< MueLu::Hierarchy< Scalar, LocalOrdinal, GlobalOrdinal, Node > > CreateXpetraPreconditioner(Teuchos::RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > op, const Teuchos::ParameterList &inParamList, Teuchos::RCP< Xpetra::MultiVector< double, LocalOrdinal, GlobalOrdinal, Node > > coords=Teuchos::null, Teuchos::RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > nullspace=Teuchos::null)
Helper function to create a MueLu preconditioner that can be used by Xpetra.Given an Xpetra::Matrix...
static RCP< Xpetra::MultiVector< double, LocalOrdinal, GlobalOrdinal, Node > > ExtractCoordinatesFromParameterList(ParameterList ¶mList)
void initializePrec(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOp, PreconditionerBase< Scalar > *prec, const ESupportSolveUse supportSolveUse) const
bool isCompatible(const LinearOpSourceBase< Scalar > &fwdOp) const
Teuchos::RCP< PreconditionerBase< Scalar > > createPrec() const
std::string description() const
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > ¶mList)
MueLuPreconditionerFactory()
Teuchos::RCP< const Teuchos::ParameterList > getParameterList() const
Wraps an existing MueLu::Hierarchy as a Tpetra::Operator.
Exception throws to report errors in the internal logical of the program.
Teuchos::RCP< Teuchos::ParameterList > getNonconstParameterList()
Teuchos::RCP< Teuchos::ParameterList > unsetParameterList()
Wraps an existing MueLu::Hierarchy as a Xpetra::Operator.