46 #ifndef MUELU_AGGREGATIONSTRUCTUREDALGORITHM_DEF_HPP_
47 #define MUELU_AGGREGATIONSTRUCTUREDALGORITHM_DEF_HPP_
50 #include <Teuchos_Comm.hpp>
51 #include <Teuchos_CommHelpers.hpp>
61 #include "MueLu_Aggregates.hpp"
62 #include "MueLu_IndexManager.hpp"
68 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
71 Aggregates& aggregates, std::vector<unsigned>& aggStat,
72 LO& numNonAggregatedNodes)
const {
73 Monitor m(*
this,
"BuildAggregates");
76 if(
const char* dbg = std::getenv(
"MUELU_STRUCTUREDALGORITHM_DEBUG")) {
77 out = Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
91 *out <<
"Extract data for ghosted nodes" << std::endl;
93 ghostedCoarseNodeCoarsePIDs, ghostedCoarseNodeCoarseGIDs);
97 LO ghostedCoarseNodeCoarseLID, aggId;
98 *out <<
"Loop over fine nodes and assign them to an aggregate and a rank" << std::endl;
103 for(
int dim = 0; dim < 3; ++dim) {
106 if(ghostedIdx[dim] - geoData->
getOffset(dim)
112 if(rem > (rate / 2)) {++coarseIdx[dim];}
118 ghostedCoarseNodeCoarseLID);
120 aggId = ghostedCoarseNodeCoarseLIDs[ghostedCoarseNodeCoarseLID];
121 vertex2AggId[nodeIdx] = aggId;
122 procWinner[nodeIdx] = ghostedCoarseNodeCoarsePIDs[ghostedCoarseNodeCoarseLID];
124 --numNonAggregatedNodes;
130 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
134 Monitor m(*
this,
"BuildAggregates");
137 if(
const char* dbg = std::getenv(
"MUELU_STRUCTUREDALGORITHM_DEBUG")) {
138 out = Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
147 int numInterpolationPoints = 0;
149 numInterpolationPoints = 1;
154 *out <<
"numInterpolationPoints=" << numInterpolationPoints << std::endl;
162 *out <<
"Compute prolongatorGraph data" << std::endl;
164 ComputeGraphDataConstant(graph, geoData, numInterpolationPoints, nnzOnRow, rowPtr, colIndex);
166 ComputeGraphDataLinear(graph, geoData, numInterpolationPoints, nnzOnRow, rowPtr, colIndex);
171 *out <<
"Compute domain and column maps of the CrsGraph" << std::endl;
173 *out <<
"Extract data for ghosted nodes" << std::endl;
178 ghostedCoarseNodeCoarsePIDs, ghostedCoarseNodeCoarseGIDs);
184 ghostedCoarseNodeCoarseGIDs(),
189 LO coarseNodeIdx = 0;
190 Array<GO> coarseNodeCoarseGIDs, coarseNodeFineGIDs;
192 for(
LO nodeIdx = 0; nodeIdx < ghostedCoarseNodeCoarseGIDs.
size(); ++nodeIdx) {
193 if(ghostedCoarseNodeCoarsePIDs[nodeIdx] == colMap->getComm()->getRank()) {
194 coarseNodeCoarseGIDs[coarseNodeIdx] = ghostedCoarseNodeCoarseGIDs[nodeIdx];
198 domainMap = MapFactory::Build(graph.
GetDomainMap()->lib(),
200 coarseNodeCoarseGIDs(),
204 coarseCoordinatesMap = MapFactory::Build(graph.
GetDomainMap()->lib(),
206 coarseNodeCoarseGIDs(),
210 coarseCoordinatesFineMap = MapFactory::Build(graph.
GetDomainMap()->lib(),
212 coarseNodeFineGIDs(),
231 coarseCoordinatesMap = MapFactory::Build(graph.
GetDomainMap()->lib(),
237 coarseCoordinatesFineMap = MapFactory::Build(graph.
GetDomainMap()->lib(),
239 coarseNodeFineGIDs(),
250 myGraph->insertLocalIndices(nodeIdx, colIndex(rowPtr[nodeIdx], nnzOnRow[nodeIdx]) );
257 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
264 if(
const char* dbg = std::getenv(
"MUELU_STRUCTUREDALGORITHM_DEBUG")) {
265 out = Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
275 ghostedCoarseNodeCoarsePIDs, ghostedCoarseNodeCoarseGIDs);
277 LO ghostedCoarseNodeCoarseLID, rem, rate;
281 nnzOnRow[nodeIdx] = Teuchos::as<size_t>(1);
282 rowPtr[nodeIdx + 1] = rowPtr[nodeIdx] + 1;
287 for(
int dim = 0; dim < 3; ++dim) {
290 if(ghostedIdx[dim] - geoData->
getOffset(dim)
296 if(rem > (rate / 2)) {++coarseIdx[dim];}
302 ghostedCoarseNodeCoarseLID);
303 colIndex[rowPtr[nodeIdx]] = ghostedCoarseNodeCoarseLIDs[ghostedCoarseNodeCoarseLID];
309 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
316 if(
const char* dbg = std::getenv(
"MUELU_STRUCTUREDALGORITHM_DEBUG")) {
317 out = Teuchos::fancyOStream(Teuchos::rcpFromRef(std::cout));
334 for(
int dim=0; dim < numDimensions; dim++){
337 if(ghostedIdx[dim] - geoData->
getOffset(dim)
343 if(ijkRem[dim] > (rate / 2)) {++coarseIdx[dim];}
350 bool allCoarse =
true;
352 for(
int dim = 0; dim < numDimensions; ++dim) {
353 isCoarse[dim] =
false;
355 isCoarse[dim] =
true;
360 isCoarse[dim] =
true;
363 isCoarse[dim] =
true;
373 colIndex[rowPtr[nodeIdx]]);
374 nnzOnRow[nodeIdx] = Teuchos::as<size_t>(1);
375 rowPtr[nodeIdx + 1] = rowPtr[nodeIdx] + 1;
379 nnzOnRow[nodeIdx] = Teuchos::as<size_t>( numInterpolationPoints );
380 rowPtr[nodeIdx + 1] = rowPtr[nodeIdx] + Teuchos::as<LO>( numInterpolationPoints );
382 for(
int dim = 0; dim < numDimensions; ++dim) {
389 if(numDimensions > 1) {
392 if(numDimensions > 2) {
396 geoData->
getCoarseNodeGhostedLID(coarseIdx[0]+1, coarseIdx[1]+1, coarseIdx[2]+1, colIndex[ rowPtr[nodeIdx]+7]);