46 #ifndef MUELU_UTILITIES_DEF_HPP 47 #define MUELU_UTILITIES_DEF_HPP 49 #include <Teuchos_DefaultComm.hpp> 50 #include <Teuchos_ParameterList.hpp> 54 #ifdef HAVE_MUELU_EPETRA 56 # include "Epetra_MpiComm.h" 60 #if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_EPETRAEXT) 72 #ifdef HAVE_MUELU_TPETRA 73 #include <MatrixMarket_Tpetra.hpp> 74 #include <Tpetra_RowMatrixTransposer.hpp> 75 #include <TpetraExt_MatrixMatrix.hpp> 81 #ifdef HAVE_MUELU_EPETRA 102 #if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_ML) 103 #include <ml_operator.h> 104 #include <ml_epetra_utils.h> 109 #ifdef HAVE_MUELU_EPETRA 114 #ifdef HAVE_MUELU_EPETRA 116 template<
typename SC,
typename LO,
typename GO,
typename NO>
120 #ifdef HAVE_MUELU_EPETRA 121 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
124 if (tmpVec == Teuchos::null)
125 throw Exceptions::BadCast(
"Cast from Xpetra::MultiVector to Xpetra::EpetraMultiVector failed");
126 return tmpVec->getEpetra_MultiVector();
129 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
132 if (tmpVec == Teuchos::null)
133 throw Exceptions::BadCast(
"Cast from Xpetra::MultiVector to Xpetra::EpetraMultiVector failed");
134 return tmpVec->getEpetra_MultiVector();
137 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
143 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
149 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
152 if (crsOp == Teuchos::null)
155 if (tmp_ECrsMtx == Teuchos::null)
156 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
157 return tmp_ECrsMtx->getEpetra_CrsMatrix();
160 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
163 if (crsOp == Teuchos::null)
166 if (tmp_ECrsMtx == Teuchos::null)
167 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
168 return tmp_ECrsMtx->getEpetra_CrsMatrixNonConst();
171 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
178 }
catch (std::bad_cast) {
179 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
181 }
catch (std::bad_cast) {
186 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
193 }
catch (std::bad_cast) {
194 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
196 }
catch (std::bad_cast) {
201 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
204 if (xeMap == Teuchos::null)
205 throw Exceptions::BadCast(
"Utilities::Map2EpetraMap : Cast from Xpetra::Map to Xpetra::EpetraMap failed");
206 return xeMap->getEpetra_Map();
210 #ifdef HAVE_MUELU_TPETRA 211 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
212 RCP<const Tpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> >
215 if (tmpVec == Teuchos::null)
216 throw Exceptions::BadCast(
"Cast from Xpetra::MultiVector to Xpetra::TpetraMultiVector failed");
217 return tmpVec->getTpetra_MultiVector();
220 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
223 if (tmpVec == Teuchos::null)
224 throw Exceptions::BadCast(
"Cast from Xpetra::MultiVector to Xpetra::TpetraMultiVector failed");
225 return tmpVec->getTpetra_MultiVector();
228 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
234 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
240 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
241 const Tpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>&
247 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
251 if (crsOp == Teuchos::null)
254 if (tmp_ECrsMtx == Teuchos::null)
255 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix failed");
256 return tmp_ECrsMtx->getTpetra_CrsMatrix();
259 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
262 if (crsOp == Teuchos::null)
265 if (tmp_ECrsMtx == Teuchos::null)
266 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix failed");
267 return tmp_ECrsMtx->getTpetra_CrsMatrixNonConst();
270 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
277 }
catch (std::bad_cast) {
278 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix failed");
280 }
catch (std::bad_cast) {
285 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
292 }
catch (std::bad_cast) {
293 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix failed");
295 }
catch (std::bad_cast) {
300 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
303 if (crsOp == Teuchos::null)
306 RCP<const Xpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > crsMat = crsOp->getCrsMatrix();
308 RCP<const Xpetra::TpetraBlockCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > tmp_BlockCrs;
309 if(!tmp_Crs.is_null()) {
314 if (tmp_BlockCrs.is_null())
315 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix and Xpetra::TpetraBlockCrsMatrix failed");
316 return tmp_BlockCrs->getTpetra_BlockCrsMatrixNonConst();
320 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
323 if (crsOp == Teuchos::null)
326 RCP<const Xpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > crsMat = crsOp->getCrsMatrix();
328 RCP<const Xpetra::TpetraBlockCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > tmp_BlockCrs;
329 if(!tmp_Crs.is_null()) {
334 if (tmp_BlockCrs.is_null())
335 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix and Xpetra::TpetraBlockCrsMatrix failed");
336 return tmp_BlockCrs->getTpetra_BlockCrsMatrixNonConst();
341 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
344 if (tmp_TMap == Teuchos::null)
345 throw Exceptions::BadCast(
"Utilities::Map2TpetraMap : Cast from Xpetra::Map to Xpetra::TpetraMap failed");
346 return tmp_TMap->getTpetra_Map();
350 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
353 bool doOptimizeStorage)
355 Scalar one = Teuchos::ScalarTraits<Scalar>::one();
356 Teuchos::ArrayRCP<Scalar> sv(scalingVector.size());
358 for (
int i = 0; i < scalingVector.size(); ++i)
359 sv[i] = one / scalingVector[i];
361 for (
int i = 0; i < scalingVector.size(); ++i)
362 sv[i] = scalingVector[i];
367 MyOldScaleMatrix_Tpetra(Op, sv, doFillComplete, doOptimizeStorage);
371 MyOldScaleMatrix_Epetra(Op, sv, doFillComplete, doOptimizeStorage);
380 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
385 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
388 bool doOptimizeStorage)
390 #ifdef HAVE_MUELU_TPETRA 392 Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& tpOp = Op2NonConstTpetraCrs(Op);
394 const RCP<const Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node> > rowMap = tpOp.getRowMap();
395 const RCP<const Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node> > domainMap = tpOp.getDomainMap();
396 const RCP<const Tpetra::Map<LocalOrdinal,GlobalOrdinal,Node> > rangeMap = tpOp.getRangeMap();
398 size_t maxRowSize = tpOp.getNodeMaxNumRowEntries();
399 if (maxRowSize == Teuchos::as<size_t>(-1))
402 std::vector<Scalar> scaledVals(maxRowSize);
403 if (tpOp.isFillComplete())
407 Teuchos::ArrayView<const LocalOrdinal> cols;
408 Teuchos::ArrayView<const Scalar> vals;
410 for (
size_t i = 0; i < rowMap->getNodeNumElements(); ++i) {
411 tpOp.getLocalRowView(i, cols, vals);
412 size_t nnz = tpOp.getNumEntriesInLocalRow(i);
413 if (nnz > maxRowSize) {
415 scaledVals.resize(maxRowSize);
417 for (
size_t j = 0; j < nnz; ++j)
418 scaledVals[j] = vals[j]*scalingVector[i];
421 Teuchos::ArrayView<const Scalar> valview(&scaledVals[0], nnz);
422 tpOp.replaceLocalValues(i, cols, valview);
427 Teuchos::ArrayView<const GlobalOrdinal> cols;
428 Teuchos::ArrayView<const Scalar> vals;
430 for (
size_t i = 0; i < rowMap->getNodeNumElements(); ++i) {
431 GlobalOrdinal gid = rowMap->getGlobalElement(i);
432 tpOp.getGlobalRowView(gid, cols, vals);
433 size_t nnz = tpOp.getNumEntriesInGlobalRow(gid);
434 if (nnz > maxRowSize) {
436 scaledVals.resize(maxRowSize);
439 for (
size_t j = 0; j < nnz; ++j)
440 scaledVals[j] = vals[j]*scalingVector[i];
443 Teuchos::ArrayView<const Scalar> valview(&scaledVals[0], nnz);
444 tpOp.replaceGlobalValues(gid, cols, valview);
449 if (doFillComplete) {
450 if (domainMap == Teuchos::null || rangeMap == Teuchos::null)
451 throw Exceptions::RuntimeError(
"In Utilities::Scaling: cannot fillComplete because the domain and/or range map hasn't been defined");
453 RCP<Teuchos::ParameterList> params = rcp(
new Teuchos::ParameterList());
454 params->set(
"Optimize Storage", doOptimizeStorage);
455 params->set(
"No Nonlocal Changes",
true);
466 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
467 RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> >
470 #if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_EPETRAEXT) 471 std::string TorE =
"epetra";
473 std::string TorE =
"tpetra";
476 #if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_EPETRAEXT) 485 #ifdef HAVE_MUELU_TPETRA 486 if (TorE ==
"tpetra") {
490 RCP<Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > A;
491 Tpetra::RowMatrixTransposer<Scalar, LocalOrdinal, GlobalOrdinal, Node> transposer(rcpFromRef(tpetraOp),label);
492 A = transposer.createTranspose();
497 if (!AAAA->isFillComplete())
500 if (Op.
IsView(
"stridedMaps"))
501 AAAA->CreateView(
"stridedMaps", Teuchos::rcpFromRef(Op),
true);
505 }
catch (std::exception& e) {
506 std::cout <<
"threw exception '" << e.what() <<
"'" << std::endl;
513 std::cout <<
"Utilities::Transpose() not implemented for Epetra" << std::endl;
514 return Teuchos::null;
519 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
520 RCP<Xpetra::MultiVector<double,LocalOrdinal,GlobalOrdinal,Node> >
524 RCP<Xpetra::MultiVector<double,LocalOrdinal,GlobalOrdinal,Node> > coordinates = Teuchos::null;
527 if(paramList.isParameter (
"Coordinates") ==
false)
530 #if defined(HAVE_MUELU_TPETRA) 536 #if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_FLOAT) 537 typedef Tpetra::MultiVector<float, LocalOrdinal, GlobalOrdinal, Node> tfMV;
538 RCP<tfMV> floatCoords = Teuchos::null;
544 #if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_DOUBLE) 545 typedef Tpetra::MultiVector<double, LocalOrdinal, GlobalOrdinal, Node> tdMV;
546 RCP<tdMV> doubleCoords = Teuchos::null;
547 if (paramList.isType<RCP<tdMV> >(
"Coordinates")) {
549 doubleCoords = paramList.get<RCP<tdMV> >(
"Coordinates");
550 paramList.remove(
"Coordinates");
552 #if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_FLOAT) 553 else if (paramList.isType<RCP<tfMV> >(
"Coordinates")) {
555 floatCoords = paramList.get<RCP<tfMV> >(
"Coordinates");
556 paramList.remove(
"Coordinates");
557 doubleCoords = rcp(
new tdMV(floatCoords->getMap(), floatCoords->getNumVectors()));
562 if(doubleCoords != Teuchos::null) {
565 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(coordinates));
566 TEUCHOS_TEST_FOR_EXCEPT(doubleCoords->getNumVectors() != coordinates->getNumVectors());
571 throw Exceptions::RuntimeError(
"ExtractCoordinatesFromParameterList: The coordinates vector in parameter list is expected to be a Tpetra multivector with SC=double or float.");
573 #endif // endif HAVE_TPETRA 580 #define MUELU_UTILITIES_SHORT 581 #endif // MUELU_UTILITIES_DEF_HPP Exception indicating invalid cast attempted.
static const RCP< const Tpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > Map2TpetraMap(const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > &map)
RCP< Epetra_CrsMatrix > getEpetra_CrsMatrixNonConst() const
static RCP< const Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > MV2TpetraMV(RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > const vec)
Helper utility to pull out the underlying Tpetra objects from an Xpetra object.
static void MyOldScaleMatrix(Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Op, const Teuchos::ArrayRCP< const Scalar > &scalingVector, bool doInverse=true, bool doFillComplete=true, bool doOptimizeStorage=true)
static RCP< const Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op2TpetraCrs(RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
static const Epetra_Map & Map2EpetraMap(const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > &map)
static RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Transpose(Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Op, bool optimizeTranspose=false, const std::string &label=std::string())
virtual const RCP< const Map > & getRowMap() const
static RCP< Epetra_CrsMatrix > Op2NonConstEpetraCrs(RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
static RCP< const Epetra_MultiVector > MV2EpetraMV(RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > const vec)
Helper utility to pull out the underlying Epetra objects from an Xpetra object.
Namespace for MueLu classes and methods.
static RCP< Xpetra::MultiVector< double, LocalOrdinal, GlobalOrdinal, Node > > ExtractCoordinatesFromParameterList(ParameterList ¶mList)
static RCP< Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > MV2NonConstTpetraMV(RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > vec)
virtual void fillComplete(const RCP< const Map > &domainMap, const RCP< const Map > &rangeMap, const RCP< ParameterList > ¶ms=null)=0
RCP< Epetra_MultiVector > getEpetra_MultiVector() const
static RCP< Epetra_MultiVector > MV2NonConstEpetraMV(RCP< Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > vec)
static RCP< Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op2NonConstTpetraCrs(RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
RCP< Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsMatrixNonConst() const
static RCP< Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > MV2NonConstTpetraMV2(Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &vec)
RCP< CrsMatrix > getCrsMatrix() const
static RCP< const Tpetra::RowMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op2TpetraRow(RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
RCP< Xpetra::CrsMatrixWrap< SC, LO, GO, NO > > Convert_Epetra_CrsMatrix_ToXpetra_CrsMatrixWrap(RCP< Epetra_CrsMatrix > &epAB)
RCP< const Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsMatrix() const
void deep_copy(const DynRankView< DT, DP... > &dst, typename ViewTraits< DT, DP... >::const_value_type &value, typename std::enable_if< std::is_same< typename ViewTraits< DT, DP... >::specialize, void >::value >::type *=0)
static void MyOldScaleMatrix_Epetra(Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Op, const Teuchos::ArrayRCP< Scalar > &scalingVector, bool doFillComplete, bool doOptimizeStorage)
RCP< const Epetra_CrsMatrix > getEpetra_CrsMatrix() const
static RCP< const Epetra_CrsMatrix > Op2EpetraCrs(RCP< const Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
virtual bool isLocallyIndexed() const=0
virtual Teuchos::RCP< const Map > getRangeMap() const=0
bool IsView(const viewLabel_t viewLabel) const
Exception throws to report errors in the internal logical of the program.
RCP< Tpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_MultiVector() const
static RCP< Tpetra::RowMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op2NonConstTpetraRow(RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Op)
static void MyOldScaleMatrix_Tpetra(Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Op, const Teuchos::ArrayRCP< Scalar > &scalingVector, bool doFillComplete, bool doOptimizeStorage)
virtual Teuchos::RCP< const Map > getDomainMap() const=0