10#ifndef XPETRA_TPETRACRSGRAPH_DEF_HPP
11#define XPETRA_TPETRACRSGRAPH_DEF_HPP
15#include "Tpetra_CrsGraph.hpp"
20#include "Xpetra_TpetraMap.hpp"
21#include "Xpetra_TpetraImport.hpp"
22#include "Xpetra_TpetraExport.hpp"
26template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
28 : graph_(
Teuchos::
rcp(new
Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>(
toTpetra(rowMap), maxNumEntriesPerRow, params))) {}
30template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
34template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
38template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
42template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
55 graph_ = Tpetra::importAndFillCompleteCrsGraph<MyTpetraCrsGraph>(v,
toTpetra(importer), myDomainMap, myRangeMap, params);
56 bool restrictComm =
false;
57 if (!params.
is_null()) restrictComm = params->
get(
"Restrict Communicator", restrictComm);
58 if (restrictComm && graph_->getRowMap().is_null()) graph_ = Teuchos::null;
61template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
65 const typename local_graph_type::row_map_type &rowPointers,
66 const typename local_graph_type::entries_type::non_const_type &columnIndices,
70template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
74 const local_graph_type &lclGraph,
78template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
88template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
97template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
100template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
103 graph_->insertGlobalIndices(globalRow, indices);
106template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
109 graph_->insertLocalIndices(localRow, indices);
112template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
115 graph_->removeLocalIndices(localRow);
118template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
121 rowptr.
resize(getLocalNumRows() + 1);
122 colind.
resize(numNonZeros);
125template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
128 graph_->setAllIndices(rowptr, colind);
131template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
134 rowptr = Kokkos::Compat::persistingView(graph_->getLocalRowPtrsHost());
135 colind = Kokkos::Compat::persistingView(graph_->getLocalIndicesHost());
138template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
144template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
147 graph_->fillComplete(params);
150template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
161 if (importer != Teuchos::null) {
163 myImport = tImporter.getTpetra_Import();
165 if (exporter != Teuchos::null) {
167 myExport = tExporter.getTpetra_Export();
170 graph_->expertStaticFillComplete(
toTpetra(domainMap),
toTpetra(rangeMap), myImport, myExport, params);
173template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
176 return graph_->getComm();
179template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
182 return toXpetra(graph_->getRowMap());
185template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
188 return toXpetra(graph_->getColMap());
191template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
194 return toXpetra(graph_->getDomainMap());
197template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
200 return toXpetra(graph_->getRangeMap());
203template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
206 return toXpetra(graph_->getImporter());
209template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
212 return toXpetra(graph_->getExporter());
215template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
218 return graph_->getGlobalNumRows();
221template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
224 return graph_->getGlobalNumCols();
227template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
230 return graph_->getLocalNumRows();
233template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
236 return graph_->getLocalNumCols();
239template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
242 return graph_->getIndexBase();
245template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
248 return graph_->getGlobalNumEntries();
251template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
254 return graph_->getLocalNumEntries();
257template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
260 return graph_->getNumEntriesInGlobalRow(globalRow);
263template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
266 return graph_->getNumEntriesInLocalRow(localRow);
269template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
271 XPETRA_MONITOR(
"TpetraCrsGraph::getNumAllocatedEntriesInGlobalRow");
272 return graph_->getNumAllocatedEntriesInGlobalRow(globalRow);
275template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
277 XPETRA_MONITOR(
"TpetraCrsGraph::getNumAllocatedEntriesInLocalRow");
278 return graph_->getNumAllocatedEntriesInLocalRow(localRow);
281template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
284 return graph_->getGlobalMaxNumRowEntries();
287template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
290 return graph_->getLocalMaxNumRowEntries();
293template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
296 return graph_->hasColMap();
299template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
302 return graph_->isLocallyIndexed();
305template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
308 return graph_->isGloballyIndexed();
311template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
314 return graph_->isFillComplete();
317template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
320 return graph_->isStorageOptimized();
323template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
331template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
339template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
344template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
346 return getTpetra_CrsGraph()->getLocalGraphDevice();
349template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
351 getTpetra_CrsGraph()->getLocalDiagOffsets(offsets);
354template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
357 graph_->computeGlobalConstants();
360template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
363 return graph_->description();
366template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
369 graph_->describe(out, verbLevel);
372template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
375 return Kokkos::Compat::persistingView(graph_->getLocalRowPtrsHost());
378template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
381 return rcp(
new TpetraMap(graph_->getMap()));
384template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
396template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
406template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
417template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
428template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
432template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
#define XPETRA_MONITOR(funcName)
#define XPETRA_DYNAMIC_CAST(type, obj, newObj, exceptionMsg)
void resize(const size_type n, const T &val=T())
void getGlobalRowView(const global_ordinal_type gblRow, global_inds_host_view_type &gblColInds) const override
void getLocalRowView(const LocalOrdinal lclRow, local_inds_host_view_type &lclColInds) const override
virtual local_graph_type::host_mirror_type getLocalGraphHost() const =0
Get the local graph.
KokkosSparse::StaticCrsGraph< LocalOrdinal, Kokkos::LayoutLeft, device_type, void, size_t > local_graph_type
void setAllIndices(const ArrayRCP< size_t > &rowptr, const ArrayRCP< LocalOrdinal > &colind)
Sets the 1D pointer arrays of the graph.
void fillComplete(const RCP< const Map > &domainMap, const RCP< const Map > &rangeMap, const RCP< ParameterList > ¶ms=null)
Signal that data entry is complete, specifying domain and range maps.
RCP< const Comm< int > > getComm() const
Returns the communicator.
size_t getLocalMaxNumRowEntries() const
Maximum number of entries in all rows owned by the calling process.
RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsGraph() const
Get the underlying Tpetra graph.
void insertLocalIndices(const LocalOrdinal localRow, const ArrayView< const LocalOrdinal > &indices)
Insert local indices into the graph.
void allocateAllIndices(size_t numNonZeros, ArrayRCP< size_t > &rowptr, ArrayRCP< LocalOrdinal > &colind)
Allocates the 1D pointer arrays of the graph.
void insertGlobalIndices(GlobalOrdinal globalRow, const ArrayView< const GlobalOrdinal > &indices)
Insert global indices into the graph.
global_size_t getGlobalNumCols() const
Returns the number of global columns in the graph.
void expertStaticFillComplete(const Teuchos::RCP< const map_type > &domainMap, const Teuchos::RCP< const map_type > &rangeMap, const Teuchos::RCP< const Import > &importer=Teuchos::null, const Teuchos::RCP< const Export > &exporter=Teuchos::null, const Teuchos::RCP< Teuchos::ParameterList > ¶ms=Teuchos::null)
Expert version of fillComplete.
TpetraCrsGraph(const RCP< const Map > &rowMap, size_t maxNumEntriesPerRow, const RCP< ParameterList > ¶ms=null)
Constructor specifying fixed number of entries for each row.
size_t getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const
Returns the current number of entries on this node in the specified global row.
RCP< const Import > getImporter() const
Returns the importer associated with this graph.
void removeLocalIndices(LocalOrdinal localRow)
Remove all graph indices from the specified local row.
typename Xpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node >::local_graph_type local_graph_type
bool isLocallyIndexed() const
Whether column indices are stored using local indices on the calling process.
size_t getNumAllocatedEntriesInLocalRow(LocalOrdinal localRow) const
Returns the current number of allocated entries on this node in the specified local row.
virtual ~TpetraCrsGraph()
Destructor.
size_t getGlobalMaxNumRowEntries() const
Maximum number of entries in all rows over all processes.
size_t getLocalNumCols() const
Returns the number of columns connected to the locally owned rows of this graph.
size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const
Returns the current number of entries on this node in the specified local row.
bool isFillComplete() const
Whether fillComplete() has been called and the graph is in compute mode.
local_graph_type getLocalGraphDevice() const
Access the local KokkosSparse::StaticCrsGraph data for device use.
size_t getNumAllocatedEntriesInGlobalRow(GlobalOrdinal globalRow) const
Returns the current number of allocated entries for this node in the specified global row .
local_graph_type::host_mirror_type getLocalGraphHost() const
Access the local KokkosSparse::StaticCrsGraph data for host use.
size_t getLocalNumRows() const
Returns the number of graph rows owned on the calling node.
void computeGlobalConstants()
Force the computation of global constants if we don't have them.
bool hasColMap() const
Whether the graph has a column Map.
void doExport(const DistObject< GlobalOrdinal, LocalOrdinal, GlobalOrdinal, Node > &dest, const Import &importer, CombineMode CM)
Export.
void getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView< const GlobalOrdinal > &Indices) const
Return a const, nonpersisting view of global indices in the given row.
RCP< const Map > getRowMap() const
Returns the Map that describes the row distribution in this graph.
RCP< const Map > getRangeMap() const
Returns the Map associated with the domain of this graph.
size_t getLocalNumEntries() const
Returns the local number of entries in the graph.
bool isStorageOptimized() const
Returns true if storage has been optimized.
void getAllIndices(ArrayRCP< const size_t > &rowptr, ArrayRCP< const LocalOrdinal > &colind) const
Gets the 1D pointer arrays of the graph.
void getLocalDiagOffsets(const Kokkos::View< size_t *, typename Node::device_type, Kokkos::MemoryUnmanaged > &offsets) const
Get offsets of the diagonal entries in the matrix.
void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Print the object with some verbosity level to an FancyOStream object.
void doImport(const DistObject< GlobalOrdinal, LocalOrdinal, GlobalOrdinal, Node > &source, const Import &importer, CombineMode CM)
Import.
Teuchos::RCP< const Map > getMap() const
Implements DistObject interface.
bool isGloballyIndexed() const
Whether column indices are stored using global indices on the calling process.
std::string description() const
Return a simple one-line description of this object.
GlobalOrdinal getIndexBase() const
Returns the index base for global indices for this graph.
RCP< const Export > getExporter() const
Returns the exporter associated with this graph.
ArrayRCP< const size_t > getNodeRowPtrs() const
Get an ArrayRCP of the row-offsets.
RCP< const Map > getDomainMap() const
Returns the Map associated with the domain of this graph.
void getLocalRowView(LocalOrdinal LocalRow, ArrayView< const LocalOrdinal > &indices) const
Return a const, nonpersisting view of local indices in the given row.
global_size_t getGlobalNumEntries() const
Returns the global number of entries in the graph.
global_size_t getGlobalNumRows() const
Returns the number of global rows in the graph.
RCP< const Map > getColMap() const
Returns the Map that describes the column distribution in this graph.
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toXpetra(RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > graph)
size_t global_size_t
Global size_t object.
RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toTpetra(const RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > &graph)
CombineMode
Xpetra::Combine Mode enumerable type.