Xpetra_StridedMap_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

// WARNING: This code is experimental. Backwards compatibility should not be expected.

#ifndef XPETRA_STRIDEDMAP_DECL_HPP
#define XPETRA_STRIDEDMAP_DECL_HPP

#include <Tpetra_KokkosCompat_DefaultNode.hpp>
#include <Teuchos_Describable.hpp>

#include "Xpetra_ConfigDefs.hpp"
#include "Xpetra_Map_decl.hpp"

namespace Xpetra {

template <class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class StridedMap : public virtual 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_STRIDEDMAP_SHORT
#include "Xpetra_UseShortNamesOrdinal.hpp"

 public:


  StridedMap(UnderlyingLib xlib,
             global_size_t numGlobalElements,
             GlobalOrdinal indexBase,
             std::vector<size_t>& stridingInfo,
             const Teuchos::RCP<const Teuchos::Comm<int>>& comm,
             LocalOrdinal stridedBlockId = -1,  // FIXME (mfh 03 Sep 2014) This breaks for unsigned LocalOrdinal
             GlobalOrdinal offset        = 0,
             LocalGlobal lg              = GloballyDistributed);


  StridedMap(UnderlyingLib xlib,
             global_size_t numGlobalElements,
             size_t numLocalElements,
             GlobalOrdinal indexBase,
             std::vector<size_t>& stridingInfo,
             const Teuchos::RCP<const Teuchos::Comm<int>>& comm,
             LocalOrdinal stridedBlockId = -1,
             GlobalOrdinal offset        = 0);

  StridedMap(UnderlyingLib xlib,
             global_size_t numGlobalElements,
             const Teuchos::ArrayView<const GlobalOrdinal>& elementList,
             GlobalOrdinal indexBase,
             std::vector<size_t>& stridingInfo,
             const Teuchos::RCP<const Teuchos::Comm<int>>& comm,
             LocalOrdinal stridedBlockId = -1);

  StridedMap(const RCP<const Map>& map,
             std::vector<size_t>& stridingInfo,
             GlobalOrdinal /* indexBase */,
             LocalOrdinal stridedBlockId = -1,
             GlobalOrdinal offset        = 0);

  virtual ~StridedMap();




  std::vector<size_t> getStridingData() const;

  void setStridingData(std::vector<size_t> stridingInfo);

  size_t getFixedBlockSize() const;

  LocalOrdinal getStridedBlockId() const;

  bool isStrided() const;

  bool isBlocked() const;

  GlobalOrdinal getOffset() const;

  void setOffset(GlobalOrdinal offset);

  // returns number of strided block id which gid belongs to.
  size_t GID2StridingBlockId(GlobalOrdinal gid) const;



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

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

  local_map_type getLocalMap() const {
    return map_->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  // HAVE_XPETRA_TPETRA ELSE


  /* // function currently not needed but maybe useful
  std::vector<GlobalOrdinal> NodeId2GlobalDofIds(GlobalOrdinal nodeId) const {
    TEUCHOS_TEST_FOR_EXCEPTION(stridingInfo_.size() == 0, Exceptions::RuntimeError, "StridedMap::NodeId2GlobalDofIds:
  stridingInfo not valid: stridingInfo.size() = 0?"); std::vector<GlobalOrdinal> dofs; if(stridedBlockId_ > -1) {
        TEUCHOS_TEST_FOR_EXCEPTION(stridingInfo_[stridedBlockId_] == 0, Exceptions::RuntimeError,
  "StridedMap::NodeId2GlobalDofIds: stridingInfo not valid: stridingInfo[stridedBlockId] = 0?");

        // determine nStridedOffset
        size_t nStridedOffset = 0;
        for(int j=0; j<stridedBlockId_; j++) {
          nStridedOffset += stridingInfo_[j];
        }

        for(size_t i = 0; i<stridingInfo_[stridedBlockId_]; i++) {
          GlobalOrdinal gid =
              nodeId * Teuchos::as<GlobalOrdinal>(getFixedBlockSize()) +
              offset_ +
              Teuchos::as<GlobalOrdinal>(nStridedOffset) +
              Teuchos::as<GlobalOrdinal>(i);
          dofs.push_back(gid);
        }
    } else {
      for(size_t i = 0; i<getFixedBlockSize(); i++) {
        GlobalOrdinal gid =
            nodeId * Teuchos::as<GlobalOrdinal>(getFixedBlockSize()) +
            offset_ +
            Teuchos::as<GlobalOrdinal>(i);
        dofs.push_back(gid);
      }
    }
    return dofs;
  }*/

 private:
  virtual bool CheckConsistency();

 private:
  RCP<const Xpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> map_;

  std::vector<size_t> stridingInfo_;

  LocalOrdinal stridedBlockId_;

  GlobalOrdinal offset_;

  GlobalOrdinal indexBase_;

 public:


  global_size_t getGlobalNumElements() const;

  size_t getLocalNumElements() const;

  GlobalOrdinal getIndexBase() const;

  LocalOrdinal getMinLocalIndex() const;

  LocalOrdinal getMaxLocalIndex() const;

  GlobalOrdinal getMinGlobalIndex() const;

  GlobalOrdinal getMaxGlobalIndex() const;

  GlobalOrdinal getMinAllGlobalIndex() const;

  GlobalOrdinal getMaxAllGlobalIndex() const;

  LocalOrdinal getLocalElement(GlobalOrdinal globalIndex) const;

  GlobalOrdinal getGlobalElement(LocalOrdinal localIndex) const;

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

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

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

  global_indices_array_device_type getMyGlobalIndicesDevice() const;

  bool isNodeLocalElement(LocalOrdinal localIndex) const;

  bool isNodeGlobalElement(GlobalOrdinal globalIndex) const;

  bool isContiguous() const;

  bool isDistributed() const;


  bool isCompatible(const Map& map) const;

  bool isSameAs(const Map& map) const;

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

  RCP<const Map> removeEmptyProcesses() const;

  RCP<const Map> replaceCommWithSubset(const Teuchos::RCP<const Teuchos::Comm<int>>& newComm) const;

  std::string description() const;

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

  UnderlyingLib lib() const;

};  // StridedMap class

}  // namespace Xpetra

#define XPETRA_STRIDEDMAP_SHORT
#endif  // XPETRA_STRIDEDMAP_DECL_HPP