Xpetra_TpetraRowMatrix.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//             Xpetra: A linear algebra interface package
//
// Copyright 2012 NTESS and the Xpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef XPETRA_TPETRAROWMATRIX_HPP
#define XPETRA_TPETRAROWMATRIX_HPP
 
/* this file is automatically generated - do not edit (see script/tpetra.py) */
 
// WARNING: This code is experimental. Backwards compatibility should not be expected.
 
#include <Teuchos_Describable.hpp>
#include "Xpetra_ConfigDefs.hpp"
#include "Xpetra_Map.hpp"
#include "Xpetra_Vector.hpp"
 
#include "Xpetra_TpetraMap.hpp"
 
#include "Tpetra_RowMatrix.hpp"
 
#include "Xpetra_RowMatrix.hpp"
#include "Xpetra_Exceptions.hpp"
 
namespace Xpetra {
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class TpetraRowMatrix
  : virtual public RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
 public:
 
 
  virtual ~TpetraRowMatrix() = default;
 
 
 
 
  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getRowMap() const {
    XPETRA_MONITOR("TpetraRowMatrix::getRowMap");
    return toXpetra(mtx_->getRowMap());
  }
 
  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getColMap() const {
    XPETRA_MONITOR("TpetraRowMatrix::getColMap");
    return toXpetra(mtx_->getColMap());
  }
 
  Tpetra::global_size_t getGlobalNumRows() const {
    XPETRA_MONITOR("TpetraRowMatrix::getGlobalNumRows");
    return mtx_->getGlobalNumRows();
  }
 
  Tpetra::global_size_t getGlobalNumCols() const {
    XPETRA_MONITOR("TpetraRowMatrix::getGlobalNumCols");
    return mtx_->getGlobalNumCols();
  }
 
  size_t getLocalNumRows() const {
    XPETRA_MONITOR("TpetraRowMatrix::getLocalNumRows");
    return mtx_->getLocalNumRows();
  }
 
  size_t getLocalNumCols() const {
    XPETRA_MONITOR("TpetraRowMatrix::getLocalNumCols");
    return mtx_->getLocalNumCols();
  }
 
  Tpetra::global_size_t getGlobalNumEntries() const {
    XPETRA_MONITOR("TpetraRowMatrix::getGlobalNumEntries");
    return mtx_->getGlobalNumEntries();
  }
 
  size_t getNodeNumEntries() const {
    XPETRA_MONITOR("TpetraRowMatrix::getNodeNumEntries");
    return mtx_->getNodeNumEntries();
  }
 
  size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const {
    XPETRA_MONITOR("TpetraRowMatrix::getNumEntriesInLocalRow");
    return mtx_->getNumEntriesInLocalRow(localRow);
  }
 
  size_t getGlobalMaxNumRowEntries() const {
    XPETRA_MONITOR("TpetraRowMatrix::getGlobalMaxNumRowEntries");
    return mtx_->getGlobalMaxNumRowEntries();
  }
 
  size_t getLocalMaxNumRowEntries() const {
    XPETRA_MONITOR("TpetraRowMatrix::getLocalMaxNumRowEntries");
    return mtx_->getLocalMaxNumRowEntries();
  }
 
  bool isLocallyIndexed() const {
    XPETRA_MONITOR("TpetraRowMatrix::isLocallyIndexed");
    return mtx_->isLocallyIndexed();
  }
 
  bool isGloballyIndexed() const {
    XPETRA_MONITOR("TpetraRowMatrix::isGloballyIndexed");
    return mtx_->isGloballyIndexed();
  }
 
  bool isFillComplete() const {
    XPETRA_MONITOR("TpetraRowMatrix::isFillComplete");
    return mtx_->isFillComplete();
  }
 
  bool supportsRowViews() const {
    XPETRA_MONITOR("TpetraRowMatrix::supportsRowViews");
    return mtx_->supportsRowViews();
  }
 
 
 
 
  void getLocalRowCopy(LocalOrdinal LocalRow, const Teuchos::ArrayView<LocalOrdinal> &Indices, const Teuchos::ArrayView<Scalar> &Values, size_t &NumEntries) const {
    XPETRA_MONITOR("TpetraRowMatrix::getLocalRowCopy");
    typename Tpetra::RowGraph<LocalOrdinal, GlobalOrdinal, Node>::nonconst_local_inds_host_view_type indices("indices", Indices.size());
    typename Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type values("values", Values.size());
 
    mtx_->getLocalRowCopy(LocalRow, indices, values, NumEntries);
    for (size_t i = 0; i < NumEntries; ++i) {
      Indices[i] = indices(i);
      Values[i]  = values(i);
    }
  }
 
  void getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView<const GlobalOrdinal> &indices, ArrayView<const Scalar> &values) const {
    XPETRA_MONITOR("TpetraRowMatrix::getGlobalRowView");
    typename Tpetra::RowGraph<LocalOrdinal, GlobalOrdinal, Node>::global_inds_host_view_type k_indices;
    typename Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type k_values;
 
    mtx_->getGlobalRowView(GlobalRow, k_indices, k_values);
    indices = ArrayView<const GlobalOrdinal>(k_indices.data(), k_indices.extent(0));
    values  = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(k_values.data()), k_values.extent(0));
  }
 
  void getLocalRowView(LocalOrdinal LocalRow, ArrayView<const LocalOrdinal> &indices, ArrayView<const Scalar> &values) const {
    XPETRA_MONITOR("TpetraRowMatrix::getLocalRowView");
    typename Tpetra::RowGraph<LocalOrdinal, GlobalOrdinal, Node>::local_inds_host_view_type k_indices;
    typename Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type k_values;
 
    mtx_->getLocalRowView(LocalRow, k_indices, k_values);
    indices = ArrayView<const LocalOrdinal>(k_indices.data(), k_indices.extent(0));
    values  = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(k_values.data()), k_values.extent(0));
  }
 
  void getLocalDiagCopy(Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node> &diag) const { TEUCHOS_TEST_FOR_EXCEPTION(1, Xpetra::Exceptions::NotImplemented, "TODO"); }
  //{ XPETRA_MONITOR("TpetraRowMatrix::getLocalDiagCopy"); mtx_->getLocalDiagCopy(diag); }
 
 
 
 
  typename ScalarTraits<Scalar>::magnitudeType getFrobeniusNorm() const {
    XPETRA_MONITOR("TpetraRowMatrix::getFrobeniusNorm");
    return mtx_->getFrobeniusNorm();
  }
 
 
 
 
  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getDomainMap() const {
    XPETRA_MONITOR("TpetraRowMatrix::getDomainMap");
    return toXpetra(mtx_->getDomainMap());
  }
 
  const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getRangeMap() const {
    XPETRA_MONITOR("TpetraRowMatrix::getRangeMap");
    return toXpetra(mtx_->getRangeMap());
  }
 
  void apply(const MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> &X, MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> &Y, Teuchos::ETransp mode = Teuchos::NO_TRANS, Scalar alpha = Teuchos::ScalarTraits<Scalar>::one(), Scalar beta = Teuchos::ScalarTraits<Scalar>::zero()) const { TEUCHOS_TEST_FOR_EXCEPTION(1, Xpetra::Exceptions::NotImplemented, "TODO"); }
  //{ XPETRA_MONITOR("TpetraRowMatrix::apply"); mtx_->apply(X, Y, mode, alpha, beta); }
 
 
 
 
  TpetraRowMatrix() = default;
 
  TpetraRowMatrix(const Teuchos::RCP<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > &mtx)
    : mtx_(mtx) {}
 
  void setTpetra_RowMatrix(const Teuchos::RCP<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > &mtx) { mtx_ = mtx; }
 
  RCP<const Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > getTpetra_RowMatrix() const { return mtx_; }
 
  RCP<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > getTpetra_RowMatrixNonConst() const { return mtx_; }  // TODO: remove
 
 
 private:
  RCP<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > mtx_;
 
};  // TpetraRowMatrix class
 
}  // namespace Xpetra
 
#define XPETRA_TPETRAROWMATRIX_SHORT
#endif  // XPETRA_TPETRAROWMATRIX_HPP