xref: /petsc/src/dm/impls/plex/adaptors/parmmg/parmmgadapt.c (revision d71ae5a4db6382e7f06317b8d368875286fe9008)
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*d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMAdaptMetric_ParMmg_Plex(DM dm, Vec vertexMetric, DMLabel bdLabel, DMLabel rgLabel, DM *dmNew)
14*d71ae5a4SJacob Faibussowitsch {
150a7a67b9SPierre Jolivet   MPI_Comm           comm;
162bc0b47dSJoe Wallwork   const char        *bdName = "_boundary_";
179fe9e680SJoe Wallwork   const char        *rgName = "_regions_";
182bc0b47dSJoe Wallwork   DM                 udm, cdm;
19c1dc6da0SJoe Wallwork   DMLabel            bdLabelNew, rgLabelNew;
209fe9e680SJoe Wallwork   const char        *bdLabelName, *rgLabelName;
21c1dc6da0SJoe Wallwork   IS                 globalVertexNum;
222bc0b47dSJoe Wallwork   PetscSection       coordSection;
232bc0b47dSJoe Wallwork   Vec                coordinates;
242bc0b47dSJoe Wallwork   PetscSF            sf;
252bc0b47dSJoe Wallwork   const PetscScalar *coords, *met;
26ae8b063eSJoe Wallwork   PetscReal         *vertices, *metric, *verticesNew, *verticesNewLoc, gradationFactor, hausdorffNumber;
27c1dc6da0SJoe Wallwork   PetscInt          *cells, *cellsNew, *cellTags, *cellTagsNew, *verTags, *verTagsNew;
28c1dc6da0SJoe Wallwork   PetscInt          *bdFaces, *faceTags, *facesNew, *faceTagsNew;
29c1dc6da0SJoe Wallwork   PetscInt          *corners, *requiredCells, *requiredVer, *ridges, *requiredFaces;
30c1dc6da0SJoe Wallwork   PetscInt           cStart, cEnd, c, numCells, fStart, fEnd, f, numFaceTags, vStart, vEnd, v, numVertices;
31c1dc6da0SJoe Wallwork   PetscInt           dim, off, coff, maxConeSize, bdSize, i, j, k, Neq, verbosity, numIter;
322bc0b47dSJoe Wallwork   PetscInt          *numVerInterfaces, *ngbRanks, *verNgbRank, *interfaces_lv, *interfaces_gv, *intOffset;
33c1dc6da0SJoe Wallwork   PetscInt           niranks, nrranks, numNgbRanks, numVerNgbRanksTotal, count, sliceSize, p, r, n, lv, gv;
34c1dc6da0SJoe Wallwork   PetscInt          *gv_new, *owners, *verticesNewSorted, pStart, pEnd;
352bc0b47dSJoe Wallwork   PetscInt           numCellsNew, numVerticesNew, numCornersNew, numFacesNew, numVerticesNewLoc;
36c1dc6da0SJoe Wallwork   const PetscInt    *gV, *ioffset, *irootloc, *roffset, *rmine, *rremote;
3776f360caSJoe Wallwork   PetscBool          flg = PETSC_FALSE, noInsert, noSwap, noMove, noSurf, isotropic, uniform;
382bc0b47dSJoe Wallwork   const PetscMPIInt *iranks, *rranks;
392bc0b47dSJoe Wallwork   PetscMPIInt        numProcs, rank;
402bc0b47dSJoe Wallwork   PMMG_pParMesh      parmesh = NULL;
412bc0b47dSJoe Wallwork 
422bc0b47dSJoe Wallwork   PetscFunctionBegin;
43b2c4de1dSPierre Jolivet   PetscCall(PetscCitationsRegister(ParMmgCitation, &ParMmgCite));
449566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
459566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &numProcs));
469566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
472bc0b47dSJoe Wallwork   if (bdLabel) {
489566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)bdLabel, &bdLabelName));
499566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(bdLabelName, bdName, &flg));
5028b400f6SJacob Faibussowitsch     PetscCheck(!flg, comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for boundary facets", bdLabelName);
512bc0b47dSJoe Wallwork   }
529fe9e680SJoe Wallwork   if (rgLabel) {
539566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)rgLabel, &rgLabelName));
549566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(rgLabelName, rgName, &flg));
5528b400f6SJacob Faibussowitsch     PetscCheck(!flg, comm, PETSC_ERR_ARG_WRONG, "\"%s\" cannot be used as label for element tags", rgLabelName);
569fe9e680SJoe Wallwork   }
572bc0b47dSJoe Wallwork 
582bc0b47dSJoe Wallwork   /* Get mesh information */
599566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
605f80ce2aSJacob Faibussowitsch   PetscCheck(dim == 3, comm, PETSC_ERR_ARG_OUTOFRANGE, "ParMmg only works in 3D.");
612bc0b47dSJoe Wallwork   Neq = (dim * (dim + 1)) / 2;
629566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
639566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
649566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
659566063dSJacob Faibussowitsch   PetscCall(DMPlexUninterpolate(dm, &udm));
669566063dSJacob Faibussowitsch   PetscCall(DMPlexGetMaxSizes(udm, &maxConeSize, NULL));
672bc0b47dSJoe Wallwork   numCells    = cEnd - cStart;
682bc0b47dSJoe Wallwork   numVertices = vEnd - vStart;
692bc0b47dSJoe Wallwork 
702bc0b47dSJoe Wallwork   /* Get cell offsets */
719566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(numCells * maxConeSize, &cells));
722bc0b47dSJoe Wallwork   for (c = 0, coff = 0; c < numCells; ++c) {
732bc0b47dSJoe Wallwork     const PetscInt *cone;
742bc0b47dSJoe Wallwork     PetscInt        coneSize, cl;
752bc0b47dSJoe Wallwork 
769566063dSJacob Faibussowitsch     PetscCall(DMPlexGetConeSize(udm, c, &coneSize));
779566063dSJacob Faibussowitsch     PetscCall(DMPlexGetCone(udm, c, &cone));
782bc0b47dSJoe Wallwork     for (cl = 0; cl < coneSize; ++cl) cells[coff++] = cone[cl] - vStart + 1;
792bc0b47dSJoe Wallwork   }
802bc0b47dSJoe Wallwork 
812bc0b47dSJoe Wallwork   /* Get vertex coordinate array */
829566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
839566063dSJacob Faibussowitsch   PetscCall(DMGetLocalSection(cdm, &coordSection));
849566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
859566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(coordinates, &coords));
869566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numVertices * Neq, &metric, dim * numVertices, &vertices));
872bc0b47dSJoe Wallwork   for (v = 0; v < vEnd - vStart; ++v) {
889566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(coordSection, v + vStart, &off));
892bc0b47dSJoe Wallwork     for (i = 0; i < dim; ++i) vertices[dim * v + i] = PetscRealPart(coords[off + i]);
902bc0b47dSJoe Wallwork   }
919566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(coordinates, &coords));
922bc0b47dSJoe Wallwork 
93c1dc6da0SJoe Wallwork   /* Get face tags */
94c1dc6da0SJoe Wallwork   if (!bdLabel) {
95c1dc6da0SJoe Wallwork     flg = PETSC_TRUE;
969566063dSJacob Faibussowitsch     PetscCall(DMLabelCreate(PETSC_COMM_SELF, bdName, &bdLabel));
979566063dSJacob Faibussowitsch     PetscCall(DMPlexMarkBoundaryFaces(dm, 1, bdLabel));
98c1dc6da0SJoe Wallwork   }
999566063dSJacob Faibussowitsch   PetscCall(DMLabelGetBounds(bdLabel, &pStart, &pEnd));
100c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
101c1dc6da0SJoe Wallwork     PetscBool hasPoint;
102c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
1032bc0b47dSJoe Wallwork 
1049566063dSJacob Faibussowitsch     PetscCall(DMLabelHasPoint(bdLabel, f, &hasPoint));
105c1dc6da0SJoe Wallwork     if ((!hasPoint) || (f < fStart) || (f >= fEnd)) continue;
106c1dc6da0SJoe Wallwork     numFaceTags++;
107c1dc6da0SJoe Wallwork 
1089566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1092bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize * 2; cl += 2) {
1102bc0b47dSJoe Wallwork       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) ++bdSize;
1112bc0b47dSJoe Wallwork     }
1129566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1132bc0b47dSJoe Wallwork   }
1149566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(bdSize, &bdFaces, numFaceTags, &faceTags));
115c1dc6da0SJoe Wallwork   for (f = pStart, bdSize = 0, numFaceTags = 0; f < pEnd; ++f) {
116c1dc6da0SJoe Wallwork     PetscBool hasPoint;
117c1dc6da0SJoe Wallwork     PetscInt *closure = NULL, closureSize, cl;
1182bc0b47dSJoe Wallwork 
1199566063dSJacob Faibussowitsch     PetscCall(DMLabelHasPoint(bdLabel, f, &hasPoint));
120c1dc6da0SJoe Wallwork     if ((!hasPoint) || (f < fStart) || (f >= fEnd)) continue;
121c1dc6da0SJoe Wallwork 
1229566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1232bc0b47dSJoe Wallwork     for (cl = 0; cl < closureSize * 2; cl += 2) {
1242bc0b47dSJoe Wallwork       if ((closure[cl] >= vStart) && (closure[cl] < vEnd)) bdFaces[bdSize++] = closure[cl] - vStart + 1;
1252bc0b47dSJoe Wallwork     }
1269566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &closureSize, &closure));
1279566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(bdLabel, f, &faceTags[numFaceTags++]));
1282bc0b47dSJoe Wallwork   }
1292bc0b47dSJoe Wallwork 
1309fe9e680SJoe Wallwork   /* Get cell tags */
1319566063dSJacob Faibussowitsch   PetscCall(PetscCalloc2(numVertices, &verTags, numCells, &cellTags));
1329fe9e680SJoe Wallwork   if (rgLabel) {
1339566063dSJacob Faibussowitsch     for (c = cStart; c < cEnd; ++c) PetscCall(DMLabelGetValue(rgLabel, c, &cellTags[c]));
1349fe9e680SJoe Wallwork   }
1359fe9e680SJoe Wallwork 
1362bc0b47dSJoe Wallwork   /* Get metric */
1379566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(vertexMetric, NULL, "-adapt_metric_view"));
1389566063dSJacob Faibussowitsch   PetscCall(VecGetArrayRead(vertexMetric, &met));
1399566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricIsIsotropic(dm, &isotropic));
1409566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricIsUniform(dm, &uniform));
1412bc0b47dSJoe Wallwork   for (v = 0; v < (vEnd - vStart); ++v) {
1422bc0b47dSJoe Wallwork     for (i = 0, k = 0; i < dim; ++i) {
1430a7a67b9SPierre Jolivet       for (j = i; j < dim; ++j, ++k) {
14493520041SJoe Wallwork         if (isotropic) {
14593520041SJoe Wallwork           if (i == j) {
14693520041SJoe Wallwork             if (uniform) metric[Neq * v + k] = PetscRealPart(met[0]);
14793520041SJoe Wallwork             else metric[Neq * v + k] = PetscRealPart(met[v]);
14893520041SJoe Wallwork           } else metric[Neq * v + k] = 0.0;
1490a7a67b9SPierre Jolivet         } else metric[Neq * v + k] = PetscRealPart(met[dim * dim * v + dim * i + j]);
1502bc0b47dSJoe Wallwork       }
1512bc0b47dSJoe Wallwork     }
1522bc0b47dSJoe Wallwork   }
1539566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayRead(vertexMetric, &met));
1542bc0b47dSJoe Wallwork 
1552bc0b47dSJoe Wallwork   /* Build ParMMG communicators: the list of vertices between two partitions  */
1562bc0b47dSJoe Wallwork   niranks = nrranks = 0;
1572bc0b47dSJoe Wallwork   numNgbRanks       = 0;
1582bc0b47dSJoe Wallwork   if (numProcs > 1) {
1599566063dSJacob Faibussowitsch     PetscCall(DMGetPointSF(dm, &sf));
1609566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
1619566063dSJacob Faibussowitsch     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc));
1629566063dSJacob Faibussowitsch     PetscCall(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote));
1639566063dSJacob Faibussowitsch     PetscCall(PetscCalloc1(numProcs, &numVerInterfaces));
1642bc0b47dSJoe Wallwork 
165c51fff1bSJoe Wallwork     /* Count number of roots associated with each leaf */
1662bc0b47dSJoe Wallwork     for (r = 0; r < niranks; ++r) {
1672bc0b47dSJoe Wallwork       for (i = ioffset[r], count = 0; i < ioffset[r + 1]; ++i) {
1682bc0b47dSJoe Wallwork         if (irootloc[i] >= vStart && irootloc[i] < vEnd) count++;
1692bc0b47dSJoe Wallwork       }
1702bc0b47dSJoe Wallwork       numVerInterfaces[iranks[r]] += count;
1712bc0b47dSJoe Wallwork     }
172c51fff1bSJoe Wallwork 
173c51fff1bSJoe Wallwork     /* Count number of leaves associated with each root */
1742bc0b47dSJoe Wallwork     for (r = 0; r < nrranks; ++r) {
1752bc0b47dSJoe Wallwork       for (i = roffset[r], count = 0; i < roffset[r + 1]; ++i) {
1762bc0b47dSJoe Wallwork         if (rmine[i] >= vStart && rmine[i] < vEnd) count++;
1772bc0b47dSJoe Wallwork       }
1782bc0b47dSJoe Wallwork       numVerInterfaces[rranks[r]] += count;
1792bc0b47dSJoe Wallwork     }
180c51fff1bSJoe Wallwork 
181c51fff1bSJoe Wallwork     /* Count global number of ranks */
1822bc0b47dSJoe Wallwork     for (p = 0; p < numProcs; ++p) {
1832bc0b47dSJoe Wallwork       if (numVerInterfaces[p]) numNgbRanks++;
1842bc0b47dSJoe Wallwork     }
185c51fff1bSJoe Wallwork 
186c51fff1bSJoe Wallwork     /* Provide numbers of vertex interfaces */
1879566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(numNgbRanks, &ngbRanks, numNgbRanks, &verNgbRank));
1882bc0b47dSJoe Wallwork     for (p = 0, n = 0; p < numProcs; ++p) {
1892bc0b47dSJoe Wallwork       if (numVerInterfaces[p]) {
1902bc0b47dSJoe Wallwork         ngbRanks[n]   = p;
1912bc0b47dSJoe Wallwork         verNgbRank[n] = numVerInterfaces[p];
1922bc0b47dSJoe Wallwork         n++;
1932bc0b47dSJoe Wallwork       }
1942bc0b47dSJoe Wallwork     }
1952bc0b47dSJoe Wallwork     numVerNgbRanksTotal = 0;
1962bc0b47dSJoe Wallwork     for (p = 0; p < numNgbRanks; ++p) numVerNgbRanksTotal += verNgbRank[p];
1972bc0b47dSJoe Wallwork 
1982bc0b47dSJoe Wallwork     /* For each neighbor, fill in interface arrays */
1999566063dSJacob Faibussowitsch     PetscCall(PetscMalloc3(numVerNgbRanksTotal, &interfaces_lv, numVerNgbRanksTotal, &interfaces_gv, numNgbRanks + 1, &intOffset));
2002bc0b47dSJoe Wallwork     intOffset[0] = 0;
2012bc0b47dSJoe Wallwork     for (p = 0, r = 0, i = 0; p < numNgbRanks; ++p) {
2022bc0b47dSJoe Wallwork       intOffset[p + 1] = intOffset[p];
203c51fff1bSJoe Wallwork 
204c51fff1bSJoe Wallwork       /* Leaf case */
2052bc0b47dSJoe Wallwork       if (iranks && iranks[i] == ngbRanks[p]) {
2062bc0b47dSJoe Wallwork         /* Add the right slice of irootloc at the right place */
2072bc0b47dSJoe Wallwork         sliceSize = ioffset[i + 1] - ioffset[i];
2082bc0b47dSJoe Wallwork         for (j = 0, count = 0; j < sliceSize; ++j) {
2095f80ce2aSJacob 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]);
2102bc0b47dSJoe Wallwork           v = irootloc[ioffset[i] + j];
2112bc0b47dSJoe Wallwork           if (v >= vStart && v < vEnd) {
2125f80ce2aSJacob 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);
2132bc0b47dSJoe Wallwork             interfaces_lv[intOffset[p + 1] + count] = v - vStart;
2142bc0b47dSJoe Wallwork             count++;
2152bc0b47dSJoe Wallwork           }
2162bc0b47dSJoe Wallwork         }
2172bc0b47dSJoe Wallwork         intOffset[p + 1] += count;
2182bc0b47dSJoe Wallwork         i++;
2192bc0b47dSJoe Wallwork       }
220c51fff1bSJoe Wallwork 
221c51fff1bSJoe Wallwork       /* Root case */
2222bc0b47dSJoe Wallwork       if (rranks && rranks[r] == ngbRanks[p]) {
2232bc0b47dSJoe Wallwork         /* Add the right slice of rmine at the right place */
2242bc0b47dSJoe Wallwork         sliceSize = roffset[r + 1] - roffset[r];
2252bc0b47dSJoe Wallwork         for (j = 0, count = 0; j < sliceSize; ++j) {
2265f80ce2aSJacob 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]);
2272bc0b47dSJoe Wallwork           v = rmine[roffset[r] + j];
2282bc0b47dSJoe Wallwork           if (v >= vStart && v < vEnd) {
2295f80ce2aSJacob 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);
2302bc0b47dSJoe Wallwork             interfaces_lv[intOffset[p + 1] + count] = v - vStart;
2312bc0b47dSJoe Wallwork             count++;
2322bc0b47dSJoe Wallwork           }
2332bc0b47dSJoe Wallwork         }
2342bc0b47dSJoe Wallwork         intOffset[p + 1] += count;
2352bc0b47dSJoe Wallwork         r++;
2362bc0b47dSJoe Wallwork       }
237c51fff1bSJoe Wallwork 
238c51fff1bSJoe Wallwork       /* Check validity of offsets */
2395f80ce2aSJacob 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]);
2402bc0b47dSJoe Wallwork     }
2419566063dSJacob Faibussowitsch     PetscCall(DMPlexGetVertexNumbering(udm, &globalVertexNum));
2429566063dSJacob Faibussowitsch     PetscCall(ISGetIndices(globalVertexNum, &gV));
2432bc0b47dSJoe Wallwork     for (i = 0; i < numVerNgbRanksTotal; ++i) {
2442bc0b47dSJoe Wallwork       v                = gV[interfaces_lv[i]];
2452bc0b47dSJoe Wallwork       interfaces_gv[i] = v < 0 ? -v - 1 : v;
2462bc0b47dSJoe Wallwork       interfaces_lv[i] += 1;
2472bc0b47dSJoe Wallwork       interfaces_gv[i] += 1;
2482bc0b47dSJoe Wallwork     }
2499566063dSJacob Faibussowitsch     PetscCall(ISRestoreIndices(globalVertexNum, &gV));
2509566063dSJacob Faibussowitsch     PetscCall(PetscFree(numVerInterfaces));
2512bc0b47dSJoe Wallwork   }
2529566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&udm));
2532bc0b47dSJoe Wallwork 
2542bc0b47dSJoe Wallwork   /* Send the data to ParMmg and remesh */
2559566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoInsertion(dm, &noInsert));
2569566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoSwapping(dm, &noSwap));
2579566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoMovement(dm, &noMove));
2589566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricNoSurf(dm, &noSurf));
2599566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetVerbosity(dm, &verbosity));
2609566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetNumIterations(dm, &numIter));
2619566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetGradationFactor(dm, &gradationFactor));
2629566063dSJacob Faibussowitsch   PetscCall(DMPlexMetricGetHausdorffNumber(dm, &hausdorffNumber));
2639566063dSJacob Faibussowitsch   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));
2649566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_meshSize(parmesh, numVertices, numCells, 0, numFaceTags, 0, 0));
2659566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_APImode, PMMG_APIDISTRIB_nodes));
2669566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_noinsert, noInsert));
2679566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_noswap, noSwap));
2689566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_nomove, noMove));
2699566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_nosurf, noSurf));
2709566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_verbose, verbosity));
2719566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_globalNum, 1));
2729566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_niter, numIter));
2739566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_dparameter(parmesh, PMMG_DPARAM_hgrad, gradationFactor));
2749566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_dparameter(parmesh, PMMG_DPARAM_hausd, hausdorffNumber));
2759566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_vertices(parmesh, vertices, verTags));
2769566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_tetrahedra(parmesh, cells, cellTags));
2779566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_triangles(parmesh, bdFaces, faceTags));
2789566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_metSize(parmesh, MMG5_Vertex, numVertices, MMG5_Tensor));
2799566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_tensorMets(parmesh, metric));
2809566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_numberOfNodeCommunicators(parmesh, numNgbRanks));
2812bc0b47dSJoe Wallwork   for (c = 0; c < numNgbRanks; ++c) {
2829566063dSJacob Faibussowitsch     PetscCallMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicatorSize(parmesh, c, ngbRanks[c], intOffset[c + 1] - intOffset[c]));
2839566063dSJacob Faibussowitsch     PetscCallMMG_NONSTANDARD(PMMG_Set_ithNodeCommunicator_nodes(parmesh, c, &interfaces_lv[intOffset[c]], &interfaces_gv[intOffset[c]], 1));
2842bc0b47dSJoe Wallwork   }
2859566063dSJacob Faibussowitsch   PetscCallMMG(PMMG_parmmglib_distributed(parmesh));
2869566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells));
2879566063dSJacob Faibussowitsch   PetscCall(PetscFree2(metric, vertices));
2889566063dSJacob Faibussowitsch   PetscCall(PetscFree2(bdFaces, faceTags));
2899566063dSJacob Faibussowitsch   PetscCall(PetscFree2(verTags, cellTags));
2902bc0b47dSJoe Wallwork   if (numProcs > 1) {
2919566063dSJacob Faibussowitsch     PetscCall(PetscFree2(ngbRanks, verNgbRank));
2929566063dSJacob Faibussowitsch     PetscCall(PetscFree3(interfaces_lv, interfaces_gv, intOffset));
2932bc0b47dSJoe Wallwork   }
2942bc0b47dSJoe Wallwork 
2952f583692SJoe Wallwork   /* Retrieve mesh from Mmg */
2962bc0b47dSJoe Wallwork   numCornersNew = 4;
2979566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Get_meshSize(parmesh, &numVerticesNew, &numCellsNew, 0, &numFacesNew, 0, 0));
2989566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(dim * numVerticesNew, &verticesNew, numVerticesNew, &verTagsNew, numVerticesNew, &corners, numVerticesNew, &requiredVer));
2999566063dSJacob Faibussowitsch   PetscCall(PetscMalloc3((dim + 1) * numCellsNew, &cellsNew, numCellsNew, &cellTagsNew, numCellsNew, &requiredCells));
3009566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(dim * numFacesNew, &facesNew, numFacesNew, &faceTagsNew, numFacesNew, &ridges, numFacesNew, &requiredFaces));
3019566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Get_vertices(parmesh, verticesNew, verTagsNew, corners, requiredVer));
3029566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Get_tetrahedra(parmesh, cellsNew, cellTagsNew, requiredCells));
3039566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Get_triangles(parmesh, facesNew, faceTagsNew, requiredFaces));
3049566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numVerticesNew, &owners, numVerticesNew, &gv_new));
3059566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Set_iparameter(parmesh, PMMG_IPARAM_globalNum, 1));
3069566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Get_verticesGloNum(parmesh, gv_new, owners));
3072bc0b47dSJoe Wallwork   for (i = 0; i < dim * numFacesNew; ++i) facesNew[i] -= 1;
3080a7a67b9SPierre Jolivet   for (i = 0; i < (dim + 1) * numCellsNew; ++i) cellsNew[i] = gv_new[cellsNew[i] - 1] - 1;
3090a7a67b9SPierre Jolivet   for (i = 0, numVerticesNewLoc = 0; i < numVerticesNew; ++i) {
3102bc0b47dSJoe Wallwork     if (owners[i] == rank) numVerticesNewLoc++;
3112bc0b47dSJoe Wallwork   }
3129566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numVerticesNewLoc * dim, &verticesNewLoc, numVerticesNew, &verticesNewSorted));
3132bc0b47dSJoe Wallwork   for (i = 0, c = 0; i < numVerticesNew; i++) {
3142bc0b47dSJoe Wallwork     if (owners[i] == rank) {
3152bc0b47dSJoe Wallwork       for (j = 0; j < dim; ++j) verticesNewLoc[dim * c + j] = verticesNew[dim * i + j];
3162bc0b47dSJoe Wallwork       c++;
3172bc0b47dSJoe Wallwork     }
3182bc0b47dSJoe Wallwork   }
3192f583692SJoe Wallwork 
3202f583692SJoe Wallwork   /* Reorder for consistency with DMPlex */
3219566063dSJacob Faibussowitsch   for (i = 0; i < numCellsNew; ++i) PetscCall(DMPlexInvertCell(DM_POLYTOPE_TETRAHEDRON, &cellsNew[4 * i]));
3222f583692SJoe Wallwork 
3232f583692SJoe Wallwork   /* Create new plex */
3249566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromCellListParallelPetsc(comm, dim, numCellsNew, numVerticesNewLoc, PETSC_DECIDE, numCornersNew, PETSC_TRUE, cellsNew, dim, verticesNewLoc, NULL, &verticesNewSorted, dmNew));
3259566063dSJacob Faibussowitsch   PetscCallMMG_NONSTANDARD(PMMG_Free_all(PMMG_ARG_start, PMMG_ARG_ppParMesh, &parmesh, PMMG_ARG_end));
3269566063dSJacob Faibussowitsch   PetscCall(PetscFree4(verticesNew, verTagsNew, corners, requiredVer));
327c1dc6da0SJoe Wallwork 
328c1dc6da0SJoe Wallwork   /* Get adapted mesh information */
3299566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(*dmNew, 0, &cStart, &cEnd));
3309566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(*dmNew, 1, &fStart, &fEnd));
3319566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(*dmNew, 0, &vStart, &vEnd));
3322bc0b47dSJoe Wallwork 
3332bc0b47dSJoe Wallwork   /* Rebuild boundary label */
3349566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(*dmNew, flg ? bdName : bdLabelName));
3359566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(*dmNew, flg ? bdName : bdLabelName, &bdLabelNew));
3362bc0b47dSJoe Wallwork   for (i = 0; i < numFacesNew; i++) {
337c1dc6da0SJoe Wallwork     PetscBool       hasTag = PETSC_FALSE;
3382bc0b47dSJoe Wallwork     PetscInt        numCoveredPoints, numFaces = 0, facePoints[3];
3392bc0b47dSJoe Wallwork     const PetscInt *coveredPoints = NULL;
3402bc0b47dSJoe Wallwork 
3412bc0b47dSJoe Wallwork     for (j = 0; j < dim; ++j) {
3422bc0b47dSJoe Wallwork       lv = facesNew[i * dim + j];
3432bc0b47dSJoe Wallwork       gv = gv_new[lv] - 1;
3449566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesNew, verticesNewSorted, &lv));
3452bc0b47dSJoe Wallwork       facePoints[j] = lv + vStart;
3462bc0b47dSJoe Wallwork     }
3479566063dSJacob Faibussowitsch     PetscCall(DMPlexGetFullJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
3482bc0b47dSJoe Wallwork     for (j = 0; j < numCoveredPoints; ++j) {
3492bc0b47dSJoe Wallwork       if (coveredPoints[j] >= fStart && coveredPoints[j] < fEnd) {
3502bc0b47dSJoe Wallwork         numFaces++;
3512bc0b47dSJoe Wallwork         f = j;
3522bc0b47dSJoe Wallwork       }
3532bc0b47dSJoe Wallwork     }
3545f80ce2aSJacob Faibussowitsch     PetscCheck(numFaces == 1, comm, PETSC_ERR_ARG_OUTOFRANGE, "%" PetscInt_FMT " vertices cannot define more than 1 facet (%" PetscInt_FMT ")", dim, numFaces);
3559566063dSJacob Faibussowitsch     PetscCall(DMLabelHasStratum(bdLabel, faceTagsNew[i], &hasTag));
3569566063dSJacob Faibussowitsch     if (hasTag) PetscCall(DMLabelSetValue(bdLabelNew, coveredPoints[f], faceTagsNew[i]));
3579566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(*dmNew, dim, facePoints, &numCoveredPoints, &coveredPoints));
3582bc0b47dSJoe Wallwork   }
3599566063dSJacob Faibussowitsch   PetscCall(PetscFree4(facesNew, faceTagsNew, ridges, requiredFaces));
3609566063dSJacob Faibussowitsch   PetscCall(PetscFree2(owners, gv_new));
3619566063dSJacob Faibussowitsch   PetscCall(PetscFree2(verticesNewLoc, verticesNewSorted));
3629566063dSJacob Faibussowitsch   if (flg) PetscCall(DMLabelDestroy(&bdLabel));
3632bc0b47dSJoe Wallwork 
3649fe9e680SJoe Wallwork   /* Rebuild cell labels */
3659566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(*dmNew, rgLabel ? rgLabelName : rgName));
3669566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(*dmNew, rgLabel ? rgLabelName : rgName, &rgLabelNew));
3679566063dSJacob Faibussowitsch   for (c = cStart; c < cEnd; ++c) PetscCall(DMLabelSetValue(rgLabelNew, c, cellTagsNew[c - cStart]));
3689566063dSJacob Faibussowitsch   PetscCall(PetscFree3(cellsNew, cellTagsNew, requiredCells));
369c1dc6da0SJoe Wallwork 
3702bc0b47dSJoe Wallwork   PetscFunctionReturn(0);
3712bc0b47dSJoe Wallwork }
372