13a074057SBarry Smith #include <petsc/private/dmpleximpl.h> /*I "petscdmplex.h" I*/ 23a074057SBarry Smith 30fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS 40fdc7489SMatthew Knepley #include <egads.h> 50fdc7489SMatthew Knepley #endif 60fdc7489SMatthew Knepley 73a074057SBarry Smith #include <ctetgen.h> 83a074057SBarry Smith 93a074057SBarry Smith /* This is to fix the tetrahedron orientation from TetGen */ 10d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexInvertCells_CTetgen(PetscInt numCells, PetscInt numCorners, PetscInt cells[]) 11d71ae5a4SJacob Faibussowitsch { 123a074057SBarry Smith PetscInt bound = numCells * numCorners, coff; 133a074057SBarry Smith 143a074057SBarry Smith PetscFunctionBegin; 159371c9d4SSatish Balay #define SWAP(a, b) \ 169371c9d4SSatish Balay do { \ 179371c9d4SSatish Balay PetscInt tmp = (a); \ 189371c9d4SSatish Balay (a) = (b); \ 199371c9d4SSatish Balay (b) = tmp; \ 209371c9d4SSatish Balay } while (0) 2196ca5757SLisandro Dalcin for (coff = 0; coff < bound; coff += numCorners) SWAP(cells[coff], cells[coff + 1]); 2296ca5757SLisandro Dalcin #undef SWAP 233a074057SBarry Smith PetscFunctionReturn(0); 243a074057SBarry Smith } 253a074057SBarry Smith 26d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMPlexGenerate_CTetgen(DM boundary, PetscBool interpolate, DM *dm) 27d71ae5a4SJacob Faibussowitsch { 283a074057SBarry Smith MPI_Comm comm; 293a074057SBarry Smith const PetscInt dim = 3; 303a074057SBarry Smith PLC *in, *out; 310fdc7489SMatthew Knepley DMUniversalLabel universal; 32af226901SMatthew G. Knepley PetscInt vStart, vEnd, v, eStart, eEnd, e, fStart, fEnd, f, defVal, verbose = 0; 330fdc7489SMatthew Knepley DMPlexInterpolatedFlag isInterpolated; 343a074057SBarry Smith PetscMPIInt rank; 353a074057SBarry Smith 363a074057SBarry Smith PetscFunctionBegin; 379566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)boundary)->prefix, "-ctetgen_verbose", &verbose, NULL)); 389566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)boundary, &comm)); 399566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 409566063dSJacob Faibussowitsch PetscCall(DMPlexIsInterpolatedCollective(boundary, &isInterpolated)); 419566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelCreate(boundary, &universal)); 42af226901SMatthew G. Knepley PetscCall(DMLabelGetDefaultValue(universal->label, &defVal)); 430fdc7489SMatthew Knepley 449566063dSJacob Faibussowitsch PetscCall(PLCCreate(&in)); 459566063dSJacob Faibussowitsch PetscCall(PLCCreate(&out)); 463a074057SBarry Smith 479566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(boundary, 0, &vStart, &vEnd)); 483a074057SBarry Smith in->numberofpoints = vEnd - vStart; 493a074057SBarry Smith if (in->numberofpoints > 0) { 503a074057SBarry Smith PetscSection coordSection; 513a074057SBarry Smith Vec coordinates; 520fdc7489SMatthew Knepley const PetscScalar *array; 533a074057SBarry Smith 549566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofpoints * dim, &in->pointlist)); 5541e9d8b5SMatthew G. Knepley PetscCall(PetscCalloc1(in->numberofpoints, &in->pointmarkerlist)); 569566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(boundary, &coordinates)); 579566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(boundary, &coordSection)); 589566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(coordinates, &array)); 593a074057SBarry Smith for (v = vStart; v < vEnd; ++v) { 603a074057SBarry Smith const PetscInt idx = v - vStart; 610fdc7489SMatthew Knepley PetscInt off, d, m; 623a074057SBarry Smith 639566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(coordSection, v, &off)); 640fdc7489SMatthew Knepley for (d = 0; d < dim; ++d) in->pointlist[idx * dim + d] = PetscRealPart(array[off + d]); 659566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, v, &m)); 66af226901SMatthew G. Knepley if (m != defVal) in->pointmarkerlist[idx] = (int)m; 673a074057SBarry Smith } 689566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(coordinates, &array)); 693a074057SBarry Smith } 703a074057SBarry Smith 719566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(boundary, 1, &eStart, &eEnd)); 720fdc7489SMatthew Knepley in->numberofedges = eEnd - eStart; 730fdc7489SMatthew Knepley if (isInterpolated == DMPLEX_INTERPOLATED_FULL && in->numberofedges > 0) { 749566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofedges * 2, &in->edgelist)); 759566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofedges, &in->edgemarkerlist)); 760fdc7489SMatthew Knepley for (e = eStart; e < eEnd; ++e) { 770fdc7489SMatthew Knepley const PetscInt idx = e - eStart; 780fdc7489SMatthew Knepley const PetscInt *cone; 790fdc7489SMatthew Knepley PetscInt coneSize, val; 800fdc7489SMatthew Knepley 819566063dSJacob Faibussowitsch PetscCall(DMPlexGetConeSize(boundary, e, &coneSize)); 829566063dSJacob Faibussowitsch PetscCall(DMPlexGetCone(boundary, e, &cone)); 830fdc7489SMatthew Knepley in->edgelist[idx * 2] = cone[0] - vStart; 840fdc7489SMatthew Knepley in->edgelist[idx * 2 + 1] = cone[1] - vStart; 850fdc7489SMatthew Knepley 869566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, e, &val)); 87af226901SMatthew G. Knepley if (val != defVal) in->edgemarkerlist[idx] = (int)val; 880fdc7489SMatthew Knepley } 890fdc7489SMatthew Knepley } 900fdc7489SMatthew Knepley 919566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(boundary, 0, &fStart, &fEnd)); 923a074057SBarry Smith in->numberoffacets = fEnd - fStart; 933a074057SBarry Smith if (in->numberoffacets > 0) { 949566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberoffacets, &in->facetlist)); 959566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberoffacets, &in->facetmarkerlist)); 963a074057SBarry Smith for (f = fStart; f < fEnd; ++f) { 973a074057SBarry Smith const PetscInt idx = f - fStart; 983a074057SBarry Smith PetscInt *points = NULL, numPoints, p, numVertices = 0, v, m = -1; 993a074057SBarry Smith polygon *poly; 1003a074057SBarry Smith 1013a074057SBarry Smith in->facetlist[idx].numberofpolygons = 1; 1029566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->facetlist[idx].numberofpolygons, &in->facetlist[idx].polygonlist)); 1033a074057SBarry Smith in->facetlist[idx].numberofholes = 0; 1043a074057SBarry Smith in->facetlist[idx].holelist = NULL; 1053a074057SBarry Smith 1069566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(boundary, f, PETSC_TRUE, &numPoints, &points)); 1073a074057SBarry Smith for (p = 0; p < numPoints * 2; p += 2) { 1083a074057SBarry Smith const PetscInt point = points[p]; 1093a074057SBarry Smith if ((point >= vStart) && (point < vEnd)) points[numVertices++] = point; 1103a074057SBarry Smith } 1113a074057SBarry Smith 1123a074057SBarry Smith poly = in->facetlist[idx].polygonlist; 1133a074057SBarry Smith poly->numberofvertices = numVertices; 1149566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(poly->numberofvertices, &poly->vertexlist)); 1153a074057SBarry Smith for (v = 0; v < numVertices; ++v) { 1163a074057SBarry Smith const PetscInt vIdx = points[v] - vStart; 1173a074057SBarry Smith poly->vertexlist[v] = vIdx; 1183a074057SBarry Smith } 1199566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, f, &m)); 120af226901SMatthew G. Knepley if (m != defVal) in->facetmarkerlist[idx] = (int)m; 1219566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(boundary, f, PETSC_TRUE, &numPoints, &points)); 1223a074057SBarry Smith } 1233a074057SBarry Smith } 124dd400576SPatrick Sanan if (rank == 0) { 1253a074057SBarry Smith TetGenOpts t; 1263a074057SBarry Smith 1279566063dSJacob Faibussowitsch PetscCall(TetGenOptsInitialize(&t)); 1283a074057SBarry Smith t.in = boundary; /* Should go away */ 1293a074057SBarry Smith t.plc = 1; 1303a074057SBarry Smith t.quality = 1; 1313a074057SBarry Smith t.edgesout = 1; 1323a074057SBarry Smith t.zeroindex = 1; 1333a074057SBarry Smith t.quiet = 1; 1343a074057SBarry Smith t.verbose = verbose; 1350fdc7489SMatthew Knepley #if 0 1360fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS 1370fdc7489SMatthew Knepley /* Need to add in more TetGen code */ 1380fdc7489SMatthew Knepley t.nobisect = 1; /* Add Y to preserve Surface Mesh for EGADS */ 1390fdc7489SMatthew Knepley #endif 1400fdc7489SMatthew Knepley #endif 1410fdc7489SMatthew Knepley 1429566063dSJacob Faibussowitsch PetscCall(TetGenCheckOpts(&t)); 1439566063dSJacob Faibussowitsch PetscCall(TetGenTetrahedralize(&t, in, out)); 1443a074057SBarry Smith } 1453a074057SBarry Smith { 1463a074057SBarry Smith const PetscInt numCorners = 4; 1473a074057SBarry Smith const PetscInt numCells = out->numberoftetrahedra; 1483a074057SBarry Smith const PetscInt numVertices = out->numberofpoints; 1490fdc7489SMatthew Knepley PetscReal *meshCoords = NULL; 1500fdc7489SMatthew Knepley PetscInt *cells = NULL; 1513a074057SBarry Smith 152a4a685f2SJacob Faibussowitsch if (sizeof(PetscReal) == sizeof(out->pointlist[0])) { 153a4a685f2SJacob Faibussowitsch meshCoords = (PetscReal *)out->pointlist; 154a4a685f2SJacob Faibussowitsch } else { 1553a074057SBarry Smith PetscInt i; 1563a074057SBarry Smith 1579566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim * numVertices, &meshCoords)); 1580fdc7489SMatthew Knepley for (i = 0; i < dim * numVertices; ++i) meshCoords[i] = (PetscReal)out->pointlist[i]; 1593a074057SBarry Smith } 160a4a685f2SJacob Faibussowitsch if (sizeof(PetscInt) == sizeof(out->tetrahedronlist[0])) { 161a4a685f2SJacob Faibussowitsch cells = (PetscInt *)out->tetrahedronlist; 162a4a685f2SJacob Faibussowitsch } else { 163a4a685f2SJacob Faibussowitsch PetscInt i; 164a4a685f2SJacob Faibussowitsch 1659566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(numCells * numCorners, &cells)); 1660fdc7489SMatthew Knepley for (i = 0; i < numCells * numCorners; i++) cells[i] = (PetscInt)out->tetrahedronlist[i]; 167a4a685f2SJacob Faibussowitsch } 1683a074057SBarry Smith 1699566063dSJacob Faibussowitsch PetscCall(DMPlexInvertCells_CTetgen(numCells, numCorners, cells)); 1709566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromCellListPetsc(comm, dim, numCells, numVertices, numCorners, interpolate, cells, dim, meshCoords, dm)); 17148a46eb9SPierre Jolivet if (sizeof(PetscReal) != sizeof(out->pointlist[0])) PetscCall(PetscFree(meshCoords)); 17248a46eb9SPierre Jolivet if (sizeof(PetscInt) != sizeof(out->tetrahedronlist[0])) PetscCall(PetscFree(cells)); 1730fdc7489SMatthew Knepley 1743a074057SBarry Smith /* Set labels */ 1759566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelCreateLabels(universal, PETSC_TRUE, *dm)); 1763a074057SBarry Smith for (v = 0; v < numVertices; ++v) { 17748a46eb9SPierre Jolivet if (out->pointmarkerlist[v]) PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, v + numCells, out->pointmarkerlist[v])); 1783a074057SBarry Smith } 1793a074057SBarry Smith if (interpolate) { 1803a074057SBarry Smith PetscInt e; 1813a074057SBarry Smith 1823a074057SBarry Smith for (e = 0; e < out->numberofedges; e++) { 1833a074057SBarry Smith if (out->edgemarkerlist[e]) { 1843a074057SBarry Smith const PetscInt vertices[2] = {out->edgelist[e * 2 + 0] + numCells, out->edgelist[e * 2 + 1] + numCells}; 1853a074057SBarry Smith const PetscInt *edges; 1863a074057SBarry Smith PetscInt numEdges; 1873a074057SBarry Smith 1889566063dSJacob Faibussowitsch PetscCall(DMPlexGetJoin(*dm, 2, vertices, &numEdges, &edges)); 18963a3b9bcSJacob Faibussowitsch PetscCheck(numEdges == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Two vertices must cover only one edge, not %" PetscInt_FMT, numEdges); 1909566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, edges[0], out->edgemarkerlist[e])); 1919566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreJoin(*dm, 2, vertices, &numEdges, &edges)); 1923a074057SBarry Smith } 1933a074057SBarry Smith } 1943a074057SBarry Smith for (f = 0; f < out->numberoftrifaces; f++) { 1953a074057SBarry Smith if (out->trifacemarkerlist[f]) { 1963a074057SBarry Smith const PetscInt vertices[3] = {out->trifacelist[f * 3 + 0] + numCells, out->trifacelist[f * 3 + 1] + numCells, out->trifacelist[f * 3 + 2] + numCells}; 1973a074057SBarry Smith const PetscInt *faces; 1983a074057SBarry Smith PetscInt numFaces; 1993a074057SBarry Smith 2009566063dSJacob Faibussowitsch PetscCall(DMPlexGetFullJoin(*dm, 3, vertices, &numFaces, &faces)); 20163a3b9bcSJacob Faibussowitsch PetscCheck(numFaces == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Three vertices must cover only one face, not %" PetscInt_FMT, numFaces); 2029566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, faces[0], out->trifacemarkerlist[f])); 2039566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreJoin(*dm, 3, vertices, &numFaces, &faces)); 2043a074057SBarry Smith } 2053a074057SBarry Smith } 2063a074057SBarry Smith } 2070fdc7489SMatthew Knepley 2080fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS 2090fdc7489SMatthew Knepley { 2100fdc7489SMatthew Knepley DMLabel bodyLabel; 2110fdc7489SMatthew Knepley PetscContainer modelObj; 2120fdc7489SMatthew Knepley PetscInt cStart, cEnd, c, eStart, eEnd, fStart, fEnd; 2130fdc7489SMatthew Knepley ego *bodies; 2140fdc7489SMatthew Knepley ego model, geom; 2150fdc7489SMatthew Knepley int Nb, oclass, mtype, *senses; 2160fdc7489SMatthew Knepley 2170fdc7489SMatthew Knepley /* Get Attached EGADS Model from Original DMPlex */ 2189566063dSJacob Faibussowitsch PetscCall(PetscObjectQuery((PetscObject)boundary, "EGADS Model", (PetscObject *)&modelObj)); 2190fdc7489SMatthew Knepley if (modelObj) { 2209566063dSJacob Faibussowitsch PetscCall(PetscContainerGetPointer(modelObj, (void **)&model)); 2219566063dSJacob Faibussowitsch PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses)); 2220fdc7489SMatthew Knepley /* Transfer EGADS Model to Volumetric Mesh */ 2239566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)*dm, "EGADS Model", (PetscObject)modelObj)); 2240fdc7489SMatthew Knepley 2250fdc7489SMatthew Knepley /* Set Cell Labels */ 2269566063dSJacob Faibussowitsch PetscCall(DMGetLabel(*dm, "EGADS Body ID", &bodyLabel)); 2279566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(*dm, 0, &cStart, &cEnd)); 2289566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(*dm, 1, &fStart, &fEnd)); 2299566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(*dm, 1, &eStart, &eEnd)); 2300fdc7489SMatthew Knepley 2310fdc7489SMatthew Knepley for (c = cStart; c < cEnd; ++c) { 2320fdc7489SMatthew Knepley PetscReal centroid[3] = {0., 0., 0.}; 2330fdc7489SMatthew Knepley PetscInt b; 2340fdc7489SMatthew Knepley 235*35cb6cd3SPierre Jolivet /* Determine what body the cell's centroid is located in */ 2360fdc7489SMatthew Knepley if (!interpolate) { 2370fdc7489SMatthew Knepley PetscSection coordSection; 2380fdc7489SMatthew Knepley Vec coordinates; 2390fdc7489SMatthew Knepley PetscScalar *coords = NULL; 2400fdc7489SMatthew Knepley PetscInt coordSize, s, d; 2410fdc7489SMatthew Knepley 2429566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(*dm, &coordinates)); 2439566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(*dm, &coordSection)); 2449566063dSJacob Faibussowitsch PetscCall(DMPlexVecGetClosure(*dm, coordSection, coordinates, c, &coordSize, &coords)); 2459371c9d4SSatish Balay for (s = 0; s < coordSize; ++s) 2469371c9d4SSatish Balay for (d = 0; d < dim; ++d) centroid[d] += coords[s * dim + d]; 2479566063dSJacob Faibussowitsch PetscCall(DMPlexVecRestoreClosure(*dm, coordSection, coordinates, c, &coordSize, &coords)); 2481baa6e33SBarry Smith } else PetscCall(DMPlexComputeCellGeometryFVM(*dm, c, NULL, centroid, NULL)); 2490fdc7489SMatthew Knepley for (b = 0; b < Nb; ++b) { 2500fdc7489SMatthew Knepley if (EG_inTopology(bodies[b], centroid) == EGADS_SUCCESS) break; 2510fdc7489SMatthew Knepley } 2520fdc7489SMatthew Knepley if (b < Nb) { 2530fdc7489SMatthew Knepley PetscInt cval = b, eVal, fVal; 2540fdc7489SMatthew Knepley PetscInt *closure = NULL, Ncl, cl; 2550fdc7489SMatthew Knepley 2569566063dSJacob Faibussowitsch PetscCall(DMLabelSetValue(bodyLabel, c, cval)); 2579566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(*dm, c, PETSC_TRUE, &Ncl, &closure)); 258c1cad2e7SMatthew G. Knepley for (cl = 0; cl < Ncl; ++cl) { 259c1cad2e7SMatthew G. Knepley const PetscInt p = closure[cl * 2]; 2600fdc7489SMatthew Knepley 2610fdc7489SMatthew Knepley if (p >= eStart && p < eEnd) { 2629566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(bodyLabel, p, &eVal)); 2639566063dSJacob Faibussowitsch if (eVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval)); 2640fdc7489SMatthew Knepley } 2650fdc7489SMatthew Knepley if (p >= fStart && p < fEnd) { 2669566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(bodyLabel, p, &fVal)); 2679566063dSJacob Faibussowitsch if (fVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval)); 2680fdc7489SMatthew Knepley } 2690fdc7489SMatthew Knepley } 2709566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(*dm, c, PETSC_TRUE, &Ncl, &closure)); 2710fdc7489SMatthew Knepley } 2720fdc7489SMatthew Knepley } 2730fdc7489SMatthew Knepley } 2740fdc7489SMatthew Knepley } 2750fdc7489SMatthew Knepley #endif 2769566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(*dm, PETSC_FALSE)); 2773a074057SBarry Smith } 2783a074057SBarry Smith 2799566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelDestroy(&universal)); 2809566063dSJacob Faibussowitsch PetscCall(PLCDestroy(&in)); 2819566063dSJacob Faibussowitsch PetscCall(PLCDestroy(&out)); 2823a074057SBarry Smith PetscFunctionReturn(0); 2833a074057SBarry Smith } 2843a074057SBarry Smith 285d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMPlexRefine_CTetgen(DM dm, PetscReal *maxVolumes, DM *dmRefined) 286d71ae5a4SJacob Faibussowitsch { 2873a074057SBarry Smith MPI_Comm comm; 2883a074057SBarry Smith const PetscInt dim = 3; 2893a074057SBarry Smith PLC *in, *out; 2900fdc7489SMatthew Knepley DMUniversalLabel universal; 291af226901SMatthew G. Knepley PetscInt vStart, vEnd, v, eStart, eEnd, e, fStart, fEnd, f, cStart, cEnd, c, defVal, verbose = 0; 2920fdc7489SMatthew Knepley DMPlexInterpolatedFlag isInterpolated; 2933a074057SBarry Smith PetscMPIInt rank; 2943a074057SBarry Smith 2953a074057SBarry Smith PetscFunctionBegin; 2969566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, ((PetscObject)dm)->prefix, "-ctetgen_verbose", &verbose, NULL)); 2979566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 2989566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 2999566063dSJacob Faibussowitsch PetscCall(DMPlexIsInterpolatedCollective(dm, &isInterpolated)); 3009566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelCreate(dm, &universal)); 301af226901SMatthew G. Knepley PetscCall(DMLabelGetDefaultValue(universal->label, &defVal)); 3020fdc7489SMatthew Knepley 3039566063dSJacob Faibussowitsch PetscCall(PLCCreate(&in)); 3049566063dSJacob Faibussowitsch PetscCall(PLCCreate(&out)); 3053a074057SBarry Smith 3069566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd)); 3073a074057SBarry Smith in->numberofpoints = vEnd - vStart; 3083a074057SBarry Smith if (in->numberofpoints > 0) { 3093a074057SBarry Smith PetscSection coordSection; 3103a074057SBarry Smith Vec coordinates; 3113a074057SBarry Smith PetscScalar *array; 3123a074057SBarry Smith 3139566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofpoints * dim, &in->pointlist)); 31441e9d8b5SMatthew G. Knepley PetscCall(PetscCalloc1(in->numberofpoints, &in->pointmarkerlist)); 3159566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &coordinates)); 3169566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 3179566063dSJacob Faibussowitsch PetscCall(VecGetArray(coordinates, &array)); 3183a074057SBarry Smith for (v = vStart; v < vEnd; ++v) { 3193a074057SBarry Smith const PetscInt idx = v - vStart; 3200fdc7489SMatthew Knepley PetscInt off, d, m; 3213a074057SBarry Smith 3229566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(coordSection, v, &off)); 3230fdc7489SMatthew Knepley for (d = 0; d < dim; ++d) in->pointlist[idx * dim + d] = PetscRealPart(array[off + d]); 3249566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, v, &m)); 325af226901SMatthew G. Knepley if (m != defVal) in->pointmarkerlist[idx] = (int)m; 3263a074057SBarry Smith } 3279566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(coordinates, &array)); 3283a074057SBarry Smith } 3293a074057SBarry Smith 3309566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, 1, &eStart, &eEnd)); 3310fdc7489SMatthew Knepley in->numberofedges = eEnd - eStart; 3320fdc7489SMatthew Knepley if (isInterpolated == DMPLEX_INTERPOLATED_FULL && in->numberofedges > 0) { 3339566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofedges * 2, &in->edgelist)); 3349566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberofedges, &in->edgemarkerlist)); 3350fdc7489SMatthew Knepley for (e = eStart; e < eEnd; ++e) { 3360fdc7489SMatthew Knepley const PetscInt idx = e - eStart; 3370fdc7489SMatthew Knepley const PetscInt *cone; 3380fdc7489SMatthew Knepley PetscInt coneSize, val; 3390fdc7489SMatthew Knepley 3409566063dSJacob Faibussowitsch PetscCall(DMPlexGetConeSize(dm, e, &coneSize)); 3419566063dSJacob Faibussowitsch PetscCall(DMPlexGetCone(dm, e, &cone)); 3420fdc7489SMatthew Knepley in->edgelist[idx * 2] = cone[0] - vStart; 3430fdc7489SMatthew Knepley in->edgelist[idx * 2 + 1] = cone[1] - vStart; 3440fdc7489SMatthew Knepley 3459566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, e, &val)); 346af226901SMatthew G. Knepley if (val != defVal) in->edgemarkerlist[idx] = (int)val; 3470fdc7489SMatthew Knepley } 3480fdc7489SMatthew Knepley } 3490fdc7489SMatthew Knepley 3509566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd)); 3510fdc7489SMatthew Knepley in->numberoftrifaces = 0; 3520fdc7489SMatthew Knepley for (f = fStart; f < fEnd; ++f) { 3530fdc7489SMatthew Knepley PetscInt supportSize; 3540fdc7489SMatthew Knepley 3559566063dSJacob Faibussowitsch PetscCall(DMPlexGetSupportSize(dm, f, &supportSize)); 3560fdc7489SMatthew Knepley if (supportSize == 1) ++in->numberoftrifaces; 3570fdc7489SMatthew Knepley } 3580fdc7489SMatthew Knepley if (isInterpolated == DMPLEX_INTERPOLATED_FULL && in->numberoftrifaces > 0) { 3590fdc7489SMatthew Knepley PetscInt tf = 0; 3600fdc7489SMatthew Knepley 3619566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberoftrifaces * 3, &in->trifacelist)); 3629566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberoftrifaces, &in->trifacemarkerlist)); 3630fdc7489SMatthew Knepley for (f = fStart; f < fEnd; ++f) { 3640fdc7489SMatthew Knepley PetscInt *points = NULL; 3650fdc7489SMatthew Knepley PetscInt supportSize, numPoints, p, Nv = 0, val; 3660fdc7489SMatthew Knepley 3679566063dSJacob Faibussowitsch PetscCall(DMPlexGetSupportSize(dm, f, &supportSize)); 3680fdc7489SMatthew Knepley if (supportSize != 1) continue; 3699566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(dm, f, PETSC_TRUE, &numPoints, &points)); 3700fdc7489SMatthew Knepley for (p = 0; p < numPoints * 2; p += 2) { 3710fdc7489SMatthew Knepley const PetscInt point = points[p]; 3720fdc7489SMatthew Knepley if ((point >= vStart) && (point < vEnd)) in->trifacelist[tf * 3 + Nv++] = point - vStart; 3730fdc7489SMatthew Knepley } 3749566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(dm, f, PETSC_TRUE, &numPoints, &points)); 37563a3b9bcSJacob Faibussowitsch PetscCheck(Nv == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %" PetscInt_FMT " has %" PetscInt_FMT " vertices, not 3", f, Nv); 3769566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(universal->label, f, &val)); 377af226901SMatthew G. Knepley if (val != defVal) in->trifacemarkerlist[tf] = (int)val; 3780fdc7489SMatthew Knepley ++tf; 3790fdc7489SMatthew Knepley } 3800fdc7489SMatthew Knepley } 3810fdc7489SMatthew Knepley 3829566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd)); 3833a074057SBarry Smith in->numberofcorners = 4; 3843a074057SBarry Smith in->numberoftetrahedra = cEnd - cStart; 3853a074057SBarry Smith in->tetrahedronvolumelist = maxVolumes; 3863a074057SBarry Smith if (in->numberoftetrahedra > 0) { 3879566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(in->numberoftetrahedra * in->numberofcorners, &in->tetrahedronlist)); 3883a074057SBarry Smith for (c = cStart; c < cEnd; ++c) { 3893a074057SBarry Smith const PetscInt idx = c - cStart; 3903a074057SBarry Smith PetscInt *closure = NULL; 3913a074057SBarry Smith PetscInt closureSize; 3923a074057SBarry Smith 3939566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 3945f80ce2aSJacob Faibussowitsch PetscCheck((closureSize == 5) || (closureSize == 15), comm, PETSC_ERR_ARG_WRONG, "Mesh has cell which is not a tetrahedron, %" PetscInt_FMT " vertices in closure", closureSize); 3950fdc7489SMatthew Knepley for (v = 0; v < 4; ++v) in->tetrahedronlist[idx * in->numberofcorners + v] = closure[(v + closureSize - 4) * 2] - vStart; 3969566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 3973a074057SBarry Smith } 3983a074057SBarry Smith } 3990fdc7489SMatthew Knepley 400dd400576SPatrick Sanan if (rank == 0) { 4013a074057SBarry Smith TetGenOpts t; 4023a074057SBarry Smith 4039566063dSJacob Faibussowitsch PetscCall(TetGenOptsInitialize(&t)); 4043a074057SBarry Smith 4053a074057SBarry Smith t.in = dm; /* Should go away */ 4063a074057SBarry Smith t.refine = 1; 4073a074057SBarry Smith t.varvolume = 1; 4083a074057SBarry Smith t.quality = 1; 4093a074057SBarry Smith t.edgesout = 1; 4103a074057SBarry Smith t.zeroindex = 1; 4113a074057SBarry Smith t.quiet = 1; 4123a074057SBarry Smith t.verbose = verbose; /* Change this */ 4133a074057SBarry Smith 4149566063dSJacob Faibussowitsch PetscCall(TetGenCheckOpts(&t)); 4159566063dSJacob Faibussowitsch PetscCall(TetGenTetrahedralize(&t, in, out)); 4163a074057SBarry Smith } 4170fdc7489SMatthew Knepley 4183a074057SBarry Smith in->tetrahedronvolumelist = NULL; 4193a074057SBarry Smith { 4203a074057SBarry Smith const PetscInt numCorners = 4; 4213a074057SBarry Smith const PetscInt numCells = out->numberoftetrahedra; 4223a074057SBarry Smith const PetscInt numVertices = out->numberofpoints; 4230fdc7489SMatthew Knepley PetscReal *meshCoords = NULL; 4240fdc7489SMatthew Knepley PetscInt *cells = NULL; 4250fdc7489SMatthew Knepley PetscBool interpolate = isInterpolated == DMPLEX_INTERPOLATED_FULL ? PETSC_TRUE : PETSC_FALSE; 4263a074057SBarry Smith 427a4a685f2SJacob Faibussowitsch if (sizeof(PetscReal) == sizeof(out->pointlist[0])) { 428a4a685f2SJacob Faibussowitsch meshCoords = (PetscReal *)out->pointlist; 429a4a685f2SJacob Faibussowitsch } else { 4303a074057SBarry Smith PetscInt i; 4313a074057SBarry Smith 4329566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim * numVertices, &meshCoords)); 4330fdc7489SMatthew Knepley for (i = 0; i < dim * numVertices; ++i) meshCoords[i] = (PetscReal)out->pointlist[i]; 4343a074057SBarry Smith } 435a4a685f2SJacob Faibussowitsch if (sizeof(PetscInt) == sizeof(out->tetrahedronlist[0])) { 436a4a685f2SJacob Faibussowitsch cells = (PetscInt *)out->tetrahedronlist; 437a4a685f2SJacob Faibussowitsch } else { 438a4a685f2SJacob Faibussowitsch PetscInt i; 439a4a685f2SJacob Faibussowitsch 4409566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(numCells * numCorners, &cells)); 4410fdc7489SMatthew Knepley for (i = 0; i < numCells * numCorners; ++i) cells[i] = (PetscInt)out->tetrahedronlist[i]; 442a4a685f2SJacob Faibussowitsch } 4433a074057SBarry Smith 4449566063dSJacob Faibussowitsch PetscCall(DMPlexInvertCells_CTetgen(numCells, numCorners, cells)); 4459566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromCellListPetsc(comm, dim, numCells, numVertices, numCorners, interpolate, cells, dim, meshCoords, dmRefined)); 4469566063dSJacob Faibussowitsch if (sizeof(PetscReal) != sizeof(out->pointlist[0])) PetscCall(PetscFree(meshCoords)); 4479566063dSJacob Faibussowitsch if (sizeof(PetscInt) != sizeof(out->tetrahedronlist[0])) PetscCall(PetscFree(cells)); 4480fdc7489SMatthew Knepley 4493a074057SBarry Smith /* Set labels */ 4509566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelCreateLabels(universal, PETSC_TRUE, *dmRefined)); 4513a074057SBarry Smith for (v = 0; v < numVertices; ++v) { 45248a46eb9SPierre Jolivet if (out->pointmarkerlist[v]) PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, v + numCells, out->pointmarkerlist[v])); 4533a074057SBarry Smith } 4543a074057SBarry Smith if (interpolate) { 4553a074057SBarry Smith PetscInt e, f; 4563a074057SBarry Smith 4573a074057SBarry Smith for (e = 0; e < out->numberofedges; e++) { 4583a074057SBarry Smith if (out->edgemarkerlist[e]) { 4593a074057SBarry Smith const PetscInt vertices[2] = {out->edgelist[e * 2 + 0] + numCells, out->edgelist[e * 2 + 1] + numCells}; 4603a074057SBarry Smith const PetscInt *edges; 4613a074057SBarry Smith PetscInt numEdges; 4623a074057SBarry Smith 4639566063dSJacob Faibussowitsch PetscCall(DMPlexGetJoin(*dmRefined, 2, vertices, &numEdges, &edges)); 46463a3b9bcSJacob Faibussowitsch PetscCheck(numEdges == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Two vertices must cover only one edge, not %" PetscInt_FMT, numEdges); 4659566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, edges[0], out->edgemarkerlist[e])); 4669566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreJoin(*dmRefined, 2, vertices, &numEdges, &edges)); 4673a074057SBarry Smith } 4683a074057SBarry Smith } 4693a074057SBarry Smith for (f = 0; f < out->numberoftrifaces; f++) { 4703a074057SBarry Smith if (out->trifacemarkerlist[f]) { 4713a074057SBarry Smith const PetscInt vertices[3] = {out->trifacelist[f * 3 + 0] + numCells, out->trifacelist[f * 3 + 1] + numCells, out->trifacelist[f * 3 + 2] + numCells}; 4723a074057SBarry Smith const PetscInt *faces; 4733a074057SBarry Smith PetscInt numFaces; 4743a074057SBarry Smith 4759566063dSJacob Faibussowitsch PetscCall(DMPlexGetFullJoin(*dmRefined, 3, vertices, &numFaces, &faces)); 47663a3b9bcSJacob Faibussowitsch PetscCheck(numFaces == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Three vertices must cover only one face, not %" PetscInt_FMT, numFaces); 4779566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, faces[0], out->trifacemarkerlist[f])); 4789566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreJoin(*dmRefined, 3, vertices, &numFaces, &faces)); 4793a074057SBarry Smith } 4803a074057SBarry Smith } 4813a074057SBarry Smith } 4820fdc7489SMatthew Knepley 4830fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS 4840fdc7489SMatthew Knepley { 4850fdc7489SMatthew Knepley DMLabel bodyLabel; 4860fdc7489SMatthew Knepley PetscContainer modelObj; 4870fdc7489SMatthew Knepley PetscInt cStart, cEnd, c, eStart, eEnd, fStart, fEnd; 4880fdc7489SMatthew Knepley ego *bodies; 4890fdc7489SMatthew Knepley ego model, geom; 4900fdc7489SMatthew Knepley int Nb, oclass, mtype, *senses; 4910fdc7489SMatthew Knepley 4920fdc7489SMatthew Knepley /* Get Attached EGADS Model from Original DMPlex */ 4939566063dSJacob Faibussowitsch PetscCall(PetscObjectQuery((PetscObject)dm, "EGADS Model", (PetscObject *)&modelObj)); 4940fdc7489SMatthew Knepley if (modelObj) { 4959566063dSJacob Faibussowitsch PetscCall(PetscContainerGetPointer(modelObj, (void **)&model)); 4969566063dSJacob Faibussowitsch PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses)); 4970fdc7489SMatthew Knepley /* Transfer EGADS Model to Volumetric Mesh */ 4989566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)*dmRefined, "EGADS Model", (PetscObject)modelObj)); 4990fdc7489SMatthew Knepley 5000fdc7489SMatthew Knepley /* Set Cell Labels */ 5019566063dSJacob Faibussowitsch PetscCall(DMGetLabel(*dmRefined, "EGADS Body ID", &bodyLabel)); 5029566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(*dmRefined, 0, &cStart, &cEnd)); 5039566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(*dmRefined, 1, &fStart, &fEnd)); 5049566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(*dmRefined, 1, &eStart, &eEnd)); 5050fdc7489SMatthew Knepley 5060fdc7489SMatthew Knepley for (c = cStart; c < cEnd; ++c) { 5070fdc7489SMatthew Knepley PetscReal centroid[3] = {0., 0., 0.}; 5080fdc7489SMatthew Knepley PetscInt b; 5090fdc7489SMatthew Knepley 510*35cb6cd3SPierre Jolivet /* Determine what body the cell's centroid is located in */ 5110fdc7489SMatthew Knepley if (!interpolate) { 5120fdc7489SMatthew Knepley PetscSection coordSection; 5130fdc7489SMatthew Knepley Vec coordinates; 5140fdc7489SMatthew Knepley PetscScalar *coords = NULL; 5150fdc7489SMatthew Knepley PetscInt coordSize, s, d; 5160fdc7489SMatthew Knepley 5179566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(*dmRefined, &coordinates)); 5189566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(*dmRefined, &coordSection)); 5199566063dSJacob Faibussowitsch PetscCall(DMPlexVecGetClosure(*dmRefined, coordSection, coordinates, c, &coordSize, &coords)); 5209371c9d4SSatish Balay for (s = 0; s < coordSize; ++s) 5219371c9d4SSatish Balay for (d = 0; d < dim; ++d) centroid[d] += coords[s * dim + d]; 5229566063dSJacob Faibussowitsch PetscCall(DMPlexVecRestoreClosure(*dmRefined, coordSection, coordinates, c, &coordSize, &coords)); 5231baa6e33SBarry Smith } else PetscCall(DMPlexComputeCellGeometryFVM(*dmRefined, c, NULL, centroid, NULL)); 5240fdc7489SMatthew Knepley for (b = 0; b < Nb; ++b) { 5250fdc7489SMatthew Knepley if (EG_inTopology(bodies[b], centroid) == EGADS_SUCCESS) break; 5260fdc7489SMatthew Knepley } 5270fdc7489SMatthew Knepley if (b < Nb) { 5280fdc7489SMatthew Knepley PetscInt cval = b, eVal, fVal; 5290fdc7489SMatthew Knepley PetscInt *closure = NULL, Ncl, cl; 5300fdc7489SMatthew Knepley 5319566063dSJacob Faibussowitsch PetscCall(DMLabelSetValue(bodyLabel, c, cval)); 5329566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(*dmRefined, c, PETSC_TRUE, &Ncl, &closure)); 5330fdc7489SMatthew Knepley for (cl = 0; cl < Ncl; cl += 2) { 5340fdc7489SMatthew Knepley const PetscInt p = closure[cl]; 5350fdc7489SMatthew Knepley 5360fdc7489SMatthew Knepley if (p >= eStart && p < eEnd) { 5379566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(bodyLabel, p, &eVal)); 5389566063dSJacob Faibussowitsch if (eVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval)); 5390fdc7489SMatthew Knepley } 5400fdc7489SMatthew Knepley if (p >= fStart && p < fEnd) { 5419566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(bodyLabel, p, &fVal)); 5429566063dSJacob Faibussowitsch if (fVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval)); 5430fdc7489SMatthew Knepley } 5440fdc7489SMatthew Knepley } 5459566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(*dmRefined, c, PETSC_TRUE, &Ncl, &closure)); 5460fdc7489SMatthew Knepley } 5470fdc7489SMatthew Knepley } 5480fdc7489SMatthew Knepley } 5490fdc7489SMatthew Knepley } 5500fdc7489SMatthew Knepley #endif 5519566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(*dmRefined, PETSC_FALSE)); 5523a074057SBarry Smith } 5539566063dSJacob Faibussowitsch PetscCall(DMUniversalLabelDestroy(&universal)); 5549566063dSJacob Faibussowitsch PetscCall(PLCDestroy(&in)); 5559566063dSJacob Faibussowitsch PetscCall(PLCDestroy(&out)); 5563a074057SBarry Smith PetscFunctionReturn(0); 5573a074057SBarry Smith } 558