xref: /petsc/src/dm/impls/plex/plexcreate.c (revision e535cce4c56ea8fc2df340b7d14fa8f4f30804fb)
1552f7358SJed Brown #define PETSCDM_DLL
2af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h> /*I   "petscdmplex.h"   I*/
3cc4c1da9SBarry Smith #include <petsc/private/hashseti.h>
40c312b8eSJed Brown #include <petscsf.h>
5cc4c1da9SBarry Smith #include <petscdmplextransform.h> /*I   "petscdmplextransform.h"   I*/
69f6c5813SMatthew G. Knepley #include <petscdmlabelephemeral.h>
7b7f5c055SJed Brown #include <petsc/private/kernels/blockmatmult.h>
8b7f5c055SJed Brown #include <petsc/private/kernels/blockinvert.h>
9552f7358SJed Brown 
10d0812dedSMatthew G. Knepley #ifdef PETSC_HAVE_UNISTD_H
11d0812dedSMatthew G. Knepley   #include <unistd.h>
12d0812dedSMatthew G. Knepley #endif
13d0812dedSMatthew G. Knepley #include <errno.h>
14d0812dedSMatthew G. Knepley 
15708be2fdSJed Brown PetscLogEvent DMPLEX_CreateFromFile, DMPLEX_CreateFromOptions, DMPLEX_BuildFromCellList, DMPLEX_BuildCoordinatesFromCellList;
1658cd63d5SVaclav Hapla 
179318fe57SMatthew G. Knepley /* External function declarations here */
189318fe57SMatthew G. Knepley static PetscErrorCode DMInitialize_Plex(DM dm);
199318fe57SMatthew G. Knepley 
20e600fa54SMatthew G. Knepley /* This copies internal things in the Plex structure that we generally want when making a new, related Plex */
21d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCopy_Internal(DM dmin, PetscBool copyPeriodicity, PetscBool copyOverlap, DM dmout)
22d71ae5a4SJacob Faibussowitsch {
234fb89dddSMatthew G. Knepley   const PetscReal     *maxCell, *Lstart, *L;
2412a88998SMatthew G. Knepley   VecType              vecType;
2512a88998SMatthew G. Knepley   MatType              matType;
265962854dSMatthew G. Knepley   PetscBool            dist, useCeed;
27adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
28e600fa54SMatthew G. Knepley 
29e600fa54SMatthew G. Knepley   PetscFunctionBegin;
3012a88998SMatthew G. Knepley   PetscCall(DMGetVecType(dmin, &vecType));
3112a88998SMatthew G. Knepley   PetscCall(DMSetVecType(dmout, vecType));
3212a88998SMatthew G. Knepley   PetscCall(DMGetMatType(dmin, &matType));
3312a88998SMatthew G. Knepley   PetscCall(DMSetMatType(dmout, matType));
34e600fa54SMatthew G. Knepley   if (copyPeriodicity) {
354fb89dddSMatthew G. Knepley     PetscCall(DMGetPeriodicity(dmin, &maxCell, &Lstart, &L));
364fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dmout, maxCell, Lstart, L));
373d0e8ed9SDavid Salac     PetscCall(DMLocalizeCoordinates(dmout));
38e600fa54SMatthew G. Knepley   }
399566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dmin, &dist));
409566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeSetDefault(dmout, dist));
416bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dmin, &reorder));
426bc1bd01Sksagiyam   PetscCall(DMPlexReorderSetDefault(dmout, reorder));
435962854dSMatthew G. Knepley   PetscCall(DMPlexGetUseCeed(dmin, &useCeed));
445962854dSMatthew G. Knepley   PetscCall(DMPlexSetUseCeed(dmout, useCeed));
45e600fa54SMatthew G. Knepley   ((DM_Plex *)dmout->data)->useHashLocation = ((DM_Plex *)dmin->data)->useHashLocation;
465962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printSetValues  = ((DM_Plex *)dmin->data)->printSetValues;
475962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFEM        = ((DM_Plex *)dmin->data)->printFEM;
485962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFVM        = ((DM_Plex *)dmin->data)->printFVM;
495962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printL2         = ((DM_Plex *)dmin->data)->printL2;
505962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printLocate     = ((DM_Plex *)dmin->data)->printLocate;
51a77a5016SMatthew G. Knepley   ((DM_Plex *)dmout->data)->printProject    = ((DM_Plex *)dmin->data)->printProject;
525962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printTol        = ((DM_Plex *)dmin->data)->printTol;
531baa6e33SBarry Smith   if (copyOverlap) PetscCall(DMPlexSetOverlap_Plex(dmout, dmin, 0));
543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
55e600fa54SMatthew G. Knepley }
56e600fa54SMatthew G. Knepley 
579318fe57SMatthew G. Knepley /* Replace dm with the contents of ndm, and then destroy ndm
589318fe57SMatthew G. Knepley    - Share the DM_Plex structure
599318fe57SMatthew G. Knepley    - Share the coordinates
609318fe57SMatthew G. Knepley    - Share the SF
619318fe57SMatthew G. Knepley */
62d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexReplace_Internal(DM dm, DM *ndm)
63d71ae5a4SJacob Faibussowitsch {
649318fe57SMatthew G. Knepley   PetscSF          sf;
659318fe57SMatthew G. Knepley   DM               dmNew = *ndm, coordDM, coarseDM;
669318fe57SMatthew G. Knepley   Vec              coords;
674fb89dddSMatthew G. Knepley   const PetscReal *maxCell, *Lstart, *L;
689318fe57SMatthew G. Knepley   PetscInt         dim, cdim;
69*e535cce4SJames Wright   PetscBool        use_natural;
709318fe57SMatthew G. Knepley 
719318fe57SMatthew G. Knepley   PetscFunctionBegin;
729318fe57SMatthew G. Knepley   if (dm == dmNew) {
739566063dSJacob Faibussowitsch     PetscCall(DMDestroy(ndm));
743ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
759318fe57SMatthew G. Knepley   }
769318fe57SMatthew G. Knepley   dm->setupcalled = dmNew->setupcalled;
77d0812dedSMatthew G. Knepley   if (!dm->hdr.name) {
78d0812dedSMatthew G. Knepley     const char *name;
79d0812dedSMatthew G. Knepley 
80d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)*ndm, &name));
81d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
82d0812dedSMatthew G. Knepley   }
839566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dmNew, &dim));
849566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
859566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dmNew, &cdim));
869566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
879566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmNew, &sf));
889566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sf));
899566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmNew, &coordDM));
909566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmNew, &coords));
919566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dm, coordDM));
929566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coords));
936858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmNew, &coordDM));
946858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmNew, &coords));
956858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dm, coordDM));
966858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dm, coords));
979318fe57SMatthew G. Knepley   /* Do not want to create the coordinate field if it does not already exist, so do not call DMGetCoordinateField() */
986858538eSMatthew G. Knepley   PetscCall(DMFieldDestroy(&dm->coordinates[0].field));
996858538eSMatthew G. Knepley   dm->coordinates[0].field            = dmNew->coordinates[0].field;
10061a622f3SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->coordFunc = ((DM_Plex *)dm->data)->coordFunc;
1014fb89dddSMatthew G. Knepley   PetscCall(DMGetPeriodicity(dmNew, &maxCell, &Lstart, &L));
1024fb89dddSMatthew G. Knepley   PetscCall(DMSetPeriodicity(dm, maxCell, Lstart, L));
103*e535cce4SJames Wright   PetscCall(DMGetNaturalSF(dmNew, &sf));
104*e535cce4SJames Wright   PetscCall(DMSetNaturalSF(dm, sf));
105*e535cce4SJames Wright   PetscCall(DMGetUseNatural(dmNew, &use_natural));
106*e535cce4SJames Wright   PetscCall(DMSetUseNatural(dm, use_natural));
1079566063dSJacob Faibussowitsch   PetscCall(DMDestroy_Plex(dm));
1089566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
1099318fe57SMatthew G. Knepley   dm->data = dmNew->data;
1109318fe57SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->refct++;
1111fca310dSJames Wright   {
1121fca310dSJames Wright     PetscInt       num_face_sfs;
1131fca310dSJames Wright     const PetscSF *sfs;
1141fca310dSJames Wright     PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &sfs));
1151fca310dSJames Wright     PetscCall(DMPlexSetIsoperiodicFaceSF(dm, num_face_sfs, (PetscSF *)sfs)); // for the compose function effect on dm
1161fca310dSJames Wright   }
1179566063dSJacob Faibussowitsch   PetscCall(DMDestroyLabelLinkList_Internal(dm));
1189566063dSJacob Faibussowitsch   PetscCall(DMCopyLabels(dmNew, dm, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
1199566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmNew, &coarseDM));
1209566063dSJacob Faibussowitsch   PetscCall(DMSetCoarseDM(dm, coarseDM));
1219566063dSJacob Faibussowitsch   PetscCall(DMDestroy(ndm));
1223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1239318fe57SMatthew G. Knepley }
1249318fe57SMatthew G. Knepley 
1259318fe57SMatthew G. Knepley /* Swap dm with the contents of dmNew
1269318fe57SMatthew G. Knepley    - Swap the DM_Plex structure
1279318fe57SMatthew G. Knepley    - Swap the coordinates
1289318fe57SMatthew G. Knepley    - Swap the point PetscSF
1299318fe57SMatthew G. Knepley */
130d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexSwap_Static(DM dmA, DM dmB)
131d71ae5a4SJacob Faibussowitsch {
1329318fe57SMatthew G. Knepley   DM          coordDMA, coordDMB;
1339318fe57SMatthew G. Knepley   Vec         coordsA, coordsB;
1349318fe57SMatthew G. Knepley   PetscSF     sfA, sfB;
1359318fe57SMatthew G. Knepley   DMField     fieldTmp;
1369318fe57SMatthew G. Knepley   void       *tmp;
1379318fe57SMatthew G. Knepley   DMLabelLink listTmp;
1389318fe57SMatthew G. Knepley   DMLabel     depthTmp;
1399318fe57SMatthew G. Knepley   PetscInt    tmpI;
1409318fe57SMatthew G. Knepley 
1419318fe57SMatthew G. Knepley   PetscFunctionBegin;
1423ba16761SJacob Faibussowitsch   if (dmA == dmB) PetscFunctionReturn(PETSC_SUCCESS);
1439566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmA, &sfA));
1449566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmB, &sfB));
1459566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)sfA));
1469566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmA, sfB));
1479566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmB, sfA));
1489566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)sfA));
1499318fe57SMatthew G. Knepley 
1509566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmA, &coordDMA));
1519566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmB, &coordDMB));
1529566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1539566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmA, coordDMB));
1549566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmB, coordDMA));
1559566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1569318fe57SMatthew G. Knepley 
1579566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmA, &coordsA));
1589566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmB, &coordsB));
1599566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordsA));
1609566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmA, coordsB));
1619566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmB, coordsA));
1629566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1639318fe57SMatthew G. Knepley 
1646858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmA, &coordDMA));
1656858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmB, &coordDMB));
1666858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1676858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmA, coordDMB));
1686858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmB, coordDMA));
1696858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1706858538eSMatthew G. Knepley 
1716858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmA, &coordsA));
1726858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmB, &coordsB));
1736858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordsA));
1746858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmA, coordsB));
1756858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmB, coordsA));
1766858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1776858538eSMatthew G. Knepley 
1786858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[0].field;
1796858538eSMatthew G. Knepley   dmA->coordinates[0].field = dmB->coordinates[0].field;
1806858538eSMatthew G. Knepley   dmB->coordinates[0].field = fieldTmp;
1816858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[1].field;
1826858538eSMatthew G. Knepley   dmA->coordinates[1].field = dmB->coordinates[1].field;
1836858538eSMatthew G. Knepley   dmB->coordinates[1].field = fieldTmp;
1849318fe57SMatthew G. Knepley   tmp                       = dmA->data;
1859318fe57SMatthew G. Knepley   dmA->data                 = dmB->data;
1869318fe57SMatthew G. Knepley   dmB->data                 = tmp;
1879318fe57SMatthew G. Knepley   listTmp                   = dmA->labels;
1889318fe57SMatthew G. Knepley   dmA->labels               = dmB->labels;
1899318fe57SMatthew G. Knepley   dmB->labels               = listTmp;
1909318fe57SMatthew G. Knepley   depthTmp                  = dmA->depthLabel;
1919318fe57SMatthew G. Knepley   dmA->depthLabel           = dmB->depthLabel;
1929318fe57SMatthew G. Knepley   dmB->depthLabel           = depthTmp;
1939318fe57SMatthew G. Knepley   depthTmp                  = dmA->celltypeLabel;
1949318fe57SMatthew G. Knepley   dmA->celltypeLabel        = dmB->celltypeLabel;
1959318fe57SMatthew G. Knepley   dmB->celltypeLabel        = depthTmp;
1969318fe57SMatthew G. Knepley   tmpI                      = dmA->levelup;
1979318fe57SMatthew G. Knepley   dmA->levelup              = dmB->levelup;
1989318fe57SMatthew G. Knepley   dmB->levelup              = tmpI;
1993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2009318fe57SMatthew G. Knepley }
2019318fe57SMatthew G. Knepley 
2023431e603SJed Brown PetscErrorCode DMPlexInterpolateInPlace_Internal(DM dm)
203d71ae5a4SJacob Faibussowitsch {
2049318fe57SMatthew G. Knepley   DM idm;
2059318fe57SMatthew G. Knepley 
2069318fe57SMatthew G. Knepley   PetscFunctionBegin;
2079566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolate(dm, &idm));
2089566063dSJacob Faibussowitsch   PetscCall(DMPlexCopyCoordinates(dm, idm));
20969d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &idm));
2103ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2119318fe57SMatthew G. Knepley }
2129318fe57SMatthew G. Knepley 
2139318fe57SMatthew G. Knepley /*@C
2149318fe57SMatthew G. Knepley   DMPlexCreateCoordinateSpace - Creates a finite element space for the coordinates
2159318fe57SMatthew G. Knepley 
21620f4b53cSBarry Smith   Collective
2179318fe57SMatthew G. Knepley 
2189318fe57SMatthew G. Knepley   Input Parameters:
21960225df5SJacob Faibussowitsch + dm        - The `DMPLEX`
22020f4b53cSBarry Smith . degree    - The degree of the finite element or `PETSC_DECIDE`
221e44f6aebSMatthew G. Knepley . project   - Flag to project current coordinates into the space
2229318fe57SMatthew G. Knepley - coordFunc - An optional function to map new points from refinement to the surface
2239318fe57SMatthew G. Knepley 
2249318fe57SMatthew G. Knepley   Level: advanced
2259318fe57SMatthew G. Knepley 
2261cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `PetscPointFunc`, `PetscFECreateLagrange()`, `DMGetCoordinateDM()`
2279318fe57SMatthew G. Knepley @*/
228e44f6aebSMatthew G. Knepley PetscErrorCode DMPlexCreateCoordinateSpace(DM dm, PetscInt degree, PetscBool project, PetscPointFunc coordFunc)
229d71ae5a4SJacob Faibussowitsch {
2309318fe57SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
231e44f6aebSMatthew G. Knepley   PetscFE  fe   = NULL;
2329318fe57SMatthew G. Knepley   DM       cdm;
2331df12153SMatthew G. Knepley   PetscInt dim, dE, qorder, height;
2349318fe57SMatthew G. Knepley 
235e44f6aebSMatthew G. Knepley   PetscFunctionBegin;
2369566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
2379566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dE));
2389318fe57SMatthew G. Knepley   qorder = degree;
239e44f6aebSMatthew G. Knepley   PetscCall(DMGetCoordinateDM(dm, &cdm));
240d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)cdm);
241dc431b0cSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-default_quadrature_order", "Quadrature order is one less than quadrature points per edge", "DMPlexCreateCoordinateSpace", qorder, &qorder, NULL, 0));
242d0609cedSBarry Smith   PetscOptionsEnd();
2431df12153SMatthew G. Knepley   PetscCall(DMPlexGetVTKCellHeight(dm, &height));
244e44f6aebSMatthew G. Knepley   if (degree >= 0) {
245e44f6aebSMatthew G. Knepley     DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
246e44f6aebSMatthew G. Knepley     PetscInt       cStart, cEnd, gct;
247dc431b0cSMatthew G. Knepley 
2481df12153SMatthew G. Knepley     PetscCall(DMPlexGetHeightStratum(dm, height, &cStart, &cEnd));
249dc431b0cSMatthew G. Knepley     if (cEnd > cStart) PetscCall(DMPlexGetCellType(dm, cStart, &ct));
250e44f6aebSMatthew G. Knepley     gct = (PetscInt)ct;
251462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &gct, 1, MPIU_INT, MPI_MIN, PetscObjectComm((PetscObject)dm)));
252e44f6aebSMatthew G. Knepley     ct = (DMPolytopeType)gct;
253e44f6aebSMatthew G. Knepley     // Work around current bug in PetscDualSpaceSetUp_Lagrange()
254e44f6aebSMatthew G. Knepley     //   Can be seen in plex_tutorials-ex10_1
255e44f6aebSMatthew G. Knepley     if (ct != DM_POLYTOPE_SEG_PRISM_TENSOR && ct != DM_POLYTOPE_TRI_PRISM_TENSOR && ct != DM_POLYTOPE_QUAD_PRISM_TENSOR) PetscCall(PetscFECreateLagrangeByCell(PETSC_COMM_SELF, dim, dE, ct, degree, qorder, &fe));
2564f9ab2b4SJed Brown   }
257e44f6aebSMatthew G. Knepley   PetscCall(DMSetCoordinateDisc(dm, fe, project));
2589566063dSJacob Faibussowitsch   PetscCall(PetscFEDestroy(&fe));
2599318fe57SMatthew G. Knepley   mesh->coordFunc = coordFunc;
2603ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2619318fe57SMatthew G. Knepley }
2629318fe57SMatthew G. Knepley 
2631df5d5c5SMatthew G. Knepley /*@
2641df5d5c5SMatthew G. Knepley   DMPlexCreateDoublet - Creates a mesh of two cells of the specified type, optionally with later refinement.
2651df5d5c5SMatthew G. Knepley 
266d083f849SBarry Smith   Collective
2671df5d5c5SMatthew G. Knepley 
2681df5d5c5SMatthew G. Knepley   Input Parameters:
269a1cb98faSBarry Smith + comm            - The communicator for the `DM` object
2701df5d5c5SMatthew G. Knepley . dim             - The spatial dimension
2711df5d5c5SMatthew G. Knepley . simplex         - Flag for simplicial cells, otherwise they are tensor product cells
2721df5d5c5SMatthew G. Knepley . interpolate     - Flag to create intermediate mesh pieces (edges, faces)
2731df5d5c5SMatthew G. Knepley - refinementLimit - A nonzero number indicates the largest admissible volume for a refined cell
2741df5d5c5SMatthew G. Knepley 
2751df5d5c5SMatthew G. Knepley   Output Parameter:
27660225df5SJacob Faibussowitsch . newdm - The `DM` object
2771df5d5c5SMatthew G. Knepley 
2781df5d5c5SMatthew G. Knepley   Level: beginner
2791df5d5c5SMatthew G. Knepley 
2801cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetType()`, `DMCreate()`
2811df5d5c5SMatthew G. Knepley @*/
282d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateDoublet(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscBool interpolate, PetscReal refinementLimit, DM *newdm)
283d71ae5a4SJacob Faibussowitsch {
2841df5d5c5SMatthew G. Knepley   DM          dm;
2851df5d5c5SMatthew G. Knepley   PetscMPIInt rank;
2861df5d5c5SMatthew G. Knepley 
2871df5d5c5SMatthew G. Knepley   PetscFunctionBegin;
2889566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, &dm));
2899566063dSJacob Faibussowitsch   PetscCall(DMSetType(dm, DMPLEX));
2909566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
29146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
2929566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
293ce78fa2fSMatthew G. Knepley   switch (dim) {
294ce78fa2fSMatthew G. Knepley   case 2:
2959566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "triangular"));
2969566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "quadrilateral"));
297ce78fa2fSMatthew G. Knepley     break;
298ce78fa2fSMatthew G. Knepley   case 3:
2999566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "tetrahedral"));
3009566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "hexahedral"));
301ce78fa2fSMatthew G. Knepley     break;
302d71ae5a4SJacob Faibussowitsch   default:
303d71ae5a4SJacob Faibussowitsch     SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
304ce78fa2fSMatthew G. Knepley   }
3051df5d5c5SMatthew G. Knepley   if (rank) {
3061df5d5c5SMatthew G. Knepley     PetscInt numPoints[2] = {0, 0};
3079566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, NULL, NULL, NULL, NULL));
3081df5d5c5SMatthew G. Knepley   } else {
3091df5d5c5SMatthew G. Knepley     switch (dim) {
3101df5d5c5SMatthew G. Knepley     case 2:
3111df5d5c5SMatthew G. Knepley       if (simplex) {
3121df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {4, 2};
3131df5d5c5SMatthew G. Knepley         PetscInt    coneSize[6]         = {3, 3, 0, 0, 0, 0};
3141df5d5c5SMatthew G. Knepley         PetscInt    cones[6]            = {2, 3, 4, 5, 4, 3};
3151df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
3161df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[8]     = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
3171df5d5c5SMatthew G. Knepley 
3189566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3191df5d5c5SMatthew G. Knepley       } else {
3201df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {6, 2};
3211df5d5c5SMatthew G. Knepley         PetscInt    coneSize[8]         = {4, 4, 0, 0, 0, 0, 0, 0};
3221df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {2, 3, 4, 5, 3, 6, 7, 4};
3231df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3241df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[12]    = {-1.0, -0.5, 0.0, -0.5, 0.0, 0.5, -1.0, 0.5, 1.0, -0.5, 1.0, 0.5};
3251df5d5c5SMatthew G. Knepley 
3269566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3271df5d5c5SMatthew G. Knepley       }
3281df5d5c5SMatthew G. Knepley       break;
3291df5d5c5SMatthew G. Knepley     case 3:
3301df5d5c5SMatthew G. Knepley       if (simplex) {
3311df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {5, 2};
3321df5d5c5SMatthew G. Knepley         PetscInt    coneSize[7]         = {4, 4, 0, 0, 0, 0, 0};
3331df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {4, 3, 5, 2, 5, 3, 4, 6};
3341df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3351df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[15]    = {-1.0, 0.0, 0.0, 0.0, -1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0};
3361df5d5c5SMatthew G. Knepley 
3379566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3381df5d5c5SMatthew G. Knepley       } else {
3391df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]         = {12, 2};
3401df5d5c5SMatthew G. Knepley         PetscInt    coneSize[14]         = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3411df5d5c5SMatthew G. Knepley         PetscInt    cones[16]            = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
3421df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3439371c9d4SSatish Balay         PetscScalar vertexCoords[36]     = {-1.0, -0.5, -0.5, -1.0, 0.5, -0.5, 0.0, 0.5, -0.5, 0.0, -0.5, -0.5, -1.0, -0.5, 0.5, 0.0, -0.5, 0.5, 0.0, 0.5, 0.5, -1.0, 0.5, 0.5, 1.0, 0.5, -0.5, 1.0, -0.5, -0.5, 1.0, -0.5, 0.5, 1.0, 0.5, 0.5};
3441df5d5c5SMatthew G. Knepley 
3459566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3461df5d5c5SMatthew G. Knepley       }
3471df5d5c5SMatthew G. Knepley       break;
348d71ae5a4SJacob Faibussowitsch     default:
349d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
3501df5d5c5SMatthew G. Knepley     }
3511df5d5c5SMatthew G. Knepley   }
35246139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
3531df5d5c5SMatthew G. Knepley   *newdm = dm;
3541df5d5c5SMatthew G. Knepley   if (refinementLimit > 0.0) {
3551df5d5c5SMatthew G. Knepley     DM          rdm;
3561df5d5c5SMatthew G. Knepley     const char *name;
3571df5d5c5SMatthew G. Knepley 
3589566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(*newdm, PETSC_FALSE));
3599566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(*newdm, refinementLimit));
3609566063dSJacob Faibussowitsch     PetscCall(DMRefine(*newdm, comm, &rdm));
3619566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)*newdm, &name));
3629566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)rdm, name));
3639566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3641df5d5c5SMatthew G. Knepley     *newdm = rdm;
3651df5d5c5SMatthew G. Knepley   }
3661df5d5c5SMatthew G. Knepley   if (interpolate) {
3675fd9971aSMatthew G. Knepley     DM idm;
3681df5d5c5SMatthew G. Knepley 
3699566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*newdm, &idm));
3709566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3711df5d5c5SMatthew G. Knepley     *newdm = idm;
3721df5d5c5SMatthew G. Knepley   }
3733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3741df5d5c5SMatthew G. Knepley }
3751df5d5c5SMatthew G. Knepley 
376d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
377d71ae5a4SJacob Faibussowitsch {
3789318fe57SMatthew G. Knepley   const PetscInt numVertices    = 2;
3799318fe57SMatthew G. Knepley   PetscInt       markerRight    = 1;
3809318fe57SMatthew G. Knepley   PetscInt       markerLeft     = 1;
3819318fe57SMatthew G. Knepley   PetscBool      markerSeparate = PETSC_FALSE;
3829318fe57SMatthew G. Knepley   Vec            coordinates;
3839318fe57SMatthew G. Knepley   PetscSection   coordSection;
3849318fe57SMatthew G. Knepley   PetscScalar   *coords;
3859318fe57SMatthew G. Knepley   PetscInt       coordSize;
3869318fe57SMatthew G. Knepley   PetscMPIInt    rank;
3879318fe57SMatthew G. Knepley   PetscInt       cdim = 1, v;
388552f7358SJed Brown 
3899318fe57SMatthew G. Knepley   PetscFunctionBegin;
3909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
3919318fe57SMatthew G. Knepley   if (markerSeparate) {
3929318fe57SMatthew G. Knepley     markerRight = 2;
3939318fe57SMatthew G. Knepley     markerLeft  = 1;
3949318fe57SMatthew G. Knepley   }
3959566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
396c5853193SPierre Jolivet   if (rank == 0) {
3979566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numVertices));
3989566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
3999566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 0, markerLeft));
4009566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 1, markerRight));
4019318fe57SMatthew G. Knepley   }
4029566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
4039566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
4049318fe57SMatthew G. Knepley   /* Build coordinates */
4059566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
4069566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
4079566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
4089566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, 0, numVertices));
4099566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
4109318fe57SMatthew G. Knepley   for (v = 0; v < numVertices; ++v) {
4119566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
4129566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
4139318fe57SMatthew G. Knepley   }
4149566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
4159566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
4169566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
4179566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
4189566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
4199566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
4209566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
4219566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
4229318fe57SMatthew G. Knepley   coords[0] = lower[0];
4239318fe57SMatthew G. Knepley   coords[1] = upper[0];
4249566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
4259566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
4269566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
4273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4289318fe57SMatthew G. Knepley }
42926492d91SMatthew G. Knepley 
430d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
431d71ae5a4SJacob Faibussowitsch {
4321df21d24SMatthew G. Knepley   const PetscInt numVertices    = (edges[0] + 1) * (edges[1] + 1);
4331df21d24SMatthew G. Knepley   const PetscInt numEdges       = edges[0] * (edges[1] + 1) + (edges[0] + 1) * edges[1];
434552f7358SJed Brown   PetscInt       markerTop      = 1;
435552f7358SJed Brown   PetscInt       markerBottom   = 1;
436552f7358SJed Brown   PetscInt       markerRight    = 1;
437552f7358SJed Brown   PetscInt       markerLeft     = 1;
438552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
439552f7358SJed Brown   Vec            coordinates;
440552f7358SJed Brown   PetscSection   coordSection;
441552f7358SJed Brown   PetscScalar   *coords;
442552f7358SJed Brown   PetscInt       coordSize;
443552f7358SJed Brown   PetscMPIInt    rank;
444552f7358SJed Brown   PetscInt       v, vx, vy;
445552f7358SJed Brown 
446552f7358SJed Brown   PetscFunctionBegin;
4479566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
448552f7358SJed Brown   if (markerSeparate) {
4491df21d24SMatthew G. Knepley     markerTop    = 3;
4501df21d24SMatthew G. Knepley     markerBottom = 1;
4511df21d24SMatthew G. Knepley     markerRight  = 2;
4521df21d24SMatthew G. Knepley     markerLeft   = 4;
453552f7358SJed Brown   }
4549566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
455dd400576SPatrick Sanan   if (rank == 0) {
456552f7358SJed Brown     PetscInt e, ex, ey;
457552f7358SJed Brown 
4589566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numEdges + numVertices));
45948a46eb9SPierre Jolivet     for (e = 0; e < numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
4609566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
461552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
462552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
463552f7358SJed Brown         PetscInt edge   = vx * edges[1] + ey + edges[0] * (edges[1] + 1);
464552f7358SJed Brown         PetscInt vertex = ey * (edges[0] + 1) + vx + numEdges;
465da80777bSKarl Rupp         PetscInt cone[2];
466552f7358SJed Brown 
4679371c9d4SSatish Balay         cone[0] = vertex;
4689371c9d4SSatish Balay         cone[1] = vertex + edges[0] + 1;
4699566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
470552f7358SJed Brown         if (vx == edges[0]) {
4719566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
4729566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
473552f7358SJed Brown           if (ey == edges[1] - 1) {
4749566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
4759566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerRight));
476552f7358SJed Brown           }
477552f7358SJed Brown         } else if (vx == 0) {
4789566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
4799566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
480552f7358SJed Brown           if (ey == edges[1] - 1) {
4819566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
4829566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerLeft));
483552f7358SJed Brown           }
484552f7358SJed Brown         }
485552f7358SJed Brown       }
486552f7358SJed Brown     }
487552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
488552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
489552f7358SJed Brown         PetscInt edge   = vy * edges[0] + ex;
490552f7358SJed Brown         PetscInt vertex = vy * (edges[0] + 1) + ex + numEdges;
491da80777bSKarl Rupp         PetscInt cone[2];
492552f7358SJed Brown 
4939371c9d4SSatish Balay         cone[0] = vertex;
4949371c9d4SSatish Balay         cone[1] = vertex + 1;
4959566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
496552f7358SJed Brown         if (vy == edges[1]) {
4979566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
4989566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
499552f7358SJed Brown           if (ex == edges[0] - 1) {
5009566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
5019566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerTop));
502552f7358SJed Brown           }
503552f7358SJed Brown         } else if (vy == 0) {
5049566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
5059566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
506552f7358SJed Brown           if (ex == edges[0] - 1) {
5079566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
5089566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerBottom));
509552f7358SJed Brown           }
510552f7358SJed Brown         }
511552f7358SJed Brown       }
512552f7358SJed Brown     }
513552f7358SJed Brown   }
5149566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
5159566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
516552f7358SJed Brown   /* Build coordinates */
5179566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 2));
5189566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
5199566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
5209566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices));
5219566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 2));
522552f7358SJed Brown   for (v = numEdges; v < numEdges + numVertices; ++v) {
5239566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 2));
5249566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 2));
525552f7358SJed Brown   }
5269566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5279566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
5289566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
5299566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
5309566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
5319566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 2));
5329566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
5339566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
534552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
535552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
536552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 0] = lower[0] + ((upper[0] - lower[0]) / edges[0]) * vx;
537552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 1] = lower[1] + ((upper[1] - lower[1]) / edges[1]) * vy;
538552f7358SJed Brown     }
539552f7358SJed Brown   }
5409566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
5419566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
5429566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
5433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
544552f7358SJed Brown }
545552f7358SJed Brown 
546d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
547d71ae5a4SJacob Faibussowitsch {
5489e8abbc3SMichael Lange   PetscInt     vertices[3], numVertices;
5497b59f5a9SMichael Lange   PetscInt     numFaces       = 2 * faces[0] * faces[1] + 2 * faces[1] * faces[2] + 2 * faces[0] * faces[2];
550c2df9bbfSMatthew G. Knepley   PetscInt     markerTop      = 1;
551c2df9bbfSMatthew G. Knepley   PetscInt     markerBottom   = 1;
552c2df9bbfSMatthew G. Knepley   PetscInt     markerFront    = 1;
553c2df9bbfSMatthew G. Knepley   PetscInt     markerBack     = 1;
554c2df9bbfSMatthew G. Knepley   PetscInt     markerRight    = 1;
555c2df9bbfSMatthew G. Knepley   PetscInt     markerLeft     = 1;
556c2df9bbfSMatthew G. Knepley   PetscBool    markerSeparate = PETSC_FALSE;
557552f7358SJed Brown   Vec          coordinates;
558552f7358SJed Brown   PetscSection coordSection;
559552f7358SJed Brown   PetscScalar *coords;
560552f7358SJed Brown   PetscInt     coordSize;
561552f7358SJed Brown   PetscMPIInt  rank;
562552f7358SJed Brown   PetscInt     v, vx, vy, vz;
5637b59f5a9SMichael Lange   PetscInt     voffset, iface = 0, cone[4];
564552f7358SJed Brown 
565552f7358SJed Brown   PetscFunctionBegin;
5661dca8a05SBarry Smith   PetscCheck(faces[0] >= 1 && faces[1] >= 1 && faces[2] >= 1, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Must have at least 1 face per side");
5679566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
568c2df9bbfSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
569c2df9bbfSMatthew G. Knepley   if (markerSeparate) {
570c2df9bbfSMatthew G. Knepley     markerBottom = 1;
571c2df9bbfSMatthew G. Knepley     markerTop    = 2;
572c2df9bbfSMatthew G. Knepley     markerFront  = 3;
573c2df9bbfSMatthew G. Knepley     markerBack   = 4;
574c2df9bbfSMatthew G. Knepley     markerRight  = 5;
575c2df9bbfSMatthew G. Knepley     markerLeft   = 6;
576c2df9bbfSMatthew G. Knepley   }
5779371c9d4SSatish Balay   vertices[0] = faces[0] + 1;
5789371c9d4SSatish Balay   vertices[1] = faces[1] + 1;
5799371c9d4SSatish Balay   vertices[2] = faces[2] + 1;
5809e8abbc3SMichael Lange   numVertices = vertices[0] * vertices[1] * vertices[2];
581dd400576SPatrick Sanan   if (rank == 0) {
582552f7358SJed Brown     PetscInt f;
583552f7358SJed Brown 
5849566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numFaces + numVertices));
58548a46eb9SPierre Jolivet     for (f = 0; f < numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
5869566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
5877b59f5a9SMichael Lange 
5887b59f5a9SMichael Lange     /* Side 0 (Top) */
5897b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
5907b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
5917b59f5a9SMichael Lange         voffset = numFaces + vertices[0] * vertices[1] * (vertices[2] - 1) + vy * vertices[0] + vx;
5929371c9d4SSatish Balay         cone[0] = voffset;
5939371c9d4SSatish Balay         cone[1] = voffset + 1;
5949371c9d4SSatish Balay         cone[2] = voffset + vertices[0] + 1;
5959371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
5969566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
597c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerTop));
598c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerTop));
599c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerTop));
600c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerTop));
601c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerTop));
6027b59f5a9SMichael Lange         iface++;
603552f7358SJed Brown       }
604552f7358SJed Brown     }
6057b59f5a9SMichael Lange 
6067b59f5a9SMichael Lange     /* Side 1 (Bottom) */
6077b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
6087b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6097b59f5a9SMichael Lange         voffset = numFaces + vy * (faces[0] + 1) + vx;
6109371c9d4SSatish Balay         cone[0] = voffset + 1;
6119371c9d4SSatish Balay         cone[1] = voffset;
6129371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6139371c9d4SSatish Balay         cone[3] = voffset + vertices[0] + 1;
6149566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
615c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBottom));
616c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBottom));
617c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBottom));
618c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerBottom));
619c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerBottom));
6207b59f5a9SMichael Lange         iface++;
621552f7358SJed Brown       }
622552f7358SJed Brown     }
6237b59f5a9SMichael Lange 
6247b59f5a9SMichael Lange     /* Side 2 (Front) */
6257b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6267b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6277b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vx;
6289371c9d4SSatish Balay         cone[0] = voffset;
6299371c9d4SSatish Balay         cone[1] = voffset + 1;
6309371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + 1;
6319371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1];
6329566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
633c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerFront));
634c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerFront));
635c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerFront));
636c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerFront));
637c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerFront));
6387b59f5a9SMichael Lange         iface++;
639552f7358SJed Brown       }
6407b59f5a9SMichael Lange     }
6417b59f5a9SMichael Lange 
6427b59f5a9SMichael Lange     /* Side 3 (Back) */
6437b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6447b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6457b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vertices[0] * (vertices[1] - 1) + vx;
6469371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6479371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1] + 1;
6489371c9d4SSatish Balay         cone[2] = voffset + 1;
6499371c9d4SSatish Balay         cone[3] = voffset;
6509566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
651c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBack));
652c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBack));
653c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBack));
654c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerBack));
655c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerBack));
6567b59f5a9SMichael Lange         iface++;
6577b59f5a9SMichael Lange       }
6587b59f5a9SMichael Lange     }
6597b59f5a9SMichael Lange 
6607b59f5a9SMichael Lange     /* Side 4 (Left) */
6617b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6627b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
6637b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0];
6649371c9d4SSatish Balay         cone[0] = voffset;
6659371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1];
6669371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + vertices[0];
6679371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
6689566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
669c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerLeft));
670c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerLeft));
671c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerLeft));
672c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[1] + 0, markerLeft));
673c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerLeft));
6747b59f5a9SMichael Lange         iface++;
6757b59f5a9SMichael Lange       }
6767b59f5a9SMichael Lange     }
6777b59f5a9SMichael Lange 
6787b59f5a9SMichael Lange     /* Side 5 (Right) */
6797b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6807b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
681aab5bcd8SJed Brown         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0] + faces[0];
6829371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6839371c9d4SSatish Balay         cone[1] = voffset;
6849371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6859371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1] + vertices[0];
6869566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
687c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerRight));
688c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerRight));
689c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerRight));
690c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerRight));
691c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerRight));
6927b59f5a9SMichael Lange         iface++;
6937b59f5a9SMichael Lange       }
694552f7358SJed Brown     }
695552f7358SJed Brown   }
6969566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
6979566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
698552f7358SJed Brown   /* Build coordinates */
6999566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 3));
7009566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
7019566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
7029566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices));
7039566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 3));
704552f7358SJed Brown   for (v = numFaces; v < numFaces + numVertices; ++v) {
7059566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 3));
7069566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 3));
707552f7358SJed Brown   }
7089566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
7099566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
7109566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
7119566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
7129566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
7139566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 3));
7149566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
7159566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
716552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
717552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
718552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
719552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 0] = lower[0] + ((upper[0] - lower[0]) / faces[0]) * vx;
720552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 1] = lower[1] + ((upper[1] - lower[1]) / faces[1]) * vy;
721552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 2] = lower[2] + ((upper[2] - lower[2]) / faces[2]) * vz;
722552f7358SJed Brown       }
723552f7358SJed Brown     }
724552f7358SJed Brown   }
7259566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
7269566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
7279566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
7283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
729552f7358SJed Brown }
730552f7358SJed Brown 
731d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate)
732d71ae5a4SJacob Faibussowitsch {
7339318fe57SMatthew G. Knepley   PetscFunctionBegin;
7349318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
73546139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
7369566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim - 1));
7379566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim));
7389318fe57SMatthew G. Knepley   switch (dim) {
739d71ae5a4SJacob Faibussowitsch   case 1:
740d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(dm, lower, upper, faces));
741d71ae5a4SJacob Faibussowitsch     break;
742d71ae5a4SJacob Faibussowitsch   case 2:
743d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(dm, lower, upper, faces));
744d71ae5a4SJacob Faibussowitsch     break;
745d71ae5a4SJacob Faibussowitsch   case 3:
746d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(dm, lower, upper, faces));
747d71ae5a4SJacob Faibussowitsch     break;
748d71ae5a4SJacob Faibussowitsch   default:
749d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Dimension not supported: %" PetscInt_FMT, dim);
7509318fe57SMatthew G. Knepley   }
75146139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
7529566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
7533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7549318fe57SMatthew G. Knepley }
7559318fe57SMatthew G. Knepley 
7569318fe57SMatthew G. Knepley /*@C
7579318fe57SMatthew G. Knepley   DMPlexCreateBoxSurfaceMesh - Creates a mesh on the surface of the tensor product of unit intervals (box) using tensor cells (hexahedra).
7589318fe57SMatthew G. Knepley 
7599318fe57SMatthew G. Knepley   Collective
7609318fe57SMatthew G. Knepley 
7619318fe57SMatthew G. Knepley   Input Parameters:
762a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
76320f4b53cSBarry Smith . dim         - The spatial dimension of the box, so the resulting mesh is has dimension `dim`-1
76420f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
76520f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
76620f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
7679318fe57SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
7689318fe57SMatthew G. Knepley 
7699318fe57SMatthew G. Knepley   Output Parameter:
770a1cb98faSBarry Smith . dm - The `DM` object
7719318fe57SMatthew G. Knepley 
7729318fe57SMatthew G. Knepley   Level: beginner
7739318fe57SMatthew G. Knepley 
7741cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateBoxMesh()`, `DMPlexCreateFromFile()`, `DMSetType()`, `DMCreate()`
7759318fe57SMatthew G. Knepley @*/
776d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBoxSurfaceMesh(MPI_Comm comm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate, DM *dm)
777d71ae5a4SJacob Faibussowitsch {
7789318fe57SMatthew G. Knepley   PetscInt  fac[3] = {1, 1, 1};
7799318fe57SMatthew G. Knepley   PetscReal low[3] = {0, 0, 0};
7809318fe57SMatthew G. Knepley   PetscReal upp[3] = {1, 1, 1};
7819318fe57SMatthew G. Knepley 
7829318fe57SMatthew G. Knepley   PetscFunctionBegin;
7839566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
7849566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
7859566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(*dm, dim, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, interpolate));
7863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7879318fe57SMatthew G. Knepley }
7889318fe57SMatthew G. Knepley 
789d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateLineMesh_Internal(DM dm, PetscInt segments, PetscReal lower, PetscReal upper, DMBoundaryType bd)
790d71ae5a4SJacob Faibussowitsch {
791fdbf62faSLisandro Dalcin   PetscInt     i, fStart, fEnd, numCells = 0, numVerts = 0;
792fdbf62faSLisandro Dalcin   PetscInt     numPoints[2], *coneSize, *cones, *coneOrientations;
793fdbf62faSLisandro Dalcin   PetscScalar *vertexCoords;
794fdbf62faSLisandro Dalcin   PetscReal    L, maxCell;
795fdbf62faSLisandro Dalcin   PetscBool    markerSeparate = PETSC_FALSE;
796fdbf62faSLisandro Dalcin   PetscInt     markerLeft = 1, faceMarkerLeft = 1;
797fdbf62faSLisandro Dalcin   PetscInt     markerRight = 1, faceMarkerRight = 2;
798fdbf62faSLisandro Dalcin   PetscBool    wrap = (bd == DM_BOUNDARY_PERIODIC || bd == DM_BOUNDARY_TWIST) ? PETSC_TRUE : PETSC_FALSE;
799fdbf62faSLisandro Dalcin   PetscMPIInt  rank;
800fdbf62faSLisandro Dalcin 
801fdbf62faSLisandro Dalcin   PetscFunctionBegin;
8024f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
803fdbf62faSLisandro Dalcin 
8049566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, 1));
8059566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
8069566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
807fdbf62faSLisandro Dalcin 
8089566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
809dd400576SPatrick Sanan   if (rank == 0) numCells = segments;
810dd400576SPatrick Sanan   if (rank == 0) numVerts = segments + (wrap ? 0 : 1);
811fdbf62faSLisandro Dalcin 
8129371c9d4SSatish Balay   numPoints[0] = numVerts;
8139371c9d4SSatish Balay   numPoints[1] = numCells;
8149566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(numCells + numVerts, &coneSize, numCells * 2, &cones, numCells + numVerts, &coneOrientations, numVerts, &vertexCoords));
8159566063dSJacob Faibussowitsch   PetscCall(PetscArrayzero(coneOrientations, numCells + numVerts));
816ad540459SPierre Jolivet   for (i = 0; i < numCells; ++i) coneSize[i] = 2;
817ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) coneSize[numCells + i] = 0;
8189371c9d4SSatish Balay   for (i = 0; i < numCells; ++i) {
8199371c9d4SSatish Balay     cones[2 * i]     = numCells + i % numVerts;
8209371c9d4SSatish Balay     cones[2 * i + 1] = numCells + (i + 1) % numVerts;
8219371c9d4SSatish Balay   }
822ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) vertexCoords[i] = lower + (upper - lower) * ((PetscReal)i / (PetscReal)numCells);
8239566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
8249566063dSJacob Faibussowitsch   PetscCall(PetscFree4(coneSize, cones, coneOrientations, vertexCoords));
825fdbf62faSLisandro Dalcin 
8269566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
8279371c9d4SSatish Balay   if (markerSeparate) {
8289371c9d4SSatish Balay     markerLeft  = faceMarkerLeft;
8299371c9d4SSatish Balay     markerRight = faceMarkerRight;
8309371c9d4SSatish Balay   }
831dd400576SPatrick Sanan   if (!wrap && rank == 0) {
8329566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
8339566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fStart, markerLeft));
8349566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fEnd - 1, markerRight));
8359566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fStart, faceMarkerLeft));
8369566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fEnd - 1, faceMarkerRight));
837fdbf62faSLisandro Dalcin   }
838fdbf62faSLisandro Dalcin   if (wrap) {
839fdbf62faSLisandro Dalcin     L       = upper - lower;
840fdbf62faSLisandro Dalcin     maxCell = (PetscReal)1.1 * (L / (PetscReal)PetscMax(1, segments));
8414fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, &maxCell, &lower, &L));
842fdbf62faSLisandro Dalcin   }
8439566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
8443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
845fdbf62faSLisandro Dalcin }
846fdbf62faSLisandro Dalcin 
8474054ae39SJames Wright // Creates "Face Sets" label based on the standard box labeling conventions
8484054ae39SJames Wright static PetscErrorCode DMPlexSetBoxLabel_Internal(DM dm)
8494054ae39SJames Wright {
8506ff49feeSJames Wright   DM              cdm;
8516ff49feeSJames Wright   PetscSection    csection;
8526ff49feeSJames Wright   Vec             coordinates;
8534054ae39SJames Wright   DMLabel         label;
8546ff49feeSJames Wright   IS              faces_is;
8554054ae39SJames Wright   PetscInt        dim, num_face;
8564054ae39SJames Wright   const PetscInt *faces;
8574054ae39SJames Wright   PetscInt        faceMarkerBottom, faceMarkerTop, faceMarkerFront, faceMarkerBack, faceMarkerRight, faceMarkerLeft;
8584054ae39SJames Wright 
8594054ae39SJames Wright   PetscFunctionBeginUser;
8604054ae39SJames Wright   PetscCall(DMGetDimension(dm, &dim));
861d7c1f440SPierre Jolivet   PetscCheck((dim == 2) || (dim == 3), PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "DMPlex box labeling only supports 2D and 3D meshes, received DM of dimension %" PetscInt_FMT, dim);
8624054ae39SJames Wright   // Get Face Sets label
8634054ae39SJames Wright   PetscCall(DMGetLabel(dm, "Face Sets", &label));
8644054ae39SJames Wright   if (label) {
8654054ae39SJames Wright     PetscCall(DMLabelReset(label));
8664054ae39SJames Wright   } else {
8674054ae39SJames Wright     PetscCall(DMCreateLabel(dm, "Face Sets"));
8684054ae39SJames Wright     PetscCall(DMGetLabel(dm, "Face Sets", &label));
8694054ae39SJames Wright   }
8704054ae39SJames Wright   PetscCall(DMPlexMarkBoundaryFaces(dm, 1, label));
8716ff49feeSJames Wright   PetscCall(DMGetStratumIS(dm, "Face Sets", 1, &faces_is));
8726ff49feeSJames Wright   if (!faces_is) PetscFunctionReturn(PETSC_SUCCESS); // No faces on rank
8734054ae39SJames Wright 
8744054ae39SJames Wright   switch (dim) {
8754054ae39SJames Wright   case 2:
8764054ae39SJames Wright     faceMarkerTop    = 3;
8774054ae39SJames Wright     faceMarkerBottom = 1;
8784054ae39SJames Wright     faceMarkerRight  = 2;
8794054ae39SJames Wright     faceMarkerLeft   = 4;
8804054ae39SJames Wright     break;
8814054ae39SJames Wright   case 3:
8824054ae39SJames Wright     faceMarkerBottom = 1;
8834054ae39SJames Wright     faceMarkerTop    = 2;
8844054ae39SJames Wright     faceMarkerFront  = 3;
8854054ae39SJames Wright     faceMarkerBack   = 4;
8864054ae39SJames Wright     faceMarkerRight  = 5;
8874054ae39SJames Wright     faceMarkerLeft   = 6;
8884054ae39SJames Wright     break;
8894054ae39SJames Wright   default:
8904054ae39SJames Wright     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
8914054ae39SJames Wright   }
8924054ae39SJames Wright 
8936ff49feeSJames Wright   PetscCall(ISGetLocalSize(faces_is, &num_face));
8946ff49feeSJames Wright   PetscCall(ISGetIndices(faces_is, &faces));
8956ff49feeSJames Wright   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
8966ff49feeSJames Wright   PetscCall(DMGetCoordinateDM(dm, &cdm));
8976ff49feeSJames Wright   PetscCall(DMGetLocalSection(cdm, &csection));
8984054ae39SJames Wright   for (PetscInt f = 0; f < num_face; ++f) {
8996ff49feeSJames Wright     PetscScalar *coords = NULL;
9006ff49feeSJames Wright     PetscInt     face = faces[f], flip = 1, label_value = -1, coords_size;
9014054ae39SJames Wright 
9024054ae39SJames Wright     { // Determine if orientation of face is flipped
9034054ae39SJames Wright       PetscInt        num_cells_support, num_faces, start = -1;
9044054ae39SJames Wright       const PetscInt *orients, *cell_faces, *cells;
9054054ae39SJames Wright 
9064054ae39SJames Wright       PetscCall(DMPlexGetSupport(dm, face, &cells));
9074054ae39SJames Wright       PetscCall(DMPlexGetSupportSize(dm, face, &num_cells_support));
9084054ae39SJames Wright       PetscCheck(num_cells_support == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Expected one cell in support of exterior face, but got %" PetscInt_FMT " cells", num_cells_support);
9094054ae39SJames Wright       PetscCall(DMPlexGetCone(dm, cells[0], &cell_faces));
9104054ae39SJames Wright       PetscCall(DMPlexGetConeSize(dm, cells[0], &num_faces));
9114054ae39SJames Wright       for (PetscInt i = 0; i < num_faces; i++) {
9124054ae39SJames Wright         if (cell_faces[i] == face) start = i;
9134054ae39SJames Wright       }
9144054ae39SJames Wright       PetscCheck(start >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT, "Could not find face %" PetscInt_FMT " in cone of its support", face);
9154054ae39SJames Wright       PetscCall(DMPlexGetConeOrientation(dm, cells[0], &orients));
9164054ae39SJames Wright       if (orients[start] < 0) flip = -1;
9174054ae39SJames Wright     }
9184054ae39SJames Wright 
9196ff49feeSJames Wright     // Cannot use DMPlexComputeCellGeometryFVM() for high-order geometry, so must calculate normal vectors manually
9206ff49feeSJames Wright     // Use the vertices (depth 0) of coordinate DM to calculate normal vector
9216ff49feeSJames Wright     PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
9224054ae39SJames Wright     switch (dim) {
9234054ae39SJames Wright     case 2: {
9246ff49feeSJames Wright       PetscScalar vec[2];
9256ff49feeSJames Wright 
9266ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) vec[d] = flip * (PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]));
9276ff49feeSJames Wright       PetscScalar normal[] = {vec[1], -vec[0]};
9286ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
9296ff49feeSJames Wright         label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
9304054ae39SJames Wright       } else {
9316ff49feeSJames Wright         label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerTop : faceMarkerBottom;
9324054ae39SJames Wright       }
9334054ae39SJames Wright     } break;
9344054ae39SJames Wright     case 3: {
9356ff49feeSJames Wright       PetscScalar vec1[3], vec2[3], normal[3];
9366ff49feeSJames Wright 
9376ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) {
9386ff49feeSJames Wright         vec1[d] = PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]);
9396ff49feeSJames Wright         vec2[d] = PetscRealPart(coords[2 * dim + d]) - PetscRealPart(coords[1 * dim + d]);
9406ff49feeSJames Wright       }
9416ff49feeSJames Wright 
9426ff49feeSJames Wright       // Calculate normal vector via cross-product
9436ff49feeSJames Wright       normal[0] = flip * ((vec1[1] * vec2[2]) - (vec1[2] * vec2[1]));
9446ff49feeSJames Wright       normal[1] = flip * ((vec1[2] * vec2[0]) - (vec1[0] * vec2[2]));
9456ff49feeSJames Wright       normal[2] = flip * ((vec1[0] * vec2[1]) - (vec1[1] * vec2[0]));
9466ff49feeSJames Wright 
9476ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
9486ff49feeSJames Wright         if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[2])) {
9496ff49feeSJames Wright           label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
9504054ae39SJames Wright         } else {
9516ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
9524054ae39SJames Wright         }
9534054ae39SJames Wright       } else {
9546ff49feeSJames Wright         if (PetscAbsScalar(normal[1]) > PetscAbsScalar(normal[2])) {
9556ff49feeSJames Wright           label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerBack : faceMarkerFront;
9564054ae39SJames Wright         } else {
9576ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
9584054ae39SJames Wright         }
9594054ae39SJames Wright       }
9604054ae39SJames Wright     } break;
9614054ae39SJames Wright     }
9624054ae39SJames Wright 
9634054ae39SJames Wright     PetscInt previous_label_value; // always 1 due to DMPlexMarkBoundaryFaces call above
9644054ae39SJames Wright     PetscCall(DMGetLabelValue(dm, "Face Sets", face, &previous_label_value));
9654054ae39SJames Wright     PetscCall(DMClearLabelValue(dm, "Face Sets", face, previous_label_value));
9664054ae39SJames Wright     PetscCall(DMSetLabelValue(dm, "Face Sets", face, label_value));
9676ff49feeSJames Wright     PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
9684054ae39SJames Wright   }
9696ff49feeSJames Wright   PetscCall(ISRestoreIndices(faces_is, &faces));
9706ff49feeSJames Wright   PetscCall(ISDestroy(&faces_is));
9714054ae39SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
9724054ae39SJames Wright }
9734054ae39SJames Wright 
974d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Simplex_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate)
975d71ae5a4SJacob Faibussowitsch {
9769318fe57SMatthew G. Knepley   DM      boundary, vol;
977c22d3578SMatthew G. Knepley   DMLabel bdlabel;
978d6218766SMatthew G. Knepley 
979d6218766SMatthew G. Knepley   PetscFunctionBegin;
9804f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
981c22d3578SMatthew G. Knepley   for (PetscInt i = 0; i < dim; ++i) PetscCheck(periodicity[i] == DM_BOUNDARY_NONE, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Periodicity is not supported for simplex meshes");
9829566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &boundary));
9839566063dSJacob Faibussowitsch   PetscCall(DMSetType(boundary, DMPLEX));
9849566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(boundary, dim, faces, lower, upper, PETSC_FALSE));
9859566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(boundary, NULL, interpolate, &vol));
986c22d3578SMatthew G. Knepley   PetscCall(DMGetLabel(vol, "marker", &bdlabel));
987c22d3578SMatthew G. Knepley   if (bdlabel) PetscCall(DMPlexLabelComplete(vol, bdlabel));
9885de52c6dSVaclav Hapla   PetscCall(DMPlexCopy_Internal(dm, PETSC_TRUE, PETSC_FALSE, vol));
98969d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
9904054ae39SJames Wright   if (interpolate) {
9914054ae39SJames Wright     PetscCall(DMPlexInterpolateInPlace_Internal(dm));
9924054ae39SJames Wright     PetscCall(DMPlexSetBoxLabel_Internal(dm));
9934054ae39SJames Wright   }
9949566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&boundary));
9953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
996d6218766SMatthew G. Knepley }
997d6218766SMatthew G. Knepley 
998d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCubeMesh_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY, DMBoundaryType bdZ)
999d71ae5a4SJacob Faibussowitsch {
1000ed0e4b50SMatthew G. Knepley   DMLabel     cutLabel  = NULL;
1001f4eb4c5dSMatthew G. Knepley   PetscInt    markerTop = 1, faceMarkerTop = 1;
1002f4eb4c5dSMatthew G. Knepley   PetscInt    markerBottom = 1, faceMarkerBottom = 1;
1003f4eb4c5dSMatthew G. Knepley   PetscInt    markerFront = 1, faceMarkerFront = 1;
1004f4eb4c5dSMatthew G. Knepley   PetscInt    markerBack = 1, faceMarkerBack = 1;
1005f4eb4c5dSMatthew G. Knepley   PetscInt    markerRight = 1, faceMarkerRight = 1;
1006f4eb4c5dSMatthew G. Knepley   PetscInt    markerLeft = 1, faceMarkerLeft = 1;
10073dfda0b1SToby Isaac   PetscInt    dim;
1008d8211ee3SMatthew G. Knepley   PetscBool   markerSeparate = PETSC_FALSE, cutMarker = PETSC_FALSE;
10093dfda0b1SToby Isaac   PetscMPIInt rank;
10103dfda0b1SToby Isaac 
10113dfda0b1SToby Isaac   PetscFunctionBegin;
10129566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
10139566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
10149566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
10159566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
10169566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
10179371c9d4SSatish Balay   if (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST || bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST || bdZ == DM_BOUNDARY_PERIODIC || bdZ == DM_BOUNDARY_TWIST) {
10189371c9d4SSatish Balay     if (cutMarker) {
10199371c9d4SSatish Balay       PetscCall(DMCreateLabel(dm, "periodic_cut"));
10209371c9d4SSatish Balay       PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
10219371c9d4SSatish Balay     }
1022d8211ee3SMatthew G. Knepley   }
10233dfda0b1SToby Isaac   switch (dim) {
10243dfda0b1SToby Isaac   case 2:
1025f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 3;
1026f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1027f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 2;
1028f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 4;
10293dfda0b1SToby Isaac     break;
10303dfda0b1SToby Isaac   case 3:
1031f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1032f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 2;
1033f4eb4c5dSMatthew G. Knepley     faceMarkerFront  = 3;
1034f4eb4c5dSMatthew G. Knepley     faceMarkerBack   = 4;
1035f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 5;
1036f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 6;
10373dfda0b1SToby Isaac     break;
1038d71ae5a4SJacob Faibussowitsch   default:
1039d71ae5a4SJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
10403dfda0b1SToby Isaac   }
10419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
1042f4eb4c5dSMatthew G. Knepley   if (markerSeparate) {
1043f4eb4c5dSMatthew G. Knepley     markerBottom = faceMarkerBottom;
1044f4eb4c5dSMatthew G. Knepley     markerTop    = faceMarkerTop;
1045f4eb4c5dSMatthew G. Knepley     markerFront  = faceMarkerFront;
1046f4eb4c5dSMatthew G. Knepley     markerBack   = faceMarkerBack;
1047f4eb4c5dSMatthew G. Knepley     markerRight  = faceMarkerRight;
1048f4eb4c5dSMatthew G. Knepley     markerLeft   = faceMarkerLeft;
10493dfda0b1SToby Isaac   }
10503dfda0b1SToby Isaac   {
1051dd400576SPatrick Sanan     const PetscInt numXEdges    = rank == 0 ? edges[0] : 0;
1052dd400576SPatrick Sanan     const PetscInt numYEdges    = rank == 0 ? edges[1] : 0;
1053dd400576SPatrick Sanan     const PetscInt numZEdges    = rank == 0 ? edges[2] : 0;
1054dd400576SPatrick Sanan     const PetscInt numXVertices = rank == 0 ? (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST ? edges[0] : edges[0] + 1) : 0;
1055dd400576SPatrick Sanan     const PetscInt numYVertices = rank == 0 ? (bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[1] : edges[1] + 1) : 0;
1056dd400576SPatrick Sanan     const PetscInt numZVertices = rank == 0 ? (bdZ == DM_BOUNDARY_PERIODIC || bdZ == DM_BOUNDARY_TWIST ? edges[2] : edges[2] + 1) : 0;
10573dfda0b1SToby Isaac     const PetscInt numCells     = numXEdges * numYEdges * numZEdges;
10583dfda0b1SToby Isaac     const PetscInt numXFaces    = numYEdges * numZEdges;
10593dfda0b1SToby Isaac     const PetscInt numYFaces    = numXEdges * numZEdges;
10603dfda0b1SToby Isaac     const PetscInt numZFaces    = numXEdges * numYEdges;
10613dfda0b1SToby Isaac     const PetscInt numTotXFaces = numXVertices * numXFaces;
10623dfda0b1SToby Isaac     const PetscInt numTotYFaces = numYVertices * numYFaces;
10633dfda0b1SToby Isaac     const PetscInt numTotZFaces = numZVertices * numZFaces;
10643dfda0b1SToby Isaac     const PetscInt numFaces     = numTotXFaces + numTotYFaces + numTotZFaces;
10653dfda0b1SToby Isaac     const PetscInt numTotXEdges = numXEdges * numYVertices * numZVertices;
10663dfda0b1SToby Isaac     const PetscInt numTotYEdges = numYEdges * numXVertices * numZVertices;
10673dfda0b1SToby Isaac     const PetscInt numTotZEdges = numZEdges * numXVertices * numYVertices;
10683dfda0b1SToby Isaac     const PetscInt numVertices  = numXVertices * numYVertices * numZVertices;
10693dfda0b1SToby Isaac     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numTotZEdges;
10703dfda0b1SToby Isaac     const PetscInt firstVertex  = (dim == 2) ? numFaces : numCells;
10713dfda0b1SToby Isaac     const PetscInt firstXFace   = (dim == 2) ? 0 : numCells + numVertices;
10723dfda0b1SToby Isaac     const PetscInt firstYFace   = firstXFace + numTotXFaces;
10733dfda0b1SToby Isaac     const PetscInt firstZFace   = firstYFace + numTotYFaces;
10743dfda0b1SToby Isaac     const PetscInt firstXEdge   = numCells + numFaces + numVertices;
10753dfda0b1SToby Isaac     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
10763dfda0b1SToby Isaac     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
10773dfda0b1SToby Isaac     Vec            coordinates;
10783dfda0b1SToby Isaac     PetscSection   coordSection;
10793dfda0b1SToby Isaac     PetscScalar   *coords;
10803dfda0b1SToby Isaac     PetscInt       coordSize;
10813dfda0b1SToby Isaac     PetscInt       v, vx, vy, vz;
10823dfda0b1SToby Isaac     PetscInt       c, f, fx, fy, fz, e, ex, ey, ez;
10833dfda0b1SToby Isaac 
10849566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numFaces + numEdges + numVertices));
108548a46eb9SPierre Jolivet     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
108648a46eb9SPierre Jolivet     for (f = firstXFace; f < firstXFace + numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
108748a46eb9SPierre Jolivet     for (e = firstXEdge; e < firstXEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
10889566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
10893dfda0b1SToby Isaac     /* Build cells */
10903dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
10913dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
10923dfda0b1SToby Isaac         for (fx = 0; fx < numXEdges; ++fx) {
10933dfda0b1SToby Isaac           PetscInt cell  = (fz * numYEdges + fy) * numXEdges + fx;
10943dfda0b1SToby Isaac           PetscInt faceB = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
10953dfda0b1SToby Isaac           PetscInt faceT = firstZFace + (fy * numXEdges + fx) * numZVertices + ((fz + 1) % numZVertices);
10963dfda0b1SToby Isaac           PetscInt faceF = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
10973dfda0b1SToby Isaac           PetscInt faceK = firstYFace + (fz * numXEdges + fx) * numYVertices + ((fy + 1) % numYVertices);
10983dfda0b1SToby Isaac           PetscInt faceL = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
10993dfda0b1SToby Isaac           PetscInt faceR = firstXFace + (fz * numYEdges + fy) * numXVertices + ((fx + 1) % numXVertices);
11003dfda0b1SToby Isaac           /* B,  T,  F,  K,  R,  L */
1101b5a892a1SMatthew G. Knepley           PetscInt ornt[6] = {-2, 0, 0, -3, 0, -2}; /* ??? */
110242206facSLisandro Dalcin           PetscInt cone[6];
11033dfda0b1SToby Isaac 
11043dfda0b1SToby Isaac           /* no boundary twisting in 3D */
11059371c9d4SSatish Balay           cone[0] = faceB;
11069371c9d4SSatish Balay           cone[1] = faceT;
11079371c9d4SSatish Balay           cone[2] = faceF;
11089371c9d4SSatish Balay           cone[3] = faceK;
11099371c9d4SSatish Balay           cone[4] = faceR;
11109371c9d4SSatish Balay           cone[5] = faceL;
11119566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, cell, cone));
11129566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, cell, ornt));
11139566063dSJacob Faibussowitsch           if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11149566063dSJacob Faibussowitsch           if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11159566063dSJacob Faibussowitsch           if (bdZ != DM_BOUNDARY_NONE && fz == numZEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11163dfda0b1SToby Isaac         }
11173dfda0b1SToby Isaac       }
11183dfda0b1SToby Isaac     }
11193dfda0b1SToby Isaac     /* Build x faces */
11203dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
11213dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
11223dfda0b1SToby Isaac         for (fx = 0; fx < numXVertices; ++fx) {
11233dfda0b1SToby Isaac           PetscInt face    = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
11243dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
11253dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (((fy + 1) % numYVertices) * numXVertices + fx) * numZEdges + fz;
11263dfda0b1SToby Isaac           PetscInt edgeB   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
11273dfda0b1SToby Isaac           PetscInt edgeT   = firstYEdge + (((fz + 1) % numZVertices) * numXVertices + fx) * numYEdges + fy;
1128b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11293dfda0b1SToby Isaac           PetscInt cone[4];
11303dfda0b1SToby Isaac 
11313dfda0b1SToby Isaac           if (dim == 3) {
11323dfda0b1SToby Isaac             /* markers */
11333dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
11343dfda0b1SToby Isaac               if (fx == numXVertices - 1) {
11359566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerRight));
11369566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerRight));
11379371c9d4SSatish Balay               } else if (fx == 0) {
11389566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerLeft));
11399566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerLeft));
11403dfda0b1SToby Isaac               }
11413dfda0b1SToby Isaac             }
11423dfda0b1SToby Isaac           }
11439371c9d4SSatish Balay           cone[0] = edgeB;
11449371c9d4SSatish Balay           cone[1] = edgeR;
11459371c9d4SSatish Balay           cone[2] = edgeT;
11469371c9d4SSatish Balay           cone[3] = edgeL;
11479566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
11489566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
11493dfda0b1SToby Isaac         }
11503dfda0b1SToby Isaac       }
11513dfda0b1SToby Isaac     }
11523dfda0b1SToby Isaac     /* Build y faces */
11533dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
115442206facSLisandro Dalcin       for (fx = 0; fx < numXEdges; ++fx) {
11553dfda0b1SToby Isaac         for (fy = 0; fy < numYVertices; ++fy) {
11563dfda0b1SToby Isaac           PetscInt face    = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
11573dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
11583dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (fy * numXVertices + ((fx + 1) % numXVertices)) * numZEdges + fz;
11593dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
11603dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (((fz + 1) % numZVertices) * numYVertices + fy) * numXEdges + fx;
1161b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11623dfda0b1SToby Isaac           PetscInt cone[4];
11633dfda0b1SToby Isaac 
11643dfda0b1SToby Isaac           if (dim == 3) {
11653dfda0b1SToby Isaac             /* markers */
11663dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
11673dfda0b1SToby Isaac               if (fy == numYVertices - 1) {
11689566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBack));
11699566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBack));
11709371c9d4SSatish Balay               } else if (fy == 0) {
11719566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerFront));
11729566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerFront));
11733dfda0b1SToby Isaac               }
11743dfda0b1SToby Isaac             }
11753dfda0b1SToby Isaac           }
11769371c9d4SSatish Balay           cone[0] = edgeB;
11779371c9d4SSatish Balay           cone[1] = edgeR;
11789371c9d4SSatish Balay           cone[2] = edgeT;
11799371c9d4SSatish Balay           cone[3] = edgeL;
11809566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
11819566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
11823dfda0b1SToby Isaac         }
11833dfda0b1SToby Isaac       }
11843dfda0b1SToby Isaac     }
11853dfda0b1SToby Isaac     /* Build z faces */
11863dfda0b1SToby Isaac     for (fy = 0; fy < numYEdges; ++fy) {
11873dfda0b1SToby Isaac       for (fx = 0; fx < numXEdges; ++fx) {
11883dfda0b1SToby Isaac         for (fz = 0; fz < numZVertices; fz++) {
11893dfda0b1SToby Isaac           PetscInt face    = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
11903dfda0b1SToby Isaac           PetscInt edgeL   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
11913dfda0b1SToby Isaac           PetscInt edgeR   = firstYEdge + (fz * numXVertices + ((fx + 1) % numXVertices)) * numYEdges + fy;
11923dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
11933dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (fz * numYVertices + ((fy + 1) % numYVertices)) * numXEdges + fx;
1194b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11953dfda0b1SToby Isaac           PetscInt cone[4];
11963dfda0b1SToby Isaac 
11973dfda0b1SToby Isaac           if (dim == 2) {
11989371c9d4SSatish Balay             if (bdX == DM_BOUNDARY_TWIST && fx == numXEdges - 1) {
11999371c9d4SSatish Balay               edgeR += numYEdges - 1 - 2 * fy;
12009371c9d4SSatish Balay               ornt[1] = -1;
12019371c9d4SSatish Balay             }
12029371c9d4SSatish Balay             if (bdY == DM_BOUNDARY_TWIST && fy == numYEdges - 1) {
12039371c9d4SSatish Balay               edgeT += numXEdges - 1 - 2 * fx;
12049371c9d4SSatish Balay               ornt[2] = 0;
12059371c9d4SSatish Balay             }
12069566063dSJacob Faibussowitsch             if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
12079566063dSJacob Faibussowitsch             if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
1208d1c88043SMatthew G. Knepley           } else {
12093dfda0b1SToby Isaac             /* markers */
12103dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
12113dfda0b1SToby Isaac               if (fz == numZVertices - 1) {
12129566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerTop));
12139566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerTop));
12149371c9d4SSatish Balay               } else if (fz == 0) {
12159566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBottom));
12169566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBottom));
12173dfda0b1SToby Isaac               }
12183dfda0b1SToby Isaac             }
12193dfda0b1SToby Isaac           }
12209371c9d4SSatish Balay           cone[0] = edgeB;
12219371c9d4SSatish Balay           cone[1] = edgeR;
12229371c9d4SSatish Balay           cone[2] = edgeT;
12239371c9d4SSatish Balay           cone[3] = edgeL;
12249566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
12259566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
12263dfda0b1SToby Isaac         }
12273dfda0b1SToby Isaac       }
12283dfda0b1SToby Isaac     }
12293dfda0b1SToby Isaac     /* Build Z edges*/
12303dfda0b1SToby Isaac     for (vy = 0; vy < numYVertices; vy++) {
12313dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
12323dfda0b1SToby Isaac         for (ez = 0; ez < numZEdges; ez++) {
12333dfda0b1SToby Isaac           const PetscInt edge    = firstZEdge + (vy * numXVertices + vx) * numZEdges + ez;
12343dfda0b1SToby Isaac           const PetscInt vertexB = firstVertex + (ez * numYVertices + vy) * numXVertices + vx;
12353dfda0b1SToby Isaac           const PetscInt vertexT = firstVertex + (((ez + 1) % numZVertices) * numYVertices + vy) * numXVertices + vx;
12363dfda0b1SToby Isaac           PetscInt       cone[2];
12373dfda0b1SToby Isaac 
12389371c9d4SSatish Balay           cone[0] = vertexB;
12399371c9d4SSatish Balay           cone[1] = vertexT;
1240c2df9bbfSMatthew G. Knepley           PetscCall(DMPlexSetCone(dm, edge, cone));
12413dfda0b1SToby Isaac           if (dim == 3) {
12423dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
12433dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
12449566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1245c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1246c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1247c2df9bbfSMatthew G. Knepley               } else if (vx == 0) {
12489566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1249c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1250c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
12513dfda0b1SToby Isaac               }
12523dfda0b1SToby Isaac             }
12533dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
12543dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
12559566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1256c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1257c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1258c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
12599566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1260c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1261c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
12623dfda0b1SToby Isaac               }
12633dfda0b1SToby Isaac             }
12643dfda0b1SToby Isaac           }
12653dfda0b1SToby Isaac         }
12663dfda0b1SToby Isaac       }
12673dfda0b1SToby Isaac     }
12683dfda0b1SToby Isaac     /* Build Y edges*/
12693dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
12703dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
12713dfda0b1SToby Isaac         for (ey = 0; ey < numYEdges; ey++) {
12723dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdY == DM_BOUNDARY_TWIST && ey == numYEdges - 1) ? (numXVertices - vx - 1) : (vz * numYVertices + ((ey + 1) % numYVertices)) * numXVertices + vx;
12733dfda0b1SToby Isaac           const PetscInt edge    = firstYEdge + (vz * numXVertices + vx) * numYEdges + ey;
12743dfda0b1SToby Isaac           const PetscInt vertexF = firstVertex + (vz * numYVertices + ey) * numXVertices + vx;
12753dfda0b1SToby Isaac           const PetscInt vertexK = firstVertex + nextv;
12763dfda0b1SToby Isaac           PetscInt       cone[2];
12773dfda0b1SToby Isaac 
12789371c9d4SSatish Balay           cone[0] = vertexF;
12799371c9d4SSatish Balay           cone[1] = vertexK;
12809566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
12813dfda0b1SToby Isaac           if (dim == 2) {
12823dfda0b1SToby Isaac             if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
12833dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
12849566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight));
12859566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
12869566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1287c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1288d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
12899566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft));
12909566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
12919566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1292c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
12933dfda0b1SToby Isaac               }
1294d8211ee3SMatthew G. Knepley             } else {
12954c67ea77SStefano Zampini               if (vx == 0 && cutLabel) {
12969566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
12979566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1298c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
12993dfda0b1SToby Isaac               }
1300d8211ee3SMatthew G. Knepley             }
1301d8211ee3SMatthew G. Knepley           } else {
13023dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
13033dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
13049566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1305c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1306c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1307d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
13089566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1309c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1310c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
13113dfda0b1SToby Isaac               }
13123dfda0b1SToby Isaac             }
13133dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
13143dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
13159566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1316c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1317c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1318d8211ee3SMatthew G. Knepley               } else if (vz == 0) {
13199566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1320c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1321c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13223dfda0b1SToby Isaac               }
13233dfda0b1SToby Isaac             }
13243dfda0b1SToby Isaac           }
13253dfda0b1SToby Isaac         }
13263dfda0b1SToby Isaac       }
13273dfda0b1SToby Isaac     }
13283dfda0b1SToby Isaac     /* Build X edges*/
13293dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
13303dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; vy++) {
13313dfda0b1SToby Isaac         for (ex = 0; ex < numXEdges; ex++) {
13323dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdX == DM_BOUNDARY_TWIST && ex == numXEdges - 1) ? (numYVertices - vy - 1) * numXVertices : (vz * numYVertices + vy) * numXVertices + (ex + 1) % numXVertices;
13333dfda0b1SToby Isaac           const PetscInt edge    = firstXEdge + (vz * numYVertices + vy) * numXEdges + ex;
13343dfda0b1SToby Isaac           const PetscInt vertexL = firstVertex + (vz * numYVertices + vy) * numXVertices + ex;
13353dfda0b1SToby Isaac           const PetscInt vertexR = firstVertex + nextv;
13363dfda0b1SToby Isaac           PetscInt       cone[2];
13373dfda0b1SToby Isaac 
13389371c9d4SSatish Balay           cone[0] = vertexL;
13399371c9d4SSatish Balay           cone[1] = vertexR;
13409566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
13413dfda0b1SToby Isaac           if (dim == 2) {
13423dfda0b1SToby Isaac             if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
13433dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
13449566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop));
13459566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
13469566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1347c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1348d8211ee3SMatthew G. Knepley               } else if (vy == 0) {
13499566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom));
13509566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
13519566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1352c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13533dfda0b1SToby Isaac               }
1354d8211ee3SMatthew G. Knepley             } else {
13554c67ea77SStefano Zampini               if (vy == 0 && cutLabel) {
13569566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
13579566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1358c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
13593dfda0b1SToby Isaac               }
1360d8211ee3SMatthew G. Knepley             }
1361d8211ee3SMatthew G. Knepley           } else {
13623dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
13633dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
13649566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1365c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1366c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1367c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
13689566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1369c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1370c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
13713dfda0b1SToby Isaac               }
13723dfda0b1SToby Isaac             }
13733dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
13743dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
13759566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1376c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1377c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1378c2df9bbfSMatthew G. Knepley               } else if (vz == 0) {
13799566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1380c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1381c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13823dfda0b1SToby Isaac               }
13833dfda0b1SToby Isaac             }
13843dfda0b1SToby Isaac           }
13853dfda0b1SToby Isaac         }
13863dfda0b1SToby Isaac       }
13873dfda0b1SToby Isaac     }
13889566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
13899566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
13903dfda0b1SToby Isaac     /* Build coordinates */
13919566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
13929566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
13939566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
13949566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVertices));
13953dfda0b1SToby Isaac     for (v = firstVertex; v < firstVertex + numVertices; ++v) {
13969566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
13979566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
13983dfda0b1SToby Isaac     }
13999566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
14009566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
14019566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
14029566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
14039566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
14049566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
14059566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
14069566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
14073dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; ++vz) {
14083dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; ++vy) {
14093dfda0b1SToby Isaac         for (vx = 0; vx < numXVertices; ++vx) {
14103dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * vx;
14113dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * vy;
1412ad540459SPierre Jolivet           if (dim == 3) coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 2] = lower[2] + ((upper[2] - lower[2]) / numZEdges) * vz;
14133dfda0b1SToby Isaac         }
14143dfda0b1SToby Isaac       }
14153dfda0b1SToby Isaac     }
14169566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
14179566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
14189566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
14193dfda0b1SToby Isaac   }
14203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14213dfda0b1SToby Isaac }
14223dfda0b1SToby Isaac 
1423d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Tensor_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1424d71ae5a4SJacob Faibussowitsch {
14259318fe57SMatthew G. Knepley   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
14269318fe57SMatthew G. Knepley   PetscInt       fac[3] = {0, 0, 0}, d;
1427552f7358SJed Brown 
1428552f7358SJed Brown   PetscFunctionBegin;
14294f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
14309318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
14319566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
14329371c9d4SSatish Balay   for (d = 0; d < dim; ++d) {
14339371c9d4SSatish Balay     fac[d] = faces[d];
14349371c9d4SSatish Balay     bdt[d] = periodicity[d];
14359371c9d4SSatish Balay   }
14369566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateCubeMesh_Internal(dm, lower, upper, fac, bdt[0], bdt[1], bdt[2]));
14379371c9d4SSatish Balay   if (periodicity[0] == DM_BOUNDARY_PERIODIC || periodicity[0] == DM_BOUNDARY_TWIST || periodicity[1] == DM_BOUNDARY_PERIODIC || periodicity[1] == DM_BOUNDARY_TWIST || (dim > 2 && (periodicity[2] == DM_BOUNDARY_PERIODIC || periodicity[2] == DM_BOUNDARY_TWIST))) {
14386858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
14396858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
1440552f7358SJed Brown 
14419318fe57SMatthew G. Knepley     for (d = 0; d < dim; ++d) {
14426858538eSMatthew G. Knepley       if (periodicity[d] != DM_BOUNDARY_NONE) {
14439318fe57SMatthew G. Knepley         L[d]       = upper[d] - lower[d];
14449318fe57SMatthew G. Knepley         maxCell[d] = 1.1 * (L[d] / PetscMax(1, faces[d]));
1445768d5fceSMatthew G. Knepley       }
14466858538eSMatthew G. Knepley     }
14474fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
1448768d5fceSMatthew G. Knepley   }
14499566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
14503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14519318fe57SMatthew G. Knepley }
14529318fe57SMatthew G. Knepley 
14535dca41c3SJed Brown static PetscErrorCode DMPlexCreateBoxMesh_Internal(DM dm, DMPlexShape shape, PetscInt dim, PetscBool simplex, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate)
1454d71ae5a4SJacob Faibussowitsch {
14559318fe57SMatthew G. Knepley   PetscFunctionBegin;
145646139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
14575dca41c3SJed Brown   if (shape == DM_SHAPE_ZBOX) PetscCall(DMPlexCreateBoxMesh_Tensor_SFC_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
14586725e60dSJed Brown   else if (dim == 1) PetscCall(DMPlexCreateLineMesh_Internal(dm, faces[0], lower[0], upper[0], periodicity[0]));
14599566063dSJacob Faibussowitsch   else if (simplex) PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
14609566063dSJacob Faibussowitsch   else PetscCall(DMPlexCreateBoxMesh_Tensor_Internal(dm, dim, faces, lower, upper, periodicity));
14619318fe57SMatthew G. Knepley   if (!interpolate && dim > 1 && !simplex) {
1462768d5fceSMatthew G. Knepley     DM udm;
1463768d5fceSMatthew G. Knepley 
14649566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(dm, &udm));
14659566063dSJacob Faibussowitsch     PetscCall(DMPlexCopyCoordinates(dm, udm));
146669d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &udm));
1467768d5fceSMatthew G. Knepley   }
146846139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
14693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1470c8c68bd8SToby Isaac }
1471c8c68bd8SToby Isaac 
14725d83a8b1SBarry Smith /*@
1473768d5fceSMatthew G. Knepley   DMPlexCreateBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using simplices or tensor cells (hexahedra).
1474768d5fceSMatthew G. Knepley 
1475d083f849SBarry Smith   Collective
1476768d5fceSMatthew G. Knepley 
1477768d5fceSMatthew G. Knepley   Input Parameters:
1478a1cb98faSBarry Smith + comm               - The communicator for the `DM` object
1479768d5fceSMatthew G. Knepley . dim                - The spatial dimension
1480a1cb98faSBarry Smith . simplex            - `PETSC_TRUE` for simplices, `PETSC_FALSE` for tensor cells
148120f4b53cSBarry Smith . faces              - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
148220f4b53cSBarry Smith . lower              - The lower left corner, or `NULL` for (0, 0, 0)
148320f4b53cSBarry Smith . upper              - The upper right corner, or `NULL` for (1, 1, 1)
148420f4b53cSBarry Smith . periodicity        - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
148542108689Sksagiyam . interpolate        - Flag to create intermediate mesh pieces (edges, faces)
148642108689Sksagiyam . localizationHeight - Flag to localize edges and faces in addition to cells; only significant for periodic meshes
148742108689Sksagiyam - sparseLocalize     - Flag to localize coordinates only for cells near the periodic boundary; only significant for periodic meshes
1488768d5fceSMatthew G. Knepley 
1489768d5fceSMatthew G. Knepley   Output Parameter:
1490a1cb98faSBarry Smith . dm - The `DM` object
1491768d5fceSMatthew G. Knepley 
1492768d5fceSMatthew G. Knepley   Level: beginner
1493768d5fceSMatthew G. Knepley 
1494a1cb98faSBarry Smith   Note:
1495a1cb98faSBarry Smith   To customize this mesh using options, use
1496a1cb98faSBarry Smith .vb
1497a1cb98faSBarry Smith   DMCreate(comm, &dm);
1498a1cb98faSBarry Smith   DMSetType(dm, DMPLEX);
1499a1cb98faSBarry Smith   DMSetFromOptions(dm);
1500a1cb98faSBarry Smith .ve
1501a1cb98faSBarry Smith   and use the options in `DMSetFromOptions()`.
1502a1cb98faSBarry Smith 
1503a4e35b19SJacob Faibussowitsch   Here is the numbering returned for 2 faces in each direction for tensor cells\:
1504a1cb98faSBarry Smith .vb
1505a1cb98faSBarry Smith  10---17---11---18----12
1506a1cb98faSBarry Smith   |         |         |
1507a1cb98faSBarry Smith   |         |         |
1508a1cb98faSBarry Smith  20    2   22    3    24
1509a1cb98faSBarry Smith   |         |         |
1510a1cb98faSBarry Smith   |         |         |
1511a1cb98faSBarry Smith   7---15----8---16----9
1512a1cb98faSBarry Smith   |         |         |
1513a1cb98faSBarry Smith   |         |         |
1514a1cb98faSBarry Smith  19    0   21    1   23
1515a1cb98faSBarry Smith   |         |         |
1516a1cb98faSBarry Smith   |         |         |
1517a1cb98faSBarry Smith   4---13----5---14----6
1518a1cb98faSBarry Smith .ve
1519a1cb98faSBarry Smith   and for simplicial cells
1520a1cb98faSBarry Smith .vb
1521a1cb98faSBarry Smith  14----8---15----9----16
1522a1cb98faSBarry Smith   |\     5  |\      7 |
1523a1cb98faSBarry Smith   | \       | \       |
1524a1cb98faSBarry Smith  13   2    14    3    15
1525a1cb98faSBarry Smith   | 4   \   | 6   \   |
1526a1cb98faSBarry Smith   |       \ |       \ |
1527a1cb98faSBarry Smith  11----6---12----7----13
1528a1cb98faSBarry Smith   |\        |\        |
1529a1cb98faSBarry Smith   | \    1  | \     3 |
1530a1cb98faSBarry Smith  10   0    11    1    12
1531a1cb98faSBarry Smith   | 0   \   | 2   \   |
1532a1cb98faSBarry Smith   |       \ |       \ |
1533a1cb98faSBarry Smith   8----4----9----5----10
1534a1cb98faSBarry Smith .ve
1535a1cb98faSBarry Smith 
15361cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
1537768d5fceSMatthew G. Knepley @*/
153842108689Sksagiyam PetscErrorCode DMPlexCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate, PetscInt localizationHeight, PetscBool sparseLocalize, DM *dm)
1539d71ae5a4SJacob Faibussowitsch {
15409318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
1541fdbf62faSLisandro Dalcin   PetscReal      low[3] = {0, 0, 0};
1542fdbf62faSLisandro Dalcin   PetscReal      upp[3] = {1, 1, 1};
1543fdbf62faSLisandro Dalcin   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
1544552f7358SJed Brown 
1545768d5fceSMatthew G. Knepley   PetscFunctionBegin;
15469566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
15479566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
15485dca41c3SJed Brown   PetscCall(DMPlexCreateBoxMesh_Internal(*dm, DM_SHAPE_BOX, dim, simplex, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt, interpolate));
154942108689Sksagiyam   if (periodicity) {
155042108689Sksagiyam     DM cdm;
155142108689Sksagiyam 
155242108689Sksagiyam     PetscCall(DMGetCoordinateDM(*dm, &cdm));
155342108689Sksagiyam     PetscCall(DMPlexSetMaxProjectionHeight(cdm, localizationHeight));
155442108689Sksagiyam     PetscCall(DMSetSparseLocalize(*dm, sparseLocalize));
155542108689Sksagiyam     PetscCall(DMLocalizeCoordinates(*dm));
155642108689Sksagiyam   }
15573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15589318fe57SMatthew G. Knepley }
1559fdbf62faSLisandro Dalcin 
1560d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeBoxMesh_Internal(DM dm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1561d71ae5a4SJacob Faibussowitsch {
15629318fe57SMatthew G. Knepley   DM       bdm, vol;
15639318fe57SMatthew G. Knepley   PetscInt i;
15649318fe57SMatthew G. Knepley 
15659318fe57SMatthew G. Knepley   PetscFunctionBegin;
15661fcf445aSMatthew G. Knepley   // TODO Now we can support periodicity
156708401ef6SPierre Jolivet   for (i = 0; i < 3; ++i) PetscCheck(periodicity[i] == DM_BOUNDARY_NONE, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Periodicity not yet supported");
15689566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &bdm));
15699566063dSJacob Faibussowitsch   PetscCall(DMSetType(bdm, DMPLEX));
15709566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(bdm, 2));
157146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, bdm, 0, 0, 0));
15729566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(bdm, 2, faces, lower, upper, periodicity, PETSC_TRUE));
15731fcf445aSMatthew G. Knepley   PetscCall(DMPlexExtrude(bdm, faces[2], upper[2] - lower[2], PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, NULL, NULL, &vol));
157446139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, bdm, 0, 0, 0));
15759566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&bdm));
157669d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
15779318fe57SMatthew G. Knepley   if (lower[2] != 0.0) {
15789318fe57SMatthew G. Knepley     Vec          v;
15799318fe57SMatthew G. Knepley     PetscScalar *x;
15809318fe57SMatthew G. Knepley     PetscInt     cDim, n;
15819318fe57SMatthew G. Knepley 
15829566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &v));
15839566063dSJacob Faibussowitsch     PetscCall(VecGetBlockSize(v, &cDim));
15849566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(v, &n));
15859566063dSJacob Faibussowitsch     PetscCall(VecGetArray(v, &x));
15869318fe57SMatthew G. Knepley     x += cDim;
15879318fe57SMatthew G. Knepley     for (i = 0; i < n; i += cDim) x[i] += lower[2];
15889566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(v, &x));
15899566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, v));
15909318fe57SMatthew G. Knepley   }
15913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1592552f7358SJed Brown }
1593552f7358SJed Brown 
159400dabe28SStefano Zampini /*@
159539f4f5dbSPierre Jolivet   DMPlexCreateWedgeBoxMesh - Creates a 3-D mesh tessellating the (x,y) plane and extruding in the third direction using wedge cells.
159600dabe28SStefano Zampini 
1597d083f849SBarry Smith   Collective
159800dabe28SStefano Zampini 
159900dabe28SStefano Zampini   Input Parameters:
1600a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
160120f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1, 1, 1)
160220f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
160320f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
160420f4b53cSBarry Smith . periodicity - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
1605a1cb98faSBarry Smith . orderHeight - If `PETSC_TRUE`, orders the extruded cells in the height first. Otherwise, orders the cell on the layers first
160600dabe28SStefano Zampini - interpolate - Flag to create intermediate mesh pieces (edges, faces)
160700dabe28SStefano Zampini 
160800dabe28SStefano Zampini   Output Parameter:
1609a1cb98faSBarry Smith . dm - The `DM` object
161000dabe28SStefano Zampini 
161100dabe28SStefano Zampini   Level: beginner
161200dabe28SStefano Zampini 
16131cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateWedgeCylinderMesh()`, `DMExtrude()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
161400dabe28SStefano Zampini @*/
1615d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeBoxMesh(MPI_Comm comm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool orderHeight, PetscBool interpolate, DM *dm)
1616d71ae5a4SJacob Faibussowitsch {
16179318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
161800dabe28SStefano Zampini   PetscReal      low[3] = {0, 0, 0};
161900dabe28SStefano Zampini   PetscReal      upp[3] = {1, 1, 1};
162000dabe28SStefano Zampini   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
162100dabe28SStefano Zampini 
162200dabe28SStefano Zampini   PetscFunctionBegin;
16239566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
16249566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
16259566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeBoxMesh_Internal(*dm, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt));
1626d410b0cfSMatthew G. Knepley   if (!interpolate) {
1627d410b0cfSMatthew G. Knepley     DM udm;
162800dabe28SStefano Zampini 
16299566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(*dm, &udm));
163069d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(*dm, &udm));
163100dabe28SStefano Zampini   }
16327ff04441SMatthew G. Knepley   if (periodicity) PetscCall(DMLocalizeCoordinates(*dm));
16333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
163400dabe28SStefano Zampini }
163500dabe28SStefano Zampini 
1636cfb853baSMatthew G. Knepley /*
1637cfb853baSMatthew G. Knepley   DMPlexTensorPointLexicographic_Private - Returns all tuples of size 'len' with nonnegative integers that are all less than or equal to 'max' for that dimension.
1638cfb853baSMatthew G. Knepley 
1639cfb853baSMatthew G. Knepley   Input Parameters:
1640cfb853baSMatthew G. Knepley + len - The length of the tuple
1641cfb853baSMatthew G. Knepley . max - The maximum for each dimension, so values are in [0, max)
1642cfb853baSMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition
1643cfb853baSMatthew G. Knepley 
1644cfb853baSMatthew G. Knepley   Output Parameter:
164520f4b53cSBarry Smith . tup - A tuple of `len` integers whose entries are at most `max`
1646cfb853baSMatthew G. Knepley 
1647cfb853baSMatthew G. Knepley   Level: developer
1648cfb853baSMatthew G. Knepley 
164920f4b53cSBarry Smith   Note:
165020f4b53cSBarry Smith   Ordering is lexicographic with lowest index as least significant in ordering.
165120f4b53cSBarry Smith   e.g. for len == 2 and max == 2, this will return, in order, {0,0}, {1,0}, {2,0}, {0,1}, {1,1}, {2,1}, {0,2}, {1,2}, {2,2}.
165220f4b53cSBarry Smith 
1653cfb853baSMatthew G. Knepley .seealso: PetscDualSpaceTensorPointLexicographic_Internal(), PetscDualSpaceLatticePointLexicographic_Internal()
1654cfb853baSMatthew G. Knepley */
1655cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexTensorPointLexicographic_Private(PetscInt len, const PetscInt max[], PetscInt tup[])
1656cfb853baSMatthew G. Knepley {
1657cfb853baSMatthew G. Knepley   PetscInt i;
1658cfb853baSMatthew G. Knepley 
1659cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1660cfb853baSMatthew G. Knepley   for (i = 0; i < len; ++i) {
1661cfb853baSMatthew G. Knepley     if (tup[i] < max[i] - 1) {
1662cfb853baSMatthew G. Knepley       break;
1663cfb853baSMatthew G. Knepley     } else {
1664cfb853baSMatthew G. Knepley       tup[i] = 0;
1665cfb853baSMatthew G. Knepley     }
1666cfb853baSMatthew G. Knepley   }
1667cfb853baSMatthew G. Knepley   if (i == len) tup[i - 1] = max[i - 1];
1668cfb853baSMatthew G. Knepley   else ++tup[i];
16693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1670cfb853baSMatthew G. Knepley }
1671cfb853baSMatthew G. Knepley 
1672cfb853baSMatthew G. Knepley static PetscInt TupleToIndex_Private(PetscInt len, const PetscInt max[], const PetscInt tup[])
1673cfb853baSMatthew G. Knepley {
1674cfb853baSMatthew G. Knepley   PetscInt i, idx = tup[len - 1];
1675cfb853baSMatthew G. Knepley 
1676cfb853baSMatthew G. Knepley   for (i = len - 2; i >= 0; --i) {
1677cfb853baSMatthew G. Knepley     idx *= max[i];
1678cfb853baSMatthew G. Knepley     idx += tup[i];
1679cfb853baSMatthew G. Knepley   }
1680cfb853baSMatthew G. Knepley   return idx;
1681cfb853baSMatthew G. Knepley }
1682cfb853baSMatthew G. Knepley 
1683cfb853baSMatthew G. Knepley static PetscErrorCode DestroyExtent_Private(void *extent)
1684cfb853baSMatthew G. Knepley {
1685cfb853baSMatthew G. Knepley   return PetscFree(extent);
1686cfb853baSMatthew G. Knepley }
1687cfb853baSMatthew G. Knepley 
1688cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexCreateHypercubicMesh_Internal(DM dm, PetscInt dim, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], const DMBoundaryType bd[])
1689cfb853baSMatthew G. Knepley {
1690cfb853baSMatthew G. Knepley   Vec          coordinates;
1691cfb853baSMatthew G. Knepley   PetscSection coordSection;
1692cfb853baSMatthew G. Knepley   DMLabel      cutLabel    = NULL;
1693cfb853baSMatthew G. Knepley   PetscBool    cutMarker   = PETSC_FALSE;
1694cfb853baSMatthew G. Knepley   PetscBool    periodic    = PETSC_FALSE;
1695cfb853baSMatthew G. Knepley   PetscInt     numCells    = 1, c;
1696cfb853baSMatthew G. Knepley   PetscInt     numVertices = 1, v;
1697cfb853baSMatthew G. Knepley   PetscScalar *coords;
1698cfb853baSMatthew G. Knepley   PetscInt    *vertices, *vert, *vtmp, *supp, cone[2];
1699cfb853baSMatthew G. Knepley   PetscInt     d, e, cell = 0, coordSize;
1700cfb853baSMatthew G. Knepley   PetscMPIInt  rank;
1701cfb853baSMatthew G. Knepley 
1702cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1703cfb853baSMatthew G. Knepley   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1704cfb853baSMatthew G. Knepley   PetscCall(DMSetDimension(dm, dim));
1705cfb853baSMatthew G. Knepley   PetscCall(PetscCalloc4(dim, &vertices, dim, &vert, dim, &vtmp, 2 * dim, &supp));
1706cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
1707cfb853baSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
1708cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) periodic = (periodic || bd[d] == DM_BOUNDARY_PERIODIC) ? PETSC_TRUE : PETSC_FALSE;
1709cfb853baSMatthew G. Knepley   if (periodic && cutMarker) {
1710cfb853baSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, "periodic_cut"));
1711cfb853baSMatthew G. Knepley     PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
1712cfb853baSMatthew G. Knepley   }
1713cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) PetscCheck(bd[d] == DM_BOUNDARY_PERIODIC, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Hypercubic mesh must be periodic now");
1714cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
1715cfb853baSMatthew G. Knepley     vertices[d] = edges[d];
1716cfb853baSMatthew G. Knepley     numVertices *= vertices[d];
1717cfb853baSMatthew G. Knepley   }
1718cfb853baSMatthew G. Knepley   numCells = numVertices * dim;
1719cfb853baSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
1720cfb853baSMatthew G. Knepley   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, 2));
1721cfb853baSMatthew G. Knepley   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetSupportSize(dm, v, 2 * dim));
1722cfb853baSMatthew G. Knepley   /* TODO Loop over boundary and reset support sizes */
1723cfb853baSMatthew G. Knepley   PetscCall(DMSetUp(dm)); /* Allocate space for cones and supports */
1724cfb853baSMatthew G. Knepley   /* Build cell cones and vertex supports */
1725cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "celltype"));
1726cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
1727cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert) + numCells;
1728cfb853baSMatthew G. Knepley     PetscInt       s      = 0;
1729cfb853baSMatthew G. Knepley 
17303ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "Vertex %" PetscInt_FMT ":", vertex));
17313ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vert[d]));
17323ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1733cfb853baSMatthew G. Knepley     PetscCall(DMPlexSetCellType(dm, vertex, DM_POLYTOPE_POINT));
1734cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) {
1735cfb853baSMatthew G. Knepley       for (e = 0; e < dim; ++e) vtmp[e] = vert[e];
1736cfb853baSMatthew G. Knepley       vtmp[d] = (vert[d] + 1) % vertices[d];
1737cfb853baSMatthew G. Knepley       cone[0] = vertex;
1738cfb853baSMatthew G. Knepley       cone[1] = TupleToIndex_Private(dim, vertices, vtmp) + numCells;
17393ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Vertex %" PetscInt_FMT ":", cone[1]));
17403ba16761SJacob Faibussowitsch       for (e = 0; e < dim; ++e) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vtmp[e]));
17413ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1742cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, cell, cone));
1743cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCellType(dm, cell, DM_POLYTOPE_SEGMENT));
17443ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Edge %" PetscInt_FMT " (%" PetscInt_FMT " %" PetscInt_FMT ")\n", cell, cone[0], cone[1]));
1745cfb853baSMatthew G. Knepley       ++cell;
1746cfb853baSMatthew G. Knepley     }
1747cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) {
1748cfb853baSMatthew G. Knepley       for (e = 0; e < dim; ++e) vtmp[e] = vert[e];
1749cfb853baSMatthew G. Knepley       vtmp[d]   = (vert[d] + vertices[d] - 1) % vertices[d];
1750cfb853baSMatthew G. Knepley       supp[s++] = TupleToIndex_Private(dim, vertices, vtmp) * dim + d;
1751cfb853baSMatthew G. Knepley       supp[s++] = (vertex - numCells) * dim + d;
1752cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetSupport(dm, vertex, supp));
1753cfb853baSMatthew G. Knepley     }
1754cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
1755cfb853baSMatthew G. Knepley   }
1756cfb853baSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
1757cfb853baSMatthew G. Knepley   /* Build coordinates */
1758cfb853baSMatthew G. Knepley   PetscCall(DMGetCoordinateSection(dm, &coordSection));
1759cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetNumFields(coordSection, 1));
1760cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
1761cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
1762cfb853baSMatthew G. Knepley   for (v = numCells; v < numCells + numVertices; ++v) {
1763cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetDof(coordSection, v, dim));
1764cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
1765cfb853baSMatthew G. Knepley   }
1766cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetUp(coordSection));
1767cfb853baSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
1768cfb853baSMatthew G. Knepley   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
1769cfb853baSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
1770cfb853baSMatthew G. Knepley   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
1771cfb853baSMatthew G. Knepley   PetscCall(VecSetBlockSize(coordinates, dim));
1772cfb853baSMatthew G. Knepley   PetscCall(VecSetType(coordinates, VECSTANDARD));
1773cfb853baSMatthew G. Knepley   PetscCall(VecGetArray(coordinates, &coords));
1774cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) vert[d] = 0;
1775cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
1776cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert);
1777cfb853baSMatthew G. Knepley 
1778cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) coords[vertex * dim + d] = lower[d] + ((upper[d] - lower[d]) / vertices[d]) * vert[d];
17793ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "Vertex %" PetscInt_FMT ":", vertex));
17803ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vert[d]));
17813ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %g", (double)PetscRealPart(coords[vertex * dim + d])));
17823ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1783cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
1784cfb853baSMatthew G. Knepley   }
1785cfb853baSMatthew G. Knepley   PetscCall(VecRestoreArray(coordinates, &coords));
1786cfb853baSMatthew G. Knepley   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
1787cfb853baSMatthew G. Knepley   PetscCall(VecDestroy(&coordinates));
1788cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(vertices, vert, vtmp, supp));
1789cfb853baSMatthew G. Knepley   //PetscCall(DMSetPeriodicity(dm, NULL, lower, upper));
1790cfb853baSMatthew G. Knepley   // Attach the extent
1791cfb853baSMatthew G. Knepley   {
1792cfb853baSMatthew G. Knepley     PetscContainer c;
1793cfb853baSMatthew G. Knepley     PetscInt      *extent;
1794cfb853baSMatthew G. Knepley 
1795cfb853baSMatthew G. Knepley     PetscCall(PetscMalloc1(dim, &extent));
1796cfb853baSMatthew G. Knepley     for (PetscInt d = 0; d < dim; ++d) extent[d] = edges[d];
1797cfb853baSMatthew G. Knepley     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
1798cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetUserDestroy(c, DestroyExtent_Private));
1799cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetPointer(c, extent));
1800cfb853baSMatthew G. Knepley     PetscCall(PetscObjectCompose((PetscObject)dm, "_extent", (PetscObject)c));
1801cfb853baSMatthew G. Knepley     PetscCall(PetscContainerDestroy(&c));
1802cfb853baSMatthew G. Knepley   }
18033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1804cfb853baSMatthew G. Knepley }
1805cfb853baSMatthew G. Knepley 
1806cfb853baSMatthew G. Knepley /*@C
1807aaa8cc7dSPierre Jolivet   DMPlexCreateHypercubicMesh - Creates a periodic mesh on the tensor product of unit intervals using only vertices and edges.
1808cfb853baSMatthew G. Knepley 
1809cfb853baSMatthew G. Knepley   Collective
1810cfb853baSMatthew G. Knepley 
1811cfb853baSMatthew G. Knepley   Input Parameters:
1812cfb853baSMatthew G. Knepley + comm  - The communicator for the DM object
1813cfb853baSMatthew G. Knepley . dim   - The spatial dimension
181420f4b53cSBarry Smith . edges - Number of edges per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
181520f4b53cSBarry Smith . lower - The lower left corner, or `NULL` for (0, 0, 0)
181620f4b53cSBarry Smith - upper - The upper right corner, or `NULL` for (1, 1, 1)
1817cfb853baSMatthew G. Knepley 
1818cfb853baSMatthew G. Knepley   Output Parameter:
1819cfb853baSMatthew G. Knepley . dm - The DM object
1820cfb853baSMatthew G. Knepley 
182120f4b53cSBarry Smith   Level: beginner
182220f4b53cSBarry Smith 
182320f4b53cSBarry Smith   Note:
182420f4b53cSBarry Smith   If you want to customize this mesh using options, you just need to
182520f4b53cSBarry Smith .vb
182620f4b53cSBarry Smith   DMCreate(comm, &dm);
182720f4b53cSBarry Smith   DMSetType(dm, DMPLEX);
182820f4b53cSBarry Smith   DMSetFromOptions(dm);
182920f4b53cSBarry Smith .ve
183020f4b53cSBarry Smith   and use the options on the `DMSetFromOptions()` page.
1831cfb853baSMatthew G. Knepley 
1832cfb853baSMatthew G. Knepley   The vertices are numbered is lexicographic order, and the dim edges exiting a vertex in the positive orthant are number consecutively,
183320f4b53cSBarry Smith .vb
183420f4b53cSBarry Smith  18--0-19--2-20--4-18
183520f4b53cSBarry Smith   |     |     |     |
183620f4b53cSBarry Smith  13    15    17    13
183720f4b53cSBarry Smith   |     |     |     |
183820f4b53cSBarry Smith  24-12-25-14-26-16-24
183920f4b53cSBarry Smith   |     |     |     |
184020f4b53cSBarry Smith   7     9    11     7
184120f4b53cSBarry Smith   |     |     |     |
184220f4b53cSBarry Smith  21--6-22--8-23-10-21
184320f4b53cSBarry Smith   |     |     |     |
184420f4b53cSBarry Smith   1     3     5     1
184520f4b53cSBarry Smith   |     |     |     |
184620f4b53cSBarry Smith  18--0-19--2-20--4-18
184720f4b53cSBarry Smith .ve
1848cfb853baSMatthew G. Knepley 
184976fbde31SPierre Jolivet .seealso: `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
1850cfb853baSMatthew G. Knepley @*/
1851cfb853baSMatthew G. Knepley PetscErrorCode DMPlexCreateHypercubicMesh(MPI_Comm comm, PetscInt dim, const PetscInt edges[], const PetscReal lower[], const PetscReal upper[], DM *dm)
1852cfb853baSMatthew G. Knepley {
1853cfb853baSMatthew G. Knepley   PetscInt       *edg;
1854cfb853baSMatthew G. Knepley   PetscReal      *low, *upp;
1855cfb853baSMatthew G. Knepley   DMBoundaryType *bdt;
1856cfb853baSMatthew G. Knepley   PetscInt        d;
1857cfb853baSMatthew G. Knepley 
1858cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1859cfb853baSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
1860cfb853baSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
1861cfb853baSMatthew G. Knepley   PetscCall(PetscMalloc4(dim, &edg, dim, &low, dim, &upp, dim, &bdt));
1862cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
1863cfb853baSMatthew G. Knepley     edg[d] = edges ? edges[d] : 1;
1864cfb853baSMatthew G. Knepley     low[d] = lower ? lower[d] : 0.;
1865cfb853baSMatthew G. Knepley     upp[d] = upper ? upper[d] : 1.;
1866cfb853baSMatthew G. Knepley     bdt[d] = DM_BOUNDARY_PERIODIC;
1867cfb853baSMatthew G. Knepley   }
1868cfb853baSMatthew G. Knepley   PetscCall(DMPlexCreateHypercubicMesh_Internal(*dm, dim, low, upp, edg, bdt));
1869cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(edg, low, upp, bdt));
18703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1871cfb853baSMatthew G. Knepley }
1872cfb853baSMatthew G. Knepley 
1873cc4c1da9SBarry Smith /*@
1874a1cb98faSBarry Smith   DMPlexSetOptionsPrefix - Sets the prefix used for searching for all `DM` options in the database.
1875a9074c1eSMatthew G. Knepley 
187620f4b53cSBarry Smith   Logically Collective
1877a9074c1eSMatthew G. Knepley 
1878a9074c1eSMatthew G. Knepley   Input Parameters:
187920f4b53cSBarry Smith + dm     - the `DM` context
1880a9074c1eSMatthew G. Knepley - prefix - the prefix to prepend to all option names
1881a9074c1eSMatthew G. Knepley 
1882a1cb98faSBarry Smith   Level: advanced
1883a1cb98faSBarry Smith 
1884a1cb98faSBarry Smith   Note:
1885a9074c1eSMatthew G. Knepley   A hyphen (-) must NOT be given at the beginning of the prefix name.
1886a9074c1eSMatthew G. Knepley   The first character of all runtime options is AUTOMATICALLY the hyphen.
1887a9074c1eSMatthew G. Knepley 
18881cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `SNESSetFromOptions()`
1889a9074c1eSMatthew G. Knepley @*/
1890d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexSetOptionsPrefix(DM dm, const char prefix[])
1891d71ae5a4SJacob Faibussowitsch {
1892a9074c1eSMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
1893a9074c1eSMatthew G. Knepley 
1894a9074c1eSMatthew G. Knepley   PetscFunctionBegin;
1895a9074c1eSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
18969566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, prefix));
18979566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)mesh->partitioner, prefix));
18983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1899a9074c1eSMatthew G. Knepley }
1900a9074c1eSMatthew G. Knepley 
19019318fe57SMatthew G. Knepley /* Remap geometry to cylinder
190261a622f3SMatthew G. Knepley    TODO: This only works for a single refinement, then it is broken
190361a622f3SMatthew G. Knepley 
19049318fe57SMatthew G. Knepley      Interior square: Linear interpolation is correct
19059318fe57SMatthew G. Knepley      The other cells all have vertices on rays from the origin. We want to uniformly expand the spacing
19069318fe57SMatthew G. Knepley      such that the last vertex is on the unit circle. So the closest and farthest vertices are at distance
19070510c589SMatthew G. Knepley 
19089318fe57SMatthew G. Knepley        phi     = arctan(y/x)
19099318fe57SMatthew G. Knepley        d_close = sqrt(1/8 + 1/4 sin^2(phi))
19109318fe57SMatthew G. Knepley        d_far   = sqrt(1/2 + sin^2(phi))
19110510c589SMatthew G. Knepley 
19129318fe57SMatthew G. Knepley      so we remap them using
19130510c589SMatthew G. Knepley 
19149318fe57SMatthew G. Knepley        x_new = x_close + (x - x_close) (1 - d_close) / (d_far - d_close)
19159318fe57SMatthew G. Knepley        y_new = y_close + (y - y_close) (1 - d_close) / (d_far - d_close)
19160510c589SMatthew G. Knepley 
19179318fe57SMatthew G. Knepley      If pi/4 < phi < 3pi/4 or -3pi/4 < phi < -pi/4, then we switch x and y.
19189318fe57SMatthew G. Knepley */
1919d71ae5a4SJacob Faibussowitsch static void snapToCylinder(PetscInt dim, PetscInt Nf, PetscInt NfAux, const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[])
1920d71ae5a4SJacob Faibussowitsch {
19219318fe57SMatthew G. Knepley   const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
19229318fe57SMatthew G. Knepley   const PetscReal ds2 = 0.5 * dis;
192322cc497dSMatthew G. Knepley 
19249318fe57SMatthew G. Knepley   if ((PetscAbsScalar(u[0]) <= ds2) && (PetscAbsScalar(u[1]) <= ds2)) {
19259318fe57SMatthew G. Knepley     f0[0] = u[0];
19269318fe57SMatthew G. Knepley     f0[1] = u[1];
19279318fe57SMatthew G. Knepley   } else {
19289318fe57SMatthew G. Knepley     PetscReal phi, sinp, cosp, dc, df, x, y, xc, yc;
19290510c589SMatthew G. Knepley 
19309318fe57SMatthew G. Knepley     x    = PetscRealPart(u[0]);
19319318fe57SMatthew G. Knepley     y    = PetscRealPart(u[1]);
19329318fe57SMatthew G. Knepley     phi  = PetscAtan2Real(y, x);
19339318fe57SMatthew G. Knepley     sinp = PetscSinReal(phi);
19349318fe57SMatthew G. Knepley     cosp = PetscCosReal(phi);
19359318fe57SMatthew G. Knepley     if ((PetscAbsReal(phi) > PETSC_PI / 4.0) && (PetscAbsReal(phi) < 3.0 * PETSC_PI / 4.0)) {
19369318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / sinp);
19379318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / sinp);
19389318fe57SMatthew G. Knepley       xc = ds2 * x / PetscAbsReal(y);
19399318fe57SMatthew G. Knepley       yc = ds2 * PetscSignReal(y);
19409318fe57SMatthew G. Knepley     } else {
19419318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / cosp);
19429318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / cosp);
19439318fe57SMatthew G. Knepley       xc = ds2 * PetscSignReal(x);
19449318fe57SMatthew G. Knepley       yc = ds2 * y / PetscAbsReal(x);
19459318fe57SMatthew G. Knepley     }
19469318fe57SMatthew G. Knepley     f0[0] = xc + (u[0] - xc) * (1.0 - dc) / (df - dc);
19479318fe57SMatthew G. Knepley     f0[1] = yc + (u[1] - yc) * (1.0 - dc) / (df - dc);
19489318fe57SMatthew G. Knepley   }
19499318fe57SMatthew G. Knepley   f0[2] = u[2];
19509318fe57SMatthew G. Knepley }
19510510c589SMatthew G. Knepley 
195249704ca5SMatthew G. Knepley static PetscErrorCode DMPlexCreateHexCylinderMesh_Internal(DM dm, DMBoundaryType periodicZ, PetscInt Nr)
1953d71ae5a4SJacob Faibussowitsch {
19540510c589SMatthew G. Knepley   const PetscInt dim = 3;
19559318fe57SMatthew G. Knepley   PetscInt       numCells, numVertices;
1956d8c47e87SMatthew G. Knepley   PetscMPIInt    rank;
19570510c589SMatthew G. Knepley 
19580510c589SMatthew G. Knepley   PetscFunctionBegin;
195946139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
19609566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
19619566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
19620510c589SMatthew G. Knepley   /* Create topology */
19630510c589SMatthew G. Knepley   {
19640510c589SMatthew G. Knepley     PetscInt cone[8], c;
19650510c589SMatthew G. Knepley 
1966dd400576SPatrick Sanan     numCells    = rank == 0 ? 5 : 0;
1967dd400576SPatrick Sanan     numVertices = rank == 0 ? 16 : 0;
1968006a8963SMatthew G. Knepley     if (periodicZ == DM_BOUNDARY_PERIODIC) {
1969ae8bcbbbSMatthew G. Knepley       numCells *= 3;
1970dd400576SPatrick Sanan       numVertices = rank == 0 ? 24 : 0;
1971006a8963SMatthew G. Knepley     }
19729566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
19739566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 8));
19749566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
1975dd400576SPatrick Sanan     if (rank == 0) {
1976006a8963SMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
19779371c9d4SSatish Balay         cone[0] = 15;
19789371c9d4SSatish Balay         cone[1] = 18;
19799371c9d4SSatish Balay         cone[2] = 17;
19809371c9d4SSatish Balay         cone[3] = 16;
19819371c9d4SSatish Balay         cone[4] = 31;
19829371c9d4SSatish Balay         cone[5] = 32;
19839371c9d4SSatish Balay         cone[6] = 33;
19849371c9d4SSatish Balay         cone[7] = 34;
19859566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
19869371c9d4SSatish Balay         cone[0] = 16;
19879371c9d4SSatish Balay         cone[1] = 17;
19889371c9d4SSatish Balay         cone[2] = 24;
19899371c9d4SSatish Balay         cone[3] = 23;
19909371c9d4SSatish Balay         cone[4] = 32;
19919371c9d4SSatish Balay         cone[5] = 36;
19929371c9d4SSatish Balay         cone[6] = 37;
19939371c9d4SSatish Balay         cone[7] = 33; /* 22 25 26 21 */
19949566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
19959371c9d4SSatish Balay         cone[0] = 18;
19969371c9d4SSatish Balay         cone[1] = 27;
19979371c9d4SSatish Balay         cone[2] = 24;
19989371c9d4SSatish Balay         cone[3] = 17;
19999371c9d4SSatish Balay         cone[4] = 34;
20009371c9d4SSatish Balay         cone[5] = 33;
20019371c9d4SSatish Balay         cone[6] = 37;
20029371c9d4SSatish Balay         cone[7] = 38;
20039566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
20049371c9d4SSatish Balay         cone[0] = 29;
20059371c9d4SSatish Balay         cone[1] = 27;
20069371c9d4SSatish Balay         cone[2] = 18;
20079371c9d4SSatish Balay         cone[3] = 15;
20089371c9d4SSatish Balay         cone[4] = 35;
20099371c9d4SSatish Balay         cone[5] = 31;
20109371c9d4SSatish Balay         cone[6] = 34;
20119371c9d4SSatish Balay         cone[7] = 38;
20129566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
20139371c9d4SSatish Balay         cone[0] = 29;
20149371c9d4SSatish Balay         cone[1] = 15;
20159371c9d4SSatish Balay         cone[2] = 16;
20169371c9d4SSatish Balay         cone[3] = 23;
20179371c9d4SSatish Balay         cone[4] = 35;
20189371c9d4SSatish Balay         cone[5] = 36;
20199371c9d4SSatish Balay         cone[6] = 32;
20209371c9d4SSatish Balay         cone[7] = 31;
20219566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2022006a8963SMatthew G. Knepley 
20239371c9d4SSatish Balay         cone[0] = 31;
20249371c9d4SSatish Balay         cone[1] = 34;
20259371c9d4SSatish Balay         cone[2] = 33;
20269371c9d4SSatish Balay         cone[3] = 32;
20279371c9d4SSatish Balay         cone[4] = 19;
20289371c9d4SSatish Balay         cone[5] = 22;
20299371c9d4SSatish Balay         cone[6] = 21;
20309371c9d4SSatish Balay         cone[7] = 20;
20319566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
20329371c9d4SSatish Balay         cone[0] = 32;
20339371c9d4SSatish Balay         cone[1] = 33;
20349371c9d4SSatish Balay         cone[2] = 37;
20359371c9d4SSatish Balay         cone[3] = 36;
20369371c9d4SSatish Balay         cone[4] = 22;
20379371c9d4SSatish Balay         cone[5] = 25;
20389371c9d4SSatish Balay         cone[6] = 26;
20399371c9d4SSatish Balay         cone[7] = 21;
20409566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 6, cone));
20419371c9d4SSatish Balay         cone[0] = 34;
20429371c9d4SSatish Balay         cone[1] = 38;
20439371c9d4SSatish Balay         cone[2] = 37;
20449371c9d4SSatish Balay         cone[3] = 33;
20459371c9d4SSatish Balay         cone[4] = 20;
20469371c9d4SSatish Balay         cone[5] = 21;
20479371c9d4SSatish Balay         cone[6] = 26;
20489371c9d4SSatish Balay         cone[7] = 28;
20499566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 7, cone));
20509371c9d4SSatish Balay         cone[0] = 35;
20519371c9d4SSatish Balay         cone[1] = 38;
20529371c9d4SSatish Balay         cone[2] = 34;
20539371c9d4SSatish Balay         cone[3] = 31;
20549371c9d4SSatish Balay         cone[4] = 30;
20559371c9d4SSatish Balay         cone[5] = 19;
20569371c9d4SSatish Balay         cone[6] = 20;
20579371c9d4SSatish Balay         cone[7] = 28;
20589566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 8, cone));
20599371c9d4SSatish Balay         cone[0] = 35;
20609371c9d4SSatish Balay         cone[1] = 31;
20619371c9d4SSatish Balay         cone[2] = 32;
20629371c9d4SSatish Balay         cone[3] = 36;
20639371c9d4SSatish Balay         cone[4] = 30;
20649371c9d4SSatish Balay         cone[5] = 25;
20659371c9d4SSatish Balay         cone[6] = 22;
20669371c9d4SSatish Balay         cone[7] = 19;
20679566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 9, cone));
2068ae8bcbbbSMatthew G. Knepley 
20699371c9d4SSatish Balay         cone[0] = 19;
20709371c9d4SSatish Balay         cone[1] = 20;
20719371c9d4SSatish Balay         cone[2] = 21;
20729371c9d4SSatish Balay         cone[3] = 22;
20739371c9d4SSatish Balay         cone[4] = 15;
20749371c9d4SSatish Balay         cone[5] = 16;
20759371c9d4SSatish Balay         cone[6] = 17;
20769371c9d4SSatish Balay         cone[7] = 18;
20779566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 10, cone));
20789371c9d4SSatish Balay         cone[0] = 22;
20799371c9d4SSatish Balay         cone[1] = 21;
20809371c9d4SSatish Balay         cone[2] = 26;
20819371c9d4SSatish Balay         cone[3] = 25;
20829371c9d4SSatish Balay         cone[4] = 16;
20839371c9d4SSatish Balay         cone[5] = 23;
20849371c9d4SSatish Balay         cone[6] = 24;
20859371c9d4SSatish Balay         cone[7] = 17;
20869566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 11, cone));
20879371c9d4SSatish Balay         cone[0] = 20;
20889371c9d4SSatish Balay         cone[1] = 28;
20899371c9d4SSatish Balay         cone[2] = 26;
20909371c9d4SSatish Balay         cone[3] = 21;
20919371c9d4SSatish Balay         cone[4] = 18;
20929371c9d4SSatish Balay         cone[5] = 17;
20939371c9d4SSatish Balay         cone[6] = 24;
20949371c9d4SSatish Balay         cone[7] = 27;
20959566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 12, cone));
20969371c9d4SSatish Balay         cone[0] = 30;
20979371c9d4SSatish Balay         cone[1] = 28;
20989371c9d4SSatish Balay         cone[2] = 20;
20999371c9d4SSatish Balay         cone[3] = 19;
21009371c9d4SSatish Balay         cone[4] = 29;
21019371c9d4SSatish Balay         cone[5] = 15;
21029371c9d4SSatish Balay         cone[6] = 18;
21039371c9d4SSatish Balay         cone[7] = 27;
21049566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 13, cone));
21059371c9d4SSatish Balay         cone[0] = 30;
21069371c9d4SSatish Balay         cone[1] = 19;
21079371c9d4SSatish Balay         cone[2] = 22;
21089371c9d4SSatish Balay         cone[3] = 25;
21099371c9d4SSatish Balay         cone[4] = 29;
21109371c9d4SSatish Balay         cone[5] = 23;
21119371c9d4SSatish Balay         cone[6] = 16;
21129371c9d4SSatish Balay         cone[7] = 15;
21139566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
2114006a8963SMatthew G. Knepley       } else {
21159371c9d4SSatish Balay         cone[0] = 5;
21169371c9d4SSatish Balay         cone[1] = 8;
21179371c9d4SSatish Balay         cone[2] = 7;
21189371c9d4SSatish Balay         cone[3] = 6;
21199371c9d4SSatish Balay         cone[4] = 9;
21209371c9d4SSatish Balay         cone[5] = 12;
21219371c9d4SSatish Balay         cone[6] = 11;
21229371c9d4SSatish Balay         cone[7] = 10;
21239566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
21249371c9d4SSatish Balay         cone[0] = 6;
21259371c9d4SSatish Balay         cone[1] = 7;
21269371c9d4SSatish Balay         cone[2] = 14;
21279371c9d4SSatish Balay         cone[3] = 13;
21289371c9d4SSatish Balay         cone[4] = 12;
21299371c9d4SSatish Balay         cone[5] = 15;
21309371c9d4SSatish Balay         cone[6] = 16;
21319371c9d4SSatish Balay         cone[7] = 11;
21329566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
21339371c9d4SSatish Balay         cone[0] = 8;
21349371c9d4SSatish Balay         cone[1] = 17;
21359371c9d4SSatish Balay         cone[2] = 14;
21369371c9d4SSatish Balay         cone[3] = 7;
21379371c9d4SSatish Balay         cone[4] = 10;
21389371c9d4SSatish Balay         cone[5] = 11;
21399371c9d4SSatish Balay         cone[6] = 16;
21409371c9d4SSatish Balay         cone[7] = 18;
21419566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
21429371c9d4SSatish Balay         cone[0] = 19;
21439371c9d4SSatish Balay         cone[1] = 17;
21449371c9d4SSatish Balay         cone[2] = 8;
21459371c9d4SSatish Balay         cone[3] = 5;
21469371c9d4SSatish Balay         cone[4] = 20;
21479371c9d4SSatish Balay         cone[5] = 9;
21489371c9d4SSatish Balay         cone[6] = 10;
21499371c9d4SSatish Balay         cone[7] = 18;
21509566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
21519371c9d4SSatish Balay         cone[0] = 19;
21529371c9d4SSatish Balay         cone[1] = 5;
21539371c9d4SSatish Balay         cone[2] = 6;
21549371c9d4SSatish Balay         cone[3] = 13;
21559371c9d4SSatish Balay         cone[4] = 20;
21569371c9d4SSatish Balay         cone[5] = 15;
21579371c9d4SSatish Balay         cone[6] = 12;
21589371c9d4SSatish Balay         cone[7] = 9;
21599566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2160006a8963SMatthew G. Knepley       }
2161d8c47e87SMatthew G. Knepley     }
21629566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
21639566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
21640510c589SMatthew G. Knepley   }
2165dbc1dc17SMatthew G. Knepley   /* Create cube geometry */
21660510c589SMatthew G. Knepley   {
21670510c589SMatthew G. Knepley     Vec             coordinates;
21680510c589SMatthew G. Knepley     PetscSection    coordSection;
21690510c589SMatthew G. Knepley     PetscScalar    *coords;
21700510c589SMatthew G. Knepley     PetscInt        coordSize, v;
21710510c589SMatthew G. Knepley     const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
21720510c589SMatthew G. Knepley     const PetscReal ds2 = dis / 2.0;
21730510c589SMatthew G. Knepley 
21740510c589SMatthew G. Knepley     /* Build coordinates */
21759566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
21769566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
21779566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
21789566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
21790510c589SMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
21809566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
21819566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
21820510c589SMatthew G. Knepley     }
21839566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
21849566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
21859566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
21869566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
21879566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
21889566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
21899566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
21909566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
2191dd400576SPatrick Sanan     if (rank == 0) {
21929371c9d4SSatish Balay       coords[0 * dim + 0]  = -ds2;
21939371c9d4SSatish Balay       coords[0 * dim + 1]  = -ds2;
21949371c9d4SSatish Balay       coords[0 * dim + 2]  = 0.0;
21959371c9d4SSatish Balay       coords[1 * dim + 0]  = ds2;
21969371c9d4SSatish Balay       coords[1 * dim + 1]  = -ds2;
21979371c9d4SSatish Balay       coords[1 * dim + 2]  = 0.0;
21989371c9d4SSatish Balay       coords[2 * dim + 0]  = ds2;
21999371c9d4SSatish Balay       coords[2 * dim + 1]  = ds2;
22009371c9d4SSatish Balay       coords[2 * dim + 2]  = 0.0;
22019371c9d4SSatish Balay       coords[3 * dim + 0]  = -ds2;
22029371c9d4SSatish Balay       coords[3 * dim + 1]  = ds2;
22039371c9d4SSatish Balay       coords[3 * dim + 2]  = 0.0;
22049371c9d4SSatish Balay       coords[4 * dim + 0]  = -ds2;
22059371c9d4SSatish Balay       coords[4 * dim + 1]  = -ds2;
22069371c9d4SSatish Balay       coords[4 * dim + 2]  = 1.0;
22079371c9d4SSatish Balay       coords[5 * dim + 0]  = -ds2;
22089371c9d4SSatish Balay       coords[5 * dim + 1]  = ds2;
22099371c9d4SSatish Balay       coords[5 * dim + 2]  = 1.0;
22109371c9d4SSatish Balay       coords[6 * dim + 0]  = ds2;
22119371c9d4SSatish Balay       coords[6 * dim + 1]  = ds2;
22129371c9d4SSatish Balay       coords[6 * dim + 2]  = 1.0;
22139371c9d4SSatish Balay       coords[7 * dim + 0]  = ds2;
22149371c9d4SSatish Balay       coords[7 * dim + 1]  = -ds2;
22159371c9d4SSatish Balay       coords[7 * dim + 2]  = 1.0;
22169371c9d4SSatish Balay       coords[8 * dim + 0]  = dis;
22179371c9d4SSatish Balay       coords[8 * dim + 1]  = -dis;
22189371c9d4SSatish Balay       coords[8 * dim + 2]  = 0.0;
22199371c9d4SSatish Balay       coords[9 * dim + 0]  = dis;
22209371c9d4SSatish Balay       coords[9 * dim + 1]  = dis;
22219371c9d4SSatish Balay       coords[9 * dim + 2]  = 0.0;
22229371c9d4SSatish Balay       coords[10 * dim + 0] = dis;
22239371c9d4SSatish Balay       coords[10 * dim + 1] = -dis;
22249371c9d4SSatish Balay       coords[10 * dim + 2] = 1.0;
22259371c9d4SSatish Balay       coords[11 * dim + 0] = dis;
22269371c9d4SSatish Balay       coords[11 * dim + 1] = dis;
22279371c9d4SSatish Balay       coords[11 * dim + 2] = 1.0;
22289371c9d4SSatish Balay       coords[12 * dim + 0] = -dis;
22299371c9d4SSatish Balay       coords[12 * dim + 1] = dis;
22309371c9d4SSatish Balay       coords[12 * dim + 2] = 0.0;
22319371c9d4SSatish Balay       coords[13 * dim + 0] = -dis;
22329371c9d4SSatish Balay       coords[13 * dim + 1] = dis;
22339371c9d4SSatish Balay       coords[13 * dim + 2] = 1.0;
22349371c9d4SSatish Balay       coords[14 * dim + 0] = -dis;
22359371c9d4SSatish Balay       coords[14 * dim + 1] = -dis;
22369371c9d4SSatish Balay       coords[14 * dim + 2] = 0.0;
22379371c9d4SSatish Balay       coords[15 * dim + 0] = -dis;
22389371c9d4SSatish Balay       coords[15 * dim + 1] = -dis;
22399371c9d4SSatish Balay       coords[15 * dim + 2] = 1.0;
2240ae8bcbbbSMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
22419371c9d4SSatish Balay         /* 15 31 19 */ coords[16 * dim + 0] = -ds2;
22429371c9d4SSatish Balay         coords[16 * dim + 1]                = -ds2;
22439371c9d4SSatish Balay         coords[16 * dim + 2]                = 0.5;
22449371c9d4SSatish Balay         /* 16 32 22 */ coords[17 * dim + 0] = ds2;
22459371c9d4SSatish Balay         coords[17 * dim + 1]                = -ds2;
22469371c9d4SSatish Balay         coords[17 * dim + 2]                = 0.5;
22479371c9d4SSatish Balay         /* 17 33 21 */ coords[18 * dim + 0] = ds2;
22489371c9d4SSatish Balay         coords[18 * dim + 1]                = ds2;
22499371c9d4SSatish Balay         coords[18 * dim + 2]                = 0.5;
22509371c9d4SSatish Balay         /* 18 34 20 */ coords[19 * dim + 0] = -ds2;
22519371c9d4SSatish Balay         coords[19 * dim + 1]                = ds2;
22529371c9d4SSatish Balay         coords[19 * dim + 2]                = 0.5;
22539371c9d4SSatish Balay         /* 29 35 30 */ coords[20 * dim + 0] = -dis;
22549371c9d4SSatish Balay         coords[20 * dim + 1]                = -dis;
22559371c9d4SSatish Balay         coords[20 * dim + 2]                = 0.5;
22569371c9d4SSatish Balay         /* 23 36 25 */ coords[21 * dim + 0] = dis;
22579371c9d4SSatish Balay         coords[21 * dim + 1]                = -dis;
22589371c9d4SSatish Balay         coords[21 * dim + 2]                = 0.5;
22599371c9d4SSatish Balay         /* 24 37 26 */ coords[22 * dim + 0] = dis;
22609371c9d4SSatish Balay         coords[22 * dim + 1]                = dis;
22619371c9d4SSatish Balay         coords[22 * dim + 2]                = 0.5;
22629371c9d4SSatish Balay         /* 27 38 28 */ coords[23 * dim + 0] = -dis;
22639371c9d4SSatish Balay         coords[23 * dim + 1]                = dis;
22649371c9d4SSatish Balay         coords[23 * dim + 2]                = 0.5;
2265ae8bcbbbSMatthew G. Knepley       }
2266d8c47e87SMatthew G. Knepley     }
22679566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
22689566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
22699566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
22700510c589SMatthew G. Knepley   }
2271006a8963SMatthew G. Knepley   /* Create periodicity */
2272006a8963SMatthew G. Knepley   if (periodicZ == DM_BOUNDARY_PERIODIC || periodicZ == DM_BOUNDARY_TWIST) {
22736858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
22746858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
2275006a8963SMatthew G. Knepley     PetscReal lower[3]   = {0.0, 0.0, 0.0};
2276ae8bcbbbSMatthew G. Knepley     PetscReal upper[3]   = {1.0, 1.0, 1.5};
22776858538eSMatthew G. Knepley     PetscInt  numZCells  = 3;
2278006a8963SMatthew G. Knepley 
22796858538eSMatthew G. Knepley     L[2]       = upper[2] - lower[2];
22806858538eSMatthew G. Knepley     maxCell[2] = 1.1 * (L[2] / numZCells);
22814fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
2282006a8963SMatthew G. Knepley   }
2283dbc1dc17SMatthew G. Knepley   {
22849318fe57SMatthew G. Knepley     DM          cdm;
22859318fe57SMatthew G. Knepley     PetscDS     cds;
22869318fe57SMatthew G. Knepley     PetscScalar c[2] = {1.0, 1.0};
2287dbc1dc17SMatthew G. Knepley 
228849704ca5SMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
22899566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
22909566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
22919566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 2, c));
2292dbc1dc17SMatthew G. Knepley   }
229346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
229446139095SJed Brown 
22959318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
22969566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(dm));
229749704ca5SMatthew G. Knepley 
229849704ca5SMatthew G. Knepley   char        oldprefix[PETSC_MAX_PATH_LEN];
229949704ca5SMatthew G. Knepley   const char *prefix;
230049704ca5SMatthew G. Knepley 
230149704ca5SMatthew G. Knepley   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
230249704ca5SMatthew G. Knepley   PetscCall(PetscStrncpy(oldprefix, prefix, PETSC_MAX_PATH_LEN));
230349704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, "petsc_cyl_ref_"));
230449704ca5SMatthew G. Knepley   for (PetscInt r = 0; r < PetscMax(0, Nr); ++r) {
230549704ca5SMatthew G. Knepley     DM rdm;
230649704ca5SMatthew G. Knepley 
230749704ca5SMatthew G. Knepley     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
230849704ca5SMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &rdm));
230949704ca5SMatthew G. Knepley   }
231049704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldprefix));
231149704ca5SMatthew G. Knepley   PetscCall(DMPlexRemapGeometry(dm, 0.0, snapToCylinder));
231249704ca5SMatthew G. Knepley 
231349704ca5SMatthew G. Knepley   DMLabel         bdlabel, edgelabel;
231449704ca5SMatthew G. Knepley   IS              faceIS;
231549704ca5SMatthew G. Knepley   const PetscInt *faces;
231649704ca5SMatthew G. Knepley   PetscInt        Nf;
231749704ca5SMatthew G. Knepley 
231849704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
231949704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
232049704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "generatrix"));
232149704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "generatrix", &edgelabel));
232249704ca5SMatthew G. Knepley   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
232349704ca5SMatthew G. Knepley   // Remove faces on top and bottom
232449704ca5SMatthew G. Knepley   PetscCall(DMLabelGetStratumIS(bdlabel, 1, &faceIS));
2325ba1b3593SJeremy L Thompson   if (faceIS) {
232649704ca5SMatthew G. Knepley     PetscCall(ISGetLocalSize(faceIS, &Nf));
232749704ca5SMatthew G. Knepley     PetscCall(ISGetIndices(faceIS, &faces));
232849704ca5SMatthew G. Knepley     for (PetscInt f = 0; f < Nf; ++f) {
232949704ca5SMatthew G. Knepley       PetscReal vol, normal[3];
233049704ca5SMatthew G. Knepley 
233149704ca5SMatthew G. Knepley       PetscCall(DMPlexComputeCellGeometryFVM(dm, faces[f], &vol, NULL, normal));
233249704ca5SMatthew G. Knepley       if (PetscAbsReal(normal[2]) < PETSC_SMALL) PetscCall(DMLabelSetValue(edgelabel, faces[f], 1));
233349704ca5SMatthew G. Knepley     }
233449704ca5SMatthew G. Knepley     PetscCall(ISRestoreIndices(faceIS, &faces));
233549704ca5SMatthew G. Knepley     PetscCall(ISDestroy(&faceIS));
2336ba1b3593SJeremy L Thompson   }
233749704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, bdlabel));
233849704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, edgelabel));
23393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23400510c589SMatthew G. Knepley }
23410510c589SMatthew G. Knepley 
234224119c2aSMatthew G. Knepley /*@
23439318fe57SMatthew G. Knepley   DMPlexCreateHexCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using hexahedra.
234424119c2aSMatthew G. Knepley 
2345d083f849SBarry Smith   Collective
234624119c2aSMatthew G. Knepley 
234724119c2aSMatthew G. Knepley   Input Parameters:
2348a1cb98faSBarry Smith + comm      - The communicator for the `DM` object
234949704ca5SMatthew G. Knepley . periodicZ - The boundary type for the Z direction
235049704ca5SMatthew G. Knepley - Nr        - The number of refinements to carry out
235124119c2aSMatthew G. Knepley 
235224119c2aSMatthew G. Knepley   Output Parameter:
235320f4b53cSBarry Smith . dm - The `DM` object
235424119c2aSMatthew G. Knepley 
235524119c2aSMatthew G. Knepley   Level: beginner
235624119c2aSMatthew G. Knepley 
2357a1cb98faSBarry Smith   Note:
2358a4e35b19SJacob Faibussowitsch   Here is the output numbering looking from the bottom of the cylinder\:
2359a1cb98faSBarry Smith .vb
2360a1cb98faSBarry Smith        17-----14
2361a1cb98faSBarry Smith         |     |
2362a1cb98faSBarry Smith         |  2  |
2363a1cb98faSBarry Smith         |     |
2364a1cb98faSBarry Smith  17-----8-----7-----14
2365a1cb98faSBarry Smith   |     |     |     |
2366a1cb98faSBarry Smith   |  3  |  0  |  1  |
2367a1cb98faSBarry Smith   |     |     |     |
2368a1cb98faSBarry Smith  19-----5-----6-----13
2369a1cb98faSBarry Smith         |     |
2370a1cb98faSBarry Smith         |  4  |
2371a1cb98faSBarry Smith         |     |
2372a1cb98faSBarry Smith        19-----13
2373a1cb98faSBarry Smith 
2374a1cb98faSBarry Smith  and up through the top
2375a1cb98faSBarry Smith 
2376a1cb98faSBarry Smith        18-----16
2377a1cb98faSBarry Smith         |     |
2378a1cb98faSBarry Smith         |  2  |
2379a1cb98faSBarry Smith         |     |
2380a1cb98faSBarry Smith  18----10----11-----16
2381a1cb98faSBarry Smith   |     |     |     |
2382a1cb98faSBarry Smith   |  3  |  0  |  1  |
2383a1cb98faSBarry Smith   |     |     |     |
2384a1cb98faSBarry Smith  20-----9----12-----15
2385a1cb98faSBarry Smith         |     |
2386a1cb98faSBarry Smith         |  4  |
2387a1cb98faSBarry Smith         |     |
2388a1cb98faSBarry Smith        20-----15
2389a1cb98faSBarry Smith .ve
2390a1cb98faSBarry Smith 
23911cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
239224119c2aSMatthew G. Knepley @*/
239349704ca5SMatthew G. Knepley PetscErrorCode DMPlexCreateHexCylinderMesh(MPI_Comm comm, DMBoundaryType periodicZ, PetscInt Nr, DM *dm)
2394d71ae5a4SJacob Faibussowitsch {
23959318fe57SMatthew G. Knepley   PetscFunctionBegin;
239649704ca5SMatthew G. Knepley   PetscAssertPointer(dm, 4);
23979566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
23989566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
239949704ca5SMatthew G. Knepley   PetscCall(DMPlexCreateHexCylinderMesh_Internal(*dm, periodicZ, Nr));
24003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
24019318fe57SMatthew G. Knepley }
24029318fe57SMatthew G. Knepley 
2403d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeCylinderMesh_Internal(DM dm, PetscInt n, PetscBool interpolate)
2404d71ae5a4SJacob Faibussowitsch {
240524119c2aSMatthew G. Knepley   const PetscInt dim = 3;
2406412e9a14SMatthew G. Knepley   PetscInt       numCells, numVertices, v;
24079fe9f049SMatthew G. Knepley   PetscMPIInt    rank;
240824119c2aSMatthew G. Knepley 
240924119c2aSMatthew G. Knepley   PetscFunctionBegin;
241063a3b9bcSJacob Faibussowitsch   PetscCheck(n >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Number of wedges %" PetscInt_FMT " cannot be negative", n);
241146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
24129566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
24139566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
2414412e9a14SMatthew G. Knepley   /* Must create the celltype label here so that we do not automatically try to compute the types */
24159566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "celltype"));
241624119c2aSMatthew G. Knepley   /* Create topology */
241724119c2aSMatthew G. Knepley   {
241824119c2aSMatthew G. Knepley     PetscInt cone[6], c;
241924119c2aSMatthew G. Knepley 
2420dd400576SPatrick Sanan     numCells    = rank == 0 ? n : 0;
2421dd400576SPatrick Sanan     numVertices = rank == 0 ? 2 * (n + 1) : 0;
24229566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
24239566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
24249566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
242524119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
24269371c9d4SSatish Balay       cone[0] = c + n * 1;
24279371c9d4SSatish Balay       cone[1] = (c + 1) % n + n * 1;
24289371c9d4SSatish Balay       cone[2] = 0 + 3 * n;
24299371c9d4SSatish Balay       cone[3] = c + n * 2;
24309371c9d4SSatish Balay       cone[4] = (c + 1) % n + n * 2;
24319371c9d4SSatish Balay       cone[5] = 1 + 3 * n;
24329566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(dm, c, cone));
24339566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCellType(dm, c, DM_POLYTOPE_TRI_PRISM_TENSOR));
243424119c2aSMatthew G. Knepley     }
24359566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
24369566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
243724119c2aSMatthew G. Knepley   }
243848a46eb9SPierre Jolivet   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(dm, v, DM_POLYTOPE_POINT));
243924119c2aSMatthew G. Knepley   /* Create cylinder geometry */
244024119c2aSMatthew G. Knepley   {
244124119c2aSMatthew G. Knepley     Vec          coordinates;
244224119c2aSMatthew G. Knepley     PetscSection coordSection;
244324119c2aSMatthew G. Knepley     PetscScalar *coords;
2444412e9a14SMatthew G. Knepley     PetscInt     coordSize, c;
244524119c2aSMatthew G. Knepley 
244624119c2aSMatthew G. Knepley     /* Build coordinates */
24479566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
24489566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
24499566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
24509566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
245124119c2aSMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
24529566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
24539566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
245424119c2aSMatthew G. Knepley     }
24559566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
24569566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
24579566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
24589566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
24599566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
24609566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
24619566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
24629566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
246324119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
24649371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
24659371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
24669371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 2] = 1.0;
24679371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
24689371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
24699371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 2] = 0.0;
247024119c2aSMatthew G. Knepley     }
2471dd400576SPatrick Sanan     if (rank == 0) {
24729371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 0] = 0.0;
24739371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 1] = 0.0;
24749371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 2] = 1.0;
24759371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 0] = 0.0;
24769371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 1] = 0.0;
24779371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 2] = 0.0;
24789fe9f049SMatthew G. Knepley     }
24799566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
24809566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
24819566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
248224119c2aSMatthew G. Knepley   }
248346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
24849318fe57SMatthew G. Knepley   /* Interpolate */
24859566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
24863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
24879318fe57SMatthew G. Knepley }
24889318fe57SMatthew G. Knepley 
24899318fe57SMatthew G. Knepley /*@
24909318fe57SMatthew G. Knepley   DMPlexCreateWedgeCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using wedges.
24919318fe57SMatthew G. Knepley 
24929318fe57SMatthew G. Knepley   Collective
24939318fe57SMatthew G. Knepley 
24949318fe57SMatthew G. Knepley   Input Parameters:
2495a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
24969318fe57SMatthew G. Knepley . n           - The number of wedges around the origin
24979318fe57SMatthew G. Knepley - interpolate - Create edges and faces
24989318fe57SMatthew G. Knepley 
24999318fe57SMatthew G. Knepley   Output Parameter:
2500a1cb98faSBarry Smith . dm - The `DM` object
25019318fe57SMatthew G. Knepley 
25029318fe57SMatthew G. Knepley   Level: beginner
25039318fe57SMatthew G. Knepley 
25041cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
25059318fe57SMatthew G. Knepley @*/
2506d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeCylinderMesh(MPI_Comm comm, PetscInt n, PetscBool interpolate, DM *dm)
2507d71ae5a4SJacob Faibussowitsch {
25089318fe57SMatthew G. Knepley   PetscFunctionBegin;
25094f572ea9SToby Isaac   PetscAssertPointer(dm, 4);
25109566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
25119566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
25129566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(*dm, n, interpolate));
25133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
251424119c2aSMatthew G. Knepley }
251524119c2aSMatthew G. Knepley 
2516d71ae5a4SJacob Faibussowitsch static inline PetscReal DiffNormReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2517d71ae5a4SJacob Faibussowitsch {
251865a81367SMatthew G. Knepley   PetscReal prod = 0.0;
251965a81367SMatthew G. Knepley   PetscInt  i;
252065a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += PetscSqr(x[i] - y[i]);
252165a81367SMatthew G. Knepley   return PetscSqrtReal(prod);
252265a81367SMatthew G. Knepley }
2523dd2b43ebSStefano Zampini 
2524d71ae5a4SJacob Faibussowitsch static inline PetscReal DotReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2525d71ae5a4SJacob Faibussowitsch {
252665a81367SMatthew G. Knepley   PetscReal prod = 0.0;
252765a81367SMatthew G. Knepley   PetscInt  i;
252865a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += x[i] * y[i];
252965a81367SMatthew G. Knepley   return prod;
253065a81367SMatthew G. Knepley }
253165a81367SMatthew G. Knepley 
253251a74b61SMatthew G. Knepley /* The first constant is the sphere radius */
2533d71ae5a4SJacob Faibussowitsch static void snapToSphere(PetscInt dim, PetscInt Nf, PetscInt NfAux, const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[])
2534d71ae5a4SJacob Faibussowitsch {
253551a74b61SMatthew G. Knepley   PetscReal r     = PetscRealPart(constants[0]);
253651a74b61SMatthew G. Knepley   PetscReal norm2 = 0.0, fac;
253751a74b61SMatthew G. Knepley   PetscInt  n     = uOff[1] - uOff[0], d;
253851a74b61SMatthew G. Knepley 
253951a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) norm2 += PetscSqr(PetscRealPart(u[d]));
254051a74b61SMatthew G. Knepley   fac = r / PetscSqrtReal(norm2);
254151a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) f0[d] = u[d] * fac;
254251a74b61SMatthew G. Knepley }
254351a74b61SMatthew G. Knepley 
2544d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateSphereMesh_Internal(DM dm, PetscInt dim, PetscBool simplex, PetscReal R)
2545d71ae5a4SJacob Faibussowitsch {
254665a81367SMatthew G. Knepley   const PetscInt embedDim = dim + 1;
254765a81367SMatthew G. Knepley   PetscSection   coordSection;
254865a81367SMatthew G. Knepley   Vec            coordinates;
254965a81367SMatthew G. Knepley   PetscScalar   *coords;
255065a81367SMatthew G. Knepley   PetscReal     *coordsIn;
255107c565c5SJose E. Roman   PetscInt       numCells, numEdges, numVerts = 0, firstVertex = 0, v, firstEdge, coordSize, d, e;
255265a81367SMatthew G. Knepley   PetscMPIInt    rank;
255365a81367SMatthew G. Knepley 
255465a81367SMatthew G. Knepley   PetscFunctionBegin;
25559318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveBool(dm, simplex, 3);
255646139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
25579566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
25589566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim + 1));
25599566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
256065a81367SMatthew G. Knepley   switch (dim) {
25615c344501SMatthew G. Knepley   case 1:
25625c344501SMatthew G. Knepley     numCells = 16;
25635c344501SMatthew G. Knepley     numVerts = numCells;
25645c344501SMatthew G. Knepley 
25655c344501SMatthew G. Knepley     // Build Topology
25665c344501SMatthew G. Knepley     PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
25675c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
25685c344501SMatthew G. Knepley     PetscCall(DMSetUp(dm));
25695c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; ++c) {
25705c344501SMatthew G. Knepley       PetscInt cone[2];
25715c344501SMatthew G. Knepley 
25725c344501SMatthew G. Knepley       cone[0] = c + numCells;
25735c344501SMatthew G. Knepley       cone[1] = (c + 1) % numVerts + numCells;
25745c344501SMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, c, cone));
25755c344501SMatthew G. Knepley     }
25765c344501SMatthew G. Knepley     PetscCall(DMPlexSymmetrize(dm));
25775c344501SMatthew G. Knepley     PetscCall(DMPlexStratify(dm));
25785c344501SMatthew G. Knepley     PetscCall(PetscMalloc1(numVerts * embedDim, &coordsIn));
25795c344501SMatthew G. Knepley     for (PetscInt v = 0; v < numVerts; ++v) {
25805c344501SMatthew G. Knepley       const PetscReal rad = 2. * PETSC_PI * v / numVerts;
25815c344501SMatthew G. Knepley 
25825c344501SMatthew G. Knepley       coordsIn[v * embedDim + 0] = PetscCosReal(rad);
25835c344501SMatthew G. Knepley       coordsIn[v * embedDim + 1] = PetscSinReal(rad);
25845c344501SMatthew G. Knepley     }
25855c344501SMatthew G. Knepley     break;
258665a81367SMatthew G. Knepley   case 2:
258765a81367SMatthew G. Knepley     if (simplex) {
258851a74b61SMatthew G. Knepley       const PetscReal radius    = PetscSqrtReal(1 + PETSC_PHI * PETSC_PHI) / (1.0 + PETSC_PHI);
258951a74b61SMatthew G. Knepley       const PetscReal edgeLen   = 2.0 / (1.0 + PETSC_PHI) * (R / radius);
259065a81367SMatthew G. Knepley       const PetscInt  degree    = 5;
259151a74b61SMatthew G. Knepley       PetscReal       vertex[3] = {0.0, 1.0 / (1.0 + PETSC_PHI), PETSC_PHI / (1.0 + PETSC_PHI)};
259265a81367SMatthew G. Knepley       PetscInt        s[3]      = {1, 1, 1};
259365a81367SMatthew G. Knepley       PetscInt        cone[3];
259407c565c5SJose E. Roman       PetscInt       *graph;
259565a81367SMatthew G. Knepley 
25969371c9d4SSatish Balay       vertex[0] *= R / radius;
25979371c9d4SSatish Balay       vertex[1] *= R / radius;
25989371c9d4SSatish Balay       vertex[2] *= R / radius;
2599dd400576SPatrick Sanan       numCells    = rank == 0 ? 20 : 0;
2600dd400576SPatrick Sanan       numVerts    = rank == 0 ? 12 : 0;
260165a81367SMatthew G. Knepley       firstVertex = numCells;
260251a74b61SMatthew G. Knepley       /* Use icosahedron, which for a R-sphere has coordinates which are all cyclic permutations of
260365a81367SMatthew G. Knepley 
260465a81367SMatthew G. Knepley            (0, \pm 1/\phi+1, \pm \phi/\phi+1)
260565a81367SMatthew G. Knepley 
260665a81367SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
260751a74b61SMatthew G. Knepley          length is then given by 2/(1+\phi) = 2 * 0.38197 = 0.76393.
260865a81367SMatthew G. Knepley       */
260965a81367SMatthew G. Knepley       /* Construct vertices */
26109566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2611dd400576SPatrick Sanan       if (rank == 0) {
261207c565c5SJose E. Roman         for (PetscInt p = 0, i = 0; p < embedDim; ++p) {
261365a81367SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
261465a81367SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
261565a81367SMatthew G. Knepley               for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertex[(d + p) % embedDim];
261665a81367SMatthew G. Knepley               ++i;
261765a81367SMatthew G. Knepley             }
261865a81367SMatthew G. Knepley           }
261965a81367SMatthew G. Knepley         }
262045da822fSValeria Barra       }
262165a81367SMatthew G. Knepley       /* Construct graph */
26229566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
262307c565c5SJose E. Roman       for (PetscInt i = 0; i < numVerts; ++i) {
262407c565c5SJose E. Roman         PetscInt k = 0;
262507c565c5SJose E. Roman         for (PetscInt j = 0; j < numVerts; ++j) {
26269371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
26279371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
26289371c9d4SSatish Balay             ++k;
26299371c9d4SSatish Balay           }
263065a81367SMatthew G. Knepley         }
263163a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid icosahedron, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
263265a81367SMatthew G. Knepley       }
263365a81367SMatthew G. Knepley       /* Build Topology */
26349566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
263507c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
26369566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
263765a81367SMatthew G. Knepley       /* Cells */
263807c565c5SJose E. Roman       for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
263907c565c5SJose E. Roman         for (PetscInt j = 0; j < i; ++j) {
264007c565c5SJose E. Roman           for (PetscInt k = 0; k < j; ++k) {
264165a81367SMatthew G. Knepley             if (graph[i * numVerts + j] && graph[j * numVerts + k] && graph[k * numVerts + i]) {
26429371c9d4SSatish Balay               cone[0] = firstVertex + i;
26439371c9d4SSatish Balay               cone[1] = firstVertex + j;
26449371c9d4SSatish Balay               cone[2] = firstVertex + k;
264565a81367SMatthew G. Knepley               /* Check orientation */
264665a81367SMatthew G. Knepley               {
26479371c9d4SSatish Balay                 const PetscInt epsilon[3][3][3] = {
26489371c9d4SSatish Balay                   {{0, 0, 0},  {0, 0, 1},  {0, -1, 0}},
26499371c9d4SSatish Balay                   {{0, 0, -1}, {0, 0, 0},  {1, 0, 0} },
26509371c9d4SSatish Balay                   {{0, 1, 0},  {-1, 0, 0}, {0, 0, 0} }
26519371c9d4SSatish Balay                 };
265265a81367SMatthew G. Knepley                 PetscReal normal[3];
265365a81367SMatthew G. Knepley                 PetscInt  e, f;
265465a81367SMatthew G. Knepley 
265565a81367SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) {
265665a81367SMatthew G. Knepley                   normal[d] = 0.0;
265765a81367SMatthew G. Knepley                   for (e = 0; e < embedDim; ++e) {
2658ad540459SPierre Jolivet                     for (f = 0; f < embedDim; ++f) normal[d] += epsilon[d][e][f] * (coordsIn[j * embedDim + e] - coordsIn[i * embedDim + e]) * (coordsIn[k * embedDim + f] - coordsIn[i * embedDim + f]);
265965a81367SMatthew G. Knepley                   }
266065a81367SMatthew G. Knepley                 }
26619371c9d4SSatish Balay                 if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
26629371c9d4SSatish Balay                   PetscInt tmp = cone[1];
26639371c9d4SSatish Balay                   cone[1]      = cone[2];
26649371c9d4SSatish Balay                   cone[2]      = tmp;
266565a81367SMatthew G. Knepley                 }
266665a81367SMatthew G. Knepley               }
26679566063dSJacob Faibussowitsch               PetscCall(DMPlexSetCone(dm, c++, cone));
266865a81367SMatthew G. Knepley             }
266965a81367SMatthew G. Knepley           }
267065a81367SMatthew G. Knepley         }
267165a81367SMatthew G. Knepley       }
26729566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
26739566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
26749566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
267565a81367SMatthew G. Knepley     } else {
26762829fed8SMatthew G. Knepley       /*
26772829fed8SMatthew G. Knepley         12-21--13
26782829fed8SMatthew G. Knepley          |     |
26792829fed8SMatthew G. Knepley         25  4  24
26802829fed8SMatthew G. Knepley          |     |
26812829fed8SMatthew G. Knepley   12-25--9-16--8-24--13
26822829fed8SMatthew G. Knepley    |     |     |     |
26832829fed8SMatthew G. Knepley   23  5 17  0 15  3  22
26842829fed8SMatthew G. Knepley    |     |     |     |
26852829fed8SMatthew G. Knepley   10-20--6-14--7-19--11
26862829fed8SMatthew G. Knepley          |     |
26872829fed8SMatthew G. Knepley         20  1  19
26882829fed8SMatthew G. Knepley          |     |
26892829fed8SMatthew G. Knepley         10-18--11
26902829fed8SMatthew G. Knepley          |     |
26912829fed8SMatthew G. Knepley         23  2  22
26922829fed8SMatthew G. Knepley          |     |
26932829fed8SMatthew G. Knepley         12-21--13
26942829fed8SMatthew G. Knepley        */
26952829fed8SMatthew G. Knepley       PetscInt cone[4], ornt[4];
26962829fed8SMatthew G. Knepley 
2697dd400576SPatrick Sanan       numCells    = rank == 0 ? 6 : 0;
2698dd400576SPatrick Sanan       numEdges    = rank == 0 ? 12 : 0;
2699dd400576SPatrick Sanan       numVerts    = rank == 0 ? 8 : 0;
270065a81367SMatthew G. Knepley       firstVertex = numCells;
270165a81367SMatthew G. Knepley       firstEdge   = numCells + numVerts;
27022829fed8SMatthew G. Knepley       /* Build Topology */
27039566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numEdges + numVerts));
270407c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 4));
270548a46eb9SPierre Jolivet       for (e = firstEdge; e < firstEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
27069566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
2707dd400576SPatrick Sanan       if (rank == 0) {
27082829fed8SMatthew G. Knepley         /* Cell 0 */
27099371c9d4SSatish Balay         cone[0] = 14;
27109371c9d4SSatish Balay         cone[1] = 15;
27119371c9d4SSatish Balay         cone[2] = 16;
27129371c9d4SSatish Balay         cone[3] = 17;
27139566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
27149371c9d4SSatish Balay         ornt[0] = 0;
27159371c9d4SSatish Balay         ornt[1] = 0;
27169371c9d4SSatish Balay         ornt[2] = 0;
27179371c9d4SSatish Balay         ornt[3] = 0;
27189566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 0, ornt));
27192829fed8SMatthew G. Knepley         /* Cell 1 */
27209371c9d4SSatish Balay         cone[0] = 18;
27219371c9d4SSatish Balay         cone[1] = 19;
27229371c9d4SSatish Balay         cone[2] = 14;
27239371c9d4SSatish Balay         cone[3] = 20;
27249566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
27259371c9d4SSatish Balay         ornt[0] = 0;
27269371c9d4SSatish Balay         ornt[1] = 0;
27279371c9d4SSatish Balay         ornt[2] = -1;
27289371c9d4SSatish Balay         ornt[3] = 0;
27299566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 1, ornt));
27302829fed8SMatthew G. Knepley         /* Cell 2 */
27319371c9d4SSatish Balay         cone[0] = 21;
27329371c9d4SSatish Balay         cone[1] = 22;
27339371c9d4SSatish Balay         cone[2] = 18;
27349371c9d4SSatish Balay         cone[3] = 23;
27359566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
27369371c9d4SSatish Balay         ornt[0] = 0;
27379371c9d4SSatish Balay         ornt[1] = 0;
27389371c9d4SSatish Balay         ornt[2] = -1;
27399371c9d4SSatish Balay         ornt[3] = 0;
27409566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 2, ornt));
27412829fed8SMatthew G. Knepley         /* Cell 3 */
27429371c9d4SSatish Balay         cone[0] = 19;
27439371c9d4SSatish Balay         cone[1] = 22;
27449371c9d4SSatish Balay         cone[2] = 24;
27459371c9d4SSatish Balay         cone[3] = 15;
27469566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
27479371c9d4SSatish Balay         ornt[0] = -1;
27489371c9d4SSatish Balay         ornt[1] = -1;
27499371c9d4SSatish Balay         ornt[2] = 0;
27509371c9d4SSatish Balay         ornt[3] = -1;
27519566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 3, ornt));
27522829fed8SMatthew G. Knepley         /* Cell 4 */
27539371c9d4SSatish Balay         cone[0] = 16;
27549371c9d4SSatish Balay         cone[1] = 24;
27559371c9d4SSatish Balay         cone[2] = 21;
27569371c9d4SSatish Balay         cone[3] = 25;
27579566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
27589371c9d4SSatish Balay         ornt[0] = -1;
27599371c9d4SSatish Balay         ornt[1] = -1;
27609371c9d4SSatish Balay         ornt[2] = -1;
27619371c9d4SSatish Balay         ornt[3] = 0;
27629566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 4, ornt));
27632829fed8SMatthew G. Knepley         /* Cell 5 */
27649371c9d4SSatish Balay         cone[0] = 20;
27659371c9d4SSatish Balay         cone[1] = 17;
27669371c9d4SSatish Balay         cone[2] = 25;
27679371c9d4SSatish Balay         cone[3] = 23;
27689566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
27699371c9d4SSatish Balay         ornt[0] = -1;
27709371c9d4SSatish Balay         ornt[1] = -1;
27719371c9d4SSatish Balay         ornt[2] = -1;
27729371c9d4SSatish Balay         ornt[3] = -1;
27739566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 5, ornt));
27742829fed8SMatthew G. Knepley         /* Edges */
27759371c9d4SSatish Balay         cone[0] = 6;
27769371c9d4SSatish Balay         cone[1] = 7;
27779566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
27789371c9d4SSatish Balay         cone[0] = 7;
27799371c9d4SSatish Balay         cone[1] = 8;
27809566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 15, cone));
27819371c9d4SSatish Balay         cone[0] = 8;
27829371c9d4SSatish Balay         cone[1] = 9;
27839566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 16, cone));
27849371c9d4SSatish Balay         cone[0] = 9;
27859371c9d4SSatish Balay         cone[1] = 6;
27869566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 17, cone));
27879371c9d4SSatish Balay         cone[0] = 10;
27889371c9d4SSatish Balay         cone[1] = 11;
27899566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 18, cone));
27909371c9d4SSatish Balay         cone[0] = 11;
27919371c9d4SSatish Balay         cone[1] = 7;
27929566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 19, cone));
27939371c9d4SSatish Balay         cone[0] = 6;
27949371c9d4SSatish Balay         cone[1] = 10;
27959566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 20, cone));
27969371c9d4SSatish Balay         cone[0] = 12;
27979371c9d4SSatish Balay         cone[1] = 13;
27989566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 21, cone));
27999371c9d4SSatish Balay         cone[0] = 13;
28009371c9d4SSatish Balay         cone[1] = 11;
28019566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 22, cone));
28029371c9d4SSatish Balay         cone[0] = 10;
28039371c9d4SSatish Balay         cone[1] = 12;
28049566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 23, cone));
28059371c9d4SSatish Balay         cone[0] = 13;
28069371c9d4SSatish Balay         cone[1] = 8;
28079566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 24, cone));
28089371c9d4SSatish Balay         cone[0] = 12;
28099371c9d4SSatish Balay         cone[1] = 9;
28109566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 25, cone));
281145da822fSValeria Barra       }
28129566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
28139566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
28142829fed8SMatthew G. Knepley       /* Build coordinates */
28159566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2816dd400576SPatrick Sanan       if (rank == 0) {
28179371c9d4SSatish Balay         coordsIn[0 * embedDim + 0] = -R;
28189371c9d4SSatish Balay         coordsIn[0 * embedDim + 1] = R;
28199371c9d4SSatish Balay         coordsIn[0 * embedDim + 2] = -R;
28209371c9d4SSatish Balay         coordsIn[1 * embedDim + 0] = R;
28219371c9d4SSatish Balay         coordsIn[1 * embedDim + 1] = R;
28229371c9d4SSatish Balay         coordsIn[1 * embedDim + 2] = -R;
28239371c9d4SSatish Balay         coordsIn[2 * embedDim + 0] = R;
28249371c9d4SSatish Balay         coordsIn[2 * embedDim + 1] = -R;
28259371c9d4SSatish Balay         coordsIn[2 * embedDim + 2] = -R;
28269371c9d4SSatish Balay         coordsIn[3 * embedDim + 0] = -R;
28279371c9d4SSatish Balay         coordsIn[3 * embedDim + 1] = -R;
28289371c9d4SSatish Balay         coordsIn[3 * embedDim + 2] = -R;
28299371c9d4SSatish Balay         coordsIn[4 * embedDim + 0] = -R;
28309371c9d4SSatish Balay         coordsIn[4 * embedDim + 1] = R;
28319371c9d4SSatish Balay         coordsIn[4 * embedDim + 2] = R;
28329371c9d4SSatish Balay         coordsIn[5 * embedDim + 0] = R;
28339371c9d4SSatish Balay         coordsIn[5 * embedDim + 1] = R;
28349371c9d4SSatish Balay         coordsIn[5 * embedDim + 2] = R;
28359371c9d4SSatish Balay         coordsIn[6 * embedDim + 0] = -R;
28369371c9d4SSatish Balay         coordsIn[6 * embedDim + 1] = -R;
28379371c9d4SSatish Balay         coordsIn[6 * embedDim + 2] = R;
28389371c9d4SSatish Balay         coordsIn[7 * embedDim + 0] = R;
28399371c9d4SSatish Balay         coordsIn[7 * embedDim + 1] = -R;
28409371c9d4SSatish Balay         coordsIn[7 * embedDim + 2] = R;
284165a81367SMatthew G. Knepley       }
284245da822fSValeria Barra     }
284365a81367SMatthew G. Knepley     break;
284465a81367SMatthew G. Knepley   case 3:
2845116ded15SMatthew G. Knepley     if (simplex) {
2846116ded15SMatthew G. Knepley       const PetscReal edgeLen         = 1.0 / PETSC_PHI;
284751a74b61SMatthew G. Knepley       PetscReal       vertexA[4]      = {0.5, 0.5, 0.5, 0.5};
284851a74b61SMatthew G. Knepley       PetscReal       vertexB[4]      = {1.0, 0.0, 0.0, 0.0};
284951a74b61SMatthew G. Knepley       PetscReal       vertexC[4]      = {0.5, 0.5 * PETSC_PHI, 0.5 / PETSC_PHI, 0.0};
2850116ded15SMatthew G. Knepley       const PetscInt  degree          = 12;
2851116ded15SMatthew G. Knepley       PetscInt        s[4]            = {1, 1, 1};
28529371c9d4SSatish Balay       PetscInt        evenPerm[12][4] = {
28539371c9d4SSatish Balay         {0, 1, 2, 3},
28549371c9d4SSatish Balay         {0, 2, 3, 1},
28559371c9d4SSatish Balay         {0, 3, 1, 2},
28569371c9d4SSatish Balay         {1, 0, 3, 2},
28579371c9d4SSatish Balay         {1, 2, 0, 3},
28589371c9d4SSatish Balay         {1, 3, 2, 0},
28599371c9d4SSatish Balay         {2, 0, 1, 3},
28609371c9d4SSatish Balay         {2, 1, 3, 0},
28619371c9d4SSatish Balay         {2, 3, 0, 1},
28629371c9d4SSatish Balay         {3, 0, 2, 1},
28639371c9d4SSatish Balay         {3, 1, 0, 2},
28649371c9d4SSatish Balay         {3, 2, 1, 0}
28659371c9d4SSatish Balay       };
2866116ded15SMatthew G. Knepley       PetscInt  cone[4];
2867116ded15SMatthew G. Knepley       PetscInt *graph, p, i, j, k, l;
2868116ded15SMatthew G. Knepley 
28699371c9d4SSatish Balay       vertexA[0] *= R;
28709371c9d4SSatish Balay       vertexA[1] *= R;
28719371c9d4SSatish Balay       vertexA[2] *= R;
28729371c9d4SSatish Balay       vertexA[3] *= R;
28739371c9d4SSatish Balay       vertexB[0] *= R;
28749371c9d4SSatish Balay       vertexB[1] *= R;
28759371c9d4SSatish Balay       vertexB[2] *= R;
28769371c9d4SSatish Balay       vertexB[3] *= R;
28779371c9d4SSatish Balay       vertexC[0] *= R;
28789371c9d4SSatish Balay       vertexC[1] *= R;
28799371c9d4SSatish Balay       vertexC[2] *= R;
28809371c9d4SSatish Balay       vertexC[3] *= R;
2881dd400576SPatrick Sanan       numCells    = rank == 0 ? 600 : 0;
2882dd400576SPatrick Sanan       numVerts    = rank == 0 ? 120 : 0;
2883116ded15SMatthew G. Knepley       firstVertex = numCells;
2884116ded15SMatthew G. Knepley       /* Use the 600-cell, which for a unit sphere has coordinates which are
2885116ded15SMatthew G. Knepley 
2886116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm 1,    \pm 1, \pm 1)                          16
2887116ded15SMatthew G. Knepley                (\pm 1,    0,       0,      0)  all cyclic permutations   8
2888116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm phi, \pm 1/phi, 0)  all even permutations    96
2889116ded15SMatthew G. Knepley 
2890116ded15SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
28916333ae4fSvaleriabarra          length is then given by 1/\phi = 0.61803.
2892116ded15SMatthew G. Knepley 
2893116ded15SMatthew G. Knepley          http://buzzard.pugetsound.edu/sage-practice/ch03s03.html
2894116ded15SMatthew G. Knepley          http://mathworld.wolfram.com/600-Cell.html
2895116ded15SMatthew G. Knepley       */
2896116ded15SMatthew G. Knepley       /* Construct vertices */
28979566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2898116ded15SMatthew G. Knepley       i = 0;
2899dd400576SPatrick Sanan       if (rank == 0) {
2900116ded15SMatthew G. Knepley         for (s[0] = -1; s[0] < 2; s[0] += 2) {
2901116ded15SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
2902116ded15SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
2903116ded15SMatthew G. Knepley               for (s[3] = -1; s[3] < 2; s[3] += 2) {
2904116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[d] * vertexA[d];
2905116ded15SMatthew G. Knepley                 ++i;
2906116ded15SMatthew G. Knepley               }
2907116ded15SMatthew G. Knepley             }
2908116ded15SMatthew G. Knepley           }
2909116ded15SMatthew G. Knepley         }
2910116ded15SMatthew G. Knepley         for (p = 0; p < embedDim; ++p) {
2911116ded15SMatthew G. Knepley           s[1] = s[2] = s[3] = 1;
2912116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
2913116ded15SMatthew G. Knepley             for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertexB[(d + p) % embedDim];
2914116ded15SMatthew G. Knepley             ++i;
2915116ded15SMatthew G. Knepley           }
2916116ded15SMatthew G. Knepley         }
2917116ded15SMatthew G. Knepley         for (p = 0; p < 12; ++p) {
2918116ded15SMatthew G. Knepley           s[3] = 1;
2919116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
2920116ded15SMatthew G. Knepley             for (s[1] = -1; s[1] < 2; s[1] += 2) {
2921116ded15SMatthew G. Knepley               for (s[2] = -1; s[2] < 2; s[2] += 2) {
2922116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[evenPerm[p][d]] * vertexC[evenPerm[p][d]];
2923116ded15SMatthew G. Knepley                 ++i;
2924116ded15SMatthew G. Knepley               }
2925116ded15SMatthew G. Knepley             }
2926116ded15SMatthew G. Knepley           }
2927116ded15SMatthew G. Knepley         }
292845da822fSValeria Barra       }
292963a3b9bcSJacob Faibussowitsch       PetscCheck(i == numVerts, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertices %" PetscInt_FMT " != %" PetscInt_FMT, i, numVerts);
2930116ded15SMatthew G. Knepley       /* Construct graph */
29319566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
2932116ded15SMatthew G. Knepley       for (i = 0; i < numVerts; ++i) {
2933116ded15SMatthew G. Knepley         for (j = 0, k = 0; j < numVerts; ++j) {
29349371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
29359371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
29369371c9d4SSatish Balay             ++k;
29379371c9d4SSatish Balay           }
2938116ded15SMatthew G. Knepley         }
293963a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
2940116ded15SMatthew G. Knepley       }
2941116ded15SMatthew G. Knepley       /* Build Topology */
29429566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
294307c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
29449566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
2945116ded15SMatthew G. Knepley       /* Cells */
2946dd400576SPatrick Sanan       if (rank == 0) {
294707c565c5SJose E. Roman         for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
2948116ded15SMatthew G. Knepley           for (j = 0; j < i; ++j) {
2949116ded15SMatthew G. Knepley             for (k = 0; k < j; ++k) {
2950116ded15SMatthew G. Knepley               for (l = 0; l < k; ++l) {
29519371c9d4SSatish Balay                 if (graph[i * numVerts + j] && graph[j * numVerts + k] && graph[k * numVerts + i] && graph[l * numVerts + i] && graph[l * numVerts + j] && graph[l * numVerts + k]) {
29529371c9d4SSatish Balay                   cone[0] = firstVertex + i;
29539371c9d4SSatish Balay                   cone[1] = firstVertex + j;
29549371c9d4SSatish Balay                   cone[2] = firstVertex + k;
29559371c9d4SSatish Balay                   cone[3] = firstVertex + l;
2956116ded15SMatthew G. Knepley                   /* Check orientation: https://ef.gy/linear-algebra:normal-vectors-in-higher-dimensional-spaces */
2957116ded15SMatthew G. Knepley                   {
29589371c9d4SSatish Balay                     const PetscInt epsilon[4][4][4][4] = {
29599371c9d4SSatish Balay                       {{{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}},  {{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 1}, {0, 0, -1, 0}}, {{0, 0, 0, 0}, {0, 0, 0, -1}, {0, 0, 0, 0}, {0, 1, 0, 0}}, {{0, 0, 0, 0}, {0, 0, 1, 0}, {0, -1, 0, 0}, {0, 0, 0, 0}}},
2960116ded15SMatthew G. Knepley 
29619371c9d4SSatish Balay                       {{{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, -1}, {0, 0, 1, 0}}, {{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}},  {{0, 0, 0, 1}, {0, 0, 0, 0}, {0, 0, 0, 0}, {-1, 0, 0, 0}}, {{0, 0, -1, 0}, {0, 0, 0, 0}, {1, 0, 0, 0}, {0, 0, 0, 0}}},
2962116ded15SMatthew G. Knepley 
29639371c9d4SSatish Balay                       {{{0, 0, 0, 0}, {0, 0, 0, 1}, {0, 0, 0, 0}, {0, -1, 0, 0}}, {{0, 0, 0, -1}, {0, 0, 0, 0}, {0, 0, 0, 0}, {1, 0, 0, 0}}, {{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}},  {{0, 1, 0, 0}, {-1, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}}},
2964116ded15SMatthew G. Knepley 
29659371c9d4SSatish Balay                       {{{0, 0, 0, 0}, {0, 0, -1, 0}, {0, 1, 0, 0}, {0, 0, 0, 0}}, {{0, 0, 1, 0}, {0, 0, 0, 0}, {-1, 0, 0, 0}, {0, 0, 0, 0}}, {{0, -1, 0, 0}, {1, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}}, {{0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}, {0, 0, 0, 0}} }
29669371c9d4SSatish Balay                     };
2967116ded15SMatthew G. Knepley                     PetscReal normal[4];
2968116ded15SMatthew G. Knepley                     PetscInt  e, f, g;
2969116ded15SMatthew G. Knepley 
2970116ded15SMatthew G. Knepley                     for (d = 0; d < embedDim; ++d) {
2971116ded15SMatthew G. Knepley                       normal[d] = 0.0;
2972116ded15SMatthew G. Knepley                       for (e = 0; e < embedDim; ++e) {
2973116ded15SMatthew G. Knepley                         for (f = 0; f < embedDim; ++f) {
2974116ded15SMatthew G. Knepley                           for (g = 0; g < embedDim; ++g) {
2975116ded15SMatthew G. Knepley                             normal[d] += epsilon[d][e][f][g] * (coordsIn[j * embedDim + e] - coordsIn[i * embedDim + e]) * (coordsIn[k * embedDim + f] - coordsIn[i * embedDim + f]) * (coordsIn[l * embedDim + f] - coordsIn[i * embedDim + f]);
2976116ded15SMatthew G. Knepley                           }
2977116ded15SMatthew G. Knepley                         }
2978116ded15SMatthew G. Knepley                       }
2979116ded15SMatthew G. Knepley                     }
29809371c9d4SSatish Balay                     if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
29819371c9d4SSatish Balay                       PetscInt tmp = cone[1];
29829371c9d4SSatish Balay                       cone[1]      = cone[2];
29839371c9d4SSatish Balay                       cone[2]      = tmp;
29849371c9d4SSatish Balay                     }
2985116ded15SMatthew G. Knepley                   }
29869566063dSJacob Faibussowitsch                   PetscCall(DMPlexSetCone(dm, c++, cone));
2987116ded15SMatthew G. Knepley                 }
2988116ded15SMatthew G. Knepley               }
2989116ded15SMatthew G. Knepley             }
2990116ded15SMatthew G. Knepley           }
2991116ded15SMatthew G. Knepley         }
299245da822fSValeria Barra       }
29939566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
29949566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
29959566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
2996116ded15SMatthew G. Knepley     }
2997f4d061e9SPierre Jolivet     break;
2998d71ae5a4SJacob Faibussowitsch   default:
2999d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension for sphere: %" PetscInt_FMT, dim);
300065a81367SMatthew G. Knepley   }
300165a81367SMatthew G. Knepley   /* Create coordinates */
30029566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
30039566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
30049566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, embedDim));
30059566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVerts));
30062829fed8SMatthew G. Knepley   for (v = firstVertex; v < firstVertex + numVerts; ++v) {
30079566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, embedDim));
30089566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, embedDim));
30092829fed8SMatthew G. Knepley   }
30109566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
30119566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
30129566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
30139566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, embedDim));
30149566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
30159566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
30169566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
30179566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
30189371c9d4SSatish Balay   for (v = 0; v < numVerts; ++v)
3019ad540459SPierre Jolivet     for (d = 0; d < embedDim; ++d) coords[v * embedDim + d] = coordsIn[v * embedDim + d];
30209566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
30219566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
30229566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
30239566063dSJacob Faibussowitsch   PetscCall(PetscFree(coordsIn));
302451a74b61SMatthew G. Knepley   {
302551a74b61SMatthew G. Knepley     DM          cdm;
302651a74b61SMatthew G. Knepley     PetscDS     cds;
30279318fe57SMatthew G. Knepley     PetscScalar c = R;
302851a74b61SMatthew G. Knepley 
3029e44f6aebSMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, snapToSphere));
30309566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
30319566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
30329566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 1, &c));
303351a74b61SMatthew G. Knepley   }
303446139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
30359318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
30369566063dSJacob Faibussowitsch   if (simplex) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
30373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30389318fe57SMatthew G. Knepley }
30399318fe57SMatthew G. Knepley 
3040b7f5c055SJed Brown typedef void (*TPSEvaluateFunc)(const PetscReal[], PetscReal *, PetscReal[], PetscReal (*)[3]);
3041b7f5c055SJed Brown 
3042b7f5c055SJed Brown /*
3043b7f5c055SJed Brown  The Schwarz P implicit surface is
3044b7f5c055SJed Brown 
3045b7f5c055SJed Brown      f(x) = cos(x0) + cos(x1) + cos(x2) = 0
3046b7f5c055SJed Brown */
3047d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_SchwarzP(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3048d71ae5a4SJacob Faibussowitsch {
3049b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(y[0] * PETSC_PI), PetscCosReal(y[1] * PETSC_PI), PetscCosReal(y[2] * PETSC_PI)};
3050b7f5c055SJed Brown   PetscReal g[3] = {-PetscSinReal(y[0] * PETSC_PI), -PetscSinReal(y[1] * PETSC_PI), -PetscSinReal(y[2] * PETSC_PI)};
3051b7f5c055SJed Brown   f[0]           = c[0] + c[1] + c[2];
3052b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3053b7f5c055SJed Brown     grad[i] = PETSC_PI * g[i];
3054ad540459SPierre Jolivet     for (PetscInt j = 0; j < 3; j++) hess[i][j] = (i == j) ? -PetscSqr(PETSC_PI) * c[i] : 0.;
3055b7f5c055SJed Brown   }
3056b7f5c055SJed Brown }
3057b7f5c055SJed Brown 
30584663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3059d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_SchwarzP(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3060d71ae5a4SJacob Faibussowitsch {
3061ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) u[i] = -PETSC_PI * PetscSinReal(x[i] * PETSC_PI);
30623ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
30634663dae6SJed Brown }
30644663dae6SJed Brown 
3065b7f5c055SJed Brown /*
3066b7f5c055SJed Brown  The Gyroid implicit surface is
3067b7f5c055SJed Brown 
3068b7f5c055SJed Brown  f(x,y,z) = sin(pi * x) * cos (pi * (y + 1/2))  + sin(pi * (y + 1/2)) * cos(pi * (z + 1/4)) + sin(pi * (z + 1/4)) * cos(pi * x)
3069b7f5c055SJed Brown 
3070b7f5c055SJed Brown */
3071d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_Gyroid(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3072d71ae5a4SJacob Faibussowitsch {
3073b7f5c055SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * y[0]), PetscSinReal(PETSC_PI * (y[1] + .5)), PetscSinReal(PETSC_PI * (y[2] + .25))};
3074b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * y[0]), PetscCosReal(PETSC_PI * (y[1] + .5)), PetscCosReal(PETSC_PI * (y[2] + .25))};
3075b7f5c055SJed Brown   f[0]           = s[0] * c[1] + s[1] * c[2] + s[2] * c[0];
3076b7f5c055SJed Brown   grad[0]        = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
3077b7f5c055SJed Brown   grad[1]        = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
3078b7f5c055SJed Brown   grad[2]        = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
3079b7f5c055SJed Brown   hess[0][0]     = -PetscSqr(PETSC_PI) * (s[0] * c[1] + s[2] * c[0]);
3080b7f5c055SJed Brown   hess[0][1]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3081b7f5c055SJed Brown   hess[0][2]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3082b7f5c055SJed Brown   hess[1][0]     = -PetscSqr(PETSC_PI) * (s[1] * c[2] + s[0] * c[1]);
3083b7f5c055SJed Brown   hess[1][1]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3084b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3085b7f5c055SJed Brown   hess[2][0]     = -PetscSqr(PETSC_PI) * (s[2] * c[0] + s[1] * c[2]);
3086b7f5c055SJed Brown   hess[2][1]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3087b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3088b7f5c055SJed Brown }
3089b7f5c055SJed Brown 
30904663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3091d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_Gyroid(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3092d71ae5a4SJacob Faibussowitsch {
30934663dae6SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * x[0]), PetscSinReal(PETSC_PI * (x[1] + .5)), PetscSinReal(PETSC_PI * (x[2] + .25))};
30944663dae6SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * x[0]), PetscCosReal(PETSC_PI * (x[1] + .5)), PetscCosReal(PETSC_PI * (x[2] + .25))};
30954663dae6SJed Brown   u[0]           = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
30964663dae6SJed Brown   u[1]           = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
30974663dae6SJed Brown   u[2]           = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
30983ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
30994663dae6SJed Brown }
31004663dae6SJed Brown 
3101b7f5c055SJed Brown /*
3102b7f5c055SJed Brown    We wish to solve
3103b7f5c055SJed Brown 
3104b7f5c055SJed Brown          min_y || y - x ||^2  subject to f(y) = 0
3105b7f5c055SJed Brown 
3106b7f5c055SJed Brown    Let g(y) = grad(f).  The minimization problem is equivalent to asking to satisfy
3107b7f5c055SJed Brown    f(y) = 0 and (y-x) is parallel to g(y).  We do this by using Householder QR to obtain a basis for the
3108b7f5c055SJed Brown    tangent space and ask for both components in the tangent space to be zero.
3109b7f5c055SJed Brown 
3110b7f5c055SJed Brown    Take g to be a column vector and compute the "full QR" factorization Q R = g,
3111b7f5c055SJed Brown    where Q = I - 2 n n^T is a symmetric orthogonal matrix.
3112b7f5c055SJed Brown    The first column of Q is parallel to g so the remaining two columns span the null space.
3113b7f5c055SJed Brown    Let Qn = Q[:,1:] be those remaining columns.  Then Qn Qn^T is an orthogonal projector into the tangent space.
3114da81f932SPierre Jolivet    Since Q is symmetric, this is equivalent to multiplying by Q and taking the last two entries.
3115b7f5c055SJed Brown    In total, we have a system of 3 equations in 3 unknowns:
3116b7f5c055SJed Brown 
3117b7f5c055SJed Brown      f(y) = 0                       1 equation
3118b7f5c055SJed Brown      Qn^T (y - x) = 0               2 equations
3119b7f5c055SJed Brown 
3120b7f5c055SJed Brown    Here, we compute the residual and Jacobian of this system.
3121b7f5c055SJed Brown */
3122d71ae5a4SJacob Faibussowitsch static void TPSNearestPointResJac(TPSEvaluateFunc feval, const PetscScalar x[], const PetscScalar y[], PetscScalar res[], PetscScalar J[])
3123d71ae5a4SJacob Faibussowitsch {
3124b7f5c055SJed Brown   PetscReal yreal[3] = {PetscRealPart(y[0]), PetscRealPart(y[1]), PetscRealPart(y[2])};
3125b7f5c055SJed Brown   PetscReal d[3]     = {PetscRealPart(y[0] - x[0]), PetscRealPart(y[1] - x[1]), PetscRealPart(y[2] - x[2])};
31262f0490c0SSatish Balay   PetscReal f, grad[3], n[3], norm, norm_y[3], nd, nd_y[3], sign;
31279371c9d4SSatish Balay   PetscReal n_y[3][3] = {
31289371c9d4SSatish Balay     {0, 0, 0},
31299371c9d4SSatish Balay     {0, 0, 0},
31309371c9d4SSatish Balay     {0, 0, 0}
31319371c9d4SSatish Balay   };
3132b7f5c055SJed Brown 
3133b7f5c055SJed Brown   feval(yreal, &f, grad, n_y);
3134b7f5c055SJed Brown 
3135b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n[i] = grad[i];
3136b7f5c055SJed Brown   norm = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3137ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) norm_y[i] = 1. / norm * n[i] * n_y[i][i];
3138b7f5c055SJed Brown 
3139b7f5c055SJed Brown   // Define the Householder reflector
3140b7f5c055SJed Brown   sign = n[0] >= 0 ? 1. : -1.;
3141b7f5c055SJed Brown   n[0] += norm * sign;
3142b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n_y[0][i] += norm_y[i] * sign;
3143b7f5c055SJed Brown 
3144b7f5c055SJed Brown   norm      = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3145b7f5c055SJed Brown   norm_y[0] = 1. / norm * (n[0] * n_y[0][0]);
3146b7f5c055SJed Brown   norm_y[1] = 1. / norm * (n[0] * n_y[0][1] + n[1] * n_y[1][1]);
3147b7f5c055SJed Brown   norm_y[2] = 1. / norm * (n[0] * n_y[0][2] + n[2] * n_y[2][2]);
3148b7f5c055SJed Brown 
3149b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3150b7f5c055SJed Brown     n[i] /= norm;
3151b7f5c055SJed Brown     for (PetscInt j = 0; j < 3; j++) {
3152b7f5c055SJed Brown       // note that n[i] is n_old[i]/norm when executing the code below
3153b7f5c055SJed Brown       n_y[i][j] = n_y[i][j] / norm - n[i] / norm * norm_y[j];
3154b7f5c055SJed Brown     }
3155b7f5c055SJed Brown   }
3156b7f5c055SJed Brown 
3157b7f5c055SJed Brown   nd = n[0] * d[0] + n[1] * d[1] + n[2] * d[2];
3158b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) nd_y[i] = n[i] + n_y[0][i] * d[0] + n_y[1][i] * d[1] + n_y[2][i] * d[2];
3159b7f5c055SJed Brown 
3160b7f5c055SJed Brown   res[0] = f;
3161b7f5c055SJed Brown   res[1] = d[1] - 2 * n[1] * nd;
3162b7f5c055SJed Brown   res[2] = d[2] - 2 * n[2] * nd;
3163b7f5c055SJed Brown   // J[j][i] is J_{ij} (column major)
3164b7f5c055SJed Brown   for (PetscInt j = 0; j < 3; j++) {
3165b7f5c055SJed Brown     J[0 + j * 3] = grad[j];
3166b7f5c055SJed Brown     J[1 + j * 3] = (j == 1) * 1. - 2 * (n_y[1][j] * nd + n[1] * nd_y[j]);
3167b7f5c055SJed Brown     J[2 + j * 3] = (j == 2) * 1. - 2 * (n_y[2][j] * nd + n[2] * nd_y[j]);
3168b7f5c055SJed Brown   }
3169b7f5c055SJed Brown }
3170b7f5c055SJed Brown 
3171b7f5c055SJed Brown /*
3172b7f5c055SJed Brown    Project x to the nearest point on the implicit surface using Newton's method.
3173b7f5c055SJed Brown */
3174d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSNearestPoint(TPSEvaluateFunc feval, PetscScalar x[])
3175d71ae5a4SJacob Faibussowitsch {
3176b7f5c055SJed Brown   PetscScalar y[3] = {x[0], x[1], x[2]}; // Initial guess
3177b7f5c055SJed Brown 
3178b7f5c055SJed Brown   PetscFunctionBegin;
3179b7f5c055SJed Brown   for (PetscInt iter = 0; iter < 10; iter++) {
3180b7f5c055SJed Brown     PetscScalar res[3], J[9];
3181b7f5c055SJed Brown     PetscReal   resnorm;
3182b7f5c055SJed Brown     TPSNearestPointResJac(feval, x, y, res, J);
3183b7f5c055SJed Brown     resnorm = PetscSqrtReal(PetscSqr(PetscRealPart(res[0])) + PetscSqr(PetscRealPart(res[1])) + PetscSqr(PetscRealPart(res[2])));
3184b7f5c055SJed Brown     if (0) { // Turn on this monitor if you need to confirm quadratic convergence
318563a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "[%" PetscInt_FMT "] res [%g %g %g]\n", iter, (double)PetscRealPart(res[0]), (double)PetscRealPart(res[1]), (double)PetscRealPart(res[2])));
3186b7f5c055SJed Brown     }
3187b7f5c055SJed Brown     if (resnorm < PETSC_SMALL) break;
3188b7f5c055SJed Brown 
3189b7f5c055SJed Brown     // Take the Newton step
31909566063dSJacob Faibussowitsch     PetscCall(PetscKernel_A_gets_inverse_A_3(J, 0., PETSC_FALSE, NULL));
3191b7f5c055SJed Brown     PetscKernel_v_gets_v_minus_A_times_w_3(y, J, res);
3192b7f5c055SJed Brown   }
3193b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) x[i] = y[i];
31943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3195b7f5c055SJed Brown }
3196b7f5c055SJed Brown 
3197b7f5c055SJed Brown const char *const DMPlexTPSTypes[] = {"SCHWARZ_P", "GYROID", "DMPlexTPSType", "DMPLEX_TPS_", NULL};
3198b7f5c055SJed Brown 
3199d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateTPSMesh_Internal(DM dm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness)
3200d71ae5a4SJacob Faibussowitsch {
3201b7f5c055SJed Brown   PetscMPIInt rank;
3202b7f5c055SJed Brown   PetscInt    topoDim = 2, spaceDim = 3, numFaces = 0, numVertices = 0, numEdges = 0;
3203b7f5c055SJed Brown   PetscInt(*edges)[2] = NULL, *edgeSets = NULL;
3204b7f5c055SJed Brown   PetscInt           *cells_flat = NULL;
3205b7f5c055SJed Brown   PetscReal          *vtxCoords  = NULL;
3206b7f5c055SJed Brown   TPSEvaluateFunc     evalFunc   = NULL;
32078434afd1SBarry Smith   PetscSimplePointFn *normalFunc = NULL;
3208b7f5c055SJed Brown   DMLabel             label;
3209b7f5c055SJed Brown 
3210b7f5c055SJed Brown   PetscFunctionBegin;
321146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
32129566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
321363a3b9bcSJacob Faibussowitsch   PetscCheck((layers != 0) ^ (thickness == 0.), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_INCOMP, "Layers %" PetscInt_FMT " must be nonzero iff thickness %g is nonzero", layers, (double)thickness);
3214b7f5c055SJed Brown   switch (tpstype) {
3215b7f5c055SJed Brown   case DMPLEX_TPS_SCHWARZ_P:
3216b7f5c055SJed Brown     PetscCheck(!periodic || (periodic[0] == DM_BOUNDARY_NONE && periodic[1] == DM_BOUNDARY_NONE && periodic[2] == DM_BOUNDARY_NONE), PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Schwarz P does not support periodic meshes");
3217c5853193SPierre Jolivet     if (rank == 0) {
3218b7f5c055SJed Brown       PetscInt(*cells)[6][4][4] = NULL; // [junction, junction-face, cell, conn]
3219b7f5c055SJed Brown       PetscInt  Njunctions = 0, Ncuts = 0, Npipes[3], vcount;
3220b7f5c055SJed Brown       PetscReal L = 1;
3221b7f5c055SJed Brown 
3222b7f5c055SJed Brown       Npipes[0]   = (extent[0] + 1) * extent[1] * extent[2];
3223b7f5c055SJed Brown       Npipes[1]   = extent[0] * (extent[1] + 1) * extent[2];
3224b7f5c055SJed Brown       Npipes[2]   = extent[0] * extent[1] * (extent[2] + 1);
3225b7f5c055SJed Brown       Njunctions  = extent[0] * extent[1] * extent[2];
3226b7f5c055SJed Brown       Ncuts       = 2 * (extent[0] * extent[1] + extent[1] * extent[2] + extent[2] * extent[0]);
3227b7f5c055SJed Brown       numVertices = 4 * (Npipes[0] + Npipes[1] + Npipes[2]) + 8 * Njunctions;
32289566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(3 * numVertices, &vtxCoords));
32299566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Njunctions, &cells));
32309566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edges));
32319566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edgeSets));
3232b7f5c055SJed Brown       // x-normal pipes
3233b7f5c055SJed Brown       vcount = 0;
3234b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0] + 1; i++) {
3235b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3236b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3237b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3238b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * i - 1) * L;
3239b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3240b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3241b7f5c055SJed Brown             }
3242b7f5c055SJed Brown           }
3243b7f5c055SJed Brown         }
3244b7f5c055SJed Brown       }
3245b7f5c055SJed Brown       // y-normal pipes
3246b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3247b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1] + 1; j++) {
3248b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3249b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3250b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3251b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * j - 1) * L;
3252b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3253b7f5c055SJed Brown             }
3254b7f5c055SJed Brown           }
3255b7f5c055SJed Brown         }
3256b7f5c055SJed Brown       }
3257b7f5c055SJed Brown       // z-normal pipes
3258b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3259b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3260b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2] + 1; k++) {
3261b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3262b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3263b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3264b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * k - 1) * L;
3265b7f5c055SJed Brown             }
3266b7f5c055SJed Brown           }
3267b7f5c055SJed Brown         }
3268b7f5c055SJed Brown       }
3269b7f5c055SJed Brown       // junctions
3270b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3271b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3272b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3273b7f5c055SJed Brown             const PetscInt J = (i * extent[1] + j) * extent[2] + k, Jvoff = (Npipes[0] + Npipes[1] + Npipes[2]) * 4 + J * 8;
3274b7f5c055SJed Brown             PetscCheck(vcount / 3 == Jvoff, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected vertex count");
3275b7f5c055SJed Brown             for (PetscInt ii = 0; ii < 2; ii++) {
3276b7f5c055SJed Brown               for (PetscInt jj = 0; jj < 2; jj++) {
3277b7f5c055SJed Brown                 for (PetscInt kk = 0; kk < 2; kk++) {
3278b7f5c055SJed Brown                   double Ls           = (1 - sqrt(2) / 4) * L;
3279b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * i * L + (2 * ii - 1) * Ls;
3280b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * j * L + (2 * jj - 1) * Ls;
3281b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * k * L + (2 * kk - 1) * Ls;
3282b7f5c055SJed Brown                 }
3283b7f5c055SJed Brown               }
3284b7f5c055SJed Brown             }
3285b7f5c055SJed Brown             const PetscInt jfaces[3][2][4] = {
3286b7f5c055SJed Brown               {{3, 1, 0, 2}, {7, 5, 4, 6}}, // x-aligned
3287b7f5c055SJed Brown               {{5, 4, 0, 1}, {7, 6, 2, 3}}, // y-aligned
3288b7f5c055SJed Brown               {{6, 2, 0, 4}, {7, 3, 1, 5}}  // z-aligned
3289b7f5c055SJed Brown             };
3290b7f5c055SJed Brown             const PetscInt pipe_lo[3] = {// vertex numbers of pipes
32919371c9d4SSatish Balay                                          ((i * extent[1] + j) * extent[2] + k) * 4, ((i * (extent[1] + 1) + j) * extent[2] + k + Npipes[0]) * 4, ((i * extent[1] + j) * (extent[2] + 1) + k + Npipes[0] + Npipes[1]) * 4};
3292b7f5c055SJed Brown             const PetscInt pipe_hi[3] = {// vertex numbers of pipes
32939371c9d4SSatish Balay                                          (((i + 1) * extent[1] + j) * extent[2] + k) * 4, ((i * (extent[1] + 1) + j + 1) * extent[2] + k + Npipes[0]) * 4, ((i * extent[1] + j) * (extent[2] + 1) + k + 1 + Npipes[0] + Npipes[1]) * 4};
3294b7f5c055SJed Brown             for (PetscInt dir = 0; dir < 3; dir++) { // x,y,z
3295b7f5c055SJed Brown               const PetscInt ijk[3] = {i, j, k};
3296b7f5c055SJed Brown               for (PetscInt l = 0; l < 4; l++) { // rotations
3297b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][0] = pipe_lo[dir] + l;
3298b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][1] = Jvoff + jfaces[dir][0][l];
3299b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][2] = Jvoff + jfaces[dir][0][(l - 1 + 4) % 4];
3300b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][3] = pipe_lo[dir] + (l - 1 + 4) % 4;
3301b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][0] = Jvoff + jfaces[dir][1][l];
3302b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][1] = pipe_hi[dir] + l;
3303b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][2] = pipe_hi[dir] + (l - 1 + 4) % 4;
3304b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][3] = Jvoff + jfaces[dir][1][(l - 1 + 4) % 4];
3305b7f5c055SJed Brown                 if (ijk[dir] == 0) {
3306b7f5c055SJed Brown                   edges[numEdges][0] = pipe_lo[dir] + l;
3307b7f5c055SJed Brown                   edges[numEdges][1] = pipe_lo[dir] + (l + 1) % 4;
3308b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 1;
3309b7f5c055SJed Brown                   numEdges++;
3310b7f5c055SJed Brown                 }
3311b7f5c055SJed Brown                 if (ijk[dir] + 1 == extent[dir]) {
3312b7f5c055SJed Brown                   edges[numEdges][0] = pipe_hi[dir] + l;
3313b7f5c055SJed Brown                   edges[numEdges][1] = pipe_hi[dir] + (l + 1) % 4;
3314b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 2;
3315b7f5c055SJed Brown                   numEdges++;
3316b7f5c055SJed Brown                 }
3317b7f5c055SJed Brown               }
3318b7f5c055SJed Brown             }
3319b7f5c055SJed Brown           }
3320b7f5c055SJed Brown         }
3321b7f5c055SJed Brown       }
332263a3b9bcSJacob Faibussowitsch       PetscCheck(numEdges == Ncuts * 4, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Edge count %" PetscInt_FMT " incompatible with number of cuts %" PetscInt_FMT, numEdges, Ncuts);
3323b7f5c055SJed Brown       numFaces   = 24 * Njunctions;
3324b7f5c055SJed Brown       cells_flat = cells[0][0][0];
3325b7f5c055SJed Brown     }
3326b7f5c055SJed Brown     evalFunc   = TPSEvaluate_SchwarzP;
33274663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_SchwarzP;
3328b7f5c055SJed Brown     break;
3329b7f5c055SJed Brown   case DMPLEX_TPS_GYROID:
3330c5853193SPierre Jolivet     if (rank == 0) {
3331b7f5c055SJed Brown       // This is a coarse mesh approximation of the gyroid shifted to being the zero of the level set
3332b7f5c055SJed Brown       //
3333b7f5c055SJed Brown       //     sin(pi*x)*cos(pi*(y+1/2)) + sin(pi*(y+1/2))*cos(pi*(z+1/4)) + sin(pi*(z+1/4))*cos(x)
3334b7f5c055SJed Brown       //
3335b7f5c055SJed Brown       // on the cell [0,2]^3.
3336b7f5c055SJed Brown       //
3337b7f5c055SJed Brown       // Think about dividing that cell into four columns, and focus on the column [0,1]x[0,1]x[0,2].
3338b7f5c055SJed Brown       // If you looked at the gyroid in that column at different slices of z you would see that it kind of spins
3339b7f5c055SJed Brown       // like a boomerang:
3340b7f5c055SJed Brown       //
3341b7f5c055SJed Brown       //     z = 0          z = 1/4        z = 1/2        z = 3/4     //
3342b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3343b7f5c055SJed Brown       //                                                              //
3344b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3345b7f5c055SJed Brown       //      \                                   /            \      //
3346b7f5c055SJed Brown       //       \            `-_   _-'            /              }     //
3347b7f5c055SJed Brown       //        *-_            `-'            _-'              /      //
3348b7f5c055SJed Brown       //     +     `-+      +       +      +-'     +      +   /   +   //
3349b7f5c055SJed Brown       //                                                              //
3350b7f5c055SJed Brown       //                                                              //
3351b7f5c055SJed Brown       //     z = 1          z = 5/4        z = 3/2        z = 7/4     //
3352b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3353b7f5c055SJed Brown       //                                                              //
3354b7f5c055SJed Brown       //     +-_     +      +       +      +     _-+      +   /   +   //
3355b7f5c055SJed Brown       //        `-_            _-_            _-`            /        //
3356b7f5c055SJed Brown       //           \        _-'   `-_        /              {         //
3357b7f5c055SJed Brown       //            \                       /                \        //
3358b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3359b7f5c055SJed Brown       //
3360b7f5c055SJed Brown       //
3361b7f5c055SJed Brown       // This course mesh approximates each of these slices by two line segments,
3362b7f5c055SJed Brown       // and then connects the segments in consecutive layers with quadrilateral faces.
3363b7f5c055SJed Brown       // All of the end points of the segments are multiples of 1/4 except for the
3364b7f5c055SJed Brown       // point * in the picture for z = 0 above and the similar points in other layers.
3365b7f5c055SJed Brown       // That point is at (gamma, gamma, 0), where gamma is calculated below.
3366b7f5c055SJed Brown       //
3367b7f5c055SJed Brown       // The column  [1,2]x[1,2]x[0,2] looks the same as this column;
3368b7f5c055SJed Brown       // The columns [1,2]x[0,1]x[0,2] and [0,1]x[1,2]x[0,2] are mirror images.
3369b7f5c055SJed Brown       //
3370b7f5c055SJed Brown       // As for how this method turned into the names given to the vertices:
3371b7f5c055SJed Brown       // that was not systematic, it was just the way it worked out in my handwritten notes.
3372b7f5c055SJed Brown 
3373b7f5c055SJed Brown       PetscInt facesPerBlock = 64;
3374b7f5c055SJed Brown       PetscInt vertsPerBlock = 56;
3375b7f5c055SJed Brown       PetscInt extentPlus[3];
3376b7f5c055SJed Brown       PetscInt numBlocks, numBlocksPlus;
33779371c9d4SSatish Balay       const PetscInt A = 0, B = 1, C = 2, D = 3, E = 4, F = 5, G = 6, H = 7, II = 8, J = 9, K = 10, L = 11, M = 12, N = 13, O = 14, P = 15, Q = 16, R = 17, S = 18, T = 19, U = 20, V = 21, W = 22, X = 23, Y = 24, Z = 25, Ap = 26, Bp = 27, Cp = 28, Dp = 29, Ep = 30, Fp = 31, Gp = 32, Hp = 33, Ip = 34, Jp = 35, Kp = 36, Lp = 37, Mp = 38, Np = 39, Op = 40, Pp = 41, Qp = 42, Rp = 43, Sp = 44, Tp = 45, Up = 46, Vp = 47, Wp = 48, Xp = 49, Yp = 50, Zp = 51, Aq = 52, Bq = 53, Cq = 54, Dq = 55;
33789371c9d4SSatish Balay       const PetscInt pattern[64][4] = {
33799371c9d4SSatish Balay         /* face to vertex within the coarse discretization of a single gyroid block */
3380b7f5c055SJed Brown         /* layer 0 */
33819371c9d4SSatish Balay         {A,           C,           K,           G          },
33829371c9d4SSatish Balay         {C,           B,           II,          K          },
33839371c9d4SSatish Balay         {D,           A,           H,           L          },
33849371c9d4SSatish Balay         {B + 56 * 1,  D,           L,           J          },
33859371c9d4SSatish Balay         {E,           B + 56 * 1,  J,           N          },
33869371c9d4SSatish Balay         {A + 56 * 2,  E,           N,           H + 56 * 2 },
33879371c9d4SSatish Balay         {F,           A + 56 * 2,  G + 56 * 2,  M          },
33889371c9d4SSatish Balay         {B,           F,           M,           II         },
3389b7f5c055SJed Brown         /* layer 1 */
33909371c9d4SSatish Balay         {G,           K,           Q,           O          },
33919371c9d4SSatish Balay         {K,           II,          P,           Q          },
33929371c9d4SSatish Balay         {L,           H,           O + 56 * 1,  R          },
33939371c9d4SSatish Balay         {J,           L,           R,           P          },
33949371c9d4SSatish Balay         {N,           J,           P,           S          },
33959371c9d4SSatish Balay         {H + 56 * 2,  N,           S,           O + 56 * 3 },
33969371c9d4SSatish Balay         {M,           G + 56 * 2,  O + 56 * 2,  T          },
33979371c9d4SSatish Balay         {II,          M,           T,           P          },
3398b7f5c055SJed Brown         /* layer 2 */
33999371c9d4SSatish Balay         {O,           Q,           Y,           U          },
34009371c9d4SSatish Balay         {Q,           P,           W,           Y          },
34019371c9d4SSatish Balay         {R,           O + 56 * 1,  U + 56 * 1,  Ap         },
34029371c9d4SSatish Balay         {P,           R,           Ap,          W          },
34039371c9d4SSatish Balay         {S,           P,           X,           Bp         },
34049371c9d4SSatish Balay         {O + 56 * 3,  S,           Bp,          V + 56 * 1 },
34059371c9d4SSatish Balay         {T,           O + 56 * 2,  V,           Z          },
34069371c9d4SSatish Balay         {P,           T,           Z,           X          },
3407b7f5c055SJed Brown         /* layer 3 */
34089371c9d4SSatish Balay         {U,           Y,           Ep,          Dp         },
34099371c9d4SSatish Balay         {Y,           W,           Cp,          Ep         },
34109371c9d4SSatish Balay         {Ap,          U + 56 * 1,  Dp + 56 * 1, Gp         },
34119371c9d4SSatish Balay         {W,           Ap,          Gp,          Cp         },
34129371c9d4SSatish Balay         {Bp,          X,           Cp + 56 * 2, Fp         },
34139371c9d4SSatish Balay         {V + 56 * 1,  Bp,          Fp,          Dp + 56 * 1},
34149371c9d4SSatish Balay         {Z,           V,           Dp,          Hp         },
34159371c9d4SSatish Balay         {X,           Z,           Hp,          Cp + 56 * 2},
3416b7f5c055SJed Brown         /* layer 4 */
34179371c9d4SSatish Balay         {Dp,          Ep,          Mp,          Kp         },
34189371c9d4SSatish Balay         {Ep,          Cp,          Ip,          Mp         },
34199371c9d4SSatish Balay         {Gp,          Dp + 56 * 1, Lp,          Np         },
34209371c9d4SSatish Balay         {Cp,          Gp,          Np,          Jp         },
34219371c9d4SSatish Balay         {Fp,          Cp + 56 * 2, Jp + 56 * 2, Pp         },
34229371c9d4SSatish Balay         {Dp + 56 * 1, Fp,          Pp,          Lp         },
34239371c9d4SSatish Balay         {Hp,          Dp,          Kp,          Op         },
34249371c9d4SSatish Balay         {Cp + 56 * 2, Hp,          Op,          Ip + 56 * 2},
3425b7f5c055SJed Brown         /* layer 5 */
34269371c9d4SSatish Balay         {Kp,          Mp,          Sp,          Rp         },
34279371c9d4SSatish Balay         {Mp,          Ip,          Qp,          Sp         },
34289371c9d4SSatish Balay         {Np,          Lp,          Rp,          Tp         },
34299371c9d4SSatish Balay         {Jp,          Np,          Tp,          Qp + 56 * 1},
34309371c9d4SSatish Balay         {Pp,          Jp + 56 * 2, Qp + 56 * 3, Up         },
34319371c9d4SSatish Balay         {Lp,          Pp,          Up,          Rp         },
34329371c9d4SSatish Balay         {Op,          Kp,          Rp,          Vp         },
34339371c9d4SSatish Balay         {Ip + 56 * 2, Op,          Vp,          Qp + 56 * 2},
3434b7f5c055SJed Brown         /* layer 6 */
34359371c9d4SSatish Balay         {Rp,          Sp,          Aq,          Yp         },
34369371c9d4SSatish Balay         {Sp,          Qp,          Wp,          Aq         },
34379371c9d4SSatish Balay         {Tp,          Rp,          Yp,          Cq         },
34389371c9d4SSatish Balay         {Qp + 56 * 1, Tp,          Cq,          Wp + 56 * 1},
34399371c9d4SSatish Balay         {Up,          Qp + 56 * 3, Xp + 56 * 1, Dq         },
34409371c9d4SSatish Balay         {Rp,          Up,          Dq,          Zp         },
34419371c9d4SSatish Balay         {Vp,          Rp,          Zp,          Bq         },
34429371c9d4SSatish Balay         {Qp + 56 * 2, Vp,          Bq,          Xp         },
3443b7f5c055SJed Brown         /* layer 7 (the top is the periodic image of the bottom of layer 0) */
34449371c9d4SSatish Balay         {Yp,          Aq,          C + 56 * 4,  A + 56 * 4 },
34459371c9d4SSatish Balay         {Aq,          Wp,          B + 56 * 4,  C + 56 * 4 },
34469371c9d4SSatish Balay         {Cq,          Yp,          A + 56 * 4,  D + 56 * 4 },
34479371c9d4SSatish Balay         {Wp + 56 * 1, Cq,          D + 56 * 4,  B + 56 * 5 },
34489371c9d4SSatish Balay         {Dq,          Xp + 56 * 1, B + 56 * 5,  E + 56 * 4 },
34499371c9d4SSatish Balay         {Zp,          Dq,          E + 56 * 4,  A + 56 * 6 },
34509371c9d4SSatish Balay         {Bq,          Zp,          A + 56 * 6,  F + 56 * 4 },
34519371c9d4SSatish Balay         {Xp,          Bq,          F + 56 * 4,  B + 56 * 4 }
3452b7f5c055SJed Brown       };
3453b7f5c055SJed Brown       const PetscReal gamma                = PetscAcosReal((PetscSqrtReal(3.) - 1.) / PetscSqrtReal(2.)) / PETSC_PI;
34549371c9d4SSatish Balay       const PetscReal patternCoords[56][3] = {
3455bee3fc89SBarry Smith         {1.,        0.,        0.  }, /* A  */
3456bee3fc89SBarry Smith         {0.,        1.,        0.  }, /* B  */
3457bee3fc89SBarry Smith         {gamma,     gamma,     0.  }, /* C  */
3458bee3fc89SBarry Smith         {1 + gamma, 1 - gamma, 0.  }, /* D  */
3459bee3fc89SBarry Smith         {2 - gamma, 2 - gamma, 0.  }, /* E  */
3460bee3fc89SBarry Smith         {1 - gamma, 1 + gamma, 0.  }, /* F  */
3461b7f5c055SJed Brown 
3462bee3fc89SBarry Smith         {.5,        0,         .25 }, /* G  */
3463bee3fc89SBarry Smith         {1.5,       0.,        .25 }, /* H  */
3464bee3fc89SBarry Smith         {.5,        1.,        .25 }, /* II */
3465bee3fc89SBarry Smith         {1.5,       1.,        .25 }, /* J  */
3466bee3fc89SBarry Smith         {.25,       .5,        .25 }, /* K  */
3467bee3fc89SBarry Smith         {1.25,      .5,        .25 }, /* L  */
3468bee3fc89SBarry Smith         {.75,       1.5,       .25 }, /* M  */
3469bee3fc89SBarry Smith         {1.75,      1.5,       .25 }, /* N  */
3470b7f5c055SJed Brown 
3471bee3fc89SBarry Smith         {0.,        0.,        .5  }, /* O  */
3472bee3fc89SBarry Smith         {1.,        1.,        .5  }, /* P  */
3473bee3fc89SBarry Smith         {gamma,     1 - gamma, .5  }, /* Q  */
3474bee3fc89SBarry Smith         {1 + gamma, gamma,     .5  }, /* R  */
3475bee3fc89SBarry Smith         {2 - gamma, 1 + gamma, .5  }, /* S  */
3476bee3fc89SBarry Smith         {1 - gamma, 2 - gamma, .5  }, /* T  */
3477b7f5c055SJed Brown 
3478bee3fc89SBarry Smith         {0.,        .5,        .75 }, /* U  */
3479bee3fc89SBarry Smith         {0.,        1.5,       .75 }, /* V  */
3480bee3fc89SBarry Smith         {1.,        .5,        .75 }, /* W  */
3481bee3fc89SBarry Smith         {1.,        1.5,       .75 }, /* X  */
3482bee3fc89SBarry Smith         {.5,        .75,       .75 }, /* Y  */
3483bee3fc89SBarry Smith         {.5,        1.75,      .75 }, /* Z  */
3484bee3fc89SBarry Smith         {1.5,       .25,       .75 }, /* Ap */
3485bee3fc89SBarry Smith         {1.5,       1.25,      .75 }, /* Bp */
3486b7f5c055SJed Brown 
3487bee3fc89SBarry Smith         {1.,        0.,        1.  }, /* Cp */
3488bee3fc89SBarry Smith         {0.,        1.,        1.  }, /* Dp */
3489bee3fc89SBarry Smith         {1 - gamma, 1 - gamma, 1.  }, /* Ep */
3490bee3fc89SBarry Smith         {1 + gamma, 1 + gamma, 1.  }, /* Fp */
3491bee3fc89SBarry Smith         {2 - gamma, gamma,     1.  }, /* Gp */
3492bee3fc89SBarry Smith         {gamma,     2 - gamma, 1.  }, /* Hp */
3493b7f5c055SJed Brown 
3494bee3fc89SBarry Smith         {.5,        0.,        1.25}, /* Ip */
3495bee3fc89SBarry Smith         {1.5,       0.,        1.25}, /* Jp */
3496bee3fc89SBarry Smith         {.5,        1.,        1.25}, /* Kp */
3497bee3fc89SBarry Smith         {1.5,       1.,        1.25}, /* Lp */
3498bee3fc89SBarry Smith         {.75,       .5,        1.25}, /* Mp */
3499bee3fc89SBarry Smith         {1.75,      .5,        1.25}, /* Np */
3500bee3fc89SBarry Smith         {.25,       1.5,       1.25}, /* Op */
3501bee3fc89SBarry Smith         {1.25,      1.5,       1.25}, /* Pp */
3502b7f5c055SJed Brown 
3503bee3fc89SBarry Smith         {0.,        0.,        1.5 }, /* Qp */
3504bee3fc89SBarry Smith         {1.,        1.,        1.5 }, /* Rp */
3505bee3fc89SBarry Smith         {1 - gamma, gamma,     1.5 }, /* Sp */
3506bee3fc89SBarry Smith         {2 - gamma, 1 - gamma, 1.5 }, /* Tp */
3507bee3fc89SBarry Smith         {1 + gamma, 2 - gamma, 1.5 }, /* Up */
3508bee3fc89SBarry Smith         {gamma,     1 + gamma, 1.5 }, /* Vp */
3509b7f5c055SJed Brown 
3510bee3fc89SBarry Smith         {0.,        .5,        1.75}, /* Wp */
3511bee3fc89SBarry Smith         {0.,        1.5,       1.75}, /* Xp */
3512bee3fc89SBarry Smith         {1.,        .5,        1.75}, /* Yp */
3513bee3fc89SBarry Smith         {1.,        1.5,       1.75}, /* Zp */
3514bee3fc89SBarry Smith         {.5,        .25,       1.75}, /* Aq */
3515bee3fc89SBarry Smith         {.5,        1.25,      1.75}, /* Bq */
3516bee3fc89SBarry Smith         {1.5,       .75,       1.75}, /* Cq */
3517bee3fc89SBarry Smith         {1.5,       1.75,      1.75}, /* Dq */
3518b7f5c055SJed Brown       };
3519b7f5c055SJed Brown       PetscInt(*cells)[64][4] = NULL;
3520b7f5c055SJed Brown       PetscBool *seen;
3521b7f5c055SJed Brown       PetscInt  *vertToTrueVert;
3522b7f5c055SJed Brown       PetscInt   count;
3523b7f5c055SJed Brown 
3524b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) extentPlus[i] = extent[i] + 1;
3525b7f5c055SJed Brown       numBlocks = 1;
3526b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocks *= extent[i];
3527b7f5c055SJed Brown       numBlocksPlus = 1;
3528b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocksPlus *= extentPlus[i];
3529b7f5c055SJed Brown       numFaces = numBlocks * facesPerBlock;
35309566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocks, &cells));
35319566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numBlocksPlus * vertsPerBlock, &seen));
3532b7f5c055SJed Brown       for (PetscInt k = 0; k < extent[2]; k++) {
3533b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3534b7f5c055SJed Brown           for (PetscInt i = 0; i < extent[0]; i++) {
3535b7f5c055SJed Brown             for (PetscInt f = 0; f < facesPerBlock; f++) {
3536b7f5c055SJed Brown               for (PetscInt v = 0; v < 4; v++) {
3537b7f5c055SJed Brown                 PetscInt vertRaw     = pattern[f][v];
3538b7f5c055SJed Brown                 PetscInt blockidx    = vertRaw / 56;
3539b7f5c055SJed Brown                 PetscInt patternvert = vertRaw % 56;
3540b7f5c055SJed Brown                 PetscInt xplus       = (blockidx & 1);
3541b7f5c055SJed Brown                 PetscInt yplus       = (blockidx & 2) >> 1;
3542b7f5c055SJed Brown                 PetscInt zplus       = (blockidx & 4) >> 2;
3543b7f5c055SJed Brown                 PetscInt zcoord      = (periodic && periodic[2] == DM_BOUNDARY_PERIODIC) ? ((k + zplus) % extent[2]) : (k + zplus);
3544b7f5c055SJed Brown                 PetscInt ycoord      = (periodic && periodic[1] == DM_BOUNDARY_PERIODIC) ? ((j + yplus) % extent[1]) : (j + yplus);
3545b7f5c055SJed Brown                 PetscInt xcoord      = (periodic && periodic[0] == DM_BOUNDARY_PERIODIC) ? ((i + xplus) % extent[0]) : (i + xplus);
3546b7f5c055SJed Brown                 PetscInt vert        = ((zcoord * extentPlus[1] + ycoord) * extentPlus[0] + xcoord) * 56 + patternvert;
3547b7f5c055SJed Brown 
3548b7f5c055SJed Brown                 cells[(k * extent[1] + j) * extent[0] + i][f][v] = vert;
3549b7f5c055SJed Brown                 seen[vert]                                       = PETSC_TRUE;
3550b7f5c055SJed Brown               }
3551b7f5c055SJed Brown             }
3552b7f5c055SJed Brown           }
3553b7f5c055SJed Brown         }
3554b7f5c055SJed Brown       }
35559371c9d4SSatish Balay       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++)
35569371c9d4SSatish Balay         if (seen[i]) numVertices++;
3557b7f5c055SJed Brown       count = 0;
35589566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocksPlus * vertsPerBlock, &vertToTrueVert));
35599566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * 3, &vtxCoords));
3560b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++) vertToTrueVert[i] = -1;
3561b7f5c055SJed Brown       for (PetscInt k = 0; k < extentPlus[2]; k++) {
3562b7f5c055SJed Brown         for (PetscInt j = 0; j < extentPlus[1]; j++) {
3563b7f5c055SJed Brown           for (PetscInt i = 0; i < extentPlus[0]; i++) {
3564b7f5c055SJed Brown             for (PetscInt v = 0; v < vertsPerBlock; v++) {
3565b7f5c055SJed Brown               PetscInt vIdx = ((k * extentPlus[1] + j) * extentPlus[0] + i) * vertsPerBlock + v;
3566b7f5c055SJed Brown 
3567b7f5c055SJed Brown               if (seen[vIdx]) {
3568b7f5c055SJed Brown                 PetscInt thisVert;
3569b7f5c055SJed Brown 
3570b7f5c055SJed Brown                 vertToTrueVert[vIdx] = thisVert = count++;
3571b7f5c055SJed Brown 
3572b7f5c055SJed Brown                 for (PetscInt d = 0; d < 3; d++) vtxCoords[3 * thisVert + d] = patternCoords[v][d];
3573b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 0] += i * 2;
3574b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 1] += j * 2;
3575b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 2] += k * 2;
3576b7f5c055SJed Brown               }
3577b7f5c055SJed Brown             }
3578b7f5c055SJed Brown           }
3579b7f5c055SJed Brown         }
3580b7f5c055SJed Brown       }
3581b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocks; i++) {
3582b7f5c055SJed Brown         for (PetscInt f = 0; f < facesPerBlock; f++) {
3583ad540459SPierre Jolivet           for (PetscInt v = 0; v < 4; v++) cells[i][f][v] = vertToTrueVert[cells[i][f][v]];
3584b7f5c055SJed Brown         }
3585b7f5c055SJed Brown       }
35869566063dSJacob Faibussowitsch       PetscCall(PetscFree(vertToTrueVert));
35879566063dSJacob Faibussowitsch       PetscCall(PetscFree(seen));
3588b7f5c055SJed Brown       cells_flat = cells[0][0];
3589b7f5c055SJed Brown       numEdges   = 0;
3590b7f5c055SJed Brown       for (PetscInt i = 0; i < numFaces; i++) {
3591b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3592b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3593b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3594b7f5c055SJed Brown 
3595b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3596b7f5c055SJed Brown             if (!periodic || periodic[0] != DM_BOUNDARY_PERIODIC) {
3597b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) numEdges++;
3598b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) numEdges++;
3599b7f5c055SJed Brown             }
3600b7f5c055SJed Brown           }
3601b7f5c055SJed Brown         }
3602b7f5c055SJed Brown       }
36039566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edges));
36049566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edgeSets));
3605b7f5c055SJed Brown       for (PetscInt edge = 0, i = 0; i < numFaces; i++) {
3606b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3607b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3608b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3609b7f5c055SJed Brown 
3610b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3611b7f5c055SJed Brown             if (!periodic || periodic[d] != DM_BOUNDARY_PERIODIC) {
3612b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) {
3613b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3614b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3615b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d;
3616b7f5c055SJed Brown               }
3617b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) {
3618b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3619b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3620b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d + 1;
3621b7f5c055SJed Brown               }
3622b7f5c055SJed Brown             }
3623b7f5c055SJed Brown           }
3624b7f5c055SJed Brown         }
3625b7f5c055SJed Brown       }
3626b7f5c055SJed Brown     }
3627b7f5c055SJed Brown     evalFunc   = TPSEvaluate_Gyroid;
36284663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_Gyroid;
3629b7f5c055SJed Brown     break;
3630b7f5c055SJed Brown   }
3631b7f5c055SJed Brown 
36329566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, topoDim));
3633c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(dm, numFaces, numVertices, 4, cells_flat));
36349566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(dm, 0, 0, 0, NULL));
36359566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells_flat));
3636b7f5c055SJed Brown   {
3637b7f5c055SJed Brown     DM idm;
36389566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(dm, &idm));
363969d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &idm));
3640b7f5c055SJed Brown   }
3641c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, vtxCoords));
36429566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, NULL));
36439566063dSJacob Faibussowitsch   PetscCall(PetscFree(vtxCoords));
3644b7f5c055SJed Brown 
36459566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
36469566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
3647b7f5c055SJed Brown   for (PetscInt e = 0; e < numEdges; e++) {
3648b7f5c055SJed Brown     PetscInt        njoin;
3649b7f5c055SJed Brown     const PetscInt *join, verts[] = {numFaces + edges[e][0], numFaces + edges[e][1]};
36509566063dSJacob Faibussowitsch     PetscCall(DMPlexGetJoin(dm, 2, verts, &njoin, &join));
365163a3b9bcSJacob Faibussowitsch     PetscCheck(njoin == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected unique join of vertices %" PetscInt_FMT " and %" PetscInt_FMT, edges[e][0], edges[e][1]);
36529566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(label, join[0], edgeSets[e]));
36539566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(dm, 2, verts, &njoin, &join));
3654b7f5c055SJed Brown   }
36559566063dSJacob Faibussowitsch   PetscCall(PetscFree(edges));
36569566063dSJacob Faibussowitsch   PetscCall(PetscFree(edgeSets));
36571436d7faSJed Brown   if (tps_distribute) {
36581436d7faSJed Brown     DM               pdm = NULL;
36591436d7faSJed Brown     PetscPartitioner part;
36601436d7faSJed Brown 
36619566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
36629566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
36639566063dSJacob Faibussowitsch     PetscCall(DMPlexDistribute(dm, 0, NULL, &pdm));
366448a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
36651436d7faSJed Brown     // Do not auto-distribute again
36669566063dSJacob Faibussowitsch     PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
36671436d7faSJed Brown   }
3668b7f5c055SJed Brown 
36699566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
3670b7f5c055SJed Brown   for (PetscInt refine = 0; refine < refinements; refine++) {
3671b7f5c055SJed Brown     PetscInt     m;
3672b7f5c055SJed Brown     DM           dmf;
3673b7f5c055SJed Brown     Vec          X;
3674b7f5c055SJed Brown     PetscScalar *x;
36759566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, MPI_COMM_NULL, &dmf));
367669d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmf));
3677b7f5c055SJed Brown 
36789566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &X));
36799566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(X, &m));
36809566063dSJacob Faibussowitsch     PetscCall(VecGetArray(X, &x));
368148a46eb9SPierre Jolivet     for (PetscInt i = 0; i < m; i += 3) PetscCall(TPSNearestPoint(evalFunc, &x[i]));
36829566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(X, &x));
3683b7f5c055SJed Brown   }
3684b7f5c055SJed Brown 
3685b7f5c055SJed Brown   // Face Sets has already been propagated to new vertices during refinement; this propagates to the initial vertices.
36869566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
36879566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, label));
3688b7f5c055SJed Brown 
368946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
369046139095SJed Brown 
3691b7f5c055SJed Brown   if (thickness > 0) {
36924663dae6SJed Brown     DM              edm, cdm, ecdm;
36934663dae6SJed Brown     DMPlexTransform tr;
36944663dae6SJed Brown     const char     *prefix;
36954663dae6SJed Brown     PetscOptions    options;
36964663dae6SJed Brown     // Code from DMPlexExtrude
36974663dae6SJed Brown     PetscCall(DMPlexTransformCreate(PetscObjectComm((PetscObject)dm), &tr));
36984663dae6SJed Brown     PetscCall(DMPlexTransformSetDM(tr, dm));
36994663dae6SJed Brown     PetscCall(DMPlexTransformSetType(tr, DMPLEXEXTRUDE));
37004663dae6SJed Brown     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
37014663dae6SJed Brown     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tr, prefix));
37024663dae6SJed Brown     PetscCall(PetscObjectGetOptions((PetscObject)dm, &options));
37034663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, options));
37044663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetLayers(tr, layers));
37054663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetThickness(tr, thickness));
37064663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetTensor(tr, PETSC_FALSE));
37074663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetSymmetric(tr, PETSC_TRUE));
37084663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetNormalFunction(tr, normalFunc));
37094663dae6SJed Brown     PetscCall(DMPlexTransformSetFromOptions(tr));
37104663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, NULL));
37114663dae6SJed Brown     PetscCall(DMPlexTransformSetUp(tr));
37124663dae6SJed Brown     PetscCall(PetscObjectViewFromOptions((PetscObject)tr, NULL, "-dm_plex_tps_transform_view"));
37134663dae6SJed Brown     PetscCall(DMPlexTransformApply(tr, dm, &edm));
37144663dae6SJed Brown     PetscCall(DMCopyDisc(dm, edm));
37154663dae6SJed Brown     PetscCall(DMGetCoordinateDM(dm, &cdm));
37164663dae6SJed Brown     PetscCall(DMGetCoordinateDM(edm, &ecdm));
37174663dae6SJed Brown     PetscCall(DMCopyDisc(cdm, ecdm));
37184663dae6SJed Brown     PetscCall(DMPlexTransformCreateDiscLabels(tr, edm));
37194663dae6SJed Brown     PetscCall(DMPlexTransformDestroy(&tr));
3720a77a5016SMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, edm));
372169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
3722b7f5c055SJed Brown   }
37233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3724b7f5c055SJed Brown }
3725b7f5c055SJed Brown 
3726b7f5c055SJed Brown /*@
3727b7f5c055SJed Brown   DMPlexCreateTPSMesh - Create a distributed, interpolated mesh of a triply-periodic surface
3728b7f5c055SJed Brown 
3729b7f5c055SJed Brown   Collective
3730b7f5c055SJed Brown 
3731b7f5c055SJed Brown   Input Parameters:
3732a1cb98faSBarry Smith + comm           - The communicator for the `DM` object
3733b7f5c055SJed Brown . tpstype        - Type of triply-periodic surface
3734b7f5c055SJed Brown . extent         - Array of length 3 containing number of periods in each direction
373520f4b53cSBarry Smith . periodic       - array of length 3 with periodicity, or `NULL` for non-periodic
37361436d7faSJed Brown . tps_distribute - Distribute 2D manifold mesh prior to refinement and extrusion (more scalable)
3737817da375SSatish Balay . refinements    - Number of factor-of-2 refinements of 2D manifold mesh
37381436d7faSJed Brown . layers         - Number of cell layers extruded in normal direction
3739817da375SSatish Balay - thickness      - Thickness in normal direction
3740b7f5c055SJed Brown 
3741b7f5c055SJed Brown   Output Parameter:
3742a1cb98faSBarry Smith . dm - The `DM` object
3743a1cb98faSBarry Smith 
3744a1cb98faSBarry Smith   Level: beginner
3745b7f5c055SJed Brown 
3746b7f5c055SJed Brown   Notes:
374715229ffcSPierre Jolivet   This meshes the surface of the Schwarz P or Gyroid surfaces.  Schwarz P is the simplest member of the triply-periodic minimal surfaces.
37481d27aa22SBarry Smith   <https://en.wikipedia.org/wiki/Schwarz_minimal_surface#Schwarz_P_(%22Primitive%22)> and can be cut with "clean" boundaries.
37491d27aa22SBarry Smith   The Gyroid <https://en.wikipedia.org/wiki/Gyroid> is another triply-periodic minimal surface with applications in additive manufacturing; it is much more difficult to "cut" since there are no planes of symmetry.
3750b7f5c055SJed Brown   Our implementation creates a very coarse mesh of the surface and refines (by 4-way splitting) as many times as requested.
3751b7f5c055SJed Brown   On each refinement, all vertices are projected to their nearest point on the surface.
3752b7f5c055SJed Brown   This projection could readily be extended to related surfaces.
3753b7f5c055SJed Brown 
37541d27aa22SBarry Smith   See {cite}`maskery2018insights`
37551d27aa22SBarry Smith 
37561d27aa22SBarry Smith   The face (edge) sets for the Schwarz P surface are numbered $1(-x), 2(+x), 3(-y), 4(+y), 5(-z), 6(+z)$.
37571d27aa22SBarry Smith   When the mesh is refined, "Face Sets" contain the new vertices (created during refinement).
37581d27aa22SBarry Smith   Use `DMPlexLabelComplete()` to propagate to coarse-level vertices.
3759b7f5c055SJed Brown 
376060225df5SJacob Faibussowitsch   Developer Notes:
3761b7f5c055SJed Brown   The Gyroid mesh does not currently mark boundary sets.
3762b7f5c055SJed Brown 
37631cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMSetType()`, `DMCreate()`
3764b7f5c055SJed Brown @*/
3765d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateTPSMesh(MPI_Comm comm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness, DM *dm)
3766d71ae5a4SJacob Faibussowitsch {
3767b7f5c055SJed Brown   PetscFunctionBegin;
37689566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
37699566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
37709566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateTPSMesh_Internal(*dm, tpstype, extent, periodic, tps_distribute, refinements, layers, thickness));
37713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3772b7f5c055SJed Brown }
3773b7f5c055SJed Brown 
37749318fe57SMatthew G. Knepley /*@
37759318fe57SMatthew G. Knepley   DMPlexCreateSphereMesh - Creates a mesh on the d-dimensional sphere, S^d.
37769318fe57SMatthew G. Knepley 
37779318fe57SMatthew G. Knepley   Collective
37789318fe57SMatthew G. Knepley 
37799318fe57SMatthew G. Knepley   Input Parameters:
3780a1cb98faSBarry Smith + comm    - The communicator for the `DM` object
37819318fe57SMatthew G. Knepley . dim     - The dimension
37829318fe57SMatthew G. Knepley . simplex - Use simplices, or tensor product cells
37839318fe57SMatthew G. Knepley - R       - The radius
37849318fe57SMatthew G. Knepley 
37859318fe57SMatthew G. Knepley   Output Parameter:
3786a1cb98faSBarry Smith . dm - The `DM` object
37879318fe57SMatthew G. Knepley 
37889318fe57SMatthew G. Knepley   Level: beginner
37899318fe57SMatthew G. Knepley 
37901cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBallMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
37919318fe57SMatthew G. Knepley @*/
3792d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateSphereMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscReal R, DM *dm)
3793d71ae5a4SJacob Faibussowitsch {
37949318fe57SMatthew G. Knepley   PetscFunctionBegin;
37954f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
37969566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
37979566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
37989566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(*dm, dim, simplex, R));
37993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
38009318fe57SMatthew G. Knepley }
38019318fe57SMatthew G. Knepley 
3802d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBallMesh_Internal(DM dm, PetscInt dim, PetscReal R)
3803d71ae5a4SJacob Faibussowitsch {
38049318fe57SMatthew G. Knepley   DM          sdm, vol;
38059318fe57SMatthew G. Knepley   DMLabel     bdlabel;
3806dd2b43ebSStefano Zampini   const char *prefix;
38079318fe57SMatthew G. Knepley 
38089318fe57SMatthew G. Knepley   PetscFunctionBegin;
38099566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &sdm));
38109566063dSJacob Faibussowitsch   PetscCall(DMSetType(sdm, DMPLEX));
3811dd2b43ebSStefano Zampini   PetscCall(DMGetOptionsPrefix(dm, &prefix));
3812dd2b43ebSStefano Zampini   PetscCall(DMSetOptionsPrefix(sdm, prefix));
3813dd2b43ebSStefano Zampini   PetscCall(DMAppendOptionsPrefix(sdm, "bd_"));
3814dd2b43ebSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(sdm, PETSC_FALSE));
38159566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(sdm, dim - 1, PETSC_TRUE, R));
38169566063dSJacob Faibussowitsch   PetscCall(DMSetFromOptions(sdm));
38179566063dSJacob Faibussowitsch   PetscCall(DMViewFromOptions(sdm, NULL, "-dm_view"));
38189566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(sdm, NULL, PETSC_TRUE, &vol));
38199566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&sdm));
382069d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
38219566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
38229566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
38239566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
38249566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, bdlabel));
38253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
382651a74b61SMatthew G. Knepley }
382751a74b61SMatthew G. Knepley 
382851a74b61SMatthew G. Knepley /*@
382951a74b61SMatthew G. Knepley   DMPlexCreateBallMesh - Creates a simplex mesh on the d-dimensional ball, B^d.
383051a74b61SMatthew G. Knepley 
383151a74b61SMatthew G. Knepley   Collective
383251a74b61SMatthew G. Knepley 
383351a74b61SMatthew G. Knepley   Input Parameters:
3834a1cb98faSBarry Smith + comm - The communicator for the `DM` object
383551a74b61SMatthew G. Knepley . dim  - The dimension
383651a74b61SMatthew G. Knepley - R    - The radius
383751a74b61SMatthew G. Knepley 
383851a74b61SMatthew G. Knepley   Output Parameter:
3839a1cb98faSBarry Smith . dm - The `DM` object
384051a74b61SMatthew G. Knepley 
3841a1cb98faSBarry Smith   Options Database Key:
384260225df5SJacob Faibussowitsch . bd_dm_refine - This will refine the surface mesh preserving the sphere geometry
384351a74b61SMatthew G. Knepley 
384451a74b61SMatthew G. Knepley   Level: beginner
384551a74b61SMatthew G. Knepley 
38461cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
384751a74b61SMatthew G. Knepley @*/
3848d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBallMesh(MPI_Comm comm, PetscInt dim, PetscReal R, DM *dm)
3849d71ae5a4SJacob Faibussowitsch {
385051a74b61SMatthew G. Knepley   PetscFunctionBegin;
38519566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
38529566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
38539566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBallMesh_Internal(*dm, dim, R));
38543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
38552829fed8SMatthew G. Knepley }
38562829fed8SMatthew G. Knepley 
3857d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateReferenceCell_Internal(DM rdm, DMPolytopeType ct)
3858d71ae5a4SJacob Faibussowitsch {
38590a6ba040SMatthew G. Knepley   PetscFunctionBegin;
38609318fe57SMatthew G. Knepley   switch (ct) {
38619371c9d4SSatish Balay   case DM_POLYTOPE_POINT: {
38629318fe57SMatthew G. Knepley     PetscInt    numPoints[1]        = {1};
38639318fe57SMatthew G. Knepley     PetscInt    coneSize[1]         = {0};
38649318fe57SMatthew G. Knepley     PetscInt    cones[1]            = {0};
38659318fe57SMatthew G. Knepley     PetscInt    coneOrientations[1] = {0};
38669318fe57SMatthew G. Knepley     PetscScalar vertexCoords[1]     = {0.0};
38679318fe57SMatthew G. Knepley 
38689566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 0));
38699566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38709371c9d4SSatish Balay   } break;
38719371c9d4SSatish Balay   case DM_POLYTOPE_SEGMENT: {
38729318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
38739318fe57SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
38749318fe57SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
38759318fe57SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
38769318fe57SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
38779318fe57SMatthew G. Knepley 
38789566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
38799566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38809371c9d4SSatish Balay   } break;
38819371c9d4SSatish Balay   case DM_POLYTOPE_POINT_PRISM_TENSOR: {
3882b5a892a1SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
3883b5a892a1SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
3884b5a892a1SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
3885b5a892a1SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
3886b5a892a1SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
3887b5a892a1SMatthew G. Knepley 
38889566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
38899566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38909371c9d4SSatish Balay   } break;
38919371c9d4SSatish Balay   case DM_POLYTOPE_TRIANGLE: {
38929318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {3, 1};
38939318fe57SMatthew G. Knepley     PetscInt    coneSize[4]         = {3, 0, 0, 0};
38949318fe57SMatthew G. Knepley     PetscInt    cones[3]            = {1, 2, 3};
38959318fe57SMatthew G. Knepley     PetscInt    coneOrientations[3] = {0, 0, 0};
38969318fe57SMatthew G. Knepley     PetscScalar vertexCoords[6]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0};
38979318fe57SMatthew G. Knepley 
38989566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
38999566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39009371c9d4SSatish Balay   } break;
39019371c9d4SSatish Balay   case DM_POLYTOPE_QUADRILATERAL: {
39029318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39039318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
39049318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39059318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
39069318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0};
39079318fe57SMatthew G. Knepley 
39089566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
39099566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39109371c9d4SSatish Balay   } break;
39119371c9d4SSatish Balay   case DM_POLYTOPE_SEG_PRISM_TENSOR: {
39129318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39139318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
39149318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39159318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
39169318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0};
39179318fe57SMatthew G. Knepley 
39189566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
39199566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39209371c9d4SSatish Balay   } break;
39219371c9d4SSatish Balay   case DM_POLYTOPE_TETRAHEDRON: {
39229318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39239318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
3924f0edb160SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39259318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
3926f0edb160SMatthew G. Knepley     PetscScalar vertexCoords[12]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0};
39279318fe57SMatthew G. Knepley 
39289566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39299566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39309371c9d4SSatish Balay   } break;
39319371c9d4SSatish Balay   case DM_POLYTOPE_HEXAHEDRON: {
39329318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
39339318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
3934f0edb160SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
39359318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39369371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0};
39379318fe57SMatthew G. Knepley 
39389566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39399566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39409371c9d4SSatish Balay   } break;
39419371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM: {
39429318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
39439318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
3944f0edb160SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
39459318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
39469371c9d4SSatish Balay     PetscScalar vertexCoords[18]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, 1.0, 1.0};
39479318fe57SMatthew G. Knepley 
39489566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39499566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39509371c9d4SSatish Balay   } break;
39519371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM_TENSOR: {
39529318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
39539318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
39549318fe57SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
39559318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
39569371c9d4SSatish Balay     PetscScalar vertexCoords[18]    = {-1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, 1.0, 1.0};
39579318fe57SMatthew G. Knepley 
39589566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39599566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39609371c9d4SSatish Balay   } break;
39619371c9d4SSatish Balay   case DM_POLYTOPE_QUAD_PRISM_TENSOR: {
39629318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
39639318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
39649318fe57SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
39659318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39669371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0};
39679318fe57SMatthew G. Knepley 
39689566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39699566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39709371c9d4SSatish Balay   } break;
39719371c9d4SSatish Balay   case DM_POLYTOPE_PYRAMID: {
39729318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {5, 1};
39739318fe57SMatthew G. Knepley     PetscInt    coneSize[6]         = {5, 0, 0, 0, 0, 0};
3974f0edb160SMatthew G. Knepley     PetscInt    cones[5]            = {1, 2, 3, 4, 5};
39759318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39769371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0, 1.0};
39779318fe57SMatthew G. Knepley 
39789566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39799566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39809371c9d4SSatish Balay   } break;
3981d71ae5a4SJacob Faibussowitsch   default:
3982d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)rdm), PETSC_ERR_ARG_WRONG, "Cannot create reference cell for cell type %s", DMPolytopeTypes[ct]);
39839318fe57SMatthew G. Knepley   }
39849318fe57SMatthew G. Knepley   {
39859318fe57SMatthew G. Knepley     PetscInt Nv, v;
39869318fe57SMatthew G. Knepley 
39879318fe57SMatthew G. Knepley     /* Must create the celltype label here so that we do not automatically try to compute the types */
39889566063dSJacob Faibussowitsch     PetscCall(DMCreateLabel(rdm, "celltype"));
39899566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCellType(rdm, 0, ct));
39909566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(rdm, NULL, &Nv));
39919566063dSJacob Faibussowitsch     for (v = 1; v < Nv; ++v) PetscCall(DMPlexSetCellType(rdm, v, DM_POLYTOPE_POINT));
39929318fe57SMatthew G. Knepley   }
39939566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(rdm));
39949566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)rdm, DMPolytopeTypes[ct]));
39953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
39960a6ba040SMatthew G. Knepley }
39970a6ba040SMatthew G. Knepley 
39989318fe57SMatthew G. Knepley /*@
3999a1cb98faSBarry Smith   DMPlexCreateReferenceCell - Create a `DMPLEX` with the appropriate FEM reference cell
40009318fe57SMatthew G. Knepley 
40019318fe57SMatthew G. Knepley   Collective
40029318fe57SMatthew G. Knepley 
40039318fe57SMatthew G. Knepley   Input Parameters:
40049318fe57SMatthew G. Knepley + comm - The communicator
40059318fe57SMatthew G. Knepley - ct   - The cell type of the reference cell
40069318fe57SMatthew G. Knepley 
40079318fe57SMatthew G. Knepley   Output Parameter:
40089318fe57SMatthew G. Knepley . refdm - The reference cell
40099318fe57SMatthew G. Knepley 
40109318fe57SMatthew G. Knepley   Level: intermediate
40119318fe57SMatthew G. Knepley 
401242747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`
40139318fe57SMatthew G. Knepley @*/
4014d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, DMPolytopeType ct, DM *refdm)
4015d71ae5a4SJacob Faibussowitsch {
40160a6ba040SMatthew G. Knepley   PetscFunctionBegin;
40179566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, refdm));
40189566063dSJacob Faibussowitsch   PetscCall(DMSetType(*refdm, DMPLEX));
40199566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceCell_Internal(*refdm, ct));
40203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40219318fe57SMatthew G. Knepley }
402279a015ccSMatthew G. Knepley 
4023d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoundaryLabel_Private(DM dm, const char name[])
4024d71ae5a4SJacob Faibussowitsch {
40259318fe57SMatthew G. Knepley   DM        plex;
40269318fe57SMatthew G. Knepley   DMLabel   label;
40279318fe57SMatthew G. Knepley   PetscBool hasLabel;
40280a6ba040SMatthew G. Knepley 
4029c22d3578SMatthew G. Knepley   PetscFunctionBegin;
40309566063dSJacob Faibussowitsch   PetscCall(DMHasLabel(dm, name, &hasLabel));
40313ba16761SJacob Faibussowitsch   if (hasLabel) PetscFunctionReturn(PETSC_SUCCESS);
40329566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, name));
40339566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, name, &label));
40349566063dSJacob Faibussowitsch   PetscCall(DMConvert(dm, DMPLEX, &plex));
40359566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(plex, 1, label));
40361c8afea9SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(plex, label));
40379566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&plex));
40383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40399318fe57SMatthew G. Knepley }
4040acdc6f61SToby Isaac 
4041669647acSMatthew G. Knepley /*
4042669647acSMatthew G. Knepley   We use the last coordinate as the radius, the inner radius is lower[dim-1] and the outer radius is upper[dim-1]. Then we map the first coordinate around the circle.
4043669647acSMatthew G. Knepley 
4044669647acSMatthew G. Knepley     (x, y) -> (r, theta) = (x[1], (x[0] - lower[0]) * 2\pi/(upper[0] - lower[0]))
4045669647acSMatthew G. Knepley */
4046d71ae5a4SJacob Faibussowitsch static void boxToAnnulus(PetscInt dim, PetscInt Nf, PetscInt NfAux, const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[])
4047d71ae5a4SJacob Faibussowitsch {
4048669647acSMatthew G. Knepley   const PetscReal low = PetscRealPart(constants[0]);
4049669647acSMatthew G. Knepley   const PetscReal upp = PetscRealPart(constants[1]);
4050669647acSMatthew G. Knepley   const PetscReal r   = PetscRealPart(u[1]);
4051669647acSMatthew G. Knepley   const PetscReal th  = 2. * PETSC_PI * (PetscRealPart(u[0]) - low) / (upp - low);
4052669647acSMatthew G. Knepley 
4053669647acSMatthew G. Knepley   f0[0] = r * PetscCosReal(th);
4054669647acSMatthew G. Knepley   f0[1] = r * PetscSinReal(th);
4055669647acSMatthew G. Knepley }
4056669647acSMatthew G. Knepley 
40575390be7dSMatthew G. Knepley // Insert vertices and their joins, marked by depth
40585390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Vertices(DM dm, DMLabel label, DMLabel vlabel, PetscInt val, PetscInt n, const PetscInt vertices[])
40595390be7dSMatthew G. Knepley {
40605390be7dSMatthew G. Knepley   PetscFunctionBegin;
40615390be7dSMatthew G. Knepley   PetscCall(DMPlexMarkSubmesh_Interpolated(dm, vlabel, val, PETSC_FALSE, PETSC_FALSE, label, NULL));
40625390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
40635390be7dSMatthew G. Knepley }
40645390be7dSMatthew G. Knepley 
40655390be7dSMatthew G. Knepley // Insert faces and their closures, marked by depth
40665390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Faces(DM dm, DMLabel label, PetscInt n, const PetscInt faces[])
40675390be7dSMatthew G. Knepley {
40685390be7dSMatthew G. Knepley   PetscFunctionBegin;
40695390be7dSMatthew G. Knepley   for (PetscInt p = 0; p < n; ++p) {
40705390be7dSMatthew G. Knepley     const PetscInt point   = faces[p];
40715390be7dSMatthew G. Knepley     PetscInt      *closure = NULL;
40725390be7dSMatthew G. Knepley     PetscInt       clSize, pdepth;
40735390be7dSMatthew G. Knepley 
40745390be7dSMatthew G. Knepley     PetscCall(DMPlexGetPointDepth(dm, point, &pdepth));
40755390be7dSMatthew G. Knepley     PetscCall(DMLabelSetValue(label, point, pdepth));
40765390be7dSMatthew G. Knepley     PetscCall(DMPlexGetTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
40775390be7dSMatthew G. Knepley     for (PetscInt cl = 0; cl < clSize * 2; cl += 2) {
40785390be7dSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, closure[cl], &pdepth));
40795390be7dSMatthew G. Knepley       PetscCall(DMLabelSetValue(label, closure[cl], pdepth));
40805390be7dSMatthew G. Knepley     }
40815390be7dSMatthew G. Knepley     PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
40825390be7dSMatthew G. Knepley   }
40835390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
40845390be7dSMatthew G. Knepley }
40855390be7dSMatthew G. Knepley 
40864e22dd4cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions, const char pre[], const char name[], const char *option[], const char *value[], PetscBool *flg);
40874e22dd4cSMatthew G. Knepley 
40885dca41c3SJed Brown const char *const DMPlexShapes[] = {"box", "box_surface", "ball", "sphere", "cylinder", "schwarz_p", "gyroid", "doublet", "annulus", "hypercubic", "zbox", "unknown", "DMPlexShape", "DM_SHAPE_", NULL};
40899318fe57SMatthew G. Knepley 
4090d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateFromOptions_Internal(PetscOptionItems *PetscOptionsObject, PetscBool *useCoordSpace, DM dm)
4091d71ae5a4SJacob Faibussowitsch {
40929318fe57SMatthew G. Knepley   DMPlexShape    shape   = DM_SHAPE_BOX;
40939318fe57SMatthew G. Knepley   DMPolytopeType cell    = DM_POLYTOPE_TRIANGLE;
40949318fe57SMatthew G. Knepley   PetscInt       dim     = 2;
40959318fe57SMatthew G. Knepley   PetscBool      simplex = PETSC_TRUE, interpolate = PETSC_TRUE, adjCone = PETSC_FALSE, adjClosure = PETSC_TRUE, refDomain = PETSC_FALSE;
4096d0812dedSMatthew G. Knepley   PetscBool      flg, flg2, fflg, strflg, bdfflg, nameflg;
40979318fe57SMatthew G. Knepley   MPI_Comm       comm;
4098ed5e4e85SVaclav Hapla   char           filename[PETSC_MAX_PATH_LEN]   = "<unspecified>";
4099ed5e4e85SVaclav Hapla   char           bdFilename[PETSC_MAX_PATH_LEN] = "<unspecified>";
4100ed5e4e85SVaclav Hapla   char           plexname[PETSC_MAX_PATH_LEN]   = "";
41014e22dd4cSMatthew G. Knepley   const char    *option;
41029318fe57SMatthew G. Knepley 
41039318fe57SMatthew G. Knepley   PetscFunctionBegin;
4104708be2fdSJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
41059566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
41069318fe57SMatthew G. Knepley   /* TODO Turn this into a registration interface */
41079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_filename", "File containing a mesh", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &fflg));
4108d0812dedSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_file_contents", "Contents of a file format in a string", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &strflg));
41099566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_filename", "File containing a mesh boundary", "DMPlexCreateFromFile", bdFilename, bdFilename, sizeof(bdFilename), &bdfflg));
41109566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_name", "Name of the mesh in the file", "DMPlexCreateFromFile", plexname, plexname, sizeof(plexname), &nameflg));
41119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_cell", "Cell shape", "", DMPolytopeTypes, (PetscEnum)cell, (PetscEnum *)&cell, NULL));
41129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_reference_cell_domain", "Use a reference cell domain", "", refDomain, &refDomain, NULL));
41139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_shape", "Shape for built-in mesh", "", DMPlexShapes, (PetscEnum)shape, (PetscEnum *)&shape, &flg));
41149566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_dim", "Topological dimension of the mesh", "DMGetDimension", dim, &dim, &flg, 0));
41159566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_simplex", "Mesh cell shape", "", simplex, &simplex, &flg));
41169566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate", "Flag to create edges and faces automatically", "", interpolate, &interpolate, &flg));
41179566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_cone", "Set adjacency direction", "DMSetBasicAdjacency", adjCone, &adjCone, &flg));
41189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_closure", "Set adjacency size", "DMSetBasicAdjacency", adjClosure, &adjClosure, &flg2));
41199566063dSJacob Faibussowitsch   if (flg || flg2) PetscCall(DMSetBasicAdjacency(dm, adjCone, adjClosure));
41209318fe57SMatthew G. Knepley 
412161a622f3SMatthew G. Knepley   switch (cell) {
412261a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT:
412361a622f3SMatthew G. Knepley   case DM_POLYTOPE_SEGMENT:
412461a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT_PRISM_TENSOR:
412561a622f3SMatthew G. Knepley   case DM_POLYTOPE_TRIANGLE:
412661a622f3SMatthew G. Knepley   case DM_POLYTOPE_QUADRILATERAL:
412761a622f3SMatthew G. Knepley   case DM_POLYTOPE_TETRAHEDRON:
4128d71ae5a4SJacob Faibussowitsch   case DM_POLYTOPE_HEXAHEDRON:
4129d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_TRUE;
4130d71ae5a4SJacob Faibussowitsch     break;
4131d71ae5a4SJacob Faibussowitsch   default:
4132d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_FALSE;
4133d71ae5a4SJacob Faibussowitsch     break;
413461a622f3SMatthew G. Knepley   }
413561a622f3SMatthew G. Knepley 
41369318fe57SMatthew G. Knepley   if (fflg) {
41379318fe57SMatthew G. Knepley     DM          dmnew;
41381e4a82c4SMatthew G. Knepley     const char *name;
41399318fe57SMatthew G. Knepley 
41401e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
41411e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), filename, nameflg ? plexname : name, interpolate, &dmnew));
41425de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
414369d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
41449318fe57SMatthew G. Knepley   } else if (refDomain) {
41459566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateReferenceCell_Internal(dm, cell));
41469318fe57SMatthew G. Knepley   } else if (bdfflg) {
41479318fe57SMatthew G. Knepley     DM          bdm, dmnew;
41481e4a82c4SMatthew G. Knepley     const char *name;
41499318fe57SMatthew G. Knepley 
41501e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
41511e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), bdFilename, nameflg ? plexname : name, interpolate, &bdm));
41529566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)bdm, "bd_"));
41539566063dSJacob Faibussowitsch     PetscCall(DMSetFromOptions(bdm));
41549566063dSJacob Faibussowitsch     PetscCall(DMPlexGenerate(bdm, NULL, interpolate, &dmnew));
41559566063dSJacob Faibussowitsch     PetscCall(DMDestroy(&bdm));
41565de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
415769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
4158d0812dedSMatthew G. Knepley   } else if (strflg) {
4159d0812dedSMatthew G. Knepley     DM          dmnew;
4160d0812dedSMatthew G. Knepley     PetscViewer viewer;
4161d0812dedSMatthew G. Knepley     const char *contents;
4162d0812dedSMatthew G. Knepley     char       *strname;
4163d0812dedSMatthew G. Knepley     char        tmpdir[PETSC_MAX_PATH_LEN];
4164d0812dedSMatthew G. Knepley     char        tmpfilename[PETSC_MAX_PATH_LEN];
4165d0812dedSMatthew G. Knepley     char        name[PETSC_MAX_PATH_LEN];
4166d0812dedSMatthew G. Knepley     MPI_Comm    comm;
4167d0812dedSMatthew G. Knepley     PetscMPIInt rank;
4168d0812dedSMatthew G. Knepley 
4169d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4170d0812dedSMatthew G. Knepley     PetscCallMPI(MPI_Comm_rank(comm, &rank));
4171d0812dedSMatthew G. Knepley     PetscCall(PetscStrchr(filename, ':', &strname));
4172d0812dedSMatthew G. Knepley     PetscCheck(strname, comm, PETSC_ERR_ARG_WRONG, "File contents must have the form \"ext:string_name\", not %s", filename);
4173d0812dedSMatthew G. Knepley     strname[0] = '\0';
4174d0812dedSMatthew G. Knepley     ++strname;
4175d0812dedSMatthew G. Knepley     PetscCall(PetscDLSym(NULL, strname, (void **)&contents));
4176d0812dedSMatthew G. Knepley     PetscCheck(contents, comm, PETSC_ERR_ARG_WRONG, "Could not locate mesh string %s", strname);
4177d0812dedSMatthew G. Knepley     PetscCall(PetscGetTmp(comm, tmpdir, PETSC_MAX_PATH_LEN));
4178ed32af8cSMatthew G. Knepley     PetscCall(PetscStrlcat(tmpdir, "/meshXXXXXX", PETSC_MAX_PATH_LEN));
4179ed32af8cSMatthew G. Knepley     PetscCall(PetscMkdtemp(tmpdir));
4180ed32af8cSMatthew G. Knepley     PetscCall(PetscSNPrintf(tmpfilename, PETSC_MAX_PATH_LEN, "%s/mesh.%s", tmpdir, filename));
4181d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIOpen(comm, tmpfilename, &viewer));
4182d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIPrintf(viewer, "%s\n", contents));
4183d0812dedSMatthew G. Knepley     PetscCall(PetscViewerDestroy(&viewer));
4184d0812dedSMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), tmpfilename, plexname, interpolate, &dmnew));
4185ed32af8cSMatthew G. Knepley     PetscCall(PetscRMTree(tmpdir));
4186d0812dedSMatthew G. Knepley     PetscCall(PetscSNPrintf(name, PETSC_MAX_PATH_LEN, "%s Mesh", strname));
4187d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
4188d0812dedSMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
4189d0812dedSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
41909318fe57SMatthew G. Knepley   } else {
41919566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)dm, DMPlexShapes[shape]));
41929318fe57SMatthew G. Knepley     switch (shape) {
4193669647acSMatthew G. Knepley     case DM_SHAPE_BOX:
41945dca41c3SJed Brown     case DM_SHAPE_ZBOX:
4195669647acSMatthew G. Knepley     case DM_SHAPE_ANNULUS: {
41969318fe57SMatthew G. Knepley       PetscInt       faces[3]  = {0, 0, 0};
41979318fe57SMatthew G. Knepley       PetscReal      lower[3]  = {0, 0, 0};
41989318fe57SMatthew G. Knepley       PetscReal      upper[3]  = {1, 1, 1};
41999318fe57SMatthew G. Knepley       DMBoundaryType bdt[3]    = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4200669647acSMatthew G. Knepley       PetscBool      isAnnular = shape == DM_SHAPE_ANNULUS ? PETSC_TRUE : PETSC_FALSE;
42019318fe57SMatthew G. Knepley       PetscInt       i, n;
42029318fe57SMatthew G. Knepley 
42039318fe57SMatthew G. Knepley       n = dim;
42049318fe57SMatthew G. Knepley       for (i = 0; i < dim; ++i) faces[i] = (dim == 1 ? 1 : 4 - dim);
42059566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
42069318fe57SMatthew G. Knepley       n = 3;
42079566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
420863a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
42099318fe57SMatthew G. Knepley       n = 3;
42109566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
421163a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
42129318fe57SMatthew G. Knepley       n = 3;
42139566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
421463a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4215669647acSMatthew G. Knepley 
4216669647acSMatthew G. Knepley       PetscCheck(!isAnnular || dim == 2, comm, PETSC_ERR_ARG_OUTOFRANGE, "Only two dimensional annuli have been implemented");
4217669647acSMatthew G. Knepley       if (isAnnular)
4218669647acSMatthew G. Knepley         for (i = 0; i < dim - 1; ++i) bdt[i] = DM_BOUNDARY_PERIODIC;
4219669647acSMatthew G. Knepley 
42209318fe57SMatthew G. Knepley       switch (cell) {
422161a622f3SMatthew G. Knepley       case DM_POLYTOPE_TRI_PRISM_TENSOR:
42229566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeBoxMesh_Internal(dm, faces, lower, upper, bdt));
4223d410b0cfSMatthew G. Knepley         if (!interpolate) {
4224d410b0cfSMatthew G. Knepley           DM udm;
4225d410b0cfSMatthew G. Knepley 
42269566063dSJacob Faibussowitsch           PetscCall(DMPlexUninterpolate(dm, &udm));
422769d8a87bSksagiyam           PetscCall(DMPlexReplace_Internal(dm, &udm));
4228d410b0cfSMatthew G. Knepley         }
42299318fe57SMatthew G. Knepley         break;
4230d71ae5a4SJacob Faibussowitsch       default:
42315dca41c3SJed Brown         PetscCall(DMPlexCreateBoxMesh_Internal(dm, shape, dim, simplex, faces, lower, upper, bdt, interpolate));
4232d71ae5a4SJacob Faibussowitsch         break;
42339318fe57SMatthew G. Knepley       }
4234669647acSMatthew G. Knepley       if (isAnnular) {
4235669647acSMatthew G. Knepley         DM          cdm;
4236669647acSMatthew G. Knepley         PetscDS     cds;
4237669647acSMatthew G. Knepley         PetscScalar bounds[2] = {lower[0], upper[0]};
4238669647acSMatthew G. Knepley 
4239669647acSMatthew G. Knepley         // Fix coordinates for annular region
4240669647acSMatthew G. Knepley         PetscCall(DMSetPeriodicity(dm, NULL, NULL, NULL));
4241669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinatesLocal(dm, NULL));
4242669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinates(dm, NULL));
4243e44f6aebSMatthew G. Knepley         PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
4244669647acSMatthew G. Knepley         PetscCall(DMGetCoordinateDM(dm, &cdm));
4245669647acSMatthew G. Knepley         PetscCall(DMGetDS(cdm, &cds));
4246669647acSMatthew G. Knepley         PetscCall(PetscDSSetConstants(cds, 2, bounds));
4247669647acSMatthew G. Knepley         PetscCall(DMPlexRemapGeometry(dm, 0.0, boxToAnnulus));
4248669647acSMatthew G. Knepley       }
42499371c9d4SSatish Balay     } break;
42509371c9d4SSatish Balay     case DM_SHAPE_BOX_SURFACE: {
42519318fe57SMatthew G. Knepley       PetscInt  faces[3] = {0, 0, 0};
42529318fe57SMatthew G. Knepley       PetscReal lower[3] = {0, 0, 0};
42539318fe57SMatthew G. Knepley       PetscReal upper[3] = {1, 1, 1};
42549318fe57SMatthew G. Knepley       PetscInt  i, n;
42559318fe57SMatthew G. Knepley 
42569318fe57SMatthew G. Knepley       n = dim + 1;
42579318fe57SMatthew G. Knepley       for (i = 0; i < dim + 1; ++i) faces[i] = (dim + 1 == 1 ? 1 : 4 - (dim + 1));
42589566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
42599318fe57SMatthew G. Knepley       n = 3;
42609566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
426163a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim + 1), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim + 1);
42629318fe57SMatthew G. Knepley       n = 3;
42639566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
426463a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim + 1), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim + 1);
42659566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(dm, dim + 1, faces, lower, upper, interpolate));
42669371c9d4SSatish Balay     } break;
42679371c9d4SSatish Balay     case DM_SHAPE_SPHERE: {
42689318fe57SMatthew G. Knepley       PetscReal R = 1.0;
42699318fe57SMatthew G. Knepley 
42709566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_sphere_radius", "Radius of the sphere", "", R, &R, &flg));
42719566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateSphereMesh_Internal(dm, dim, simplex, R));
42729371c9d4SSatish Balay     } break;
42739371c9d4SSatish Balay     case DM_SHAPE_BALL: {
42749318fe57SMatthew G. Knepley       PetscReal R = 1.0;
42759318fe57SMatthew G. Knepley 
42769566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_ball_radius", "Radius of the ball", "", R, &R, &flg));
42779566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBallMesh_Internal(dm, dim, R));
42789371c9d4SSatish Balay     } break;
42799371c9d4SSatish Balay     case DM_SHAPE_CYLINDER: {
42809318fe57SMatthew G. Knepley       DMBoundaryType bdt = DM_BOUNDARY_NONE;
42819318fe57SMatthew G. Knepley       PetscInt       Nw  = 6;
428249704ca5SMatthew G. Knepley       PetscInt       Nr  = 0;
42839318fe57SMatthew G. Knepley 
42849566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnum("-dm_plex_cylinder_bd", "Boundary type in the z direction", "", DMBoundaryTypes, (PetscEnum)bdt, (PetscEnum *)&bdt, NULL));
42859566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_wedges", "Number of wedges around the cylinder", "", Nw, &Nw, NULL));
428649704ca5SMatthew G. Knepley       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_refine", "Number of refinements before projection", "", Nr, &Nr, NULL));
42879318fe57SMatthew G. Knepley       switch (cell) {
4288d71ae5a4SJacob Faibussowitsch       case DM_POLYTOPE_TRI_PRISM_TENSOR:
4289d71ae5a4SJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(dm, Nw, interpolate));
4290d71ae5a4SJacob Faibussowitsch         break;
4291d71ae5a4SJacob Faibussowitsch       default:
429249704ca5SMatthew G. Knepley         PetscCall(DMPlexCreateHexCylinderMesh_Internal(dm, bdt, Nr));
4293d71ae5a4SJacob Faibussowitsch         break;
42949318fe57SMatthew G. Knepley       }
42959371c9d4SSatish Balay     } break;
4296b7f5c055SJed Brown     case DM_SHAPE_SCHWARZ_P: // fallthrough
42979371c9d4SSatish Balay     case DM_SHAPE_GYROID: {
4298b7f5c055SJed Brown       PetscInt       extent[3] = {1, 1, 1}, refine = 0, layers = 0, three;
4299b7f5c055SJed Brown       PetscReal      thickness   = 0.;
4300b7f5c055SJed Brown       DMBoundaryType periodic[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4301b7f5c055SJed Brown       DMPlexTPSType  tps_type    = shape == DM_SHAPE_SCHWARZ_P ? DMPLEX_TPS_SCHWARZ_P : DMPLEX_TPS_GYROID;
43021436d7faSJed Brown       PetscBool      tps_distribute;
43039566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_tps_extent", "Number of replicas for each of three dimensions", NULL, extent, (three = 3, &three), NULL));
43049566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_refine", "Number of refinements", NULL, refine, &refine, NULL));
43059566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_tps_periodic", "Periodicity in each of three dimensions", NULL, DMBoundaryTypes, (PetscEnum *)periodic, (three = 3, &three), NULL));
43069566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_layers", "Number of layers in volumetric extrusion (or zero to not extrude)", NULL, layers, &layers, NULL));
43079566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_tps_thickness", "Thickness of volumetric extrusion", NULL, thickness, &thickness, NULL));
43089566063dSJacob Faibussowitsch       PetscCall(DMPlexDistributeGetDefault(dm, &tps_distribute));
43099566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_tps_distribute", "Distribute the 2D mesh prior to refinement and extrusion", NULL, tps_distribute, &tps_distribute, NULL));
43109566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateTPSMesh_Internal(dm, tps_type, extent, periodic, tps_distribute, refine, layers, thickness));
43119371c9d4SSatish Balay     } break;
43129371c9d4SSatish Balay     case DM_SHAPE_DOUBLET: {
431305bd46c0SStefano Zampini       DM        dmnew;
431405bd46c0SStefano Zampini       PetscReal rl = 0.0;
431505bd46c0SStefano Zampini 
431605bd46c0SStefano Zampini       PetscCall(PetscOptionsReal("-dm_plex_doublet_refinementlimit", "Refinement limit", NULL, rl, &rl, NULL));
431705bd46c0SStefano Zampini       PetscCall(DMPlexCreateDoublet(PetscObjectComm((PetscObject)dm), dim, simplex, interpolate, rl, &dmnew));
43185de52c6dSVaclav Hapla       PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
431969d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &dmnew));
43209371c9d4SSatish Balay     } break;
4321cfb853baSMatthew G. Knepley     case DM_SHAPE_HYPERCUBIC: {
4322cfb853baSMatthew G. Knepley       PetscInt       *edges;
4323cfb853baSMatthew G. Knepley       PetscReal      *lower, *upper;
4324cfb853baSMatthew G. Knepley       DMBoundaryType *bdt;
4325cfb853baSMatthew G. Knepley       PetscInt        n, d;
4326cfb853baSMatthew G. Knepley 
4327cfb853baSMatthew G. Knepley       *useCoordSpace = PETSC_FALSE;
4328cfb853baSMatthew G. Knepley       PetscCall(PetscMalloc4(dim, &edges, dim, &lower, dim, &upper, dim, &bdt));
4329cfb853baSMatthew G. Knepley       for (d = 0; d < dim; ++d) {
4330cfb853baSMatthew G. Knepley         edges[d] = 1;
4331cfb853baSMatthew G. Knepley         lower[d] = 0.;
4332cfb853baSMatthew G. Knepley         upper[d] = 1.;
4333cfb853baSMatthew G. Knepley         bdt[d]   = DM_BOUNDARY_PERIODIC;
4334cfb853baSMatthew G. Knepley       }
4335cfb853baSMatthew G. Knepley       n = dim;
4336cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", edges, &n, &flg));
4337cfb853baSMatthew G. Knepley       n = dim;
4338cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
4339cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4340cfb853baSMatthew G. Knepley       n = dim;
4341cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
4342cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4343cfb853baSMatthew G. Knepley       n = dim;
4344cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
4345cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4346cfb853baSMatthew G. Knepley       PetscCall(DMPlexCreateHypercubicMesh_Internal(dm, dim, lower, upper, edges, bdt));
4347cfb853baSMatthew G. Knepley       PetscCall(PetscFree4(edges, lower, upper, bdt));
4348cfb853baSMatthew G. Knepley     } break;
4349d71ae5a4SJacob Faibussowitsch     default:
4350d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_SUP, "Domain shape %s is unsupported", DMPlexShapes[shape]);
43519318fe57SMatthew G. Knepley     }
43529318fe57SMatthew G. Knepley   }
43539566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
435448a46eb9SPierre Jolivet   if (!((PetscObject)dm)->name && nameflg) PetscCall(PetscObjectSetName((PetscObject)dm, plexname));
43554e22dd4cSMatthew G. Knepley   // Allow label creation
43564e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_label_", &option, NULL, &flg));
43574e22dd4cSMatthew G. Knepley   if (flg) {
43584e22dd4cSMatthew G. Knepley     DMLabel     label;
43594e22dd4cSMatthew G. Knepley     PetscInt    points[1024], n = 1024;
43604e22dd4cSMatthew G. Knepley     char        fulloption[PETSC_MAX_PATH_LEN];
43614e22dd4cSMatthew G. Knepley     const char *name = &option[14];
43624e22dd4cSMatthew G. Knepley 
43634e22dd4cSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, name));
43644e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, name, &label));
43654e22dd4cSMatthew G. Knepley     fulloption[0] = '-';
43664e22dd4cSMatthew G. Knepley     fulloption[1] = 0;
43674e22dd4cSMatthew G. Knepley     PetscCall(PetscStrlcat(fulloption, option, PETSC_MAX_PATH_LEN));
43684e22dd4cSMatthew G. Knepley     PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, NULL));
43694e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < n; ++p) PetscCall(DMLabelSetValue(label, points[p], 1));
43704e22dd4cSMatthew G. Knepley   }
4371dd0eeac9SMatthew G. Knepley   // Allow cohesive label creation
4372dd0eeac9SMatthew G. Knepley   //   Faces are input, completed, and all points are marked with their depth
4373dd0eeac9SMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_cohesive_label_", &option, NULL, &flg));
4374dd0eeac9SMatthew G. Knepley   if (flg) {
4375dd0eeac9SMatthew G. Knepley     DMLabel   label;
4376dd0eeac9SMatthew G. Knepley     PetscInt  points[1024], n, pStart, pEnd, Nl = 1;
43775390be7dSMatthew G. Knepley     PetscBool noCreate = PETSC_FALSE;
4378dd0eeac9SMatthew G. Knepley     char      fulloption[PETSC_MAX_PATH_LEN];
4379dd0eeac9SMatthew G. Knepley     char      name[PETSC_MAX_PATH_LEN];
4380dd0eeac9SMatthew G. Knepley     size_t    len;
4381dd0eeac9SMatthew G. Knepley 
4382dd0eeac9SMatthew G. Knepley     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4383dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrncpy(name, &option[23], PETSC_MAX_PATH_LEN));
4384dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrlen(name, &len));
4385dd0eeac9SMatthew G. Knepley     if (name[len - 1] == '0') Nl = 10;
4386dd0eeac9SMatthew G. Knepley     for (PetscInt l = 0; l < Nl; ++l) {
43876497c311SBarry Smith       if (l > 0) name[len - 1] = (char)('0' + l);
4388dd0eeac9SMatthew G. Knepley       fulloption[0] = 0;
4389dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, "-dm_plex_cohesive_label_", 32));
4390dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, name, PETSC_MAX_PATH_LEN - 32));
4391dd0eeac9SMatthew G. Knepley       n = 1024;
4392dd0eeac9SMatthew G. Knepley       PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, &flg));
4393dd0eeac9SMatthew G. Knepley       if (!flg) break;
43945390be7dSMatthew G. Knepley       PetscCall(DMHasLabel(dm, name, &noCreate));
43955390be7dSMatthew G. Knepley       if (noCreate) {
43965390be7dSMatthew G. Knepley         DMLabel         inlabel;
43975390be7dSMatthew G. Knepley         IS              pointIS;
43985390be7dSMatthew G. Knepley         const PetscInt *lpoints;
43995390be7dSMatthew G. Knepley         PetscInt        pdep, ln, inval = points[0];
44005390be7dSMatthew G. Knepley         char            newname[PETSC_MAX_PATH_LEN];
44015390be7dSMatthew G. Knepley 
44025390be7dSMatthew G. Knepley         PetscCheck(n == 1, comm, PETSC_ERR_ARG_WRONG, "Must specify a label value with this option");
44035390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &inlabel));
44045390be7dSMatthew G. Knepley         PetscCall(DMLabelGetStratumIS(inlabel, inval, &pointIS));
44055390be7dSMatthew G. Knepley         PetscCall(ISGetLocalSize(pointIS, &ln));
44065390be7dSMatthew G. Knepley         PetscCall(ISGetIndices(pointIS, &lpoints));
44075390be7dSMatthew G. Knepley         PetscCall(DMPlexGetPointDepth(dm, lpoints[0], &pdep));
44085390be7dSMatthew G. Knepley         PetscCall(PetscSNPrintf(newname, PETSC_MAX_PATH_LEN, "%s%" PetscInt_FMT, name, points[0]));
44095390be7dSMatthew G. Knepley         PetscCall(DMCreateLabel(dm, newname));
44105390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, newname, &label));
44115390be7dSMatthew G. Knepley         if (!pdep) PetscCall(ProcessCohesiveLabel_Vertices(dm, label, inlabel, inval, ln, lpoints));
44125390be7dSMatthew G. Knepley         else PetscCall(ProcessCohesiveLabel_Faces(dm, label, ln, lpoints));
44135390be7dSMatthew G. Knepley         PetscCall(ISRestoreIndices(pointIS, &lpoints));
44145390be7dSMatthew G. Knepley         PetscCall(ISDestroy(&pointIS));
44155390be7dSMatthew G. Knepley       } else {
4416dd0eeac9SMatthew G. Knepley         PetscCall(DMCreateLabel(dm, name));
4417dd0eeac9SMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &label));
4418dd0eeac9SMatthew G. Knepley         if (pStart >= pEnd) n = 0;
44195390be7dSMatthew G. Knepley         PetscCall(ProcessCohesiveLabel_Faces(dm, label, n, points));
4420dd0eeac9SMatthew G. Knepley       }
4421dd0eeac9SMatthew G. Knepley       PetscCall(DMPlexOrientLabel(dm, label));
44220542aa8cSMatthew G. Knepley       PetscCall(DMPlexLabelCohesiveComplete(dm, label, NULL, 1, PETSC_FALSE, PETSC_FALSE, NULL));
4423dd0eeac9SMatthew G. Knepley     }
4424dd0eeac9SMatthew G. Knepley   }
44255390be7dSMatthew G. Knepley   PetscCall(DMViewFromOptions(dm, NULL, "-created_dm_view"));
4426708be2fdSJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
44273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44280a6ba040SMatthew G. Knepley }
44290a6ba040SMatthew G. Knepley 
4430d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_NonRefinement_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4431d71ae5a4SJacob Faibussowitsch {
44320a6ba040SMatthew G. Knepley   DM_Plex  *mesh = (DM_Plex *)dm->data;
44337f9d8d6cSVaclav Hapla   PetscBool flg, flg2;
44349318fe57SMatthew G. Knepley   char      bdLabel[PETSC_MAX_PATH_LEN];
4435adc21957SMatthew G. Knepley   char      method[PETSC_MAX_PATH_LEN];
44360a6ba040SMatthew G. Knepley 
44370a6ba040SMatthew G. Knepley   PetscFunctionBegin;
44380a6ba040SMatthew G. Knepley   /* Handle viewing */
44399566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMPlexMatSetClosure", PETSC_FALSE, &mesh->printSetValues, NULL));
44405962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fem", "Debug output level for all fem computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printFEM, NULL, 0));
44415962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fvm", "Debug output level for all fvm computations", "DMPlexSNESComputeResidualFVM", 0, &mesh->printFVM, NULL, 0));
44429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMPlexSNESComputeResidualFEM", mesh->printTol, &mesh->printTol, NULL));
44439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_l2", "Debug output level all L2 diff computations", "DMComputeL2Diff", 0, &mesh->printL2, NULL, 0));
4444f5867de0SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_locate", "Debug output level all point location computations", "DMLocatePoints", 0, &mesh->printLocate, NULL, 0));
4445a77a5016SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_project", "Debug output level all projection computations", "DMPlexProject", 0, &mesh->printProject, NULL, 0));
44469566063dSJacob Faibussowitsch   PetscCall(DMMonitorSetFromOptions(dm, "-dm_plex_monitor_throughput", "Monitor the simulation throughput", "DMPlexMonitorThroughput", DMPlexMonitorThroughput, NULL, &flg));
44479566063dSJacob Faibussowitsch   if (flg) PetscCall(PetscLogDefaultBegin());
44489318fe57SMatthew G. Knepley   /* Labeling */
44499566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_label", "Label to mark the mesh boundary", "", bdLabel, bdLabel, sizeof(bdLabel), &flg));
44509566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexCreateBoundaryLabel_Private(dm, bdLabel));
4451953fc75cSMatthew G. Knepley   /* Point Location */
44529566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMInterpolate", PETSC_FALSE, &mesh->useHashLocation, NULL));
44530848f4b5SMatthew G. Knepley   /* Partitioning and distribution */
44549566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", PETSC_FALSE, &mesh->partitionBalance, NULL));
4455d02c7345SMatthew G. Knepley   /* Reordering */
4456adc21957SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_reorder_section", "Compute point permutation for local section", "DMReorderSectionSetDefault", PETSC_FALSE, &flg2, &flg));
4457adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetDefault(dm, flg2 ? DM_REORDER_DEFAULT_TRUE : DM_REORDER_DEFAULT_FALSE));
4458adc21957SMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_reorder_section_type", "Reordering method for local section", "DMReorderSectionSetType", method, method, PETSC_MAX_PATH_LEN, &flg));
4459adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetType(dm, method));
44602e62ab5aSMatthew G. Knepley   /* Generation and remeshing */
44619566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_remesh_bd", "Allow changes to the boundary on remeshing", "DMAdapt", PETSC_FALSE, &mesh->remeshBd, NULL));
4462b29cfa1cSToby Isaac   /* Projection behavior */
4463d5b43468SJose E. Roman   PetscCall(PetscOptionsBoundedInt("-dm_plex_max_projection_height", "Maximum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL, 0));
44649566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_regular_refinement", "Use special nested projection algorithm for regular refinement", "DMPlexSetRegularRefinement", mesh->regularRefinement, &mesh->regularRefinement, NULL));
4465f12cf164SMatthew G. Knepley   /* Checking structure */
4466f12cf164SMatthew G. Knepley   {
44677f9d8d6cSVaclav Hapla     PetscBool all = PETSC_FALSE;
4468f12cf164SMatthew G. Knepley 
44697f9d8d6cSVaclav Hapla     PetscCall(PetscOptionsBool("-dm_plex_check_all", "Perform all basic checks", "DMPlexCheck", PETSC_FALSE, &all, NULL));
44707f9d8d6cSVaclav Hapla     if (all) {
44717f9d8d6cSVaclav Hapla       PetscCall(DMPlexCheck(dm));
44727f9d8d6cSVaclav Hapla     } else {
44739566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_symmetry", "Check that the adjacency information in the mesh is symmetric", "DMPlexCheckSymmetry", PETSC_FALSE, &flg, &flg2));
44747f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSymmetry(dm));
44759566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_skeleton", "Check that each cell has the correct number of vertices (only for homogeneous simplex or tensor meshes)", "DMPlexCheckSkeleton", PETSC_FALSE, &flg, &flg2));
44767f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSkeleton(dm, 0));
44779566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_faces", "Check that the faces of each cell give a vertex order this is consistent with what we expect from the cell type", "DMPlexCheckFaces", PETSC_FALSE, &flg, &flg2));
44787f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckFaces(dm, 0));
44799566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_geometry", "Check that cells have positive volume", "DMPlexCheckGeometry", PETSC_FALSE, &flg, &flg2));
44807f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckGeometry(dm));
44819566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_pointsf", "Check some necessary conditions for PointSF", "DMPlexCheckPointSF", PETSC_FALSE, &flg, &flg2));
4482d7d32a9aSMatthew G. Knepley       if (flg && flg2) PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE));
44839566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_interface_cones", "Check points on inter-partition interfaces have conforming order of cone points", "DMPlexCheckInterfaceCones", PETSC_FALSE, &flg, &flg2));
44847f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckInterfaceCones(dm));
44857f9d8d6cSVaclav Hapla     }
44869566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_plex_check_cell_shape", "Check cell shape", "DMPlexCheckCellShape", PETSC_FALSE, &flg, &flg2));
44879566063dSJacob Faibussowitsch     if (flg && flg2) PetscCall(DMPlexCheckCellShape(dm, PETSC_TRUE, PETSC_DETERMINE));
4488f12cf164SMatthew G. Knepley   }
44899318fe57SMatthew G. Knepley   {
44909318fe57SMatthew G. Knepley     PetscReal scale = 1.0;
44914f3833eaSMatthew G. Knepley 
44929566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-dm_plex_scale", "Scale factor for mesh coordinates", "DMPlexScale", scale, &scale, &flg));
44939318fe57SMatthew G. Knepley     if (flg) {
44949318fe57SMatthew G. Knepley       Vec coordinates, coordinatesLocal;
44959318fe57SMatthew G. Knepley 
44969566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(dm, &coordinates));
44979566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal));
44989566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinates, scale));
44999566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinatesLocal, scale));
45009318fe57SMatthew G. Knepley     }
45019318fe57SMatthew G. Knepley   }
45029566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerSetFromOptions(mesh->partitioner));
45033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
450468d4fef7SMatthew G. Knepley }
450568d4fef7SMatthew G. Knepley 
4506d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_Overlap_Plex(DM dm, PetscOptionItems *PetscOptionsObject, PetscInt *overlap)
4507d71ae5a4SJacob Faibussowitsch {
4508c506a872SMatthew G. Knepley   PetscInt  numOvLabels = 16, numOvExLabels = 16;
4509c506a872SMatthew G. Knepley   char     *ovLabelNames[16], *ovExLabelNames[16];
4510c506a872SMatthew G. Knepley   PetscInt  numOvValues = 16, numOvExValues = 16, l;
4511c506a872SMatthew G. Knepley   PetscBool flg;
4512c506a872SMatthew G. Knepley 
4513c506a872SMatthew G. Knepley   PetscFunctionBegin;
4514c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", *overlap, overlap, NULL, 0));
4515c506a872SMatthew G. Knepley   PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_labels", "List of overlap label names", "DMPlexDistribute", ovLabelNames, &numOvLabels, &flg));
4516c506a872SMatthew G. Knepley   if (!flg) numOvLabels = 0;
4517c506a872SMatthew G. Knepley   if (numOvLabels) {
4518c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvLabels = numOvLabels;
4519c506a872SMatthew G. Knepley     for (l = 0; l < numOvLabels; ++l) {
4520c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovLabelNames[l], &((DM_Plex *)dm->data)->ovLabels[l]));
4521c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovLabelNames[l]);
4522c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovLabelNames[l]));
4523c506a872SMatthew G. Knepley     }
4524c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_values", "List of overlap label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovValues, &numOvValues, &flg));
4525c506a872SMatthew G. Knepley     if (!flg) numOvValues = 0;
4526c506a872SMatthew G. Knepley     PetscCheck(numOvLabels == numOvValues, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "The number of labels %" PetscInt_FMT " must match the number of values %" PetscInt_FMT, numOvLabels, numOvValues);
4527c506a872SMatthew G. Knepley 
4528c506a872SMatthew G. Knepley     PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_exclude_labels", "List of overlap exclude label names", "DMPlexDistribute", ovExLabelNames, &numOvExLabels, &flg));
4529c506a872SMatthew G. Knepley     if (!flg) numOvExLabels = 0;
4530c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvExLabels = numOvExLabels;
4531c506a872SMatthew G. Knepley     for (l = 0; l < numOvExLabels; ++l) {
4532c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovExLabelNames[l], &((DM_Plex *)dm->data)->ovExLabels[l]));
4533c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovExLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovExLabelNames[l]);
4534c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovExLabelNames[l]));
4535c506a872SMatthew G. Knepley     }
4536c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_exclude_values", "List of overlap exclude label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovExValues, &numOvExValues, &flg));
4537c506a872SMatthew G. Knepley     if (!flg) numOvExValues = 0;
4538c506a872SMatthew G. Knepley     PetscCheck(numOvExLabels == numOvExValues, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "The number of exclude labels %" PetscInt_FMT " must match the number of values %" PetscInt_FMT, numOvExLabels, numOvExValues);
4539c506a872SMatthew G. Knepley   }
45403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4541c506a872SMatthew G. Knepley }
4542c506a872SMatthew G. Knepley 
4543d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMSetFromOptions_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4544d71ae5a4SJacob Faibussowitsch {
4545bdf63967SMatthew G. Knepley   PetscFunctionList    ordlist;
4546bdf63967SMatthew G. Knepley   char                 oname[256];
45474e22dd4cSMatthew G. Knepley   char                 sublabelname[PETSC_MAX_PATH_LEN] = "";
4548adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
4549d410b0cfSMatthew G. Knepley   PetscReal            volume    = -1.0;
45509318fe57SMatthew G. Knepley   PetscInt             prerefine = 0, refine = 0, r, coarsen = 0, overlap = 0, extLayers = 0, dim;
45511b742f01SMatthew G. Knepley   PetscBool            uniformOrig = PETSC_FALSE, created = PETSC_FALSE, uniform = PETSC_TRUE, distribute, saveSF = PETSC_FALSE, interpolate = PETSC_TRUE, coordSpace = PETSC_TRUE, remap = PETSC_TRUE, ghostCells = PETSC_FALSE, isHierarchy, flg;
455268d4fef7SMatthew G. Knepley 
455368d4fef7SMatthew G. Knepley   PetscFunctionBegin;
4554d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "DMPlex Options");
4555dd4c3f67SMatthew G. Knepley   if (dm->cloneOpts) goto non_refine;
45569318fe57SMatthew G. Knepley   /* Handle automatic creation */
45579566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
45586bc1bd01Sksagiyam   if (dim < 0) {
45596bc1bd01Sksagiyam     PetscCall(DMPlexCreateFromOptions_Internal(PetscOptionsObject, &coordSpace, dm));
45606bc1bd01Sksagiyam     created = PETSC_TRUE;
45616bc1bd01Sksagiyam   }
45626bc1bd01Sksagiyam   PetscCall(DMGetDimension(dm, &dim));
4563d89e6e46SMatthew G. Knepley   /* Handle interpolation before distribution */
45649566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate_pre", "Flag to interpolate mesh before distribution", "", interpolate, &interpolate, &flg));
4565d89e6e46SMatthew G. Knepley   if (flg) {
4566d89e6e46SMatthew G. Knepley     DMPlexInterpolatedFlag interpolated;
4567d89e6e46SMatthew G. Knepley 
45689566063dSJacob Faibussowitsch     PetscCall(DMPlexIsInterpolated(dm, &interpolated));
4569d89e6e46SMatthew G. Knepley     if (interpolated == DMPLEX_INTERPOLATED_FULL && !interpolate) {
4570d89e6e46SMatthew G. Knepley       DM udm;
4571d89e6e46SMatthew G. Knepley 
45729566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(dm, &udm));
457369d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &udm));
4574d89e6e46SMatthew G. Knepley     } else if (interpolated != DMPLEX_INTERPOLATED_FULL && interpolate) {
4575d89e6e46SMatthew G. Knepley       DM idm;
4576d89e6e46SMatthew G. Knepley 
45779566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(dm, &idm));
457869d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &idm));
4579d89e6e46SMatthew G. Knepley     }
4580d89e6e46SMatthew G. Knepley   }
45814e22dd4cSMatthew G. Knepley   // Handle submesh selection before distribution
45824e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_submesh", "Label to use for submesh selection", "", sublabelname, sublabelname, PETSC_MAX_PATH_LEN, &flg));
45834e22dd4cSMatthew G. Knepley   if (flg) {
45844e22dd4cSMatthew G. Knepley     DM              subdm;
45854e22dd4cSMatthew G. Knepley     DMLabel         label;
45864e22dd4cSMatthew G. Knepley     IS              valueIS, pointIS;
45874e22dd4cSMatthew G. Knepley     const PetscInt *values, *points;
45884e22dd4cSMatthew G. Knepley     PetscBool       markedFaces = PETSC_FALSE;
45894e22dd4cSMatthew G. Knepley     PetscInt        Nv, value, Np;
45904e22dd4cSMatthew G. Knepley 
45914e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, sublabelname, &label));
45924e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetNumValues(label, &Nv));
45934e22dd4cSMatthew G. Knepley     PetscCheck(Nv == 1, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Only a single label value is currently supported for submesh selection, not %" PetscInt_FMT, Nv);
45944e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetValueIS(label, &valueIS));
45954e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(valueIS, &values));
45964e22dd4cSMatthew G. Knepley     value = values[0];
45974e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(valueIS, &values));
45984e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&valueIS));
45994e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumSize(label, value, &Np));
46004e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumIS(label, value, &pointIS));
46014e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(pointIS, &points));
46024e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < Np; ++p) {
46034e22dd4cSMatthew G. Knepley       PetscInt pdepth;
46044e22dd4cSMatthew G. Knepley 
46054e22dd4cSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, points[p], &pdepth));
46064e22dd4cSMatthew G. Knepley       if (pdepth) {
46074e22dd4cSMatthew G. Knepley         markedFaces = PETSC_TRUE;
46084e22dd4cSMatthew G. Knepley         break;
46094e22dd4cSMatthew G. Knepley       }
46104e22dd4cSMatthew G. Knepley     }
46114e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(pointIS, &points));
46124e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&pointIS));
46134e22dd4cSMatthew G. Knepley     PetscCall(DMPlexCreateSubmesh(dm, label, value, markedFaces, &subdm));
46144e22dd4cSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &subdm));
46154e22dd4cSMatthew G. Knepley     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
46164e22dd4cSMatthew G. Knepley   }
46179b44eab4SMatthew G. Knepley   /* Handle DMPlex refinement before distribution */
46189566063dSJacob Faibussowitsch   PetscCall(DMPlexGetRefinementUniform(dm, &uniformOrig));
46199566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_pre", "The number of refinements before distribution", "DMCreate", prerefine, &prerefine, NULL, 0));
46209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap_pre", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
46219566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_uniform_pre", "Flag for uniform refinement before distribution", "DMCreate", uniform, &uniform, &flg));
46229566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexSetRefinementUniform(dm, uniform));
46239566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_refine_volume_limit_pre", "The maximum cell volume after refinement before distribution", "DMCreate", volume, &volume, &flg));
46249318fe57SMatthew G. Knepley   if (flg) {
46259566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(dm, PETSC_FALSE));
46269566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(dm, volume));
46279318fe57SMatthew G. Knepley     prerefine = PetscMax(prerefine, 1);
46289318fe57SMatthew G. Knepley   }
4629b23db253SStefano Zampini   if (prerefine) PetscCall(DMLocalizeCoordinates(dm));
46309b44eab4SMatthew G. Knepley   for (r = 0; r < prerefine; ++r) {
46319b44eab4SMatthew G. Knepley     DM             rdm;
46329b44eab4SMatthew G. Knepley     PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
46339b44eab4SMatthew G. Knepley 
4634dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
46359566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
463669d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &rdm));
4637dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
463861a622f3SMatthew G. Knepley     if (coordFunc && remap) {
46399566063dSJacob Faibussowitsch       PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
46409b44eab4SMatthew G. Knepley       ((DM_Plex *)dm->data)->coordFunc = coordFunc;
46419b44eab4SMatthew G. Knepley     }
46429b44eab4SMatthew G. Knepley   }
46439566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, uniformOrig));
46449318fe57SMatthew G. Knepley   /* Handle DMPlex extrusion before distribution */
46459566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_extrude", "The number of layers to extrude", "", extLayers, &extLayers, NULL, 0));
46469318fe57SMatthew G. Knepley   if (extLayers) {
46479318fe57SMatthew G. Knepley     DM edm;
46489318fe57SMatthew G. Knepley 
46499566063dSJacob Faibussowitsch     PetscCall(DMExtrude(dm, extLayers, &edm));
465069d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
465148d16a33SMatthew G. Knepley     ((DM_Plex *)dm->data)->coordFunc = NULL;
4652dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4653d410b0cfSMatthew G. Knepley     extLayers = 0;
46545e17fc22SAidan Hamilton     PetscCall(DMGetDimension(dm, &dim));
46559318fe57SMatthew G. Knepley   }
4656bdf63967SMatthew G. Knepley   /* Handle DMPlex reordering before distribution */
46576bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dm, &reorder));
46589566063dSJacob Faibussowitsch   PetscCall(MatGetOrderingList(&ordlist));
46596bc1bd01Sksagiyam   PetscCall(PetscStrncpy(oname, MATORDERINGNATURAL, sizeof(oname)));
46609566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-dm_plex_reorder", "Set mesh reordering type", "DMPlexGetOrdering", ordlist, MATORDERINGNATURAL, oname, sizeof(oname), &flg));
4661adc21957SMatthew G. Knepley   if (reorder == DM_REORDER_DEFAULT_TRUE || flg) {
4662bdf63967SMatthew G. Knepley     DM pdm;
4663bdf63967SMatthew G. Knepley     IS perm;
4664bdf63967SMatthew G. Knepley 
46659566063dSJacob Faibussowitsch     PetscCall(DMPlexGetOrdering(dm, oname, NULL, &perm));
46669566063dSJacob Faibussowitsch     PetscCall(DMPlexPermute(dm, perm, &pdm));
46679566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&perm));
466869d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &pdm));
4669dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4670bdf63967SMatthew G. Knepley   }
46719b44eab4SMatthew G. Knepley   /* Handle DMPlex distribution */
46729566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dm, &distribute));
4673c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute", "Flag to redistribute a mesh among processes", "DMPlexDistribute", distribute, &distribute, NULL));
4674a286e215SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute_save_sf", "Flag to save the migration SF", "DMPlexSetMigrationSF", saveSF, &saveSF, NULL));
4675dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_Overlap_Plex(dm, PetscOptionsObject, &overlap));
46769b44eab4SMatthew G. Knepley   if (distribute) {
46779b44eab4SMatthew G. Knepley     DM               pdm = NULL;
46789b44eab4SMatthew G. Knepley     PetscPartitioner part;
4679a286e215SMatthew G. Knepley     PetscSF          sfMigration;
46809b44eab4SMatthew G. Knepley 
46819566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
46829566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
4683a286e215SMatthew G. Knepley     PetscCall(DMPlexDistribute(dm, overlap, &sfMigration, &pdm));
468448a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
4685a286e215SMatthew G. Knepley     if (saveSF) PetscCall(DMPlexSetMigrationSF(dm, sfMigration));
4686a286e215SMatthew G. Knepley     PetscCall(PetscSFDestroy(&sfMigration));
46879b44eab4SMatthew G. Knepley   }
46884054ae39SJames Wright 
46894054ae39SJames Wright   {
46904054ae39SJames Wright     PetscBool useBoxLabel = PETSC_FALSE;
46914054ae39SJames Wright     PetscCall(PetscOptionsBool("-dm_plex_box_label", "Create 'Face Sets' assuming boundary faces align with cartesian directions", "DMCreate", useBoxLabel, &useBoxLabel, NULL));
46924054ae39SJames Wright     if (useBoxLabel) PetscCall(DMPlexSetBoxLabel_Internal(dm));
46934054ae39SJames Wright   }
4694d2b2dc1eSMatthew G. Knepley   /* Must check CEED options before creating function space for coordinates */
4695d2b2dc1eSMatthew G. Knepley   {
4696d2b2dc1eSMatthew G. Knepley     PetscBool useCeed = PETSC_FALSE, flg;
4697d2b2dc1eSMatthew G. Knepley 
4698d2b2dc1eSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_plex_use_ceed", "Use LibCEED as the FEM backend", "DMPlexSetUseCeed", useCeed, &useCeed, &flg));
4699d2b2dc1eSMatthew G. Knepley     if (flg) PetscCall(DMPlexSetUseCeed(dm, useCeed));
4700d2b2dc1eSMatthew G. Knepley   }
47019318fe57SMatthew G. Knepley   /* Create coordinate space */
47029318fe57SMatthew G. Knepley   if (created) {
470361a622f3SMatthew G. Knepley     DM_Plex  *mesh   = (DM_Plex *)dm->data;
4704e44f6aebSMatthew G. Knepley     PetscInt  degree = 1, deg;
47055515ebd3SMatthew G. Knepley     PetscInt  height = 0;
47065515ebd3SMatthew G. Knepley     DM        cdm;
4707c3db174cSMatthew G. Knepley     PetscBool flg, localize = PETSC_TRUE, sparseLocalize = PETSC_TRUE;
47089318fe57SMatthew G. Knepley 
47099566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_coord_space", "Use an FEM space for coordinates", "", coordSpace, &coordSpace, &flg));
47109566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-dm_coord_petscspace_degree", "FEM degree for coordinate space", "", degree, &degree, NULL));
4711e44f6aebSMatthew G. Knepley     PetscCall(DMGetCoordinateDegree_Internal(dm, &deg));
4712e44f6aebSMatthew G. Knepley     if (coordSpace && deg <= 1) PetscCall(DMPlexCreateCoordinateSpace(dm, degree, PETSC_TRUE, mesh->coordFunc));
47135515ebd3SMatthew G. Knepley     PetscCall(DMGetCoordinateDM(dm, &cdm));
471461a622f3SMatthew G. Knepley     if (flg && !coordSpace) {
471561a622f3SMatthew G. Knepley       PetscDS      cds;
471661a622f3SMatthew G. Knepley       PetscObject  obj;
471761a622f3SMatthew G. Knepley       PetscClassId id;
471861a622f3SMatthew G. Knepley 
47199566063dSJacob Faibussowitsch       PetscCall(DMGetDS(cdm, &cds));
47209566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
47219566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
472261a622f3SMatthew G. Knepley       if (id == PETSCFE_CLASSID) {
472361a622f3SMatthew G. Knepley         PetscContainer dummy;
472461a622f3SMatthew G. Knepley 
47259566063dSJacob Faibussowitsch         PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &dummy));
47269566063dSJacob Faibussowitsch         PetscCall(PetscObjectSetName((PetscObject)dummy, "coordinates"));
47279566063dSJacob Faibussowitsch         PetscCall(DMSetField(cdm, 0, NULL, (PetscObject)dummy));
47289566063dSJacob Faibussowitsch         PetscCall(PetscContainerDestroy(&dummy));
47299566063dSJacob Faibussowitsch         PetscCall(DMClearDS(cdm));
473061a622f3SMatthew G. Knepley       }
473161a622f3SMatthew G. Knepley       mesh->coordFunc = NULL;
473261a622f3SMatthew G. Knepley     }
4733c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_localize", "Localize mesh coordinates", "", localize, &localize, NULL));
4734c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_sparse_localize", "Localize only necessary cells", "DMSetSparseLocalize", sparseLocalize, &sparseLocalize, &flg));
4735c3db174cSMatthew G. Knepley     if (flg) PetscCall(DMSetSparseLocalize(dm, sparseLocalize));
47365515ebd3SMatthew G. Knepley     PetscCall(PetscOptionsInt("-dm_localize_height", "Localize edges and faces in addition to cells", "", height, &height, &flg));
47375515ebd3SMatthew G. Knepley     if (flg) PetscCall(DMPlexSetMaxProjectionHeight(cdm, height));
4738c3db174cSMatthew G. Knepley     if (localize) PetscCall(DMLocalizeCoordinates(dm));
47399318fe57SMatthew G. Knepley   }
474068d4fef7SMatthew G. Knepley   /* Handle DMPlex refinement */
474161a622f3SMatthew G. Knepley   remap = PETSC_TRUE;
47429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL, 0));
47439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
47449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy, 0));
47459566063dSJacob Faibussowitsch   if (refine) PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
474668d4fef7SMatthew G. Knepley   if (refine && isHierarchy) {
4747acdc6f61SToby Isaac     DM *dms, coarseDM;
474868d4fef7SMatthew G. Knepley 
47499566063dSJacob Faibussowitsch     PetscCall(DMGetCoarseDM(dm, &coarseDM));
47509566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)coarseDM));
47519566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(refine, &dms));
47529566063dSJacob Faibussowitsch     PetscCall(DMRefineHierarchy(dm, refine, dms));
475368d4fef7SMatthew G. Knepley     /* Total hack since we do not pass in a pointer */
47549566063dSJacob Faibussowitsch     PetscCall(DMPlexSwap_Static(dm, dms[refine - 1]));
475568d4fef7SMatthew G. Knepley     if (refine == 1) {
47569566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[0]));
47579566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
475868d4fef7SMatthew G. Knepley     } else {
47599566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[refine - 2]));
47609566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
47619566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dms[0], dms[refine - 1]));
47629566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dms[0], PETSC_TRUE));
476368d4fef7SMatthew G. Knepley     }
47649566063dSJacob Faibussowitsch     PetscCall(DMSetCoarseDM(dms[refine - 1], coarseDM));
47659566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)coarseDM));
476668d4fef7SMatthew G. Knepley     /* Free DMs */
476768d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
4768dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
47699566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
477068d4fef7SMatthew G. Knepley     }
47719566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
477268d4fef7SMatthew G. Knepley   } else {
477368d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
47749318fe57SMatthew G. Knepley       DM             rdm;
477551a74b61SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
477668d4fef7SMatthew G. Knepley 
4777dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
47789566063dSJacob Faibussowitsch       PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
477968d4fef7SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
478069d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
4781dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
478261a622f3SMatthew G. Knepley       if (coordFunc && remap) {
47839566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
478451a74b61SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
478551a74b61SMatthew G. Knepley       }
478668d4fef7SMatthew G. Knepley     }
478768d4fef7SMatthew G. Knepley   }
47883cf6fe12SMatthew G. Knepley   /* Handle DMPlex coarsening */
47899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL, 0));
47909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy, 0));
4791b653a561SMatthew G. Knepley   if (coarsen && isHierarchy) {
4792b653a561SMatthew G. Knepley     DM *dms;
4793b653a561SMatthew G. Knepley 
47949566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(coarsen, &dms));
47959566063dSJacob Faibussowitsch     PetscCall(DMCoarsenHierarchy(dm, coarsen, dms));
4796b653a561SMatthew G. Knepley     /* Free DMs */
4797b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
4798dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
47999566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
4800b653a561SMatthew G. Knepley     }
48019566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
4802b653a561SMatthew G. Knepley   } else {
4803b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
48049318fe57SMatthew G. Knepley       DM             cdm;
48059318fe57SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
48063cf6fe12SMatthew G. Knepley 
4807dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
48089566063dSJacob Faibussowitsch       PetscCall(DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &cdm));
48093cf6fe12SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
481069d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &cdm));
4811dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
48129318fe57SMatthew G. Knepley       if (coordFunc) {
48139566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
48149318fe57SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
48159318fe57SMatthew G. Knepley       }
48163cf6fe12SMatthew G. Knepley     }
4817b653a561SMatthew G. Knepley   }
4818be664eb1SMatthew G. Knepley   // Handle coordinate remapping
4819be664eb1SMatthew G. Knepley   remap = PETSC_FALSE;
4820be664eb1SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_coord_remap", "Flag to control coordinate remapping", "", remap, &remap, NULL));
4821be664eb1SMatthew G. Knepley   if (remap) {
4822be664eb1SMatthew G. Knepley     DMPlexCoordMap map     = DM_COORD_MAP_NONE;
4823be664eb1SMatthew G. Knepley     PetscPointFunc mapFunc = NULL;
4824be664eb1SMatthew G. Knepley     PetscScalar    params[16];
4825f45b553cSPierre Jolivet     PetscInt       Np = PETSC_STATIC_ARRAY_LENGTH(params), cdim;
4826be664eb1SMatthew G. Knepley     MPI_Comm       comm;
4827be664eb1SMatthew G. Knepley 
4828be664eb1SMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4829be664eb1SMatthew G. Knepley     PetscCall(DMGetCoordinateDim(dm, &cdim));
4830be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsScalarArray("-dm_coord_map_params", "Parameters for the coordinate remapping", "", params, &Np, &flg));
4831be664eb1SMatthew G. Knepley     if (!flg) Np = 0;
4832be664eb1SMatthew G. Knepley     // TODO Allow user to pass a map function by name
4833be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsEnum("-dm_coord_map", "Coordinate mapping for built-in mesh", "", DMPlexCoordMaps, (PetscEnum)map, (PetscEnum *)&map, &flg));
4834be664eb1SMatthew G. Knepley     if (flg) {
4835be664eb1SMatthew G. Knepley       switch (map) {
4836be664eb1SMatthew G. Knepley       case DM_COORD_MAP_NONE:
4837be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
4838be664eb1SMatthew G. Knepley         break;
4839be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHEAR:
4840be664eb1SMatthew G. Knepley         mapFunc = coordMap_shear;
4841be664eb1SMatthew G. Knepley         if (!Np) {
4842be664eb1SMatthew G. Knepley           Np        = cdim + 1;
4843be664eb1SMatthew G. Knepley           params[0] = 0;
4844be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
4845be664eb1SMatthew G. Knepley         }
4846be664eb1SMatthew G. Knepley         PetscCheck(Np == cdim + 1, comm, PETSC_ERR_ARG_WRONG, "The shear coordinate map must have cdim + 1 = %" PetscInt_FMT " parameters, not %" PetscInt_FMT, cdim + 1, Np);
4847be664eb1SMatthew G. Knepley         break;
4848be664eb1SMatthew G. Knepley       case DM_COORD_MAP_FLARE:
4849be664eb1SMatthew G. Knepley         mapFunc = coordMap_flare;
4850be664eb1SMatthew G. Knepley         if (!Np) {
4851be664eb1SMatthew G. Knepley           Np        = cdim + 1;
4852be664eb1SMatthew G. Knepley           params[0] = 0;
4853be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
4854be664eb1SMatthew G. Knepley         }
4855be664eb1SMatthew G. Knepley         PetscCheck(Np == cdim + 1, comm, PETSC_ERR_ARG_WRONG, "The flare coordinate map must have cdim + 1 = %" PetscInt_FMT " parameters, not %" PetscInt_FMT, cdim + 1, Np);
4856be664eb1SMatthew G. Knepley         break;
4857be664eb1SMatthew G. Knepley       case DM_COORD_MAP_ANNULUS:
4858be664eb1SMatthew G. Knepley         mapFunc = coordMap_annulus;
4859be664eb1SMatthew G. Knepley         if (!Np) {
4860be664eb1SMatthew G. Knepley           Np        = 2;
4861be664eb1SMatthew G. Knepley           params[0] = 1.;
4862be664eb1SMatthew G. Knepley           params[1] = 2.;
4863be664eb1SMatthew G. Knepley         }
4864be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The annulus coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
4865be664eb1SMatthew G. Knepley         break;
4866be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHELL:
4867be664eb1SMatthew G. Knepley         mapFunc = coordMap_shell;
4868be664eb1SMatthew G. Knepley         if (!Np) {
4869be664eb1SMatthew G. Knepley           Np        = 2;
4870be664eb1SMatthew G. Knepley           params[0] = 1.;
4871be664eb1SMatthew G. Knepley           params[1] = 2.;
4872be664eb1SMatthew G. Knepley         }
4873be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The spherical shell coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
4874be664eb1SMatthew G. Knepley         break;
4875be664eb1SMatthew G. Knepley       default:
4876be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
4877be664eb1SMatthew G. Knepley       }
4878be664eb1SMatthew G. Knepley     }
4879be664eb1SMatthew G. Knepley     if (Np) {
4880be664eb1SMatthew G. Knepley       DM      cdm;
4881be664eb1SMatthew G. Knepley       PetscDS cds;
4882be664eb1SMatthew G. Knepley 
4883be664eb1SMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
4884be664eb1SMatthew G. Knepley       PetscCall(DMGetDS(cdm, &cds));
4885be664eb1SMatthew G. Knepley       PetscCall(PetscDSSetConstants(cds, Np, params));
4886be664eb1SMatthew G. Knepley     }
4887be664eb1SMatthew G. Knepley     PetscCall(DMPlexRemapGeometry(dm, 0.0, mapFunc));
4888be664eb1SMatthew G. Knepley   }
4889909dfd52SMatthew G. Knepley   /* Handle ghost cells */
48909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_create_fv_ghost_cells", "Flag to create finite volume ghost cells on the boundary", "DMCreate", ghostCells, &ghostCells, NULL));
4891909dfd52SMatthew G. Knepley   if (ghostCells) {
4892909dfd52SMatthew G. Knepley     DM   gdm;
4893909dfd52SMatthew G. Knepley     char lname[PETSC_MAX_PATH_LEN];
4894909dfd52SMatthew G. Knepley 
4895909dfd52SMatthew G. Knepley     lname[0] = '\0';
48969566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-dm_plex_fv_ghost_cells_label", "Label name for ghost cells boundary", "DMCreate", lname, lname, sizeof(lname), &flg));
48979566063dSJacob Faibussowitsch     PetscCall(DMPlexConstructGhostCells(dm, flg ? lname : NULL, NULL, &gdm));
489869d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &gdm));
4899909dfd52SMatthew G. Knepley   }
49006913077dSMatthew G. Knepley   /* Handle 1D order */
4901adc21957SMatthew G. Knepley   if (reorder != DM_REORDER_DEFAULT_FALSE && dim == 1) {
49026913077dSMatthew G. Knepley     DM           cdm, rdm;
49036913077dSMatthew G. Knepley     PetscDS      cds;
49046913077dSMatthew G. Knepley     PetscObject  obj;
49056913077dSMatthew G. Knepley     PetscClassId id = PETSC_OBJECT_CLASSID;
49066913077dSMatthew G. Knepley     IS           perm;
49076bc1bd01Sksagiyam     PetscInt     Nf;
49086913077dSMatthew G. Knepley     PetscBool    distributed;
49096913077dSMatthew G. Knepley 
49109566063dSJacob Faibussowitsch     PetscCall(DMPlexIsDistributed(dm, &distributed));
49119566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
49129566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
49139566063dSJacob Faibussowitsch     PetscCall(PetscDSGetNumFields(cds, &Nf));
49146913077dSMatthew G. Knepley     if (Nf) {
49159566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
49169566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
49176913077dSMatthew G. Knepley     }
49186bc1bd01Sksagiyam     if (!distributed && id != PETSCFE_CLASSID) {
49199566063dSJacob Faibussowitsch       PetscCall(DMPlexGetOrdering1D(dm, &perm));
49209566063dSJacob Faibussowitsch       PetscCall(DMPlexPermute(dm, perm, &rdm));
492169d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
49229566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&perm));
49236913077dSMatthew G. Knepley     }
49246913077dSMatthew G. Knepley   }
49253cf6fe12SMatthew G. Knepley /* Handle */
4926dd4c3f67SMatthew G. Knepley non_refine:
4927dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
492822d6dc08SStefano Zampini   char    *phases[16];
492922d6dc08SStefano Zampini   PetscInt Nphases = 16;
493022d6dc08SStefano Zampini   PetscCall(PetscOptionsStringArray("-dm_plex_option_phases", "Option phase prefixes", "DMSetFromOptions", phases, &Nphases, &flg));
4931d0609cedSBarry Smith   PetscOptionsHeadEnd();
493222d6dc08SStefano Zampini 
493322d6dc08SStefano Zampini   // Phases
493422d6dc08SStefano Zampini   if (flg) {
493522d6dc08SStefano Zampini     const char *oldPrefix;
493622d6dc08SStefano Zampini 
493722d6dc08SStefano Zampini     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &oldPrefix));
493822d6dc08SStefano Zampini     for (PetscInt ph = 0; ph < Nphases; ++ph) {
493922d6dc08SStefano Zampini       PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)dm, phases[ph]));
494022d6dc08SStefano Zampini       PetscCall(PetscInfo(dm, "Options phase %s for DM %s\n", phases[ph], dm->hdr.name));
494122d6dc08SStefano Zampini       PetscCall(DMSetFromOptions(dm));
494222d6dc08SStefano Zampini       PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldPrefix));
494322d6dc08SStefano Zampini       PetscCall(PetscFree(phases[ph]));
494422d6dc08SStefano Zampini     }
494522d6dc08SStefano Zampini   }
49463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49470a6ba040SMatthew G. Knepley }
49480a6ba040SMatthew G. Knepley 
4949d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_Plex(DM dm, Vec *vec)
4950d71ae5a4SJacob Faibussowitsch {
4951552f7358SJed Brown   PetscFunctionBegin;
49529566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector_Section_Private(dm, vec));
49539566063dSJacob Faibussowitsch   /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */
49549566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex));
49559566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_Plex_Native));
49569566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex));
49579566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_Plex_Native));
49583ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4959552f7358SJed Brown }
4960552f7358SJed Brown 
4961d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_Plex(DM dm, Vec *vec)
4962d71ae5a4SJacob Faibussowitsch {
4963552f7358SJed Brown   PetscFunctionBegin;
49649566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector_Section_Private(dm, vec));
49659566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex_Local));
49669566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex_Local));
49673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4968552f7358SJed Brown }
4969552f7358SJed Brown 
4970d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
4971d71ae5a4SJacob Faibussowitsch {
4972793f3fe5SMatthew G. Knepley   PetscInt depth, d;
4973793f3fe5SMatthew G. Knepley 
4974793f3fe5SMatthew G. Knepley   PetscFunctionBegin;
49759566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm, &depth));
4976793f3fe5SMatthew G. Knepley   if (depth == 1) {
49779566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &d));
49789566063dSJacob Faibussowitsch     if (dim == 0) PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
49799566063dSJacob Faibussowitsch     else if (dim == d) PetscCall(DMPlexGetDepthStratum(dm, 1, pStart, pEnd));
49809371c9d4SSatish Balay     else {
49819371c9d4SSatish Balay       *pStart = 0;
49829371c9d4SSatish Balay       *pEnd   = 0;
49839371c9d4SSatish Balay     }
4984793f3fe5SMatthew G. Knepley   } else {
49859566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
4986793f3fe5SMatthew G. Knepley   }
49873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4988793f3fe5SMatthew G. Knepley }
4989793f3fe5SMatthew G. Knepley 
4990d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetNeighbors_Plex(DM dm, PetscInt *nranks, const PetscMPIInt *ranks[])
4991d71ae5a4SJacob Faibussowitsch {
4992502a2867SDave May   PetscSF            sf;
49936497c311SBarry Smith   PetscMPIInt        niranks, njranks;
49946497c311SBarry Smith   PetscInt           n;
49950a19bb7dSprj-   const PetscMPIInt *iranks, *jranks;
49960a19bb7dSprj-   DM_Plex           *data = (DM_Plex *)dm->data;
4997502a2867SDave May 
49982f356facSMatthew G. Knepley   PetscFunctionBegin;
49999566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dm, &sf));
50000a19bb7dSprj-   if (!data->neighbors) {
50019566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
50029566063dSJacob Faibussowitsch     PetscCall(PetscSFGetRootRanks(sf, &njranks, &jranks, NULL, NULL, NULL));
50039566063dSJacob Faibussowitsch     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL));
50049566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors));
50059566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + 1, jranks, njranks));
50069566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks));
50070a19bb7dSprj-     n = njranks + niranks;
50089566063dSJacob Faibussowitsch     PetscCall(PetscSortRemoveDupsMPIInt(&n, data->neighbors + 1));
50090a19bb7dSprj-     /* The following cast should never fail: can't have more neighbors than PETSC_MPI_INT_MAX */
50109566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(n, data->neighbors));
50110a19bb7dSprj-   }
50120a19bb7dSprj-   if (nranks) *nranks = data->neighbors[0];
50130a19bb7dSprj-   if (ranks) {
50140a19bb7dSprj-     if (data->neighbors[0]) *ranks = data->neighbors + 1;
50150a19bb7dSprj-     else *ranks = NULL;
50160a19bb7dSprj-   }
50173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5018502a2867SDave May }
5019502a2867SDave May 
50201eb70e55SToby Isaac PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM, DM, Mat, Vec, Vec);
50211eb70e55SToby Isaac 
5022d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_Plex(DM dm)
5023d71ae5a4SJacob Faibussowitsch {
5024552f7358SJed Brown   PetscFunctionBegin;
5025552f7358SJed Brown   dm->ops->view                      = DMView_Plex;
50262c40f234SMatthew G. Knepley   dm->ops->load                      = DMLoad_Plex;
5027552f7358SJed Brown   dm->ops->setfromoptions            = DMSetFromOptions_Plex;
502838221697SMatthew G. Knepley   dm->ops->clone                     = DMClone_Plex;
5029552f7358SJed Brown   dm->ops->setup                     = DMSetUp_Plex;
50301bb6d2a8SBarry Smith   dm->ops->createlocalsection        = DMCreateLocalSection_Plex;
5031adc21957SMatthew G. Knepley   dm->ops->createsectionpermutation  = DMCreateSectionPermutation_Plex;
503266ad2231SToby Isaac   dm->ops->createdefaultconstraints  = DMCreateDefaultConstraints_Plex;
5033552f7358SJed Brown   dm->ops->createglobalvector        = DMCreateGlobalVector_Plex;
5034552f7358SJed Brown   dm->ops->createlocalvector         = DMCreateLocalVector_Plex;
5035184d77edSJed Brown   dm->ops->getlocaltoglobalmapping   = NULL;
50360298fd71SBarry Smith   dm->ops->createfieldis             = NULL;
5037552f7358SJed Brown   dm->ops->createcoordinatedm        = DMCreateCoordinateDM_Plex;
5038f19dbd58SToby Isaac   dm->ops->createcoordinatefield     = DMCreateCoordinateField_Plex;
50390a6ba040SMatthew G. Knepley   dm->ops->getcoloring               = NULL;
5040552f7358SJed Brown   dm->ops->creatematrix              = DMCreateMatrix_Plex;
5041bceba477SMatthew G. Knepley   dm->ops->createinterpolation       = DMCreateInterpolation_Plex;
5042bd041c0cSMatthew G. Knepley   dm->ops->createmassmatrix          = DMCreateMassMatrix_Plex;
5043b4937a87SMatthew G. Knepley   dm->ops->createmassmatrixlumped    = DMCreateMassMatrixLumped_Plex;
50445a84ad33SLisandro Dalcin   dm->ops->createinjection           = DMCreateInjection_Plex;
5045552f7358SJed Brown   dm->ops->refine                    = DMRefine_Plex;
50460a6ba040SMatthew G. Knepley   dm->ops->coarsen                   = DMCoarsen_Plex;
50470a6ba040SMatthew G. Knepley   dm->ops->refinehierarchy           = DMRefineHierarchy_Plex;
5048b653a561SMatthew G. Knepley   dm->ops->coarsenhierarchy          = DMCoarsenHierarchy_Plex;
5049d410b0cfSMatthew G. Knepley   dm->ops->extrude                   = DMExtrude_Plex;
50500298fd71SBarry Smith   dm->ops->globaltolocalbegin        = NULL;
50510298fd71SBarry Smith   dm->ops->globaltolocalend          = NULL;
50520298fd71SBarry Smith   dm->ops->localtoglobalbegin        = NULL;
50530298fd71SBarry Smith   dm->ops->localtoglobalend          = NULL;
5054552f7358SJed Brown   dm->ops->destroy                   = DMDestroy_Plex;
5055552f7358SJed Brown   dm->ops->createsubdm               = DMCreateSubDM_Plex;
50562adcc780SMatthew G. Knepley   dm->ops->createsuperdm             = DMCreateSuperDM_Plex;
5057793f3fe5SMatthew G. Knepley   dm->ops->getdimpoints              = DMGetDimPoints_Plex;
5058552f7358SJed Brown   dm->ops->locatepoints              = DMLocatePoints_Plex;
50590709b2feSToby Isaac   dm->ops->projectfunctionlocal      = DMProjectFunctionLocal_Plex;
50600709b2feSToby Isaac   dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_Plex;
5061bfc4295aSToby Isaac   dm->ops->projectfieldlocal         = DMProjectFieldLocal_Plex;
50628c6c5593SMatthew G. Knepley   dm->ops->projectfieldlabellocal    = DMProjectFieldLabelLocal_Plex;
5063ece3a9fcSMatthew G. Knepley   dm->ops->projectbdfieldlabellocal  = DMProjectBdFieldLabelLocal_Plex;
50640709b2feSToby Isaac   dm->ops->computel2diff             = DMComputeL2Diff_Plex;
5065b698f381SToby Isaac   dm->ops->computel2gradientdiff     = DMComputeL2GradientDiff_Plex;
50662a16baeaSToby Isaac   dm->ops->computel2fielddiff        = DMComputeL2FieldDiff_Plex;
506728d58a37SPierre Jolivet   dm->ops->getneighbors              = DMGetNeighbors_Plex;
50686c6a6b79SMatthew G. Knepley   dm->ops->getlocalboundingbox       = DMGetLocalBoundingBox_Coordinates;
5069907a3e9cSStefano Zampini   dm->ops->createdomaindecomposition = DMCreateDomainDecomposition_Plex;
5070907a3e9cSStefano Zampini   dm->ops->createddscatters          = DMCreateDomainDecompositionScatters_Plex;
50719566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_Plex));
50726c51210dSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", DMPlexInsertTimeDerivativeBoundaryValues_Plex));
50739566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", DMSetUpGLVisViewer_Plex));
50749566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_Plex));
50759566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", DMPlexDistributeGetDefault_Plex));
50769566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", DMPlexDistributeSetDefault_Plex));
50776bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", DMPlexReorderGetDefault_Plex));
50786bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", DMPlexReorderSetDefault_Plex));
5079adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetDefault_C", DMReorderSectionGetDefault_Plex));
5080adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetDefault_C", DMReorderSectionSetDefault_Plex));
5081adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetType_C", DMReorderSectionGetType_Plex));
5082adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetType_C", DMReorderSectionSetType_Plex));
50839566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", DMInterpolateSolution_Plex));
5084c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMPlexGetOverlap_Plex));
5085c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", DMPlexSetOverlap_Plex));
5086d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", DMPlexGetUseCeed_Plex));
5087d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", DMPlexSetUseCeed_Plex));
50883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5089552f7358SJed Brown }
5090552f7358SJed Brown 
5091d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode DMClone_Plex(DM dm, DM *newdm)
5092d71ae5a4SJacob Faibussowitsch {
509363a16f15SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *)dm->data;
50941fca310dSJames Wright   const PetscSF *face_sfs;
50951fca310dSJames Wright   PetscInt       num_face_sfs;
509663a16f15SMatthew G. Knepley 
509763a16f15SMatthew G. Knepley   PetscFunctionBegin;
509863a16f15SMatthew G. Knepley   mesh->refct++;
509963a16f15SMatthew G. Knepley   (*newdm)->data = mesh;
51001fca310dSJames Wright   PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &face_sfs));
51011fca310dSJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(*newdm, num_face_sfs, (PetscSF *)face_sfs));
51029566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)*newdm, DMPLEX));
51039566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(*newdm));
51043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
510563a16f15SMatthew G. Knepley }
510663a16f15SMatthew G. Knepley 
51078818961aSMatthew G Knepley /*MC
5108a1cb98faSBarry Smith   DMPLEX = "plex" - A `DM` object that encapsulates an unstructured mesh, or CW Complex, which can be expressed using a Hasse Diagram.
510920f4b53cSBarry Smith                     In the local representation, `Vec`s contain all unknowns in the interior and shared boundary. This is
51108818961aSMatthew G Knepley                     specified by a PetscSection object. Ownership in the global representation is determined by
5111a1cb98faSBarry Smith                     ownership of the underlying `DMPLEX` points. This is specified by another `PetscSection` object.
51128818961aSMatthew G Knepley 
5113e5893cccSMatthew G. Knepley   Options Database Keys:
5114250712c9SMatthew G. Knepley + -dm_refine_pre                     - Refine mesh before distribution
5115250712c9SMatthew G. Knepley + -dm_refine_uniform_pre             - Choose uniform or generator-based refinement
5116250712c9SMatthew G. Knepley + -dm_refine_volume_limit_pre        - Cell volume limit after pre-refinement using generator
5117250712c9SMatthew G. Knepley . -dm_distribute                     - Distribute mesh across processes
5118250712c9SMatthew G. Knepley . -dm_distribute_overlap             - Number of cells to overlap for distribution
5119250712c9SMatthew G. Knepley . -dm_refine                         - Refine mesh after distribution
5120c3db174cSMatthew G. Knepley . -dm_localize <bool>                - Whether to localize coordinates for periodic meshes
5121c3db174cSMatthew G. Knepley . -dm_sparse_localize <bool>         - Whether to only localize cells on the periodic boundary
5122250712c9SMatthew G. Knepley . -dm_plex_hash_location             - Use grid hashing for point location
5123ddce0771SMatthew G. Knepley . -dm_plex_hash_box_faces <n,m,p>    - The number of divisions in each direction of the grid hash
5124f12cf164SMatthew G. Knepley . -dm_plex_partition_balance         - Attempt to evenly divide points on partition boundary between processes
5125f12cf164SMatthew G. Knepley . -dm_plex_remesh_bd                 - Allow changes to the boundary on remeshing
5126d5b43468SJose E. Roman . -dm_plex_max_projection_height     - Maximum mesh point height used to project locally
5127f12cf164SMatthew G. Knepley . -dm_plex_regular_refinement        - Use special nested projection algorithm for regular refinement
5128d02c7345SMatthew G. Knepley . -dm_plex_reorder_section           - Use specialized blocking if available
5129aaa8cc7dSPierre Jolivet . -dm_plex_check_all                 - Perform all checks below
5130f12cf164SMatthew G. Knepley . -dm_plex_check_symmetry            - Check that the adjacency information in the mesh is symmetric
5131f12cf164SMatthew G. Knepley . -dm_plex_check_skeleton <celltype> - Check that each cell has the correct number of vertices
5132f12cf164SMatthew G. Knepley . -dm_plex_check_faces <celltype>    - Check that the faces of each cell give a vertex order this is consistent with what we expect from the cell type
5133f12cf164SMatthew G. Knepley . -dm_plex_check_geometry            - Check that cells have positive volume
5134f12cf164SMatthew G. Knepley . -dm_view :mesh.tex:ascii_latex     - View the mesh in LaTeX/TikZ
5135e5893cccSMatthew G. Knepley . -dm_plex_view_scale <num>          - Scale the TikZ
51365962854dSMatthew G. Knepley . -dm_plex_print_fem <num>           - View FEM assembly information, such as element vectors and matrices
51375962854dSMatthew G. Knepley - -dm_plex_print_fvm <num>           - View FVM assembly information, such as flux updates
5138e5893cccSMatthew G. Knepley 
51398818961aSMatthew G Knepley   Level: intermediate
51408818961aSMatthew G Knepley 
51411cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMPlexCreate()`, `DMCreate()`, `DMSetType()`, `PetscSection`
51428818961aSMatthew G Knepley M*/
51438818961aSMatthew G Knepley 
5144d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm)
5145d71ae5a4SJacob Faibussowitsch {
5146552f7358SJed Brown   DM_Plex *mesh;
5147412e9a14SMatthew G. Knepley   PetscInt unit;
5148552f7358SJed Brown 
5149552f7358SJed Brown   PetscFunctionBegin;
5150f39ec787SMatthew G. Knepley   PetscCall(PetscCitationsRegister(PlexCitation, &Plexcite));
5151552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
51524dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mesh));
5153adc21957SMatthew G. Knepley   dm->reorderSection = DM_REORDER_DEFAULT_NOTSET;
5154552f7358SJed Brown   dm->data           = mesh;
5155552f7358SJed Brown 
5156552f7358SJed Brown   mesh->refct = 1;
51579566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection));
51589566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection));
5159552f7358SJed Brown   mesh->refinementUniform      = PETSC_TRUE;
5160552f7358SJed Brown   mesh->refinementLimit        = -1.0;
5161e600fa54SMatthew G. Knepley   mesh->distDefault            = PETSC_TRUE;
5162adc21957SMatthew G. Knepley   mesh->reorderDefault         = DM_REORDER_DEFAULT_NOTSET;
51631d1f2f2aSksagiyam   mesh->distributionName       = NULL;
51647d0f5628SVaclav Hapla   mesh->interpolated           = DMPLEX_INTERPOLATED_INVALID;
51657d0f5628SVaclav Hapla   mesh->interpolatedCollective = DMPLEX_INTERPOLATED_INVALID;
5166552f7358SJed Brown 
51679566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner));
51682e62ab5aSMatthew G. Knepley   mesh->remeshBd = PETSC_FALSE;
5169d9deefdfSMatthew G. Knepley 
51708865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
5171552f7358SJed Brown 
5172df0420ecSMatthew G. Knepley   mesh->depthState    = -1;
5173ba2698f1SMatthew G. Knepley   mesh->celltypeState = -1;
51746113b454SMatthew G. Knepley   mesh->printTol      = 1.0e-10;
5175c29ce622SStefano Zampini   mesh->nonempty_comm = MPI_COMM_SELF;
5176552f7358SJed Brown 
51779566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
51783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5179552f7358SJed Brown }
5180552f7358SJed Brown 
5181552f7358SJed Brown /*@
5182a1cb98faSBarry Smith   DMPlexCreate - Creates a `DMPLEX` object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
5183552f7358SJed Brown 
5184d083f849SBarry Smith   Collective
5185552f7358SJed Brown 
5186552f7358SJed Brown   Input Parameter:
5187a1cb98faSBarry Smith . comm - The communicator for the `DMPLEX` object
5188552f7358SJed Brown 
5189552f7358SJed Brown   Output Parameter:
5190a1cb98faSBarry Smith . mesh - The `DMPLEX` object
5191552f7358SJed Brown 
5192552f7358SJed Brown   Level: beginner
5193552f7358SJed Brown 
519442747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMCreate()`, `DMSetType()`
5195552f7358SJed Brown @*/
5196d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
5197d71ae5a4SJacob Faibussowitsch {
5198552f7358SJed Brown   PetscFunctionBegin;
51994f572ea9SToby Isaac   PetscAssertPointer(mesh, 2);
52009566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, mesh));
52019566063dSJacob Faibussowitsch   PetscCall(DMSetType(*mesh, DMPLEX));
52023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5203552f7358SJed Brown }
5204552f7358SJed Brown 
5205b09969d6SVaclav Hapla /*@C
5206b0fe842aSMatthew G. Knepley   DMPlexBuildFromCellListParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) where all cells have the same celltype
5207a1cb98faSBarry Smith 
520820f4b53cSBarry Smith   Collective; No Fortran Support
5209b09969d6SVaclav Hapla 
5210b09969d6SVaclav Hapla   Input Parameters:
5211a1cb98faSBarry Smith + dm          - The `DM`
5212b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5213a1cb98faSBarry Smith . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5214a1cb98faSBarry Smith . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5215b09969d6SVaclav Hapla . numCorners  - The number of vertices for each cell
52165e488331SVaclav Hapla - cells       - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5217b09969d6SVaclav Hapla 
5218be8c289dSNicolas Barral   Output Parameters:
5219a1cb98faSBarry Smith + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5220be8c289dSNicolas Barral - verticesAdjSaved - (Optional) vertex adjacency array
5221b09969d6SVaclav Hapla 
5222b09969d6SVaclav Hapla   Level: advanced
5223b09969d6SVaclav Hapla 
5224a1cb98faSBarry Smith   Notes:
5225a1cb98faSBarry Smith   Two triangles sharing a face
5226a1cb98faSBarry Smith .vb
5227a1cb98faSBarry Smith 
5228a1cb98faSBarry Smith         2
5229a1cb98faSBarry Smith       / | \
5230a1cb98faSBarry Smith      /  |  \
5231a1cb98faSBarry Smith     /   |   \
5232a1cb98faSBarry Smith    0  0 | 1  3
5233a1cb98faSBarry Smith     \   |   /
5234a1cb98faSBarry Smith      \  |  /
5235a1cb98faSBarry Smith       \ | /
5236a1cb98faSBarry Smith         1
5237a1cb98faSBarry Smith .ve
5238a1cb98faSBarry Smith   would have input
5239a1cb98faSBarry Smith .vb
5240a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5241a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5242a1cb98faSBarry Smith .ve
5243a1cb98faSBarry Smith   which would result in the `DMPLEX`
5244a1cb98faSBarry Smith .vb
5245a1cb98faSBarry Smith 
5246a1cb98faSBarry Smith         4
5247a1cb98faSBarry Smith       / | \
5248a1cb98faSBarry Smith      /  |  \
5249a1cb98faSBarry Smith     /   |   \
5250a1cb98faSBarry Smith    2  0 | 1  5
5251a1cb98faSBarry Smith     \   |   /
5252a1cb98faSBarry Smith      \  |  /
5253a1cb98faSBarry Smith       \ | /
5254a1cb98faSBarry Smith         3
5255a1cb98faSBarry Smith .ve
5256a1cb98faSBarry Smith 
5257a1cb98faSBarry Smith   Vertices are implicitly numbered consecutively 0,...,NVertices.
5258a1cb98faSBarry Smith   Each rank owns a chunk of numVertices consecutive vertices.
5259a1cb98faSBarry Smith   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5260a1cb98faSBarry Smith   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
5261a1cb98faSBarry Smith   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5262a1cb98faSBarry Smith 
5263a1cb98faSBarry Smith   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5264a1cb98faSBarry Smith 
52651cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5266a1cb98faSBarry Smith           `PetscSF`
5267b09969d6SVaclav Hapla @*/
5268d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellListParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, const PetscInt cells[], PetscSF *vertexSF, PetscInt **verticesAdjSaved)
5269d71ae5a4SJacob Faibussowitsch {
52702464107aSksagiyam   PetscSF     sfPoint;
52712464107aSksagiyam   PetscLayout layout;
527282fb893eSVaclav Hapla   PetscInt    numVerticesAdj, *verticesAdj, *cones, c, p;
5273a47d0d45SMatthew G. Knepley 
5274a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
527525b6865aSVaclav Hapla   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
52769566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
527725b6865aSVaclav Hapla   /* Get/check global number of vertices */
527825b6865aSVaclav Hapla   {
527925b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
528025b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
528125b6865aSVaclav Hapla 
528225b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
52831690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
52849371c9d4SSatish Balay     for (i = 0; i < len; i++)
52859371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
528625b6865aSVaclav Hapla     ++NVerticesInCells;
5287462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
528825b6865aSVaclav Hapla 
528925b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
52909371c9d4SSatish Balay     else
52919371c9d4SSatish Balay       PetscCheck(NVertices == PETSC_DECIDE || NVertices >= NVerticesInCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Specified global number of vertices %" PetscInt_FMT " must be greater than or equal to the number of vertices in cells %" PetscInt_FMT, NVertices, NVerticesInCells);
529225b6865aSVaclav Hapla   }
52939079aca8SVaclav Hapla   /* Count locally unique vertices */
52949079aca8SVaclav Hapla   {
52959079aca8SVaclav Hapla     PetscHSetI vhash;
52969079aca8SVaclav Hapla     PetscInt   off = 0;
52979079aca8SVaclav Hapla 
52989566063dSJacob Faibussowitsch     PetscCall(PetscHSetICreate(&vhash));
5299a47d0d45SMatthew G. Knepley     for (c = 0; c < numCells; ++c) {
530048a46eb9SPierre Jolivet       for (p = 0; p < numCorners; ++p) PetscCall(PetscHSetIAdd(vhash, cells[c * numCorners + p]));
5301a47d0d45SMatthew G. Knepley     }
53029566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
53039566063dSJacob Faibussowitsch     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5304ad540459SPierre Jolivet     else verticesAdj = *verticesAdjSaved;
53059566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
53069566063dSJacob Faibussowitsch     PetscCall(PetscHSetIDestroy(&vhash));
530763a3b9bcSJacob Faibussowitsch     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
5308a47d0d45SMatthew G. Knepley   }
53099566063dSJacob Faibussowitsch   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5310a47d0d45SMatthew G. Knepley   /* Create cones */
53119566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
53129566063dSJacob Faibussowitsch   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
53139566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
53149566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5315a47d0d45SMatthew G. Knepley   for (c = 0; c < numCells; ++c) {
5316a47d0d45SMatthew G. Knepley     for (p = 0; p < numCorners; ++p) {
5317a47d0d45SMatthew G. Knepley       const PetscInt gv = cells[c * numCorners + p];
5318a47d0d45SMatthew G. Knepley       PetscInt       lv;
5319a47d0d45SMatthew G. Knepley 
53209079aca8SVaclav Hapla       /* Positions within verticesAdj form 0-based local vertex numbering;
53219079aca8SVaclav Hapla          we need to shift it by numCells to get correct DAG points (cells go first) */
53229566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
532363a3b9bcSJacob Faibussowitsch       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5324961cfab0SVaclav Hapla       cones[c * numCorners + p] = lv + numCells;
5325a47d0d45SMatthew G. Knepley     }
5326a47d0d45SMatthew G. Knepley   }
53272464107aSksagiyam   /* Build point sf */
53289566063dSJacob Faibussowitsch   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
53299566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetSize(layout, NVertices));
53309566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
53319566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetBlockSize(layout, 1));
53329566063dSJacob Faibussowitsch   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
53339566063dSJacob Faibussowitsch   PetscCall(PetscLayoutDestroy(&layout));
53349566063dSJacob Faibussowitsch   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
53359566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
53362464107aSksagiyam   if (dm->sf) {
53372464107aSksagiyam     const char *prefix;
53382464107aSksagiyam 
53399566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
53409566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
53412464107aSksagiyam   }
53429566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sfPoint));
53439566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&sfPoint));
5344f4f49eeaSPierre Jolivet   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
5345a47d0d45SMatthew G. Knepley   /* Fill in the rest of the topology structure */
53469566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
53479566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
53489566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
53493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5350a47d0d45SMatthew G. Knepley }
5351a47d0d45SMatthew G. Knepley 
5352b0fe842aSMatthew G. Knepley /*@C
5353b0fe842aSMatthew G. Knepley   DMPlexBuildFromCellSectionParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) allowing multiple celltypes
5354b0fe842aSMatthew G. Knepley 
5355b0fe842aSMatthew G. Knepley   Collective; No Fortran Support
5356b0fe842aSMatthew G. Knepley 
5357b0fe842aSMatthew G. Knepley   Input Parameters:
5358b0fe842aSMatthew G. Knepley + dm          - The `DM`
5359b0fe842aSMatthew G. Knepley . numCells    - The number of cells owned by this process
5360b0fe842aSMatthew G. Knepley . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5361b0fe842aSMatthew G. Knepley . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5362b0fe842aSMatthew G. Knepley . cellSection - The `PetscSection` giving the number of vertices for each cell (layout of cells)
5363b0fe842aSMatthew G. Knepley - cells       - An array of the global vertex numbers for each cell
5364b0fe842aSMatthew G. Knepley 
5365b0fe842aSMatthew G. Knepley   Output Parameters:
5366b0fe842aSMatthew G. Knepley + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5367b0fe842aSMatthew G. Knepley - verticesAdjSaved - (Optional) vertex adjacency array
5368b0fe842aSMatthew G. Knepley 
5369b0fe842aSMatthew G. Knepley   Level: advanced
5370b0fe842aSMatthew G. Knepley 
5371b0fe842aSMatthew G. Knepley   Notes:
5372b0fe842aSMatthew G. Knepley   A triangle and quadrilateral sharing a face
5373b0fe842aSMatthew G. Knepley .vb
5374b0fe842aSMatthew G. Knepley         2----------3
5375b0fe842aSMatthew G. Knepley       / |          |
5376b0fe842aSMatthew G. Knepley      /  |          |
5377b0fe842aSMatthew G. Knepley     /   |          |
5378b0fe842aSMatthew G. Knepley    0  0 |     1    |
5379b0fe842aSMatthew G. Knepley     \   |          |
5380b0fe842aSMatthew G. Knepley      \  |          |
5381b0fe842aSMatthew G. Knepley       \ |          |
5382b0fe842aSMatthew G. Knepley         1----------4
5383b0fe842aSMatthew G. Knepley .ve
5384b0fe842aSMatthew G. Knepley   would have input
5385b0fe842aSMatthew G. Knepley .vb
5386b0fe842aSMatthew G. Knepley   numCells = 2, numVertices = 5
5387b0fe842aSMatthew G. Knepley   cells = [0 1 2  1 4 3 2]
5388b0fe842aSMatthew G. Knepley .ve
5389b0fe842aSMatthew G. Knepley   which would result in the `DMPLEX`
5390b0fe842aSMatthew G. Knepley .vb
5391b0fe842aSMatthew G. Knepley         4----------5
5392b0fe842aSMatthew G. Knepley       / |          |
5393b0fe842aSMatthew G. Knepley      /  |          |
5394b0fe842aSMatthew G. Knepley     /   |          |
5395b0fe842aSMatthew G. Knepley    2  0 |     1    |
5396b0fe842aSMatthew G. Knepley     \   |          |
5397b0fe842aSMatthew G. Knepley      \  |          |
5398b0fe842aSMatthew G. Knepley       \ |          |
5399b0fe842aSMatthew G. Knepley         3----------6
5400b0fe842aSMatthew G. Knepley .ve
5401b0fe842aSMatthew G. Knepley 
5402b0fe842aSMatthew G. Knepley   Vertices are implicitly numbered consecutively 0,...,NVertices.
5403b0fe842aSMatthew G. Knepley   Each rank owns a chunk of numVertices consecutive vertices.
5404b0fe842aSMatthew G. Knepley   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5405b0fe842aSMatthew G. Knepley   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
5406b0fe842aSMatthew G. Knepley   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5407b0fe842aSMatthew G. Knepley 
5408b0fe842aSMatthew G. Knepley   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5409b0fe842aSMatthew G. Knepley 
5410b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexCreateFromCellSectionParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5411b0fe842aSMatthew G. Knepley           `PetscSF`
5412b0fe842aSMatthew G. Knepley @*/
5413b0fe842aSMatthew G. Knepley PetscErrorCode DMPlexBuildFromCellSectionParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, const PetscInt cells[], PetscSF *vertexSF, PetscInt **verticesAdjSaved)
5414b0fe842aSMatthew G. Knepley {
5415b0fe842aSMatthew G. Knepley   PetscSF     sfPoint;
5416b0fe842aSMatthew G. Knepley   PetscLayout layout;
5417b0fe842aSMatthew G. Knepley   PetscInt    numVerticesAdj, *verticesAdj, *cones, cStart, cEnd, len;
5418b0fe842aSMatthew G. Knepley 
5419b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
5420b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
5421b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
5422b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetChart(cellSection, &cStart, &cEnd));
5423b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(cellSection, &len));
5424b0fe842aSMatthew G. Knepley   PetscCheck(cStart == 0 && cEnd == numCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Section chart [%" PetscInt_FMT ", %" PetscInt_FMT ") should be [0, %" PetscInt_FMT ")", cStart, cEnd, numCells);
5425b0fe842aSMatthew G. Knepley   /* Get/check global number of vertices */
5426b0fe842aSMatthew G. Knepley   {
5427b0fe842aSMatthew G. Knepley     PetscInt NVerticesInCells;
5428b0fe842aSMatthew G. Knepley 
5429b0fe842aSMatthew G. Knepley     /* NVerticesInCells = max(cells) + 1 */
5430b0fe842aSMatthew G. Knepley     NVerticesInCells = PETSC_MIN_INT;
5431b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++)
5432b0fe842aSMatthew G. Knepley       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
5433b0fe842aSMatthew G. Knepley     ++NVerticesInCells;
5434b0fe842aSMatthew G. Knepley     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
5435b0fe842aSMatthew G. Knepley 
5436b0fe842aSMatthew G. Knepley     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
5437b0fe842aSMatthew G. Knepley     else
5438b0fe842aSMatthew G. Knepley       PetscCheck(NVertices == PETSC_DECIDE || NVertices >= NVerticesInCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Specified global number of vertices %" PetscInt_FMT " must be greater than or equal to the number of vertices in cells %" PetscInt_FMT, NVertices, NVerticesInCells);
5439b0fe842aSMatthew G. Knepley   }
5440b0fe842aSMatthew G. Knepley   /* Count locally unique vertices */
5441b0fe842aSMatthew G. Knepley   {
5442b0fe842aSMatthew G. Knepley     PetscHSetI vhash;
5443b0fe842aSMatthew G. Knepley     PetscInt   off = 0;
5444b0fe842aSMatthew G. Knepley 
5445b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetICreate(&vhash));
5446b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++) PetscCall(PetscHSetIAdd(vhash, cells[i]));
5447b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
5448b0fe842aSMatthew G. Knepley     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5449b0fe842aSMatthew G. Knepley     else verticesAdj = *verticesAdjSaved;
5450b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
5451b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIDestroy(&vhash));
5452b0fe842aSMatthew G. Knepley     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
5453b0fe842aSMatthew G. Knepley   }
5454b0fe842aSMatthew G. Knepley   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5455b0fe842aSMatthew G. Knepley   /* Create cones */
5456b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
5457b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
5458b0fe842aSMatthew G. Knepley     PetscInt dof;
5459b0fe842aSMatthew G. Knepley 
5460b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
5461b0fe842aSMatthew G. Knepley     PetscCall(DMPlexSetConeSize(dm, c, dof));
5462b0fe842aSMatthew G. Knepley   }
5463b0fe842aSMatthew G. Knepley   PetscCall(DMSetUp(dm));
5464b0fe842aSMatthew G. Knepley   PetscCall(DMPlexGetCones(dm, &cones));
5465b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
5466b0fe842aSMatthew G. Knepley     PetscInt dof, off;
5467b0fe842aSMatthew G. Knepley 
5468b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
5469b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetOffset(cellSection, c, &off));
5470b0fe842aSMatthew G. Knepley     for (PetscInt p = off; p < off + dof; ++p) {
5471b0fe842aSMatthew G. Knepley       const PetscInt gv = cells[p];
5472b0fe842aSMatthew G. Knepley       PetscInt       lv;
5473b0fe842aSMatthew G. Knepley 
5474b0fe842aSMatthew G. Knepley       /* Positions within verticesAdj form 0-based local vertex numbering;
5475b0fe842aSMatthew G. Knepley          we need to shift it by numCells to get correct DAG points (cells go first) */
5476b0fe842aSMatthew G. Knepley       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
5477b0fe842aSMatthew G. Knepley       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5478b0fe842aSMatthew G. Knepley       cones[p] = lv + numCells;
5479b0fe842aSMatthew G. Knepley     }
5480b0fe842aSMatthew G. Knepley   }
5481b0fe842aSMatthew G. Knepley   /* Build point sf */
5482b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
5483b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetSize(layout, NVertices));
5484b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
5485b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetBlockSize(layout, 1));
5486b0fe842aSMatthew G. Knepley   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
5487b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutDestroy(&layout));
5488b0fe842aSMatthew G. Knepley   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
5489b0fe842aSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
5490b0fe842aSMatthew G. Knepley   if (dm->sf) {
5491b0fe842aSMatthew G. Knepley     const char *prefix;
5492b0fe842aSMatthew G. Knepley 
5493b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
5494b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
5495b0fe842aSMatthew G. Knepley   }
5496b0fe842aSMatthew G. Knepley   PetscCall(DMSetPointSF(dm, sfPoint));
5497b0fe842aSMatthew G. Knepley   PetscCall(PetscSFDestroy(&sfPoint));
5498b0fe842aSMatthew G. Knepley   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
5499b0fe842aSMatthew G. Knepley   /* Fill in the rest of the topology structure */
5500b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSymmetrize(dm));
5501b0fe842aSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
5502b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
5503b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
5504b0fe842aSMatthew G. Knepley }
5505b0fe842aSMatthew G. Knepley 
5506cc4c1da9SBarry Smith /*@
5507a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellListParallel - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
5508a1cb98faSBarry Smith 
550920f4b53cSBarry Smith   Collective; No Fortran Support
5510b09969d6SVaclav Hapla 
5511b09969d6SVaclav Hapla   Input Parameters:
5512a1cb98faSBarry Smith + dm           - The `DM`
5513b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
5514a1cb98faSBarry Smith . sfVert       - `PetscSF` describing complete vertex ownership
5515b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5516b09969d6SVaclav Hapla 
5517b09969d6SVaclav Hapla   Level: advanced
5518b09969d6SVaclav Hapla 
55191cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellListParallel()`
5520b09969d6SVaclav Hapla @*/
5521d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellListParallel(DM dm, PetscInt spaceDim, PetscSF sfVert, const PetscReal vertexCoords[])
5522d71ae5a4SJacob Faibussowitsch {
5523a47d0d45SMatthew G. Knepley   PetscSection coordSection;
5524a47d0d45SMatthew G. Knepley   Vec          coordinates;
5525a47d0d45SMatthew G. Knepley   PetscScalar *coords;
55261edcf0b2SVaclav Hapla   PetscInt     numVertices, numVerticesAdj, coordSize, v, vStart, vEnd;
5527a47d0d45SMatthew G. Knepley 
5528a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
55299566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
55309566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
55311dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
55329566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
55339566063dSJacob Faibussowitsch   PetscCall(PetscSFGetGraph(sfVert, &numVertices, &numVerticesAdj, NULL, NULL));
55341dca8a05SBarry Smith   PetscCheck(vEnd - vStart == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Supplied sfVert has wrong number of leaves = %" PetscInt_FMT " != %" PetscInt_FMT " = vEnd - vStart", numVerticesAdj, vEnd - vStart);
55359566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
55369566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
55379566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
55389566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
55391edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
55409566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
55419566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
5542a47d0d45SMatthew G. Knepley   }
55439566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
55449566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
55459566063dSJacob Faibussowitsch   PetscCall(VecCreate(PetscObjectComm((PetscObject)dm), &coordinates));
55469566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
55479566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
55489566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
55499566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
55509566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
5551a47d0d45SMatthew G. Knepley   {
5552a47d0d45SMatthew G. Knepley     MPI_Datatype coordtype;
5553a47d0d45SMatthew G. Knepley 
5554a47d0d45SMatthew G. Knepley     /* Need a temp buffer for coords if we have complex/single */
55556497c311SBarry Smith     PetscCallMPI(MPI_Type_contiguous((PetscMPIInt)spaceDim, MPIU_SCALAR, &coordtype));
55569566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_commit(&coordtype));
555721016a8bSBarry Smith #if defined(PETSC_USE_COMPLEX)
555821016a8bSBarry Smith     {
555921016a8bSBarry Smith       PetscScalar *svertexCoords;
556021016a8bSBarry Smith       PetscInt     i;
55619566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * spaceDim, &svertexCoords));
55623612f820SVaclav Hapla       for (i = 0; i < numVertices * spaceDim; i++) svertexCoords[i] = vertexCoords[i];
55639566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
55649566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
55659566063dSJacob Faibussowitsch       PetscCall(PetscFree(svertexCoords));
556621016a8bSBarry Smith     }
556721016a8bSBarry Smith #else
55689566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastBegin(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
55699566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastEnd(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
557021016a8bSBarry Smith #endif
55719566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_free(&coordtype));
5572a47d0d45SMatthew G. Knepley   }
55739566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
55749566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
55759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
55769566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
55773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5578a47d0d45SMatthew G. Knepley }
5579a47d0d45SMatthew G. Knepley 
5580c3edce3dSSatish Balay /*@
5581b0fe842aSMatthew G. Knepley   DMPlexCreateFromCellListParallelPetsc - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) where all cells have the same celltype
5582a1cb98faSBarry Smith 
5583a1cb98faSBarry Smith   Collective
5584a47d0d45SMatthew G. Knepley 
5585a47d0d45SMatthew G. Knepley   Input Parameters:
5586a47d0d45SMatthew G. Knepley + comm         - The communicator
5587a47d0d45SMatthew G. Knepley . dim          - The topological dimension of the mesh
5588a47d0d45SMatthew G. Knepley . numCells     - The number of cells owned by this process
5589a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
5590a1cb98faSBarry Smith . NVertices    - The global number of vertices, or `PETSC_DECIDE`
5591a47d0d45SMatthew G. Knepley . numCorners   - The number of vertices for each cell
5592a47d0d45SMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
5593a47d0d45SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5594a47d0d45SMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
5595a47d0d45SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5596a47d0d45SMatthew G. Knepley 
5597d8d19677SJose E. Roman   Output Parameters:
5598a1cb98faSBarry Smith + dm          - The `DM`
5599a1cb98faSBarry Smith . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
560060225df5SJacob Faibussowitsch - verticesAdj - (Optional) vertex adjacency array
5601a47d0d45SMatthew G. Knepley 
5602b09969d6SVaclav Hapla   Level: intermediate
5603a47d0d45SMatthew G. Knepley 
5604a1cb98faSBarry Smith   Notes:
5605a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
5606a1cb98faSBarry Smith   `DMPlexBuildFromCellListParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
5607a1cb98faSBarry Smith 
5608a1cb98faSBarry Smith   See `DMPlexBuildFromCellListParallel()` for an example and details about the topology-related parameters.
5609a1cb98faSBarry Smith 
5610a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
5611a1cb98faSBarry Smith 
56121cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
5613a47d0d45SMatthew G. Knepley @*/
5614d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromCellListParallelPetsc(MPI_Comm comm, PetscInt dim, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, PetscBool interpolate, const PetscInt cells[], PetscInt spaceDim, const PetscReal vertexCoords[], PetscSF *vertexSF, PetscInt **verticesAdj, DM *dm)
5615d71ae5a4SJacob Faibussowitsch {
5616a47d0d45SMatthew G. Knepley   PetscSF sfVert;
5617a47d0d45SMatthew G. Knepley 
5618a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
56199566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
56209566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
5621a47d0d45SMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
5622064a246eSJacob Faibussowitsch   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
56239566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
56249566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildFromCellListParallel(*dm, numCells, numVertices, NVertices, numCorners, cells, &sfVert, verticesAdj));
5625a47d0d45SMatthew G. Knepley   if (interpolate) {
56265fd9971aSMatthew G. Knepley     DM idm;
5627a47d0d45SMatthew G. Knepley 
56289566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
56299566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
5630a47d0d45SMatthew G. Knepley     *dm = idm;
5631a47d0d45SMatthew G. Knepley   }
56329566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
563318d54ad4SMichael Lange   if (vertexSF) *vertexSF = sfVert;
56349566063dSJacob Faibussowitsch   else PetscCall(PetscSFDestroy(&sfVert));
56353ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5636a47d0d45SMatthew G. Knepley }
5637a47d0d45SMatthew G. Knepley 
5638cc4c1da9SBarry Smith /*@
5639b0fe842aSMatthew G. Knepley   DMPlexCreateFromCellSectionParallel - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) and supports multiple celltypes
5640b0fe842aSMatthew G. Knepley 
5641b0fe842aSMatthew G. Knepley   Collective
5642b0fe842aSMatthew G. Knepley 
5643b0fe842aSMatthew G. Knepley   Input Parameters:
5644b0fe842aSMatthew G. Knepley + comm         - The communicator
5645b0fe842aSMatthew G. Knepley . dim          - The topological dimension of the mesh
5646b0fe842aSMatthew G. Knepley . numCells     - The number of cells owned by this process
5647b0fe842aSMatthew G. Knepley . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
5648b0fe842aSMatthew G. Knepley . NVertices    - The global number of vertices, or `PETSC_DECIDE`
5649b0fe842aSMatthew G. Knepley . cellSection  - The `PetscSection` giving the number of vertices for each cell (layout of cells)
5650b0fe842aSMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
5651b0fe842aSMatthew G. Knepley . cells        - An array of the global vertex numbers for each cell
5652b0fe842aSMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
5653b0fe842aSMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5654b0fe842aSMatthew G. Knepley 
5655b0fe842aSMatthew G. Knepley   Output Parameters:
5656b0fe842aSMatthew G. Knepley + dm          - The `DM`
5657b0fe842aSMatthew G. Knepley . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
5658b0fe842aSMatthew G. Knepley - verticesAdj - (Optional) vertex adjacency array
5659b0fe842aSMatthew G. Knepley 
5660b0fe842aSMatthew G. Knepley   Level: intermediate
5661b0fe842aSMatthew G. Knepley 
5662b0fe842aSMatthew G. Knepley   Notes:
5663b0fe842aSMatthew G. Knepley   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
5664b0fe842aSMatthew G. Knepley   `DMPlexBuildFromCellSectionParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
5665b0fe842aSMatthew G. Knepley 
5666b0fe842aSMatthew G. Knepley   See `DMPlexBuildFromCellSectionParallel()` for an example and details about the topology-related parameters.
5667b0fe842aSMatthew G. Knepley 
5668b0fe842aSMatthew G. Knepley   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
5669b0fe842aSMatthew G. Knepley 
5670b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
5671b0fe842aSMatthew G. Knepley @*/
5672b0fe842aSMatthew G. Knepley PetscErrorCode DMPlexCreateFromCellSectionParallel(MPI_Comm comm, PetscInt dim, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, PetscBool interpolate, const PetscInt cells[], PetscInt spaceDim, const PetscReal vertexCoords[], PetscSF *vertexSF, PetscInt **verticesAdj, DM *dm)
5673b0fe842aSMatthew G. Knepley {
5674b0fe842aSMatthew G. Knepley   PetscSF sfVert;
5675b0fe842aSMatthew G. Knepley 
5676b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
5677b0fe842aSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
5678b0fe842aSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
5679b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
5680b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
5681b0fe842aSMatthew G. Knepley   PetscCall(DMSetDimension(*dm, dim));
5682b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildFromCellSectionParallel(*dm, numCells, numVertices, NVertices, cellSection, cells, &sfVert, verticesAdj));
5683b0fe842aSMatthew G. Knepley   if (interpolate) {
5684b0fe842aSMatthew G. Knepley     DM idm;
5685b0fe842aSMatthew G. Knepley 
5686b0fe842aSMatthew G. Knepley     PetscCall(DMPlexInterpolate(*dm, &idm));
5687b0fe842aSMatthew G. Knepley     PetscCall(DMDestroy(dm));
5688b0fe842aSMatthew G. Knepley     *dm = idm;
5689b0fe842aSMatthew G. Knepley   }
5690b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
5691b0fe842aSMatthew G. Knepley   if (vertexSF) *vertexSF = sfVert;
5692b0fe842aSMatthew G. Knepley   else PetscCall(PetscSFDestroy(&sfVert));
5693b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
5694b0fe842aSMatthew G. Knepley }
5695b0fe842aSMatthew G. Knepley 
5696b0fe842aSMatthew G. Knepley /*@
5697a1cb98faSBarry Smith   DMPlexBuildFromCellList - Build `DMPLEX` topology from a list of vertices for each cell (common mesh generator output)
5698a1cb98faSBarry Smith 
569920f4b53cSBarry Smith   Collective; No Fortran Support
57009298eaa6SMatthew G Knepley 
57019298eaa6SMatthew G Knepley   Input Parameters:
5702a1cb98faSBarry Smith + dm          - The `DM`
5703b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5704a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DETERMINE`
57059298eaa6SMatthew G Knepley . numCorners  - The number of vertices for each cell
5706a3b724e8SBarry Smith - cells       - An array of `numCells` x `numCorners` numbers, the global vertex numbers for each cell
57079298eaa6SMatthew G Knepley 
5708b09969d6SVaclav Hapla   Level: advanced
57099298eaa6SMatthew G Knepley 
5710b09969d6SVaclav Hapla   Notes:
5711b09969d6SVaclav Hapla   Two triangles sharing a face
5712a1cb98faSBarry Smith .vb
57139298eaa6SMatthew G Knepley 
5714a1cb98faSBarry Smith         2
5715a1cb98faSBarry Smith       / | \
5716a1cb98faSBarry Smith      /  |  \
5717a1cb98faSBarry Smith     /   |   \
5718a1cb98faSBarry Smith    0  0 | 1  3
5719a1cb98faSBarry Smith     \   |   /
5720a1cb98faSBarry Smith      \  |  /
5721a1cb98faSBarry Smith       \ | /
5722a1cb98faSBarry Smith         1
5723a1cb98faSBarry Smith .ve
5724a1cb98faSBarry Smith   would have input
5725a1cb98faSBarry Smith .vb
5726a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5727a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5728a1cb98faSBarry Smith .ve
5729a1cb98faSBarry Smith   which would result in the `DMPLEX`
5730a1cb98faSBarry Smith .vb
5731a1cb98faSBarry Smith 
5732a1cb98faSBarry Smith         4
5733a1cb98faSBarry Smith       / | \
5734a1cb98faSBarry Smith      /  |  \
5735a1cb98faSBarry Smith     /   |   \
5736a1cb98faSBarry Smith    2  0 | 1  5
5737a1cb98faSBarry Smith     \   |   /
5738a1cb98faSBarry Smith      \  |  /
5739a1cb98faSBarry Smith       \ | /
5740a1cb98faSBarry Smith         3
5741a1cb98faSBarry Smith .ve
5742a1cb98faSBarry Smith 
5743a1cb98faSBarry Smith   If numVertices is `PETSC_DETERMINE`, it is computed by PETSc as the maximum vertex index in cells + 1.
574425b6865aSVaclav Hapla 
57451cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListPetsc()`
5746b09969d6SVaclav Hapla @*/
5747d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellList(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const PetscInt cells[])
5748d71ae5a4SJacob Faibussowitsch {
5749961cfab0SVaclav Hapla   PetscInt *cones, c, p, dim;
5750b09969d6SVaclav Hapla 
5751b09969d6SVaclav Hapla   PetscFunctionBegin;
57529566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
57539566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
575425b6865aSVaclav Hapla   /* Get/check global number of vertices */
575525b6865aSVaclav Hapla   {
575625b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
575725b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
575825b6865aSVaclav Hapla 
575925b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
57601690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
57619371c9d4SSatish Balay     for (i = 0; i < len; i++)
57629371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
576325b6865aSVaclav Hapla     ++NVerticesInCells;
576425b6865aSVaclav Hapla 
576525b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE) numVertices = NVerticesInCells;
57669371c9d4SSatish Balay     else
57679371c9d4SSatish Balay       PetscCheck(numVertices >= NVerticesInCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Specified number of vertices %" PetscInt_FMT " must be greater than or equal to the number of vertices in cells %" PetscInt_FMT, numVertices, NVerticesInCells);
576825b6865aSVaclav Hapla   }
57699566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
577048a46eb9SPierre Jolivet   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
57719566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
57729566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5773b09969d6SVaclav Hapla   for (c = 0; c < numCells; ++c) {
5774ad540459SPierre Jolivet     for (p = 0; p < numCorners; ++p) cones[c * numCorners + p] = cells[c * numCorners + p] + numCells;
5775b09969d6SVaclav Hapla   }
57769566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
57779566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
57789566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
57793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5780b09969d6SVaclav Hapla }
5781b09969d6SVaclav Hapla 
5782cc4c1da9SBarry Smith /*@
5783a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellList - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
5784a1cb98faSBarry Smith 
5785cc4c1da9SBarry Smith   Collective
5786b09969d6SVaclav Hapla 
5787b09969d6SVaclav Hapla   Input Parameters:
5788a1cb98faSBarry Smith + dm           - The `DM`
5789b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
5790b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5791b09969d6SVaclav Hapla 
5792b09969d6SVaclav Hapla   Level: advanced
5793b09969d6SVaclav Hapla 
57941cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellList()`
5795b09969d6SVaclav Hapla @*/
5796d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellList(DM dm, PetscInt spaceDim, const PetscReal vertexCoords[])
5797d71ae5a4SJacob Faibussowitsch {
5798b09969d6SVaclav Hapla   PetscSection coordSection;
5799b09969d6SVaclav Hapla   Vec          coordinates;
5800b09969d6SVaclav Hapla   DM           cdm;
5801b09969d6SVaclav Hapla   PetscScalar *coords;
58021edcf0b2SVaclav Hapla   PetscInt     v, vStart, vEnd, d;
5803b09969d6SVaclav Hapla 
5804b09969d6SVaclav Hapla   PetscFunctionBegin;
58059566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
58069566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
58071dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
58089566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
58099566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
58109566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
58119566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
58129566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
58131edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
58149566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
58159566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
5816b09969d6SVaclav Hapla   }
58179566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5818b09969d6SVaclav Hapla 
58199566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
58209566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(cdm, &coordinates));
58219566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
58229566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
58239566063dSJacob Faibussowitsch   PetscCall(VecGetArrayWrite(coordinates, &coords));
58241edcf0b2SVaclav Hapla   for (v = 0; v < vEnd - vStart; ++v) {
5825ad540459SPierre Jolivet     for (d = 0; d < spaceDim; ++d) coords[v * spaceDim + d] = vertexCoords[v * spaceDim + d];
5826b09969d6SVaclav Hapla   }
58279566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayWrite(coordinates, &coords));
58289566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
58299566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
58309566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
58313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5832b09969d6SVaclav Hapla }
5833b09969d6SVaclav Hapla 
5834b09969d6SVaclav Hapla /*@
5835a1cb98faSBarry Smith   DMPlexCreateFromCellListPetsc - Create `DMPLEX` from a list of vertices for each cell (common mesh generator output), but only process 0 takes in the input
58363df08285SMatthew G. Knepley 
5837a1cb98faSBarry Smith   Collective
5838b09969d6SVaclav Hapla 
5839b09969d6SVaclav Hapla   Input Parameters:
5840b09969d6SVaclav Hapla + comm         - The communicator
5841b09969d6SVaclav Hapla . dim          - The topological dimension of the mesh
58423df08285SMatthew G. Knepley . numCells     - The number of cells, only on process 0
5843a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`, only on process 0
58443df08285SMatthew G. Knepley . numCorners   - The number of vertices for each cell, only on process 0
5845b09969d6SVaclav Hapla . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
58463df08285SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the vertices for each cell, only on process 0
5847b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
58483df08285SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex, only on process 0
5849b09969d6SVaclav Hapla 
5850b09969d6SVaclav Hapla   Output Parameter:
5851a1cb98faSBarry Smith . dm - The `DM`, which only has points on process 0
585225b6865aSVaclav Hapla 
5853b09969d6SVaclav Hapla   Level: intermediate
5854b09969d6SVaclav Hapla 
5855a1cb98faSBarry Smith   Notes:
5856a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, `DMPlexBuildFromCellList()`,
5857a1cb98faSBarry Smith   `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellList()`
5858a1cb98faSBarry Smith 
5859a1cb98faSBarry Smith   See `DMPlexBuildFromCellList()` for an example and details about the topology-related parameters.
5860a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellList()` for details about the geometry-related parameters.
5861a1cb98faSBarry Smith   See `DMPlexCreateFromCellListParallelPetsc()` for parallel input
5862a1cb98faSBarry Smith 
58631cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellList()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
58649298eaa6SMatthew G Knepley @*/
5865d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromCellListPetsc(MPI_Comm comm, PetscInt dim, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, PetscBool interpolate, const PetscInt cells[], PetscInt spaceDim, const PetscReal vertexCoords[], DM *dm)
5866d71ae5a4SJacob Faibussowitsch {
58673df08285SMatthew G. Knepley   PetscMPIInt rank;
58689298eaa6SMatthew G Knepley 
58699298eaa6SMatthew G Knepley   PetscFunctionBegin;
587028b400f6SJacob Faibussowitsch   PetscCheck(dim, comm, PETSC_ERR_ARG_OUTOFRANGE, "This is not appropriate for 0-dimensional meshes. Consider either creating the DM using DMPlexCreateFromDAG(), by hand, or using DMSwarm.");
58719566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
58729566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
58739566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
58749566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
5875c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(*dm, numCells, numVertices, numCorners, cells));
58769566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(*dm, 0, 0, 0, NULL));
58779298eaa6SMatthew G Knepley   if (interpolate) {
58785fd9971aSMatthew G. Knepley     DM idm;
58799298eaa6SMatthew G Knepley 
58809566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
58819566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
58829298eaa6SMatthew G Knepley     *dm = idm;
58839298eaa6SMatthew G Knepley   }
5884c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, vertexCoords));
58859566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, NULL));
58863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
58879298eaa6SMatthew G Knepley }
58889298eaa6SMatthew G Knepley 
5889939f6067SMatthew G. Knepley /*@
589020f4b53cSBarry Smith   DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a `DM`
5891939f6067SMatthew G. Knepley 
5892939f6067SMatthew G. Knepley   Input Parameters:
589320f4b53cSBarry Smith + dm               - The empty `DM` object, usually from `DMCreate()` and `DMSetDimension()`
5894939f6067SMatthew G. Knepley . depth            - The depth of the DAG
589520f4b53cSBarry Smith . numPoints        - Array of size depth + 1 containing the number of points at each `depth`
5896939f6067SMatthew G. Knepley . coneSize         - The cone size of each point
5897939f6067SMatthew G. Knepley . cones            - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point
5898939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point
589920f4b53cSBarry Smith - vertexCoords     - An array of `numPoints`[0]*spacedim numbers representing the coordinates of each vertex, with spacedim the value set via `DMSetCoordinateDim()`
5900939f6067SMatthew G. Knepley 
5901939f6067SMatthew G. Knepley   Output Parameter:
590220f4b53cSBarry Smith . dm - The `DM`
590320f4b53cSBarry Smith 
590420f4b53cSBarry Smith   Level: advanced
5905939f6067SMatthew G. Knepley 
5906a1cb98faSBarry Smith   Note:
5907a1cb98faSBarry Smith   Two triangles sharing a face would have input
5908a1cb98faSBarry Smith .vb
5909a1cb98faSBarry Smith   depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0]
5910a1cb98faSBarry Smith   cones = [2 3 4  3 5 4], coneOrientations = [0 0 0  0 0 0]
5911a1cb98faSBarry Smith  vertexCoords = [-1.0 0.0  0.0 -1.0  0.0 1.0  1.0 0.0]
5912a1cb98faSBarry Smith .ve
5913939f6067SMatthew G. Knepley   which would result in the DMPlex
5914a1cb98faSBarry Smith .vb
5915a1cb98faSBarry Smith         4
5916a1cb98faSBarry Smith       / | \
5917a1cb98faSBarry Smith      /  |  \
5918a1cb98faSBarry Smith     /   |   \
5919a1cb98faSBarry Smith    2  0 | 1  5
5920a1cb98faSBarry Smith     \   |   /
5921a1cb98faSBarry Smith      \  |  /
5922a1cb98faSBarry Smith       \ | /
5923a1cb98faSBarry Smith         3
5924a1cb98faSBarry Smith .ve
5925a1cb98faSBarry Smith   Notice that all points are numbered consecutively, unlike `DMPlexCreateFromCellListPetsc()`
5926939f6067SMatthew G. Knepley 
59271cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
5928939f6067SMatthew G. Knepley @*/
5929d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
5930d71ae5a4SJacob Faibussowitsch {
59319298eaa6SMatthew G Knepley   Vec          coordinates;
59329298eaa6SMatthew G Knepley   PetscSection coordSection;
59339298eaa6SMatthew G Knepley   PetscScalar *coords;
5934811e8653SToby Isaac   PetscInt     coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off;
59359298eaa6SMatthew G Knepley 
59369298eaa6SMatthew G Knepley   PetscFunctionBegin;
59379566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
59389566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dimEmbed));
593963a3b9bcSJacob Faibussowitsch   PetscCheck(dimEmbed >= dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Embedding dimension %" PetscInt_FMT " cannot be less than intrinsic dimension %" PetscInt_FMT, dimEmbed, dim);
59409298eaa6SMatthew G Knepley   for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
59419566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, pStart, pEnd));
59429298eaa6SMatthew G Knepley   for (p = pStart; p < pEnd; ++p) {
59439566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeSize(dm, p, coneSize[p - pStart]));
5944ad540459SPierre Jolivet     if (firstVertex < 0 && !coneSize[p - pStart]) firstVertex = p - pStart;
594597e052ccSToby Isaac   }
59461dca8a05SBarry Smith   PetscCheck(firstVertex >= 0 || !numPoints[0], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Expected %" PetscInt_FMT " vertices but could not find any", numPoints[0]);
59479566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm)); /* Allocate space for cones */
59489298eaa6SMatthew G Knepley   for (p = pStart, off = 0; p < pEnd; off += coneSize[p - pStart], ++p) {
59499566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCone(dm, p, &cones[off]));
59509566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeOrientation(dm, p, &coneOrientations[off]));
59519298eaa6SMatthew G Knepley   }
59529566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
59539566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
59549298eaa6SMatthew G Knepley   /* Build coordinates */
59559566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
59569566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
59579566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed));
59589566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numPoints[0]));
59599298eaa6SMatthew G Knepley   for (v = firstVertex; v < firstVertex + numPoints[0]; ++v) {
59609566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed));
59619566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed));
59629298eaa6SMatthew G Knepley   }
59639566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
59649566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
59659566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
59669566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
59679566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
59689566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, dimEmbed));
59699566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
59709318fe57SMatthew G. Knepley   if (vertexCoords) {
59719566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
59729298eaa6SMatthew G Knepley     for (v = 0; v < numPoints[0]; ++v) {
59739298eaa6SMatthew G Knepley       PetscInt off;
59749298eaa6SMatthew G Knepley 
59759566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSection, v + firstVertex, &off));
5976ad540459SPierre Jolivet       for (d = 0; d < dimEmbed; ++d) coords[off + d] = vertexCoords[v * dimEmbed + d];
59779298eaa6SMatthew G Knepley     }
59789318fe57SMatthew G. Knepley   }
59799566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
59809566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
59819566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
59823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
59839298eaa6SMatthew G Knepley }
59848415267dSToby Isaac 
5985a4e35b19SJacob Faibussowitsch /*
5986a1cb98faSBarry Smith   DMPlexCreateCellVertexFromFile - Create a `DMPLEX` mesh from a simple cell-vertex file.
5987a1cb98faSBarry Smith 
5988a1cb98faSBarry Smith   Collective
59898ca92349SMatthew G. Knepley 
59908ca92349SMatthew G. Knepley + comm        - The MPI communicator
59918ca92349SMatthew G. Knepley . filename    - Name of the .dat file
59928ca92349SMatthew G. Knepley - interpolate - Create faces and edges in the mesh
59938ca92349SMatthew G. Knepley 
59948ca92349SMatthew G. Knepley   Output Parameter:
5995a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
59968ca92349SMatthew G. Knepley 
59978ca92349SMatthew G. Knepley   Level: beginner
59988ca92349SMatthew G. Knepley 
5999a1cb98faSBarry Smith   Note:
6000a1cb98faSBarry Smith   The format is the simplest possible:
6001a1cb98faSBarry Smith .vb
6002d0812dedSMatthew G. Knepley   dim Ne Nv Nc Nl
6003d0812dedSMatthew G. Knepley   v_1 v_2 ... v_Nc
6004d0812dedSMatthew G. Knepley   ...
6005d0812dedSMatthew G. Knepley   x y z marker_1 ... marker_Nl
6006a1cb98faSBarry Smith .ve
6007a1cb98faSBarry Smith 
6008a1cb98faSBarry Smith   Developer Note:
6009a1cb98faSBarry Smith   Should use a `PetscViewer` not a filename
6010a1cb98faSBarry Smith 
60116afe31f6SMartin Diehl .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateGmsh()`, `DMPlexCreate()`
6012a4e35b19SJacob Faibussowitsch */
6013ff6a9541SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCellVertexFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
6014d71ae5a4SJacob Faibussowitsch {
60158ca92349SMatthew G. Knepley   DMLabel      marker;
60168ca92349SMatthew G. Knepley   PetscViewer  viewer;
60178ca92349SMatthew G. Knepley   Vec          coordinates;
60188ca92349SMatthew G. Knepley   PetscSection coordSection;
60198ca92349SMatthew G. Knepley   PetscScalar *coords;
60208ca92349SMatthew G. Knepley   char         line[PETSC_MAX_PATH_LEN];
6021d0812dedSMatthew G. Knepley   PetscInt     cdim, coordSize, v, c, d;
60228ca92349SMatthew G. Knepley   PetscMPIInt  rank;
6023d0812dedSMatthew G. Knepley   int          snum, dim, Nv, Nc, Ncn, Nl;
60248ca92349SMatthew G. Knepley 
60258ca92349SMatthew G. Knepley   PetscFunctionBegin;
60269566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
60279566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(comm, &viewer));
60289566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII));
60299566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
60309566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetName(viewer, filename));
6031dd400576SPatrick Sanan   if (rank == 0) {
6032d0812dedSMatthew G. Knepley     PetscCall(PetscViewerRead(viewer, line, 5, NULL, PETSC_STRING));
6033d0812dedSMatthew G. Knepley     snum = sscanf(line, "%d %d %d %d %d", &dim, &Nc, &Nv, &Ncn, &Nl);
6034d0812dedSMatthew G. Knepley     PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
603525ce1634SJed Brown   } else {
6036f8d5e320SMatthew G. Knepley     Nc = Nv = Ncn = Nl = 0;
60378ca92349SMatthew G. Knepley   }
6038d0812dedSMatthew G. Knepley   PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm));
6039d0812dedSMatthew G. Knepley   cdim = (PetscInt)dim;
60409566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
60419566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
60429566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(*dm, 0, Nc + Nv));
6043d0812dedSMatthew G. Knepley   PetscCall(DMSetDimension(*dm, (PetscInt)dim));
60449566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(*dm, cdim));
60458ca92349SMatthew G. Knepley   /* Read topology */
6046dd400576SPatrick Sanan   if (rank == 0) {
6047f8d5e320SMatthew G. Knepley     char     format[PETSC_MAX_PATH_LEN];
6048f8d5e320SMatthew G. Knepley     PetscInt cone[8];
60498ca92349SMatthew G. Knepley     int      vbuf[8], v;
60508ca92349SMatthew G. Knepley 
60519371c9d4SSatish Balay     for (c = 0; c < Ncn; ++c) {
60529371c9d4SSatish Balay       format[c * 3 + 0] = '%';
60539371c9d4SSatish Balay       format[c * 3 + 1] = 'd';
60549371c9d4SSatish Balay       format[c * 3 + 2] = ' ';
60559371c9d4SSatish Balay     }
6056f8d5e320SMatthew G. Knepley     format[Ncn * 3 - 1] = '\0';
60579566063dSJacob Faibussowitsch     for (c = 0; c < Nc; ++c) PetscCall(DMPlexSetConeSize(*dm, c, Ncn));
60589566063dSJacob Faibussowitsch     PetscCall(DMSetUp(*dm));
60598ca92349SMatthew G. Knepley     for (c = 0; c < Nc; ++c) {
60609566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, Ncn, NULL, PETSC_STRING));
6061f8d5e320SMatthew G. Knepley       switch (Ncn) {
6062d71ae5a4SJacob Faibussowitsch       case 2:
6063d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1]);
6064d71ae5a4SJacob Faibussowitsch         break;
6065d71ae5a4SJacob Faibussowitsch       case 3:
6066d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2]);
6067d71ae5a4SJacob Faibussowitsch         break;
6068d71ae5a4SJacob Faibussowitsch       case 4:
6069d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3]);
6070d71ae5a4SJacob Faibussowitsch         break;
6071d71ae5a4SJacob Faibussowitsch       case 6:
6072d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5]);
6073d71ae5a4SJacob Faibussowitsch         break;
6074d71ae5a4SJacob Faibussowitsch       case 8:
6075d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5], &vbuf[6], &vbuf[7]);
6076d71ae5a4SJacob Faibussowitsch         break;
6077d71ae5a4SJacob Faibussowitsch       default:
6078d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No cell shape with %d vertices", Ncn);
6079f8d5e320SMatthew G. Knepley       }
608008401ef6SPierre Jolivet       PetscCheck(snum == Ncn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6081f8d5e320SMatthew G. Knepley       for (v = 0; v < Ncn; ++v) cone[v] = vbuf[v] + Nc;
60828ca92349SMatthew G. Knepley       /* Hexahedra are inverted */
6083f8d5e320SMatthew G. Knepley       if (Ncn == 8) {
60848ca92349SMatthew G. Knepley         PetscInt tmp = cone[1];
60858ca92349SMatthew G. Knepley         cone[1]      = cone[3];
60868ca92349SMatthew G. Knepley         cone[3]      = tmp;
60878ca92349SMatthew G. Knepley       }
60889566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(*dm, c, cone));
60898ca92349SMatthew G. Knepley     }
60908ca92349SMatthew G. Knepley   }
60919566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(*dm));
60929566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(*dm));
60938ca92349SMatthew G. Knepley   /* Read coordinates */
60949566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(*dm, &coordSection));
60959566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
60969566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
60979566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, Nc, Nc + Nv));
60988ca92349SMatthew G. Knepley   for (v = Nc; v < Nc + Nv; ++v) {
60999566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
61009566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
61018ca92349SMatthew G. Knepley   }
61029566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
61039566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
61049566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
61059566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
61069566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
61079566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
61089566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
61099566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
6110dd400576SPatrick Sanan   if (rank == 0) {
6111f8d5e320SMatthew G. Knepley     char   format[PETSC_MAX_PATH_LEN];
61128ca92349SMatthew G. Knepley     double x[3];
6113f8d5e320SMatthew G. Knepley     int    l, val[3];
61148ca92349SMatthew G. Knepley 
6115f8d5e320SMatthew G. Knepley     if (Nl) {
61169371c9d4SSatish Balay       for (l = 0; l < Nl; ++l) {
61179371c9d4SSatish Balay         format[l * 3 + 0] = '%';
61189371c9d4SSatish Balay         format[l * 3 + 1] = 'd';
61199371c9d4SSatish Balay         format[l * 3 + 2] = ' ';
61209371c9d4SSatish Balay       }
6121f8d5e320SMatthew G. Knepley       format[Nl * 3 - 1] = '\0';
61229566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
61239566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &marker));
6124f8d5e320SMatthew G. Knepley     }
61258ca92349SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
61269566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, 3 + Nl, NULL, PETSC_STRING));
6127f8d5e320SMatthew G. Knepley       snum = sscanf(line, "%lg %lg %lg", &x[0], &x[1], &x[2]);
612808401ef6SPierre Jolivet       PetscCheck(snum == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6129f8d5e320SMatthew G. Knepley       switch (Nl) {
6130d71ae5a4SJacob Faibussowitsch       case 0:
6131d71ae5a4SJacob Faibussowitsch         snum = 0;
6132d71ae5a4SJacob Faibussowitsch         break;
6133d71ae5a4SJacob Faibussowitsch       case 1:
6134d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0]);
6135d71ae5a4SJacob Faibussowitsch         break;
6136d71ae5a4SJacob Faibussowitsch       case 2:
6137d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1]);
6138d71ae5a4SJacob Faibussowitsch         break;
6139d71ae5a4SJacob Faibussowitsch       case 3:
6140d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1], &val[2]);
6141d71ae5a4SJacob Faibussowitsch         break;
6142d71ae5a4SJacob Faibussowitsch       default:
6143d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Request support for %d labels", Nl);
6144f8d5e320SMatthew G. Knepley       }
614508401ef6SPierre Jolivet       PetscCheck(snum == Nl, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
61468ca92349SMatthew G. Knepley       for (d = 0; d < cdim; ++d) coords[v * cdim + d] = x[d];
61479566063dSJacob Faibussowitsch       for (l = 0; l < Nl; ++l) PetscCall(DMLabelSetValue(marker, v + Nc, val[l]));
61488ca92349SMatthew G. Knepley     }
61498ca92349SMatthew G. Knepley   }
61509566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
61519566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
61529566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
61539566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&viewer));
61548ca92349SMatthew G. Knepley   if (interpolate) {
61558ca92349SMatthew G. Knepley     DM      idm;
61568ca92349SMatthew G. Knepley     DMLabel bdlabel;
61578ca92349SMatthew G. Knepley 
61589566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
61599566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
61608ca92349SMatthew G. Knepley     *dm = idm;
61618ca92349SMatthew G. Knepley 
6162f8d5e320SMatthew G. Knepley     if (!Nl) {
61639566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
61649566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &bdlabel));
61659566063dSJacob Faibussowitsch       PetscCall(DMPlexMarkBoundaryFaces(*dm, PETSC_DETERMINE, bdlabel));
61669566063dSJacob Faibussowitsch       PetscCall(DMPlexLabelComplete(*dm, bdlabel));
61678ca92349SMatthew G. Knepley     }
6168f8d5e320SMatthew G. Knepley   }
61693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
61708ca92349SMatthew G. Knepley }
61718ca92349SMatthew G. Knepley 
6172cc4c1da9SBarry Smith /*@
6173a1cb98faSBarry Smith   DMPlexCreateFromFile - This takes a filename and produces a `DM`
6174a1cb98faSBarry Smith 
6175a1cb98faSBarry Smith   Collective
6176ca522641SMatthew G. Knepley 
6177ca522641SMatthew G. Knepley   Input Parameters:
6178ca522641SMatthew G. Knepley + comm        - The communicator
6179ca522641SMatthew G. Knepley . filename    - A file name
6180a1cb98faSBarry Smith . plexname    - The object name of the resulting `DM`, also used for intra-datafile lookup by some formats
6181ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
6182ca522641SMatthew G. Knepley 
6183ca522641SMatthew G. Knepley   Output Parameter:
6184a1cb98faSBarry Smith . dm - The `DM`
6185ca522641SMatthew G. Knepley 
6186a1cb98faSBarry Smith   Options Database Key:
6187a1cb98faSBarry Smith . -dm_plex_create_from_hdf5_xdmf - use the `PETSC_VIEWER_HDF5_XDMF` format for reading HDF5
618802ef0d99SVaclav Hapla 
618937fdd005SBarry Smith   Use `-dm_plex_create_ prefix` to pass options to the internal `PetscViewer`, e.g.
6190bca97951SVaclav Hapla $ -dm_plex_create_viewer_hdf5_collective
6191bca97951SVaclav Hapla 
6192ca522641SMatthew G. Knepley   Level: beginner
6193ca522641SMatthew G. Knepley 
6194a1cb98faSBarry Smith   Notes:
6195a1cb98faSBarry Smith   Using `PETSCVIEWERHDF5` type with `PETSC_VIEWER_HDF5_PETSC` format, one can save multiple `DMPLEX`
6196a1cb98faSBarry Smith   meshes in a single HDF5 file. This in turn requires one to name the `DMPLEX` object with `PetscObjectSetName()`
6197a1cb98faSBarry Smith   before saving it with `DMView()` and before loading it with `DMLoad()` for identification of the mesh object.
6198a1cb98faSBarry Smith   The input parameter name is thus used to name the `DMPLEX` object when `DMPlexCreateFromFile()` internally
6199a1cb98faSBarry Smith   calls `DMLoad()`. Currently, name is ignored for other viewer types and/or formats.
6200a1cb98faSBarry Smith 
62011cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`, `PetscObjectSetName()`, `DMView()`, `DMLoad()`
6202ca522641SMatthew G. Knepley @*/
6203d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], const char plexname[], PetscBool interpolate, DM *dm)
6204d71ae5a4SJacob Faibussowitsch {
6205ef3a5affSJacob Faibussowitsch   const char  extGmsh[]      = ".msh";
6206ef3a5affSJacob Faibussowitsch   const char  extGmsh2[]     = ".msh2";
6207ef3a5affSJacob Faibussowitsch   const char  extGmsh4[]     = ".msh4";
6208ef3a5affSJacob Faibussowitsch   const char  extCGNS[]      = ".cgns";
6209ef3a5affSJacob Faibussowitsch   const char  extExodus[]    = ".exo";
6210ef3a5affSJacob Faibussowitsch   const char  extExodus_e[]  = ".e";
6211ef3a5affSJacob Faibussowitsch   const char  extGenesis[]   = ".gen";
6212ef3a5affSJacob Faibussowitsch   const char  extFluent[]    = ".cas";
6213ef3a5affSJacob Faibussowitsch   const char  extHDF5[]      = ".h5";
62146f2c871aSStefano Zampini   const char  extXDMFHDF5[]  = ".xdmf.h5";
6215ef3a5affSJacob Faibussowitsch   const char  extPLY[]       = ".ply";
6216ef3a5affSJacob Faibussowitsch   const char  extEGADSLite[] = ".egadslite";
6217ef3a5affSJacob Faibussowitsch   const char  extEGADS[]     = ".egads";
6218ef3a5affSJacob Faibussowitsch   const char  extIGES[]      = ".igs";
6219ef3a5affSJacob Faibussowitsch   const char  extSTEP[]      = ".stp";
6220ef3a5affSJacob Faibussowitsch   const char  extCV[]        = ".dat";
6221ca522641SMatthew G. Knepley   size_t      len;
62226afe31f6SMartin Diehl   PetscBool   isGmsh, isGmsh2, isGmsh4, isCGNS, isExodus, isGenesis, isFluent, isHDF5, isPLY, isEGADSLite, isEGADS, isIGES, isSTEP, isCV, isXDMFHDF5;
6223ca522641SMatthew G. Knepley   PetscMPIInt rank;
6224ca522641SMatthew G. Knepley 
6225ca522641SMatthew G. Knepley   PetscFunctionBegin;
62264f572ea9SToby Isaac   PetscAssertPointer(filename, 2);
62274f572ea9SToby Isaac   if (plexname) PetscAssertPointer(plexname, 3);
62284f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
62299566063dSJacob Faibussowitsch   PetscCall(DMInitializePackage());
62309566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromFile, 0, 0, 0, 0));
62319566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
62329566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(filename, &len));
623328b400f6SJacob Faibussowitsch   PetscCheck(len, comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path");
6234ef3a5affSJacob Faibussowitsch 
62359371c9d4SSatish Balay #define CheckExtension(extension__, is_extension__) \
62369371c9d4SSatish Balay   do { \
6237274aaeaaSJacob Faibussowitsch     PetscAssert(sizeof(extension__), comm, PETSC_ERR_PLIB, "Zero-size extension: %s", extension__); \
6238274aaeaaSJacob Faibussowitsch     /* don't count the null-terminator at the end */ \
6239274aaeaaSJacob Faibussowitsch     const size_t ext_len = sizeof(extension__) - 1; \
6240274aaeaaSJacob Faibussowitsch     if (len < ext_len) { \
6241ef3a5affSJacob Faibussowitsch       is_extension__ = PETSC_FALSE; \
6242ef3a5affSJacob Faibussowitsch     } else { \
6243274aaeaaSJacob Faibussowitsch       PetscCall(PetscStrncmp(filename + len - ext_len, extension__, ext_len, &is_extension__)); \
6244ef3a5affSJacob Faibussowitsch     } \
6245ef3a5affSJacob Faibussowitsch   } while (0)
6246ef3a5affSJacob Faibussowitsch 
6247ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh, isGmsh);
6248ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh2, isGmsh2);
6249ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh4, isGmsh4);
6250ef3a5affSJacob Faibussowitsch   CheckExtension(extCGNS, isCGNS);
6251ef3a5affSJacob Faibussowitsch   CheckExtension(extExodus, isExodus);
6252ef3a5affSJacob Faibussowitsch   if (!isExodus) CheckExtension(extExodus_e, isExodus);
6253ef3a5affSJacob Faibussowitsch   CheckExtension(extGenesis, isGenesis);
6254ef3a5affSJacob Faibussowitsch   CheckExtension(extFluent, isFluent);
6255ef3a5affSJacob Faibussowitsch   CheckExtension(extHDF5, isHDF5);
6256ef3a5affSJacob Faibussowitsch   CheckExtension(extPLY, isPLY);
6257ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADSLite, isEGADSLite);
6258ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADS, isEGADS);
6259ef3a5affSJacob Faibussowitsch   CheckExtension(extIGES, isIGES);
6260ef3a5affSJacob Faibussowitsch   CheckExtension(extSTEP, isSTEP);
6261ef3a5affSJacob Faibussowitsch   CheckExtension(extCV, isCV);
62626f2c871aSStefano Zampini   CheckExtension(extXDMFHDF5, isXDMFHDF5);
6263ef3a5affSJacob Faibussowitsch 
6264ef3a5affSJacob Faibussowitsch #undef CheckExtension
6265ef3a5affSJacob Faibussowitsch 
6266de78e4feSLisandro Dalcin   if (isGmsh || isGmsh2 || isGmsh4) {
62679566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateGmshFromFile(comm, filename, interpolate, dm));
6268ca522641SMatthew G. Knepley   } else if (isCGNS) {
62699566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm));
627090c68965SMatthew G. Knepley   } else if (isExodus || isGenesis) {
62719566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateExodusFromFile(comm, filename, interpolate, dm));
62722f0bd6dcSMichael Lange   } else if (isFluent) {
62739566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFluentFromFile(comm, filename, interpolate, dm));
6274cc2f8f65SMatthew G. Knepley   } else if (isHDF5) {
6275cc2f8f65SMatthew G. Knepley     PetscViewer viewer;
6276cc2f8f65SMatthew G. Knepley 
627743b242b4SVaclav Hapla     /* PETSC_VIEWER_HDF5_XDMF is used if the filename ends with .xdmf.h5, or if -dm_plex_create_from_hdf5_xdmf option is present */
62786f2c871aSStefano Zampini     PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_create_from_hdf5_xdmf", &isXDMFHDF5, NULL));
62799566063dSJacob Faibussowitsch     PetscCall(PetscViewerCreate(comm, &viewer));
62809566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetType(viewer, PETSCVIEWERHDF5));
62819566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetOptionsPrefix(viewer, "dm_plex_create_"));
62829566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetFromOptions(viewer));
62839566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
62849566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetName(viewer, filename));
6285cd7e8a5eSksagiyam 
62869566063dSJacob Faibussowitsch     PetscCall(DMCreate(comm, dm));
6287f4f49eeaSPierre Jolivet     PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
62889566063dSJacob Faibussowitsch     PetscCall(DMSetType(*dm, DMPLEX));
62896f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPushFormat(viewer, PETSC_VIEWER_HDF5_XDMF));
62909566063dSJacob Faibussowitsch     PetscCall(DMLoad(*dm, viewer));
62916f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPopFormat(viewer));
62929566063dSJacob Faibussowitsch     PetscCall(PetscViewerDestroy(&viewer));
62935fd9971aSMatthew G. Knepley 
62945fd9971aSMatthew G. Knepley     if (interpolate) {
62955fd9971aSMatthew G. Knepley       DM idm;
62965fd9971aSMatthew G. Knepley 
62979566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(*dm, &idm));
62989566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
62995fd9971aSMatthew G. Knepley       *dm = idm;
63005fd9971aSMatthew G. Knepley     }
6301f2801cd6SMatthew G. Knepley   } else if (isPLY) {
63029566063dSJacob Faibussowitsch     PetscCall(DMPlexCreatePLYFromFile(comm, filename, interpolate, dm));
6303c1cad2e7SMatthew G. Knepley   } else if (isEGADSLite || isEGADS || isIGES || isSTEP) {
63049566063dSJacob Faibussowitsch     if (isEGADSLite) PetscCall(DMPlexCreateEGADSLiteFromFile(comm, filename, dm));
63059566063dSJacob Faibussowitsch     else PetscCall(DMPlexCreateEGADSFromFile(comm, filename, dm));
63067bee2925SMatthew Knepley     if (!interpolate) {
63077bee2925SMatthew Knepley       DM udm;
63087bee2925SMatthew Knepley 
63099566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(*dm, &udm));
63109566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
63117bee2925SMatthew Knepley       *dm = udm;
63127bee2925SMatthew Knepley     }
63138ca92349SMatthew G. Knepley   } else if (isCV) {
63149566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCellVertexFromFile(comm, filename, interpolate, dm));
631598921bdaSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename);
63169566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(plexname, &len));
6317f4f49eeaSPierre Jolivet   if (len) PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
63189566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromFile, 0, 0, 0, 0));
63193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6320ca522641SMatthew G. Knepley }
632112b8a6daSStefano Zampini 
6322cc4c1da9SBarry Smith /*@
63239f6c5813SMatthew G. Knepley   DMPlexCreateEphemeral - This takes a `DMPlexTransform` and a base `DMPlex` and produces an ephemeral `DM`, meaning one that is created on the fly in response to queries.
63249f6c5813SMatthew G. Knepley 
63250528010dSStefano Zampini   Input Parameters:
63260528010dSStefano Zampini + tr     - The `DMPlexTransform`
63270528010dSStefano Zampini - prefix - An options prefix, or NULL
63289f6c5813SMatthew G. Knepley 
63299f6c5813SMatthew G. Knepley   Output Parameter:
63309f6c5813SMatthew G. Knepley . dm - The `DM`
63319f6c5813SMatthew G. Knepley 
63329f6c5813SMatthew G. Knepley   Level: beginner
63339f6c5813SMatthew G. Knepley 
633420f4b53cSBarry Smith   Notes:
633520f4b53cSBarry Smith   An emphemeral mesh is one that is not stored concretely, as in the default `DMPLEX` implementation, but rather is produced on the fly in response to queries, using information from the transform and the base mesh.
633620f4b53cSBarry Smith 
63379f6c5813SMatthew G. Knepley .seealso: `DMPlexCreateFromFile`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
63389f6c5813SMatthew G. Knepley @*/
63390528010dSStefano Zampini PetscErrorCode DMPlexCreateEphemeral(DMPlexTransform tr, const char prefix[], DM *dm)
63409f6c5813SMatthew G. Knepley {
63410528010dSStefano Zampini   DM           bdm, bcdm, cdm;
63420528010dSStefano Zampini   Vec          coordinates, coordinatesNew;
63430528010dSStefano Zampini   PetscSection cs;
6344817b2c36SMatthew G. Knepley   PetscInt     cdim, Nl;
63459f6c5813SMatthew G. Knepley 
63469f6c5813SMatthew G. Knepley   PetscFunctionBegin;
63479f6c5813SMatthew G. Knepley   PetscCall(DMCreate(PetscObjectComm((PetscObject)tr), dm));
63489f6c5813SMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
63490528010dSStefano Zampini   ((DM_Plex *)(*dm)->data)->interpolated = DMPLEX_INTERPOLATED_FULL;
63500528010dSStefano Zampini   // Handle coordinates
63510528010dSStefano Zampini   PetscCall(DMPlexTransformGetDM(tr, &bdm));
6352817b2c36SMatthew G. Knepley   PetscCall(DMPlexTransformSetDimensions(tr, bdm, *dm));
6353817b2c36SMatthew G. Knepley   PetscCall(DMGetCoordinateDim(*dm, &cdim));
63540528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(bdm, &bcdm));
63550528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(*dm, &cdm));
63560528010dSStefano Zampini   PetscCall(DMCopyDisc(bcdm, cdm));
63570528010dSStefano Zampini   PetscCall(DMGetLocalSection(cdm, &cs));
63580528010dSStefano Zampini   PetscCall(PetscSectionSetNumFields(cs, 1));
63590528010dSStefano Zampini   PetscCall(PetscSectionSetFieldComponents(cs, 0, cdim));
63600528010dSStefano Zampini   PetscCall(DMGetCoordinatesLocal(bdm, &coordinates));
63610528010dSStefano Zampini   PetscCall(VecDuplicate(coordinates, &coordinatesNew));
63620528010dSStefano Zampini   PetscCall(VecCopy(coordinates, coordinatesNew));
63630528010dSStefano Zampini   PetscCall(DMSetCoordinatesLocal(*dm, coordinatesNew));
63640528010dSStefano Zampini   PetscCall(VecDestroy(&coordinatesNew));
63659f6c5813SMatthew G. Knepley 
63669f6c5813SMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)tr));
63679f6c5813SMatthew G. Knepley   PetscCall(DMPlexTransformDestroy(&((DM_Plex *)(*dm)->data)->tr));
63689f6c5813SMatthew G. Knepley   ((DM_Plex *)(*dm)->data)->tr = tr;
63690528010dSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(*dm, PETSC_FALSE));
63700528010dSStefano Zampini   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*dm, prefix));
63710528010dSStefano Zampini   PetscCall(DMSetFromOptions(*dm));
63729f6c5813SMatthew G. Knepley 
63739f6c5813SMatthew G. Knepley   PetscCall(DMGetNumLabels(bdm, &Nl));
63749f6c5813SMatthew G. Knepley   for (PetscInt l = 0; l < Nl; ++l) {
63759f6c5813SMatthew G. Knepley     DMLabel     label, labelNew;
63769f6c5813SMatthew G. Knepley     const char *lname;
63779f6c5813SMatthew G. Knepley     PetscBool   isDepth, isCellType;
63789f6c5813SMatthew G. Knepley 
63799f6c5813SMatthew G. Knepley     PetscCall(DMGetLabelName(bdm, l, &lname));
63809f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "depth", &isDepth));
63819f6c5813SMatthew G. Knepley     if (isDepth) continue;
63829f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "celltype", &isCellType));
63839f6c5813SMatthew G. Knepley     if (isCellType) continue;
63849f6c5813SMatthew G. Knepley     PetscCall(DMCreateLabel(*dm, lname));
63859f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(bdm, lname, &label));
63869f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(*dm, lname, &labelNew));
63879f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetType(labelNew, DMLABELEPHEMERAL));
63889f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetLabel(labelNew, label));
63899f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetTransform(labelNew, tr));
63909f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetUp(labelNew));
63919f6c5813SMatthew G. Knepley   }
63923ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
63939f6c5813SMatthew G. Knepley }
6394