xref: /petsc/src/dm/impls/plex/generators/ctetgen/ctetgenerate.c (revision 41e9d8b5bae7cddec70f4add541c9e66dc5b0939)
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 */
10a4a685f2SJacob Faibussowitsch static PetscErrorCode DMPlexInvertCells_CTetgen(PetscInt numCells, PetscInt numCorners, PetscInt cells[])
113a074057SBarry Smith {
123a074057SBarry Smith   PetscInt bound = numCells*numCorners, coff;
133a074057SBarry Smith 
143a074057SBarry Smith   PetscFunctionBegin;
15a4a685f2SJacob Faibussowitsch #define SWAP(a,b) do { PetscInt tmp = (a); (a) = (b); (b) = tmp; } while (0)
1696ca5757SLisandro Dalcin   for (coff = 0; coff < bound; coff += numCorners) SWAP(cells[coff],cells[coff+1]);
1796ca5757SLisandro Dalcin #undef SWAP
183a074057SBarry Smith   PetscFunctionReturn(0);
193a074057SBarry Smith }
203a074057SBarry Smith 
213a074057SBarry Smith PETSC_EXTERN PetscErrorCode DMPlexGenerate_CTetgen(DM boundary, PetscBool interpolate, DM *dm)
223a074057SBarry Smith {
233a074057SBarry Smith   MPI_Comm               comm;
243a074057SBarry Smith   const PetscInt         dim = 3;
253a074057SBarry Smith   PLC                   *in, *out;
260fdc7489SMatthew Knepley   DMUniversalLabel       universal;
27af226901SMatthew G. Knepley   PetscInt               vStart, vEnd, v, eStart, eEnd, e, fStart, fEnd, f, defVal, verbose = 0;
280fdc7489SMatthew Knepley   DMPlexInterpolatedFlag isInterpolated;
293a074057SBarry Smith   PetscMPIInt            rank;
303a074057SBarry Smith 
313a074057SBarry Smith   PetscFunctionBegin;
329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL,((PetscObject) boundary)->prefix, "-ctetgen_verbose", &verbose, NULL));
339566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)boundary,&comm));
349566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
359566063dSJacob Faibussowitsch   PetscCall(DMPlexIsInterpolatedCollective(boundary, &isInterpolated));
369566063dSJacob Faibussowitsch   PetscCall(DMUniversalLabelCreate(boundary, &universal));
37af226901SMatthew G. Knepley   PetscCall(DMLabelGetDefaultValue(universal->label, &defVal));
380fdc7489SMatthew Knepley 
399566063dSJacob Faibussowitsch   PetscCall(PLCCreate(&in));
409566063dSJacob Faibussowitsch   PetscCall(PLCCreate(&out));
413a074057SBarry Smith 
429566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(boundary, 0, &vStart, &vEnd));
433a074057SBarry Smith   in->numberofpoints = vEnd - vStart;
443a074057SBarry Smith   if (in->numberofpoints > 0) {
453a074057SBarry Smith     PetscSection       coordSection;
463a074057SBarry Smith     Vec                coordinates;
470fdc7489SMatthew Knepley     const PetscScalar *array;
483a074057SBarry Smith 
499566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(in->numberofpoints*dim, &in->pointlist));
50*41e9d8b5SMatthew G. Knepley     PetscCall(PetscCalloc1(in->numberofpoints,     &in->pointmarkerlist));
519566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(boundary, &coordinates));
529566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(boundary, &coordSection));
539566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(coordinates, &array));
543a074057SBarry Smith     for (v = vStart; v < vEnd; ++v) {
553a074057SBarry Smith       const PetscInt idx = v - vStart;
560fdc7489SMatthew Knepley       PetscInt       off, d, m;
573a074057SBarry Smith 
589566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSection, v, &off));
590fdc7489SMatthew Knepley       for (d = 0; d < dim; ++d) in->pointlist[idx*dim + d] = PetscRealPart(array[off+d]);
609566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValue(universal->label, v, &m));
61af226901SMatthew G. Knepley       if (m != defVal) in->pointmarkerlist[idx] = (int) m;
623a074057SBarry Smith     }
639566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(coordinates, &array));
643a074057SBarry Smith   }
653a074057SBarry Smith 
669566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(boundary, 1, &eStart, &eEnd));
670fdc7489SMatthew Knepley   in->numberofedges = eEnd - eStart;
680fdc7489SMatthew Knepley   if (isInterpolated == DMPLEX_INTERPOLATED_FULL && in->numberofedges > 0) {
699566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(in->numberofedges*2, &in->edgelist));
709566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(in->numberofedges,   &in->edgemarkerlist));
710fdc7489SMatthew Knepley     for (e = eStart; e < eEnd; ++e) {
720fdc7489SMatthew Knepley       const PetscInt  idx = e - eStart;
730fdc7489SMatthew Knepley       const PetscInt *cone;
740fdc7489SMatthew Knepley       PetscInt        coneSize, val;
750fdc7489SMatthew Knepley 
769566063dSJacob Faibussowitsch       PetscCall(DMPlexGetConeSize(boundary, e, &coneSize));
779566063dSJacob Faibussowitsch       PetscCall(DMPlexGetCone(boundary, e, &cone));
780fdc7489SMatthew Knepley       in->edgelist[idx*2]     = cone[0] - vStart;
790fdc7489SMatthew Knepley       in->edgelist[idx*2 + 1] = cone[1] - vStart;
800fdc7489SMatthew Knepley 
819566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValue(universal->label, e, &val));
82af226901SMatthew G. Knepley       if (val != defVal) in->edgemarkerlist[idx] = (int) val;
830fdc7489SMatthew Knepley     }
840fdc7489SMatthew Knepley   }
850fdc7489SMatthew Knepley 
869566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(boundary, 0, &fStart, &fEnd));
873a074057SBarry Smith   in->numberoffacets = fEnd - fStart;
883a074057SBarry Smith   if (in->numberoffacets > 0) {
899566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(in->numberoffacets, &in->facetlist));
909566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(in->numberoffacets, &in->facetmarkerlist));
913a074057SBarry Smith     for (f = fStart; f < fEnd; ++f) {
923a074057SBarry Smith       const PetscInt idx    = f - fStart;
933a074057SBarry Smith       PetscInt      *points = NULL, numPoints, p, numVertices = 0, v, m = -1;
943a074057SBarry Smith       polygon       *poly;
953a074057SBarry Smith 
963a074057SBarry Smith       in->facetlist[idx].numberofpolygons = 1;
979566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(in->facetlist[idx].numberofpolygons, &in->facetlist[idx].polygonlist));
983a074057SBarry Smith       in->facetlist[idx].numberofholes    = 0;
993a074057SBarry Smith       in->facetlist[idx].holelist         = NULL;
1003a074057SBarry Smith 
1019566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(boundary, f, PETSC_TRUE, &numPoints, &points));
1023a074057SBarry Smith       for (p = 0; p < numPoints*2; p += 2) {
1033a074057SBarry Smith         const PetscInt point = points[p];
1043a074057SBarry Smith         if ((point >= vStart) && (point < vEnd)) points[numVertices++] = point;
1053a074057SBarry Smith       }
1063a074057SBarry Smith 
1073a074057SBarry Smith       poly                   = in->facetlist[idx].polygonlist;
1083a074057SBarry Smith       poly->numberofvertices = numVertices;
1099566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(poly->numberofvertices, &poly->vertexlist));
1103a074057SBarry Smith       for (v = 0; v < numVertices; ++v) {
1113a074057SBarry Smith         const PetscInt vIdx = points[v] - vStart;
1123a074057SBarry Smith         poly->vertexlist[v] = vIdx;
1133a074057SBarry Smith       }
1149566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValue(universal->label, f, &m));
115af226901SMatthew G. Knepley       if (m != defVal) in->facetmarkerlist[idx] = (int) m;
1169566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(boundary, f, PETSC_TRUE, &numPoints, &points));
1173a074057SBarry Smith     }
1183a074057SBarry Smith   }
119dd400576SPatrick Sanan   if (rank == 0) {
1203a074057SBarry Smith     TetGenOpts t;
1213a074057SBarry Smith 
1229566063dSJacob Faibussowitsch     PetscCall(TetGenOptsInitialize(&t));
1233a074057SBarry Smith     t.in        = boundary; /* Should go away */
1243a074057SBarry Smith     t.plc       = 1;
1253a074057SBarry Smith     t.quality   = 1;
1263a074057SBarry Smith     t.edgesout  = 1;
1273a074057SBarry Smith     t.zeroindex = 1;
1283a074057SBarry Smith     t.quiet     = 1;
1293a074057SBarry Smith     t.verbose   = verbose;
1300fdc7489SMatthew Knepley #if 0
1310fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS
1320fdc7489SMatthew Knepley     /* Need to add in more TetGen code */
1330fdc7489SMatthew Knepley     t.nobisect  = 1; /* Add Y to preserve Surface Mesh for EGADS */
1340fdc7489SMatthew Knepley #endif
1350fdc7489SMatthew Knepley #endif
1360fdc7489SMatthew Knepley 
1379566063dSJacob Faibussowitsch     PetscCall(TetGenCheckOpts(&t));
1389566063dSJacob Faibussowitsch     PetscCall(TetGenTetrahedralize(&t, in, out));
1393a074057SBarry Smith   }
1403a074057SBarry Smith   {
1413a074057SBarry Smith     const PetscInt numCorners  = 4;
1423a074057SBarry Smith     const PetscInt numCells    = out->numberoftetrahedra;
1433a074057SBarry Smith     const PetscInt numVertices = out->numberofpoints;
1440fdc7489SMatthew Knepley     PetscReal      *meshCoords = NULL;
1450fdc7489SMatthew Knepley     PetscInt       *cells      = NULL;
1463a074057SBarry Smith 
147a4a685f2SJacob Faibussowitsch     if (sizeof (PetscReal) == sizeof (out->pointlist[0])) {
148a4a685f2SJacob Faibussowitsch       meshCoords = (PetscReal *) out->pointlist;
149a4a685f2SJacob Faibussowitsch     } else {
1503a074057SBarry Smith       PetscInt i;
1513a074057SBarry Smith 
1529566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(dim * numVertices, &meshCoords));
1530fdc7489SMatthew Knepley       for (i = 0; i < dim * numVertices; ++i) meshCoords[i] = (PetscReal) out->pointlist[i];
1543a074057SBarry Smith     }
155a4a685f2SJacob Faibussowitsch     if (sizeof (PetscInt) == sizeof (out->tetrahedronlist[0])) {
156a4a685f2SJacob Faibussowitsch       cells = (PetscInt *) out->tetrahedronlist;
157a4a685f2SJacob Faibussowitsch     } else {
158a4a685f2SJacob Faibussowitsch       PetscInt i;
159a4a685f2SJacob Faibussowitsch 
1609566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numCells * numCorners, &cells));
1610fdc7489SMatthew Knepley       for (i = 0; i < numCells * numCorners; i++) cells[i] = (PetscInt) out->tetrahedronlist[i];
162a4a685f2SJacob Faibussowitsch     }
1633a074057SBarry Smith 
1649566063dSJacob Faibussowitsch     PetscCall(DMPlexInvertCells_CTetgen(numCells, numCorners, cells));
1659566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromCellListPetsc(comm, dim, numCells, numVertices, numCorners, interpolate, cells, dim, meshCoords, dm));
166a4a685f2SJacob Faibussowitsch     if (sizeof (PetscReal) != sizeof (out->pointlist[0])) {
1679566063dSJacob Faibussowitsch       PetscCall(PetscFree(meshCoords));
1683a074057SBarry Smith     }
169a4a685f2SJacob Faibussowitsch     if (sizeof (PetscInt) != sizeof (out->tetrahedronlist[0])) {
1709566063dSJacob Faibussowitsch       PetscCall(PetscFree(cells));
171a4a685f2SJacob Faibussowitsch     }
1720fdc7489SMatthew Knepley 
1733a074057SBarry Smith     /* Set labels */
1749566063dSJacob Faibussowitsch     PetscCall(DMUniversalLabelCreateLabels(universal, PETSC_TRUE, *dm));
1753a074057SBarry Smith     for (v = 0; v < numVertices; ++v) {
1763a074057SBarry Smith       if (out->pointmarkerlist[v]) {
1779566063dSJacob Faibussowitsch         PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, v+numCells, out->pointmarkerlist[v]));
1783a074057SBarry Smith       }
1793a074057SBarry Smith     }
1803a074057SBarry Smith     if (interpolate) {
1813a074057SBarry Smith       PetscInt e;
1823a074057SBarry Smith 
1833a074057SBarry Smith       for (e = 0; e < out->numberofedges; e++) {
1843a074057SBarry Smith         if (out->edgemarkerlist[e]) {
1853a074057SBarry Smith           const PetscInt  vertices[2] = {out->edgelist[e*2+0]+numCells, out->edgelist[e*2+1]+numCells};
1863a074057SBarry Smith           const PetscInt *edges;
1873a074057SBarry Smith           PetscInt        numEdges;
1883a074057SBarry Smith 
1899566063dSJacob Faibussowitsch           PetscCall(DMPlexGetJoin(*dm, 2, vertices, &numEdges, &edges));
19063a3b9bcSJacob Faibussowitsch           PetscCheck(numEdges == 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Two vertices must cover only one edge, not %" PetscInt_FMT, numEdges);
1919566063dSJacob Faibussowitsch           PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, edges[0], out->edgemarkerlist[e]));
1929566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreJoin(*dm, 2, vertices, &numEdges, &edges));
1933a074057SBarry Smith         }
1943a074057SBarry Smith       }
1953a074057SBarry Smith       for (f = 0; f < out->numberoftrifaces; f++) {
1963a074057SBarry Smith         if (out->trifacemarkerlist[f]) {
1973a074057SBarry Smith           const PetscInt  vertices[3] = {out->trifacelist[f*3+0]+numCells, out->trifacelist[f*3+1]+numCells, out->trifacelist[f*3+2]+numCells};
1983a074057SBarry Smith           const PetscInt *faces;
1993a074057SBarry Smith           PetscInt        numFaces;
2003a074057SBarry Smith 
2019566063dSJacob Faibussowitsch           PetscCall(DMPlexGetFullJoin(*dm, 3, vertices, &numFaces, &faces));
20263a3b9bcSJacob Faibussowitsch           PetscCheck(numFaces == 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Three vertices must cover only one face, not %" PetscInt_FMT, numFaces);
2039566063dSJacob Faibussowitsch           PetscCall(DMUniversalLabelSetLabelValue(universal, *dm, PETSC_TRUE, faces[0], out->trifacemarkerlist[f]));
2049566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreJoin(*dm, 3, vertices, &numFaces, &faces));
2053a074057SBarry Smith         }
2063a074057SBarry Smith       }
2073a074057SBarry Smith     }
2080fdc7489SMatthew Knepley 
2090fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS
2100fdc7489SMatthew Knepley     {
2110fdc7489SMatthew Knepley       DMLabel        bodyLabel;
2120fdc7489SMatthew Knepley       PetscContainer modelObj;
2130fdc7489SMatthew Knepley       PetscInt       cStart, cEnd, c, eStart, eEnd, fStart, fEnd;
2140fdc7489SMatthew Knepley       ego           *bodies;
2150fdc7489SMatthew Knepley       ego            model, geom;
2160fdc7489SMatthew Knepley       int            Nb, oclass, mtype, *senses;
2170fdc7489SMatthew Knepley 
2180fdc7489SMatthew Knepley       /* Get Attached EGADS Model from Original DMPlex */
2199566063dSJacob Faibussowitsch       PetscCall(PetscObjectQuery((PetscObject) boundary, "EGADS Model", (PetscObject *) &modelObj));
2200fdc7489SMatthew Knepley       if (modelObj) {
2219566063dSJacob Faibussowitsch         PetscCall(PetscContainerGetPointer(modelObj, (void **) &model));
2229566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses));
2230fdc7489SMatthew Knepley         /* Transfer EGADS Model to Volumetric Mesh */
2249566063dSJacob Faibussowitsch         PetscCall(PetscObjectCompose((PetscObject) *dm, "EGADS Model", (PetscObject) modelObj));
2250fdc7489SMatthew Knepley 
2260fdc7489SMatthew Knepley         /* Set Cell Labels */
2279566063dSJacob Faibussowitsch         PetscCall(DMGetLabel(*dm, "EGADS Body ID", &bodyLabel));
2289566063dSJacob Faibussowitsch         PetscCall(DMPlexGetHeightStratum(*dm, 0, &cStart, &cEnd));
2299566063dSJacob Faibussowitsch         PetscCall(DMPlexGetHeightStratum(*dm, 1, &fStart, &fEnd));
2309566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthStratum(*dm, 1, &eStart, &eEnd));
2310fdc7489SMatthew Knepley 
2320fdc7489SMatthew Knepley         for (c = cStart; c < cEnd; ++c) {
2330fdc7489SMatthew Knepley           PetscReal centroid[3] = {0., 0., 0.};
2340fdc7489SMatthew Knepley           PetscInt  b;
2350fdc7489SMatthew Knepley 
2360fdc7489SMatthew Knepley           /* Deterimine what body the cell's centroid is located in */
2370fdc7489SMatthew Knepley           if (!interpolate) {
2380fdc7489SMatthew Knepley             PetscSection   coordSection;
2390fdc7489SMatthew Knepley             Vec            coordinates;
2400fdc7489SMatthew Knepley             PetscScalar   *coords = NULL;
2410fdc7489SMatthew Knepley             PetscInt       coordSize, s, d;
2420fdc7489SMatthew Knepley 
2439566063dSJacob Faibussowitsch             PetscCall(DMGetCoordinatesLocal(*dm, &coordinates));
2449566063dSJacob Faibussowitsch             PetscCall(DMGetCoordinateSection(*dm, &coordSection));
2459566063dSJacob Faibussowitsch             PetscCall(DMPlexVecGetClosure(*dm, coordSection, coordinates, c, &coordSize, &coords));
2460fdc7489SMatthew Knepley             for (s = 0; s < coordSize; ++s) 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 
2853a074057SBarry Smith PETSC_EXTERN PetscErrorCode DMPlexRefine_CTetgen(DM dm, PetscReal *maxVolumes, DM *dmRefined)
2863a074057SBarry Smith {
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));
314*41e9d8b5SMatthew 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) {
4523a074057SBarry Smith       if (out->pointmarkerlist[v]) {
4539566063dSJacob Faibussowitsch         PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, v+numCells, out->pointmarkerlist[v]));
4543a074057SBarry Smith       }
4553a074057SBarry Smith     }
4563a074057SBarry Smith     if (interpolate) {
4573a074057SBarry Smith       PetscInt e, f;
4583a074057SBarry Smith 
4593a074057SBarry Smith       for (e = 0; e < out->numberofedges; e++) {
4603a074057SBarry Smith         if (out->edgemarkerlist[e]) {
4613a074057SBarry Smith           const PetscInt  vertices[2] = {out->edgelist[e*2+0]+numCells, out->edgelist[e*2+1]+numCells};
4623a074057SBarry Smith           const PetscInt *edges;
4633a074057SBarry Smith           PetscInt        numEdges;
4643a074057SBarry Smith 
4659566063dSJacob Faibussowitsch           PetscCall(DMPlexGetJoin(*dmRefined, 2, vertices, &numEdges, &edges));
46663a3b9bcSJacob Faibussowitsch           PetscCheck(numEdges == 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Two vertices must cover only one edge, not %" PetscInt_FMT, numEdges);
4679566063dSJacob Faibussowitsch           PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, edges[0], out->edgemarkerlist[e]));
4689566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreJoin(*dmRefined, 2, vertices, &numEdges, &edges));
4693a074057SBarry Smith         }
4703a074057SBarry Smith       }
4713a074057SBarry Smith       for (f = 0; f < out->numberoftrifaces; f++) {
4723a074057SBarry Smith         if (out->trifacemarkerlist[f]) {
4733a074057SBarry Smith           const PetscInt  vertices[3] = {out->trifacelist[f*3+0]+numCells, out->trifacelist[f*3+1]+numCells, out->trifacelist[f*3+2]+numCells};
4743a074057SBarry Smith           const PetscInt *faces;
4753a074057SBarry Smith           PetscInt        numFaces;
4763a074057SBarry Smith 
4779566063dSJacob Faibussowitsch           PetscCall(DMPlexGetFullJoin(*dmRefined, 3, vertices, &numFaces, &faces));
47863a3b9bcSJacob Faibussowitsch           PetscCheck(numFaces == 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Three vertices must cover only one face, not %" PetscInt_FMT, numFaces);
4799566063dSJacob Faibussowitsch           PetscCall(DMUniversalLabelSetLabelValue(universal, *dmRefined, PETSC_TRUE, faces[0], out->trifacemarkerlist[f]));
4809566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreJoin(*dmRefined, 3, vertices, &numFaces, &faces));
4813a074057SBarry Smith         }
4823a074057SBarry Smith       }
4833a074057SBarry Smith     }
4840fdc7489SMatthew Knepley 
4850fdc7489SMatthew Knepley #ifdef PETSC_HAVE_EGADS
4860fdc7489SMatthew Knepley     {
4870fdc7489SMatthew Knepley       DMLabel        bodyLabel;
4880fdc7489SMatthew Knepley       PetscContainer modelObj;
4890fdc7489SMatthew Knepley       PetscInt       cStart, cEnd, c, eStart, eEnd, fStart, fEnd;
4900fdc7489SMatthew Knepley       ego           *bodies;
4910fdc7489SMatthew Knepley       ego            model, geom;
4920fdc7489SMatthew Knepley       int            Nb, oclass, mtype, *senses;
4930fdc7489SMatthew Knepley 
4940fdc7489SMatthew Knepley       /* Get Attached EGADS Model from Original DMPlex */
4959566063dSJacob Faibussowitsch       PetscCall(PetscObjectQuery((PetscObject) dm, "EGADS Model", (PetscObject *) &modelObj));
4960fdc7489SMatthew Knepley       if (modelObj) {
4979566063dSJacob Faibussowitsch         PetscCall(PetscContainerGetPointer(modelObj, (void **) &model));
4989566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses));
4990fdc7489SMatthew Knepley         /* Transfer EGADS Model to Volumetric Mesh */
5009566063dSJacob Faibussowitsch         PetscCall(PetscObjectCompose((PetscObject) *dmRefined, "EGADS Model", (PetscObject) modelObj));
5010fdc7489SMatthew Knepley 
5020fdc7489SMatthew Knepley         /* Set Cell Labels */
5039566063dSJacob Faibussowitsch         PetscCall(DMGetLabel(*dmRefined, "EGADS Body ID", &bodyLabel));
5049566063dSJacob Faibussowitsch         PetscCall(DMPlexGetHeightStratum(*dmRefined, 0, &cStart, &cEnd));
5059566063dSJacob Faibussowitsch         PetscCall(DMPlexGetHeightStratum(*dmRefined, 1, &fStart, &fEnd));
5069566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthStratum(*dmRefined, 1, &eStart, &eEnd));
5070fdc7489SMatthew Knepley 
5080fdc7489SMatthew Knepley         for (c = cStart; c < cEnd; ++c) {
5090fdc7489SMatthew Knepley           PetscReal centroid[3] = {0., 0., 0.};
5100fdc7489SMatthew Knepley           PetscInt  b;
5110fdc7489SMatthew Knepley 
5120fdc7489SMatthew Knepley           /* Deterimine what body the cell's centroid is located in */
5130fdc7489SMatthew Knepley           if (!interpolate) {
5140fdc7489SMatthew Knepley             PetscSection   coordSection;
5150fdc7489SMatthew Knepley             Vec            coordinates;
5160fdc7489SMatthew Knepley             PetscScalar   *coords = NULL;
5170fdc7489SMatthew Knepley             PetscInt       coordSize, s, d;
5180fdc7489SMatthew Knepley 
5199566063dSJacob Faibussowitsch             PetscCall(DMGetCoordinatesLocal(*dmRefined, &coordinates));
5209566063dSJacob Faibussowitsch             PetscCall(DMGetCoordinateSection(*dmRefined, &coordSection));
5219566063dSJacob Faibussowitsch             PetscCall(DMPlexVecGetClosure(*dmRefined, coordSection, coordinates, c, &coordSize, &coords));
5220fdc7489SMatthew Knepley             for (s = 0; s < coordSize; ++s) for (d = 0; d < dim; ++d) centroid[d] += coords[s*dim+d];
5239566063dSJacob Faibussowitsch             PetscCall(DMPlexVecRestoreClosure(*dmRefined, coordSection, coordinates, c, &coordSize, &coords));
5241baa6e33SBarry Smith           } else PetscCall(DMPlexComputeCellGeometryFVM(*dmRefined, c, NULL, centroid, NULL));
5250fdc7489SMatthew Knepley           for (b = 0; b < Nb; ++b) {
5260fdc7489SMatthew Knepley             if (EG_inTopology(bodies[b], centroid) == EGADS_SUCCESS) break;
5270fdc7489SMatthew Knepley           }
5280fdc7489SMatthew Knepley           if (b < Nb) {
5290fdc7489SMatthew Knepley             PetscInt   cval = b, eVal, fVal;
5300fdc7489SMatthew Knepley             PetscInt *closure = NULL, Ncl, cl;
5310fdc7489SMatthew Knepley 
5329566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(bodyLabel, c, cval));
5339566063dSJacob Faibussowitsch             PetscCall(DMPlexGetTransitiveClosure(*dmRefined, c, PETSC_TRUE, &Ncl, &closure));
5340fdc7489SMatthew Knepley             for (cl = 0; cl < Ncl; cl += 2) {
5350fdc7489SMatthew Knepley               const PetscInt p = closure[cl];
5360fdc7489SMatthew Knepley 
5370fdc7489SMatthew Knepley               if (p >= eStart && p < eEnd) {
5389566063dSJacob Faibussowitsch                 PetscCall(DMLabelGetValue(bodyLabel, p, &eVal));
5399566063dSJacob Faibussowitsch                 if (eVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval));
5400fdc7489SMatthew Knepley               }
5410fdc7489SMatthew Knepley               if (p >= fStart && p < fEnd) {
5429566063dSJacob Faibussowitsch                 PetscCall(DMLabelGetValue(bodyLabel, p, &fVal));
5439566063dSJacob Faibussowitsch                 if (fVal < 0) PetscCall(DMLabelSetValue(bodyLabel, p, cval));
5440fdc7489SMatthew Knepley               }
5450fdc7489SMatthew Knepley             }
5469566063dSJacob Faibussowitsch             PetscCall(DMPlexRestoreTransitiveClosure(*dmRefined, c, PETSC_TRUE, &Ncl, &closure));
5470fdc7489SMatthew Knepley           }
5480fdc7489SMatthew Knepley         }
5490fdc7489SMatthew Knepley       }
5500fdc7489SMatthew Knepley     }
5510fdc7489SMatthew Knepley #endif
5529566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(*dmRefined, PETSC_FALSE));
5533a074057SBarry Smith   }
5549566063dSJacob Faibussowitsch   PetscCall(DMUniversalLabelDestroy(&universal));
5559566063dSJacob Faibussowitsch   PetscCall(PLCDestroy(&in));
5569566063dSJacob Faibussowitsch   PetscCall(PLCDestroy(&out));
5573a074057SBarry Smith   PetscFunctionReturn(0);
5583a074057SBarry Smith }
559