xref: /petsc/src/dm/impls/plex/plexcreate.c (revision 8d2ec52a4c95501bf60d703362f516a18e423877)
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;
30835f2295SStefano Zampini   if (dmin == dmout) PetscFunctionReturn(PETSC_SUCCESS);
3112a88998SMatthew G. Knepley   PetscCall(DMGetVecType(dmin, &vecType));
3212a88998SMatthew G. Knepley   PetscCall(DMSetVecType(dmout, vecType));
3312a88998SMatthew G. Knepley   PetscCall(DMGetMatType(dmin, &matType));
3412a88998SMatthew G. Knepley   PetscCall(DMSetMatType(dmout, matType));
35e600fa54SMatthew G. Knepley   if (copyPeriodicity) {
364fb89dddSMatthew G. Knepley     PetscCall(DMGetPeriodicity(dmin, &maxCell, &Lstart, &L));
374fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dmout, maxCell, Lstart, L));
383d0e8ed9SDavid Salac     PetscCall(DMLocalizeCoordinates(dmout));
39e600fa54SMatthew G. Knepley   }
409566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dmin, &dist));
419566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeSetDefault(dmout, dist));
426bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dmin, &reorder));
436bc1bd01Sksagiyam   PetscCall(DMPlexReorderSetDefault(dmout, reorder));
445962854dSMatthew G. Knepley   PetscCall(DMPlexGetUseCeed(dmin, &useCeed));
455962854dSMatthew G. Knepley   PetscCall(DMPlexSetUseCeed(dmout, useCeed));
46e600fa54SMatthew G. Knepley   ((DM_Plex *)dmout->data)->useHashLocation = ((DM_Plex *)dmin->data)->useHashLocation;
475962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printSetValues  = ((DM_Plex *)dmin->data)->printSetValues;
485962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFEM        = ((DM_Plex *)dmin->data)->printFEM;
495962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFVM        = ((DM_Plex *)dmin->data)->printFVM;
505962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printL2         = ((DM_Plex *)dmin->data)->printL2;
515962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printLocate     = ((DM_Plex *)dmin->data)->printLocate;
52a77a5016SMatthew G. Knepley   ((DM_Plex *)dmout->data)->printProject    = ((DM_Plex *)dmin->data)->printProject;
535962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printTol        = ((DM_Plex *)dmin->data)->printTol;
541baa6e33SBarry Smith   if (copyOverlap) PetscCall(DMPlexSetOverlap_Plex(dmout, dmin, 0));
553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
56e600fa54SMatthew G. Knepley }
57e600fa54SMatthew G. Knepley 
589318fe57SMatthew G. Knepley /* Replace dm with the contents of ndm, and then destroy ndm
599318fe57SMatthew G. Knepley    - Share the DM_Plex structure
609318fe57SMatthew G. Knepley    - Share the coordinates
619318fe57SMatthew G. Knepley    - Share the SF
629318fe57SMatthew G. Knepley */
63d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexReplace_Internal(DM dm, DM *ndm)
64d71ae5a4SJacob Faibussowitsch {
659318fe57SMatthew G. Knepley   PetscSF          sf;
669318fe57SMatthew G. Knepley   DM               dmNew = *ndm, coordDM, coarseDM;
679318fe57SMatthew G. Knepley   Vec              coords;
684fb89dddSMatthew G. Knepley   const PetscReal *maxCell, *Lstart, *L;
699318fe57SMatthew G. Knepley   PetscInt         dim, cdim;
70e535cce4SJames Wright   PetscBool        use_natural;
719318fe57SMatthew G. Knepley 
729318fe57SMatthew G. Knepley   PetscFunctionBegin;
739318fe57SMatthew G. Knepley   if (dm == dmNew) {
749566063dSJacob Faibussowitsch     PetscCall(DMDestroy(ndm));
753ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
769318fe57SMatthew G. Knepley   }
779318fe57SMatthew G. Knepley   dm->setupcalled = dmNew->setupcalled;
78d0812dedSMatthew G. Knepley   if (!dm->hdr.name) {
79d0812dedSMatthew G. Knepley     const char *name;
80d0812dedSMatthew G. Knepley 
81d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)*ndm, &name));
82d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
83d0812dedSMatthew G. Knepley   }
849566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dmNew, &dim));
859566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
869566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dmNew, &cdim));
879566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
889566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmNew, &sf));
899566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sf));
909566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmNew, &coordDM));
919566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmNew, &coords));
929566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dm, coordDM));
939566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coords));
946858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmNew, &coordDM));
956858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmNew, &coords));
966858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dm, coordDM));
976858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dm, coords));
989318fe57SMatthew G. Knepley   /* Do not want to create the coordinate field if it does not already exist, so do not call DMGetCoordinateField() */
996858538eSMatthew G. Knepley   PetscCall(DMFieldDestroy(&dm->coordinates[0].field));
1006858538eSMatthew G. Knepley   dm->coordinates[0].field            = dmNew->coordinates[0].field;
10161a622f3SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->coordFunc = ((DM_Plex *)dm->data)->coordFunc;
1024fb89dddSMatthew G. Knepley   PetscCall(DMGetPeriodicity(dmNew, &maxCell, &Lstart, &L));
1034fb89dddSMatthew G. Knepley   PetscCall(DMSetPeriodicity(dm, maxCell, Lstart, L));
104e535cce4SJames Wright   PetscCall(DMGetNaturalSF(dmNew, &sf));
105e535cce4SJames Wright   PetscCall(DMSetNaturalSF(dm, sf));
106e535cce4SJames Wright   PetscCall(DMGetUseNatural(dmNew, &use_natural));
107e535cce4SJames Wright   PetscCall(DMSetUseNatural(dm, use_natural));
1089566063dSJacob Faibussowitsch   PetscCall(DMDestroy_Plex(dm));
1099566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
1109318fe57SMatthew G. Knepley   dm->data = dmNew->data;
1119318fe57SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->refct++;
1121fca310dSJames Wright   {
1131fca310dSJames Wright     PetscInt       num_face_sfs;
1141fca310dSJames Wright     const PetscSF *sfs;
1151fca310dSJames Wright     PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &sfs));
1161fca310dSJames Wright     PetscCall(DMPlexSetIsoperiodicFaceSF(dm, num_face_sfs, (PetscSF *)sfs)); // for the compose function effect on dm
1171fca310dSJames Wright   }
1189566063dSJacob Faibussowitsch   PetscCall(DMDestroyLabelLinkList_Internal(dm));
1199566063dSJacob Faibussowitsch   PetscCall(DMCopyLabels(dmNew, dm, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
1209566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmNew, &coarseDM));
1219566063dSJacob Faibussowitsch   PetscCall(DMSetCoarseDM(dm, coarseDM));
1229566063dSJacob Faibussowitsch   PetscCall(DMDestroy(ndm));
1233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1249318fe57SMatthew G. Knepley }
1259318fe57SMatthew G. Knepley 
1269318fe57SMatthew G. Knepley /* Swap dm with the contents of dmNew
1279318fe57SMatthew G. Knepley    - Swap the DM_Plex structure
1289318fe57SMatthew G. Knepley    - Swap the coordinates
1299318fe57SMatthew G. Knepley    - Swap the point PetscSF
1309318fe57SMatthew G. Knepley */
131d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexSwap_Static(DM dmA, DM dmB)
132d71ae5a4SJacob Faibussowitsch {
1339318fe57SMatthew G. Knepley   DM          coordDMA, coordDMB;
1349318fe57SMatthew G. Knepley   Vec         coordsA, coordsB;
1359318fe57SMatthew G. Knepley   PetscSF     sfA, sfB;
1369318fe57SMatthew G. Knepley   DMField     fieldTmp;
1379318fe57SMatthew G. Knepley   void       *tmp;
1389318fe57SMatthew G. Knepley   DMLabelLink listTmp;
1399318fe57SMatthew G. Knepley   DMLabel     depthTmp;
1409318fe57SMatthew G. Knepley   PetscInt    tmpI;
1419318fe57SMatthew G. Knepley 
1429318fe57SMatthew G. Knepley   PetscFunctionBegin;
1433ba16761SJacob Faibussowitsch   if (dmA == dmB) PetscFunctionReturn(PETSC_SUCCESS);
1449566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmA, &sfA));
1459566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmB, &sfB));
1469566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)sfA));
1479566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmA, sfB));
1489566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmB, sfA));
1499566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)sfA));
1509318fe57SMatthew G. Knepley 
1519566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmA, &coordDMA));
1529566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmB, &coordDMB));
1539566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1549566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmA, coordDMB));
1559566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmB, coordDMA));
1569566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1579318fe57SMatthew G. Knepley 
1589566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmA, &coordsA));
1599566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmB, &coordsB));
1609566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordsA));
1619566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmA, coordsB));
1629566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmB, coordsA));
1639566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1649318fe57SMatthew G. Knepley 
1656858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmA, &coordDMA));
1666858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmB, &coordDMB));
1676858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1686858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmA, coordDMB));
1696858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmB, coordDMA));
1706858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1716858538eSMatthew G. Knepley 
1726858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmA, &coordsA));
1736858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmB, &coordsB));
1746858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordsA));
1756858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmA, coordsB));
1766858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmB, coordsA));
1776858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1786858538eSMatthew G. Knepley 
1796858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[0].field;
1806858538eSMatthew G. Knepley   dmA->coordinates[0].field = dmB->coordinates[0].field;
1816858538eSMatthew G. Knepley   dmB->coordinates[0].field = fieldTmp;
1826858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[1].field;
1836858538eSMatthew G. Knepley   dmA->coordinates[1].field = dmB->coordinates[1].field;
1846858538eSMatthew G. Knepley   dmB->coordinates[1].field = fieldTmp;
1859318fe57SMatthew G. Knepley   tmp                       = dmA->data;
1869318fe57SMatthew G. Knepley   dmA->data                 = dmB->data;
1879318fe57SMatthew G. Knepley   dmB->data                 = tmp;
1889318fe57SMatthew G. Knepley   listTmp                   = dmA->labels;
1899318fe57SMatthew G. Knepley   dmA->labels               = dmB->labels;
1909318fe57SMatthew G. Knepley   dmB->labels               = listTmp;
1919318fe57SMatthew G. Knepley   depthTmp                  = dmA->depthLabel;
1929318fe57SMatthew G. Knepley   dmA->depthLabel           = dmB->depthLabel;
1939318fe57SMatthew G. Knepley   dmB->depthLabel           = depthTmp;
1949318fe57SMatthew G. Knepley   depthTmp                  = dmA->celltypeLabel;
1959318fe57SMatthew G. Knepley   dmA->celltypeLabel        = dmB->celltypeLabel;
1969318fe57SMatthew G. Knepley   dmB->celltypeLabel        = depthTmp;
1979318fe57SMatthew G. Knepley   tmpI                      = dmA->levelup;
1989318fe57SMatthew G. Knepley   dmA->levelup              = dmB->levelup;
1999318fe57SMatthew G. Knepley   dmB->levelup              = tmpI;
2003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2019318fe57SMatthew G. Knepley }
2029318fe57SMatthew G. Knepley 
2033431e603SJed Brown PetscErrorCode DMPlexInterpolateInPlace_Internal(DM dm)
204d71ae5a4SJacob Faibussowitsch {
2059318fe57SMatthew G. Knepley   DM idm;
2069318fe57SMatthew G. Knepley 
2079318fe57SMatthew G. Knepley   PetscFunctionBegin;
2089566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolate(dm, &idm));
2099566063dSJacob Faibussowitsch   PetscCall(DMPlexCopyCoordinates(dm, idm));
21069d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &idm));
2113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2129318fe57SMatthew G. Knepley }
2139318fe57SMatthew G. Knepley 
2149318fe57SMatthew G. Knepley /*@C
2159318fe57SMatthew G. Knepley   DMPlexCreateCoordinateSpace - Creates a finite element space for the coordinates
2169318fe57SMatthew G. Knepley 
21720f4b53cSBarry Smith   Collective
2189318fe57SMatthew G. Knepley 
2199318fe57SMatthew G. Knepley   Input Parameters:
22060225df5SJacob Faibussowitsch + dm        - The `DMPLEX`
22120f4b53cSBarry Smith . degree    - The degree of the finite element or `PETSC_DECIDE`
222e44f6aebSMatthew G. Knepley . project   - Flag to project current coordinates into the space
2239318fe57SMatthew G. Knepley - coordFunc - An optional function to map new points from refinement to the surface
2249318fe57SMatthew G. Knepley 
2259318fe57SMatthew G. Knepley   Level: advanced
2269318fe57SMatthew G. Knepley 
2271cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `PetscPointFunc`, `PetscFECreateLagrange()`, `DMGetCoordinateDM()`
2289318fe57SMatthew G. Knepley @*/
229e44f6aebSMatthew G. Knepley PetscErrorCode DMPlexCreateCoordinateSpace(DM dm, PetscInt degree, PetscBool project, PetscPointFunc coordFunc)
230d71ae5a4SJacob Faibussowitsch {
2319318fe57SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
232e44f6aebSMatthew G. Knepley   PetscFE  fe   = NULL;
2339318fe57SMatthew G. Knepley   DM       cdm;
2341df12153SMatthew G. Knepley   PetscInt dim, dE, qorder, height;
2359318fe57SMatthew G. Knepley 
236e44f6aebSMatthew G. Knepley   PetscFunctionBegin;
2379566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
2389566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dE));
2399318fe57SMatthew G. Knepley   qorder = degree;
240e44f6aebSMatthew G. Knepley   PetscCall(DMGetCoordinateDM(dm, &cdm));
241d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)cdm);
242dc431b0cSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-default_quadrature_order", "Quadrature order is one less than quadrature points per edge", "DMPlexCreateCoordinateSpace", qorder, &qorder, NULL, 0));
243d0609cedSBarry Smith   PetscOptionsEnd();
2441df12153SMatthew G. Knepley   PetscCall(DMPlexGetVTKCellHeight(dm, &height));
245e44f6aebSMatthew G. Knepley   if (degree >= 0) {
246e44f6aebSMatthew G. Knepley     DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
247e44f6aebSMatthew G. Knepley     PetscInt       cStart, cEnd, gct;
248dc431b0cSMatthew G. Knepley 
2491df12153SMatthew G. Knepley     PetscCall(DMPlexGetHeightStratum(dm, height, &cStart, &cEnd));
250dc431b0cSMatthew G. Knepley     if (cEnd > cStart) PetscCall(DMPlexGetCellType(dm, cStart, &ct));
251e44f6aebSMatthew G. Knepley     gct = (PetscInt)ct;
252462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &gct, 1, MPIU_INT, MPI_MIN, PetscObjectComm((PetscObject)dm)));
253e44f6aebSMatthew G. Knepley     ct = (DMPolytopeType)gct;
254e44f6aebSMatthew G. Knepley     // Work around current bug in PetscDualSpaceSetUp_Lagrange()
255e44f6aebSMatthew G. Knepley     //   Can be seen in plex_tutorials-ex10_1
256e44f6aebSMatthew 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));
2574f9ab2b4SJed Brown   }
258e44f6aebSMatthew G. Knepley   PetscCall(DMSetCoordinateDisc(dm, fe, project));
2599566063dSJacob Faibussowitsch   PetscCall(PetscFEDestroy(&fe));
2609318fe57SMatthew G. Knepley   mesh->coordFunc = coordFunc;
2613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2629318fe57SMatthew G. Knepley }
2639318fe57SMatthew G. Knepley 
2641df5d5c5SMatthew G. Knepley /*@
2651df5d5c5SMatthew G. Knepley   DMPlexCreateDoublet - Creates a mesh of two cells of the specified type, optionally with later refinement.
2661df5d5c5SMatthew G. Knepley 
267d083f849SBarry Smith   Collective
2681df5d5c5SMatthew G. Knepley 
2691df5d5c5SMatthew G. Knepley   Input Parameters:
270a1cb98faSBarry Smith + comm            - The communicator for the `DM` object
2711df5d5c5SMatthew G. Knepley . dim             - The spatial dimension
2721df5d5c5SMatthew G. Knepley . simplex         - Flag for simplicial cells, otherwise they are tensor product cells
2731df5d5c5SMatthew G. Knepley . interpolate     - Flag to create intermediate mesh pieces (edges, faces)
2741df5d5c5SMatthew G. Knepley - refinementLimit - A nonzero number indicates the largest admissible volume for a refined cell
2751df5d5c5SMatthew G. Knepley 
2761df5d5c5SMatthew G. Knepley   Output Parameter:
27760225df5SJacob Faibussowitsch . newdm - The `DM` object
2781df5d5c5SMatthew G. Knepley 
2791df5d5c5SMatthew G. Knepley   Level: beginner
2801df5d5c5SMatthew G. Knepley 
2811cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetType()`, `DMCreate()`
2821df5d5c5SMatthew G. Knepley @*/
283d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateDoublet(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscBool interpolate, PetscReal refinementLimit, DM *newdm)
284d71ae5a4SJacob Faibussowitsch {
2851df5d5c5SMatthew G. Knepley   DM          dm;
2861df5d5c5SMatthew G. Knepley   PetscMPIInt rank;
2871df5d5c5SMatthew G. Knepley 
2881df5d5c5SMatthew G. Knepley   PetscFunctionBegin;
2899566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, &dm));
2909566063dSJacob Faibussowitsch   PetscCall(DMSetType(dm, DMPLEX));
2919566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
29246139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
2939566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
294ce78fa2fSMatthew G. Knepley   switch (dim) {
295ce78fa2fSMatthew G. Knepley   case 2:
2969566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "triangular"));
2979566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "quadrilateral"));
298ce78fa2fSMatthew G. Knepley     break;
299ce78fa2fSMatthew G. Knepley   case 3:
3009566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "tetrahedral"));
3019566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "hexahedral"));
302ce78fa2fSMatthew G. Knepley     break;
303d71ae5a4SJacob Faibussowitsch   default:
304d71ae5a4SJacob Faibussowitsch     SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
305ce78fa2fSMatthew G. Knepley   }
3061df5d5c5SMatthew G. Knepley   if (rank) {
3071df5d5c5SMatthew G. Knepley     PetscInt numPoints[2] = {0, 0};
3089566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, NULL, NULL, NULL, NULL));
3091df5d5c5SMatthew G. Knepley   } else {
3101df5d5c5SMatthew G. Knepley     switch (dim) {
3111df5d5c5SMatthew G. Knepley     case 2:
3121df5d5c5SMatthew G. Knepley       if (simplex) {
3131df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {4, 2};
3141df5d5c5SMatthew G. Knepley         PetscInt    coneSize[6]         = {3, 3, 0, 0, 0, 0};
3151df5d5c5SMatthew G. Knepley         PetscInt    cones[6]            = {2, 3, 4, 5, 4, 3};
3161df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
3171df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[8]     = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
3181df5d5c5SMatthew G. Knepley 
3199566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3201df5d5c5SMatthew G. Knepley       } else {
3211df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {6, 2};
3221df5d5c5SMatthew G. Knepley         PetscInt    coneSize[8]         = {4, 4, 0, 0, 0, 0, 0, 0};
3231df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {2, 3, 4, 5, 3, 6, 7, 4};
3241df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3251df5d5c5SMatthew 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};
3261df5d5c5SMatthew G. Knepley 
3279566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3281df5d5c5SMatthew G. Knepley       }
3291df5d5c5SMatthew G. Knepley       break;
3301df5d5c5SMatthew G. Knepley     case 3:
3311df5d5c5SMatthew G. Knepley       if (simplex) {
3321df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {5, 2};
3331df5d5c5SMatthew G. Knepley         PetscInt    coneSize[7]         = {4, 4, 0, 0, 0, 0, 0};
3341df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {4, 3, 5, 2, 5, 3, 4, 6};
3351df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3361df5d5c5SMatthew 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};
3371df5d5c5SMatthew G. Knepley 
3389566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3391df5d5c5SMatthew G. Knepley       } else {
3401df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]         = {12, 2};
3411df5d5c5SMatthew G. Knepley         PetscInt    coneSize[14]         = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3421df5d5c5SMatthew G. Knepley         PetscInt    cones[16]            = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
3431df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3449371c9d4SSatish 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};
3451df5d5c5SMatthew G. Knepley 
3469566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3471df5d5c5SMatthew G. Knepley       }
3481df5d5c5SMatthew G. Knepley       break;
349d71ae5a4SJacob Faibussowitsch     default:
350d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
3511df5d5c5SMatthew G. Knepley     }
3521df5d5c5SMatthew G. Knepley   }
35346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
3541df5d5c5SMatthew G. Knepley   *newdm = dm;
3551df5d5c5SMatthew G. Knepley   if (refinementLimit > 0.0) {
3561df5d5c5SMatthew G. Knepley     DM          rdm;
3571df5d5c5SMatthew G. Knepley     const char *name;
3581df5d5c5SMatthew G. Knepley 
3599566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(*newdm, PETSC_FALSE));
3609566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(*newdm, refinementLimit));
3619566063dSJacob Faibussowitsch     PetscCall(DMRefine(*newdm, comm, &rdm));
3629566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)*newdm, &name));
3639566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)rdm, name));
3649566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3651df5d5c5SMatthew G. Knepley     *newdm = rdm;
3661df5d5c5SMatthew G. Knepley   }
3671df5d5c5SMatthew G. Knepley   if (interpolate) {
3685fd9971aSMatthew G. Knepley     DM idm;
3691df5d5c5SMatthew G. Knepley 
3709566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*newdm, &idm));
3719566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3721df5d5c5SMatthew G. Knepley     *newdm = idm;
3731df5d5c5SMatthew G. Knepley   }
3743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3751df5d5c5SMatthew G. Knepley }
3761df5d5c5SMatthew G. Knepley 
377d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
378d71ae5a4SJacob Faibussowitsch {
3799318fe57SMatthew G. Knepley   const PetscInt numVertices    = 2;
3809318fe57SMatthew G. Knepley   PetscInt       markerRight    = 1;
3819318fe57SMatthew G. Knepley   PetscInt       markerLeft     = 1;
3829318fe57SMatthew G. Knepley   PetscBool      markerSeparate = PETSC_FALSE;
3839318fe57SMatthew G. Knepley   Vec            coordinates;
3849318fe57SMatthew G. Knepley   PetscSection   coordSection;
3859318fe57SMatthew G. Knepley   PetscScalar   *coords;
3869318fe57SMatthew G. Knepley   PetscInt       coordSize;
3879318fe57SMatthew G. Knepley   PetscMPIInt    rank;
3889318fe57SMatthew G. Knepley   PetscInt       cdim = 1, v;
389552f7358SJed Brown 
3909318fe57SMatthew G. Knepley   PetscFunctionBegin;
3919566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
3929318fe57SMatthew G. Knepley   if (markerSeparate) {
3939318fe57SMatthew G. Knepley     markerRight = 2;
3949318fe57SMatthew G. Knepley     markerLeft  = 1;
3959318fe57SMatthew G. Knepley   }
3969566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
397c5853193SPierre Jolivet   if (rank == 0) {
3989566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numVertices));
3999566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
4009566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 0, markerLeft));
4019566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 1, markerRight));
4029318fe57SMatthew G. Knepley   }
4039566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
4049566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
4059318fe57SMatthew G. Knepley   /* Build coordinates */
4069566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
4079566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
4089566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
4099566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, 0, numVertices));
4109566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
4119318fe57SMatthew G. Knepley   for (v = 0; v < numVertices; ++v) {
4129566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
4139566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
4149318fe57SMatthew G. Knepley   }
4159566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
4169566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
4179566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
4189566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
4199566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
4209566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
4219566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
4229566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
4239318fe57SMatthew G. Knepley   coords[0] = lower[0];
4249318fe57SMatthew G. Knepley   coords[1] = upper[0];
4259566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
4269566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
4279566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
4283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4299318fe57SMatthew G. Knepley }
43026492d91SMatthew G. Knepley 
431d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
432d71ae5a4SJacob Faibussowitsch {
4331df21d24SMatthew G. Knepley   const PetscInt numVertices    = (edges[0] + 1) * (edges[1] + 1);
4341df21d24SMatthew G. Knepley   const PetscInt numEdges       = edges[0] * (edges[1] + 1) + (edges[0] + 1) * edges[1];
435552f7358SJed Brown   PetscInt       markerTop      = 1;
436552f7358SJed Brown   PetscInt       markerBottom   = 1;
437552f7358SJed Brown   PetscInt       markerRight    = 1;
438552f7358SJed Brown   PetscInt       markerLeft     = 1;
439552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
440552f7358SJed Brown   Vec            coordinates;
441552f7358SJed Brown   PetscSection   coordSection;
442552f7358SJed Brown   PetscScalar   *coords;
443552f7358SJed Brown   PetscInt       coordSize;
444552f7358SJed Brown   PetscMPIInt    rank;
445552f7358SJed Brown   PetscInt       v, vx, vy;
446552f7358SJed Brown 
447552f7358SJed Brown   PetscFunctionBegin;
4489566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
449552f7358SJed Brown   if (markerSeparate) {
4501df21d24SMatthew G. Knepley     markerTop    = 3;
4511df21d24SMatthew G. Knepley     markerBottom = 1;
4521df21d24SMatthew G. Knepley     markerRight  = 2;
4531df21d24SMatthew G. Knepley     markerLeft   = 4;
454552f7358SJed Brown   }
4559566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
456dd400576SPatrick Sanan   if (rank == 0) {
457552f7358SJed Brown     PetscInt e, ex, ey;
458552f7358SJed Brown 
4599566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numEdges + numVertices));
46048a46eb9SPierre Jolivet     for (e = 0; e < numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
4619566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
462552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
463552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
464552f7358SJed Brown         PetscInt edge   = vx * edges[1] + ey + edges[0] * (edges[1] + 1);
465552f7358SJed Brown         PetscInt vertex = ey * (edges[0] + 1) + vx + numEdges;
466da80777bSKarl Rupp         PetscInt cone[2];
467552f7358SJed Brown 
4689371c9d4SSatish Balay         cone[0] = vertex;
4699371c9d4SSatish Balay         cone[1] = vertex + edges[0] + 1;
4709566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
471552f7358SJed Brown         if (vx == edges[0]) {
4729566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
4739566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
474552f7358SJed Brown           if (ey == edges[1] - 1) {
4759566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
4769566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerRight));
477552f7358SJed Brown           }
478552f7358SJed Brown         } else if (vx == 0) {
4799566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
4809566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
481552f7358SJed Brown           if (ey == edges[1] - 1) {
4829566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
4839566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerLeft));
484552f7358SJed Brown           }
485552f7358SJed Brown         }
486552f7358SJed Brown       }
487552f7358SJed Brown     }
488552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
489552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
490552f7358SJed Brown         PetscInt edge   = vy * edges[0] + ex;
491552f7358SJed Brown         PetscInt vertex = vy * (edges[0] + 1) + ex + numEdges;
492da80777bSKarl Rupp         PetscInt cone[2];
493552f7358SJed Brown 
4949371c9d4SSatish Balay         cone[0] = vertex;
4959371c9d4SSatish Balay         cone[1] = vertex + 1;
4969566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
497552f7358SJed Brown         if (vy == edges[1]) {
4989566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
4999566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
500552f7358SJed Brown           if (ex == edges[0] - 1) {
5019566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
5029566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerTop));
503552f7358SJed Brown           }
504552f7358SJed Brown         } else if (vy == 0) {
5059566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
5069566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
507552f7358SJed Brown           if (ex == edges[0] - 1) {
5089566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
5099566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerBottom));
510552f7358SJed Brown           }
511552f7358SJed Brown         }
512552f7358SJed Brown       }
513552f7358SJed Brown     }
514552f7358SJed Brown   }
5159566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
5169566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
517552f7358SJed Brown   /* Build coordinates */
5189566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 2));
5199566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
5209566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
5219566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices));
5229566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 2));
523552f7358SJed Brown   for (v = numEdges; v < numEdges + numVertices; ++v) {
5249566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 2));
5259566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 2));
526552f7358SJed Brown   }
5279566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5289566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
5299566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
5309566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
5319566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
5329566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 2));
5339566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
5349566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
535552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
536552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
537552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 0] = lower[0] + ((upper[0] - lower[0]) / edges[0]) * vx;
538552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 1] = lower[1] + ((upper[1] - lower[1]) / edges[1]) * vy;
539552f7358SJed Brown     }
540552f7358SJed Brown   }
5419566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
5429566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
5439566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
5443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
545552f7358SJed Brown }
546552f7358SJed Brown 
547d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
548d71ae5a4SJacob Faibussowitsch {
5499e8abbc3SMichael Lange   PetscInt     vertices[3], numVertices;
5507b59f5a9SMichael Lange   PetscInt     numFaces       = 2 * faces[0] * faces[1] + 2 * faces[1] * faces[2] + 2 * faces[0] * faces[2];
551c2df9bbfSMatthew G. Knepley   PetscInt     markerTop      = 1;
552c2df9bbfSMatthew G. Knepley   PetscInt     markerBottom   = 1;
553c2df9bbfSMatthew G. Knepley   PetscInt     markerFront    = 1;
554c2df9bbfSMatthew G. Knepley   PetscInt     markerBack     = 1;
555c2df9bbfSMatthew G. Knepley   PetscInt     markerRight    = 1;
556c2df9bbfSMatthew G. Knepley   PetscInt     markerLeft     = 1;
557c2df9bbfSMatthew G. Knepley   PetscBool    markerSeparate = PETSC_FALSE;
558552f7358SJed Brown   Vec          coordinates;
559552f7358SJed Brown   PetscSection coordSection;
560552f7358SJed Brown   PetscScalar *coords;
561552f7358SJed Brown   PetscInt     coordSize;
562552f7358SJed Brown   PetscMPIInt  rank;
563552f7358SJed Brown   PetscInt     v, vx, vy, vz;
5647b59f5a9SMichael Lange   PetscInt     voffset, iface = 0, cone[4];
565552f7358SJed Brown 
566552f7358SJed Brown   PetscFunctionBegin;
5671dca8a05SBarry 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");
5689566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
569c2df9bbfSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
570c2df9bbfSMatthew G. Knepley   if (markerSeparate) {
571c2df9bbfSMatthew G. Knepley     markerBottom = 1;
572c2df9bbfSMatthew G. Knepley     markerTop    = 2;
573c2df9bbfSMatthew G. Knepley     markerFront  = 3;
574c2df9bbfSMatthew G. Knepley     markerBack   = 4;
575c2df9bbfSMatthew G. Knepley     markerRight  = 5;
576c2df9bbfSMatthew G. Knepley     markerLeft   = 6;
577c2df9bbfSMatthew G. Knepley   }
5789371c9d4SSatish Balay   vertices[0] = faces[0] + 1;
5799371c9d4SSatish Balay   vertices[1] = faces[1] + 1;
5809371c9d4SSatish Balay   vertices[2] = faces[2] + 1;
5819e8abbc3SMichael Lange   numVertices = vertices[0] * vertices[1] * vertices[2];
582dd400576SPatrick Sanan   if (rank == 0) {
583552f7358SJed Brown     PetscInt f;
584552f7358SJed Brown 
5859566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numFaces + numVertices));
58648a46eb9SPierre Jolivet     for (f = 0; f < numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
5879566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
5887b59f5a9SMichael Lange 
5897b59f5a9SMichael Lange     /* Side 0 (Top) */
5907b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
5917b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
5927b59f5a9SMichael Lange         voffset = numFaces + vertices[0] * vertices[1] * (vertices[2] - 1) + vy * vertices[0] + vx;
5939371c9d4SSatish Balay         cone[0] = voffset;
5949371c9d4SSatish Balay         cone[1] = voffset + 1;
5959371c9d4SSatish Balay         cone[2] = voffset + vertices[0] + 1;
5969371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
5979566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
598c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerTop));
599c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerTop));
600c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerTop));
601c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerTop));
602c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerTop));
6037b59f5a9SMichael Lange         iface++;
604552f7358SJed Brown       }
605552f7358SJed Brown     }
6067b59f5a9SMichael Lange 
6077b59f5a9SMichael Lange     /* Side 1 (Bottom) */
6087b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
6097b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6107b59f5a9SMichael Lange         voffset = numFaces + vy * (faces[0] + 1) + vx;
6119371c9d4SSatish Balay         cone[0] = voffset + 1;
6129371c9d4SSatish Balay         cone[1] = voffset;
6139371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6149371c9d4SSatish Balay         cone[3] = voffset + vertices[0] + 1;
6159566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
616c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBottom));
617c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBottom));
618c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBottom));
619c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerBottom));
620c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerBottom));
6217b59f5a9SMichael Lange         iface++;
622552f7358SJed Brown       }
623552f7358SJed Brown     }
6247b59f5a9SMichael Lange 
6257b59f5a9SMichael Lange     /* Side 2 (Front) */
6267b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6277b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6287b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vx;
6299371c9d4SSatish Balay         cone[0] = voffset;
6309371c9d4SSatish Balay         cone[1] = voffset + 1;
6319371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + 1;
6329371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1];
6339566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
634c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerFront));
635c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerFront));
636c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerFront));
637c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerFront));
638c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerFront));
6397b59f5a9SMichael Lange         iface++;
640552f7358SJed Brown       }
6417b59f5a9SMichael Lange     }
6427b59f5a9SMichael Lange 
6437b59f5a9SMichael Lange     /* Side 3 (Back) */
6447b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6457b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6467b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vertices[0] * (vertices[1] - 1) + vx;
6479371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6489371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1] + 1;
6499371c9d4SSatish Balay         cone[2] = voffset + 1;
6509371c9d4SSatish Balay         cone[3] = voffset;
6519566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
652c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBack));
653c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBack));
654c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBack));
655c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerBack));
656c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerBack));
6577b59f5a9SMichael Lange         iface++;
6587b59f5a9SMichael Lange       }
6597b59f5a9SMichael Lange     }
6607b59f5a9SMichael Lange 
6617b59f5a9SMichael Lange     /* Side 4 (Left) */
6627b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6637b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
6647b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0];
6659371c9d4SSatish Balay         cone[0] = voffset;
6669371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1];
6679371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + vertices[0];
6689371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
6699566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
670c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerLeft));
671c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerLeft));
672c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerLeft));
673c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[1] + 0, markerLeft));
674c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerLeft));
6757b59f5a9SMichael Lange         iface++;
6767b59f5a9SMichael Lange       }
6777b59f5a9SMichael Lange     }
6787b59f5a9SMichael Lange 
6797b59f5a9SMichael Lange     /* Side 5 (Right) */
6807b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6817b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
682aab5bcd8SJed Brown         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0] + faces[0];
6839371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6849371c9d4SSatish Balay         cone[1] = voffset;
6859371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6869371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1] + vertices[0];
6879566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
688c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerRight));
689c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerRight));
690c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerRight));
691c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerRight));
692c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerRight));
6937b59f5a9SMichael Lange         iface++;
6947b59f5a9SMichael Lange       }
695552f7358SJed Brown     }
696552f7358SJed Brown   }
6979566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
6989566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
699552f7358SJed Brown   /* Build coordinates */
7009566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 3));
7019566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
7029566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
7039566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices));
7049566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 3));
705552f7358SJed Brown   for (v = numFaces; v < numFaces + numVertices; ++v) {
7069566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 3));
7079566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 3));
708552f7358SJed Brown   }
7099566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
7109566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
7119566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
7129566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
7139566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
7149566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 3));
7159566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
7169566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
717552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
718552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
719552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
720552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 0] = lower[0] + ((upper[0] - lower[0]) / faces[0]) * vx;
721552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 1] = lower[1] + ((upper[1] - lower[1]) / faces[1]) * vy;
722552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 2] = lower[2] + ((upper[2] - lower[2]) / faces[2]) * vz;
723552f7358SJed Brown       }
724552f7358SJed Brown     }
725552f7358SJed Brown   }
7269566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
7279566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
7289566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
7293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
730552f7358SJed Brown }
731552f7358SJed Brown 
732d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate)
733d71ae5a4SJacob Faibussowitsch {
7349318fe57SMatthew G. Knepley   PetscFunctionBegin;
7359318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
73646139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
7379566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim - 1));
7389566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim));
7399318fe57SMatthew G. Knepley   switch (dim) {
740d71ae5a4SJacob Faibussowitsch   case 1:
741d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(dm, lower, upper, faces));
742d71ae5a4SJacob Faibussowitsch     break;
743d71ae5a4SJacob Faibussowitsch   case 2:
744d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(dm, lower, upper, faces));
745d71ae5a4SJacob Faibussowitsch     break;
746d71ae5a4SJacob Faibussowitsch   case 3:
747d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(dm, lower, upper, faces));
748d71ae5a4SJacob Faibussowitsch     break;
749d71ae5a4SJacob Faibussowitsch   default:
750d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Dimension not supported: %" PetscInt_FMT, dim);
7519318fe57SMatthew G. Knepley   }
75246139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
7539566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
7543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7559318fe57SMatthew G. Knepley }
7569318fe57SMatthew G. Knepley 
7579318fe57SMatthew G. Knepley /*@C
7589318fe57SMatthew G. Knepley   DMPlexCreateBoxSurfaceMesh - Creates a mesh on the surface of the tensor product of unit intervals (box) using tensor cells (hexahedra).
7599318fe57SMatthew G. Knepley 
7609318fe57SMatthew G. Knepley   Collective
7619318fe57SMatthew G. Knepley 
7629318fe57SMatthew G. Knepley   Input Parameters:
763a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
76420f4b53cSBarry Smith . dim         - The spatial dimension of the box, so the resulting mesh is has dimension `dim`-1
76520f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
76620f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
76720f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
7689318fe57SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
7699318fe57SMatthew G. Knepley 
7709318fe57SMatthew G. Knepley   Output Parameter:
771a1cb98faSBarry Smith . dm - The `DM` object
7729318fe57SMatthew G. Knepley 
7739318fe57SMatthew G. Knepley   Level: beginner
7749318fe57SMatthew G. Knepley 
7751cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateBoxMesh()`, `DMPlexCreateFromFile()`, `DMSetType()`, `DMCreate()`
7769318fe57SMatthew G. Knepley @*/
777d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBoxSurfaceMesh(MPI_Comm comm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate, DM *dm)
778d71ae5a4SJacob Faibussowitsch {
7799318fe57SMatthew G. Knepley   PetscInt  fac[3] = {1, 1, 1};
7809318fe57SMatthew G. Knepley   PetscReal low[3] = {0, 0, 0};
7819318fe57SMatthew G. Knepley   PetscReal upp[3] = {1, 1, 1};
7829318fe57SMatthew G. Knepley 
7839318fe57SMatthew G. Knepley   PetscFunctionBegin;
7849566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
7859566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
7869566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(*dm, dim, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, interpolate));
7873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7889318fe57SMatthew G. Knepley }
7899318fe57SMatthew G. Knepley 
790d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateLineMesh_Internal(DM dm, PetscInt segments, PetscReal lower, PetscReal upper, DMBoundaryType bd)
791d71ae5a4SJacob Faibussowitsch {
792fdbf62faSLisandro Dalcin   PetscInt     i, fStart, fEnd, numCells = 0, numVerts = 0;
793fdbf62faSLisandro Dalcin   PetscInt     numPoints[2], *coneSize, *cones, *coneOrientations;
794fdbf62faSLisandro Dalcin   PetscScalar *vertexCoords;
795fdbf62faSLisandro Dalcin   PetscReal    L, maxCell;
796fdbf62faSLisandro Dalcin   PetscBool    markerSeparate = PETSC_FALSE;
797fdbf62faSLisandro Dalcin   PetscInt     markerLeft = 1, faceMarkerLeft = 1;
798fdbf62faSLisandro Dalcin   PetscInt     markerRight = 1, faceMarkerRight = 2;
799fdbf62faSLisandro Dalcin   PetscBool    wrap = (bd == DM_BOUNDARY_PERIODIC || bd == DM_BOUNDARY_TWIST) ? PETSC_TRUE : PETSC_FALSE;
800fdbf62faSLisandro Dalcin   PetscMPIInt  rank;
801fdbf62faSLisandro Dalcin 
802fdbf62faSLisandro Dalcin   PetscFunctionBegin;
8034f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
804fdbf62faSLisandro Dalcin 
8059566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, 1));
8069566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
8079566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
808fdbf62faSLisandro Dalcin 
8099566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
810dd400576SPatrick Sanan   if (rank == 0) numCells = segments;
811dd400576SPatrick Sanan   if (rank == 0) numVerts = segments + (wrap ? 0 : 1);
812fdbf62faSLisandro Dalcin 
8139371c9d4SSatish Balay   numPoints[0] = numVerts;
8149371c9d4SSatish Balay   numPoints[1] = numCells;
8159566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(numCells + numVerts, &coneSize, numCells * 2, &cones, numCells + numVerts, &coneOrientations, numVerts, &vertexCoords));
8169566063dSJacob Faibussowitsch   PetscCall(PetscArrayzero(coneOrientations, numCells + numVerts));
817ad540459SPierre Jolivet   for (i = 0; i < numCells; ++i) coneSize[i] = 2;
818ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) coneSize[numCells + i] = 0;
8199371c9d4SSatish Balay   for (i = 0; i < numCells; ++i) {
8209371c9d4SSatish Balay     cones[2 * i]     = numCells + i % numVerts;
8219371c9d4SSatish Balay     cones[2 * i + 1] = numCells + (i + 1) % numVerts;
8229371c9d4SSatish Balay   }
823ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) vertexCoords[i] = lower + (upper - lower) * ((PetscReal)i / (PetscReal)numCells);
8249566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
8259566063dSJacob Faibussowitsch   PetscCall(PetscFree4(coneSize, cones, coneOrientations, vertexCoords));
826fdbf62faSLisandro Dalcin 
8279566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
8289371c9d4SSatish Balay   if (markerSeparate) {
8299371c9d4SSatish Balay     markerLeft  = faceMarkerLeft;
8309371c9d4SSatish Balay     markerRight = faceMarkerRight;
8319371c9d4SSatish Balay   }
832dd400576SPatrick Sanan   if (!wrap && rank == 0) {
8339566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
8349566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fStart, markerLeft));
8359566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fEnd - 1, markerRight));
8369566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fStart, faceMarkerLeft));
8379566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fEnd - 1, faceMarkerRight));
838fdbf62faSLisandro Dalcin   }
839fdbf62faSLisandro Dalcin   if (wrap) {
840fdbf62faSLisandro Dalcin     L       = upper - lower;
841fdbf62faSLisandro Dalcin     maxCell = (PetscReal)1.1 * (L / (PetscReal)PetscMax(1, segments));
8424fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, &maxCell, &lower, &L));
843fdbf62faSLisandro Dalcin   }
8449566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
8453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
846fdbf62faSLisandro Dalcin }
847fdbf62faSLisandro Dalcin 
8484054ae39SJames Wright // Creates "Face Sets" label based on the standard box labeling conventions
849d7d2d1d2SJames Wright static PetscErrorCode DMPlexSetBoxLabel_Internal(DM dm, const DMBoundaryType periodicity[])
8504054ae39SJames Wright {
8516ff49feeSJames Wright   DM              cdm;
8526ff49feeSJames Wright   PetscSection    csection;
8536ff49feeSJames Wright   Vec             coordinates;
8544054ae39SJames Wright   DMLabel         label;
8556ff49feeSJames Wright   IS              faces_is;
8562b4f33d9SJames Wright   PetscInt        dim, num_face = 0;
8574054ae39SJames Wright   const PetscInt *faces;
8584054ae39SJames Wright   PetscInt        faceMarkerBottom, faceMarkerTop, faceMarkerFront, faceMarkerBack, faceMarkerRight, faceMarkerLeft;
8594054ae39SJames Wright 
8604054ae39SJames Wright   PetscFunctionBeginUser;
8614054ae39SJames Wright   PetscCall(DMGetDimension(dm, &dim));
862d7c1f440SPierre 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);
8634054ae39SJames Wright   // Get Face Sets label
8644054ae39SJames Wright   PetscCall(DMGetLabel(dm, "Face Sets", &label));
8654054ae39SJames Wright   if (label) {
8664054ae39SJames Wright     PetscCall(DMLabelReset(label));
8674054ae39SJames Wright   } else {
8684054ae39SJames Wright     PetscCall(DMCreateLabel(dm, "Face Sets"));
8694054ae39SJames Wright     PetscCall(DMGetLabel(dm, "Face Sets", &label));
8704054ae39SJames Wright   }
8714054ae39SJames Wright   PetscCall(DMPlexMarkBoundaryFaces(dm, 1, label));
8726ff49feeSJames Wright   PetscCall(DMGetStratumIS(dm, "Face Sets", 1, &faces_is));
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 
8932b4f33d9SJames Wright   if (faces_is) PetscCall(ISGetLocalSize(faces_is, &num_face));
8942b4f33d9SJames Wright   if (faces_is) 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   }
9692b4f33d9SJames Wright   if (faces_is) PetscCall(ISRestoreIndices(faces_is, &faces));
9706ff49feeSJames Wright   PetscCall(ISDestroy(&faces_is));
971d7d2d1d2SJames Wright 
972d7d2d1d2SJames Wright   // Create Isoperiodic SF from newly-created face labels
973d7d2d1d2SJames Wright   PetscSF     periodicsfs[3];
974d7d2d1d2SJames Wright   PetscInt    periodic_sf_index  = 0;
975d7d2d1d2SJames Wright   PetscScalar transform[3][4][4] = {{{0.}}};
976d7d2d1d2SJames Wright   for (PetscInt d = 0; d < dim; d++) {
977d7d2d1d2SJames Wright     IS              donor_is, periodic_is;
978d7d2d1d2SJames Wright     const PetscInt *donor_faces = NULL, *periodic_faces = NULL;
979d7d2d1d2SJames Wright     PetscInt        num_donor = 0, num_periodic = 0;
980d7d2d1d2SJames Wright     PetscSF         centroidsf;
981d7d2d1d2SJames Wright     PetscReal       donor_to_periodic_distance;
982d7d2d1d2SJames Wright     const PetscInt  face_pairings[2][3][2] = {
983d7d2d1d2SJames Wright       // 2D face pairings, {donor, periodic}
984d7d2d1d2SJames Wright       {{4, 2}, {1, 3}},
985d7d2d1d2SJames Wright       // 3D face pairings
986d7d2d1d2SJames Wright       {{5, 6}, {3, 4}, {1, 2}}
987d7d2d1d2SJames Wright     };
988d7d2d1d2SJames Wright 
989d7d2d1d2SJames Wright     if (periodicity[d] != DM_BOUNDARY_PERIODIC) continue;
990d7d2d1d2SJames Wright     {
991d7d2d1d2SJames Wright       // Compute centroidsf, which is the mapping from donor faces to periodic faces
992d7d2d1d2SJames Wright       // Matches the centroid of the faces together, ignoring the periodic direction component (which should not match between donor and periodic face)
993d7d2d1d2SJames Wright       PetscInt     coords_size, centroid_comps = dim - 1;
994d7d2d1d2SJames Wright       PetscScalar *coords = NULL;
995d7d2d1d2SJames Wright       PetscReal   *donor_centroids, *periodic_centroids;
996d7d2d1d2SJames Wright       PetscReal    loc_periodic[2] = {PETSC_MIN_REAL, PETSC_MIN_REAL}, loc_periodic_global[2]; // Location of donor (0) and periodic (1) faces in periodic direction
997d7d2d1d2SJames Wright 
998d7d2d1d2SJames Wright       PetscCall(DMGetStratumIS(dm, "Face Sets", face_pairings[dim - 2][d][0], &donor_is));
999d7d2d1d2SJames Wright       PetscCall(DMGetStratumIS(dm, "Face Sets", face_pairings[dim - 2][d][1], &periodic_is));
1000d7d2d1d2SJames Wright       if (donor_is) {
1001d7d2d1d2SJames Wright         PetscCall(ISGetLocalSize(donor_is, &num_donor));
1002d7d2d1d2SJames Wright         PetscCall(ISGetIndices(donor_is, &donor_faces));
1003d7d2d1d2SJames Wright       }
1004d7d2d1d2SJames Wright       if (periodic_is) {
1005d7d2d1d2SJames Wright         PetscCall(ISGetLocalSize(periodic_is, &num_periodic));
1006d7d2d1d2SJames Wright         PetscCall(ISGetIndices(periodic_is, &periodic_faces));
1007d7d2d1d2SJames Wright       }
1008d7d2d1d2SJames Wright       PetscCall(PetscCalloc2(num_donor * centroid_comps, &donor_centroids, num_periodic * centroid_comps, &periodic_centroids));
1009d7d2d1d2SJames Wright       for (PetscInt f = 0; f < num_donor; f++) {
1010d7d2d1d2SJames Wright         PetscInt face = donor_faces[f], num_coords;
1011d7d2d1d2SJames Wright         PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
1012d7d2d1d2SJames Wright         num_coords = coords_size / dim;
1013d7d2d1d2SJames Wright         for (PetscInt c = 0; c < num_coords; c++) {
1014d7d2d1d2SJames Wright           PetscInt comp_index = 0;
1015d7d2d1d2SJames Wright           loc_periodic[0]     = PetscRealPart(coords[c * dim + d]);
1016d7d2d1d2SJames Wright           for (PetscInt i = 0; i < dim; i++) {
1017d7d2d1d2SJames Wright             if (i == d) continue; // Periodic direction not used for centroid calculation
1018d7d2d1d2SJames Wright             donor_centroids[f * centroid_comps + comp_index] += PetscRealPart(coords[c * dim + i]) / num_coords;
1019d7d2d1d2SJames Wright             comp_index++;
1020d7d2d1d2SJames Wright           }
1021d7d2d1d2SJames Wright         }
1022d7d2d1d2SJames Wright         PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
1023d7d2d1d2SJames Wright       }
1024d7d2d1d2SJames Wright 
1025d7d2d1d2SJames Wright       for (PetscInt f = 0; f < num_periodic; f++) {
1026d7d2d1d2SJames Wright         PetscInt face = periodic_faces[f], num_coords;
1027d7d2d1d2SJames Wright         PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
1028d7d2d1d2SJames Wright         num_coords = coords_size / dim;
1029d7d2d1d2SJames Wright         for (PetscInt c = 0; c < num_coords; c++) {
1030d7d2d1d2SJames Wright           PetscInt comp_index = 0;
1031d7d2d1d2SJames Wright           loc_periodic[1]     = PetscRealPart(coords[c * dim + d]);
1032d7d2d1d2SJames Wright           for (PetscInt i = 0; i < dim; i++) {
1033d7d2d1d2SJames Wright             if (i == d) continue; // Periodic direction not used for centroid calculation
1034d7d2d1d2SJames Wright             periodic_centroids[f * centroid_comps + comp_index] += PetscRealPart(coords[c * dim + i]) / num_coords;
1035d7d2d1d2SJames Wright             comp_index++;
1036d7d2d1d2SJames Wright           }
1037d7d2d1d2SJames Wright         }
1038d7d2d1d2SJames Wright         PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
1039d7d2d1d2SJames Wright       }
1040d7d2d1d2SJames Wright       PetscCallMPI(MPIU_Allreduce(loc_periodic, loc_periodic_global, 2, MPIU_REAL, MPIU_MAX, PetscObjectComm((PetscObject)dm)));
1041d7d2d1d2SJames Wright       donor_to_periodic_distance = loc_periodic_global[1] - loc_periodic_global[0];
1042d7d2d1d2SJames Wright 
1043d7d2d1d2SJames Wright       PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &centroidsf));
1044d7d2d1d2SJames Wright       PetscCall(PetscSFSetGraphFromCoordinates(centroidsf, num_donor, num_periodic, centroid_comps, 1e-10, donor_centroids, periodic_centroids));
1045d7d2d1d2SJames Wright       PetscCall(PetscSFViewFromOptions(centroidsf, NULL, "-dm_plex_box_label_centroid_sf_view"));
1046d7d2d1d2SJames Wright       PetscCall(PetscFree2(donor_centroids, periodic_centroids));
1047d7d2d1d2SJames Wright     }
1048d7d2d1d2SJames Wright 
1049d7d2d1d2SJames Wright     { // Create Isoperiodic SF using centroidsSF
1050d7d2d1d2SJames Wright       PetscInt           pStart, pEnd;
1051d7d2d1d2SJames Wright       PetscInt          *leaf_faces;
1052d7d2d1d2SJames Wright       const PetscSFNode *firemote;
1053d7d2d1d2SJames Wright       PetscSFNode       *isoperiodic_leaves;
1054d7d2d1d2SJames Wright 
1055d7d2d1d2SJames Wright       PetscCall(PetscMalloc1(num_periodic, &leaf_faces));
1056d7d2d1d2SJames Wright       PetscCall(PetscSFBcastBegin(centroidsf, MPIU_INT, donor_faces, leaf_faces, MPI_REPLACE));
1057d7d2d1d2SJames Wright       PetscCall(PetscSFBcastEnd(centroidsf, MPIU_INT, donor_faces, leaf_faces, MPI_REPLACE));
1058d7d2d1d2SJames Wright 
1059d7d2d1d2SJames Wright       PetscCall(PetscMalloc1(num_periodic, &isoperiodic_leaves));
1060d7d2d1d2SJames Wright       PetscCall(PetscSFGetGraph(centroidsf, NULL, NULL, NULL, &firemote));
1061d7d2d1d2SJames Wright       for (PetscInt l = 0; l < num_periodic; ++l) {
1062d7d2d1d2SJames Wright         isoperiodic_leaves[l].index = leaf_faces[l];
1063d7d2d1d2SJames Wright         isoperiodic_leaves[l].rank  = firemote[l].rank;
1064d7d2d1d2SJames Wright       }
1065d7d2d1d2SJames Wright 
1066d7d2d1d2SJames Wright       PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
1067d7d2d1d2SJames Wright       PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &periodicsfs[periodic_sf_index]));
1068d7d2d1d2SJames Wright       PetscCall(PetscSFSetGraph(periodicsfs[periodic_sf_index], pEnd - pStart, num_periodic, (PetscInt *)periodic_faces, PETSC_COPY_VALUES, isoperiodic_leaves, PETSC_OWN_POINTER));
1069d7d2d1d2SJames Wright       PetscCall(PetscSFViewFromOptions(periodicsfs[periodic_sf_index], NULL, "-dm_plex_box_label_periodic_sf_view"));
1070d7d2d1d2SJames Wright       PetscCall(PetscFree(leaf_faces));
1071d7d2d1d2SJames Wright     }
1072d7d2d1d2SJames Wright 
1073d7d2d1d2SJames Wright     transform[periodic_sf_index][0][0] = 1;
1074d7d2d1d2SJames Wright     transform[periodic_sf_index][1][1] = 1;
1075d7d2d1d2SJames Wright     transform[periodic_sf_index][2][2] = 1;
1076d7d2d1d2SJames Wright     transform[periodic_sf_index][3][3] = 1;
1077d7d2d1d2SJames Wright     transform[periodic_sf_index][d][3] = donor_to_periodic_distance;
1078d7d2d1d2SJames Wright 
1079d7d2d1d2SJames Wright     periodic_sf_index++;
1080d7d2d1d2SJames Wright     PetscCall(PetscSFDestroy(&centroidsf));
1081d7d2d1d2SJames Wright     if (donor_is) {
1082d7d2d1d2SJames Wright       PetscCall(ISRestoreIndices(donor_is, &donor_faces));
1083d7d2d1d2SJames Wright       PetscCall(ISDestroy(&donor_is));
1084d7d2d1d2SJames Wright     }
1085d7d2d1d2SJames Wright     if (periodic_is) {
1086d7d2d1d2SJames Wright       PetscCall(ISRestoreIndices(periodic_is, &periodic_faces));
1087d7d2d1d2SJames Wright       PetscCall(ISDestroy(&periodic_is));
1088d7d2d1d2SJames Wright     }
1089d7d2d1d2SJames Wright     PetscCall(DMClearLabelStratum(dm, "Face Sets", face_pairings[dim - 2][d][0]));
1090d7d2d1d2SJames Wright     PetscCall(DMClearLabelStratum(dm, "Face Sets", face_pairings[dim - 2][d][1]));
1091d7d2d1d2SJames Wright   }
1092d7d2d1d2SJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(dm, periodic_sf_index, periodicsfs));
1093d7d2d1d2SJames Wright   PetscCall(DMPlexSetIsoperiodicFaceTransform(dm, periodic_sf_index, (const PetscScalar *)transform));
1094d7d2d1d2SJames Wright   for (PetscInt p = 0; p < periodic_sf_index; p++) PetscCall(PetscSFDestroy(&periodicsfs[p]));
1095d7d2d1d2SJames Wright 
1096d7d2d1d2SJames Wright   { // Update coordinate DM with new Face Sets label
1097d7d2d1d2SJames Wright     DM      cdm;
1098d7d2d1d2SJames Wright     DMLabel oldFaceSets, newFaceSets;
1099d7d2d1d2SJames Wright     PetscCall(DMGetCoordinateDM(dm, &cdm));
1100d7d2d1d2SJames Wright     PetscCall(DMGetLabel(cdm, "Face Sets", &oldFaceSets));
1101d7d2d1d2SJames Wright     if (oldFaceSets) PetscCall(DMRemoveLabelBySelf(cdm, &oldFaceSets, PETSC_FALSE));
1102d7d2d1d2SJames Wright     PetscCall(DMLabelDuplicate(label, &newFaceSets));
1103d7d2d1d2SJames Wright     PetscCall(DMAddLabel(cdm, newFaceSets));
1104d7d2d1d2SJames Wright     PetscCall(DMLabelDestroy(&newFaceSets));
1105d7d2d1d2SJames Wright   }
11064054ae39SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
11074054ae39SJames Wright }
11084054ae39SJames Wright 
1109d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Simplex_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate)
1110d71ae5a4SJacob Faibussowitsch {
11119318fe57SMatthew G. Knepley   DM      boundary, vol;
1112c22d3578SMatthew G. Knepley   DMLabel bdlabel;
1113d6218766SMatthew G. Knepley 
1114d6218766SMatthew G. Knepley   PetscFunctionBegin;
11154f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
1116c22d3578SMatthew 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");
11179566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &boundary));
11189566063dSJacob Faibussowitsch   PetscCall(DMSetType(boundary, DMPLEX));
11199566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(boundary, dim, faces, lower, upper, PETSC_FALSE));
11209566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(boundary, NULL, interpolate, &vol));
1121c22d3578SMatthew G. Knepley   PetscCall(DMGetLabel(vol, "marker", &bdlabel));
1122c22d3578SMatthew G. Knepley   if (bdlabel) PetscCall(DMPlexLabelComplete(vol, bdlabel));
11235de52c6dSVaclav Hapla   PetscCall(DMPlexCopy_Internal(dm, PETSC_TRUE, PETSC_FALSE, vol));
112469d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
11254054ae39SJames Wright   if (interpolate) {
11264054ae39SJames Wright     PetscCall(DMPlexInterpolateInPlace_Internal(dm));
1127d7d2d1d2SJames Wright     PetscCall(DMPlexSetBoxLabel_Internal(dm, periodicity));
11284054ae39SJames Wright   }
11299566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&boundary));
11303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1131d6218766SMatthew G. Knepley }
1132d6218766SMatthew G. Knepley 
1133d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCubeMesh_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY, DMBoundaryType bdZ)
1134d71ae5a4SJacob Faibussowitsch {
1135ed0e4b50SMatthew G. Knepley   DMLabel     cutLabel  = NULL;
1136f4eb4c5dSMatthew G. Knepley   PetscInt    markerTop = 1, faceMarkerTop = 1;
1137f4eb4c5dSMatthew G. Knepley   PetscInt    markerBottom = 1, faceMarkerBottom = 1;
1138f4eb4c5dSMatthew G. Knepley   PetscInt    markerFront = 1, faceMarkerFront = 1;
1139f4eb4c5dSMatthew G. Knepley   PetscInt    markerBack = 1, faceMarkerBack = 1;
1140f4eb4c5dSMatthew G. Knepley   PetscInt    markerRight = 1, faceMarkerRight = 1;
1141f4eb4c5dSMatthew G. Knepley   PetscInt    markerLeft = 1, faceMarkerLeft = 1;
11423dfda0b1SToby Isaac   PetscInt    dim;
1143d8211ee3SMatthew G. Knepley   PetscBool   markerSeparate = PETSC_FALSE, cutMarker = PETSC_FALSE;
11443dfda0b1SToby Isaac   PetscMPIInt rank;
11453dfda0b1SToby Isaac 
11463dfda0b1SToby Isaac   PetscFunctionBegin;
11479566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
11489566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
11499566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
11509566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
11519566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
11529371c9d4SSatish 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) {
11539371c9d4SSatish Balay     if (cutMarker) {
11549371c9d4SSatish Balay       PetscCall(DMCreateLabel(dm, "periodic_cut"));
11559371c9d4SSatish Balay       PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
11569371c9d4SSatish Balay     }
1157d8211ee3SMatthew G. Knepley   }
11583dfda0b1SToby Isaac   switch (dim) {
11593dfda0b1SToby Isaac   case 2:
1160f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 3;
1161f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1162f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 2;
1163f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 4;
11643dfda0b1SToby Isaac     break;
11653dfda0b1SToby Isaac   case 3:
1166f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1167f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 2;
1168f4eb4c5dSMatthew G. Knepley     faceMarkerFront  = 3;
1169f4eb4c5dSMatthew G. Knepley     faceMarkerBack   = 4;
1170f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 5;
1171f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 6;
11723dfda0b1SToby Isaac     break;
1173d71ae5a4SJacob Faibussowitsch   default:
1174d71ae5a4SJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
11753dfda0b1SToby Isaac   }
11769566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
1177f4eb4c5dSMatthew G. Knepley   if (markerSeparate) {
1178f4eb4c5dSMatthew G. Knepley     markerBottom = faceMarkerBottom;
1179f4eb4c5dSMatthew G. Knepley     markerTop    = faceMarkerTop;
1180f4eb4c5dSMatthew G. Knepley     markerFront  = faceMarkerFront;
1181f4eb4c5dSMatthew G. Knepley     markerBack   = faceMarkerBack;
1182f4eb4c5dSMatthew G. Knepley     markerRight  = faceMarkerRight;
1183f4eb4c5dSMatthew G. Knepley     markerLeft   = faceMarkerLeft;
11843dfda0b1SToby Isaac   }
11853dfda0b1SToby Isaac   {
1186dd400576SPatrick Sanan     const PetscInt numXEdges    = rank == 0 ? edges[0] : 0;
1187dd400576SPatrick Sanan     const PetscInt numYEdges    = rank == 0 ? edges[1] : 0;
1188dd400576SPatrick Sanan     const PetscInt numZEdges    = rank == 0 ? edges[2] : 0;
1189dd400576SPatrick Sanan     const PetscInt numXVertices = rank == 0 ? (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST ? edges[0] : edges[0] + 1) : 0;
1190dd400576SPatrick Sanan     const PetscInt numYVertices = rank == 0 ? (bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[1] : edges[1] + 1) : 0;
1191dd400576SPatrick Sanan     const PetscInt numZVertices = rank == 0 ? (bdZ == DM_BOUNDARY_PERIODIC || bdZ == DM_BOUNDARY_TWIST ? edges[2] : edges[2] + 1) : 0;
11923dfda0b1SToby Isaac     const PetscInt numCells     = numXEdges * numYEdges * numZEdges;
11933dfda0b1SToby Isaac     const PetscInt numXFaces    = numYEdges * numZEdges;
11943dfda0b1SToby Isaac     const PetscInt numYFaces    = numXEdges * numZEdges;
11953dfda0b1SToby Isaac     const PetscInt numZFaces    = numXEdges * numYEdges;
11963dfda0b1SToby Isaac     const PetscInt numTotXFaces = numXVertices * numXFaces;
11973dfda0b1SToby Isaac     const PetscInt numTotYFaces = numYVertices * numYFaces;
11983dfda0b1SToby Isaac     const PetscInt numTotZFaces = numZVertices * numZFaces;
11993dfda0b1SToby Isaac     const PetscInt numFaces     = numTotXFaces + numTotYFaces + numTotZFaces;
12003dfda0b1SToby Isaac     const PetscInt numTotXEdges = numXEdges * numYVertices * numZVertices;
12013dfda0b1SToby Isaac     const PetscInt numTotYEdges = numYEdges * numXVertices * numZVertices;
12023dfda0b1SToby Isaac     const PetscInt numTotZEdges = numZEdges * numXVertices * numYVertices;
12033dfda0b1SToby Isaac     const PetscInt numVertices  = numXVertices * numYVertices * numZVertices;
12043dfda0b1SToby Isaac     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numTotZEdges;
12053dfda0b1SToby Isaac     const PetscInt firstVertex  = (dim == 2) ? numFaces : numCells;
12063dfda0b1SToby Isaac     const PetscInt firstXFace   = (dim == 2) ? 0 : numCells + numVertices;
12073dfda0b1SToby Isaac     const PetscInt firstYFace   = firstXFace + numTotXFaces;
12083dfda0b1SToby Isaac     const PetscInt firstZFace   = firstYFace + numTotYFaces;
12093dfda0b1SToby Isaac     const PetscInt firstXEdge   = numCells + numFaces + numVertices;
12103dfda0b1SToby Isaac     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
12113dfda0b1SToby Isaac     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
12123dfda0b1SToby Isaac     Vec            coordinates;
12133dfda0b1SToby Isaac     PetscSection   coordSection;
12143dfda0b1SToby Isaac     PetscScalar   *coords;
12153dfda0b1SToby Isaac     PetscInt       coordSize;
12163dfda0b1SToby Isaac     PetscInt       v, vx, vy, vz;
12173dfda0b1SToby Isaac     PetscInt       c, f, fx, fy, fz, e, ex, ey, ez;
12183dfda0b1SToby Isaac 
12199566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numFaces + numEdges + numVertices));
122048a46eb9SPierre Jolivet     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
122148a46eb9SPierre Jolivet     for (f = firstXFace; f < firstXFace + numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
122248a46eb9SPierre Jolivet     for (e = firstXEdge; e < firstXEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
12239566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
12243dfda0b1SToby Isaac     /* Build cells */
12253dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
12263dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
12273dfda0b1SToby Isaac         for (fx = 0; fx < numXEdges; ++fx) {
12283dfda0b1SToby Isaac           PetscInt cell  = (fz * numYEdges + fy) * numXEdges + fx;
12293dfda0b1SToby Isaac           PetscInt faceB = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
12303dfda0b1SToby Isaac           PetscInt faceT = firstZFace + (fy * numXEdges + fx) * numZVertices + ((fz + 1) % numZVertices);
12313dfda0b1SToby Isaac           PetscInt faceF = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
12323dfda0b1SToby Isaac           PetscInt faceK = firstYFace + (fz * numXEdges + fx) * numYVertices + ((fy + 1) % numYVertices);
12333dfda0b1SToby Isaac           PetscInt faceL = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
12343dfda0b1SToby Isaac           PetscInt faceR = firstXFace + (fz * numYEdges + fy) * numXVertices + ((fx + 1) % numXVertices);
12353dfda0b1SToby Isaac           /* B,  T,  F,  K,  R,  L */
1236b5a892a1SMatthew G. Knepley           PetscInt ornt[6] = {-2, 0, 0, -3, 0, -2}; /* ??? */
123742206facSLisandro Dalcin           PetscInt cone[6];
12383dfda0b1SToby Isaac 
12393dfda0b1SToby Isaac           /* no boundary twisting in 3D */
12409371c9d4SSatish Balay           cone[0] = faceB;
12419371c9d4SSatish Balay           cone[1] = faceT;
12429371c9d4SSatish Balay           cone[2] = faceF;
12439371c9d4SSatish Balay           cone[3] = faceK;
12449371c9d4SSatish Balay           cone[4] = faceR;
12459371c9d4SSatish Balay           cone[5] = faceL;
12469566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, cell, cone));
12479566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, cell, ornt));
12489566063dSJacob Faibussowitsch           if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
12499566063dSJacob Faibussowitsch           if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
12509566063dSJacob Faibussowitsch           if (bdZ != DM_BOUNDARY_NONE && fz == numZEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
12513dfda0b1SToby Isaac         }
12523dfda0b1SToby Isaac       }
12533dfda0b1SToby Isaac     }
12543dfda0b1SToby Isaac     /* Build x faces */
12553dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
12563dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
12573dfda0b1SToby Isaac         for (fx = 0; fx < numXVertices; ++fx) {
12583dfda0b1SToby Isaac           PetscInt face    = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
12593dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
12603dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (((fy + 1) % numYVertices) * numXVertices + fx) * numZEdges + fz;
12613dfda0b1SToby Isaac           PetscInt edgeB   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
12623dfda0b1SToby Isaac           PetscInt edgeT   = firstYEdge + (((fz + 1) % numZVertices) * numXVertices + fx) * numYEdges + fy;
1263b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
12643dfda0b1SToby Isaac           PetscInt cone[4];
12653dfda0b1SToby Isaac 
12663dfda0b1SToby Isaac           if (dim == 3) {
12673dfda0b1SToby Isaac             /* markers */
12683dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
12693dfda0b1SToby Isaac               if (fx == numXVertices - 1) {
12709566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerRight));
12719566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerRight));
12729371c9d4SSatish Balay               } else if (fx == 0) {
12739566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerLeft));
12749566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerLeft));
12753dfda0b1SToby Isaac               }
12763dfda0b1SToby Isaac             }
12773dfda0b1SToby Isaac           }
12789371c9d4SSatish Balay           cone[0] = edgeB;
12799371c9d4SSatish Balay           cone[1] = edgeR;
12809371c9d4SSatish Balay           cone[2] = edgeT;
12819371c9d4SSatish Balay           cone[3] = edgeL;
12829566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
12839566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
12843dfda0b1SToby Isaac         }
12853dfda0b1SToby Isaac       }
12863dfda0b1SToby Isaac     }
12873dfda0b1SToby Isaac     /* Build y faces */
12883dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
128942206facSLisandro Dalcin       for (fx = 0; fx < numXEdges; ++fx) {
12903dfda0b1SToby Isaac         for (fy = 0; fy < numYVertices; ++fy) {
12913dfda0b1SToby Isaac           PetscInt face    = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
12923dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
12933dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (fy * numXVertices + ((fx + 1) % numXVertices)) * numZEdges + fz;
12943dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
12953dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (((fz + 1) % numZVertices) * numYVertices + fy) * numXEdges + fx;
1296b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
12973dfda0b1SToby Isaac           PetscInt cone[4];
12983dfda0b1SToby Isaac 
12993dfda0b1SToby Isaac           if (dim == 3) {
13003dfda0b1SToby Isaac             /* markers */
13013dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
13023dfda0b1SToby Isaac               if (fy == numYVertices - 1) {
13039566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBack));
13049566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBack));
13059371c9d4SSatish Balay               } else if (fy == 0) {
13069566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerFront));
13079566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerFront));
13083dfda0b1SToby Isaac               }
13093dfda0b1SToby Isaac             }
13103dfda0b1SToby Isaac           }
13119371c9d4SSatish Balay           cone[0] = edgeB;
13129371c9d4SSatish Balay           cone[1] = edgeR;
13139371c9d4SSatish Balay           cone[2] = edgeT;
13149371c9d4SSatish Balay           cone[3] = edgeL;
13159566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
13169566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
13173dfda0b1SToby Isaac         }
13183dfda0b1SToby Isaac       }
13193dfda0b1SToby Isaac     }
13203dfda0b1SToby Isaac     /* Build z faces */
13213dfda0b1SToby Isaac     for (fy = 0; fy < numYEdges; ++fy) {
13223dfda0b1SToby Isaac       for (fx = 0; fx < numXEdges; ++fx) {
13233dfda0b1SToby Isaac         for (fz = 0; fz < numZVertices; fz++) {
13243dfda0b1SToby Isaac           PetscInt face    = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
13253dfda0b1SToby Isaac           PetscInt edgeL   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
13263dfda0b1SToby Isaac           PetscInt edgeR   = firstYEdge + (fz * numXVertices + ((fx + 1) % numXVertices)) * numYEdges + fy;
13273dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
13283dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (fz * numYVertices + ((fy + 1) % numYVertices)) * numXEdges + fx;
1329b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
13303dfda0b1SToby Isaac           PetscInt cone[4];
13313dfda0b1SToby Isaac 
13323dfda0b1SToby Isaac           if (dim == 2) {
13339371c9d4SSatish Balay             if (bdX == DM_BOUNDARY_TWIST && fx == numXEdges - 1) {
13349371c9d4SSatish Balay               edgeR += numYEdges - 1 - 2 * fy;
13359371c9d4SSatish Balay               ornt[1] = -1;
13369371c9d4SSatish Balay             }
13379371c9d4SSatish Balay             if (bdY == DM_BOUNDARY_TWIST && fy == numYEdges - 1) {
13389371c9d4SSatish Balay               edgeT += numXEdges - 1 - 2 * fx;
13399371c9d4SSatish Balay               ornt[2] = 0;
13409371c9d4SSatish Balay             }
13419566063dSJacob Faibussowitsch             if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
13429566063dSJacob Faibussowitsch             if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
1343d1c88043SMatthew G. Knepley           } else {
13443dfda0b1SToby Isaac             /* markers */
13453dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
13463dfda0b1SToby Isaac               if (fz == numZVertices - 1) {
13479566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerTop));
13489566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerTop));
13499371c9d4SSatish Balay               } else if (fz == 0) {
13509566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBottom));
13519566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBottom));
13523dfda0b1SToby Isaac               }
13533dfda0b1SToby Isaac             }
13543dfda0b1SToby Isaac           }
13559371c9d4SSatish Balay           cone[0] = edgeB;
13569371c9d4SSatish Balay           cone[1] = edgeR;
13579371c9d4SSatish Balay           cone[2] = edgeT;
13589371c9d4SSatish Balay           cone[3] = edgeL;
13599566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
13609566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
13613dfda0b1SToby Isaac         }
13623dfda0b1SToby Isaac       }
13633dfda0b1SToby Isaac     }
13643dfda0b1SToby Isaac     /* Build Z edges*/
13653dfda0b1SToby Isaac     for (vy = 0; vy < numYVertices; vy++) {
13663dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
13673dfda0b1SToby Isaac         for (ez = 0; ez < numZEdges; ez++) {
13683dfda0b1SToby Isaac           const PetscInt edge    = firstZEdge + (vy * numXVertices + vx) * numZEdges + ez;
13693dfda0b1SToby Isaac           const PetscInt vertexB = firstVertex + (ez * numYVertices + vy) * numXVertices + vx;
13703dfda0b1SToby Isaac           const PetscInt vertexT = firstVertex + (((ez + 1) % numZVertices) * numYVertices + vy) * numXVertices + vx;
13713dfda0b1SToby Isaac           PetscInt       cone[2];
13723dfda0b1SToby Isaac 
13739371c9d4SSatish Balay           cone[0] = vertexB;
13749371c9d4SSatish Balay           cone[1] = vertexT;
1375c2df9bbfSMatthew G. Knepley           PetscCall(DMPlexSetCone(dm, edge, cone));
13763dfda0b1SToby Isaac           if (dim == 3) {
13773dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
13783dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
13799566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1380c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1381c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1382c2df9bbfSMatthew G. Knepley               } else if (vx == 0) {
13839566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1384c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1385c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
13863dfda0b1SToby Isaac               }
13873dfda0b1SToby Isaac             }
13883dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
13893dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
13909566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1391c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1392c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1393c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
13949566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1395c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1396c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
13973dfda0b1SToby Isaac               }
13983dfda0b1SToby Isaac             }
13993dfda0b1SToby Isaac           }
14003dfda0b1SToby Isaac         }
14013dfda0b1SToby Isaac       }
14023dfda0b1SToby Isaac     }
14033dfda0b1SToby Isaac     /* Build Y edges*/
14043dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
14053dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
14063dfda0b1SToby Isaac         for (ey = 0; ey < numYEdges; ey++) {
14073dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdY == DM_BOUNDARY_TWIST && ey == numYEdges - 1) ? (numXVertices - vx - 1) : (vz * numYVertices + ((ey + 1) % numYVertices)) * numXVertices + vx;
14083dfda0b1SToby Isaac           const PetscInt edge    = firstYEdge + (vz * numXVertices + vx) * numYEdges + ey;
14093dfda0b1SToby Isaac           const PetscInt vertexF = firstVertex + (vz * numYVertices + ey) * numXVertices + vx;
14103dfda0b1SToby Isaac           const PetscInt vertexK = firstVertex + nextv;
14113dfda0b1SToby Isaac           PetscInt       cone[2];
14123dfda0b1SToby Isaac 
14139371c9d4SSatish Balay           cone[0] = vertexF;
14149371c9d4SSatish Balay           cone[1] = vertexK;
14159566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
14163dfda0b1SToby Isaac           if (dim == 2) {
14173dfda0b1SToby Isaac             if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
14183dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
14199566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight));
14209566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
14219566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1422c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1423d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
14249566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft));
14259566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
14269566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1427c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
14283dfda0b1SToby Isaac               }
1429d8211ee3SMatthew G. Knepley             } else {
14304c67ea77SStefano Zampini               if (vx == 0 && cutLabel) {
14319566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
14329566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1433c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
14343dfda0b1SToby Isaac               }
1435d8211ee3SMatthew G. Knepley             }
1436d8211ee3SMatthew G. Knepley           } else {
14373dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
14383dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
14399566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1440c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1441c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1442d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
14439566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1444c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1445c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
14463dfda0b1SToby Isaac               }
14473dfda0b1SToby Isaac             }
14483dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
14493dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
14509566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1451c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1452c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1453d8211ee3SMatthew G. Knepley               } else if (vz == 0) {
14549566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1455c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1456c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
14573dfda0b1SToby Isaac               }
14583dfda0b1SToby Isaac             }
14593dfda0b1SToby Isaac           }
14603dfda0b1SToby Isaac         }
14613dfda0b1SToby Isaac       }
14623dfda0b1SToby Isaac     }
14633dfda0b1SToby Isaac     /* Build X edges*/
14643dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
14653dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; vy++) {
14663dfda0b1SToby Isaac         for (ex = 0; ex < numXEdges; ex++) {
14673dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdX == DM_BOUNDARY_TWIST && ex == numXEdges - 1) ? (numYVertices - vy - 1) * numXVertices : (vz * numYVertices + vy) * numXVertices + (ex + 1) % numXVertices;
14683dfda0b1SToby Isaac           const PetscInt edge    = firstXEdge + (vz * numYVertices + vy) * numXEdges + ex;
14693dfda0b1SToby Isaac           const PetscInt vertexL = firstVertex + (vz * numYVertices + vy) * numXVertices + ex;
14703dfda0b1SToby Isaac           const PetscInt vertexR = firstVertex + nextv;
14713dfda0b1SToby Isaac           PetscInt       cone[2];
14723dfda0b1SToby Isaac 
14739371c9d4SSatish Balay           cone[0] = vertexL;
14749371c9d4SSatish Balay           cone[1] = vertexR;
14759566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
14763dfda0b1SToby Isaac           if (dim == 2) {
14773dfda0b1SToby Isaac             if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
14783dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
14799566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop));
14809566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
14819566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1482c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1483d8211ee3SMatthew G. Knepley               } else if (vy == 0) {
14849566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom));
14859566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
14869566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1487c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
14883dfda0b1SToby Isaac               }
1489d8211ee3SMatthew G. Knepley             } else {
14904c67ea77SStefano Zampini               if (vy == 0 && cutLabel) {
14919566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
14929566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1493c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
14943dfda0b1SToby Isaac               }
1495d8211ee3SMatthew G. Knepley             }
1496d8211ee3SMatthew G. Knepley           } else {
14973dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
14983dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
14999566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1500c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1501c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1502c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
15039566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1504c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1505c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
15063dfda0b1SToby Isaac               }
15073dfda0b1SToby Isaac             }
15083dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
15093dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
15109566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1511c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1512c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1513c2df9bbfSMatthew G. Knepley               } else if (vz == 0) {
15149566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1515c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1516c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
15173dfda0b1SToby Isaac               }
15183dfda0b1SToby Isaac             }
15193dfda0b1SToby Isaac           }
15203dfda0b1SToby Isaac         }
15213dfda0b1SToby Isaac       }
15223dfda0b1SToby Isaac     }
15239566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
15249566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
15253dfda0b1SToby Isaac     /* Build coordinates */
15269566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
15279566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
15289566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
15299566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVertices));
15303dfda0b1SToby Isaac     for (v = firstVertex; v < firstVertex + numVertices; ++v) {
15319566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
15329566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
15333dfda0b1SToby Isaac     }
15349566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
15359566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
15369566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
15379566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
15389566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
15399566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
15409566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
15419566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
15423dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; ++vz) {
15433dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; ++vy) {
15443dfda0b1SToby Isaac         for (vx = 0; vx < numXVertices; ++vx) {
15453dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * vx;
15463dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * vy;
1547ad540459SPierre Jolivet           if (dim == 3) coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 2] = lower[2] + ((upper[2] - lower[2]) / numZEdges) * vz;
15483dfda0b1SToby Isaac         }
15493dfda0b1SToby Isaac       }
15503dfda0b1SToby Isaac     }
15519566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
15529566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
15539566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
15543dfda0b1SToby Isaac   }
15553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15563dfda0b1SToby Isaac }
15573dfda0b1SToby Isaac 
1558d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Tensor_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1559d71ae5a4SJacob Faibussowitsch {
15609318fe57SMatthew G. Knepley   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
15619318fe57SMatthew G. Knepley   PetscInt       fac[3] = {0, 0, 0}, d;
1562552f7358SJed Brown 
1563552f7358SJed Brown   PetscFunctionBegin;
15644f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
15659318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
15669566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
15679371c9d4SSatish Balay   for (d = 0; d < dim; ++d) {
15689371c9d4SSatish Balay     fac[d] = faces[d];
15699371c9d4SSatish Balay     bdt[d] = periodicity[d];
15709371c9d4SSatish Balay   }
15719566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateCubeMesh_Internal(dm, lower, upper, fac, bdt[0], bdt[1], bdt[2]));
15729371c9d4SSatish 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))) {
15736858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
15746858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
1575552f7358SJed Brown 
15769318fe57SMatthew G. Knepley     for (d = 0; d < dim; ++d) {
15776858538eSMatthew G. Knepley       if (periodicity[d] != DM_BOUNDARY_NONE) {
15789318fe57SMatthew G. Knepley         L[d]       = upper[d] - lower[d];
15799318fe57SMatthew G. Knepley         maxCell[d] = 1.1 * (L[d] / PetscMax(1, faces[d]));
1580768d5fceSMatthew G. Knepley       }
15816858538eSMatthew G. Knepley     }
15824fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
1583768d5fceSMatthew G. Knepley   }
15849566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
15853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15869318fe57SMatthew G. Knepley }
15879318fe57SMatthew G. Knepley 
15885dca41c3SJed 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)
1589d71ae5a4SJacob Faibussowitsch {
15909318fe57SMatthew G. Knepley   PetscFunctionBegin;
159146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
15925dca41c3SJed Brown   if (shape == DM_SHAPE_ZBOX) PetscCall(DMPlexCreateBoxMesh_Tensor_SFC_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
15936725e60dSJed Brown   else if (dim == 1) PetscCall(DMPlexCreateLineMesh_Internal(dm, faces[0], lower[0], upper[0], periodicity[0]));
15949566063dSJacob Faibussowitsch   else if (simplex) PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
15959566063dSJacob Faibussowitsch   else PetscCall(DMPlexCreateBoxMesh_Tensor_Internal(dm, dim, faces, lower, upper, periodicity));
15969318fe57SMatthew G. Knepley   if (!interpolate && dim > 1 && !simplex) {
1597768d5fceSMatthew G. Knepley     DM udm;
1598768d5fceSMatthew G. Knepley 
15999566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(dm, &udm));
16009566063dSJacob Faibussowitsch     PetscCall(DMPlexCopyCoordinates(dm, udm));
160169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &udm));
1602768d5fceSMatthew G. Knepley   }
160346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
16043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1605c8c68bd8SToby Isaac }
1606c8c68bd8SToby Isaac 
16075d83a8b1SBarry Smith /*@
1608768d5fceSMatthew G. Knepley   DMPlexCreateBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using simplices or tensor cells (hexahedra).
1609768d5fceSMatthew G. Knepley 
1610d083f849SBarry Smith   Collective
1611768d5fceSMatthew G. Knepley 
1612768d5fceSMatthew G. Knepley   Input Parameters:
1613a1cb98faSBarry Smith + comm               - The communicator for the `DM` object
1614768d5fceSMatthew G. Knepley . dim                - The spatial dimension
1615a1cb98faSBarry Smith . simplex            - `PETSC_TRUE` for simplices, `PETSC_FALSE` for tensor cells
161620f4b53cSBarry Smith . faces              - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
161720f4b53cSBarry Smith . lower              - The lower left corner, or `NULL` for (0, 0, 0)
161820f4b53cSBarry Smith . upper              - The upper right corner, or `NULL` for (1, 1, 1)
161920f4b53cSBarry Smith . periodicity        - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
162042108689Sksagiyam . interpolate        - Flag to create intermediate mesh pieces (edges, faces)
162142108689Sksagiyam . localizationHeight - Flag to localize edges and faces in addition to cells; only significant for periodic meshes
162242108689Sksagiyam - sparseLocalize     - Flag to localize coordinates only for cells near the periodic boundary; only significant for periodic meshes
1623768d5fceSMatthew G. Knepley 
1624768d5fceSMatthew G. Knepley   Output Parameter:
1625a1cb98faSBarry Smith . dm - The `DM` object
1626768d5fceSMatthew G. Knepley 
1627768d5fceSMatthew G. Knepley   Level: beginner
1628768d5fceSMatthew G. Knepley 
1629a1cb98faSBarry Smith   Note:
1630a1cb98faSBarry Smith   To customize this mesh using options, use
1631a1cb98faSBarry Smith .vb
1632a1cb98faSBarry Smith   DMCreate(comm, &dm);
1633a1cb98faSBarry Smith   DMSetType(dm, DMPLEX);
1634a1cb98faSBarry Smith   DMSetFromOptions(dm);
1635a1cb98faSBarry Smith .ve
1636a1cb98faSBarry Smith   and use the options in `DMSetFromOptions()`.
1637a1cb98faSBarry Smith 
1638a4e35b19SJacob Faibussowitsch   Here is the numbering returned for 2 faces in each direction for tensor cells\:
1639a1cb98faSBarry Smith .vb
1640a1cb98faSBarry Smith  10---17---11---18----12
1641a1cb98faSBarry Smith   |         |         |
1642a1cb98faSBarry Smith   |         |         |
1643a1cb98faSBarry Smith  20    2   22    3    24
1644a1cb98faSBarry Smith   |         |         |
1645a1cb98faSBarry Smith   |         |         |
1646a1cb98faSBarry Smith   7---15----8---16----9
1647a1cb98faSBarry Smith   |         |         |
1648a1cb98faSBarry Smith   |         |         |
1649a1cb98faSBarry Smith  19    0   21    1   23
1650a1cb98faSBarry Smith   |         |         |
1651a1cb98faSBarry Smith   |         |         |
1652a1cb98faSBarry Smith   4---13----5---14----6
1653a1cb98faSBarry Smith .ve
1654a1cb98faSBarry Smith   and for simplicial cells
1655a1cb98faSBarry Smith .vb
1656a1cb98faSBarry Smith  14----8---15----9----16
1657a1cb98faSBarry Smith   |\     5  |\      7 |
1658a1cb98faSBarry Smith   | \       | \       |
1659a1cb98faSBarry Smith  13   2    14    3    15
1660a1cb98faSBarry Smith   | 4   \   | 6   \   |
1661a1cb98faSBarry Smith   |       \ |       \ |
1662a1cb98faSBarry Smith  11----6---12----7----13
1663a1cb98faSBarry Smith   |\        |\        |
1664a1cb98faSBarry Smith   | \    1  | \     3 |
1665a1cb98faSBarry Smith  10   0    11    1    12
1666a1cb98faSBarry Smith   | 0   \   | 2   \   |
1667a1cb98faSBarry Smith   |       \ |       \ |
1668a1cb98faSBarry Smith   8----4----9----5----10
1669a1cb98faSBarry Smith .ve
1670a1cb98faSBarry Smith 
16711cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
1672768d5fceSMatthew G. Knepley @*/
167342108689Sksagiyam 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)
1674d71ae5a4SJacob Faibussowitsch {
16759318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
1676fdbf62faSLisandro Dalcin   PetscReal      low[3] = {0, 0, 0};
1677fdbf62faSLisandro Dalcin   PetscReal      upp[3] = {1, 1, 1};
1678fdbf62faSLisandro Dalcin   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
1679552f7358SJed Brown 
1680768d5fceSMatthew G. Knepley   PetscFunctionBegin;
16819566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
16829566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
16835dca41c3SJed Brown   PetscCall(DMPlexCreateBoxMesh_Internal(*dm, DM_SHAPE_BOX, dim, simplex, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt, interpolate));
168442108689Sksagiyam   if (periodicity) {
168542108689Sksagiyam     DM cdm;
168642108689Sksagiyam 
168742108689Sksagiyam     PetscCall(DMGetCoordinateDM(*dm, &cdm));
168842108689Sksagiyam     PetscCall(DMPlexSetMaxProjectionHeight(cdm, localizationHeight));
168942108689Sksagiyam     PetscCall(DMSetSparseLocalize(*dm, sparseLocalize));
169042108689Sksagiyam     PetscCall(DMLocalizeCoordinates(*dm));
169142108689Sksagiyam   }
16923ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
16939318fe57SMatthew G. Knepley }
1694fdbf62faSLisandro Dalcin 
1695d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeBoxMesh_Internal(DM dm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1696d71ae5a4SJacob Faibussowitsch {
16979318fe57SMatthew G. Knepley   DM       bdm, vol;
16989318fe57SMatthew G. Knepley   PetscInt i;
16999318fe57SMatthew G. Knepley 
17009318fe57SMatthew G. Knepley   PetscFunctionBegin;
17011fcf445aSMatthew G. Knepley   // TODO Now we can support periodicity
170208401ef6SPierre Jolivet   for (i = 0; i < 3; ++i) PetscCheck(periodicity[i] == DM_BOUNDARY_NONE, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Periodicity not yet supported");
17039566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &bdm));
17049566063dSJacob Faibussowitsch   PetscCall(DMSetType(bdm, DMPLEX));
17059566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(bdm, 2));
170646139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, bdm, 0, 0, 0));
17079566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(bdm, 2, faces, lower, upper, periodicity, PETSC_TRUE));
17081fcf445aSMatthew G. Knepley   PetscCall(DMPlexExtrude(bdm, faces[2], upper[2] - lower[2], PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, NULL, NULL, &vol));
170946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, bdm, 0, 0, 0));
17109566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&bdm));
171169d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
17129318fe57SMatthew G. Knepley   if (lower[2] != 0.0) {
17139318fe57SMatthew G. Knepley     Vec          v;
17149318fe57SMatthew G. Knepley     PetscScalar *x;
17159318fe57SMatthew G. Knepley     PetscInt     cDim, n;
17169318fe57SMatthew G. Knepley 
17179566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &v));
17189566063dSJacob Faibussowitsch     PetscCall(VecGetBlockSize(v, &cDim));
17199566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(v, &n));
17209566063dSJacob Faibussowitsch     PetscCall(VecGetArray(v, &x));
17219318fe57SMatthew G. Knepley     x += cDim;
17229318fe57SMatthew G. Knepley     for (i = 0; i < n; i += cDim) x[i] += lower[2];
17239566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(v, &x));
17249566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, v));
17259318fe57SMatthew G. Knepley   }
17263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1727552f7358SJed Brown }
1728552f7358SJed Brown 
172900dabe28SStefano Zampini /*@
173039f4f5dbSPierre Jolivet   DMPlexCreateWedgeBoxMesh - Creates a 3-D mesh tessellating the (x,y) plane and extruding in the third direction using wedge cells.
173100dabe28SStefano Zampini 
1732d083f849SBarry Smith   Collective
173300dabe28SStefano Zampini 
173400dabe28SStefano Zampini   Input Parameters:
1735a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
173620f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1, 1, 1)
173720f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
173820f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
173920f4b53cSBarry Smith . periodicity - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
1740a1cb98faSBarry Smith . orderHeight - If `PETSC_TRUE`, orders the extruded cells in the height first. Otherwise, orders the cell on the layers first
174100dabe28SStefano Zampini - interpolate - Flag to create intermediate mesh pieces (edges, faces)
174200dabe28SStefano Zampini 
174300dabe28SStefano Zampini   Output Parameter:
1744a1cb98faSBarry Smith . dm - The `DM` object
174500dabe28SStefano Zampini 
174600dabe28SStefano Zampini   Level: beginner
174700dabe28SStefano Zampini 
17481cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateWedgeCylinderMesh()`, `DMExtrude()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
174900dabe28SStefano Zampini @*/
1750d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeBoxMesh(MPI_Comm comm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool orderHeight, PetscBool interpolate, DM *dm)
1751d71ae5a4SJacob Faibussowitsch {
17529318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
175300dabe28SStefano Zampini   PetscReal      low[3] = {0, 0, 0};
175400dabe28SStefano Zampini   PetscReal      upp[3] = {1, 1, 1};
175500dabe28SStefano Zampini   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
175600dabe28SStefano Zampini 
175700dabe28SStefano Zampini   PetscFunctionBegin;
17589566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
17599566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
17609566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeBoxMesh_Internal(*dm, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt));
1761d410b0cfSMatthew G. Knepley   if (!interpolate) {
1762d410b0cfSMatthew G. Knepley     DM udm;
176300dabe28SStefano Zampini 
17649566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(*dm, &udm));
176569d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(*dm, &udm));
176600dabe28SStefano Zampini   }
17677ff04441SMatthew G. Knepley   if (periodicity) PetscCall(DMLocalizeCoordinates(*dm));
17683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
176900dabe28SStefano Zampini }
177000dabe28SStefano Zampini 
1771cfb853baSMatthew G. Knepley /*
1772cfb853baSMatthew 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.
1773cfb853baSMatthew G. Knepley 
1774cfb853baSMatthew G. Knepley   Input Parameters:
1775cfb853baSMatthew G. Knepley + len - The length of the tuple
1776cfb853baSMatthew G. Knepley . max - The maximum for each dimension, so values are in [0, max)
1777cfb853baSMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition
1778cfb853baSMatthew G. Knepley 
1779cfb853baSMatthew G. Knepley   Output Parameter:
178020f4b53cSBarry Smith . tup - A tuple of `len` integers whose entries are at most `max`
1781cfb853baSMatthew G. Knepley 
1782cfb853baSMatthew G. Knepley   Level: developer
1783cfb853baSMatthew G. Knepley 
178420f4b53cSBarry Smith   Note:
178520f4b53cSBarry Smith   Ordering is lexicographic with lowest index as least significant in ordering.
178620f4b53cSBarry 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}.
178720f4b53cSBarry Smith 
1788cfb853baSMatthew G. Knepley .seealso: PetscDualSpaceTensorPointLexicographic_Internal(), PetscDualSpaceLatticePointLexicographic_Internal()
1789cfb853baSMatthew G. Knepley */
1790cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexTensorPointLexicographic_Private(PetscInt len, const PetscInt max[], PetscInt tup[])
1791cfb853baSMatthew G. Knepley {
1792cfb853baSMatthew G. Knepley   PetscInt i;
1793cfb853baSMatthew G. Knepley 
1794cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1795cfb853baSMatthew G. Knepley   for (i = 0; i < len; ++i) {
1796cfb853baSMatthew G. Knepley     if (tup[i] < max[i] - 1) {
1797cfb853baSMatthew G. Knepley       break;
1798cfb853baSMatthew G. Knepley     } else {
1799cfb853baSMatthew G. Knepley       tup[i] = 0;
1800cfb853baSMatthew G. Knepley     }
1801cfb853baSMatthew G. Knepley   }
1802cfb853baSMatthew G. Knepley   if (i == len) tup[i - 1] = max[i - 1];
1803cfb853baSMatthew G. Knepley   else ++tup[i];
18043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1805cfb853baSMatthew G. Knepley }
1806cfb853baSMatthew G. Knepley 
1807cfb853baSMatthew G. Knepley static PetscInt TupleToIndex_Private(PetscInt len, const PetscInt max[], const PetscInt tup[])
1808cfb853baSMatthew G. Knepley {
1809*8d2ec52aSSatish Balay   PetscInt idx = tup[len - 1];
1810cfb853baSMatthew G. Knepley 
1811*8d2ec52aSSatish Balay   for (PetscInt i = len - 2; i >= 0; --i) {
1812cfb853baSMatthew G. Knepley     idx *= max[i];
1813cfb853baSMatthew G. Knepley     idx += tup[i];
1814cfb853baSMatthew G. Knepley   }
1815cfb853baSMatthew G. Knepley   return idx;
1816cfb853baSMatthew G. Knepley }
1817cfb853baSMatthew G. Knepley 
1818*8d2ec52aSSatish Balay static void IndexToTuple_Private(PetscInt len, const PetscInt max[], PetscInt idx, PetscInt tup[])
1819cfb853baSMatthew G. Knepley {
1820*8d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
1821*8d2ec52aSSatish Balay     tup[i] = idx % max[i];
1822*8d2ec52aSSatish Balay     idx    = (idx - tup[i]) / max[i];
1823*8d2ec52aSSatish Balay   }
1824*8d2ec52aSSatish Balay }
1825*8d2ec52aSSatish Balay 
1826*8d2ec52aSSatish Balay static void TupleToRanks_Private(PetscInt len, const PetscInt max[], const PetscInt procs[], const PetscInt tup[], PetscInt ranks[])
1827*8d2ec52aSSatish Balay {
1828*8d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
1829*8d2ec52aSSatish Balay     const PetscInt div = max[i] / procs[i];
1830*8d2ec52aSSatish Balay     const PetscInt rem = max[i] % procs[i];
1831*8d2ec52aSSatish Balay     const PetscInt idx = (tup[i] < 0 ? max[i] + tup[i] : tup[i]) % max[i];
1832*8d2ec52aSSatish Balay 
1833*8d2ec52aSSatish Balay     if (idx < rem * (div + 1)) ranks[i] = idx / (div + 1);
1834*8d2ec52aSSatish Balay     else ranks[i] = rem + (idx - rem * (div + 1)) / div;
1835*8d2ec52aSSatish Balay   }
1836*8d2ec52aSSatish Balay }
1837*8d2ec52aSSatish Balay 
1838*8d2ec52aSSatish Balay static void RanksToSizes_Private(PetscInt len, const PetscInt max[], const PetscInt procs[], const PetscInt ranks[], PetscInt sizes[])
1839*8d2ec52aSSatish Balay {
1840*8d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
1841*8d2ec52aSSatish Balay     const PetscInt div = max[i] / procs[i];
1842*8d2ec52aSSatish Balay     const PetscInt rem = max[i] % procs[i];
1843*8d2ec52aSSatish Balay 
1844*8d2ec52aSSatish Balay     sizes[i] = ranks[i] < rem ? div + 1 : div;
1845*8d2ec52aSSatish Balay   }
1846*8d2ec52aSSatish Balay }
1847*8d2ec52aSSatish Balay 
1848*8d2ec52aSSatish Balay /*
1849*8d2ec52aSSatish Balay   In serial, the mesh is completely periodic. In parallel, we will include a layer of ghost vertices around the patch, so our edges will not wrap around in parallel. This will instead be handled by the SF.
1850*8d2ec52aSSatish Balay 
1851*8d2ec52aSSatish Balay   The cone for all edges will always be complete. Even in the presence of boundaries, we will keep all support sizes constant. When an edge would not exist in the support, we will create one to wrap back periodically. This allows for uniform stencils, and that edge will not be used for computation.
1852*8d2ec52aSSatish Balay 
1853*8d2ec52aSSatish Balay   All point which do not attain the vertex lower or upper bound in any dimension are owned, the rest are leaves owned by another process and present in the SF.
1854*8d2ec52aSSatish Balay */
1855*8d2ec52aSSatish Balay static PetscErrorCode DMPlexCreateHypercubicMesh_Internal(DM dm, PetscInt dim, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], PetscInt overlap, const DMBoundaryType bd[])
1856*8d2ec52aSSatish Balay {
1857*8d2ec52aSSatish Balay   const PetscInt debug = ((DM_Plex *)dm->data)->printAdj;
1858*8d2ec52aSSatish Balay   PetscSF        sf;
1859cfb853baSMatthew G. Knepley   Vec            coordinates;
1860cfb853baSMatthew G. Knepley   PetscSection   coordSection;
1861cfb853baSMatthew G. Knepley   DMLabel        cutLabel    = NULL;
1862cfb853baSMatthew G. Knepley   PetscBool      cutMarker   = PETSC_FALSE;
1863cfb853baSMatthew G. Knepley   PetscBool      periodic    = PETSC_FALSE;
1864*8d2ec52aSSatish Balay   PetscInt       numCells    = 1;
1865*8d2ec52aSSatish Balay   PetscInt       numVertices = 1;
1866*8d2ec52aSSatish Balay   PetscSFNode   *remotes;
1867cfb853baSMatthew G. Knepley   PetscScalar   *coords;
1868*8d2ec52aSSatish Balay   PetscInt      *procs;     // The number of processes along each dimension
1869*8d2ec52aSSatish Balay   PetscInt      *lrank;     // Rank in each dimension, lrank[d] \in [0, procs[d])
1870*8d2ec52aSSatish Balay   PetscInt      *ledges;    // The number of edges along each dimension for this process
1871*8d2ec52aSSatish Balay   PetscInt      *vstart;    // The first vertex along each dimension on this processes
1872*8d2ec52aSSatish Balay   PetscInt      *vertices;  // The number of vertices along each dimension on this process
1873*8d2ec52aSSatish Balay   PetscInt      *rvert;     // The global (not local) vertex number along each dimension
1874*8d2ec52aSSatish Balay   PetscInt      *rrank;     // The rank along each dimension for the process owning rvert[]
1875*8d2ec52aSSatish Balay   PetscInt      *rvertices; // The number of vertices along each dimension for the process rrank[]
1876*8d2ec52aSSatish Balay   PetscInt      *vert, *vtmp, *supp, cone[2], *leaves;
1877*8d2ec52aSSatish Balay   PetscInt       cell = 0, coordSize, Nl = 0, Nl2 = 0;
1878*8d2ec52aSSatish Balay   PetscMPIInt    rank, size;
1879*8d2ec52aSSatish Balay   MPI_Comm       comm;
1880cfb853baSMatthew G. Knepley 
1881cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1882*8d2ec52aSSatish Balay   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
1883*8d2ec52aSSatish Balay   PetscCallMPI(MPI_Comm_rank(comm, &rank));
1884*8d2ec52aSSatish Balay   PetscCallMPI(MPI_Comm_size(comm, &size));
1885cfb853baSMatthew G. Knepley   PetscCall(DMSetDimension(dm, dim));
1886*8d2ec52aSSatish Balay   PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
1887*8d2ec52aSSatish Balay   PetscCall(PetscCalloc4(dim, &procs, dim, &lrank, dim, &rrank, 2 * dim, &supp));
1888*8d2ec52aSSatish Balay   PetscCall(PetscCalloc7(dim, &ledges, dim, &vertices, dim, &rvertices, dim, &vert, dim, &rvert, dim, &vstart, dim, &vtmp));
1889cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
1890cfb853baSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
1891*8d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) periodic = (periodic || bd[d] == DM_BOUNDARY_PERIODIC) ? PETSC_TRUE : PETSC_FALSE;
1892cfb853baSMatthew G. Knepley   if (periodic && cutMarker) {
1893cfb853baSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, "periodic_cut"));
1894cfb853baSMatthew G. Knepley     PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
1895cfb853baSMatthew G. Knepley   }
1896*8d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) PetscCheck(bd[d] == DM_BOUNDARY_PERIODIC, comm, PETSC_ERR_SUP, "Hypercubic mesh must be periodic now");
1897*8d2ec52aSSatish Balay   overlap = overlap == PETSC_DETERMINE ? 1 : overlap;
1898*8d2ec52aSSatish Balay   PetscCheck(overlap >= 1, comm, PETSC_ERR_SUP, "Overlap %" PetscInt_FMT " must be greater than 0", overlap);
1899*8d2ec52aSSatish Balay   if (size > 1) {
1900*8d2ec52aSSatish Balay     PetscInt Npr = 1;
1901*8d2ec52aSSatish Balay 
1902*8d2ec52aSSatish Balay     // Make process grid
1903*8d2ec52aSSatish Balay     if (debug) PetscCall(PetscPrintf(comm, "Process grid:"));
1904*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
1905*8d2ec52aSSatish Balay       procs[d] = PetscRintReal(PetscPowReal(size, 1. / dim));
1906*8d2ec52aSSatish Balay       Npr *= procs[d];
1907*8d2ec52aSSatish Balay       if (debug) PetscCall(PetscPrintf(comm, " %" PetscInt_FMT, procs[d]));
1908*8d2ec52aSSatish Balay     }
1909*8d2ec52aSSatish Balay     if (debug) PetscCall(PetscPrintf(comm, "\n"));
1910*8d2ec52aSSatish Balay     PetscCheck(Npr == size, comm, PETSC_ERR_PLIB, "Process grid size %" PetscInt_FMT " != %d comm size", Npr, size);
1911*8d2ec52aSSatish Balay     IndexToTuple_Private(dim, procs, rank, lrank);
1912*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
1913*8d2ec52aSSatish Balay       ledges[d] = edges[d] / procs[d] + (edges[d] % procs[d] > lrank[d] ? 1 : 0);
1914*8d2ec52aSSatish Balay       vstart[d] = 0;
1915*8d2ec52aSSatish Balay       for (PetscInt r = 0; r < lrank[d]; ++r) vstart[d] += edges[d] / procs[d] + (edges[d] % procs[d] > r ? 1 : 0);
1916*8d2ec52aSSatish Balay       vstart[d] -= overlap; // For halo
1917*8d2ec52aSSatish Balay     }
1918*8d2ec52aSSatish Balay   } else {
1919*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
1920*8d2ec52aSSatish Balay       procs[d]  = 1;
1921*8d2ec52aSSatish Balay       ledges[d] = edges[d];
1922*8d2ec52aSSatish Balay     }
1923*8d2ec52aSSatish Balay   }
1924*8d2ec52aSSatish Balay   // Calculate local patch size
1925*8d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) {
1926*8d2ec52aSSatish Balay     vertices[d] = ledges[d] + (procs[d] > 1 ? 2 * overlap : 0);
1927cfb853baSMatthew G. Knepley     numVertices *= vertices[d];
1928cfb853baSMatthew G. Knepley   }
1929cfb853baSMatthew G. Knepley   numCells = numVertices * dim;
1930cfb853baSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
1931*8d2ec52aSSatish Balay   for (PetscInt c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, 2));
1932*8d2ec52aSSatish Balay   for (PetscInt v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetSupportSize(dm, v, 2 * dim));
1933cfb853baSMatthew G. Knepley   PetscCall(DMSetUp(dm)); /* Allocate space for cones and supports */
1934cfb853baSMatthew G. Knepley   /* Build cell cones and vertex supports */
1935cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "celltype"));
1936*8d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Topology for rank %d:\n", rank));
1937cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
1938cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert) + numCells;
1939cfb853baSMatthew G. Knepley     PetscInt       s      = 0;
1940*8d2ec52aSSatish Balay     PetscBool      leaf   = PETSC_FALSE;
1941cfb853baSMatthew G. Knepley 
1942*8d2ec52aSSatish Balay     if (debug) {
1943*8d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "Vertex %" PetscInt_FMT ":", vertex));
1944*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vert[d]));
1945*8d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "\n"));
1946*8d2ec52aSSatish Balay     }
1947cfb853baSMatthew G. Knepley     PetscCall(DMPlexSetCellType(dm, vertex, DM_POLYTOPE_POINT));
1948*8d2ec52aSSatish Balay     // Define edge cones
1949*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
1950*8d2ec52aSSatish Balay       for (PetscInt e = 0; e < dim; ++e) vtmp[e] = vert[e];
1951cfb853baSMatthew G. Knepley       vtmp[d] = (vert[d] + 1) % vertices[d];
1952cfb853baSMatthew G. Knepley       cone[0] = vertex;
1953cfb853baSMatthew G. Knepley       cone[1] = TupleToIndex_Private(dim, vertices, vtmp) + numCells;
1954*8d2ec52aSSatish Balay       if (debug) {
1955*8d2ec52aSSatish Balay         PetscCall(PetscSynchronizedPrintf(comm, "  Vertex %" PetscInt_FMT ":", cone[1]));
1956*8d2ec52aSSatish Balay         for (PetscInt e = 0; e < dim; ++e) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vtmp[e]));
1957*8d2ec52aSSatish Balay         PetscCall(PetscSynchronizedPrintf(comm, "\n"));
1958*8d2ec52aSSatish Balay       }
1959cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, cell, cone));
1960cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCellType(dm, cell, DM_POLYTOPE_SEGMENT));
1961*8d2ec52aSSatish Balay       if (debug) PetscCall(PetscSynchronizedPrintf(comm, "  Edge %" PetscInt_FMT " (%" PetscInt_FMT " %" PetscInt_FMT ")\n", cell, cone[0], cone[1]));
1962cfb853baSMatthew G. Knepley       ++cell;
1963*8d2ec52aSSatish Balay       // Shared vertices are any in the first or last overlap layers
1964*8d2ec52aSSatish Balay       if (vert[d] < overlap || vert[d] >= vertices[d] - overlap) leaf = PETSC_TRUE;
1965cfb853baSMatthew G. Knepley     }
1966*8d2ec52aSSatish Balay     if (size > 1 && leaf) ++Nl;
1967*8d2ec52aSSatish Balay     // Define vertex supports
1968*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
1969*8d2ec52aSSatish Balay       for (PetscInt e = 0; e < dim; ++e) vtmp[e] = vert[e];
1970cfb853baSMatthew G. Knepley       vtmp[d]   = (vert[d] + vertices[d] - 1) % vertices[d];
1971cfb853baSMatthew G. Knepley       supp[s++] = TupleToIndex_Private(dim, vertices, vtmp) * dim + d;
1972cfb853baSMatthew G. Knepley       supp[s++] = (vertex - numCells) * dim + d;
1973cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetSupport(dm, vertex, supp));
1974cfb853baSMatthew G. Knepley     }
1975cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
1976cfb853baSMatthew G. Knepley   }
1977*8d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedFlush(comm, NULL));
1978cfb853baSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
1979*8d2ec52aSSatish Balay   // Allocate for SF
1980*8d2ec52aSSatish Balay   PetscCall(PetscMalloc1(Nl, &leaves));
1981*8d2ec52aSSatish Balay   PetscCall(PetscMalloc1(Nl, &remotes));
1982*8d2ec52aSSatish Balay   // Build coordinates
1983cfb853baSMatthew G. Knepley   PetscCall(DMGetCoordinateSection(dm, &coordSection));
1984cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetNumFields(coordSection, 1));
1985cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
1986cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
1987*8d2ec52aSSatish Balay   for (PetscInt v = numCells; v < numCells + numVertices; ++v) {
1988cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetDof(coordSection, v, dim));
1989cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
1990cfb853baSMatthew G. Knepley   }
1991cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetUp(coordSection));
1992cfb853baSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
1993*8d2ec52aSSatish Balay   PetscCall(VecCreate(comm, &coordinates));
1994cfb853baSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
1995cfb853baSMatthew G. Knepley   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
1996cfb853baSMatthew G. Knepley   PetscCall(VecSetBlockSize(coordinates, dim));
1997cfb853baSMatthew G. Knepley   PetscCall(VecSetType(coordinates, VECSTANDARD));
1998cfb853baSMatthew G. Knepley   PetscCall(VecGetArray(coordinates, &coords));
1999*8d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Geometry for rank %d:\n", rank));
2000*8d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) vert[d] = 0;
2001cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
2002cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert);
2003*8d2ec52aSSatish Balay     PetscBool      leaf   = PETSC_FALSE;
2004cfb853baSMatthew G. Knepley 
2005*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
2006*8d2ec52aSSatish Balay       coords[vertex * dim + d] = lower[d] + ((upper[d] - lower[d]) / edges[d]) * (vert[d] + vstart[d]);
2007*8d2ec52aSSatish Balay       if (vert[d] < overlap || vert[d] >= vertices[d] - overlap) leaf = PETSC_TRUE;
2008*8d2ec52aSSatish Balay     }
2009*8d2ec52aSSatish Balay     if (size > 1 && leaf) {
2010*8d2ec52aSSatish Balay       PetscInt rnumCells = 1;
2011*8d2ec52aSSatish Balay 
2012*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) rvert[d] = vert[d] + vstart[d];
2013*8d2ec52aSSatish Balay       TupleToRanks_Private(dim, edges, procs, rvert, rrank);
2014*8d2ec52aSSatish Balay       leaves[Nl2]       = vertex + numCells;
2015*8d2ec52aSSatish Balay       remotes[Nl2].rank = TupleToIndex_Private(dim, procs, rrank);
2016*8d2ec52aSSatish Balay       RanksToSizes_Private(dim, edges, procs, rrank, rvertices);
2017*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) {
2018*8d2ec52aSSatish Balay         rvertices[d] += 2 * overlap; // Add halo
2019*8d2ec52aSSatish Balay         rnumCells *= rvertices[d];
2020*8d2ec52aSSatish Balay       }
2021*8d2ec52aSSatish Balay       rnumCells *= dim;
2022*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) {
2023*8d2ec52aSSatish Balay         const PetscInt diff = rrank[d] - lrank[d];
2024*8d2ec52aSSatish Balay 
2025*8d2ec52aSSatish Balay         if (!diff) rvert[d] = vert[d];                                     // Vertex is local
2026*8d2ec52aSSatish Balay         else if (rvert[d] < 0) rvert[d] = rvertices[d] - 1 + rvert[d];     // Wrap around at the bottom
2027*8d2ec52aSSatish Balay         else if (rvert[d] >= edges[d]) rvert[d] = rvert[d] - edges[d] + 1; // Wrap around at the top
2028*8d2ec52aSSatish Balay         else if (diff == -1) rvert[d] = rvertices[d] - 1 + (vert[d] - overlap);
2029*8d2ec52aSSatish Balay         else if (diff == 1) rvert[d] = (vertices[d] - vert[d] - 1) + overlap;
2030*8d2ec52aSSatish Balay         else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Process distance %" PetscInt_FMT " in direction %" PetscInt_FMT " should not be possible", diff, d);
2031*8d2ec52aSSatish Balay       }
2032*8d2ec52aSSatish Balay       remotes[Nl2].index = TupleToIndex_Private(dim, rvertices, rvert) + rnumCells;
2033*8d2ec52aSSatish Balay       if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Shared Vertex %" PetscInt_FMT " (%" PetscInt_FMT ", %" PetscInt_FMT ")\n", leaves[Nl2], remotes[Nl2].rank, remotes[Nl2].index));
2034*8d2ec52aSSatish Balay       ++Nl2;
2035*8d2ec52aSSatish Balay     }
2036*8d2ec52aSSatish Balay     if (debug) {
2037*8d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "Vertex %" PetscInt_FMT ":", vertex));
2038*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vert[d] + vstart[d]));
2039*8d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %g", (double)PetscRealPart(coords[vertex * dim + d])));
2040*8d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "\n"));
2041*8d2ec52aSSatish Balay     }
2042cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
2043cfb853baSMatthew G. Knepley   }
2044*8d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedFlush(comm, NULL));
2045cfb853baSMatthew G. Knepley   PetscCall(VecRestoreArray(coordinates, &coords));
2046cfb853baSMatthew G. Knepley   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
2047cfb853baSMatthew G. Knepley   PetscCall(VecDestroy(&coordinates));
2048*8d2ec52aSSatish Balay   // Build SF
2049*8d2ec52aSSatish Balay   PetscCheck(Nl == Nl2, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Initial number of leaves %" PetscInt_FMT " != %" PetscInt_FMT " final number", Nl, Nl2);
2050*8d2ec52aSSatish Balay   PetscCall(DMGetPointSF(dm, &sf));
2051*8d2ec52aSSatish Balay   PetscCall(PetscSFSetGraph(sf, numCells + numVertices, Nl, leaves, PETSC_OWN_POINTER, remotes, PETSC_OWN_POINTER));
2052*8d2ec52aSSatish Balay   if (debug) PetscCall(PetscSFView(sf, PETSC_VIEWER_STDOUT_WORLD));
2053cfb853baSMatthew G. Knepley   //PetscCall(DMSetPeriodicity(dm, NULL, lower, upper));
2054cfb853baSMatthew G. Knepley   // Attach the extent
2055cfb853baSMatthew G. Knepley   {
2056cfb853baSMatthew G. Knepley     PetscContainer c;
2057*8d2ec52aSSatish Balay     PetscInt      *extent, *lextent;
2058cfb853baSMatthew G. Knepley 
2059cfb853baSMatthew G. Knepley     PetscCall(PetscMalloc1(dim, &extent));
2060*8d2ec52aSSatish Balay     PetscCall(PetscMalloc1(dim, &lextent));
2061*8d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
2062*8d2ec52aSSatish Balay       extent[d]  = edges[d];
2063*8d2ec52aSSatish Balay       lextent[d] = ledges[d];
2064*8d2ec52aSSatish Balay     }
2065cfb853baSMatthew G. Knepley     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
206649abdd8aSBarry Smith     PetscCall(PetscContainerSetCtxDestroy(c, PetscCtxDestroyDefault));
2067cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetPointer(c, extent));
2068cfb853baSMatthew G. Knepley     PetscCall(PetscObjectCompose((PetscObject)dm, "_extent", (PetscObject)c));
2069cfb853baSMatthew G. Knepley     PetscCall(PetscContainerDestroy(&c));
2070*8d2ec52aSSatish Balay     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
2071*8d2ec52aSSatish Balay     PetscCall(PetscContainerSetCtxDestroy(c, PetscCtxDestroyDefault));
2072*8d2ec52aSSatish Balay     PetscCall(PetscContainerSetPointer(c, lextent));
2073*8d2ec52aSSatish Balay     PetscCall(PetscObjectCompose((PetscObject)dm, "_lextent", (PetscObject)c));
2074*8d2ec52aSSatish Balay     PetscCall(PetscContainerDestroy(&c));
2075cfb853baSMatthew G. Knepley   }
2076*8d2ec52aSSatish Balay   PetscCall(PetscFree4(procs, lrank, rrank, supp));
2077*8d2ec52aSSatish Balay   PetscCall(PetscFree7(ledges, vertices, rvertices, vert, rvert, vstart, vtmp));
20783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2079cfb853baSMatthew G. Knepley }
2080cfb853baSMatthew G. Knepley 
2081cfb853baSMatthew G. Knepley /*@C
2082aaa8cc7dSPierre Jolivet   DMPlexCreateHypercubicMesh - Creates a periodic mesh on the tensor product of unit intervals using only vertices and edges.
2083cfb853baSMatthew G. Knepley 
2084cfb853baSMatthew G. Knepley   Collective
2085cfb853baSMatthew G. Knepley 
2086cfb853baSMatthew G. Knepley   Input Parameters:
2087*8d2ec52aSSatish Balay + comm    - The communicator for the `DM` object
2088cfb853baSMatthew G. Knepley . dim     - The spatial dimension
208920f4b53cSBarry Smith . edges   - Number of edges per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
209020f4b53cSBarry Smith . lower   - The lower left corner, or `NULL` for (0, 0, 0)
2091*8d2ec52aSSatish Balay . upper   - The upper right corner, or `NULL` for (1, 1, 1)
2092*8d2ec52aSSatish Balay - overlap - The number of vertices in each direction to include in the overlap (default is 1)
2093cfb853baSMatthew G. Knepley 
2094cfb853baSMatthew G. Knepley   Output Parameter:
2095cfb853baSMatthew G. Knepley . dm - The DM object
2096cfb853baSMatthew G. Knepley 
209720f4b53cSBarry Smith   Level: beginner
209820f4b53cSBarry Smith 
209920f4b53cSBarry Smith   Note:
210020f4b53cSBarry Smith   If you want to customize this mesh using options, you just need to
210120f4b53cSBarry Smith .vb
210220f4b53cSBarry Smith   DMCreate(comm, &dm);
210320f4b53cSBarry Smith   DMSetType(dm, DMPLEX);
210420f4b53cSBarry Smith   DMSetFromOptions(dm);
210520f4b53cSBarry Smith .ve
210620f4b53cSBarry Smith   and use the options on the `DMSetFromOptions()` page.
2107cfb853baSMatthew G. Knepley 
2108cfb853baSMatthew G. Knepley   The vertices are numbered is lexicographic order, and the dim edges exiting a vertex in the positive orthant are number consecutively,
210920f4b53cSBarry Smith .vb
211020f4b53cSBarry Smith  18--0-19--2-20--4-18
211120f4b53cSBarry Smith   |     |     |     |
211220f4b53cSBarry Smith  13    15    17    13
211320f4b53cSBarry Smith   |     |     |     |
211420f4b53cSBarry Smith  24-12-25-14-26-16-24
211520f4b53cSBarry Smith   |     |     |     |
211620f4b53cSBarry Smith   7     9    11     7
211720f4b53cSBarry Smith   |     |     |     |
211820f4b53cSBarry Smith  21--6-22--8-23-10-21
211920f4b53cSBarry Smith   |     |     |     |
212020f4b53cSBarry Smith   1     3     5     1
212120f4b53cSBarry Smith   |     |     |     |
212220f4b53cSBarry Smith  18--0-19--2-20--4-18
212320f4b53cSBarry Smith .ve
2124cfb853baSMatthew G. Knepley 
212576fbde31SPierre Jolivet .seealso: `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
2126cfb853baSMatthew G. Knepley @*/
2127*8d2ec52aSSatish Balay PetscErrorCode DMPlexCreateHypercubicMesh(MPI_Comm comm, PetscInt dim, const PetscInt edges[], const PetscReal lower[], const PetscReal upper[], PetscInt overlap, DM *dm)
2128cfb853baSMatthew G. Knepley {
2129cfb853baSMatthew G. Knepley   PetscInt       *edg;
2130cfb853baSMatthew G. Knepley   PetscReal      *low, *upp;
2131cfb853baSMatthew G. Knepley   DMBoundaryType *bdt;
2132cfb853baSMatthew G. Knepley   PetscInt        d;
2133cfb853baSMatthew G. Knepley 
2134cfb853baSMatthew G. Knepley   PetscFunctionBegin;
2135cfb853baSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
2136cfb853baSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
2137cfb853baSMatthew G. Knepley   PetscCall(PetscMalloc4(dim, &edg, dim, &low, dim, &upp, dim, &bdt));
2138cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
2139cfb853baSMatthew G. Knepley     edg[d] = edges ? edges[d] : 1;
2140cfb853baSMatthew G. Knepley     low[d] = lower ? lower[d] : 0.;
2141cfb853baSMatthew G. Knepley     upp[d] = upper ? upper[d] : 1.;
2142cfb853baSMatthew G. Knepley     bdt[d] = DM_BOUNDARY_PERIODIC;
2143cfb853baSMatthew G. Knepley   }
2144*8d2ec52aSSatish Balay   PetscCall(DMPlexCreateHypercubicMesh_Internal(*dm, dim, low, upp, edg, overlap, bdt));
2145cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(edg, low, upp, bdt));
21463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2147cfb853baSMatthew G. Knepley }
2148cfb853baSMatthew G. Knepley 
2149cc4c1da9SBarry Smith /*@
2150a1cb98faSBarry Smith   DMPlexSetOptionsPrefix - Sets the prefix used for searching for all `DM` options in the database.
2151a9074c1eSMatthew G. Knepley 
215220f4b53cSBarry Smith   Logically Collective
2153a9074c1eSMatthew G. Knepley 
2154a9074c1eSMatthew G. Knepley   Input Parameters:
215520f4b53cSBarry Smith + dm     - the `DM` context
2156a9074c1eSMatthew G. Knepley - prefix - the prefix to prepend to all option names
2157a9074c1eSMatthew G. Knepley 
2158a1cb98faSBarry Smith   Level: advanced
2159a1cb98faSBarry Smith 
2160a1cb98faSBarry Smith   Note:
2161a9074c1eSMatthew G. Knepley   A hyphen (-) must NOT be given at the beginning of the prefix name.
2162a9074c1eSMatthew G. Knepley   The first character of all runtime options is AUTOMATICALLY the hyphen.
2163a9074c1eSMatthew G. Knepley 
21641cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `SNESSetFromOptions()`
2165a9074c1eSMatthew G. Knepley @*/
2166d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexSetOptionsPrefix(DM dm, const char prefix[])
2167d71ae5a4SJacob Faibussowitsch {
2168a9074c1eSMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
2169a9074c1eSMatthew G. Knepley 
2170a9074c1eSMatthew G. Knepley   PetscFunctionBegin;
2171a9074c1eSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
21729566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, prefix));
21739566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)mesh->partitioner, prefix));
21743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2175a9074c1eSMatthew G. Knepley }
2176a9074c1eSMatthew G. Knepley 
21779318fe57SMatthew G. Knepley /* Remap geometry to cylinder
217861a622f3SMatthew G. Knepley    TODO: This only works for a single refinement, then it is broken
217961a622f3SMatthew G. Knepley 
21809318fe57SMatthew G. Knepley      Interior square: Linear interpolation is correct
21819318fe57SMatthew G. Knepley      The other cells all have vertices on rays from the origin. We want to uniformly expand the spacing
21829318fe57SMatthew G. Knepley      such that the last vertex is on the unit circle. So the closest and farthest vertices are at distance
21830510c589SMatthew G. Knepley 
21849318fe57SMatthew G. Knepley        phi     = arctan(y/x)
21859318fe57SMatthew G. Knepley        d_close = sqrt(1/8 + 1/4 sin^2(phi))
21869318fe57SMatthew G. Knepley        d_far   = sqrt(1/2 + sin^2(phi))
21870510c589SMatthew G. Knepley 
21889318fe57SMatthew G. Knepley      so we remap them using
21890510c589SMatthew G. Knepley 
21909318fe57SMatthew G. Knepley        x_new = x_close + (x - x_close) (1 - d_close) / (d_far - d_close)
21919318fe57SMatthew G. Knepley        y_new = y_close + (y - y_close) (1 - d_close) / (d_far - d_close)
21920510c589SMatthew G. Knepley 
21939318fe57SMatthew G. Knepley      If pi/4 < phi < 3pi/4 or -3pi/4 < phi < -pi/4, then we switch x and y.
21949318fe57SMatthew G. Knepley */
2195d71ae5a4SJacob 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[])
2196d71ae5a4SJacob Faibussowitsch {
21979318fe57SMatthew G. Knepley   const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
21989318fe57SMatthew G. Knepley   const PetscReal ds2 = 0.5 * dis;
219922cc497dSMatthew G. Knepley 
22009318fe57SMatthew G. Knepley   if ((PetscAbsScalar(u[0]) <= ds2) && (PetscAbsScalar(u[1]) <= ds2)) {
22019318fe57SMatthew G. Knepley     f0[0] = u[0];
22029318fe57SMatthew G. Knepley     f0[1] = u[1];
22039318fe57SMatthew G. Knepley   } else {
22049318fe57SMatthew G. Knepley     PetscReal phi, sinp, cosp, dc, df, x, y, xc, yc;
22050510c589SMatthew G. Knepley 
22069318fe57SMatthew G. Knepley     x    = PetscRealPart(u[0]);
22079318fe57SMatthew G. Knepley     y    = PetscRealPart(u[1]);
22089318fe57SMatthew G. Knepley     phi  = PetscAtan2Real(y, x);
22099318fe57SMatthew G. Knepley     sinp = PetscSinReal(phi);
22109318fe57SMatthew G. Knepley     cosp = PetscCosReal(phi);
22119318fe57SMatthew G. Knepley     if ((PetscAbsReal(phi) > PETSC_PI / 4.0) && (PetscAbsReal(phi) < 3.0 * PETSC_PI / 4.0)) {
22129318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / sinp);
22139318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / sinp);
22149318fe57SMatthew G. Knepley       xc = ds2 * x / PetscAbsReal(y);
22159318fe57SMatthew G. Knepley       yc = ds2 * PetscSignReal(y);
22169318fe57SMatthew G. Knepley     } else {
22179318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / cosp);
22189318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / cosp);
22199318fe57SMatthew G. Knepley       xc = ds2 * PetscSignReal(x);
22209318fe57SMatthew G. Knepley       yc = ds2 * y / PetscAbsReal(x);
22219318fe57SMatthew G. Knepley     }
22229318fe57SMatthew G. Knepley     f0[0] = xc + (u[0] - xc) * (1.0 - dc) / (df - dc);
22239318fe57SMatthew G. Knepley     f0[1] = yc + (u[1] - yc) * (1.0 - dc) / (df - dc);
22249318fe57SMatthew G. Knepley   }
22259318fe57SMatthew G. Knepley   f0[2] = u[2];
22269318fe57SMatthew G. Knepley }
22270510c589SMatthew G. Knepley 
222849704ca5SMatthew G. Knepley static PetscErrorCode DMPlexCreateHexCylinderMesh_Internal(DM dm, DMBoundaryType periodicZ, PetscInt Nr)
2229d71ae5a4SJacob Faibussowitsch {
22300510c589SMatthew G. Knepley   const PetscInt dim = 3;
22319318fe57SMatthew G. Knepley   PetscInt       numCells, numVertices;
2232d8c47e87SMatthew G. Knepley   PetscMPIInt    rank;
22330510c589SMatthew G. Knepley 
22340510c589SMatthew G. Knepley   PetscFunctionBegin;
223546139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
22369566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
22379566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
22380510c589SMatthew G. Knepley   /* Create topology */
22390510c589SMatthew G. Knepley   {
22400510c589SMatthew G. Knepley     PetscInt cone[8], c;
22410510c589SMatthew G. Knepley 
2242dd400576SPatrick Sanan     numCells    = rank == 0 ? 5 : 0;
2243dd400576SPatrick Sanan     numVertices = rank == 0 ? 16 : 0;
2244006a8963SMatthew G. Knepley     if (periodicZ == DM_BOUNDARY_PERIODIC) {
2245ae8bcbbbSMatthew G. Knepley       numCells *= 3;
2246dd400576SPatrick Sanan       numVertices = rank == 0 ? 24 : 0;
2247006a8963SMatthew G. Knepley     }
22489566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
22499566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 8));
22509566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
2251dd400576SPatrick Sanan     if (rank == 0) {
2252006a8963SMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
22539371c9d4SSatish Balay         cone[0] = 15;
22549371c9d4SSatish Balay         cone[1] = 18;
22559371c9d4SSatish Balay         cone[2] = 17;
22569371c9d4SSatish Balay         cone[3] = 16;
22579371c9d4SSatish Balay         cone[4] = 31;
22589371c9d4SSatish Balay         cone[5] = 32;
22599371c9d4SSatish Balay         cone[6] = 33;
22609371c9d4SSatish Balay         cone[7] = 34;
22619566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
22629371c9d4SSatish Balay         cone[0] = 16;
22639371c9d4SSatish Balay         cone[1] = 17;
22649371c9d4SSatish Balay         cone[2] = 24;
22659371c9d4SSatish Balay         cone[3] = 23;
22669371c9d4SSatish Balay         cone[4] = 32;
22679371c9d4SSatish Balay         cone[5] = 36;
22689371c9d4SSatish Balay         cone[6] = 37;
22699371c9d4SSatish Balay         cone[7] = 33; /* 22 25 26 21 */
22709566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
22719371c9d4SSatish Balay         cone[0] = 18;
22729371c9d4SSatish Balay         cone[1] = 27;
22739371c9d4SSatish Balay         cone[2] = 24;
22749371c9d4SSatish Balay         cone[3] = 17;
22759371c9d4SSatish Balay         cone[4] = 34;
22769371c9d4SSatish Balay         cone[5] = 33;
22779371c9d4SSatish Balay         cone[6] = 37;
22789371c9d4SSatish Balay         cone[7] = 38;
22799566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
22809371c9d4SSatish Balay         cone[0] = 29;
22819371c9d4SSatish Balay         cone[1] = 27;
22829371c9d4SSatish Balay         cone[2] = 18;
22839371c9d4SSatish Balay         cone[3] = 15;
22849371c9d4SSatish Balay         cone[4] = 35;
22859371c9d4SSatish Balay         cone[5] = 31;
22869371c9d4SSatish Balay         cone[6] = 34;
22879371c9d4SSatish Balay         cone[7] = 38;
22889566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
22899371c9d4SSatish Balay         cone[0] = 29;
22909371c9d4SSatish Balay         cone[1] = 15;
22919371c9d4SSatish Balay         cone[2] = 16;
22929371c9d4SSatish Balay         cone[3] = 23;
22939371c9d4SSatish Balay         cone[4] = 35;
22949371c9d4SSatish Balay         cone[5] = 36;
22959371c9d4SSatish Balay         cone[6] = 32;
22969371c9d4SSatish Balay         cone[7] = 31;
22979566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2298006a8963SMatthew G. Knepley 
22999371c9d4SSatish Balay         cone[0] = 31;
23009371c9d4SSatish Balay         cone[1] = 34;
23019371c9d4SSatish Balay         cone[2] = 33;
23029371c9d4SSatish Balay         cone[3] = 32;
23039371c9d4SSatish Balay         cone[4] = 19;
23049371c9d4SSatish Balay         cone[5] = 22;
23059371c9d4SSatish Balay         cone[6] = 21;
23069371c9d4SSatish Balay         cone[7] = 20;
23079566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
23089371c9d4SSatish Balay         cone[0] = 32;
23099371c9d4SSatish Balay         cone[1] = 33;
23109371c9d4SSatish Balay         cone[2] = 37;
23119371c9d4SSatish Balay         cone[3] = 36;
23129371c9d4SSatish Balay         cone[4] = 22;
23139371c9d4SSatish Balay         cone[5] = 25;
23149371c9d4SSatish Balay         cone[6] = 26;
23159371c9d4SSatish Balay         cone[7] = 21;
23169566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 6, cone));
23179371c9d4SSatish Balay         cone[0] = 34;
23189371c9d4SSatish Balay         cone[1] = 38;
23199371c9d4SSatish Balay         cone[2] = 37;
23209371c9d4SSatish Balay         cone[3] = 33;
23219371c9d4SSatish Balay         cone[4] = 20;
23229371c9d4SSatish Balay         cone[5] = 21;
23239371c9d4SSatish Balay         cone[6] = 26;
23249371c9d4SSatish Balay         cone[7] = 28;
23259566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 7, cone));
23269371c9d4SSatish Balay         cone[0] = 35;
23279371c9d4SSatish Balay         cone[1] = 38;
23289371c9d4SSatish Balay         cone[2] = 34;
23299371c9d4SSatish Balay         cone[3] = 31;
23309371c9d4SSatish Balay         cone[4] = 30;
23319371c9d4SSatish Balay         cone[5] = 19;
23329371c9d4SSatish Balay         cone[6] = 20;
23339371c9d4SSatish Balay         cone[7] = 28;
23349566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 8, cone));
23359371c9d4SSatish Balay         cone[0] = 35;
23369371c9d4SSatish Balay         cone[1] = 31;
23379371c9d4SSatish Balay         cone[2] = 32;
23389371c9d4SSatish Balay         cone[3] = 36;
23399371c9d4SSatish Balay         cone[4] = 30;
23409371c9d4SSatish Balay         cone[5] = 25;
23419371c9d4SSatish Balay         cone[6] = 22;
23429371c9d4SSatish Balay         cone[7] = 19;
23439566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 9, cone));
2344ae8bcbbbSMatthew G. Knepley 
23459371c9d4SSatish Balay         cone[0] = 19;
23469371c9d4SSatish Balay         cone[1] = 20;
23479371c9d4SSatish Balay         cone[2] = 21;
23489371c9d4SSatish Balay         cone[3] = 22;
23499371c9d4SSatish Balay         cone[4] = 15;
23509371c9d4SSatish Balay         cone[5] = 16;
23519371c9d4SSatish Balay         cone[6] = 17;
23529371c9d4SSatish Balay         cone[7] = 18;
23539566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 10, cone));
23549371c9d4SSatish Balay         cone[0] = 22;
23559371c9d4SSatish Balay         cone[1] = 21;
23569371c9d4SSatish Balay         cone[2] = 26;
23579371c9d4SSatish Balay         cone[3] = 25;
23589371c9d4SSatish Balay         cone[4] = 16;
23599371c9d4SSatish Balay         cone[5] = 23;
23609371c9d4SSatish Balay         cone[6] = 24;
23619371c9d4SSatish Balay         cone[7] = 17;
23629566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 11, cone));
23639371c9d4SSatish Balay         cone[0] = 20;
23649371c9d4SSatish Balay         cone[1] = 28;
23659371c9d4SSatish Balay         cone[2] = 26;
23669371c9d4SSatish Balay         cone[3] = 21;
23679371c9d4SSatish Balay         cone[4] = 18;
23689371c9d4SSatish Balay         cone[5] = 17;
23699371c9d4SSatish Balay         cone[6] = 24;
23709371c9d4SSatish Balay         cone[7] = 27;
23719566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 12, cone));
23729371c9d4SSatish Balay         cone[0] = 30;
23739371c9d4SSatish Balay         cone[1] = 28;
23749371c9d4SSatish Balay         cone[2] = 20;
23759371c9d4SSatish Balay         cone[3] = 19;
23769371c9d4SSatish Balay         cone[4] = 29;
23779371c9d4SSatish Balay         cone[5] = 15;
23789371c9d4SSatish Balay         cone[6] = 18;
23799371c9d4SSatish Balay         cone[7] = 27;
23809566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 13, cone));
23819371c9d4SSatish Balay         cone[0] = 30;
23829371c9d4SSatish Balay         cone[1] = 19;
23839371c9d4SSatish Balay         cone[2] = 22;
23849371c9d4SSatish Balay         cone[3] = 25;
23859371c9d4SSatish Balay         cone[4] = 29;
23869371c9d4SSatish Balay         cone[5] = 23;
23879371c9d4SSatish Balay         cone[6] = 16;
23889371c9d4SSatish Balay         cone[7] = 15;
23899566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
2390006a8963SMatthew G. Knepley       } else {
23919371c9d4SSatish Balay         cone[0] = 5;
23929371c9d4SSatish Balay         cone[1] = 8;
23939371c9d4SSatish Balay         cone[2] = 7;
23949371c9d4SSatish Balay         cone[3] = 6;
23959371c9d4SSatish Balay         cone[4] = 9;
23969371c9d4SSatish Balay         cone[5] = 12;
23979371c9d4SSatish Balay         cone[6] = 11;
23989371c9d4SSatish Balay         cone[7] = 10;
23999566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
24009371c9d4SSatish Balay         cone[0] = 6;
24019371c9d4SSatish Balay         cone[1] = 7;
24029371c9d4SSatish Balay         cone[2] = 14;
24039371c9d4SSatish Balay         cone[3] = 13;
24049371c9d4SSatish Balay         cone[4] = 12;
24059371c9d4SSatish Balay         cone[5] = 15;
24069371c9d4SSatish Balay         cone[6] = 16;
24079371c9d4SSatish Balay         cone[7] = 11;
24089566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
24099371c9d4SSatish Balay         cone[0] = 8;
24109371c9d4SSatish Balay         cone[1] = 17;
24119371c9d4SSatish Balay         cone[2] = 14;
24129371c9d4SSatish Balay         cone[3] = 7;
24139371c9d4SSatish Balay         cone[4] = 10;
24149371c9d4SSatish Balay         cone[5] = 11;
24159371c9d4SSatish Balay         cone[6] = 16;
24169371c9d4SSatish Balay         cone[7] = 18;
24179566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
24189371c9d4SSatish Balay         cone[0] = 19;
24199371c9d4SSatish Balay         cone[1] = 17;
24209371c9d4SSatish Balay         cone[2] = 8;
24219371c9d4SSatish Balay         cone[3] = 5;
24229371c9d4SSatish Balay         cone[4] = 20;
24239371c9d4SSatish Balay         cone[5] = 9;
24249371c9d4SSatish Balay         cone[6] = 10;
24259371c9d4SSatish Balay         cone[7] = 18;
24269566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
24279371c9d4SSatish Balay         cone[0] = 19;
24289371c9d4SSatish Balay         cone[1] = 5;
24299371c9d4SSatish Balay         cone[2] = 6;
24309371c9d4SSatish Balay         cone[3] = 13;
24319371c9d4SSatish Balay         cone[4] = 20;
24329371c9d4SSatish Balay         cone[5] = 15;
24339371c9d4SSatish Balay         cone[6] = 12;
24349371c9d4SSatish Balay         cone[7] = 9;
24359566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2436006a8963SMatthew G. Knepley       }
2437d8c47e87SMatthew G. Knepley     }
24389566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
24399566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
24400510c589SMatthew G. Knepley   }
2441dbc1dc17SMatthew G. Knepley   /* Create cube geometry */
24420510c589SMatthew G. Knepley   {
24430510c589SMatthew G. Knepley     Vec             coordinates;
24440510c589SMatthew G. Knepley     PetscSection    coordSection;
24450510c589SMatthew G. Knepley     PetscScalar    *coords;
24460510c589SMatthew G. Knepley     PetscInt        coordSize, v;
24470510c589SMatthew G. Knepley     const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
24480510c589SMatthew G. Knepley     const PetscReal ds2 = dis / 2.0;
24490510c589SMatthew G. Knepley 
24500510c589SMatthew G. Knepley     /* Build coordinates */
24519566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
24529566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
24539566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
24549566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
24550510c589SMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
24569566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
24579566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
24580510c589SMatthew G. Knepley     }
24599566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
24609566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
24619566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
24629566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
24639566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
24649566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
24659566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
24669566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
2467dd400576SPatrick Sanan     if (rank == 0) {
24689371c9d4SSatish Balay       coords[0 * dim + 0]  = -ds2;
24699371c9d4SSatish Balay       coords[0 * dim + 1]  = -ds2;
24709371c9d4SSatish Balay       coords[0 * dim + 2]  = 0.0;
24719371c9d4SSatish Balay       coords[1 * dim + 0]  = ds2;
24729371c9d4SSatish Balay       coords[1 * dim + 1]  = -ds2;
24739371c9d4SSatish Balay       coords[1 * dim + 2]  = 0.0;
24749371c9d4SSatish Balay       coords[2 * dim + 0]  = ds2;
24759371c9d4SSatish Balay       coords[2 * dim + 1]  = ds2;
24769371c9d4SSatish Balay       coords[2 * dim + 2]  = 0.0;
24779371c9d4SSatish Balay       coords[3 * dim + 0]  = -ds2;
24789371c9d4SSatish Balay       coords[3 * dim + 1]  = ds2;
24799371c9d4SSatish Balay       coords[3 * dim + 2]  = 0.0;
24809371c9d4SSatish Balay       coords[4 * dim + 0]  = -ds2;
24819371c9d4SSatish Balay       coords[4 * dim + 1]  = -ds2;
24829371c9d4SSatish Balay       coords[4 * dim + 2]  = 1.0;
24839371c9d4SSatish Balay       coords[5 * dim + 0]  = -ds2;
24849371c9d4SSatish Balay       coords[5 * dim + 1]  = ds2;
24859371c9d4SSatish Balay       coords[5 * dim + 2]  = 1.0;
24869371c9d4SSatish Balay       coords[6 * dim + 0]  = ds2;
24879371c9d4SSatish Balay       coords[6 * dim + 1]  = ds2;
24889371c9d4SSatish Balay       coords[6 * dim + 2]  = 1.0;
24899371c9d4SSatish Balay       coords[7 * dim + 0]  = ds2;
24909371c9d4SSatish Balay       coords[7 * dim + 1]  = -ds2;
24919371c9d4SSatish Balay       coords[7 * dim + 2]  = 1.0;
24929371c9d4SSatish Balay       coords[8 * dim + 0]  = dis;
24939371c9d4SSatish Balay       coords[8 * dim + 1]  = -dis;
24949371c9d4SSatish Balay       coords[8 * dim + 2]  = 0.0;
24959371c9d4SSatish Balay       coords[9 * dim + 0]  = dis;
24969371c9d4SSatish Balay       coords[9 * dim + 1]  = dis;
24979371c9d4SSatish Balay       coords[9 * dim + 2]  = 0.0;
24989371c9d4SSatish Balay       coords[10 * dim + 0] = dis;
24999371c9d4SSatish Balay       coords[10 * dim + 1] = -dis;
25009371c9d4SSatish Balay       coords[10 * dim + 2] = 1.0;
25019371c9d4SSatish Balay       coords[11 * dim + 0] = dis;
25029371c9d4SSatish Balay       coords[11 * dim + 1] = dis;
25039371c9d4SSatish Balay       coords[11 * dim + 2] = 1.0;
25049371c9d4SSatish Balay       coords[12 * dim + 0] = -dis;
25059371c9d4SSatish Balay       coords[12 * dim + 1] = dis;
25069371c9d4SSatish Balay       coords[12 * dim + 2] = 0.0;
25079371c9d4SSatish Balay       coords[13 * dim + 0] = -dis;
25089371c9d4SSatish Balay       coords[13 * dim + 1] = dis;
25099371c9d4SSatish Balay       coords[13 * dim + 2] = 1.0;
25109371c9d4SSatish Balay       coords[14 * dim + 0] = -dis;
25119371c9d4SSatish Balay       coords[14 * dim + 1] = -dis;
25129371c9d4SSatish Balay       coords[14 * dim + 2] = 0.0;
25139371c9d4SSatish Balay       coords[15 * dim + 0] = -dis;
25149371c9d4SSatish Balay       coords[15 * dim + 1] = -dis;
25159371c9d4SSatish Balay       coords[15 * dim + 2] = 1.0;
2516ae8bcbbbSMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
25179371c9d4SSatish Balay         /* 15 31 19 */ coords[16 * dim + 0] = -ds2;
25189371c9d4SSatish Balay         coords[16 * dim + 1]                = -ds2;
25199371c9d4SSatish Balay         coords[16 * dim + 2]                = 0.5;
25209371c9d4SSatish Balay         /* 16 32 22 */ coords[17 * dim + 0] = ds2;
25219371c9d4SSatish Balay         coords[17 * dim + 1]                = -ds2;
25229371c9d4SSatish Balay         coords[17 * dim + 2]                = 0.5;
25239371c9d4SSatish Balay         /* 17 33 21 */ coords[18 * dim + 0] = ds2;
25249371c9d4SSatish Balay         coords[18 * dim + 1]                = ds2;
25259371c9d4SSatish Balay         coords[18 * dim + 2]                = 0.5;
25269371c9d4SSatish Balay         /* 18 34 20 */ coords[19 * dim + 0] = -ds2;
25279371c9d4SSatish Balay         coords[19 * dim + 1]                = ds2;
25289371c9d4SSatish Balay         coords[19 * dim + 2]                = 0.5;
25299371c9d4SSatish Balay         /* 29 35 30 */ coords[20 * dim + 0] = -dis;
25309371c9d4SSatish Balay         coords[20 * dim + 1]                = -dis;
25319371c9d4SSatish Balay         coords[20 * dim + 2]                = 0.5;
25329371c9d4SSatish Balay         /* 23 36 25 */ coords[21 * dim + 0] = dis;
25339371c9d4SSatish Balay         coords[21 * dim + 1]                = -dis;
25349371c9d4SSatish Balay         coords[21 * dim + 2]                = 0.5;
25359371c9d4SSatish Balay         /* 24 37 26 */ coords[22 * dim + 0] = dis;
25369371c9d4SSatish Balay         coords[22 * dim + 1]                = dis;
25379371c9d4SSatish Balay         coords[22 * dim + 2]                = 0.5;
25389371c9d4SSatish Balay         /* 27 38 28 */ coords[23 * dim + 0] = -dis;
25399371c9d4SSatish Balay         coords[23 * dim + 1]                = dis;
25409371c9d4SSatish Balay         coords[23 * dim + 2]                = 0.5;
2541ae8bcbbbSMatthew G. Knepley       }
2542d8c47e87SMatthew G. Knepley     }
25439566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
25449566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
25459566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
25460510c589SMatthew G. Knepley   }
2547006a8963SMatthew G. Knepley   /* Create periodicity */
2548006a8963SMatthew G. Knepley   if (periodicZ == DM_BOUNDARY_PERIODIC || periodicZ == DM_BOUNDARY_TWIST) {
25496858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
25506858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
2551006a8963SMatthew G. Knepley     PetscReal lower[3]   = {0.0, 0.0, 0.0};
2552ae8bcbbbSMatthew G. Knepley     PetscReal upper[3]   = {1.0, 1.0, 1.5};
25536858538eSMatthew G. Knepley     PetscInt  numZCells  = 3;
2554006a8963SMatthew G. Knepley 
25556858538eSMatthew G. Knepley     L[2]       = upper[2] - lower[2];
25566858538eSMatthew G. Knepley     maxCell[2] = 1.1 * (L[2] / numZCells);
25574fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
2558006a8963SMatthew G. Knepley   }
2559dbc1dc17SMatthew G. Knepley   {
25609318fe57SMatthew G. Knepley     DM          cdm;
25619318fe57SMatthew G. Knepley     PetscDS     cds;
25629318fe57SMatthew G. Knepley     PetscScalar c[2] = {1.0, 1.0};
2563dbc1dc17SMatthew G. Knepley 
256449704ca5SMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
25659566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
25669566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
25679566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 2, c));
2568dbc1dc17SMatthew G. Knepley   }
256946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
257046139095SJed Brown 
25719318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
25729566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(dm));
257349704ca5SMatthew G. Knepley 
257449704ca5SMatthew G. Knepley   char        oldprefix[PETSC_MAX_PATH_LEN];
257549704ca5SMatthew G. Knepley   const char *prefix;
257649704ca5SMatthew G. Knepley 
257749704ca5SMatthew G. Knepley   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
257849704ca5SMatthew G. Knepley   PetscCall(PetscStrncpy(oldprefix, prefix, PETSC_MAX_PATH_LEN));
257949704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, "petsc_cyl_ref_"));
258049704ca5SMatthew G. Knepley   for (PetscInt r = 0; r < PetscMax(0, Nr); ++r) {
258149704ca5SMatthew G. Knepley     DM rdm;
258249704ca5SMatthew G. Knepley 
258349704ca5SMatthew G. Knepley     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
258449704ca5SMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &rdm));
258549704ca5SMatthew G. Knepley   }
258649704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldprefix));
258749704ca5SMatthew G. Knepley   PetscCall(DMPlexRemapGeometry(dm, 0.0, snapToCylinder));
258849704ca5SMatthew G. Knepley 
258949704ca5SMatthew G. Knepley   DMLabel         bdlabel, edgelabel;
259049704ca5SMatthew G. Knepley   IS              faceIS;
259149704ca5SMatthew G. Knepley   const PetscInt *faces;
259249704ca5SMatthew G. Knepley   PetscInt        Nf;
259349704ca5SMatthew G. Knepley 
259449704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
259549704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
259649704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "generatrix"));
259749704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "generatrix", &edgelabel));
259849704ca5SMatthew G. Knepley   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
259949704ca5SMatthew G. Knepley   // Remove faces on top and bottom
260049704ca5SMatthew G. Knepley   PetscCall(DMLabelGetStratumIS(bdlabel, 1, &faceIS));
2601ba1b3593SJeremy L Thompson   if (faceIS) {
260249704ca5SMatthew G. Knepley     PetscCall(ISGetLocalSize(faceIS, &Nf));
260349704ca5SMatthew G. Knepley     PetscCall(ISGetIndices(faceIS, &faces));
260449704ca5SMatthew G. Knepley     for (PetscInt f = 0; f < Nf; ++f) {
260549704ca5SMatthew G. Knepley       PetscReal vol, normal[3];
260649704ca5SMatthew G. Knepley 
260749704ca5SMatthew G. Knepley       PetscCall(DMPlexComputeCellGeometryFVM(dm, faces[f], &vol, NULL, normal));
260849704ca5SMatthew G. Knepley       if (PetscAbsReal(normal[2]) < PETSC_SMALL) PetscCall(DMLabelSetValue(edgelabel, faces[f], 1));
260949704ca5SMatthew G. Knepley     }
261049704ca5SMatthew G. Knepley     PetscCall(ISRestoreIndices(faceIS, &faces));
261149704ca5SMatthew G. Knepley     PetscCall(ISDestroy(&faceIS));
2612ba1b3593SJeremy L Thompson   }
261349704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, bdlabel));
261449704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, edgelabel));
26153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
26160510c589SMatthew G. Knepley }
26170510c589SMatthew G. Knepley 
261824119c2aSMatthew G. Knepley /*@
26199318fe57SMatthew G. Knepley   DMPlexCreateHexCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using hexahedra.
262024119c2aSMatthew G. Knepley 
2621d083f849SBarry Smith   Collective
262224119c2aSMatthew G. Knepley 
262324119c2aSMatthew G. Knepley   Input Parameters:
2624a1cb98faSBarry Smith + comm      - The communicator for the `DM` object
262549704ca5SMatthew G. Knepley . periodicZ - The boundary type for the Z direction
262649704ca5SMatthew G. Knepley - Nr        - The number of refinements to carry out
262724119c2aSMatthew G. Knepley 
262824119c2aSMatthew G. Knepley   Output Parameter:
262920f4b53cSBarry Smith . dm - The `DM` object
263024119c2aSMatthew G. Knepley 
263124119c2aSMatthew G. Knepley   Level: beginner
263224119c2aSMatthew G. Knepley 
2633a1cb98faSBarry Smith   Note:
2634a4e35b19SJacob Faibussowitsch   Here is the output numbering looking from the bottom of the cylinder\:
2635a1cb98faSBarry Smith .vb
2636a1cb98faSBarry Smith        17-----14
2637a1cb98faSBarry Smith         |     |
2638a1cb98faSBarry Smith         |  2  |
2639a1cb98faSBarry Smith         |     |
2640a1cb98faSBarry Smith  17-----8-----7-----14
2641a1cb98faSBarry Smith   |     |     |     |
2642a1cb98faSBarry Smith   |  3  |  0  |  1  |
2643a1cb98faSBarry Smith   |     |     |     |
2644a1cb98faSBarry Smith  19-----5-----6-----13
2645a1cb98faSBarry Smith         |     |
2646a1cb98faSBarry Smith         |  4  |
2647a1cb98faSBarry Smith         |     |
2648a1cb98faSBarry Smith        19-----13
2649a1cb98faSBarry Smith 
2650a1cb98faSBarry Smith  and up through the top
2651a1cb98faSBarry Smith 
2652a1cb98faSBarry Smith        18-----16
2653a1cb98faSBarry Smith         |     |
2654a1cb98faSBarry Smith         |  2  |
2655a1cb98faSBarry Smith         |     |
2656a1cb98faSBarry Smith  18----10----11-----16
2657a1cb98faSBarry Smith   |     |     |     |
2658a1cb98faSBarry Smith   |  3  |  0  |  1  |
2659a1cb98faSBarry Smith   |     |     |     |
2660a1cb98faSBarry Smith  20-----9----12-----15
2661a1cb98faSBarry Smith         |     |
2662a1cb98faSBarry Smith         |  4  |
2663a1cb98faSBarry Smith         |     |
2664a1cb98faSBarry Smith        20-----15
2665a1cb98faSBarry Smith .ve
2666a1cb98faSBarry Smith 
26671cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
266824119c2aSMatthew G. Knepley @*/
266949704ca5SMatthew G. Knepley PetscErrorCode DMPlexCreateHexCylinderMesh(MPI_Comm comm, DMBoundaryType periodicZ, PetscInt Nr, DM *dm)
2670d71ae5a4SJacob Faibussowitsch {
26719318fe57SMatthew G. Knepley   PetscFunctionBegin;
267249704ca5SMatthew G. Knepley   PetscAssertPointer(dm, 4);
26739566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
26749566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
267549704ca5SMatthew G. Knepley   PetscCall(DMPlexCreateHexCylinderMesh_Internal(*dm, periodicZ, Nr));
26763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
26779318fe57SMatthew G. Knepley }
26789318fe57SMatthew G. Knepley 
2679d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeCylinderMesh_Internal(DM dm, PetscInt n, PetscBool interpolate)
2680d71ae5a4SJacob Faibussowitsch {
268124119c2aSMatthew G. Knepley   const PetscInt dim = 3;
2682412e9a14SMatthew G. Knepley   PetscInt       numCells, numVertices, v;
26839fe9f049SMatthew G. Knepley   PetscMPIInt    rank;
268424119c2aSMatthew G. Knepley 
268524119c2aSMatthew G. Knepley   PetscFunctionBegin;
268663a3b9bcSJacob Faibussowitsch   PetscCheck(n >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Number of wedges %" PetscInt_FMT " cannot be negative", n);
268746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
26889566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
26899566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
2690412e9a14SMatthew G. Knepley   /* Must create the celltype label here so that we do not automatically try to compute the types */
26919566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "celltype"));
269224119c2aSMatthew G. Knepley   /* Create topology */
269324119c2aSMatthew G. Knepley   {
269424119c2aSMatthew G. Knepley     PetscInt cone[6], c;
269524119c2aSMatthew G. Knepley 
2696dd400576SPatrick Sanan     numCells    = rank == 0 ? n : 0;
2697dd400576SPatrick Sanan     numVertices = rank == 0 ? 2 * (n + 1) : 0;
26989566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
26999566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
27009566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
270124119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
27029371c9d4SSatish Balay       cone[0] = c + n * 1;
27039371c9d4SSatish Balay       cone[1] = (c + 1) % n + n * 1;
27049371c9d4SSatish Balay       cone[2] = 0 + 3 * n;
27059371c9d4SSatish Balay       cone[3] = c + n * 2;
27069371c9d4SSatish Balay       cone[4] = (c + 1) % n + n * 2;
27079371c9d4SSatish Balay       cone[5] = 1 + 3 * n;
27089566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(dm, c, cone));
27099566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCellType(dm, c, DM_POLYTOPE_TRI_PRISM_TENSOR));
271024119c2aSMatthew G. Knepley     }
27119566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
27129566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
271324119c2aSMatthew G. Knepley   }
271448a46eb9SPierre Jolivet   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(dm, v, DM_POLYTOPE_POINT));
271524119c2aSMatthew G. Knepley   /* Create cylinder geometry */
271624119c2aSMatthew G. Knepley   {
271724119c2aSMatthew G. Knepley     Vec          coordinates;
271824119c2aSMatthew G. Knepley     PetscSection coordSection;
271924119c2aSMatthew G. Knepley     PetscScalar *coords;
2720412e9a14SMatthew G. Knepley     PetscInt     coordSize, c;
272124119c2aSMatthew G. Knepley 
272224119c2aSMatthew G. Knepley     /* Build coordinates */
27239566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
27249566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
27259566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
27269566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
272724119c2aSMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
27289566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
27299566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
273024119c2aSMatthew G. Knepley     }
27319566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
27329566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
27339566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
27349566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
27359566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
27369566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
27379566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
27389566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
273924119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
27409371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
27419371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
27429371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 2] = 1.0;
27439371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
27449371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
27459371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 2] = 0.0;
274624119c2aSMatthew G. Knepley     }
2747dd400576SPatrick Sanan     if (rank == 0) {
27489371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 0] = 0.0;
27499371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 1] = 0.0;
27509371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 2] = 1.0;
27519371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 0] = 0.0;
27529371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 1] = 0.0;
27539371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 2] = 0.0;
27549fe9f049SMatthew G. Knepley     }
27559566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
27569566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
27579566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
275824119c2aSMatthew G. Knepley   }
275946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
27609318fe57SMatthew G. Knepley   /* Interpolate */
27619566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
27623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
27639318fe57SMatthew G. Knepley }
27649318fe57SMatthew G. Knepley 
27659318fe57SMatthew G. Knepley /*@
27669318fe57SMatthew G. Knepley   DMPlexCreateWedgeCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using wedges.
27679318fe57SMatthew G. Knepley 
27689318fe57SMatthew G. Knepley   Collective
27699318fe57SMatthew G. Knepley 
27709318fe57SMatthew G. Knepley   Input Parameters:
2771a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
27729318fe57SMatthew G. Knepley . n           - The number of wedges around the origin
27739318fe57SMatthew G. Knepley - interpolate - Create edges and faces
27749318fe57SMatthew G. Knepley 
27759318fe57SMatthew G. Knepley   Output Parameter:
2776a1cb98faSBarry Smith . dm - The `DM` object
27779318fe57SMatthew G. Knepley 
27789318fe57SMatthew G. Knepley   Level: beginner
27799318fe57SMatthew G. Knepley 
27801cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
27819318fe57SMatthew G. Knepley @*/
2782d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeCylinderMesh(MPI_Comm comm, PetscInt n, PetscBool interpolate, DM *dm)
2783d71ae5a4SJacob Faibussowitsch {
27849318fe57SMatthew G. Knepley   PetscFunctionBegin;
27854f572ea9SToby Isaac   PetscAssertPointer(dm, 4);
27869566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
27879566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
27889566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(*dm, n, interpolate));
27893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
279024119c2aSMatthew G. Knepley }
279124119c2aSMatthew G. Knepley 
2792d71ae5a4SJacob Faibussowitsch static inline PetscReal DiffNormReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2793d71ae5a4SJacob Faibussowitsch {
279465a81367SMatthew G. Knepley   PetscReal prod = 0.0;
279565a81367SMatthew G. Knepley   PetscInt  i;
279665a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += PetscSqr(x[i] - y[i]);
279765a81367SMatthew G. Knepley   return PetscSqrtReal(prod);
279865a81367SMatthew G. Knepley }
2799dd2b43ebSStefano Zampini 
2800d71ae5a4SJacob Faibussowitsch static inline PetscReal DotReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2801d71ae5a4SJacob Faibussowitsch {
280265a81367SMatthew G. Knepley   PetscReal prod = 0.0;
280365a81367SMatthew G. Knepley   PetscInt  i;
280465a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += x[i] * y[i];
280565a81367SMatthew G. Knepley   return prod;
280665a81367SMatthew G. Knepley }
280765a81367SMatthew G. Knepley 
280851a74b61SMatthew G. Knepley /* The first constant is the sphere radius */
2809d71ae5a4SJacob 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[])
2810d71ae5a4SJacob Faibussowitsch {
281151a74b61SMatthew G. Knepley   PetscReal r     = PetscRealPart(constants[0]);
281251a74b61SMatthew G. Knepley   PetscReal norm2 = 0.0, fac;
281351a74b61SMatthew G. Knepley   PetscInt  n     = uOff[1] - uOff[0], d;
281451a74b61SMatthew G. Knepley 
281551a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) norm2 += PetscSqr(PetscRealPart(u[d]));
281651a74b61SMatthew G. Knepley   fac = r / PetscSqrtReal(norm2);
281751a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) f0[d] = u[d] * fac;
281851a74b61SMatthew G. Knepley }
281951a74b61SMatthew G. Knepley 
2820d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateSphereMesh_Internal(DM dm, PetscInt dim, PetscBool simplex, PetscReal R)
2821d71ae5a4SJacob Faibussowitsch {
282265a81367SMatthew G. Knepley   const PetscInt embedDim = dim + 1;
282365a81367SMatthew G. Knepley   PetscSection   coordSection;
282465a81367SMatthew G. Knepley   Vec            coordinates;
282565a81367SMatthew G. Knepley   PetscScalar   *coords;
282665a81367SMatthew G. Knepley   PetscReal     *coordsIn;
282707c565c5SJose E. Roman   PetscInt       numCells, numEdges, numVerts = 0, firstVertex = 0, v, firstEdge, coordSize, d, e;
282865a81367SMatthew G. Knepley   PetscMPIInt    rank;
282965a81367SMatthew G. Knepley 
283065a81367SMatthew G. Knepley   PetscFunctionBegin;
28319318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveBool(dm, simplex, 3);
283246139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
28339566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
28349566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim + 1));
28359566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
283665a81367SMatthew G. Knepley   switch (dim) {
28375c344501SMatthew G. Knepley   case 1:
28385c344501SMatthew G. Knepley     numCells = 16;
28395c344501SMatthew G. Knepley     numVerts = numCells;
28405c344501SMatthew G. Knepley 
28415c344501SMatthew G. Knepley     // Build Topology
28425c344501SMatthew G. Knepley     PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
28435c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
28445c344501SMatthew G. Knepley     PetscCall(DMSetUp(dm));
28455c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; ++c) {
28465c344501SMatthew G. Knepley       PetscInt cone[2];
28475c344501SMatthew G. Knepley 
28485c344501SMatthew G. Knepley       cone[0] = c + numCells;
28495c344501SMatthew G. Knepley       cone[1] = (c + 1) % numVerts + numCells;
28505c344501SMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, c, cone));
28515c344501SMatthew G. Knepley     }
28525c344501SMatthew G. Knepley     PetscCall(DMPlexSymmetrize(dm));
28535c344501SMatthew G. Knepley     PetscCall(DMPlexStratify(dm));
28545c344501SMatthew G. Knepley     PetscCall(PetscMalloc1(numVerts * embedDim, &coordsIn));
28555c344501SMatthew G. Knepley     for (PetscInt v = 0; v < numVerts; ++v) {
28565c344501SMatthew G. Knepley       const PetscReal rad = 2. * PETSC_PI * v / numVerts;
28575c344501SMatthew G. Knepley 
28585c344501SMatthew G. Knepley       coordsIn[v * embedDim + 0] = PetscCosReal(rad);
28595c344501SMatthew G. Knepley       coordsIn[v * embedDim + 1] = PetscSinReal(rad);
28605c344501SMatthew G. Knepley     }
28615c344501SMatthew G. Knepley     break;
286265a81367SMatthew G. Knepley   case 2:
286365a81367SMatthew G. Knepley     if (simplex) {
286451a74b61SMatthew G. Knepley       const PetscReal radius    = PetscSqrtReal(1 + PETSC_PHI * PETSC_PHI) / (1.0 + PETSC_PHI);
286551a74b61SMatthew G. Knepley       const PetscReal edgeLen   = 2.0 / (1.0 + PETSC_PHI) * (R / radius);
286665a81367SMatthew G. Knepley       const PetscInt  degree    = 5;
286751a74b61SMatthew G. Knepley       PetscReal       vertex[3] = {0.0, 1.0 / (1.0 + PETSC_PHI), PETSC_PHI / (1.0 + PETSC_PHI)};
286865a81367SMatthew G. Knepley       PetscInt        s[3]      = {1, 1, 1};
286965a81367SMatthew G. Knepley       PetscInt        cone[3];
287007c565c5SJose E. Roman       PetscInt       *graph;
287165a81367SMatthew G. Knepley 
28729371c9d4SSatish Balay       vertex[0] *= R / radius;
28739371c9d4SSatish Balay       vertex[1] *= R / radius;
28749371c9d4SSatish Balay       vertex[2] *= R / radius;
2875dd400576SPatrick Sanan       numCells    = rank == 0 ? 20 : 0;
2876dd400576SPatrick Sanan       numVerts    = rank == 0 ? 12 : 0;
287765a81367SMatthew G. Knepley       firstVertex = numCells;
287851a74b61SMatthew G. Knepley       /* Use icosahedron, which for a R-sphere has coordinates which are all cyclic permutations of
287965a81367SMatthew G. Knepley 
288065a81367SMatthew G. Knepley            (0, \pm 1/\phi+1, \pm \phi/\phi+1)
288165a81367SMatthew G. Knepley 
288265a81367SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
288351a74b61SMatthew G. Knepley          length is then given by 2/(1+\phi) = 2 * 0.38197 = 0.76393.
288465a81367SMatthew G. Knepley       */
288565a81367SMatthew G. Knepley       /* Construct vertices */
28869566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2887dd400576SPatrick Sanan       if (rank == 0) {
288807c565c5SJose E. Roman         for (PetscInt p = 0, i = 0; p < embedDim; ++p) {
288965a81367SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
289065a81367SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
289165a81367SMatthew G. Knepley               for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertex[(d + p) % embedDim];
289265a81367SMatthew G. Knepley               ++i;
289365a81367SMatthew G. Knepley             }
289465a81367SMatthew G. Knepley           }
289565a81367SMatthew G. Knepley         }
289645da822fSValeria Barra       }
289765a81367SMatthew G. Knepley       /* Construct graph */
28989566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
289907c565c5SJose E. Roman       for (PetscInt i = 0; i < numVerts; ++i) {
290007c565c5SJose E. Roman         PetscInt k = 0;
290107c565c5SJose E. Roman         for (PetscInt j = 0; j < numVerts; ++j) {
29029371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
29039371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
29049371c9d4SSatish Balay             ++k;
29059371c9d4SSatish Balay           }
290665a81367SMatthew G. Knepley         }
290763a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid icosahedron, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
290865a81367SMatthew G. Knepley       }
290965a81367SMatthew G. Knepley       /* Build Topology */
29109566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
291107c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
29129566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
291365a81367SMatthew G. Knepley       /* Cells */
291407c565c5SJose E. Roman       for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
291507c565c5SJose E. Roman         for (PetscInt j = 0; j < i; ++j) {
291607c565c5SJose E. Roman           for (PetscInt k = 0; k < j; ++k) {
291765a81367SMatthew G. Knepley             if (graph[i * numVerts + j] && graph[j * numVerts + k] && graph[k * numVerts + i]) {
29189371c9d4SSatish Balay               cone[0] = firstVertex + i;
29199371c9d4SSatish Balay               cone[1] = firstVertex + j;
29209371c9d4SSatish Balay               cone[2] = firstVertex + k;
292165a81367SMatthew G. Knepley               /* Check orientation */
292265a81367SMatthew G. Knepley               {
29239371c9d4SSatish Balay                 const PetscInt epsilon[3][3][3] = {
29249371c9d4SSatish Balay                   {{0, 0, 0},  {0, 0, 1},  {0, -1, 0}},
29259371c9d4SSatish Balay                   {{0, 0, -1}, {0, 0, 0},  {1, 0, 0} },
29269371c9d4SSatish Balay                   {{0, 1, 0},  {-1, 0, 0}, {0, 0, 0} }
29279371c9d4SSatish Balay                 };
292865a81367SMatthew G. Knepley                 PetscReal normal[3];
292965a81367SMatthew G. Knepley                 PetscInt  e, f;
293065a81367SMatthew G. Knepley 
293165a81367SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) {
293265a81367SMatthew G. Knepley                   normal[d] = 0.0;
293365a81367SMatthew G. Knepley                   for (e = 0; e < embedDim; ++e) {
2934ad540459SPierre 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]);
293565a81367SMatthew G. Knepley                   }
293665a81367SMatthew G. Knepley                 }
29379371c9d4SSatish Balay                 if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
29389371c9d4SSatish Balay                   PetscInt tmp = cone[1];
29399371c9d4SSatish Balay                   cone[1]      = cone[2];
29409371c9d4SSatish Balay                   cone[2]      = tmp;
294165a81367SMatthew G. Knepley                 }
294265a81367SMatthew G. Knepley               }
29439566063dSJacob Faibussowitsch               PetscCall(DMPlexSetCone(dm, c++, cone));
294465a81367SMatthew G. Knepley             }
294565a81367SMatthew G. Knepley           }
294665a81367SMatthew G. Knepley         }
294765a81367SMatthew G. Knepley       }
29489566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
29499566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
29509566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
295165a81367SMatthew G. Knepley     } else {
29522829fed8SMatthew G. Knepley       /*
29532829fed8SMatthew G. Knepley         12-21--13
29542829fed8SMatthew G. Knepley          |     |
29552829fed8SMatthew G. Knepley         25  4  24
29562829fed8SMatthew G. Knepley          |     |
29572829fed8SMatthew G. Knepley   12-25--9-16--8-24--13
29582829fed8SMatthew G. Knepley    |     |     |     |
29592829fed8SMatthew G. Knepley   23  5 17  0 15  3  22
29602829fed8SMatthew G. Knepley    |     |     |     |
29612829fed8SMatthew G. Knepley   10-20--6-14--7-19--11
29622829fed8SMatthew G. Knepley          |     |
29632829fed8SMatthew G. Knepley         20  1  19
29642829fed8SMatthew G. Knepley          |     |
29652829fed8SMatthew G. Knepley         10-18--11
29662829fed8SMatthew G. Knepley          |     |
29672829fed8SMatthew G. Knepley         23  2  22
29682829fed8SMatthew G. Knepley          |     |
29692829fed8SMatthew G. Knepley         12-21--13
29702829fed8SMatthew G. Knepley        */
29712829fed8SMatthew G. Knepley       PetscInt cone[4], ornt[4];
29722829fed8SMatthew G. Knepley 
2973dd400576SPatrick Sanan       numCells    = rank == 0 ? 6 : 0;
2974dd400576SPatrick Sanan       numEdges    = rank == 0 ? 12 : 0;
2975dd400576SPatrick Sanan       numVerts    = rank == 0 ? 8 : 0;
297665a81367SMatthew G. Knepley       firstVertex = numCells;
297765a81367SMatthew G. Knepley       firstEdge   = numCells + numVerts;
29782829fed8SMatthew G. Knepley       /* Build Topology */
29799566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numEdges + numVerts));
298007c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 4));
298148a46eb9SPierre Jolivet       for (e = firstEdge; e < firstEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
29829566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
2983dd400576SPatrick Sanan       if (rank == 0) {
29842829fed8SMatthew G. Knepley         /* Cell 0 */
29859371c9d4SSatish Balay         cone[0] = 14;
29869371c9d4SSatish Balay         cone[1] = 15;
29879371c9d4SSatish Balay         cone[2] = 16;
29889371c9d4SSatish Balay         cone[3] = 17;
29899566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
29909371c9d4SSatish Balay         ornt[0] = 0;
29919371c9d4SSatish Balay         ornt[1] = 0;
29929371c9d4SSatish Balay         ornt[2] = 0;
29939371c9d4SSatish Balay         ornt[3] = 0;
29949566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 0, ornt));
29952829fed8SMatthew G. Knepley         /* Cell 1 */
29969371c9d4SSatish Balay         cone[0] = 18;
29979371c9d4SSatish Balay         cone[1] = 19;
29989371c9d4SSatish Balay         cone[2] = 14;
29999371c9d4SSatish Balay         cone[3] = 20;
30009566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
30019371c9d4SSatish Balay         ornt[0] = 0;
30029371c9d4SSatish Balay         ornt[1] = 0;
30039371c9d4SSatish Balay         ornt[2] = -1;
30049371c9d4SSatish Balay         ornt[3] = 0;
30059566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 1, ornt));
30062829fed8SMatthew G. Knepley         /* Cell 2 */
30079371c9d4SSatish Balay         cone[0] = 21;
30089371c9d4SSatish Balay         cone[1] = 22;
30099371c9d4SSatish Balay         cone[2] = 18;
30109371c9d4SSatish Balay         cone[3] = 23;
30119566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
30129371c9d4SSatish Balay         ornt[0] = 0;
30139371c9d4SSatish Balay         ornt[1] = 0;
30149371c9d4SSatish Balay         ornt[2] = -1;
30159371c9d4SSatish Balay         ornt[3] = 0;
30169566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 2, ornt));
30172829fed8SMatthew G. Knepley         /* Cell 3 */
30189371c9d4SSatish Balay         cone[0] = 19;
30199371c9d4SSatish Balay         cone[1] = 22;
30209371c9d4SSatish Balay         cone[2] = 24;
30219371c9d4SSatish Balay         cone[3] = 15;
30229566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
30239371c9d4SSatish Balay         ornt[0] = -1;
30249371c9d4SSatish Balay         ornt[1] = -1;
30259371c9d4SSatish Balay         ornt[2] = 0;
30269371c9d4SSatish Balay         ornt[3] = -1;
30279566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 3, ornt));
30282829fed8SMatthew G. Knepley         /* Cell 4 */
30299371c9d4SSatish Balay         cone[0] = 16;
30309371c9d4SSatish Balay         cone[1] = 24;
30319371c9d4SSatish Balay         cone[2] = 21;
30329371c9d4SSatish Balay         cone[3] = 25;
30339566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
30349371c9d4SSatish Balay         ornt[0] = -1;
30359371c9d4SSatish Balay         ornt[1] = -1;
30369371c9d4SSatish Balay         ornt[2] = -1;
30379371c9d4SSatish Balay         ornt[3] = 0;
30389566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 4, ornt));
30392829fed8SMatthew G. Knepley         /* Cell 5 */
30409371c9d4SSatish Balay         cone[0] = 20;
30419371c9d4SSatish Balay         cone[1] = 17;
30429371c9d4SSatish Balay         cone[2] = 25;
30439371c9d4SSatish Balay         cone[3] = 23;
30449566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
30459371c9d4SSatish Balay         ornt[0] = -1;
30469371c9d4SSatish Balay         ornt[1] = -1;
30479371c9d4SSatish Balay         ornt[2] = -1;
30489371c9d4SSatish Balay         ornt[3] = -1;
30499566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 5, ornt));
30502829fed8SMatthew G. Knepley         /* Edges */
30519371c9d4SSatish Balay         cone[0] = 6;
30529371c9d4SSatish Balay         cone[1] = 7;
30539566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
30549371c9d4SSatish Balay         cone[0] = 7;
30559371c9d4SSatish Balay         cone[1] = 8;
30569566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 15, cone));
30579371c9d4SSatish Balay         cone[0] = 8;
30589371c9d4SSatish Balay         cone[1] = 9;
30599566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 16, cone));
30609371c9d4SSatish Balay         cone[0] = 9;
30619371c9d4SSatish Balay         cone[1] = 6;
30629566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 17, cone));
30639371c9d4SSatish Balay         cone[0] = 10;
30649371c9d4SSatish Balay         cone[1] = 11;
30659566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 18, cone));
30669371c9d4SSatish Balay         cone[0] = 11;
30679371c9d4SSatish Balay         cone[1] = 7;
30689566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 19, cone));
30699371c9d4SSatish Balay         cone[0] = 6;
30709371c9d4SSatish Balay         cone[1] = 10;
30719566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 20, cone));
30729371c9d4SSatish Balay         cone[0] = 12;
30739371c9d4SSatish Balay         cone[1] = 13;
30749566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 21, cone));
30759371c9d4SSatish Balay         cone[0] = 13;
30769371c9d4SSatish Balay         cone[1] = 11;
30779566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 22, cone));
30789371c9d4SSatish Balay         cone[0] = 10;
30799371c9d4SSatish Balay         cone[1] = 12;
30809566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 23, cone));
30819371c9d4SSatish Balay         cone[0] = 13;
30829371c9d4SSatish Balay         cone[1] = 8;
30839566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 24, cone));
30849371c9d4SSatish Balay         cone[0] = 12;
30859371c9d4SSatish Balay         cone[1] = 9;
30869566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 25, cone));
308745da822fSValeria Barra       }
30889566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
30899566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
30902829fed8SMatthew G. Knepley       /* Build coordinates */
30919566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
3092dd400576SPatrick Sanan       if (rank == 0) {
30939371c9d4SSatish Balay         coordsIn[0 * embedDim + 0] = -R;
30949371c9d4SSatish Balay         coordsIn[0 * embedDim + 1] = R;
30959371c9d4SSatish Balay         coordsIn[0 * embedDim + 2] = -R;
30969371c9d4SSatish Balay         coordsIn[1 * embedDim + 0] = R;
30979371c9d4SSatish Balay         coordsIn[1 * embedDim + 1] = R;
30989371c9d4SSatish Balay         coordsIn[1 * embedDim + 2] = -R;
30999371c9d4SSatish Balay         coordsIn[2 * embedDim + 0] = R;
31009371c9d4SSatish Balay         coordsIn[2 * embedDim + 1] = -R;
31019371c9d4SSatish Balay         coordsIn[2 * embedDim + 2] = -R;
31029371c9d4SSatish Balay         coordsIn[3 * embedDim + 0] = -R;
31039371c9d4SSatish Balay         coordsIn[3 * embedDim + 1] = -R;
31049371c9d4SSatish Balay         coordsIn[3 * embedDim + 2] = -R;
31059371c9d4SSatish Balay         coordsIn[4 * embedDim + 0] = -R;
31069371c9d4SSatish Balay         coordsIn[4 * embedDim + 1] = R;
31079371c9d4SSatish Balay         coordsIn[4 * embedDim + 2] = R;
31089371c9d4SSatish Balay         coordsIn[5 * embedDim + 0] = R;
31099371c9d4SSatish Balay         coordsIn[5 * embedDim + 1] = R;
31109371c9d4SSatish Balay         coordsIn[5 * embedDim + 2] = R;
31119371c9d4SSatish Balay         coordsIn[6 * embedDim + 0] = -R;
31129371c9d4SSatish Balay         coordsIn[6 * embedDim + 1] = -R;
31139371c9d4SSatish Balay         coordsIn[6 * embedDim + 2] = R;
31149371c9d4SSatish Balay         coordsIn[7 * embedDim + 0] = R;
31159371c9d4SSatish Balay         coordsIn[7 * embedDim + 1] = -R;
31169371c9d4SSatish Balay         coordsIn[7 * embedDim + 2] = R;
311765a81367SMatthew G. Knepley       }
311845da822fSValeria Barra     }
311965a81367SMatthew G. Knepley     break;
312065a81367SMatthew G. Knepley   case 3:
3121116ded15SMatthew G. Knepley     if (simplex) {
3122116ded15SMatthew G. Knepley       const PetscReal edgeLen         = 1.0 / PETSC_PHI;
312351a74b61SMatthew G. Knepley       PetscReal       vertexA[4]      = {0.5, 0.5, 0.5, 0.5};
312451a74b61SMatthew G. Knepley       PetscReal       vertexB[4]      = {1.0, 0.0, 0.0, 0.0};
312551a74b61SMatthew G. Knepley       PetscReal       vertexC[4]      = {0.5, 0.5 * PETSC_PHI, 0.5 / PETSC_PHI, 0.0};
3126116ded15SMatthew G. Knepley       const PetscInt  degree          = 12;
3127116ded15SMatthew G. Knepley       PetscInt        s[4]            = {1, 1, 1};
31289371c9d4SSatish Balay       PetscInt        evenPerm[12][4] = {
31299371c9d4SSatish Balay         {0, 1, 2, 3},
31309371c9d4SSatish Balay         {0, 2, 3, 1},
31319371c9d4SSatish Balay         {0, 3, 1, 2},
31329371c9d4SSatish Balay         {1, 0, 3, 2},
31339371c9d4SSatish Balay         {1, 2, 0, 3},
31349371c9d4SSatish Balay         {1, 3, 2, 0},
31359371c9d4SSatish Balay         {2, 0, 1, 3},
31369371c9d4SSatish Balay         {2, 1, 3, 0},
31379371c9d4SSatish Balay         {2, 3, 0, 1},
31389371c9d4SSatish Balay         {3, 0, 2, 1},
31399371c9d4SSatish Balay         {3, 1, 0, 2},
31409371c9d4SSatish Balay         {3, 2, 1, 0}
31419371c9d4SSatish Balay       };
3142116ded15SMatthew G. Knepley       PetscInt  cone[4];
3143116ded15SMatthew G. Knepley       PetscInt *graph, p, i, j, k, l;
3144116ded15SMatthew G. Knepley 
31459371c9d4SSatish Balay       vertexA[0] *= R;
31469371c9d4SSatish Balay       vertexA[1] *= R;
31479371c9d4SSatish Balay       vertexA[2] *= R;
31489371c9d4SSatish Balay       vertexA[3] *= R;
31499371c9d4SSatish Balay       vertexB[0] *= R;
31509371c9d4SSatish Balay       vertexB[1] *= R;
31519371c9d4SSatish Balay       vertexB[2] *= R;
31529371c9d4SSatish Balay       vertexB[3] *= R;
31539371c9d4SSatish Balay       vertexC[0] *= R;
31549371c9d4SSatish Balay       vertexC[1] *= R;
31559371c9d4SSatish Balay       vertexC[2] *= R;
31569371c9d4SSatish Balay       vertexC[3] *= R;
3157dd400576SPatrick Sanan       numCells    = rank == 0 ? 600 : 0;
3158dd400576SPatrick Sanan       numVerts    = rank == 0 ? 120 : 0;
3159116ded15SMatthew G. Knepley       firstVertex = numCells;
3160116ded15SMatthew G. Knepley       /* Use the 600-cell, which for a unit sphere has coordinates which are
3161116ded15SMatthew G. Knepley 
3162116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm 1,    \pm 1, \pm 1)                          16
3163116ded15SMatthew G. Knepley                (\pm 1,    0,       0,      0)  all cyclic permutations   8
3164116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm phi, \pm 1/phi, 0)  all even permutations    96
3165116ded15SMatthew G. Knepley 
3166116ded15SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
31676333ae4fSvaleriabarra          length is then given by 1/\phi = 0.61803.
3168116ded15SMatthew G. Knepley 
3169116ded15SMatthew G. Knepley          http://buzzard.pugetsound.edu/sage-practice/ch03s03.html
3170116ded15SMatthew G. Knepley          http://mathworld.wolfram.com/600-Cell.html
3171116ded15SMatthew G. Knepley       */
3172116ded15SMatthew G. Knepley       /* Construct vertices */
31739566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
3174116ded15SMatthew G. Knepley       i = 0;
3175dd400576SPatrick Sanan       if (rank == 0) {
3176116ded15SMatthew G. Knepley         for (s[0] = -1; s[0] < 2; s[0] += 2) {
3177116ded15SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
3178116ded15SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
3179116ded15SMatthew G. Knepley               for (s[3] = -1; s[3] < 2; s[3] += 2) {
3180116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[d] * vertexA[d];
3181116ded15SMatthew G. Knepley                 ++i;
3182116ded15SMatthew G. Knepley               }
3183116ded15SMatthew G. Knepley             }
3184116ded15SMatthew G. Knepley           }
3185116ded15SMatthew G. Knepley         }
3186116ded15SMatthew G. Knepley         for (p = 0; p < embedDim; ++p) {
3187116ded15SMatthew G. Knepley           s[1] = s[2] = s[3] = 1;
3188116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
3189116ded15SMatthew G. Knepley             for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertexB[(d + p) % embedDim];
3190116ded15SMatthew G. Knepley             ++i;
3191116ded15SMatthew G. Knepley           }
3192116ded15SMatthew G. Knepley         }
3193116ded15SMatthew G. Knepley         for (p = 0; p < 12; ++p) {
3194116ded15SMatthew G. Knepley           s[3] = 1;
3195116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
3196116ded15SMatthew G. Knepley             for (s[1] = -1; s[1] < 2; s[1] += 2) {
3197116ded15SMatthew G. Knepley               for (s[2] = -1; s[2] < 2; s[2] += 2) {
3198116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[evenPerm[p][d]] * vertexC[evenPerm[p][d]];
3199116ded15SMatthew G. Knepley                 ++i;
3200116ded15SMatthew G. Knepley               }
3201116ded15SMatthew G. Knepley             }
3202116ded15SMatthew G. Knepley           }
3203116ded15SMatthew G. Knepley         }
320445da822fSValeria Barra       }
320563a3b9bcSJacob Faibussowitsch       PetscCheck(i == numVerts, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertices %" PetscInt_FMT " != %" PetscInt_FMT, i, numVerts);
3206116ded15SMatthew G. Knepley       /* Construct graph */
32079566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
3208116ded15SMatthew G. Knepley       for (i = 0; i < numVerts; ++i) {
3209116ded15SMatthew G. Knepley         for (j = 0, k = 0; j < numVerts; ++j) {
32109371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
32119371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
32129371c9d4SSatish Balay             ++k;
32139371c9d4SSatish Balay           }
3214116ded15SMatthew G. Knepley         }
321563a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
3216116ded15SMatthew G. Knepley       }
3217116ded15SMatthew G. Knepley       /* Build Topology */
32189566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
321907c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
32209566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
3221116ded15SMatthew G. Knepley       /* Cells */
3222dd400576SPatrick Sanan       if (rank == 0) {
322307c565c5SJose E. Roman         for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
3224116ded15SMatthew G. Knepley           for (j = 0; j < i; ++j) {
3225116ded15SMatthew G. Knepley             for (k = 0; k < j; ++k) {
3226116ded15SMatthew G. Knepley               for (l = 0; l < k; ++l) {
32279371c9d4SSatish 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]) {
32289371c9d4SSatish Balay                   cone[0] = firstVertex + i;
32299371c9d4SSatish Balay                   cone[1] = firstVertex + j;
32309371c9d4SSatish Balay                   cone[2] = firstVertex + k;
32319371c9d4SSatish Balay                   cone[3] = firstVertex + l;
3232116ded15SMatthew G. Knepley                   /* Check orientation: https://ef.gy/linear-algebra:normal-vectors-in-higher-dimensional-spaces */
3233116ded15SMatthew G. Knepley                   {
32349371c9d4SSatish Balay                     const PetscInt epsilon[4][4][4][4] = {
32359371c9d4SSatish 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}}},
3236116ded15SMatthew G. Knepley 
32379371c9d4SSatish 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}}},
3238116ded15SMatthew G. Knepley 
32399371c9d4SSatish 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}}},
3240116ded15SMatthew G. Knepley 
32419371c9d4SSatish 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}} }
32429371c9d4SSatish Balay                     };
3243116ded15SMatthew G. Knepley                     PetscReal normal[4];
3244116ded15SMatthew G. Knepley                     PetscInt  e, f, g;
3245116ded15SMatthew G. Knepley 
3246116ded15SMatthew G. Knepley                     for (d = 0; d < embedDim; ++d) {
3247116ded15SMatthew G. Knepley                       normal[d] = 0.0;
3248116ded15SMatthew G. Knepley                       for (e = 0; e < embedDim; ++e) {
3249116ded15SMatthew G. Knepley                         for (f = 0; f < embedDim; ++f) {
3250116ded15SMatthew G. Knepley                           for (g = 0; g < embedDim; ++g) {
3251116ded15SMatthew 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]);
3252116ded15SMatthew G. Knepley                           }
3253116ded15SMatthew G. Knepley                         }
3254116ded15SMatthew G. Knepley                       }
3255116ded15SMatthew G. Knepley                     }
32569371c9d4SSatish Balay                     if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
32579371c9d4SSatish Balay                       PetscInt tmp = cone[1];
32589371c9d4SSatish Balay                       cone[1]      = cone[2];
32599371c9d4SSatish Balay                       cone[2]      = tmp;
32609371c9d4SSatish Balay                     }
3261116ded15SMatthew G. Knepley                   }
32629566063dSJacob Faibussowitsch                   PetscCall(DMPlexSetCone(dm, c++, cone));
3263116ded15SMatthew G. Knepley                 }
3264116ded15SMatthew G. Knepley               }
3265116ded15SMatthew G. Knepley             }
3266116ded15SMatthew G. Knepley           }
3267116ded15SMatthew G. Knepley         }
326845da822fSValeria Barra       }
32699566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
32709566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
32719566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
3272116ded15SMatthew G. Knepley     }
3273f4d061e9SPierre Jolivet     break;
3274d71ae5a4SJacob Faibussowitsch   default:
3275d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension for sphere: %" PetscInt_FMT, dim);
327665a81367SMatthew G. Knepley   }
327765a81367SMatthew G. Knepley   /* Create coordinates */
32789566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
32799566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
32809566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, embedDim));
32819566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVerts));
32822829fed8SMatthew G. Knepley   for (v = firstVertex; v < firstVertex + numVerts; ++v) {
32839566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, embedDim));
32849566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, embedDim));
32852829fed8SMatthew G. Knepley   }
32869566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
32879566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
32889566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
32899566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, embedDim));
32909566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
32919566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
32929566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
32939566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
32949371c9d4SSatish Balay   for (v = 0; v < numVerts; ++v)
3295ad540459SPierre Jolivet     for (d = 0; d < embedDim; ++d) coords[v * embedDim + d] = coordsIn[v * embedDim + d];
32969566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
32979566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
32989566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
32999566063dSJacob Faibussowitsch   PetscCall(PetscFree(coordsIn));
330051a74b61SMatthew G. Knepley   {
330151a74b61SMatthew G. Knepley     DM          cdm;
330251a74b61SMatthew G. Knepley     PetscDS     cds;
33039318fe57SMatthew G. Knepley     PetscScalar c = R;
330451a74b61SMatthew G. Knepley 
3305e44f6aebSMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, snapToSphere));
33069566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
33079566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
33089566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 1, &c));
330951a74b61SMatthew G. Knepley   }
331046139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
33119318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
33129566063dSJacob Faibussowitsch   if (simplex) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
33133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
33149318fe57SMatthew G. Knepley }
33159318fe57SMatthew G. Knepley 
3316b7f5c055SJed Brown typedef void (*TPSEvaluateFunc)(const PetscReal[], PetscReal *, PetscReal[], PetscReal (*)[3]);
3317b7f5c055SJed Brown 
3318b7f5c055SJed Brown /*
3319b7f5c055SJed Brown  The Schwarz P implicit surface is
3320b7f5c055SJed Brown 
3321b7f5c055SJed Brown      f(x) = cos(x0) + cos(x1) + cos(x2) = 0
3322b7f5c055SJed Brown */
3323d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_SchwarzP(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3324d71ae5a4SJacob Faibussowitsch {
3325b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(y[0] * PETSC_PI), PetscCosReal(y[1] * PETSC_PI), PetscCosReal(y[2] * PETSC_PI)};
3326b7f5c055SJed Brown   PetscReal g[3] = {-PetscSinReal(y[0] * PETSC_PI), -PetscSinReal(y[1] * PETSC_PI), -PetscSinReal(y[2] * PETSC_PI)};
3327b7f5c055SJed Brown   f[0]           = c[0] + c[1] + c[2];
3328b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3329b7f5c055SJed Brown     grad[i] = PETSC_PI * g[i];
3330ad540459SPierre Jolivet     for (PetscInt j = 0; j < 3; j++) hess[i][j] = (i == j) ? -PetscSqr(PETSC_PI) * c[i] : 0.;
3331b7f5c055SJed Brown   }
3332b7f5c055SJed Brown }
3333b7f5c055SJed Brown 
33344663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3335d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_SchwarzP(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3336d71ae5a4SJacob Faibussowitsch {
3337ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) u[i] = -PETSC_PI * PetscSinReal(x[i] * PETSC_PI);
33383ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
33394663dae6SJed Brown }
33404663dae6SJed Brown 
3341b7f5c055SJed Brown /*
3342b7f5c055SJed Brown  The Gyroid implicit surface is
3343b7f5c055SJed Brown 
3344b7f5c055SJed 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)
3345b7f5c055SJed Brown 
3346b7f5c055SJed Brown */
3347d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_Gyroid(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3348d71ae5a4SJacob Faibussowitsch {
3349b7f5c055SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * y[0]), PetscSinReal(PETSC_PI * (y[1] + .5)), PetscSinReal(PETSC_PI * (y[2] + .25))};
3350b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * y[0]), PetscCosReal(PETSC_PI * (y[1] + .5)), PetscCosReal(PETSC_PI * (y[2] + .25))};
3351b7f5c055SJed Brown   f[0]           = s[0] * c[1] + s[1] * c[2] + s[2] * c[0];
3352b7f5c055SJed Brown   grad[0]        = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
3353b7f5c055SJed Brown   grad[1]        = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
3354b7f5c055SJed Brown   grad[2]        = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
3355b7f5c055SJed Brown   hess[0][0]     = -PetscSqr(PETSC_PI) * (s[0] * c[1] + s[2] * c[0]);
3356b7f5c055SJed Brown   hess[0][1]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3357b7f5c055SJed Brown   hess[0][2]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3358b7f5c055SJed Brown   hess[1][0]     = -PetscSqr(PETSC_PI) * (s[1] * c[2] + s[0] * c[1]);
3359b7f5c055SJed Brown   hess[1][1]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3360b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3361b7f5c055SJed Brown   hess[2][0]     = -PetscSqr(PETSC_PI) * (s[2] * c[0] + s[1] * c[2]);
3362b7f5c055SJed Brown   hess[2][1]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3363b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3364b7f5c055SJed Brown }
3365b7f5c055SJed Brown 
33664663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3367d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_Gyroid(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3368d71ae5a4SJacob Faibussowitsch {
33694663dae6SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * x[0]), PetscSinReal(PETSC_PI * (x[1] + .5)), PetscSinReal(PETSC_PI * (x[2] + .25))};
33704663dae6SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * x[0]), PetscCosReal(PETSC_PI * (x[1] + .5)), PetscCosReal(PETSC_PI * (x[2] + .25))};
33714663dae6SJed Brown   u[0]           = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
33724663dae6SJed Brown   u[1]           = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
33734663dae6SJed Brown   u[2]           = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
33743ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
33754663dae6SJed Brown }
33764663dae6SJed Brown 
3377b7f5c055SJed Brown /*
3378b7f5c055SJed Brown    We wish to solve
3379b7f5c055SJed Brown 
3380b7f5c055SJed Brown          min_y || y - x ||^2  subject to f(y) = 0
3381b7f5c055SJed Brown 
3382b7f5c055SJed Brown    Let g(y) = grad(f).  The minimization problem is equivalent to asking to satisfy
3383b7f5c055SJed 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
3384b7f5c055SJed Brown    tangent space and ask for both components in the tangent space to be zero.
3385b7f5c055SJed Brown 
3386b7f5c055SJed Brown    Take g to be a column vector and compute the "full QR" factorization Q R = g,
3387b7f5c055SJed Brown    where Q = I - 2 n n^T is a symmetric orthogonal matrix.
3388b7f5c055SJed Brown    The first column of Q is parallel to g so the remaining two columns span the null space.
3389b7f5c055SJed Brown    Let Qn = Q[:,1:] be those remaining columns.  Then Qn Qn^T is an orthogonal projector into the tangent space.
3390da81f932SPierre Jolivet    Since Q is symmetric, this is equivalent to multiplying by Q and taking the last two entries.
3391b7f5c055SJed Brown    In total, we have a system of 3 equations in 3 unknowns:
3392b7f5c055SJed Brown 
3393b7f5c055SJed Brown      f(y) = 0                       1 equation
3394b7f5c055SJed Brown      Qn^T (y - x) = 0               2 equations
3395b7f5c055SJed Brown 
3396b7f5c055SJed Brown    Here, we compute the residual and Jacobian of this system.
3397b7f5c055SJed Brown */
3398d71ae5a4SJacob Faibussowitsch static void TPSNearestPointResJac(TPSEvaluateFunc feval, const PetscScalar x[], const PetscScalar y[], PetscScalar res[], PetscScalar J[])
3399d71ae5a4SJacob Faibussowitsch {
3400b7f5c055SJed Brown   PetscReal yreal[3] = {PetscRealPart(y[0]), PetscRealPart(y[1]), PetscRealPart(y[2])};
3401b7f5c055SJed Brown   PetscReal d[3]     = {PetscRealPart(y[0] - x[0]), PetscRealPart(y[1] - x[1]), PetscRealPart(y[2] - x[2])};
34022f0490c0SSatish Balay   PetscReal f, grad[3], n[3], norm, norm_y[3], nd, nd_y[3], sign;
34039371c9d4SSatish Balay   PetscReal n_y[3][3] = {
34049371c9d4SSatish Balay     {0, 0, 0},
34059371c9d4SSatish Balay     {0, 0, 0},
34069371c9d4SSatish Balay     {0, 0, 0}
34079371c9d4SSatish Balay   };
3408b7f5c055SJed Brown 
3409b7f5c055SJed Brown   feval(yreal, &f, grad, n_y);
3410b7f5c055SJed Brown 
3411b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n[i] = grad[i];
3412b7f5c055SJed Brown   norm = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3413ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) norm_y[i] = 1. / norm * n[i] * n_y[i][i];
3414b7f5c055SJed Brown 
3415b7f5c055SJed Brown   // Define the Householder reflector
3416b7f5c055SJed Brown   sign = n[0] >= 0 ? 1. : -1.;
3417b7f5c055SJed Brown   n[0] += norm * sign;
3418b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n_y[0][i] += norm_y[i] * sign;
3419b7f5c055SJed Brown 
3420b7f5c055SJed Brown   norm      = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3421b7f5c055SJed Brown   norm_y[0] = 1. / norm * (n[0] * n_y[0][0]);
3422b7f5c055SJed Brown   norm_y[1] = 1. / norm * (n[0] * n_y[0][1] + n[1] * n_y[1][1]);
3423b7f5c055SJed Brown   norm_y[2] = 1. / norm * (n[0] * n_y[0][2] + n[2] * n_y[2][2]);
3424b7f5c055SJed Brown 
3425b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3426b7f5c055SJed Brown     n[i] /= norm;
3427b7f5c055SJed Brown     for (PetscInt j = 0; j < 3; j++) {
3428b7f5c055SJed Brown       // note that n[i] is n_old[i]/norm when executing the code below
3429b7f5c055SJed Brown       n_y[i][j] = n_y[i][j] / norm - n[i] / norm * norm_y[j];
3430b7f5c055SJed Brown     }
3431b7f5c055SJed Brown   }
3432b7f5c055SJed Brown 
3433b7f5c055SJed Brown   nd = n[0] * d[0] + n[1] * d[1] + n[2] * d[2];
3434b7f5c055SJed 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];
3435b7f5c055SJed Brown 
3436b7f5c055SJed Brown   res[0] = f;
3437b7f5c055SJed Brown   res[1] = d[1] - 2 * n[1] * nd;
3438b7f5c055SJed Brown   res[2] = d[2] - 2 * n[2] * nd;
3439b7f5c055SJed Brown   // J[j][i] is J_{ij} (column major)
3440b7f5c055SJed Brown   for (PetscInt j = 0; j < 3; j++) {
3441b7f5c055SJed Brown     J[0 + j * 3] = grad[j];
3442b7f5c055SJed Brown     J[1 + j * 3] = (j == 1) * 1. - 2 * (n_y[1][j] * nd + n[1] * nd_y[j]);
3443b7f5c055SJed Brown     J[2 + j * 3] = (j == 2) * 1. - 2 * (n_y[2][j] * nd + n[2] * nd_y[j]);
3444b7f5c055SJed Brown   }
3445b7f5c055SJed Brown }
3446b7f5c055SJed Brown 
3447b7f5c055SJed Brown /*
3448b7f5c055SJed Brown    Project x to the nearest point on the implicit surface using Newton's method.
3449b7f5c055SJed Brown */
3450d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSNearestPoint(TPSEvaluateFunc feval, PetscScalar x[])
3451d71ae5a4SJacob Faibussowitsch {
3452b7f5c055SJed Brown   PetscScalar y[3] = {x[0], x[1], x[2]}; // Initial guess
3453b7f5c055SJed Brown 
3454b7f5c055SJed Brown   PetscFunctionBegin;
3455b7f5c055SJed Brown   for (PetscInt iter = 0; iter < 10; iter++) {
3456b7f5c055SJed Brown     PetscScalar res[3], J[9];
3457b7f5c055SJed Brown     PetscReal   resnorm;
3458b7f5c055SJed Brown     TPSNearestPointResJac(feval, x, y, res, J);
3459b7f5c055SJed Brown     resnorm = PetscSqrtReal(PetscSqr(PetscRealPart(res[0])) + PetscSqr(PetscRealPart(res[1])) + PetscSqr(PetscRealPart(res[2])));
3460b7f5c055SJed Brown     if (0) { // Turn on this monitor if you need to confirm quadratic convergence
346163a3b9bcSJacob 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])));
3462b7f5c055SJed Brown     }
3463b7f5c055SJed Brown     if (resnorm < PETSC_SMALL) break;
3464b7f5c055SJed Brown 
3465b7f5c055SJed Brown     // Take the Newton step
34669566063dSJacob Faibussowitsch     PetscCall(PetscKernel_A_gets_inverse_A_3(J, 0., PETSC_FALSE, NULL));
3467b7f5c055SJed Brown     PetscKernel_v_gets_v_minus_A_times_w_3(y, J, res);
3468b7f5c055SJed Brown   }
3469b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) x[i] = y[i];
34703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3471b7f5c055SJed Brown }
3472b7f5c055SJed Brown 
3473b7f5c055SJed Brown const char *const DMPlexTPSTypes[] = {"SCHWARZ_P", "GYROID", "DMPlexTPSType", "DMPLEX_TPS_", NULL};
3474b7f5c055SJed Brown 
3475d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateTPSMesh_Internal(DM dm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness)
3476d71ae5a4SJacob Faibussowitsch {
3477b7f5c055SJed Brown   PetscMPIInt rank;
3478b7f5c055SJed Brown   PetscInt    topoDim = 2, spaceDim = 3, numFaces = 0, numVertices = 0, numEdges = 0;
3479b7f5c055SJed Brown   PetscInt(*edges)[2] = NULL, *edgeSets = NULL;
3480b7f5c055SJed Brown   PetscInt           *cells_flat = NULL;
3481b7f5c055SJed Brown   PetscReal          *vtxCoords  = NULL;
3482b7f5c055SJed Brown   TPSEvaluateFunc     evalFunc   = NULL;
34838434afd1SBarry Smith   PetscSimplePointFn *normalFunc = NULL;
3484b7f5c055SJed Brown   DMLabel             label;
3485b7f5c055SJed Brown 
3486b7f5c055SJed Brown   PetscFunctionBegin;
348746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
34889566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
348963a3b9bcSJacob 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);
3490b7f5c055SJed Brown   switch (tpstype) {
3491b7f5c055SJed Brown   case DMPLEX_TPS_SCHWARZ_P:
3492b7f5c055SJed 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");
3493c5853193SPierre Jolivet     if (rank == 0) {
3494b7f5c055SJed Brown       PetscInt(*cells)[6][4][4] = NULL; // [junction, junction-face, cell, conn]
3495b7f5c055SJed Brown       PetscInt  Njunctions = 0, Ncuts = 0, Npipes[3], vcount;
3496b7f5c055SJed Brown       PetscReal L = 1;
3497b7f5c055SJed Brown 
3498b7f5c055SJed Brown       Npipes[0]   = (extent[0] + 1) * extent[1] * extent[2];
3499b7f5c055SJed Brown       Npipes[1]   = extent[0] * (extent[1] + 1) * extent[2];
3500b7f5c055SJed Brown       Npipes[2]   = extent[0] * extent[1] * (extent[2] + 1);
3501b7f5c055SJed Brown       Njunctions  = extent[0] * extent[1] * extent[2];
3502b7f5c055SJed Brown       Ncuts       = 2 * (extent[0] * extent[1] + extent[1] * extent[2] + extent[2] * extent[0]);
3503b7f5c055SJed Brown       numVertices = 4 * (Npipes[0] + Npipes[1] + Npipes[2]) + 8 * Njunctions;
35049566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(3 * numVertices, &vtxCoords));
35059566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Njunctions, &cells));
35069566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edges));
35079566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edgeSets));
3508b7f5c055SJed Brown       // x-normal pipes
3509b7f5c055SJed Brown       vcount = 0;
3510b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0] + 1; i++) {
3511b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3512b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3513b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3514b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * i - 1) * L;
3515b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3516b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3517b7f5c055SJed Brown             }
3518b7f5c055SJed Brown           }
3519b7f5c055SJed Brown         }
3520b7f5c055SJed Brown       }
3521b7f5c055SJed Brown       // y-normal pipes
3522b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3523b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1] + 1; j++) {
3524b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3525b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3526b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3527b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * j - 1) * L;
3528b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3529b7f5c055SJed Brown             }
3530b7f5c055SJed Brown           }
3531b7f5c055SJed Brown         }
3532b7f5c055SJed Brown       }
3533b7f5c055SJed Brown       // z-normal pipes
3534b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3535b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3536b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2] + 1; k++) {
3537b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3538b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3539b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3540b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * k - 1) * L;
3541b7f5c055SJed Brown             }
3542b7f5c055SJed Brown           }
3543b7f5c055SJed Brown         }
3544b7f5c055SJed Brown       }
3545b7f5c055SJed Brown       // junctions
3546b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3547b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3548b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3549b7f5c055SJed Brown             const PetscInt J = (i * extent[1] + j) * extent[2] + k, Jvoff = (Npipes[0] + Npipes[1] + Npipes[2]) * 4 + J * 8;
3550b7f5c055SJed Brown             PetscCheck(vcount / 3 == Jvoff, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected vertex count");
3551b7f5c055SJed Brown             for (PetscInt ii = 0; ii < 2; ii++) {
3552b7f5c055SJed Brown               for (PetscInt jj = 0; jj < 2; jj++) {
3553b7f5c055SJed Brown                 for (PetscInt kk = 0; kk < 2; kk++) {
3554b7f5c055SJed Brown                   double Ls           = (1 - sqrt(2) / 4) * L;
3555b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * i * L + (2 * ii - 1) * Ls;
3556b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * j * L + (2 * jj - 1) * Ls;
3557b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * k * L + (2 * kk - 1) * Ls;
3558b7f5c055SJed Brown                 }
3559b7f5c055SJed Brown               }
3560b7f5c055SJed Brown             }
3561b7f5c055SJed Brown             const PetscInt jfaces[3][2][4] = {
3562b7f5c055SJed Brown               {{3, 1, 0, 2}, {7, 5, 4, 6}}, // x-aligned
3563b7f5c055SJed Brown               {{5, 4, 0, 1}, {7, 6, 2, 3}}, // y-aligned
3564b7f5c055SJed Brown               {{6, 2, 0, 4}, {7, 3, 1, 5}}  // z-aligned
3565b7f5c055SJed Brown             };
3566b7f5c055SJed Brown             const PetscInt pipe_lo[3] = {// vertex numbers of pipes
35679371c9d4SSatish 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};
3568b7f5c055SJed Brown             const PetscInt pipe_hi[3] = {// vertex numbers of pipes
35699371c9d4SSatish 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};
3570b7f5c055SJed Brown             for (PetscInt dir = 0; dir < 3; dir++) { // x,y,z
3571b7f5c055SJed Brown               const PetscInt ijk[3] = {i, j, k};
3572b7f5c055SJed Brown               for (PetscInt l = 0; l < 4; l++) { // rotations
3573b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][0] = pipe_lo[dir] + l;
3574b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][1] = Jvoff + jfaces[dir][0][l];
3575b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][2] = Jvoff + jfaces[dir][0][(l - 1 + 4) % 4];
3576b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][3] = pipe_lo[dir] + (l - 1 + 4) % 4;
3577b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][0] = Jvoff + jfaces[dir][1][l];
3578b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][1] = pipe_hi[dir] + l;
3579b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][2] = pipe_hi[dir] + (l - 1 + 4) % 4;
3580b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][3] = Jvoff + jfaces[dir][1][(l - 1 + 4) % 4];
3581b7f5c055SJed Brown                 if (ijk[dir] == 0) {
3582b7f5c055SJed Brown                   edges[numEdges][0] = pipe_lo[dir] + l;
3583b7f5c055SJed Brown                   edges[numEdges][1] = pipe_lo[dir] + (l + 1) % 4;
3584b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 1;
3585b7f5c055SJed Brown                   numEdges++;
3586b7f5c055SJed Brown                 }
3587b7f5c055SJed Brown                 if (ijk[dir] + 1 == extent[dir]) {
3588b7f5c055SJed Brown                   edges[numEdges][0] = pipe_hi[dir] + l;
3589b7f5c055SJed Brown                   edges[numEdges][1] = pipe_hi[dir] + (l + 1) % 4;
3590b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 2;
3591b7f5c055SJed Brown                   numEdges++;
3592b7f5c055SJed Brown                 }
3593b7f5c055SJed Brown               }
3594b7f5c055SJed Brown             }
3595b7f5c055SJed Brown           }
3596b7f5c055SJed Brown         }
3597b7f5c055SJed Brown       }
359863a3b9bcSJacob Faibussowitsch       PetscCheck(numEdges == Ncuts * 4, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Edge count %" PetscInt_FMT " incompatible with number of cuts %" PetscInt_FMT, numEdges, Ncuts);
3599b7f5c055SJed Brown       numFaces   = 24 * Njunctions;
3600b7f5c055SJed Brown       cells_flat = cells[0][0][0];
3601b7f5c055SJed Brown     }
3602b7f5c055SJed Brown     evalFunc   = TPSEvaluate_SchwarzP;
36034663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_SchwarzP;
3604b7f5c055SJed Brown     break;
3605b7f5c055SJed Brown   case DMPLEX_TPS_GYROID:
3606c5853193SPierre Jolivet     if (rank == 0) {
3607b7f5c055SJed Brown       // This is a coarse mesh approximation of the gyroid shifted to being the zero of the level set
3608b7f5c055SJed Brown       //
3609b7f5c055SJed 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)
3610b7f5c055SJed Brown       //
3611b7f5c055SJed Brown       // on the cell [0,2]^3.
3612b7f5c055SJed Brown       //
3613b7f5c055SJed Brown       // Think about dividing that cell into four columns, and focus on the column [0,1]x[0,1]x[0,2].
3614b7f5c055SJed Brown       // If you looked at the gyroid in that column at different slices of z you would see that it kind of spins
3615b7f5c055SJed Brown       // like a boomerang:
3616b7f5c055SJed Brown       //
3617b7f5c055SJed Brown       //     z = 0          z = 1/4        z = 1/2        z = 3/4     //
3618b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3619b7f5c055SJed Brown       //                                                              //
3620b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3621b7f5c055SJed Brown       //      \                                   /            \      //
3622b7f5c055SJed Brown       //       \            `-_   _-'            /              }     //
3623b7f5c055SJed Brown       //        *-_            `-'            _-'              /      //
3624b7f5c055SJed Brown       //     +     `-+      +       +      +-'     +      +   /   +   //
3625b7f5c055SJed Brown       //                                                              //
3626b7f5c055SJed Brown       //                                                              //
3627b7f5c055SJed Brown       //     z = 1          z = 5/4        z = 3/2        z = 7/4     //
3628b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3629b7f5c055SJed Brown       //                                                              //
3630b7f5c055SJed Brown       //     +-_     +      +       +      +     _-+      +   /   +   //
3631b7f5c055SJed Brown       //        `-_            _-_            _-`            /        //
3632b7f5c055SJed Brown       //           \        _-'   `-_        /              {         //
3633b7f5c055SJed Brown       //            \                       /                \        //
3634b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3635b7f5c055SJed Brown       //
3636b7f5c055SJed Brown       //
3637b7f5c055SJed Brown       // This course mesh approximates each of these slices by two line segments,
3638b7f5c055SJed Brown       // and then connects the segments in consecutive layers with quadrilateral faces.
3639b7f5c055SJed Brown       // All of the end points of the segments are multiples of 1/4 except for the
3640b7f5c055SJed Brown       // point * in the picture for z = 0 above and the similar points in other layers.
3641b7f5c055SJed Brown       // That point is at (gamma, gamma, 0), where gamma is calculated below.
3642b7f5c055SJed Brown       //
3643b7f5c055SJed Brown       // The column  [1,2]x[1,2]x[0,2] looks the same as this column;
3644b7f5c055SJed Brown       // The columns [1,2]x[0,1]x[0,2] and [0,1]x[1,2]x[0,2] are mirror images.
3645b7f5c055SJed Brown       //
3646b7f5c055SJed Brown       // As for how this method turned into the names given to the vertices:
3647b7f5c055SJed Brown       // that was not systematic, it was just the way it worked out in my handwritten notes.
3648b7f5c055SJed Brown 
3649b7f5c055SJed Brown       PetscInt facesPerBlock = 64;
3650b7f5c055SJed Brown       PetscInt vertsPerBlock = 56;
3651b7f5c055SJed Brown       PetscInt extentPlus[3];
3652b7f5c055SJed Brown       PetscInt numBlocks, numBlocksPlus;
36539371c9d4SSatish 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;
36549371c9d4SSatish Balay       const PetscInt pattern[64][4] = {
36559371c9d4SSatish Balay         /* face to vertex within the coarse discretization of a single gyroid block */
3656b7f5c055SJed Brown         /* layer 0 */
36579371c9d4SSatish Balay         {A,           C,           K,           G          },
36589371c9d4SSatish Balay         {C,           B,           II,          K          },
36599371c9d4SSatish Balay         {D,           A,           H,           L          },
36609371c9d4SSatish Balay         {B + 56 * 1,  D,           L,           J          },
36619371c9d4SSatish Balay         {E,           B + 56 * 1,  J,           N          },
36629371c9d4SSatish Balay         {A + 56 * 2,  E,           N,           H + 56 * 2 },
36639371c9d4SSatish Balay         {F,           A + 56 * 2,  G + 56 * 2,  M          },
36649371c9d4SSatish Balay         {B,           F,           M,           II         },
3665b7f5c055SJed Brown         /* layer 1 */
36669371c9d4SSatish Balay         {G,           K,           Q,           O          },
36679371c9d4SSatish Balay         {K,           II,          P,           Q          },
36689371c9d4SSatish Balay         {L,           H,           O + 56 * 1,  R          },
36699371c9d4SSatish Balay         {J,           L,           R,           P          },
36709371c9d4SSatish Balay         {N,           J,           P,           S          },
36719371c9d4SSatish Balay         {H + 56 * 2,  N,           S,           O + 56 * 3 },
36729371c9d4SSatish Balay         {M,           G + 56 * 2,  O + 56 * 2,  T          },
36739371c9d4SSatish Balay         {II,          M,           T,           P          },
3674b7f5c055SJed Brown         /* layer 2 */
36759371c9d4SSatish Balay         {O,           Q,           Y,           U          },
36769371c9d4SSatish Balay         {Q,           P,           W,           Y          },
36779371c9d4SSatish Balay         {R,           O + 56 * 1,  U + 56 * 1,  Ap         },
36789371c9d4SSatish Balay         {P,           R,           Ap,          W          },
36799371c9d4SSatish Balay         {S,           P,           X,           Bp         },
36809371c9d4SSatish Balay         {O + 56 * 3,  S,           Bp,          V + 56 * 1 },
36819371c9d4SSatish Balay         {T,           O + 56 * 2,  V,           Z          },
36829371c9d4SSatish Balay         {P,           T,           Z,           X          },
3683b7f5c055SJed Brown         /* layer 3 */
36849371c9d4SSatish Balay         {U,           Y,           Ep,          Dp         },
36859371c9d4SSatish Balay         {Y,           W,           Cp,          Ep         },
36869371c9d4SSatish Balay         {Ap,          U + 56 * 1,  Dp + 56 * 1, Gp         },
36879371c9d4SSatish Balay         {W,           Ap,          Gp,          Cp         },
36889371c9d4SSatish Balay         {Bp,          X,           Cp + 56 * 2, Fp         },
36899371c9d4SSatish Balay         {V + 56 * 1,  Bp,          Fp,          Dp + 56 * 1},
36909371c9d4SSatish Balay         {Z,           V,           Dp,          Hp         },
36919371c9d4SSatish Balay         {X,           Z,           Hp,          Cp + 56 * 2},
3692b7f5c055SJed Brown         /* layer 4 */
36939371c9d4SSatish Balay         {Dp,          Ep,          Mp,          Kp         },
36949371c9d4SSatish Balay         {Ep,          Cp,          Ip,          Mp         },
36959371c9d4SSatish Balay         {Gp,          Dp + 56 * 1, Lp,          Np         },
36969371c9d4SSatish Balay         {Cp,          Gp,          Np,          Jp         },
36979371c9d4SSatish Balay         {Fp,          Cp + 56 * 2, Jp + 56 * 2, Pp         },
36989371c9d4SSatish Balay         {Dp + 56 * 1, Fp,          Pp,          Lp         },
36999371c9d4SSatish Balay         {Hp,          Dp,          Kp,          Op         },
37009371c9d4SSatish Balay         {Cp + 56 * 2, Hp,          Op,          Ip + 56 * 2},
3701b7f5c055SJed Brown         /* layer 5 */
37029371c9d4SSatish Balay         {Kp,          Mp,          Sp,          Rp         },
37039371c9d4SSatish Balay         {Mp,          Ip,          Qp,          Sp         },
37049371c9d4SSatish Balay         {Np,          Lp,          Rp,          Tp         },
37059371c9d4SSatish Balay         {Jp,          Np,          Tp,          Qp + 56 * 1},
37069371c9d4SSatish Balay         {Pp,          Jp + 56 * 2, Qp + 56 * 3, Up         },
37079371c9d4SSatish Balay         {Lp,          Pp,          Up,          Rp         },
37089371c9d4SSatish Balay         {Op,          Kp,          Rp,          Vp         },
37099371c9d4SSatish Balay         {Ip + 56 * 2, Op,          Vp,          Qp + 56 * 2},
3710b7f5c055SJed Brown         /* layer 6 */
37119371c9d4SSatish Balay         {Rp,          Sp,          Aq,          Yp         },
37129371c9d4SSatish Balay         {Sp,          Qp,          Wp,          Aq         },
37139371c9d4SSatish Balay         {Tp,          Rp,          Yp,          Cq         },
37149371c9d4SSatish Balay         {Qp + 56 * 1, Tp,          Cq,          Wp + 56 * 1},
37159371c9d4SSatish Balay         {Up,          Qp + 56 * 3, Xp + 56 * 1, Dq         },
37169371c9d4SSatish Balay         {Rp,          Up,          Dq,          Zp         },
37179371c9d4SSatish Balay         {Vp,          Rp,          Zp,          Bq         },
37189371c9d4SSatish Balay         {Qp + 56 * 2, Vp,          Bq,          Xp         },
3719b7f5c055SJed Brown         /* layer 7 (the top is the periodic image of the bottom of layer 0) */
37209371c9d4SSatish Balay         {Yp,          Aq,          C + 56 * 4,  A + 56 * 4 },
37219371c9d4SSatish Balay         {Aq,          Wp,          B + 56 * 4,  C + 56 * 4 },
37229371c9d4SSatish Balay         {Cq,          Yp,          A + 56 * 4,  D + 56 * 4 },
37239371c9d4SSatish Balay         {Wp + 56 * 1, Cq,          D + 56 * 4,  B + 56 * 5 },
37249371c9d4SSatish Balay         {Dq,          Xp + 56 * 1, B + 56 * 5,  E + 56 * 4 },
37259371c9d4SSatish Balay         {Zp,          Dq,          E + 56 * 4,  A + 56 * 6 },
37269371c9d4SSatish Balay         {Bq,          Zp,          A + 56 * 6,  F + 56 * 4 },
37279371c9d4SSatish Balay         {Xp,          Bq,          F + 56 * 4,  B + 56 * 4 }
3728b7f5c055SJed Brown       };
3729b7f5c055SJed Brown       const PetscReal gamma                = PetscAcosReal((PetscSqrtReal(3.) - 1.) / PetscSqrtReal(2.)) / PETSC_PI;
37309371c9d4SSatish Balay       const PetscReal patternCoords[56][3] = {
3731bee3fc89SBarry Smith         {1.,        0.,        0.  }, /* A  */
3732bee3fc89SBarry Smith         {0.,        1.,        0.  }, /* B  */
3733bee3fc89SBarry Smith         {gamma,     gamma,     0.  }, /* C  */
3734bee3fc89SBarry Smith         {1 + gamma, 1 - gamma, 0.  }, /* D  */
3735bee3fc89SBarry Smith         {2 - gamma, 2 - gamma, 0.  }, /* E  */
3736bee3fc89SBarry Smith         {1 - gamma, 1 + gamma, 0.  }, /* F  */
3737b7f5c055SJed Brown 
3738bee3fc89SBarry Smith         {.5,        0,         .25 }, /* G  */
3739bee3fc89SBarry Smith         {1.5,       0.,        .25 }, /* H  */
3740bee3fc89SBarry Smith         {.5,        1.,        .25 }, /* II */
3741bee3fc89SBarry Smith         {1.5,       1.,        .25 }, /* J  */
3742bee3fc89SBarry Smith         {.25,       .5,        .25 }, /* K  */
3743bee3fc89SBarry Smith         {1.25,      .5,        .25 }, /* L  */
3744bee3fc89SBarry Smith         {.75,       1.5,       .25 }, /* M  */
3745bee3fc89SBarry Smith         {1.75,      1.5,       .25 }, /* N  */
3746b7f5c055SJed Brown 
3747bee3fc89SBarry Smith         {0.,        0.,        .5  }, /* O  */
3748bee3fc89SBarry Smith         {1.,        1.,        .5  }, /* P  */
3749bee3fc89SBarry Smith         {gamma,     1 - gamma, .5  }, /* Q  */
3750bee3fc89SBarry Smith         {1 + gamma, gamma,     .5  }, /* R  */
3751bee3fc89SBarry Smith         {2 - gamma, 1 + gamma, .5  }, /* S  */
3752bee3fc89SBarry Smith         {1 - gamma, 2 - gamma, .5  }, /* T  */
3753b7f5c055SJed Brown 
3754bee3fc89SBarry Smith         {0.,        .5,        .75 }, /* U  */
3755bee3fc89SBarry Smith         {0.,        1.5,       .75 }, /* V  */
3756bee3fc89SBarry Smith         {1.,        .5,        .75 }, /* W  */
3757bee3fc89SBarry Smith         {1.,        1.5,       .75 }, /* X  */
3758bee3fc89SBarry Smith         {.5,        .75,       .75 }, /* Y  */
3759bee3fc89SBarry Smith         {.5,        1.75,      .75 }, /* Z  */
3760bee3fc89SBarry Smith         {1.5,       .25,       .75 }, /* Ap */
3761bee3fc89SBarry Smith         {1.5,       1.25,      .75 }, /* Bp */
3762b7f5c055SJed Brown 
3763bee3fc89SBarry Smith         {1.,        0.,        1.  }, /* Cp */
3764bee3fc89SBarry Smith         {0.,        1.,        1.  }, /* Dp */
3765bee3fc89SBarry Smith         {1 - gamma, 1 - gamma, 1.  }, /* Ep */
3766bee3fc89SBarry Smith         {1 + gamma, 1 + gamma, 1.  }, /* Fp */
3767bee3fc89SBarry Smith         {2 - gamma, gamma,     1.  }, /* Gp */
3768bee3fc89SBarry Smith         {gamma,     2 - gamma, 1.  }, /* Hp */
3769b7f5c055SJed Brown 
3770bee3fc89SBarry Smith         {.5,        0.,        1.25}, /* Ip */
3771bee3fc89SBarry Smith         {1.5,       0.,        1.25}, /* Jp */
3772bee3fc89SBarry Smith         {.5,        1.,        1.25}, /* Kp */
3773bee3fc89SBarry Smith         {1.5,       1.,        1.25}, /* Lp */
3774bee3fc89SBarry Smith         {.75,       .5,        1.25}, /* Mp */
3775bee3fc89SBarry Smith         {1.75,      .5,        1.25}, /* Np */
3776bee3fc89SBarry Smith         {.25,       1.5,       1.25}, /* Op */
3777bee3fc89SBarry Smith         {1.25,      1.5,       1.25}, /* Pp */
3778b7f5c055SJed Brown 
3779bee3fc89SBarry Smith         {0.,        0.,        1.5 }, /* Qp */
3780bee3fc89SBarry Smith         {1.,        1.,        1.5 }, /* Rp */
3781bee3fc89SBarry Smith         {1 - gamma, gamma,     1.5 }, /* Sp */
3782bee3fc89SBarry Smith         {2 - gamma, 1 - gamma, 1.5 }, /* Tp */
3783bee3fc89SBarry Smith         {1 + gamma, 2 - gamma, 1.5 }, /* Up */
3784bee3fc89SBarry Smith         {gamma,     1 + gamma, 1.5 }, /* Vp */
3785b7f5c055SJed Brown 
3786bee3fc89SBarry Smith         {0.,        .5,        1.75}, /* Wp */
3787bee3fc89SBarry Smith         {0.,        1.5,       1.75}, /* Xp */
3788bee3fc89SBarry Smith         {1.,        .5,        1.75}, /* Yp */
3789bee3fc89SBarry Smith         {1.,        1.5,       1.75}, /* Zp */
3790bee3fc89SBarry Smith         {.5,        .25,       1.75}, /* Aq */
3791bee3fc89SBarry Smith         {.5,        1.25,      1.75}, /* Bq */
3792bee3fc89SBarry Smith         {1.5,       .75,       1.75}, /* Cq */
3793bee3fc89SBarry Smith         {1.5,       1.75,      1.75}, /* Dq */
3794b7f5c055SJed Brown       };
3795b7f5c055SJed Brown       PetscInt(*cells)[64][4] = NULL;
3796b7f5c055SJed Brown       PetscBool *seen;
3797b7f5c055SJed Brown       PetscInt  *vertToTrueVert;
3798b7f5c055SJed Brown       PetscInt   count;
3799b7f5c055SJed Brown 
3800b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) extentPlus[i] = extent[i] + 1;
3801b7f5c055SJed Brown       numBlocks = 1;
3802b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocks *= extent[i];
3803b7f5c055SJed Brown       numBlocksPlus = 1;
3804b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocksPlus *= extentPlus[i];
3805b7f5c055SJed Brown       numFaces = numBlocks * facesPerBlock;
38069566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocks, &cells));
38079566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numBlocksPlus * vertsPerBlock, &seen));
3808b7f5c055SJed Brown       for (PetscInt k = 0; k < extent[2]; k++) {
3809b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3810b7f5c055SJed Brown           for (PetscInt i = 0; i < extent[0]; i++) {
3811b7f5c055SJed Brown             for (PetscInt f = 0; f < facesPerBlock; f++) {
3812b7f5c055SJed Brown               for (PetscInt v = 0; v < 4; v++) {
3813b7f5c055SJed Brown                 PetscInt vertRaw     = pattern[f][v];
3814b7f5c055SJed Brown                 PetscInt blockidx    = vertRaw / 56;
3815b7f5c055SJed Brown                 PetscInt patternvert = vertRaw % 56;
3816b7f5c055SJed Brown                 PetscInt xplus       = (blockidx & 1);
3817b7f5c055SJed Brown                 PetscInt yplus       = (blockidx & 2) >> 1;
3818b7f5c055SJed Brown                 PetscInt zplus       = (blockidx & 4) >> 2;
3819b7f5c055SJed Brown                 PetscInt zcoord      = (periodic && periodic[2] == DM_BOUNDARY_PERIODIC) ? ((k + zplus) % extent[2]) : (k + zplus);
3820b7f5c055SJed Brown                 PetscInt ycoord      = (periodic && periodic[1] == DM_BOUNDARY_PERIODIC) ? ((j + yplus) % extent[1]) : (j + yplus);
3821b7f5c055SJed Brown                 PetscInt xcoord      = (periodic && periodic[0] == DM_BOUNDARY_PERIODIC) ? ((i + xplus) % extent[0]) : (i + xplus);
3822b7f5c055SJed Brown                 PetscInt vert        = ((zcoord * extentPlus[1] + ycoord) * extentPlus[0] + xcoord) * 56 + patternvert;
3823b7f5c055SJed Brown 
3824b7f5c055SJed Brown                 cells[(k * extent[1] + j) * extent[0] + i][f][v] = vert;
3825b7f5c055SJed Brown                 seen[vert]                                       = PETSC_TRUE;
3826b7f5c055SJed Brown               }
3827b7f5c055SJed Brown             }
3828b7f5c055SJed Brown           }
3829b7f5c055SJed Brown         }
3830b7f5c055SJed Brown       }
38319371c9d4SSatish Balay       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++)
38329371c9d4SSatish Balay         if (seen[i]) numVertices++;
3833b7f5c055SJed Brown       count = 0;
38349566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocksPlus * vertsPerBlock, &vertToTrueVert));
38359566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * 3, &vtxCoords));
3836b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++) vertToTrueVert[i] = -1;
3837b7f5c055SJed Brown       for (PetscInt k = 0; k < extentPlus[2]; k++) {
3838b7f5c055SJed Brown         for (PetscInt j = 0; j < extentPlus[1]; j++) {
3839b7f5c055SJed Brown           for (PetscInt i = 0; i < extentPlus[0]; i++) {
3840b7f5c055SJed Brown             for (PetscInt v = 0; v < vertsPerBlock; v++) {
3841b7f5c055SJed Brown               PetscInt vIdx = ((k * extentPlus[1] + j) * extentPlus[0] + i) * vertsPerBlock + v;
3842b7f5c055SJed Brown 
3843b7f5c055SJed Brown               if (seen[vIdx]) {
3844b7f5c055SJed Brown                 PetscInt thisVert;
3845b7f5c055SJed Brown 
3846b7f5c055SJed Brown                 vertToTrueVert[vIdx] = thisVert = count++;
3847b7f5c055SJed Brown 
3848b7f5c055SJed Brown                 for (PetscInt d = 0; d < 3; d++) vtxCoords[3 * thisVert + d] = patternCoords[v][d];
3849b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 0] += i * 2;
3850b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 1] += j * 2;
3851b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 2] += k * 2;
3852b7f5c055SJed Brown               }
3853b7f5c055SJed Brown             }
3854b7f5c055SJed Brown           }
3855b7f5c055SJed Brown         }
3856b7f5c055SJed Brown       }
3857b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocks; i++) {
3858b7f5c055SJed Brown         for (PetscInt f = 0; f < facesPerBlock; f++) {
3859ad540459SPierre Jolivet           for (PetscInt v = 0; v < 4; v++) cells[i][f][v] = vertToTrueVert[cells[i][f][v]];
3860b7f5c055SJed Brown         }
3861b7f5c055SJed Brown       }
38629566063dSJacob Faibussowitsch       PetscCall(PetscFree(vertToTrueVert));
38639566063dSJacob Faibussowitsch       PetscCall(PetscFree(seen));
3864b7f5c055SJed Brown       cells_flat = cells[0][0];
3865b7f5c055SJed Brown       numEdges   = 0;
3866b7f5c055SJed Brown       for (PetscInt i = 0; i < numFaces; i++) {
3867b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3868b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3869b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3870b7f5c055SJed Brown 
3871b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3872b7f5c055SJed Brown             if (!periodic || periodic[0] != DM_BOUNDARY_PERIODIC) {
3873b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) numEdges++;
3874b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) numEdges++;
3875b7f5c055SJed Brown             }
3876b7f5c055SJed Brown           }
3877b7f5c055SJed Brown         }
3878b7f5c055SJed Brown       }
38799566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edges));
38809566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edgeSets));
3881b7f5c055SJed Brown       for (PetscInt edge = 0, i = 0; i < numFaces; i++) {
3882b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3883b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3884b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3885b7f5c055SJed Brown 
3886b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3887b7f5c055SJed Brown             if (!periodic || periodic[d] != DM_BOUNDARY_PERIODIC) {
3888b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) {
3889b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3890b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3891b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d;
3892b7f5c055SJed Brown               }
3893b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) {
3894b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3895b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3896b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d + 1;
3897b7f5c055SJed Brown               }
3898b7f5c055SJed Brown             }
3899b7f5c055SJed Brown           }
3900b7f5c055SJed Brown         }
3901b7f5c055SJed Brown       }
3902b7f5c055SJed Brown     }
3903b7f5c055SJed Brown     evalFunc   = TPSEvaluate_Gyroid;
39044663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_Gyroid;
3905b7f5c055SJed Brown     break;
3906b7f5c055SJed Brown   }
3907b7f5c055SJed Brown 
39089566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, topoDim));
3909c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(dm, numFaces, numVertices, 4, cells_flat));
39109566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(dm, 0, 0, 0, NULL));
39119566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells_flat));
3912b7f5c055SJed Brown   {
3913b7f5c055SJed Brown     DM idm;
39149566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(dm, &idm));
391569d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &idm));
3916b7f5c055SJed Brown   }
3917c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, vtxCoords));
39189566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, NULL));
39199566063dSJacob Faibussowitsch   PetscCall(PetscFree(vtxCoords));
3920b7f5c055SJed Brown 
39219566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
39229566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
3923b7f5c055SJed Brown   for (PetscInt e = 0; e < numEdges; e++) {
3924b7f5c055SJed Brown     PetscInt        njoin;
3925b7f5c055SJed Brown     const PetscInt *join, verts[] = {numFaces + edges[e][0], numFaces + edges[e][1]};
39269566063dSJacob Faibussowitsch     PetscCall(DMPlexGetJoin(dm, 2, verts, &njoin, &join));
392763a3b9bcSJacob 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]);
39289566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(label, join[0], edgeSets[e]));
39299566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(dm, 2, verts, &njoin, &join));
3930b7f5c055SJed Brown   }
39319566063dSJacob Faibussowitsch   PetscCall(PetscFree(edges));
39329566063dSJacob Faibussowitsch   PetscCall(PetscFree(edgeSets));
39331436d7faSJed Brown   if (tps_distribute) {
39341436d7faSJed Brown     DM               pdm = NULL;
39351436d7faSJed Brown     PetscPartitioner part;
39361436d7faSJed Brown 
39379566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
39389566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
39399566063dSJacob Faibussowitsch     PetscCall(DMPlexDistribute(dm, 0, NULL, &pdm));
394048a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
39411436d7faSJed Brown     // Do not auto-distribute again
39429566063dSJacob Faibussowitsch     PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
39431436d7faSJed Brown   }
3944b7f5c055SJed Brown 
39459566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
3946b7f5c055SJed Brown   for (PetscInt refine = 0; refine < refinements; refine++) {
3947b7f5c055SJed Brown     PetscInt     m;
3948b7f5c055SJed Brown     DM           dmf;
3949b7f5c055SJed Brown     Vec          X;
3950b7f5c055SJed Brown     PetscScalar *x;
39519566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, MPI_COMM_NULL, &dmf));
395269d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmf));
3953b7f5c055SJed Brown 
39549566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &X));
39559566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(X, &m));
39569566063dSJacob Faibussowitsch     PetscCall(VecGetArray(X, &x));
395748a46eb9SPierre Jolivet     for (PetscInt i = 0; i < m; i += 3) PetscCall(TPSNearestPoint(evalFunc, &x[i]));
39589566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(X, &x));
3959b7f5c055SJed Brown   }
3960b7f5c055SJed Brown 
3961b7f5c055SJed Brown   // Face Sets has already been propagated to new vertices during refinement; this propagates to the initial vertices.
39629566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
39639566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, label));
3964b7f5c055SJed Brown 
396546139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
396646139095SJed Brown 
3967b7f5c055SJed Brown   if (thickness > 0) {
39684663dae6SJed Brown     DM              edm, cdm, ecdm;
39694663dae6SJed Brown     DMPlexTransform tr;
39704663dae6SJed Brown     const char     *prefix;
39714663dae6SJed Brown     PetscOptions    options;
39724663dae6SJed Brown     // Code from DMPlexExtrude
39734663dae6SJed Brown     PetscCall(DMPlexTransformCreate(PetscObjectComm((PetscObject)dm), &tr));
39744663dae6SJed Brown     PetscCall(DMPlexTransformSetDM(tr, dm));
39754663dae6SJed Brown     PetscCall(DMPlexTransformSetType(tr, DMPLEXEXTRUDE));
39764663dae6SJed Brown     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
39774663dae6SJed Brown     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tr, prefix));
39784663dae6SJed Brown     PetscCall(PetscObjectGetOptions((PetscObject)dm, &options));
39794663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, options));
39804663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetLayers(tr, layers));
39814663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetThickness(tr, thickness));
39824663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetTensor(tr, PETSC_FALSE));
39834663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetSymmetric(tr, PETSC_TRUE));
39844663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetNormalFunction(tr, normalFunc));
39854663dae6SJed Brown     PetscCall(DMPlexTransformSetFromOptions(tr));
39864663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, NULL));
39874663dae6SJed Brown     PetscCall(DMPlexTransformSetUp(tr));
39884663dae6SJed Brown     PetscCall(PetscObjectViewFromOptions((PetscObject)tr, NULL, "-dm_plex_tps_transform_view"));
39894663dae6SJed Brown     PetscCall(DMPlexTransformApply(tr, dm, &edm));
39904663dae6SJed Brown     PetscCall(DMCopyDisc(dm, edm));
39914663dae6SJed Brown     PetscCall(DMGetCoordinateDM(dm, &cdm));
39924663dae6SJed Brown     PetscCall(DMGetCoordinateDM(edm, &ecdm));
39934663dae6SJed Brown     PetscCall(DMCopyDisc(cdm, ecdm));
39944663dae6SJed Brown     PetscCall(DMPlexTransformCreateDiscLabels(tr, edm));
39954663dae6SJed Brown     PetscCall(DMPlexTransformDestroy(&tr));
3996a77a5016SMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, edm));
399769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
3998b7f5c055SJed Brown   }
39993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4000b7f5c055SJed Brown }
4001b7f5c055SJed Brown 
4002b7f5c055SJed Brown /*@
4003b7f5c055SJed Brown   DMPlexCreateTPSMesh - Create a distributed, interpolated mesh of a triply-periodic surface
4004b7f5c055SJed Brown 
4005b7f5c055SJed Brown   Collective
4006b7f5c055SJed Brown 
4007b7f5c055SJed Brown   Input Parameters:
4008a1cb98faSBarry Smith + comm           - The communicator for the `DM` object
4009b7f5c055SJed Brown . tpstype        - Type of triply-periodic surface
4010b7f5c055SJed Brown . extent         - Array of length 3 containing number of periods in each direction
401120f4b53cSBarry Smith . periodic       - array of length 3 with periodicity, or `NULL` for non-periodic
40121436d7faSJed Brown . tps_distribute - Distribute 2D manifold mesh prior to refinement and extrusion (more scalable)
4013817da375SSatish Balay . refinements    - Number of factor-of-2 refinements of 2D manifold mesh
40141436d7faSJed Brown . layers         - Number of cell layers extruded in normal direction
4015817da375SSatish Balay - thickness      - Thickness in normal direction
4016b7f5c055SJed Brown 
4017b7f5c055SJed Brown   Output Parameter:
4018a1cb98faSBarry Smith . dm - The `DM` object
4019a1cb98faSBarry Smith 
4020a1cb98faSBarry Smith   Level: beginner
4021b7f5c055SJed Brown 
4022b7f5c055SJed Brown   Notes:
402315229ffcSPierre Jolivet   This meshes the surface of the Schwarz P or Gyroid surfaces.  Schwarz P is the simplest member of the triply-periodic minimal surfaces.
40241d27aa22SBarry Smith   <https://en.wikipedia.org/wiki/Schwarz_minimal_surface#Schwarz_P_(%22Primitive%22)> and can be cut with "clean" boundaries.
40251d27aa22SBarry 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.
4026b7f5c055SJed Brown   Our implementation creates a very coarse mesh of the surface and refines (by 4-way splitting) as many times as requested.
4027b7f5c055SJed Brown   On each refinement, all vertices are projected to their nearest point on the surface.
4028b7f5c055SJed Brown   This projection could readily be extended to related surfaces.
4029b7f5c055SJed Brown 
40301d27aa22SBarry Smith   See {cite}`maskery2018insights`
40311d27aa22SBarry Smith 
40321d27aa22SBarry Smith   The face (edge) sets for the Schwarz P surface are numbered $1(-x), 2(+x), 3(-y), 4(+y), 5(-z), 6(+z)$.
40331d27aa22SBarry Smith   When the mesh is refined, "Face Sets" contain the new vertices (created during refinement).
40341d27aa22SBarry Smith   Use `DMPlexLabelComplete()` to propagate to coarse-level vertices.
4035b7f5c055SJed Brown 
403660225df5SJacob Faibussowitsch   Developer Notes:
4037b7f5c055SJed Brown   The Gyroid mesh does not currently mark boundary sets.
4038b7f5c055SJed Brown 
40391cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMSetType()`, `DMCreate()`
4040b7f5c055SJed Brown @*/
4041d71ae5a4SJacob 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)
4042d71ae5a4SJacob Faibussowitsch {
4043b7f5c055SJed Brown   PetscFunctionBegin;
40449566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
40459566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
40469566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateTPSMesh_Internal(*dm, tpstype, extent, periodic, tps_distribute, refinements, layers, thickness));
40473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4048b7f5c055SJed Brown }
4049b7f5c055SJed Brown 
40509318fe57SMatthew G. Knepley /*@
40519318fe57SMatthew G. Knepley   DMPlexCreateSphereMesh - Creates a mesh on the d-dimensional sphere, S^d.
40529318fe57SMatthew G. Knepley 
40539318fe57SMatthew G. Knepley   Collective
40549318fe57SMatthew G. Knepley 
40559318fe57SMatthew G. Knepley   Input Parameters:
4056a1cb98faSBarry Smith + comm    - The communicator for the `DM` object
40579318fe57SMatthew G. Knepley . dim     - The dimension
40589318fe57SMatthew G. Knepley . simplex - Use simplices, or tensor product cells
40599318fe57SMatthew G. Knepley - R       - The radius
40609318fe57SMatthew G. Knepley 
40619318fe57SMatthew G. Knepley   Output Parameter:
4062a1cb98faSBarry Smith . dm - The `DM` object
40639318fe57SMatthew G. Knepley 
40649318fe57SMatthew G. Knepley   Level: beginner
40659318fe57SMatthew G. Knepley 
40661cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBallMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
40679318fe57SMatthew G. Knepley @*/
4068d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateSphereMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscReal R, DM *dm)
4069d71ae5a4SJacob Faibussowitsch {
40709318fe57SMatthew G. Knepley   PetscFunctionBegin;
40714f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
40729566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
40739566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
40749566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(*dm, dim, simplex, R));
40753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40769318fe57SMatthew G. Knepley }
40779318fe57SMatthew G. Knepley 
4078d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBallMesh_Internal(DM dm, PetscInt dim, PetscReal R)
4079d71ae5a4SJacob Faibussowitsch {
40809318fe57SMatthew G. Knepley   DM          sdm, vol;
40819318fe57SMatthew G. Knepley   DMLabel     bdlabel;
4082dd2b43ebSStefano Zampini   const char *prefix;
40839318fe57SMatthew G. Knepley 
40849318fe57SMatthew G. Knepley   PetscFunctionBegin;
40859566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &sdm));
40869566063dSJacob Faibussowitsch   PetscCall(DMSetType(sdm, DMPLEX));
4087dd2b43ebSStefano Zampini   PetscCall(DMGetOptionsPrefix(dm, &prefix));
4088dd2b43ebSStefano Zampini   PetscCall(DMSetOptionsPrefix(sdm, prefix));
4089dd2b43ebSStefano Zampini   PetscCall(DMAppendOptionsPrefix(sdm, "bd_"));
4090dd2b43ebSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(sdm, PETSC_FALSE));
40919566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(sdm, dim - 1, PETSC_TRUE, R));
40929566063dSJacob Faibussowitsch   PetscCall(DMSetFromOptions(sdm));
40939566063dSJacob Faibussowitsch   PetscCall(DMViewFromOptions(sdm, NULL, "-dm_view"));
40949566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(sdm, NULL, PETSC_TRUE, &vol));
40959566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&sdm));
409669d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
40979566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
40989566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
40999566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
41009566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, bdlabel));
41013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
410251a74b61SMatthew G. Knepley }
410351a74b61SMatthew G. Knepley 
410451a74b61SMatthew G. Knepley /*@
410551a74b61SMatthew G. Knepley   DMPlexCreateBallMesh - Creates a simplex mesh on the d-dimensional ball, B^d.
410651a74b61SMatthew G. Knepley 
410751a74b61SMatthew G. Knepley   Collective
410851a74b61SMatthew G. Knepley 
410951a74b61SMatthew G. Knepley   Input Parameters:
4110a1cb98faSBarry Smith + comm - The communicator for the `DM` object
411151a74b61SMatthew G. Knepley . dim  - The dimension
411251a74b61SMatthew G. Knepley - R    - The radius
411351a74b61SMatthew G. Knepley 
411451a74b61SMatthew G. Knepley   Output Parameter:
4115a1cb98faSBarry Smith . dm - The `DM` object
411651a74b61SMatthew G. Knepley 
4117a1cb98faSBarry Smith   Options Database Key:
411860225df5SJacob Faibussowitsch . bd_dm_refine - This will refine the surface mesh preserving the sphere geometry
411951a74b61SMatthew G. Knepley 
412051a74b61SMatthew G. Knepley   Level: beginner
412151a74b61SMatthew G. Knepley 
41221cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
412351a74b61SMatthew G. Knepley @*/
4124d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBallMesh(MPI_Comm comm, PetscInt dim, PetscReal R, DM *dm)
4125d71ae5a4SJacob Faibussowitsch {
412651a74b61SMatthew G. Knepley   PetscFunctionBegin;
41279566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
41289566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
41299566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBallMesh_Internal(*dm, dim, R));
41303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
41312829fed8SMatthew G. Knepley }
41322829fed8SMatthew G. Knepley 
4133d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateReferenceCell_Internal(DM rdm, DMPolytopeType ct)
4134d71ae5a4SJacob Faibussowitsch {
41350a6ba040SMatthew G. Knepley   PetscFunctionBegin;
41369318fe57SMatthew G. Knepley   switch (ct) {
41379371c9d4SSatish Balay   case DM_POLYTOPE_POINT: {
41389318fe57SMatthew G. Knepley     PetscInt    numPoints[1]        = {1};
41399318fe57SMatthew G. Knepley     PetscInt    coneSize[1]         = {0};
41409318fe57SMatthew G. Knepley     PetscInt    cones[1]            = {0};
41419318fe57SMatthew G. Knepley     PetscInt    coneOrientations[1] = {0};
41429318fe57SMatthew G. Knepley     PetscScalar vertexCoords[1]     = {0.0};
41439318fe57SMatthew G. Knepley 
41449566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 0));
41459566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41469371c9d4SSatish Balay   } break;
41479371c9d4SSatish Balay   case DM_POLYTOPE_SEGMENT: {
41489318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
41499318fe57SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
41509318fe57SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
41519318fe57SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
41529318fe57SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
41539318fe57SMatthew G. Knepley 
41549566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
41559566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41569371c9d4SSatish Balay   } break;
41579371c9d4SSatish Balay   case DM_POLYTOPE_POINT_PRISM_TENSOR: {
4158b5a892a1SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
4159b5a892a1SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
4160b5a892a1SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
4161b5a892a1SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
4162b5a892a1SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
4163b5a892a1SMatthew G. Knepley 
41649566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
41659566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41669371c9d4SSatish Balay   } break;
41679371c9d4SSatish Balay   case DM_POLYTOPE_TRIANGLE: {
41689318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {3, 1};
41699318fe57SMatthew G. Knepley     PetscInt    coneSize[4]         = {3, 0, 0, 0};
41709318fe57SMatthew G. Knepley     PetscInt    cones[3]            = {1, 2, 3};
41719318fe57SMatthew G. Knepley     PetscInt    coneOrientations[3] = {0, 0, 0};
41729318fe57SMatthew G. Knepley     PetscScalar vertexCoords[6]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0};
41739318fe57SMatthew G. Knepley 
41749566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
41759566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41769371c9d4SSatish Balay   } break;
41779371c9d4SSatish Balay   case DM_POLYTOPE_QUADRILATERAL: {
41789318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
41799318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
41809318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
41819318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
41829318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0};
41839318fe57SMatthew G. Knepley 
41849566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
41859566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41869371c9d4SSatish Balay   } break;
41879371c9d4SSatish Balay   case DM_POLYTOPE_SEG_PRISM_TENSOR: {
41889318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
41899318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
41909318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
41919318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
41929318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0};
41939318fe57SMatthew G. Knepley 
41949566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
41959566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
41969371c9d4SSatish Balay   } break;
41979371c9d4SSatish Balay   case DM_POLYTOPE_TETRAHEDRON: {
41989318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
41999318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
4200f0edb160SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
42019318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
4202f0edb160SMatthew 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};
42039318fe57SMatthew G. Knepley 
42049566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42059566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42069371c9d4SSatish Balay   } break;
42079371c9d4SSatish Balay   case DM_POLYTOPE_HEXAHEDRON: {
42089318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
42099318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
4210f0edb160SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
42119318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
42129371c9d4SSatish 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};
42139318fe57SMatthew G. Knepley 
42149566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42159566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42169371c9d4SSatish Balay   } break;
42179371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM: {
42189318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
42199318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
4220f0edb160SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
42219318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
42229371c9d4SSatish 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};
42239318fe57SMatthew G. Knepley 
42249566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42259566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42269371c9d4SSatish Balay   } break;
42279371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM_TENSOR: {
42289318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
42299318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
42309318fe57SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
42319318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
42329371c9d4SSatish 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};
42339318fe57SMatthew G. Knepley 
42349566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42359566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42369371c9d4SSatish Balay   } break;
42379371c9d4SSatish Balay   case DM_POLYTOPE_QUAD_PRISM_TENSOR: {
42389318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
42399318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
42409318fe57SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
42419318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
42429371c9d4SSatish 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};
42439318fe57SMatthew G. Knepley 
42449566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42459566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42469371c9d4SSatish Balay   } break;
42479371c9d4SSatish Balay   case DM_POLYTOPE_PYRAMID: {
42489318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {5, 1};
42499318fe57SMatthew G. Knepley     PetscInt    coneSize[6]         = {5, 0, 0, 0, 0, 0};
4250f0edb160SMatthew G. Knepley     PetscInt    cones[5]            = {1, 2, 3, 4, 5};
42519318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
42529371c9d4SSatish 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};
42539318fe57SMatthew G. Knepley 
42549566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
42559566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
42569371c9d4SSatish Balay   } break;
4257d71ae5a4SJacob Faibussowitsch   default:
4258d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)rdm), PETSC_ERR_ARG_WRONG, "Cannot create reference cell for cell type %s", DMPolytopeTypes[ct]);
42599318fe57SMatthew G. Knepley   }
42609318fe57SMatthew G. Knepley   {
42619318fe57SMatthew G. Knepley     PetscInt Nv, v;
42629318fe57SMatthew G. Knepley 
42639318fe57SMatthew G. Knepley     /* Must create the celltype label here so that we do not automatically try to compute the types */
42649566063dSJacob Faibussowitsch     PetscCall(DMCreateLabel(rdm, "celltype"));
42659566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCellType(rdm, 0, ct));
42669566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(rdm, NULL, &Nv));
42679566063dSJacob Faibussowitsch     for (v = 1; v < Nv; ++v) PetscCall(DMPlexSetCellType(rdm, v, DM_POLYTOPE_POINT));
42689318fe57SMatthew G. Knepley   }
42699566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(rdm));
42709566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)rdm, DMPolytopeTypes[ct]));
42713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
42720a6ba040SMatthew G. Knepley }
42730a6ba040SMatthew G. Knepley 
42749318fe57SMatthew G. Knepley /*@
4275a1cb98faSBarry Smith   DMPlexCreateReferenceCell - Create a `DMPLEX` with the appropriate FEM reference cell
42769318fe57SMatthew G. Knepley 
42779318fe57SMatthew G. Knepley   Collective
42789318fe57SMatthew G. Knepley 
42799318fe57SMatthew G. Knepley   Input Parameters:
42809318fe57SMatthew G. Knepley + comm - The communicator
42819318fe57SMatthew G. Knepley - ct   - The cell type of the reference cell
42829318fe57SMatthew G. Knepley 
42839318fe57SMatthew G. Knepley   Output Parameter:
42849318fe57SMatthew G. Knepley . refdm - The reference cell
42859318fe57SMatthew G. Knepley 
42869318fe57SMatthew G. Knepley   Level: intermediate
42879318fe57SMatthew G. Knepley 
428842747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`
42899318fe57SMatthew G. Knepley @*/
4290d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, DMPolytopeType ct, DM *refdm)
4291d71ae5a4SJacob Faibussowitsch {
42920a6ba040SMatthew G. Knepley   PetscFunctionBegin;
42939566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, refdm));
42949566063dSJacob Faibussowitsch   PetscCall(DMSetType(*refdm, DMPLEX));
42959566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceCell_Internal(*refdm, ct));
42963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
42979318fe57SMatthew G. Knepley }
429879a015ccSMatthew G. Knepley 
4299d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoundaryLabel_Private(DM dm, const char name[])
4300d71ae5a4SJacob Faibussowitsch {
43019318fe57SMatthew G. Knepley   DM        plex;
43029318fe57SMatthew G. Knepley   DMLabel   label;
43039318fe57SMatthew G. Knepley   PetscBool hasLabel;
43040a6ba040SMatthew G. Knepley 
4305c22d3578SMatthew G. Knepley   PetscFunctionBegin;
43069566063dSJacob Faibussowitsch   PetscCall(DMHasLabel(dm, name, &hasLabel));
43073ba16761SJacob Faibussowitsch   if (hasLabel) PetscFunctionReturn(PETSC_SUCCESS);
43089566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, name));
43099566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, name, &label));
43109566063dSJacob Faibussowitsch   PetscCall(DMConvert(dm, DMPLEX, &plex));
43119566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(plex, 1, label));
43121c8afea9SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(plex, label));
43139566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&plex));
43143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43159318fe57SMatthew G. Knepley }
4316acdc6f61SToby Isaac 
4317669647acSMatthew G. Knepley /*
4318669647acSMatthew 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.
4319669647acSMatthew G. Knepley 
4320669647acSMatthew G. Knepley     (x, y) -> (r, theta) = (x[1], (x[0] - lower[0]) * 2\pi/(upper[0] - lower[0]))
4321669647acSMatthew G. Knepley */
4322d71ae5a4SJacob 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[])
4323d71ae5a4SJacob Faibussowitsch {
4324669647acSMatthew G. Knepley   const PetscReal low = PetscRealPart(constants[0]);
4325669647acSMatthew G. Knepley   const PetscReal upp = PetscRealPart(constants[1]);
4326669647acSMatthew G. Knepley   const PetscReal r   = PetscRealPart(u[1]);
4327669647acSMatthew G. Knepley   const PetscReal th  = 2. * PETSC_PI * (PetscRealPart(u[0]) - low) / (upp - low);
4328669647acSMatthew G. Knepley 
4329669647acSMatthew G. Knepley   f0[0] = r * PetscCosReal(th);
4330669647acSMatthew G. Knepley   f0[1] = r * PetscSinReal(th);
4331669647acSMatthew G. Knepley }
4332669647acSMatthew G. Knepley 
43335390be7dSMatthew G. Knepley // Insert vertices and their joins, marked by depth
43345390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Vertices(DM dm, DMLabel label, DMLabel vlabel, PetscInt val, PetscInt n, const PetscInt vertices[])
43355390be7dSMatthew G. Knepley {
43365390be7dSMatthew G. Knepley   PetscFunctionBegin;
43375390be7dSMatthew G. Knepley   PetscCall(DMPlexMarkSubmesh_Interpolated(dm, vlabel, val, PETSC_FALSE, PETSC_FALSE, label, NULL));
43385390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
43395390be7dSMatthew G. Knepley }
43405390be7dSMatthew G. Knepley 
43415390be7dSMatthew G. Knepley // Insert faces and their closures, marked by depth
43425390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Faces(DM dm, DMLabel label, PetscInt n, const PetscInt faces[])
43435390be7dSMatthew G. Knepley {
43445390be7dSMatthew G. Knepley   PetscFunctionBegin;
43455390be7dSMatthew G. Knepley   for (PetscInt p = 0; p < n; ++p) {
43465390be7dSMatthew G. Knepley     const PetscInt point   = faces[p];
43475390be7dSMatthew G. Knepley     PetscInt      *closure = NULL;
43485390be7dSMatthew G. Knepley     PetscInt       clSize, pdepth;
43495390be7dSMatthew G. Knepley 
43505390be7dSMatthew G. Knepley     PetscCall(DMPlexGetPointDepth(dm, point, &pdepth));
43515390be7dSMatthew G. Knepley     PetscCall(DMLabelSetValue(label, point, pdepth));
43525390be7dSMatthew G. Knepley     PetscCall(DMPlexGetTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
43535390be7dSMatthew G. Knepley     for (PetscInt cl = 0; cl < clSize * 2; cl += 2) {
43545390be7dSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, closure[cl], &pdepth));
43555390be7dSMatthew G. Knepley       PetscCall(DMLabelSetValue(label, closure[cl], pdepth));
43565390be7dSMatthew G. Knepley     }
43575390be7dSMatthew G. Knepley     PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
43585390be7dSMatthew G. Knepley   }
43595390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
43605390be7dSMatthew G. Knepley }
43615390be7dSMatthew G. Knepley 
43624e22dd4cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions, const char pre[], const char name[], const char *option[], const char *value[], PetscBool *flg);
43634e22dd4cSMatthew G. Knepley 
43645dca41c3SJed Brown const char *const DMPlexShapes[] = {"box", "box_surface", "ball", "sphere", "cylinder", "schwarz_p", "gyroid", "doublet", "annulus", "hypercubic", "zbox", "unknown", "DMPlexShape", "DM_SHAPE_", NULL};
43659318fe57SMatthew G. Knepley 
4366d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateFromOptions_Internal(PetscOptionItems *PetscOptionsObject, PetscBool *useCoordSpace, DM dm)
4367d71ae5a4SJacob Faibussowitsch {
43689318fe57SMatthew G. Knepley   DMPlexShape    shape   = DM_SHAPE_BOX;
43699318fe57SMatthew G. Knepley   DMPolytopeType cell    = DM_POLYTOPE_TRIANGLE;
43709318fe57SMatthew G. Knepley   PetscInt       dim     = 2;
4371b9da1bb3SMatthew G. Knepley   PetscBool      simplex = PETSC_TRUE, interpolate = PETSC_TRUE, orient = PETSC_FALSE, adjCone = PETSC_FALSE, adjClosure = PETSC_TRUE, refDomain = PETSC_FALSE;
4372d0812dedSMatthew G. Knepley   PetscBool      flg, flg2, fflg, strflg, bdfflg, nameflg;
43739318fe57SMatthew G. Knepley   MPI_Comm       comm;
4374ed5e4e85SVaclav Hapla   char           filename[PETSC_MAX_PATH_LEN]   = "<unspecified>";
4375ed5e4e85SVaclav Hapla   char           bdFilename[PETSC_MAX_PATH_LEN] = "<unspecified>";
4376ed5e4e85SVaclav Hapla   char           plexname[PETSC_MAX_PATH_LEN]   = "";
43774e22dd4cSMatthew G. Knepley   const char    *option;
43789318fe57SMatthew G. Knepley 
43799318fe57SMatthew G. Knepley   PetscFunctionBegin;
4380708be2fdSJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
43819566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
43829318fe57SMatthew G. Knepley   /* TODO Turn this into a registration interface */
43839566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_filename", "File containing a mesh", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &fflg));
4384d0812dedSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_file_contents", "Contents of a file format in a string", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &strflg));
43859566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_filename", "File containing a mesh boundary", "DMPlexCreateFromFile", bdFilename, bdFilename, sizeof(bdFilename), &bdfflg));
43869566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_name", "Name of the mesh in the file", "DMPlexCreateFromFile", plexname, plexname, sizeof(plexname), &nameflg));
43879566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_cell", "Cell shape", "", DMPolytopeTypes, (PetscEnum)cell, (PetscEnum *)&cell, NULL));
43889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_reference_cell_domain", "Use a reference cell domain", "", refDomain, &refDomain, NULL));
43899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_shape", "Shape for built-in mesh", "", DMPlexShapes, (PetscEnum)shape, (PetscEnum *)&shape, &flg));
43909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_dim", "Topological dimension of the mesh", "DMGetDimension", dim, &dim, &flg, 0));
43919566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_simplex", "Mesh cell shape", "", simplex, &simplex, &flg));
43929566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate", "Flag to create edges and faces automatically", "", interpolate, &interpolate, &flg));
4393b9da1bb3SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_plex_orient", "Orient the constructed mesh", "DMPlexOrient", orient, &orient, &flg));
43949566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_cone", "Set adjacency direction", "DMSetBasicAdjacency", adjCone, &adjCone, &flg));
43959566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_closure", "Set adjacency size", "DMSetBasicAdjacency", adjClosure, &adjClosure, &flg2));
43969566063dSJacob Faibussowitsch   if (flg || flg2) PetscCall(DMSetBasicAdjacency(dm, adjCone, adjClosure));
43973f3e541fSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_adj", "Debug output level all adjacency computations", "", 0, &((DM_Plex *)dm->data)->printAdj, NULL, 0));
43989318fe57SMatthew G. Knepley 
439961a622f3SMatthew G. Knepley   switch (cell) {
440061a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT:
440161a622f3SMatthew G. Knepley   case DM_POLYTOPE_SEGMENT:
440261a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT_PRISM_TENSOR:
440361a622f3SMatthew G. Knepley   case DM_POLYTOPE_TRIANGLE:
440461a622f3SMatthew G. Knepley   case DM_POLYTOPE_QUADRILATERAL:
440561a622f3SMatthew G. Knepley   case DM_POLYTOPE_TETRAHEDRON:
4406d71ae5a4SJacob Faibussowitsch   case DM_POLYTOPE_HEXAHEDRON:
4407d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_TRUE;
4408d71ae5a4SJacob Faibussowitsch     break;
4409d71ae5a4SJacob Faibussowitsch   default:
4410d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_FALSE;
4411d71ae5a4SJacob Faibussowitsch     break;
441261a622f3SMatthew G. Knepley   }
441361a622f3SMatthew G. Knepley 
44149318fe57SMatthew G. Knepley   if (fflg) {
44159318fe57SMatthew G. Knepley     DM          dmnew;
44161e4a82c4SMatthew G. Knepley     const char *name;
44179318fe57SMatthew G. Knepley 
44181e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
44191e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), filename, nameflg ? plexname : name, interpolate, &dmnew));
44205de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
442169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
44229318fe57SMatthew G. Knepley   } else if (refDomain) {
44239566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateReferenceCell_Internal(dm, cell));
44249318fe57SMatthew G. Knepley   } else if (bdfflg) {
44259318fe57SMatthew G. Knepley     DM          bdm, dmnew;
44261e4a82c4SMatthew G. Knepley     const char *name;
44279318fe57SMatthew G. Knepley 
44281e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
44291e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), bdFilename, nameflg ? plexname : name, interpolate, &bdm));
44309566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)bdm, "bd_"));
44319566063dSJacob Faibussowitsch     PetscCall(DMSetFromOptions(bdm));
44329566063dSJacob Faibussowitsch     PetscCall(DMPlexGenerate(bdm, NULL, interpolate, &dmnew));
44339566063dSJacob Faibussowitsch     PetscCall(DMDestroy(&bdm));
44345de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
443569d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
4436d0812dedSMatthew G. Knepley   } else if (strflg) {
4437d0812dedSMatthew G. Knepley     DM          dmnew;
4438d0812dedSMatthew G. Knepley     PetscViewer viewer;
4439d0812dedSMatthew G. Knepley     const char *contents;
4440d0812dedSMatthew G. Knepley     char       *strname;
4441d0812dedSMatthew G. Knepley     char        tmpdir[PETSC_MAX_PATH_LEN];
4442d0812dedSMatthew G. Knepley     char        tmpfilename[PETSC_MAX_PATH_LEN];
4443d0812dedSMatthew G. Knepley     char        name[PETSC_MAX_PATH_LEN];
4444d0812dedSMatthew G. Knepley     MPI_Comm    comm;
4445d0812dedSMatthew G. Knepley     PetscMPIInt rank;
4446d0812dedSMatthew G. Knepley 
4447d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4448d0812dedSMatthew G. Knepley     PetscCallMPI(MPI_Comm_rank(comm, &rank));
4449d0812dedSMatthew G. Knepley     PetscCall(PetscStrchr(filename, ':', &strname));
4450d0812dedSMatthew G. Knepley     PetscCheck(strname, comm, PETSC_ERR_ARG_WRONG, "File contents must have the form \"ext:string_name\", not %s", filename);
4451d0812dedSMatthew G. Knepley     strname[0] = '\0';
4452d0812dedSMatthew G. Knepley     ++strname;
4453d0812dedSMatthew G. Knepley     PetscCall(PetscDLSym(NULL, strname, (void **)&contents));
4454d0812dedSMatthew G. Knepley     PetscCheck(contents, comm, PETSC_ERR_ARG_WRONG, "Could not locate mesh string %s", strname);
4455d0812dedSMatthew G. Knepley     PetscCall(PetscGetTmp(comm, tmpdir, PETSC_MAX_PATH_LEN));
4456ed32af8cSMatthew G. Knepley     PetscCall(PetscStrlcat(tmpdir, "/meshXXXXXX", PETSC_MAX_PATH_LEN));
4457ed32af8cSMatthew G. Knepley     PetscCall(PetscMkdtemp(tmpdir));
4458ed32af8cSMatthew G. Knepley     PetscCall(PetscSNPrintf(tmpfilename, PETSC_MAX_PATH_LEN, "%s/mesh.%s", tmpdir, filename));
4459d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIOpen(comm, tmpfilename, &viewer));
4460d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIPrintf(viewer, "%s\n", contents));
4461d0812dedSMatthew G. Knepley     PetscCall(PetscViewerDestroy(&viewer));
4462d0812dedSMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), tmpfilename, plexname, interpolate, &dmnew));
4463ed32af8cSMatthew G. Knepley     PetscCall(PetscRMTree(tmpdir));
4464d0812dedSMatthew G. Knepley     PetscCall(PetscSNPrintf(name, PETSC_MAX_PATH_LEN, "%s Mesh", strname));
4465d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
4466d0812dedSMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
4467d0812dedSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
44689318fe57SMatthew G. Knepley   } else {
44699566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)dm, DMPlexShapes[shape]));
44709318fe57SMatthew G. Knepley     switch (shape) {
4471669647acSMatthew G. Knepley     case DM_SHAPE_BOX:
44725dca41c3SJed Brown     case DM_SHAPE_ZBOX:
4473669647acSMatthew G. Knepley     case DM_SHAPE_ANNULUS: {
44749318fe57SMatthew G. Knepley       PetscInt       faces[3]  = {0, 0, 0};
44759318fe57SMatthew G. Knepley       PetscReal      lower[3]  = {0, 0, 0};
44769318fe57SMatthew G. Knepley       PetscReal      upper[3]  = {1, 1, 1};
44779318fe57SMatthew G. Knepley       DMBoundaryType bdt[3]    = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4478669647acSMatthew G. Knepley       PetscBool      isAnnular = shape == DM_SHAPE_ANNULUS ? PETSC_TRUE : PETSC_FALSE;
44799318fe57SMatthew G. Knepley       PetscInt       i, n;
44809318fe57SMatthew G. Knepley 
44819318fe57SMatthew G. Knepley       n = dim;
44829318fe57SMatthew G. Knepley       for (i = 0; i < dim; ++i) faces[i] = (dim == 1 ? 1 : 4 - dim);
44839566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
44849318fe57SMatthew G. Knepley       n = 3;
44859566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
448663a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
44879318fe57SMatthew G. Knepley       n = 3;
44889566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
448963a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
44909318fe57SMatthew G. Knepley       n = 3;
44919566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
449263a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4493669647acSMatthew G. Knepley 
4494669647acSMatthew G. Knepley       PetscCheck(!isAnnular || dim == 2, comm, PETSC_ERR_ARG_OUTOFRANGE, "Only two dimensional annuli have been implemented");
4495669647acSMatthew G. Knepley       if (isAnnular)
4496669647acSMatthew G. Knepley         for (i = 0; i < dim - 1; ++i) bdt[i] = DM_BOUNDARY_PERIODIC;
4497669647acSMatthew G. Knepley 
44989318fe57SMatthew G. Knepley       switch (cell) {
449961a622f3SMatthew G. Knepley       case DM_POLYTOPE_TRI_PRISM_TENSOR:
45009566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeBoxMesh_Internal(dm, faces, lower, upper, bdt));
4501d410b0cfSMatthew G. Knepley         if (!interpolate) {
4502d410b0cfSMatthew G. Knepley           DM udm;
4503d410b0cfSMatthew G. Knepley 
45049566063dSJacob Faibussowitsch           PetscCall(DMPlexUninterpolate(dm, &udm));
450569d8a87bSksagiyam           PetscCall(DMPlexReplace_Internal(dm, &udm));
4506d410b0cfSMatthew G. Knepley         }
45079318fe57SMatthew G. Knepley         break;
4508d71ae5a4SJacob Faibussowitsch       default:
45095dca41c3SJed Brown         PetscCall(DMPlexCreateBoxMesh_Internal(dm, shape, dim, simplex, faces, lower, upper, bdt, interpolate));
4510d71ae5a4SJacob Faibussowitsch         break;
45119318fe57SMatthew G. Knepley       }
4512669647acSMatthew G. Knepley       if (isAnnular) {
4513669647acSMatthew G. Knepley         DM          cdm;
4514669647acSMatthew G. Knepley         PetscDS     cds;
4515669647acSMatthew G. Knepley         PetscScalar bounds[2] = {lower[0], upper[0]};
4516669647acSMatthew G. Knepley 
4517669647acSMatthew G. Knepley         // Fix coordinates for annular region
4518669647acSMatthew G. Knepley         PetscCall(DMSetPeriodicity(dm, NULL, NULL, NULL));
4519669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinatesLocal(dm, NULL));
4520669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinates(dm, NULL));
4521e44f6aebSMatthew G. Knepley         PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
4522669647acSMatthew G. Knepley         PetscCall(DMGetCoordinateDM(dm, &cdm));
4523669647acSMatthew G. Knepley         PetscCall(DMGetDS(cdm, &cds));
4524669647acSMatthew G. Knepley         PetscCall(PetscDSSetConstants(cds, 2, bounds));
4525669647acSMatthew G. Knepley         PetscCall(DMPlexRemapGeometry(dm, 0.0, boxToAnnulus));
4526669647acSMatthew G. Knepley       }
45279371c9d4SSatish Balay     } break;
45289371c9d4SSatish Balay     case DM_SHAPE_BOX_SURFACE: {
45299318fe57SMatthew G. Knepley       PetscInt  faces[3] = {0, 0, 0};
45309318fe57SMatthew G. Knepley       PetscReal lower[3] = {0, 0, 0};
45319318fe57SMatthew G. Knepley       PetscReal upper[3] = {1, 1, 1};
45329318fe57SMatthew G. Knepley       PetscInt  i, n;
45339318fe57SMatthew G. Knepley 
45349318fe57SMatthew G. Knepley       n = dim + 1;
45359318fe57SMatthew G. Knepley       for (i = 0; i < dim + 1; ++i) faces[i] = (dim + 1 == 1 ? 1 : 4 - (dim + 1));
45369566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
45379318fe57SMatthew G. Knepley       n = 3;
45389566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
453963a3b9bcSJacob 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);
45409318fe57SMatthew G. Knepley       n = 3;
45419566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
454263a3b9bcSJacob 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);
45439566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(dm, dim + 1, faces, lower, upper, interpolate));
45449371c9d4SSatish Balay     } break;
45459371c9d4SSatish Balay     case DM_SHAPE_SPHERE: {
45469318fe57SMatthew G. Knepley       PetscReal R = 1.0;
45479318fe57SMatthew G. Knepley 
45489566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_sphere_radius", "Radius of the sphere", "", R, &R, &flg));
45499566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateSphereMesh_Internal(dm, dim, simplex, R));
45509371c9d4SSatish Balay     } break;
45519371c9d4SSatish Balay     case DM_SHAPE_BALL: {
45529318fe57SMatthew G. Knepley       PetscReal R = 1.0;
45539318fe57SMatthew G. Knepley 
45549566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_ball_radius", "Radius of the ball", "", R, &R, &flg));
45559566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBallMesh_Internal(dm, dim, R));
45569371c9d4SSatish Balay     } break;
45579371c9d4SSatish Balay     case DM_SHAPE_CYLINDER: {
45589318fe57SMatthew G. Knepley       DMBoundaryType bdt = DM_BOUNDARY_NONE;
45599318fe57SMatthew G. Knepley       PetscInt       Nw  = 6;
456049704ca5SMatthew G. Knepley       PetscInt       Nr  = 0;
45619318fe57SMatthew G. Knepley 
45629566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnum("-dm_plex_cylinder_bd", "Boundary type in the z direction", "", DMBoundaryTypes, (PetscEnum)bdt, (PetscEnum *)&bdt, NULL));
45639566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_wedges", "Number of wedges around the cylinder", "", Nw, &Nw, NULL));
456449704ca5SMatthew G. Knepley       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_refine", "Number of refinements before projection", "", Nr, &Nr, NULL));
45659318fe57SMatthew G. Knepley       switch (cell) {
4566d71ae5a4SJacob Faibussowitsch       case DM_POLYTOPE_TRI_PRISM_TENSOR:
4567d71ae5a4SJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(dm, Nw, interpolate));
4568d71ae5a4SJacob Faibussowitsch         break;
4569d71ae5a4SJacob Faibussowitsch       default:
457049704ca5SMatthew G. Knepley         PetscCall(DMPlexCreateHexCylinderMesh_Internal(dm, bdt, Nr));
4571d71ae5a4SJacob Faibussowitsch         break;
45729318fe57SMatthew G. Knepley       }
45739371c9d4SSatish Balay     } break;
4574b7f5c055SJed Brown     case DM_SHAPE_SCHWARZ_P: // fallthrough
45759371c9d4SSatish Balay     case DM_SHAPE_GYROID: {
4576b7f5c055SJed Brown       PetscInt       extent[3] = {1, 1, 1}, refine = 0, layers = 0, three;
4577b7f5c055SJed Brown       PetscReal      thickness   = 0.;
4578b7f5c055SJed Brown       DMBoundaryType periodic[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4579b7f5c055SJed Brown       DMPlexTPSType  tps_type    = shape == DM_SHAPE_SCHWARZ_P ? DMPLEX_TPS_SCHWARZ_P : DMPLEX_TPS_GYROID;
45801436d7faSJed Brown       PetscBool      tps_distribute;
45819566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_tps_extent", "Number of replicas for each of three dimensions", NULL, extent, (three = 3, &three), NULL));
45829566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_refine", "Number of refinements", NULL, refine, &refine, NULL));
45839566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_tps_periodic", "Periodicity in each of three dimensions", NULL, DMBoundaryTypes, (PetscEnum *)periodic, (three = 3, &three), NULL));
45849566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_layers", "Number of layers in volumetric extrusion (or zero to not extrude)", NULL, layers, &layers, NULL));
45859566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_tps_thickness", "Thickness of volumetric extrusion", NULL, thickness, &thickness, NULL));
45869566063dSJacob Faibussowitsch       PetscCall(DMPlexDistributeGetDefault(dm, &tps_distribute));
45879566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_tps_distribute", "Distribute the 2D mesh prior to refinement and extrusion", NULL, tps_distribute, &tps_distribute, NULL));
45889566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateTPSMesh_Internal(dm, tps_type, extent, periodic, tps_distribute, refine, layers, thickness));
45899371c9d4SSatish Balay     } break;
45909371c9d4SSatish Balay     case DM_SHAPE_DOUBLET: {
459105bd46c0SStefano Zampini       DM        dmnew;
459205bd46c0SStefano Zampini       PetscReal rl = 0.0;
459305bd46c0SStefano Zampini 
459405bd46c0SStefano Zampini       PetscCall(PetscOptionsReal("-dm_plex_doublet_refinementlimit", "Refinement limit", NULL, rl, &rl, NULL));
459505bd46c0SStefano Zampini       PetscCall(DMPlexCreateDoublet(PetscObjectComm((PetscObject)dm), dim, simplex, interpolate, rl, &dmnew));
45965de52c6dSVaclav Hapla       PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
459769d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &dmnew));
45989371c9d4SSatish Balay     } break;
4599cfb853baSMatthew G. Knepley     case DM_SHAPE_HYPERCUBIC: {
4600*8d2ec52aSSatish Balay       PetscInt       *edges, overlap = 1;
4601cfb853baSMatthew G. Knepley       PetscReal      *lower, *upper;
4602cfb853baSMatthew G. Knepley       DMBoundaryType *bdt;
4603cfb853baSMatthew G. Knepley       PetscInt        n, d;
4604cfb853baSMatthew G. Knepley 
4605cfb853baSMatthew G. Knepley       *useCoordSpace = PETSC_FALSE;
4606cfb853baSMatthew G. Knepley       PetscCall(PetscMalloc4(dim, &edges, dim, &lower, dim, &upper, dim, &bdt));
4607cfb853baSMatthew G. Knepley       for (d = 0; d < dim; ++d) {
4608cfb853baSMatthew G. Knepley         edges[d] = 1;
4609cfb853baSMatthew G. Knepley         lower[d] = 0.;
4610cfb853baSMatthew G. Knepley         upper[d] = 1.;
4611cfb853baSMatthew G. Knepley         bdt[d]   = DM_BOUNDARY_PERIODIC;
4612cfb853baSMatthew G. Knepley       }
4613cfb853baSMatthew G. Knepley       n = dim;
4614cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", edges, &n, &flg));
4615cfb853baSMatthew G. Knepley       n = dim;
4616cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
4617cfb853baSMatthew 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);
4618cfb853baSMatthew G. Knepley       n = dim;
4619cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
4620cfb853baSMatthew 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);
4621cfb853baSMatthew G. Knepley       n = dim;
4622cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
4623cfb853baSMatthew 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);
4624*8d2ec52aSSatish Balay       PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", overlap, &overlap, NULL, 0));
4625*8d2ec52aSSatish Balay       PetscCall(DMPlexCreateHypercubicMesh_Internal(dm, dim, lower, upper, edges, overlap, bdt));
4626cfb853baSMatthew G. Knepley       PetscCall(PetscFree4(edges, lower, upper, bdt));
4627cfb853baSMatthew G. Knepley     } break;
4628d71ae5a4SJacob Faibussowitsch     default:
4629d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_SUP, "Domain shape %s is unsupported", DMPlexShapes[shape]);
46309318fe57SMatthew G. Knepley     }
46319318fe57SMatthew G. Knepley   }
46329566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
463348a46eb9SPierre Jolivet   if (!((PetscObject)dm)->name && nameflg) PetscCall(PetscObjectSetName((PetscObject)dm, plexname));
4634b9da1bb3SMatthew G. Knepley   if (orient) PetscCall(DMPlexOrient(dm));
46354e22dd4cSMatthew G. Knepley   // Allow label creation
46364e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_label_", &option, NULL, &flg));
46374e22dd4cSMatthew G. Knepley   if (flg) {
46384e22dd4cSMatthew G. Knepley     DMLabel     label;
46394e22dd4cSMatthew G. Knepley     PetscInt    points[1024], n = 1024;
46404e22dd4cSMatthew G. Knepley     char        fulloption[PETSC_MAX_PATH_LEN];
46414e22dd4cSMatthew G. Knepley     const char *name = &option[14];
46424e22dd4cSMatthew G. Knepley 
46434e22dd4cSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, name));
46444e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, name, &label));
46454e22dd4cSMatthew G. Knepley     fulloption[0] = '-';
46464e22dd4cSMatthew G. Knepley     fulloption[1] = 0;
46474e22dd4cSMatthew G. Knepley     PetscCall(PetscStrlcat(fulloption, option, PETSC_MAX_PATH_LEN));
46484e22dd4cSMatthew G. Knepley     PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, NULL));
46494e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < n; ++p) PetscCall(DMLabelSetValue(label, points[p], 1));
46504e22dd4cSMatthew G. Knepley   }
4651dd0eeac9SMatthew G. Knepley   // Allow cohesive label creation
4652dd0eeac9SMatthew G. Knepley   //   Faces are input, completed, and all points are marked with their depth
4653dd0eeac9SMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_cohesive_label_", &option, NULL, &flg));
4654dd0eeac9SMatthew G. Knepley   if (flg) {
4655dd0eeac9SMatthew G. Knepley     DMLabel   label;
4656dd0eeac9SMatthew G. Knepley     PetscInt  points[1024], n, pStart, pEnd, Nl = 1;
46575390be7dSMatthew G. Knepley     PetscBool noCreate = PETSC_FALSE;
4658dd0eeac9SMatthew G. Knepley     char      fulloption[PETSC_MAX_PATH_LEN];
4659dd0eeac9SMatthew G. Knepley     char      name[PETSC_MAX_PATH_LEN];
4660dd0eeac9SMatthew G. Knepley     size_t    len;
4661dd0eeac9SMatthew G. Knepley 
4662dd0eeac9SMatthew G. Knepley     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4663dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrncpy(name, &option[23], PETSC_MAX_PATH_LEN));
4664dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrlen(name, &len));
4665dd0eeac9SMatthew G. Knepley     if (name[len - 1] == '0') Nl = 10;
4666dd0eeac9SMatthew G. Knepley     for (PetscInt l = 0; l < Nl; ++l) {
46676497c311SBarry Smith       if (l > 0) name[len - 1] = (char)('0' + l);
4668dd0eeac9SMatthew G. Knepley       fulloption[0] = 0;
4669dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, "-dm_plex_cohesive_label_", 32));
4670dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, name, PETSC_MAX_PATH_LEN - 32));
4671dd0eeac9SMatthew G. Knepley       n = 1024;
4672dd0eeac9SMatthew G. Knepley       PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, &flg));
4673dd0eeac9SMatthew G. Knepley       if (!flg) break;
46745390be7dSMatthew G. Knepley       PetscCall(DMHasLabel(dm, name, &noCreate));
46755390be7dSMatthew G. Knepley       if (noCreate) {
46765390be7dSMatthew G. Knepley         DMLabel         inlabel;
46775390be7dSMatthew G. Knepley         IS              pointIS;
46785390be7dSMatthew G. Knepley         const PetscInt *lpoints;
46795390be7dSMatthew G. Knepley         PetscInt        pdep, ln, inval = points[0];
46805390be7dSMatthew G. Knepley         char            newname[PETSC_MAX_PATH_LEN];
46815390be7dSMatthew G. Knepley 
46825390be7dSMatthew G. Knepley         PetscCheck(n == 1, comm, PETSC_ERR_ARG_WRONG, "Must specify a label value with this option");
46835390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &inlabel));
46845390be7dSMatthew G. Knepley         PetscCall(DMLabelGetStratumIS(inlabel, inval, &pointIS));
46855390be7dSMatthew G. Knepley         PetscCall(ISGetLocalSize(pointIS, &ln));
46865390be7dSMatthew G. Knepley         PetscCall(ISGetIndices(pointIS, &lpoints));
46875390be7dSMatthew G. Knepley         PetscCall(DMPlexGetPointDepth(dm, lpoints[0], &pdep));
46885390be7dSMatthew G. Knepley         PetscCall(PetscSNPrintf(newname, PETSC_MAX_PATH_LEN, "%s%" PetscInt_FMT, name, points[0]));
46895390be7dSMatthew G. Knepley         PetscCall(DMCreateLabel(dm, newname));
46905390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, newname, &label));
46915390be7dSMatthew G. Knepley         if (!pdep) PetscCall(ProcessCohesiveLabel_Vertices(dm, label, inlabel, inval, ln, lpoints));
46925390be7dSMatthew G. Knepley         else PetscCall(ProcessCohesiveLabel_Faces(dm, label, ln, lpoints));
46935390be7dSMatthew G. Knepley         PetscCall(ISRestoreIndices(pointIS, &lpoints));
46945390be7dSMatthew G. Knepley         PetscCall(ISDestroy(&pointIS));
46955390be7dSMatthew G. Knepley       } else {
4696dd0eeac9SMatthew G. Knepley         PetscCall(DMCreateLabel(dm, name));
4697dd0eeac9SMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &label));
4698dd0eeac9SMatthew G. Knepley         if (pStart >= pEnd) n = 0;
46995390be7dSMatthew G. Knepley         PetscCall(ProcessCohesiveLabel_Faces(dm, label, n, points));
4700dd0eeac9SMatthew G. Knepley       }
4701dd0eeac9SMatthew G. Knepley       PetscCall(DMPlexOrientLabel(dm, label));
47020542aa8cSMatthew G. Knepley       PetscCall(DMPlexLabelCohesiveComplete(dm, label, NULL, 1, PETSC_FALSE, PETSC_FALSE, NULL));
4703dd0eeac9SMatthew G. Knepley     }
4704dd0eeac9SMatthew G. Knepley   }
47055390be7dSMatthew G. Knepley   PetscCall(DMViewFromOptions(dm, NULL, "-created_dm_view"));
4706708be2fdSJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
47073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47080a6ba040SMatthew G. Knepley }
47090a6ba040SMatthew G. Knepley 
4710d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_NonRefinement_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4711d71ae5a4SJacob Faibussowitsch {
47120a6ba040SMatthew G. Knepley   DM_Plex  *mesh = (DM_Plex *)dm->data;
47137f9d8d6cSVaclav Hapla   PetscBool flg, flg2;
47149318fe57SMatthew G. Knepley   char      bdLabel[PETSC_MAX_PATH_LEN];
4715adc21957SMatthew G. Knepley   char      method[PETSC_MAX_PATH_LEN];
47160a6ba040SMatthew G. Knepley 
47170a6ba040SMatthew G. Knepley   PetscFunctionBegin;
47180a6ba040SMatthew G. Knepley   /* Handle viewing */
47199566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMPlexMatSetClosure", PETSC_FALSE, &mesh->printSetValues, NULL));
47205962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fem", "Debug output level for all fem computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printFEM, NULL, 0));
47215962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fvm", "Debug output level for all fvm computations", "DMPlexSNESComputeResidualFVM", 0, &mesh->printFVM, NULL, 0));
47223f3e541fSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_adj", "Debug output level all adjacency computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printAdj, NULL, 0));
47239566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMPlexSNESComputeResidualFEM", mesh->printTol, &mesh->printTol, NULL));
47249566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_l2", "Debug output level all L2 diff computations", "DMComputeL2Diff", 0, &mesh->printL2, NULL, 0));
4725f5867de0SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_locate", "Debug output level all point location computations", "DMLocatePoints", 0, &mesh->printLocate, NULL, 0));
4726a77a5016SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_project", "Debug output level all projection computations", "DMPlexProject", 0, &mesh->printProject, NULL, 0));
47279566063dSJacob Faibussowitsch   PetscCall(DMMonitorSetFromOptions(dm, "-dm_plex_monitor_throughput", "Monitor the simulation throughput", "DMPlexMonitorThroughput", DMPlexMonitorThroughput, NULL, &flg));
47289566063dSJacob Faibussowitsch   if (flg) PetscCall(PetscLogDefaultBegin());
47299318fe57SMatthew G. Knepley   /* Labeling */
47309566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_label", "Label to mark the mesh boundary", "", bdLabel, bdLabel, sizeof(bdLabel), &flg));
47319566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexCreateBoundaryLabel_Private(dm, bdLabel));
4732953fc75cSMatthew G. Knepley   /* Point Location */
47339566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMInterpolate", PETSC_FALSE, &mesh->useHashLocation, NULL));
47340848f4b5SMatthew G. Knepley   /* Partitioning and distribution */
47359566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", PETSC_FALSE, &mesh->partitionBalance, NULL));
4736d02c7345SMatthew G. Knepley   /* Reordering */
4737adc21957SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_reorder_section", "Compute point permutation for local section", "DMReorderSectionSetDefault", PETSC_FALSE, &flg2, &flg));
4738adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetDefault(dm, flg2 ? DM_REORDER_DEFAULT_TRUE : DM_REORDER_DEFAULT_FALSE));
4739adc21957SMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_reorder_section_type", "Reordering method for local section", "DMReorderSectionSetType", method, method, PETSC_MAX_PATH_LEN, &flg));
4740adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetType(dm, method));
47412e62ab5aSMatthew G. Knepley   /* Generation and remeshing */
47429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_remesh_bd", "Allow changes to the boundary on remeshing", "DMAdapt", PETSC_FALSE, &mesh->remeshBd, NULL));
4743b29cfa1cSToby Isaac   /* Projection behavior */
4744d5b43468SJose E. Roman   PetscCall(PetscOptionsBoundedInt("-dm_plex_max_projection_height", "Maximum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL, 0));
47459566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_regular_refinement", "Use special nested projection algorithm for regular refinement", "DMPlexSetRegularRefinement", mesh->regularRefinement, &mesh->regularRefinement, NULL));
4746f12cf164SMatthew G. Knepley   /* Checking structure */
4747f12cf164SMatthew G. Knepley   {
47487f9d8d6cSVaclav Hapla     PetscBool all = PETSC_FALSE;
4749f12cf164SMatthew G. Knepley 
47507f9d8d6cSVaclav Hapla     PetscCall(PetscOptionsBool("-dm_plex_check_all", "Perform all basic checks", "DMPlexCheck", PETSC_FALSE, &all, NULL));
47517f9d8d6cSVaclav Hapla     if (all) {
47527f9d8d6cSVaclav Hapla       PetscCall(DMPlexCheck(dm));
47537f9d8d6cSVaclav Hapla     } else {
47549566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_symmetry", "Check that the adjacency information in the mesh is symmetric", "DMPlexCheckSymmetry", PETSC_FALSE, &flg, &flg2));
47557f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSymmetry(dm));
47569566063dSJacob 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));
47577f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSkeleton(dm, 0));
47589566063dSJacob 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));
47597f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckFaces(dm, 0));
47609566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_geometry", "Check that cells have positive volume", "DMPlexCheckGeometry", PETSC_FALSE, &flg, &flg2));
47617f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckGeometry(dm));
47629566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_pointsf", "Check some necessary conditions for PointSF", "DMPlexCheckPointSF", PETSC_FALSE, &flg, &flg2));
4763d7d32a9aSMatthew G. Knepley       if (flg && flg2) PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE));
47649566063dSJacob 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));
47657f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckInterfaceCones(dm));
47667f9d8d6cSVaclav Hapla     }
47679566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_plex_check_cell_shape", "Check cell shape", "DMPlexCheckCellShape", PETSC_FALSE, &flg, &flg2));
47689566063dSJacob Faibussowitsch     if (flg && flg2) PetscCall(DMPlexCheckCellShape(dm, PETSC_TRUE, PETSC_DETERMINE));
4769f12cf164SMatthew G. Knepley   }
47709318fe57SMatthew G. Knepley   {
47719318fe57SMatthew G. Knepley     PetscReal scale = 1.0;
47724f3833eaSMatthew G. Knepley 
47739566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-dm_plex_scale", "Scale factor for mesh coordinates", "DMPlexScale", scale, &scale, &flg));
47749318fe57SMatthew G. Knepley     if (flg) {
47759318fe57SMatthew G. Knepley       Vec coordinates, coordinatesLocal;
47769318fe57SMatthew G. Knepley 
47779566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(dm, &coordinates));
47789566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal));
47799566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinates, scale));
47809566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinatesLocal, scale));
47819318fe57SMatthew G. Knepley     }
47829318fe57SMatthew G. Knepley   }
47839566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerSetFromOptions(mesh->partitioner));
47843ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
478568d4fef7SMatthew G. Knepley }
478668d4fef7SMatthew G. Knepley 
4787d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_Overlap_Plex(DM dm, PetscOptionItems *PetscOptionsObject, PetscInt *overlap)
4788d71ae5a4SJacob Faibussowitsch {
4789c506a872SMatthew G. Knepley   PetscInt  numOvLabels = 16, numOvExLabels = 16;
4790c506a872SMatthew G. Knepley   char     *ovLabelNames[16], *ovExLabelNames[16];
4791c506a872SMatthew G. Knepley   PetscInt  numOvValues = 16, numOvExValues = 16, l;
4792c506a872SMatthew G. Knepley   PetscBool flg;
4793c506a872SMatthew G. Knepley 
4794c506a872SMatthew G. Knepley   PetscFunctionBegin;
4795c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", *overlap, overlap, NULL, 0));
4796c506a872SMatthew G. Knepley   PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_labels", "List of overlap label names", "DMPlexDistribute", ovLabelNames, &numOvLabels, &flg));
4797c506a872SMatthew G. Knepley   if (!flg) numOvLabels = 0;
4798c506a872SMatthew G. Knepley   if (numOvLabels) {
4799c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvLabels = numOvLabels;
4800c506a872SMatthew G. Knepley     for (l = 0; l < numOvLabels; ++l) {
4801c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovLabelNames[l], &((DM_Plex *)dm->data)->ovLabels[l]));
4802c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovLabelNames[l]);
4803c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovLabelNames[l]));
4804c506a872SMatthew G. Knepley     }
4805c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_values", "List of overlap label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovValues, &numOvValues, &flg));
4806c506a872SMatthew G. Knepley     if (!flg) numOvValues = 0;
4807c506a872SMatthew 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);
4808c506a872SMatthew G. Knepley 
4809c506a872SMatthew G. Knepley     PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_exclude_labels", "List of overlap exclude label names", "DMPlexDistribute", ovExLabelNames, &numOvExLabels, &flg));
4810c506a872SMatthew G. Knepley     if (!flg) numOvExLabels = 0;
4811c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvExLabels = numOvExLabels;
4812c506a872SMatthew G. Knepley     for (l = 0; l < numOvExLabels; ++l) {
4813c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovExLabelNames[l], &((DM_Plex *)dm->data)->ovExLabels[l]));
4814c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovExLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovExLabelNames[l]);
4815c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovExLabelNames[l]));
4816c506a872SMatthew G. Knepley     }
4817c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_exclude_values", "List of overlap exclude label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovExValues, &numOvExValues, &flg));
4818c506a872SMatthew G. Knepley     if (!flg) numOvExValues = 0;
4819c506a872SMatthew 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);
4820c506a872SMatthew G. Knepley   }
48213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4822c506a872SMatthew G. Knepley }
4823c506a872SMatthew G. Knepley 
4824d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMSetFromOptions_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4825d71ae5a4SJacob Faibussowitsch {
4826bdf63967SMatthew G. Knepley   PetscFunctionList    ordlist;
4827bdf63967SMatthew G. Knepley   char                 oname[256];
48284e22dd4cSMatthew G. Knepley   char                 sublabelname[PETSC_MAX_PATH_LEN] = "";
4829adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
4830d410b0cfSMatthew G. Knepley   PetscReal            volume    = -1.0;
48319318fe57SMatthew G. Knepley   PetscInt             prerefine = 0, refine = 0, r, coarsen = 0, overlap = 0, extLayers = 0, dim;
48321b742f01SMatthew 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;
483368d4fef7SMatthew G. Knepley 
483468d4fef7SMatthew G. Knepley   PetscFunctionBegin;
4835d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "DMPlex Options");
4836dd4c3f67SMatthew G. Knepley   if (dm->cloneOpts) goto non_refine;
48379318fe57SMatthew G. Knepley   /* Handle automatic creation */
48389566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
48396bc1bd01Sksagiyam   if (dim < 0) {
48406bc1bd01Sksagiyam     PetscCall(DMPlexCreateFromOptions_Internal(PetscOptionsObject, &coordSpace, dm));
48416bc1bd01Sksagiyam     created = PETSC_TRUE;
48426bc1bd01Sksagiyam   }
48436bc1bd01Sksagiyam   PetscCall(DMGetDimension(dm, &dim));
4844d89e6e46SMatthew G. Knepley   /* Handle interpolation before distribution */
48459566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate_pre", "Flag to interpolate mesh before distribution", "", interpolate, &interpolate, &flg));
4846d89e6e46SMatthew G. Knepley   if (flg) {
4847d89e6e46SMatthew G. Knepley     DMPlexInterpolatedFlag interpolated;
4848d89e6e46SMatthew G. Knepley 
48499566063dSJacob Faibussowitsch     PetscCall(DMPlexIsInterpolated(dm, &interpolated));
4850d89e6e46SMatthew G. Knepley     if (interpolated == DMPLEX_INTERPOLATED_FULL && !interpolate) {
4851d89e6e46SMatthew G. Knepley       DM udm;
4852d89e6e46SMatthew G. Knepley 
48539566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(dm, &udm));
485469d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &udm));
4855d89e6e46SMatthew G. Knepley     } else if (interpolated != DMPLEX_INTERPOLATED_FULL && interpolate) {
4856d89e6e46SMatthew G. Knepley       DM idm;
4857d89e6e46SMatthew G. Knepley 
48589566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(dm, &idm));
485969d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &idm));
4860d89e6e46SMatthew G. Knepley     }
4861d89e6e46SMatthew G. Knepley   }
48624e22dd4cSMatthew G. Knepley   // Handle submesh selection before distribution
48634e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_submesh", "Label to use for submesh selection", "", sublabelname, sublabelname, PETSC_MAX_PATH_LEN, &flg));
48644e22dd4cSMatthew G. Knepley   if (flg) {
48654e22dd4cSMatthew G. Knepley     DM              subdm;
48664e22dd4cSMatthew G. Knepley     DMLabel         label;
48674e22dd4cSMatthew G. Knepley     IS              valueIS, pointIS;
48684e22dd4cSMatthew G. Knepley     const PetscInt *values, *points;
48694e22dd4cSMatthew G. Knepley     PetscBool       markedFaces = PETSC_FALSE;
48704e22dd4cSMatthew G. Knepley     PetscInt        Nv, value, Np;
48714e22dd4cSMatthew G. Knepley 
48724e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, sublabelname, &label));
48734e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetNumValues(label, &Nv));
48744e22dd4cSMatthew 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);
48754e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetValueIS(label, &valueIS));
48764e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(valueIS, &values));
48774e22dd4cSMatthew G. Knepley     value = values[0];
48784e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(valueIS, &values));
48794e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&valueIS));
48804e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumSize(label, value, &Np));
48814e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumIS(label, value, &pointIS));
48824e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(pointIS, &points));
48834e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < Np; ++p) {
48844e22dd4cSMatthew G. Knepley       PetscInt pdepth;
48854e22dd4cSMatthew G. Knepley 
48864e22dd4cSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, points[p], &pdepth));
48874e22dd4cSMatthew G. Knepley       if (pdepth) {
48884e22dd4cSMatthew G. Knepley         markedFaces = PETSC_TRUE;
48894e22dd4cSMatthew G. Knepley         break;
48904e22dd4cSMatthew G. Knepley       }
48914e22dd4cSMatthew G. Knepley     }
48924e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(pointIS, &points));
48934e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&pointIS));
48944e22dd4cSMatthew G. Knepley     PetscCall(DMPlexCreateSubmesh(dm, label, value, markedFaces, &subdm));
48954e22dd4cSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &subdm));
48964e22dd4cSMatthew G. Knepley     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
48974e22dd4cSMatthew G. Knepley   }
48989b44eab4SMatthew G. Knepley   /* Handle DMPlex refinement before distribution */
48999566063dSJacob Faibussowitsch   PetscCall(DMPlexGetRefinementUniform(dm, &uniformOrig));
49009566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_pre", "The number of refinements before distribution", "DMCreate", prerefine, &prerefine, NULL, 0));
49019566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap_pre", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
49029566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_uniform_pre", "Flag for uniform refinement before distribution", "DMCreate", uniform, &uniform, &flg));
49039566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexSetRefinementUniform(dm, uniform));
49049566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_refine_volume_limit_pre", "The maximum cell volume after refinement before distribution", "DMCreate", volume, &volume, &flg));
49059318fe57SMatthew G. Knepley   if (flg) {
49069566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(dm, PETSC_FALSE));
49079566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(dm, volume));
49089318fe57SMatthew G. Knepley     prerefine = PetscMax(prerefine, 1);
49099318fe57SMatthew G. Knepley   }
4910b23db253SStefano Zampini   if (prerefine) PetscCall(DMLocalizeCoordinates(dm));
49119b44eab4SMatthew G. Knepley   for (r = 0; r < prerefine; ++r) {
49129b44eab4SMatthew G. Knepley     DM             rdm;
49139b44eab4SMatthew G. Knepley     PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
49149b44eab4SMatthew G. Knepley 
4915dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
49169566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
491769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &rdm));
4918dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
491961a622f3SMatthew G. Knepley     if (coordFunc && remap) {
49209566063dSJacob Faibussowitsch       PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
49219b44eab4SMatthew G. Knepley       ((DM_Plex *)dm->data)->coordFunc = coordFunc;
49229b44eab4SMatthew G. Knepley     }
49239b44eab4SMatthew G. Knepley   }
49249566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, uniformOrig));
49259318fe57SMatthew G. Knepley   /* Handle DMPlex extrusion before distribution */
49269566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_extrude", "The number of layers to extrude", "", extLayers, &extLayers, NULL, 0));
49279318fe57SMatthew G. Knepley   if (extLayers) {
49289318fe57SMatthew G. Knepley     DM edm;
49299318fe57SMatthew G. Knepley 
49309566063dSJacob Faibussowitsch     PetscCall(DMExtrude(dm, extLayers, &edm));
493169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
493248d16a33SMatthew G. Knepley     ((DM_Plex *)dm->data)->coordFunc = NULL;
4933dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4934d410b0cfSMatthew G. Knepley     extLayers = 0;
49355e17fc22SAidan Hamilton     PetscCall(DMGetDimension(dm, &dim));
49369318fe57SMatthew G. Knepley   }
4937bdf63967SMatthew G. Knepley   /* Handle DMPlex reordering before distribution */
49386bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dm, &reorder));
49399566063dSJacob Faibussowitsch   PetscCall(MatGetOrderingList(&ordlist));
49406bc1bd01Sksagiyam   PetscCall(PetscStrncpy(oname, MATORDERINGNATURAL, sizeof(oname)));
49419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-dm_plex_reorder", "Set mesh reordering type", "DMPlexGetOrdering", ordlist, MATORDERINGNATURAL, oname, sizeof(oname), &flg));
4942adc21957SMatthew G. Knepley   if (reorder == DM_REORDER_DEFAULT_TRUE || flg) {
4943bdf63967SMatthew G. Knepley     DM pdm;
4944bdf63967SMatthew G. Knepley     IS perm;
4945bdf63967SMatthew G. Knepley 
49469566063dSJacob Faibussowitsch     PetscCall(DMPlexGetOrdering(dm, oname, NULL, &perm));
49479566063dSJacob Faibussowitsch     PetscCall(DMPlexPermute(dm, perm, &pdm));
49489566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&perm));
494969d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &pdm));
4950dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4951bdf63967SMatthew G. Knepley   }
49529b44eab4SMatthew G. Knepley   /* Handle DMPlex distribution */
49539566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dm, &distribute));
4954c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute", "Flag to redistribute a mesh among processes", "DMPlexDistribute", distribute, &distribute, NULL));
4955a286e215SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute_save_sf", "Flag to save the migration SF", "DMPlexSetMigrationSF", saveSF, &saveSF, NULL));
4956dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_Overlap_Plex(dm, PetscOptionsObject, &overlap));
49579b44eab4SMatthew G. Knepley   if (distribute) {
49589b44eab4SMatthew G. Knepley     DM               pdm = NULL;
49599b44eab4SMatthew G. Knepley     PetscPartitioner part;
4960a286e215SMatthew G. Knepley     PetscSF          sfMigration;
49619b44eab4SMatthew G. Knepley 
49629566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
49639566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
4964a286e215SMatthew G. Knepley     PetscCall(DMPlexDistribute(dm, overlap, &sfMigration, &pdm));
49655d2873a6SJames Wright     if (pdm) {
49665d2873a6SJames Wright       // Delete the local section to force the existing one to be rebuilt with the distributed DM
49675d2873a6SJames Wright       PetscCall(DMSetLocalSection(dm, pdm->localSection));
49685d2873a6SJames Wright       PetscCall(DMPlexReplace_Internal(dm, &pdm));
49695d2873a6SJames Wright     }
4970a286e215SMatthew G. Knepley     if (saveSF) PetscCall(DMPlexSetMigrationSF(dm, sfMigration));
4971a286e215SMatthew G. Knepley     PetscCall(PetscSFDestroy(&sfMigration));
49729b44eab4SMatthew G. Knepley   }
49734054ae39SJames Wright 
49744054ae39SJames Wright   {
49754054ae39SJames Wright     PetscBool useBoxLabel = PETSC_FALSE;
49764054ae39SJames Wright     PetscCall(PetscOptionsBool("-dm_plex_box_label", "Create 'Face Sets' assuming boundary faces align with cartesian directions", "DMCreate", useBoxLabel, &useBoxLabel, NULL));
4977d7d2d1d2SJames Wright     if (useBoxLabel) {
4978d7d2d1d2SJames Wright       PetscInt       n      = 3;
4979d7d2d1d2SJames Wright       DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4980d7d2d1d2SJames Wright 
4981d7d2d1d2SJames Wright       PetscCall(PetscOptionsEnumArray("-dm_plex_box_label_bd", "Boundary type for each dimension when using -dm_plex_box_label", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
4982d7d2d1d2SJames Wright       PetscCheck(!flg || !(n != dim), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4983d7d2d1d2SJames Wright       PetscCall(DMPlexSetBoxLabel_Internal(dm, bdt));
4984d7d2d1d2SJames Wright     }
49854054ae39SJames Wright   }
4986d2b2dc1eSMatthew G. Knepley   /* Must check CEED options before creating function space for coordinates */
4987d2b2dc1eSMatthew G. Knepley   {
4988d2b2dc1eSMatthew G. Knepley     PetscBool useCeed = PETSC_FALSE, flg;
4989d2b2dc1eSMatthew G. Knepley 
4990d2b2dc1eSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_plex_use_ceed", "Use LibCEED as the FEM backend", "DMPlexSetUseCeed", useCeed, &useCeed, &flg));
4991d2b2dc1eSMatthew G. Knepley     if (flg) PetscCall(DMPlexSetUseCeed(dm, useCeed));
4992d2b2dc1eSMatthew G. Knepley   }
49939318fe57SMatthew G. Knepley   /* Create coordinate space */
49949318fe57SMatthew G. Knepley   if (created) {
499561a622f3SMatthew G. Knepley     DM_Plex  *mesh   = (DM_Plex *)dm->data;
4996e44f6aebSMatthew G. Knepley     PetscInt  degree = 1, deg;
49975515ebd3SMatthew G. Knepley     PetscInt  height = 0;
49985515ebd3SMatthew G. Knepley     DM        cdm;
4999c3db174cSMatthew G. Knepley     PetscBool flg, localize = PETSC_TRUE, sparseLocalize = PETSC_TRUE;
50009318fe57SMatthew G. Knepley 
50019566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_coord_space", "Use an FEM space for coordinates", "", coordSpace, &coordSpace, &flg));
50029566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-dm_coord_petscspace_degree", "FEM degree for coordinate space", "", degree, &degree, NULL));
5003e44f6aebSMatthew G. Knepley     PetscCall(DMGetCoordinateDegree_Internal(dm, &deg));
5004e44f6aebSMatthew G. Knepley     if (coordSpace && deg <= 1) PetscCall(DMPlexCreateCoordinateSpace(dm, degree, PETSC_TRUE, mesh->coordFunc));
50055515ebd3SMatthew G. Knepley     PetscCall(DMGetCoordinateDM(dm, &cdm));
500661a622f3SMatthew G. Knepley     if (flg && !coordSpace) {
500761a622f3SMatthew G. Knepley       PetscDS      cds;
500861a622f3SMatthew G. Knepley       PetscObject  obj;
500961a622f3SMatthew G. Knepley       PetscClassId id;
501061a622f3SMatthew G. Knepley 
50119566063dSJacob Faibussowitsch       PetscCall(DMGetDS(cdm, &cds));
50129566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
50139566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
501461a622f3SMatthew G. Knepley       if (id == PETSCFE_CLASSID) {
501561a622f3SMatthew G. Knepley         PetscContainer dummy;
501661a622f3SMatthew G. Knepley 
50179566063dSJacob Faibussowitsch         PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &dummy));
50189566063dSJacob Faibussowitsch         PetscCall(PetscObjectSetName((PetscObject)dummy, "coordinates"));
50199566063dSJacob Faibussowitsch         PetscCall(DMSetField(cdm, 0, NULL, (PetscObject)dummy));
50209566063dSJacob Faibussowitsch         PetscCall(PetscContainerDestroy(&dummy));
50219566063dSJacob Faibussowitsch         PetscCall(DMClearDS(cdm));
502261a622f3SMatthew G. Knepley       }
502361a622f3SMatthew G. Knepley       mesh->coordFunc = NULL;
502461a622f3SMatthew G. Knepley     }
5025c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_localize", "Localize mesh coordinates", "", localize, &localize, NULL));
5026c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_sparse_localize", "Localize only necessary cells", "DMSetSparseLocalize", sparseLocalize, &sparseLocalize, &flg));
5027c3db174cSMatthew G. Knepley     if (flg) PetscCall(DMSetSparseLocalize(dm, sparseLocalize));
50285515ebd3SMatthew G. Knepley     PetscCall(PetscOptionsInt("-dm_localize_height", "Localize edges and faces in addition to cells", "", height, &height, &flg));
50295515ebd3SMatthew G. Knepley     if (flg) PetscCall(DMPlexSetMaxProjectionHeight(cdm, height));
5030c3db174cSMatthew G. Knepley     if (localize) PetscCall(DMLocalizeCoordinates(dm));
50319318fe57SMatthew G. Knepley   }
503268d4fef7SMatthew G. Knepley   /* Handle DMPlex refinement */
503361a622f3SMatthew G. Knepley   remap = PETSC_TRUE;
50349566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL, 0));
50359566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
50369566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy, 0));
50379566063dSJacob Faibussowitsch   if (refine) PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
503868d4fef7SMatthew G. Knepley   if (refine && isHierarchy) {
5039acdc6f61SToby Isaac     DM *dms, coarseDM;
504068d4fef7SMatthew G. Knepley 
50419566063dSJacob Faibussowitsch     PetscCall(DMGetCoarseDM(dm, &coarseDM));
50429566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)coarseDM));
50439566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(refine, &dms));
50449566063dSJacob Faibussowitsch     PetscCall(DMRefineHierarchy(dm, refine, dms));
504568d4fef7SMatthew G. Knepley     /* Total hack since we do not pass in a pointer */
50469566063dSJacob Faibussowitsch     PetscCall(DMPlexSwap_Static(dm, dms[refine - 1]));
504768d4fef7SMatthew G. Knepley     if (refine == 1) {
50489566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[0]));
50499566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
505068d4fef7SMatthew G. Knepley     } else {
50519566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[refine - 2]));
50529566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
50539566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dms[0], dms[refine - 1]));
50549566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dms[0], PETSC_TRUE));
505568d4fef7SMatthew G. Knepley     }
50569566063dSJacob Faibussowitsch     PetscCall(DMSetCoarseDM(dms[refine - 1], coarseDM));
50579566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)coarseDM));
505868d4fef7SMatthew G. Knepley     /* Free DMs */
505968d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
5060dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
50619566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
506268d4fef7SMatthew G. Knepley     }
50639566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
506468d4fef7SMatthew G. Knepley   } else {
506568d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
50669318fe57SMatthew G. Knepley       DM             rdm;
506751a74b61SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
506868d4fef7SMatthew G. Knepley 
5069dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
50709566063dSJacob Faibussowitsch       PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
507168d4fef7SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
507269d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
5073dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
507461a622f3SMatthew G. Knepley       if (coordFunc && remap) {
50759566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
507651a74b61SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
507751a74b61SMatthew G. Knepley       }
507868d4fef7SMatthew G. Knepley     }
507968d4fef7SMatthew G. Knepley   }
50803cf6fe12SMatthew G. Knepley   /* Handle DMPlex coarsening */
50819566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL, 0));
50829566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy, 0));
5083b653a561SMatthew G. Knepley   if (coarsen && isHierarchy) {
5084b653a561SMatthew G. Knepley     DM *dms;
5085b653a561SMatthew G. Knepley 
50869566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(coarsen, &dms));
50879566063dSJacob Faibussowitsch     PetscCall(DMCoarsenHierarchy(dm, coarsen, dms));
5088b653a561SMatthew G. Knepley     /* Free DMs */
5089b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
5090dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
50919566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
5092b653a561SMatthew G. Knepley     }
50939566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
5094b653a561SMatthew G. Knepley   } else {
5095b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
50969318fe57SMatthew G. Knepley       DM             cdm;
50979318fe57SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
50983cf6fe12SMatthew G. Knepley 
5099dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
51009566063dSJacob Faibussowitsch       PetscCall(DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &cdm));
51013cf6fe12SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
510269d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &cdm));
5103dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
51049318fe57SMatthew G. Knepley       if (coordFunc) {
51059566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
51069318fe57SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
51079318fe57SMatthew G. Knepley       }
51083cf6fe12SMatthew G. Knepley     }
5109b653a561SMatthew G. Knepley   }
5110be664eb1SMatthew G. Knepley   // Handle coordinate remapping
5111be664eb1SMatthew G. Knepley   remap = PETSC_FALSE;
5112be664eb1SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_coord_remap", "Flag to control coordinate remapping", "", remap, &remap, NULL));
5113be664eb1SMatthew G. Knepley   if (remap) {
5114be664eb1SMatthew G. Knepley     DMPlexCoordMap map     = DM_COORD_MAP_NONE;
5115be664eb1SMatthew G. Knepley     PetscPointFunc mapFunc = NULL;
5116be664eb1SMatthew G. Knepley     PetscScalar    params[16];
5117f45b553cSPierre Jolivet     PetscInt       Np = PETSC_STATIC_ARRAY_LENGTH(params), cdim;
5118be664eb1SMatthew G. Knepley     MPI_Comm       comm;
5119be664eb1SMatthew G. Knepley 
5120be664eb1SMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
5121be664eb1SMatthew G. Knepley     PetscCall(DMGetCoordinateDim(dm, &cdim));
5122be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsScalarArray("-dm_coord_map_params", "Parameters for the coordinate remapping", "", params, &Np, &flg));
5123be664eb1SMatthew G. Knepley     if (!flg) Np = 0;
5124be664eb1SMatthew G. Knepley     // TODO Allow user to pass a map function by name
5125be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsEnum("-dm_coord_map", "Coordinate mapping for built-in mesh", "", DMPlexCoordMaps, (PetscEnum)map, (PetscEnum *)&map, &flg));
5126be664eb1SMatthew G. Knepley     if (flg) {
5127be664eb1SMatthew G. Knepley       switch (map) {
5128be664eb1SMatthew G. Knepley       case DM_COORD_MAP_NONE:
5129be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
5130be664eb1SMatthew G. Knepley         break;
5131be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHEAR:
5132be664eb1SMatthew G. Knepley         mapFunc = coordMap_shear;
5133be664eb1SMatthew G. Knepley         if (!Np) {
5134be664eb1SMatthew G. Knepley           Np        = cdim + 1;
5135be664eb1SMatthew G. Knepley           params[0] = 0;
5136be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
5137be664eb1SMatthew G. Knepley         }
5138be664eb1SMatthew 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);
5139be664eb1SMatthew G. Knepley         break;
5140be664eb1SMatthew G. Knepley       case DM_COORD_MAP_FLARE:
5141be664eb1SMatthew G. Knepley         mapFunc = coordMap_flare;
5142be664eb1SMatthew G. Knepley         if (!Np) {
5143be664eb1SMatthew G. Knepley           Np        = cdim + 1;
5144be664eb1SMatthew G. Knepley           params[0] = 0;
5145be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
5146be664eb1SMatthew G. Knepley         }
5147be664eb1SMatthew 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);
5148be664eb1SMatthew G. Knepley         break;
5149be664eb1SMatthew G. Knepley       case DM_COORD_MAP_ANNULUS:
5150be664eb1SMatthew G. Knepley         mapFunc = coordMap_annulus;
5151be664eb1SMatthew G. Knepley         if (!Np) {
5152be664eb1SMatthew G. Knepley           Np        = 2;
5153be664eb1SMatthew G. Knepley           params[0] = 1.;
5154be664eb1SMatthew G. Knepley           params[1] = 2.;
5155be664eb1SMatthew G. Knepley         }
5156be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The annulus coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
5157be664eb1SMatthew G. Knepley         break;
5158be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHELL:
5159be664eb1SMatthew G. Knepley         mapFunc = coordMap_shell;
5160be664eb1SMatthew G. Knepley         if (!Np) {
5161be664eb1SMatthew G. Knepley           Np        = 2;
5162be664eb1SMatthew G. Knepley           params[0] = 1.;
5163be664eb1SMatthew G. Knepley           params[1] = 2.;
5164be664eb1SMatthew G. Knepley         }
5165be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The spherical shell coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
5166be664eb1SMatthew G. Knepley         break;
5167be664eb1SMatthew G. Knepley       default:
5168be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
5169be664eb1SMatthew G. Knepley       }
5170be664eb1SMatthew G. Knepley     }
5171be664eb1SMatthew G. Knepley     if (Np) {
5172be664eb1SMatthew G. Knepley       DM      cdm;
5173be664eb1SMatthew G. Knepley       PetscDS cds;
5174be664eb1SMatthew G. Knepley 
5175be664eb1SMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
5176be664eb1SMatthew G. Knepley       PetscCall(DMGetDS(cdm, &cds));
5177be664eb1SMatthew G. Knepley       PetscCall(PetscDSSetConstants(cds, Np, params));
5178be664eb1SMatthew G. Knepley     }
5179be664eb1SMatthew G. Knepley     PetscCall(DMPlexRemapGeometry(dm, 0.0, mapFunc));
5180be664eb1SMatthew G. Knepley   }
5181909dfd52SMatthew G. Knepley   /* Handle ghost cells */
51829566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_create_fv_ghost_cells", "Flag to create finite volume ghost cells on the boundary", "DMCreate", ghostCells, &ghostCells, NULL));
5183909dfd52SMatthew G. Knepley   if (ghostCells) {
5184909dfd52SMatthew G. Knepley     DM   gdm;
5185909dfd52SMatthew G. Knepley     char lname[PETSC_MAX_PATH_LEN];
5186909dfd52SMatthew G. Knepley 
5187909dfd52SMatthew G. Knepley     lname[0] = '\0';
51889566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-dm_plex_fv_ghost_cells_label", "Label name for ghost cells boundary", "DMCreate", lname, lname, sizeof(lname), &flg));
51899566063dSJacob Faibussowitsch     PetscCall(DMPlexConstructGhostCells(dm, flg ? lname : NULL, NULL, &gdm));
519069d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &gdm));
5191909dfd52SMatthew G. Knepley   }
51926913077dSMatthew G. Knepley   /* Handle 1D order */
5193adc21957SMatthew G. Knepley   if (reorder != DM_REORDER_DEFAULT_FALSE && dim == 1) {
51946913077dSMatthew G. Knepley     DM           cdm, rdm;
51956913077dSMatthew G. Knepley     PetscDS      cds;
51966913077dSMatthew G. Knepley     PetscObject  obj;
51976913077dSMatthew G. Knepley     PetscClassId id = PETSC_OBJECT_CLASSID;
51986913077dSMatthew G. Knepley     IS           perm;
51996bc1bd01Sksagiyam     PetscInt     Nf;
52006913077dSMatthew G. Knepley     PetscBool    distributed;
52016913077dSMatthew G. Knepley 
52029566063dSJacob Faibussowitsch     PetscCall(DMPlexIsDistributed(dm, &distributed));
52039566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
52049566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
52059566063dSJacob Faibussowitsch     PetscCall(PetscDSGetNumFields(cds, &Nf));
52066913077dSMatthew G. Knepley     if (Nf) {
52079566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
52089566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
52096913077dSMatthew G. Knepley     }
52106bc1bd01Sksagiyam     if (!distributed && id != PETSCFE_CLASSID) {
52119566063dSJacob Faibussowitsch       PetscCall(DMPlexGetOrdering1D(dm, &perm));
52129566063dSJacob Faibussowitsch       PetscCall(DMPlexPermute(dm, perm, &rdm));
521369d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
52149566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&perm));
52156913077dSMatthew G. Knepley     }
52166913077dSMatthew G. Knepley   }
52173cf6fe12SMatthew G. Knepley /* Handle */
5218dd4c3f67SMatthew G. Knepley non_refine:
5219dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
522022d6dc08SStefano Zampini   char    *phases[16];
522122d6dc08SStefano Zampini   PetscInt Nphases = 16;
522222d6dc08SStefano Zampini   PetscCall(PetscOptionsStringArray("-dm_plex_option_phases", "Option phase prefixes", "DMSetFromOptions", phases, &Nphases, &flg));
5223d0609cedSBarry Smith   PetscOptionsHeadEnd();
522422d6dc08SStefano Zampini 
522522d6dc08SStefano Zampini   // Phases
522622d6dc08SStefano Zampini   if (flg) {
522722d6dc08SStefano Zampini     const char *oldPrefix;
522822d6dc08SStefano Zampini 
522922d6dc08SStefano Zampini     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &oldPrefix));
523022d6dc08SStefano Zampini     for (PetscInt ph = 0; ph < Nphases; ++ph) {
523122d6dc08SStefano Zampini       PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)dm, phases[ph]));
523222d6dc08SStefano Zampini       PetscCall(PetscInfo(dm, "Options phase %s for DM %s\n", phases[ph], dm->hdr.name));
523322d6dc08SStefano Zampini       PetscCall(DMSetFromOptions(dm));
523422d6dc08SStefano Zampini       PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldPrefix));
523522d6dc08SStefano Zampini       PetscCall(PetscFree(phases[ph]));
523622d6dc08SStefano Zampini     }
523722d6dc08SStefano Zampini   }
52383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
52390a6ba040SMatthew G. Knepley }
52400a6ba040SMatthew G. Knepley 
5241d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_Plex(DM dm, Vec *vec)
5242d71ae5a4SJacob Faibussowitsch {
5243552f7358SJed Brown   PetscFunctionBegin;
52449566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector_Section_Private(dm, vec));
52459566063dSJacob Faibussowitsch   /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */
52469566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex));
52479566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_Plex_Native));
52489566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex));
52499566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_Plex_Native));
52503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5251552f7358SJed Brown }
5252552f7358SJed Brown 
5253d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_Plex(DM dm, Vec *vec)
5254d71ae5a4SJacob Faibussowitsch {
5255552f7358SJed Brown   PetscFunctionBegin;
52569566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector_Section_Private(dm, vec));
52579566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex_Local));
52589566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex_Local));
52593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5260552f7358SJed Brown }
5261552f7358SJed Brown 
5262d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
5263d71ae5a4SJacob Faibussowitsch {
5264793f3fe5SMatthew G. Knepley   PetscInt depth, d;
5265793f3fe5SMatthew G. Knepley 
5266793f3fe5SMatthew G. Knepley   PetscFunctionBegin;
52679566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm, &depth));
5268793f3fe5SMatthew G. Knepley   if (depth == 1) {
52699566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &d));
52709566063dSJacob Faibussowitsch     if (dim == 0) PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
52719566063dSJacob Faibussowitsch     else if (dim == d) PetscCall(DMPlexGetDepthStratum(dm, 1, pStart, pEnd));
52729371c9d4SSatish Balay     else {
52739371c9d4SSatish Balay       *pStart = 0;
52749371c9d4SSatish Balay       *pEnd   = 0;
52759371c9d4SSatish Balay     }
5276793f3fe5SMatthew G. Knepley   } else {
52779566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
5278793f3fe5SMatthew G. Knepley   }
52793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5280793f3fe5SMatthew G. Knepley }
5281793f3fe5SMatthew G. Knepley 
5282d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetNeighbors_Plex(DM dm, PetscInt *nranks, const PetscMPIInt *ranks[])
5283d71ae5a4SJacob Faibussowitsch {
5284502a2867SDave May   PetscSF            sf;
52856497c311SBarry Smith   PetscMPIInt        niranks, njranks;
52866497c311SBarry Smith   PetscInt           n;
52870a19bb7dSprj-   const PetscMPIInt *iranks, *jranks;
52880a19bb7dSprj-   DM_Plex           *data = (DM_Plex *)dm->data;
5289502a2867SDave May 
52902f356facSMatthew G. Knepley   PetscFunctionBegin;
52919566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dm, &sf));
52920a19bb7dSprj-   if (!data->neighbors) {
52939566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
52949566063dSJacob Faibussowitsch     PetscCall(PetscSFGetRootRanks(sf, &njranks, &jranks, NULL, NULL, NULL));
52959566063dSJacob Faibussowitsch     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL));
52969566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors));
52979566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + 1, jranks, njranks));
52989566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks));
52990a19bb7dSprj-     n = njranks + niranks;
53009566063dSJacob Faibussowitsch     PetscCall(PetscSortRemoveDupsMPIInt(&n, data->neighbors + 1));
53010a19bb7dSprj-     /* The following cast should never fail: can't have more neighbors than PETSC_MPI_INT_MAX */
53029566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(n, data->neighbors));
53030a19bb7dSprj-   }
53040a19bb7dSprj-   if (nranks) *nranks = data->neighbors[0];
53050a19bb7dSprj-   if (ranks) {
53060a19bb7dSprj-     if (data->neighbors[0]) *ranks = data->neighbors + 1;
53070a19bb7dSprj-     else *ranks = NULL;
53080a19bb7dSprj-   }
53093ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5310502a2867SDave May }
5311502a2867SDave May 
53121eb70e55SToby Isaac PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM, DM, Mat, Vec, Vec);
53131eb70e55SToby Isaac 
5314d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_Plex(DM dm)
5315d71ae5a4SJacob Faibussowitsch {
5316552f7358SJed Brown   PetscFunctionBegin;
5317552f7358SJed Brown   dm->ops->view                      = DMView_Plex;
53182c40f234SMatthew G. Knepley   dm->ops->load                      = DMLoad_Plex;
5319552f7358SJed Brown   dm->ops->setfromoptions            = DMSetFromOptions_Plex;
532038221697SMatthew G. Knepley   dm->ops->clone                     = DMClone_Plex;
5321552f7358SJed Brown   dm->ops->setup                     = DMSetUp_Plex;
53221bb6d2a8SBarry Smith   dm->ops->createlocalsection        = DMCreateLocalSection_Plex;
5323adc21957SMatthew G. Knepley   dm->ops->createsectionpermutation  = DMCreateSectionPermutation_Plex;
532466ad2231SToby Isaac   dm->ops->createdefaultconstraints  = DMCreateDefaultConstraints_Plex;
5325552f7358SJed Brown   dm->ops->createglobalvector        = DMCreateGlobalVector_Plex;
5326552f7358SJed Brown   dm->ops->createlocalvector         = DMCreateLocalVector_Plex;
5327184d77edSJed Brown   dm->ops->getlocaltoglobalmapping   = NULL;
53280298fd71SBarry Smith   dm->ops->createfieldis             = NULL;
5329552f7358SJed Brown   dm->ops->createcoordinatedm        = DMCreateCoordinateDM_Plex;
5330f19dbd58SToby Isaac   dm->ops->createcoordinatefield     = DMCreateCoordinateField_Plex;
53310a6ba040SMatthew G. Knepley   dm->ops->getcoloring               = NULL;
5332552f7358SJed Brown   dm->ops->creatematrix              = DMCreateMatrix_Plex;
5333bceba477SMatthew G. Knepley   dm->ops->createinterpolation       = DMCreateInterpolation_Plex;
5334bd041c0cSMatthew G. Knepley   dm->ops->createmassmatrix          = DMCreateMassMatrix_Plex;
5335b4937a87SMatthew G. Knepley   dm->ops->createmassmatrixlumped    = DMCreateMassMatrixLumped_Plex;
53365a84ad33SLisandro Dalcin   dm->ops->createinjection           = DMCreateInjection_Plex;
5337552f7358SJed Brown   dm->ops->refine                    = DMRefine_Plex;
53380a6ba040SMatthew G. Knepley   dm->ops->coarsen                   = DMCoarsen_Plex;
53390a6ba040SMatthew G. Knepley   dm->ops->refinehierarchy           = DMRefineHierarchy_Plex;
5340b653a561SMatthew G. Knepley   dm->ops->coarsenhierarchy          = DMCoarsenHierarchy_Plex;
5341d410b0cfSMatthew G. Knepley   dm->ops->extrude                   = DMExtrude_Plex;
53420298fd71SBarry Smith   dm->ops->globaltolocalbegin        = NULL;
53430298fd71SBarry Smith   dm->ops->globaltolocalend          = NULL;
53440298fd71SBarry Smith   dm->ops->localtoglobalbegin        = NULL;
53450298fd71SBarry Smith   dm->ops->localtoglobalend          = NULL;
5346552f7358SJed Brown   dm->ops->destroy                   = DMDestroy_Plex;
5347552f7358SJed Brown   dm->ops->createsubdm               = DMCreateSubDM_Plex;
53482adcc780SMatthew G. Knepley   dm->ops->createsuperdm             = DMCreateSuperDM_Plex;
5349793f3fe5SMatthew G. Knepley   dm->ops->getdimpoints              = DMGetDimPoints_Plex;
5350552f7358SJed Brown   dm->ops->locatepoints              = DMLocatePoints_Plex;
53510709b2feSToby Isaac   dm->ops->projectfunctionlocal      = DMProjectFunctionLocal_Plex;
53520709b2feSToby Isaac   dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_Plex;
5353bfc4295aSToby Isaac   dm->ops->projectfieldlocal         = DMProjectFieldLocal_Plex;
53548c6c5593SMatthew G. Knepley   dm->ops->projectfieldlabellocal    = DMProjectFieldLabelLocal_Plex;
5355ece3a9fcSMatthew G. Knepley   dm->ops->projectbdfieldlabellocal  = DMProjectBdFieldLabelLocal_Plex;
53560709b2feSToby Isaac   dm->ops->computel2diff             = DMComputeL2Diff_Plex;
5357b698f381SToby Isaac   dm->ops->computel2gradientdiff     = DMComputeL2GradientDiff_Plex;
53582a16baeaSToby Isaac   dm->ops->computel2fielddiff        = DMComputeL2FieldDiff_Plex;
535928d58a37SPierre Jolivet   dm->ops->getneighbors              = DMGetNeighbors_Plex;
53606c6a6b79SMatthew G. Knepley   dm->ops->getlocalboundingbox       = DMGetLocalBoundingBox_Coordinates;
5361907a3e9cSStefano Zampini   dm->ops->createdomaindecomposition = DMCreateDomainDecomposition_Plex;
5362907a3e9cSStefano Zampini   dm->ops->createddscatters          = DMCreateDomainDecompositionScatters_Plex;
53639566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_Plex));
53646c51210dSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", DMPlexInsertTimeDerivativeBoundaryValues_Plex));
53659566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", DMSetUpGLVisViewer_Plex));
53669566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_Plex));
53679566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", DMPlexDistributeGetDefault_Plex));
53689566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", DMPlexDistributeSetDefault_Plex));
53696bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", DMPlexReorderGetDefault_Plex));
53706bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", DMPlexReorderSetDefault_Plex));
5371adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetDefault_C", DMReorderSectionGetDefault_Plex));
5372adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetDefault_C", DMReorderSectionSetDefault_Plex));
5373adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetType_C", DMReorderSectionGetType_Plex));
5374adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetType_C", DMReorderSectionSetType_Plex));
53759566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", DMInterpolateSolution_Plex));
5376c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMPlexGetOverlap_Plex));
5377c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", DMPlexSetOverlap_Plex));
5378d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", DMPlexGetUseCeed_Plex));
5379d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", DMPlexSetUseCeed_Plex));
53803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5381552f7358SJed Brown }
5382552f7358SJed Brown 
5383d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode DMClone_Plex(DM dm, DM *newdm)
5384d71ae5a4SJacob Faibussowitsch {
538563a16f15SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *)dm->data;
53861fca310dSJames Wright   const PetscSF *face_sfs;
53871fca310dSJames Wright   PetscInt       num_face_sfs;
538863a16f15SMatthew G. Knepley 
538963a16f15SMatthew G. Knepley   PetscFunctionBegin;
539063a16f15SMatthew G. Knepley   mesh->refct++;
539163a16f15SMatthew G. Knepley   (*newdm)->data = mesh;
53921fca310dSJames Wright   PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &face_sfs));
53931fca310dSJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(*newdm, num_face_sfs, (PetscSF *)face_sfs));
53949566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)*newdm, DMPLEX));
53959566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(*newdm));
53963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
539763a16f15SMatthew G. Knepley }
539863a16f15SMatthew G. Knepley 
53998818961aSMatthew G Knepley /*MC
5400a1cb98faSBarry Smith   DMPLEX = "plex" - A `DM` object that encapsulates an unstructured mesh, or CW Complex, which can be expressed using a Hasse Diagram.
540120f4b53cSBarry Smith                     In the local representation, `Vec`s contain all unknowns in the interior and shared boundary. This is
54028818961aSMatthew G Knepley                     specified by a PetscSection object. Ownership in the global representation is determined by
5403a1cb98faSBarry Smith                     ownership of the underlying `DMPLEX` points. This is specified by another `PetscSection` object.
54048818961aSMatthew G Knepley 
5405e5893cccSMatthew G. Knepley   Options Database Keys:
5406250712c9SMatthew G. Knepley + -dm_refine_pre                     - Refine mesh before distribution
5407250712c9SMatthew G. Knepley + -dm_refine_uniform_pre             - Choose uniform or generator-based refinement
5408250712c9SMatthew G. Knepley + -dm_refine_volume_limit_pre        - Cell volume limit after pre-refinement using generator
5409250712c9SMatthew G. Knepley . -dm_distribute                     - Distribute mesh across processes
5410250712c9SMatthew G. Knepley . -dm_distribute_overlap             - Number of cells to overlap for distribution
5411250712c9SMatthew G. Knepley . -dm_refine                         - Refine mesh after distribution
5412c3db174cSMatthew G. Knepley . -dm_localize <bool>                - Whether to localize coordinates for periodic meshes
5413c3db174cSMatthew G. Knepley . -dm_sparse_localize <bool>         - Whether to only localize cells on the periodic boundary
5414250712c9SMatthew G. Knepley . -dm_plex_hash_location             - Use grid hashing for point location
5415ddce0771SMatthew G. Knepley . -dm_plex_hash_box_faces <n,m,p>    - The number of divisions in each direction of the grid hash
5416f12cf164SMatthew G. Knepley . -dm_plex_partition_balance         - Attempt to evenly divide points on partition boundary between processes
5417f12cf164SMatthew G. Knepley . -dm_plex_remesh_bd                 - Allow changes to the boundary on remeshing
5418d5b43468SJose E. Roman . -dm_plex_max_projection_height     - Maximum mesh point height used to project locally
5419f12cf164SMatthew G. Knepley . -dm_plex_regular_refinement        - Use special nested projection algorithm for regular refinement
5420d02c7345SMatthew G. Knepley . -dm_plex_reorder_section           - Use specialized blocking if available
5421aaa8cc7dSPierre Jolivet . -dm_plex_check_all                 - Perform all checks below
5422f12cf164SMatthew G. Knepley . -dm_plex_check_symmetry            - Check that the adjacency information in the mesh is symmetric
5423f12cf164SMatthew G. Knepley . -dm_plex_check_skeleton <celltype> - Check that each cell has the correct number of vertices
5424f12cf164SMatthew 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
5425f12cf164SMatthew G. Knepley . -dm_plex_check_geometry            - Check that cells have positive volume
5426f12cf164SMatthew G. Knepley . -dm_view :mesh.tex:ascii_latex     - View the mesh in LaTeX/TikZ
5427e5893cccSMatthew G. Knepley . -dm_plex_view_scale <num>          - Scale the TikZ
54285962854dSMatthew G. Knepley . -dm_plex_print_fem <num>           - View FEM assembly information, such as element vectors and matrices
54295962854dSMatthew G. Knepley - -dm_plex_print_fvm <num>           - View FVM assembly information, such as flux updates
5430e5893cccSMatthew G. Knepley 
54318818961aSMatthew G Knepley   Level: intermediate
54328818961aSMatthew G Knepley 
54331cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMPlexCreate()`, `DMCreate()`, `DMSetType()`, `PetscSection`
54348818961aSMatthew G Knepley M*/
54358818961aSMatthew G Knepley 
5436d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm)
5437d71ae5a4SJacob Faibussowitsch {
5438552f7358SJed Brown   DM_Plex *mesh;
5439412e9a14SMatthew G. Knepley   PetscInt unit;
5440552f7358SJed Brown 
5441552f7358SJed Brown   PetscFunctionBegin;
5442f39ec787SMatthew G. Knepley   PetscCall(PetscCitationsRegister(PlexCitation, &Plexcite));
5443552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
54444dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mesh));
5445adc21957SMatthew G. Knepley   dm->reorderSection = DM_REORDER_DEFAULT_NOTSET;
5446552f7358SJed Brown   dm->data           = mesh;
5447552f7358SJed Brown 
5448552f7358SJed Brown   mesh->refct = 1;
54499566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection));
54509566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection));
5451552f7358SJed Brown   mesh->refinementUniform      = PETSC_TRUE;
5452552f7358SJed Brown   mesh->refinementLimit        = -1.0;
5453e600fa54SMatthew G. Knepley   mesh->distDefault            = PETSC_TRUE;
5454adc21957SMatthew G. Knepley   mesh->reorderDefault         = DM_REORDER_DEFAULT_NOTSET;
54551d1f2f2aSksagiyam   mesh->distributionName       = NULL;
54567d0f5628SVaclav Hapla   mesh->interpolated           = DMPLEX_INTERPOLATED_INVALID;
54577d0f5628SVaclav Hapla   mesh->interpolatedCollective = DMPLEX_INTERPOLATED_INVALID;
5458552f7358SJed Brown 
54599566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner));
54602e62ab5aSMatthew G. Knepley   mesh->remeshBd = PETSC_FALSE;
5461d9deefdfSMatthew G. Knepley 
54628865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
5463552f7358SJed Brown 
5464df0420ecSMatthew G. Knepley   mesh->depthState    = -1;
5465ba2698f1SMatthew G. Knepley   mesh->celltypeState = -1;
54666113b454SMatthew G. Knepley   mesh->printTol      = 1.0e-10;
5467c29ce622SStefano Zampini   mesh->nonempty_comm = MPI_COMM_SELF;
5468552f7358SJed Brown 
54699566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
54703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5471552f7358SJed Brown }
5472552f7358SJed Brown 
5473552f7358SJed Brown /*@
5474a1cb98faSBarry Smith   DMPlexCreate - Creates a `DMPLEX` object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
5475552f7358SJed Brown 
5476d083f849SBarry Smith   Collective
5477552f7358SJed Brown 
5478552f7358SJed Brown   Input Parameter:
5479a1cb98faSBarry Smith . comm - The communicator for the `DMPLEX` object
5480552f7358SJed Brown 
5481552f7358SJed Brown   Output Parameter:
5482a1cb98faSBarry Smith . mesh - The `DMPLEX` object
5483552f7358SJed Brown 
5484552f7358SJed Brown   Level: beginner
5485552f7358SJed Brown 
548642747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMCreate()`, `DMSetType()`
5487552f7358SJed Brown @*/
5488d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
5489d71ae5a4SJacob Faibussowitsch {
5490552f7358SJed Brown   PetscFunctionBegin;
54914f572ea9SToby Isaac   PetscAssertPointer(mesh, 2);
54929566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, mesh));
54939566063dSJacob Faibussowitsch   PetscCall(DMSetType(*mesh, DMPLEX));
54943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5495552f7358SJed Brown }
5496552f7358SJed Brown 
5497b09969d6SVaclav Hapla /*@C
5498b0fe842aSMatthew 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
5499a1cb98faSBarry Smith 
550020f4b53cSBarry Smith   Collective; No Fortran Support
5501b09969d6SVaclav Hapla 
5502b09969d6SVaclav Hapla   Input Parameters:
5503a1cb98faSBarry Smith + dm          - The `DM`
5504b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5505a1cb98faSBarry Smith . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5506a1cb98faSBarry Smith . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5507b09969d6SVaclav Hapla . numCorners  - The number of vertices for each cell
55085e488331SVaclav Hapla - cells       - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5509b09969d6SVaclav Hapla 
5510be8c289dSNicolas Barral   Output Parameters:
5511a1cb98faSBarry Smith + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5512be8c289dSNicolas Barral - verticesAdjSaved - (Optional) vertex adjacency array
5513b09969d6SVaclav Hapla 
5514b09969d6SVaclav Hapla   Level: advanced
5515b09969d6SVaclav Hapla 
5516a1cb98faSBarry Smith   Notes:
5517a1cb98faSBarry Smith   Two triangles sharing a face
5518a1cb98faSBarry Smith .vb
5519a1cb98faSBarry Smith 
5520a1cb98faSBarry Smith         2
5521a1cb98faSBarry Smith       / | \
5522a1cb98faSBarry Smith      /  |  \
5523a1cb98faSBarry Smith     /   |   \
5524a1cb98faSBarry Smith    0  0 | 1  3
5525a1cb98faSBarry Smith     \   |   /
5526a1cb98faSBarry Smith      \  |  /
5527a1cb98faSBarry Smith       \ | /
5528a1cb98faSBarry Smith         1
5529a1cb98faSBarry Smith .ve
5530a1cb98faSBarry Smith   would have input
5531a1cb98faSBarry Smith .vb
5532a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5533a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5534a1cb98faSBarry Smith .ve
5535a1cb98faSBarry Smith   which would result in the `DMPLEX`
5536a1cb98faSBarry Smith .vb
5537a1cb98faSBarry Smith 
5538a1cb98faSBarry Smith         4
5539a1cb98faSBarry Smith       / | \
5540a1cb98faSBarry Smith      /  |  \
5541a1cb98faSBarry Smith     /   |   \
5542a1cb98faSBarry Smith    2  0 | 1  5
5543a1cb98faSBarry Smith     \   |   /
5544a1cb98faSBarry Smith      \  |  /
5545a1cb98faSBarry Smith       \ | /
5546a1cb98faSBarry Smith         3
5547a1cb98faSBarry Smith .ve
5548a1cb98faSBarry Smith 
5549a1cb98faSBarry Smith   Vertices are implicitly numbered consecutively 0,...,NVertices.
5550a1cb98faSBarry Smith   Each rank owns a chunk of numVertices consecutive vertices.
5551a1cb98faSBarry Smith   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5552a1cb98faSBarry Smith   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
5553a1cb98faSBarry Smith   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5554a1cb98faSBarry Smith 
5555a1cb98faSBarry Smith   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5556a1cb98faSBarry Smith 
55571cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5558a1cb98faSBarry Smith           `PetscSF`
5559b09969d6SVaclav Hapla @*/
5560d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellListParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, const PetscInt cells[], PetscSF *vertexSF, PetscInt **verticesAdjSaved)
5561d71ae5a4SJacob Faibussowitsch {
55622464107aSksagiyam   PetscSF     sfPoint;
55632464107aSksagiyam   PetscLayout layout;
556482fb893eSVaclav Hapla   PetscInt    numVerticesAdj, *verticesAdj, *cones, c, p;
5565a47d0d45SMatthew G. Knepley 
5566a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
556725b6865aSVaclav Hapla   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
55689566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
556925b6865aSVaclav Hapla   /* Get/check global number of vertices */
557025b6865aSVaclav Hapla   {
557125b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
557225b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
557325b6865aSVaclav Hapla 
557425b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
55751690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
55769371c9d4SSatish Balay     for (i = 0; i < len; i++)
55779371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
557825b6865aSVaclav Hapla     ++NVerticesInCells;
5579462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
558025b6865aSVaclav Hapla 
558125b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
55829371c9d4SSatish Balay     else
55839371c9d4SSatish 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);
558425b6865aSVaclav Hapla   }
55859079aca8SVaclav Hapla   /* Count locally unique vertices */
55869079aca8SVaclav Hapla   {
55879079aca8SVaclav Hapla     PetscHSetI vhash;
55889079aca8SVaclav Hapla     PetscInt   off = 0;
55899079aca8SVaclav Hapla 
55909566063dSJacob Faibussowitsch     PetscCall(PetscHSetICreate(&vhash));
5591a47d0d45SMatthew G. Knepley     for (c = 0; c < numCells; ++c) {
559248a46eb9SPierre Jolivet       for (p = 0; p < numCorners; ++p) PetscCall(PetscHSetIAdd(vhash, cells[c * numCorners + p]));
5593a47d0d45SMatthew G. Knepley     }
55949566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
55959566063dSJacob Faibussowitsch     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5596ad540459SPierre Jolivet     else verticesAdj = *verticesAdjSaved;
55979566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
55989566063dSJacob Faibussowitsch     PetscCall(PetscHSetIDestroy(&vhash));
559963a3b9bcSJacob Faibussowitsch     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
5600a47d0d45SMatthew G. Knepley   }
56019566063dSJacob Faibussowitsch   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5602a47d0d45SMatthew G. Knepley   /* Create cones */
56039566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
56049566063dSJacob Faibussowitsch   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
56059566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
56069566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5607a47d0d45SMatthew G. Knepley   for (c = 0; c < numCells; ++c) {
5608a47d0d45SMatthew G. Knepley     for (p = 0; p < numCorners; ++p) {
5609a47d0d45SMatthew G. Knepley       const PetscInt gv = cells[c * numCorners + p];
5610a47d0d45SMatthew G. Knepley       PetscInt       lv;
5611a47d0d45SMatthew G. Knepley 
56129079aca8SVaclav Hapla       /* Positions within verticesAdj form 0-based local vertex numbering;
56139079aca8SVaclav Hapla          we need to shift it by numCells to get correct DAG points (cells go first) */
56149566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
561563a3b9bcSJacob Faibussowitsch       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5616961cfab0SVaclav Hapla       cones[c * numCorners + p] = lv + numCells;
5617a47d0d45SMatthew G. Knepley     }
5618a47d0d45SMatthew G. Knepley   }
56192464107aSksagiyam   /* Build point sf */
56209566063dSJacob Faibussowitsch   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
56219566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetSize(layout, NVertices));
56229566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
56239566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetBlockSize(layout, 1));
56249566063dSJacob Faibussowitsch   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
56259566063dSJacob Faibussowitsch   PetscCall(PetscLayoutDestroy(&layout));
56269566063dSJacob Faibussowitsch   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
56279566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
56282464107aSksagiyam   if (dm->sf) {
56292464107aSksagiyam     const char *prefix;
56302464107aSksagiyam 
56319566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
56329566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
56332464107aSksagiyam   }
56349566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sfPoint));
56359566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&sfPoint));
5636f4f49eeaSPierre Jolivet   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
5637a47d0d45SMatthew G. Knepley   /* Fill in the rest of the topology structure */
56389566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
56399566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
56409566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
56413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5642a47d0d45SMatthew G. Knepley }
5643a47d0d45SMatthew G. Knepley 
5644b0fe842aSMatthew G. Knepley /*@C
5645b0fe842aSMatthew G. Knepley   DMPlexBuildFromCellSectionParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) allowing multiple celltypes
5646b0fe842aSMatthew G. Knepley 
5647b0fe842aSMatthew G. Knepley   Collective; No Fortran Support
5648b0fe842aSMatthew G. Knepley 
5649b0fe842aSMatthew G. Knepley   Input Parameters:
5650b0fe842aSMatthew G. Knepley + dm          - The `DM`
5651b0fe842aSMatthew G. Knepley . numCells    - The number of cells owned by this process
5652b0fe842aSMatthew G. Knepley . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5653b0fe842aSMatthew G. Knepley . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5654b0fe842aSMatthew G. Knepley . cellSection - The `PetscSection` giving the number of vertices for each cell (layout of cells)
5655b0fe842aSMatthew G. Knepley - cells       - An array of the global vertex numbers for each cell
5656b0fe842aSMatthew G. Knepley 
5657b0fe842aSMatthew G. Knepley   Output Parameters:
5658b0fe842aSMatthew G. Knepley + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5659b0fe842aSMatthew G. Knepley - verticesAdjSaved - (Optional) vertex adjacency array
5660b0fe842aSMatthew G. Knepley 
5661b0fe842aSMatthew G. Knepley   Level: advanced
5662b0fe842aSMatthew G. Knepley 
5663b0fe842aSMatthew G. Knepley   Notes:
5664b0fe842aSMatthew G. Knepley   A triangle and quadrilateral sharing a face
5665b0fe842aSMatthew G. Knepley .vb
5666b0fe842aSMatthew G. Knepley         2----------3
5667b0fe842aSMatthew G. Knepley       / |          |
5668b0fe842aSMatthew G. Knepley      /  |          |
5669b0fe842aSMatthew G. Knepley     /   |          |
5670b0fe842aSMatthew G. Knepley    0  0 |     1    |
5671b0fe842aSMatthew G. Knepley     \   |          |
5672b0fe842aSMatthew G. Knepley      \  |          |
5673b0fe842aSMatthew G. Knepley       \ |          |
5674b0fe842aSMatthew G. Knepley         1----------4
5675b0fe842aSMatthew G. Knepley .ve
5676b0fe842aSMatthew G. Knepley   would have input
5677b0fe842aSMatthew G. Knepley .vb
5678b0fe842aSMatthew G. Knepley   numCells = 2, numVertices = 5
5679b0fe842aSMatthew G. Knepley   cells = [0 1 2  1 4 3 2]
5680b0fe842aSMatthew G. Knepley .ve
5681b0fe842aSMatthew G. Knepley   which would result in the `DMPLEX`
5682b0fe842aSMatthew G. Knepley .vb
5683b0fe842aSMatthew G. Knepley         4----------5
5684b0fe842aSMatthew G. Knepley       / |          |
5685b0fe842aSMatthew G. Knepley      /  |          |
5686b0fe842aSMatthew G. Knepley     /   |          |
5687b0fe842aSMatthew G. Knepley    2  0 |     1    |
5688b0fe842aSMatthew G. Knepley     \   |          |
5689b0fe842aSMatthew G. Knepley      \  |          |
5690b0fe842aSMatthew G. Knepley       \ |          |
5691b0fe842aSMatthew G. Knepley         3----------6
5692b0fe842aSMatthew G. Knepley .ve
5693b0fe842aSMatthew G. Knepley 
5694b0fe842aSMatthew G. Knepley   Vertices are implicitly numbered consecutively 0,...,NVertices.
5695b0fe842aSMatthew G. Knepley   Each rank owns a chunk of numVertices consecutive vertices.
5696b0fe842aSMatthew G. Knepley   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5697b0fe842aSMatthew 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.
5698b0fe842aSMatthew G. Knepley   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5699b0fe842aSMatthew G. Knepley 
5700b0fe842aSMatthew G. Knepley   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5701b0fe842aSMatthew G. Knepley 
5702b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexCreateFromCellSectionParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5703b0fe842aSMatthew G. Knepley           `PetscSF`
5704b0fe842aSMatthew G. Knepley @*/
5705b0fe842aSMatthew G. Knepley PetscErrorCode DMPlexBuildFromCellSectionParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, const PetscInt cells[], PetscSF *vertexSF, PetscInt **verticesAdjSaved)
5706b0fe842aSMatthew G. Knepley {
5707b0fe842aSMatthew G. Knepley   PetscSF     sfPoint;
5708b0fe842aSMatthew G. Knepley   PetscLayout layout;
5709b0fe842aSMatthew G. Knepley   PetscInt    numVerticesAdj, *verticesAdj, *cones, cStart, cEnd, len;
5710b0fe842aSMatthew G. Knepley 
5711b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
5712b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
5713b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
5714b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetChart(cellSection, &cStart, &cEnd));
5715b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(cellSection, &len));
5716b0fe842aSMatthew 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);
5717b0fe842aSMatthew G. Knepley   /* Get/check global number of vertices */
5718b0fe842aSMatthew G. Knepley   {
5719b0fe842aSMatthew G. Knepley     PetscInt NVerticesInCells;
5720b0fe842aSMatthew G. Knepley 
5721b0fe842aSMatthew G. Knepley     /* NVerticesInCells = max(cells) + 1 */
5722b0fe842aSMatthew G. Knepley     NVerticesInCells = PETSC_MIN_INT;
5723b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++)
5724b0fe842aSMatthew G. Knepley       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
5725b0fe842aSMatthew G. Knepley     ++NVerticesInCells;
5726b0fe842aSMatthew G. Knepley     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
5727b0fe842aSMatthew G. Knepley 
5728b0fe842aSMatthew G. Knepley     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
5729b0fe842aSMatthew G. Knepley     else
5730b0fe842aSMatthew 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);
5731b0fe842aSMatthew G. Knepley   }
5732b0fe842aSMatthew G. Knepley   /* Count locally unique vertices */
5733b0fe842aSMatthew G. Knepley   {
5734b0fe842aSMatthew G. Knepley     PetscHSetI vhash;
5735b0fe842aSMatthew G. Knepley     PetscInt   off = 0;
5736b0fe842aSMatthew G. Knepley 
5737b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetICreate(&vhash));
5738b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++) PetscCall(PetscHSetIAdd(vhash, cells[i]));
5739b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
5740b0fe842aSMatthew G. Knepley     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5741b0fe842aSMatthew G. Knepley     else verticesAdj = *verticesAdjSaved;
5742b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
5743b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIDestroy(&vhash));
5744b0fe842aSMatthew 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);
5745b0fe842aSMatthew G. Knepley   }
5746b0fe842aSMatthew G. Knepley   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5747b0fe842aSMatthew G. Knepley   /* Create cones */
5748b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
5749b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
5750b0fe842aSMatthew G. Knepley     PetscInt dof;
5751b0fe842aSMatthew G. Knepley 
5752b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
5753b0fe842aSMatthew G. Knepley     PetscCall(DMPlexSetConeSize(dm, c, dof));
5754b0fe842aSMatthew G. Knepley   }
5755b0fe842aSMatthew G. Knepley   PetscCall(DMSetUp(dm));
5756b0fe842aSMatthew G. Knepley   PetscCall(DMPlexGetCones(dm, &cones));
5757b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
5758b0fe842aSMatthew G. Knepley     PetscInt dof, off;
5759b0fe842aSMatthew G. Knepley 
5760b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
5761b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetOffset(cellSection, c, &off));
5762b0fe842aSMatthew G. Knepley     for (PetscInt p = off; p < off + dof; ++p) {
5763b0fe842aSMatthew G. Knepley       const PetscInt gv = cells[p];
5764b0fe842aSMatthew G. Knepley       PetscInt       lv;
5765b0fe842aSMatthew G. Knepley 
5766b0fe842aSMatthew G. Knepley       /* Positions within verticesAdj form 0-based local vertex numbering;
5767b0fe842aSMatthew G. Knepley          we need to shift it by numCells to get correct DAG points (cells go first) */
5768b0fe842aSMatthew G. Knepley       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
5769b0fe842aSMatthew G. Knepley       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5770b0fe842aSMatthew G. Knepley       cones[p] = lv + numCells;
5771b0fe842aSMatthew G. Knepley     }
5772b0fe842aSMatthew G. Knepley   }
5773b0fe842aSMatthew G. Knepley   /* Build point sf */
5774b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
5775b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetSize(layout, NVertices));
5776b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
5777b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetBlockSize(layout, 1));
5778b0fe842aSMatthew G. Knepley   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
5779b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutDestroy(&layout));
5780b0fe842aSMatthew G. Knepley   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
5781b0fe842aSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
5782b0fe842aSMatthew G. Knepley   if (dm->sf) {
5783b0fe842aSMatthew G. Knepley     const char *prefix;
5784b0fe842aSMatthew G. Knepley 
5785b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
5786b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
5787b0fe842aSMatthew G. Knepley   }
5788b0fe842aSMatthew G. Knepley   PetscCall(DMSetPointSF(dm, sfPoint));
5789b0fe842aSMatthew G. Knepley   PetscCall(PetscSFDestroy(&sfPoint));
5790b0fe842aSMatthew G. Knepley   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
5791b0fe842aSMatthew G. Knepley   /* Fill in the rest of the topology structure */
5792b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSymmetrize(dm));
5793b0fe842aSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
5794b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
5795b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
5796b0fe842aSMatthew G. Knepley }
5797b0fe842aSMatthew G. Knepley 
5798cc4c1da9SBarry Smith /*@
5799a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellListParallel - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
5800a1cb98faSBarry Smith 
580120f4b53cSBarry Smith   Collective; No Fortran Support
5802b09969d6SVaclav Hapla 
5803b09969d6SVaclav Hapla   Input Parameters:
5804a1cb98faSBarry Smith + dm           - The `DM`
5805b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
5806a1cb98faSBarry Smith . sfVert       - `PetscSF` describing complete vertex ownership
5807b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5808b09969d6SVaclav Hapla 
5809b09969d6SVaclav Hapla   Level: advanced
5810b09969d6SVaclav Hapla 
58111cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellListParallel()`
5812b09969d6SVaclav Hapla @*/
5813d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellListParallel(DM dm, PetscInt spaceDim, PetscSF sfVert, const PetscReal vertexCoords[])
5814d71ae5a4SJacob Faibussowitsch {
5815a47d0d45SMatthew G. Knepley   PetscSection coordSection;
5816a47d0d45SMatthew G. Knepley   Vec          coordinates;
5817a47d0d45SMatthew G. Knepley   PetscScalar *coords;
58181edcf0b2SVaclav Hapla   PetscInt     numVertices, numVerticesAdj, coordSize, v, vStart, vEnd;
5819835f2295SStefano Zampini   PetscMPIInt  spaceDimi;
5820a47d0d45SMatthew G. Knepley 
5821a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
58229566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
58239566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
58241dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
58259566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
58269566063dSJacob Faibussowitsch   PetscCall(PetscSFGetGraph(sfVert, &numVertices, &numVerticesAdj, NULL, NULL));
58271dca8a05SBarry 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);
58289566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
58299566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
58309566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
58319566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
58321edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
58339566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
58349566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
5835a47d0d45SMatthew G. Knepley   }
58369566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
58379566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
58389566063dSJacob Faibussowitsch   PetscCall(VecCreate(PetscObjectComm((PetscObject)dm), &coordinates));
58399566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
58409566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
58419566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
58429566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
58439566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
5844a47d0d45SMatthew G. Knepley   {
5845a47d0d45SMatthew G. Knepley     MPI_Datatype coordtype;
5846a47d0d45SMatthew G. Knepley 
5847a47d0d45SMatthew G. Knepley     /* Need a temp buffer for coords if we have complex/single */
5848835f2295SStefano Zampini     PetscCall(PetscMPIIntCast(spaceDim, &spaceDimi));
5849835f2295SStefano Zampini     PetscCallMPI(MPI_Type_contiguous(spaceDimi, MPIU_SCALAR, &coordtype));
58509566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_commit(&coordtype));
585121016a8bSBarry Smith #if defined(PETSC_USE_COMPLEX)
585221016a8bSBarry Smith     {
585321016a8bSBarry Smith       PetscScalar *svertexCoords;
585421016a8bSBarry Smith       PetscInt     i;
58559566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * spaceDim, &svertexCoords));
58563612f820SVaclav Hapla       for (i = 0; i < numVertices * spaceDim; i++) svertexCoords[i] = vertexCoords[i];
58579566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
58589566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
58599566063dSJacob Faibussowitsch       PetscCall(PetscFree(svertexCoords));
586021016a8bSBarry Smith     }
586121016a8bSBarry Smith #else
58629566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastBegin(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
58639566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastEnd(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
586421016a8bSBarry Smith #endif
58659566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_free(&coordtype));
5866a47d0d45SMatthew G. Knepley   }
58679566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
58689566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
58699566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
58709566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
58713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5872a47d0d45SMatthew G. Knepley }
5873a47d0d45SMatthew G. Knepley 
5874c3edce3dSSatish Balay /*@
5875b0fe842aSMatthew 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
5876a1cb98faSBarry Smith 
5877a1cb98faSBarry Smith   Collective
5878a47d0d45SMatthew G. Knepley 
5879a47d0d45SMatthew G. Knepley   Input Parameters:
5880a47d0d45SMatthew G. Knepley + comm         - The communicator
5881a47d0d45SMatthew G. Knepley . dim          - The topological dimension of the mesh
5882a47d0d45SMatthew G. Knepley . numCells     - The number of cells owned by this process
5883a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
5884a1cb98faSBarry Smith . NVertices    - The global number of vertices, or `PETSC_DECIDE`
5885a47d0d45SMatthew G. Knepley . numCorners   - The number of vertices for each cell
5886a47d0d45SMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
5887a47d0d45SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5888a47d0d45SMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
5889a47d0d45SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5890a47d0d45SMatthew G. Knepley 
5891d8d19677SJose E. Roman   Output Parameters:
5892a1cb98faSBarry Smith + dm          - The `DM`
5893a1cb98faSBarry Smith . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
589460225df5SJacob Faibussowitsch - verticesAdj - (Optional) vertex adjacency array
5895a47d0d45SMatthew G. Knepley 
5896b09969d6SVaclav Hapla   Level: intermediate
5897a47d0d45SMatthew G. Knepley 
5898a1cb98faSBarry Smith   Notes:
5899a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
5900a1cb98faSBarry Smith   `DMPlexBuildFromCellListParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
5901a1cb98faSBarry Smith 
5902a1cb98faSBarry Smith   See `DMPlexBuildFromCellListParallel()` for an example and details about the topology-related parameters.
5903a1cb98faSBarry Smith 
5904a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
5905a1cb98faSBarry Smith 
59061cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
5907a47d0d45SMatthew G. Knepley @*/
5908d71ae5a4SJacob 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)
5909d71ae5a4SJacob Faibussowitsch {
5910a47d0d45SMatthew G. Knepley   PetscSF sfVert;
5911a47d0d45SMatthew G. Knepley 
5912a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
59139566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
59149566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
5915a47d0d45SMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
5916064a246eSJacob Faibussowitsch   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
59179566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
59189566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildFromCellListParallel(*dm, numCells, numVertices, NVertices, numCorners, cells, &sfVert, verticesAdj));
5919a47d0d45SMatthew G. Knepley   if (interpolate) {
59205fd9971aSMatthew G. Knepley     DM idm;
5921a47d0d45SMatthew G. Knepley 
59229566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
59239566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
5924a47d0d45SMatthew G. Knepley     *dm = idm;
5925a47d0d45SMatthew G. Knepley   }
59269566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
592718d54ad4SMichael Lange   if (vertexSF) *vertexSF = sfVert;
59289566063dSJacob Faibussowitsch   else PetscCall(PetscSFDestroy(&sfVert));
59293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5930a47d0d45SMatthew G. Knepley }
5931a47d0d45SMatthew G. Knepley 
5932cc4c1da9SBarry Smith /*@
5933b0fe842aSMatthew G. Knepley   DMPlexCreateFromCellSectionParallel - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) and supports multiple celltypes
5934b0fe842aSMatthew G. Knepley 
5935b0fe842aSMatthew G. Knepley   Collective
5936b0fe842aSMatthew G. Knepley 
5937b0fe842aSMatthew G. Knepley   Input Parameters:
5938b0fe842aSMatthew G. Knepley + comm         - The communicator
5939b0fe842aSMatthew G. Knepley . dim          - The topological dimension of the mesh
5940b0fe842aSMatthew G. Knepley . numCells     - The number of cells owned by this process
5941b0fe842aSMatthew G. Knepley . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
5942b0fe842aSMatthew G. Knepley . NVertices    - The global number of vertices, or `PETSC_DECIDE`
5943b0fe842aSMatthew G. Knepley . cellSection  - The `PetscSection` giving the number of vertices for each cell (layout of cells)
5944b0fe842aSMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
5945b0fe842aSMatthew G. Knepley . cells        - An array of the global vertex numbers for each cell
5946b0fe842aSMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
5947b0fe842aSMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5948b0fe842aSMatthew G. Knepley 
5949b0fe842aSMatthew G. Knepley   Output Parameters:
5950b0fe842aSMatthew G. Knepley + dm          - The `DM`
5951b0fe842aSMatthew G. Knepley . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
5952b0fe842aSMatthew G. Knepley - verticesAdj - (Optional) vertex adjacency array
5953b0fe842aSMatthew G. Knepley 
5954b0fe842aSMatthew G. Knepley   Level: intermediate
5955b0fe842aSMatthew G. Knepley 
5956b0fe842aSMatthew G. Knepley   Notes:
5957b0fe842aSMatthew G. Knepley   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
5958b0fe842aSMatthew G. Knepley   `DMPlexBuildFromCellSectionParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
5959b0fe842aSMatthew G. Knepley 
5960b0fe842aSMatthew G. Knepley   See `DMPlexBuildFromCellSectionParallel()` for an example and details about the topology-related parameters.
5961b0fe842aSMatthew G. Knepley 
5962b0fe842aSMatthew G. Knepley   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
5963b0fe842aSMatthew G. Knepley 
5964b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
5965b0fe842aSMatthew G. Knepley @*/
5966b0fe842aSMatthew 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)
5967b0fe842aSMatthew G. Knepley {
5968b0fe842aSMatthew G. Knepley   PetscSF sfVert;
5969b0fe842aSMatthew G. Knepley 
5970b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
5971b0fe842aSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
5972b0fe842aSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
5973b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
5974b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
5975b0fe842aSMatthew G. Knepley   PetscCall(DMSetDimension(*dm, dim));
5976b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildFromCellSectionParallel(*dm, numCells, numVertices, NVertices, cellSection, cells, &sfVert, verticesAdj));
5977b0fe842aSMatthew G. Knepley   if (interpolate) {
5978b0fe842aSMatthew G. Knepley     DM idm;
5979b0fe842aSMatthew G. Knepley 
5980b0fe842aSMatthew G. Knepley     PetscCall(DMPlexInterpolate(*dm, &idm));
5981b0fe842aSMatthew G. Knepley     PetscCall(DMDestroy(dm));
5982b0fe842aSMatthew G. Knepley     *dm = idm;
5983b0fe842aSMatthew G. Knepley   }
5984b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
5985b0fe842aSMatthew G. Knepley   if (vertexSF) *vertexSF = sfVert;
5986b0fe842aSMatthew G. Knepley   else PetscCall(PetscSFDestroy(&sfVert));
5987b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
5988b0fe842aSMatthew G. Knepley }
5989b0fe842aSMatthew G. Knepley 
5990b0fe842aSMatthew G. Knepley /*@
5991a1cb98faSBarry Smith   DMPlexBuildFromCellList - Build `DMPLEX` topology from a list of vertices for each cell (common mesh generator output)
5992a1cb98faSBarry Smith 
599320f4b53cSBarry Smith   Collective; No Fortran Support
59949298eaa6SMatthew G Knepley 
59959298eaa6SMatthew G Knepley   Input Parameters:
5996a1cb98faSBarry Smith + dm          - The `DM`
5997b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5998a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DETERMINE`
59999298eaa6SMatthew G Knepley . numCorners  - The number of vertices for each cell
6000a3b724e8SBarry Smith - cells       - An array of `numCells` x `numCorners` numbers, the global vertex numbers for each cell
60019298eaa6SMatthew G Knepley 
6002b09969d6SVaclav Hapla   Level: advanced
60039298eaa6SMatthew G Knepley 
6004b09969d6SVaclav Hapla   Notes:
6005b09969d6SVaclav Hapla   Two triangles sharing a face
6006a1cb98faSBarry Smith .vb
60079298eaa6SMatthew G Knepley 
6008a1cb98faSBarry Smith         2
6009a1cb98faSBarry Smith       / | \
6010a1cb98faSBarry Smith      /  |  \
6011a1cb98faSBarry Smith     /   |   \
6012a1cb98faSBarry Smith    0  0 | 1  3
6013a1cb98faSBarry Smith     \   |   /
6014a1cb98faSBarry Smith      \  |  /
6015a1cb98faSBarry Smith       \ | /
6016a1cb98faSBarry Smith         1
6017a1cb98faSBarry Smith .ve
6018a1cb98faSBarry Smith   would have input
6019a1cb98faSBarry Smith .vb
6020a1cb98faSBarry Smith   numCells = 2, numVertices = 4
6021a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
6022a1cb98faSBarry Smith .ve
6023a1cb98faSBarry Smith   which would result in the `DMPLEX`
6024a1cb98faSBarry Smith .vb
6025a1cb98faSBarry Smith 
6026a1cb98faSBarry Smith         4
6027a1cb98faSBarry Smith       / | \
6028a1cb98faSBarry Smith      /  |  \
6029a1cb98faSBarry Smith     /   |   \
6030a1cb98faSBarry Smith    2  0 | 1  5
6031a1cb98faSBarry Smith     \   |   /
6032a1cb98faSBarry Smith      \  |  /
6033a1cb98faSBarry Smith       \ | /
6034a1cb98faSBarry Smith         3
6035a1cb98faSBarry Smith .ve
6036a1cb98faSBarry Smith 
6037a1cb98faSBarry Smith   If numVertices is `PETSC_DETERMINE`, it is computed by PETSc as the maximum vertex index in cells + 1.
603825b6865aSVaclav Hapla 
60391cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListPetsc()`
6040b09969d6SVaclav Hapla @*/
6041d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellList(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const PetscInt cells[])
6042d71ae5a4SJacob Faibussowitsch {
6043961cfab0SVaclav Hapla   PetscInt *cones, c, p, dim;
6044b09969d6SVaclav Hapla 
6045b09969d6SVaclav Hapla   PetscFunctionBegin;
60469566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
60479566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
604825b6865aSVaclav Hapla   /* Get/check global number of vertices */
604925b6865aSVaclav Hapla   {
605025b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
605125b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
605225b6865aSVaclav Hapla 
605325b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
60541690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
60559371c9d4SSatish Balay     for (i = 0; i < len; i++)
60569371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
605725b6865aSVaclav Hapla     ++NVerticesInCells;
605825b6865aSVaclav Hapla 
605925b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE) numVertices = NVerticesInCells;
60609371c9d4SSatish Balay     else
60619371c9d4SSatish 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);
606225b6865aSVaclav Hapla   }
60639566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
606448a46eb9SPierre Jolivet   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
60659566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
60669566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
6067b09969d6SVaclav Hapla   for (c = 0; c < numCells; ++c) {
6068ad540459SPierre Jolivet     for (p = 0; p < numCorners; ++p) cones[c * numCorners + p] = cells[c * numCorners + p] + numCells;
6069b09969d6SVaclav Hapla   }
60709566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
60719566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
60729566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
60733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6074b09969d6SVaclav Hapla }
6075b09969d6SVaclav Hapla 
6076cc4c1da9SBarry Smith /*@
6077a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellList - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
6078a1cb98faSBarry Smith 
6079cc4c1da9SBarry Smith   Collective
6080b09969d6SVaclav Hapla 
6081b09969d6SVaclav Hapla   Input Parameters:
6082a1cb98faSBarry Smith + dm           - The `DM`
6083b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
6084b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
6085b09969d6SVaclav Hapla 
6086b09969d6SVaclav Hapla   Level: advanced
6087b09969d6SVaclav Hapla 
60881cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellList()`
6089b09969d6SVaclav Hapla @*/
6090d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellList(DM dm, PetscInt spaceDim, const PetscReal vertexCoords[])
6091d71ae5a4SJacob Faibussowitsch {
6092b09969d6SVaclav Hapla   PetscSection coordSection;
6093b09969d6SVaclav Hapla   Vec          coordinates;
6094b09969d6SVaclav Hapla   DM           cdm;
6095b09969d6SVaclav Hapla   PetscScalar *coords;
60961edcf0b2SVaclav Hapla   PetscInt     v, vStart, vEnd, d;
6097b09969d6SVaclav Hapla 
6098b09969d6SVaclav Hapla   PetscFunctionBegin;
60999566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
61009566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
61011dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
61029566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
61039566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
61049566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
61059566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
61069566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
61071edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
61089566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
61099566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
6110b09969d6SVaclav Hapla   }
61119566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
6112b09969d6SVaclav Hapla 
61139566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
61149566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(cdm, &coordinates));
61159566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
61169566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
61179566063dSJacob Faibussowitsch   PetscCall(VecGetArrayWrite(coordinates, &coords));
61181edcf0b2SVaclav Hapla   for (v = 0; v < vEnd - vStart; ++v) {
6119ad540459SPierre Jolivet     for (d = 0; d < spaceDim; ++d) coords[v * spaceDim + d] = vertexCoords[v * spaceDim + d];
6120b09969d6SVaclav Hapla   }
61219566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayWrite(coordinates, &coords));
61229566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
61239566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
61249566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
61253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6126b09969d6SVaclav Hapla }
6127b09969d6SVaclav Hapla 
6128b09969d6SVaclav Hapla /*@
6129a1cb98faSBarry Smith   DMPlexCreateFromCellListPetsc - Create `DMPLEX` from a list of vertices for each cell (common mesh generator output), but only process 0 takes in the input
61303df08285SMatthew G. Knepley 
6131a1cb98faSBarry Smith   Collective
6132b09969d6SVaclav Hapla 
6133b09969d6SVaclav Hapla   Input Parameters:
6134b09969d6SVaclav Hapla + comm         - The communicator
6135b09969d6SVaclav Hapla . dim          - The topological dimension of the mesh
61363df08285SMatthew G. Knepley . numCells     - The number of cells, only on process 0
6137a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`, only on process 0
61383df08285SMatthew G. Knepley . numCorners   - The number of vertices for each cell, only on process 0
6139b09969d6SVaclav Hapla . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
61403df08285SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the vertices for each cell, only on process 0
6141b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
61423df08285SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex, only on process 0
6143b09969d6SVaclav Hapla 
6144b09969d6SVaclav Hapla   Output Parameter:
6145a1cb98faSBarry Smith . dm - The `DM`, which only has points on process 0
614625b6865aSVaclav Hapla 
6147b09969d6SVaclav Hapla   Level: intermediate
6148b09969d6SVaclav Hapla 
6149a1cb98faSBarry Smith   Notes:
6150a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, `DMPlexBuildFromCellList()`,
6151a1cb98faSBarry Smith   `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellList()`
6152a1cb98faSBarry Smith 
6153a1cb98faSBarry Smith   See `DMPlexBuildFromCellList()` for an example and details about the topology-related parameters.
6154a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellList()` for details about the geometry-related parameters.
6155a1cb98faSBarry Smith   See `DMPlexCreateFromCellListParallelPetsc()` for parallel input
6156a1cb98faSBarry Smith 
61571cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellList()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
61589298eaa6SMatthew G Knepley @*/
6159d71ae5a4SJacob 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)
6160d71ae5a4SJacob Faibussowitsch {
61613df08285SMatthew G. Knepley   PetscMPIInt rank;
61629298eaa6SMatthew G Knepley 
61639298eaa6SMatthew G Knepley   PetscFunctionBegin;
616428b400f6SJacob 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.");
61659566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
61669566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
61679566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
61689566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
6169c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(*dm, numCells, numVertices, numCorners, cells));
61709566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(*dm, 0, 0, 0, NULL));
61719298eaa6SMatthew G Knepley   if (interpolate) {
61725fd9971aSMatthew G. Knepley     DM idm;
61739298eaa6SMatthew G Knepley 
61749566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
61759566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
61769298eaa6SMatthew G Knepley     *dm = idm;
61779298eaa6SMatthew G Knepley   }
6178c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, vertexCoords));
61799566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, NULL));
61803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
61819298eaa6SMatthew G Knepley }
61829298eaa6SMatthew G Knepley 
6183939f6067SMatthew G. Knepley /*@
618420f4b53cSBarry Smith   DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a `DM`
6185939f6067SMatthew G. Knepley 
6186939f6067SMatthew G. Knepley   Input Parameters:
618720f4b53cSBarry Smith + dm               - The empty `DM` object, usually from `DMCreate()` and `DMSetDimension()`
6188939f6067SMatthew G. Knepley . depth            - The depth of the DAG
618920f4b53cSBarry Smith . numPoints        - Array of size depth + 1 containing the number of points at each `depth`
6190939f6067SMatthew G. Knepley . coneSize         - The cone size of each point
6191939f6067SMatthew G. Knepley . cones            - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point
6192939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point
619320f4b53cSBarry Smith - vertexCoords     - An array of `numPoints`[0]*spacedim numbers representing the coordinates of each vertex, with spacedim the value set via `DMSetCoordinateDim()`
6194939f6067SMatthew G. Knepley 
6195939f6067SMatthew G. Knepley   Output Parameter:
619620f4b53cSBarry Smith . dm - The `DM`
619720f4b53cSBarry Smith 
619820f4b53cSBarry Smith   Level: advanced
6199939f6067SMatthew G. Knepley 
6200a1cb98faSBarry Smith   Note:
6201a1cb98faSBarry Smith   Two triangles sharing a face would have input
6202a1cb98faSBarry Smith .vb
6203a1cb98faSBarry Smith   depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0]
6204a1cb98faSBarry Smith   cones = [2 3 4  3 5 4], coneOrientations = [0 0 0  0 0 0]
6205a1cb98faSBarry Smith  vertexCoords = [-1.0 0.0  0.0 -1.0  0.0 1.0  1.0 0.0]
6206a1cb98faSBarry Smith .ve
6207939f6067SMatthew G. Knepley   which would result in the DMPlex
6208a1cb98faSBarry Smith .vb
6209a1cb98faSBarry Smith         4
6210a1cb98faSBarry Smith       / | \
6211a1cb98faSBarry Smith      /  |  \
6212a1cb98faSBarry Smith     /   |   \
6213a1cb98faSBarry Smith    2  0 | 1  5
6214a1cb98faSBarry Smith     \   |   /
6215a1cb98faSBarry Smith      \  |  /
6216a1cb98faSBarry Smith       \ | /
6217a1cb98faSBarry Smith         3
6218a1cb98faSBarry Smith .ve
6219a1cb98faSBarry Smith   Notice that all points are numbered consecutively, unlike `DMPlexCreateFromCellListPetsc()`
6220939f6067SMatthew G. Knepley 
62211cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
6222939f6067SMatthew G. Knepley @*/
6223d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
6224d71ae5a4SJacob Faibussowitsch {
62259298eaa6SMatthew G Knepley   Vec          coordinates;
62269298eaa6SMatthew G Knepley   PetscSection coordSection;
62279298eaa6SMatthew G Knepley   PetscScalar *coords;
6228811e8653SToby Isaac   PetscInt     coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off;
62299298eaa6SMatthew G Knepley 
62309298eaa6SMatthew G Knepley   PetscFunctionBegin;
62319566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
62329566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dimEmbed));
623363a3b9bcSJacob Faibussowitsch   PetscCheck(dimEmbed >= dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Embedding dimension %" PetscInt_FMT " cannot be less than intrinsic dimension %" PetscInt_FMT, dimEmbed, dim);
62349298eaa6SMatthew G Knepley   for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
62359566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, pStart, pEnd));
62369298eaa6SMatthew G Knepley   for (p = pStart; p < pEnd; ++p) {
62379566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeSize(dm, p, coneSize[p - pStart]));
6238ad540459SPierre Jolivet     if (firstVertex < 0 && !coneSize[p - pStart]) firstVertex = p - pStart;
623997e052ccSToby Isaac   }
62401dca8a05SBarry Smith   PetscCheck(firstVertex >= 0 || !numPoints[0], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Expected %" PetscInt_FMT " vertices but could not find any", numPoints[0]);
62419566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm)); /* Allocate space for cones */
62429298eaa6SMatthew G Knepley   for (p = pStart, off = 0; p < pEnd; off += coneSize[p - pStart], ++p) {
62439566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCone(dm, p, &cones[off]));
62449566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeOrientation(dm, p, &coneOrientations[off]));
62459298eaa6SMatthew G Knepley   }
62469566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
62479566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
62489298eaa6SMatthew G Knepley   /* Build coordinates */
62499566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
62509566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
62519566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed));
62529566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numPoints[0]));
62539298eaa6SMatthew G Knepley   for (v = firstVertex; v < firstVertex + numPoints[0]; ++v) {
62549566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed));
62559566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed));
62569298eaa6SMatthew G Knepley   }
62579566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
62589566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
62599566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
62609566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
62619566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
62629566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, dimEmbed));
62639566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
62649318fe57SMatthew G. Knepley   if (vertexCoords) {
62659566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
62669298eaa6SMatthew G Knepley     for (v = 0; v < numPoints[0]; ++v) {
62679298eaa6SMatthew G Knepley       PetscInt off;
62689298eaa6SMatthew G Knepley 
62699566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSection, v + firstVertex, &off));
6270ad540459SPierre Jolivet       for (d = 0; d < dimEmbed; ++d) coords[off + d] = vertexCoords[v * dimEmbed + d];
62719298eaa6SMatthew G Knepley     }
62729318fe57SMatthew G. Knepley   }
62739566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
62749566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
62759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
62763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
62779298eaa6SMatthew G Knepley }
62788415267dSToby Isaac 
6279a4e35b19SJacob Faibussowitsch /*
6280a1cb98faSBarry Smith   DMPlexCreateCellVertexFromFile - Create a `DMPLEX` mesh from a simple cell-vertex file.
6281a1cb98faSBarry Smith 
6282a1cb98faSBarry Smith   Collective
62838ca92349SMatthew G. Knepley 
62848ca92349SMatthew G. Knepley + comm        - The MPI communicator
62858ca92349SMatthew G. Knepley . filename    - Name of the .dat file
62868ca92349SMatthew G. Knepley - interpolate - Create faces and edges in the mesh
62878ca92349SMatthew G. Knepley 
62888ca92349SMatthew G. Knepley   Output Parameter:
6289a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
62908ca92349SMatthew G. Knepley 
62918ca92349SMatthew G. Knepley   Level: beginner
62928ca92349SMatthew G. Knepley 
6293a1cb98faSBarry Smith   Note:
6294a1cb98faSBarry Smith   The format is the simplest possible:
6295a1cb98faSBarry Smith .vb
6296d0812dedSMatthew G. Knepley   dim Ne Nv Nc Nl
6297d0812dedSMatthew G. Knepley   v_1 v_2 ... v_Nc
6298d0812dedSMatthew G. Knepley   ...
6299d0812dedSMatthew G. Knepley   x y z marker_1 ... marker_Nl
6300a1cb98faSBarry Smith .ve
6301a1cb98faSBarry Smith 
6302a1cb98faSBarry Smith   Developer Note:
6303a1cb98faSBarry Smith   Should use a `PetscViewer` not a filename
6304a1cb98faSBarry Smith 
63056afe31f6SMartin Diehl .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateGmsh()`, `DMPlexCreate()`
6306a4e35b19SJacob Faibussowitsch */
6307ff6a9541SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCellVertexFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
6308d71ae5a4SJacob Faibussowitsch {
63098ca92349SMatthew G. Knepley   DMLabel      marker;
63108ca92349SMatthew G. Knepley   PetscViewer  viewer;
63118ca92349SMatthew G. Knepley   Vec          coordinates;
63128ca92349SMatthew G. Knepley   PetscSection coordSection;
63138ca92349SMatthew G. Knepley   PetscScalar *coords;
63148ca92349SMatthew G. Knepley   char         line[PETSC_MAX_PATH_LEN];
6315d0812dedSMatthew G. Knepley   PetscInt     cdim, coordSize, v, c, d;
63168ca92349SMatthew G. Knepley   PetscMPIInt  rank;
6317d0812dedSMatthew G. Knepley   int          snum, dim, Nv, Nc, Ncn, Nl;
63188ca92349SMatthew G. Knepley 
63198ca92349SMatthew G. Knepley   PetscFunctionBegin;
63209566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
63219566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(comm, &viewer));
63229566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII));
63239566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
63249566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetName(viewer, filename));
6325dd400576SPatrick Sanan   if (rank == 0) {
6326d0812dedSMatthew G. Knepley     PetscCall(PetscViewerRead(viewer, line, 5, NULL, PETSC_STRING));
6327d0812dedSMatthew G. Knepley     snum = sscanf(line, "%d %d %d %d %d", &dim, &Nc, &Nv, &Ncn, &Nl);
6328d0812dedSMatthew G. Knepley     PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
632925ce1634SJed Brown   } else {
6330f8d5e320SMatthew G. Knepley     Nc = Nv = Ncn = Nl = 0;
63318ca92349SMatthew G. Knepley   }
6332d0812dedSMatthew G. Knepley   PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm));
6333835f2295SStefano Zampini   cdim = dim;
63349566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
63359566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
63369566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(*dm, 0, Nc + Nv));
6337835f2295SStefano Zampini   PetscCall(DMSetDimension(*dm, dim));
63389566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(*dm, cdim));
63398ca92349SMatthew G. Knepley   /* Read topology */
6340dd400576SPatrick Sanan   if (rank == 0) {
6341f8d5e320SMatthew G. Knepley     char     format[PETSC_MAX_PATH_LEN];
6342f8d5e320SMatthew G. Knepley     PetscInt cone[8];
63438ca92349SMatthew G. Knepley     int      vbuf[8], v;
63448ca92349SMatthew G. Knepley 
63459371c9d4SSatish Balay     for (c = 0; c < Ncn; ++c) {
63469371c9d4SSatish Balay       format[c * 3 + 0] = '%';
63479371c9d4SSatish Balay       format[c * 3 + 1] = 'd';
63489371c9d4SSatish Balay       format[c * 3 + 2] = ' ';
63499371c9d4SSatish Balay     }
6350f8d5e320SMatthew G. Knepley     format[Ncn * 3 - 1] = '\0';
63519566063dSJacob Faibussowitsch     for (c = 0; c < Nc; ++c) PetscCall(DMPlexSetConeSize(*dm, c, Ncn));
63529566063dSJacob Faibussowitsch     PetscCall(DMSetUp(*dm));
63538ca92349SMatthew G. Knepley     for (c = 0; c < Nc; ++c) {
63549566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, Ncn, NULL, PETSC_STRING));
6355f8d5e320SMatthew G. Knepley       switch (Ncn) {
6356d71ae5a4SJacob Faibussowitsch       case 2:
6357d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1]);
6358d71ae5a4SJacob Faibussowitsch         break;
6359d71ae5a4SJacob Faibussowitsch       case 3:
6360d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2]);
6361d71ae5a4SJacob Faibussowitsch         break;
6362d71ae5a4SJacob Faibussowitsch       case 4:
6363d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3]);
6364d71ae5a4SJacob Faibussowitsch         break;
6365d71ae5a4SJacob Faibussowitsch       case 6:
6366d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5]);
6367d71ae5a4SJacob Faibussowitsch         break;
6368d71ae5a4SJacob Faibussowitsch       case 8:
6369d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5], &vbuf[6], &vbuf[7]);
6370d71ae5a4SJacob Faibussowitsch         break;
6371d71ae5a4SJacob Faibussowitsch       default:
6372d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No cell shape with %d vertices", Ncn);
6373f8d5e320SMatthew G. Knepley       }
637408401ef6SPierre Jolivet       PetscCheck(snum == Ncn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6375f8d5e320SMatthew G. Knepley       for (v = 0; v < Ncn; ++v) cone[v] = vbuf[v] + Nc;
63768ca92349SMatthew G. Knepley       /* Hexahedra are inverted */
6377f8d5e320SMatthew G. Knepley       if (Ncn == 8) {
63788ca92349SMatthew G. Knepley         PetscInt tmp = cone[1];
63798ca92349SMatthew G. Knepley         cone[1]      = cone[3];
63808ca92349SMatthew G. Knepley         cone[3]      = tmp;
63818ca92349SMatthew G. Knepley       }
63829566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(*dm, c, cone));
63838ca92349SMatthew G. Knepley     }
63848ca92349SMatthew G. Knepley   }
63859566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(*dm));
63869566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(*dm));
63878ca92349SMatthew G. Knepley   /* Read coordinates */
63889566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(*dm, &coordSection));
63899566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
63909566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
63919566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, Nc, Nc + Nv));
63928ca92349SMatthew G. Knepley   for (v = Nc; v < Nc + Nv; ++v) {
63939566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
63949566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
63958ca92349SMatthew G. Knepley   }
63969566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
63979566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
63989566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
63999566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
64009566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
64019566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
64029566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
64039566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
6404dd400576SPatrick Sanan   if (rank == 0) {
6405f8d5e320SMatthew G. Knepley     char   format[PETSC_MAX_PATH_LEN];
64068ca92349SMatthew G. Knepley     double x[3];
6407f8d5e320SMatthew G. Knepley     int    l, val[3];
64088ca92349SMatthew G. Knepley 
6409f8d5e320SMatthew G. Knepley     if (Nl) {
64109371c9d4SSatish Balay       for (l = 0; l < Nl; ++l) {
64119371c9d4SSatish Balay         format[l * 3 + 0] = '%';
64129371c9d4SSatish Balay         format[l * 3 + 1] = 'd';
64139371c9d4SSatish Balay         format[l * 3 + 2] = ' ';
64149371c9d4SSatish Balay       }
6415f8d5e320SMatthew G. Knepley       format[Nl * 3 - 1] = '\0';
64169566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
64179566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &marker));
6418f8d5e320SMatthew G. Knepley     }
64198ca92349SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
64209566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, 3 + Nl, NULL, PETSC_STRING));
6421f8d5e320SMatthew G. Knepley       snum = sscanf(line, "%lg %lg %lg", &x[0], &x[1], &x[2]);
642208401ef6SPierre Jolivet       PetscCheck(snum == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6423f8d5e320SMatthew G. Knepley       switch (Nl) {
6424d71ae5a4SJacob Faibussowitsch       case 0:
6425d71ae5a4SJacob Faibussowitsch         snum = 0;
6426d71ae5a4SJacob Faibussowitsch         break;
6427d71ae5a4SJacob Faibussowitsch       case 1:
6428d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0]);
6429d71ae5a4SJacob Faibussowitsch         break;
6430d71ae5a4SJacob Faibussowitsch       case 2:
6431d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1]);
6432d71ae5a4SJacob Faibussowitsch         break;
6433d71ae5a4SJacob Faibussowitsch       case 3:
6434d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1], &val[2]);
6435d71ae5a4SJacob Faibussowitsch         break;
6436d71ae5a4SJacob Faibussowitsch       default:
6437d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Request support for %d labels", Nl);
6438f8d5e320SMatthew G. Knepley       }
643908401ef6SPierre Jolivet       PetscCheck(snum == Nl, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
64408ca92349SMatthew G. Knepley       for (d = 0; d < cdim; ++d) coords[v * cdim + d] = x[d];
64419566063dSJacob Faibussowitsch       for (l = 0; l < Nl; ++l) PetscCall(DMLabelSetValue(marker, v + Nc, val[l]));
64428ca92349SMatthew G. Knepley     }
64438ca92349SMatthew G. Knepley   }
64449566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
64459566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
64469566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
64479566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&viewer));
64488ca92349SMatthew G. Knepley   if (interpolate) {
64498ca92349SMatthew G. Knepley     DM      idm;
64508ca92349SMatthew G. Knepley     DMLabel bdlabel;
64518ca92349SMatthew G. Knepley 
64529566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
64539566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
64548ca92349SMatthew G. Knepley     *dm = idm;
64558ca92349SMatthew G. Knepley 
6456f8d5e320SMatthew G. Knepley     if (!Nl) {
64579566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
64589566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &bdlabel));
64599566063dSJacob Faibussowitsch       PetscCall(DMPlexMarkBoundaryFaces(*dm, PETSC_DETERMINE, bdlabel));
64609566063dSJacob Faibussowitsch       PetscCall(DMPlexLabelComplete(*dm, bdlabel));
64618ca92349SMatthew G. Knepley     }
6462f8d5e320SMatthew G. Knepley   }
64633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
64648ca92349SMatthew G. Knepley }
64658ca92349SMatthew G. Knepley 
6466cc4c1da9SBarry Smith /*@
6467a1cb98faSBarry Smith   DMPlexCreateFromFile - This takes a filename and produces a `DM`
6468a1cb98faSBarry Smith 
6469a1cb98faSBarry Smith   Collective
6470ca522641SMatthew G. Knepley 
6471ca522641SMatthew G. Knepley   Input Parameters:
6472ca522641SMatthew G. Knepley + comm        - The communicator
6473ca522641SMatthew G. Knepley . filename    - A file name
6474a1cb98faSBarry Smith . plexname    - The object name of the resulting `DM`, also used for intra-datafile lookup by some formats
6475ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
6476ca522641SMatthew G. Knepley 
6477ca522641SMatthew G. Knepley   Output Parameter:
6478a1cb98faSBarry Smith . dm - The `DM`
6479ca522641SMatthew G. Knepley 
6480a1cb98faSBarry Smith   Options Database Key:
6481a1cb98faSBarry Smith . -dm_plex_create_from_hdf5_xdmf - use the `PETSC_VIEWER_HDF5_XDMF` format for reading HDF5
648202ef0d99SVaclav Hapla 
648337fdd005SBarry Smith   Use `-dm_plex_create_ prefix` to pass options to the internal `PetscViewer`, e.g.
6484bca97951SVaclav Hapla $ -dm_plex_create_viewer_hdf5_collective
6485bca97951SVaclav Hapla 
6486ca522641SMatthew G. Knepley   Level: beginner
6487ca522641SMatthew G. Knepley 
6488a1cb98faSBarry Smith   Notes:
6489a1cb98faSBarry Smith   Using `PETSCVIEWERHDF5` type with `PETSC_VIEWER_HDF5_PETSC` format, one can save multiple `DMPLEX`
6490a1cb98faSBarry Smith   meshes in a single HDF5 file. This in turn requires one to name the `DMPLEX` object with `PetscObjectSetName()`
6491a1cb98faSBarry Smith   before saving it with `DMView()` and before loading it with `DMLoad()` for identification of the mesh object.
6492a1cb98faSBarry Smith   The input parameter name is thus used to name the `DMPLEX` object when `DMPlexCreateFromFile()` internally
6493a1cb98faSBarry Smith   calls `DMLoad()`. Currently, name is ignored for other viewer types and/or formats.
6494a1cb98faSBarry Smith 
64951cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`, `PetscObjectSetName()`, `DMView()`, `DMLoad()`
6496ca522641SMatthew G. Knepley @*/
6497d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], const char plexname[], PetscBool interpolate, DM *dm)
6498d71ae5a4SJacob Faibussowitsch {
6499ef3a5affSJacob Faibussowitsch   const char  extGmsh[]      = ".msh";
6500ef3a5affSJacob Faibussowitsch   const char  extGmsh2[]     = ".msh2";
6501ef3a5affSJacob Faibussowitsch   const char  extGmsh4[]     = ".msh4";
6502ef3a5affSJacob Faibussowitsch   const char  extCGNS[]      = ".cgns";
6503ef3a5affSJacob Faibussowitsch   const char  extExodus[]    = ".exo";
6504ef3a5affSJacob Faibussowitsch   const char  extExodus_e[]  = ".e";
6505ef3a5affSJacob Faibussowitsch   const char  extGenesis[]   = ".gen";
6506ef3a5affSJacob Faibussowitsch   const char  extFluent[]    = ".cas";
6507ef3a5affSJacob Faibussowitsch   const char  extHDF5[]      = ".h5";
65086f2c871aSStefano Zampini   const char  extXDMFHDF5[]  = ".xdmf.h5";
6509ef3a5affSJacob Faibussowitsch   const char  extPLY[]       = ".ply";
6510ef3a5affSJacob Faibussowitsch   const char  extEGADSLite[] = ".egadslite";
6511ef3a5affSJacob Faibussowitsch   const char  extEGADS[]     = ".egads";
6512ef3a5affSJacob Faibussowitsch   const char  extIGES[]      = ".igs";
6513ef3a5affSJacob Faibussowitsch   const char  extSTEP[]      = ".stp";
6514ef3a5affSJacob Faibussowitsch   const char  extCV[]        = ".dat";
6515ca522641SMatthew G. Knepley   size_t      len;
65166afe31f6SMartin Diehl   PetscBool   isGmsh, isGmsh2, isGmsh4, isCGNS, isExodus, isGenesis, isFluent, isHDF5, isPLY, isEGADSLite, isEGADS, isIGES, isSTEP, isCV, isXDMFHDF5;
6517ca522641SMatthew G. Knepley   PetscMPIInt rank;
6518ca522641SMatthew G. Knepley 
6519ca522641SMatthew G. Knepley   PetscFunctionBegin;
65204f572ea9SToby Isaac   PetscAssertPointer(filename, 2);
65214f572ea9SToby Isaac   if (plexname) PetscAssertPointer(plexname, 3);
65224f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
65239566063dSJacob Faibussowitsch   PetscCall(DMInitializePackage());
65249566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromFile, 0, 0, 0, 0));
65259566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
65269566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(filename, &len));
652728b400f6SJacob Faibussowitsch   PetscCheck(len, comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path");
6528ef3a5affSJacob Faibussowitsch 
65299371c9d4SSatish Balay #define CheckExtension(extension__, is_extension__) \
65309371c9d4SSatish Balay   do { \
6531274aaeaaSJacob Faibussowitsch     PetscAssert(sizeof(extension__), comm, PETSC_ERR_PLIB, "Zero-size extension: %s", extension__); \
6532274aaeaaSJacob Faibussowitsch     /* don't count the null-terminator at the end */ \
6533274aaeaaSJacob Faibussowitsch     const size_t ext_len = sizeof(extension__) - 1; \
6534274aaeaaSJacob Faibussowitsch     if (len < ext_len) { \
6535ef3a5affSJacob Faibussowitsch       is_extension__ = PETSC_FALSE; \
6536ef3a5affSJacob Faibussowitsch     } else { \
6537274aaeaaSJacob Faibussowitsch       PetscCall(PetscStrncmp(filename + len - ext_len, extension__, ext_len, &is_extension__)); \
6538ef3a5affSJacob Faibussowitsch     } \
6539ef3a5affSJacob Faibussowitsch   } while (0)
6540ef3a5affSJacob Faibussowitsch 
6541ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh, isGmsh);
6542ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh2, isGmsh2);
6543ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh4, isGmsh4);
6544ef3a5affSJacob Faibussowitsch   CheckExtension(extCGNS, isCGNS);
6545ef3a5affSJacob Faibussowitsch   CheckExtension(extExodus, isExodus);
6546ef3a5affSJacob Faibussowitsch   if (!isExodus) CheckExtension(extExodus_e, isExodus);
6547ef3a5affSJacob Faibussowitsch   CheckExtension(extGenesis, isGenesis);
6548ef3a5affSJacob Faibussowitsch   CheckExtension(extFluent, isFluent);
6549ef3a5affSJacob Faibussowitsch   CheckExtension(extHDF5, isHDF5);
6550ef3a5affSJacob Faibussowitsch   CheckExtension(extPLY, isPLY);
6551ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADSLite, isEGADSLite);
6552ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADS, isEGADS);
6553ef3a5affSJacob Faibussowitsch   CheckExtension(extIGES, isIGES);
6554ef3a5affSJacob Faibussowitsch   CheckExtension(extSTEP, isSTEP);
6555ef3a5affSJacob Faibussowitsch   CheckExtension(extCV, isCV);
65566f2c871aSStefano Zampini   CheckExtension(extXDMFHDF5, isXDMFHDF5);
6557ef3a5affSJacob Faibussowitsch 
6558ef3a5affSJacob Faibussowitsch #undef CheckExtension
6559ef3a5affSJacob Faibussowitsch 
6560de78e4feSLisandro Dalcin   if (isGmsh || isGmsh2 || isGmsh4) {
65619566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateGmshFromFile(comm, filename, interpolate, dm));
6562ca522641SMatthew G. Knepley   } else if (isCGNS) {
65639566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm));
656490c68965SMatthew G. Knepley   } else if (isExodus || isGenesis) {
65659566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateExodusFromFile(comm, filename, interpolate, dm));
65662f0bd6dcSMichael Lange   } else if (isFluent) {
65679566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFluentFromFile(comm, filename, interpolate, dm));
6568cc2f8f65SMatthew G. Knepley   } else if (isHDF5) {
6569cc2f8f65SMatthew G. Knepley     PetscViewer viewer;
6570cc2f8f65SMatthew G. Knepley 
657143b242b4SVaclav 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 */
65726f2c871aSStefano Zampini     PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_create_from_hdf5_xdmf", &isXDMFHDF5, NULL));
65739566063dSJacob Faibussowitsch     PetscCall(PetscViewerCreate(comm, &viewer));
65749566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetType(viewer, PETSCVIEWERHDF5));
65759566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetOptionsPrefix(viewer, "dm_plex_create_"));
65769566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetFromOptions(viewer));
65779566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
65789566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetName(viewer, filename));
6579cd7e8a5eSksagiyam 
65809566063dSJacob Faibussowitsch     PetscCall(DMCreate(comm, dm));
6581f4f49eeaSPierre Jolivet     PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
65829566063dSJacob Faibussowitsch     PetscCall(DMSetType(*dm, DMPLEX));
65836f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPushFormat(viewer, PETSC_VIEWER_HDF5_XDMF));
65849566063dSJacob Faibussowitsch     PetscCall(DMLoad(*dm, viewer));
65856f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPopFormat(viewer));
65869566063dSJacob Faibussowitsch     PetscCall(PetscViewerDestroy(&viewer));
65875fd9971aSMatthew G. Knepley 
65885fd9971aSMatthew G. Knepley     if (interpolate) {
65895fd9971aSMatthew G. Knepley       DM idm;
65905fd9971aSMatthew G. Knepley 
65919566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(*dm, &idm));
65929566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
65935fd9971aSMatthew G. Knepley       *dm = idm;
65945fd9971aSMatthew G. Knepley     }
6595f2801cd6SMatthew G. Knepley   } else if (isPLY) {
65969566063dSJacob Faibussowitsch     PetscCall(DMPlexCreatePLYFromFile(comm, filename, interpolate, dm));
6597c1cad2e7SMatthew G. Knepley   } else if (isEGADSLite || isEGADS || isIGES || isSTEP) {
65989566063dSJacob Faibussowitsch     if (isEGADSLite) PetscCall(DMPlexCreateEGADSLiteFromFile(comm, filename, dm));
65999566063dSJacob Faibussowitsch     else PetscCall(DMPlexCreateEGADSFromFile(comm, filename, dm));
66007bee2925SMatthew Knepley     if (!interpolate) {
66017bee2925SMatthew Knepley       DM udm;
66027bee2925SMatthew Knepley 
66039566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(*dm, &udm));
66049566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
66057bee2925SMatthew Knepley       *dm = udm;
66067bee2925SMatthew Knepley     }
66078ca92349SMatthew G. Knepley   } else if (isCV) {
66089566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCellVertexFromFile(comm, filename, interpolate, dm));
660998921bdaSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename);
66109566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(plexname, &len));
6611f4f49eeaSPierre Jolivet   if (len) PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
66129566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromFile, 0, 0, 0, 0));
66133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6614ca522641SMatthew G. Knepley }
661512b8a6daSStefano Zampini 
6616cc4c1da9SBarry Smith /*@
66179f6c5813SMatthew 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.
66189f6c5813SMatthew G. Knepley 
66190528010dSStefano Zampini   Input Parameters:
66200528010dSStefano Zampini + tr     - The `DMPlexTransform`
66210528010dSStefano Zampini - prefix - An options prefix, or NULL
66229f6c5813SMatthew G. Knepley 
66239f6c5813SMatthew G. Knepley   Output Parameter:
66249f6c5813SMatthew G. Knepley . dm - The `DM`
66259f6c5813SMatthew G. Knepley 
66269f6c5813SMatthew G. Knepley   Level: beginner
66279f6c5813SMatthew G. Knepley 
662820f4b53cSBarry Smith   Notes:
662920f4b53cSBarry 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.
663020f4b53cSBarry Smith 
66319f6c5813SMatthew G. Knepley .seealso: `DMPlexCreateFromFile`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
66329f6c5813SMatthew G. Knepley @*/
66330528010dSStefano Zampini PetscErrorCode DMPlexCreateEphemeral(DMPlexTransform tr, const char prefix[], DM *dm)
66349f6c5813SMatthew G. Knepley {
66350528010dSStefano Zampini   DM           bdm, bcdm, cdm;
66360528010dSStefano Zampini   Vec          coordinates, coordinatesNew;
66370528010dSStefano Zampini   PetscSection cs;
6638817b2c36SMatthew G. Knepley   PetscInt     cdim, Nl;
66399f6c5813SMatthew G. Knepley 
66409f6c5813SMatthew G. Knepley   PetscFunctionBegin;
66419f6c5813SMatthew G. Knepley   PetscCall(DMCreate(PetscObjectComm((PetscObject)tr), dm));
66429f6c5813SMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
66430528010dSStefano Zampini   ((DM_Plex *)(*dm)->data)->interpolated = DMPLEX_INTERPOLATED_FULL;
66440528010dSStefano Zampini   // Handle coordinates
66450528010dSStefano Zampini   PetscCall(DMPlexTransformGetDM(tr, &bdm));
6646817b2c36SMatthew G. Knepley   PetscCall(DMPlexTransformSetDimensions(tr, bdm, *dm));
6647817b2c36SMatthew G. Knepley   PetscCall(DMGetCoordinateDim(*dm, &cdim));
66480528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(bdm, &bcdm));
66490528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(*dm, &cdm));
66500528010dSStefano Zampini   PetscCall(DMCopyDisc(bcdm, cdm));
66510528010dSStefano Zampini   PetscCall(DMGetLocalSection(cdm, &cs));
66520528010dSStefano Zampini   PetscCall(PetscSectionSetNumFields(cs, 1));
66530528010dSStefano Zampini   PetscCall(PetscSectionSetFieldComponents(cs, 0, cdim));
66540528010dSStefano Zampini   PetscCall(DMGetCoordinatesLocal(bdm, &coordinates));
66550528010dSStefano Zampini   PetscCall(VecDuplicate(coordinates, &coordinatesNew));
66560528010dSStefano Zampini   PetscCall(VecCopy(coordinates, coordinatesNew));
66570528010dSStefano Zampini   PetscCall(DMSetCoordinatesLocal(*dm, coordinatesNew));
66580528010dSStefano Zampini   PetscCall(VecDestroy(&coordinatesNew));
66599f6c5813SMatthew G. Knepley 
66609f6c5813SMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)tr));
66619f6c5813SMatthew G. Knepley   PetscCall(DMPlexTransformDestroy(&((DM_Plex *)(*dm)->data)->tr));
66629f6c5813SMatthew G. Knepley   ((DM_Plex *)(*dm)->data)->tr = tr;
66630528010dSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(*dm, PETSC_FALSE));
66640528010dSStefano Zampini   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*dm, prefix));
66650528010dSStefano Zampini   PetscCall(DMSetFromOptions(*dm));
66669f6c5813SMatthew G. Knepley 
66679f6c5813SMatthew G. Knepley   PetscCall(DMGetNumLabels(bdm, &Nl));
66689f6c5813SMatthew G. Knepley   for (PetscInt l = 0; l < Nl; ++l) {
66699f6c5813SMatthew G. Knepley     DMLabel     label, labelNew;
66709f6c5813SMatthew G. Knepley     const char *lname;
66719f6c5813SMatthew G. Knepley     PetscBool   isDepth, isCellType;
66729f6c5813SMatthew G. Knepley 
66739f6c5813SMatthew G. Knepley     PetscCall(DMGetLabelName(bdm, l, &lname));
66749f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "depth", &isDepth));
66759f6c5813SMatthew G. Knepley     if (isDepth) continue;
66769f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "celltype", &isCellType));
66779f6c5813SMatthew G. Knepley     if (isCellType) continue;
66789f6c5813SMatthew G. Knepley     PetscCall(DMCreateLabel(*dm, lname));
66799f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(bdm, lname, &label));
66809f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(*dm, lname, &labelNew));
66819f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetType(labelNew, DMLABELEPHEMERAL));
66829f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetLabel(labelNew, label));
66839f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetTransform(labelNew, tr));
66849f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetUp(labelNew));
66859f6c5813SMatthew G. Knepley   }
66863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
66879f6c5813SMatthew G. Knepley }
6688