Tpetra_Reindex_CrsMatrix_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//          Tpetra: Templated Linear Algebra Services Package
//
// Copyright 2008 NTESS and the Tpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef TPETRA_REINDEX_CRSMATRIX_DECL_HPP
#define TPETRA_REINDEX_CRSMATRIX_DECL_HPP

 
#include <Tpetra_Transform.hpp>
#include <Tpetra_CrsMatrix.hpp>
 
namespace Tpetra {


template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
class Reindex_CrsMatrix : public ViewTransform<CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> > {
 public:
  using NewType      = typename ViewTransform<CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> >::NewType;
  using OriginalType = typename ViewTransform<CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> >::OriginalType;


  Reindex_CrsMatrix(Teuchos::RCP<Map<LocalOrdinal, GlobalOrdinal, Node> const> newRowMap);


  ~Reindex_CrsMatrix();


  NewType operator()(OriginalType const& origMatrix);
 
 private:
  Teuchos::RCP<Map<LocalOrdinal, GlobalOrdinal, Node> const> newRowMap_;
  Teuchos::RCP<Map<LocalOrdinal, GlobalOrdinal, Node> > newColMap_;
};
 
}  // namespace Tpetra
 
#endif  // TPETRA_REINDEX_CRSMATRIX_DECL_HPP