52 #ifndef _ZOLTAN2_MODELHELPERS_HPP_ 53 #define _ZOLTAN2_MODELHELPERS_HPP_ 59 template <
typename User>
60 RCP<Tpetra::CrsMatrix<int,
61 typename MeshAdapter<User>::lno_t,
63 typename MeshAdapter<User>::node_t> >
65 const RCP<
const Comm<int> > comm,
68 typedef typename MeshAdapter<User>::gno_t gno_t;
69 typedef typename MeshAdapter<User>::lno_t lno_t;
72 typedef int nonzero_t;
73 typedef Tpetra::CrsMatrix<nonzero_t,lno_t,gno_t,node_t> sparse_matrix_type;
74 typedef Tpetra::Map<lno_t, gno_t, node_t> map_type;
76 const GST dummy = Teuchos::OrdinalTraits<GST>::invalid ();
79 if (ia->availAdjs(sourcetarget, through)) {
80 using Tpetra::DefaultPlatform;
88 const lno_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 adjsMatrix = rcp (
new sparse_matrix_type (sourcetargetMapG,
168 Array<nonzero_t> justOneA(maxcols, 1);
169 ArrayView<const gno_t> adjacencyIdsAV(adjacencyIds, offsets[LocalNumIDs]);
171 for (
size_t localElement=0; localElement<LocalNumIDs; ++localElement){
173 size_t ncols = offsets[localElement+1] - offsets[localElement];
174 adjsMatrix->insertGlobalValues(Ids[localElement],
175 adjacencyIdsAV(offsets[localElement], ncols),
180 adjsMatrix->fillComplete (oneToOneTMap,
181 adjsMatrix->getRowMap());
184 RCP<sparse_matrix_type> secondAdjs =
185 rcp (
new sparse_matrix_type(adjsMatrix->getRowMap(),0));
186 Tpetra::MatrixMatrix::Multiply(*adjsMatrix,
false,*adjsMatrix,
190 return RCP<sparse_matrix_type>();
193 template <
typename User>
195 const RCP<
const Comm<int> > comm,
198 const typename MeshAdapter<User>::lno_t *&offsets,
199 const typename MeshAdapter<User>::gno_t *&adjacencyIds)
201 typedef typename MeshAdapter<User>::gno_t gno_t;
202 typedef typename MeshAdapter<User>::lno_t lno_t;
205 typedef int nonzero_t;
206 typedef Tpetra::CrsMatrix<nonzero_t,lno_t,gno_t,node_t> sparse_matrix_type;
210 if (secondAdjs!=RCP<sparse_matrix_type>()) {
211 Array<gno_t> Indices;
212 Array<nonzero_t> Values;
216 gno_t
const *Ids=NULL;
217 ia->getIDsViewOf(sourcetarget, Ids);
220 size_t LocalNumIDs = ia->getLocalNumOf(sourcetarget);
221 lno_t *start =
new lno_t [LocalNumIDs+1];
222 std::vector<gno_t> adj;
224 for (
size_t localElement=0; localElement<LocalNumIDs; ++localElement){
225 start[localElement] = nadj;
226 const gno_t globalRow = Ids[localElement];
227 size_t NumEntries = secondAdjs->getNumEntriesInGlobalRow (globalRow);
228 Indices.resize (NumEntries);
229 Values.resize (NumEntries);
230 secondAdjs->getGlobalRowCopy (globalRow,Indices(),Values(),NumEntries);
232 for (
size_t j = 0; j < NumEntries; ++j) {
233 if(globalRow != Indices[j]) {
234 adj.push_back(Indices[j]);
241 start[LocalNumIDs] = nadj;
243 gno_t *adj_ =
new gno_t [nadj];
245 for (
size_t i=0; i < nadj; i++) {
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)
InputTraits< User >::gno_t gno_t
Defines the MeshAdapter interface.
MeshAdapter defines the interface for mesh input.
void get2ndAdjsViewFromAdjs(const Teuchos::RCP< const MeshAdapter< User > > &ia, const RCP< const Comm< int > > comm, Zoltan2::MeshEntityType sourcetarget, Zoltan2::MeshEntityType through, const typename MeshAdapter< User >::lno_t *&offsets, const typename MeshAdapter< User >::gno_t *&adjacencyIds)
MeshEntityType
Enumerate entity types for meshes: Regions, Faces, Edges, or Vertices.