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