Tpetra_EpetraRowMatrix.hpp

// @HEADER
// *****************************************************************************
//          Tpetra: Templated Linear Algebra Services Package
//
// Copyright 2008 NTESS and the Tpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef TPETRA_EPETRAROWMATRIX_HPP
#define TPETRA_EPETRAROWMATRIX_HPP
 
#include "TpetraCore_config.h"
 
#if !defined(TPETRA_ENABLE_DEPRECATED_CODE)
#error This file is deprecated due to Epetra removal and will be removed
#endif
 
#if defined(TPETRA_ENABLE_DEPRECATED_CODE) && defined(HAVE_TPETRA_EPETRA)
 
#include <Epetra_Comm.h>
#include <Epetra_BasicRowMatrix.h>
#include <Tpetra_CrsMatrix.hpp>
#include <Teuchos_TestForException.hpp>
#include <memory>  // std::shared_ptr
#include <stdexcept>
#include <type_traits>
#include <vector>
 
namespace Tpetra {
namespace Details {
 
// Epetra_MpiComm actually has reference-counting value semantics,
// just like std::shared_ptr.  We only return
// std::shared_ptr<Epetra_Comm> because Epetra_Comm is an abstract
// base class, so we must return it by pointer.
TPETRA_DEPRECATED_MSG("epetra removal")
std::shared_ptr<Epetra_Comm>
makeEpetraCommFromTeuchosComm(const Teuchos::Comm<int> &teuchosComm);
 
}  // namespace Details
}  // namespace Tpetra
 
namespace {  // (anonymous)
 
template <class EpetraGlobalOrdinalType, class TpetraMapType>
TPETRA_DEPRECATED_MSG("epetra removal")
Epetra_Map
    tpetraToEpetraMapTmpl(const TpetraMapType &tpetraMap) {
  using Teuchos::ArrayView;
  typedef typename TpetraMapType::global_ordinal_type TGO;
  typedef typename TpetraMapType::local_ordinal_type LO;
  typedef EpetraGlobalOrdinalType EGO;
 
  const TGO gblNumInds                   = static_cast<TGO>(tpetraMap.getGlobalNumElements());
  const LO lclNumInds                    = static_cast<LO>(tpetraMap.getLocalNumElements());
  ArrayView<const TGO> global_index_list = tpetraMap.getLocalElementList();
 
  std::vector<EGO> global_index_list_epetra;
  const EGO *global_index_list_epetra_ptr = NULL;
  if (std::is_same<TGO, EGO>::value) {
    global_index_list_epetra_ptr =
        reinterpret_cast<const EGO *>(global_index_list.getRawPtr());
  } else {
    global_index_list_epetra.resize(lclNumInds);
    for (LO k = 0; k < lclNumInds; ++k) {
      // TODO (mfh 11 Oct 2017) Detect overflow from TGO to EGO.
      global_index_list_epetra[k] = static_cast<EGO>(global_index_list[k]);
    }
    global_index_list_epetra_ptr = global_index_list_epetra.data();
  }
  const EGO indexBase = tpetraMap.getIndexBase();
  std::shared_ptr<Epetra_Comm> epetraComm =
      Tpetra::Details::makeEpetraCommFromTeuchosComm(*(tpetraMap.getComm()));
  // It's OK for the shared_ptr to fall out of scope.  Subclasses of
  // Epetra_Comm have reference-counted value semantics, so passing
  // the Epetra_Comm by const reference into Epetra_Map's constructor
  // is safe.
  //
  // TODO (mfh 11 Oct 2017) Detect overflow from TGO to EGO, or from
  // LO to int.
  return Epetra_Map(static_cast<EGO>(gblNumInds),
                    static_cast<int>(lclNumInds),
                    global_index_list_epetra_ptr, indexBase, *epetraComm);
}
 
}  // namespace
 
namespace Tpetra {
 
template <class TpetraMatrixType>
class
    TPETRA_DEPRECATED_MSG("epetra removal")
        EpetraRowMatrix : public Epetra_BasicRowMatrix {
 public:
  EpetraRowMatrix(const Teuchos::RCP<TpetraMatrixType> &mat, const Epetra_Comm &comm);
  virtual ~EpetraRowMatrix(){};
 
  int ExtractMyRowCopy(int MyRow, int Length, int &NumEntries, double *Values, int *Indices) const;
 
  // not implemented
  int ExtractMyEntryView(int CurEntry, double *&Value, int &RowIndex, int &ColIndex);
 
  // not implemented
  int ExtractMyEntryView(int CurEntry, double const *&Value, int &RowIndex, int &ColIndex) const;
 
  int NumMyRowEntries(int MyRow, int &NumEntries) const;
 
 private:
  Teuchos::RCP<TpetraMatrixType> tpetra_matrix_;
};  // class EpetraRowMatrix
 
template <class TpetraMatrixType>
EpetraRowMatrix<TpetraMatrixType>::EpetraRowMatrix(
    const Teuchos::RCP<TpetraMatrixType> &mat, const Epetra_Comm &comm)
  : Epetra_BasicRowMatrix(comm)
  , tpetra_matrix_(mat) {
  using Teuchos::RCP;
  typedef typename TpetraMatrixType::map_type tpetra_map_type;
#if !defined(EPETRA_NO_64BIT_GLOBAL_INDICES)
  // Prefer using Epetra64 if it is enabled.
  typedef long long EGO;
#elif !defined(EPETRA_NO_32BIT_GLOBAL_INDICES)
  // We don't have Epetra64, but we do have 32-bit indices.
  typedef int EGO;
#else
#error "Epetra was not configured correctly.  Neither 64-bit nor 32-bit indices are enabled."
#endif
  const char tfecfFuncName[] = "EpetraRowMatrix: ";
 
  TEUCHOS_TEST_FOR_EXCEPTION_CLASS_FUNC(mat.is_null(), std::invalid_argument,
                                        "The input Tpetra matrix is null.");
  TEUCHOS_TEST_FOR_EXCEPTION_CLASS_FUNC(mat->getRowMap().is_null(), std::invalid_argument,
                                        "The input Tpetra matrix's row Map is null.");
  TEUCHOS_TEST_FOR_EXCEPTION_CLASS_FUNC(mat->getColMap().is_null(), std::invalid_argument,
                                        "The input Tpetra matrix's column Map is null.");
 
  RCP<const tpetra_map_type> tpetraRowMap = mat->getRowMap();
  Epetra_Map epetraRowMap =
      tpetraToEpetraMapTmpl<EGO, tpetra_map_type>(*tpetraRowMap);
  RCP<const tpetra_map_type> tpetraColMap = mat->getColMap();
  Epetra_Map epetraColMap =
      tpetraToEpetraMapTmpl<EGO, tpetra_map_type>(*tpetraColMap);
  this->SetMaps(epetraRowMap, epetraColMap);
}
 
template <class TpetraMatrixType>
int EpetraRowMatrix<TpetraMatrixType>::ExtractMyRowCopy(int MyRow, int Length, int &NumEntries, double *Values, int *Indices) const {
  using inds_view = typename TpetraMatrixType::nonconst_local_inds_host_view_type;
  using vals_view = typename TpetraMatrixType::nonconst_values_host_view_type;
  static_assert(std::is_same<typename TpetraMatrixType::scalar_type, double>::value,
                "This code assumes that Tpetra::CrsMatrix's scalar_type is int.");
  static_assert(std::is_same<typename TpetraMatrixType::local_ordinal_type, int>::value,
                "This code assumes that Tpetra::CrsMatrix's local_ordinal_type is int.");
  inds_view IndicesView(Indices, Length);
  vals_view ValuesView(Values, Length);
  size_t num_entries = NumEntries;
  tpetra_matrix_->getLocalRowCopy(MyRow, IndicesView, ValuesView, num_entries);
  NumEntries = num_entries;
  return 0;
}
 
template <class TpetraMatrixType>
int EpetraRowMatrix<TpetraMatrixType>::ExtractMyEntryView(int CurEntry, double *&Value, int &RowIndex, int &ColIndex) {
  // not implemented
  return -1;
}
 
template <class TpetraMatrixType>
int EpetraRowMatrix<TpetraMatrixType>::ExtractMyEntryView(int CurEntry, double const *&Value, int &RowIndex, int &ColIndex) const {
  // not implemented
  return -1;
}
 
template <class TpetraMatrixType>
int EpetraRowMatrix<TpetraMatrixType>::NumMyRowEntries(int MyRow, int &NumEntries) const {
  NumEntries = tpetra_matrix_->getNumEntriesInLocalRow(MyRow);
  return 0;
}
 
}  // namespace Tpetra
 
#endif  // defined(TPETRA_ENABLE_DEPRECATED_CODE) && defined(HAVE_TPETRA_EPETRA)
 
// here is the include-guard #endif:
 
#endif