xref: /petsc/src/dm/impls/plex/adaptors/parmmg/parmmgadapt.c (revision 8a7f912ab8467fae8fe0f39ed87215e84b6a29d1)
15f80ce2aSJacob Faibussowitsch #include "../mmgcommon.h" /*I      "petscdmplex.h"   I*/
22bc0b47dSJoe Wallwork #include <parmmg/libparmmg.h>
32bc0b47dSJoe Wallwork 
4b2c4de1dSPierre Jolivet PetscBool  ParMmgCite       = PETSC_FALSE;
5b2c4de1dSPierre Jolivet const char ParMmgCitation[] = "@techreport{cirrottola:hal-02386837,\n"
6b2c4de1dSPierre Jolivet                               "  title       = {Parallel unstructured mesh adaptation using iterative remeshing and repartitioning},\n"
7b2c4de1dSPierre Jolivet                               "  institution = {Inria Bordeaux},\n"
8b2c4de1dSPierre Jolivet                               "  author      = {L. Cirrottola and A. Froehly},\n"
9b2c4de1dSPierre Jolivet                               "  number      = {9307},\n"
10b2c4de1dSPierre Jolivet                               "  note        = {\\url{https://hal.inria.fr/hal-02386837}},\n"
11b2c4de1dSPierre Jolivet                               "  year        = {2019}\n}\n";
12b2c4de1dSPierre Jolivet 
13*8a7f912aSStephan Kramer /*
14*8a7f912aSStephan Kramer  Coupling code for the ParMmg metric-based mesh adaptation package.
15*8a7f912aSStephan Kramer */
16d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMAdaptMetric_ParMmg_Plex(DM dm, Vec vertexMetric, DMLabel bdLabel, DMLabel rgLabel, DM *dmNew)
17d71ae5a4SJacob Faibussowitsch {
180a7a67b9SPierre Jolivet   MPI_Comm           comm;
192bc0b47dSJoe Wallwork   const char        *bdName = "_boundary_";
209fe9e680SJoe Wallwork   const char        *rgName = "_regions_";
212bc0b47dSJoe Wallwork   DM                 udm, cdm;
22c1dc6da0SJoe Wallwork   DMLabel            bdLabelNew, rgLabelNew;
239fe9e680SJoe Wallwork   const char        *bdLabelName, *rgLabelName;
24c1dc6da0SJoe Wallwork   IS                 globalVertexNum;
252bc0b47dSJoe Wallwork   PetscSection       coordSection;
262bc0b47dSJoe Wallwork   Vec                coordinates;
272bc0b47dSJoe Wallwork   PetscSF            sf;
282bc0b47dSJoe Wallwork   const PetscScalar *coords, *met;
29ae8b063eSJoe Wallwork   PetscReal         *vertices, *metric, *verticesNew, *verticesNewLoc, gradationFactor, hausdorffNumber;
30c1dc6da0SJoe Wallwork   PetscInt          *cells, *cellsNew, *cellTags, *cellTagsNew, *verTags, *verTagsNew;
31c1dc6da0SJoe Wallwork   PetscInt          *bdFaces, *faceTags, *facesNew, *faceTagsNew;
32c1dc6da0SJoe Wallwork   PetscInt          *corners, *requiredCells, *requiredVer, *ridges, *requiredFaces;
33c1dc6da0SJoe Wallwork   PetscInt           cStart, cEnd, c, numCells, fStart, fEnd, f, numFaceTags, vStart, vEnd, v, numVertices;
34*8a7f912aSStephan Kramer   PetscInt           numCellsNotShared, *cIsLeaf, numUsedVertices, *vertexNumber, *fIsIncluded;
35c1dc6da0SJoe Wallwork   PetscInt           dim, off, coff, maxConeSize, bdSize, i, j, k, Neq, verbosity, numIter;
36*8a7f912aSStephan Kramer   PetscInt          *interfaces_lv, *interfaces_gv, *interfacesOffset;
37*8a7f912aSStephan Kramer   PetscInt           niranks, nrranks, numNgbRanks, r, lv, gv;
38c1dc6da0SJoe Wallwork   PetscInt          *gv_new, *owners, *verticesNewSorted, pStart, pEnd;
392bc0b47dSJoe Wallwork   PetscInt           numCellsNew, numVerticesNew, numCornersNew, numFacesNew, numVerticesNewLoc;
40c1dc6da0SJoe Wallwork   const PetscInt    *gV, *ioffset, *irootloc, *roffset, *rmine, *rremote;
4176f360caSJoe Wallwork   PetscBool          flg = PETSC_FALSE, noInsert, noSwap, noMove, noSurf, isotropic, uniform;
422bc0b47dSJoe Wallwork   const PetscMPIInt *iranks, *rranks;
432bc0b47dSJoe Wallwork   PetscMPIInt        numProcs, rank;
442bc0b47dSJoe Wallwork   PMMG_pParMesh      parmesh = NULL;
452bc0b47dSJoe Wallwork 
46*8a7f912aSStephan Kramer   // DEVELOPER NOTE: ParMmg wants to know the rank of every process which is sharing a given point and
47*8a7f912aSStephan Kramer   //                 for this information to be conveyed to every process that is sharing that point.
48*8a7f912aSStephan Kramer 
492bc0b47dSJoe Wallwork   PetscFunctionBegin;
50b2c4de1dSPierre Jolivet   PetscCall(PetscCitationsRegister(ParMmgCitation, &ParMmgCite));
519566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
529566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &numProcs));
539566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
542bc0b47dSJoe Wallwork   if (bdLabel) {
559566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)bdLabel, &bdLabelName));
569566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(bdLabelName, bdName, &flg));
5728b400f6SJacob Faibussowitsch     PetscCheck(!flg, comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for boundary facets", bdLabelName);
582bc0b47dSJoe Wallwork   }
599fe9e680SJoe Wallwork   if (rgLabel) {
609566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)rgLabel, &rgLabelName));
619566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(rgLabelName, rgName, &flg));
6228b400f6SJacob Faibussowitsch     PetscCheck(!flg, comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for element tags", rgLabelName);
639fe9e680SJoe Wallwork   }
642bc0b47dSJoe Wallwork 
652bc0b47dSJoe Wallwork   /* Get mesh information */
669566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
675f80ce2aSJacob Faibussowitsch   PetscCheck(dim == 3, comm, PETSC_ERR_ARG_OUTOFRANGE, "ParMmg only works in 3D.");
682bc0b47dSJoe Wallwork   Neq = (dim * (dim + 1)) / 2;
699566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
709566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
719566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
729566063dSJacob Faibussowitsch   PetscCall(DMPlexUninterpolate(dm, &udm));
739566063dSJacob Faibussowitsch   PetscCall(DMPlexGetMaxSizes(udm, &maxConeSize, NULL));
742bc0b47dSJoe Wallwork   numCells    = cEnd - cStart;
752bc0b47dSJoe Wallwork   numVertices = vEnd - vStart;
762bc0b47dSJoe Wallwork 
77*8a7f912aSStephan Kramer   /* Get parallel data; work out which cells are owned and which are leaves */
78*8a7f912aSStephan Kramer   PetscCall(PetscCalloc1(numCells, &cIsLeaf));
79*8a7f912aSStephan Kramer   numCellsNotShared = numCells;
80*8a7f912aSStephan Kramer   niranks = nrranks = 0;
81*8a7f912aSStephan Kramer   if (numProcs > 1) {
82*8a7f912aSStephan Kramer     PetscCall(DMGetPointSF(dm, &sf));
83*8a7f912aSStephan Kramer     PetscCall(PetscSFSetUp(sf));
84*8a7f912aSStephan Kramer     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc));
85*8a7f912aSStephan Kramer     PetscCall(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote));
86*8a7f912aSStephan Kramer     for (r = 0; r < nrranks; ++r) {
87*8a7f912aSStephan Kramer       for (i = roffset[r]; i < roffset[r + 1]; ++i) {
88*8a7f912aSStephan Kramer         if (rmine[i] >= cStart && rmine[i] < cEnd) {
89*8a7f912aSStephan Kramer           cIsLeaf[rmine[i] - cStart] = 1;
90*8a7f912aSStephan Kramer           numCellsNotShared--;
91*8a7f912aSStephan Kramer         }
92*8a7f912aSStephan Kramer       }
93*8a7f912aSStephan Kramer     }
94*8a7f912aSStephan Kramer   }
95*8a7f912aSStephan Kramer 
96*8a7f912aSStephan Kramer   /*
97*8a7f912aSStephan Kramer     Create a vertex numbering for ParMmg starting at 1. Vertices not included in any
98*8a7f912aSStephan Kramer     owned cell remain 0 and will be removed. Using this numbering, create cells.
99*8a7f912aSStephan Kramer   */
100*8a7f912aSStephan Kramer   numUsedVertices = 0;
101*8a7f912aSStephan Kramer   PetscCall(PetscCalloc1(numVertices, &vertexNumber));
102*8a7f912aSStephan Kramer   PetscCall(PetscMalloc1(numCellsNotShared * maxConeSize, &cells));
1032bc0b47dSJoe Wallwork   for (c = 0, coff = 0; c < numCells; ++c) {
1042bc0b47dSJoe Wallwork     const PetscInt *cone;
1052bc0b47dSJoe Wallwork     PetscInt        coneSize, cl;
1062bc0b47dSJoe Wallwork 
107*8a7f912aSStephan Kramer     if (!cIsLeaf[c]) {
108*8a7f912aSStephan Kramer       PetscCall(DMPlexGetConeSize(udm, cStart + c, &coneSize));
109*8a7f912aSStephan Kramer       PetscCall(DMPlexGetCone(udm, cStart + c, &cone));
110*8a7f912aSStephan Kramer       for (cl = 0; cl < coneSize; ++cl) {
111*8a7f912aSStephan Kramer         if (!vertexNumber[cone[cl] - vStart]) vertexNumber[cone[cl] - vStart] = ++numUsedVertices;
112*8a7f912aSStephan Kramer         cells[coff++] = vertexNumber[cone[cl] - vStart];
113*8a7f912aSStephan Kramer       }
114*8a7f912aSStephan Kramer     }
1152bc0b47dSJoe Wallwork   }
1162bc0b47dSJoe Wallwork 
117*8a7f912aSStephan Kramer   /* Get array of vertex coordinates */
1189566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
1199566063dSJacob Faibussowitsch   PetscCall(DMGetLocalSection(cdm, &coordSection));
1209566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
1219566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(coordinates, &coords));
122*8a7f912aSStephan Kramer   PetscCall(PetscMalloc2(numUsedVertices * Neq, &metric, dim * numUsedVertices, &vertices));
1232bc0b47dSJoe Wallwork   for (v = 0; v < vEnd - vStart; ++v) {
1249566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(coordSection, v + vStart, &off));
125*8a7f912aSStephan Kramer     if (vertexNumber[v]) {
126*8a7f912aSStephan Kramer       for (i = 0; i < dim; ++i) vertices[dim * (vertexNumber[v] - 1) + i] = PetscRealPart(coords[off + i]);
127*8a7f912aSStephan Kramer     }
1282bc0b47dSJoe Wallwork   }
1299566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(coordinates, &coords));
1302bc0b47dSJoe Wallwork 
131c1dc6da0SJoe Wallwork   /* Get face tags */
132c1dc6da0SJoe Wallwork   if (!bdLabel) {
133c1dc6da0SJoe Wallwork     flg = PETSC_TRUE;
1349566063dSJacob Faibussowitsch     PetscCall(DMLabelCreate(PETSC_COMM_SELF, bdName, &bdLabel));
1359566063dSJacob Faibussowitsch     PetscCall(DMPlexMarkBoundaryFaces(dm, 1, bdLabel));
136c1dc6da0SJoe Wallwork   }
1379566063dSJacob Faibussowitsch   PetscCall(DMLabelGetBounds(bdLabel, &pStart, &pEnd));
138*8a7f912aSStephan Kramer   PetscCall(PetscCalloc1(pEnd - pStart, &fIsIncluded));
139c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
140c1dc6da0SJoe Wallwork     PetscBool hasPoint;
141c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
1422bc0b47dSJoe Wallwork 
1439566063dSJacob Faibussowitsch     PetscCall(DMLabelHasPoint(bdLabel, f, &hasPoint));
144c1dc6da0SJoe Wallwork     if ((!hasPoint) || (f < fStart) || (f >= fEnd)) continue;
145*8a7f912aSStephan Kramer 
146*8a7f912aSStephan Kramer     /* Only faces adjacent to an owned (non-leaf) cell are included */
147*8a7f912aSStephan Kramer     PetscInt        nnbrs;
148*8a7f912aSStephan Kramer     const PetscInt *nbrs;
149*8a7f912aSStephan Kramer     PetscCall(DMPlexGetSupportSize(dm, f, &nnbrs));
150*8a7f912aSStephan Kramer     PetscCall(DMPlexGetSupport(dm, f, &nbrs));
151*8a7f912aSStephan Kramer     for (c = 0; c < nnbrs; ++c) fIsIncluded[f - pStart] = fIsIncluded[f - pStart] || !cIsLeaf[nbrs[c]];
152*8a7f912aSStephan Kramer     if (!fIsIncluded[f - pStart]) continue;
153*8a7f912aSStephan Kramer 
154c1dc6da0SJoe Wallwork     numFaceTags++;
155c1dc6da0SJoe Wallwork 
1569566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1572bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize * 2; cl += 2) {
1582bc0b47dSJoe Wallwork       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) ++bdSize;
1592bc0b47dSJoe Wallwork     }
1609566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1612bc0b47dSJoe Wallwork   }
1629566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(bdSize, &bdFaces, numFaceTags, &faceTags));
163c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
164c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
1652bc0b47dSJoe Wallwork 
166*8a7f912aSStephan Kramer     if (!fIsIncluded[f - pStart]) continue;
167c1dc6da0SJoe Wallwork 
1689566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1692bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize * 2; cl += 2) {
170*8a7f912aSStephan Kramer       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) bdFaces[bdSize++] = vertexNumber[closure[cl] - vStart];
1712bc0b47dSJoe Wallwork     }
1729566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1739566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(bdLabel, f, &faceTags[numFaceTags++]));
1742bc0b47dSJoe Wallwork   }
175*8a7f912aSStephan Kramer   PetscCall(PetscFree(fIsIncluded));
1762bc0b47dSJoe Wallwork 
1779fe9e680SJoe Wallwork   /* Get cell tags */
178*8a7f912aSStephan Kramer   PetscCall(PetscCalloc2(numUsedVertices, &verTags, numCellsNotShared, &cellTags));
1799fe9e680SJoe Wallwork   if (rgLabel) {
180*8a7f912aSStephan Kramer     for (c = cStart, coff = 0; c < cEnd; ++c) {
181*8a7f912aSStephan Kramer       if (!cIsLeaf[c - cStart]) { PetscCall(DMLabelGetValue(rgLabel, c, &cellTags[coff++])); }
1829fe9e680SJoe Wallwork     }
183*8a7f912aSStephan Kramer   }
184*8a7f912aSStephan Kramer   PetscCall(PetscFree(cIsLeaf));
1859fe9e680SJoe Wallwork 
186*8a7f912aSStephan Kramer   /* Get metric, using only the upper triangular part */
1879566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(vertexMetric, NULL, "-adapt_metric_view"));
1889566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(vertexMetric, &met));
1899566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricIsIsotropic(dm, &isotropic));
1909566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricIsUniform(dm, &uniform));
1912bc0b47dSJoe Wallwork   for (v = 0; v < (vEnd - vStart); ++v) {
192*8a7f912aSStephan Kramer     PetscInt vv = vertexNumber[v];
193*8a7f912aSStephan Kramer     if (!vv--) continue;
1942bc0b47dSJoe Wallwork     for (i = 0, k = 0; i < dim; ++i) {
1950a7a67b9SPierre Jolivet       for (j = i; j < dim; ++j, ++k) {
19693520041SJoe Wallwork         if (isotropic) {
19793520041SJoe Wallwork           if (i == j) {
198*8a7f912aSStephan Kramer             if (uniform) metric[Neq * vv + k] = PetscRealPart(met[0]);
199*8a7f912aSStephan Kramer             else metric[Neq * vv + k] = PetscRealPart(met[v]);
200*8a7f912aSStephan Kramer           } else metric[Neq * vv + k] = 0.0;
201*8a7f912aSStephan Kramer         } else metric[Neq * vv + k] = PetscRealPart(met[dim * dim * v + dim * i + j]);
2022bc0b47dSJoe Wallwork       }
2032bc0b47dSJoe Wallwork     }
2042bc0b47dSJoe Wallwork   }
2059566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(vertexMetric, &met));
2062bc0b47dSJoe Wallwork 
207*8a7f912aSStephan Kramer   /* Build ParMmg communicators: the list of vertices between two partitions  */
2082bc0b47dSJoe Wallwork   numNgbRanks = 0;
2092bc0b47dSJoe Wallwork   if (numProcs > 1) {
210*8a7f912aSStephan Kramer     DM              rankdm;
211*8a7f912aSStephan Kramer     PetscSection    rankSection, rankGlobalSection;
212*8a7f912aSStephan Kramer     PetscSF         rankSF;
213*8a7f912aSStephan Kramer     const PetscInt *degree;
214*8a7f912aSStephan Kramer     PetscInt       *rankOfUsedVertices, *rankOfUsedMultiRootLeaves, *usedCopies;
215*8a7f912aSStephan Kramer     PetscInt       *rankArray, *rankGlobalArray, *interfacesPerRank;
216*8a7f912aSStephan Kramer     PetscInt        offset, mrl, rootDegreeCnt, s, shareCnt, gv;
2172bc0b47dSJoe Wallwork 
218*8a7f912aSStephan Kramer     PetscCall(PetscSFComputeDegreeBegin(sf, &degree));
219*8a7f912aSStephan Kramer     PetscCall(PetscSFComputeDegreeEnd(sf, &degree));
220*8a7f912aSStephan Kramer     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
221*8a7f912aSStephan Kramer     for (i = 0, rootDegreeCnt = 0; i < pEnd - pStart; ++i) rootDegreeCnt += degree[i];
222*8a7f912aSStephan Kramer 
223*8a7f912aSStephan Kramer     /* rankOfUsedVertices, point-array: rank+1 if vertex and in use */
224*8a7f912aSStephan Kramer     PetscCall(PetscCalloc1(pEnd - pStart, &rankOfUsedVertices));
225*8a7f912aSStephan Kramer     for (i = 0; i < pEnd - pStart; ++i) rankOfUsedVertices[i] = -1;
226*8a7f912aSStephan Kramer     for (i = vStart; i < vEnd; ++i) {
227*8a7f912aSStephan Kramer       if (vertexNumber[i - vStart]) rankOfUsedVertices[i] = rank;
2282bc0b47dSJoe Wallwork     }
229c51fff1bSJoe Wallwork 
230*8a7f912aSStephan Kramer     /* rankOfUsedMultiRootLeaves, multiroot-array: rank if vertex and in use, else -1 */
231*8a7f912aSStephan Kramer     PetscCall(PetscMalloc1(rootDegreeCnt, &rankOfUsedMultiRootLeaves));
232*8a7f912aSStephan Kramer     PetscCall(PetscSFGatherBegin(sf, MPIU_INT, rankOfUsedVertices, rankOfUsedMultiRootLeaves));
233*8a7f912aSStephan Kramer     PetscCall(PetscSFGatherEnd(sf, MPIU_INT, rankOfUsedVertices, rankOfUsedMultiRootLeaves));
234*8a7f912aSStephan Kramer     PetscCall(PetscFree(rankOfUsedVertices));
235*8a7f912aSStephan Kramer 
236*8a7f912aSStephan Kramer     /* usedCopies, point-array: if vertex, shared by how many processes */
237*8a7f912aSStephan Kramer     PetscCall(PetscCalloc1(pEnd - pStart, &usedCopies));
238*8a7f912aSStephan Kramer     for (i = 0, mrl = 0; i < vStart - pStart; i++) mrl += degree[i];
239*8a7f912aSStephan Kramer     for (i = vStart - pStart; i < vEnd - pStart; ++i) {
240*8a7f912aSStephan Kramer       for (j = 0; j < degree[i]; j++, mrl++) {
241*8a7f912aSStephan Kramer         if (rankOfUsedMultiRootLeaves[mrl] != -1) usedCopies[i]++;
2422bc0b47dSJoe Wallwork       }
243*8a7f912aSStephan Kramer       if (vertexNumber[i - vStart + pStart]) usedCopies[i]++;
244*8a7f912aSStephan Kramer     }
245*8a7f912aSStephan Kramer     PetscCall(PetscSFBcastBegin(sf, MPIU_INT, usedCopies, usedCopies, MPI_REPLACE));
246*8a7f912aSStephan Kramer     PetscCall(PetscSFBcastEnd(sf, MPIU_INT, usedCopies, usedCopies, MPI_REPLACE));
247*8a7f912aSStephan Kramer 
248*8a7f912aSStephan Kramer     /* Create a section to store ranks of vertices shared by more than one process */
249*8a7f912aSStephan Kramer     PetscCall(PetscSectionCreate(comm, &rankSection));
250*8a7f912aSStephan Kramer     PetscCall(PetscSectionSetNumFields(rankSection, 1));
251*8a7f912aSStephan Kramer     PetscCall(PetscSectionSetChart(rankSection, pStart, pEnd));
252*8a7f912aSStephan Kramer     for (i = vStart - pStart; i < vEnd - pStart; ++i) {
253*8a7f912aSStephan Kramer       if (usedCopies[i] > 1) { PetscCall(PetscSectionSetDof(rankSection, i + pStart, usedCopies[i])); }
254*8a7f912aSStephan Kramer     }
255*8a7f912aSStephan Kramer     PetscCall(PetscSectionSetUp(rankSection));
256*8a7f912aSStephan Kramer     PetscCall(PetscSectionCreateGlobalSection(rankSection, sf, PETSC_FALSE, PETSC_FALSE, PETSC_TRUE, &rankGlobalSection));
257*8a7f912aSStephan Kramer 
258*8a7f912aSStephan Kramer     PetscCall(PetscSectionGetStorageSize(rankGlobalSection, &s));
259*8a7f912aSStephan Kramer     PetscCall(PetscMalloc1(s, &rankGlobalArray));
260*8a7f912aSStephan Kramer     for (i = 0, mrl = 0; i < vStart - pStart; i++) mrl += degree[i];
261*8a7f912aSStephan Kramer     for (i = vStart - pStart, k = 0; i < vEnd - pStart; ++i) {
262*8a7f912aSStephan Kramer       if (usedCopies[i] > 1 && degree[i]) {
263*8a7f912aSStephan Kramer         PetscCall(PetscSectionGetOffset(rankSection, k, &offset));
264*8a7f912aSStephan Kramer         if (vertexNumber[i - vStart + pStart]) rankGlobalArray[k++] = rank;
265*8a7f912aSStephan Kramer         for (j = 0; j < degree[i]; j++, mrl++) {
266*8a7f912aSStephan Kramer           if (rankOfUsedMultiRootLeaves[mrl] != -1) { rankGlobalArray[k++] = rankOfUsedMultiRootLeaves[mrl]; }
267*8a7f912aSStephan Kramer         }
268*8a7f912aSStephan Kramer       } else mrl += degree[i];
269*8a7f912aSStephan Kramer     }
270*8a7f912aSStephan Kramer     PetscCall(PetscFree(rankOfUsedMultiRootLeaves));
271*8a7f912aSStephan Kramer     PetscCall(PetscFree(usedCopies));
272*8a7f912aSStephan Kramer     PetscCall(PetscSectionDestroy(&rankGlobalSection));
273*8a7f912aSStephan Kramer 
274*8a7f912aSStephan Kramer     /*
275*8a7f912aSStephan Kramer       Broadcast the array of ranks.
276*8a7f912aSStephan Kramer         (We want all processes to know all the ranks that are looking at each point.
277*8a7f912aSStephan Kramer         Above, we tell the roots. Here, the roots tell the leaves.)
278*8a7f912aSStephan Kramer     */
279*8a7f912aSStephan Kramer     PetscCall(DMClone(dm, &rankdm));
280*8a7f912aSStephan Kramer     PetscCall(DMSetLocalSection(rankdm, rankSection));
281*8a7f912aSStephan Kramer     PetscCall(DMGetSectionSF(rankdm, &rankSF));
282*8a7f912aSStephan Kramer     PetscCall(PetscSectionGetStorageSize(rankSection, &s));
283*8a7f912aSStephan Kramer     PetscCall(PetscMalloc1(s, &rankArray));
284*8a7f912aSStephan Kramer     PetscCall(PetscSFBcastBegin(rankSF, MPI_INT, rankGlobalArray, rankArray, MPI_REPLACE));
285*8a7f912aSStephan Kramer     PetscCall(PetscSFBcastEnd(rankSF, MPI_INT, rankGlobalArray, rankArray, MPI_REPLACE));
286*8a7f912aSStephan Kramer     PetscCall(PetscFree(rankGlobalArray));
287*8a7f912aSStephan Kramer     PetscCall(DMDestroy(&rankdm));
288*8a7f912aSStephan Kramer 
289*8a7f912aSStephan Kramer     /* Count the number of interfaces per rank, not including those on the root */
290*8a7f912aSStephan Kramer     PetscCall(PetscCalloc1(numProcs, &interfacesPerRank));
291*8a7f912aSStephan Kramer     for (v = vStart; v < vEnd; v++) {
292*8a7f912aSStephan Kramer       if (vertexNumber[v - vStart]) {
293*8a7f912aSStephan Kramer         PetscCall(PetscSectionGetDof(rankSection, v, &shareCnt));
294*8a7f912aSStephan Kramer         if (shareCnt) {
295*8a7f912aSStephan Kramer           PetscCall(PetscSectionGetOffset(rankSection, v, &offset));
296*8a7f912aSStephan Kramer           for (j = 0; j < shareCnt; j++) { interfacesPerRank[rankArray[offset + j]]++; }
297*8a7f912aSStephan Kramer         }
298*8a7f912aSStephan Kramer       }
299*8a7f912aSStephan Kramer     }
300*8a7f912aSStephan Kramer     for (r = 0, k = 0, interfacesPerRank[rank] = 0; r < numProcs; r++) k += interfacesPerRank[r];
301*8a7f912aSStephan Kramer 
302*8a7f912aSStephan Kramer     /* Get the degree of the vertex */
303*8a7f912aSStephan Kramer     PetscCall(PetscMalloc3(k, &interfaces_lv, k, &interfaces_gv, numProcs + 1, &interfacesOffset));
304*8a7f912aSStephan Kramer     interfacesOffset[0] = 0;
305*8a7f912aSStephan Kramer     for (r = 0; r < numProcs; r++) {
306*8a7f912aSStephan Kramer       interfacesOffset[r + 1] = interfacesOffset[r] + interfacesPerRank[r];
307*8a7f912aSStephan Kramer       if (interfacesPerRank[r]) numNgbRanks++;
308*8a7f912aSStephan Kramer       interfacesPerRank[r] = 0;
3092bc0b47dSJoe Wallwork     }
310c51fff1bSJoe Wallwork 
311*8a7f912aSStephan Kramer     /* Get the local and global vertex numbers at interfaces */
312*8a7f912aSStephan Kramer     PetscCall(DMPlexGetVertexNumbering(dm, &globalVertexNum));
3139566063dSJacob Faibussowitsch     PetscCall(ISGetIndices(globalVertexNum, &gV));
314*8a7f912aSStephan Kramer     for (v = vStart; v < vEnd; v++) {
315*8a7f912aSStephan Kramer       if (vertexNumber[v - vStart]) {
316*8a7f912aSStephan Kramer         PetscCall(PetscSectionGetDof(rankSection, v, &shareCnt));
317*8a7f912aSStephan Kramer         if (shareCnt) {
318*8a7f912aSStephan Kramer           PetscCall(PetscSectionGetOffset(rankSection, v, &offset));
319*8a7f912aSStephan Kramer           for (j = 0; j < shareCnt; j++) {
320*8a7f912aSStephan Kramer             r = rankArray[offset + j];
321*8a7f912aSStephan Kramer             if (r == rank) continue;
322*8a7f912aSStephan Kramer             k                = interfacesOffset[r] + interfacesPerRank[r]++;
323*8a7f912aSStephan Kramer             interfaces_lv[k] = vertexNumber[v - vStart];
324*8a7f912aSStephan Kramer             gv               = gV[v - vStart];
325*8a7f912aSStephan Kramer             interfaces_gv[k] = gv < 0 ? -gv : gv + 1;
3262bc0b47dSJoe Wallwork           }
327*8a7f912aSStephan Kramer         }
328*8a7f912aSStephan Kramer       }
329*8a7f912aSStephan Kramer     }
330*8a7f912aSStephan Kramer     PetscCall(PetscFree(interfacesPerRank));
331*8a7f912aSStephan Kramer     PetscCall(PetscFree(rankArray));
3329566063dSJacob Faibussowitsch     PetscCall(ISRestoreIndices(globalVertexNum, &gV));
333*8a7f912aSStephan Kramer     PetscCall(PetscSectionDestroy(&rankSection));
3342bc0b47dSJoe Wallwork   }
3359566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&udm));
336*8a7f912aSStephan Kramer   PetscCall(PetscFree(vertexNumber));
3372bc0b47dSJoe Wallwork 
3382bc0b47dSJoe Wallwork   /* Send the data to ParMmg and remesh */
3399566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoInsertion(dm, &noInsert));
3409566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoSwapping(dm, &noSwap));
3419566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoMovement(dm, &noMove));
3429566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoSurf(dm, &noSurf));
3439566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetVerbosity(dm, &verbosity));
3449566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetNumIterations(dm, &numIter));
3459566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetGradationFactor(dm, &gradationFactor));
3469566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetHausdorffNumber(dm, &hausdorffNumber));
347b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Init_parMesh, PMMG_ARG_start, PMMG_ARG_ppParMesh, &parmesh, PMMG_ARG_pMesh, PMMG_ARG_pMet, PMMG_ARG_dim, 3, PMMG_ARG_MPIComm, comm, PMMG_ARG_end);
348*8a7f912aSStephan Kramer   PetscCallMMG_NONSTANDARD(PMMG_Set_meshSize, parmesh, numUsedVertices, numCellsNotShared, 0, numFaceTags, 0, 0);
349b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_APImode, PMMG_APIDISTRIB_nodes);
350b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_noinsert, noInsert);
351b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_noswap, noSwap);
352b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_nomove, noMove);
353b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_nosurf, noSurf);
354b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_verbose, verbosity);
355b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_globalNum, 1);
356b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_niter, numIter);
357b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_dparameter, parmesh, PMMG_DPARAM_hgrad, gradationFactor);
358b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_dparameter, parmesh, PMMG_DPARAM_hausd, hausdorffNumber);
359b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_vertices, parmesh, vertices, verTags);
360b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_tetrahedra, parmesh, cells, cellTags);
361b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_triangles, parmesh, bdFaces, faceTags);
362*8a7f912aSStephan Kramer   PetscCallMMG_NONSTANDARD(PMMG_Set_metSize, parmesh, MMG5_Vertex, numUsedVertices, MMG5_Tensor);
363b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_tensorMets, parmesh, metric);
364b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_numberOfNodeCommunicators, parmesh, numNgbRanks);
365*8a7f912aSStephan Kramer   for (r = 0, c = 0; r < numProcs; ++r) {
366*8a7f912aSStephan Kramer     if (interfacesOffset[r + 1] > interfacesOffset[r]) {
367*8a7f912aSStephan Kramer       PetscCallMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicatorSize, parmesh, c, r, interfacesOffset[r + 1] - interfacesOffset[r]);
368*8a7f912aSStephan Kramer       PetscCallMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicator_nodes, parmesh, c++, &interfaces_lv[interfacesOffset[r]], &interfaces_gv[interfacesOffset[r]], 1);
369*8a7f912aSStephan Kramer     }
3702bc0b47dSJoe Wallwork   }
371b3377c29SStefano Zampini   PetscCallMMG(PMMG_parmmglib_distributed, parmesh);
3729566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells));
3739566063dSJacob Faibussowitsch   PetscCall(PetscFree2(metric, vertices));
3749566063dSJacob Faibussowitsch   PetscCall(PetscFree2(bdFaces, faceTags));
3759566063dSJacob Faibussowitsch   PetscCall(PetscFree2(verTags, cellTags));
376*8a7f912aSStephan Kramer   if (numProcs > 1) { PetscCall(PetscFree3(interfaces_lv, interfaces_gv, interfacesOffset)); }
3772bc0b47dSJoe Wallwork 
3782f583692SJoe Wallwork   /* Retrieve mesh from Mmg */
3792bc0b47dSJoe Wallwork   numCornersNew = 4;
380b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Get_meshSize, parmesh, &numVerticesNew, &numCellsNew, 0, &numFacesNew, 0, 0);
3819566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(dim * numVerticesNew, &verticesNew, numVerticesNew, &verTagsNew, numVerticesNew, &corners, numVerticesNew, &requiredVer));
3829566063dSJacob Faibussowitsch   PetscCall(PetscMalloc3((dim + 1) * numCellsNew, &cellsNew, numCellsNew, &cellTagsNew, numCellsNew, &requiredCells));
3839566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(dim * numFacesNew, &facesNew, numFacesNew, &faceTagsNew, numFacesNew, &ridges, numFacesNew, &requiredFaces));
384b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Get_vertices, parmesh, verticesNew, verTagsNew, corners, requiredVer);
385b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Get_tetrahedra, parmesh, cellsNew, cellTagsNew, requiredCells);
386b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Get_triangles, parmesh, facesNew, faceTagsNew, requiredFaces);
3879566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numVerticesNew, &owners, numVerticesNew, &gv_new));
388b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_globalNum, 1);
389b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Get_verticesGloNum, parmesh, gv_new, owners);
3902bc0b47dSJoe Wallwork   for (i = 0; i < dim * numFacesNew; ++i) facesNew[i] -= 1;
3910a7a67b9SPierre Jolivet   for (i = 0; i < (dim + 1) * numCellsNew; ++i) cellsNew[i] = gv_new[cellsNew[i] - 1] - 1;
3920a7a67b9SPierre Jolivet   for (i = 0, numVerticesNewLoc = 0; i < numVerticesNew; ++i) {
3932bc0b47dSJoe Wallwork     if (owners[i] == rank) numVerticesNewLoc++;
3942bc0b47dSJoe Wallwork   }
3959566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numVerticesNewLoc * dim, &verticesNewLoc, numVerticesNew, &verticesNewSorted));
3962bc0b47dSJoe Wallwork   for (i = 0, c = 0; i < numVerticesNew; i++) {
3972bc0b47dSJoe Wallwork     if (owners[i] == rank) {
3982bc0b47dSJoe Wallwork       for (j = 0; j < dim; ++j) verticesNewLoc[dim * c + j] = verticesNew[dim * i + j];
3992bc0b47dSJoe Wallwork       c++;
4002bc0b47dSJoe Wallwork     }
4012bc0b47dSJoe Wallwork   }
4022f583692SJoe Wallwork 
4032f583692SJoe Wallwork   /* Reorder for consistency with DMPlex */
4049566063dSJacob Faibussowitsch   for (i = 0; i < numCellsNew; ++i) PetscCall(DMPlexInvertCell(DM_POLYTOPE_TETRAHEDRON, &cellsNew[4 * i]));
4052f583692SJoe Wallwork 
4062f583692SJoe Wallwork   /* Create new plex */
4079566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromCellListParallelPetsc(comm, dim, numCellsNew, numVerticesNewLoc, PETSC_DECIDE, numCornersNew, PETSC_TRUE, cellsNew, dim, verticesNewLoc, NULL, &verticesNewSorted, dmNew));
408b3377c29SStefano Zampini   PetscCallMMG_NONSTANDARD(PMMG_Free_all, PMMG_ARG_start, PMMG_ARG_ppParMesh, &parmesh, PMMG_ARG_end);
4099566063dSJacob Faibussowitsch   PetscCall(PetscFree4(verticesNew, verTagsNew, corners, requiredVer));
410c1dc6da0SJoe Wallwork 
411c1dc6da0SJoe Wallwork   /* Get adapted mesh information */
4129566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(*dmNew, 0, &cStart, &cEnd));
4139566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(*dmNew, 1, &fStart, &fEnd));
4149566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(*dmNew, 0, &vStart, &vEnd));
4152bc0b47dSJoe Wallwork 
4162bc0b47dSJoe Wallwork   /* Rebuild boundary label */
4179566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(*dmNew, flg ? bdName : bdLabelName));
4189566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(*dmNew, flg ? bdName : bdLabelName, &bdLabelNew));
4192bc0b47dSJoe Wallwork   for (i = 0; i < numFacesNew; i++) {
420c1dc6da0SJoe Wallwork     PetscBool       hasTag = PETSC_FALSE;
4212bc0b47dSJoe Wallwork     PetscInt        numCoveredPoints, numFaces = 0, facePoints[3];
4222bc0b47dSJoe Wallwork     const PetscInt *coveredPoints = NULL;
4232bc0b47dSJoe Wallwork 
4242bc0b47dSJoe Wallwork     for (j = 0; j < dim; ++j) {
4252bc0b47dSJoe Wallwork       lv = facesNew[i * dim + j];
4262bc0b47dSJoe Wallwork       gv = gv_new[lv] - 1;
4279566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesNew, verticesNewSorted, &lv));
4282bc0b47dSJoe Wallwork       facePoints[j] = lv + vStart;
4292bc0b47dSJoe Wallwork     }
4309566063dSJacob Faibussowitsch     PetscCall(DMPlexGetFullJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
4312bc0b47dSJoe Wallwork     for (j = 0; j < numCoveredPoints; ++j) {
4322bc0b47dSJoe Wallwork       if (coveredPoints[j] >= fStart && coveredPoints[j] < fEnd) {
4332bc0b47dSJoe Wallwork         numFaces++;
4342bc0b47dSJoe Wallwork         f = j;
4352bc0b47dSJoe Wallwork       }
4362bc0b47dSJoe Wallwork     }
4375f80ce2aSJacob Faibussowitsch     PetscCheck(numFaces == 1, comm, PETSC_ERR_ARG_OUTOFRANGE, "%" PetscInt_FMT " vertices cannot define more than 1 facet (%" PetscInt_FMT ")", dim, numFaces);
4389566063dSJacob Faibussowitsch     PetscCall(DMLabelHasStratum(bdLabel, faceTagsNew[i], &hasTag));
4399566063dSJacob Faibussowitsch     if (hasTag) PetscCall(DMLabelSetValue(bdLabelNew, coveredPoints[f], faceTagsNew[i]));
4409566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
4412bc0b47dSJoe Wallwork   }
4429566063dSJacob Faibussowitsch   PetscCall(PetscFree4(facesNew, faceTagsNew, ridges, requiredFaces));
4439566063dSJacob Faibussowitsch   PetscCall(PetscFree2(owners, gv_new));
4449566063dSJacob Faibussowitsch   PetscCall(PetscFree2(verticesNewLoc, verticesNewSorted));
4459566063dSJacob Faibussowitsch   if (flg) PetscCall(DMLabelDestroy(&bdLabel));
4462bc0b47dSJoe Wallwork 
4479fe9e680SJoe Wallwork   /* Rebuild cell labels */
4489566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(*dmNew, rgLabel ? rgLabelName : rgName));
4499566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(*dmNew, rgLabel ? rgLabelName : rgName, &rgLabelNew));
4509566063dSJacob Faibussowitsch   for (c = cStart; c < cEnd; ++c) PetscCall(DMLabelSetValue(rgLabelNew, c, cellTagsNew[c - cStart]));
4519566063dSJacob Faibussowitsch   PetscCall(PetscFree3(cellsNew, cellTagsNew, requiredCells));
4523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4532bc0b47dSJoe Wallwork }
454