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, °ree)); 219*8a7f912aSStephan Kramer PetscCall(PetscSFComputeDegreeEnd(sf, °ree)); 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