Xpetra_BlockedMap_decl.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_XPETRA_SUP_BLOCKEDMAP_XPETRA_BLOCKEDMAP_DECL_HPP_
#define PACKAGES_XPETRA_SUP_BLOCKEDMAP_XPETRA_BLOCKEDMAP_DECL_HPP_

#include "Xpetra_ConfigDefs.hpp"

#include "Xpetra_Import.hpp"
#include "Xpetra_Map_decl.hpp"

namespace Xpetra {

template <class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class BlockedMap : public Map<LocalOrdinal, GlobalOrdinal, Node> {
 public:
  typedef LocalOrdinal local_ordinal_type;
  typedef GlobalOrdinal global_ordinal_type;
  typedef Node node_type;
  typedef typename Map<LocalOrdinal, GlobalOrdinal, Node>::global_indices_array_device_type global_indices_array_device_type;

 private:
#undef XPETRA_BLOCKEDMAP_SHORT
#include "Xpetra_UseShortNamesOrdinal.hpp"

 public:



  BlockedMap();

  BlockedMap(const RCP<const Map>& fullmap, const std::vector<RCP<const Map>>& maps, bool bThyraMode = false);

  BlockedMap(const std::vector<RCP<const Map>>& maps, const std::vector<RCP<const Map>>& thyramaps);

  BlockedMap(const BlockedMap& input);

  virtual ~BlockedMap();



  virtual global_size_t getGlobalNumElements() const;

  virtual size_t getLocalNumElements() const;

  virtual GlobalOrdinal getIndexBase() const;

  virtual LocalOrdinal getMinLocalIndex() const;

  virtual LocalOrdinal getMaxLocalIndex() const;

  virtual GlobalOrdinal getMinGlobalIndex() const;

  virtual GlobalOrdinal getMaxGlobalIndex() const;

  virtual GlobalOrdinal getMinAllGlobalIndex() const;

  virtual GlobalOrdinal getMaxAllGlobalIndex() const;

  virtual LocalOrdinal getLocalElement(GlobalOrdinal globalIndex) const;

  virtual GlobalOrdinal getGlobalElement(LocalOrdinal localIndex) const;

  virtual LookupStatus getRemoteIndexList(const Teuchos::ArrayView<const GlobalOrdinal>& /* GIDList    */,
                                          const Teuchos::ArrayView<int>& /* nodeIDList */,
                                          const Teuchos::ArrayView<LocalOrdinal>& /* LIDList    */) const;

  virtual LookupStatus getRemoteIndexList(const Teuchos::ArrayView<const GlobalOrdinal>& /* GIDList    */,
                                          const Teuchos::ArrayView<int>& /* nodeIDList */) const;

  virtual Teuchos::ArrayView<const GlobalOrdinal> getLocalElementList() const;

  virtual global_indices_array_device_type getMyGlobalIndicesDevice() const;




  virtual bool isNodeLocalElement(LocalOrdinal localIndex) const;

  virtual bool isNodeGlobalElement(GlobalOrdinal globalIndex) const;

  virtual bool isContiguous() const;

  virtual bool isDistributed() const;

  virtual bool isCompatible(const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>& map) const;

  virtual bool isSameAs(const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>& map) const;




  virtual Teuchos::RCP<const Teuchos::Comm<int>> getComm() const;


  BlockedMap<LocalOrdinal, GlobalOrdinal, Node>& operator=(const BlockedMap& rhs);



  /*virtual size_t getLocalLength() const {
    throw Xpetra::Exceptions::RuntimeError("BlockedMap::getLocalLength: routine not implemented.");
    return 0;
  }*/

  /*virtual global_size_t getGlobalLength() const {
    throw Xpetra::Exceptions::RuntimeError("BlockedMap::getGlobalLength: routine not implemented.");
    return 0;
  }*/

  virtual bool getThyraMode() const;




  virtual RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> removeEmptyProcesses() const;

  virtual RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
  replaceCommWithSubset(const Teuchos::RCP<const Teuchos::Comm<int>>& /* newComm */) const;




  virtual UnderlyingLib lib() const;

  // TODO: find a better solution for this hack
  // The problem is that EpetraMap, TpetraMap and StridedMap all inherit Map. To have proper toEpetra() we
  // need to understand the type of underlying matrix. But in src/Map we have no knowledge of StridedMaps, so
  // we cannot check for it by casting. This function allows us to avoid the restriction, as StridedMap redefines
  // it to return the base map.
  virtual RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
  getMap() const;


  size_t getNumMaps() const;

  const RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
  getMap(size_t i, bool bThyraMode = false) const;

  const RCP<Xpetra::Import<LocalOrdinal, GlobalOrdinal, Node>>
  getImporter(size_t i) const;

  const RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
  getFullMap() const;

  size_t getMapIndexForGID(GlobalOrdinal gid) const;

#ifdef HAVE_XPETRA_TPETRA
  using local_map_type = typename Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>::local_map_type;

  local_map_type getLocalMap() const { return fullmap_->getLocalMap(); }

#else  // HAVE_XPETRA_TPETRA
#ifdef __GNUC__
#warning "Xpetra Kokkos interface for CrsMatrix is enabled (HAVE_XPETRA_KOKKOS_REFACTOR) but Tpetra is disabled. The Kokkos interface needs Tpetra to be enabled, too."
#endif  // __GNUC__
#endif  // #else !HAVE_XPETRA_TPETRA




  virtual std::string description() const;

  virtual void describe(Teuchos::FancyOStream& out, const Teuchos::EVerbosityLevel verbLevel = Teuchos::Describable::verbLevel_default) const;


 protected:
  virtual void assign(const BlockedMap& input);

  static Teuchos::RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>
  concatenateMaps(const std::vector<Teuchos::RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>>>& subMaps);

 private:
  bool CheckConsistency() const;

 private:
  RCP<const Map> fullmap_;
  std::vector<RCP<const Map>> maps_;
  std::vector<RCP<Import>> importers_;
  bool bThyraMode_;                        //< boolean flag: use Thyra numbering for local sub-block maps. default = false (for Xpetra mode)
  std::vector<RCP<const Map>> thyraMaps_;  //< store Thyra-style numbering maps here in Thyra mode. In Xpetra mode this vector is empty.
};                                         // BlockedMap class

}  // namespace Xpetra

#define XPETRA_BLOCKEDMAP_SHORT

#endif /* PACKAGES_XPETRA_SUP_BLOCKEDMAP_XPETRA_BLOCKEDMAP_DECL_HPP_ */