xref: /petsc/src/dm/impls/plex/plexcreate.c (revision 1690c2ae071c7584458d4e437df7b47bc4686b3c)
1552f7358SJed Brown #define PETSCDM_DLL
2af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h> /*I   "petscdmplex.h"   I*/
3cc4c1da9SBarry Smith #include <petsc/private/hashseti.h>
40c312b8eSJed Brown #include <petscsf.h>
5cc4c1da9SBarry Smith #include <petscdmplextransform.h> /*I   "petscdmplextransform.h"   I*/
69f6c5813SMatthew G. Knepley #include <petscdmlabelephemeral.h>
7b7f5c055SJed Brown #include <petsc/private/kernels/blockmatmult.h>
8b7f5c055SJed Brown #include <petsc/private/kernels/blockinvert.h>
9552f7358SJed Brown 
10d0812dedSMatthew G. Knepley #ifdef PETSC_HAVE_UNISTD_H
11d0812dedSMatthew G. Knepley   #include <unistd.h>
12d0812dedSMatthew G. Knepley #endif
13d0812dedSMatthew G. Knepley #include <errno.h>
14d0812dedSMatthew G. Knepley 
15708be2fdSJed Brown PetscLogEvent DMPLEX_CreateFromFile, DMPLEX_CreateFromOptions, DMPLEX_BuildFromCellList, DMPLEX_BuildCoordinatesFromCellList;
1658cd63d5SVaclav Hapla 
179318fe57SMatthew G. Knepley /* External function declarations here */
189318fe57SMatthew G. Knepley static PetscErrorCode DMInitialize_Plex(DM dm);
199318fe57SMatthew G. Knepley 
20e600fa54SMatthew G. Knepley /* This copies internal things in the Plex structure that we generally want when making a new, related Plex */
21d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCopy_Internal(DM dmin, PetscBool copyPeriodicity, PetscBool copyOverlap, DM dmout)
22d71ae5a4SJacob Faibussowitsch {
234fb89dddSMatthew G. Knepley   const PetscReal     *maxCell, *Lstart, *L;
2412a88998SMatthew G. Knepley   VecType              vecType;
2512a88998SMatthew G. Knepley   MatType              matType;
265962854dSMatthew G. Knepley   PetscBool            dist, useCeed;
27adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
28e600fa54SMatthew G. Knepley 
29e600fa54SMatthew G. Knepley   PetscFunctionBegin;
3012a88998SMatthew G. Knepley   PetscCall(DMGetVecType(dmin, &vecType));
3112a88998SMatthew G. Knepley   PetscCall(DMSetVecType(dmout, vecType));
3212a88998SMatthew G. Knepley   PetscCall(DMGetMatType(dmin, &matType));
3312a88998SMatthew G. Knepley   PetscCall(DMSetMatType(dmout, matType));
34e600fa54SMatthew G. Knepley   if (copyPeriodicity) {
354fb89dddSMatthew G. Knepley     PetscCall(DMGetPeriodicity(dmin, &maxCell, &Lstart, &L));
364fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dmout, maxCell, Lstart, L));
373d0e8ed9SDavid Salac     PetscCall(DMLocalizeCoordinates(dmout));
38e600fa54SMatthew G. Knepley   }
399566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dmin, &dist));
409566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeSetDefault(dmout, dist));
416bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dmin, &reorder));
426bc1bd01Sksagiyam   PetscCall(DMPlexReorderSetDefault(dmout, reorder));
435962854dSMatthew G. Knepley   PetscCall(DMPlexGetUseCeed(dmin, &useCeed));
445962854dSMatthew G. Knepley   PetscCall(DMPlexSetUseCeed(dmout, useCeed));
45e600fa54SMatthew G. Knepley   ((DM_Plex *)dmout->data)->useHashLocation = ((DM_Plex *)dmin->data)->useHashLocation;
465962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printSetValues  = ((DM_Plex *)dmin->data)->printSetValues;
475962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFEM        = ((DM_Plex *)dmin->data)->printFEM;
485962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFVM        = ((DM_Plex *)dmin->data)->printFVM;
495962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printL2         = ((DM_Plex *)dmin->data)->printL2;
505962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printLocate     = ((DM_Plex *)dmin->data)->printLocate;
515962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printTol        = ((DM_Plex *)dmin->data)->printTol;
521baa6e33SBarry Smith   if (copyOverlap) PetscCall(DMPlexSetOverlap_Plex(dmout, dmin, 0));
533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
54e600fa54SMatthew G. Knepley }
55e600fa54SMatthew G. Knepley 
569318fe57SMatthew G. Knepley /* Replace dm with the contents of ndm, and then destroy ndm
579318fe57SMatthew G. Knepley    - Share the DM_Plex structure
589318fe57SMatthew G. Knepley    - Share the coordinates
599318fe57SMatthew G. Knepley    - Share the SF
609318fe57SMatthew G. Knepley */
61d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexReplace_Internal(DM dm, DM *ndm)
62d71ae5a4SJacob Faibussowitsch {
639318fe57SMatthew G. Knepley   PetscSF          sf;
649318fe57SMatthew G. Knepley   DM               dmNew = *ndm, coordDM, coarseDM;
659318fe57SMatthew G. Knepley   Vec              coords;
664fb89dddSMatthew G. Knepley   const PetscReal *maxCell, *Lstart, *L;
679318fe57SMatthew G. Knepley   PetscInt         dim, cdim;
689318fe57SMatthew G. Knepley 
699318fe57SMatthew G. Knepley   PetscFunctionBegin;
709318fe57SMatthew G. Knepley   if (dm == dmNew) {
719566063dSJacob Faibussowitsch     PetscCall(DMDestroy(ndm));
723ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
739318fe57SMatthew G. Knepley   }
749318fe57SMatthew G. Knepley   dm->setupcalled = dmNew->setupcalled;
75d0812dedSMatthew G. Knepley   if (!dm->hdr.name) {
76d0812dedSMatthew G. Knepley     const char *name;
77d0812dedSMatthew G. Knepley 
78d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)*ndm, &name));
79d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
80d0812dedSMatthew G. Knepley   }
819566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dmNew, &dim));
829566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
839566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dmNew, &cdim));
849566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
859566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmNew, &sf));
869566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sf));
879566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmNew, &coordDM));
889566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmNew, &coords));
899566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dm, coordDM));
909566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coords));
916858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmNew, &coordDM));
926858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmNew, &coords));
936858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dm, coordDM));
946858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dm, coords));
959318fe57SMatthew G. Knepley   /* Do not want to create the coordinate field if it does not already exist, so do not call DMGetCoordinateField() */
966858538eSMatthew G. Knepley   PetscCall(DMFieldDestroy(&dm->coordinates[0].field));
976858538eSMatthew G. Knepley   dm->coordinates[0].field            = dmNew->coordinates[0].field;
9861a622f3SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->coordFunc = ((DM_Plex *)dm->data)->coordFunc;
994fb89dddSMatthew G. Knepley   PetscCall(DMGetPeriodicity(dmNew, &maxCell, &Lstart, &L));
1004fb89dddSMatthew G. Knepley   PetscCall(DMSetPeriodicity(dm, maxCell, Lstart, L));
1016925d1c4SMatthew G. Knepley   PetscCall(DMPlexGetGlobalToNaturalSF(dmNew, &sf));
1026925d1c4SMatthew G. Knepley   PetscCall(DMPlexSetGlobalToNaturalSF(dm, sf));
1039566063dSJacob Faibussowitsch   PetscCall(DMDestroy_Plex(dm));
1049566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
1059318fe57SMatthew G. Knepley   dm->data = dmNew->data;
1069318fe57SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->refct++;
1071fca310dSJames Wright   {
1081fca310dSJames Wright     PetscInt       num_face_sfs;
1091fca310dSJames Wright     const PetscSF *sfs;
1101fca310dSJames Wright     PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &sfs));
1111fca310dSJames Wright     PetscCall(DMPlexSetIsoperiodicFaceSF(dm, num_face_sfs, (PetscSF *)sfs)); // for the compose function effect on dm
1121fca310dSJames Wright   }
1139566063dSJacob Faibussowitsch   PetscCall(DMDestroyLabelLinkList_Internal(dm));
1149566063dSJacob Faibussowitsch   PetscCall(DMCopyLabels(dmNew, dm, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
1159566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmNew, &coarseDM));
1169566063dSJacob Faibussowitsch   PetscCall(DMSetCoarseDM(dm, coarseDM));
1179566063dSJacob Faibussowitsch   PetscCall(DMDestroy(ndm));
1183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1199318fe57SMatthew G. Knepley }
1209318fe57SMatthew G. Knepley 
1219318fe57SMatthew G. Knepley /* Swap dm with the contents of dmNew
1229318fe57SMatthew G. Knepley    - Swap the DM_Plex structure
1239318fe57SMatthew G. Knepley    - Swap the coordinates
1249318fe57SMatthew G. Knepley    - Swap the point PetscSF
1259318fe57SMatthew G. Knepley */
126d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexSwap_Static(DM dmA, DM dmB)
127d71ae5a4SJacob Faibussowitsch {
1289318fe57SMatthew G. Knepley   DM          coordDMA, coordDMB;
1299318fe57SMatthew G. Knepley   Vec         coordsA, coordsB;
1309318fe57SMatthew G. Knepley   PetscSF     sfA, sfB;
1319318fe57SMatthew G. Knepley   DMField     fieldTmp;
1329318fe57SMatthew G. Knepley   void       *tmp;
1339318fe57SMatthew G. Knepley   DMLabelLink listTmp;
1349318fe57SMatthew G. Knepley   DMLabel     depthTmp;
1359318fe57SMatthew G. Knepley   PetscInt    tmpI;
1369318fe57SMatthew G. Knepley 
1379318fe57SMatthew G. Knepley   PetscFunctionBegin;
1383ba16761SJacob Faibussowitsch   if (dmA == dmB) PetscFunctionReturn(PETSC_SUCCESS);
1399566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmA, &sfA));
1409566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmB, &sfB));
1419566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)sfA));
1429566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmA, sfB));
1439566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmB, sfA));
1449566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)sfA));
1459318fe57SMatthew G. Knepley 
1469566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmA, &coordDMA));
1479566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmB, &coordDMB));
1489566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1499566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmA, coordDMB));
1509566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmB, coordDMA));
1519566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1529318fe57SMatthew G. Knepley 
1539566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmA, &coordsA));
1549566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmB, &coordsB));
1559566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordsA));
1569566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmA, coordsB));
1579566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmB, coordsA));
1589566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1599318fe57SMatthew G. Knepley 
1606858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmA, &coordDMA));
1616858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmB, &coordDMB));
1626858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordDMA));
1636858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmA, coordDMB));
1646858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmB, coordDMA));
1656858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
1666858538eSMatthew G. Knepley 
1676858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmA, &coordsA));
1686858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmB, &coordsB));
1696858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordsA));
1706858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmA, coordsB));
1716858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmB, coordsA));
1726858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordsA));
1736858538eSMatthew G. Knepley 
1746858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[0].field;
1756858538eSMatthew G. Knepley   dmA->coordinates[0].field = dmB->coordinates[0].field;
1766858538eSMatthew G. Knepley   dmB->coordinates[0].field = fieldTmp;
1776858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[1].field;
1786858538eSMatthew G. Knepley   dmA->coordinates[1].field = dmB->coordinates[1].field;
1796858538eSMatthew G. Knepley   dmB->coordinates[1].field = fieldTmp;
1809318fe57SMatthew G. Knepley   tmp                       = dmA->data;
1819318fe57SMatthew G. Knepley   dmA->data                 = dmB->data;
1829318fe57SMatthew G. Knepley   dmB->data                 = tmp;
1839318fe57SMatthew G. Knepley   listTmp                   = dmA->labels;
1849318fe57SMatthew G. Knepley   dmA->labels               = dmB->labels;
1859318fe57SMatthew G. Knepley   dmB->labels               = listTmp;
1869318fe57SMatthew G. Knepley   depthTmp                  = dmA->depthLabel;
1879318fe57SMatthew G. Knepley   dmA->depthLabel           = dmB->depthLabel;
1889318fe57SMatthew G. Knepley   dmB->depthLabel           = depthTmp;
1899318fe57SMatthew G. Knepley   depthTmp                  = dmA->celltypeLabel;
1909318fe57SMatthew G. Knepley   dmA->celltypeLabel        = dmB->celltypeLabel;
1919318fe57SMatthew G. Knepley   dmB->celltypeLabel        = depthTmp;
1929318fe57SMatthew G. Knepley   tmpI                      = dmA->levelup;
1939318fe57SMatthew G. Knepley   dmA->levelup              = dmB->levelup;
1949318fe57SMatthew G. Knepley   dmB->levelup              = tmpI;
1953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1969318fe57SMatthew G. Knepley }
1979318fe57SMatthew G. Knepley 
1983431e603SJed Brown PetscErrorCode DMPlexInterpolateInPlace_Internal(DM dm)
199d71ae5a4SJacob Faibussowitsch {
2009318fe57SMatthew G. Knepley   DM idm;
2019318fe57SMatthew G. Knepley 
2029318fe57SMatthew G. Knepley   PetscFunctionBegin;
2039566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolate(dm, &idm));
2049566063dSJacob Faibussowitsch   PetscCall(DMPlexCopyCoordinates(dm, idm));
20569d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &idm));
2063ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2079318fe57SMatthew G. Knepley }
2089318fe57SMatthew G. Knepley 
2099318fe57SMatthew G. Knepley /*@C
2109318fe57SMatthew G. Knepley   DMPlexCreateCoordinateSpace - Creates a finite element space for the coordinates
2119318fe57SMatthew G. Knepley 
21220f4b53cSBarry Smith   Collective
2139318fe57SMatthew G. Knepley 
2149318fe57SMatthew G. Knepley   Input Parameters:
21560225df5SJacob Faibussowitsch + dm        - The `DMPLEX`
21620f4b53cSBarry Smith . degree    - The degree of the finite element or `PETSC_DECIDE`
217e44f6aebSMatthew G. Knepley . project   - Flag to project current coordinates into the space
2189318fe57SMatthew G. Knepley - coordFunc - An optional function to map new points from refinement to the surface
2199318fe57SMatthew G. Knepley 
2209318fe57SMatthew G. Knepley   Level: advanced
2219318fe57SMatthew G. Knepley 
2221cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `PetscPointFunc`, `PetscFECreateLagrange()`, `DMGetCoordinateDM()`
2239318fe57SMatthew G. Knepley @*/
224e44f6aebSMatthew G. Knepley PetscErrorCode DMPlexCreateCoordinateSpace(DM dm, PetscInt degree, PetscBool project, PetscPointFunc coordFunc)
225d71ae5a4SJacob Faibussowitsch {
2269318fe57SMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
227e44f6aebSMatthew G. Knepley   PetscFE  fe   = NULL;
2289318fe57SMatthew G. Knepley   DM       cdm;
2291df12153SMatthew G. Knepley   PetscInt dim, dE, qorder, height;
2309318fe57SMatthew G. Knepley 
231e44f6aebSMatthew G. Knepley   PetscFunctionBegin;
2329566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
2339566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dE));
2349318fe57SMatthew G. Knepley   qorder = degree;
235e44f6aebSMatthew G. Knepley   PetscCall(DMGetCoordinateDM(dm, &cdm));
236d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)cdm);
237dc431b0cSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-default_quadrature_order", "Quadrature order is one less than quadrature points per edge", "DMPlexCreateCoordinateSpace", qorder, &qorder, NULL, 0));
238d0609cedSBarry Smith   PetscOptionsEnd();
2391df12153SMatthew G. Knepley   PetscCall(DMPlexGetVTKCellHeight(dm, &height));
240e44f6aebSMatthew G. Knepley   if (degree >= 0) {
241e44f6aebSMatthew G. Knepley     DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
242e44f6aebSMatthew G. Knepley     PetscInt       cStart, cEnd, gct;
243dc431b0cSMatthew G. Knepley 
2441df12153SMatthew G. Knepley     PetscCall(DMPlexGetHeightStratum(dm, height, &cStart, &cEnd));
245dc431b0cSMatthew G. Knepley     if (cEnd > cStart) PetscCall(DMPlexGetCellType(dm, cStart, &ct));
246e44f6aebSMatthew G. Knepley     gct = (PetscInt)ct;
247e44f6aebSMatthew G. Knepley     PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &gct, 1, MPIU_INT, MPI_MIN, PetscObjectComm((PetscObject)dm)));
248e44f6aebSMatthew G. Knepley     ct = (DMPolytopeType)gct;
249e44f6aebSMatthew G. Knepley     // Work around current bug in PetscDualSpaceSetUp_Lagrange()
250e44f6aebSMatthew G. Knepley     //   Can be seen in plex_tutorials-ex10_1
251e44f6aebSMatthew 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));
2524f9ab2b4SJed Brown   }
253e44f6aebSMatthew G. Knepley   PetscCall(DMSetCoordinateDisc(dm, fe, project));
2549566063dSJacob Faibussowitsch   PetscCall(PetscFEDestroy(&fe));
2559318fe57SMatthew G. Knepley   mesh->coordFunc = coordFunc;
2563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2579318fe57SMatthew G. Knepley }
2589318fe57SMatthew G. Knepley 
2591df5d5c5SMatthew G. Knepley /*@
2601df5d5c5SMatthew G. Knepley   DMPlexCreateDoublet - Creates a mesh of two cells of the specified type, optionally with later refinement.
2611df5d5c5SMatthew G. Knepley 
262d083f849SBarry Smith   Collective
2631df5d5c5SMatthew G. Knepley 
2641df5d5c5SMatthew G. Knepley   Input Parameters:
265a1cb98faSBarry Smith + comm            - The communicator for the `DM` object
2661df5d5c5SMatthew G. Knepley . dim             - The spatial dimension
2671df5d5c5SMatthew G. Knepley . simplex         - Flag for simplicial cells, otherwise they are tensor product cells
2681df5d5c5SMatthew G. Knepley . interpolate     - Flag to create intermediate mesh pieces (edges, faces)
2691df5d5c5SMatthew G. Knepley - refinementLimit - A nonzero number indicates the largest admissible volume for a refined cell
2701df5d5c5SMatthew G. Knepley 
2711df5d5c5SMatthew G. Knepley   Output Parameter:
27260225df5SJacob Faibussowitsch . newdm - The `DM` object
2731df5d5c5SMatthew G. Knepley 
2741df5d5c5SMatthew G. Knepley   Level: beginner
2751df5d5c5SMatthew G. Knepley 
2761cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetType()`, `DMCreate()`
2771df5d5c5SMatthew G. Knepley @*/
278d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateDoublet(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscBool interpolate, PetscReal refinementLimit, DM *newdm)
279d71ae5a4SJacob Faibussowitsch {
2801df5d5c5SMatthew G. Knepley   DM          dm;
2811df5d5c5SMatthew G. Knepley   PetscMPIInt rank;
2821df5d5c5SMatthew G. Knepley 
2831df5d5c5SMatthew G. Knepley   PetscFunctionBegin;
2849566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, &dm));
2859566063dSJacob Faibussowitsch   PetscCall(DMSetType(dm, DMPLEX));
2869566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
28746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
2889566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
289ce78fa2fSMatthew G. Knepley   switch (dim) {
290ce78fa2fSMatthew G. Knepley   case 2:
2919566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "triangular"));
2929566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "quadrilateral"));
293ce78fa2fSMatthew G. Knepley     break;
294ce78fa2fSMatthew G. Knepley   case 3:
2959566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "tetrahedral"));
2969566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "hexahedral"));
297ce78fa2fSMatthew G. Knepley     break;
298d71ae5a4SJacob Faibussowitsch   default:
299d71ae5a4SJacob Faibussowitsch     SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
300ce78fa2fSMatthew G. Knepley   }
3011df5d5c5SMatthew G. Knepley   if (rank) {
3021df5d5c5SMatthew G. Knepley     PetscInt numPoints[2] = {0, 0};
3039566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, NULL, NULL, NULL, NULL));
3041df5d5c5SMatthew G. Knepley   } else {
3051df5d5c5SMatthew G. Knepley     switch (dim) {
3061df5d5c5SMatthew G. Knepley     case 2:
3071df5d5c5SMatthew G. Knepley       if (simplex) {
3081df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {4, 2};
3091df5d5c5SMatthew G. Knepley         PetscInt    coneSize[6]         = {3, 3, 0, 0, 0, 0};
3101df5d5c5SMatthew G. Knepley         PetscInt    cones[6]            = {2, 3, 4, 5, 4, 3};
3111df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
3121df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[8]     = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
3131df5d5c5SMatthew G. Knepley 
3149566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3151df5d5c5SMatthew G. Knepley       } else {
3161df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {6, 2};
3171df5d5c5SMatthew G. Knepley         PetscInt    coneSize[8]         = {4, 4, 0, 0, 0, 0, 0, 0};
3181df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {2, 3, 4, 5, 3, 6, 7, 4};
3191df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3201df5d5c5SMatthew 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};
3211df5d5c5SMatthew G. Knepley 
3229566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3231df5d5c5SMatthew G. Knepley       }
3241df5d5c5SMatthew G. Knepley       break;
3251df5d5c5SMatthew G. Knepley     case 3:
3261df5d5c5SMatthew G. Knepley       if (simplex) {
3271df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {5, 2};
3281df5d5c5SMatthew G. Knepley         PetscInt    coneSize[7]         = {4, 4, 0, 0, 0, 0, 0};
3291df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {4, 3, 5, 2, 5, 3, 4, 6};
3301df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
3311df5d5c5SMatthew 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};
3321df5d5c5SMatthew G. Knepley 
3339566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3341df5d5c5SMatthew G. Knepley       } else {
3351df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]         = {12, 2};
3361df5d5c5SMatthew G. Knepley         PetscInt    coneSize[14]         = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3371df5d5c5SMatthew G. Knepley         PetscInt    cones[16]            = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
3381df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
3399371c9d4SSatish 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};
3401df5d5c5SMatthew G. Knepley 
3419566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
3421df5d5c5SMatthew G. Knepley       }
3431df5d5c5SMatthew G. Knepley       break;
344d71ae5a4SJacob Faibussowitsch     default:
345d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
3461df5d5c5SMatthew G. Knepley     }
3471df5d5c5SMatthew G. Knepley   }
34846139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
3491df5d5c5SMatthew G. Knepley   *newdm = dm;
3501df5d5c5SMatthew G. Knepley   if (refinementLimit > 0.0) {
3511df5d5c5SMatthew G. Knepley     DM          rdm;
3521df5d5c5SMatthew G. Knepley     const char *name;
3531df5d5c5SMatthew G. Knepley 
3549566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(*newdm, PETSC_FALSE));
3559566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(*newdm, refinementLimit));
3569566063dSJacob Faibussowitsch     PetscCall(DMRefine(*newdm, comm, &rdm));
3579566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)*newdm, &name));
3589566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)rdm, name));
3599566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3601df5d5c5SMatthew G. Knepley     *newdm = rdm;
3611df5d5c5SMatthew G. Knepley   }
3621df5d5c5SMatthew G. Knepley   if (interpolate) {
3635fd9971aSMatthew G. Knepley     DM idm;
3641df5d5c5SMatthew G. Knepley 
3659566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*newdm, &idm));
3669566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
3671df5d5c5SMatthew G. Knepley     *newdm = idm;
3681df5d5c5SMatthew G. Knepley   }
3693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3701df5d5c5SMatthew G. Knepley }
3711df5d5c5SMatthew G. Knepley 
372d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
373d71ae5a4SJacob Faibussowitsch {
3749318fe57SMatthew G. Knepley   const PetscInt numVertices    = 2;
3759318fe57SMatthew G. Knepley   PetscInt       markerRight    = 1;
3769318fe57SMatthew G. Knepley   PetscInt       markerLeft     = 1;
3779318fe57SMatthew G. Knepley   PetscBool      markerSeparate = PETSC_FALSE;
3789318fe57SMatthew G. Knepley   Vec            coordinates;
3799318fe57SMatthew G. Knepley   PetscSection   coordSection;
3809318fe57SMatthew G. Knepley   PetscScalar   *coords;
3819318fe57SMatthew G. Knepley   PetscInt       coordSize;
3829318fe57SMatthew G. Knepley   PetscMPIInt    rank;
3839318fe57SMatthew G. Knepley   PetscInt       cdim = 1, v;
384552f7358SJed Brown 
3859318fe57SMatthew G. Knepley   PetscFunctionBegin;
3869566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
3879318fe57SMatthew G. Knepley   if (markerSeparate) {
3889318fe57SMatthew G. Knepley     markerRight = 2;
3899318fe57SMatthew G. Knepley     markerLeft  = 1;
3909318fe57SMatthew G. Knepley   }
3919566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
392c5853193SPierre Jolivet   if (rank == 0) {
3939566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numVertices));
3949566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
3959566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 0, markerLeft));
3969566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 1, markerRight));
3979318fe57SMatthew G. Knepley   }
3989566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
3999566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
4009318fe57SMatthew G. Knepley   /* Build coordinates */
4019566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
4029566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
4039566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
4049566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, 0, numVertices));
4059566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
4069318fe57SMatthew G. Knepley   for (v = 0; v < numVertices; ++v) {
4079566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
4089566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
4099318fe57SMatthew G. Knepley   }
4109566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
4119566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
4129566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
4139566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
4149566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
4159566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
4169566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
4179566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
4189318fe57SMatthew G. Knepley   coords[0] = lower[0];
4199318fe57SMatthew G. Knepley   coords[1] = upper[0];
4209566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
4219566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
4229566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
4233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4249318fe57SMatthew G. Knepley }
42526492d91SMatthew G. Knepley 
426d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
427d71ae5a4SJacob Faibussowitsch {
4281df21d24SMatthew G. Knepley   const PetscInt numVertices    = (edges[0] + 1) * (edges[1] + 1);
4291df21d24SMatthew G. Knepley   const PetscInt numEdges       = edges[0] * (edges[1] + 1) + (edges[0] + 1) * edges[1];
430552f7358SJed Brown   PetscInt       markerTop      = 1;
431552f7358SJed Brown   PetscInt       markerBottom   = 1;
432552f7358SJed Brown   PetscInt       markerRight    = 1;
433552f7358SJed Brown   PetscInt       markerLeft     = 1;
434552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
435552f7358SJed Brown   Vec            coordinates;
436552f7358SJed Brown   PetscSection   coordSection;
437552f7358SJed Brown   PetscScalar   *coords;
438552f7358SJed Brown   PetscInt       coordSize;
439552f7358SJed Brown   PetscMPIInt    rank;
440552f7358SJed Brown   PetscInt       v, vx, vy;
441552f7358SJed Brown 
442552f7358SJed Brown   PetscFunctionBegin;
4439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
444552f7358SJed Brown   if (markerSeparate) {
4451df21d24SMatthew G. Knepley     markerTop    = 3;
4461df21d24SMatthew G. Knepley     markerBottom = 1;
4471df21d24SMatthew G. Knepley     markerRight  = 2;
4481df21d24SMatthew G. Knepley     markerLeft   = 4;
449552f7358SJed Brown   }
4509566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
451dd400576SPatrick Sanan   if (rank == 0) {
452552f7358SJed Brown     PetscInt e, ex, ey;
453552f7358SJed Brown 
4549566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numEdges + numVertices));
45548a46eb9SPierre Jolivet     for (e = 0; e < numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
4569566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
457552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
458552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
459552f7358SJed Brown         PetscInt edge   = vx * edges[1] + ey + edges[0] * (edges[1] + 1);
460552f7358SJed Brown         PetscInt vertex = ey * (edges[0] + 1) + vx + numEdges;
461da80777bSKarl Rupp         PetscInt cone[2];
462552f7358SJed Brown 
4639371c9d4SSatish Balay         cone[0] = vertex;
4649371c9d4SSatish Balay         cone[1] = vertex + edges[0] + 1;
4659566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
466552f7358SJed Brown         if (vx == edges[0]) {
4679566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
4689566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
469552f7358SJed Brown           if (ey == edges[1] - 1) {
4709566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
4719566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerRight));
472552f7358SJed Brown           }
473552f7358SJed Brown         } else if (vx == 0) {
4749566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
4759566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
476552f7358SJed Brown           if (ey == edges[1] - 1) {
4779566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
4789566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerLeft));
479552f7358SJed Brown           }
480552f7358SJed Brown         }
481552f7358SJed Brown       }
482552f7358SJed Brown     }
483552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
484552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
485552f7358SJed Brown         PetscInt edge   = vy * edges[0] + ex;
486552f7358SJed Brown         PetscInt vertex = vy * (edges[0] + 1) + ex + numEdges;
487da80777bSKarl Rupp         PetscInt cone[2];
488552f7358SJed Brown 
4899371c9d4SSatish Balay         cone[0] = vertex;
4909371c9d4SSatish Balay         cone[1] = vertex + 1;
4919566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
492552f7358SJed Brown         if (vy == edges[1]) {
4939566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
4949566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
495552f7358SJed Brown           if (ex == edges[0] - 1) {
4969566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
4979566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerTop));
498552f7358SJed Brown           }
499552f7358SJed Brown         } else if (vy == 0) {
5009566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
5019566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
502552f7358SJed Brown           if (ex == edges[0] - 1) {
5039566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
5049566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerBottom));
505552f7358SJed Brown           }
506552f7358SJed Brown         }
507552f7358SJed Brown       }
508552f7358SJed Brown     }
509552f7358SJed Brown   }
5109566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
5119566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
512552f7358SJed Brown   /* Build coordinates */
5139566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 2));
5149566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
5159566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
5169566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices));
5179566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 2));
518552f7358SJed Brown   for (v = numEdges; v < numEdges + numVertices; ++v) {
5199566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 2));
5209566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 2));
521552f7358SJed Brown   }
5229566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5239566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
5249566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
5259566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
5269566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
5279566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 2));
5289566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
5299566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
530552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
531552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
532552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 0] = lower[0] + ((upper[0] - lower[0]) / edges[0]) * vx;
533552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 1] = lower[1] + ((upper[1] - lower[1]) / edges[1]) * vy;
534552f7358SJed Brown     }
535552f7358SJed Brown   }
5369566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
5379566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
5389566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
5393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
540552f7358SJed Brown }
541552f7358SJed Brown 
542d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
543d71ae5a4SJacob Faibussowitsch {
5449e8abbc3SMichael Lange   PetscInt     vertices[3], numVertices;
5457b59f5a9SMichael Lange   PetscInt     numFaces       = 2 * faces[0] * faces[1] + 2 * faces[1] * faces[2] + 2 * faces[0] * faces[2];
546c2df9bbfSMatthew G. Knepley   PetscInt     markerTop      = 1;
547c2df9bbfSMatthew G. Knepley   PetscInt     markerBottom   = 1;
548c2df9bbfSMatthew G. Knepley   PetscInt     markerFront    = 1;
549c2df9bbfSMatthew G. Knepley   PetscInt     markerBack     = 1;
550c2df9bbfSMatthew G. Knepley   PetscInt     markerRight    = 1;
551c2df9bbfSMatthew G. Knepley   PetscInt     markerLeft     = 1;
552c2df9bbfSMatthew G. Knepley   PetscBool    markerSeparate = PETSC_FALSE;
553552f7358SJed Brown   Vec          coordinates;
554552f7358SJed Brown   PetscSection coordSection;
555552f7358SJed Brown   PetscScalar *coords;
556552f7358SJed Brown   PetscInt     coordSize;
557552f7358SJed Brown   PetscMPIInt  rank;
558552f7358SJed Brown   PetscInt     v, vx, vy, vz;
5597b59f5a9SMichael Lange   PetscInt     voffset, iface = 0, cone[4];
560552f7358SJed Brown 
561552f7358SJed Brown   PetscFunctionBegin;
5621dca8a05SBarry 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");
5639566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
564c2df9bbfSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
565c2df9bbfSMatthew G. Knepley   if (markerSeparate) {
566c2df9bbfSMatthew G. Knepley     markerBottom = 1;
567c2df9bbfSMatthew G. Knepley     markerTop    = 2;
568c2df9bbfSMatthew G. Knepley     markerFront  = 3;
569c2df9bbfSMatthew G. Knepley     markerBack   = 4;
570c2df9bbfSMatthew G. Knepley     markerRight  = 5;
571c2df9bbfSMatthew G. Knepley     markerLeft   = 6;
572c2df9bbfSMatthew G. Knepley   }
5739371c9d4SSatish Balay   vertices[0] = faces[0] + 1;
5749371c9d4SSatish Balay   vertices[1] = faces[1] + 1;
5759371c9d4SSatish Balay   vertices[2] = faces[2] + 1;
5769e8abbc3SMichael Lange   numVertices = vertices[0] * vertices[1] * vertices[2];
577dd400576SPatrick Sanan   if (rank == 0) {
578552f7358SJed Brown     PetscInt f;
579552f7358SJed Brown 
5809566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numFaces + numVertices));
58148a46eb9SPierre Jolivet     for (f = 0; f < numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
5829566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
5837b59f5a9SMichael Lange 
5847b59f5a9SMichael Lange     /* Side 0 (Top) */
5857b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
5867b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
5877b59f5a9SMichael Lange         voffset = numFaces + vertices[0] * vertices[1] * (vertices[2] - 1) + vy * vertices[0] + vx;
5889371c9d4SSatish Balay         cone[0] = voffset;
5899371c9d4SSatish Balay         cone[1] = voffset + 1;
5909371c9d4SSatish Balay         cone[2] = voffset + vertices[0] + 1;
5919371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
5929566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
593c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerTop));
594c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerTop));
595c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerTop));
596c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerTop));
597c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerTop));
5987b59f5a9SMichael Lange         iface++;
599552f7358SJed Brown       }
600552f7358SJed Brown     }
6017b59f5a9SMichael Lange 
6027b59f5a9SMichael Lange     /* Side 1 (Bottom) */
6037b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
6047b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6057b59f5a9SMichael Lange         voffset = numFaces + vy * (faces[0] + 1) + vx;
6069371c9d4SSatish Balay         cone[0] = voffset + 1;
6079371c9d4SSatish Balay         cone[1] = voffset;
6089371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6099371c9d4SSatish Balay         cone[3] = voffset + vertices[0] + 1;
6109566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
611c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBottom));
612c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBottom));
613c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBottom));
614c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerBottom));
615c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerBottom));
6167b59f5a9SMichael Lange         iface++;
617552f7358SJed Brown       }
618552f7358SJed Brown     }
6197b59f5a9SMichael Lange 
6207b59f5a9SMichael Lange     /* Side 2 (Front) */
6217b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6227b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6237b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vx;
6249371c9d4SSatish Balay         cone[0] = voffset;
6259371c9d4SSatish Balay         cone[1] = voffset + 1;
6269371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + 1;
6279371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1];
6289566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
629c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerFront));
630c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerFront));
631c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerFront));
632c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerFront));
633c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerFront));
6347b59f5a9SMichael Lange         iface++;
635552f7358SJed Brown       }
6367b59f5a9SMichael Lange     }
6377b59f5a9SMichael Lange 
6387b59f5a9SMichael Lange     /* Side 3 (Back) */
6397b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6407b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
6417b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vertices[0] * (vertices[1] - 1) + vx;
6429371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6439371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1] + 1;
6449371c9d4SSatish Balay         cone[2] = voffset + 1;
6459371c9d4SSatish Balay         cone[3] = voffset;
6469566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
647c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBack));
648c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBack));
649c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBack));
650c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerBack));
651c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerBack));
6527b59f5a9SMichael Lange         iface++;
6537b59f5a9SMichael Lange       }
6547b59f5a9SMichael Lange     }
6557b59f5a9SMichael Lange 
6567b59f5a9SMichael Lange     /* Side 4 (Left) */
6577b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6587b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
6597b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0];
6609371c9d4SSatish Balay         cone[0] = voffset;
6619371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1];
6629371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + vertices[0];
6639371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
6649566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
665c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerLeft));
666c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerLeft));
667c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerLeft));
668c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[1] + 0, markerLeft));
669c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerLeft));
6707b59f5a9SMichael Lange         iface++;
6717b59f5a9SMichael Lange       }
6727b59f5a9SMichael Lange     }
6737b59f5a9SMichael Lange 
6747b59f5a9SMichael Lange     /* Side 5 (Right) */
6757b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
6767b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
677aab5bcd8SJed Brown         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0] + faces[0];
6789371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
6799371c9d4SSatish Balay         cone[1] = voffset;
6809371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
6819371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1] + vertices[0];
6829566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
683c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerRight));
684c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerRight));
685c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerRight));
686c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerRight));
687c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerRight));
6887b59f5a9SMichael Lange         iface++;
6897b59f5a9SMichael Lange       }
690552f7358SJed Brown     }
691552f7358SJed Brown   }
6929566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
6939566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
694552f7358SJed Brown   /* Build coordinates */
6959566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 3));
6969566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
6979566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
6989566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices));
6999566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 3));
700552f7358SJed Brown   for (v = numFaces; v < numFaces + numVertices; ++v) {
7019566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 3));
7029566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 3));
703552f7358SJed Brown   }
7049566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
7059566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
7069566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
7079566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
7089566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
7099566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 3));
7109566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
7119566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
712552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
713552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
714552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
715552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 0] = lower[0] + ((upper[0] - lower[0]) / faces[0]) * vx;
716552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 1] = lower[1] + ((upper[1] - lower[1]) / faces[1]) * vy;
717552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 2] = lower[2] + ((upper[2] - lower[2]) / faces[2]) * vz;
718552f7358SJed Brown       }
719552f7358SJed Brown     }
720552f7358SJed Brown   }
7219566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
7229566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
7239566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
7243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
725552f7358SJed Brown }
726552f7358SJed Brown 
727d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate)
728d71ae5a4SJacob Faibussowitsch {
7299318fe57SMatthew G. Knepley   PetscFunctionBegin;
7309318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
73146139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
7329566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim - 1));
7339566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim));
7349318fe57SMatthew G. Knepley   switch (dim) {
735d71ae5a4SJacob Faibussowitsch   case 1:
736d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(dm, lower, upper, faces));
737d71ae5a4SJacob Faibussowitsch     break;
738d71ae5a4SJacob Faibussowitsch   case 2:
739d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(dm, lower, upper, faces));
740d71ae5a4SJacob Faibussowitsch     break;
741d71ae5a4SJacob Faibussowitsch   case 3:
742d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(dm, lower, upper, faces));
743d71ae5a4SJacob Faibussowitsch     break;
744d71ae5a4SJacob Faibussowitsch   default:
745d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Dimension not supported: %" PetscInt_FMT, dim);
7469318fe57SMatthew G. Knepley   }
74746139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
7489566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
7493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7509318fe57SMatthew G. Knepley }
7519318fe57SMatthew G. Knepley 
7529318fe57SMatthew G. Knepley /*@C
7539318fe57SMatthew G. Knepley   DMPlexCreateBoxSurfaceMesh - Creates a mesh on the surface of the tensor product of unit intervals (box) using tensor cells (hexahedra).
7549318fe57SMatthew G. Knepley 
7559318fe57SMatthew G. Knepley   Collective
7569318fe57SMatthew G. Knepley 
7579318fe57SMatthew G. Knepley   Input Parameters:
758a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
75920f4b53cSBarry Smith . dim         - The spatial dimension of the box, so the resulting mesh is has dimension `dim`-1
76020f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
76120f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
76220f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
7639318fe57SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
7649318fe57SMatthew G. Knepley 
7659318fe57SMatthew G. Knepley   Output Parameter:
766a1cb98faSBarry Smith . dm - The `DM` object
7679318fe57SMatthew G. Knepley 
7689318fe57SMatthew G. Knepley   Level: beginner
7699318fe57SMatthew G. Knepley 
7701cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateBoxMesh()`, `DMPlexCreateFromFile()`, `DMSetType()`, `DMCreate()`
7719318fe57SMatthew G. Knepley @*/
772d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBoxSurfaceMesh(MPI_Comm comm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate, DM *dm)
773d71ae5a4SJacob Faibussowitsch {
7749318fe57SMatthew G. Knepley   PetscInt  fac[3] = {1, 1, 1};
7759318fe57SMatthew G. Knepley   PetscReal low[3] = {0, 0, 0};
7769318fe57SMatthew G. Knepley   PetscReal upp[3] = {1, 1, 1};
7779318fe57SMatthew G. Knepley 
7789318fe57SMatthew G. Knepley   PetscFunctionBegin;
7799566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
7809566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
7819566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(*dm, dim, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, interpolate));
7823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
7839318fe57SMatthew G. Knepley }
7849318fe57SMatthew G. Knepley 
785d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateLineMesh_Internal(DM dm, PetscInt segments, PetscReal lower, PetscReal upper, DMBoundaryType bd)
786d71ae5a4SJacob Faibussowitsch {
787fdbf62faSLisandro Dalcin   PetscInt     i, fStart, fEnd, numCells = 0, numVerts = 0;
788fdbf62faSLisandro Dalcin   PetscInt     numPoints[2], *coneSize, *cones, *coneOrientations;
789fdbf62faSLisandro Dalcin   PetscScalar *vertexCoords;
790fdbf62faSLisandro Dalcin   PetscReal    L, maxCell;
791fdbf62faSLisandro Dalcin   PetscBool    markerSeparate = PETSC_FALSE;
792fdbf62faSLisandro Dalcin   PetscInt     markerLeft = 1, faceMarkerLeft = 1;
793fdbf62faSLisandro Dalcin   PetscInt     markerRight = 1, faceMarkerRight = 2;
794fdbf62faSLisandro Dalcin   PetscBool    wrap = (bd == DM_BOUNDARY_PERIODIC || bd == DM_BOUNDARY_TWIST) ? PETSC_TRUE : PETSC_FALSE;
795fdbf62faSLisandro Dalcin   PetscMPIInt  rank;
796fdbf62faSLisandro Dalcin 
797fdbf62faSLisandro Dalcin   PetscFunctionBegin;
7984f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
799fdbf62faSLisandro Dalcin 
8009566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, 1));
8019566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
8029566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
803fdbf62faSLisandro Dalcin 
8049566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
805dd400576SPatrick Sanan   if (rank == 0) numCells = segments;
806dd400576SPatrick Sanan   if (rank == 0) numVerts = segments + (wrap ? 0 : 1);
807fdbf62faSLisandro Dalcin 
8089371c9d4SSatish Balay   numPoints[0] = numVerts;
8099371c9d4SSatish Balay   numPoints[1] = numCells;
8109566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(numCells + numVerts, &coneSize, numCells * 2, &cones, numCells + numVerts, &coneOrientations, numVerts, &vertexCoords));
8119566063dSJacob Faibussowitsch   PetscCall(PetscArrayzero(coneOrientations, numCells + numVerts));
812ad540459SPierre Jolivet   for (i = 0; i < numCells; ++i) coneSize[i] = 2;
813ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) coneSize[numCells + i] = 0;
8149371c9d4SSatish Balay   for (i = 0; i < numCells; ++i) {
8159371c9d4SSatish Balay     cones[2 * i]     = numCells + i % numVerts;
8169371c9d4SSatish Balay     cones[2 * i + 1] = numCells + (i + 1) % numVerts;
8179371c9d4SSatish Balay   }
818ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) vertexCoords[i] = lower + (upper - lower) * ((PetscReal)i / (PetscReal)numCells);
8199566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
8209566063dSJacob Faibussowitsch   PetscCall(PetscFree4(coneSize, cones, coneOrientations, vertexCoords));
821fdbf62faSLisandro Dalcin 
8229566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
8239371c9d4SSatish Balay   if (markerSeparate) {
8249371c9d4SSatish Balay     markerLeft  = faceMarkerLeft;
8259371c9d4SSatish Balay     markerRight = faceMarkerRight;
8269371c9d4SSatish Balay   }
827dd400576SPatrick Sanan   if (!wrap && rank == 0) {
8289566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
8299566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fStart, markerLeft));
8309566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fEnd - 1, markerRight));
8319566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fStart, faceMarkerLeft));
8329566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fEnd - 1, faceMarkerRight));
833fdbf62faSLisandro Dalcin   }
834fdbf62faSLisandro Dalcin   if (wrap) {
835fdbf62faSLisandro Dalcin     L       = upper - lower;
836fdbf62faSLisandro Dalcin     maxCell = (PetscReal)1.1 * (L / (PetscReal)PetscMax(1, segments));
8374fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, &maxCell, &lower, &L));
838fdbf62faSLisandro Dalcin   }
8399566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
8403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
841fdbf62faSLisandro Dalcin }
842fdbf62faSLisandro Dalcin 
8434054ae39SJames Wright // Creates "Face Sets" label based on the standard box labeling conventions
8444054ae39SJames Wright static PetscErrorCode DMPlexSetBoxLabel_Internal(DM dm)
8454054ae39SJames Wright {
8466ff49feeSJames Wright   DM              cdm;
8476ff49feeSJames Wright   PetscSection    csection;
8486ff49feeSJames Wright   Vec             coordinates;
8494054ae39SJames Wright   DMLabel         label;
8506ff49feeSJames Wright   IS              faces_is;
8514054ae39SJames Wright   PetscInt        dim, num_face;
8524054ae39SJames Wright   const PetscInt *faces;
8534054ae39SJames Wright   PetscInt        faceMarkerBottom, faceMarkerTop, faceMarkerFront, faceMarkerBack, faceMarkerRight, faceMarkerLeft;
8544054ae39SJames Wright 
8554054ae39SJames Wright   PetscFunctionBeginUser;
8564054ae39SJames Wright   PetscCall(DMGetDimension(dm, &dim));
8576ff49feeSJames Wright   PetscCheck((dim == 2) || (dim == 3), PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "DMPlex box labeling only supports 2D and 3D meshes, recieved DM of dimension %" PetscInt_FMT, dim);
8584054ae39SJames Wright   // Get Face Sets label
8594054ae39SJames Wright   PetscCall(DMGetLabel(dm, "Face Sets", &label));
8604054ae39SJames Wright   if (label) {
8614054ae39SJames Wright     PetscCall(DMLabelReset(label));
8624054ae39SJames Wright   } else {
8634054ae39SJames Wright     PetscCall(DMCreateLabel(dm, "Face Sets"));
8644054ae39SJames Wright     PetscCall(DMGetLabel(dm, "Face Sets", &label));
8654054ae39SJames Wright   }
8664054ae39SJames Wright   PetscCall(DMPlexMarkBoundaryFaces(dm, 1, label));
8676ff49feeSJames Wright   PetscCall(DMGetStratumIS(dm, "Face Sets", 1, &faces_is));
8686ff49feeSJames Wright   if (!faces_is) PetscFunctionReturn(PETSC_SUCCESS); // No faces on rank
8694054ae39SJames Wright 
8704054ae39SJames Wright   switch (dim) {
8714054ae39SJames Wright   case 2:
8724054ae39SJames Wright     faceMarkerTop    = 3;
8734054ae39SJames Wright     faceMarkerBottom = 1;
8744054ae39SJames Wright     faceMarkerRight  = 2;
8754054ae39SJames Wright     faceMarkerLeft   = 4;
8764054ae39SJames Wright     break;
8774054ae39SJames Wright   case 3:
8784054ae39SJames Wright     faceMarkerBottom = 1;
8794054ae39SJames Wright     faceMarkerTop    = 2;
8804054ae39SJames Wright     faceMarkerFront  = 3;
8814054ae39SJames Wright     faceMarkerBack   = 4;
8824054ae39SJames Wright     faceMarkerRight  = 5;
8834054ae39SJames Wright     faceMarkerLeft   = 6;
8844054ae39SJames Wright     break;
8854054ae39SJames Wright   default:
8864054ae39SJames Wright     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
8874054ae39SJames Wright   }
8884054ae39SJames Wright 
8896ff49feeSJames Wright   PetscCall(ISGetLocalSize(faces_is, &num_face));
8906ff49feeSJames Wright   PetscCall(ISGetIndices(faces_is, &faces));
8916ff49feeSJames Wright   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
8926ff49feeSJames Wright   PetscCall(DMGetCoordinateDM(dm, &cdm));
8936ff49feeSJames Wright   PetscCall(DMGetLocalSection(cdm, &csection));
8944054ae39SJames Wright   for (PetscInt f = 0; f < num_face; ++f) {
8956ff49feeSJames Wright     PetscScalar *coords = NULL;
8966ff49feeSJames Wright     PetscInt     face = faces[f], flip = 1, label_value = -1, coords_size;
8974054ae39SJames Wright 
8984054ae39SJames Wright     { // Determine if orientation of face is flipped
8994054ae39SJames Wright       PetscInt        num_cells_support, num_faces, start = -1;
9004054ae39SJames Wright       const PetscInt *orients, *cell_faces, *cells;
9014054ae39SJames Wright 
9024054ae39SJames Wright       PetscCall(DMPlexGetSupport(dm, face, &cells));
9034054ae39SJames Wright       PetscCall(DMPlexGetSupportSize(dm, face, &num_cells_support));
9044054ae39SJames 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);
9054054ae39SJames Wright       PetscCall(DMPlexGetCone(dm, cells[0], &cell_faces));
9064054ae39SJames Wright       PetscCall(DMPlexGetConeSize(dm, cells[0], &num_faces));
9074054ae39SJames Wright       for (PetscInt i = 0; i < num_faces; i++) {
9084054ae39SJames Wright         if (cell_faces[i] == face) start = i;
9094054ae39SJames Wright       }
9104054ae39SJames Wright       PetscCheck(start >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT, "Could not find face %" PetscInt_FMT " in cone of its support", face);
9114054ae39SJames Wright       PetscCall(DMPlexGetConeOrientation(dm, cells[0], &orients));
9124054ae39SJames Wright       if (orients[start] < 0) flip = -1;
9134054ae39SJames Wright     }
9144054ae39SJames Wright 
9156ff49feeSJames Wright     // Cannot use DMPlexComputeCellGeometryFVM() for high-order geometry, so must calculate normal vectors manually
9166ff49feeSJames Wright     // Use the vertices (depth 0) of coordinate DM to calculate normal vector
9176ff49feeSJames Wright     PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
9184054ae39SJames Wright     switch (dim) {
9194054ae39SJames Wright     case 2: {
9206ff49feeSJames Wright       PetscScalar vec[2];
9216ff49feeSJames Wright 
9226ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) vec[d] = flip * (PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]));
9236ff49feeSJames Wright       PetscScalar normal[] = {vec[1], -vec[0]};
9246ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
9256ff49feeSJames Wright         label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
9264054ae39SJames Wright       } else {
9276ff49feeSJames Wright         label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerTop : faceMarkerBottom;
9284054ae39SJames Wright       }
9294054ae39SJames Wright     } break;
9304054ae39SJames Wright     case 3: {
9316ff49feeSJames Wright       PetscScalar vec1[3], vec2[3], normal[3];
9326ff49feeSJames Wright 
9336ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) {
9346ff49feeSJames Wright         vec1[d] = PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]);
9356ff49feeSJames Wright         vec2[d] = PetscRealPart(coords[2 * dim + d]) - PetscRealPart(coords[1 * dim + d]);
9366ff49feeSJames Wright       }
9376ff49feeSJames Wright 
9386ff49feeSJames Wright       // Calculate normal vector via cross-product
9396ff49feeSJames Wright       normal[0] = flip * ((vec1[1] * vec2[2]) - (vec1[2] * vec2[1]));
9406ff49feeSJames Wright       normal[1] = flip * ((vec1[2] * vec2[0]) - (vec1[0] * vec2[2]));
9416ff49feeSJames Wright       normal[2] = flip * ((vec1[0] * vec2[1]) - (vec1[1] * vec2[0]));
9426ff49feeSJames Wright 
9436ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
9446ff49feeSJames Wright         if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[2])) {
9456ff49feeSJames Wright           label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
9464054ae39SJames Wright         } else {
9476ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
9484054ae39SJames Wright         }
9494054ae39SJames Wright       } else {
9506ff49feeSJames Wright         if (PetscAbsScalar(normal[1]) > PetscAbsScalar(normal[2])) {
9516ff49feeSJames Wright           label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerBack : faceMarkerFront;
9524054ae39SJames Wright         } else {
9536ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
9544054ae39SJames Wright         }
9554054ae39SJames Wright       }
9564054ae39SJames Wright     } break;
9574054ae39SJames Wright     }
9584054ae39SJames Wright 
9594054ae39SJames Wright     PetscInt previous_label_value; // always 1 due to DMPlexMarkBoundaryFaces call above
9604054ae39SJames Wright     PetscCall(DMGetLabelValue(dm, "Face Sets", face, &previous_label_value));
9614054ae39SJames Wright     PetscCall(DMClearLabelValue(dm, "Face Sets", face, previous_label_value));
9624054ae39SJames Wright     PetscCall(DMSetLabelValue(dm, "Face Sets", face, label_value));
9636ff49feeSJames Wright     PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
9644054ae39SJames Wright   }
9656ff49feeSJames Wright   PetscCall(ISRestoreIndices(faces_is, &faces));
9666ff49feeSJames Wright   PetscCall(ISDestroy(&faces_is));
9674054ae39SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
9684054ae39SJames Wright }
9694054ae39SJames Wright 
970d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Simplex_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate)
971d71ae5a4SJacob Faibussowitsch {
9729318fe57SMatthew G. Knepley   DM      boundary, vol;
973c22d3578SMatthew G. Knepley   DMLabel bdlabel;
974d6218766SMatthew G. Knepley 
975d6218766SMatthew G. Knepley   PetscFunctionBegin;
9764f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
977c22d3578SMatthew 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");
9789566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &boundary));
9799566063dSJacob Faibussowitsch   PetscCall(DMSetType(boundary, DMPLEX));
9809566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(boundary, dim, faces, lower, upper, PETSC_FALSE));
9819566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(boundary, NULL, interpolate, &vol));
982c22d3578SMatthew G. Knepley   PetscCall(DMGetLabel(vol, "marker", &bdlabel));
983c22d3578SMatthew G. Knepley   if (bdlabel) PetscCall(DMPlexLabelComplete(vol, bdlabel));
9845de52c6dSVaclav Hapla   PetscCall(DMPlexCopy_Internal(dm, PETSC_TRUE, PETSC_FALSE, vol));
98569d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
9864054ae39SJames Wright   if (interpolate) {
9874054ae39SJames Wright     PetscCall(DMPlexInterpolateInPlace_Internal(dm));
9884054ae39SJames Wright     PetscCall(DMPlexSetBoxLabel_Internal(dm));
9894054ae39SJames Wright   }
9909566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&boundary));
9913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
992d6218766SMatthew G. Knepley }
993d6218766SMatthew G. Knepley 
994d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCubeMesh_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY, DMBoundaryType bdZ)
995d71ae5a4SJacob Faibussowitsch {
996ed0e4b50SMatthew G. Knepley   DMLabel     cutLabel  = NULL;
997f4eb4c5dSMatthew G. Knepley   PetscInt    markerTop = 1, faceMarkerTop = 1;
998f4eb4c5dSMatthew G. Knepley   PetscInt    markerBottom = 1, faceMarkerBottom = 1;
999f4eb4c5dSMatthew G. Knepley   PetscInt    markerFront = 1, faceMarkerFront = 1;
1000f4eb4c5dSMatthew G. Knepley   PetscInt    markerBack = 1, faceMarkerBack = 1;
1001f4eb4c5dSMatthew G. Knepley   PetscInt    markerRight = 1, faceMarkerRight = 1;
1002f4eb4c5dSMatthew G. Knepley   PetscInt    markerLeft = 1, faceMarkerLeft = 1;
10033dfda0b1SToby Isaac   PetscInt    dim;
1004d8211ee3SMatthew G. Knepley   PetscBool   markerSeparate = PETSC_FALSE, cutMarker = PETSC_FALSE;
10053dfda0b1SToby Isaac   PetscMPIInt rank;
10063dfda0b1SToby Isaac 
10073dfda0b1SToby Isaac   PetscFunctionBegin;
10089566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
10099566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
10109566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
10119566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
10129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
10139371c9d4SSatish 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) {
10149371c9d4SSatish Balay     if (cutMarker) {
10159371c9d4SSatish Balay       PetscCall(DMCreateLabel(dm, "periodic_cut"));
10169371c9d4SSatish Balay       PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
10179371c9d4SSatish Balay     }
1018d8211ee3SMatthew G. Knepley   }
10193dfda0b1SToby Isaac   switch (dim) {
10203dfda0b1SToby Isaac   case 2:
1021f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 3;
1022f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1023f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 2;
1024f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 4;
10253dfda0b1SToby Isaac     break;
10263dfda0b1SToby Isaac   case 3:
1027f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1028f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 2;
1029f4eb4c5dSMatthew G. Knepley     faceMarkerFront  = 3;
1030f4eb4c5dSMatthew G. Knepley     faceMarkerBack   = 4;
1031f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 5;
1032f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 6;
10333dfda0b1SToby Isaac     break;
1034d71ae5a4SJacob Faibussowitsch   default:
1035d71ae5a4SJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
10363dfda0b1SToby Isaac   }
10379566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
1038f4eb4c5dSMatthew G. Knepley   if (markerSeparate) {
1039f4eb4c5dSMatthew G. Knepley     markerBottom = faceMarkerBottom;
1040f4eb4c5dSMatthew G. Knepley     markerTop    = faceMarkerTop;
1041f4eb4c5dSMatthew G. Knepley     markerFront  = faceMarkerFront;
1042f4eb4c5dSMatthew G. Knepley     markerBack   = faceMarkerBack;
1043f4eb4c5dSMatthew G. Knepley     markerRight  = faceMarkerRight;
1044f4eb4c5dSMatthew G. Knepley     markerLeft   = faceMarkerLeft;
10453dfda0b1SToby Isaac   }
10463dfda0b1SToby Isaac   {
1047dd400576SPatrick Sanan     const PetscInt numXEdges    = rank == 0 ? edges[0] : 0;
1048dd400576SPatrick Sanan     const PetscInt numYEdges    = rank == 0 ? edges[1] : 0;
1049dd400576SPatrick Sanan     const PetscInt numZEdges    = rank == 0 ? edges[2] : 0;
1050dd400576SPatrick Sanan     const PetscInt numXVertices = rank == 0 ? (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST ? edges[0] : edges[0] + 1) : 0;
1051dd400576SPatrick Sanan     const PetscInt numYVertices = rank == 0 ? (bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[1] : edges[1] + 1) : 0;
1052dd400576SPatrick Sanan     const PetscInt numZVertices = rank == 0 ? (bdZ == DM_BOUNDARY_PERIODIC || bdZ == DM_BOUNDARY_TWIST ? edges[2] : edges[2] + 1) : 0;
10533dfda0b1SToby Isaac     const PetscInt numCells     = numXEdges * numYEdges * numZEdges;
10543dfda0b1SToby Isaac     const PetscInt numXFaces    = numYEdges * numZEdges;
10553dfda0b1SToby Isaac     const PetscInt numYFaces    = numXEdges * numZEdges;
10563dfda0b1SToby Isaac     const PetscInt numZFaces    = numXEdges * numYEdges;
10573dfda0b1SToby Isaac     const PetscInt numTotXFaces = numXVertices * numXFaces;
10583dfda0b1SToby Isaac     const PetscInt numTotYFaces = numYVertices * numYFaces;
10593dfda0b1SToby Isaac     const PetscInt numTotZFaces = numZVertices * numZFaces;
10603dfda0b1SToby Isaac     const PetscInt numFaces     = numTotXFaces + numTotYFaces + numTotZFaces;
10613dfda0b1SToby Isaac     const PetscInt numTotXEdges = numXEdges * numYVertices * numZVertices;
10623dfda0b1SToby Isaac     const PetscInt numTotYEdges = numYEdges * numXVertices * numZVertices;
10633dfda0b1SToby Isaac     const PetscInt numTotZEdges = numZEdges * numXVertices * numYVertices;
10643dfda0b1SToby Isaac     const PetscInt numVertices  = numXVertices * numYVertices * numZVertices;
10653dfda0b1SToby Isaac     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numTotZEdges;
10663dfda0b1SToby Isaac     const PetscInt firstVertex  = (dim == 2) ? numFaces : numCells;
10673dfda0b1SToby Isaac     const PetscInt firstXFace   = (dim == 2) ? 0 : numCells + numVertices;
10683dfda0b1SToby Isaac     const PetscInt firstYFace   = firstXFace + numTotXFaces;
10693dfda0b1SToby Isaac     const PetscInt firstZFace   = firstYFace + numTotYFaces;
10703dfda0b1SToby Isaac     const PetscInt firstXEdge   = numCells + numFaces + numVertices;
10713dfda0b1SToby Isaac     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
10723dfda0b1SToby Isaac     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
10733dfda0b1SToby Isaac     Vec            coordinates;
10743dfda0b1SToby Isaac     PetscSection   coordSection;
10753dfda0b1SToby Isaac     PetscScalar   *coords;
10763dfda0b1SToby Isaac     PetscInt       coordSize;
10773dfda0b1SToby Isaac     PetscInt       v, vx, vy, vz;
10783dfda0b1SToby Isaac     PetscInt       c, f, fx, fy, fz, e, ex, ey, ez;
10793dfda0b1SToby Isaac 
10809566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numFaces + numEdges + numVertices));
108148a46eb9SPierre Jolivet     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
108248a46eb9SPierre Jolivet     for (f = firstXFace; f < firstXFace + numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
108348a46eb9SPierre Jolivet     for (e = firstXEdge; e < firstXEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
10849566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
10853dfda0b1SToby Isaac     /* Build cells */
10863dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
10873dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
10883dfda0b1SToby Isaac         for (fx = 0; fx < numXEdges; ++fx) {
10893dfda0b1SToby Isaac           PetscInt cell  = (fz * numYEdges + fy) * numXEdges + fx;
10903dfda0b1SToby Isaac           PetscInt faceB = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
10913dfda0b1SToby Isaac           PetscInt faceT = firstZFace + (fy * numXEdges + fx) * numZVertices + ((fz + 1) % numZVertices);
10923dfda0b1SToby Isaac           PetscInt faceF = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
10933dfda0b1SToby Isaac           PetscInt faceK = firstYFace + (fz * numXEdges + fx) * numYVertices + ((fy + 1) % numYVertices);
10943dfda0b1SToby Isaac           PetscInt faceL = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
10953dfda0b1SToby Isaac           PetscInt faceR = firstXFace + (fz * numYEdges + fy) * numXVertices + ((fx + 1) % numXVertices);
10963dfda0b1SToby Isaac           /* B,  T,  F,  K,  R,  L */
1097b5a892a1SMatthew G. Knepley           PetscInt ornt[6] = {-2, 0, 0, -3, 0, -2}; /* ??? */
109842206facSLisandro Dalcin           PetscInt cone[6];
10993dfda0b1SToby Isaac 
11003dfda0b1SToby Isaac           /* no boundary twisting in 3D */
11019371c9d4SSatish Balay           cone[0] = faceB;
11029371c9d4SSatish Balay           cone[1] = faceT;
11039371c9d4SSatish Balay           cone[2] = faceF;
11049371c9d4SSatish Balay           cone[3] = faceK;
11059371c9d4SSatish Balay           cone[4] = faceR;
11069371c9d4SSatish Balay           cone[5] = faceL;
11079566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, cell, cone));
11089566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, cell, ornt));
11099566063dSJacob Faibussowitsch           if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11109566063dSJacob Faibussowitsch           if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11119566063dSJacob Faibussowitsch           if (bdZ != DM_BOUNDARY_NONE && fz == numZEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
11123dfda0b1SToby Isaac         }
11133dfda0b1SToby Isaac       }
11143dfda0b1SToby Isaac     }
11153dfda0b1SToby Isaac     /* Build x faces */
11163dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
11173dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
11183dfda0b1SToby Isaac         for (fx = 0; fx < numXVertices; ++fx) {
11193dfda0b1SToby Isaac           PetscInt face    = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
11203dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
11213dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (((fy + 1) % numYVertices) * numXVertices + fx) * numZEdges + fz;
11223dfda0b1SToby Isaac           PetscInt edgeB   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
11233dfda0b1SToby Isaac           PetscInt edgeT   = firstYEdge + (((fz + 1) % numZVertices) * numXVertices + fx) * numYEdges + fy;
1124b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11253dfda0b1SToby Isaac           PetscInt cone[4];
11263dfda0b1SToby Isaac 
11273dfda0b1SToby Isaac           if (dim == 3) {
11283dfda0b1SToby Isaac             /* markers */
11293dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
11303dfda0b1SToby Isaac               if (fx == numXVertices - 1) {
11319566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerRight));
11329566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerRight));
11339371c9d4SSatish Balay               } else if (fx == 0) {
11349566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerLeft));
11359566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerLeft));
11363dfda0b1SToby Isaac               }
11373dfda0b1SToby Isaac             }
11383dfda0b1SToby Isaac           }
11399371c9d4SSatish Balay           cone[0] = edgeB;
11409371c9d4SSatish Balay           cone[1] = edgeR;
11419371c9d4SSatish Balay           cone[2] = edgeT;
11429371c9d4SSatish Balay           cone[3] = edgeL;
11439566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
11449566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
11453dfda0b1SToby Isaac         }
11463dfda0b1SToby Isaac       }
11473dfda0b1SToby Isaac     }
11483dfda0b1SToby Isaac     /* Build y faces */
11493dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
115042206facSLisandro Dalcin       for (fx = 0; fx < numXEdges; ++fx) {
11513dfda0b1SToby Isaac         for (fy = 0; fy < numYVertices; ++fy) {
11523dfda0b1SToby Isaac           PetscInt face    = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
11533dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
11543dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (fy * numXVertices + ((fx + 1) % numXVertices)) * numZEdges + fz;
11553dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
11563dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (((fz + 1) % numZVertices) * numYVertices + fy) * numXEdges + fx;
1157b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11583dfda0b1SToby Isaac           PetscInt cone[4];
11593dfda0b1SToby Isaac 
11603dfda0b1SToby Isaac           if (dim == 3) {
11613dfda0b1SToby Isaac             /* markers */
11623dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
11633dfda0b1SToby Isaac               if (fy == numYVertices - 1) {
11649566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBack));
11659566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBack));
11669371c9d4SSatish Balay               } else if (fy == 0) {
11679566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerFront));
11689566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerFront));
11693dfda0b1SToby Isaac               }
11703dfda0b1SToby Isaac             }
11713dfda0b1SToby Isaac           }
11729371c9d4SSatish Balay           cone[0] = edgeB;
11739371c9d4SSatish Balay           cone[1] = edgeR;
11749371c9d4SSatish Balay           cone[2] = edgeT;
11759371c9d4SSatish Balay           cone[3] = edgeL;
11769566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
11779566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
11783dfda0b1SToby Isaac         }
11793dfda0b1SToby Isaac       }
11803dfda0b1SToby Isaac     }
11813dfda0b1SToby Isaac     /* Build z faces */
11823dfda0b1SToby Isaac     for (fy = 0; fy < numYEdges; ++fy) {
11833dfda0b1SToby Isaac       for (fx = 0; fx < numXEdges; ++fx) {
11843dfda0b1SToby Isaac         for (fz = 0; fz < numZVertices; fz++) {
11853dfda0b1SToby Isaac           PetscInt face    = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
11863dfda0b1SToby Isaac           PetscInt edgeL   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
11873dfda0b1SToby Isaac           PetscInt edgeR   = firstYEdge + (fz * numXVertices + ((fx + 1) % numXVertices)) * numYEdges + fy;
11883dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
11893dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (fz * numYVertices + ((fy + 1) % numYVertices)) * numXEdges + fx;
1190b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
11913dfda0b1SToby Isaac           PetscInt cone[4];
11923dfda0b1SToby Isaac 
11933dfda0b1SToby Isaac           if (dim == 2) {
11949371c9d4SSatish Balay             if (bdX == DM_BOUNDARY_TWIST && fx == numXEdges - 1) {
11959371c9d4SSatish Balay               edgeR += numYEdges - 1 - 2 * fy;
11969371c9d4SSatish Balay               ornt[1] = -1;
11979371c9d4SSatish Balay             }
11989371c9d4SSatish Balay             if (bdY == DM_BOUNDARY_TWIST && fy == numYEdges - 1) {
11999371c9d4SSatish Balay               edgeT += numXEdges - 1 - 2 * fx;
12009371c9d4SSatish Balay               ornt[2] = 0;
12019371c9d4SSatish Balay             }
12029566063dSJacob Faibussowitsch             if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
12039566063dSJacob Faibussowitsch             if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
1204d1c88043SMatthew G. Knepley           } else {
12053dfda0b1SToby Isaac             /* markers */
12063dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
12073dfda0b1SToby Isaac               if (fz == numZVertices - 1) {
12089566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerTop));
12099566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerTop));
12109371c9d4SSatish Balay               } else if (fz == 0) {
12119566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBottom));
12129566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBottom));
12133dfda0b1SToby Isaac               }
12143dfda0b1SToby Isaac             }
12153dfda0b1SToby Isaac           }
12169371c9d4SSatish Balay           cone[0] = edgeB;
12179371c9d4SSatish Balay           cone[1] = edgeR;
12189371c9d4SSatish Balay           cone[2] = edgeT;
12199371c9d4SSatish Balay           cone[3] = edgeL;
12209566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
12219566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
12223dfda0b1SToby Isaac         }
12233dfda0b1SToby Isaac       }
12243dfda0b1SToby Isaac     }
12253dfda0b1SToby Isaac     /* Build Z edges*/
12263dfda0b1SToby Isaac     for (vy = 0; vy < numYVertices; vy++) {
12273dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
12283dfda0b1SToby Isaac         for (ez = 0; ez < numZEdges; ez++) {
12293dfda0b1SToby Isaac           const PetscInt edge    = firstZEdge + (vy * numXVertices + vx) * numZEdges + ez;
12303dfda0b1SToby Isaac           const PetscInt vertexB = firstVertex + (ez * numYVertices + vy) * numXVertices + vx;
12313dfda0b1SToby Isaac           const PetscInt vertexT = firstVertex + (((ez + 1) % numZVertices) * numYVertices + vy) * numXVertices + vx;
12323dfda0b1SToby Isaac           PetscInt       cone[2];
12333dfda0b1SToby Isaac 
12349371c9d4SSatish Balay           cone[0] = vertexB;
12359371c9d4SSatish Balay           cone[1] = vertexT;
1236c2df9bbfSMatthew G. Knepley           PetscCall(DMPlexSetCone(dm, edge, cone));
12373dfda0b1SToby Isaac           if (dim == 3) {
12383dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
12393dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
12409566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1241c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1242c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1243c2df9bbfSMatthew G. Knepley               } else if (vx == 0) {
12449566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1245c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1246c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
12473dfda0b1SToby Isaac               }
12483dfda0b1SToby Isaac             }
12493dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
12503dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
12519566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1252c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1253c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1254c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
12559566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1256c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1257c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
12583dfda0b1SToby Isaac               }
12593dfda0b1SToby Isaac             }
12603dfda0b1SToby Isaac           }
12613dfda0b1SToby Isaac         }
12623dfda0b1SToby Isaac       }
12633dfda0b1SToby Isaac     }
12643dfda0b1SToby Isaac     /* Build Y edges*/
12653dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
12663dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
12673dfda0b1SToby Isaac         for (ey = 0; ey < numYEdges; ey++) {
12683dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdY == DM_BOUNDARY_TWIST && ey == numYEdges - 1) ? (numXVertices - vx - 1) : (vz * numYVertices + ((ey + 1) % numYVertices)) * numXVertices + vx;
12693dfda0b1SToby Isaac           const PetscInt edge    = firstYEdge + (vz * numXVertices + vx) * numYEdges + ey;
12703dfda0b1SToby Isaac           const PetscInt vertexF = firstVertex + (vz * numYVertices + ey) * numXVertices + vx;
12713dfda0b1SToby Isaac           const PetscInt vertexK = firstVertex + nextv;
12723dfda0b1SToby Isaac           PetscInt       cone[2];
12733dfda0b1SToby Isaac 
12749371c9d4SSatish Balay           cone[0] = vertexF;
12759371c9d4SSatish Balay           cone[1] = vertexK;
12769566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
12773dfda0b1SToby Isaac           if (dim == 2) {
12783dfda0b1SToby Isaac             if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
12793dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
12809566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight));
12819566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
12829566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1283c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1284d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
12859566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft));
12869566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
12879566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1288c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
12893dfda0b1SToby Isaac               }
1290d8211ee3SMatthew G. Knepley             } else {
12914c67ea77SStefano Zampini               if (vx == 0 && cutLabel) {
12929566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
12939566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1294c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
12953dfda0b1SToby Isaac               }
1296d8211ee3SMatthew G. Knepley             }
1297d8211ee3SMatthew G. Knepley           } else {
12983dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
12993dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
13009566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1301c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1302c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1303d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
13049566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1305c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1306c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
13073dfda0b1SToby Isaac               }
13083dfda0b1SToby Isaac             }
13093dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
13103dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
13119566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1312c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1313c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1314d8211ee3SMatthew G. Knepley               } else if (vz == 0) {
13159566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1316c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1317c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13183dfda0b1SToby Isaac               }
13193dfda0b1SToby Isaac             }
13203dfda0b1SToby Isaac           }
13213dfda0b1SToby Isaac         }
13223dfda0b1SToby Isaac       }
13233dfda0b1SToby Isaac     }
13243dfda0b1SToby Isaac     /* Build X edges*/
13253dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
13263dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; vy++) {
13273dfda0b1SToby Isaac         for (ex = 0; ex < numXEdges; ex++) {
13283dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdX == DM_BOUNDARY_TWIST && ex == numXEdges - 1) ? (numYVertices - vy - 1) * numXVertices : (vz * numYVertices + vy) * numXVertices + (ex + 1) % numXVertices;
13293dfda0b1SToby Isaac           const PetscInt edge    = firstXEdge + (vz * numYVertices + vy) * numXEdges + ex;
13303dfda0b1SToby Isaac           const PetscInt vertexL = firstVertex + (vz * numYVertices + vy) * numXVertices + ex;
13313dfda0b1SToby Isaac           const PetscInt vertexR = firstVertex + nextv;
13323dfda0b1SToby Isaac           PetscInt       cone[2];
13333dfda0b1SToby Isaac 
13349371c9d4SSatish Balay           cone[0] = vertexL;
13359371c9d4SSatish Balay           cone[1] = vertexR;
13369566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
13373dfda0b1SToby Isaac           if (dim == 2) {
13383dfda0b1SToby Isaac             if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
13393dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
13409566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop));
13419566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
13429566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1343c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1344d8211ee3SMatthew G. Knepley               } else if (vy == 0) {
13459566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom));
13469566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
13479566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1348c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13493dfda0b1SToby Isaac               }
1350d8211ee3SMatthew G. Knepley             } else {
13514c67ea77SStefano Zampini               if (vy == 0 && cutLabel) {
13529566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
13539566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1354c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
13553dfda0b1SToby Isaac               }
1356d8211ee3SMatthew G. Knepley             }
1357d8211ee3SMatthew G. Knepley           } else {
13583dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
13593dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
13609566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1361c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1362c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1363c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
13649566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1365c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1366c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
13673dfda0b1SToby Isaac               }
13683dfda0b1SToby Isaac             }
13693dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
13703dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
13719566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1372c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1373c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1374c2df9bbfSMatthew G. Knepley               } else if (vz == 0) {
13759566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1376c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1377c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
13783dfda0b1SToby Isaac               }
13793dfda0b1SToby Isaac             }
13803dfda0b1SToby Isaac           }
13813dfda0b1SToby Isaac         }
13823dfda0b1SToby Isaac       }
13833dfda0b1SToby Isaac     }
13849566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
13859566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
13863dfda0b1SToby Isaac     /* Build coordinates */
13879566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
13889566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
13899566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
13909566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVertices));
13913dfda0b1SToby Isaac     for (v = firstVertex; v < firstVertex + numVertices; ++v) {
13929566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
13939566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
13943dfda0b1SToby Isaac     }
13959566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
13969566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
13979566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
13989566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
13999566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
14009566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
14019566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
14029566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
14033dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; ++vz) {
14043dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; ++vy) {
14053dfda0b1SToby Isaac         for (vx = 0; vx < numXVertices; ++vx) {
14063dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * vx;
14073dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * vy;
1408ad540459SPierre Jolivet           if (dim == 3) coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 2] = lower[2] + ((upper[2] - lower[2]) / numZEdges) * vz;
14093dfda0b1SToby Isaac         }
14103dfda0b1SToby Isaac       }
14113dfda0b1SToby Isaac     }
14129566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
14139566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
14149566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
14153dfda0b1SToby Isaac   }
14163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14173dfda0b1SToby Isaac }
14183dfda0b1SToby Isaac 
1419d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Tensor_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1420d71ae5a4SJacob Faibussowitsch {
14219318fe57SMatthew G. Knepley   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
14229318fe57SMatthew G. Knepley   PetscInt       fac[3] = {0, 0, 0}, d;
1423552f7358SJed Brown 
1424552f7358SJed Brown   PetscFunctionBegin;
14254f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
14269318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
14279566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
14289371c9d4SSatish Balay   for (d = 0; d < dim; ++d) {
14299371c9d4SSatish Balay     fac[d] = faces[d];
14309371c9d4SSatish Balay     bdt[d] = periodicity[d];
14319371c9d4SSatish Balay   }
14329566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateCubeMesh_Internal(dm, lower, upper, fac, bdt[0], bdt[1], bdt[2]));
14339371c9d4SSatish 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))) {
14346858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
14356858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
1436552f7358SJed Brown 
14379318fe57SMatthew G. Knepley     for (d = 0; d < dim; ++d) {
14386858538eSMatthew G. Knepley       if (periodicity[d] != DM_BOUNDARY_NONE) {
14399318fe57SMatthew G. Knepley         L[d]       = upper[d] - lower[d];
14409318fe57SMatthew G. Knepley         maxCell[d] = 1.1 * (L[d] / PetscMax(1, faces[d]));
1441768d5fceSMatthew G. Knepley       }
14426858538eSMatthew G. Knepley     }
14434fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
1444768d5fceSMatthew G. Knepley   }
14459566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
14463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14479318fe57SMatthew G. Knepley }
14489318fe57SMatthew G. Knepley 
14495dca41c3SJed 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)
1450d71ae5a4SJacob Faibussowitsch {
14519318fe57SMatthew G. Knepley   PetscFunctionBegin;
145246139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
14535dca41c3SJed Brown   if (shape == DM_SHAPE_ZBOX) PetscCall(DMPlexCreateBoxMesh_Tensor_SFC_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
14546725e60dSJed Brown   else if (dim == 1) PetscCall(DMPlexCreateLineMesh_Internal(dm, faces[0], lower[0], upper[0], periodicity[0]));
14559566063dSJacob Faibussowitsch   else if (simplex) PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
14569566063dSJacob Faibussowitsch   else PetscCall(DMPlexCreateBoxMesh_Tensor_Internal(dm, dim, faces, lower, upper, periodicity));
14579318fe57SMatthew G. Knepley   if (!interpolate && dim > 1 && !simplex) {
1458768d5fceSMatthew G. Knepley     DM udm;
1459768d5fceSMatthew G. Knepley 
14609566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(dm, &udm));
14619566063dSJacob Faibussowitsch     PetscCall(DMPlexCopyCoordinates(dm, udm));
146269d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &udm));
1463768d5fceSMatthew G. Knepley   }
146446139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
14653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1466c8c68bd8SToby Isaac }
1467c8c68bd8SToby Isaac 
14685d83a8b1SBarry Smith /*@
1469768d5fceSMatthew G. Knepley   DMPlexCreateBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using simplices or tensor cells (hexahedra).
1470768d5fceSMatthew G. Knepley 
1471d083f849SBarry Smith   Collective
1472768d5fceSMatthew G. Knepley 
1473768d5fceSMatthew G. Knepley   Input Parameters:
1474a1cb98faSBarry Smith + comm               - The communicator for the `DM` object
1475768d5fceSMatthew G. Knepley . dim                - The spatial dimension
1476a1cb98faSBarry Smith . simplex            - `PETSC_TRUE` for simplices, `PETSC_FALSE` for tensor cells
147720f4b53cSBarry Smith . faces              - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
147820f4b53cSBarry Smith . lower              - The lower left corner, or `NULL` for (0, 0, 0)
147920f4b53cSBarry Smith . upper              - The upper right corner, or `NULL` for (1, 1, 1)
148020f4b53cSBarry Smith . periodicity        - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
148142108689Sksagiyam . interpolate        - Flag to create intermediate mesh pieces (edges, faces)
148242108689Sksagiyam . localizationHeight - Flag to localize edges and faces in addition to cells; only significant for periodic meshes
148342108689Sksagiyam - sparseLocalize     - Flag to localize coordinates only for cells near the periodic boundary; only significant for periodic meshes
1484768d5fceSMatthew G. Knepley 
1485768d5fceSMatthew G. Knepley   Output Parameter:
1486a1cb98faSBarry Smith . dm - The `DM` object
1487768d5fceSMatthew G. Knepley 
1488768d5fceSMatthew G. Knepley   Level: beginner
1489768d5fceSMatthew G. Knepley 
1490a1cb98faSBarry Smith   Note:
1491a1cb98faSBarry Smith   To customize this mesh using options, use
1492a1cb98faSBarry Smith .vb
1493a1cb98faSBarry Smith   DMCreate(comm, &dm);
1494a1cb98faSBarry Smith   DMSetType(dm, DMPLEX);
1495a1cb98faSBarry Smith   DMSetFromOptions(dm);
1496a1cb98faSBarry Smith .ve
1497a1cb98faSBarry Smith   and use the options in `DMSetFromOptions()`.
1498a1cb98faSBarry Smith 
1499a4e35b19SJacob Faibussowitsch   Here is the numbering returned for 2 faces in each direction for tensor cells\:
1500a1cb98faSBarry Smith .vb
1501a1cb98faSBarry Smith  10---17---11---18----12
1502a1cb98faSBarry Smith   |         |         |
1503a1cb98faSBarry Smith   |         |         |
1504a1cb98faSBarry Smith  20    2   22    3    24
1505a1cb98faSBarry Smith   |         |         |
1506a1cb98faSBarry Smith   |         |         |
1507a1cb98faSBarry Smith   7---15----8---16----9
1508a1cb98faSBarry Smith   |         |         |
1509a1cb98faSBarry Smith   |         |         |
1510a1cb98faSBarry Smith  19    0   21    1   23
1511a1cb98faSBarry Smith   |         |         |
1512a1cb98faSBarry Smith   |         |         |
1513a1cb98faSBarry Smith   4---13----5---14----6
1514a1cb98faSBarry Smith .ve
1515a1cb98faSBarry Smith   and for simplicial cells
1516a1cb98faSBarry Smith .vb
1517a1cb98faSBarry Smith  14----8---15----9----16
1518a1cb98faSBarry Smith   |\     5  |\      7 |
1519a1cb98faSBarry Smith   | \       | \       |
1520a1cb98faSBarry Smith  13   2    14    3    15
1521a1cb98faSBarry Smith   | 4   \   | 6   \   |
1522a1cb98faSBarry Smith   |       \ |       \ |
1523a1cb98faSBarry Smith  11----6---12----7----13
1524a1cb98faSBarry Smith   |\        |\        |
1525a1cb98faSBarry Smith   | \    1  | \     3 |
1526a1cb98faSBarry Smith  10   0    11    1    12
1527a1cb98faSBarry Smith   | 0   \   | 2   \   |
1528a1cb98faSBarry Smith   |       \ |       \ |
1529a1cb98faSBarry Smith   8----4----9----5----10
1530a1cb98faSBarry Smith .ve
1531a1cb98faSBarry Smith 
15321cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
1533768d5fceSMatthew G. Knepley @*/
153442108689Sksagiyam 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)
1535d71ae5a4SJacob Faibussowitsch {
15369318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
1537fdbf62faSLisandro Dalcin   PetscReal      low[3] = {0, 0, 0};
1538fdbf62faSLisandro Dalcin   PetscReal      upp[3] = {1, 1, 1};
1539fdbf62faSLisandro Dalcin   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
1540552f7358SJed Brown 
1541768d5fceSMatthew G. Knepley   PetscFunctionBegin;
15429566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
15439566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
15445dca41c3SJed Brown   PetscCall(DMPlexCreateBoxMesh_Internal(*dm, DM_SHAPE_BOX, dim, simplex, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt, interpolate));
154542108689Sksagiyam   if (periodicity) {
154642108689Sksagiyam     DM cdm;
154742108689Sksagiyam 
154842108689Sksagiyam     PetscCall(DMGetCoordinateDM(*dm, &cdm));
154942108689Sksagiyam     PetscCall(DMPlexSetMaxProjectionHeight(cdm, localizationHeight));
155042108689Sksagiyam     PetscCall(DMSetSparseLocalize(*dm, sparseLocalize));
155142108689Sksagiyam     PetscCall(DMLocalizeCoordinates(*dm));
155242108689Sksagiyam   }
15533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15549318fe57SMatthew G. Knepley }
1555fdbf62faSLisandro Dalcin 
1556d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeBoxMesh_Internal(DM dm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1557d71ae5a4SJacob Faibussowitsch {
15589318fe57SMatthew G. Knepley   DM       bdm, vol;
15599318fe57SMatthew G. Knepley   PetscInt i;
15609318fe57SMatthew G. Knepley 
15619318fe57SMatthew G. Knepley   PetscFunctionBegin;
15621fcf445aSMatthew G. Knepley   // TODO Now we can support periodicity
156308401ef6SPierre Jolivet   for (i = 0; i < 3; ++i) PetscCheck(periodicity[i] == DM_BOUNDARY_NONE, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Periodicity not yet supported");
15649566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &bdm));
15659566063dSJacob Faibussowitsch   PetscCall(DMSetType(bdm, DMPLEX));
15669566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(bdm, 2));
156746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, bdm, 0, 0, 0));
15689566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(bdm, 2, faces, lower, upper, periodicity, PETSC_TRUE));
15691fcf445aSMatthew G. Knepley   PetscCall(DMPlexExtrude(bdm, faces[2], upper[2] - lower[2], PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, NULL, NULL, &vol));
157046139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, bdm, 0, 0, 0));
15719566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&bdm));
157269d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
15739318fe57SMatthew G. Knepley   if (lower[2] != 0.0) {
15749318fe57SMatthew G. Knepley     Vec          v;
15759318fe57SMatthew G. Knepley     PetscScalar *x;
15769318fe57SMatthew G. Knepley     PetscInt     cDim, n;
15779318fe57SMatthew G. Knepley 
15789566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &v));
15799566063dSJacob Faibussowitsch     PetscCall(VecGetBlockSize(v, &cDim));
15809566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(v, &n));
15819566063dSJacob Faibussowitsch     PetscCall(VecGetArray(v, &x));
15829318fe57SMatthew G. Knepley     x += cDim;
15839318fe57SMatthew G. Knepley     for (i = 0; i < n; i += cDim) x[i] += lower[2];
15849566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(v, &x));
15859566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, v));
15869318fe57SMatthew G. Knepley   }
15873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1588552f7358SJed Brown }
1589552f7358SJed Brown 
159000dabe28SStefano Zampini /*@
159139f4f5dbSPierre Jolivet   DMPlexCreateWedgeBoxMesh - Creates a 3-D mesh tessellating the (x,y) plane and extruding in the third direction using wedge cells.
159200dabe28SStefano Zampini 
1593d083f849SBarry Smith   Collective
159400dabe28SStefano Zampini 
159500dabe28SStefano Zampini   Input Parameters:
1596a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
159720f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1, 1, 1)
159820f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
159920f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
160020f4b53cSBarry Smith . periodicity - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
1601a1cb98faSBarry Smith . orderHeight - If `PETSC_TRUE`, orders the extruded cells in the height first. Otherwise, orders the cell on the layers first
160200dabe28SStefano Zampini - interpolate - Flag to create intermediate mesh pieces (edges, faces)
160300dabe28SStefano Zampini 
160400dabe28SStefano Zampini   Output Parameter:
1605a1cb98faSBarry Smith . dm - The `DM` object
160600dabe28SStefano Zampini 
160700dabe28SStefano Zampini   Level: beginner
160800dabe28SStefano Zampini 
16091cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateWedgeCylinderMesh()`, `DMExtrude()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
161000dabe28SStefano Zampini @*/
1611d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeBoxMesh(MPI_Comm comm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool orderHeight, PetscBool interpolate, DM *dm)
1612d71ae5a4SJacob Faibussowitsch {
16139318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
161400dabe28SStefano Zampini   PetscReal      low[3] = {0, 0, 0};
161500dabe28SStefano Zampini   PetscReal      upp[3] = {1, 1, 1};
161600dabe28SStefano Zampini   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
161700dabe28SStefano Zampini 
161800dabe28SStefano Zampini   PetscFunctionBegin;
16199566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
16209566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
16219566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeBoxMesh_Internal(*dm, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt));
1622d410b0cfSMatthew G. Knepley   if (!interpolate) {
1623d410b0cfSMatthew G. Knepley     DM udm;
162400dabe28SStefano Zampini 
16259566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(*dm, &udm));
162669d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(*dm, &udm));
162700dabe28SStefano Zampini   }
16287ff04441SMatthew G. Knepley   if (periodicity) PetscCall(DMLocalizeCoordinates(*dm));
16293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
163000dabe28SStefano Zampini }
163100dabe28SStefano Zampini 
1632cfb853baSMatthew G. Knepley /*
1633cfb853baSMatthew 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.
1634cfb853baSMatthew G. Knepley 
1635cfb853baSMatthew G. Knepley   Input Parameters:
1636cfb853baSMatthew G. Knepley + len - The length of the tuple
1637cfb853baSMatthew G. Knepley . max - The maximum for each dimension, so values are in [0, max)
1638cfb853baSMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition
1639cfb853baSMatthew G. Knepley 
1640cfb853baSMatthew G. Knepley   Output Parameter:
164120f4b53cSBarry Smith . tup - A tuple of `len` integers whose entries are at most `max`
1642cfb853baSMatthew G. Knepley 
1643cfb853baSMatthew G. Knepley   Level: developer
1644cfb853baSMatthew G. Knepley 
164520f4b53cSBarry Smith   Note:
164620f4b53cSBarry Smith   Ordering is lexicographic with lowest index as least significant in ordering.
164720f4b53cSBarry 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}.
164820f4b53cSBarry Smith 
1649cfb853baSMatthew G. Knepley .seealso: PetscDualSpaceTensorPointLexicographic_Internal(), PetscDualSpaceLatticePointLexicographic_Internal()
1650cfb853baSMatthew G. Knepley */
1651cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexTensorPointLexicographic_Private(PetscInt len, const PetscInt max[], PetscInt tup[])
1652cfb853baSMatthew G. Knepley {
1653cfb853baSMatthew G. Knepley   PetscInt i;
1654cfb853baSMatthew G. Knepley 
1655cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1656cfb853baSMatthew G. Knepley   for (i = 0; i < len; ++i) {
1657cfb853baSMatthew G. Knepley     if (tup[i] < max[i] - 1) {
1658cfb853baSMatthew G. Knepley       break;
1659cfb853baSMatthew G. Knepley     } else {
1660cfb853baSMatthew G. Knepley       tup[i] = 0;
1661cfb853baSMatthew G. Knepley     }
1662cfb853baSMatthew G. Knepley   }
1663cfb853baSMatthew G. Knepley   if (i == len) tup[i - 1] = max[i - 1];
1664cfb853baSMatthew G. Knepley   else ++tup[i];
16653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1666cfb853baSMatthew G. Knepley }
1667cfb853baSMatthew G. Knepley 
1668cfb853baSMatthew G. Knepley static PetscInt TupleToIndex_Private(PetscInt len, const PetscInt max[], const PetscInt tup[])
1669cfb853baSMatthew G. Knepley {
1670cfb853baSMatthew G. Knepley   PetscInt i, idx = tup[len - 1];
1671cfb853baSMatthew G. Knepley 
1672cfb853baSMatthew G. Knepley   for (i = len - 2; i >= 0; --i) {
1673cfb853baSMatthew G. Knepley     idx *= max[i];
1674cfb853baSMatthew G. Knepley     idx += tup[i];
1675cfb853baSMatthew G. Knepley   }
1676cfb853baSMatthew G. Knepley   return idx;
1677cfb853baSMatthew G. Knepley }
1678cfb853baSMatthew G. Knepley 
1679cfb853baSMatthew G. Knepley static PetscErrorCode DestroyExtent_Private(void *extent)
1680cfb853baSMatthew G. Knepley {
1681cfb853baSMatthew G. Knepley   return PetscFree(extent);
1682cfb853baSMatthew G. Knepley }
1683cfb853baSMatthew G. Knepley 
1684cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexCreateHypercubicMesh_Internal(DM dm, PetscInt dim, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], const DMBoundaryType bd[])
1685cfb853baSMatthew G. Knepley {
1686cfb853baSMatthew G. Knepley   Vec          coordinates;
1687cfb853baSMatthew G. Knepley   PetscSection coordSection;
1688cfb853baSMatthew G. Knepley   DMLabel      cutLabel    = NULL;
1689cfb853baSMatthew G. Knepley   PetscBool    cutMarker   = PETSC_FALSE;
1690cfb853baSMatthew G. Knepley   PetscBool    periodic    = PETSC_FALSE;
1691cfb853baSMatthew G. Knepley   PetscInt     numCells    = 1, c;
1692cfb853baSMatthew G. Knepley   PetscInt     numVertices = 1, v;
1693cfb853baSMatthew G. Knepley   PetscScalar *coords;
1694cfb853baSMatthew G. Knepley   PetscInt    *vertices, *vert, *vtmp, *supp, cone[2];
1695cfb853baSMatthew G. Knepley   PetscInt     d, e, cell = 0, coordSize;
1696cfb853baSMatthew G. Knepley   PetscMPIInt  rank;
1697cfb853baSMatthew G. Knepley 
1698cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1699cfb853baSMatthew G. Knepley   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1700cfb853baSMatthew G. Knepley   PetscCall(DMSetDimension(dm, dim));
1701cfb853baSMatthew G. Knepley   PetscCall(PetscCalloc4(dim, &vertices, dim, &vert, dim, &vtmp, 2 * dim, &supp));
1702cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
1703cfb853baSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
1704cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) periodic = (periodic || bd[d] == DM_BOUNDARY_PERIODIC) ? PETSC_TRUE : PETSC_FALSE;
1705cfb853baSMatthew G. Knepley   if (periodic && cutMarker) {
1706cfb853baSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, "periodic_cut"));
1707cfb853baSMatthew G. Knepley     PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
1708cfb853baSMatthew G. Knepley   }
1709cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) PetscCheck(bd[d] == DM_BOUNDARY_PERIODIC, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Hypercubic mesh must be periodic now");
1710cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
1711cfb853baSMatthew G. Knepley     vertices[d] = edges[d];
1712cfb853baSMatthew G. Knepley     numVertices *= vertices[d];
1713cfb853baSMatthew G. Knepley   }
1714cfb853baSMatthew G. Knepley   numCells = numVertices * dim;
1715cfb853baSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
1716cfb853baSMatthew G. Knepley   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, 2));
1717cfb853baSMatthew G. Knepley   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetSupportSize(dm, v, 2 * dim));
1718cfb853baSMatthew G. Knepley   /* TODO Loop over boundary and reset support sizes */
1719cfb853baSMatthew G. Knepley   PetscCall(DMSetUp(dm)); /* Allocate space for cones and supports */
1720cfb853baSMatthew G. Knepley   /* Build cell cones and vertex supports */
1721cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "celltype"));
1722cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
1723cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert) + numCells;
1724cfb853baSMatthew G. Knepley     PetscInt       s      = 0;
1725cfb853baSMatthew G. Knepley 
17263ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "Vertex %" PetscInt_FMT ":", vertex));
17273ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vert[d]));
17283ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1729cfb853baSMatthew G. Knepley     PetscCall(DMPlexSetCellType(dm, vertex, DM_POLYTOPE_POINT));
1730cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) {
1731cfb853baSMatthew G. Knepley       for (e = 0; e < dim; ++e) vtmp[e] = vert[e];
1732cfb853baSMatthew G. Knepley       vtmp[d] = (vert[d] + 1) % vertices[d];
1733cfb853baSMatthew G. Knepley       cone[0] = vertex;
1734cfb853baSMatthew G. Knepley       cone[1] = TupleToIndex_Private(dim, vertices, vtmp) + numCells;
17353ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Vertex %" PetscInt_FMT ":", cone[1]));
17363ba16761SJacob Faibussowitsch       for (e = 0; e < dim; ++e) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vtmp[e]));
17373ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1738cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, cell, cone));
1739cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCellType(dm, cell, DM_POLYTOPE_SEGMENT));
17403ba16761SJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Edge %" PetscInt_FMT " (%" PetscInt_FMT " %" PetscInt_FMT ")\n", cell, cone[0], cone[1]));
1741cfb853baSMatthew G. Knepley       ++cell;
1742cfb853baSMatthew G. Knepley     }
1743cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) {
1744cfb853baSMatthew G. Knepley       for (e = 0; e < dim; ++e) vtmp[e] = vert[e];
1745cfb853baSMatthew G. Knepley       vtmp[d]   = (vert[d] + vertices[d] - 1) % vertices[d];
1746cfb853baSMatthew G. Knepley       supp[s++] = TupleToIndex_Private(dim, vertices, vtmp) * dim + d;
1747cfb853baSMatthew G. Knepley       supp[s++] = (vertex - numCells) * dim + d;
1748cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetSupport(dm, vertex, supp));
1749cfb853baSMatthew G. Knepley     }
1750cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
1751cfb853baSMatthew G. Knepley   }
1752cfb853baSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
1753cfb853baSMatthew G. Knepley   /* Build coordinates */
1754cfb853baSMatthew G. Knepley   PetscCall(DMGetCoordinateSection(dm, &coordSection));
1755cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetNumFields(coordSection, 1));
1756cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
1757cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
1758cfb853baSMatthew G. Knepley   for (v = numCells; v < numCells + numVertices; ++v) {
1759cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetDof(coordSection, v, dim));
1760cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
1761cfb853baSMatthew G. Knepley   }
1762cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetUp(coordSection));
1763cfb853baSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
1764cfb853baSMatthew G. Knepley   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
1765cfb853baSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
1766cfb853baSMatthew G. Knepley   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
1767cfb853baSMatthew G. Knepley   PetscCall(VecSetBlockSize(coordinates, dim));
1768cfb853baSMatthew G. Knepley   PetscCall(VecSetType(coordinates, VECSTANDARD));
1769cfb853baSMatthew G. Knepley   PetscCall(VecGetArray(coordinates, &coords));
1770cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) vert[d] = 0;
1771cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
1772cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert);
1773cfb853baSMatthew G. Knepley 
1774cfb853baSMatthew G. Knepley     for (d = 0; d < dim; ++d) coords[vertex * dim + d] = lower[d] + ((upper[d] - lower[d]) / vertices[d]) * vert[d];
17753ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "Vertex %" PetscInt_FMT ":", vertex));
17763ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, vert[d]));
17773ba16761SJacob Faibussowitsch     for (d = 0; d < dim; ++d) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %g", (double)PetscRealPart(coords[vertex * dim + d])));
17783ba16761SJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
1779cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
1780cfb853baSMatthew G. Knepley   }
1781cfb853baSMatthew G. Knepley   PetscCall(VecRestoreArray(coordinates, &coords));
1782cfb853baSMatthew G. Knepley   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
1783cfb853baSMatthew G. Knepley   PetscCall(VecDestroy(&coordinates));
1784cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(vertices, vert, vtmp, supp));
1785cfb853baSMatthew G. Knepley   //PetscCall(DMSetPeriodicity(dm, NULL, lower, upper));
1786cfb853baSMatthew G. Knepley   // Attach the extent
1787cfb853baSMatthew G. Knepley   {
1788cfb853baSMatthew G. Knepley     PetscContainer c;
1789cfb853baSMatthew G. Knepley     PetscInt      *extent;
1790cfb853baSMatthew G. Knepley 
1791cfb853baSMatthew G. Knepley     PetscCall(PetscMalloc1(dim, &extent));
1792cfb853baSMatthew G. Knepley     for (PetscInt d = 0; d < dim; ++d) extent[d] = edges[d];
1793cfb853baSMatthew G. Knepley     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
1794cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetUserDestroy(c, DestroyExtent_Private));
1795cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetPointer(c, extent));
1796cfb853baSMatthew G. Knepley     PetscCall(PetscObjectCompose((PetscObject)dm, "_extent", (PetscObject)c));
1797cfb853baSMatthew G. Knepley     PetscCall(PetscContainerDestroy(&c));
1798cfb853baSMatthew G. Knepley   }
17993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1800cfb853baSMatthew G. Knepley }
1801cfb853baSMatthew G. Knepley 
1802cfb853baSMatthew G. Knepley /*@C
1803aaa8cc7dSPierre Jolivet   DMPlexCreateHypercubicMesh - Creates a periodic mesh on the tensor product of unit intervals using only vertices and edges.
1804cfb853baSMatthew G. Knepley 
1805cfb853baSMatthew G. Knepley   Collective
1806cfb853baSMatthew G. Knepley 
1807cfb853baSMatthew G. Knepley   Input Parameters:
1808cfb853baSMatthew G. Knepley + comm  - The communicator for the DM object
1809cfb853baSMatthew G. Knepley . dim   - The spatial dimension
181020f4b53cSBarry Smith . edges - Number of edges per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
181120f4b53cSBarry Smith . lower - The lower left corner, or `NULL` for (0, 0, 0)
181220f4b53cSBarry Smith - upper - The upper right corner, or `NULL` for (1, 1, 1)
1813cfb853baSMatthew G. Knepley 
1814cfb853baSMatthew G. Knepley   Output Parameter:
1815cfb853baSMatthew G. Knepley . dm - The DM object
1816cfb853baSMatthew G. Knepley 
181720f4b53cSBarry Smith   Level: beginner
181820f4b53cSBarry Smith 
181920f4b53cSBarry Smith   Note:
182020f4b53cSBarry Smith   If you want to customize this mesh using options, you just need to
182120f4b53cSBarry Smith .vb
182220f4b53cSBarry Smith   DMCreate(comm, &dm);
182320f4b53cSBarry Smith   DMSetType(dm, DMPLEX);
182420f4b53cSBarry Smith   DMSetFromOptions(dm);
182520f4b53cSBarry Smith .ve
182620f4b53cSBarry Smith   and use the options on the `DMSetFromOptions()` page.
1827cfb853baSMatthew G. Knepley 
1828cfb853baSMatthew G. Knepley   The vertices are numbered is lexicographic order, and the dim edges exiting a vertex in the positive orthant are number consecutively,
182920f4b53cSBarry Smith .vb
183020f4b53cSBarry Smith  18--0-19--2-20--4-18
183120f4b53cSBarry Smith   |     |     |     |
183220f4b53cSBarry Smith  13    15    17    13
183320f4b53cSBarry Smith   |     |     |     |
183420f4b53cSBarry Smith  24-12-25-14-26-16-24
183520f4b53cSBarry Smith   |     |     |     |
183620f4b53cSBarry Smith   7     9    11     7
183720f4b53cSBarry Smith   |     |     |     |
183820f4b53cSBarry Smith  21--6-22--8-23-10-21
183920f4b53cSBarry Smith   |     |     |     |
184020f4b53cSBarry Smith   1     3     5     1
184120f4b53cSBarry Smith   |     |     |     |
184220f4b53cSBarry Smith  18--0-19--2-20--4-18
184320f4b53cSBarry Smith .ve
1844cfb853baSMatthew G. Knepley 
184576fbde31SPierre Jolivet .seealso: `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
1846cfb853baSMatthew G. Knepley @*/
1847cfb853baSMatthew G. Knepley PetscErrorCode DMPlexCreateHypercubicMesh(MPI_Comm comm, PetscInt dim, const PetscInt edges[], const PetscReal lower[], const PetscReal upper[], DM *dm)
1848cfb853baSMatthew G. Knepley {
1849cfb853baSMatthew G. Knepley   PetscInt       *edg;
1850cfb853baSMatthew G. Knepley   PetscReal      *low, *upp;
1851cfb853baSMatthew G. Knepley   DMBoundaryType *bdt;
1852cfb853baSMatthew G. Knepley   PetscInt        d;
1853cfb853baSMatthew G. Knepley 
1854cfb853baSMatthew G. Knepley   PetscFunctionBegin;
1855cfb853baSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
1856cfb853baSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
1857cfb853baSMatthew G. Knepley   PetscCall(PetscMalloc4(dim, &edg, dim, &low, dim, &upp, dim, &bdt));
1858cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
1859cfb853baSMatthew G. Knepley     edg[d] = edges ? edges[d] : 1;
1860cfb853baSMatthew G. Knepley     low[d] = lower ? lower[d] : 0.;
1861cfb853baSMatthew G. Knepley     upp[d] = upper ? upper[d] : 1.;
1862cfb853baSMatthew G. Knepley     bdt[d] = DM_BOUNDARY_PERIODIC;
1863cfb853baSMatthew G. Knepley   }
1864cfb853baSMatthew G. Knepley   PetscCall(DMPlexCreateHypercubicMesh_Internal(*dm, dim, low, upp, edg, bdt));
1865cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(edg, low, upp, bdt));
18663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1867cfb853baSMatthew G. Knepley }
1868cfb853baSMatthew G. Knepley 
1869cc4c1da9SBarry Smith /*@
1870a1cb98faSBarry Smith   DMPlexSetOptionsPrefix - Sets the prefix used for searching for all `DM` options in the database.
1871a9074c1eSMatthew G. Knepley 
187220f4b53cSBarry Smith   Logically Collective
1873a9074c1eSMatthew G. Knepley 
1874a9074c1eSMatthew G. Knepley   Input Parameters:
187520f4b53cSBarry Smith + dm     - the `DM` context
1876a9074c1eSMatthew G. Knepley - prefix - the prefix to prepend to all option names
1877a9074c1eSMatthew G. Knepley 
1878a1cb98faSBarry Smith   Level: advanced
1879a1cb98faSBarry Smith 
1880a1cb98faSBarry Smith   Note:
1881a9074c1eSMatthew G. Knepley   A hyphen (-) must NOT be given at the beginning of the prefix name.
1882a9074c1eSMatthew G. Knepley   The first character of all runtime options is AUTOMATICALLY the hyphen.
1883a9074c1eSMatthew G. Knepley 
18841cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `SNESSetFromOptions()`
1885a9074c1eSMatthew G. Knepley @*/
1886d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexSetOptionsPrefix(DM dm, const char prefix[])
1887d71ae5a4SJacob Faibussowitsch {
1888a9074c1eSMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
1889a9074c1eSMatthew G. Knepley 
1890a9074c1eSMatthew G. Knepley   PetscFunctionBegin;
1891a9074c1eSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
18929566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, prefix));
18939566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)mesh->partitioner, prefix));
18943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1895a9074c1eSMatthew G. Knepley }
1896a9074c1eSMatthew G. Knepley 
18979318fe57SMatthew G. Knepley /* Remap geometry to cylinder
189861a622f3SMatthew G. Knepley    TODO: This only works for a single refinement, then it is broken
189961a622f3SMatthew G. Knepley 
19009318fe57SMatthew G. Knepley      Interior square: Linear interpolation is correct
19019318fe57SMatthew G. Knepley      The other cells all have vertices on rays from the origin. We want to uniformly expand the spacing
19029318fe57SMatthew G. Knepley      such that the last vertex is on the unit circle. So the closest and farthest vertices are at distance
19030510c589SMatthew G. Knepley 
19049318fe57SMatthew G. Knepley        phi     = arctan(y/x)
19059318fe57SMatthew G. Knepley        d_close = sqrt(1/8 + 1/4 sin^2(phi))
19069318fe57SMatthew G. Knepley        d_far   = sqrt(1/2 + sin^2(phi))
19070510c589SMatthew G. Knepley 
19089318fe57SMatthew G. Knepley      so we remap them using
19090510c589SMatthew G. Knepley 
19109318fe57SMatthew G. Knepley        x_new = x_close + (x - x_close) (1 - d_close) / (d_far - d_close)
19119318fe57SMatthew G. Knepley        y_new = y_close + (y - y_close) (1 - d_close) / (d_far - d_close)
19120510c589SMatthew G. Knepley 
19139318fe57SMatthew G. Knepley      If pi/4 < phi < 3pi/4 or -3pi/4 < phi < -pi/4, then we switch x and y.
19149318fe57SMatthew G. Knepley */
1915d71ae5a4SJacob 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[])
1916d71ae5a4SJacob Faibussowitsch {
19179318fe57SMatthew G. Knepley   const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
19189318fe57SMatthew G. Knepley   const PetscReal ds2 = 0.5 * dis;
191922cc497dSMatthew G. Knepley 
19209318fe57SMatthew G. Knepley   if ((PetscAbsScalar(u[0]) <= ds2) && (PetscAbsScalar(u[1]) <= ds2)) {
19219318fe57SMatthew G. Knepley     f0[0] = u[0];
19229318fe57SMatthew G. Knepley     f0[1] = u[1];
19239318fe57SMatthew G. Knepley   } else {
19249318fe57SMatthew G. Knepley     PetscReal phi, sinp, cosp, dc, df, x, y, xc, yc;
19250510c589SMatthew G. Knepley 
19269318fe57SMatthew G. Knepley     x    = PetscRealPart(u[0]);
19279318fe57SMatthew G. Knepley     y    = PetscRealPart(u[1]);
19289318fe57SMatthew G. Knepley     phi  = PetscAtan2Real(y, x);
19299318fe57SMatthew G. Knepley     sinp = PetscSinReal(phi);
19309318fe57SMatthew G. Knepley     cosp = PetscCosReal(phi);
19319318fe57SMatthew G. Knepley     if ((PetscAbsReal(phi) > PETSC_PI / 4.0) && (PetscAbsReal(phi) < 3.0 * PETSC_PI / 4.0)) {
19329318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / sinp);
19339318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / sinp);
19349318fe57SMatthew G. Knepley       xc = ds2 * x / PetscAbsReal(y);
19359318fe57SMatthew G. Knepley       yc = ds2 * PetscSignReal(y);
19369318fe57SMatthew G. Knepley     } else {
19379318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / cosp);
19389318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / cosp);
19399318fe57SMatthew G. Knepley       xc = ds2 * PetscSignReal(x);
19409318fe57SMatthew G. Knepley       yc = ds2 * y / PetscAbsReal(x);
19419318fe57SMatthew G. Knepley     }
19429318fe57SMatthew G. Knepley     f0[0] = xc + (u[0] - xc) * (1.0 - dc) / (df - dc);
19439318fe57SMatthew G. Knepley     f0[1] = yc + (u[1] - yc) * (1.0 - dc) / (df - dc);
19449318fe57SMatthew G. Knepley   }
19459318fe57SMatthew G. Knepley   f0[2] = u[2];
19469318fe57SMatthew G. Knepley }
19470510c589SMatthew G. Knepley 
194849704ca5SMatthew G. Knepley static PetscErrorCode DMPlexCreateHexCylinderMesh_Internal(DM dm, DMBoundaryType periodicZ, PetscInt Nr)
1949d71ae5a4SJacob Faibussowitsch {
19500510c589SMatthew G. Knepley   const PetscInt dim = 3;
19519318fe57SMatthew G. Knepley   PetscInt       numCells, numVertices;
1952d8c47e87SMatthew G. Knepley   PetscMPIInt    rank;
19530510c589SMatthew G. Knepley 
19540510c589SMatthew G. Knepley   PetscFunctionBegin;
195546139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
19569566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
19579566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
19580510c589SMatthew G. Knepley   /* Create topology */
19590510c589SMatthew G. Knepley   {
19600510c589SMatthew G. Knepley     PetscInt cone[8], c;
19610510c589SMatthew G. Knepley 
1962dd400576SPatrick Sanan     numCells    = rank == 0 ? 5 : 0;
1963dd400576SPatrick Sanan     numVertices = rank == 0 ? 16 : 0;
1964006a8963SMatthew G. Knepley     if (periodicZ == DM_BOUNDARY_PERIODIC) {
1965ae8bcbbbSMatthew G. Knepley       numCells *= 3;
1966dd400576SPatrick Sanan       numVertices = rank == 0 ? 24 : 0;
1967006a8963SMatthew G. Knepley     }
19689566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
19699566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 8));
19709566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
1971dd400576SPatrick Sanan     if (rank == 0) {
1972006a8963SMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
19739371c9d4SSatish Balay         cone[0] = 15;
19749371c9d4SSatish Balay         cone[1] = 18;
19759371c9d4SSatish Balay         cone[2] = 17;
19769371c9d4SSatish Balay         cone[3] = 16;
19779371c9d4SSatish Balay         cone[4] = 31;
19789371c9d4SSatish Balay         cone[5] = 32;
19799371c9d4SSatish Balay         cone[6] = 33;
19809371c9d4SSatish Balay         cone[7] = 34;
19819566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
19829371c9d4SSatish Balay         cone[0] = 16;
19839371c9d4SSatish Balay         cone[1] = 17;
19849371c9d4SSatish Balay         cone[2] = 24;
19859371c9d4SSatish Balay         cone[3] = 23;
19869371c9d4SSatish Balay         cone[4] = 32;
19879371c9d4SSatish Balay         cone[5] = 36;
19889371c9d4SSatish Balay         cone[6] = 37;
19899371c9d4SSatish Balay         cone[7] = 33; /* 22 25 26 21 */
19909566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
19919371c9d4SSatish Balay         cone[0] = 18;
19929371c9d4SSatish Balay         cone[1] = 27;
19939371c9d4SSatish Balay         cone[2] = 24;
19949371c9d4SSatish Balay         cone[3] = 17;
19959371c9d4SSatish Balay         cone[4] = 34;
19969371c9d4SSatish Balay         cone[5] = 33;
19979371c9d4SSatish Balay         cone[6] = 37;
19989371c9d4SSatish Balay         cone[7] = 38;
19999566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
20009371c9d4SSatish Balay         cone[0] = 29;
20019371c9d4SSatish Balay         cone[1] = 27;
20029371c9d4SSatish Balay         cone[2] = 18;
20039371c9d4SSatish Balay         cone[3] = 15;
20049371c9d4SSatish Balay         cone[4] = 35;
20059371c9d4SSatish Balay         cone[5] = 31;
20069371c9d4SSatish Balay         cone[6] = 34;
20079371c9d4SSatish Balay         cone[7] = 38;
20089566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
20099371c9d4SSatish Balay         cone[0] = 29;
20109371c9d4SSatish Balay         cone[1] = 15;
20119371c9d4SSatish Balay         cone[2] = 16;
20129371c9d4SSatish Balay         cone[3] = 23;
20139371c9d4SSatish Balay         cone[4] = 35;
20149371c9d4SSatish Balay         cone[5] = 36;
20159371c9d4SSatish Balay         cone[6] = 32;
20169371c9d4SSatish Balay         cone[7] = 31;
20179566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2018006a8963SMatthew G. Knepley 
20199371c9d4SSatish Balay         cone[0] = 31;
20209371c9d4SSatish Balay         cone[1] = 34;
20219371c9d4SSatish Balay         cone[2] = 33;
20229371c9d4SSatish Balay         cone[3] = 32;
20239371c9d4SSatish Balay         cone[4] = 19;
20249371c9d4SSatish Balay         cone[5] = 22;
20259371c9d4SSatish Balay         cone[6] = 21;
20269371c9d4SSatish Balay         cone[7] = 20;
20279566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
20289371c9d4SSatish Balay         cone[0] = 32;
20299371c9d4SSatish Balay         cone[1] = 33;
20309371c9d4SSatish Balay         cone[2] = 37;
20319371c9d4SSatish Balay         cone[3] = 36;
20329371c9d4SSatish Balay         cone[4] = 22;
20339371c9d4SSatish Balay         cone[5] = 25;
20349371c9d4SSatish Balay         cone[6] = 26;
20359371c9d4SSatish Balay         cone[7] = 21;
20369566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 6, cone));
20379371c9d4SSatish Balay         cone[0] = 34;
20389371c9d4SSatish Balay         cone[1] = 38;
20399371c9d4SSatish Balay         cone[2] = 37;
20409371c9d4SSatish Balay         cone[3] = 33;
20419371c9d4SSatish Balay         cone[4] = 20;
20429371c9d4SSatish Balay         cone[5] = 21;
20439371c9d4SSatish Balay         cone[6] = 26;
20449371c9d4SSatish Balay         cone[7] = 28;
20459566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 7, cone));
20469371c9d4SSatish Balay         cone[0] = 35;
20479371c9d4SSatish Balay         cone[1] = 38;
20489371c9d4SSatish Balay         cone[2] = 34;
20499371c9d4SSatish Balay         cone[3] = 31;
20509371c9d4SSatish Balay         cone[4] = 30;
20519371c9d4SSatish Balay         cone[5] = 19;
20529371c9d4SSatish Balay         cone[6] = 20;
20539371c9d4SSatish Balay         cone[7] = 28;
20549566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 8, cone));
20559371c9d4SSatish Balay         cone[0] = 35;
20569371c9d4SSatish Balay         cone[1] = 31;
20579371c9d4SSatish Balay         cone[2] = 32;
20589371c9d4SSatish Balay         cone[3] = 36;
20599371c9d4SSatish Balay         cone[4] = 30;
20609371c9d4SSatish Balay         cone[5] = 25;
20619371c9d4SSatish Balay         cone[6] = 22;
20629371c9d4SSatish Balay         cone[7] = 19;
20639566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 9, cone));
2064ae8bcbbbSMatthew G. Knepley 
20659371c9d4SSatish Balay         cone[0] = 19;
20669371c9d4SSatish Balay         cone[1] = 20;
20679371c9d4SSatish Balay         cone[2] = 21;
20689371c9d4SSatish Balay         cone[3] = 22;
20699371c9d4SSatish Balay         cone[4] = 15;
20709371c9d4SSatish Balay         cone[5] = 16;
20719371c9d4SSatish Balay         cone[6] = 17;
20729371c9d4SSatish Balay         cone[7] = 18;
20739566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 10, cone));
20749371c9d4SSatish Balay         cone[0] = 22;
20759371c9d4SSatish Balay         cone[1] = 21;
20769371c9d4SSatish Balay         cone[2] = 26;
20779371c9d4SSatish Balay         cone[3] = 25;
20789371c9d4SSatish Balay         cone[4] = 16;
20799371c9d4SSatish Balay         cone[5] = 23;
20809371c9d4SSatish Balay         cone[6] = 24;
20819371c9d4SSatish Balay         cone[7] = 17;
20829566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 11, cone));
20839371c9d4SSatish Balay         cone[0] = 20;
20849371c9d4SSatish Balay         cone[1] = 28;
20859371c9d4SSatish Balay         cone[2] = 26;
20869371c9d4SSatish Balay         cone[3] = 21;
20879371c9d4SSatish Balay         cone[4] = 18;
20889371c9d4SSatish Balay         cone[5] = 17;
20899371c9d4SSatish Balay         cone[6] = 24;
20909371c9d4SSatish Balay         cone[7] = 27;
20919566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 12, cone));
20929371c9d4SSatish Balay         cone[0] = 30;
20939371c9d4SSatish Balay         cone[1] = 28;
20949371c9d4SSatish Balay         cone[2] = 20;
20959371c9d4SSatish Balay         cone[3] = 19;
20969371c9d4SSatish Balay         cone[4] = 29;
20979371c9d4SSatish Balay         cone[5] = 15;
20989371c9d4SSatish Balay         cone[6] = 18;
20999371c9d4SSatish Balay         cone[7] = 27;
21009566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 13, cone));
21019371c9d4SSatish Balay         cone[0] = 30;
21029371c9d4SSatish Balay         cone[1] = 19;
21039371c9d4SSatish Balay         cone[2] = 22;
21049371c9d4SSatish Balay         cone[3] = 25;
21059371c9d4SSatish Balay         cone[4] = 29;
21069371c9d4SSatish Balay         cone[5] = 23;
21079371c9d4SSatish Balay         cone[6] = 16;
21089371c9d4SSatish Balay         cone[7] = 15;
21099566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
2110006a8963SMatthew G. Knepley       } else {
21119371c9d4SSatish Balay         cone[0] = 5;
21129371c9d4SSatish Balay         cone[1] = 8;
21139371c9d4SSatish Balay         cone[2] = 7;
21149371c9d4SSatish Balay         cone[3] = 6;
21159371c9d4SSatish Balay         cone[4] = 9;
21169371c9d4SSatish Balay         cone[5] = 12;
21179371c9d4SSatish Balay         cone[6] = 11;
21189371c9d4SSatish Balay         cone[7] = 10;
21199566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
21209371c9d4SSatish Balay         cone[0] = 6;
21219371c9d4SSatish Balay         cone[1] = 7;
21229371c9d4SSatish Balay         cone[2] = 14;
21239371c9d4SSatish Balay         cone[3] = 13;
21249371c9d4SSatish Balay         cone[4] = 12;
21259371c9d4SSatish Balay         cone[5] = 15;
21269371c9d4SSatish Balay         cone[6] = 16;
21279371c9d4SSatish Balay         cone[7] = 11;
21289566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
21299371c9d4SSatish Balay         cone[0] = 8;
21309371c9d4SSatish Balay         cone[1] = 17;
21319371c9d4SSatish Balay         cone[2] = 14;
21329371c9d4SSatish Balay         cone[3] = 7;
21339371c9d4SSatish Balay         cone[4] = 10;
21349371c9d4SSatish Balay         cone[5] = 11;
21359371c9d4SSatish Balay         cone[6] = 16;
21369371c9d4SSatish Balay         cone[7] = 18;
21379566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
21389371c9d4SSatish Balay         cone[0] = 19;
21399371c9d4SSatish Balay         cone[1] = 17;
21409371c9d4SSatish Balay         cone[2] = 8;
21419371c9d4SSatish Balay         cone[3] = 5;
21429371c9d4SSatish Balay         cone[4] = 20;
21439371c9d4SSatish Balay         cone[5] = 9;
21449371c9d4SSatish Balay         cone[6] = 10;
21459371c9d4SSatish Balay         cone[7] = 18;
21469566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
21479371c9d4SSatish Balay         cone[0] = 19;
21489371c9d4SSatish Balay         cone[1] = 5;
21499371c9d4SSatish Balay         cone[2] = 6;
21509371c9d4SSatish Balay         cone[3] = 13;
21519371c9d4SSatish Balay         cone[4] = 20;
21529371c9d4SSatish Balay         cone[5] = 15;
21539371c9d4SSatish Balay         cone[6] = 12;
21549371c9d4SSatish Balay         cone[7] = 9;
21559566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2156006a8963SMatthew G. Knepley       }
2157d8c47e87SMatthew G. Knepley     }
21589566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
21599566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
21600510c589SMatthew G. Knepley   }
2161dbc1dc17SMatthew G. Knepley   /* Create cube geometry */
21620510c589SMatthew G. Knepley   {
21630510c589SMatthew G. Knepley     Vec             coordinates;
21640510c589SMatthew G. Knepley     PetscSection    coordSection;
21650510c589SMatthew G. Knepley     PetscScalar    *coords;
21660510c589SMatthew G. Knepley     PetscInt        coordSize, v;
21670510c589SMatthew G. Knepley     const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
21680510c589SMatthew G. Knepley     const PetscReal ds2 = dis / 2.0;
21690510c589SMatthew G. Knepley 
21700510c589SMatthew G. Knepley     /* Build coordinates */
21719566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
21729566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
21739566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
21749566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
21750510c589SMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
21769566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
21779566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
21780510c589SMatthew G. Knepley     }
21799566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
21809566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
21819566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
21829566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
21839566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
21849566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
21859566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
21869566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
2187dd400576SPatrick Sanan     if (rank == 0) {
21889371c9d4SSatish Balay       coords[0 * dim + 0]  = -ds2;
21899371c9d4SSatish Balay       coords[0 * dim + 1]  = -ds2;
21909371c9d4SSatish Balay       coords[0 * dim + 2]  = 0.0;
21919371c9d4SSatish Balay       coords[1 * dim + 0]  = ds2;
21929371c9d4SSatish Balay       coords[1 * dim + 1]  = -ds2;
21939371c9d4SSatish Balay       coords[1 * dim + 2]  = 0.0;
21949371c9d4SSatish Balay       coords[2 * dim + 0]  = ds2;
21959371c9d4SSatish Balay       coords[2 * dim + 1]  = ds2;
21969371c9d4SSatish Balay       coords[2 * dim + 2]  = 0.0;
21979371c9d4SSatish Balay       coords[3 * dim + 0]  = -ds2;
21989371c9d4SSatish Balay       coords[3 * dim + 1]  = ds2;
21999371c9d4SSatish Balay       coords[3 * dim + 2]  = 0.0;
22009371c9d4SSatish Balay       coords[4 * dim + 0]  = -ds2;
22019371c9d4SSatish Balay       coords[4 * dim + 1]  = -ds2;
22029371c9d4SSatish Balay       coords[4 * dim + 2]  = 1.0;
22039371c9d4SSatish Balay       coords[5 * dim + 0]  = -ds2;
22049371c9d4SSatish Balay       coords[5 * dim + 1]  = ds2;
22059371c9d4SSatish Balay       coords[5 * dim + 2]  = 1.0;
22069371c9d4SSatish Balay       coords[6 * dim + 0]  = ds2;
22079371c9d4SSatish Balay       coords[6 * dim + 1]  = ds2;
22089371c9d4SSatish Balay       coords[6 * dim + 2]  = 1.0;
22099371c9d4SSatish Balay       coords[7 * dim + 0]  = ds2;
22109371c9d4SSatish Balay       coords[7 * dim + 1]  = -ds2;
22119371c9d4SSatish Balay       coords[7 * dim + 2]  = 1.0;
22129371c9d4SSatish Balay       coords[8 * dim + 0]  = dis;
22139371c9d4SSatish Balay       coords[8 * dim + 1]  = -dis;
22149371c9d4SSatish Balay       coords[8 * dim + 2]  = 0.0;
22159371c9d4SSatish Balay       coords[9 * dim + 0]  = dis;
22169371c9d4SSatish Balay       coords[9 * dim + 1]  = dis;
22179371c9d4SSatish Balay       coords[9 * dim + 2]  = 0.0;
22189371c9d4SSatish Balay       coords[10 * dim + 0] = dis;
22199371c9d4SSatish Balay       coords[10 * dim + 1] = -dis;
22209371c9d4SSatish Balay       coords[10 * dim + 2] = 1.0;
22219371c9d4SSatish Balay       coords[11 * dim + 0] = dis;
22229371c9d4SSatish Balay       coords[11 * dim + 1] = dis;
22239371c9d4SSatish Balay       coords[11 * dim + 2] = 1.0;
22249371c9d4SSatish Balay       coords[12 * dim + 0] = -dis;
22259371c9d4SSatish Balay       coords[12 * dim + 1] = dis;
22269371c9d4SSatish Balay       coords[12 * dim + 2] = 0.0;
22279371c9d4SSatish Balay       coords[13 * dim + 0] = -dis;
22289371c9d4SSatish Balay       coords[13 * dim + 1] = dis;
22299371c9d4SSatish Balay       coords[13 * dim + 2] = 1.0;
22309371c9d4SSatish Balay       coords[14 * dim + 0] = -dis;
22319371c9d4SSatish Balay       coords[14 * dim + 1] = -dis;
22329371c9d4SSatish Balay       coords[14 * dim + 2] = 0.0;
22339371c9d4SSatish Balay       coords[15 * dim + 0] = -dis;
22349371c9d4SSatish Balay       coords[15 * dim + 1] = -dis;
22359371c9d4SSatish Balay       coords[15 * dim + 2] = 1.0;
2236ae8bcbbbSMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
22379371c9d4SSatish Balay         /* 15 31 19 */ coords[16 * dim + 0] = -ds2;
22389371c9d4SSatish Balay         coords[16 * dim + 1]                = -ds2;
22399371c9d4SSatish Balay         coords[16 * dim + 2]                = 0.5;
22409371c9d4SSatish Balay         /* 16 32 22 */ coords[17 * dim + 0] = ds2;
22419371c9d4SSatish Balay         coords[17 * dim + 1]                = -ds2;
22429371c9d4SSatish Balay         coords[17 * dim + 2]                = 0.5;
22439371c9d4SSatish Balay         /* 17 33 21 */ coords[18 * dim + 0] = ds2;
22449371c9d4SSatish Balay         coords[18 * dim + 1]                = ds2;
22459371c9d4SSatish Balay         coords[18 * dim + 2]                = 0.5;
22469371c9d4SSatish Balay         /* 18 34 20 */ coords[19 * dim + 0] = -ds2;
22479371c9d4SSatish Balay         coords[19 * dim + 1]                = ds2;
22489371c9d4SSatish Balay         coords[19 * dim + 2]                = 0.5;
22499371c9d4SSatish Balay         /* 29 35 30 */ coords[20 * dim + 0] = -dis;
22509371c9d4SSatish Balay         coords[20 * dim + 1]                = -dis;
22519371c9d4SSatish Balay         coords[20 * dim + 2]                = 0.5;
22529371c9d4SSatish Balay         /* 23 36 25 */ coords[21 * dim + 0] = dis;
22539371c9d4SSatish Balay         coords[21 * dim + 1]                = -dis;
22549371c9d4SSatish Balay         coords[21 * dim + 2]                = 0.5;
22559371c9d4SSatish Balay         /* 24 37 26 */ coords[22 * dim + 0] = dis;
22569371c9d4SSatish Balay         coords[22 * dim + 1]                = dis;
22579371c9d4SSatish Balay         coords[22 * dim + 2]                = 0.5;
22589371c9d4SSatish Balay         /* 27 38 28 */ coords[23 * dim + 0] = -dis;
22599371c9d4SSatish Balay         coords[23 * dim + 1]                = dis;
22609371c9d4SSatish Balay         coords[23 * dim + 2]                = 0.5;
2261ae8bcbbbSMatthew G. Knepley       }
2262d8c47e87SMatthew G. Knepley     }
22639566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
22649566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
22659566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
22660510c589SMatthew G. Knepley   }
2267006a8963SMatthew G. Knepley   /* Create periodicity */
2268006a8963SMatthew G. Knepley   if (periodicZ == DM_BOUNDARY_PERIODIC || periodicZ == DM_BOUNDARY_TWIST) {
22696858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
22706858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
2271006a8963SMatthew G. Knepley     PetscReal lower[3]   = {0.0, 0.0, 0.0};
2272ae8bcbbbSMatthew G. Knepley     PetscReal upper[3]   = {1.0, 1.0, 1.5};
22736858538eSMatthew G. Knepley     PetscInt  numZCells  = 3;
2274006a8963SMatthew G. Knepley 
22756858538eSMatthew G. Knepley     L[2]       = upper[2] - lower[2];
22766858538eSMatthew G. Knepley     maxCell[2] = 1.1 * (L[2] / numZCells);
22774fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
2278006a8963SMatthew G. Knepley   }
2279dbc1dc17SMatthew G. Knepley   {
22809318fe57SMatthew G. Knepley     DM          cdm;
22819318fe57SMatthew G. Knepley     PetscDS     cds;
22829318fe57SMatthew G. Knepley     PetscScalar c[2] = {1.0, 1.0};
2283dbc1dc17SMatthew G. Knepley 
228449704ca5SMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
22859566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
22869566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
22879566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 2, c));
2288dbc1dc17SMatthew G. Knepley   }
228946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
229046139095SJed Brown 
22919318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
22929566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(dm));
229349704ca5SMatthew G. Knepley 
229449704ca5SMatthew G. Knepley   char        oldprefix[PETSC_MAX_PATH_LEN];
229549704ca5SMatthew G. Knepley   const char *prefix;
229649704ca5SMatthew G. Knepley 
229749704ca5SMatthew G. Knepley   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
229849704ca5SMatthew G. Knepley   PetscCall(PetscStrncpy(oldprefix, prefix, PETSC_MAX_PATH_LEN));
229949704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, "petsc_cyl_ref_"));
230049704ca5SMatthew G. Knepley   for (PetscInt r = 0; r < PetscMax(0, Nr); ++r) {
230149704ca5SMatthew G. Knepley     DM rdm;
230249704ca5SMatthew G. Knepley 
230349704ca5SMatthew G. Knepley     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
230449704ca5SMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &rdm));
230549704ca5SMatthew G. Knepley   }
230649704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldprefix));
230749704ca5SMatthew G. Knepley   PetscCall(DMPlexRemapGeometry(dm, 0.0, snapToCylinder));
230849704ca5SMatthew G. Knepley 
230949704ca5SMatthew G. Knepley   DMLabel         bdlabel, edgelabel;
231049704ca5SMatthew G. Knepley   IS              faceIS;
231149704ca5SMatthew G. Knepley   const PetscInt *faces;
231249704ca5SMatthew G. Knepley   PetscInt        Nf;
231349704ca5SMatthew G. Knepley 
231449704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
231549704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
231649704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "generatrix"));
231749704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "generatrix", &edgelabel));
231849704ca5SMatthew G. Knepley   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
231949704ca5SMatthew G. Knepley   // Remove faces on top and bottom
232049704ca5SMatthew G. Knepley   PetscCall(DMLabelGetStratumIS(bdlabel, 1, &faceIS));
2321ba1b3593SJeremy L Thompson   if (faceIS) {
232249704ca5SMatthew G. Knepley     PetscCall(ISGetLocalSize(faceIS, &Nf));
232349704ca5SMatthew G. Knepley     PetscCall(ISGetIndices(faceIS, &faces));
232449704ca5SMatthew G. Knepley     for (PetscInt f = 0; f < Nf; ++f) {
232549704ca5SMatthew G. Knepley       PetscReal vol, normal[3];
232649704ca5SMatthew G. Knepley 
232749704ca5SMatthew G. Knepley       PetscCall(DMPlexComputeCellGeometryFVM(dm, faces[f], &vol, NULL, normal));
232849704ca5SMatthew G. Knepley       if (PetscAbsReal(normal[2]) < PETSC_SMALL) PetscCall(DMLabelSetValue(edgelabel, faces[f], 1));
232949704ca5SMatthew G. Knepley     }
233049704ca5SMatthew G. Knepley     PetscCall(ISRestoreIndices(faceIS, &faces));
233149704ca5SMatthew G. Knepley     PetscCall(ISDestroy(&faceIS));
2332ba1b3593SJeremy L Thompson   }
233349704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, bdlabel));
233449704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, edgelabel));
23353ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23360510c589SMatthew G. Knepley }
23370510c589SMatthew G. Knepley 
233824119c2aSMatthew G. Knepley /*@
23399318fe57SMatthew G. Knepley   DMPlexCreateHexCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using hexahedra.
234024119c2aSMatthew G. Knepley 
2341d083f849SBarry Smith   Collective
234224119c2aSMatthew G. Knepley 
234324119c2aSMatthew G. Knepley   Input Parameters:
2344a1cb98faSBarry Smith + comm      - The communicator for the `DM` object
234549704ca5SMatthew G. Knepley . periodicZ - The boundary type for the Z direction
234649704ca5SMatthew G. Knepley - Nr        - The number of refinements to carry out
234724119c2aSMatthew G. Knepley 
234824119c2aSMatthew G. Knepley   Output Parameter:
234920f4b53cSBarry Smith . dm - The `DM` object
235024119c2aSMatthew G. Knepley 
235124119c2aSMatthew G. Knepley   Level: beginner
235224119c2aSMatthew G. Knepley 
2353a1cb98faSBarry Smith   Note:
2354a4e35b19SJacob Faibussowitsch   Here is the output numbering looking from the bottom of the cylinder\:
2355a1cb98faSBarry Smith .vb
2356a1cb98faSBarry Smith        17-----14
2357a1cb98faSBarry Smith         |     |
2358a1cb98faSBarry Smith         |  2  |
2359a1cb98faSBarry Smith         |     |
2360a1cb98faSBarry Smith  17-----8-----7-----14
2361a1cb98faSBarry Smith   |     |     |     |
2362a1cb98faSBarry Smith   |  3  |  0  |  1  |
2363a1cb98faSBarry Smith   |     |     |     |
2364a1cb98faSBarry Smith  19-----5-----6-----13
2365a1cb98faSBarry Smith         |     |
2366a1cb98faSBarry Smith         |  4  |
2367a1cb98faSBarry Smith         |     |
2368a1cb98faSBarry Smith        19-----13
2369a1cb98faSBarry Smith 
2370a1cb98faSBarry Smith  and up through the top
2371a1cb98faSBarry Smith 
2372a1cb98faSBarry Smith        18-----16
2373a1cb98faSBarry Smith         |     |
2374a1cb98faSBarry Smith         |  2  |
2375a1cb98faSBarry Smith         |     |
2376a1cb98faSBarry Smith  18----10----11-----16
2377a1cb98faSBarry Smith   |     |     |     |
2378a1cb98faSBarry Smith   |  3  |  0  |  1  |
2379a1cb98faSBarry Smith   |     |     |     |
2380a1cb98faSBarry Smith  20-----9----12-----15
2381a1cb98faSBarry Smith         |     |
2382a1cb98faSBarry Smith         |  4  |
2383a1cb98faSBarry Smith         |     |
2384a1cb98faSBarry Smith        20-----15
2385a1cb98faSBarry Smith .ve
2386a1cb98faSBarry Smith 
23871cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
238824119c2aSMatthew G. Knepley @*/
238949704ca5SMatthew G. Knepley PetscErrorCode DMPlexCreateHexCylinderMesh(MPI_Comm comm, DMBoundaryType periodicZ, PetscInt Nr, DM *dm)
2390d71ae5a4SJacob Faibussowitsch {
23919318fe57SMatthew G. Knepley   PetscFunctionBegin;
239249704ca5SMatthew G. Knepley   PetscAssertPointer(dm, 4);
23939566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
23949566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
239549704ca5SMatthew G. Knepley   PetscCall(DMPlexCreateHexCylinderMesh_Internal(*dm, periodicZ, Nr));
23963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23979318fe57SMatthew G. Knepley }
23989318fe57SMatthew G. Knepley 
2399d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeCylinderMesh_Internal(DM dm, PetscInt n, PetscBool interpolate)
2400d71ae5a4SJacob Faibussowitsch {
240124119c2aSMatthew G. Knepley   const PetscInt dim = 3;
2402412e9a14SMatthew G. Knepley   PetscInt       numCells, numVertices, v;
24039fe9f049SMatthew G. Knepley   PetscMPIInt    rank;
240424119c2aSMatthew G. Knepley 
240524119c2aSMatthew G. Knepley   PetscFunctionBegin;
240663a3b9bcSJacob Faibussowitsch   PetscCheck(n >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Number of wedges %" PetscInt_FMT " cannot be negative", n);
240746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
24089566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
24099566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
2410412e9a14SMatthew G. Knepley   /* Must create the celltype label here so that we do not automatically try to compute the types */
24119566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "celltype"));
241224119c2aSMatthew G. Knepley   /* Create topology */
241324119c2aSMatthew G. Knepley   {
241424119c2aSMatthew G. Knepley     PetscInt cone[6], c;
241524119c2aSMatthew G. Knepley 
2416dd400576SPatrick Sanan     numCells    = rank == 0 ? n : 0;
2417dd400576SPatrick Sanan     numVertices = rank == 0 ? 2 * (n + 1) : 0;
24189566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
24199566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
24209566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
242124119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
24229371c9d4SSatish Balay       cone[0] = c + n * 1;
24239371c9d4SSatish Balay       cone[1] = (c + 1) % n + n * 1;
24249371c9d4SSatish Balay       cone[2] = 0 + 3 * n;
24259371c9d4SSatish Balay       cone[3] = c + n * 2;
24269371c9d4SSatish Balay       cone[4] = (c + 1) % n + n * 2;
24279371c9d4SSatish Balay       cone[5] = 1 + 3 * n;
24289566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(dm, c, cone));
24299566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCellType(dm, c, DM_POLYTOPE_TRI_PRISM_TENSOR));
243024119c2aSMatthew G. Knepley     }
24319566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
24329566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
243324119c2aSMatthew G. Knepley   }
243448a46eb9SPierre Jolivet   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(dm, v, DM_POLYTOPE_POINT));
243524119c2aSMatthew G. Knepley   /* Create cylinder geometry */
243624119c2aSMatthew G. Knepley   {
243724119c2aSMatthew G. Knepley     Vec          coordinates;
243824119c2aSMatthew G. Knepley     PetscSection coordSection;
243924119c2aSMatthew G. Knepley     PetscScalar *coords;
2440412e9a14SMatthew G. Knepley     PetscInt     coordSize, c;
244124119c2aSMatthew G. Knepley 
244224119c2aSMatthew G. Knepley     /* Build coordinates */
24439566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
24449566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
24459566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
24469566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
244724119c2aSMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
24489566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
24499566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
245024119c2aSMatthew G. Knepley     }
24519566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
24529566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
24539566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
24549566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
24559566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
24569566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
24579566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
24589566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
245924119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
24609371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
24619371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
24629371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 2] = 1.0;
24639371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
24649371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
24659371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 2] = 0.0;
246624119c2aSMatthew G. Knepley     }
2467dd400576SPatrick Sanan     if (rank == 0) {
24689371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 0] = 0.0;
24699371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 1] = 0.0;
24709371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 2] = 1.0;
24719371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 0] = 0.0;
24729371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 1] = 0.0;
24739371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 2] = 0.0;
24749fe9f049SMatthew G. Knepley     }
24759566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
24769566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
24779566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
247824119c2aSMatthew G. Knepley   }
247946139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
24809318fe57SMatthew G. Knepley   /* Interpolate */
24819566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
24823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
24839318fe57SMatthew G. Knepley }
24849318fe57SMatthew G. Knepley 
24859318fe57SMatthew G. Knepley /*@
24869318fe57SMatthew G. Knepley   DMPlexCreateWedgeCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using wedges.
24879318fe57SMatthew G. Knepley 
24889318fe57SMatthew G. Knepley   Collective
24899318fe57SMatthew G. Knepley 
24909318fe57SMatthew G. Knepley   Input Parameters:
2491a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
24929318fe57SMatthew G. Knepley . n           - The number of wedges around the origin
24939318fe57SMatthew G. Knepley - interpolate - Create edges and faces
24949318fe57SMatthew G. Knepley 
24959318fe57SMatthew G. Knepley   Output Parameter:
2496a1cb98faSBarry Smith . dm - The `DM` object
24979318fe57SMatthew G. Knepley 
24989318fe57SMatthew G. Knepley   Level: beginner
24999318fe57SMatthew G. Knepley 
25001cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
25019318fe57SMatthew G. Knepley @*/
2502d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeCylinderMesh(MPI_Comm comm, PetscInt n, PetscBool interpolate, DM *dm)
2503d71ae5a4SJacob Faibussowitsch {
25049318fe57SMatthew G. Knepley   PetscFunctionBegin;
25054f572ea9SToby Isaac   PetscAssertPointer(dm, 4);
25069566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
25079566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
25089566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(*dm, n, interpolate));
25093ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
251024119c2aSMatthew G. Knepley }
251124119c2aSMatthew G. Knepley 
2512d71ae5a4SJacob Faibussowitsch static inline PetscReal DiffNormReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2513d71ae5a4SJacob Faibussowitsch {
251465a81367SMatthew G. Knepley   PetscReal prod = 0.0;
251565a81367SMatthew G. Knepley   PetscInt  i;
251665a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += PetscSqr(x[i] - y[i]);
251765a81367SMatthew G. Knepley   return PetscSqrtReal(prod);
251865a81367SMatthew G. Knepley }
2519dd2b43ebSStefano Zampini 
2520d71ae5a4SJacob Faibussowitsch static inline PetscReal DotReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
2521d71ae5a4SJacob Faibussowitsch {
252265a81367SMatthew G. Knepley   PetscReal prod = 0.0;
252365a81367SMatthew G. Knepley   PetscInt  i;
252465a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += x[i] * y[i];
252565a81367SMatthew G. Knepley   return prod;
252665a81367SMatthew G. Knepley }
252765a81367SMatthew G. Knepley 
252851a74b61SMatthew G. Knepley /* The first constant is the sphere radius */
2529d71ae5a4SJacob 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[])
2530d71ae5a4SJacob Faibussowitsch {
253151a74b61SMatthew G. Knepley   PetscReal r     = PetscRealPart(constants[0]);
253251a74b61SMatthew G. Knepley   PetscReal norm2 = 0.0, fac;
253351a74b61SMatthew G. Knepley   PetscInt  n     = uOff[1] - uOff[0], d;
253451a74b61SMatthew G. Knepley 
253551a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) norm2 += PetscSqr(PetscRealPart(u[d]));
253651a74b61SMatthew G. Knepley   fac = r / PetscSqrtReal(norm2);
253751a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) f0[d] = u[d] * fac;
253851a74b61SMatthew G. Knepley }
253951a74b61SMatthew G. Knepley 
2540d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateSphereMesh_Internal(DM dm, PetscInt dim, PetscBool simplex, PetscReal R)
2541d71ae5a4SJacob Faibussowitsch {
254265a81367SMatthew G. Knepley   const PetscInt embedDim = dim + 1;
254365a81367SMatthew G. Knepley   PetscSection   coordSection;
254465a81367SMatthew G. Knepley   Vec            coordinates;
254565a81367SMatthew G. Knepley   PetscScalar   *coords;
254665a81367SMatthew G. Knepley   PetscReal     *coordsIn;
254707c565c5SJose E. Roman   PetscInt       numCells, numEdges, numVerts = 0, firstVertex = 0, v, firstEdge, coordSize, d, e;
254865a81367SMatthew G. Knepley   PetscMPIInt    rank;
254965a81367SMatthew G. Knepley 
255065a81367SMatthew G. Knepley   PetscFunctionBegin;
25519318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveBool(dm, simplex, 3);
255246139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
25539566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
25549566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim + 1));
25559566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
255665a81367SMatthew G. Knepley   switch (dim) {
25575c344501SMatthew G. Knepley   case 1:
25585c344501SMatthew G. Knepley     numCells = 16;
25595c344501SMatthew G. Knepley     numVerts = numCells;
25605c344501SMatthew G. Knepley 
25615c344501SMatthew G. Knepley     // Build Topology
25625c344501SMatthew G. Knepley     PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
25635c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
25645c344501SMatthew G. Knepley     PetscCall(DMSetUp(dm));
25655c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; ++c) {
25665c344501SMatthew G. Knepley       PetscInt cone[2];
25675c344501SMatthew G. Knepley 
25685c344501SMatthew G. Knepley       cone[0] = c + numCells;
25695c344501SMatthew G. Knepley       cone[1] = (c + 1) % numVerts + numCells;
25705c344501SMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, c, cone));
25715c344501SMatthew G. Knepley     }
25725c344501SMatthew G. Knepley     PetscCall(DMPlexSymmetrize(dm));
25735c344501SMatthew G. Knepley     PetscCall(DMPlexStratify(dm));
25745c344501SMatthew G. Knepley     PetscCall(PetscMalloc1(numVerts * embedDim, &coordsIn));
25755c344501SMatthew G. Knepley     for (PetscInt v = 0; v < numVerts; ++v) {
25765c344501SMatthew G. Knepley       const PetscReal rad = 2. * PETSC_PI * v / numVerts;
25775c344501SMatthew G. Knepley 
25785c344501SMatthew G. Knepley       coordsIn[v * embedDim + 0] = PetscCosReal(rad);
25795c344501SMatthew G. Knepley       coordsIn[v * embedDim + 1] = PetscSinReal(rad);
25805c344501SMatthew G. Knepley     }
25815c344501SMatthew G. Knepley     break;
258265a81367SMatthew G. Knepley   case 2:
258365a81367SMatthew G. Knepley     if (simplex) {
258451a74b61SMatthew G. Knepley       const PetscReal radius    = PetscSqrtReal(1 + PETSC_PHI * PETSC_PHI) / (1.0 + PETSC_PHI);
258551a74b61SMatthew G. Knepley       const PetscReal edgeLen   = 2.0 / (1.0 + PETSC_PHI) * (R / radius);
258665a81367SMatthew G. Knepley       const PetscInt  degree    = 5;
258751a74b61SMatthew G. Knepley       PetscReal       vertex[3] = {0.0, 1.0 / (1.0 + PETSC_PHI), PETSC_PHI / (1.0 + PETSC_PHI)};
258865a81367SMatthew G. Knepley       PetscInt        s[3]      = {1, 1, 1};
258965a81367SMatthew G. Knepley       PetscInt        cone[3];
259007c565c5SJose E. Roman       PetscInt       *graph;
259165a81367SMatthew G. Knepley 
25929371c9d4SSatish Balay       vertex[0] *= R / radius;
25939371c9d4SSatish Balay       vertex[1] *= R / radius;
25949371c9d4SSatish Balay       vertex[2] *= R / radius;
2595dd400576SPatrick Sanan       numCells    = rank == 0 ? 20 : 0;
2596dd400576SPatrick Sanan       numVerts    = rank == 0 ? 12 : 0;
259765a81367SMatthew G. Knepley       firstVertex = numCells;
259851a74b61SMatthew G. Knepley       /* Use icosahedron, which for a R-sphere has coordinates which are all cyclic permutations of
259965a81367SMatthew G. Knepley 
260065a81367SMatthew G. Knepley            (0, \pm 1/\phi+1, \pm \phi/\phi+1)
260165a81367SMatthew G. Knepley 
260265a81367SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
260351a74b61SMatthew G. Knepley          length is then given by 2/(1+\phi) = 2 * 0.38197 = 0.76393.
260465a81367SMatthew G. Knepley       */
260565a81367SMatthew G. Knepley       /* Construct vertices */
26069566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2607dd400576SPatrick Sanan       if (rank == 0) {
260807c565c5SJose E. Roman         for (PetscInt p = 0, i = 0; p < embedDim; ++p) {
260965a81367SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
261065a81367SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
261165a81367SMatthew G. Knepley               for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertex[(d + p) % embedDim];
261265a81367SMatthew G. Knepley               ++i;
261365a81367SMatthew G. Knepley             }
261465a81367SMatthew G. Knepley           }
261565a81367SMatthew G. Knepley         }
261645da822fSValeria Barra       }
261765a81367SMatthew G. Knepley       /* Construct graph */
26189566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
261907c565c5SJose E. Roman       for (PetscInt i = 0; i < numVerts; ++i) {
262007c565c5SJose E. Roman         PetscInt k = 0;
262107c565c5SJose E. Roman         for (PetscInt j = 0; j < numVerts; ++j) {
26229371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
26239371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
26249371c9d4SSatish Balay             ++k;
26259371c9d4SSatish Balay           }
262665a81367SMatthew G. Knepley         }
262763a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid icosahedron, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
262865a81367SMatthew G. Knepley       }
262965a81367SMatthew G. Knepley       /* Build Topology */
26309566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
263107c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
26329566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
263365a81367SMatthew G. Knepley       /* Cells */
263407c565c5SJose E. Roman       for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
263507c565c5SJose E. Roman         for (PetscInt j = 0; j < i; ++j) {
263607c565c5SJose E. Roman           for (PetscInt k = 0; k < j; ++k) {
263765a81367SMatthew G. Knepley             if (graph[i * numVerts + j] && graph[j * numVerts + k] && graph[k * numVerts + i]) {
26389371c9d4SSatish Balay               cone[0] = firstVertex + i;
26399371c9d4SSatish Balay               cone[1] = firstVertex + j;
26409371c9d4SSatish Balay               cone[2] = firstVertex + k;
264165a81367SMatthew G. Knepley               /* Check orientation */
264265a81367SMatthew G. Knepley               {
26439371c9d4SSatish Balay                 const PetscInt epsilon[3][3][3] = {
26449371c9d4SSatish Balay                   {{0, 0, 0},  {0, 0, 1},  {0, -1, 0}},
26459371c9d4SSatish Balay                   {{0, 0, -1}, {0, 0, 0},  {1, 0, 0} },
26469371c9d4SSatish Balay                   {{0, 1, 0},  {-1, 0, 0}, {0, 0, 0} }
26479371c9d4SSatish Balay                 };
264865a81367SMatthew G. Knepley                 PetscReal normal[3];
264965a81367SMatthew G. Knepley                 PetscInt  e, f;
265065a81367SMatthew G. Knepley 
265165a81367SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) {
265265a81367SMatthew G. Knepley                   normal[d] = 0.0;
265365a81367SMatthew G. Knepley                   for (e = 0; e < embedDim; ++e) {
2654ad540459SPierre 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]);
265565a81367SMatthew G. Knepley                   }
265665a81367SMatthew G. Knepley                 }
26579371c9d4SSatish Balay                 if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
26589371c9d4SSatish Balay                   PetscInt tmp = cone[1];
26599371c9d4SSatish Balay                   cone[1]      = cone[2];
26609371c9d4SSatish Balay                   cone[2]      = tmp;
266165a81367SMatthew G. Knepley                 }
266265a81367SMatthew G. Knepley               }
26639566063dSJacob Faibussowitsch               PetscCall(DMPlexSetCone(dm, c++, cone));
266465a81367SMatthew G. Knepley             }
266565a81367SMatthew G. Knepley           }
266665a81367SMatthew G. Knepley         }
266765a81367SMatthew G. Knepley       }
26689566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
26699566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
26709566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
267165a81367SMatthew G. Knepley     } else {
26722829fed8SMatthew G. Knepley       /*
26732829fed8SMatthew G. Knepley         12-21--13
26742829fed8SMatthew G. Knepley          |     |
26752829fed8SMatthew G. Knepley         25  4  24
26762829fed8SMatthew G. Knepley          |     |
26772829fed8SMatthew G. Knepley   12-25--9-16--8-24--13
26782829fed8SMatthew G. Knepley    |     |     |     |
26792829fed8SMatthew G. Knepley   23  5 17  0 15  3  22
26802829fed8SMatthew G. Knepley    |     |     |     |
26812829fed8SMatthew G. Knepley   10-20--6-14--7-19--11
26822829fed8SMatthew G. Knepley          |     |
26832829fed8SMatthew G. Knepley         20  1  19
26842829fed8SMatthew G. Knepley          |     |
26852829fed8SMatthew G. Knepley         10-18--11
26862829fed8SMatthew G. Knepley          |     |
26872829fed8SMatthew G. Knepley         23  2  22
26882829fed8SMatthew G. Knepley          |     |
26892829fed8SMatthew G. Knepley         12-21--13
26902829fed8SMatthew G. Knepley        */
26912829fed8SMatthew G. Knepley       PetscInt cone[4], ornt[4];
26922829fed8SMatthew G. Knepley 
2693dd400576SPatrick Sanan       numCells    = rank == 0 ? 6 : 0;
2694dd400576SPatrick Sanan       numEdges    = rank == 0 ? 12 : 0;
2695dd400576SPatrick Sanan       numVerts    = rank == 0 ? 8 : 0;
269665a81367SMatthew G. Knepley       firstVertex = numCells;
269765a81367SMatthew G. Knepley       firstEdge   = numCells + numVerts;
26982829fed8SMatthew G. Knepley       /* Build Topology */
26999566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numEdges + numVerts));
270007c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 4));
270148a46eb9SPierre Jolivet       for (e = firstEdge; e < firstEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
27029566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
2703dd400576SPatrick Sanan       if (rank == 0) {
27042829fed8SMatthew G. Knepley         /* Cell 0 */
27059371c9d4SSatish Balay         cone[0] = 14;
27069371c9d4SSatish Balay         cone[1] = 15;
27079371c9d4SSatish Balay         cone[2] = 16;
27089371c9d4SSatish Balay         cone[3] = 17;
27099566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
27109371c9d4SSatish Balay         ornt[0] = 0;
27119371c9d4SSatish Balay         ornt[1] = 0;
27129371c9d4SSatish Balay         ornt[2] = 0;
27139371c9d4SSatish Balay         ornt[3] = 0;
27149566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 0, ornt));
27152829fed8SMatthew G. Knepley         /* Cell 1 */
27169371c9d4SSatish Balay         cone[0] = 18;
27179371c9d4SSatish Balay         cone[1] = 19;
27189371c9d4SSatish Balay         cone[2] = 14;
27199371c9d4SSatish Balay         cone[3] = 20;
27209566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
27219371c9d4SSatish Balay         ornt[0] = 0;
27229371c9d4SSatish Balay         ornt[1] = 0;
27239371c9d4SSatish Balay         ornt[2] = -1;
27249371c9d4SSatish Balay         ornt[3] = 0;
27259566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 1, ornt));
27262829fed8SMatthew G. Knepley         /* Cell 2 */
27279371c9d4SSatish Balay         cone[0] = 21;
27289371c9d4SSatish Balay         cone[1] = 22;
27299371c9d4SSatish Balay         cone[2] = 18;
27309371c9d4SSatish Balay         cone[3] = 23;
27319566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
27329371c9d4SSatish Balay         ornt[0] = 0;
27339371c9d4SSatish Balay         ornt[1] = 0;
27349371c9d4SSatish Balay         ornt[2] = -1;
27359371c9d4SSatish Balay         ornt[3] = 0;
27369566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 2, ornt));
27372829fed8SMatthew G. Knepley         /* Cell 3 */
27389371c9d4SSatish Balay         cone[0] = 19;
27399371c9d4SSatish Balay         cone[1] = 22;
27409371c9d4SSatish Balay         cone[2] = 24;
27419371c9d4SSatish Balay         cone[3] = 15;
27429566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
27439371c9d4SSatish Balay         ornt[0] = -1;
27449371c9d4SSatish Balay         ornt[1] = -1;
27459371c9d4SSatish Balay         ornt[2] = 0;
27469371c9d4SSatish Balay         ornt[3] = -1;
27479566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 3, ornt));
27482829fed8SMatthew G. Knepley         /* Cell 4 */
27499371c9d4SSatish Balay         cone[0] = 16;
27509371c9d4SSatish Balay         cone[1] = 24;
27519371c9d4SSatish Balay         cone[2] = 21;
27529371c9d4SSatish Balay         cone[3] = 25;
27539566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
27549371c9d4SSatish Balay         ornt[0] = -1;
27559371c9d4SSatish Balay         ornt[1] = -1;
27569371c9d4SSatish Balay         ornt[2] = -1;
27579371c9d4SSatish Balay         ornt[3] = 0;
27589566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 4, ornt));
27592829fed8SMatthew G. Knepley         /* Cell 5 */
27609371c9d4SSatish Balay         cone[0] = 20;
27619371c9d4SSatish Balay         cone[1] = 17;
27629371c9d4SSatish Balay         cone[2] = 25;
27639371c9d4SSatish Balay         cone[3] = 23;
27649566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
27659371c9d4SSatish Balay         ornt[0] = -1;
27669371c9d4SSatish Balay         ornt[1] = -1;
27679371c9d4SSatish Balay         ornt[2] = -1;
27689371c9d4SSatish Balay         ornt[3] = -1;
27699566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 5, ornt));
27702829fed8SMatthew G. Knepley         /* Edges */
27719371c9d4SSatish Balay         cone[0] = 6;
27729371c9d4SSatish Balay         cone[1] = 7;
27739566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
27749371c9d4SSatish Balay         cone[0] = 7;
27759371c9d4SSatish Balay         cone[1] = 8;
27769566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 15, cone));
27779371c9d4SSatish Balay         cone[0] = 8;
27789371c9d4SSatish Balay         cone[1] = 9;
27799566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 16, cone));
27809371c9d4SSatish Balay         cone[0] = 9;
27819371c9d4SSatish Balay         cone[1] = 6;
27829566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 17, cone));
27839371c9d4SSatish Balay         cone[0] = 10;
27849371c9d4SSatish Balay         cone[1] = 11;
27859566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 18, cone));
27869371c9d4SSatish Balay         cone[0] = 11;
27879371c9d4SSatish Balay         cone[1] = 7;
27889566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 19, cone));
27899371c9d4SSatish Balay         cone[0] = 6;
27909371c9d4SSatish Balay         cone[1] = 10;
27919566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 20, cone));
27929371c9d4SSatish Balay         cone[0] = 12;
27939371c9d4SSatish Balay         cone[1] = 13;
27949566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 21, cone));
27959371c9d4SSatish Balay         cone[0] = 13;
27969371c9d4SSatish Balay         cone[1] = 11;
27979566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 22, cone));
27989371c9d4SSatish Balay         cone[0] = 10;
27999371c9d4SSatish Balay         cone[1] = 12;
28009566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 23, cone));
28019371c9d4SSatish Balay         cone[0] = 13;
28029371c9d4SSatish Balay         cone[1] = 8;
28039566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 24, cone));
28049371c9d4SSatish Balay         cone[0] = 12;
28059371c9d4SSatish Balay         cone[1] = 9;
28069566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 25, cone));
280745da822fSValeria Barra       }
28089566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
28099566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
28102829fed8SMatthew G. Knepley       /* Build coordinates */
28119566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2812dd400576SPatrick Sanan       if (rank == 0) {
28139371c9d4SSatish Balay         coordsIn[0 * embedDim + 0] = -R;
28149371c9d4SSatish Balay         coordsIn[0 * embedDim + 1] = R;
28159371c9d4SSatish Balay         coordsIn[0 * embedDim + 2] = -R;
28169371c9d4SSatish Balay         coordsIn[1 * embedDim + 0] = R;
28179371c9d4SSatish Balay         coordsIn[1 * embedDim + 1] = R;
28189371c9d4SSatish Balay         coordsIn[1 * embedDim + 2] = -R;
28199371c9d4SSatish Balay         coordsIn[2 * embedDim + 0] = R;
28209371c9d4SSatish Balay         coordsIn[2 * embedDim + 1] = -R;
28219371c9d4SSatish Balay         coordsIn[2 * embedDim + 2] = -R;
28229371c9d4SSatish Balay         coordsIn[3 * embedDim + 0] = -R;
28239371c9d4SSatish Balay         coordsIn[3 * embedDim + 1] = -R;
28249371c9d4SSatish Balay         coordsIn[3 * embedDim + 2] = -R;
28259371c9d4SSatish Balay         coordsIn[4 * embedDim + 0] = -R;
28269371c9d4SSatish Balay         coordsIn[4 * embedDim + 1] = R;
28279371c9d4SSatish Balay         coordsIn[4 * embedDim + 2] = R;
28289371c9d4SSatish Balay         coordsIn[5 * embedDim + 0] = R;
28299371c9d4SSatish Balay         coordsIn[5 * embedDim + 1] = R;
28309371c9d4SSatish Balay         coordsIn[5 * embedDim + 2] = R;
28319371c9d4SSatish Balay         coordsIn[6 * embedDim + 0] = -R;
28329371c9d4SSatish Balay         coordsIn[6 * embedDim + 1] = -R;
28339371c9d4SSatish Balay         coordsIn[6 * embedDim + 2] = R;
28349371c9d4SSatish Balay         coordsIn[7 * embedDim + 0] = R;
28359371c9d4SSatish Balay         coordsIn[7 * embedDim + 1] = -R;
28369371c9d4SSatish Balay         coordsIn[7 * embedDim + 2] = R;
283765a81367SMatthew G. Knepley       }
283845da822fSValeria Barra     }
283965a81367SMatthew G. Knepley     break;
284065a81367SMatthew G. Knepley   case 3:
2841116ded15SMatthew G. Knepley     if (simplex) {
2842116ded15SMatthew G. Knepley       const PetscReal edgeLen         = 1.0 / PETSC_PHI;
284351a74b61SMatthew G. Knepley       PetscReal       vertexA[4]      = {0.5, 0.5, 0.5, 0.5};
284451a74b61SMatthew G. Knepley       PetscReal       vertexB[4]      = {1.0, 0.0, 0.0, 0.0};
284551a74b61SMatthew G. Knepley       PetscReal       vertexC[4]      = {0.5, 0.5 * PETSC_PHI, 0.5 / PETSC_PHI, 0.0};
2846116ded15SMatthew G. Knepley       const PetscInt  degree          = 12;
2847116ded15SMatthew G. Knepley       PetscInt        s[4]            = {1, 1, 1};
28489371c9d4SSatish Balay       PetscInt        evenPerm[12][4] = {
28499371c9d4SSatish Balay         {0, 1, 2, 3},
28509371c9d4SSatish Balay         {0, 2, 3, 1},
28519371c9d4SSatish Balay         {0, 3, 1, 2},
28529371c9d4SSatish Balay         {1, 0, 3, 2},
28539371c9d4SSatish Balay         {1, 2, 0, 3},
28549371c9d4SSatish Balay         {1, 3, 2, 0},
28559371c9d4SSatish Balay         {2, 0, 1, 3},
28569371c9d4SSatish Balay         {2, 1, 3, 0},
28579371c9d4SSatish Balay         {2, 3, 0, 1},
28589371c9d4SSatish Balay         {3, 0, 2, 1},
28599371c9d4SSatish Balay         {3, 1, 0, 2},
28609371c9d4SSatish Balay         {3, 2, 1, 0}
28619371c9d4SSatish Balay       };
2862116ded15SMatthew G. Knepley       PetscInt  cone[4];
2863116ded15SMatthew G. Knepley       PetscInt *graph, p, i, j, k, l;
2864116ded15SMatthew G. Knepley 
28659371c9d4SSatish Balay       vertexA[0] *= R;
28669371c9d4SSatish Balay       vertexA[1] *= R;
28679371c9d4SSatish Balay       vertexA[2] *= R;
28689371c9d4SSatish Balay       vertexA[3] *= R;
28699371c9d4SSatish Balay       vertexB[0] *= R;
28709371c9d4SSatish Balay       vertexB[1] *= R;
28719371c9d4SSatish Balay       vertexB[2] *= R;
28729371c9d4SSatish Balay       vertexB[3] *= R;
28739371c9d4SSatish Balay       vertexC[0] *= R;
28749371c9d4SSatish Balay       vertexC[1] *= R;
28759371c9d4SSatish Balay       vertexC[2] *= R;
28769371c9d4SSatish Balay       vertexC[3] *= R;
2877dd400576SPatrick Sanan       numCells    = rank == 0 ? 600 : 0;
2878dd400576SPatrick Sanan       numVerts    = rank == 0 ? 120 : 0;
2879116ded15SMatthew G. Knepley       firstVertex = numCells;
2880116ded15SMatthew G. Knepley       /* Use the 600-cell, which for a unit sphere has coordinates which are
2881116ded15SMatthew G. Knepley 
2882116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm 1,    \pm 1, \pm 1)                          16
2883116ded15SMatthew G. Knepley                (\pm 1,    0,       0,      0)  all cyclic permutations   8
2884116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm phi, \pm 1/phi, 0)  all even permutations    96
2885116ded15SMatthew G. Knepley 
2886116ded15SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
28876333ae4fSvaleriabarra          length is then given by 1/\phi = 0.61803.
2888116ded15SMatthew G. Knepley 
2889116ded15SMatthew G. Knepley          http://buzzard.pugetsound.edu/sage-practice/ch03s03.html
2890116ded15SMatthew G. Knepley          http://mathworld.wolfram.com/600-Cell.html
2891116ded15SMatthew G. Knepley       */
2892116ded15SMatthew G. Knepley       /* Construct vertices */
28939566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
2894116ded15SMatthew G. Knepley       i = 0;
2895dd400576SPatrick Sanan       if (rank == 0) {
2896116ded15SMatthew G. Knepley         for (s[0] = -1; s[0] < 2; s[0] += 2) {
2897116ded15SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
2898116ded15SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
2899116ded15SMatthew G. Knepley               for (s[3] = -1; s[3] < 2; s[3] += 2) {
2900116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[d] * vertexA[d];
2901116ded15SMatthew G. Knepley                 ++i;
2902116ded15SMatthew G. Knepley               }
2903116ded15SMatthew G. Knepley             }
2904116ded15SMatthew G. Knepley           }
2905116ded15SMatthew G. Knepley         }
2906116ded15SMatthew G. Knepley         for (p = 0; p < embedDim; ++p) {
2907116ded15SMatthew G. Knepley           s[1] = s[2] = s[3] = 1;
2908116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
2909116ded15SMatthew G. Knepley             for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertexB[(d + p) % embedDim];
2910116ded15SMatthew G. Knepley             ++i;
2911116ded15SMatthew G. Knepley           }
2912116ded15SMatthew G. Knepley         }
2913116ded15SMatthew G. Knepley         for (p = 0; p < 12; ++p) {
2914116ded15SMatthew G. Knepley           s[3] = 1;
2915116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
2916116ded15SMatthew G. Knepley             for (s[1] = -1; s[1] < 2; s[1] += 2) {
2917116ded15SMatthew G. Knepley               for (s[2] = -1; s[2] < 2; s[2] += 2) {
2918116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[evenPerm[p][d]] * vertexC[evenPerm[p][d]];
2919116ded15SMatthew G. Knepley                 ++i;
2920116ded15SMatthew G. Knepley               }
2921116ded15SMatthew G. Knepley             }
2922116ded15SMatthew G. Knepley           }
2923116ded15SMatthew G. Knepley         }
292445da822fSValeria Barra       }
292563a3b9bcSJacob Faibussowitsch       PetscCheck(i == numVerts, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertices %" PetscInt_FMT " != %" PetscInt_FMT, i, numVerts);
2926116ded15SMatthew G. Knepley       /* Construct graph */
29279566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
2928116ded15SMatthew G. Knepley       for (i = 0; i < numVerts; ++i) {
2929116ded15SMatthew G. Knepley         for (j = 0, k = 0; j < numVerts; ++j) {
29309371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
29319371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
29329371c9d4SSatish Balay             ++k;
29339371c9d4SSatish Balay           }
2934116ded15SMatthew G. Knepley         }
293563a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
2936116ded15SMatthew G. Knepley       }
2937116ded15SMatthew G. Knepley       /* Build Topology */
29389566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
293907c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
29409566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
2941116ded15SMatthew G. Knepley       /* Cells */
2942dd400576SPatrick Sanan       if (rank == 0) {
294307c565c5SJose E. Roman         for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
2944116ded15SMatthew G. Knepley           for (j = 0; j < i; ++j) {
2945116ded15SMatthew G. Knepley             for (k = 0; k < j; ++k) {
2946116ded15SMatthew G. Knepley               for (l = 0; l < k; ++l) {
29479371c9d4SSatish 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]) {
29489371c9d4SSatish Balay                   cone[0] = firstVertex + i;
29499371c9d4SSatish Balay                   cone[1] = firstVertex + j;
29509371c9d4SSatish Balay                   cone[2] = firstVertex + k;
29519371c9d4SSatish Balay                   cone[3] = firstVertex + l;
2952116ded15SMatthew G. Knepley                   /* Check orientation: https://ef.gy/linear-algebra:normal-vectors-in-higher-dimensional-spaces */
2953116ded15SMatthew G. Knepley                   {
29549371c9d4SSatish Balay                     const PetscInt epsilon[4][4][4][4] = {
29559371c9d4SSatish 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}}},
2956116ded15SMatthew G. Knepley 
29579371c9d4SSatish 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}}},
2958116ded15SMatthew G. Knepley 
29599371c9d4SSatish 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}}},
2960116ded15SMatthew G. Knepley 
29619371c9d4SSatish 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}} }
29629371c9d4SSatish Balay                     };
2963116ded15SMatthew G. Knepley                     PetscReal normal[4];
2964116ded15SMatthew G. Knepley                     PetscInt  e, f, g;
2965116ded15SMatthew G. Knepley 
2966116ded15SMatthew G. Knepley                     for (d = 0; d < embedDim; ++d) {
2967116ded15SMatthew G. Knepley                       normal[d] = 0.0;
2968116ded15SMatthew G. Knepley                       for (e = 0; e < embedDim; ++e) {
2969116ded15SMatthew G. Knepley                         for (f = 0; f < embedDim; ++f) {
2970116ded15SMatthew G. Knepley                           for (g = 0; g < embedDim; ++g) {
2971116ded15SMatthew 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]);
2972116ded15SMatthew G. Knepley                           }
2973116ded15SMatthew G. Knepley                         }
2974116ded15SMatthew G. Knepley                       }
2975116ded15SMatthew G. Knepley                     }
29769371c9d4SSatish Balay                     if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
29779371c9d4SSatish Balay                       PetscInt tmp = cone[1];
29789371c9d4SSatish Balay                       cone[1]      = cone[2];
29799371c9d4SSatish Balay                       cone[2]      = tmp;
29809371c9d4SSatish Balay                     }
2981116ded15SMatthew G. Knepley                   }
29829566063dSJacob Faibussowitsch                   PetscCall(DMPlexSetCone(dm, c++, cone));
2983116ded15SMatthew G. Knepley                 }
2984116ded15SMatthew G. Knepley               }
2985116ded15SMatthew G. Knepley             }
2986116ded15SMatthew G. Knepley           }
2987116ded15SMatthew G. Knepley         }
298845da822fSValeria Barra       }
29899566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
29909566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
29919566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
2992116ded15SMatthew G. Knepley     }
2993f4d061e9SPierre Jolivet     break;
2994d71ae5a4SJacob Faibussowitsch   default:
2995d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension for sphere: %" PetscInt_FMT, dim);
299665a81367SMatthew G. Knepley   }
299765a81367SMatthew G. Knepley   /* Create coordinates */
29989566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
29999566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
30009566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, embedDim));
30019566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVerts));
30022829fed8SMatthew G. Knepley   for (v = firstVertex; v < firstVertex + numVerts; ++v) {
30039566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, embedDim));
30049566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, embedDim));
30052829fed8SMatthew G. Knepley   }
30069566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
30079566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
30089566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
30099566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, embedDim));
30109566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
30119566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
30129566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
30139566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
30149371c9d4SSatish Balay   for (v = 0; v < numVerts; ++v)
3015ad540459SPierre Jolivet     for (d = 0; d < embedDim; ++d) coords[v * embedDim + d] = coordsIn[v * embedDim + d];
30169566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
30179566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
30189566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
30199566063dSJacob Faibussowitsch   PetscCall(PetscFree(coordsIn));
302051a74b61SMatthew G. Knepley   {
302151a74b61SMatthew G. Knepley     DM          cdm;
302251a74b61SMatthew G. Knepley     PetscDS     cds;
30239318fe57SMatthew G. Knepley     PetscScalar c = R;
302451a74b61SMatthew G. Knepley 
3025e44f6aebSMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, snapToSphere));
30269566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
30279566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
30289566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 1, &c));
302951a74b61SMatthew G. Knepley   }
303046139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
30319318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
30329566063dSJacob Faibussowitsch   if (simplex) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
30333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30349318fe57SMatthew G. Knepley }
30359318fe57SMatthew G. Knepley 
3036b7f5c055SJed Brown typedef void (*TPSEvaluateFunc)(const PetscReal[], PetscReal *, PetscReal[], PetscReal (*)[3]);
3037b7f5c055SJed Brown 
3038b7f5c055SJed Brown /*
3039b7f5c055SJed Brown  The Schwarz P implicit surface is
3040b7f5c055SJed Brown 
3041b7f5c055SJed Brown      f(x) = cos(x0) + cos(x1) + cos(x2) = 0
3042b7f5c055SJed Brown */
3043d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_SchwarzP(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3044d71ae5a4SJacob Faibussowitsch {
3045b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(y[0] * PETSC_PI), PetscCosReal(y[1] * PETSC_PI), PetscCosReal(y[2] * PETSC_PI)};
3046b7f5c055SJed Brown   PetscReal g[3] = {-PetscSinReal(y[0] * PETSC_PI), -PetscSinReal(y[1] * PETSC_PI), -PetscSinReal(y[2] * PETSC_PI)};
3047b7f5c055SJed Brown   f[0]           = c[0] + c[1] + c[2];
3048b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3049b7f5c055SJed Brown     grad[i] = PETSC_PI * g[i];
3050ad540459SPierre Jolivet     for (PetscInt j = 0; j < 3; j++) hess[i][j] = (i == j) ? -PetscSqr(PETSC_PI) * c[i] : 0.;
3051b7f5c055SJed Brown   }
3052b7f5c055SJed Brown }
3053b7f5c055SJed Brown 
30544663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3055d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_SchwarzP(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3056d71ae5a4SJacob Faibussowitsch {
3057ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) u[i] = -PETSC_PI * PetscSinReal(x[i] * PETSC_PI);
30583ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
30594663dae6SJed Brown }
30604663dae6SJed Brown 
3061b7f5c055SJed Brown /*
3062b7f5c055SJed Brown  The Gyroid implicit surface is
3063b7f5c055SJed Brown 
3064b7f5c055SJed 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)
3065b7f5c055SJed Brown 
3066b7f5c055SJed Brown */
3067d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_Gyroid(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3068d71ae5a4SJacob Faibussowitsch {
3069b7f5c055SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * y[0]), PetscSinReal(PETSC_PI * (y[1] + .5)), PetscSinReal(PETSC_PI * (y[2] + .25))};
3070b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * y[0]), PetscCosReal(PETSC_PI * (y[1] + .5)), PetscCosReal(PETSC_PI * (y[2] + .25))};
3071b7f5c055SJed Brown   f[0]           = s[0] * c[1] + s[1] * c[2] + s[2] * c[0];
3072b7f5c055SJed Brown   grad[0]        = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
3073b7f5c055SJed Brown   grad[1]        = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
3074b7f5c055SJed Brown   grad[2]        = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
3075b7f5c055SJed Brown   hess[0][0]     = -PetscSqr(PETSC_PI) * (s[0] * c[1] + s[2] * c[0]);
3076b7f5c055SJed Brown   hess[0][1]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3077b7f5c055SJed Brown   hess[0][2]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3078b7f5c055SJed Brown   hess[1][0]     = -PetscSqr(PETSC_PI) * (s[1] * c[2] + s[0] * c[1]);
3079b7f5c055SJed Brown   hess[1][1]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3080b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3081b7f5c055SJed Brown   hess[2][0]     = -PetscSqr(PETSC_PI) * (s[2] * c[0] + s[1] * c[2]);
3082b7f5c055SJed Brown   hess[2][1]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3083b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3084b7f5c055SJed Brown }
3085b7f5c055SJed Brown 
30864663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3087d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_Gyroid(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3088d71ae5a4SJacob Faibussowitsch {
30894663dae6SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * x[0]), PetscSinReal(PETSC_PI * (x[1] + .5)), PetscSinReal(PETSC_PI * (x[2] + .25))};
30904663dae6SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * x[0]), PetscCosReal(PETSC_PI * (x[1] + .5)), PetscCosReal(PETSC_PI * (x[2] + .25))};
30914663dae6SJed Brown   u[0]           = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
30924663dae6SJed Brown   u[1]           = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
30934663dae6SJed Brown   u[2]           = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
30943ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
30954663dae6SJed Brown }
30964663dae6SJed Brown 
3097b7f5c055SJed Brown /*
3098b7f5c055SJed Brown    We wish to solve
3099b7f5c055SJed Brown 
3100b7f5c055SJed Brown          min_y || y - x ||^2  subject to f(y) = 0
3101b7f5c055SJed Brown 
3102b7f5c055SJed Brown    Let g(y) = grad(f).  The minimization problem is equivalent to asking to satisfy
3103b7f5c055SJed 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
3104b7f5c055SJed Brown    tangent space and ask for both components in the tangent space to be zero.
3105b7f5c055SJed Brown 
3106b7f5c055SJed Brown    Take g to be a column vector and compute the "full QR" factorization Q R = g,
3107b7f5c055SJed Brown    where Q = I - 2 n n^T is a symmetric orthogonal matrix.
3108b7f5c055SJed Brown    The first column of Q is parallel to g so the remaining two columns span the null space.
3109b7f5c055SJed Brown    Let Qn = Q[:,1:] be those remaining columns.  Then Qn Qn^T is an orthogonal projector into the tangent space.
3110da81f932SPierre Jolivet    Since Q is symmetric, this is equivalent to multiplying by Q and taking the last two entries.
3111b7f5c055SJed Brown    In total, we have a system of 3 equations in 3 unknowns:
3112b7f5c055SJed Brown 
3113b7f5c055SJed Brown      f(y) = 0                       1 equation
3114b7f5c055SJed Brown      Qn^T (y - x) = 0               2 equations
3115b7f5c055SJed Brown 
3116b7f5c055SJed Brown    Here, we compute the residual and Jacobian of this system.
3117b7f5c055SJed Brown */
3118d71ae5a4SJacob Faibussowitsch static void TPSNearestPointResJac(TPSEvaluateFunc feval, const PetscScalar x[], const PetscScalar y[], PetscScalar res[], PetscScalar J[])
3119d71ae5a4SJacob Faibussowitsch {
3120b7f5c055SJed Brown   PetscReal yreal[3] = {PetscRealPart(y[0]), PetscRealPart(y[1]), PetscRealPart(y[2])};
3121b7f5c055SJed Brown   PetscReal d[3]     = {PetscRealPart(y[0] - x[0]), PetscRealPart(y[1] - x[1]), PetscRealPart(y[2] - x[2])};
31222f0490c0SSatish Balay   PetscReal f, grad[3], n[3], norm, norm_y[3], nd, nd_y[3], sign;
31239371c9d4SSatish Balay   PetscReal n_y[3][3] = {
31249371c9d4SSatish Balay     {0, 0, 0},
31259371c9d4SSatish Balay     {0, 0, 0},
31269371c9d4SSatish Balay     {0, 0, 0}
31279371c9d4SSatish Balay   };
3128b7f5c055SJed Brown 
3129b7f5c055SJed Brown   feval(yreal, &f, grad, n_y);
3130b7f5c055SJed Brown 
3131b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n[i] = grad[i];
3132b7f5c055SJed Brown   norm = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3133ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) norm_y[i] = 1. / norm * n[i] * n_y[i][i];
3134b7f5c055SJed Brown 
3135b7f5c055SJed Brown   // Define the Householder reflector
3136b7f5c055SJed Brown   sign = n[0] >= 0 ? 1. : -1.;
3137b7f5c055SJed Brown   n[0] += norm * sign;
3138b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n_y[0][i] += norm_y[i] * sign;
3139b7f5c055SJed Brown 
3140b7f5c055SJed Brown   norm      = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3141b7f5c055SJed Brown   norm_y[0] = 1. / norm * (n[0] * n_y[0][0]);
3142b7f5c055SJed Brown   norm_y[1] = 1. / norm * (n[0] * n_y[0][1] + n[1] * n_y[1][1]);
3143b7f5c055SJed Brown   norm_y[2] = 1. / norm * (n[0] * n_y[0][2] + n[2] * n_y[2][2]);
3144b7f5c055SJed Brown 
3145b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3146b7f5c055SJed Brown     n[i] /= norm;
3147b7f5c055SJed Brown     for (PetscInt j = 0; j < 3; j++) {
3148b7f5c055SJed Brown       // note that n[i] is n_old[i]/norm when executing the code below
3149b7f5c055SJed Brown       n_y[i][j] = n_y[i][j] / norm - n[i] / norm * norm_y[j];
3150b7f5c055SJed Brown     }
3151b7f5c055SJed Brown   }
3152b7f5c055SJed Brown 
3153b7f5c055SJed Brown   nd = n[0] * d[0] + n[1] * d[1] + n[2] * d[2];
3154b7f5c055SJed 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];
3155b7f5c055SJed Brown 
3156b7f5c055SJed Brown   res[0] = f;
3157b7f5c055SJed Brown   res[1] = d[1] - 2 * n[1] * nd;
3158b7f5c055SJed Brown   res[2] = d[2] - 2 * n[2] * nd;
3159b7f5c055SJed Brown   // J[j][i] is J_{ij} (column major)
3160b7f5c055SJed Brown   for (PetscInt j = 0; j < 3; j++) {
3161b7f5c055SJed Brown     J[0 + j * 3] = grad[j];
3162b7f5c055SJed Brown     J[1 + j * 3] = (j == 1) * 1. - 2 * (n_y[1][j] * nd + n[1] * nd_y[j]);
3163b7f5c055SJed Brown     J[2 + j * 3] = (j == 2) * 1. - 2 * (n_y[2][j] * nd + n[2] * nd_y[j]);
3164b7f5c055SJed Brown   }
3165b7f5c055SJed Brown }
3166b7f5c055SJed Brown 
3167b7f5c055SJed Brown /*
3168b7f5c055SJed Brown    Project x to the nearest point on the implicit surface using Newton's method.
3169b7f5c055SJed Brown */
3170d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSNearestPoint(TPSEvaluateFunc feval, PetscScalar x[])
3171d71ae5a4SJacob Faibussowitsch {
3172b7f5c055SJed Brown   PetscScalar y[3] = {x[0], x[1], x[2]}; // Initial guess
3173b7f5c055SJed Brown 
3174b7f5c055SJed Brown   PetscFunctionBegin;
3175b7f5c055SJed Brown   for (PetscInt iter = 0; iter < 10; iter++) {
3176b7f5c055SJed Brown     PetscScalar res[3], J[9];
3177b7f5c055SJed Brown     PetscReal   resnorm;
3178b7f5c055SJed Brown     TPSNearestPointResJac(feval, x, y, res, J);
3179b7f5c055SJed Brown     resnorm = PetscSqrtReal(PetscSqr(PetscRealPart(res[0])) + PetscSqr(PetscRealPart(res[1])) + PetscSqr(PetscRealPart(res[2])));
3180b7f5c055SJed Brown     if (0) { // Turn on this monitor if you need to confirm quadratic convergence
318163a3b9bcSJacob 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])));
3182b7f5c055SJed Brown     }
3183b7f5c055SJed Brown     if (resnorm < PETSC_SMALL) break;
3184b7f5c055SJed Brown 
3185b7f5c055SJed Brown     // Take the Newton step
31869566063dSJacob Faibussowitsch     PetscCall(PetscKernel_A_gets_inverse_A_3(J, 0., PETSC_FALSE, NULL));
3187b7f5c055SJed Brown     PetscKernel_v_gets_v_minus_A_times_w_3(y, J, res);
3188b7f5c055SJed Brown   }
3189b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) x[i] = y[i];
31903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3191b7f5c055SJed Brown }
3192b7f5c055SJed Brown 
3193b7f5c055SJed Brown const char *const DMPlexTPSTypes[] = {"SCHWARZ_P", "GYROID", "DMPlexTPSType", "DMPLEX_TPS_", NULL};
3194b7f5c055SJed Brown 
3195d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateTPSMesh_Internal(DM dm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness)
3196d71ae5a4SJacob Faibussowitsch {
3197b7f5c055SJed Brown   PetscMPIInt rank;
3198b7f5c055SJed Brown   PetscInt    topoDim = 2, spaceDim = 3, numFaces = 0, numVertices = 0, numEdges = 0;
3199b7f5c055SJed Brown   PetscInt(*edges)[2] = NULL, *edgeSets = NULL;
3200b7f5c055SJed Brown   PetscInt           *cells_flat = NULL;
3201b7f5c055SJed Brown   PetscReal          *vtxCoords  = NULL;
3202b7f5c055SJed Brown   TPSEvaluateFunc     evalFunc   = NULL;
32038434afd1SBarry Smith   PetscSimplePointFn *normalFunc = NULL;
3204b7f5c055SJed Brown   DMLabel             label;
3205b7f5c055SJed Brown 
3206b7f5c055SJed Brown   PetscFunctionBegin;
320746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
32089566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
320963a3b9bcSJacob 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);
3210b7f5c055SJed Brown   switch (tpstype) {
3211b7f5c055SJed Brown   case DMPLEX_TPS_SCHWARZ_P:
3212b7f5c055SJed 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");
3213c5853193SPierre Jolivet     if (rank == 0) {
3214b7f5c055SJed Brown       PetscInt(*cells)[6][4][4] = NULL; // [junction, junction-face, cell, conn]
3215b7f5c055SJed Brown       PetscInt  Njunctions = 0, Ncuts = 0, Npipes[3], vcount;
3216b7f5c055SJed Brown       PetscReal L = 1;
3217b7f5c055SJed Brown 
3218b7f5c055SJed Brown       Npipes[0]   = (extent[0] + 1) * extent[1] * extent[2];
3219b7f5c055SJed Brown       Npipes[1]   = extent[0] * (extent[1] + 1) * extent[2];
3220b7f5c055SJed Brown       Npipes[2]   = extent[0] * extent[1] * (extent[2] + 1);
3221b7f5c055SJed Brown       Njunctions  = extent[0] * extent[1] * extent[2];
3222b7f5c055SJed Brown       Ncuts       = 2 * (extent[0] * extent[1] + extent[1] * extent[2] + extent[2] * extent[0]);
3223b7f5c055SJed Brown       numVertices = 4 * (Npipes[0] + Npipes[1] + Npipes[2]) + 8 * Njunctions;
32249566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(3 * numVertices, &vtxCoords));
32259566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Njunctions, &cells));
32269566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edges));
32279566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edgeSets));
3228b7f5c055SJed Brown       // x-normal pipes
3229b7f5c055SJed Brown       vcount = 0;
3230b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0] + 1; i++) {
3231b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3232b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3233b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3234b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * i - 1) * L;
3235b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3236b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3237b7f5c055SJed Brown             }
3238b7f5c055SJed Brown           }
3239b7f5c055SJed Brown         }
3240b7f5c055SJed Brown       }
3241b7f5c055SJed Brown       // y-normal pipes
3242b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3243b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1] + 1; j++) {
3244b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3245b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3246b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3247b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * j - 1) * L;
3248b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3249b7f5c055SJed Brown             }
3250b7f5c055SJed Brown           }
3251b7f5c055SJed Brown         }
3252b7f5c055SJed Brown       }
3253b7f5c055SJed Brown       // z-normal pipes
3254b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3255b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3256b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2] + 1; k++) {
3257b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3258b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3259b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3260b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * k - 1) * L;
3261b7f5c055SJed Brown             }
3262b7f5c055SJed Brown           }
3263b7f5c055SJed Brown         }
3264b7f5c055SJed Brown       }
3265b7f5c055SJed Brown       // junctions
3266b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3267b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3268b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3269b7f5c055SJed Brown             const PetscInt J = (i * extent[1] + j) * extent[2] + k, Jvoff = (Npipes[0] + Npipes[1] + Npipes[2]) * 4 + J * 8;
3270b7f5c055SJed Brown             PetscCheck(vcount / 3 == Jvoff, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected vertex count");
3271b7f5c055SJed Brown             for (PetscInt ii = 0; ii < 2; ii++) {
3272b7f5c055SJed Brown               for (PetscInt jj = 0; jj < 2; jj++) {
3273b7f5c055SJed Brown                 for (PetscInt kk = 0; kk < 2; kk++) {
3274b7f5c055SJed Brown                   double Ls           = (1 - sqrt(2) / 4) * L;
3275b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * i * L + (2 * ii - 1) * Ls;
3276b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * j * L + (2 * jj - 1) * Ls;
3277b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * k * L + (2 * kk - 1) * Ls;
3278b7f5c055SJed Brown                 }
3279b7f5c055SJed Brown               }
3280b7f5c055SJed Brown             }
3281b7f5c055SJed Brown             const PetscInt jfaces[3][2][4] = {
3282b7f5c055SJed Brown               {{3, 1, 0, 2}, {7, 5, 4, 6}}, // x-aligned
3283b7f5c055SJed Brown               {{5, 4, 0, 1}, {7, 6, 2, 3}}, // y-aligned
3284b7f5c055SJed Brown               {{6, 2, 0, 4}, {7, 3, 1, 5}}  // z-aligned
3285b7f5c055SJed Brown             };
3286b7f5c055SJed Brown             const PetscInt pipe_lo[3] = {// vertex numbers of pipes
32879371c9d4SSatish 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};
3288b7f5c055SJed Brown             const PetscInt pipe_hi[3] = {// vertex numbers of pipes
32899371c9d4SSatish 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};
3290b7f5c055SJed Brown             for (PetscInt dir = 0; dir < 3; dir++) { // x,y,z
3291b7f5c055SJed Brown               const PetscInt ijk[3] = {i, j, k};
3292b7f5c055SJed Brown               for (PetscInt l = 0; l < 4; l++) { // rotations
3293b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][0] = pipe_lo[dir] + l;
3294b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][1] = Jvoff + jfaces[dir][0][l];
3295b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][2] = Jvoff + jfaces[dir][0][(l - 1 + 4) % 4];
3296b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][3] = pipe_lo[dir] + (l - 1 + 4) % 4;
3297b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][0] = Jvoff + jfaces[dir][1][l];
3298b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][1] = pipe_hi[dir] + l;
3299b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][2] = pipe_hi[dir] + (l - 1 + 4) % 4;
3300b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][3] = Jvoff + jfaces[dir][1][(l - 1 + 4) % 4];
3301b7f5c055SJed Brown                 if (ijk[dir] == 0) {
3302b7f5c055SJed Brown                   edges[numEdges][0] = pipe_lo[dir] + l;
3303b7f5c055SJed Brown                   edges[numEdges][1] = pipe_lo[dir] + (l + 1) % 4;
3304b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 1;
3305b7f5c055SJed Brown                   numEdges++;
3306b7f5c055SJed Brown                 }
3307b7f5c055SJed Brown                 if (ijk[dir] + 1 == extent[dir]) {
3308b7f5c055SJed Brown                   edges[numEdges][0] = pipe_hi[dir] + l;
3309b7f5c055SJed Brown                   edges[numEdges][1] = pipe_hi[dir] + (l + 1) % 4;
3310b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 2;
3311b7f5c055SJed Brown                   numEdges++;
3312b7f5c055SJed Brown                 }
3313b7f5c055SJed Brown               }
3314b7f5c055SJed Brown             }
3315b7f5c055SJed Brown           }
3316b7f5c055SJed Brown         }
3317b7f5c055SJed Brown       }
331863a3b9bcSJacob Faibussowitsch       PetscCheck(numEdges == Ncuts * 4, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Edge count %" PetscInt_FMT " incompatible with number of cuts %" PetscInt_FMT, numEdges, Ncuts);
3319b7f5c055SJed Brown       numFaces   = 24 * Njunctions;
3320b7f5c055SJed Brown       cells_flat = cells[0][0][0];
3321b7f5c055SJed Brown     }
3322b7f5c055SJed Brown     evalFunc   = TPSEvaluate_SchwarzP;
33234663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_SchwarzP;
3324b7f5c055SJed Brown     break;
3325b7f5c055SJed Brown   case DMPLEX_TPS_GYROID:
3326c5853193SPierre Jolivet     if (rank == 0) {
3327b7f5c055SJed Brown       // This is a coarse mesh approximation of the gyroid shifted to being the zero of the level set
3328b7f5c055SJed Brown       //
3329b7f5c055SJed 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)
3330b7f5c055SJed Brown       //
3331b7f5c055SJed Brown       // on the cell [0,2]^3.
3332b7f5c055SJed Brown       //
3333b7f5c055SJed Brown       // Think about dividing that cell into four columns, and focus on the column [0,1]x[0,1]x[0,2].
3334b7f5c055SJed Brown       // If you looked at the gyroid in that column at different slices of z you would see that it kind of spins
3335b7f5c055SJed Brown       // like a boomerang:
3336b7f5c055SJed Brown       //
3337b7f5c055SJed Brown       //     z = 0          z = 1/4        z = 1/2        z = 3/4     //
3338b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3339b7f5c055SJed Brown       //                                                              //
3340b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3341b7f5c055SJed Brown       //      \                                   /            \      //
3342b7f5c055SJed Brown       //       \            `-_   _-'            /              }     //
3343b7f5c055SJed Brown       //        *-_            `-'            _-'              /      //
3344b7f5c055SJed Brown       //     +     `-+      +       +      +-'     +      +   /   +   //
3345b7f5c055SJed Brown       //                                                              //
3346b7f5c055SJed Brown       //                                                              //
3347b7f5c055SJed Brown       //     z = 1          z = 5/4        z = 3/2        z = 7/4     //
3348b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3349b7f5c055SJed Brown       //                                                              //
3350b7f5c055SJed Brown       //     +-_     +      +       +      +     _-+      +   /   +   //
3351b7f5c055SJed Brown       //        `-_            _-_            _-`            /        //
3352b7f5c055SJed Brown       //           \        _-'   `-_        /              {         //
3353b7f5c055SJed Brown       //            \                       /                \        //
3354b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3355b7f5c055SJed Brown       //
3356b7f5c055SJed Brown       //
3357b7f5c055SJed Brown       // This course mesh approximates each of these slices by two line segments,
3358b7f5c055SJed Brown       // and then connects the segments in consecutive layers with quadrilateral faces.
3359b7f5c055SJed Brown       // All of the end points of the segments are multiples of 1/4 except for the
3360b7f5c055SJed Brown       // point * in the picture for z = 0 above and the similar points in other layers.
3361b7f5c055SJed Brown       // That point is at (gamma, gamma, 0), where gamma is calculated below.
3362b7f5c055SJed Brown       //
3363b7f5c055SJed Brown       // The column  [1,2]x[1,2]x[0,2] looks the same as this column;
3364b7f5c055SJed Brown       // The columns [1,2]x[0,1]x[0,2] and [0,1]x[1,2]x[0,2] are mirror images.
3365b7f5c055SJed Brown       //
3366b7f5c055SJed Brown       // As for how this method turned into the names given to the vertices:
3367b7f5c055SJed Brown       // that was not systematic, it was just the way it worked out in my handwritten notes.
3368b7f5c055SJed Brown 
3369b7f5c055SJed Brown       PetscInt facesPerBlock = 64;
3370b7f5c055SJed Brown       PetscInt vertsPerBlock = 56;
3371b7f5c055SJed Brown       PetscInt extentPlus[3];
3372b7f5c055SJed Brown       PetscInt numBlocks, numBlocksPlus;
33739371c9d4SSatish 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;
33749371c9d4SSatish Balay       const PetscInt pattern[64][4] = {
33759371c9d4SSatish Balay         /* face to vertex within the coarse discretization of a single gyroid block */
3376b7f5c055SJed Brown         /* layer 0 */
33779371c9d4SSatish Balay         {A,           C,           K,           G          },
33789371c9d4SSatish Balay         {C,           B,           II,          K          },
33799371c9d4SSatish Balay         {D,           A,           H,           L          },
33809371c9d4SSatish Balay         {B + 56 * 1,  D,           L,           J          },
33819371c9d4SSatish Balay         {E,           B + 56 * 1,  J,           N          },
33829371c9d4SSatish Balay         {A + 56 * 2,  E,           N,           H + 56 * 2 },
33839371c9d4SSatish Balay         {F,           A + 56 * 2,  G + 56 * 2,  M          },
33849371c9d4SSatish Balay         {B,           F,           M,           II         },
3385b7f5c055SJed Brown         /* layer 1 */
33869371c9d4SSatish Balay         {G,           K,           Q,           O          },
33879371c9d4SSatish Balay         {K,           II,          P,           Q          },
33889371c9d4SSatish Balay         {L,           H,           O + 56 * 1,  R          },
33899371c9d4SSatish Balay         {J,           L,           R,           P          },
33909371c9d4SSatish Balay         {N,           J,           P,           S          },
33919371c9d4SSatish Balay         {H + 56 * 2,  N,           S,           O + 56 * 3 },
33929371c9d4SSatish Balay         {M,           G + 56 * 2,  O + 56 * 2,  T          },
33939371c9d4SSatish Balay         {II,          M,           T,           P          },
3394b7f5c055SJed Brown         /* layer 2 */
33959371c9d4SSatish Balay         {O,           Q,           Y,           U          },
33969371c9d4SSatish Balay         {Q,           P,           W,           Y          },
33979371c9d4SSatish Balay         {R,           O + 56 * 1,  U + 56 * 1,  Ap         },
33989371c9d4SSatish Balay         {P,           R,           Ap,          W          },
33999371c9d4SSatish Balay         {S,           P,           X,           Bp         },
34009371c9d4SSatish Balay         {O + 56 * 3,  S,           Bp,          V + 56 * 1 },
34019371c9d4SSatish Balay         {T,           O + 56 * 2,  V,           Z          },
34029371c9d4SSatish Balay         {P,           T,           Z,           X          },
3403b7f5c055SJed Brown         /* layer 3 */
34049371c9d4SSatish Balay         {U,           Y,           Ep,          Dp         },
34059371c9d4SSatish Balay         {Y,           W,           Cp,          Ep         },
34069371c9d4SSatish Balay         {Ap,          U + 56 * 1,  Dp + 56 * 1, Gp         },
34079371c9d4SSatish Balay         {W,           Ap,          Gp,          Cp         },
34089371c9d4SSatish Balay         {Bp,          X,           Cp + 56 * 2, Fp         },
34099371c9d4SSatish Balay         {V + 56 * 1,  Bp,          Fp,          Dp + 56 * 1},
34109371c9d4SSatish Balay         {Z,           V,           Dp,          Hp         },
34119371c9d4SSatish Balay         {X,           Z,           Hp,          Cp + 56 * 2},
3412b7f5c055SJed Brown         /* layer 4 */
34139371c9d4SSatish Balay         {Dp,          Ep,          Mp,          Kp         },
34149371c9d4SSatish Balay         {Ep,          Cp,          Ip,          Mp         },
34159371c9d4SSatish Balay         {Gp,          Dp + 56 * 1, Lp,          Np         },
34169371c9d4SSatish Balay         {Cp,          Gp,          Np,          Jp         },
34179371c9d4SSatish Balay         {Fp,          Cp + 56 * 2, Jp + 56 * 2, Pp         },
34189371c9d4SSatish Balay         {Dp + 56 * 1, Fp,          Pp,          Lp         },
34199371c9d4SSatish Balay         {Hp,          Dp,          Kp,          Op         },
34209371c9d4SSatish Balay         {Cp + 56 * 2, Hp,          Op,          Ip + 56 * 2},
3421b7f5c055SJed Brown         /* layer 5 */
34229371c9d4SSatish Balay         {Kp,          Mp,          Sp,          Rp         },
34239371c9d4SSatish Balay         {Mp,          Ip,          Qp,          Sp         },
34249371c9d4SSatish Balay         {Np,          Lp,          Rp,          Tp         },
34259371c9d4SSatish Balay         {Jp,          Np,          Tp,          Qp + 56 * 1},
34269371c9d4SSatish Balay         {Pp,          Jp + 56 * 2, Qp + 56 * 3, Up         },
34279371c9d4SSatish Balay         {Lp,          Pp,          Up,          Rp         },
34289371c9d4SSatish Balay         {Op,          Kp,          Rp,          Vp         },
34299371c9d4SSatish Balay         {Ip + 56 * 2, Op,          Vp,          Qp + 56 * 2},
3430b7f5c055SJed Brown         /* layer 6 */
34319371c9d4SSatish Balay         {Rp,          Sp,          Aq,          Yp         },
34329371c9d4SSatish Balay         {Sp,          Qp,          Wp,          Aq         },
34339371c9d4SSatish Balay         {Tp,          Rp,          Yp,          Cq         },
34349371c9d4SSatish Balay         {Qp + 56 * 1, Tp,          Cq,          Wp + 56 * 1},
34359371c9d4SSatish Balay         {Up,          Qp + 56 * 3, Xp + 56 * 1, Dq         },
34369371c9d4SSatish Balay         {Rp,          Up,          Dq,          Zp         },
34379371c9d4SSatish Balay         {Vp,          Rp,          Zp,          Bq         },
34389371c9d4SSatish Balay         {Qp + 56 * 2, Vp,          Bq,          Xp         },
3439b7f5c055SJed Brown         /* layer 7 (the top is the periodic image of the bottom of layer 0) */
34409371c9d4SSatish Balay         {Yp,          Aq,          C + 56 * 4,  A + 56 * 4 },
34419371c9d4SSatish Balay         {Aq,          Wp,          B + 56 * 4,  C + 56 * 4 },
34429371c9d4SSatish Balay         {Cq,          Yp,          A + 56 * 4,  D + 56 * 4 },
34439371c9d4SSatish Balay         {Wp + 56 * 1, Cq,          D + 56 * 4,  B + 56 * 5 },
34449371c9d4SSatish Balay         {Dq,          Xp + 56 * 1, B + 56 * 5,  E + 56 * 4 },
34459371c9d4SSatish Balay         {Zp,          Dq,          E + 56 * 4,  A + 56 * 6 },
34469371c9d4SSatish Balay         {Bq,          Zp,          A + 56 * 6,  F + 56 * 4 },
34479371c9d4SSatish Balay         {Xp,          Bq,          F + 56 * 4,  B + 56 * 4 }
3448b7f5c055SJed Brown       };
3449b7f5c055SJed Brown       const PetscReal gamma                = PetscAcosReal((PetscSqrtReal(3.) - 1.) / PetscSqrtReal(2.)) / PETSC_PI;
34509371c9d4SSatish Balay       const PetscReal patternCoords[56][3] = {
3451bee3fc89SBarry Smith         {1.,        0.,        0.  }, /* A  */
3452bee3fc89SBarry Smith         {0.,        1.,        0.  }, /* B  */
3453bee3fc89SBarry Smith         {gamma,     gamma,     0.  }, /* C  */
3454bee3fc89SBarry Smith         {1 + gamma, 1 - gamma, 0.  }, /* D  */
3455bee3fc89SBarry Smith         {2 - gamma, 2 - gamma, 0.  }, /* E  */
3456bee3fc89SBarry Smith         {1 - gamma, 1 + gamma, 0.  }, /* F  */
3457b7f5c055SJed Brown 
3458bee3fc89SBarry Smith         {.5,        0,         .25 }, /* G  */
3459bee3fc89SBarry Smith         {1.5,       0.,        .25 }, /* H  */
3460bee3fc89SBarry Smith         {.5,        1.,        .25 }, /* II */
3461bee3fc89SBarry Smith         {1.5,       1.,        .25 }, /* J  */
3462bee3fc89SBarry Smith         {.25,       .5,        .25 }, /* K  */
3463bee3fc89SBarry Smith         {1.25,      .5,        .25 }, /* L  */
3464bee3fc89SBarry Smith         {.75,       1.5,       .25 }, /* M  */
3465bee3fc89SBarry Smith         {1.75,      1.5,       .25 }, /* N  */
3466b7f5c055SJed Brown 
3467bee3fc89SBarry Smith         {0.,        0.,        .5  }, /* O  */
3468bee3fc89SBarry Smith         {1.,        1.,        .5  }, /* P  */
3469bee3fc89SBarry Smith         {gamma,     1 - gamma, .5  }, /* Q  */
3470bee3fc89SBarry Smith         {1 + gamma, gamma,     .5  }, /* R  */
3471bee3fc89SBarry Smith         {2 - gamma, 1 + gamma, .5  }, /* S  */
3472bee3fc89SBarry Smith         {1 - gamma, 2 - gamma, .5  }, /* T  */
3473b7f5c055SJed Brown 
3474bee3fc89SBarry Smith         {0.,        .5,        .75 }, /* U  */
3475bee3fc89SBarry Smith         {0.,        1.5,       .75 }, /* V  */
3476bee3fc89SBarry Smith         {1.,        .5,        .75 }, /* W  */
3477bee3fc89SBarry Smith         {1.,        1.5,       .75 }, /* X  */
3478bee3fc89SBarry Smith         {.5,        .75,       .75 }, /* Y  */
3479bee3fc89SBarry Smith         {.5,        1.75,      .75 }, /* Z  */
3480bee3fc89SBarry Smith         {1.5,       .25,       .75 }, /* Ap */
3481bee3fc89SBarry Smith         {1.5,       1.25,      .75 }, /* Bp */
3482b7f5c055SJed Brown 
3483bee3fc89SBarry Smith         {1.,        0.,        1.  }, /* Cp */
3484bee3fc89SBarry Smith         {0.,        1.,        1.  }, /* Dp */
3485bee3fc89SBarry Smith         {1 - gamma, 1 - gamma, 1.  }, /* Ep */
3486bee3fc89SBarry Smith         {1 + gamma, 1 + gamma, 1.  }, /* Fp */
3487bee3fc89SBarry Smith         {2 - gamma, gamma,     1.  }, /* Gp */
3488bee3fc89SBarry Smith         {gamma,     2 - gamma, 1.  }, /* Hp */
3489b7f5c055SJed Brown 
3490bee3fc89SBarry Smith         {.5,        0.,        1.25}, /* Ip */
3491bee3fc89SBarry Smith         {1.5,       0.,        1.25}, /* Jp */
3492bee3fc89SBarry Smith         {.5,        1.,        1.25}, /* Kp */
3493bee3fc89SBarry Smith         {1.5,       1.,        1.25}, /* Lp */
3494bee3fc89SBarry Smith         {.75,       .5,        1.25}, /* Mp */
3495bee3fc89SBarry Smith         {1.75,      .5,        1.25}, /* Np */
3496bee3fc89SBarry Smith         {.25,       1.5,       1.25}, /* Op */
3497bee3fc89SBarry Smith         {1.25,      1.5,       1.25}, /* Pp */
3498b7f5c055SJed Brown 
3499bee3fc89SBarry Smith         {0.,        0.,        1.5 }, /* Qp */
3500bee3fc89SBarry Smith         {1.,        1.,        1.5 }, /* Rp */
3501bee3fc89SBarry Smith         {1 - gamma, gamma,     1.5 }, /* Sp */
3502bee3fc89SBarry Smith         {2 - gamma, 1 - gamma, 1.5 }, /* Tp */
3503bee3fc89SBarry Smith         {1 + gamma, 2 - gamma, 1.5 }, /* Up */
3504bee3fc89SBarry Smith         {gamma,     1 + gamma, 1.5 }, /* Vp */
3505b7f5c055SJed Brown 
3506bee3fc89SBarry Smith         {0.,        .5,        1.75}, /* Wp */
3507bee3fc89SBarry Smith         {0.,        1.5,       1.75}, /* Xp */
3508bee3fc89SBarry Smith         {1.,        .5,        1.75}, /* Yp */
3509bee3fc89SBarry Smith         {1.,        1.5,       1.75}, /* Zp */
3510bee3fc89SBarry Smith         {.5,        .25,       1.75}, /* Aq */
3511bee3fc89SBarry Smith         {.5,        1.25,      1.75}, /* Bq */
3512bee3fc89SBarry Smith         {1.5,       .75,       1.75}, /* Cq */
3513bee3fc89SBarry Smith         {1.5,       1.75,      1.75}, /* Dq */
3514b7f5c055SJed Brown       };
3515b7f5c055SJed Brown       PetscInt(*cells)[64][4] = NULL;
3516b7f5c055SJed Brown       PetscBool *seen;
3517b7f5c055SJed Brown       PetscInt  *vertToTrueVert;
3518b7f5c055SJed Brown       PetscInt   count;
3519b7f5c055SJed Brown 
3520b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) extentPlus[i] = extent[i] + 1;
3521b7f5c055SJed Brown       numBlocks = 1;
3522b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocks *= extent[i];
3523b7f5c055SJed Brown       numBlocksPlus = 1;
3524b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocksPlus *= extentPlus[i];
3525b7f5c055SJed Brown       numFaces = numBlocks * facesPerBlock;
35269566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocks, &cells));
35279566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numBlocksPlus * vertsPerBlock, &seen));
3528b7f5c055SJed Brown       for (PetscInt k = 0; k < extent[2]; k++) {
3529b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3530b7f5c055SJed Brown           for (PetscInt i = 0; i < extent[0]; i++) {
3531b7f5c055SJed Brown             for (PetscInt f = 0; f < facesPerBlock; f++) {
3532b7f5c055SJed Brown               for (PetscInt v = 0; v < 4; v++) {
3533b7f5c055SJed Brown                 PetscInt vertRaw     = pattern[f][v];
3534b7f5c055SJed Brown                 PetscInt blockidx    = vertRaw / 56;
3535b7f5c055SJed Brown                 PetscInt patternvert = vertRaw % 56;
3536b7f5c055SJed Brown                 PetscInt xplus       = (blockidx & 1);
3537b7f5c055SJed Brown                 PetscInt yplus       = (blockidx & 2) >> 1;
3538b7f5c055SJed Brown                 PetscInt zplus       = (blockidx & 4) >> 2;
3539b7f5c055SJed Brown                 PetscInt zcoord      = (periodic && periodic[2] == DM_BOUNDARY_PERIODIC) ? ((k + zplus) % extent[2]) : (k + zplus);
3540b7f5c055SJed Brown                 PetscInt ycoord      = (periodic && periodic[1] == DM_BOUNDARY_PERIODIC) ? ((j + yplus) % extent[1]) : (j + yplus);
3541b7f5c055SJed Brown                 PetscInt xcoord      = (periodic && periodic[0] == DM_BOUNDARY_PERIODIC) ? ((i + xplus) % extent[0]) : (i + xplus);
3542b7f5c055SJed Brown                 PetscInt vert        = ((zcoord * extentPlus[1] + ycoord) * extentPlus[0] + xcoord) * 56 + patternvert;
3543b7f5c055SJed Brown 
3544b7f5c055SJed Brown                 cells[(k * extent[1] + j) * extent[0] + i][f][v] = vert;
3545b7f5c055SJed Brown                 seen[vert]                                       = PETSC_TRUE;
3546b7f5c055SJed Brown               }
3547b7f5c055SJed Brown             }
3548b7f5c055SJed Brown           }
3549b7f5c055SJed Brown         }
3550b7f5c055SJed Brown       }
35519371c9d4SSatish Balay       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++)
35529371c9d4SSatish Balay         if (seen[i]) numVertices++;
3553b7f5c055SJed Brown       count = 0;
35549566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocksPlus * vertsPerBlock, &vertToTrueVert));
35559566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * 3, &vtxCoords));
3556b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++) vertToTrueVert[i] = -1;
3557b7f5c055SJed Brown       for (PetscInt k = 0; k < extentPlus[2]; k++) {
3558b7f5c055SJed Brown         for (PetscInt j = 0; j < extentPlus[1]; j++) {
3559b7f5c055SJed Brown           for (PetscInt i = 0; i < extentPlus[0]; i++) {
3560b7f5c055SJed Brown             for (PetscInt v = 0; v < vertsPerBlock; v++) {
3561b7f5c055SJed Brown               PetscInt vIdx = ((k * extentPlus[1] + j) * extentPlus[0] + i) * vertsPerBlock + v;
3562b7f5c055SJed Brown 
3563b7f5c055SJed Brown               if (seen[vIdx]) {
3564b7f5c055SJed Brown                 PetscInt thisVert;
3565b7f5c055SJed Brown 
3566b7f5c055SJed Brown                 vertToTrueVert[vIdx] = thisVert = count++;
3567b7f5c055SJed Brown 
3568b7f5c055SJed Brown                 for (PetscInt d = 0; d < 3; d++) vtxCoords[3 * thisVert + d] = patternCoords[v][d];
3569b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 0] += i * 2;
3570b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 1] += j * 2;
3571b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 2] += k * 2;
3572b7f5c055SJed Brown               }
3573b7f5c055SJed Brown             }
3574b7f5c055SJed Brown           }
3575b7f5c055SJed Brown         }
3576b7f5c055SJed Brown       }
3577b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocks; i++) {
3578b7f5c055SJed Brown         for (PetscInt f = 0; f < facesPerBlock; f++) {
3579ad540459SPierre Jolivet           for (PetscInt v = 0; v < 4; v++) cells[i][f][v] = vertToTrueVert[cells[i][f][v]];
3580b7f5c055SJed Brown         }
3581b7f5c055SJed Brown       }
35829566063dSJacob Faibussowitsch       PetscCall(PetscFree(vertToTrueVert));
35839566063dSJacob Faibussowitsch       PetscCall(PetscFree(seen));
3584b7f5c055SJed Brown       cells_flat = cells[0][0];
3585b7f5c055SJed Brown       numEdges   = 0;
3586b7f5c055SJed Brown       for (PetscInt i = 0; i < numFaces; i++) {
3587b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3588b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3589b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3590b7f5c055SJed Brown 
3591b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3592b7f5c055SJed Brown             if (!periodic || periodic[0] != DM_BOUNDARY_PERIODIC) {
3593b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) numEdges++;
3594b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) numEdges++;
3595b7f5c055SJed Brown             }
3596b7f5c055SJed Brown           }
3597b7f5c055SJed Brown         }
3598b7f5c055SJed Brown       }
35999566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edges));
36009566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edgeSets));
3601b7f5c055SJed Brown       for (PetscInt edge = 0, i = 0; i < numFaces; i++) {
3602b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
3603b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
3604b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
3605b7f5c055SJed Brown 
3606b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
3607b7f5c055SJed Brown             if (!periodic || periodic[d] != DM_BOUNDARY_PERIODIC) {
3608b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) {
3609b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3610b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3611b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d;
3612b7f5c055SJed Brown               }
3613b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) {
3614b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
3615b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
3616b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d + 1;
3617b7f5c055SJed Brown               }
3618b7f5c055SJed Brown             }
3619b7f5c055SJed Brown           }
3620b7f5c055SJed Brown         }
3621b7f5c055SJed Brown       }
3622b7f5c055SJed Brown     }
3623b7f5c055SJed Brown     evalFunc   = TPSEvaluate_Gyroid;
36244663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_Gyroid;
3625b7f5c055SJed Brown     break;
3626b7f5c055SJed Brown   }
3627b7f5c055SJed Brown 
36289566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, topoDim));
3629c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(dm, numFaces, numVertices, 4, cells_flat));
36309566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(dm, 0, 0, 0, NULL));
36319566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells_flat));
3632b7f5c055SJed Brown   {
3633b7f5c055SJed Brown     DM idm;
36349566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(dm, &idm));
363569d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &idm));
3636b7f5c055SJed Brown   }
3637c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, vtxCoords));
36389566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, NULL));
36399566063dSJacob Faibussowitsch   PetscCall(PetscFree(vtxCoords));
3640b7f5c055SJed Brown 
36419566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
36429566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
3643b7f5c055SJed Brown   for (PetscInt e = 0; e < numEdges; e++) {
3644b7f5c055SJed Brown     PetscInt        njoin;
3645b7f5c055SJed Brown     const PetscInt *join, verts[] = {numFaces + edges[e][0], numFaces + edges[e][1]};
36469566063dSJacob Faibussowitsch     PetscCall(DMPlexGetJoin(dm, 2, verts, &njoin, &join));
364763a3b9bcSJacob 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]);
36489566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(label, join[0], edgeSets[e]));
36499566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(dm, 2, verts, &njoin, &join));
3650b7f5c055SJed Brown   }
36519566063dSJacob Faibussowitsch   PetscCall(PetscFree(edges));
36529566063dSJacob Faibussowitsch   PetscCall(PetscFree(edgeSets));
36531436d7faSJed Brown   if (tps_distribute) {
36541436d7faSJed Brown     DM               pdm = NULL;
36551436d7faSJed Brown     PetscPartitioner part;
36561436d7faSJed Brown 
36579566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
36589566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
36599566063dSJacob Faibussowitsch     PetscCall(DMPlexDistribute(dm, 0, NULL, &pdm));
366048a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
36611436d7faSJed Brown     // Do not auto-distribute again
36629566063dSJacob Faibussowitsch     PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
36631436d7faSJed Brown   }
3664b7f5c055SJed Brown 
36659566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
3666b7f5c055SJed Brown   for (PetscInt refine = 0; refine < refinements; refine++) {
3667b7f5c055SJed Brown     PetscInt     m;
3668b7f5c055SJed Brown     DM           dmf;
3669b7f5c055SJed Brown     Vec          X;
3670b7f5c055SJed Brown     PetscScalar *x;
36719566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, MPI_COMM_NULL, &dmf));
367269d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmf));
3673b7f5c055SJed Brown 
36749566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &X));
36759566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(X, &m));
36769566063dSJacob Faibussowitsch     PetscCall(VecGetArray(X, &x));
367748a46eb9SPierre Jolivet     for (PetscInt i = 0; i < m; i += 3) PetscCall(TPSNearestPoint(evalFunc, &x[i]));
36789566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(X, &x));
3679b7f5c055SJed Brown   }
3680b7f5c055SJed Brown 
3681b7f5c055SJed Brown   // Face Sets has already been propagated to new vertices during refinement; this propagates to the initial vertices.
36829566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
36839566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, label));
3684b7f5c055SJed Brown 
368546139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
368646139095SJed Brown 
3687b7f5c055SJed Brown   if (thickness > 0) {
36884663dae6SJed Brown     DM              edm, cdm, ecdm;
36894663dae6SJed Brown     DMPlexTransform tr;
36904663dae6SJed Brown     const char     *prefix;
36914663dae6SJed Brown     PetscOptions    options;
36924663dae6SJed Brown     // Code from DMPlexExtrude
36934663dae6SJed Brown     PetscCall(DMPlexTransformCreate(PetscObjectComm((PetscObject)dm), &tr));
36944663dae6SJed Brown     PetscCall(DMPlexTransformSetDM(tr, dm));
36954663dae6SJed Brown     PetscCall(DMPlexTransformSetType(tr, DMPLEXEXTRUDE));
36964663dae6SJed Brown     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
36974663dae6SJed Brown     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tr, prefix));
36984663dae6SJed Brown     PetscCall(PetscObjectGetOptions((PetscObject)dm, &options));
36994663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, options));
37004663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetLayers(tr, layers));
37014663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetThickness(tr, thickness));
37024663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetTensor(tr, PETSC_FALSE));
37034663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetSymmetric(tr, PETSC_TRUE));
37044663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetNormalFunction(tr, normalFunc));
37054663dae6SJed Brown     PetscCall(DMPlexTransformSetFromOptions(tr));
37064663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, NULL));
37074663dae6SJed Brown     PetscCall(DMPlexTransformSetUp(tr));
37084663dae6SJed Brown     PetscCall(PetscObjectViewFromOptions((PetscObject)tr, NULL, "-dm_plex_tps_transform_view"));
37094663dae6SJed Brown     PetscCall(DMPlexTransformApply(tr, dm, &edm));
37104663dae6SJed Brown     PetscCall(DMCopyDisc(dm, edm));
37114663dae6SJed Brown     PetscCall(DMGetCoordinateDM(dm, &cdm));
37124663dae6SJed Brown     PetscCall(DMGetCoordinateDM(edm, &ecdm));
37134663dae6SJed Brown     PetscCall(DMCopyDisc(cdm, ecdm));
37144663dae6SJed Brown     PetscCall(DMPlexTransformCreateDiscLabels(tr, edm));
37154663dae6SJed Brown     PetscCall(DMPlexTransformDestroy(&tr));
37164663dae6SJed Brown     if (edm) {
37174663dae6SJed Brown       ((DM_Plex *)edm->data)->printFEM    = ((DM_Plex *)dm->data)->printFEM;
37184663dae6SJed Brown       ((DM_Plex *)edm->data)->printL2     = ((DM_Plex *)dm->data)->printL2;
3719f5867de0SMatthew G. Knepley       ((DM_Plex *)edm->data)->printLocate = ((DM_Plex *)dm->data)->printLocate;
37204663dae6SJed Brown     }
372169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
3722b7f5c055SJed Brown   }
37233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3724b7f5c055SJed Brown }
3725b7f5c055SJed Brown 
3726b7f5c055SJed Brown /*@
3727b7f5c055SJed Brown   DMPlexCreateTPSMesh - Create a distributed, interpolated mesh of a triply-periodic surface
3728b7f5c055SJed Brown 
3729b7f5c055SJed Brown   Collective
3730b7f5c055SJed Brown 
3731b7f5c055SJed Brown   Input Parameters:
3732a1cb98faSBarry Smith + comm           - The communicator for the `DM` object
3733b7f5c055SJed Brown . tpstype        - Type of triply-periodic surface
3734b7f5c055SJed Brown . extent         - Array of length 3 containing number of periods in each direction
373520f4b53cSBarry Smith . periodic       - array of length 3 with periodicity, or `NULL` for non-periodic
37361436d7faSJed Brown . tps_distribute - Distribute 2D manifold mesh prior to refinement and extrusion (more scalable)
3737817da375SSatish Balay . refinements    - Number of factor-of-2 refinements of 2D manifold mesh
37381436d7faSJed Brown . layers         - Number of cell layers extruded in normal direction
3739817da375SSatish Balay - thickness      - Thickness in normal direction
3740b7f5c055SJed Brown 
3741b7f5c055SJed Brown   Output Parameter:
3742a1cb98faSBarry Smith . dm - The `DM` object
3743a1cb98faSBarry Smith 
3744a1cb98faSBarry Smith   Level: beginner
3745b7f5c055SJed Brown 
3746b7f5c055SJed Brown   Notes:
374715229ffcSPierre Jolivet   This meshes the surface of the Schwarz P or Gyroid surfaces.  Schwarz P is the simplest member of the triply-periodic minimal surfaces.
37481d27aa22SBarry Smith   <https://en.wikipedia.org/wiki/Schwarz_minimal_surface#Schwarz_P_(%22Primitive%22)> and can be cut with "clean" boundaries.
37491d27aa22SBarry Smith   The Gyroid <https://en.wikipedia.org/wiki/Gyroid> is another triply-periodic minimal surface with applications in additive manufacturing; it is much more difficult to "cut" since there are no planes of symmetry.
3750b7f5c055SJed Brown   Our implementation creates a very coarse mesh of the surface and refines (by 4-way splitting) as many times as requested.
3751b7f5c055SJed Brown   On each refinement, all vertices are projected to their nearest point on the surface.
3752b7f5c055SJed Brown   This projection could readily be extended to related surfaces.
3753b7f5c055SJed Brown 
37541d27aa22SBarry Smith   See {cite}`maskery2018insights`
37551d27aa22SBarry Smith 
37561d27aa22SBarry Smith   The face (edge) sets for the Schwarz P surface are numbered $1(-x), 2(+x), 3(-y), 4(+y), 5(-z), 6(+z)$.
37571d27aa22SBarry Smith   When the mesh is refined, "Face Sets" contain the new vertices (created during refinement).
37581d27aa22SBarry Smith   Use `DMPlexLabelComplete()` to propagate to coarse-level vertices.
3759b7f5c055SJed Brown 
376060225df5SJacob Faibussowitsch   Developer Notes:
3761b7f5c055SJed Brown   The Gyroid mesh does not currently mark boundary sets.
3762b7f5c055SJed Brown 
37631cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMSetType()`, `DMCreate()`
3764b7f5c055SJed Brown @*/
3765d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateTPSMesh(MPI_Comm comm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness, DM *dm)
3766d71ae5a4SJacob Faibussowitsch {
3767b7f5c055SJed Brown   PetscFunctionBegin;
37689566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
37699566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
37709566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateTPSMesh_Internal(*dm, tpstype, extent, periodic, tps_distribute, refinements, layers, thickness));
37713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3772b7f5c055SJed Brown }
3773b7f5c055SJed Brown 
37749318fe57SMatthew G. Knepley /*@
37759318fe57SMatthew G. Knepley   DMPlexCreateSphereMesh - Creates a mesh on the d-dimensional sphere, S^d.
37769318fe57SMatthew G. Knepley 
37779318fe57SMatthew G. Knepley   Collective
37789318fe57SMatthew G. Knepley 
37799318fe57SMatthew G. Knepley   Input Parameters:
3780a1cb98faSBarry Smith + comm    - The communicator for the `DM` object
37819318fe57SMatthew G. Knepley . dim     - The dimension
37829318fe57SMatthew G. Knepley . simplex - Use simplices, or tensor product cells
37839318fe57SMatthew G. Knepley - R       - The radius
37849318fe57SMatthew G. Knepley 
37859318fe57SMatthew G. Knepley   Output Parameter:
3786a1cb98faSBarry Smith . dm - The `DM` object
37879318fe57SMatthew G. Knepley 
37889318fe57SMatthew G. Knepley   Level: beginner
37899318fe57SMatthew G. Knepley 
37901cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBallMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
37919318fe57SMatthew G. Knepley @*/
3792d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateSphereMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscReal R, DM *dm)
3793d71ae5a4SJacob Faibussowitsch {
37949318fe57SMatthew G. Knepley   PetscFunctionBegin;
37954f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
37969566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
37979566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
37989566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(*dm, dim, simplex, R));
37993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
38009318fe57SMatthew G. Knepley }
38019318fe57SMatthew G. Knepley 
3802d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBallMesh_Internal(DM dm, PetscInt dim, PetscReal R)
3803d71ae5a4SJacob Faibussowitsch {
38049318fe57SMatthew G. Knepley   DM          sdm, vol;
38059318fe57SMatthew G. Knepley   DMLabel     bdlabel;
3806dd2b43ebSStefano Zampini   const char *prefix;
38079318fe57SMatthew G. Knepley 
38089318fe57SMatthew G. Knepley   PetscFunctionBegin;
38099566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &sdm));
38109566063dSJacob Faibussowitsch   PetscCall(DMSetType(sdm, DMPLEX));
3811dd2b43ebSStefano Zampini   PetscCall(DMGetOptionsPrefix(dm, &prefix));
3812dd2b43ebSStefano Zampini   PetscCall(DMSetOptionsPrefix(sdm, prefix));
3813dd2b43ebSStefano Zampini   PetscCall(DMAppendOptionsPrefix(sdm, "bd_"));
3814dd2b43ebSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(sdm, PETSC_FALSE));
38159566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(sdm, dim - 1, PETSC_TRUE, R));
38169566063dSJacob Faibussowitsch   PetscCall(DMSetFromOptions(sdm));
38179566063dSJacob Faibussowitsch   PetscCall(DMViewFromOptions(sdm, NULL, "-dm_view"));
38189566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(sdm, NULL, PETSC_TRUE, &vol));
38199566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&sdm));
382069d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
38219566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
38229566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
38239566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
38249566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, bdlabel));
38253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
382651a74b61SMatthew G. Knepley }
382751a74b61SMatthew G. Knepley 
382851a74b61SMatthew G. Knepley /*@
382951a74b61SMatthew G. Knepley   DMPlexCreateBallMesh - Creates a simplex mesh on the d-dimensional ball, B^d.
383051a74b61SMatthew G. Knepley 
383151a74b61SMatthew G. Knepley   Collective
383251a74b61SMatthew G. Knepley 
383351a74b61SMatthew G. Knepley   Input Parameters:
3834a1cb98faSBarry Smith + comm - The communicator for the `DM` object
383551a74b61SMatthew G. Knepley . dim  - The dimension
383651a74b61SMatthew G. Knepley - R    - The radius
383751a74b61SMatthew G. Knepley 
383851a74b61SMatthew G. Knepley   Output Parameter:
3839a1cb98faSBarry Smith . dm - The `DM` object
384051a74b61SMatthew G. Knepley 
3841a1cb98faSBarry Smith   Options Database Key:
384260225df5SJacob Faibussowitsch . bd_dm_refine - This will refine the surface mesh preserving the sphere geometry
384351a74b61SMatthew G. Knepley 
384451a74b61SMatthew G. Knepley   Level: beginner
384551a74b61SMatthew G. Knepley 
38461cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
384751a74b61SMatthew G. Knepley @*/
3848d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBallMesh(MPI_Comm comm, PetscInt dim, PetscReal R, DM *dm)
3849d71ae5a4SJacob Faibussowitsch {
385051a74b61SMatthew G. Knepley   PetscFunctionBegin;
38519566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
38529566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
38539566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBallMesh_Internal(*dm, dim, R));
38543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
38552829fed8SMatthew G. Knepley }
38562829fed8SMatthew G. Knepley 
3857d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateReferenceCell_Internal(DM rdm, DMPolytopeType ct)
3858d71ae5a4SJacob Faibussowitsch {
38590a6ba040SMatthew G. Knepley   PetscFunctionBegin;
38609318fe57SMatthew G. Knepley   switch (ct) {
38619371c9d4SSatish Balay   case DM_POLYTOPE_POINT: {
38629318fe57SMatthew G. Knepley     PetscInt    numPoints[1]        = {1};
38639318fe57SMatthew G. Knepley     PetscInt    coneSize[1]         = {0};
38649318fe57SMatthew G. Knepley     PetscInt    cones[1]            = {0};
38659318fe57SMatthew G. Knepley     PetscInt    coneOrientations[1] = {0};
38669318fe57SMatthew G. Knepley     PetscScalar vertexCoords[1]     = {0.0};
38679318fe57SMatthew G. Knepley 
38689566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 0));
38699566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38709371c9d4SSatish Balay   } break;
38719371c9d4SSatish Balay   case DM_POLYTOPE_SEGMENT: {
38729318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
38739318fe57SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
38749318fe57SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
38759318fe57SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
38769318fe57SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
38779318fe57SMatthew G. Knepley 
38789566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
38799566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38809371c9d4SSatish Balay   } break;
38819371c9d4SSatish Balay   case DM_POLYTOPE_POINT_PRISM_TENSOR: {
3882b5a892a1SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
3883b5a892a1SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
3884b5a892a1SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
3885b5a892a1SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
3886b5a892a1SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
3887b5a892a1SMatthew G. Knepley 
38889566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
38899566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
38909371c9d4SSatish Balay   } break;
38919371c9d4SSatish Balay   case DM_POLYTOPE_TRIANGLE: {
38929318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {3, 1};
38939318fe57SMatthew G. Knepley     PetscInt    coneSize[4]         = {3, 0, 0, 0};
38949318fe57SMatthew G. Knepley     PetscInt    cones[3]            = {1, 2, 3};
38959318fe57SMatthew G. Knepley     PetscInt    coneOrientations[3] = {0, 0, 0};
38969318fe57SMatthew G. Knepley     PetscScalar vertexCoords[6]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0};
38979318fe57SMatthew G. Knepley 
38989566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
38999566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39009371c9d4SSatish Balay   } break;
39019371c9d4SSatish Balay   case DM_POLYTOPE_QUADRILATERAL: {
39029318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39039318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
39049318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39059318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
39069318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0};
39079318fe57SMatthew G. Knepley 
39089566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
39099566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39109371c9d4SSatish Balay   } break;
39119371c9d4SSatish Balay   case DM_POLYTOPE_SEG_PRISM_TENSOR: {
39129318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39139318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
39149318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39159318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
39169318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0};
39179318fe57SMatthew G. Knepley 
39189566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
39199566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39209371c9d4SSatish Balay   } break;
39219371c9d4SSatish Balay   case DM_POLYTOPE_TETRAHEDRON: {
39229318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
39239318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
3924f0edb160SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
39259318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
3926f0edb160SMatthew G. Knepley     PetscScalar vertexCoords[12]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0};
39279318fe57SMatthew G. Knepley 
39289566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39299566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39309371c9d4SSatish Balay   } break;
39319371c9d4SSatish Balay   case DM_POLYTOPE_HEXAHEDRON: {
39329318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
39339318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
3934f0edb160SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
39359318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39369371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0};
39379318fe57SMatthew G. Knepley 
39389566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39399566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39409371c9d4SSatish Balay   } break;
39419371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM: {
39429318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
39439318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
3944f0edb160SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
39459318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
39469371c9d4SSatish Balay     PetscScalar vertexCoords[18]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, -1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, 1.0, 1.0};
39479318fe57SMatthew G. Knepley 
39489566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39499566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39509371c9d4SSatish Balay   } break;
39519371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM_TENSOR: {
39529318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
39539318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
39549318fe57SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
39559318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
39569371c9d4SSatish Balay     PetscScalar vertexCoords[18]    = {-1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, 1.0, 1.0};
39579318fe57SMatthew G. Knepley 
39589566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39599566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39609371c9d4SSatish Balay   } break;
39619371c9d4SSatish Balay   case DM_POLYTOPE_QUAD_PRISM_TENSOR: {
39629318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
39639318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
39649318fe57SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
39659318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39669371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0, -1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0};
39679318fe57SMatthew G. Knepley 
39689566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39699566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39709371c9d4SSatish Balay   } break;
39719371c9d4SSatish Balay   case DM_POLYTOPE_PYRAMID: {
39729318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {5, 1};
39739318fe57SMatthew G. Knepley     PetscInt    coneSize[6]         = {5, 0, 0, 0, 0, 0};
3974f0edb160SMatthew G. Knepley     PetscInt    cones[5]            = {1, 2, 3, 4, 5};
39759318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
39769371c9d4SSatish Balay     PetscScalar vertexCoords[24]    = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, -1.0, 0.0, 0.0, 1.0};
39779318fe57SMatthew G. Knepley 
39789566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
39799566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
39809371c9d4SSatish Balay   } break;
3981d71ae5a4SJacob Faibussowitsch   default:
3982d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)rdm), PETSC_ERR_ARG_WRONG, "Cannot create reference cell for cell type %s", DMPolytopeTypes[ct]);
39839318fe57SMatthew G. Knepley   }
39849318fe57SMatthew G. Knepley   {
39859318fe57SMatthew G. Knepley     PetscInt Nv, v;
39869318fe57SMatthew G. Knepley 
39879318fe57SMatthew G. Knepley     /* Must create the celltype label here so that we do not automatically try to compute the types */
39889566063dSJacob Faibussowitsch     PetscCall(DMCreateLabel(rdm, "celltype"));
39899566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCellType(rdm, 0, ct));
39909566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(rdm, NULL, &Nv));
39919566063dSJacob Faibussowitsch     for (v = 1; v < Nv; ++v) PetscCall(DMPlexSetCellType(rdm, v, DM_POLYTOPE_POINT));
39929318fe57SMatthew G. Knepley   }
39939566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(rdm));
39949566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)rdm, DMPolytopeTypes[ct]));
39953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
39960a6ba040SMatthew G. Knepley }
39970a6ba040SMatthew G. Knepley 
39989318fe57SMatthew G. Knepley /*@
3999a1cb98faSBarry Smith   DMPlexCreateReferenceCell - Create a `DMPLEX` with the appropriate FEM reference cell
40009318fe57SMatthew G. Knepley 
40019318fe57SMatthew G. Knepley   Collective
40029318fe57SMatthew G. Knepley 
40039318fe57SMatthew G. Knepley   Input Parameters:
40049318fe57SMatthew G. Knepley + comm - The communicator
40059318fe57SMatthew G. Knepley - ct   - The cell type of the reference cell
40069318fe57SMatthew G. Knepley 
40079318fe57SMatthew G. Knepley   Output Parameter:
40089318fe57SMatthew G. Knepley . refdm - The reference cell
40099318fe57SMatthew G. Knepley 
40109318fe57SMatthew G. Knepley   Level: intermediate
40119318fe57SMatthew G. Knepley 
401242747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`
40139318fe57SMatthew G. Knepley @*/
4014d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, DMPolytopeType ct, DM *refdm)
4015d71ae5a4SJacob Faibussowitsch {
40160a6ba040SMatthew G. Knepley   PetscFunctionBegin;
40179566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, refdm));
40189566063dSJacob Faibussowitsch   PetscCall(DMSetType(*refdm, DMPLEX));
40199566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceCell_Internal(*refdm, ct));
40203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40219318fe57SMatthew G. Knepley }
402279a015ccSMatthew G. Knepley 
4023d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoundaryLabel_Private(DM dm, const char name[])
4024d71ae5a4SJacob Faibussowitsch {
40259318fe57SMatthew G. Knepley   DM        plex;
40269318fe57SMatthew G. Knepley   DMLabel   label;
40279318fe57SMatthew G. Knepley   PetscBool hasLabel;
40280a6ba040SMatthew G. Knepley 
4029c22d3578SMatthew G. Knepley   PetscFunctionBegin;
40309566063dSJacob Faibussowitsch   PetscCall(DMHasLabel(dm, name, &hasLabel));
40313ba16761SJacob Faibussowitsch   if (hasLabel) PetscFunctionReturn(PETSC_SUCCESS);
40329566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, name));
40339566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, name, &label));
40349566063dSJacob Faibussowitsch   PetscCall(DMConvert(dm, DMPLEX, &plex));
40359566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(plex, 1, label));
40361c8afea9SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(plex, label));
40379566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&plex));
40383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40399318fe57SMatthew G. Knepley }
4040acdc6f61SToby Isaac 
4041669647acSMatthew G. Knepley /*
4042669647acSMatthew G. Knepley   We use the last coordinate as the radius, the inner radius is lower[dim-1] and the outer radius is upper[dim-1]. Then we map the first coordinate around the circle.
4043669647acSMatthew G. Knepley 
4044669647acSMatthew G. Knepley     (x, y) -> (r, theta) = (x[1], (x[0] - lower[0]) * 2\pi/(upper[0] - lower[0]))
4045669647acSMatthew G. Knepley */
4046d71ae5a4SJacob Faibussowitsch static void boxToAnnulus(PetscInt dim, PetscInt Nf, PetscInt NfAux, const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[])
4047d71ae5a4SJacob Faibussowitsch {
4048669647acSMatthew G. Knepley   const PetscReal low = PetscRealPart(constants[0]);
4049669647acSMatthew G. Knepley   const PetscReal upp = PetscRealPart(constants[1]);
4050669647acSMatthew G. Knepley   const PetscReal r   = PetscRealPart(u[1]);
4051669647acSMatthew G. Knepley   const PetscReal th  = 2. * PETSC_PI * (PetscRealPart(u[0]) - low) / (upp - low);
4052669647acSMatthew G. Knepley 
4053669647acSMatthew G. Knepley   f0[0] = r * PetscCosReal(th);
4054669647acSMatthew G. Knepley   f0[1] = r * PetscSinReal(th);
4055669647acSMatthew G. Knepley }
4056669647acSMatthew G. Knepley 
40575390be7dSMatthew G. Knepley // Insert vertices and their joins, marked by depth
40585390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Vertices(DM dm, DMLabel label, DMLabel vlabel, PetscInt val, PetscInt n, const PetscInt vertices[])
40595390be7dSMatthew G. Knepley {
40605390be7dSMatthew G. Knepley   PetscFunctionBegin;
40615390be7dSMatthew G. Knepley   PetscCall(DMPlexMarkSubmesh_Interpolated(dm, vlabel, val, PETSC_FALSE, PETSC_FALSE, label, NULL));
40625390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
40635390be7dSMatthew G. Knepley }
40645390be7dSMatthew G. Knepley 
40655390be7dSMatthew G. Knepley // Insert faces and their closures, marked by depth
40665390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Faces(DM dm, DMLabel label, PetscInt n, const PetscInt faces[])
40675390be7dSMatthew G. Knepley {
40685390be7dSMatthew G. Knepley   PetscFunctionBegin;
40695390be7dSMatthew G. Knepley   for (PetscInt p = 0; p < n; ++p) {
40705390be7dSMatthew G. Knepley     const PetscInt point   = faces[p];
40715390be7dSMatthew G. Knepley     PetscInt      *closure = NULL;
40725390be7dSMatthew G. Knepley     PetscInt       clSize, pdepth;
40735390be7dSMatthew G. Knepley 
40745390be7dSMatthew G. Knepley     PetscCall(DMPlexGetPointDepth(dm, point, &pdepth));
40755390be7dSMatthew G. Knepley     PetscCall(DMLabelSetValue(label, point, pdepth));
40765390be7dSMatthew G. Knepley     PetscCall(DMPlexGetTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
40775390be7dSMatthew G. Knepley     for (PetscInt cl = 0; cl < clSize * 2; cl += 2) {
40785390be7dSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, closure[cl], &pdepth));
40795390be7dSMatthew G. Knepley       PetscCall(DMLabelSetValue(label, closure[cl], pdepth));
40805390be7dSMatthew G. Knepley     }
40815390be7dSMatthew G. Knepley     PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
40825390be7dSMatthew G. Knepley   }
40835390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
40845390be7dSMatthew G. Knepley }
40855390be7dSMatthew G. Knepley 
40864e22dd4cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions, const char pre[], const char name[], const char *option[], const char *value[], PetscBool *flg);
40874e22dd4cSMatthew G. Knepley 
40885dca41c3SJed Brown const char *const DMPlexShapes[] = {"box", "box_surface", "ball", "sphere", "cylinder", "schwarz_p", "gyroid", "doublet", "annulus", "hypercubic", "zbox", "unknown", "DMPlexShape", "DM_SHAPE_", NULL};
40899318fe57SMatthew G. Knepley 
4090d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateFromOptions_Internal(PetscOptionItems *PetscOptionsObject, PetscBool *useCoordSpace, DM dm)
4091d71ae5a4SJacob Faibussowitsch {
40929318fe57SMatthew G. Knepley   DMPlexShape    shape   = DM_SHAPE_BOX;
40939318fe57SMatthew G. Knepley   DMPolytopeType cell    = DM_POLYTOPE_TRIANGLE;
40949318fe57SMatthew G. Knepley   PetscInt       dim     = 2;
40959318fe57SMatthew G. Knepley   PetscBool      simplex = PETSC_TRUE, interpolate = PETSC_TRUE, adjCone = PETSC_FALSE, adjClosure = PETSC_TRUE, refDomain = PETSC_FALSE;
4096d0812dedSMatthew G. Knepley   PetscBool      flg, flg2, fflg, strflg, bdfflg, nameflg;
40979318fe57SMatthew G. Knepley   MPI_Comm       comm;
4098ed5e4e85SVaclav Hapla   char           filename[PETSC_MAX_PATH_LEN]   = "<unspecified>";
4099ed5e4e85SVaclav Hapla   char           bdFilename[PETSC_MAX_PATH_LEN] = "<unspecified>";
4100ed5e4e85SVaclav Hapla   char           plexname[PETSC_MAX_PATH_LEN]   = "";
41014e22dd4cSMatthew G. Knepley   const char    *option;
41029318fe57SMatthew G. Knepley 
41039318fe57SMatthew G. Knepley   PetscFunctionBegin;
4104708be2fdSJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
41059566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
41069318fe57SMatthew G. Knepley   /* TODO Turn this into a registration interface */
41079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_filename", "File containing a mesh", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &fflg));
4108d0812dedSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_file_contents", "Contents of a file format in a string", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &strflg));
41099566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_filename", "File containing a mesh boundary", "DMPlexCreateFromFile", bdFilename, bdFilename, sizeof(bdFilename), &bdfflg));
41109566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_name", "Name of the mesh in the file", "DMPlexCreateFromFile", plexname, plexname, sizeof(plexname), &nameflg));
41119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_cell", "Cell shape", "", DMPolytopeTypes, (PetscEnum)cell, (PetscEnum *)&cell, NULL));
41129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_reference_cell_domain", "Use a reference cell domain", "", refDomain, &refDomain, NULL));
41139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_shape", "Shape for built-in mesh", "", DMPlexShapes, (PetscEnum)shape, (PetscEnum *)&shape, &flg));
41149566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_dim", "Topological dimension of the mesh", "DMGetDimension", dim, &dim, &flg, 0));
41159566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_simplex", "Mesh cell shape", "", simplex, &simplex, &flg));
41169566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate", "Flag to create edges and faces automatically", "", interpolate, &interpolate, &flg));
41179566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_cone", "Set adjacency direction", "DMSetBasicAdjacency", adjCone, &adjCone, &flg));
41189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_closure", "Set adjacency size", "DMSetBasicAdjacency", adjClosure, &adjClosure, &flg2));
41199566063dSJacob Faibussowitsch   if (flg || flg2) PetscCall(DMSetBasicAdjacency(dm, adjCone, adjClosure));
41209318fe57SMatthew G. Knepley 
412161a622f3SMatthew G. Knepley   switch (cell) {
412261a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT:
412361a622f3SMatthew G. Knepley   case DM_POLYTOPE_SEGMENT:
412461a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT_PRISM_TENSOR:
412561a622f3SMatthew G. Knepley   case DM_POLYTOPE_TRIANGLE:
412661a622f3SMatthew G. Knepley   case DM_POLYTOPE_QUADRILATERAL:
412761a622f3SMatthew G. Knepley   case DM_POLYTOPE_TETRAHEDRON:
4128d71ae5a4SJacob Faibussowitsch   case DM_POLYTOPE_HEXAHEDRON:
4129d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_TRUE;
4130d71ae5a4SJacob Faibussowitsch     break;
4131d71ae5a4SJacob Faibussowitsch   default:
4132d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_FALSE;
4133d71ae5a4SJacob Faibussowitsch     break;
413461a622f3SMatthew G. Knepley   }
413561a622f3SMatthew G. Knepley 
41369318fe57SMatthew G. Knepley   if (fflg) {
41379318fe57SMatthew G. Knepley     DM          dmnew;
41381e4a82c4SMatthew G. Knepley     const char *name;
41399318fe57SMatthew G. Knepley 
41401e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
41411e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), filename, nameflg ? plexname : name, interpolate, &dmnew));
41425de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
414369d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
41449318fe57SMatthew G. Knepley   } else if (refDomain) {
41459566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateReferenceCell_Internal(dm, cell));
41469318fe57SMatthew G. Knepley   } else if (bdfflg) {
41479318fe57SMatthew G. Knepley     DM          bdm, dmnew;
41481e4a82c4SMatthew G. Knepley     const char *name;
41499318fe57SMatthew G. Knepley 
41501e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
41511e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), bdFilename, nameflg ? plexname : name, interpolate, &bdm));
41529566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)bdm, "bd_"));
41539566063dSJacob Faibussowitsch     PetscCall(DMSetFromOptions(bdm));
41549566063dSJacob Faibussowitsch     PetscCall(DMPlexGenerate(bdm, NULL, interpolate, &dmnew));
41559566063dSJacob Faibussowitsch     PetscCall(DMDestroy(&bdm));
41565de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
415769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
4158d0812dedSMatthew G. Knepley   } else if (strflg) {
4159d0812dedSMatthew G. Knepley     DM          dmnew;
4160d0812dedSMatthew G. Knepley     PetscViewer viewer;
4161d0812dedSMatthew G. Knepley     const char *contents;
4162d0812dedSMatthew G. Knepley     char       *strname;
4163d0812dedSMatthew G. Knepley     char        tmpdir[PETSC_MAX_PATH_LEN];
4164d0812dedSMatthew G. Knepley     char        tmpfilename[PETSC_MAX_PATH_LEN];
4165d0812dedSMatthew G. Knepley     char        name[PETSC_MAX_PATH_LEN];
4166d0812dedSMatthew G. Knepley     MPI_Comm    comm;
4167d0812dedSMatthew G. Knepley     PetscMPIInt rank;
4168d0812dedSMatthew G. Knepley 
4169d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4170d0812dedSMatthew G. Knepley     PetscCallMPI(MPI_Comm_rank(comm, &rank));
4171d0812dedSMatthew G. Knepley     PetscCall(PetscStrchr(filename, ':', &strname));
4172d0812dedSMatthew G. Knepley     PetscCheck(strname, comm, PETSC_ERR_ARG_WRONG, "File contents must have the form \"ext:string_name\", not %s", filename);
4173d0812dedSMatthew G. Knepley     strname[0] = '\0';
4174d0812dedSMatthew G. Knepley     ++strname;
4175d0812dedSMatthew G. Knepley     PetscCall(PetscDLSym(NULL, strname, (void **)&contents));
4176d0812dedSMatthew G. Knepley     PetscCheck(contents, comm, PETSC_ERR_ARG_WRONG, "Could not locate mesh string %s", strname);
4177d0812dedSMatthew G. Knepley     PetscCall(PetscGetTmp(comm, tmpdir, PETSC_MAX_PATH_LEN));
4178ed32af8cSMatthew G. Knepley     PetscCall(PetscStrlcat(tmpdir, "/meshXXXXXX", PETSC_MAX_PATH_LEN));
4179ed32af8cSMatthew G. Knepley     PetscCall(PetscMkdtemp(tmpdir));
4180ed32af8cSMatthew G. Knepley     PetscCall(PetscSNPrintf(tmpfilename, PETSC_MAX_PATH_LEN, "%s/mesh.%s", tmpdir, filename));
4181d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIOpen(comm, tmpfilename, &viewer));
4182d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIPrintf(viewer, "%s\n", contents));
4183d0812dedSMatthew G. Knepley     PetscCall(PetscViewerDestroy(&viewer));
4184d0812dedSMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), tmpfilename, plexname, interpolate, &dmnew));
4185ed32af8cSMatthew G. Knepley     PetscCall(PetscRMTree(tmpdir));
4186d0812dedSMatthew G. Knepley     PetscCall(PetscSNPrintf(name, PETSC_MAX_PATH_LEN, "%s Mesh", strname));
4187d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
4188d0812dedSMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
4189d0812dedSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
41909318fe57SMatthew G. Knepley   } else {
41919566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)dm, DMPlexShapes[shape]));
41929318fe57SMatthew G. Knepley     switch (shape) {
4193669647acSMatthew G. Knepley     case DM_SHAPE_BOX:
41945dca41c3SJed Brown     case DM_SHAPE_ZBOX:
4195669647acSMatthew G. Knepley     case DM_SHAPE_ANNULUS: {
41969318fe57SMatthew G. Knepley       PetscInt       faces[3]  = {0, 0, 0};
41979318fe57SMatthew G. Knepley       PetscReal      lower[3]  = {0, 0, 0};
41989318fe57SMatthew G. Knepley       PetscReal      upper[3]  = {1, 1, 1};
41999318fe57SMatthew G. Knepley       DMBoundaryType bdt[3]    = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4200669647acSMatthew G. Knepley       PetscBool      isAnnular = shape == DM_SHAPE_ANNULUS ? PETSC_TRUE : PETSC_FALSE;
42019318fe57SMatthew G. Knepley       PetscInt       i, n;
42029318fe57SMatthew G. Knepley 
42039318fe57SMatthew G. Knepley       n = dim;
42049318fe57SMatthew G. Knepley       for (i = 0; i < dim; ++i) faces[i] = (dim == 1 ? 1 : 4 - dim);
42059566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
42069318fe57SMatthew G. Knepley       n = 3;
42079566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
420863a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
42099318fe57SMatthew G. Knepley       n = 3;
42109566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
421163a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
42129318fe57SMatthew G. Knepley       n = 3;
42139566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
421463a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4215669647acSMatthew G. Knepley 
4216669647acSMatthew G. Knepley       PetscCheck(!isAnnular || dim == 2, comm, PETSC_ERR_ARG_OUTOFRANGE, "Only two dimensional annuli have been implemented");
4217669647acSMatthew G. Knepley       if (isAnnular)
4218669647acSMatthew G. Knepley         for (i = 0; i < dim - 1; ++i) bdt[i] = DM_BOUNDARY_PERIODIC;
4219669647acSMatthew G. Knepley 
42209318fe57SMatthew G. Knepley       switch (cell) {
422161a622f3SMatthew G. Knepley       case DM_POLYTOPE_TRI_PRISM_TENSOR:
42229566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeBoxMesh_Internal(dm, faces, lower, upper, bdt));
4223d410b0cfSMatthew G. Knepley         if (!interpolate) {
4224d410b0cfSMatthew G. Knepley           DM udm;
4225d410b0cfSMatthew G. Knepley 
42269566063dSJacob Faibussowitsch           PetscCall(DMPlexUninterpolate(dm, &udm));
422769d8a87bSksagiyam           PetscCall(DMPlexReplace_Internal(dm, &udm));
4228d410b0cfSMatthew G. Knepley         }
42299318fe57SMatthew G. Knepley         break;
4230d71ae5a4SJacob Faibussowitsch       default:
42315dca41c3SJed Brown         PetscCall(DMPlexCreateBoxMesh_Internal(dm, shape, dim, simplex, faces, lower, upper, bdt, interpolate));
4232d71ae5a4SJacob Faibussowitsch         break;
42339318fe57SMatthew G. Knepley       }
4234669647acSMatthew G. Knepley       if (isAnnular) {
4235669647acSMatthew G. Knepley         DM          cdm;
4236669647acSMatthew G. Knepley         PetscDS     cds;
4237669647acSMatthew G. Knepley         PetscScalar bounds[2] = {lower[0], upper[0]};
4238669647acSMatthew G. Knepley 
4239669647acSMatthew G. Knepley         // Fix coordinates for annular region
4240669647acSMatthew G. Knepley         PetscCall(DMSetPeriodicity(dm, NULL, NULL, NULL));
4241669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinatesLocal(dm, NULL));
4242669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinates(dm, NULL));
4243e44f6aebSMatthew G. Knepley         PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
4244669647acSMatthew G. Knepley         PetscCall(DMGetCoordinateDM(dm, &cdm));
4245669647acSMatthew G. Knepley         PetscCall(DMGetDS(cdm, &cds));
4246669647acSMatthew G. Knepley         PetscCall(PetscDSSetConstants(cds, 2, bounds));
4247669647acSMatthew G. Knepley         PetscCall(DMPlexRemapGeometry(dm, 0.0, boxToAnnulus));
4248669647acSMatthew G. Knepley       }
42499371c9d4SSatish Balay     } break;
42509371c9d4SSatish Balay     case DM_SHAPE_BOX_SURFACE: {
42519318fe57SMatthew G. Knepley       PetscInt  faces[3] = {0, 0, 0};
42529318fe57SMatthew G. Knepley       PetscReal lower[3] = {0, 0, 0};
42539318fe57SMatthew G. Knepley       PetscReal upper[3] = {1, 1, 1};
42549318fe57SMatthew G. Knepley       PetscInt  i, n;
42559318fe57SMatthew G. Knepley 
42569318fe57SMatthew G. Knepley       n = dim + 1;
42579318fe57SMatthew G. Knepley       for (i = 0; i < dim + 1; ++i) faces[i] = (dim + 1 == 1 ? 1 : 4 - (dim + 1));
42589566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
42599318fe57SMatthew G. Knepley       n = 3;
42609566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
426163a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim + 1), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim + 1);
42629318fe57SMatthew G. Knepley       n = 3;
42639566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
426463a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim + 1), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim + 1);
42659566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(dm, dim + 1, faces, lower, upper, interpolate));
42669371c9d4SSatish Balay     } break;
42679371c9d4SSatish Balay     case DM_SHAPE_SPHERE: {
42689318fe57SMatthew G. Knepley       PetscReal R = 1.0;
42699318fe57SMatthew G. Knepley 
42709566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_sphere_radius", "Radius of the sphere", "", R, &R, &flg));
42719566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateSphereMesh_Internal(dm, dim, simplex, R));
42729371c9d4SSatish Balay     } break;
42739371c9d4SSatish Balay     case DM_SHAPE_BALL: {
42749318fe57SMatthew G. Knepley       PetscReal R = 1.0;
42759318fe57SMatthew G. Knepley 
42769566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_ball_radius", "Radius of the ball", "", R, &R, &flg));
42779566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBallMesh_Internal(dm, dim, R));
42789371c9d4SSatish Balay     } break;
42799371c9d4SSatish Balay     case DM_SHAPE_CYLINDER: {
42809318fe57SMatthew G. Knepley       DMBoundaryType bdt = DM_BOUNDARY_NONE;
42819318fe57SMatthew G. Knepley       PetscInt       Nw  = 6;
428249704ca5SMatthew G. Knepley       PetscInt       Nr  = 0;
42839318fe57SMatthew G. Knepley 
42849566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnum("-dm_plex_cylinder_bd", "Boundary type in the z direction", "", DMBoundaryTypes, (PetscEnum)bdt, (PetscEnum *)&bdt, NULL));
42859566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_wedges", "Number of wedges around the cylinder", "", Nw, &Nw, NULL));
428649704ca5SMatthew G. Knepley       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_refine", "Number of refinements before projection", "", Nr, &Nr, NULL));
42879318fe57SMatthew G. Knepley       switch (cell) {
4288d71ae5a4SJacob Faibussowitsch       case DM_POLYTOPE_TRI_PRISM_TENSOR:
4289d71ae5a4SJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(dm, Nw, interpolate));
4290d71ae5a4SJacob Faibussowitsch         break;
4291d71ae5a4SJacob Faibussowitsch       default:
429249704ca5SMatthew G. Knepley         PetscCall(DMPlexCreateHexCylinderMesh_Internal(dm, bdt, Nr));
4293d71ae5a4SJacob Faibussowitsch         break;
42949318fe57SMatthew G. Knepley       }
42959371c9d4SSatish Balay     } break;
4296b7f5c055SJed Brown     case DM_SHAPE_SCHWARZ_P: // fallthrough
42979371c9d4SSatish Balay     case DM_SHAPE_GYROID: {
4298b7f5c055SJed Brown       PetscInt       extent[3] = {1, 1, 1}, refine = 0, layers = 0, three;
4299b7f5c055SJed Brown       PetscReal      thickness   = 0.;
4300b7f5c055SJed Brown       DMBoundaryType periodic[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4301b7f5c055SJed Brown       DMPlexTPSType  tps_type    = shape == DM_SHAPE_SCHWARZ_P ? DMPLEX_TPS_SCHWARZ_P : DMPLEX_TPS_GYROID;
43021436d7faSJed Brown       PetscBool      tps_distribute;
43039566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_tps_extent", "Number of replicas for each of three dimensions", NULL, extent, (three = 3, &three), NULL));
43049566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_refine", "Number of refinements", NULL, refine, &refine, NULL));
43059566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_tps_periodic", "Periodicity in each of three dimensions", NULL, DMBoundaryTypes, (PetscEnum *)periodic, (three = 3, &three), NULL));
43069566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_layers", "Number of layers in volumetric extrusion (or zero to not extrude)", NULL, layers, &layers, NULL));
43079566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_tps_thickness", "Thickness of volumetric extrusion", NULL, thickness, &thickness, NULL));
43089566063dSJacob Faibussowitsch       PetscCall(DMPlexDistributeGetDefault(dm, &tps_distribute));
43099566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_tps_distribute", "Distribute the 2D mesh prior to refinement and extrusion", NULL, tps_distribute, &tps_distribute, NULL));
43109566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateTPSMesh_Internal(dm, tps_type, extent, periodic, tps_distribute, refine, layers, thickness));
43119371c9d4SSatish Balay     } break;
43129371c9d4SSatish Balay     case DM_SHAPE_DOUBLET: {
431305bd46c0SStefano Zampini       DM        dmnew;
431405bd46c0SStefano Zampini       PetscReal rl = 0.0;
431505bd46c0SStefano Zampini 
431605bd46c0SStefano Zampini       PetscCall(PetscOptionsReal("-dm_plex_doublet_refinementlimit", "Refinement limit", NULL, rl, &rl, NULL));
431705bd46c0SStefano Zampini       PetscCall(DMPlexCreateDoublet(PetscObjectComm((PetscObject)dm), dim, simplex, interpolate, rl, &dmnew));
43185de52c6dSVaclav Hapla       PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
431969d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &dmnew));
43209371c9d4SSatish Balay     } break;
4321cfb853baSMatthew G. Knepley     case DM_SHAPE_HYPERCUBIC: {
4322cfb853baSMatthew G. Knepley       PetscInt       *edges;
4323cfb853baSMatthew G. Knepley       PetscReal      *lower, *upper;
4324cfb853baSMatthew G. Knepley       DMBoundaryType *bdt;
4325cfb853baSMatthew G. Knepley       PetscInt        n, d;
4326cfb853baSMatthew G. Knepley 
4327cfb853baSMatthew G. Knepley       *useCoordSpace = PETSC_FALSE;
4328cfb853baSMatthew G. Knepley       PetscCall(PetscMalloc4(dim, &edges, dim, &lower, dim, &upper, dim, &bdt));
4329cfb853baSMatthew G. Knepley       for (d = 0; d < dim; ++d) {
4330cfb853baSMatthew G. Knepley         edges[d] = 1;
4331cfb853baSMatthew G. Knepley         lower[d] = 0.;
4332cfb853baSMatthew G. Knepley         upper[d] = 1.;
4333cfb853baSMatthew G. Knepley         bdt[d]   = DM_BOUNDARY_PERIODIC;
4334cfb853baSMatthew G. Knepley       }
4335cfb853baSMatthew G. Knepley       n = dim;
4336cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", edges, &n, &flg));
4337cfb853baSMatthew G. Knepley       n = dim;
4338cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
4339cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4340cfb853baSMatthew G. Knepley       n = dim;
4341cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
4342cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4343cfb853baSMatthew G. Knepley       n = dim;
4344cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
4345cfb853baSMatthew G. Knepley       PetscCheck(!flg || n == dim, comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4346cfb853baSMatthew G. Knepley       PetscCall(DMPlexCreateHypercubicMesh_Internal(dm, dim, lower, upper, edges, bdt));
4347cfb853baSMatthew G. Knepley       PetscCall(PetscFree4(edges, lower, upper, bdt));
4348cfb853baSMatthew G. Knepley     } break;
4349d71ae5a4SJacob Faibussowitsch     default:
4350d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_SUP, "Domain shape %s is unsupported", DMPlexShapes[shape]);
43519318fe57SMatthew G. Knepley     }
43529318fe57SMatthew G. Knepley   }
43539566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
435448a46eb9SPierre Jolivet   if (!((PetscObject)dm)->name && nameflg) PetscCall(PetscObjectSetName((PetscObject)dm, plexname));
43554e22dd4cSMatthew G. Knepley   // Allow label creation
43564e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_label_", &option, NULL, &flg));
43574e22dd4cSMatthew G. Knepley   if (flg) {
43584e22dd4cSMatthew G. Knepley     DMLabel     label;
43594e22dd4cSMatthew G. Knepley     PetscInt    points[1024], n = 1024;
43604e22dd4cSMatthew G. Knepley     char        fulloption[PETSC_MAX_PATH_LEN];
43614e22dd4cSMatthew G. Knepley     const char *name = &option[14];
43624e22dd4cSMatthew G. Knepley 
43634e22dd4cSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, name));
43644e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, name, &label));
43654e22dd4cSMatthew G. Knepley     fulloption[0] = '-';
43664e22dd4cSMatthew G. Knepley     fulloption[1] = 0;
43674e22dd4cSMatthew G. Knepley     PetscCall(PetscStrlcat(fulloption, option, PETSC_MAX_PATH_LEN));
43684e22dd4cSMatthew G. Knepley     PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, NULL));
43694e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < n; ++p) PetscCall(DMLabelSetValue(label, points[p], 1));
43704e22dd4cSMatthew G. Knepley   }
4371dd0eeac9SMatthew G. Knepley   // Allow cohesive label creation
4372dd0eeac9SMatthew G. Knepley   //   Faces are input, completed, and all points are marked with their depth
4373dd0eeac9SMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_cohesive_label_", &option, NULL, &flg));
4374dd0eeac9SMatthew G. Knepley   if (flg) {
4375dd0eeac9SMatthew G. Knepley     DMLabel   label;
4376dd0eeac9SMatthew G. Knepley     PetscInt  points[1024], n, pStart, pEnd, Nl = 1;
43775390be7dSMatthew G. Knepley     PetscBool noCreate = PETSC_FALSE;
4378dd0eeac9SMatthew G. Knepley     char      fulloption[PETSC_MAX_PATH_LEN];
4379dd0eeac9SMatthew G. Knepley     char      name[PETSC_MAX_PATH_LEN];
4380dd0eeac9SMatthew G. Knepley     size_t    len;
4381dd0eeac9SMatthew G. Knepley 
4382dd0eeac9SMatthew G. Knepley     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4383dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrncpy(name, &option[23], PETSC_MAX_PATH_LEN));
4384dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrlen(name, &len));
4385dd0eeac9SMatthew G. Knepley     if (name[len - 1] == '0') Nl = 10;
4386dd0eeac9SMatthew G. Knepley     for (PetscInt l = 0; l < Nl; ++l) {
43876497c311SBarry Smith       if (l > 0) name[len - 1] = (char)('0' + l);
4388dd0eeac9SMatthew G. Knepley       fulloption[0] = 0;
4389dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, "-dm_plex_cohesive_label_", 32));
4390dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, name, PETSC_MAX_PATH_LEN - 32));
4391dd0eeac9SMatthew G. Knepley       n = 1024;
4392dd0eeac9SMatthew G. Knepley       PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, &flg));
4393dd0eeac9SMatthew G. Knepley       if (!flg) break;
43945390be7dSMatthew G. Knepley       PetscCall(DMHasLabel(dm, name, &noCreate));
43955390be7dSMatthew G. Knepley       if (noCreate) {
43965390be7dSMatthew G. Knepley         DMLabel         inlabel;
43975390be7dSMatthew G. Knepley         IS              pointIS;
43985390be7dSMatthew G. Knepley         const PetscInt *lpoints;
43995390be7dSMatthew G. Knepley         PetscInt        pdep, ln, inval = points[0];
44005390be7dSMatthew G. Knepley         char            newname[PETSC_MAX_PATH_LEN];
44015390be7dSMatthew G. Knepley 
44025390be7dSMatthew G. Knepley         PetscCheck(n == 1, comm, PETSC_ERR_ARG_WRONG, "Must specify a label value with this option");
44035390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &inlabel));
44045390be7dSMatthew G. Knepley         PetscCall(DMLabelGetStratumIS(inlabel, inval, &pointIS));
44055390be7dSMatthew G. Knepley         PetscCall(ISGetLocalSize(pointIS, &ln));
44065390be7dSMatthew G. Knepley         PetscCall(ISGetIndices(pointIS, &lpoints));
44075390be7dSMatthew G. Knepley         PetscCall(DMPlexGetPointDepth(dm, lpoints[0], &pdep));
44085390be7dSMatthew G. Knepley         PetscCall(PetscSNPrintf(newname, PETSC_MAX_PATH_LEN, "%s%" PetscInt_FMT, name, points[0]));
44095390be7dSMatthew G. Knepley         PetscCall(DMCreateLabel(dm, newname));
44105390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, newname, &label));
44115390be7dSMatthew G. Knepley         if (!pdep) PetscCall(ProcessCohesiveLabel_Vertices(dm, label, inlabel, inval, ln, lpoints));
44125390be7dSMatthew G. Knepley         else PetscCall(ProcessCohesiveLabel_Faces(dm, label, ln, lpoints));
44135390be7dSMatthew G. Knepley         PetscCall(ISRestoreIndices(pointIS, &lpoints));
44145390be7dSMatthew G. Knepley         PetscCall(ISDestroy(&pointIS));
44155390be7dSMatthew G. Knepley       } else {
4416dd0eeac9SMatthew G. Knepley         PetscCall(DMCreateLabel(dm, name));
4417dd0eeac9SMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &label));
4418dd0eeac9SMatthew G. Knepley         if (pStart >= pEnd) n = 0;
44195390be7dSMatthew G. Knepley         PetscCall(ProcessCohesiveLabel_Faces(dm, label, n, points));
4420dd0eeac9SMatthew G. Knepley       }
4421dd0eeac9SMatthew G. Knepley       PetscCall(DMPlexOrientLabel(dm, label));
44220542aa8cSMatthew G. Knepley       PetscCall(DMPlexLabelCohesiveComplete(dm, label, NULL, 1, PETSC_FALSE, PETSC_FALSE, NULL));
4423dd0eeac9SMatthew G. Knepley     }
4424dd0eeac9SMatthew G. Knepley   }
44255390be7dSMatthew G. Knepley   PetscCall(DMViewFromOptions(dm, NULL, "-created_dm_view"));
4426708be2fdSJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
44273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44280a6ba040SMatthew G. Knepley }
44290a6ba040SMatthew G. Knepley 
4430d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_NonRefinement_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4431d71ae5a4SJacob Faibussowitsch {
44320a6ba040SMatthew G. Knepley   DM_Plex  *mesh = (DM_Plex *)dm->data;
44337f9d8d6cSVaclav Hapla   PetscBool flg, flg2;
44349318fe57SMatthew G. Knepley   char      bdLabel[PETSC_MAX_PATH_LEN];
4435adc21957SMatthew G. Knepley   char      method[PETSC_MAX_PATH_LEN];
44360a6ba040SMatthew G. Knepley 
44370a6ba040SMatthew G. Knepley   PetscFunctionBegin;
44380a6ba040SMatthew G. Knepley   /* Handle viewing */
44399566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMPlexMatSetClosure", PETSC_FALSE, &mesh->printSetValues, NULL));
44405962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fem", "Debug output level for all fem computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printFEM, NULL, 0));
44415962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fvm", "Debug output level for all fvm computations", "DMPlexSNESComputeResidualFVM", 0, &mesh->printFVM, NULL, 0));
44429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMPlexSNESComputeResidualFEM", mesh->printTol, &mesh->printTol, NULL));
44439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_l2", "Debug output level all L2 diff computations", "DMComputeL2Diff", 0, &mesh->printL2, NULL, 0));
4444f5867de0SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_locate", "Debug output level all point location computations", "DMLocatePoints", 0, &mesh->printLocate, NULL, 0));
44459566063dSJacob Faibussowitsch   PetscCall(DMMonitorSetFromOptions(dm, "-dm_plex_monitor_throughput", "Monitor the simulation throughput", "DMPlexMonitorThroughput", DMPlexMonitorThroughput, NULL, &flg));
44469566063dSJacob Faibussowitsch   if (flg) PetscCall(PetscLogDefaultBegin());
44479318fe57SMatthew G. Knepley   /* Labeling */
44489566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_label", "Label to mark the mesh boundary", "", bdLabel, bdLabel, sizeof(bdLabel), &flg));
44499566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexCreateBoundaryLabel_Private(dm, bdLabel));
4450953fc75cSMatthew G. Knepley   /* Point Location */
44519566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMInterpolate", PETSC_FALSE, &mesh->useHashLocation, NULL));
44520848f4b5SMatthew G. Knepley   /* Partitioning and distribution */
44539566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", PETSC_FALSE, &mesh->partitionBalance, NULL));
4454d02c7345SMatthew G. Knepley   /* Reordering */
4455adc21957SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_reorder_section", "Compute point permutation for local section", "DMReorderSectionSetDefault", PETSC_FALSE, &flg2, &flg));
4456adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetDefault(dm, flg2 ? DM_REORDER_DEFAULT_TRUE : DM_REORDER_DEFAULT_FALSE));
4457adc21957SMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_reorder_section_type", "Reordering method for local section", "DMReorderSectionSetType", method, method, PETSC_MAX_PATH_LEN, &flg));
4458adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetType(dm, method));
44592e62ab5aSMatthew G. Knepley   /* Generation and remeshing */
44609566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_remesh_bd", "Allow changes to the boundary on remeshing", "DMAdapt", PETSC_FALSE, &mesh->remeshBd, NULL));
4461b29cfa1cSToby Isaac   /* Projection behavior */
4462d5b43468SJose E. Roman   PetscCall(PetscOptionsBoundedInt("-dm_plex_max_projection_height", "Maximum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL, 0));
44639566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_regular_refinement", "Use special nested projection algorithm for regular refinement", "DMPlexSetRegularRefinement", mesh->regularRefinement, &mesh->regularRefinement, NULL));
4464f12cf164SMatthew G. Knepley   /* Checking structure */
4465f12cf164SMatthew G. Knepley   {
44667f9d8d6cSVaclav Hapla     PetscBool all = PETSC_FALSE;
4467f12cf164SMatthew G. Knepley 
44687f9d8d6cSVaclav Hapla     PetscCall(PetscOptionsBool("-dm_plex_check_all", "Perform all basic checks", "DMPlexCheck", PETSC_FALSE, &all, NULL));
44697f9d8d6cSVaclav Hapla     if (all) {
44707f9d8d6cSVaclav Hapla       PetscCall(DMPlexCheck(dm));
44717f9d8d6cSVaclav Hapla     } else {
44729566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_symmetry", "Check that the adjacency information in the mesh is symmetric", "DMPlexCheckSymmetry", PETSC_FALSE, &flg, &flg2));
44737f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSymmetry(dm));
44749566063dSJacob 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));
44757f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSkeleton(dm, 0));
44769566063dSJacob 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));
44777f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckFaces(dm, 0));
44789566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_geometry", "Check that cells have positive volume", "DMPlexCheckGeometry", PETSC_FALSE, &flg, &flg2));
44797f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckGeometry(dm));
44809566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_pointsf", "Check some necessary conditions for PointSF", "DMPlexCheckPointSF", PETSC_FALSE, &flg, &flg2));
4481d7d32a9aSMatthew G. Knepley       if (flg && flg2) PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE));
44829566063dSJacob 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));
44837f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckInterfaceCones(dm));
44847f9d8d6cSVaclav Hapla     }
44859566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_plex_check_cell_shape", "Check cell shape", "DMPlexCheckCellShape", PETSC_FALSE, &flg, &flg2));
44869566063dSJacob Faibussowitsch     if (flg && flg2) PetscCall(DMPlexCheckCellShape(dm, PETSC_TRUE, PETSC_DETERMINE));
4487f12cf164SMatthew G. Knepley   }
44889318fe57SMatthew G. Knepley   {
44899318fe57SMatthew G. Knepley     PetscReal scale = 1.0;
44904f3833eaSMatthew G. Knepley 
44919566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-dm_plex_scale", "Scale factor for mesh coordinates", "DMPlexScale", scale, &scale, &flg));
44929318fe57SMatthew G. Knepley     if (flg) {
44939318fe57SMatthew G. Knepley       Vec coordinates, coordinatesLocal;
44949318fe57SMatthew G. Knepley 
44959566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(dm, &coordinates));
44969566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal));
44979566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinates, scale));
44989566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinatesLocal, scale));
44999318fe57SMatthew G. Knepley     }
45009318fe57SMatthew G. Knepley   }
45019566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerSetFromOptions(mesh->partitioner));
45023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
450368d4fef7SMatthew G. Knepley }
450468d4fef7SMatthew G. Knepley 
4505d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetFromOptions_Overlap_Plex(DM dm, PetscOptionItems *PetscOptionsObject, PetscInt *overlap)
4506d71ae5a4SJacob Faibussowitsch {
4507c506a872SMatthew G. Knepley   PetscInt  numOvLabels = 16, numOvExLabels = 16;
4508c506a872SMatthew G. Knepley   char     *ovLabelNames[16], *ovExLabelNames[16];
4509c506a872SMatthew G. Knepley   PetscInt  numOvValues = 16, numOvExValues = 16, l;
4510c506a872SMatthew G. Knepley   PetscBool flg;
4511c506a872SMatthew G. Knepley 
4512c506a872SMatthew G. Knepley   PetscFunctionBegin;
4513c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", *overlap, overlap, NULL, 0));
4514c506a872SMatthew G. Knepley   PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_labels", "List of overlap label names", "DMPlexDistribute", ovLabelNames, &numOvLabels, &flg));
4515c506a872SMatthew G. Knepley   if (!flg) numOvLabels = 0;
4516c506a872SMatthew G. Knepley   if (numOvLabels) {
4517c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvLabels = numOvLabels;
4518c506a872SMatthew G. Knepley     for (l = 0; l < numOvLabels; ++l) {
4519c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovLabelNames[l], &((DM_Plex *)dm->data)->ovLabels[l]));
4520c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovLabelNames[l]);
4521c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovLabelNames[l]));
4522c506a872SMatthew G. Knepley     }
4523c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_values", "List of overlap label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovValues, &numOvValues, &flg));
4524c506a872SMatthew G. Knepley     if (!flg) numOvValues = 0;
4525c506a872SMatthew 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);
4526c506a872SMatthew G. Knepley 
4527c506a872SMatthew G. Knepley     PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_exclude_labels", "List of overlap exclude label names", "DMPlexDistribute", ovExLabelNames, &numOvExLabels, &flg));
4528c506a872SMatthew G. Knepley     if (!flg) numOvExLabels = 0;
4529c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvExLabels = numOvExLabels;
4530c506a872SMatthew G. Knepley     for (l = 0; l < numOvExLabels; ++l) {
4531c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovExLabelNames[l], &((DM_Plex *)dm->data)->ovExLabels[l]));
4532c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovExLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovExLabelNames[l]);
4533c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovExLabelNames[l]));
4534c506a872SMatthew G. Knepley     }
4535c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_exclude_values", "List of overlap exclude label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovExValues, &numOvExValues, &flg));
4536c506a872SMatthew G. Knepley     if (!flg) numOvExValues = 0;
4537c506a872SMatthew 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);
4538c506a872SMatthew G. Knepley   }
45393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4540c506a872SMatthew G. Knepley }
4541c506a872SMatthew G. Knepley 
4542d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMSetFromOptions_Plex(DM dm, PetscOptionItems *PetscOptionsObject)
4543d71ae5a4SJacob Faibussowitsch {
4544bdf63967SMatthew G. Knepley   PetscFunctionList    ordlist;
4545bdf63967SMatthew G. Knepley   char                 oname[256];
45464e22dd4cSMatthew G. Knepley   char                 sublabelname[PETSC_MAX_PATH_LEN] = "";
4547adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
4548d410b0cfSMatthew G. Knepley   PetscReal            volume    = -1.0;
45499318fe57SMatthew G. Knepley   PetscInt             prerefine = 0, refine = 0, r, coarsen = 0, overlap = 0, extLayers = 0, dim;
45501b742f01SMatthew 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;
455168d4fef7SMatthew G. Knepley 
455268d4fef7SMatthew G. Knepley   PetscFunctionBegin;
4553d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "DMPlex Options");
4554dd4c3f67SMatthew G. Knepley   if (dm->cloneOpts) goto non_refine;
45559318fe57SMatthew G. Knepley   /* Handle automatic creation */
45569566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
45576bc1bd01Sksagiyam   if (dim < 0) {
45586bc1bd01Sksagiyam     PetscCall(DMPlexCreateFromOptions_Internal(PetscOptionsObject, &coordSpace, dm));
45596bc1bd01Sksagiyam     created = PETSC_TRUE;
45606bc1bd01Sksagiyam   }
45616bc1bd01Sksagiyam   PetscCall(DMGetDimension(dm, &dim));
4562d89e6e46SMatthew G. Knepley   /* Handle interpolation before distribution */
45639566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate_pre", "Flag to interpolate mesh before distribution", "", interpolate, &interpolate, &flg));
4564d89e6e46SMatthew G. Knepley   if (flg) {
4565d89e6e46SMatthew G. Knepley     DMPlexInterpolatedFlag interpolated;
4566d89e6e46SMatthew G. Knepley 
45679566063dSJacob Faibussowitsch     PetscCall(DMPlexIsInterpolated(dm, &interpolated));
4568d89e6e46SMatthew G. Knepley     if (interpolated == DMPLEX_INTERPOLATED_FULL && !interpolate) {
4569d89e6e46SMatthew G. Knepley       DM udm;
4570d89e6e46SMatthew G. Knepley 
45719566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(dm, &udm));
457269d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &udm));
4573d89e6e46SMatthew G. Knepley     } else if (interpolated != DMPLEX_INTERPOLATED_FULL && interpolate) {
4574d89e6e46SMatthew G. Knepley       DM idm;
4575d89e6e46SMatthew G. Knepley 
45769566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(dm, &idm));
457769d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &idm));
4578d89e6e46SMatthew G. Knepley     }
4579d89e6e46SMatthew G. Knepley   }
45804e22dd4cSMatthew G. Knepley   // Handle submesh selection before distribution
45814e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_submesh", "Label to use for submesh selection", "", sublabelname, sublabelname, PETSC_MAX_PATH_LEN, &flg));
45824e22dd4cSMatthew G. Knepley   if (flg) {
45834e22dd4cSMatthew G. Knepley     DM              subdm;
45844e22dd4cSMatthew G. Knepley     DMLabel         label;
45854e22dd4cSMatthew G. Knepley     IS              valueIS, pointIS;
45864e22dd4cSMatthew G. Knepley     const PetscInt *values, *points;
45874e22dd4cSMatthew G. Knepley     PetscBool       markedFaces = PETSC_FALSE;
45884e22dd4cSMatthew G. Knepley     PetscInt        Nv, value, Np;
45894e22dd4cSMatthew G. Knepley 
45904e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, sublabelname, &label));
45914e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetNumValues(label, &Nv));
45924e22dd4cSMatthew 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);
45934e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetValueIS(label, &valueIS));
45944e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(valueIS, &values));
45954e22dd4cSMatthew G. Knepley     value = values[0];
45964e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(valueIS, &values));
45974e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&valueIS));
45984e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumSize(label, value, &Np));
45994e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumIS(label, value, &pointIS));
46004e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(pointIS, &points));
46014e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < Np; ++p) {
46024e22dd4cSMatthew G. Knepley       PetscInt pdepth;
46034e22dd4cSMatthew G. Knepley 
46044e22dd4cSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, points[p], &pdepth));
46054e22dd4cSMatthew G. Knepley       if (pdepth) {
46064e22dd4cSMatthew G. Knepley         markedFaces = PETSC_TRUE;
46074e22dd4cSMatthew G. Knepley         break;
46084e22dd4cSMatthew G. Knepley       }
46094e22dd4cSMatthew G. Knepley     }
46104e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(pointIS, &points));
46114e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&pointIS));
46124e22dd4cSMatthew G. Knepley     PetscCall(DMPlexCreateSubmesh(dm, label, value, markedFaces, &subdm));
46134e22dd4cSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &subdm));
46144e22dd4cSMatthew G. Knepley     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
46154e22dd4cSMatthew G. Knepley   }
46169b44eab4SMatthew G. Knepley   /* Handle DMPlex refinement before distribution */
46179566063dSJacob Faibussowitsch   PetscCall(DMPlexGetRefinementUniform(dm, &uniformOrig));
46189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_pre", "The number of refinements before distribution", "DMCreate", prerefine, &prerefine, NULL, 0));
46199566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap_pre", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
46209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_uniform_pre", "Flag for uniform refinement before distribution", "DMCreate", uniform, &uniform, &flg));
46219566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexSetRefinementUniform(dm, uniform));
46229566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_refine_volume_limit_pre", "The maximum cell volume after refinement before distribution", "DMCreate", volume, &volume, &flg));
46239318fe57SMatthew G. Knepley   if (flg) {
46249566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(dm, PETSC_FALSE));
46259566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(dm, volume));
46269318fe57SMatthew G. Knepley     prerefine = PetscMax(prerefine, 1);
46279318fe57SMatthew G. Knepley   }
4628b23db253SStefano Zampini   if (prerefine) PetscCall(DMLocalizeCoordinates(dm));
46299b44eab4SMatthew G. Knepley   for (r = 0; r < prerefine; ++r) {
46309b44eab4SMatthew G. Knepley     DM             rdm;
46319b44eab4SMatthew G. Knepley     PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
46329b44eab4SMatthew G. Knepley 
4633dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
46349566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
463569d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &rdm));
4636dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
463761a622f3SMatthew G. Knepley     if (coordFunc && remap) {
46389566063dSJacob Faibussowitsch       PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
46399b44eab4SMatthew G. Knepley       ((DM_Plex *)dm->data)->coordFunc = coordFunc;
46409b44eab4SMatthew G. Knepley     }
46419b44eab4SMatthew G. Knepley   }
46429566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, uniformOrig));
46439318fe57SMatthew G. Knepley   /* Handle DMPlex extrusion before distribution */
46449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_extrude", "The number of layers to extrude", "", extLayers, &extLayers, NULL, 0));
46459318fe57SMatthew G. Knepley   if (extLayers) {
46469318fe57SMatthew G. Knepley     DM edm;
46479318fe57SMatthew G. Knepley 
46489566063dSJacob Faibussowitsch     PetscCall(DMExtrude(dm, extLayers, &edm));
464969d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
465048d16a33SMatthew G. Knepley     ((DM_Plex *)dm->data)->coordFunc = NULL;
4651dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4652d410b0cfSMatthew G. Knepley     extLayers = 0;
46535e17fc22SAidan Hamilton     PetscCall(DMGetDimension(dm, &dim));
46549318fe57SMatthew G. Knepley   }
4655bdf63967SMatthew G. Knepley   /* Handle DMPlex reordering before distribution */
46566bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dm, &reorder));
46579566063dSJacob Faibussowitsch   PetscCall(MatGetOrderingList(&ordlist));
46586bc1bd01Sksagiyam   PetscCall(PetscStrncpy(oname, MATORDERINGNATURAL, sizeof(oname)));
46599566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-dm_plex_reorder", "Set mesh reordering type", "DMPlexGetOrdering", ordlist, MATORDERINGNATURAL, oname, sizeof(oname), &flg));
4660adc21957SMatthew G. Knepley   if (reorder == DM_REORDER_DEFAULT_TRUE || flg) {
4661bdf63967SMatthew G. Knepley     DM pdm;
4662bdf63967SMatthew G. Knepley     IS perm;
4663bdf63967SMatthew G. Knepley 
46649566063dSJacob Faibussowitsch     PetscCall(DMPlexGetOrdering(dm, oname, NULL, &perm));
46659566063dSJacob Faibussowitsch     PetscCall(DMPlexPermute(dm, perm, &pdm));
46669566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&perm));
466769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &pdm));
4668dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4669bdf63967SMatthew G. Knepley   }
46709b44eab4SMatthew G. Knepley   /* Handle DMPlex distribution */
46719566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dm, &distribute));
4672c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute", "Flag to redistribute a mesh among processes", "DMPlexDistribute", distribute, &distribute, NULL));
4673a286e215SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute_save_sf", "Flag to save the migration SF", "DMPlexSetMigrationSF", saveSF, &saveSF, NULL));
4674dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_Overlap_Plex(dm, PetscOptionsObject, &overlap));
46759b44eab4SMatthew G. Knepley   if (distribute) {
46769b44eab4SMatthew G. Knepley     DM               pdm = NULL;
46779b44eab4SMatthew G. Knepley     PetscPartitioner part;
4678a286e215SMatthew G. Knepley     PetscSF          sfMigration;
46799b44eab4SMatthew G. Knepley 
46809566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
46819566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
4682a286e215SMatthew G. Knepley     PetscCall(DMPlexDistribute(dm, overlap, &sfMigration, &pdm));
468348a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
4684a286e215SMatthew G. Knepley     if (saveSF) PetscCall(DMPlexSetMigrationSF(dm, sfMigration));
4685a286e215SMatthew G. Knepley     PetscCall(PetscSFDestroy(&sfMigration));
46869b44eab4SMatthew G. Knepley   }
46874054ae39SJames Wright 
46884054ae39SJames Wright   {
46894054ae39SJames Wright     PetscBool useBoxLabel = PETSC_FALSE;
46904054ae39SJames Wright     PetscCall(PetscOptionsBool("-dm_plex_box_label", "Create 'Face Sets' assuming boundary faces align with cartesian directions", "DMCreate", useBoxLabel, &useBoxLabel, NULL));
46914054ae39SJames Wright     if (useBoxLabel) PetscCall(DMPlexSetBoxLabel_Internal(dm));
46924054ae39SJames Wright   }
4693d2b2dc1eSMatthew G. Knepley   /* Must check CEED options before creating function space for coordinates */
4694d2b2dc1eSMatthew G. Knepley   {
4695d2b2dc1eSMatthew G. Knepley     PetscBool useCeed = PETSC_FALSE, flg;
4696d2b2dc1eSMatthew G. Knepley 
4697d2b2dc1eSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_plex_use_ceed", "Use LibCEED as the FEM backend", "DMPlexSetUseCeed", useCeed, &useCeed, &flg));
4698d2b2dc1eSMatthew G. Knepley     if (flg) PetscCall(DMPlexSetUseCeed(dm, useCeed));
4699d2b2dc1eSMatthew G. Knepley   }
47009318fe57SMatthew G. Knepley   /* Create coordinate space */
47019318fe57SMatthew G. Knepley   if (created) {
470261a622f3SMatthew G. Knepley     DM_Plex  *mesh   = (DM_Plex *)dm->data;
4703e44f6aebSMatthew G. Knepley     PetscInt  degree = 1, deg;
47045515ebd3SMatthew G. Knepley     PetscInt  height = 0;
47055515ebd3SMatthew G. Knepley     DM        cdm;
4706c3db174cSMatthew G. Knepley     PetscBool flg, localize = PETSC_TRUE, sparseLocalize = PETSC_TRUE;
47079318fe57SMatthew G. Knepley 
47089566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_coord_space", "Use an FEM space for coordinates", "", coordSpace, &coordSpace, &flg));
47099566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-dm_coord_petscspace_degree", "FEM degree for coordinate space", "", degree, &degree, NULL));
4710e44f6aebSMatthew G. Knepley     PetscCall(DMGetCoordinateDegree_Internal(dm, &deg));
4711e44f6aebSMatthew G. Knepley     if (coordSpace && deg <= 1) PetscCall(DMPlexCreateCoordinateSpace(dm, degree, PETSC_TRUE, mesh->coordFunc));
47125515ebd3SMatthew G. Knepley     PetscCall(DMGetCoordinateDM(dm, &cdm));
471361a622f3SMatthew G. Knepley     if (flg && !coordSpace) {
471461a622f3SMatthew G. Knepley       PetscDS      cds;
471561a622f3SMatthew G. Knepley       PetscObject  obj;
471661a622f3SMatthew G. Knepley       PetscClassId id;
471761a622f3SMatthew G. Knepley 
47189566063dSJacob Faibussowitsch       PetscCall(DMGetDS(cdm, &cds));
47199566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
47209566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
472161a622f3SMatthew G. Knepley       if (id == PETSCFE_CLASSID) {
472261a622f3SMatthew G. Knepley         PetscContainer dummy;
472361a622f3SMatthew G. Knepley 
47249566063dSJacob Faibussowitsch         PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &dummy));
47259566063dSJacob Faibussowitsch         PetscCall(PetscObjectSetName((PetscObject)dummy, "coordinates"));
47269566063dSJacob Faibussowitsch         PetscCall(DMSetField(cdm, 0, NULL, (PetscObject)dummy));
47279566063dSJacob Faibussowitsch         PetscCall(PetscContainerDestroy(&dummy));
47289566063dSJacob Faibussowitsch         PetscCall(DMClearDS(cdm));
472961a622f3SMatthew G. Knepley       }
473061a622f3SMatthew G. Knepley       mesh->coordFunc = NULL;
473161a622f3SMatthew G. Knepley     }
4732c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_localize", "Localize mesh coordinates", "", localize, &localize, NULL));
4733c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_sparse_localize", "Localize only necessary cells", "DMSetSparseLocalize", sparseLocalize, &sparseLocalize, &flg));
4734c3db174cSMatthew G. Knepley     if (flg) PetscCall(DMSetSparseLocalize(dm, sparseLocalize));
47355515ebd3SMatthew G. Knepley     PetscCall(PetscOptionsInt("-dm_localize_height", "Localize edges and faces in addition to cells", "", height, &height, &flg));
47365515ebd3SMatthew G. Knepley     if (flg) PetscCall(DMPlexSetMaxProjectionHeight(cdm, height));
4737c3db174cSMatthew G. Knepley     if (localize) PetscCall(DMLocalizeCoordinates(dm));
47389318fe57SMatthew G. Knepley   }
473968d4fef7SMatthew G. Knepley   /* Handle DMPlex refinement */
474061a622f3SMatthew G. Knepley   remap = PETSC_TRUE;
47419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL, 0));
47429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
47439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy, 0));
47449566063dSJacob Faibussowitsch   if (refine) PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
474568d4fef7SMatthew G. Knepley   if (refine && isHierarchy) {
4746acdc6f61SToby Isaac     DM *dms, coarseDM;
474768d4fef7SMatthew G. Knepley 
47489566063dSJacob Faibussowitsch     PetscCall(DMGetCoarseDM(dm, &coarseDM));
47499566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)coarseDM));
47509566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(refine, &dms));
47519566063dSJacob Faibussowitsch     PetscCall(DMRefineHierarchy(dm, refine, dms));
475268d4fef7SMatthew G. Knepley     /* Total hack since we do not pass in a pointer */
47539566063dSJacob Faibussowitsch     PetscCall(DMPlexSwap_Static(dm, dms[refine - 1]));
475468d4fef7SMatthew G. Knepley     if (refine == 1) {
47559566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[0]));
47569566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
475768d4fef7SMatthew G. Knepley     } else {
47589566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[refine - 2]));
47599566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
47609566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dms[0], dms[refine - 1]));
47619566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dms[0], PETSC_TRUE));
476268d4fef7SMatthew G. Knepley     }
47639566063dSJacob Faibussowitsch     PetscCall(DMSetCoarseDM(dms[refine - 1], coarseDM));
47649566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)coarseDM));
476568d4fef7SMatthew G. Knepley     /* Free DMs */
476668d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
4767dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
47689566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
476968d4fef7SMatthew G. Knepley     }
47709566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
477168d4fef7SMatthew G. Knepley   } else {
477268d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
47739318fe57SMatthew G. Knepley       DM             rdm;
477451a74b61SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
477568d4fef7SMatthew G. Knepley 
4776dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
47779566063dSJacob Faibussowitsch       PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
477868d4fef7SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
477969d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
4780dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
478161a622f3SMatthew G. Knepley       if (coordFunc && remap) {
47829566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
478351a74b61SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
478451a74b61SMatthew G. Knepley       }
478568d4fef7SMatthew G. Knepley     }
478668d4fef7SMatthew G. Knepley   }
47873cf6fe12SMatthew G. Knepley   /* Handle DMPlex coarsening */
47889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL, 0));
47899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy, 0));
4790b653a561SMatthew G. Knepley   if (coarsen && isHierarchy) {
4791b653a561SMatthew G. Knepley     DM *dms;
4792b653a561SMatthew G. Knepley 
47939566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(coarsen, &dms));
47949566063dSJacob Faibussowitsch     PetscCall(DMCoarsenHierarchy(dm, coarsen, dms));
4795b653a561SMatthew G. Knepley     /* Free DMs */
4796b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
4797dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
47989566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
4799b653a561SMatthew G. Knepley     }
48009566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
4801b653a561SMatthew G. Knepley   } else {
4802b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
48039318fe57SMatthew G. Knepley       DM             cdm;
48049318fe57SMatthew G. Knepley       PetscPointFunc coordFunc = ((DM_Plex *)dm->data)->coordFunc;
48053cf6fe12SMatthew G. Knepley 
4806dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
48079566063dSJacob Faibussowitsch       PetscCall(DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &cdm));
48083cf6fe12SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
480969d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &cdm));
4810dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
48119318fe57SMatthew G. Knepley       if (coordFunc) {
48129566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
48139318fe57SMatthew G. Knepley         ((DM_Plex *)dm->data)->coordFunc = coordFunc;
48149318fe57SMatthew G. Knepley       }
48153cf6fe12SMatthew G. Knepley     }
4816b653a561SMatthew G. Knepley   }
4817be664eb1SMatthew G. Knepley   // Handle coordinate remapping
4818be664eb1SMatthew G. Knepley   remap = PETSC_FALSE;
4819be664eb1SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_coord_remap", "Flag to control coordinate remapping", "", remap, &remap, NULL));
4820be664eb1SMatthew G. Knepley   if (remap) {
4821be664eb1SMatthew G. Knepley     DMPlexCoordMap map     = DM_COORD_MAP_NONE;
4822be664eb1SMatthew G. Knepley     PetscPointFunc mapFunc = NULL;
4823be664eb1SMatthew G. Knepley     PetscScalar    params[16];
4824f45b553cSPierre Jolivet     PetscInt       Np = PETSC_STATIC_ARRAY_LENGTH(params), cdim;
4825be664eb1SMatthew G. Knepley     MPI_Comm       comm;
4826be664eb1SMatthew G. Knepley 
4827be664eb1SMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4828be664eb1SMatthew G. Knepley     PetscCall(DMGetCoordinateDim(dm, &cdim));
4829be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsScalarArray("-dm_coord_map_params", "Parameters for the coordinate remapping", "", params, &Np, &flg));
4830be664eb1SMatthew G. Knepley     if (!flg) Np = 0;
4831be664eb1SMatthew G. Knepley     // TODO Allow user to pass a map function by name
4832be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsEnum("-dm_coord_map", "Coordinate mapping for built-in mesh", "", DMPlexCoordMaps, (PetscEnum)map, (PetscEnum *)&map, &flg));
4833be664eb1SMatthew G. Knepley     if (flg) {
4834be664eb1SMatthew G. Knepley       switch (map) {
4835be664eb1SMatthew G. Knepley       case DM_COORD_MAP_NONE:
4836be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
4837be664eb1SMatthew G. Knepley         break;
4838be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHEAR:
4839be664eb1SMatthew G. Knepley         mapFunc = coordMap_shear;
4840be664eb1SMatthew G. Knepley         if (!Np) {
4841be664eb1SMatthew G. Knepley           Np        = cdim + 1;
4842be664eb1SMatthew G. Knepley           params[0] = 0;
4843be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
4844be664eb1SMatthew G. Knepley         }
4845be664eb1SMatthew 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);
4846be664eb1SMatthew G. Knepley         break;
4847be664eb1SMatthew G. Knepley       case DM_COORD_MAP_FLARE:
4848be664eb1SMatthew G. Knepley         mapFunc = coordMap_flare;
4849be664eb1SMatthew G. Knepley         if (!Np) {
4850be664eb1SMatthew G. Knepley           Np        = cdim + 1;
4851be664eb1SMatthew G. Knepley           params[0] = 0;
4852be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
4853be664eb1SMatthew G. Knepley         }
4854be664eb1SMatthew 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);
4855be664eb1SMatthew G. Knepley         break;
4856be664eb1SMatthew G. Knepley       case DM_COORD_MAP_ANNULUS:
4857be664eb1SMatthew G. Knepley         mapFunc = coordMap_annulus;
4858be664eb1SMatthew G. Knepley         if (!Np) {
4859be664eb1SMatthew G. Knepley           Np        = 2;
4860be664eb1SMatthew G. Knepley           params[0] = 1.;
4861be664eb1SMatthew G. Knepley           params[1] = 2.;
4862be664eb1SMatthew G. Knepley         }
4863be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The annulus coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
4864be664eb1SMatthew G. Knepley         break;
4865be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHELL:
4866be664eb1SMatthew G. Knepley         mapFunc = coordMap_shell;
4867be664eb1SMatthew G. Knepley         if (!Np) {
4868be664eb1SMatthew G. Knepley           Np        = 2;
4869be664eb1SMatthew G. Knepley           params[0] = 1.;
4870be664eb1SMatthew G. Knepley           params[1] = 2.;
4871be664eb1SMatthew G. Knepley         }
4872be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The spherical shell coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
4873be664eb1SMatthew G. Knepley         break;
4874be664eb1SMatthew G. Knepley       default:
4875be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
4876be664eb1SMatthew G. Knepley       }
4877be664eb1SMatthew G. Knepley     }
4878be664eb1SMatthew G. Knepley     if (Np) {
4879be664eb1SMatthew G. Knepley       DM      cdm;
4880be664eb1SMatthew G. Knepley       PetscDS cds;
4881be664eb1SMatthew G. Knepley 
4882be664eb1SMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
4883be664eb1SMatthew G. Knepley       PetscCall(DMGetDS(cdm, &cds));
4884be664eb1SMatthew G. Knepley       PetscCall(PetscDSSetConstants(cds, Np, params));
4885be664eb1SMatthew G. Knepley     }
4886be664eb1SMatthew G. Knepley     PetscCall(DMPlexRemapGeometry(dm, 0.0, mapFunc));
4887be664eb1SMatthew G. Knepley   }
4888909dfd52SMatthew G. Knepley   /* Handle ghost cells */
48899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_create_fv_ghost_cells", "Flag to create finite volume ghost cells on the boundary", "DMCreate", ghostCells, &ghostCells, NULL));
4890909dfd52SMatthew G. Knepley   if (ghostCells) {
4891909dfd52SMatthew G. Knepley     DM   gdm;
4892909dfd52SMatthew G. Knepley     char lname[PETSC_MAX_PATH_LEN];
4893909dfd52SMatthew G. Knepley 
4894909dfd52SMatthew G. Knepley     lname[0] = '\0';
48959566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-dm_plex_fv_ghost_cells_label", "Label name for ghost cells boundary", "DMCreate", lname, lname, sizeof(lname), &flg));
48969566063dSJacob Faibussowitsch     PetscCall(DMPlexConstructGhostCells(dm, flg ? lname : NULL, NULL, &gdm));
489769d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &gdm));
4898909dfd52SMatthew G. Knepley   }
48996913077dSMatthew G. Knepley   /* Handle 1D order */
4900adc21957SMatthew G. Knepley   if (reorder != DM_REORDER_DEFAULT_FALSE && dim == 1) {
49016913077dSMatthew G. Knepley     DM           cdm, rdm;
49026913077dSMatthew G. Knepley     PetscDS      cds;
49036913077dSMatthew G. Knepley     PetscObject  obj;
49046913077dSMatthew G. Knepley     PetscClassId id = PETSC_OBJECT_CLASSID;
49056913077dSMatthew G. Knepley     IS           perm;
49066bc1bd01Sksagiyam     PetscInt     Nf;
49076913077dSMatthew G. Knepley     PetscBool    distributed;
49086913077dSMatthew G. Knepley 
49099566063dSJacob Faibussowitsch     PetscCall(DMPlexIsDistributed(dm, &distributed));
49109566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
49119566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
49129566063dSJacob Faibussowitsch     PetscCall(PetscDSGetNumFields(cds, &Nf));
49136913077dSMatthew G. Knepley     if (Nf) {
49149566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
49159566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
49166913077dSMatthew G. Knepley     }
49176bc1bd01Sksagiyam     if (!distributed && id != PETSCFE_CLASSID) {
49189566063dSJacob Faibussowitsch       PetscCall(DMPlexGetOrdering1D(dm, &perm));
49199566063dSJacob Faibussowitsch       PetscCall(DMPlexPermute(dm, perm, &rdm));
492069d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
49219566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&perm));
49226913077dSMatthew G. Knepley     }
49236913077dSMatthew G. Knepley   }
49243cf6fe12SMatthew G. Knepley /* Handle */
4925dd4c3f67SMatthew G. Knepley non_refine:
4926dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
4927d0609cedSBarry Smith   PetscOptionsHeadEnd();
49283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49290a6ba040SMatthew G. Knepley }
49300a6ba040SMatthew G. Knepley 
4931d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_Plex(DM dm, Vec *vec)
4932d71ae5a4SJacob Faibussowitsch {
4933552f7358SJed Brown   PetscFunctionBegin;
49349566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector_Section_Private(dm, vec));
49359566063dSJacob Faibussowitsch   /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */
49369566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex));
49379566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_Plex_Native));
49389566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex));
49399566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_Plex_Native));
49403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4941552f7358SJed Brown }
4942552f7358SJed Brown 
4943d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_Plex(DM dm, Vec *vec)
4944d71ae5a4SJacob Faibussowitsch {
4945552f7358SJed Brown   PetscFunctionBegin;
49469566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector_Section_Private(dm, vec));
49479566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex_Local));
49489566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex_Local));
49493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4950552f7358SJed Brown }
4951552f7358SJed Brown 
4952d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
4953d71ae5a4SJacob Faibussowitsch {
4954793f3fe5SMatthew G. Knepley   PetscInt depth, d;
4955793f3fe5SMatthew G. Knepley 
4956793f3fe5SMatthew G. Knepley   PetscFunctionBegin;
49579566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm, &depth));
4958793f3fe5SMatthew G. Knepley   if (depth == 1) {
49599566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &d));
49609566063dSJacob Faibussowitsch     if (dim == 0) PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
49619566063dSJacob Faibussowitsch     else if (dim == d) PetscCall(DMPlexGetDepthStratum(dm, 1, pStart, pEnd));
49629371c9d4SSatish Balay     else {
49639371c9d4SSatish Balay       *pStart = 0;
49649371c9d4SSatish Balay       *pEnd   = 0;
49659371c9d4SSatish Balay     }
4966793f3fe5SMatthew G. Knepley   } else {
49679566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
4968793f3fe5SMatthew G. Knepley   }
49693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4970793f3fe5SMatthew G. Knepley }
4971793f3fe5SMatthew G. Knepley 
4972d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetNeighbors_Plex(DM dm, PetscInt *nranks, const PetscMPIInt *ranks[])
4973d71ae5a4SJacob Faibussowitsch {
4974502a2867SDave May   PetscSF            sf;
49756497c311SBarry Smith   PetscMPIInt        niranks, njranks;
49766497c311SBarry Smith   PetscInt           n;
49770a19bb7dSprj-   const PetscMPIInt *iranks, *jranks;
49780a19bb7dSprj-   DM_Plex           *data = (DM_Plex *)dm->data;
4979502a2867SDave May 
49802f356facSMatthew G. Knepley   PetscFunctionBegin;
49819566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dm, &sf));
49820a19bb7dSprj-   if (!data->neighbors) {
49839566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
49849566063dSJacob Faibussowitsch     PetscCall(PetscSFGetRootRanks(sf, &njranks, &jranks, NULL, NULL, NULL));
49859566063dSJacob Faibussowitsch     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL));
49869566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors));
49879566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + 1, jranks, njranks));
49889566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks));
49890a19bb7dSprj-     n = njranks + niranks;
49909566063dSJacob Faibussowitsch     PetscCall(PetscSortRemoveDupsMPIInt(&n, data->neighbors + 1));
49910a19bb7dSprj-     /* The following cast should never fail: can't have more neighbors than PETSC_MPI_INT_MAX */
49929566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(n, data->neighbors));
49930a19bb7dSprj-   }
49940a19bb7dSprj-   if (nranks) *nranks = data->neighbors[0];
49950a19bb7dSprj-   if (ranks) {
49960a19bb7dSprj-     if (data->neighbors[0]) *ranks = data->neighbors + 1;
49970a19bb7dSprj-     else *ranks = NULL;
49980a19bb7dSprj-   }
49993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5000502a2867SDave May }
5001502a2867SDave May 
50021eb70e55SToby Isaac PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM, DM, Mat, Vec, Vec);
50031eb70e55SToby Isaac 
5004d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_Plex(DM dm)
5005d71ae5a4SJacob Faibussowitsch {
5006552f7358SJed Brown   PetscFunctionBegin;
5007552f7358SJed Brown   dm->ops->view                      = DMView_Plex;
50082c40f234SMatthew G. Knepley   dm->ops->load                      = DMLoad_Plex;
5009552f7358SJed Brown   dm->ops->setfromoptions            = DMSetFromOptions_Plex;
501038221697SMatthew G. Knepley   dm->ops->clone                     = DMClone_Plex;
5011552f7358SJed Brown   dm->ops->setup                     = DMSetUp_Plex;
50121bb6d2a8SBarry Smith   dm->ops->createlocalsection        = DMCreateLocalSection_Plex;
5013adc21957SMatthew G. Knepley   dm->ops->createsectionpermutation  = DMCreateSectionPermutation_Plex;
501466ad2231SToby Isaac   dm->ops->createdefaultconstraints  = DMCreateDefaultConstraints_Plex;
5015552f7358SJed Brown   dm->ops->createglobalvector        = DMCreateGlobalVector_Plex;
5016552f7358SJed Brown   dm->ops->createlocalvector         = DMCreateLocalVector_Plex;
5017184d77edSJed Brown   dm->ops->getlocaltoglobalmapping   = NULL;
50180298fd71SBarry Smith   dm->ops->createfieldis             = NULL;
5019552f7358SJed Brown   dm->ops->createcoordinatedm        = DMCreateCoordinateDM_Plex;
5020f19dbd58SToby Isaac   dm->ops->createcoordinatefield     = DMCreateCoordinateField_Plex;
50210a6ba040SMatthew G. Knepley   dm->ops->getcoloring               = NULL;
5022552f7358SJed Brown   dm->ops->creatematrix              = DMCreateMatrix_Plex;
5023bceba477SMatthew G. Knepley   dm->ops->createinterpolation       = DMCreateInterpolation_Plex;
5024bd041c0cSMatthew G. Knepley   dm->ops->createmassmatrix          = DMCreateMassMatrix_Plex;
5025b4937a87SMatthew G. Knepley   dm->ops->createmassmatrixlumped    = DMCreateMassMatrixLumped_Plex;
50265a84ad33SLisandro Dalcin   dm->ops->createinjection           = DMCreateInjection_Plex;
5027552f7358SJed Brown   dm->ops->refine                    = DMRefine_Plex;
50280a6ba040SMatthew G. Knepley   dm->ops->coarsen                   = DMCoarsen_Plex;
50290a6ba040SMatthew G. Knepley   dm->ops->refinehierarchy           = DMRefineHierarchy_Plex;
5030b653a561SMatthew G. Knepley   dm->ops->coarsenhierarchy          = DMCoarsenHierarchy_Plex;
5031d410b0cfSMatthew G. Knepley   dm->ops->extrude                   = DMExtrude_Plex;
50320298fd71SBarry Smith   dm->ops->globaltolocalbegin        = NULL;
50330298fd71SBarry Smith   dm->ops->globaltolocalend          = NULL;
50340298fd71SBarry Smith   dm->ops->localtoglobalbegin        = NULL;
50350298fd71SBarry Smith   dm->ops->localtoglobalend          = NULL;
5036552f7358SJed Brown   dm->ops->destroy                   = DMDestroy_Plex;
5037552f7358SJed Brown   dm->ops->createsubdm               = DMCreateSubDM_Plex;
50382adcc780SMatthew G. Knepley   dm->ops->createsuperdm             = DMCreateSuperDM_Plex;
5039793f3fe5SMatthew G. Knepley   dm->ops->getdimpoints              = DMGetDimPoints_Plex;
5040552f7358SJed Brown   dm->ops->locatepoints              = DMLocatePoints_Plex;
50410709b2feSToby Isaac   dm->ops->projectfunctionlocal      = DMProjectFunctionLocal_Plex;
50420709b2feSToby Isaac   dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_Plex;
5043bfc4295aSToby Isaac   dm->ops->projectfieldlocal         = DMProjectFieldLocal_Plex;
50448c6c5593SMatthew G. Knepley   dm->ops->projectfieldlabellocal    = DMProjectFieldLabelLocal_Plex;
5045ece3a9fcSMatthew G. Knepley   dm->ops->projectbdfieldlabellocal  = DMProjectBdFieldLabelLocal_Plex;
50460709b2feSToby Isaac   dm->ops->computel2diff             = DMComputeL2Diff_Plex;
5047b698f381SToby Isaac   dm->ops->computel2gradientdiff     = DMComputeL2GradientDiff_Plex;
50482a16baeaSToby Isaac   dm->ops->computel2fielddiff        = DMComputeL2FieldDiff_Plex;
504928d58a37SPierre Jolivet   dm->ops->getneighbors              = DMGetNeighbors_Plex;
50506c6a6b79SMatthew G. Knepley   dm->ops->getlocalboundingbox       = DMGetLocalBoundingBox_Coordinates;
5051907a3e9cSStefano Zampini   dm->ops->createdomaindecomposition = DMCreateDomainDecomposition_Plex;
5052907a3e9cSStefano Zampini   dm->ops->createddscatters          = DMCreateDomainDecompositionScatters_Plex;
50539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_Plex));
50546c51210dSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", DMPlexInsertTimeDerivativeBoundaryValues_Plex));
50559566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", DMSetUpGLVisViewer_Plex));
50569566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_Plex));
50579566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", DMPlexDistributeGetDefault_Plex));
50589566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", DMPlexDistributeSetDefault_Plex));
50596bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", DMPlexReorderGetDefault_Plex));
50606bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", DMPlexReorderSetDefault_Plex));
5061adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetDefault_C", DMReorderSectionGetDefault_Plex));
5062adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetDefault_C", DMReorderSectionSetDefault_Plex));
5063adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetType_C", DMReorderSectionGetType_Plex));
5064adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetType_C", DMReorderSectionSetType_Plex));
50659566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", DMInterpolateSolution_Plex));
5066c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMPlexGetOverlap_Plex));
5067c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", DMPlexSetOverlap_Plex));
5068d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", DMPlexGetUseCeed_Plex));
5069d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", DMPlexSetUseCeed_Plex));
50703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5071552f7358SJed Brown }
5072552f7358SJed Brown 
5073d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode DMClone_Plex(DM dm, DM *newdm)
5074d71ae5a4SJacob Faibussowitsch {
507563a16f15SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *)dm->data;
50761fca310dSJames Wright   const PetscSF *face_sfs;
50771fca310dSJames Wright   PetscInt       num_face_sfs;
507863a16f15SMatthew G. Knepley 
507963a16f15SMatthew G. Knepley   PetscFunctionBegin;
508063a16f15SMatthew G. Knepley   mesh->refct++;
508163a16f15SMatthew G. Knepley   (*newdm)->data = mesh;
50821fca310dSJames Wright   PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &face_sfs));
50831fca310dSJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(*newdm, num_face_sfs, (PetscSF *)face_sfs));
50849566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)*newdm, DMPLEX));
50859566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(*newdm));
50863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
508763a16f15SMatthew G. Knepley }
508863a16f15SMatthew G. Knepley 
50898818961aSMatthew G Knepley /*MC
5090a1cb98faSBarry Smith   DMPLEX = "plex" - A `DM` object that encapsulates an unstructured mesh, or CW Complex, which can be expressed using a Hasse Diagram.
509120f4b53cSBarry Smith                     In the local representation, `Vec`s contain all unknowns in the interior and shared boundary. This is
50928818961aSMatthew G Knepley                     specified by a PetscSection object. Ownership in the global representation is determined by
5093a1cb98faSBarry Smith                     ownership of the underlying `DMPLEX` points. This is specified by another `PetscSection` object.
50948818961aSMatthew G Knepley 
5095e5893cccSMatthew G. Knepley   Options Database Keys:
5096250712c9SMatthew G. Knepley + -dm_refine_pre                     - Refine mesh before distribution
5097250712c9SMatthew G. Knepley + -dm_refine_uniform_pre             - Choose uniform or generator-based refinement
5098250712c9SMatthew G. Knepley + -dm_refine_volume_limit_pre        - Cell volume limit after pre-refinement using generator
5099250712c9SMatthew G. Knepley . -dm_distribute                     - Distribute mesh across processes
5100250712c9SMatthew G. Knepley . -dm_distribute_overlap             - Number of cells to overlap for distribution
5101250712c9SMatthew G. Knepley . -dm_refine                         - Refine mesh after distribution
5102c3db174cSMatthew G. Knepley . -dm_localize <bool>                - Whether to localize coordinates for periodic meshes
5103c3db174cSMatthew G. Knepley . -dm_sparse_localize <bool>         - Whether to only localize cells on the periodic boundary
5104250712c9SMatthew G. Knepley . -dm_plex_hash_location             - Use grid hashing for point location
5105ddce0771SMatthew G. Knepley . -dm_plex_hash_box_faces <n,m,p>    - The number of divisions in each direction of the grid hash
5106f12cf164SMatthew G. Knepley . -dm_plex_partition_balance         - Attempt to evenly divide points on partition boundary between processes
5107f12cf164SMatthew G. Knepley . -dm_plex_remesh_bd                 - Allow changes to the boundary on remeshing
5108d5b43468SJose E. Roman . -dm_plex_max_projection_height     - Maximum mesh point height used to project locally
5109f12cf164SMatthew G. Knepley . -dm_plex_regular_refinement        - Use special nested projection algorithm for regular refinement
5110d02c7345SMatthew G. Knepley . -dm_plex_reorder_section           - Use specialized blocking if available
5111aaa8cc7dSPierre Jolivet . -dm_plex_check_all                 - Perform all checks below
5112f12cf164SMatthew G. Knepley . -dm_plex_check_symmetry            - Check that the adjacency information in the mesh is symmetric
5113f12cf164SMatthew G. Knepley . -dm_plex_check_skeleton <celltype> - Check that each cell has the correct number of vertices
5114f12cf164SMatthew 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
5115f12cf164SMatthew G. Knepley . -dm_plex_check_geometry            - Check that cells have positive volume
5116f12cf164SMatthew G. Knepley . -dm_view :mesh.tex:ascii_latex     - View the mesh in LaTeX/TikZ
5117e5893cccSMatthew G. Knepley . -dm_plex_view_scale <num>          - Scale the TikZ
51185962854dSMatthew G. Knepley . -dm_plex_print_fem <num>           - View FEM assembly information, such as element vectors and matrices
51195962854dSMatthew G. Knepley - -dm_plex_print_fvm <num>           - View FVM assembly information, such as flux updates
5120e5893cccSMatthew G. Knepley 
51218818961aSMatthew G Knepley   Level: intermediate
51228818961aSMatthew G Knepley 
51231cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMPlexCreate()`, `DMCreate()`, `DMSetType()`, `PetscSection`
51248818961aSMatthew G Knepley M*/
51258818961aSMatthew G Knepley 
5126d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm)
5127d71ae5a4SJacob Faibussowitsch {
5128552f7358SJed Brown   DM_Plex *mesh;
5129412e9a14SMatthew G. Knepley   PetscInt unit;
5130552f7358SJed Brown 
5131552f7358SJed Brown   PetscFunctionBegin;
5132f39ec787SMatthew G. Knepley   PetscCall(PetscCitationsRegister(PlexCitation, &Plexcite));
5133552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
51344dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mesh));
5135adc21957SMatthew G. Knepley   dm->reorderSection = DM_REORDER_DEFAULT_NOTSET;
5136552f7358SJed Brown   dm->data           = mesh;
5137552f7358SJed Brown 
5138552f7358SJed Brown   mesh->refct = 1;
51399566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection));
51409566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection));
5141552f7358SJed Brown   mesh->refinementUniform      = PETSC_TRUE;
5142552f7358SJed Brown   mesh->refinementLimit        = -1.0;
5143e600fa54SMatthew G. Knepley   mesh->distDefault            = PETSC_TRUE;
5144adc21957SMatthew G. Knepley   mesh->reorderDefault         = DM_REORDER_DEFAULT_NOTSET;
51451d1f2f2aSksagiyam   mesh->distributionName       = NULL;
51467d0f5628SVaclav Hapla   mesh->interpolated           = DMPLEX_INTERPOLATED_INVALID;
51477d0f5628SVaclav Hapla   mesh->interpolatedCollective = DMPLEX_INTERPOLATED_INVALID;
5148552f7358SJed Brown 
51499566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner));
51502e62ab5aSMatthew G. Knepley   mesh->remeshBd = PETSC_FALSE;
5151d9deefdfSMatthew G. Knepley 
51528865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
5153552f7358SJed Brown 
5154df0420ecSMatthew G. Knepley   mesh->depthState    = -1;
5155ba2698f1SMatthew G. Knepley   mesh->celltypeState = -1;
51566113b454SMatthew G. Knepley   mesh->printTol      = 1.0e-10;
5157c29ce622SStefano Zampini   mesh->nonempty_comm = MPI_COMM_SELF;
5158552f7358SJed Brown 
51599566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
51603ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5161552f7358SJed Brown }
5162552f7358SJed Brown 
5163552f7358SJed Brown /*@
5164a1cb98faSBarry Smith   DMPlexCreate - Creates a `DMPLEX` object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
5165552f7358SJed Brown 
5166d083f849SBarry Smith   Collective
5167552f7358SJed Brown 
5168552f7358SJed Brown   Input Parameter:
5169a1cb98faSBarry Smith . comm - The communicator for the `DMPLEX` object
5170552f7358SJed Brown 
5171552f7358SJed Brown   Output Parameter:
5172a1cb98faSBarry Smith . mesh - The `DMPLEX` object
5173552f7358SJed Brown 
5174552f7358SJed Brown   Level: beginner
5175552f7358SJed Brown 
517642747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMCreate()`, `DMSetType()`
5177552f7358SJed Brown @*/
5178d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
5179d71ae5a4SJacob Faibussowitsch {
5180552f7358SJed Brown   PetscFunctionBegin;
51814f572ea9SToby Isaac   PetscAssertPointer(mesh, 2);
51829566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, mesh));
51839566063dSJacob Faibussowitsch   PetscCall(DMSetType(*mesh, DMPLEX));
51843ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5185552f7358SJed Brown }
5186552f7358SJed Brown 
5187b09969d6SVaclav Hapla /*@C
5188a1cb98faSBarry Smith   DMPlexBuildFromCellListParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output)
5189a1cb98faSBarry Smith 
519020f4b53cSBarry Smith   Collective; No Fortran Support
5191b09969d6SVaclav Hapla 
5192b09969d6SVaclav Hapla   Input Parameters:
5193a1cb98faSBarry Smith + dm          - The `DM`
5194b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5195a1cb98faSBarry Smith . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5196a1cb98faSBarry Smith . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5197b09969d6SVaclav Hapla . numCorners  - The number of vertices for each cell
51985e488331SVaclav Hapla - cells       - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5199b09969d6SVaclav Hapla 
5200be8c289dSNicolas Barral   Output Parameters:
5201a1cb98faSBarry Smith + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5202be8c289dSNicolas Barral - verticesAdjSaved - (Optional) vertex adjacency array
5203b09969d6SVaclav Hapla 
5204b09969d6SVaclav Hapla   Level: advanced
5205b09969d6SVaclav Hapla 
5206a1cb98faSBarry Smith   Notes:
5207a1cb98faSBarry Smith   Two triangles sharing a face
5208a1cb98faSBarry Smith .vb
5209a1cb98faSBarry Smith 
5210a1cb98faSBarry Smith         2
5211a1cb98faSBarry Smith       / | \
5212a1cb98faSBarry Smith      /  |  \
5213a1cb98faSBarry Smith     /   |   \
5214a1cb98faSBarry Smith    0  0 | 1  3
5215a1cb98faSBarry Smith     \   |   /
5216a1cb98faSBarry Smith      \  |  /
5217a1cb98faSBarry Smith       \ | /
5218a1cb98faSBarry Smith         1
5219a1cb98faSBarry Smith .ve
5220a1cb98faSBarry Smith   would have input
5221a1cb98faSBarry Smith .vb
5222a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5223a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5224a1cb98faSBarry Smith .ve
5225a1cb98faSBarry Smith   which would result in the `DMPLEX`
5226a1cb98faSBarry Smith .vb
5227a1cb98faSBarry Smith 
5228a1cb98faSBarry Smith         4
5229a1cb98faSBarry Smith       / | \
5230a1cb98faSBarry Smith      /  |  \
5231a1cb98faSBarry Smith     /   |   \
5232a1cb98faSBarry Smith    2  0 | 1  5
5233a1cb98faSBarry Smith     \   |   /
5234a1cb98faSBarry Smith      \  |  /
5235a1cb98faSBarry Smith       \ | /
5236a1cb98faSBarry Smith         3
5237a1cb98faSBarry Smith .ve
5238a1cb98faSBarry Smith 
5239a1cb98faSBarry Smith   Vertices are implicitly numbered consecutively 0,...,NVertices.
5240a1cb98faSBarry Smith   Each rank owns a chunk of numVertices consecutive vertices.
5241a1cb98faSBarry Smith   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5242a1cb98faSBarry Smith   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
5243a1cb98faSBarry Smith   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5244a1cb98faSBarry Smith 
5245a1cb98faSBarry Smith   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5246a1cb98faSBarry Smith 
52471cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5248a1cb98faSBarry Smith           `PetscSF`
5249b09969d6SVaclav Hapla @*/
5250d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellListParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, const PetscInt cells[], PetscSF *vertexSF, PetscInt **verticesAdjSaved)
5251d71ae5a4SJacob Faibussowitsch {
52522464107aSksagiyam   PetscSF     sfPoint;
52532464107aSksagiyam   PetscLayout layout;
525482fb893eSVaclav Hapla   PetscInt    numVerticesAdj, *verticesAdj, *cones, c, p;
5255a47d0d45SMatthew G. Knepley 
5256a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
525725b6865aSVaclav Hapla   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
52589566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
525925b6865aSVaclav Hapla   /* Get/check global number of vertices */
526025b6865aSVaclav Hapla   {
526125b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
526225b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
526325b6865aSVaclav Hapla 
526425b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
5265*1690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
52669371c9d4SSatish Balay     for (i = 0; i < len; i++)
52679371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
526825b6865aSVaclav Hapla     ++NVerticesInCells;
5269712fec58SPierre Jolivet     PetscCall(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
527025b6865aSVaclav Hapla 
527125b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
52729371c9d4SSatish Balay     else
52739371c9d4SSatish 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);
527425b6865aSVaclav Hapla   }
52759079aca8SVaclav Hapla   /* Count locally unique vertices */
52769079aca8SVaclav Hapla   {
52779079aca8SVaclav Hapla     PetscHSetI vhash;
52789079aca8SVaclav Hapla     PetscInt   off = 0;
52799079aca8SVaclav Hapla 
52809566063dSJacob Faibussowitsch     PetscCall(PetscHSetICreate(&vhash));
5281a47d0d45SMatthew G. Knepley     for (c = 0; c < numCells; ++c) {
528248a46eb9SPierre Jolivet       for (p = 0; p < numCorners; ++p) PetscCall(PetscHSetIAdd(vhash, cells[c * numCorners + p]));
5283a47d0d45SMatthew G. Knepley     }
52849566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
52859566063dSJacob Faibussowitsch     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5286ad540459SPierre Jolivet     else verticesAdj = *verticesAdjSaved;
52879566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
52889566063dSJacob Faibussowitsch     PetscCall(PetscHSetIDestroy(&vhash));
528963a3b9bcSJacob Faibussowitsch     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
5290a47d0d45SMatthew G. Knepley   }
52919566063dSJacob Faibussowitsch   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5292a47d0d45SMatthew G. Knepley   /* Create cones */
52939566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
52949566063dSJacob Faibussowitsch   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
52959566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
52969566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5297a47d0d45SMatthew G. Knepley   for (c = 0; c < numCells; ++c) {
5298a47d0d45SMatthew G. Knepley     for (p = 0; p < numCorners; ++p) {
5299a47d0d45SMatthew G. Knepley       const PetscInt gv = cells[c * numCorners + p];
5300a47d0d45SMatthew G. Knepley       PetscInt       lv;
5301a47d0d45SMatthew G. Knepley 
53029079aca8SVaclav Hapla       /* Positions within verticesAdj form 0-based local vertex numbering;
53039079aca8SVaclav Hapla          we need to shift it by numCells to get correct DAG points (cells go first) */
53049566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
530563a3b9bcSJacob Faibussowitsch       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5306961cfab0SVaclav Hapla       cones[c * numCorners + p] = lv + numCells;
5307a47d0d45SMatthew G. Knepley     }
5308a47d0d45SMatthew G. Knepley   }
53092464107aSksagiyam   /* Build point sf */
53109566063dSJacob Faibussowitsch   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
53119566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetSize(layout, NVertices));
53129566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
53139566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetBlockSize(layout, 1));
53149566063dSJacob Faibussowitsch   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
53159566063dSJacob Faibussowitsch   PetscCall(PetscLayoutDestroy(&layout));
53169566063dSJacob Faibussowitsch   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
53179566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
53182464107aSksagiyam   if (dm->sf) {
53192464107aSksagiyam     const char *prefix;
53202464107aSksagiyam 
53219566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
53229566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
53232464107aSksagiyam   }
53249566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sfPoint));
53259566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&sfPoint));
5326f4f49eeaSPierre Jolivet   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
5327a47d0d45SMatthew G. Knepley   /* Fill in the rest of the topology structure */
53289566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
53299566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
53309566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
53313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5332a47d0d45SMatthew G. Knepley }
5333a47d0d45SMatthew G. Knepley 
5334cc4c1da9SBarry Smith /*@
5335a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellListParallel - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
5336a1cb98faSBarry Smith 
533720f4b53cSBarry Smith   Collective; No Fortran Support
5338b09969d6SVaclav Hapla 
5339b09969d6SVaclav Hapla   Input Parameters:
5340a1cb98faSBarry Smith + dm           - The `DM`
5341b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
5342a1cb98faSBarry Smith . sfVert       - `PetscSF` describing complete vertex ownership
5343b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5344b09969d6SVaclav Hapla 
5345b09969d6SVaclav Hapla   Level: advanced
5346b09969d6SVaclav Hapla 
53471cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellListParallel()`
5348b09969d6SVaclav Hapla @*/
5349d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellListParallel(DM dm, PetscInt spaceDim, PetscSF sfVert, const PetscReal vertexCoords[])
5350d71ae5a4SJacob Faibussowitsch {
5351a47d0d45SMatthew G. Knepley   PetscSection coordSection;
5352a47d0d45SMatthew G. Knepley   Vec          coordinates;
5353a47d0d45SMatthew G. Knepley   PetscScalar *coords;
53541edcf0b2SVaclav Hapla   PetscInt     numVertices, numVerticesAdj, coordSize, v, vStart, vEnd;
5355a47d0d45SMatthew G. Knepley 
5356a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
53579566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
53589566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
53591dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
53609566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
53619566063dSJacob Faibussowitsch   PetscCall(PetscSFGetGraph(sfVert, &numVertices, &numVerticesAdj, NULL, NULL));
53621dca8a05SBarry 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);
53639566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
53649566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
53659566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
53669566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
53671edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
53689566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
53699566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
5370a47d0d45SMatthew G. Knepley   }
53719566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
53729566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
53739566063dSJacob Faibussowitsch   PetscCall(VecCreate(PetscObjectComm((PetscObject)dm), &coordinates));
53749566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
53759566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
53769566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
53779566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
53789566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
5379a47d0d45SMatthew G. Knepley   {
5380a47d0d45SMatthew G. Knepley     MPI_Datatype coordtype;
5381a47d0d45SMatthew G. Knepley 
5382a47d0d45SMatthew G. Knepley     /* Need a temp buffer for coords if we have complex/single */
53836497c311SBarry Smith     PetscCallMPI(MPI_Type_contiguous((PetscMPIInt)spaceDim, MPIU_SCALAR, &coordtype));
53849566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_commit(&coordtype));
538521016a8bSBarry Smith #if defined(PETSC_USE_COMPLEX)
538621016a8bSBarry Smith     {
538721016a8bSBarry Smith       PetscScalar *svertexCoords;
538821016a8bSBarry Smith       PetscInt     i;
53899566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * spaceDim, &svertexCoords));
53903612f820SVaclav Hapla       for (i = 0; i < numVertices * spaceDim; i++) svertexCoords[i] = vertexCoords[i];
53919566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
53929566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
53939566063dSJacob Faibussowitsch       PetscCall(PetscFree(svertexCoords));
539421016a8bSBarry Smith     }
539521016a8bSBarry Smith #else
53969566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastBegin(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
53979566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastEnd(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
539821016a8bSBarry Smith #endif
53999566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_free(&coordtype));
5400a47d0d45SMatthew G. Knepley   }
54019566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
54029566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
54039566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
54049566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
54053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5406a47d0d45SMatthew G. Knepley }
5407a47d0d45SMatthew G. Knepley 
5408c3edce3dSSatish Balay /*@
5409a1cb98faSBarry Smith   DMPlexCreateFromCellListParallelPetsc - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output)
5410a1cb98faSBarry Smith 
5411a1cb98faSBarry Smith   Collective
5412a47d0d45SMatthew G. Knepley 
5413a47d0d45SMatthew G. Knepley   Input Parameters:
5414a47d0d45SMatthew G. Knepley + comm         - The communicator
5415a47d0d45SMatthew G. Knepley . dim          - The topological dimension of the mesh
5416a47d0d45SMatthew G. Knepley . numCells     - The number of cells owned by this process
5417a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
5418a1cb98faSBarry Smith . NVertices    - The global number of vertices, or `PETSC_DECIDE`
5419a47d0d45SMatthew G. Knepley . numCorners   - The number of vertices for each cell
5420a47d0d45SMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
5421a47d0d45SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5422a47d0d45SMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
5423a47d0d45SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5424a47d0d45SMatthew G. Knepley 
5425d8d19677SJose E. Roman   Output Parameters:
5426a1cb98faSBarry Smith + dm          - The `DM`
5427a1cb98faSBarry Smith . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
542860225df5SJacob Faibussowitsch - verticesAdj - (Optional) vertex adjacency array
5429a47d0d45SMatthew G. Knepley 
5430b09969d6SVaclav Hapla   Level: intermediate
5431a47d0d45SMatthew G. Knepley 
5432a1cb98faSBarry Smith   Notes:
5433a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
5434a1cb98faSBarry Smith   `DMPlexBuildFromCellListParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
5435a1cb98faSBarry Smith 
5436a1cb98faSBarry Smith   See `DMPlexBuildFromCellListParallel()` for an example and details about the topology-related parameters.
5437a1cb98faSBarry Smith 
5438a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
5439a1cb98faSBarry Smith 
54401cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
5441a47d0d45SMatthew G. Knepley @*/
5442d71ae5a4SJacob 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)
5443d71ae5a4SJacob Faibussowitsch {
5444a47d0d45SMatthew G. Knepley   PetscSF sfVert;
5445a47d0d45SMatthew G. Knepley 
5446a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
54479566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
54489566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
5449a47d0d45SMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
5450064a246eSJacob Faibussowitsch   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
54519566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
54529566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildFromCellListParallel(*dm, numCells, numVertices, NVertices, numCorners, cells, &sfVert, verticesAdj));
5453a47d0d45SMatthew G. Knepley   if (interpolate) {
54545fd9971aSMatthew G. Knepley     DM idm;
5455a47d0d45SMatthew G. Knepley 
54569566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
54579566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
5458a47d0d45SMatthew G. Knepley     *dm = idm;
5459a47d0d45SMatthew G. Knepley   }
54609566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
546118d54ad4SMichael Lange   if (vertexSF) *vertexSF = sfVert;
54629566063dSJacob Faibussowitsch   else PetscCall(PetscSFDestroy(&sfVert));
54633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5464a47d0d45SMatthew G. Knepley }
5465a47d0d45SMatthew G. Knepley 
5466cc4c1da9SBarry Smith /*@
5467a1cb98faSBarry Smith   DMPlexBuildFromCellList - Build `DMPLEX` topology from a list of vertices for each cell (common mesh generator output)
5468a1cb98faSBarry Smith 
546920f4b53cSBarry Smith   Collective; No Fortran Support
54709298eaa6SMatthew G Knepley 
54719298eaa6SMatthew G Knepley   Input Parameters:
5472a1cb98faSBarry Smith + dm          - The `DM`
5473b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5474a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DETERMINE`
54759298eaa6SMatthew G Knepley . numCorners  - The number of vertices for each cell
5476a3b724e8SBarry Smith - cells       - An array of `numCells` x `numCorners` numbers, the global vertex numbers for each cell
54779298eaa6SMatthew G Knepley 
5478b09969d6SVaclav Hapla   Level: advanced
54799298eaa6SMatthew G Knepley 
5480b09969d6SVaclav Hapla   Notes:
5481b09969d6SVaclav Hapla   Two triangles sharing a face
5482a1cb98faSBarry Smith .vb
54839298eaa6SMatthew G Knepley 
5484a1cb98faSBarry Smith         2
5485a1cb98faSBarry Smith       / | \
5486a1cb98faSBarry Smith      /  |  \
5487a1cb98faSBarry Smith     /   |   \
5488a1cb98faSBarry Smith    0  0 | 1  3
5489a1cb98faSBarry Smith     \   |   /
5490a1cb98faSBarry Smith      \  |  /
5491a1cb98faSBarry Smith       \ | /
5492a1cb98faSBarry Smith         1
5493a1cb98faSBarry Smith .ve
5494a1cb98faSBarry Smith   would have input
5495a1cb98faSBarry Smith .vb
5496a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5497a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5498a1cb98faSBarry Smith .ve
5499a1cb98faSBarry Smith   which would result in the `DMPLEX`
5500a1cb98faSBarry Smith .vb
5501a1cb98faSBarry Smith 
5502a1cb98faSBarry Smith         4
5503a1cb98faSBarry Smith       / | \
5504a1cb98faSBarry Smith      /  |  \
5505a1cb98faSBarry Smith     /   |   \
5506a1cb98faSBarry Smith    2  0 | 1  5
5507a1cb98faSBarry Smith     \   |   /
5508a1cb98faSBarry Smith      \  |  /
5509a1cb98faSBarry Smith       \ | /
5510a1cb98faSBarry Smith         3
5511a1cb98faSBarry Smith .ve
5512a1cb98faSBarry Smith 
5513a1cb98faSBarry Smith   If numVertices is `PETSC_DETERMINE`, it is computed by PETSc as the maximum vertex index in cells + 1.
551425b6865aSVaclav Hapla 
55151cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListPetsc()`
5516b09969d6SVaclav Hapla @*/
5517d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellList(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const PetscInt cells[])
5518d71ae5a4SJacob Faibussowitsch {
5519961cfab0SVaclav Hapla   PetscInt *cones, c, p, dim;
5520b09969d6SVaclav Hapla 
5521b09969d6SVaclav Hapla   PetscFunctionBegin;
55229566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
55239566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
552425b6865aSVaclav Hapla   /* Get/check global number of vertices */
552525b6865aSVaclav Hapla   {
552625b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
552725b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
552825b6865aSVaclav Hapla 
552925b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
5530*1690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
55319371c9d4SSatish Balay     for (i = 0; i < len; i++)
55329371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
553325b6865aSVaclav Hapla     ++NVerticesInCells;
553425b6865aSVaclav Hapla 
553525b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE) numVertices = NVerticesInCells;
55369371c9d4SSatish Balay     else
55379371c9d4SSatish 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);
553825b6865aSVaclav Hapla   }
55399566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
554048a46eb9SPierre Jolivet   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
55419566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
55429566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5543b09969d6SVaclav Hapla   for (c = 0; c < numCells; ++c) {
5544ad540459SPierre Jolivet     for (p = 0; p < numCorners; ++p) cones[c * numCorners + p] = cells[c * numCorners + p] + numCells;
5545b09969d6SVaclav Hapla   }
55469566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
55479566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
55489566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
55493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5550b09969d6SVaclav Hapla }
5551b09969d6SVaclav Hapla 
5552cc4c1da9SBarry Smith /*@
5553a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellList - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
5554a1cb98faSBarry Smith 
5555cc4c1da9SBarry Smith   Collective
5556b09969d6SVaclav Hapla 
5557b09969d6SVaclav Hapla   Input Parameters:
5558a1cb98faSBarry Smith + dm           - The `DM`
5559b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
5560b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
5561b09969d6SVaclav Hapla 
5562b09969d6SVaclav Hapla   Level: advanced
5563b09969d6SVaclav Hapla 
55641cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellList()`
5565b09969d6SVaclav Hapla @*/
5566d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellList(DM dm, PetscInt spaceDim, const PetscReal vertexCoords[])
5567d71ae5a4SJacob Faibussowitsch {
5568b09969d6SVaclav Hapla   PetscSection coordSection;
5569b09969d6SVaclav Hapla   Vec          coordinates;
5570b09969d6SVaclav Hapla   DM           cdm;
5571b09969d6SVaclav Hapla   PetscScalar *coords;
55721edcf0b2SVaclav Hapla   PetscInt     v, vStart, vEnd, d;
5573b09969d6SVaclav Hapla 
5574b09969d6SVaclav Hapla   PetscFunctionBegin;
55759566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
55769566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
55771dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
55789566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
55799566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
55809566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
55819566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
55829566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
55831edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
55849566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
55859566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
5586b09969d6SVaclav Hapla   }
55879566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5588b09969d6SVaclav Hapla 
55899566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
55909566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(cdm, &coordinates));
55919566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
55929566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
55939566063dSJacob Faibussowitsch   PetscCall(VecGetArrayWrite(coordinates, &coords));
55941edcf0b2SVaclav Hapla   for (v = 0; v < vEnd - vStart; ++v) {
5595ad540459SPierre Jolivet     for (d = 0; d < spaceDim; ++d) coords[v * spaceDim + d] = vertexCoords[v * spaceDim + d];
5596b09969d6SVaclav Hapla   }
55979566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayWrite(coordinates, &coords));
55989566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
55999566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
56009566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
56013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5602b09969d6SVaclav Hapla }
5603b09969d6SVaclav Hapla 
5604b09969d6SVaclav Hapla /*@
5605a1cb98faSBarry Smith   DMPlexCreateFromCellListPetsc - Create `DMPLEX` from a list of vertices for each cell (common mesh generator output), but only process 0 takes in the input
56063df08285SMatthew G. Knepley 
5607a1cb98faSBarry Smith   Collective
5608b09969d6SVaclav Hapla 
5609b09969d6SVaclav Hapla   Input Parameters:
5610b09969d6SVaclav Hapla + comm         - The communicator
5611b09969d6SVaclav Hapla . dim          - The topological dimension of the mesh
56123df08285SMatthew G. Knepley . numCells     - The number of cells, only on process 0
5613a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`, only on process 0
56143df08285SMatthew G. Knepley . numCorners   - The number of vertices for each cell, only on process 0
5615b09969d6SVaclav Hapla . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
56163df08285SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the vertices for each cell, only on process 0
5617b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
56183df08285SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex, only on process 0
5619b09969d6SVaclav Hapla 
5620b09969d6SVaclav Hapla   Output Parameter:
5621a1cb98faSBarry Smith . dm - The `DM`, which only has points on process 0
562225b6865aSVaclav Hapla 
5623b09969d6SVaclav Hapla   Level: intermediate
5624b09969d6SVaclav Hapla 
5625a1cb98faSBarry Smith   Notes:
5626a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, `DMPlexBuildFromCellList()`,
5627a1cb98faSBarry Smith   `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellList()`
5628a1cb98faSBarry Smith 
5629a1cb98faSBarry Smith   See `DMPlexBuildFromCellList()` for an example and details about the topology-related parameters.
5630a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellList()` for details about the geometry-related parameters.
5631a1cb98faSBarry Smith   See `DMPlexCreateFromCellListParallelPetsc()` for parallel input
5632a1cb98faSBarry Smith 
56331cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellList()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
56349298eaa6SMatthew G Knepley @*/
5635d71ae5a4SJacob 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)
5636d71ae5a4SJacob Faibussowitsch {
56373df08285SMatthew G. Knepley   PetscMPIInt rank;
56389298eaa6SMatthew G Knepley 
56399298eaa6SMatthew G Knepley   PetscFunctionBegin;
564028b400f6SJacob 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.");
56419566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
56429566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
56439566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
56449566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
5645c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(*dm, numCells, numVertices, numCorners, cells));
56469566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(*dm, 0, 0, 0, NULL));
56479298eaa6SMatthew G Knepley   if (interpolate) {
56485fd9971aSMatthew G. Knepley     DM idm;
56499298eaa6SMatthew G Knepley 
56509566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
56519566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
56529298eaa6SMatthew G Knepley     *dm = idm;
56539298eaa6SMatthew G Knepley   }
5654c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, vertexCoords));
56559566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, NULL));
56563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
56579298eaa6SMatthew G Knepley }
56589298eaa6SMatthew G Knepley 
5659939f6067SMatthew G. Knepley /*@
566020f4b53cSBarry Smith   DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a `DM`
5661939f6067SMatthew G. Knepley 
5662939f6067SMatthew G. Knepley   Input Parameters:
566320f4b53cSBarry Smith + dm               - The empty `DM` object, usually from `DMCreate()` and `DMSetDimension()`
5664939f6067SMatthew G. Knepley . depth            - The depth of the DAG
566520f4b53cSBarry Smith . numPoints        - Array of size depth + 1 containing the number of points at each `depth`
5666939f6067SMatthew G. Knepley . coneSize         - The cone size of each point
5667939f6067SMatthew G. Knepley . cones            - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point
5668939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point
566920f4b53cSBarry Smith - vertexCoords     - An array of `numPoints`[0]*spacedim numbers representing the coordinates of each vertex, with spacedim the value set via `DMSetCoordinateDim()`
5670939f6067SMatthew G. Knepley 
5671939f6067SMatthew G. Knepley   Output Parameter:
567220f4b53cSBarry Smith . dm - The `DM`
567320f4b53cSBarry Smith 
567420f4b53cSBarry Smith   Level: advanced
5675939f6067SMatthew G. Knepley 
5676a1cb98faSBarry Smith   Note:
5677a1cb98faSBarry Smith   Two triangles sharing a face would have input
5678a1cb98faSBarry Smith .vb
5679a1cb98faSBarry Smith   depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0]
5680a1cb98faSBarry Smith   cones = [2 3 4  3 5 4], coneOrientations = [0 0 0  0 0 0]
5681a1cb98faSBarry Smith  vertexCoords = [-1.0 0.0  0.0 -1.0  0.0 1.0  1.0 0.0]
5682a1cb98faSBarry Smith .ve
5683939f6067SMatthew G. Knepley   which would result in the DMPlex
5684a1cb98faSBarry Smith .vb
5685a1cb98faSBarry Smith         4
5686a1cb98faSBarry Smith       / | \
5687a1cb98faSBarry Smith      /  |  \
5688a1cb98faSBarry Smith     /   |   \
5689a1cb98faSBarry Smith    2  0 | 1  5
5690a1cb98faSBarry Smith     \   |   /
5691a1cb98faSBarry Smith      \  |  /
5692a1cb98faSBarry Smith       \ | /
5693a1cb98faSBarry Smith         3
5694a1cb98faSBarry Smith .ve
5695a1cb98faSBarry Smith   Notice that all points are numbered consecutively, unlike `DMPlexCreateFromCellListPetsc()`
5696939f6067SMatthew G. Knepley 
56971cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
5698939f6067SMatthew G. Knepley @*/
5699d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
5700d71ae5a4SJacob Faibussowitsch {
57019298eaa6SMatthew G Knepley   Vec          coordinates;
57029298eaa6SMatthew G Knepley   PetscSection coordSection;
57039298eaa6SMatthew G Knepley   PetscScalar *coords;
5704811e8653SToby Isaac   PetscInt     coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off;
57059298eaa6SMatthew G Knepley 
57069298eaa6SMatthew G Knepley   PetscFunctionBegin;
57079566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
57089566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dimEmbed));
570963a3b9bcSJacob Faibussowitsch   PetscCheck(dimEmbed >= dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Embedding dimension %" PetscInt_FMT " cannot be less than intrinsic dimension %" PetscInt_FMT, dimEmbed, dim);
57109298eaa6SMatthew G Knepley   for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
57119566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, pStart, pEnd));
57129298eaa6SMatthew G Knepley   for (p = pStart; p < pEnd; ++p) {
57139566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeSize(dm, p, coneSize[p - pStart]));
5714ad540459SPierre Jolivet     if (firstVertex < 0 && !coneSize[p - pStart]) firstVertex = p - pStart;
571597e052ccSToby Isaac   }
57161dca8a05SBarry Smith   PetscCheck(firstVertex >= 0 || !numPoints[0], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Expected %" PetscInt_FMT " vertices but could not find any", numPoints[0]);
57179566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm)); /* Allocate space for cones */
57189298eaa6SMatthew G Knepley   for (p = pStart, off = 0; p < pEnd; off += coneSize[p - pStart], ++p) {
57199566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCone(dm, p, &cones[off]));
57209566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeOrientation(dm, p, &coneOrientations[off]));
57219298eaa6SMatthew G Knepley   }
57229566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
57239566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
57249298eaa6SMatthew G Knepley   /* Build coordinates */
57259566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
57269566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
57279566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed));
57289566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numPoints[0]));
57299298eaa6SMatthew G Knepley   for (v = firstVertex; v < firstVertex + numPoints[0]; ++v) {
57309566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed));
57319566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed));
57329298eaa6SMatthew G Knepley   }
57339566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
57349566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
57359566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
57369566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
57379566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
57389566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, dimEmbed));
57399566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
57409318fe57SMatthew G. Knepley   if (vertexCoords) {
57419566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
57429298eaa6SMatthew G Knepley     for (v = 0; v < numPoints[0]; ++v) {
57439298eaa6SMatthew G Knepley       PetscInt off;
57449298eaa6SMatthew G Knepley 
57459566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSection, v + firstVertex, &off));
5746ad540459SPierre Jolivet       for (d = 0; d < dimEmbed; ++d) coords[off + d] = vertexCoords[v * dimEmbed + d];
57479298eaa6SMatthew G Knepley     }
57489318fe57SMatthew G. Knepley   }
57499566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
57509566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
57519566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
57523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
57539298eaa6SMatthew G Knepley }
57548415267dSToby Isaac 
5755a4e35b19SJacob Faibussowitsch /*
5756a1cb98faSBarry Smith   DMPlexCreateCellVertexFromFile - Create a `DMPLEX` mesh from a simple cell-vertex file.
5757a1cb98faSBarry Smith 
5758a1cb98faSBarry Smith   Collective
57598ca92349SMatthew G. Knepley 
57608ca92349SMatthew G. Knepley + comm        - The MPI communicator
57618ca92349SMatthew G. Knepley . filename    - Name of the .dat file
57628ca92349SMatthew G. Knepley - interpolate - Create faces and edges in the mesh
57638ca92349SMatthew G. Knepley 
57648ca92349SMatthew G. Knepley   Output Parameter:
5765a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
57668ca92349SMatthew G. Knepley 
57678ca92349SMatthew G. Knepley   Level: beginner
57688ca92349SMatthew G. Knepley 
5769a1cb98faSBarry Smith   Note:
5770a1cb98faSBarry Smith   The format is the simplest possible:
5771a1cb98faSBarry Smith .vb
5772d0812dedSMatthew G. Knepley   dim Ne Nv Nc Nl
5773d0812dedSMatthew G. Knepley   v_1 v_2 ... v_Nc
5774d0812dedSMatthew G. Knepley   ...
5775d0812dedSMatthew G. Knepley   x y z marker_1 ... marker_Nl
5776a1cb98faSBarry Smith .ve
5777a1cb98faSBarry Smith 
5778a1cb98faSBarry Smith   Developer Note:
5779a1cb98faSBarry Smith   Should use a `PetscViewer` not a filename
5780a1cb98faSBarry Smith 
57816afe31f6SMartin Diehl .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateGmsh()`, `DMPlexCreate()`
5782a4e35b19SJacob Faibussowitsch */
5783ff6a9541SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCellVertexFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
5784d71ae5a4SJacob Faibussowitsch {
57858ca92349SMatthew G. Knepley   DMLabel      marker;
57868ca92349SMatthew G. Knepley   PetscViewer  viewer;
57878ca92349SMatthew G. Knepley   Vec          coordinates;
57888ca92349SMatthew G. Knepley   PetscSection coordSection;
57898ca92349SMatthew G. Knepley   PetscScalar *coords;
57908ca92349SMatthew G. Knepley   char         line[PETSC_MAX_PATH_LEN];
5791d0812dedSMatthew G. Knepley   PetscInt     cdim, coordSize, v, c, d;
57928ca92349SMatthew G. Knepley   PetscMPIInt  rank;
5793d0812dedSMatthew G. Knepley   int          snum, dim, Nv, Nc, Ncn, Nl;
57948ca92349SMatthew G. Knepley 
57958ca92349SMatthew G. Knepley   PetscFunctionBegin;
57969566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
57979566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(comm, &viewer));
57989566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII));
57999566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
58009566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetName(viewer, filename));
5801dd400576SPatrick Sanan   if (rank == 0) {
5802d0812dedSMatthew G. Knepley     PetscCall(PetscViewerRead(viewer, line, 5, NULL, PETSC_STRING));
5803d0812dedSMatthew G. Knepley     snum = sscanf(line, "%d %d %d %d %d", &dim, &Nc, &Nv, &Ncn, &Nl);
5804d0812dedSMatthew G. Knepley     PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
580525ce1634SJed Brown   } else {
5806f8d5e320SMatthew G. Knepley     Nc = Nv = Ncn = Nl = 0;
58078ca92349SMatthew G. Knepley   }
5808d0812dedSMatthew G. Knepley   PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm));
5809d0812dedSMatthew G. Knepley   cdim = (PetscInt)dim;
58109566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
58119566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
58129566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(*dm, 0, Nc + Nv));
5813d0812dedSMatthew G. Knepley   PetscCall(DMSetDimension(*dm, (PetscInt)dim));
58149566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(*dm, cdim));
58158ca92349SMatthew G. Knepley   /* Read topology */
5816dd400576SPatrick Sanan   if (rank == 0) {
5817f8d5e320SMatthew G. Knepley     char     format[PETSC_MAX_PATH_LEN];
5818f8d5e320SMatthew G. Knepley     PetscInt cone[8];
58198ca92349SMatthew G. Knepley     int      vbuf[8], v;
58208ca92349SMatthew G. Knepley 
58219371c9d4SSatish Balay     for (c = 0; c < Ncn; ++c) {
58229371c9d4SSatish Balay       format[c * 3 + 0] = '%';
58239371c9d4SSatish Balay       format[c * 3 + 1] = 'd';
58249371c9d4SSatish Balay       format[c * 3 + 2] = ' ';
58259371c9d4SSatish Balay     }
5826f8d5e320SMatthew G. Knepley     format[Ncn * 3 - 1] = '\0';
58279566063dSJacob Faibussowitsch     for (c = 0; c < Nc; ++c) PetscCall(DMPlexSetConeSize(*dm, c, Ncn));
58289566063dSJacob Faibussowitsch     PetscCall(DMSetUp(*dm));
58298ca92349SMatthew G. Knepley     for (c = 0; c < Nc; ++c) {
58309566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, Ncn, NULL, PETSC_STRING));
5831f8d5e320SMatthew G. Knepley       switch (Ncn) {
5832d71ae5a4SJacob Faibussowitsch       case 2:
5833d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1]);
5834d71ae5a4SJacob Faibussowitsch         break;
5835d71ae5a4SJacob Faibussowitsch       case 3:
5836d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2]);
5837d71ae5a4SJacob Faibussowitsch         break;
5838d71ae5a4SJacob Faibussowitsch       case 4:
5839d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3]);
5840d71ae5a4SJacob Faibussowitsch         break;
5841d71ae5a4SJacob Faibussowitsch       case 6:
5842d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5]);
5843d71ae5a4SJacob Faibussowitsch         break;
5844d71ae5a4SJacob Faibussowitsch       case 8:
5845d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5], &vbuf[6], &vbuf[7]);
5846d71ae5a4SJacob Faibussowitsch         break;
5847d71ae5a4SJacob Faibussowitsch       default:
5848d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No cell shape with %d vertices", Ncn);
5849f8d5e320SMatthew G. Knepley       }
585008401ef6SPierre Jolivet       PetscCheck(snum == Ncn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
5851f8d5e320SMatthew G. Knepley       for (v = 0; v < Ncn; ++v) cone[v] = vbuf[v] + Nc;
58528ca92349SMatthew G. Knepley       /* Hexahedra are inverted */
5853f8d5e320SMatthew G. Knepley       if (Ncn == 8) {
58548ca92349SMatthew G. Knepley         PetscInt tmp = cone[1];
58558ca92349SMatthew G. Knepley         cone[1]      = cone[3];
58568ca92349SMatthew G. Knepley         cone[3]      = tmp;
58578ca92349SMatthew G. Knepley       }
58589566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(*dm, c, cone));
58598ca92349SMatthew G. Knepley     }
58608ca92349SMatthew G. Knepley   }
58619566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(*dm));
58629566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(*dm));
58638ca92349SMatthew G. Knepley   /* Read coordinates */
58649566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(*dm, &coordSection));
58659566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
58669566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
58679566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, Nc, Nc + Nv));
58688ca92349SMatthew G. Knepley   for (v = Nc; v < Nc + Nv; ++v) {
58699566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
58709566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
58718ca92349SMatthew G. Knepley   }
58729566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
58739566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
58749566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
58759566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
58769566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
58779566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
58789566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
58799566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
5880dd400576SPatrick Sanan   if (rank == 0) {
5881f8d5e320SMatthew G. Knepley     char   format[PETSC_MAX_PATH_LEN];
58828ca92349SMatthew G. Knepley     double x[3];
5883f8d5e320SMatthew G. Knepley     int    l, val[3];
58848ca92349SMatthew G. Knepley 
5885f8d5e320SMatthew G. Knepley     if (Nl) {
58869371c9d4SSatish Balay       for (l = 0; l < Nl; ++l) {
58879371c9d4SSatish Balay         format[l * 3 + 0] = '%';
58889371c9d4SSatish Balay         format[l * 3 + 1] = 'd';
58899371c9d4SSatish Balay         format[l * 3 + 2] = ' ';
58909371c9d4SSatish Balay       }
5891f8d5e320SMatthew G. Knepley       format[Nl * 3 - 1] = '\0';
58929566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
58939566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &marker));
5894f8d5e320SMatthew G. Knepley     }
58958ca92349SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
58969566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, 3 + Nl, NULL, PETSC_STRING));
5897f8d5e320SMatthew G. Knepley       snum = sscanf(line, "%lg %lg %lg", &x[0], &x[1], &x[2]);
589808401ef6SPierre Jolivet       PetscCheck(snum == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
5899f8d5e320SMatthew G. Knepley       switch (Nl) {
5900d71ae5a4SJacob Faibussowitsch       case 0:
5901d71ae5a4SJacob Faibussowitsch         snum = 0;
5902d71ae5a4SJacob Faibussowitsch         break;
5903d71ae5a4SJacob Faibussowitsch       case 1:
5904d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0]);
5905d71ae5a4SJacob Faibussowitsch         break;
5906d71ae5a4SJacob Faibussowitsch       case 2:
5907d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1]);
5908d71ae5a4SJacob Faibussowitsch         break;
5909d71ae5a4SJacob Faibussowitsch       case 3:
5910d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1], &val[2]);
5911d71ae5a4SJacob Faibussowitsch         break;
5912d71ae5a4SJacob Faibussowitsch       default:
5913d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Request support for %d labels", Nl);
5914f8d5e320SMatthew G. Knepley       }
591508401ef6SPierre Jolivet       PetscCheck(snum == Nl, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
59168ca92349SMatthew G. Knepley       for (d = 0; d < cdim; ++d) coords[v * cdim + d] = x[d];
59179566063dSJacob Faibussowitsch       for (l = 0; l < Nl; ++l) PetscCall(DMLabelSetValue(marker, v + Nc, val[l]));
59188ca92349SMatthew G. Knepley     }
59198ca92349SMatthew G. Knepley   }
59209566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
59219566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
59229566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
59239566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&viewer));
59248ca92349SMatthew G. Knepley   if (interpolate) {
59258ca92349SMatthew G. Knepley     DM      idm;
59268ca92349SMatthew G. Knepley     DMLabel bdlabel;
59278ca92349SMatthew G. Knepley 
59289566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
59299566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
59308ca92349SMatthew G. Knepley     *dm = idm;
59318ca92349SMatthew G. Knepley 
5932f8d5e320SMatthew G. Knepley     if (!Nl) {
59339566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
59349566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &bdlabel));
59359566063dSJacob Faibussowitsch       PetscCall(DMPlexMarkBoundaryFaces(*dm, PETSC_DETERMINE, bdlabel));
59369566063dSJacob Faibussowitsch       PetscCall(DMPlexLabelComplete(*dm, bdlabel));
59378ca92349SMatthew G. Knepley     }
5938f8d5e320SMatthew G. Knepley   }
59393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
59408ca92349SMatthew G. Knepley }
59418ca92349SMatthew G. Knepley 
5942cc4c1da9SBarry Smith /*@
5943a1cb98faSBarry Smith   DMPlexCreateFromFile - This takes a filename and produces a `DM`
5944a1cb98faSBarry Smith 
5945a1cb98faSBarry Smith   Collective
5946ca522641SMatthew G. Knepley 
5947ca522641SMatthew G. Knepley   Input Parameters:
5948ca522641SMatthew G. Knepley + comm        - The communicator
5949ca522641SMatthew G. Knepley . filename    - A file name
5950a1cb98faSBarry Smith . plexname    - The object name of the resulting `DM`, also used for intra-datafile lookup by some formats
5951ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
5952ca522641SMatthew G. Knepley 
5953ca522641SMatthew G. Knepley   Output Parameter:
5954a1cb98faSBarry Smith . dm - The `DM`
5955ca522641SMatthew G. Knepley 
5956a1cb98faSBarry Smith   Options Database Key:
5957a1cb98faSBarry Smith . -dm_plex_create_from_hdf5_xdmf - use the `PETSC_VIEWER_HDF5_XDMF` format for reading HDF5
595802ef0d99SVaclav Hapla 
595937fdd005SBarry Smith   Use `-dm_plex_create_ prefix` to pass options to the internal `PetscViewer`, e.g.
5960bca97951SVaclav Hapla $ -dm_plex_create_viewer_hdf5_collective
5961bca97951SVaclav Hapla 
5962ca522641SMatthew G. Knepley   Level: beginner
5963ca522641SMatthew G. Knepley 
5964a1cb98faSBarry Smith   Notes:
5965a1cb98faSBarry Smith   Using `PETSCVIEWERHDF5` type with `PETSC_VIEWER_HDF5_PETSC` format, one can save multiple `DMPLEX`
5966a1cb98faSBarry Smith   meshes in a single HDF5 file. This in turn requires one to name the `DMPLEX` object with `PetscObjectSetName()`
5967a1cb98faSBarry Smith   before saving it with `DMView()` and before loading it with `DMLoad()` for identification of the mesh object.
5968a1cb98faSBarry Smith   The input parameter name is thus used to name the `DMPLEX` object when `DMPlexCreateFromFile()` internally
5969a1cb98faSBarry Smith   calls `DMLoad()`. Currently, name is ignored for other viewer types and/or formats.
5970a1cb98faSBarry Smith 
59711cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`, `PetscObjectSetName()`, `DMView()`, `DMLoad()`
5972ca522641SMatthew G. Knepley @*/
5973d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], const char plexname[], PetscBool interpolate, DM *dm)
5974d71ae5a4SJacob Faibussowitsch {
5975ef3a5affSJacob Faibussowitsch   const char  extGmsh[]      = ".msh";
5976ef3a5affSJacob Faibussowitsch   const char  extGmsh2[]     = ".msh2";
5977ef3a5affSJacob Faibussowitsch   const char  extGmsh4[]     = ".msh4";
5978ef3a5affSJacob Faibussowitsch   const char  extCGNS[]      = ".cgns";
5979ef3a5affSJacob Faibussowitsch   const char  extExodus[]    = ".exo";
5980ef3a5affSJacob Faibussowitsch   const char  extExodus_e[]  = ".e";
5981ef3a5affSJacob Faibussowitsch   const char  extGenesis[]   = ".gen";
5982ef3a5affSJacob Faibussowitsch   const char  extFluent[]    = ".cas";
5983ef3a5affSJacob Faibussowitsch   const char  extHDF5[]      = ".h5";
59846f2c871aSStefano Zampini   const char  extXDMFHDF5[]  = ".xdmf.h5";
5985ef3a5affSJacob Faibussowitsch   const char  extPLY[]       = ".ply";
5986ef3a5affSJacob Faibussowitsch   const char  extEGADSLite[] = ".egadslite";
5987ef3a5affSJacob Faibussowitsch   const char  extEGADS[]     = ".egads";
5988ef3a5affSJacob Faibussowitsch   const char  extIGES[]      = ".igs";
5989ef3a5affSJacob Faibussowitsch   const char  extSTEP[]      = ".stp";
5990ef3a5affSJacob Faibussowitsch   const char  extCV[]        = ".dat";
5991ca522641SMatthew G. Knepley   size_t      len;
59926afe31f6SMartin Diehl   PetscBool   isGmsh, isGmsh2, isGmsh4, isCGNS, isExodus, isGenesis, isFluent, isHDF5, isPLY, isEGADSLite, isEGADS, isIGES, isSTEP, isCV, isXDMFHDF5;
5993ca522641SMatthew G. Knepley   PetscMPIInt rank;
5994ca522641SMatthew G. Knepley 
5995ca522641SMatthew G. Knepley   PetscFunctionBegin;
59964f572ea9SToby Isaac   PetscAssertPointer(filename, 2);
59974f572ea9SToby Isaac   if (plexname) PetscAssertPointer(plexname, 3);
59984f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
59999566063dSJacob Faibussowitsch   PetscCall(DMInitializePackage());
60009566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromFile, 0, 0, 0, 0));
60019566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
60029566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(filename, &len));
600328b400f6SJacob Faibussowitsch   PetscCheck(len, comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path");
6004ef3a5affSJacob Faibussowitsch 
60059371c9d4SSatish Balay #define CheckExtension(extension__, is_extension__) \
60069371c9d4SSatish Balay   do { \
6007274aaeaaSJacob Faibussowitsch     PetscAssert(sizeof(extension__), comm, PETSC_ERR_PLIB, "Zero-size extension: %s", extension__); \
6008274aaeaaSJacob Faibussowitsch     /* don't count the null-terminator at the end */ \
6009274aaeaaSJacob Faibussowitsch     const size_t ext_len = sizeof(extension__) - 1; \
6010274aaeaaSJacob Faibussowitsch     if (len < ext_len) { \
6011ef3a5affSJacob Faibussowitsch       is_extension__ = PETSC_FALSE; \
6012ef3a5affSJacob Faibussowitsch     } else { \
6013274aaeaaSJacob Faibussowitsch       PetscCall(PetscStrncmp(filename + len - ext_len, extension__, ext_len, &is_extension__)); \
6014ef3a5affSJacob Faibussowitsch     } \
6015ef3a5affSJacob Faibussowitsch   } while (0)
6016ef3a5affSJacob Faibussowitsch 
6017ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh, isGmsh);
6018ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh2, isGmsh2);
6019ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh4, isGmsh4);
6020ef3a5affSJacob Faibussowitsch   CheckExtension(extCGNS, isCGNS);
6021ef3a5affSJacob Faibussowitsch   CheckExtension(extExodus, isExodus);
6022ef3a5affSJacob Faibussowitsch   if (!isExodus) CheckExtension(extExodus_e, isExodus);
6023ef3a5affSJacob Faibussowitsch   CheckExtension(extGenesis, isGenesis);
6024ef3a5affSJacob Faibussowitsch   CheckExtension(extFluent, isFluent);
6025ef3a5affSJacob Faibussowitsch   CheckExtension(extHDF5, isHDF5);
6026ef3a5affSJacob Faibussowitsch   CheckExtension(extPLY, isPLY);
6027ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADSLite, isEGADSLite);
6028ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADS, isEGADS);
6029ef3a5affSJacob Faibussowitsch   CheckExtension(extIGES, isIGES);
6030ef3a5affSJacob Faibussowitsch   CheckExtension(extSTEP, isSTEP);
6031ef3a5affSJacob Faibussowitsch   CheckExtension(extCV, isCV);
60326f2c871aSStefano Zampini   CheckExtension(extXDMFHDF5, isXDMFHDF5);
6033ef3a5affSJacob Faibussowitsch 
6034ef3a5affSJacob Faibussowitsch #undef CheckExtension
6035ef3a5affSJacob Faibussowitsch 
6036de78e4feSLisandro Dalcin   if (isGmsh || isGmsh2 || isGmsh4) {
60379566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateGmshFromFile(comm, filename, interpolate, dm));
6038ca522641SMatthew G. Knepley   } else if (isCGNS) {
60399566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm));
604090c68965SMatthew G. Knepley   } else if (isExodus || isGenesis) {
60419566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateExodusFromFile(comm, filename, interpolate, dm));
60422f0bd6dcSMichael Lange   } else if (isFluent) {
60439566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFluentFromFile(comm, filename, interpolate, dm));
6044cc2f8f65SMatthew G. Knepley   } else if (isHDF5) {
6045cc2f8f65SMatthew G. Knepley     PetscViewer viewer;
6046cc2f8f65SMatthew G. Knepley 
604743b242b4SVaclav 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 */
60486f2c871aSStefano Zampini     PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_create_from_hdf5_xdmf", &isXDMFHDF5, NULL));
60499566063dSJacob Faibussowitsch     PetscCall(PetscViewerCreate(comm, &viewer));
60509566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetType(viewer, PETSCVIEWERHDF5));
60519566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetOptionsPrefix(viewer, "dm_plex_create_"));
60529566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetFromOptions(viewer));
60539566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
60549566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetName(viewer, filename));
6055cd7e8a5eSksagiyam 
60569566063dSJacob Faibussowitsch     PetscCall(DMCreate(comm, dm));
6057f4f49eeaSPierre Jolivet     PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
60589566063dSJacob Faibussowitsch     PetscCall(DMSetType(*dm, DMPLEX));
60596f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPushFormat(viewer, PETSC_VIEWER_HDF5_XDMF));
60609566063dSJacob Faibussowitsch     PetscCall(DMLoad(*dm, viewer));
60616f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPopFormat(viewer));
60629566063dSJacob Faibussowitsch     PetscCall(PetscViewerDestroy(&viewer));
60635fd9971aSMatthew G. Knepley 
60645fd9971aSMatthew G. Knepley     if (interpolate) {
60655fd9971aSMatthew G. Knepley       DM idm;
60665fd9971aSMatthew G. Knepley 
60679566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(*dm, &idm));
60689566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
60695fd9971aSMatthew G. Knepley       *dm = idm;
60705fd9971aSMatthew G. Knepley     }
6071f2801cd6SMatthew G. Knepley   } else if (isPLY) {
60729566063dSJacob Faibussowitsch     PetscCall(DMPlexCreatePLYFromFile(comm, filename, interpolate, dm));
6073c1cad2e7SMatthew G. Knepley   } else if (isEGADSLite || isEGADS || isIGES || isSTEP) {
60749566063dSJacob Faibussowitsch     if (isEGADSLite) PetscCall(DMPlexCreateEGADSLiteFromFile(comm, filename, dm));
60759566063dSJacob Faibussowitsch     else PetscCall(DMPlexCreateEGADSFromFile(comm, filename, dm));
60767bee2925SMatthew Knepley     if (!interpolate) {
60777bee2925SMatthew Knepley       DM udm;
60787bee2925SMatthew Knepley 
60799566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(*dm, &udm));
60809566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
60817bee2925SMatthew Knepley       *dm = udm;
60827bee2925SMatthew Knepley     }
60838ca92349SMatthew G. Knepley   } else if (isCV) {
60849566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCellVertexFromFile(comm, filename, interpolate, dm));
608598921bdaSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename);
60869566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(plexname, &len));
6087f4f49eeaSPierre Jolivet   if (len) PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
60889566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromFile, 0, 0, 0, 0));
60893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6090ca522641SMatthew G. Knepley }
609112b8a6daSStefano Zampini 
6092cc4c1da9SBarry Smith /*@
60939f6c5813SMatthew 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.
60949f6c5813SMatthew G. Knepley 
60950528010dSStefano Zampini   Input Parameters:
60960528010dSStefano Zampini + tr     - The `DMPlexTransform`
60970528010dSStefano Zampini - prefix - An options prefix, or NULL
60989f6c5813SMatthew G. Knepley 
60999f6c5813SMatthew G. Knepley   Output Parameter:
61009f6c5813SMatthew G. Knepley . dm - The `DM`
61019f6c5813SMatthew G. Knepley 
61029f6c5813SMatthew G. Knepley   Level: beginner
61039f6c5813SMatthew G. Knepley 
610420f4b53cSBarry Smith   Notes:
610520f4b53cSBarry 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.
610620f4b53cSBarry Smith 
61079f6c5813SMatthew G. Knepley .seealso: `DMPlexCreateFromFile`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
61089f6c5813SMatthew G. Knepley @*/
61090528010dSStefano Zampini PetscErrorCode DMPlexCreateEphemeral(DMPlexTransform tr, const char prefix[], DM *dm)
61109f6c5813SMatthew G. Knepley {
61110528010dSStefano Zampini   DM           bdm, bcdm, cdm;
61120528010dSStefano Zampini   Vec          coordinates, coordinatesNew;
61130528010dSStefano Zampini   PetscSection cs;
6114817b2c36SMatthew G. Knepley   PetscInt     cdim, Nl;
61159f6c5813SMatthew G. Knepley 
61169f6c5813SMatthew G. Knepley   PetscFunctionBegin;
61179f6c5813SMatthew G. Knepley   PetscCall(DMCreate(PetscObjectComm((PetscObject)tr), dm));
61189f6c5813SMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
61190528010dSStefano Zampini   ((DM_Plex *)(*dm)->data)->interpolated = DMPLEX_INTERPOLATED_FULL;
61200528010dSStefano Zampini   // Handle coordinates
61210528010dSStefano Zampini   PetscCall(DMPlexTransformGetDM(tr, &bdm));
6122817b2c36SMatthew G. Knepley   PetscCall(DMPlexTransformSetDimensions(tr, bdm, *dm));
6123817b2c36SMatthew G. Knepley   PetscCall(DMGetCoordinateDim(*dm, &cdim));
61240528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(bdm, &bcdm));
61250528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(*dm, &cdm));
61260528010dSStefano Zampini   PetscCall(DMCopyDisc(bcdm, cdm));
61270528010dSStefano Zampini   PetscCall(DMGetLocalSection(cdm, &cs));
61280528010dSStefano Zampini   PetscCall(PetscSectionSetNumFields(cs, 1));
61290528010dSStefano Zampini   PetscCall(PetscSectionSetFieldComponents(cs, 0, cdim));
61300528010dSStefano Zampini   PetscCall(DMGetCoordinatesLocal(bdm, &coordinates));
61310528010dSStefano Zampini   PetscCall(VecDuplicate(coordinates, &coordinatesNew));
61320528010dSStefano Zampini   PetscCall(VecCopy(coordinates, coordinatesNew));
61330528010dSStefano Zampini   PetscCall(DMSetCoordinatesLocal(*dm, coordinatesNew));
61340528010dSStefano Zampini   PetscCall(VecDestroy(&coordinatesNew));
61359f6c5813SMatthew G. Knepley 
61369f6c5813SMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)tr));
61379f6c5813SMatthew G. Knepley   PetscCall(DMPlexTransformDestroy(&((DM_Plex *)(*dm)->data)->tr));
61389f6c5813SMatthew G. Knepley   ((DM_Plex *)(*dm)->data)->tr = tr;
61390528010dSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(*dm, PETSC_FALSE));
61400528010dSStefano Zampini   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*dm, prefix));
61410528010dSStefano Zampini   PetscCall(DMSetFromOptions(*dm));
61429f6c5813SMatthew G. Knepley 
61439f6c5813SMatthew G. Knepley   PetscCall(DMGetNumLabels(bdm, &Nl));
61449f6c5813SMatthew G. Knepley   for (PetscInt l = 0; l < Nl; ++l) {
61459f6c5813SMatthew G. Knepley     DMLabel     label, labelNew;
61469f6c5813SMatthew G. Knepley     const char *lname;
61479f6c5813SMatthew G. Knepley     PetscBool   isDepth, isCellType;
61489f6c5813SMatthew G. Knepley 
61499f6c5813SMatthew G. Knepley     PetscCall(DMGetLabelName(bdm, l, &lname));
61509f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "depth", &isDepth));
61519f6c5813SMatthew G. Knepley     if (isDepth) continue;
61529f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "celltype", &isCellType));
61539f6c5813SMatthew G. Knepley     if (isCellType) continue;
61549f6c5813SMatthew G. Knepley     PetscCall(DMCreateLabel(*dm, lname));
61559f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(bdm, lname, &label));
61569f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(*dm, lname, &labelNew));
61579f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetType(labelNew, DMLABELEPHEMERAL));
61589f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetLabel(labelNew, label));
61599f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetTransform(labelNew, tr));
61609f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetUp(labelNew));
61619f6c5813SMatthew G. Knepley   }
61623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
61639f6c5813SMatthew G. Knepley }
6164