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 */
DMPlexInvertCells_CTetgen(PetscInt numCells,PetscInt numCorners,PetscInt cells[])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
233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS);
243a074057SBarry Smith }
253a074057SBarry Smith
DMPlexGenerate_CTetgen(DM boundary,PetscBool interpolate,DM * dm)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) {
220*2a8381b2SBarry Smith PetscCall(PetscContainerGetPointer(modelObj, &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
23535cb6cd3SPierre 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));
2823ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS);
2833a074057SBarry Smith }
2843a074057SBarry Smith
DMPlexRefine_CTetgen(DM dm,PetscReal * maxVolumes,DM * dmRefined)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) {
495*2a8381b2SBarry Smith PetscCall(PetscContainerGetPointer(modelObj, &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
51035cb6cd3SPierre 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));
5563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS);
5573a074057SBarry Smith }
558