46 #ifndef THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP 47 #define THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP 51 #ifdef HAVE_MUELU_STRATIMIKOS 54 #include "Thyra_DefaultPreconditioner.hpp" 55 #include "Thyra_TpetraLinearOp.hpp" 56 #include "Thyra_TpetraThyraWrappers.hpp" 58 #include "Teuchos_Ptr.hpp" 59 #include "Teuchos_TestForException.hpp" 60 #include "Teuchos_Assert.hpp" 61 #include "Teuchos_Time.hpp" 63 #include <Xpetra_CrsMatrixWrap.hpp> 64 #include <Xpetra_CrsMatrix.hpp> 65 #include <Xpetra_Matrix.hpp> 66 #include <Xpetra_ThyraUtils.hpp> 68 #include <MueLu_Hierarchy.hpp> 70 #include <MueLu_HierarchyHelpers.hpp> 71 #include <MueLu_ParameterListInterpreter.hpp> 72 #include <MueLu_MLParameterListInterpreter.hpp> 74 #ifdef HAVE_MUELU_TPETRA 75 #include <MueLu_TpetraOperator.hpp> 77 #ifdef HAVE_MUELU_EPETRA 85 using Teuchos::ParameterList;
90 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
92 paramList_(rcp(new ParameterList()))
97 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
99 const RCP<const LinearOpBase<Scalar> > fwdOp = fwdOpSrc.getOp();
101 #ifdef HAVE_MUELU_TPETRA 102 if (Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::isTpetra(fwdOp))
return true;
105 #ifdef HAVE_MUELU_EPETRA 106 if (Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node>::isEpetra(fwdOp))
return true;
112 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
114 return Teuchos::rcp(
new DefaultPreconditioner<Scalar>);
117 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
119 initializePrec(
const RCP<
const LinearOpSourceBase<Scalar> >& fwdOpSrc, PreconditionerBase<Scalar>* prec,
const ESupportSolveUse supportSolveUse)
const {
120 using Teuchos::rcp_dynamic_cast;
123 typedef Xpetra::Operator<Scalar, LocalOrdinal, GlobalOrdinal, Node> XpOp;
124 typedef Xpetra::ThyraUtils<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpThyUtils;
125 typedef Xpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpCrsMat;
126 typedef Xpetra::Matrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpMat;
127 typedef Xpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> XpMultVec;
128 typedef Xpetra::MultiVector<double,LocalOrdinal,GlobalOrdinal,Node> XpMultVecDouble;
129 typedef Thyra::LinearOpBase<Scalar> ThyLinOpBase;
130 #ifdef HAVE_MUELU_TPETRA 132 typedef Tpetra::Operator<Scalar,LocalOrdinal,GlobalOrdinal,Node> TpOp;
133 typedef Thyra::TpetraLinearOp<Scalar,LocalOrdinal,GlobalOrdinal,Node> ThyTpLinOp;
135 #ifdef HAVE_MUELU_EPETRA 137 typedef Thyra::EpetraLinearOp ThyEpLinOp;
142 TEUCHOS_ASSERT(Teuchos::nonnull(fwdOpSrc));
144 TEUCHOS_ASSERT(prec);
150 const RCP<const ThyLinOpBase> fwdOp = fwdOpSrc->getOp();
151 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(fwdOp));
154 bool bIsEpetra = XpThyUtils::isEpetra(fwdOp);
155 bool bIsTpetra = XpThyUtils::isTpetra(fwdOp);
156 TEUCHOS_TEST_FOR_EXCEPT(bIsEpetra == bIsTpetra);
158 RCP<const XpCrsMat > xpetraFwdCrsMat = XpThyUtils::toXpetra(fwdOp);
159 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMat));
162 RCP<XpCrsMat> xpetraFwdCrsMatNonConst = Teuchos::rcp_const_cast<XpCrsMat>(xpetraFwdCrsMat);
163 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(xpetraFwdCrsMatNonConst));
166 RCP<XpMat> A = rcp(
new Xpetra::CrsMatrixWrap<Scalar,LocalOrdinal,GlobalOrdinal,Node>(xpetraFwdCrsMatNonConst));
167 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(A));
170 const Teuchos::Ptr<DefaultPreconditioner<Scalar> > defaultPrec = Teuchos::ptr(
dynamic_cast<DefaultPreconditioner<Scalar> *
>(prec));
171 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(defaultPrec));
174 RCP<ThyLinOpBase> thyra_precOp = Teuchos::null;
175 thyra_precOp = rcp_dynamic_cast<Thyra::LinearOpBase<Scalar> >(defaultPrec->getNonconstUnspecifiedPrecOp(),
true);
178 RCP<MueLu::Hierarchy<Scalar,LocalOrdinal,GlobalOrdinal,Node> > H = Teuchos::null;
183 const bool startingOver = (thyra_precOp.is_null() || !paramList.isParameter(
"reuse: type") || paramList.get<std::string>(
"reuse: type") ==
"none");
185 if (startingOver ==
true) {
187 Teuchos::RCP<XpMultVecDouble> coordinates = Teuchos::null;
188 #ifdef HAVE_MUELU_TPETRA 192 # if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_MUELU_INST_FLOAT_INT_INT) 193 typedef Tpetra::MultiVector<float, LocalOrdinal, GlobalOrdinal, Node> tfMV;
194 RCP<tfMV> floatCoords = Teuchos::null;
196 typedef Tpetra::MultiVector<double, LocalOrdinal, GlobalOrdinal, Node> tdMV;
197 RCP<tdMV> doubleCoords = Teuchos::null;
198 if (paramList.isType<RCP<tdMV> >(
"Coordinates")) {
199 doubleCoords = paramList.get<RCP<tdMV> >(
"Coordinates");
200 paramList.remove(
"Coordinates");
202 # if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_MUELU_INST_FLOAT_INT_INT) 203 else if (paramList.isType<RCP<tfMV> >(
"Coordinates")) {
204 floatCoords = paramList.get<RCP<tfMV> >(
"Coordinates");
205 paramList.remove(
"Coordinates");
206 doubleCoords = rcp(
new tdMV(floatCoords->getMap(), floatCoords->getNumVectors()));
207 deep_copy(*doubleCoords, *floatCoords);
210 if(doubleCoords != Teuchos::null) {
211 coordinates = MueLu::TpetraMultiVector_To_XpetraMultiVector<double,LocalOrdinal,GlobalOrdinal,Node>(doubleCoords);
212 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(coordinates));
213 TEUCHOS_TEST_FOR_EXCEPT(doubleCoords->getNumVectors() != coordinates->getNumVectors());
216 #endif // HAVE_MUELU_TPETRA 218 #ifdef HAVE_MUELU_EPETRA 220 RCP<Epetra_MultiVector> doubleCoords;
221 if (paramList.isType<RCP<Epetra_MultiVector> >(
"Coordinates")) {
222 doubleCoords = paramList.get<RCP<Epetra_MultiVector> >(
"Coordinates");
223 paramList.remove(
"Coordinates");
224 RCP<Xpetra::EpetraMultiVector> epCoordinates = Teuchos::rcp(
new Xpetra::EpetraMultiVector(doubleCoords));
225 RCP<Xpetra::MultiVector<double,int,int,Node> > epCoordinatesMult = rcp_dynamic_cast<Xpetra::MultiVector<double,int,int,Node> >(epCoordinates);
226 coordinates = rcp_dynamic_cast<Xpetra::MultiVector<double,LocalOrdinal,GlobalOrdinal,Node> >(epCoordinatesMult);
227 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(coordinates));
228 TEUCHOS_TEST_FOR_EXCEPT(doubleCoords->NumVectors() != Teuchos::as<int>(coordinates->getNumVectors()));
231 #endif // HAVE_MUELU_EPETRA 235 RCP<XpMultVec> nullspace = Teuchos::null;
236 #ifdef HAVE_MUELU_TPETRA 238 typedef Tpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> tMV;
239 RCP<tMV> tpetra_nullspace = Teuchos::null;
240 if (paramList.isType<Teuchos::RCP<tMV> >(
"Nullspace")) {
241 tpetra_nullspace = paramList.get<RCP<tMV> >(
"Nullspace");
242 paramList.remove(
"Nullspace");
243 nullspace = MueLu::TpetraMultiVector_To_XpetraMultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>(tpetra_nullspace);
244 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(nullspace));
248 #ifdef HAVE_MUELU_EPETRA 250 RCP<Epetra_MultiVector> epetra_nullspace = Teuchos::null;
251 if (paramList.isType<RCP<Epetra_MultiVector> >(
"Nullspace")) {
252 epetra_nullspace = paramList.get<RCP<Epetra_MultiVector> >(
"Nullspace");
253 paramList.remove(
"Nullspace");
254 RCP<Xpetra::EpetraMultiVector> xpEpNullspace = Teuchos::rcp(
new Xpetra::EpetraMultiVector(epetra_nullspace));
255 RCP<Xpetra::MultiVector<double,int,int,Node> > xpEpNullspaceMult = rcp_dynamic_cast<Xpetra::MultiVector<double,int,int,Node> >(xpEpNullspace);
256 nullspace = rcp_dynamic_cast<XpMultVec>(xpEpNullspaceMult);
257 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(nullspace));
269 #ifdef HAVE_MUELU_TPETRA 272 RCP<ThyTpLinOp> tpetr_precOp = rcp_dynamic_cast<ThyTpLinOp>(thyra_precOp);
273 RCP<MueTpOp> muelu_precOp = rcp_dynamic_cast<MueTpOp>(tpetr_precOp->getTpetraOperator(),
true);
275 H = muelu_precOp->GetHierarchy();
278 #ifdef HAVE_MUELU_TPETRA 280 RCP<ThyEpLinOp> epetr_precOp = rcp_dynamic_cast<ThyEpLinOp>(thyra_precOp);
281 RCP<MueEpOp> muelu_precOp = rcp_dynamic_cast<MueEpOp>(epetr_precOp->epetra_op(),
true);
287 "MueLu::ThyraPreconditionerFactory: Hierarchy has no levels in it");
289 "MueLu::ThyraPreconditionerFactory: Hierarchy has no fine level operator");
290 RCP<MueLu::Level> level0 = H->
GetLevel(0);
291 RCP<XpOp> O0 = level0->Get<RCP<XpOp> >(
"A");
292 RCP<XpMat> A0 = rcp_dynamic_cast<XpMat>(O0);
298 A->SetFixedBlockSize(A0->GetFixedBlockSize());
308 RCP<ThyLinOpBase > thyraPrecOp = Teuchos::null;
309 #ifdef HAVE_MUELU_TPETRA 311 RCP<MueTpOp> muelu_tpetraOp = rcp(
new MueTpOp(H));
312 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(muelu_tpetraOp));
313 thyraPrecOp = Thyra::createLinearOp(RCP<TpOp>(muelu_tpetraOp));
317 #ifdef HAVE_MUELU_EPETRA 319 RCP<MueLu::Hierarchy<double,int,int> > epetraH =
321 RCP<MueEpOp> muelu_epetraOp = rcp(
new MueEpOp(epetraH));
322 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(muelu_epetraOp));
324 set_extra_data(fwdOp,
"IFPF::fwdOp", Teuchos::inOutArg(muelu_epetraOp), Teuchos::POST_DESTROY,
false);
325 RCP<ThyEpLinOp> thyra_epetraOp = Thyra::nonconstEpetraLinearOp(muelu_epetraOp, NOTRANS, EPETRA_OP_APPLY_APPLY_INVERSE, EPETRA_OP_ADJOINT_UNSUPPORTED);
326 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(thyra_epetraOp));
327 thyraPrecOp = rcp_dynamic_cast<ThyLinOpBase>(thyra_epetraOp);
330 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(thyraPrecOp));
332 defaultPrec->initializeUnspecified(thyraPrecOp);
336 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
338 CreateXpetraPreconditioner(Teuchos::RCP<Xpetra::Matrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > op,
const Teuchos::ParameterList& inParamList, Teuchos::RCP<Xpetra::MultiVector<double, LocalOrdinal, GlobalOrdinal, Node> > coords, Teuchos::RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> > nullspace)
const {
343 Teuchos::ParameterList paramList = inParamList;
345 bool hasParamList = paramList.numParams();
347 RCP<HierarchyManager> mueLuFactory;
349 std::string syntaxStr =
"parameterlist: syntax";
350 if (hasParamList && paramList.isParameter(syntaxStr) && paramList.get<std::string>(syntaxStr) ==
"ml") {
351 paramList.remove(syntaxStr);
357 RCP<Hierarchy> H = mueLuFactory->CreateHierarchy();
358 H->setlib(op->getDomainMap()->lib());
362 H->GetLevel(0)->Set(
"A", op);
365 if (coords != Teuchos::null) {
366 H->GetLevel(0)->Set(
"Coordinates", coords);
370 if (nullspace == Teuchos::null) {
371 int nPDE = MueLu::MasterList::getDefault<int>(
"number of equations");
372 if (paramList.isSublist(
"Matrix")) {
374 const Teuchos::ParameterList& operatorList = paramList.sublist(
"Matrix");
375 if (operatorList.isParameter(
"PDE equations"))
376 nPDE = operatorList.get<
int>(
"PDE equations");
378 }
else if (paramList.isParameter(
"number of equations")) {
380 nPDE = paramList.get<
int>(
"number of equations");
383 nullspace = Xpetra::MultiVectorFactory<Scalar,LocalOrdinal,GlobalOrdinal,Node>::Build(op->getDomainMap(), nPDE);
385 nullspace->putScalar(Teuchos::ScalarTraits<Scalar>::one());
388 for (
int i = 0; i < nPDE; i++) {
389 Teuchos::ArrayRCP<Scalar> nsData = nullspace->getDataNonConst(i);
390 for (
int j = 0; j < nsData.size(); j++) {
391 GlobalOrdinal GID = op->getDomainMap()->getGlobalElement(j) - op->getDomainMap()->getIndexBase();
393 if ((GID-i) % nPDE == 0)
394 nsData[j] = Teuchos::ScalarTraits<Scalar>::one();
399 H->GetLevel(0)->Set(
"Nullspace", nullspace);
402 Teuchos::ParameterList nonSerialList,dummyList;
406 mueLuFactory->SetupHierarchy(*H);
411 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
413 uninitializePrec(PreconditionerBase<Scalar>* prec, RCP<
const LinearOpSourceBase<Scalar> >* fwdOp, ESupportSolveUse* supportSolveUse)
const {
414 TEUCHOS_ASSERT(prec);
417 const Teuchos::Ptr<DefaultPreconditioner<Scalar> > defaultPrec = Teuchos::ptr(
dynamic_cast<DefaultPreconditioner<Scalar> *
>(prec));
418 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(defaultPrec));
422 *fwdOp = Teuchos::null;
425 if (supportSolveUse) {
427 *supportSolveUse = Thyra::SUPPORT_SOLVE_UNSPECIFIED;
430 defaultPrec->uninitialize();
435 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
437 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(paramList));
441 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
446 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
448 RCP<ParameterList> savedParamList =
paramList_;
450 return savedParamList;
453 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
458 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
460 static RCP<const ParameterList> validPL;
462 if (Teuchos::is_null(validPL))
463 validPL = rcp(
new ParameterList());
470 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
472 return "Thyra::MueLuPreconditionerFactory";
476 #endif // HAVE_MUELU_STRATIMIKOS 478 #endif // ifdef THYRA_MUELU_PRECONDITIONER_FACTORY_DEF_HPP RCP< Level > & GetLevel(const int levelID=0)
Retrieve a certain level from hierarchy.
static void AddNonSerializableDataToHierarchy(HierarchyManager &HM, Hierarchy &H, const ParameterList &nonSerialList)
void initializePrec(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOp, PreconditionerBase< Scalar > *prec, const ESupportSolveUse supportSolveUse) const
void uninitializePrec(PreconditionerBase< Scalar > *prec, Teuchos::RCP< const LinearOpSourceBase< Scalar > > *fwdOp, ESupportSolveUse *supportSolveUse) const
Teuchos::RCP< Teuchos::ParameterList > paramList_
Turns a MueLu::Hierarchy into a Epetra_Operator. It allows MueLu to be used as a preconditioner for A...
bool isCompatible(const LinearOpSourceBase< Scalar > &fwdOp) const
Teuchos::RCP< const Teuchos::ParameterList > getParameterList() const
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > ¶mList)
MueLuPreconditionerFactory()
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
std::string description() const
Class that accepts ML-style parameters and builds a MueLu preconditioner. This interpreter uses the s...
Teuchos::RCP< MueLu::Hierarchy< Scalar, LocalOrdinal, GlobalOrdinal, Node > > CreateXpetraPreconditioner(Teuchos::RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > op, const Teuchos::ParameterList ¶mList, Teuchos::RCP< Xpetra::MultiVector< double, LocalOrdinal, GlobalOrdinal, Node > > coords, Teuchos::RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > nullspace) const
Teuchos::RCP< PreconditionerBase< Scalar > > createPrec() 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()
Provides methods to build a multigrid hierarchy and apply multigrid cycles.
long ExtractNonSerializableData(const Teuchos::ParameterList &inList, Teuchos::ParameterList &serialList, Teuchos::ParameterList &nonSerialList)