xref: /petsc/src/dm/impls/plex/adaptors/parmmg/parmmgadapt.c (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
1*5f80ce2aSJacob Faibussowitsch #include "../mmgcommon.h"   /*I      "petscdmplex.h"   I*/
22bc0b47dSJoe Wallwork #include <parmmg/libparmmg.h>
32bc0b47dSJoe Wallwork 
49fe9e680SJoe Wallwork PETSC_EXTERN PetscErrorCode DMAdaptMetric_ParMmg_Plex(DM dm, Vec vertexMetric, DMLabel bdLabel, DMLabel rgLabel, DM *dmNew)
52bc0b47dSJoe Wallwork {
60a7a67b9SPierre Jolivet   MPI_Comm           comm;
72bc0b47dSJoe Wallwork   const char        *bdName = "_boundary_";
89fe9e680SJoe Wallwork   const char        *rgName = "_regions_";
92bc0b47dSJoe Wallwork   DM                 udm, cdm;
10c1dc6da0SJoe Wallwork   DMLabel            bdLabelNew, rgLabelNew;
119fe9e680SJoe Wallwork   const char        *bdLabelName, *rgLabelName;
12c1dc6da0SJoe Wallwork   IS                 globalVertexNum;
132bc0b47dSJoe Wallwork   PetscSection       coordSection;
142bc0b47dSJoe Wallwork   Vec                coordinates;
152bc0b47dSJoe Wallwork   PetscSF            sf;
162bc0b47dSJoe Wallwork   const PetscScalar *coords, *met;
17ae8b063eSJoe Wallwork   PetscReal         *vertices, *metric, *verticesNew, *verticesNewLoc, gradationFactor, hausdorffNumber;
18c1dc6da0SJoe Wallwork   PetscInt          *cells, *cellsNew, *cellTags, *cellTagsNew, *verTags, *verTagsNew;
19c1dc6da0SJoe Wallwork   PetscInt          *bdFaces, *faceTags, *facesNew, *faceTagsNew;
20c1dc6da0SJoe Wallwork   PetscInt          *corners, *requiredCells, *requiredVer, *ridges, *requiredFaces;
21c1dc6da0SJoe Wallwork   PetscInt           cStart, cEnd, c, numCells, fStart, fEnd, f, numFaceTags, vStart, vEnd, v, numVertices;
22c1dc6da0SJoe Wallwork   PetscInt           dim, off, coff, maxConeSize, bdSize, i, j, k, Neq, verbosity, numIter;
232bc0b47dSJoe Wallwork   PetscInt          *numVerInterfaces, *ngbRanks, *verNgbRank, *interfaces_lv, *interfaces_gv, *intOffset;
24c1dc6da0SJoe Wallwork   PetscInt           niranks, nrranks, numNgbRanks, numVerNgbRanksTotal, count, sliceSize, p, r, n, lv, gv;
25c1dc6da0SJoe Wallwork   PetscInt          *gv_new, *owners, *verticesNewSorted, pStart, pEnd;
262bc0b47dSJoe Wallwork   PetscInt           numCellsNew, numVerticesNew, numCornersNew, numFacesNew, numVerticesNewLoc;
27c1dc6da0SJoe Wallwork   const PetscInt    *gV, *ioffset, *irootloc, *roffset, *rmine, *rremote;
2876f360caSJoe Wallwork   PetscBool          flg = PETSC_FALSE, noInsert, noSwap, noMove, noSurf, isotropic, uniform;
292bc0b47dSJoe Wallwork   const PetscMPIInt *iranks, *rranks;
302bc0b47dSJoe Wallwork   PetscMPIInt        numProcs, rank;
312bc0b47dSJoe Wallwork   PMMG_pParMesh      parmesh = NULL;
322bc0b47dSJoe Wallwork 
332bc0b47dSJoe Wallwork   PetscFunctionBegin;
34*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectGetComm((PetscObject) dm, &comm));
35*5f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(comm, &numProcs));
36*5f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(comm, &rank));
372bc0b47dSJoe Wallwork   if (bdLabel) {
38*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectGetName((PetscObject) bdLabel, &bdLabelName));
39*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrcmp(bdLabelName, bdName, &flg));
402c71b3e2SJacob Faibussowitsch     PetscCheckFalse(flg,comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for boundary facets", bdLabelName);
412bc0b47dSJoe Wallwork   }
429fe9e680SJoe Wallwork   if (rgLabel) {
43*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectGetName((PetscObject) rgLabel, &rgLabelName));
44*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrcmp(rgLabelName, rgName, &flg));
452c71b3e2SJacob Faibussowitsch     PetscCheckFalse(flg,comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for element tags", rgLabelName);
469fe9e680SJoe Wallwork   }
472bc0b47dSJoe Wallwork 
482bc0b47dSJoe Wallwork   /* Get mesh information */
49*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetDimension(dm, &dim));
50*5f80ce2aSJacob Faibussowitsch   PetscCheck(dim == 3,comm, PETSC_ERR_ARG_OUTOFRANGE, "ParMmg only works in 3D.");
512bc0b47dSJoe Wallwork   Neq  = (dim*(dim+1))/2;
52*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
53*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
54*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
55*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexUninterpolate(dm, &udm));
56*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetMaxSizes(udm, &maxConeSize, NULL));
572bc0b47dSJoe Wallwork   numCells    = cEnd - cStart;
582bc0b47dSJoe Wallwork   numVertices = vEnd - vStart;
592bc0b47dSJoe Wallwork 
602bc0b47dSJoe Wallwork   /* Get cell offsets */
61*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(numCells*maxConeSize, &cells));
622bc0b47dSJoe Wallwork   for (c = 0, coff = 0; c < numCells; ++c) {
632bc0b47dSJoe Wallwork     const PetscInt *cone;
642bc0b47dSJoe Wallwork     PetscInt        coneSize, cl;
652bc0b47dSJoe Wallwork 
66*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetConeSize(udm, c, &coneSize));
67*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetCone(udm, c, &cone));
682bc0b47dSJoe Wallwork     for (cl = 0; cl < coneSize; ++cl) cells[coff++] = cone[cl] - vStart+1;
692bc0b47dSJoe Wallwork   }
702bc0b47dSJoe Wallwork 
712bc0b47dSJoe Wallwork   /* Get vertex coordinate array */
72*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetCoordinateDM(dm, &cdm));
73*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetLocalSection(cdm, &coordSection));
74*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetCoordinatesLocal(dm, &coordinates));
75*5f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(coordinates, &coords));
76*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(numVertices*Neq, &metric, dim*numVertices, &vertices));
772bc0b47dSJoe Wallwork   for (v = 0; v < vEnd-vStart; ++v) {
78*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSectionGetOffset(coordSection, v+vStart, &off));
792bc0b47dSJoe Wallwork     for (i = 0; i < dim; ++i) vertices[dim*v+i] = PetscRealPart(coords[off+i]);
802bc0b47dSJoe Wallwork   }
81*5f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(coordinates, &coords));
822bc0b47dSJoe Wallwork 
83c1dc6da0SJoe Wallwork   /* Get face tags */
84c1dc6da0SJoe Wallwork   if (!bdLabel) {
85c1dc6da0SJoe Wallwork     flg = PETSC_TRUE;
86*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMLabelCreate(PETSC_COMM_SELF, bdName, &bdLabel));
87*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexMarkBoundaryFaces(dm, 1, bdLabel));
88c1dc6da0SJoe Wallwork   }
89*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMLabelGetBounds(bdLabel, &pStart, &pEnd));
90c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
91c1dc6da0SJoe Wallwork     PetscBool hasPoint;
92c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
932bc0b47dSJoe Wallwork 
94*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMLabelHasPoint(bdLabel, f, &hasPoint));
95c1dc6da0SJoe Wallwork     if ((!hasPoint) || (f < fStart) || (f >= fEnd)) continue;
96c1dc6da0SJoe Wallwork     numFaceTags++;
97c1dc6da0SJoe Wallwork 
98*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
992bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize*2; cl += 2) {
1002bc0b47dSJoe Wallwork       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) ++bdSize;
1012bc0b47dSJoe Wallwork     }
102*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1032bc0b47dSJoe Wallwork   }
104*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(bdSize, &bdFaces, numFaceTags, &faceTags));
105c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
106c1dc6da0SJoe Wallwork     PetscBool hasPoint;
107c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
1082bc0b47dSJoe Wallwork 
109*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMLabelHasPoint(bdLabel, f, &hasPoint));
110c1dc6da0SJoe Wallwork     if ((!hasPoint) || (f < fStart) || (f >= fEnd)) continue;
111c1dc6da0SJoe Wallwork 
112*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1132bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize*2; cl += 2) {
1142bc0b47dSJoe Wallwork       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) bdFaces[bdSize++] = closure[cl] - vStart+1;
1152bc0b47dSJoe Wallwork     }
116*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
117*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMLabelGetValue(bdLabel, f, &faceTags[numFaceTags++]));
1182bc0b47dSJoe Wallwork   }
1192bc0b47dSJoe Wallwork 
1209fe9e680SJoe Wallwork   /* Get cell tags */
121*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscCalloc2(numVertices, &verTags, numCells, &cellTags));
1229fe9e680SJoe Wallwork   if (rgLabel) {
123*5f80ce2aSJacob Faibussowitsch     for (c = cStart; c < cEnd; ++c) CHKERRQ(DMLabelGetValue(rgLabel, c, &cellTags[c]));
1249fe9e680SJoe Wallwork   }
1259fe9e680SJoe Wallwork 
1262bc0b47dSJoe Wallwork   /* Get metric */
127*5f80ce2aSJacob Faibussowitsch   CHKERRQ(VecViewFromOptions(vertexMetric, NULL, "-adapt_metric_view"));
128*5f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(vertexMetric, &met));
129*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricIsIsotropic(dm, &isotropic));
130*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricIsUniform(dm, &uniform));
1312bc0b47dSJoe Wallwork   for (v = 0; v < (vEnd-vStart); ++v) {
1322bc0b47dSJoe Wallwork     for (i = 0, k = 0; i < dim; ++i) {
1330a7a67b9SPierre Jolivet       for (j = i; j < dim; ++j, ++k) {
13493520041SJoe Wallwork         if (isotropic) {
13593520041SJoe Wallwork           if (i == j) {
13693520041SJoe Wallwork             if (uniform) metric[Neq*v+k] = PetscRealPart(met[0]);
13793520041SJoe Wallwork             else metric[Neq*v+k] = PetscRealPart(met[v]);
13893520041SJoe Wallwork           } else metric[Neq*v+k] = 0.0;
1390a7a67b9SPierre Jolivet         } else metric[Neq*v+k] = PetscRealPart(met[dim*dim*v+dim*i+j]);
1402bc0b47dSJoe Wallwork       }
1412bc0b47dSJoe Wallwork     }
1422bc0b47dSJoe Wallwork   }
143*5f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(vertexMetric, &met));
1442bc0b47dSJoe Wallwork 
1452bc0b47dSJoe Wallwork   /* Build ParMMG communicators: the list of vertices between two partitions  */
1462bc0b47dSJoe Wallwork   niranks = nrranks = 0;
1472bc0b47dSJoe Wallwork   numNgbRanks = 0;
1482bc0b47dSJoe Wallwork   if (numProcs > 1) {
149*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMGetPointSF(dm, &sf));
150*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSFSetUp(sf));
151*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc));
152*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote));
153*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscCalloc1(numProcs, &numVerInterfaces));
1542bc0b47dSJoe Wallwork 
155c51fff1bSJoe Wallwork     /* Count number of roots associated with each leaf */
1562bc0b47dSJoe Wallwork     for (r = 0; r < niranks; ++r) {
1572bc0b47dSJoe Wallwork       for (i=ioffset[r], count=0; i<ioffset[r+1]; ++i) {
1582bc0b47dSJoe Wallwork         if (irootloc[i] >= vStart && irootloc[i] < vEnd) count++;
1592bc0b47dSJoe Wallwork       }
1602bc0b47dSJoe Wallwork       numVerInterfaces[iranks[r]] += count;
1612bc0b47dSJoe Wallwork     }
162c51fff1bSJoe Wallwork 
163c51fff1bSJoe Wallwork     /* Count number of leaves associated with each root */
1642bc0b47dSJoe Wallwork     for (r = 0; r < nrranks; ++r) {
1652bc0b47dSJoe Wallwork       for (i=roffset[r], count=0; i<roffset[r+1]; ++i) {
1662bc0b47dSJoe Wallwork         if (rmine[i] >= vStart && rmine[i] < vEnd) count++;
1672bc0b47dSJoe Wallwork       }
1682bc0b47dSJoe Wallwork       numVerInterfaces[rranks[r]] += count;
1692bc0b47dSJoe Wallwork     }
170c51fff1bSJoe Wallwork 
171c51fff1bSJoe Wallwork     /* Count global number of ranks */
1722bc0b47dSJoe Wallwork     for (p = 0; p < numProcs; ++p) {
1732bc0b47dSJoe Wallwork       if (numVerInterfaces[p]) numNgbRanks++;
1742bc0b47dSJoe Wallwork     }
175c51fff1bSJoe Wallwork 
176c51fff1bSJoe Wallwork     /* Provide numbers of vertex interfaces */
177*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc2(numNgbRanks, &ngbRanks, numNgbRanks, &verNgbRank));
1782bc0b47dSJoe Wallwork     for (p = 0, n = 0; p < numProcs; ++p) {
1792bc0b47dSJoe Wallwork       if (numVerInterfaces[p]) {
1802bc0b47dSJoe Wallwork         ngbRanks[n] = p;
1812bc0b47dSJoe Wallwork         verNgbRank[n] = numVerInterfaces[p];
1822bc0b47dSJoe Wallwork         n++;
1832bc0b47dSJoe Wallwork       }
1842bc0b47dSJoe Wallwork     }
1852bc0b47dSJoe Wallwork     numVerNgbRanksTotal = 0;
1862bc0b47dSJoe Wallwork     for (p = 0; p < numNgbRanks; ++p) numVerNgbRanksTotal += verNgbRank[p];
1872bc0b47dSJoe Wallwork 
1882bc0b47dSJoe Wallwork     /* For each neighbor, fill in interface arrays */
189*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc3(numVerNgbRanksTotal, &interfaces_lv, numVerNgbRanksTotal, &interfaces_gv,  numNgbRanks+1, &intOffset));
1902bc0b47dSJoe Wallwork     intOffset[0] = 0;
1912bc0b47dSJoe Wallwork     for (p = 0, r = 0, i = 0; p < numNgbRanks; ++p) {
1922bc0b47dSJoe Wallwork       intOffset[p+1] = intOffset[p];
193c51fff1bSJoe Wallwork 
194c51fff1bSJoe Wallwork       /* Leaf case */
1952bc0b47dSJoe Wallwork       if (iranks && iranks[i] == ngbRanks[p]) {
1962bc0b47dSJoe Wallwork 
1972bc0b47dSJoe Wallwork         /* Add the right slice of irootloc at the right place */
1982bc0b47dSJoe Wallwork         sliceSize = ioffset[i+1]-ioffset[i];
1992bc0b47dSJoe Wallwork         for (j = 0, count = 0; j < sliceSize; ++j) {
200*5f80ce2aSJacob Faibussowitsch           PetscCheck(ioffset[i]+j < ioffset[niranks],comm, PETSC_ERR_ARG_OUTOFRANGE, "Leaf index %" PetscInt_FMT " out of range (expected < %" PetscInt_FMT ")", ioffset[i]+j, ioffset[niranks]);
2012bc0b47dSJoe Wallwork           v = irootloc[ioffset[i]+j];
2022bc0b47dSJoe Wallwork           if (v >= vStart && v < vEnd) {
203*5f80ce2aSJacob Faibussowitsch             PetscCheck(intOffset[p+1]+count < numVerNgbRanksTotal,comm, PETSC_ERR_ARG_OUTOFRANGE, "Leaf interface index %" PetscInt_FMT " out of range (expected < %" PetscInt_FMT ")", intOffset[p+1]+count, numVerNgbRanksTotal);
2042bc0b47dSJoe Wallwork             interfaces_lv[intOffset[p+1]+count] = v-vStart;
2052bc0b47dSJoe Wallwork             count++;
2062bc0b47dSJoe Wallwork           }
2072bc0b47dSJoe Wallwork         }
2082bc0b47dSJoe Wallwork         intOffset[p+1] += count;
2092bc0b47dSJoe Wallwork         i++;
2102bc0b47dSJoe Wallwork       }
211c51fff1bSJoe Wallwork 
212c51fff1bSJoe Wallwork       /* Root case */
2132bc0b47dSJoe Wallwork       if (rranks && rranks[r] == ngbRanks[p]) {
2142bc0b47dSJoe Wallwork 
2152bc0b47dSJoe Wallwork         /* Add the right slice of rmine at the right place */
2162bc0b47dSJoe Wallwork         sliceSize = roffset[r+1]-roffset[r];
2172bc0b47dSJoe Wallwork         for (j = 0, count = 0; j < sliceSize; ++j) {
218*5f80ce2aSJacob Faibussowitsch           PetscCheck(roffset[r]+j < roffset[nrranks],comm, PETSC_ERR_ARG_OUTOFRANGE, "Root index %" PetscInt_FMT " out of range (expected < %" PetscInt_FMT ")", roffset[r]+j, roffset[nrranks]);
2192bc0b47dSJoe Wallwork           v = rmine[roffset[r]+j];
2202bc0b47dSJoe Wallwork           if (v >= vStart && v < vEnd) {
221*5f80ce2aSJacob Faibussowitsch             PetscCheck(intOffset[p+1]+count < numVerNgbRanksTotal,comm, PETSC_ERR_ARG_OUTOFRANGE, "Root interface index %" PetscInt_FMT " out of range (expected < %" PetscInt_FMT ")", intOffset[p+1]+count, numVerNgbRanksTotal);
2222bc0b47dSJoe Wallwork             interfaces_lv[intOffset[p+1]+count] = v-vStart;
2232bc0b47dSJoe Wallwork             count++;
2242bc0b47dSJoe Wallwork           }
2252bc0b47dSJoe Wallwork         }
2262bc0b47dSJoe Wallwork         intOffset[p+1] += count;
2272bc0b47dSJoe Wallwork         r++;
2282bc0b47dSJoe Wallwork       }
229c51fff1bSJoe Wallwork 
230c51fff1bSJoe Wallwork       /* Check validity of offsets */
231*5f80ce2aSJacob Faibussowitsch       PetscCheck(intOffset[p+1] == intOffset[p]+verNgbRank[p],comm, PETSC_ERR_ARG_OUTOFRANGE, "Missing offsets (expected %" PetscInt_FMT ", got %" PetscInt_FMT ")", intOffset[p]+verNgbRank[p], intOffset[p+1]);
2322bc0b47dSJoe Wallwork     }
233*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetVertexNumbering(udm, &globalVertexNum));
234*5f80ce2aSJacob Faibussowitsch     CHKERRQ(ISGetIndices(globalVertexNum, &gV));
2352bc0b47dSJoe Wallwork     for (i = 0; i < numVerNgbRanksTotal; ++i) {
2362bc0b47dSJoe Wallwork       v = gV[interfaces_lv[i]];
2372bc0b47dSJoe Wallwork       interfaces_gv[i] = v < 0 ? -v-1 : v;
2382bc0b47dSJoe Wallwork       interfaces_lv[i] += 1;
2392bc0b47dSJoe Wallwork       interfaces_gv[i] += 1;
2402bc0b47dSJoe Wallwork     }
241*5f80ce2aSJacob Faibussowitsch     CHKERRQ(ISRestoreIndices(globalVertexNum, &gV));
242*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree(numVerInterfaces));
2432bc0b47dSJoe Wallwork   }
244*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMDestroy(&udm));
2452bc0b47dSJoe Wallwork 
2462bc0b47dSJoe Wallwork   /* Send the data to ParMmg and remesh */
247*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricNoInsertion(dm, &noInsert));
248*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricNoSwapping(dm, &noSwap));
249*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricNoMovement(dm, &noMove));
250*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricNoSurf(dm, &noSurf));
251*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricGetVerbosity(dm, &verbosity));
252*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricGetNumIterations(dm, &numIter));
253*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricGetGradationFactor(dm, &gradationFactor));
254*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexMetricGetHausdorffNumber(dm, &hausdorffNumber));
255*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_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));
256*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_meshSize(parmesh, numVertices, numCells, 0, numFaceTags, 0, 0));
257*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_APImode, PMMG_APIDISTRIB_nodes));
258*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_noinsert, noInsert));
259*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_noswap, noSwap));
260*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_nomove, noMove));
261*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_nosurf, noSurf));
262*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_verbose, verbosity));
263*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_globalNum, 1));
264*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_niter, numIter));
265*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_dparameter(parmesh, PMMG_DPARAM_hgrad, gradationFactor));
266*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_dparameter(parmesh, PMMG_DPARAM_hausd, hausdorffNumber));
267*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_vertices(parmesh, vertices, verTags));
268*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_tetrahedra(parmesh, cells, cellTags));
269*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_triangles(parmesh, bdFaces, faceTags));
270*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_metSize(parmesh, MMG5_Vertex, numVertices, MMG5_Tensor));
271*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_tensorMets(parmesh, metric));
272*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_numberOfNodeCommunicators(parmesh, numNgbRanks));
2732bc0b47dSJoe Wallwork   for (c = 0; c < numNgbRanks; ++c) {
274*5f80ce2aSJacob Faibussowitsch     CHKERRMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicatorSize, parmesh, c, ngbRanks[c], intOffset[c+1]-intOffset[c]);
275*5f80ce2aSJacob Faibussowitsch     CHKERRMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicator_nodes, parmesh, c, &interfaces_lv[intOffset[c]], &interfaces_gv[intOffset[c]], 1);
2762bc0b47dSJoe Wallwork   }
277*5f80ce2aSJacob Faibussowitsch   CHKERRMMG(PMMG_parmmglib_distributed, parmesh);
278*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(cells));
279*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(metric, vertices));
280*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(bdFaces, faceTags));
281*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(verTags, cellTags));
2822bc0b47dSJoe Wallwork   if (numProcs > 1) {
283*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree2(ngbRanks, verNgbRank));
284*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree3(interfaces_lv, interfaces_gv, intOffset));
2852bc0b47dSJoe Wallwork   }
2862bc0b47dSJoe Wallwork 
2872f583692SJoe Wallwork   /* Retrieve mesh from Mmg */
2882bc0b47dSJoe Wallwork   numCornersNew = 4;
289*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Get_meshSize, parmesh, &numVerticesNew, &numCellsNew, 0, &numFacesNew, 0, 0);
290*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc4(dim*numVerticesNew, &verticesNew, numVerticesNew, &verTagsNew, numVerticesNew, &corners, numVerticesNew, &requiredVer));
291*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc3((dim+1)*numCellsNew, &cellsNew, numCellsNew, &cellTagsNew, numCellsNew, &requiredCells));
292*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc4(dim*numFacesNew, &facesNew, numFacesNew, &faceTagsNew, numFacesNew, &ridges, numFacesNew, &requiredFaces));
293*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Get_vertices, parmesh, verticesNew, verTagsNew, corners, requiredVer);
294*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Get_tetrahedra, parmesh, cellsNew, cellTagsNew, requiredCells);
295*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Get_triangles, parmesh, facesNew, faceTagsNew, requiredFaces);
296*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(numVerticesNew, &owners, numVerticesNew, &gv_new));
297*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Set_iparameter, parmesh, PMMG_IPARAM_globalNum, 1);
298*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Get_verticesGloNum, parmesh, gv_new, owners);
2992bc0b47dSJoe Wallwork   for (i = 0; i < dim*numFacesNew; ++i) facesNew[i] -= 1;
3000a7a67b9SPierre Jolivet   for (i = 0; i < (dim+1)*numCellsNew; ++i) cellsNew[i] = gv_new[cellsNew[i]-1]-1;
3010a7a67b9SPierre Jolivet   for (i = 0, numVerticesNewLoc = 0; i < numVerticesNew; ++i) {
3022bc0b47dSJoe Wallwork     if (owners[i] == rank) numVerticesNewLoc++;
3032bc0b47dSJoe Wallwork   }
304*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(numVerticesNewLoc*dim, &verticesNewLoc, numVerticesNew, &verticesNewSorted));
3052bc0b47dSJoe Wallwork   for (i = 0, c = 0; i < numVerticesNew; i++) {
3062bc0b47dSJoe Wallwork     if (owners[i] == rank) {
3072bc0b47dSJoe Wallwork       for (j=0; j<dim; ++j) verticesNewLoc[dim*c+j] = verticesNew[dim*i+j];
3082bc0b47dSJoe Wallwork       c++;
3092bc0b47dSJoe Wallwork     }
3102bc0b47dSJoe Wallwork   }
3112f583692SJoe Wallwork 
3122f583692SJoe Wallwork   /* Reorder for consistency with DMPlex */
313*5f80ce2aSJacob Faibussowitsch   for (i = 0; i < numCellsNew; ++i) CHKERRQ(DMPlexInvertCell(DM_POLYTOPE_TETRAHEDRON, &cellsNew[4*i]));
3142f583692SJoe Wallwork 
3152f583692SJoe Wallwork   /* Create new plex */
316*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexCreateFromCellListParallelPetsc(comm, dim, numCellsNew, numVerticesNewLoc, PETSC_DECIDE, numCornersNew, PETSC_TRUE, cellsNew, dim, verticesNewLoc, NULL, &verticesNewSorted, dmNew));
317*5f80ce2aSJacob Faibussowitsch   CHKERRMMG_NONSTANDARD(PMMG_Free_all, PMMG_ARG_start, PMMG_ARG_ppParMesh, &parmesh, PMMG_ARG_end);
318*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree4(verticesNew, verTagsNew, corners, requiredVer));
319c1dc6da0SJoe Wallwork 
320c1dc6da0SJoe Wallwork   /* Get adapted mesh information */
321*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetHeightStratum(*dmNew, 0, &cStart, &cEnd));
322*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetHeightStratum(*dmNew, 1, &fStart, &fEnd));
323*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMPlexGetDepthStratum(*dmNew, 0, &vStart, &vEnd));
3242bc0b47dSJoe Wallwork 
3252bc0b47dSJoe Wallwork   /* Rebuild boundary label */
326*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMCreateLabel(*dmNew, flg ? bdName : bdLabelName));
327*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetLabel(*dmNew, flg ? bdName : bdLabelName, &bdLabelNew));
3282bc0b47dSJoe Wallwork   for (i = 0; i < numFacesNew; i++) {
329c1dc6da0SJoe Wallwork     PetscBool       hasTag = PETSC_FALSE;
3302bc0b47dSJoe Wallwork     PetscInt        numCoveredPoints, numFaces = 0, facePoints[3];
3312bc0b47dSJoe Wallwork     const PetscInt *coveredPoints = NULL;
3322bc0b47dSJoe Wallwork 
3332bc0b47dSJoe Wallwork     for (j = 0; j < dim; ++j) {
3342bc0b47dSJoe Wallwork       lv = facesNew[i*dim+j];
3352bc0b47dSJoe Wallwork       gv = gv_new[lv]-1;
336*5f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFindInt(gv, numVerticesNew, verticesNewSorted, &lv));
3372bc0b47dSJoe Wallwork       facePoints[j] = lv+vStart;
3382bc0b47dSJoe Wallwork     }
339*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexGetFullJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
3402bc0b47dSJoe Wallwork     for (j = 0; j < numCoveredPoints; ++j) {
3412bc0b47dSJoe Wallwork       if (coveredPoints[j] >= fStart && coveredPoints[j] < fEnd) {
3422bc0b47dSJoe Wallwork         numFaces++;
3432bc0b47dSJoe Wallwork         f = j;
3442bc0b47dSJoe Wallwork       }
3452bc0b47dSJoe Wallwork     }
346*5f80ce2aSJacob Faibussowitsch     PetscCheck(numFaces == 1,comm, PETSC_ERR_ARG_OUTOFRANGE, "%" PetscInt_FMT " vertices cannot define more than 1 facet (%" PetscInt_FMT ")", dim, numFaces);
347*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMLabelHasStratum(bdLabel, faceTagsNew[i], &hasTag));
348*5f80ce2aSJacob Faibussowitsch     if (hasTag) CHKERRQ(DMLabelSetValue(bdLabelNew, coveredPoints[f], faceTagsNew[i]));
349*5f80ce2aSJacob Faibussowitsch     CHKERRQ(DMPlexRestoreJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
3502bc0b47dSJoe Wallwork   }
351*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree4(facesNew, faceTagsNew, ridges, requiredFaces));
352*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(owners, gv_new));
353*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(verticesNewLoc, verticesNewSorted));
354*5f80ce2aSJacob Faibussowitsch   if (flg) CHKERRQ(DMLabelDestroy(&bdLabel));
3552bc0b47dSJoe Wallwork 
3569fe9e680SJoe Wallwork   /* Rebuild cell labels */
357*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMCreateLabel(*dmNew, rgLabel ? rgLabelName : rgName));
358*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMGetLabel(*dmNew, rgLabel ? rgLabelName : rgName, &rgLabelNew));
359*5f80ce2aSJacob Faibussowitsch   for (c = cStart; c < cEnd; ++c) CHKERRQ(DMLabelSetValue(rgLabelNew, c, cellTagsNew[c-cStart]));
360*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree3(cellsNew, cellTagsNew, requiredCells));
361c1dc6da0SJoe Wallwork 
3622bc0b47dSJoe Wallwork   PetscFunctionReturn(0);
3632bc0b47dSJoe Wallwork }
364