49 #ifndef XPETRA_STRIDEDMAPFACTORY_DEF_HPP
50 #define XPETRA_STRIDEDMAPFACTORY_DEF_HPP
58 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
59 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
63 GlobalOrdinal indexBase,
64 std::vector<size_t>& stridingInfo,
65 const Teuchos::RCP<
const Teuchos::Comm<int>>& comm,
66 LocalOrdinal stridedBlockId,
72 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
73 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
77 size_t numLocalElements,
78 GlobalOrdinal indexBase,
79 std::vector<size_t>& stridingInfo,
80 const Teuchos::RCP<
const Teuchos::Comm<int>>& comm,
81 LocalOrdinal stridedBlockId,
82 GlobalOrdinal offset) {
83 return rcp(
new StridedMap(lib, numGlobalElements, numLocalElements, indexBase, stridingInfo, comm, stridedBlockId, offset));
86 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
87 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
89 Build(
const RCP<const Map>& map,
90 std::vector<size_t>& stridingInfo,
91 LocalOrdinal stridedBlockId,
92 GlobalOrdinal offset) {
93 return rcp(
new StridedMap(map, stridingInfo, map->getIndexBase(), stridedBlockId, offset));
96 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
97 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
99 Build(
const RCP<const StridedMap>& map, LocalOrdinal stridedBlockId) {
101 "Xpetra::StridedMapFactory::Build: constructor expects stridedBlockId > -1.");
104 "Xpetra::StridedMapFactory::Build: constructor expects a full map (stridedBlockId == -1).");
106 std::vector<size_t> stridingInfo = map->getStridingData();
108 Teuchos::ArrayView<const GlobalOrdinal> dofGids = map->getLocalElementList();
111 size_t nStridedOffset = 0;
112 for (
int j = 0; j < map->getStridedBlockId(); j++) {
113 nStridedOffset += stridingInfo[j];
116 const size_t numMyBlockDofs = (stridingInfo[stridedBlockId] * map->getLocalNumElements()) / map->getFixedBlockSize();
118 std::vector<GlobalOrdinal> subBlockDofGids(numMyBlockDofs);
121 LocalOrdinal ind = 0;
122 for (
typename Teuchos::ArrayView<const GlobalOrdinal>::iterator it = dofGids.begin(); it != dofGids.end(); ++it) {
123 if (map->GID2StridingBlockId(*it) == Teuchos::as<size_t>(stridedBlockId)) {
124 subBlockDofGids[ind++] = *it;
128 const Teuchos::ArrayView<const GlobalOrdinal> subBlockDofGids_view(&subBlockDofGids[0], subBlockDofGids.size());
131 Teuchos::OrdinalTraits<global_size_t>::invalid(),
132 subBlockDofGids_view,
139 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
140 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
149 for (LocalOrdinal i = 0; i < N; i++)
150 newElements[i] = oldElements[i];
156 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
157 RCP<Xpetra::StridedMap<LocalOrdinal, GlobalOrdinal, Node>>
161 const Teuchos::ArrayView<const GlobalOrdinal>& elementList,
162 GlobalOrdinal indexBase,
163 std::vector<size_t>& stridingInfo,
164 const Teuchos::RCP<
const Teuchos::Comm<int>>& comm,
165 LocalOrdinal stridedBlockId,
167 return rcp(
new StridedMap(lib, numGlobalElements, elementList, indexBase, stridingInfo, comm, stridedBlockId));
172 #endif // XPETRA_STRIDEDMAPFACTORY_DEF_HPP
Teuchos::ArrayView< const GlobalOrdinal > getLocalElementList() const
Return a list of the global indices owned by this node.
LocalOrdinal getStridedBlockId() const
size_t getLocalNumElements() const
Returns the number of elements belonging to the calling node.
Exception throws to report errors in the internal logical of the program.
Teuchos::RCP< const Teuchos::Comm< int > > getComm() const
Get the Comm object for this Map.
size_t global_size_t
Global size_t object.
GlobalOrdinal getIndexBase() const
Returns the index base for this Map.
global_size_t getGlobalNumElements() const
Returns the number of elements in this Map.
std::vector< size_t > getStridingData() const
UnderlyingLib lib() const
Get the library used by this object (Tpetra or Epetra?)
#define XPETRA_MONITOR(funcName)
static RCP< Xpetra::StridedMap< LocalOrdinal, GlobalOrdinal, Node > > Build(UnderlyingLib lib, global_size_t numGlobalElements, GlobalOrdinal indexBase, std::vector< size_t > &stridingInfo, const Teuchos::RCP< const Teuchos::Comm< int >> &comm, LocalOrdinal stridedBlockId=-1, GlobalOrdinal offset=0, LocalGlobal lg=Xpetra::GloballyDistributed)
Map constructor with Xpetra-defined contiguous uniform distribution.
Class that stores a strided map.