52 #ifndef _ZOLTAN2_MODELHELPERS_HPP_
53 #define _ZOLTAN2_MODELHELPERS_HPP_
59 template <
typename User>
60 RCP<Tpetra::CrsMatrix<int,
65 const RCP<
const Comm<int> > comm,
73 typedef int nonzero_t;
74 typedef Tpetra::CrsMatrix<nonzero_t,lno_t,gno_t,node_t> sparse_matrix_type;
75 typedef Tpetra::Map<lno_t, gno_t, node_t> map_type;
77 const GST dummy = Teuchos::OrdinalTraits<GST>::invalid ();
80 if (ia->availAdjs(sourcetarget, through)) {
88 const offset_t *offsets=NULL;
89 const gno_t *adjacencyIds=NULL;
90 ia->getAdjsView(sourcetarget, through, offsets, adjacencyIds);
92 const gno_t *Ids=NULL;
93 ia->getIDsViewOf(sourcetarget, Ids);
95 const gno_t *throughIds=NULL;
96 ia->getIDsViewOf(through, throughIds);
98 size_t LocalNumIDs = ia->getLocalNumOf(sourcetarget);
104 RCP<const map_type> sourcetargetMapG;
105 RCP<const map_type> throughMapG;
108 size_t LocalNumOfThrough = ia->getLocalNumOf(through);
113 min[0] = std::numeric_limits<gno_t>::max();
114 for (
size_t i = 0; i < LocalNumIDs; ++i) {
115 if (Ids[i] < min[0]) {
118 size_t ncols = offsets[i+1] - offsets[i];
119 if (ncols > maxcols) maxcols = ncols;
123 min[1] = std::numeric_limits<gno_t>::max();
124 for (
size_t i = 0; i < LocalNumOfThrough; ++i) {
125 if (throughIds[i] < min[1]) {
126 min[1] = throughIds[i];
131 Teuchos::reduceAll<int, gno_t>(*comm, Teuchos::REDUCE_MIN, 2, min, gmin);
134 ArrayView<const gno_t> sourceTargetGIDs(Ids, LocalNumIDs);
135 sourcetargetMapG = rcp(
new map_type(dummy,
136 sourceTargetGIDs, gmin[0], comm));
150 ArrayView<const gno_t> throughGIDs(throughIds, LocalNumOfThrough);
151 throughMapG = rcp (
new map_type(dummy,
152 throughGIDs, gmin[1], comm));
155 RCP<const map_type> oneToOneTMap =
156 Tpetra::createOneToOne<lno_t, gno_t, node_t>(throughMapG);
162 RCP<sparse_matrix_type> adjsMatrix;
165 Array<size_t> rowlens(LocalNumIDs);
166 for (
size_t localElement = 0; localElement < LocalNumIDs; localElement++)
167 rowlens[localElement] = offsets[localElement+1] - offsets[localElement];
168 adjsMatrix = rcp (
new sparse_matrix_type (sourcetargetMapG,
171 Array<nonzero_t> justOneA(maxcols, 1);
172 ArrayView<const gno_t> adjacencyIdsAV(adjacencyIds, offsets[LocalNumIDs]);
174 for (
size_t localElement=0; localElement<LocalNumIDs; ++localElement){
176 size_t ncols = offsets[localElement+1] - offsets[localElement];
177 adjsMatrix->insertGlobalValues(Ids[localElement],
178 adjacencyIdsAV(offsets[localElement], ncols),
183 adjsMatrix->fillComplete (oneToOneTMap,
184 adjsMatrix->getRowMap());
187 RCP<sparse_matrix_type> secondAdjs =
188 rcp (
new sparse_matrix_type(adjsMatrix->getRowMap(),0));
189 Tpetra::MatrixMatrix::Multiply(*adjsMatrix,
false,*adjsMatrix,
193 return RCP<sparse_matrix_type>();
196 template <
typename User>
199 const RCP<
const Comm<int> > comm,
210 typedef int nonzero_t;
211 typedef Tpetra::CrsMatrix<nonzero_t,lno_t,gno_t,node_t> sparse_matrix_type;
216 size_t LocalNumIDs = ia->getLocalNumOf(sourcetarget);
217 lno_t *start =
new lno_t [LocalNumIDs+1];
219 if (secondAdjs!=RCP<sparse_matrix_type>()) {
223 gno_t
const *Ids=NULL;
224 ia->getIDsViewOf(sourcetarget, Ids);
226 std::vector<gno_t> adj;
228 for (
size_t localElement=0; localElement<LocalNumIDs; ++localElement){
229 start[localElement] = nadj;
230 const gno_t globalRow = Ids[localElement];
231 size_t NumEntries = secondAdjs->getNumEntriesInGlobalRow (globalRow);
232 typename sparse_matrix_type::nonconst_global_inds_host_view_type Indices(
"Indices", NumEntries);
233 typename sparse_matrix_type::nonconst_values_host_view_type Values(
"Values", NumEntries);
234 secondAdjs->getGlobalRowCopy (globalRow,Indices,Values,NumEntries);
236 for (
size_t j = 0; j < NumEntries; ++j) {
237 if(globalRow != Indices[j]) {
238 adj.push_back(Indices[j]);
245 start[LocalNumIDs] = nadj;
247 gno_t *adj_ =
new gno_t [nadj];
249 for (
size_t i=0; i < nadj; i++) {
253 offsets = arcp<offset_t>(start, 0, LocalNumIDs+1,
true);
254 adjacencyIds = arcp<gno_t>(adj_, 0, nadj,
true);
258 for (
size_t i = 0; i <= LocalNumIDs; i++)
261 offsets = arcp<offset_t>(start, 0, LocalNumIDs+1,
true);
262 adjacencyIds = Teuchos::null;
RCP< Tpetra::CrsMatrix< int, typename MeshAdapter< User >::lno_t, typename MeshAdapter< User >::gno_t, typename MeshAdapter< User >::node_t > > get2ndAdjsMatFromAdjs(const Teuchos::RCP< const MeshAdapter< User > > &ia, const RCP< const Comm< int > > comm, Zoltan2::MeshEntityType sourcetarget, Zoltan2::MeshEntityType through)
void get2ndAdjsViewFromAdjs(const Teuchos::RCP< const MeshAdapter< User > > &ia, const RCP< const Comm< int > > comm, Zoltan2::MeshEntityType sourcetarget, Zoltan2::MeshEntityType through, Teuchos::ArrayRCP< typename MeshAdapter< User >::offset_t > &offsets, Teuchos::ArrayRCP< typename MeshAdapter< User >::gno_t > &adjacencyIds)
Defines the MeshAdapter interface.
MeshAdapter defines the interface for mesh input.
map_t::global_ordinal_type gno_t
typename InputTraits< User >::node_t node_t
typename InputTraits< User >::gno_t gno_t
map_t::local_ordinal_type lno_t
typename InputTraits< User >::offset_t offset_t
MeshEntityType
Enumerate entity types for meshes: Regions, Faces, Edges, or Vertices.
Tpetra::global_size_t global_size_t
typename InputTraits< User >::lno_t lno_t