MueLu_ImportUtils.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 PACKAGES_MUELU_IMPORT_UTILS_HPP_
#define PACKAGES_MUELU_IMPORT_UTILS_HPP_
 
#include "Xpetra_ConfigDefs.hpp"
#include "Xpetra_Exceptions.hpp"
#include "Xpetra_Map.hpp"  // definition of UnderlyingLib
#include "Xpetra_Import.hpp"
#include "Teuchos_Array.hpp"
#include "Teuchos_ArrayView.hpp"
 
#include <utility>
 
#include "Xpetra_TpetraImport.hpp"
#include "Tpetra_Import_Util.hpp"
 
namespace MueLu {

template <class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class ImportUtils {
#undef MUELU_IMPORTUTILS_SHORT
 
 public:
  void
  getPidGidPairs(const Xpetra::Import<LocalOrdinal, GlobalOrdinal, Node>& Importer,
                 Teuchos::Array<std::pair<int, GlobalOrdinal> >& gpids,
                 bool use_minus_one_for_local) {
    Xpetra::UnderlyingLib lib = Importer.getSourceMap()->lib();
    if (lib == Xpetra::UseTpetra) {
      Tpetra::Import_Util::getPidGidPairs(Xpetra::toTpetra(Importer), gpids, use_minus_one_for_local);
    }
  }

  void
  getPids(const Xpetra::Import<LocalOrdinal, GlobalOrdinal, Node>& Importer,
          Teuchos::Array<int>& pids,
          bool use_minus_one_for_local) {
    Xpetra::UnderlyingLib lib = Importer.getSourceMap()->lib();
    if (lib == Xpetra::UseTpetra) {
      Tpetra::Import_Util::getPids(Xpetra::toTpetra(Importer), pids, use_minus_one_for_local);
    }
  }

  // Like the above, but without the resize
  void
  getPids(const Xpetra::Import<LocalOrdinal, GlobalOrdinal, Node>& Importer,
          Teuchos::ArrayView<int>& pids,
          bool use_minus_one_for_local) {
    Xpetra::UnderlyingLib lib = Importer.getSourceMap()->lib();
    if (lib == Xpetra::UseTpetra) {
      Tpetra::Import_Util::getPids(Xpetra::toTpetra(Importer), pids, use_minus_one_for_local);
    }
  }

  void
  getRemotePIDs(const Xpetra::Import<LocalOrdinal, GlobalOrdinal, Node>& Importer,
                Teuchos::Array<int>& RemotePIDs) {
    Xpetra::UnderlyingLib lib = Importer.getSourceMap()->lib();
    if (lib == Xpetra::UseTpetra) {
      Tpetra::Import_Util::getRemotePIDs(Xpetra::toTpetra(Importer), RemotePIDs);
    }
  }
 
};  // end class ImportUtils
 
}  // namespace MueLu
 
#define MUELU_IMPORTUTILS_SHORT
 
#endif  // PACKAGES_MUELU_IMPORT_UTILS_HPP_