xref: /petsc/src/dm/impls/plex/plexcreate.c (revision 014689415f8092083d18be3809cf56ed1d7bc937)
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 
205552b385SBrandon PETSC_EXTERN PetscErrorCode DMPlexCheckEGADS_Private(DM dm)
215552b385SBrandon {
225552b385SBrandon   PetscObject modelObj;
235552b385SBrandon 
245552b385SBrandon   PetscFunctionBegin;
255552b385SBrandon   PetscCall(PetscObjectQuery((PetscObject)dm, "EGADS Model", &modelObj));
265552b385SBrandon   PetscCheck(modelObj, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Input DM must have attached EGADS Geometry Model");
275552b385SBrandon   PetscFunctionReturn(PETSC_SUCCESS);
285552b385SBrandon }
295552b385SBrandon 
305552b385SBrandon static PetscErrorCode DMPlexCopyContext_Private(DM dmin, const char name[], DM dmout)
315552b385SBrandon {
325552b385SBrandon   PetscObject obj;
335552b385SBrandon 
345552b385SBrandon   PetscFunctionBegin;
355552b385SBrandon   PetscCall(PetscObjectQuery((PetscObject)dmin, name, &obj));
365552b385SBrandon   if (obj) PetscCall(PetscObjectCompose((PetscObject)dmout, name, obj));
375552b385SBrandon   PetscFunctionReturn(PETSC_SUCCESS);
385552b385SBrandon }
395552b385SBrandon 
405552b385SBrandon static PetscErrorCode DMPlexSwapContext_Private(DM dmA, const char name[], DM dmB)
415552b385SBrandon {
425552b385SBrandon   PetscObject objA, objB;
435552b385SBrandon 
445552b385SBrandon   PetscFunctionBegin;
455552b385SBrandon   PetscCall(PetscObjectQuery((PetscObject)dmA, name, &objA));
465552b385SBrandon   PetscCall(PetscObjectQuery((PetscObject)dmB, name, &objB));
475552b385SBrandon   PetscCall(PetscObjectReference(objA));
485552b385SBrandon   PetscCall(PetscObjectReference(objB));
495552b385SBrandon   PetscCall(PetscObjectCompose((PetscObject)dmA, name, objB));
505552b385SBrandon   PetscCall(PetscObjectCompose((PetscObject)dmB, name, objA));
515552b385SBrandon   PetscCall(PetscObjectDereference(objA));
525552b385SBrandon   PetscCall(PetscObjectDereference(objB));
535552b385SBrandon   PetscFunctionReturn(PETSC_SUCCESS);
545552b385SBrandon }
555552b385SBrandon 
565552b385SBrandon PetscErrorCode DMPlexCopyEGADSInfo_Internal(DM dmin, DM dmout)
575552b385SBrandon {
585552b385SBrandon   PetscFunctionBegin;
595552b385SBrandon   PetscCall(DMPlexCopyContext_Private(dmin, "EGADS Model", dmout));
605552b385SBrandon   PetscCall(DMPlexCopyContext_Private(dmin, "EGADS Context", dmout));
615552b385SBrandon   PetscCall(DMPlexCopyContext_Private(dmin, "EGADSlite Model", dmout));
625552b385SBrandon   PetscCall(DMPlexCopyContext_Private(dmin, "EGADSlite Context", dmout));
635552b385SBrandon   PetscFunctionReturn(PETSC_SUCCESS);
645552b385SBrandon }
655552b385SBrandon 
665552b385SBrandon static PetscErrorCode DMPlexSwapEGADSInfo_Private(DM dmA, DM dmB)
675552b385SBrandon {
685552b385SBrandon   PetscFunctionBegin;
695552b385SBrandon   PetscCall(DMPlexSwapContext_Private(dmA, "EGADS Model", dmB));
705552b385SBrandon   PetscCall(DMPlexSwapContext_Private(dmA, "EGADS Context", dmB));
715552b385SBrandon   PetscCall(DMPlexSwapContext_Private(dmA, "EGADSlite Model", dmB));
725552b385SBrandon   PetscCall(DMPlexSwapContext_Private(dmA, "EGADSlite Context", dmB));
735552b385SBrandon   PetscFunctionReturn(PETSC_SUCCESS);
745552b385SBrandon }
755552b385SBrandon 
76e600fa54SMatthew G. Knepley /* This copies internal things in the Plex structure that we generally want when making a new, related Plex */
77d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCopy_Internal(DM dmin, PetscBool copyPeriodicity, PetscBool copyOverlap, DM dmout)
78d71ae5a4SJacob Faibussowitsch {
794fb89dddSMatthew G. Knepley   const PetscReal     *maxCell, *Lstart, *L;
8012a88998SMatthew G. Knepley   VecType              vecType;
8112a88998SMatthew G. Knepley   MatType              matType;
82129f447cSJames Wright   PetscBool            dist, useCeed, balance_partition;
83adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
84e600fa54SMatthew G. Knepley 
85e600fa54SMatthew G. Knepley   PetscFunctionBegin;
86835f2295SStefano Zampini   if (dmin == dmout) PetscFunctionReturn(PETSC_SUCCESS);
8712a88998SMatthew G. Knepley   PetscCall(DMGetVecType(dmin, &vecType));
8812a88998SMatthew G. Knepley   PetscCall(DMSetVecType(dmout, vecType));
8912a88998SMatthew G. Knepley   PetscCall(DMGetMatType(dmin, &matType));
9012a88998SMatthew G. Knepley   PetscCall(DMSetMatType(dmout, matType));
91e600fa54SMatthew G. Knepley   if (copyPeriodicity) {
924fb89dddSMatthew G. Knepley     PetscCall(DMGetPeriodicity(dmin, &maxCell, &Lstart, &L));
934fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dmout, maxCell, Lstart, L));
943d0e8ed9SDavid Salac     PetscCall(DMLocalizeCoordinates(dmout));
95e600fa54SMatthew G. Knepley   }
969566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dmin, &dist));
979566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeSetDefault(dmout, dist));
986bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dmin, &reorder));
996bc1bd01Sksagiyam   PetscCall(DMPlexReorderSetDefault(dmout, reorder));
1005962854dSMatthew G. Knepley   PetscCall(DMPlexGetUseCeed(dmin, &useCeed));
1015962854dSMatthew G. Knepley   PetscCall(DMPlexSetUseCeed(dmout, useCeed));
102129f447cSJames Wright   PetscCall(DMPlexGetPartitionBalance(dmin, &balance_partition));
103129f447cSJames Wright   PetscCall(DMPlexSetPartitionBalance(dmout, balance_partition));
104e600fa54SMatthew G. Knepley   ((DM_Plex *)dmout->data)->useHashLocation = ((DM_Plex *)dmin->data)->useHashLocation;
1055962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printSetValues  = ((DM_Plex *)dmin->data)->printSetValues;
1065962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFEM        = ((DM_Plex *)dmin->data)->printFEM;
1075962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printFVM        = ((DM_Plex *)dmin->data)->printFVM;
1085962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printL2         = ((DM_Plex *)dmin->data)->printL2;
1095962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printLocate     = ((DM_Plex *)dmin->data)->printLocate;
110a77a5016SMatthew G. Knepley   ((DM_Plex *)dmout->data)->printProject    = ((DM_Plex *)dmin->data)->printProject;
1115962854dSMatthew G. Knepley   ((DM_Plex *)dmout->data)->printTol        = ((DM_Plex *)dmin->data)->printTol;
1121baa6e33SBarry Smith   if (copyOverlap) PetscCall(DMPlexSetOverlap_Plex(dmout, dmin, 0));
1135552b385SBrandon   PetscCall(DMPlexCopyEGADSInfo_Internal(dmin, dmout));
1143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
115e600fa54SMatthew G. Knepley }
116e600fa54SMatthew G. Knepley 
1179318fe57SMatthew G. Knepley /* Replace dm with the contents of ndm, and then destroy ndm
1189318fe57SMatthew G. Knepley    - Share the DM_Plex structure
1199318fe57SMatthew G. Knepley    - Share the coordinates
1209318fe57SMatthew G. Knepley    - Share the SF
1219318fe57SMatthew G. Knepley */
122d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexReplace_Internal(DM dm, DM *ndm)
123d71ae5a4SJacob Faibussowitsch {
1249318fe57SMatthew G. Knepley   PetscSF          sf;
1259318fe57SMatthew G. Knepley   DM               dmNew = *ndm, coordDM, coarseDM;
1269318fe57SMatthew G. Knepley   Vec              coords;
127509b31aaSMatthew G. Knepley   PetscPointFunc   coordFunc;
1284fb89dddSMatthew G. Knepley   const PetscReal *maxCell, *Lstart, *L;
1299318fe57SMatthew G. Knepley   PetscInt         dim, cdim;
130e535cce4SJames Wright   PetscBool        use_natural;
1319318fe57SMatthew G. Knepley 
1329318fe57SMatthew G. Knepley   PetscFunctionBegin;
1339318fe57SMatthew G. Knepley   if (dm == dmNew) {
1349566063dSJacob Faibussowitsch     PetscCall(DMDestroy(ndm));
1353ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1369318fe57SMatthew G. Knepley   }
1379318fe57SMatthew G. Knepley   dm->setupcalled = dmNew->setupcalled;
138d0812dedSMatthew G. Knepley   if (!dm->hdr.name) {
139d0812dedSMatthew G. Knepley     const char *name;
140d0812dedSMatthew G. Knepley 
141d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)*ndm, &name));
142d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
143d0812dedSMatthew G. Knepley   }
1449566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dmNew, &dim));
1459566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
1469566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dmNew, &cdim));
1479566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
1489566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmNew, &sf));
1499566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sf));
1509566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmNew, &coordDM));
1519566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmNew, &coords));
1529566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dm, coordDM));
1539566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coords));
1546858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmNew, &coordDM));
1556858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmNew, &coords));
1566858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dm, coordDM));
1576858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dm, coords));
1589318fe57SMatthew G. Knepley   /* Do not want to create the coordinate field if it does not already exist, so do not call DMGetCoordinateField() */
1596858538eSMatthew G. Knepley   PetscCall(DMFieldDestroy(&dm->coordinates[0].field));
1606858538eSMatthew G. Knepley   dm->coordinates[0].field = dmNew->coordinates[0].field;
161509b31aaSMatthew G. Knepley   PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc));
162509b31aaSMatthew G. Knepley   PetscCall(DMPlexSetCoordinateMap(dmNew, coordFunc));
1634fb89dddSMatthew G. Knepley   PetscCall(DMGetPeriodicity(dmNew, &maxCell, &Lstart, &L));
1644fb89dddSMatthew G. Knepley   PetscCall(DMSetPeriodicity(dm, maxCell, Lstart, L));
165e535cce4SJames Wright   PetscCall(DMGetNaturalSF(dmNew, &sf));
166e535cce4SJames Wright   PetscCall(DMSetNaturalSF(dm, sf));
167e535cce4SJames Wright   PetscCall(DMGetUseNatural(dmNew, &use_natural));
168e535cce4SJames Wright   PetscCall(DMSetUseNatural(dm, use_natural));
1699566063dSJacob Faibussowitsch   PetscCall(DMDestroy_Plex(dm));
1709566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
1719318fe57SMatthew G. Knepley   dm->data = dmNew->data;
1729318fe57SMatthew G. Knepley   ((DM_Plex *)dmNew->data)->refct++;
1731fca310dSJames Wright   {
1741fca310dSJames Wright     PetscInt       num_face_sfs;
1751fca310dSJames Wright     const PetscSF *sfs;
1761fca310dSJames Wright     PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &sfs));
1771fca310dSJames Wright     PetscCall(DMPlexSetIsoperiodicFaceSF(dm, num_face_sfs, (PetscSF *)sfs)); // for the compose function effect on dm
1781fca310dSJames Wright   }
1799566063dSJacob Faibussowitsch   PetscCall(DMDestroyLabelLinkList_Internal(dm));
1809566063dSJacob Faibussowitsch   PetscCall(DMCopyLabels(dmNew, dm, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
1819566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmNew, &coarseDM));
1829566063dSJacob Faibussowitsch   PetscCall(DMSetCoarseDM(dm, coarseDM));
1835552b385SBrandon   PetscCall(DMPlexCopyEGADSInfo_Internal(dmNew, dm));
1849566063dSJacob Faibussowitsch   PetscCall(DMDestroy(ndm));
1853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1869318fe57SMatthew G. Knepley }
1879318fe57SMatthew G. Knepley 
1889318fe57SMatthew G. Knepley /* Swap dm with the contents of dmNew
1899318fe57SMatthew G. Knepley    - Swap the DM_Plex structure
1909318fe57SMatthew G. Knepley    - Swap the coordinates
1919318fe57SMatthew G. Knepley    - Swap the point PetscSF
1929318fe57SMatthew G. Knepley */
193d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexSwap_Static(DM dmA, DM dmB)
194d71ae5a4SJacob Faibussowitsch {
1959318fe57SMatthew G. Knepley   DM          coordDMA, coordDMB;
1969318fe57SMatthew G. Knepley   Vec         coordsA, coordsB;
1979318fe57SMatthew G. Knepley   PetscSF     sfA, sfB;
1989318fe57SMatthew G. Knepley   DMField     fieldTmp;
1999318fe57SMatthew G. Knepley   void       *tmp;
2009318fe57SMatthew G. Knepley   DMLabelLink listTmp;
2019318fe57SMatthew G. Knepley   DMLabel     depthTmp;
2029318fe57SMatthew G. Knepley   PetscInt    tmpI;
2039318fe57SMatthew G. Knepley 
2049318fe57SMatthew G. Knepley   PetscFunctionBegin;
2053ba16761SJacob Faibussowitsch   if (dmA == dmB) PetscFunctionReturn(PETSC_SUCCESS);
2069566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmA, &sfA));
2079566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dmB, &sfB));
2089566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)sfA));
2099566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmA, sfB));
2109566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dmB, sfA));
2119566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)sfA));
2129318fe57SMatthew G. Knepley 
2139566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmA, &coordDMA));
2149566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dmB, &coordDMB));
2159566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordDMA));
2169566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmA, coordDMB));
2179566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDM(dmB, coordDMA));
2189566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
2199318fe57SMatthew G. Knepley 
2209566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmA, &coordsA));
2219566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dmB, &coordsB));
2229566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)coordsA));
2239566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmA, coordsB));
2249566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dmB, coordsA));
2259566063dSJacob Faibussowitsch   PetscCall(PetscObjectDereference((PetscObject)coordsA));
2269318fe57SMatthew G. Knepley 
2276858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmA, &coordDMA));
2286858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinateDM(dmB, &coordDMB));
2296858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordDMA));
2306858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmA, coordDMB));
2316858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(dmB, coordDMA));
2326858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordDMA));
2336858538eSMatthew G. Knepley 
2346858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmA, &coordsA));
2356858538eSMatthew G. Knepley   PetscCall(DMGetCellCoordinatesLocal(dmB, &coordsB));
2366858538eSMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)coordsA));
2376858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmA, coordsB));
2386858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(dmB, coordsA));
2396858538eSMatthew G. Knepley   PetscCall(PetscObjectDereference((PetscObject)coordsA));
2406858538eSMatthew G. Knepley 
2415552b385SBrandon   PetscCall(DMPlexSwapEGADSInfo_Private(dmA, dmB));
2425552b385SBrandon 
2436858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[0].field;
2446858538eSMatthew G. Knepley   dmA->coordinates[0].field = dmB->coordinates[0].field;
2456858538eSMatthew G. Knepley   dmB->coordinates[0].field = fieldTmp;
2466858538eSMatthew G. Knepley   fieldTmp                  = dmA->coordinates[1].field;
2476858538eSMatthew G. Knepley   dmA->coordinates[1].field = dmB->coordinates[1].field;
2486858538eSMatthew G. Knepley   dmB->coordinates[1].field = fieldTmp;
2499318fe57SMatthew G. Knepley   tmp                       = dmA->data;
2509318fe57SMatthew G. Knepley   dmA->data                 = dmB->data;
2519318fe57SMatthew G. Knepley   dmB->data                 = tmp;
2529318fe57SMatthew G. Knepley   listTmp                   = dmA->labels;
2539318fe57SMatthew G. Knepley   dmA->labels               = dmB->labels;
2549318fe57SMatthew G. Knepley   dmB->labels               = listTmp;
2559318fe57SMatthew G. Knepley   depthTmp                  = dmA->depthLabel;
2569318fe57SMatthew G. Knepley   dmA->depthLabel           = dmB->depthLabel;
2579318fe57SMatthew G. Knepley   dmB->depthLabel           = depthTmp;
2589318fe57SMatthew G. Knepley   depthTmp                  = dmA->celltypeLabel;
2599318fe57SMatthew G. Knepley   dmA->celltypeLabel        = dmB->celltypeLabel;
2609318fe57SMatthew G. Knepley   dmB->celltypeLabel        = depthTmp;
2619318fe57SMatthew G. Knepley   tmpI                      = dmA->levelup;
2629318fe57SMatthew G. Knepley   dmA->levelup              = dmB->levelup;
2639318fe57SMatthew G. Knepley   dmB->levelup              = tmpI;
2643ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2659318fe57SMatthew G. Knepley }
2669318fe57SMatthew G. Knepley 
2673431e603SJed Brown PetscErrorCode DMPlexInterpolateInPlace_Internal(DM dm)
268d71ae5a4SJacob Faibussowitsch {
2699318fe57SMatthew G. Knepley   DM idm;
2709318fe57SMatthew G. Knepley 
2719318fe57SMatthew G. Knepley   PetscFunctionBegin;
2729566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolate(dm, &idm));
2739566063dSJacob Faibussowitsch   PetscCall(DMPlexCopyCoordinates(dm, idm));
27469d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &idm));
2753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2769318fe57SMatthew G. Knepley }
2779318fe57SMatthew G. Knepley 
2789318fe57SMatthew G. Knepley /*@C
2799318fe57SMatthew G. Knepley   DMPlexCreateCoordinateSpace - Creates a finite element space for the coordinates
2809318fe57SMatthew G. Knepley 
28120f4b53cSBarry Smith   Collective
2829318fe57SMatthew G. Knepley 
2839318fe57SMatthew G. Knepley   Input Parameters:
28460225df5SJacob Faibussowitsch + dm        - The `DMPLEX`
28520f4b53cSBarry Smith . degree    - The degree of the finite element or `PETSC_DECIDE`
286e44f6aebSMatthew G. Knepley . project   - Flag to project current coordinates into the space
2879318fe57SMatthew G. Knepley - coordFunc - An optional function to map new points from refinement to the surface
2889318fe57SMatthew G. Knepley 
2899318fe57SMatthew G. Knepley   Level: advanced
2909318fe57SMatthew G. Knepley 
2911cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `PetscPointFunc`, `PetscFECreateLagrange()`, `DMGetCoordinateDM()`
2929318fe57SMatthew G. Knepley @*/
293e44f6aebSMatthew G. Knepley PetscErrorCode DMPlexCreateCoordinateSpace(DM dm, PetscInt degree, PetscBool project, PetscPointFunc coordFunc)
294d71ae5a4SJacob Faibussowitsch {
295e44f6aebSMatthew G. Knepley   PetscFE  fe = NULL;
2969318fe57SMatthew G. Knepley   DM       cdm;
297ac9d17c7SMatthew G. Knepley   PetscInt dim, cdim, dE, qorder, height;
2989318fe57SMatthew G. Knepley 
299e44f6aebSMatthew G. Knepley   PetscFunctionBegin;
3009566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
301ac9d17c7SMatthew G. Knepley   cdim = dim;
3029566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dE));
3039318fe57SMatthew G. Knepley   qorder = degree;
304e44f6aebSMatthew G. Knepley   PetscCall(DMGetCoordinateDM(dm, &cdm));
305509b31aaSMatthew G. Knepley   if (!coordFunc) PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc));
306d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)cdm);
307dc431b0cSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-default_quadrature_order", "Quadrature order is one less than quadrature points per edge", "DMPlexCreateCoordinateSpace", qorder, &qorder, NULL, 0));
308ac9d17c7SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_coordinate_dim", "Set the coordinate dimension", "DMPlexCreateCoordinateSpace", cdim, &cdim, NULL, dim));
309d0609cedSBarry Smith   PetscOptionsEnd();
3101df12153SMatthew G. Knepley   PetscCall(DMPlexGetVTKCellHeight(dm, &height));
311ac9d17c7SMatthew G. Knepley   if (cdim > dim) {
312ac9d17c7SMatthew G. Knepley     DM           cdm;
313ac9d17c7SMatthew G. Knepley     PetscSection cs, csNew;
314ac9d17c7SMatthew G. Knepley     Vec          coordinates, coordinatesNew;
315ac9d17c7SMatthew G. Knepley     VecType      vectype;
316ac9d17c7SMatthew G. Knepley     IS           idx;
317ac9d17c7SMatthew G. Knepley     PetscInt    *indices;
318ac9d17c7SMatthew G. Knepley     PetscInt     bs, n;
319ac9d17c7SMatthew G. Knepley 
320ac9d17c7SMatthew G. Knepley     // Recreate coordinate section
321ac9d17c7SMatthew G. Knepley     {
322ac9d17c7SMatthew G. Knepley       const char *fieldName = NULL, *compName = NULL;
323ac9d17c7SMatthew G. Knepley       PetscInt    Nc, pStart, pEnd;
324ac9d17c7SMatthew G. Knepley 
325ac9d17c7SMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
326ac9d17c7SMatthew G. Knepley       PetscCall(DMGetLocalSection(cdm, &cs));
327ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)cs), &csNew));
328ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionSetNumFields(csNew, 1));
329ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionGetFieldName(cs, 0, &fieldName));
330ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionSetFieldName(csNew, 0, fieldName));
331ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionGetFieldComponents(cs, 0, &Nc));
332ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionSetFieldComponents(csNew, 0, cdim));
333ac9d17c7SMatthew G. Knepley       for (PetscInt c = 0; c < Nc; ++c) {
334ac9d17c7SMatthew G. Knepley         PetscCall(PetscSectionGetComponentName(cs, 0, c, &compName));
335ac9d17c7SMatthew G. Knepley         PetscCall(PetscSectionSetComponentName(csNew, 0, c, compName));
336ac9d17c7SMatthew G. Knepley       }
337ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionGetChart(cs, &pStart, &pEnd));
338ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionSetChart(csNew, pStart, pEnd));
339ac9d17c7SMatthew G. Knepley       for (PetscInt p = pStart; p < pEnd; ++p) {
340530e699aSMatthew G. Knepley         PetscInt dof;
341530e699aSMatthew G. Knepley 
342530e699aSMatthew G. Knepley         PetscCall(PetscSectionGetDof(cs, p, &dof));
343530e699aSMatthew G. Knepley         if (dof) {
344ac9d17c7SMatthew G. Knepley           PetscCall(PetscSectionSetDof(csNew, p, cdim));
345ac9d17c7SMatthew G. Knepley           PetscCall(PetscSectionSetFieldDof(csNew, p, 0, cdim));
346ac9d17c7SMatthew G. Knepley         }
347530e699aSMatthew G. Knepley       }
348ac9d17c7SMatthew G. Knepley       PetscCall(PetscSectionSetUp(csNew));
349ac9d17c7SMatthew G. Knepley     }
350ac9d17c7SMatthew G. Knepley     PetscCall(DMSetLocalSection(cdm, csNew));
351ac9d17c7SMatthew G. Knepley     PetscCall(PetscSectionDestroy(&csNew));
352530e699aSMatthew G. Knepley     // Reset coordinate dimension for coordinate DM
353530e699aSMatthew G. Knepley     PetscCall(DMSetCoordinateDim(cdm, cdim));
354530e699aSMatthew G. Knepley     PetscCall(DMSetCoordinateField(cdm, NULL));
355ac9d17c7SMatthew G. Knepley     // Inject coordinates into higher dimension
356ac9d17c7SMatthew G. Knepley     PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
357ac9d17c7SMatthew G. Knepley     PetscCall(VecGetBlockSize(coordinates, &bs));
358ac9d17c7SMatthew G. Knepley     PetscCheck(bs == dim, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "We can only inject simple coordinates into a higher dimension");
359ac9d17c7SMatthew G. Knepley     PetscCall(VecCreate(PetscObjectComm((PetscObject)coordinates), &coordinatesNew));
360ac9d17c7SMatthew G. Knepley     PetscCall(VecGetType(coordinates, &vectype));
361ac9d17c7SMatthew G. Knepley     PetscCall(VecSetType(coordinatesNew, vectype));
362ac9d17c7SMatthew G. Knepley     PetscCall(VecGetLocalSize(coordinates, &n));
363ac9d17c7SMatthew G. Knepley     PetscCheck(!(n % bs), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "We can only inject simple coordinates into a higher dimension");
364ac9d17c7SMatthew G. Knepley     n /= bs;
365ac9d17c7SMatthew G. Knepley     PetscCall(VecSetSizes(coordinatesNew, n * cdim, PETSC_DETERMINE));
366ac9d17c7SMatthew G. Knepley     PetscCall(VecSetUp(coordinatesNew));
367ac9d17c7SMatthew G. Knepley     PetscCall(PetscMalloc1(n * bs, &indices));
368ac9d17c7SMatthew G. Knepley     for (PetscInt i = 0; i < n; ++i)
369ac9d17c7SMatthew G. Knepley       for (PetscInt b = 0; b < bs; ++b) indices[i * bs + b] = i * cdim + b;
370ac9d17c7SMatthew G. Knepley     PetscCall(ISCreateGeneral(PETSC_COMM_SELF, n * bs, indices, PETSC_OWN_POINTER, &idx));
371ac9d17c7SMatthew G. Knepley     PetscCall(VecISCopy(coordinatesNew, idx, SCATTER_FORWARD, coordinates));
372ac9d17c7SMatthew G. Knepley     PetscCall(ISDestroy(&idx));
373ac9d17c7SMatthew G. Knepley     PetscCall(DMSetCoordinatesLocal(dm, coordinatesNew));
374ac9d17c7SMatthew G. Knepley     PetscCall(VecDestroy(&coordinatesNew));
375ac9d17c7SMatthew G. Knepley     PetscCall(DMSetCoordinateDim(dm, cdim));
376ac9d17c7SMatthew G. Knepley     {
377ac9d17c7SMatthew G. Knepley       PetscInt gn;
378ac9d17c7SMatthew G. Knepley 
379ac9d17c7SMatthew G. Knepley       PetscCall(DMGetCoordinates(dm, &coordinatesNew));
380ac9d17c7SMatthew G. Knepley       PetscCall(VecGetLocalSize(coordinatesNew, &gn));
381ac9d17c7SMatthew G. Knepley       PetscCheck(gn == n * cdim, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Global coordinate size %" PetscInt_FMT " != %" PetscInt_FMT "local coordinate size", gn, n * cdim);
382ac9d17c7SMatthew G. Knepley     }
383ac9d17c7SMatthew G. Knepley     dE      = cdim;
384530e699aSMatthew G. Knepley     project = PETSC_FALSE;
385ac9d17c7SMatthew G. Knepley   }
386e44f6aebSMatthew G. Knepley   if (degree >= 0) {
387e44f6aebSMatthew G. Knepley     DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
388e44f6aebSMatthew G. Knepley     PetscInt       cStart, cEnd, gct;
389dc431b0cSMatthew G. Knepley 
3901df12153SMatthew G. Knepley     PetscCall(DMPlexGetHeightStratum(dm, height, &cStart, &cEnd));
391dc431b0cSMatthew G. Knepley     if (cEnd > cStart) PetscCall(DMPlexGetCellType(dm, cStart, &ct));
392e44f6aebSMatthew G. Knepley     gct = (PetscInt)ct;
393462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &gct, 1, MPIU_INT, MPI_MIN, PetscObjectComm((PetscObject)dm)));
394e44f6aebSMatthew G. Knepley     ct = (DMPolytopeType)gct;
395e44f6aebSMatthew G. Knepley     // Work around current bug in PetscDualSpaceSetUp_Lagrange()
396e44f6aebSMatthew G. Knepley     //   Can be seen in plex_tutorials-ex10_1
397e44f6aebSMatthew 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));
3984f9ab2b4SJed Brown   }
399e44f6aebSMatthew G. Knepley   PetscCall(DMSetCoordinateDisc(dm, fe, project));
4009566063dSJacob Faibussowitsch   PetscCall(PetscFEDestroy(&fe));
4013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4029318fe57SMatthew G. Knepley }
4039318fe57SMatthew G. Knepley 
4041df5d5c5SMatthew G. Knepley /*@
4051df5d5c5SMatthew G. Knepley   DMPlexCreateDoublet - Creates a mesh of two cells of the specified type, optionally with later refinement.
4061df5d5c5SMatthew G. Knepley 
407d083f849SBarry Smith   Collective
4081df5d5c5SMatthew G. Knepley 
4091df5d5c5SMatthew G. Knepley   Input Parameters:
410a1cb98faSBarry Smith + comm            - The communicator for the `DM` object
4111df5d5c5SMatthew G. Knepley . dim             - The spatial dimension
4121df5d5c5SMatthew G. Knepley . simplex         - Flag for simplicial cells, otherwise they are tensor product cells
4131df5d5c5SMatthew G. Knepley . interpolate     - Flag to create intermediate mesh pieces (edges, faces)
4141df5d5c5SMatthew G. Knepley - refinementLimit - A nonzero number indicates the largest admissible volume for a refined cell
4151df5d5c5SMatthew G. Knepley 
4161df5d5c5SMatthew G. Knepley   Output Parameter:
41760225df5SJacob Faibussowitsch . newdm - The `DM` object
4181df5d5c5SMatthew G. Knepley 
4191df5d5c5SMatthew G. Knepley   Level: beginner
4201df5d5c5SMatthew G. Knepley 
4211cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetType()`, `DMCreate()`
4221df5d5c5SMatthew G. Knepley @*/
423d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateDoublet(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscBool interpolate, PetscReal refinementLimit, DM *newdm)
424d71ae5a4SJacob Faibussowitsch {
4251df5d5c5SMatthew G. Knepley   DM          dm;
4261df5d5c5SMatthew G. Knepley   PetscMPIInt rank;
4271df5d5c5SMatthew G. Knepley 
4281df5d5c5SMatthew G. Knepley   PetscFunctionBegin;
4299566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, &dm));
4309566063dSJacob Faibussowitsch   PetscCall(DMSetType(dm, DMPLEX));
4319566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
43246139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
4339566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
434ce78fa2fSMatthew G. Knepley   switch (dim) {
435ce78fa2fSMatthew G. Knepley   case 2:
4369566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "triangular"));
4379566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "quadrilateral"));
438ce78fa2fSMatthew G. Knepley     break;
439ce78fa2fSMatthew G. Knepley   case 3:
4409566063dSJacob Faibussowitsch     if (simplex) PetscCall(PetscObjectSetName((PetscObject)dm, "tetrahedral"));
4419566063dSJacob Faibussowitsch     else PetscCall(PetscObjectSetName((PetscObject)dm, "hexahedral"));
442ce78fa2fSMatthew G. Knepley     break;
443d71ae5a4SJacob Faibussowitsch   default:
444d71ae5a4SJacob Faibussowitsch     SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
445ce78fa2fSMatthew G. Knepley   }
4461df5d5c5SMatthew G. Knepley   if (rank) {
4471df5d5c5SMatthew G. Knepley     PetscInt numPoints[2] = {0, 0};
4489566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, NULL, NULL, NULL, NULL));
4491df5d5c5SMatthew G. Knepley   } else {
4501df5d5c5SMatthew G. Knepley     switch (dim) {
4511df5d5c5SMatthew G. Knepley     case 2:
4521df5d5c5SMatthew G. Knepley       if (simplex) {
4531df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {4, 2};
4541df5d5c5SMatthew G. Knepley         PetscInt    coneSize[6]         = {3, 3, 0, 0, 0, 0};
4551df5d5c5SMatthew G. Knepley         PetscInt    cones[6]            = {2, 3, 4, 5, 4, 3};
4561df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
4571df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[8]     = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
4581df5d5c5SMatthew G. Knepley 
4599566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
4601df5d5c5SMatthew G. Knepley       } else {
4611df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {6, 2};
4621df5d5c5SMatthew G. Knepley         PetscInt    coneSize[8]         = {4, 4, 0, 0, 0, 0, 0, 0};
4631df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {2, 3, 4, 5, 3, 6, 7, 4};
4641df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
4651df5d5c5SMatthew 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};
4661df5d5c5SMatthew G. Knepley 
4679566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
4681df5d5c5SMatthew G. Knepley       }
4691df5d5c5SMatthew G. Knepley       break;
4701df5d5c5SMatthew G. Knepley     case 3:
4711df5d5c5SMatthew G. Knepley       if (simplex) {
4721df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {5, 2};
4731df5d5c5SMatthew G. Knepley         PetscInt    coneSize[7]         = {4, 4, 0, 0, 0, 0, 0};
4741df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {4, 3, 5, 2, 5, 3, 4, 6};
4751df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
4761df5d5c5SMatthew 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};
4771df5d5c5SMatthew G. Knepley 
4789566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
4791df5d5c5SMatthew G. Knepley       } else {
4801df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]         = {12, 2};
4811df5d5c5SMatthew G. Knepley         PetscInt    coneSize[14]         = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
4821df5d5c5SMatthew G. Knepley         PetscInt    cones[16]            = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
4831df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
4849371c9d4SSatish 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};
4851df5d5c5SMatthew G. Knepley 
4869566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
4871df5d5c5SMatthew G. Knepley       }
4881df5d5c5SMatthew G. Knepley       break;
489d71ae5a4SJacob Faibussowitsch     default:
490d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %" PetscInt_FMT, dim);
4911df5d5c5SMatthew G. Knepley     }
4921df5d5c5SMatthew G. Knepley   }
49346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
4941df5d5c5SMatthew G. Knepley   *newdm = dm;
4951df5d5c5SMatthew G. Knepley   if (refinementLimit > 0.0) {
4961df5d5c5SMatthew G. Knepley     DM          rdm;
4971df5d5c5SMatthew G. Knepley     const char *name;
4981df5d5c5SMatthew G. Knepley 
4999566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(*newdm, PETSC_FALSE));
5009566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(*newdm, refinementLimit));
5019566063dSJacob Faibussowitsch     PetscCall(DMRefine(*newdm, comm, &rdm));
5029566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)*newdm, &name));
5039566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)rdm, name));
5049566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
5051df5d5c5SMatthew G. Knepley     *newdm = rdm;
5061df5d5c5SMatthew G. Knepley   }
5071df5d5c5SMatthew G. Knepley   if (interpolate) {
5085fd9971aSMatthew G. Knepley     DM idm;
5091df5d5c5SMatthew G. Knepley 
5109566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*newdm, &idm));
5119566063dSJacob Faibussowitsch     PetscCall(DMDestroy(newdm));
5121df5d5c5SMatthew G. Knepley     *newdm = idm;
5131df5d5c5SMatthew G. Knepley   }
5143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5151df5d5c5SMatthew G. Knepley }
5161df5d5c5SMatthew G. Knepley 
517d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
518d71ae5a4SJacob Faibussowitsch {
5199318fe57SMatthew G. Knepley   const PetscInt numVertices    = 2;
5209318fe57SMatthew G. Knepley   PetscInt       markerRight    = 1;
5219318fe57SMatthew G. Knepley   PetscInt       markerLeft     = 1;
5229318fe57SMatthew G. Knepley   PetscBool      markerSeparate = PETSC_FALSE;
5239318fe57SMatthew G. Knepley   Vec            coordinates;
5249318fe57SMatthew G. Knepley   PetscSection   coordSection;
5259318fe57SMatthew G. Knepley   PetscScalar   *coords;
5269318fe57SMatthew G. Knepley   PetscInt       coordSize;
5279318fe57SMatthew G. Knepley   PetscMPIInt    rank;
5289318fe57SMatthew G. Knepley   PetscInt       cdim = 1, v;
529552f7358SJed Brown 
5309318fe57SMatthew G. Knepley   PetscFunctionBegin;
5319566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
5329318fe57SMatthew G. Knepley   if (markerSeparate) {
5339318fe57SMatthew G. Knepley     markerRight = 2;
5349318fe57SMatthew G. Knepley     markerLeft  = 1;
5359318fe57SMatthew G. Knepley   }
5369566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
537c5853193SPierre Jolivet   if (rank == 0) {
5389566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numVertices));
5399566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
5409566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 0, markerLeft));
5419566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", 1, markerRight));
5429318fe57SMatthew G. Knepley   }
5439566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
5449566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
5459318fe57SMatthew G. Knepley   /* Build coordinates */
5469566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, cdim));
5479566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
5489566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
5499566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, 0, numVertices));
5509566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
5519318fe57SMatthew G. Knepley   for (v = 0; v < numVertices; ++v) {
5529566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
5539566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
5549318fe57SMatthew G. Knepley   }
5559566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
5569566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
5579566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
5589566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
5599566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
5609566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
5619566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
5629566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
5639318fe57SMatthew G. Knepley   coords[0] = lower[0];
5649318fe57SMatthew G. Knepley   coords[1] = upper[0];
5659566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
5669566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
5679566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
5683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5699318fe57SMatthew G. Knepley }
57026492d91SMatthew G. Knepley 
571d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
572d71ae5a4SJacob Faibussowitsch {
5731df21d24SMatthew G. Knepley   const PetscInt numVertices    = (edges[0] + 1) * (edges[1] + 1);
5741df21d24SMatthew G. Knepley   const PetscInt numEdges       = edges[0] * (edges[1] + 1) + (edges[0] + 1) * edges[1];
575552f7358SJed Brown   PetscInt       markerTop      = 1;
576552f7358SJed Brown   PetscInt       markerBottom   = 1;
577552f7358SJed Brown   PetscInt       markerRight    = 1;
578552f7358SJed Brown   PetscInt       markerLeft     = 1;
579552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
580552f7358SJed Brown   Vec            coordinates;
581552f7358SJed Brown   PetscSection   coordSection;
582552f7358SJed Brown   PetscScalar   *coords;
583552f7358SJed Brown   PetscInt       coordSize;
584552f7358SJed Brown   PetscMPIInt    rank;
585552f7358SJed Brown   PetscInt       v, vx, vy;
586552f7358SJed Brown 
587552f7358SJed Brown   PetscFunctionBegin;
5889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
589552f7358SJed Brown   if (markerSeparate) {
5901df21d24SMatthew G. Knepley     markerTop    = 3;
5911df21d24SMatthew G. Knepley     markerBottom = 1;
5921df21d24SMatthew G. Knepley     markerRight  = 2;
5931df21d24SMatthew G. Knepley     markerLeft   = 4;
594552f7358SJed Brown   }
5959566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
596dd400576SPatrick Sanan   if (rank == 0) {
597552f7358SJed Brown     PetscInt e, ex, ey;
598552f7358SJed Brown 
5999566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numEdges + numVertices));
60048a46eb9SPierre Jolivet     for (e = 0; e < numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
6019566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
602552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
603552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
604552f7358SJed Brown         PetscInt edge   = vx * edges[1] + ey + edges[0] * (edges[1] + 1);
605552f7358SJed Brown         PetscInt vertex = ey * (edges[0] + 1) + vx + numEdges;
606da80777bSKarl Rupp         PetscInt cone[2];
607552f7358SJed Brown 
6089371c9d4SSatish Balay         cone[0] = vertex;
6099371c9d4SSatish Balay         cone[1] = vertex + edges[0] + 1;
6109566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
611552f7358SJed Brown         if (vx == edges[0]) {
6129566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
6139566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
614552f7358SJed Brown           if (ey == edges[1] - 1) {
6159566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
6169566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerRight));
617552f7358SJed Brown           }
618552f7358SJed Brown         } else if (vx == 0) {
6199566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
6209566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
621552f7358SJed Brown           if (ey == edges[1] - 1) {
6229566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
6239566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerLeft));
624552f7358SJed Brown           }
625552f7358SJed Brown         }
626552f7358SJed Brown       }
627552f7358SJed Brown     }
628552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
629552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
630552f7358SJed Brown         PetscInt edge   = vy * edges[0] + ex;
631552f7358SJed Brown         PetscInt vertex = vy * (edges[0] + 1) + ex + numEdges;
632da80777bSKarl Rupp         PetscInt cone[2];
633552f7358SJed Brown 
6349371c9d4SSatish Balay         cone[0] = vertex;
6359371c9d4SSatish Balay         cone[1] = vertex + 1;
6369566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, edge, cone));
637552f7358SJed Brown         if (vy == edges[1]) {
6389566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
6399566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
640552f7358SJed Brown           if (ex == edges[0] - 1) {
6419566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
6429566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerTop));
643552f7358SJed Brown           }
644552f7358SJed Brown         } else if (vy == 0) {
6459566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
6469566063dSJacob Faibussowitsch           PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
647552f7358SJed Brown           if (ex == edges[0] - 1) {
6489566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
6499566063dSJacob Faibussowitsch             PetscCall(DMSetLabelValue(dm, "Face Sets", cone[1], markerBottom));
650552f7358SJed Brown           }
651552f7358SJed Brown         }
652552f7358SJed Brown       }
653552f7358SJed Brown     }
654552f7358SJed Brown   }
6559566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
6569566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
657552f7358SJed Brown   /* Build coordinates */
6589566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 2));
6599566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
6609566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
6619566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices));
6629566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 2));
663552f7358SJed Brown   for (v = numEdges; v < numEdges + numVertices; ++v) {
6649566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 2));
6659566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 2));
666552f7358SJed Brown   }
6679566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
6689566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
6699566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
6709566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
6719566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
6729566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 2));
6739566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
6749566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
675552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
676552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
677552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 0] = lower[0] + ((upper[0] - lower[0]) / edges[0]) * vx;
678552f7358SJed Brown       coords[(vy * (edges[0] + 1) + vx) * 2 + 1] = lower[1] + ((upper[1] - lower[1]) / edges[1]) * vy;
679552f7358SJed Brown     }
680552f7358SJed Brown   }
6819566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
6829566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
6839566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
6843ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
685552f7358SJed Brown }
686552f7358SJed Brown 
687d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
688d71ae5a4SJacob Faibussowitsch {
6899e8abbc3SMichael Lange   PetscInt     vertices[3], numVertices;
6907b59f5a9SMichael Lange   PetscInt     numFaces       = 2 * faces[0] * faces[1] + 2 * faces[1] * faces[2] + 2 * faces[0] * faces[2];
691c2df9bbfSMatthew G. Knepley   PetscInt     markerTop      = 1;
692c2df9bbfSMatthew G. Knepley   PetscInt     markerBottom   = 1;
693c2df9bbfSMatthew G. Knepley   PetscInt     markerFront    = 1;
694c2df9bbfSMatthew G. Knepley   PetscInt     markerBack     = 1;
695c2df9bbfSMatthew G. Knepley   PetscInt     markerRight    = 1;
696c2df9bbfSMatthew G. Knepley   PetscInt     markerLeft     = 1;
697c2df9bbfSMatthew G. Knepley   PetscBool    markerSeparate = PETSC_FALSE;
698552f7358SJed Brown   Vec          coordinates;
699552f7358SJed Brown   PetscSection coordSection;
700552f7358SJed Brown   PetscScalar *coords;
701552f7358SJed Brown   PetscInt     coordSize;
702552f7358SJed Brown   PetscMPIInt  rank;
703552f7358SJed Brown   PetscInt     v, vx, vy, vz;
7047b59f5a9SMichael Lange   PetscInt     voffset, iface = 0, cone[4];
705552f7358SJed Brown 
706552f7358SJed Brown   PetscFunctionBegin;
7071dca8a05SBarry 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");
7089566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
709c2df9bbfSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
710c2df9bbfSMatthew G. Knepley   if (markerSeparate) {
711c2df9bbfSMatthew G. Knepley     markerBottom = 1;
712c2df9bbfSMatthew G. Knepley     markerTop    = 2;
713c2df9bbfSMatthew G. Knepley     markerFront  = 3;
714c2df9bbfSMatthew G. Knepley     markerBack   = 4;
715c2df9bbfSMatthew G. Knepley     markerRight  = 5;
716c2df9bbfSMatthew G. Knepley     markerLeft   = 6;
717c2df9bbfSMatthew G. Knepley   }
7189371c9d4SSatish Balay   vertices[0] = faces[0] + 1;
7199371c9d4SSatish Balay   vertices[1] = faces[1] + 1;
7209371c9d4SSatish Balay   vertices[2] = faces[2] + 1;
7219e8abbc3SMichael Lange   numVertices = vertices[0] * vertices[1] * vertices[2];
722dd400576SPatrick Sanan   if (rank == 0) {
723552f7358SJed Brown     PetscInt f;
724552f7358SJed Brown 
7259566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numFaces + numVertices));
72648a46eb9SPierre Jolivet     for (f = 0; f < numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
7279566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
7287b59f5a9SMichael Lange 
7297b59f5a9SMichael Lange     /* Side 0 (Top) */
7307b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
7317b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
7327b59f5a9SMichael Lange         voffset = numFaces + vertices[0] * vertices[1] * (vertices[2] - 1) + vy * vertices[0] + vx;
7339371c9d4SSatish Balay         cone[0] = voffset;
7349371c9d4SSatish Balay         cone[1] = voffset + 1;
7359371c9d4SSatish Balay         cone[2] = voffset + vertices[0] + 1;
7369371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
7379566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
738c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerTop));
739c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerTop));
740c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerTop));
741c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerTop));
742c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerTop));
7437b59f5a9SMichael Lange         iface++;
744552f7358SJed Brown       }
745552f7358SJed Brown     }
7467b59f5a9SMichael Lange 
7477b59f5a9SMichael Lange     /* Side 1 (Bottom) */
7487b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
7497b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
7507b59f5a9SMichael Lange         voffset = numFaces + vy * (faces[0] + 1) + vx;
7519371c9d4SSatish Balay         cone[0] = voffset + 1;
7529371c9d4SSatish Balay         cone[1] = voffset;
7539371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
7549371c9d4SSatish Balay         cone[3] = voffset + vertices[0] + 1;
7559566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
756c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBottom));
757c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBottom));
758c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBottom));
759c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerBottom));
760c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 1, markerBottom));
7617b59f5a9SMichael Lange         iface++;
762552f7358SJed Brown       }
763552f7358SJed Brown     }
7647b59f5a9SMichael Lange 
7657b59f5a9SMichael Lange     /* Side 2 (Front) */
7667b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
7677b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
7687b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vx;
7699371c9d4SSatish Balay         cone[0] = voffset;
7709371c9d4SSatish Balay         cone[1] = voffset + 1;
7719371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + 1;
7729371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1];
7739566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
774c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerFront));
775c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerFront));
776c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerFront));
777c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerFront));
778c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerFront));
7797b59f5a9SMichael Lange         iface++;
780552f7358SJed Brown       }
7817b59f5a9SMichael Lange     }
7827b59f5a9SMichael Lange 
7837b59f5a9SMichael Lange     /* Side 3 (Back) */
7847b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
7857b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
7867b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vertices[0] * (vertices[1] - 1) + vx;
7879371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
7889371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1] + 1;
7899371c9d4SSatish Balay         cone[2] = voffset + 1;
7909371c9d4SSatish Balay         cone[3] = voffset;
7919566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
792c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerBack));
793c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerBack));
794c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 1, markerBack));
795c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerBack));
796c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 1, markerBack));
7977b59f5a9SMichael Lange         iface++;
7987b59f5a9SMichael Lange       }
7997b59f5a9SMichael Lange     }
8007b59f5a9SMichael Lange 
8017b59f5a9SMichael Lange     /* Side 4 (Left) */
8027b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
8037b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
8047b59f5a9SMichael Lange         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0];
8059371c9d4SSatish Balay         cone[0] = voffset;
8069371c9d4SSatish Balay         cone[1] = voffset + vertices[0] * vertices[1];
8079371c9d4SSatish Balay         cone[2] = voffset + vertices[0] * vertices[1] + vertices[0];
8089371c9d4SSatish Balay         cone[3] = voffset + vertices[0];
8099566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
810c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerLeft));
811c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerLeft));
812c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerLeft));
813c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[1] + 0, markerLeft));
814c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerLeft));
8157b59f5a9SMichael Lange         iface++;
8167b59f5a9SMichael Lange       }
8177b59f5a9SMichael Lange     }
8187b59f5a9SMichael Lange 
8197b59f5a9SMichael Lange     /* Side 5 (Right) */
8207b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
8217b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
822aab5bcd8SJed Brown         voffset = numFaces + vz * vertices[0] * vertices[1] + vy * vertices[0] + faces[0];
8239371c9d4SSatish Balay         cone[0] = voffset + vertices[0] * vertices[1];
8249371c9d4SSatish Balay         cone[1] = voffset;
8259371c9d4SSatish Balay         cone[2] = voffset + vertices[0];
8269371c9d4SSatish Balay         cone[3] = voffset + vertices[0] * vertices[1] + vertices[0];
8279566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, iface, cone));
828c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", iface, markerRight));
829c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + 0, markerRight));
830c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] + 0, markerRight));
831c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + 0, markerRight));
832c2df9bbfSMatthew G. Knepley         PetscCall(DMSetLabelValue(dm, "marker", voffset + vertices[0] * vertices[1] + vertices[0], markerRight));
8337b59f5a9SMichael Lange         iface++;
8347b59f5a9SMichael Lange       }
835552f7358SJed Brown     }
836552f7358SJed Brown   }
8379566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
8389566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
839552f7358SJed Brown   /* Build coordinates */
8409566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, 3));
8419566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
8429566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
8439566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices));
8449566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 3));
845552f7358SJed Brown   for (v = numFaces; v < numFaces + numVertices; ++v) {
8469566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, 3));
8479566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 3));
848552f7358SJed Brown   }
8499566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
8509566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
8519566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
8529566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
8539566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
8549566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, 3));
8559566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
8569566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
857552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
858552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
859552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
860552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 0] = lower[0] + ((upper[0] - lower[0]) / faces[0]) * vx;
861552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 1] = lower[1] + ((upper[1] - lower[1]) / faces[1]) * vy;
862552f7358SJed Brown         coords[((vz * (faces[1] + 1) + vy) * (faces[0] + 1) + vx) * 3 + 2] = lower[2] + ((upper[2] - lower[2]) / faces[2]) * vz;
863552f7358SJed Brown       }
864552f7358SJed Brown     }
865552f7358SJed Brown   }
8669566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
8679566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
8689566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
8693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
870552f7358SJed Brown }
871552f7358SJed Brown 
872d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxSurfaceMesh_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate)
873d71ae5a4SJacob Faibussowitsch {
8749318fe57SMatthew G. Knepley   PetscFunctionBegin;
8759318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
87646139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
8779566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim - 1));
8789566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim));
8799318fe57SMatthew G. Knepley   switch (dim) {
880d71ae5a4SJacob Faibussowitsch   case 1:
881d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_1D_Internal(dm, lower, upper, faces));
882d71ae5a4SJacob Faibussowitsch     break;
883d71ae5a4SJacob Faibussowitsch   case 2:
884d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_2D_Internal(dm, lower, upper, faces));
885d71ae5a4SJacob Faibussowitsch     break;
886d71ae5a4SJacob Faibussowitsch   case 3:
887d71ae5a4SJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Tensor_3D_Internal(dm, lower, upper, faces));
888d71ae5a4SJacob Faibussowitsch     break;
889d71ae5a4SJacob Faibussowitsch   default:
890d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Dimension not supported: %" PetscInt_FMT, dim);
8919318fe57SMatthew G. Knepley   }
89246139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
8939566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
8943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
8959318fe57SMatthew G. Knepley }
8969318fe57SMatthew G. Knepley 
8979318fe57SMatthew G. Knepley /*@C
8989318fe57SMatthew G. Knepley   DMPlexCreateBoxSurfaceMesh - Creates a mesh on the surface of the tensor product of unit intervals (box) using tensor cells (hexahedra).
8999318fe57SMatthew G. Knepley 
9009318fe57SMatthew G. Knepley   Collective
9019318fe57SMatthew G. Knepley 
9029318fe57SMatthew G. Knepley   Input Parameters:
903a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
90420f4b53cSBarry Smith . dim         - The spatial dimension of the box, so the resulting mesh is has dimension `dim`-1
90520f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
90620f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
90720f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
9089318fe57SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
9099318fe57SMatthew G. Knepley 
9109318fe57SMatthew G. Knepley   Output Parameter:
911a1cb98faSBarry Smith . dm - The `DM` object
9129318fe57SMatthew G. Knepley 
9139318fe57SMatthew G. Knepley   Level: beginner
9149318fe57SMatthew G. Knepley 
9151cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateBoxMesh()`, `DMPlexCreateFromFile()`, `DMSetType()`, `DMCreate()`
9169318fe57SMatthew G. Knepley @*/
917d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBoxSurfaceMesh(MPI_Comm comm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], PetscBool interpolate, DM *dm)
918d71ae5a4SJacob Faibussowitsch {
9199318fe57SMatthew G. Knepley   PetscInt  fac[3] = {1, 1, 1};
9209318fe57SMatthew G. Knepley   PetscReal low[3] = {0, 0, 0};
9219318fe57SMatthew G. Knepley   PetscReal upp[3] = {1, 1, 1};
9229318fe57SMatthew G. Knepley 
9239318fe57SMatthew G. Knepley   PetscFunctionBegin;
9249566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
9259566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
9269566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(*dm, dim, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, interpolate));
9273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
9289318fe57SMatthew G. Knepley }
9299318fe57SMatthew G. Knepley 
930d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateLineMesh_Internal(DM dm, PetscInt segments, PetscReal lower, PetscReal upper, DMBoundaryType bd)
931d71ae5a4SJacob Faibussowitsch {
932fdbf62faSLisandro Dalcin   PetscInt     i, fStart, fEnd, numCells = 0, numVerts = 0;
933fdbf62faSLisandro Dalcin   PetscInt     numPoints[2], *coneSize, *cones, *coneOrientations;
934fdbf62faSLisandro Dalcin   PetscScalar *vertexCoords;
935fdbf62faSLisandro Dalcin   PetscReal    L, maxCell;
936fdbf62faSLisandro Dalcin   PetscBool    markerSeparate = PETSC_FALSE;
937fdbf62faSLisandro Dalcin   PetscInt     markerLeft = 1, faceMarkerLeft = 1;
938fdbf62faSLisandro Dalcin   PetscInt     markerRight = 1, faceMarkerRight = 2;
939fdbf62faSLisandro Dalcin   PetscBool    wrap = (bd == DM_BOUNDARY_PERIODIC || bd == DM_BOUNDARY_TWIST) ? PETSC_TRUE : PETSC_FALSE;
940fdbf62faSLisandro Dalcin   PetscMPIInt  rank;
941fdbf62faSLisandro Dalcin 
942fdbf62faSLisandro Dalcin   PetscFunctionBegin;
9434f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
944fdbf62faSLisandro Dalcin 
9459566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, 1));
9469566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
9479566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
948fdbf62faSLisandro Dalcin 
9499566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
950dd400576SPatrick Sanan   if (rank == 0) numCells = segments;
951dd400576SPatrick Sanan   if (rank == 0) numVerts = segments + (wrap ? 0 : 1);
952fdbf62faSLisandro Dalcin 
9539371c9d4SSatish Balay   numPoints[0] = numVerts;
9549371c9d4SSatish Balay   numPoints[1] = numCells;
9559566063dSJacob Faibussowitsch   PetscCall(PetscMalloc4(numCells + numVerts, &coneSize, numCells * 2, &cones, numCells + numVerts, &coneOrientations, numVerts, &vertexCoords));
9569566063dSJacob Faibussowitsch   PetscCall(PetscArrayzero(coneOrientations, numCells + numVerts));
957ad540459SPierre Jolivet   for (i = 0; i < numCells; ++i) coneSize[i] = 2;
958ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) coneSize[numCells + i] = 0;
9599371c9d4SSatish Balay   for (i = 0; i < numCells; ++i) {
9609371c9d4SSatish Balay     cones[2 * i]     = numCells + i % numVerts;
9619371c9d4SSatish Balay     cones[2 * i + 1] = numCells + (i + 1) % numVerts;
9629371c9d4SSatish Balay   }
963ad540459SPierre Jolivet   for (i = 0; i < numVerts; ++i) vertexCoords[i] = lower + (upper - lower) * ((PetscReal)i / (PetscReal)numCells);
9649566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
9659566063dSJacob Faibussowitsch   PetscCall(PetscFree4(coneSize, cones, coneOrientations, vertexCoords));
966fdbf62faSLisandro Dalcin 
9679566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
9689371c9d4SSatish Balay   if (markerSeparate) {
9699371c9d4SSatish Balay     markerLeft  = faceMarkerLeft;
9709371c9d4SSatish Balay     markerRight = faceMarkerRight;
9719371c9d4SSatish Balay   }
972dd400576SPatrick Sanan   if (!wrap && rank == 0) {
9739566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd));
9749566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fStart, markerLeft));
9759566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "marker", fEnd - 1, markerRight));
9769566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fStart, faceMarkerLeft));
9779566063dSJacob Faibussowitsch     PetscCall(DMSetLabelValue(dm, "Face Sets", fEnd - 1, faceMarkerRight));
978fdbf62faSLisandro Dalcin   }
979fdbf62faSLisandro Dalcin   if (wrap) {
980fdbf62faSLisandro Dalcin     L       = upper - lower;
981fdbf62faSLisandro Dalcin     maxCell = (PetscReal)1.1 * (L / (PetscReal)PetscMax(1, segments));
9824fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, &maxCell, &lower, &L));
983fdbf62faSLisandro Dalcin   }
9849566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
9853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
986fdbf62faSLisandro Dalcin }
987fdbf62faSLisandro Dalcin 
9884054ae39SJames Wright // Creates "Face Sets" label based on the standard box labeling conventions
989d7d2d1d2SJames Wright static PetscErrorCode DMPlexSetBoxLabel_Internal(DM dm, const DMBoundaryType periodicity[])
9904054ae39SJames Wright {
9916ff49feeSJames Wright   DM              cdm;
9926ff49feeSJames Wright   PetscSection    csection;
9936ff49feeSJames Wright   Vec             coordinates;
9944054ae39SJames Wright   DMLabel         label;
9956ff49feeSJames Wright   IS              faces_is;
9962b4f33d9SJames Wright   PetscInt        dim, num_face = 0;
9974054ae39SJames Wright   const PetscInt *faces;
9984054ae39SJames Wright   PetscInt        faceMarkerBottom, faceMarkerTop, faceMarkerFront, faceMarkerBack, faceMarkerRight, faceMarkerLeft;
9994054ae39SJames Wright 
10004054ae39SJames Wright   PetscFunctionBeginUser;
10014054ae39SJames Wright   PetscCall(DMGetDimension(dm, &dim));
1002d7c1f440SPierre Jolivet   PetscCheck((dim == 2) || (dim == 3), PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "DMPlex box labeling only supports 2D and 3D meshes, received DM of dimension %" PetscInt_FMT, dim);
10034054ae39SJames Wright   // Get Face Sets label
10044054ae39SJames Wright   PetscCall(DMGetLabel(dm, "Face Sets", &label));
10054054ae39SJames Wright   if (label) {
10064054ae39SJames Wright     PetscCall(DMLabelReset(label));
10074054ae39SJames Wright   } else {
10084054ae39SJames Wright     PetscCall(DMCreateLabel(dm, "Face Sets"));
10094054ae39SJames Wright     PetscCall(DMGetLabel(dm, "Face Sets", &label));
10104054ae39SJames Wright   }
10114054ae39SJames Wright   PetscCall(DMPlexMarkBoundaryFaces(dm, 1, label));
10126ff49feeSJames Wright   PetscCall(DMGetStratumIS(dm, "Face Sets", 1, &faces_is));
10134054ae39SJames Wright 
10144054ae39SJames Wright   switch (dim) {
10154054ae39SJames Wright   case 2:
10164054ae39SJames Wright     faceMarkerTop    = 3;
10174054ae39SJames Wright     faceMarkerBottom = 1;
10184054ae39SJames Wright     faceMarkerRight  = 2;
10194054ae39SJames Wright     faceMarkerLeft   = 4;
10204054ae39SJames Wright     break;
10214054ae39SJames Wright   case 3:
10224054ae39SJames Wright     faceMarkerBottom = 1;
10234054ae39SJames Wright     faceMarkerTop    = 2;
10244054ae39SJames Wright     faceMarkerFront  = 3;
10254054ae39SJames Wright     faceMarkerBack   = 4;
10264054ae39SJames Wright     faceMarkerRight  = 5;
10274054ae39SJames Wright     faceMarkerLeft   = 6;
10284054ae39SJames Wright     break;
10294054ae39SJames Wright   default:
10304054ae39SJames Wright     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
10314054ae39SJames Wright   }
10324054ae39SJames Wright 
10332b4f33d9SJames Wright   if (faces_is) PetscCall(ISGetLocalSize(faces_is, &num_face));
10342b4f33d9SJames Wright   if (faces_is) PetscCall(ISGetIndices(faces_is, &faces));
10356ff49feeSJames Wright   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
10366ff49feeSJames Wright   PetscCall(DMGetCoordinateDM(dm, &cdm));
10376ff49feeSJames Wright   PetscCall(DMGetLocalSection(cdm, &csection));
10384054ae39SJames Wright   for (PetscInt f = 0; f < num_face; ++f) {
10396ff49feeSJames Wright     PetscScalar *coords = NULL;
10406ff49feeSJames Wright     PetscInt     face = faces[f], flip = 1, label_value = -1, coords_size;
10414054ae39SJames Wright 
10424054ae39SJames Wright     { // Determine if orientation of face is flipped
10434054ae39SJames Wright       PetscInt        num_cells_support, num_faces, start = -1;
10444054ae39SJames Wright       const PetscInt *orients, *cell_faces, *cells;
10454054ae39SJames Wright 
10464054ae39SJames Wright       PetscCall(DMPlexGetSupport(dm, face, &cells));
10474054ae39SJames Wright       PetscCall(DMPlexGetSupportSize(dm, face, &num_cells_support));
10484054ae39SJames 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);
10494054ae39SJames Wright       PetscCall(DMPlexGetCone(dm, cells[0], &cell_faces));
10504054ae39SJames Wright       PetscCall(DMPlexGetConeSize(dm, cells[0], &num_faces));
10514054ae39SJames Wright       for (PetscInt i = 0; i < num_faces; i++) {
10524054ae39SJames Wright         if (cell_faces[i] == face) start = i;
10534054ae39SJames Wright       }
10544054ae39SJames Wright       PetscCheck(start >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT, "Could not find face %" PetscInt_FMT " in cone of its support", face);
10554054ae39SJames Wright       PetscCall(DMPlexGetConeOrientation(dm, cells[0], &orients));
10564054ae39SJames Wright       if (orients[start] < 0) flip = -1;
10574054ae39SJames Wright     }
10584054ae39SJames Wright 
10596ff49feeSJames Wright     // Cannot use DMPlexComputeCellGeometryFVM() for high-order geometry, so must calculate normal vectors manually
10606ff49feeSJames Wright     // Use the vertices (depth 0) of coordinate DM to calculate normal vector
10616ff49feeSJames Wright     PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
10624054ae39SJames Wright     switch (dim) {
10634054ae39SJames Wright     case 2: {
10646ff49feeSJames Wright       PetscScalar vec[2];
10656ff49feeSJames Wright 
10666ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) vec[d] = flip * (PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]));
10676ff49feeSJames Wright       PetscScalar normal[] = {vec[1], -vec[0]};
10686ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
10696ff49feeSJames Wright         label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
10704054ae39SJames Wright       } else {
10716ff49feeSJames Wright         label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerTop : faceMarkerBottom;
10724054ae39SJames Wright       }
10734054ae39SJames Wright     } break;
10744054ae39SJames Wright     case 3: {
10756ff49feeSJames Wright       PetscScalar vec1[3], vec2[3], normal[3];
10766ff49feeSJames Wright 
10776ff49feeSJames Wright       for (PetscInt d = 0; d < dim; ++d) {
10786ff49feeSJames Wright         vec1[d] = PetscRealPart(coords[1 * dim + d]) - PetscRealPart(coords[0 * dim + d]);
10796ff49feeSJames Wright         vec2[d] = PetscRealPart(coords[2 * dim + d]) - PetscRealPart(coords[1 * dim + d]);
10806ff49feeSJames Wright       }
10816ff49feeSJames Wright 
10826ff49feeSJames Wright       // Calculate normal vector via cross-product
10836ff49feeSJames Wright       normal[0] = flip * ((vec1[1] * vec2[2]) - (vec1[2] * vec2[1]));
10846ff49feeSJames Wright       normal[1] = flip * ((vec1[2] * vec2[0]) - (vec1[0] * vec2[2]));
10856ff49feeSJames Wright       normal[2] = flip * ((vec1[0] * vec2[1]) - (vec1[1] * vec2[0]));
10866ff49feeSJames Wright 
10876ff49feeSJames Wright       if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[1])) {
10886ff49feeSJames Wright         if (PetscAbsScalar(normal[0]) > PetscAbsScalar(normal[2])) {
10896ff49feeSJames Wright           label_value = PetscRealPart(normal[0]) > 0 ? faceMarkerRight : faceMarkerLeft;
10904054ae39SJames Wright         } else {
10916ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
10924054ae39SJames Wright         }
10934054ae39SJames Wright       } else {
10946ff49feeSJames Wright         if (PetscAbsScalar(normal[1]) > PetscAbsScalar(normal[2])) {
10956ff49feeSJames Wright           label_value = PetscRealPart(normal[1]) > 0 ? faceMarkerBack : faceMarkerFront;
10964054ae39SJames Wright         } else {
10976ff49feeSJames Wright           label_value = PetscRealPart(normal[2]) > 0 ? faceMarkerTop : faceMarkerBottom;
10984054ae39SJames Wright         }
10994054ae39SJames Wright       }
11004054ae39SJames Wright     } break;
11014054ae39SJames Wright     }
11024054ae39SJames Wright 
11034054ae39SJames Wright     PetscInt previous_label_value; // always 1 due to DMPlexMarkBoundaryFaces call above
11044054ae39SJames Wright     PetscCall(DMGetLabelValue(dm, "Face Sets", face, &previous_label_value));
11054054ae39SJames Wright     PetscCall(DMClearLabelValue(dm, "Face Sets", face, previous_label_value));
11064054ae39SJames Wright     PetscCall(DMSetLabelValue(dm, "Face Sets", face, label_value));
11076ff49feeSJames Wright     PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
11084054ae39SJames Wright   }
11092b4f33d9SJames Wright   if (faces_is) PetscCall(ISRestoreIndices(faces_is, &faces));
11106ff49feeSJames Wright   PetscCall(ISDestroy(&faces_is));
1111d7d2d1d2SJames Wright 
1112d7d2d1d2SJames Wright   // Create Isoperiodic SF from newly-created face labels
1113d7d2d1d2SJames Wright   PetscSF     periodicsfs[3];
1114d7d2d1d2SJames Wright   PetscInt    periodic_sf_index  = 0;
1115d7d2d1d2SJames Wright   PetscScalar transform[3][4][4] = {{{0.}}};
1116d7d2d1d2SJames Wright   for (PetscInt d = 0; d < dim; d++) {
1117d7d2d1d2SJames Wright     IS              donor_is, periodic_is;
1118d7d2d1d2SJames Wright     const PetscInt *donor_faces = NULL, *periodic_faces = NULL;
1119d7d2d1d2SJames Wright     PetscInt        num_donor = 0, num_periodic = 0;
1120d7d2d1d2SJames Wright     PetscSF         centroidsf;
1121d7d2d1d2SJames Wright     PetscReal       donor_to_periodic_distance;
1122d7d2d1d2SJames Wright     const PetscInt  face_pairings[2][3][2] = {
1123d7d2d1d2SJames Wright       // 2D face pairings, {donor, periodic}
1124d7d2d1d2SJames Wright       {{4, 2}, {1, 3}},
1125d7d2d1d2SJames Wright       // 3D face pairings
1126d7d2d1d2SJames Wright       {{5, 6}, {3, 4}, {1, 2}}
1127d7d2d1d2SJames Wright     };
1128d7d2d1d2SJames Wright 
1129d7d2d1d2SJames Wright     if (periodicity[d] != DM_BOUNDARY_PERIODIC) continue;
1130d7d2d1d2SJames Wright     {
1131d7d2d1d2SJames Wright       // Compute centroidsf, which is the mapping from donor faces to periodic faces
1132d7d2d1d2SJames Wright       // Matches the centroid of the faces together, ignoring the periodic direction component (which should not match between donor and periodic face)
1133d7d2d1d2SJames Wright       PetscInt     coords_size, centroid_comps = dim - 1;
1134d7d2d1d2SJames Wright       PetscScalar *coords = NULL;
1135d7d2d1d2SJames Wright       PetscReal   *donor_centroids, *periodic_centroids;
1136d7d2d1d2SJames Wright       PetscReal    loc_periodic[2] = {PETSC_MIN_REAL, PETSC_MIN_REAL}, loc_periodic_global[2]; // Location of donor (0) and periodic (1) faces in periodic direction
1137d7d2d1d2SJames Wright 
1138d7d2d1d2SJames Wright       PetscCall(DMGetStratumIS(dm, "Face Sets", face_pairings[dim - 2][d][0], &donor_is));
1139d7d2d1d2SJames Wright       PetscCall(DMGetStratumIS(dm, "Face Sets", face_pairings[dim - 2][d][1], &periodic_is));
1140d7d2d1d2SJames Wright       if (donor_is) {
1141d7d2d1d2SJames Wright         PetscCall(ISGetLocalSize(donor_is, &num_donor));
1142d7d2d1d2SJames Wright         PetscCall(ISGetIndices(donor_is, &donor_faces));
1143d7d2d1d2SJames Wright       }
1144d7d2d1d2SJames Wright       if (periodic_is) {
1145d7d2d1d2SJames Wright         PetscCall(ISGetLocalSize(periodic_is, &num_periodic));
1146d7d2d1d2SJames Wright         PetscCall(ISGetIndices(periodic_is, &periodic_faces));
1147d7d2d1d2SJames Wright       }
1148d7d2d1d2SJames Wright       PetscCall(PetscCalloc2(num_donor * centroid_comps, &donor_centroids, num_periodic * centroid_comps, &periodic_centroids));
1149d7d2d1d2SJames Wright       for (PetscInt f = 0; f < num_donor; f++) {
1150d7d2d1d2SJames Wright         PetscInt face = donor_faces[f], num_coords;
1151d7d2d1d2SJames Wright         PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
1152d7d2d1d2SJames Wright         num_coords = coords_size / dim;
1153d7d2d1d2SJames Wright         for (PetscInt c = 0; c < num_coords; c++) {
1154d7d2d1d2SJames Wright           PetscInt comp_index = 0;
1155d7d2d1d2SJames Wright           loc_periodic[0]     = PetscRealPart(coords[c * dim + d]);
1156d7d2d1d2SJames Wright           for (PetscInt i = 0; i < dim; i++) {
1157d7d2d1d2SJames Wright             if (i == d) continue; // Periodic direction not used for centroid calculation
1158d7d2d1d2SJames Wright             donor_centroids[f * centroid_comps + comp_index] += PetscRealPart(coords[c * dim + i]) / num_coords;
1159d7d2d1d2SJames Wright             comp_index++;
1160d7d2d1d2SJames Wright           }
1161d7d2d1d2SJames Wright         }
1162d7d2d1d2SJames Wright         PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
1163d7d2d1d2SJames Wright       }
1164d7d2d1d2SJames Wright 
1165d7d2d1d2SJames Wright       for (PetscInt f = 0; f < num_periodic; f++) {
1166d7d2d1d2SJames Wright         PetscInt face = periodic_faces[f], num_coords;
1167d7d2d1d2SJames Wright         PetscCall(DMPlexVecGetClosureAtDepth_Internal(cdm, csection, coordinates, face, 0, &coords_size, &coords));
1168d7d2d1d2SJames Wright         num_coords = coords_size / dim;
1169d7d2d1d2SJames Wright         for (PetscInt c = 0; c < num_coords; c++) {
1170d7d2d1d2SJames Wright           PetscInt comp_index = 0;
1171d7d2d1d2SJames Wright           loc_periodic[1]     = PetscRealPart(coords[c * dim + d]);
1172d7d2d1d2SJames Wright           for (PetscInt i = 0; i < dim; i++) {
1173d7d2d1d2SJames Wright             if (i == d) continue; // Periodic direction not used for centroid calculation
1174d7d2d1d2SJames Wright             periodic_centroids[f * centroid_comps + comp_index] += PetscRealPart(coords[c * dim + i]) / num_coords;
1175d7d2d1d2SJames Wright             comp_index++;
1176d7d2d1d2SJames Wright           }
1177d7d2d1d2SJames Wright         }
1178d7d2d1d2SJames Wright         PetscCall(DMPlexVecRestoreClosure(cdm, csection, coordinates, face, &coords_size, &coords));
1179d7d2d1d2SJames Wright       }
1180d7d2d1d2SJames Wright       PetscCallMPI(MPIU_Allreduce(loc_periodic, loc_periodic_global, 2, MPIU_REAL, MPIU_MAX, PetscObjectComm((PetscObject)dm)));
1181d7d2d1d2SJames Wright       donor_to_periodic_distance = loc_periodic_global[1] - loc_periodic_global[0];
1182d7d2d1d2SJames Wright 
1183d7d2d1d2SJames Wright       PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &centroidsf));
1184d7d2d1d2SJames Wright       PetscCall(PetscSFSetGraphFromCoordinates(centroidsf, num_donor, num_periodic, centroid_comps, 1e-10, donor_centroids, periodic_centroids));
1185d7d2d1d2SJames Wright       PetscCall(PetscSFViewFromOptions(centroidsf, NULL, "-dm_plex_box_label_centroid_sf_view"));
1186d7d2d1d2SJames Wright       PetscCall(PetscFree2(donor_centroids, periodic_centroids));
1187d7d2d1d2SJames Wright     }
1188d7d2d1d2SJames Wright 
1189d7d2d1d2SJames Wright     { // Create Isoperiodic SF using centroidsSF
1190d7d2d1d2SJames Wright       PetscInt           pStart, pEnd;
1191d7d2d1d2SJames Wright       PetscInt          *leaf_faces;
1192d7d2d1d2SJames Wright       const PetscSFNode *firemote;
1193d7d2d1d2SJames Wright       PetscSFNode       *isoperiodic_leaves;
1194d7d2d1d2SJames Wright 
1195d7d2d1d2SJames Wright       PetscCall(PetscMalloc1(num_periodic, &leaf_faces));
1196d7d2d1d2SJames Wright       PetscCall(PetscSFBcastBegin(centroidsf, MPIU_INT, donor_faces, leaf_faces, MPI_REPLACE));
1197d7d2d1d2SJames Wright       PetscCall(PetscSFBcastEnd(centroidsf, MPIU_INT, donor_faces, leaf_faces, MPI_REPLACE));
1198d7d2d1d2SJames Wright 
1199d7d2d1d2SJames Wright       PetscCall(PetscMalloc1(num_periodic, &isoperiodic_leaves));
1200d7d2d1d2SJames Wright       PetscCall(PetscSFGetGraph(centroidsf, NULL, NULL, NULL, &firemote));
1201d7d2d1d2SJames Wright       for (PetscInt l = 0; l < num_periodic; ++l) {
1202d7d2d1d2SJames Wright         isoperiodic_leaves[l].index = leaf_faces[l];
1203d7d2d1d2SJames Wright         isoperiodic_leaves[l].rank  = firemote[l].rank;
1204d7d2d1d2SJames Wright       }
1205d7d2d1d2SJames Wright 
1206d7d2d1d2SJames Wright       PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
1207d7d2d1d2SJames Wright       PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &periodicsfs[periodic_sf_index]));
1208d7d2d1d2SJames Wright       PetscCall(PetscSFSetGraph(periodicsfs[periodic_sf_index], pEnd - pStart, num_periodic, (PetscInt *)periodic_faces, PETSC_COPY_VALUES, isoperiodic_leaves, PETSC_OWN_POINTER));
1209d7d2d1d2SJames Wright       PetscCall(PetscSFViewFromOptions(periodicsfs[periodic_sf_index], NULL, "-dm_plex_box_label_periodic_sf_view"));
1210d7d2d1d2SJames Wright       PetscCall(PetscFree(leaf_faces));
1211d7d2d1d2SJames Wright     }
1212d7d2d1d2SJames Wright 
1213d7d2d1d2SJames Wright     transform[periodic_sf_index][0][0] = 1;
1214d7d2d1d2SJames Wright     transform[periodic_sf_index][1][1] = 1;
1215d7d2d1d2SJames Wright     transform[periodic_sf_index][2][2] = 1;
1216d7d2d1d2SJames Wright     transform[periodic_sf_index][3][3] = 1;
1217d7d2d1d2SJames Wright     transform[periodic_sf_index][d][3] = donor_to_periodic_distance;
1218d7d2d1d2SJames Wright 
1219d7d2d1d2SJames Wright     periodic_sf_index++;
1220d7d2d1d2SJames Wright     PetscCall(PetscSFDestroy(&centroidsf));
1221d7d2d1d2SJames Wright     if (donor_is) {
1222d7d2d1d2SJames Wright       PetscCall(ISRestoreIndices(donor_is, &donor_faces));
1223d7d2d1d2SJames Wright       PetscCall(ISDestroy(&donor_is));
1224d7d2d1d2SJames Wright     }
1225d7d2d1d2SJames Wright     if (periodic_is) {
1226d7d2d1d2SJames Wright       PetscCall(ISRestoreIndices(periodic_is, &periodic_faces));
1227d7d2d1d2SJames Wright       PetscCall(ISDestroy(&periodic_is));
1228d7d2d1d2SJames Wright     }
1229d7d2d1d2SJames Wright     PetscCall(DMClearLabelStratum(dm, "Face Sets", face_pairings[dim - 2][d][0]));
1230d7d2d1d2SJames Wright     PetscCall(DMClearLabelStratum(dm, "Face Sets", face_pairings[dim - 2][d][1]));
1231d7d2d1d2SJames Wright   }
1232d7d2d1d2SJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(dm, periodic_sf_index, periodicsfs));
1233d7d2d1d2SJames Wright   PetscCall(DMPlexSetIsoperiodicFaceTransform(dm, periodic_sf_index, (const PetscScalar *)transform));
1234d7d2d1d2SJames Wright   for (PetscInt p = 0; p < periodic_sf_index; p++) PetscCall(PetscSFDestroy(&periodicsfs[p]));
1235d7d2d1d2SJames Wright 
1236d7d2d1d2SJames Wright   { // Update coordinate DM with new Face Sets label
1237d7d2d1d2SJames Wright     DM      cdm;
1238d7d2d1d2SJames Wright     DMLabel oldFaceSets, newFaceSets;
1239d7d2d1d2SJames Wright     PetscCall(DMGetCoordinateDM(dm, &cdm));
1240d7d2d1d2SJames Wright     PetscCall(DMGetLabel(cdm, "Face Sets", &oldFaceSets));
1241d7d2d1d2SJames Wright     if (oldFaceSets) PetscCall(DMRemoveLabelBySelf(cdm, &oldFaceSets, PETSC_FALSE));
1242d7d2d1d2SJames Wright     PetscCall(DMLabelDuplicate(label, &newFaceSets));
1243d7d2d1d2SJames Wright     PetscCall(DMAddLabel(cdm, newFaceSets));
1244d7d2d1d2SJames Wright     PetscCall(DMLabelDestroy(&newFaceSets));
1245d7d2d1d2SJames Wright   }
12464054ae39SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
12474054ae39SJames Wright }
12484054ae39SJames Wright 
1249d698cf03SStefano Zampini static PetscErrorCode DMPlexCreateSquareMesh_Simplex_CrissCross(DM dm, const PetscInt edges[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType bd[])
1250d698cf03SStefano Zampini {
1251d698cf03SStefano Zampini   PetscInt       markerTop = 1, faceMarkerTop = 3;
1252d698cf03SStefano Zampini   PetscInt       markerBottom = 1, faceMarkerBottom = 1;
1253d698cf03SStefano Zampini   PetscInt       markerRight = 1, faceMarkerRight = 2;
1254d698cf03SStefano Zampini   PetscInt       markerLeft = 1, faceMarkerLeft = 4;
1255d698cf03SStefano Zampini   PetscBool      markerSeparate = PETSC_FALSE;
1256d698cf03SStefano Zampini   DMBoundaryType bdX = bd[0], bdY = bd[1];
1257d698cf03SStefano Zampini   PetscMPIInt    rank;
1258d698cf03SStefano Zampini 
1259d698cf03SStefano Zampini   PetscFunctionBegin;
1260d698cf03SStefano Zampini   PetscCheck(bdX == DM_BOUNDARY_NONE || bdX == DM_BOUNDARY_PERIODIC, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Not implemented for boundary type %s", DMBoundaryTypes[bdX]);
1261d698cf03SStefano Zampini   PetscCheck(bdY == DM_BOUNDARY_NONE || bdY == DM_BOUNDARY_PERIODIC, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Not implemented for boundary type %s", DMBoundaryTypes[bdY]);
1262d698cf03SStefano Zampini   PetscCall(DMSetDimension(dm, 2));
1263d698cf03SStefano Zampini   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1264d698cf03SStefano Zampini   PetscCall(DMCreateLabel(dm, "marker"));
1265d698cf03SStefano Zampini   PetscCall(DMCreateLabel(dm, "Face Sets"));
1266d698cf03SStefano Zampini   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
1267d698cf03SStefano Zampini   if (markerSeparate) {
1268d698cf03SStefano Zampini     markerBottom = faceMarkerBottom;
1269d698cf03SStefano Zampini     markerTop    = faceMarkerTop;
1270d698cf03SStefano Zampini     markerRight  = faceMarkerRight;
1271d698cf03SStefano Zampini     markerLeft   = faceMarkerLeft;
1272d698cf03SStefano Zampini   }
1273d698cf03SStefano Zampini   {
1274d698cf03SStefano Zampini     const PetscInt numXEdges    = rank == 0 ? edges[0] : 0;
1275d698cf03SStefano Zampini     const PetscInt numYEdges    = rank == 0 ? edges[1] : 0;
1276d698cf03SStefano Zampini     const PetscInt numZEdges    = rank == 0 ? 4 * edges[0] * edges[1] : 0; /* Z-edges are the 4 internal edges per cell */
1277d698cf03SStefano Zampini     const PetscInt numXVertices = rank == 0 ? (bdX == DM_BOUNDARY_PERIODIC ? edges[0] : edges[0] + 1) : 0;
1278d698cf03SStefano Zampini     const PetscInt numYVertices = rank == 0 ? (bdY == DM_BOUNDARY_PERIODIC ? edges[1] : edges[1] + 1) : 0;
1279d698cf03SStefano Zampini     const PetscInt numZVertices = rank == 0 ? edges[0] * edges[1] : 0;
1280d698cf03SStefano Zampini     const PetscInt numCells     = 4 * numXEdges * numYEdges;
1281d698cf03SStefano Zampini     const PetscInt numTotXEdges = numXEdges * numYVertices;
1282d698cf03SStefano Zampini     const PetscInt numTotYEdges = numYEdges * numXVertices;
1283d698cf03SStefano Zampini     const PetscInt numVertices  = numXVertices * numYVertices + numZVertices;
1284d698cf03SStefano Zampini     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numZEdges;
1285d698cf03SStefano Zampini     const PetscInt firstVertex  = numCells;
1286d698cf03SStefano Zampini     const PetscInt firstXEdge   = numCells + numVertices;
1287d698cf03SStefano Zampini     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
1288d698cf03SStefano Zampini     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
1289d698cf03SStefano Zampini     Vec            coordinates;
1290d698cf03SStefano Zampini     PetscSection   coordSection;
1291d698cf03SStefano Zampini     PetscScalar   *coords;
1292d698cf03SStefano Zampini     PetscInt       coordSize;
1293d698cf03SStefano Zampini     PetscInt       v, vx, vy;
1294d698cf03SStefano Zampini     PetscInt       c, e, ex, ey;
1295d698cf03SStefano Zampini 
1296d698cf03SStefano Zampini     PetscCall(DMPlexSetChart(dm, 0, numCells + numEdges + numVertices));
1297d698cf03SStefano Zampini     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 3));
1298d698cf03SStefano Zampini     for (e = firstXEdge; e < firstXEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
1299d698cf03SStefano Zampini     PetscCall(DMSetUp(dm));
1300d698cf03SStefano Zampini 
1301d698cf03SStefano Zampini     /* Build cells and Z-edges */
1302d698cf03SStefano Zampini     for (ey = 0; ey < numYEdges; ++ey) {
1303d698cf03SStefano Zampini       for (ex = 0; ex < numXEdges; ++ex) {
1304d698cf03SStefano Zampini         const PetscInt exp1 = (ex + 1) % numXVertices;
1305d698cf03SStefano Zampini         const PetscInt eyp1 = (ey + 1) % numYVertices;
1306d698cf03SStefano Zampini         const PetscInt ez   = firstZEdge + 4 * (ey * numXEdges + ex);
1307d698cf03SStefano Zampini         const PetscInt vc   = firstVertex + numXVertices * numYVertices + ey * numXEdges + ex;
1308d698cf03SStefano Zampini         const PetscInt v0   = firstVertex + ey * numXVertices + ex;
1309d698cf03SStefano Zampini         const PetscInt v1   = firstVertex + ey * numXVertices + exp1;
1310d698cf03SStefano Zampini         const PetscInt v2   = firstVertex + eyp1 * numXVertices + exp1;
1311d698cf03SStefano Zampini         const PetscInt v3   = firstVertex + eyp1 * numXVertices + ex;
1312d698cf03SStefano Zampini         const PetscInt e0   = firstXEdge + ey * numXEdges + ex;
1313d698cf03SStefano Zampini         const PetscInt e1   = firstYEdge + exp1 * numYEdges + ey;
1314d698cf03SStefano Zampini         const PetscInt e2   = firstXEdge + eyp1 * numXEdges + ex;
1315d698cf03SStefano Zampini         const PetscInt e3   = firstYEdge + ex * numYEdges + ey;
1316d698cf03SStefano Zampini 
1317d698cf03SStefano Zampini         const PetscInt cones[] = {ez, e0, ez + 1, ez + 1, e1, ez + 2, ez + 2, e2, ez + 3, ez + 3, e3, ez};
1318d698cf03SStefano Zampini         const PetscInt ornts[] = {-1, 0, 0, -1, 0, 0, -1, -1, 0, -1, -1, 0};
1319d698cf03SStefano Zampini         const PetscInt verts[] = {v0, vc, v1, vc, v2, vc, v3, vc};
1320d698cf03SStefano Zampini 
1321d698cf03SStefano Zampini         for (c = 0; c < 4; c++) {
1322d698cf03SStefano Zampini           PetscInt cell = 4 * (ey * numXEdges + ex) + c;
1323d698cf03SStefano Zampini           PetscInt edge = ez + c;
1324d698cf03SStefano Zampini 
1325d698cf03SStefano Zampini           PetscCall(DMPlexSetCone(dm, cell, cones + 3 * c));
1326d698cf03SStefano Zampini           PetscCall(DMPlexSetConeOrientation(dm, cell, ornts + 3 * c));
1327d698cf03SStefano Zampini           PetscCall(DMPlexSetCone(dm, edge, verts + 2 * c));
1328d698cf03SStefano Zampini         }
1329d698cf03SStefano Zampini       }
1330d698cf03SStefano Zampini     }
1331d698cf03SStefano Zampini 
1332d698cf03SStefano Zampini     /* Build Y edges*/
1333d698cf03SStefano Zampini     for (vx = 0; vx < numXVertices; vx++) {
1334d698cf03SStefano Zampini       for (ey = 0; ey < numYEdges; ey++) {
1335d698cf03SStefano Zampini         const PetscInt edge   = firstYEdge + vx * numYEdges + ey;
1336d698cf03SStefano Zampini         const PetscInt v0     = firstVertex + ey * numXVertices + vx;
1337d698cf03SStefano Zampini         const PetscInt v1     = firstVertex + ((ey + 1) % numYVertices) * numXVertices + vx;
1338d698cf03SStefano Zampini         const PetscInt cone[] = {v0, v1};
1339d698cf03SStefano Zampini 
1340d698cf03SStefano Zampini         PetscCall(DMPlexSetCone(dm, edge, cone));
1341d698cf03SStefano Zampini         if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
1342d698cf03SStefano Zampini           if (vx == numXVertices - 1) {
1343d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight));
1344d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1345d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1346d698cf03SStefano Zampini             if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1347d698cf03SStefano Zampini           } else if (vx == 0) {
1348d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft));
1349d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1350d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1351d698cf03SStefano Zampini             if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
1352d698cf03SStefano Zampini           }
1353d698cf03SStefano Zampini         }
1354d698cf03SStefano Zampini       }
1355d698cf03SStefano Zampini     }
1356d698cf03SStefano Zampini 
1357d698cf03SStefano Zampini     /* Build X edges*/
1358d698cf03SStefano Zampini     for (vy = 0; vy < numYVertices; vy++) {
1359d698cf03SStefano Zampini       for (ex = 0; ex < numXEdges; ex++) {
1360d698cf03SStefano Zampini         const PetscInt edge   = firstXEdge + vy * numXEdges + ex;
1361d698cf03SStefano Zampini         const PetscInt v0     = firstVertex + vy * numXVertices + ex;
1362d698cf03SStefano Zampini         const PetscInt v1     = firstVertex + vy * numXVertices + (ex + 1) % numXVertices;
1363d698cf03SStefano Zampini         const PetscInt cone[] = {v0, v1};
1364d698cf03SStefano Zampini 
1365d698cf03SStefano Zampini         PetscCall(DMPlexSetCone(dm, edge, cone));
1366d698cf03SStefano Zampini         if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
1367d698cf03SStefano Zampini           if (vy == numYVertices - 1) {
1368d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop));
1369d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1370d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1371d698cf03SStefano Zampini             if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1372d698cf03SStefano Zampini           } else if (vy == 0) {
1373d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom));
1374d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1375d698cf03SStefano Zampini             PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1376d698cf03SStefano Zampini             if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
1377d698cf03SStefano Zampini           }
1378d698cf03SStefano Zampini         }
1379d698cf03SStefano Zampini       }
1380d698cf03SStefano Zampini     }
1381d698cf03SStefano Zampini 
1382d698cf03SStefano Zampini     /* Compute support, stratify, and celltype label */
1383d698cf03SStefano Zampini     PetscCall(DMPlexSymmetrize(dm));
1384d698cf03SStefano Zampini     PetscCall(DMPlexStratify(dm));
1385d698cf03SStefano Zampini     PetscCall(DMPlexComputeCellTypes(dm));
1386d698cf03SStefano Zampini 
1387d698cf03SStefano Zampini     /* Build coordinates */
1388d698cf03SStefano Zampini     PetscCall(DMGetCoordinateSection(dm, &coordSection));
1389d698cf03SStefano Zampini     PetscCall(PetscSectionSetNumFields(coordSection, 1));
1390d698cf03SStefano Zampini     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, 2));
1391d698cf03SStefano Zampini     PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVertices));
1392d698cf03SStefano Zampini     for (v = firstVertex; v < firstVertex + numVertices; ++v) {
1393d698cf03SStefano Zampini       PetscCall(PetscSectionSetDof(coordSection, v, 2));
1394d698cf03SStefano Zampini       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, 2));
1395d698cf03SStefano Zampini     }
1396d698cf03SStefano Zampini     PetscCall(PetscSectionSetUp(coordSection));
1397d698cf03SStefano Zampini     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
1398d698cf03SStefano Zampini     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
1399d698cf03SStefano Zampini     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
1400d698cf03SStefano Zampini     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
1401d698cf03SStefano Zampini     PetscCall(VecSetBlockSize(coordinates, 2));
1402d698cf03SStefano Zampini     PetscCall(VecSetType(coordinates, VECSTANDARD));
1403d698cf03SStefano Zampini     PetscCall(VecGetArray(coordinates, &coords));
1404d698cf03SStefano Zampini     for (vy = 0; vy < numYVertices; ++vy) {
1405d698cf03SStefano Zampini       for (vx = 0; vx < numXVertices; ++vx) {
1406d698cf03SStefano Zampini         coords[2 * (vy * numXVertices + vx) + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * vx;
1407d698cf03SStefano Zampini         coords[2 * (vy * numXVertices + vx) + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * vy;
1408d698cf03SStefano Zampini       }
1409d698cf03SStefano Zampini     }
1410d698cf03SStefano Zampini     for (ey = 0; ey < numYEdges; ++ey) {
1411d698cf03SStefano Zampini       for (ex = 0; ex < numXEdges; ++ex) {
1412d698cf03SStefano Zampini         const PetscInt c = ey * numXEdges + ex + numYVertices * numXVertices;
1413d698cf03SStefano Zampini 
1414d698cf03SStefano Zampini         coords[2 * c + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * (ex + 0.5);
1415d698cf03SStefano Zampini         coords[2 * c + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * (ey + 0.5);
1416d698cf03SStefano Zampini       }
1417d698cf03SStefano Zampini     }
1418d698cf03SStefano Zampini     PetscCall(VecRestoreArray(coordinates, &coords));
1419d698cf03SStefano Zampini     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
1420d698cf03SStefano Zampini     PetscCall(VecDestroy(&coordinates));
1421d698cf03SStefano Zampini 
1422d698cf03SStefano Zampini     /* handle periodic BC */
1423d698cf03SStefano Zampini     if (bdX == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_PERIODIC) {
1424d698cf03SStefano Zampini       PetscReal L[2]       = {-1., -1.};
1425d698cf03SStefano Zampini       PetscReal maxCell[2] = {-1., -1.};
1426d698cf03SStefano Zampini 
1427d698cf03SStefano Zampini       for (PetscInt d = 0; d < 2; ++d) {
1428d698cf03SStefano Zampini         if (bd[d] != DM_BOUNDARY_NONE) {
1429d698cf03SStefano Zampini           L[d]       = upper[d] - lower[d];
1430d698cf03SStefano Zampini           maxCell[d] = 1.1 * (L[d] / PetscMax(1, edges[d]));
1431d698cf03SStefano Zampini         }
1432d698cf03SStefano Zampini       }
1433d698cf03SStefano Zampini       PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
1434d698cf03SStefano Zampini     }
1435d698cf03SStefano Zampini   }
1436d698cf03SStefano Zampini   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
1437d698cf03SStefano Zampini   PetscFunctionReturn(PETSC_SUCCESS);
1438d698cf03SStefano Zampini }
1439d698cf03SStefano Zampini 
1440d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Simplex_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool interpolate)
1441d71ae5a4SJacob Faibussowitsch {
14429318fe57SMatthew G. Knepley   DM        boundary, vol;
1443c22d3578SMatthew G. Knepley   DMLabel   bdlabel;
1444d698cf03SStefano Zampini   PetscBool crisscross = PETSC_FALSE;
1445d6218766SMatthew G. Knepley 
1446d6218766SMatthew G. Knepley   PetscFunctionBegin;
14474f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
1448d698cf03SStefano Zampini   if (dim == 2) PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_box_crisscross", &crisscross, NULL));
1449d698cf03SStefano Zampini   if (crisscross) {
1450d698cf03SStefano Zampini     PetscCall(DMPlexCreateSquareMesh_Simplex_CrissCross(dm, faces, lower, upper, periodicity));
1451d698cf03SStefano Zampini   } else {
1452c22d3578SMatthew 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");
14539566063dSJacob Faibussowitsch     PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &boundary));
14549566063dSJacob Faibussowitsch     PetscCall(DMSetType(boundary, DMPLEX));
1455d698cf03SStefano Zampini     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)boundary, ((PetscObject)dm)->prefix));
14569566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(boundary, dim, faces, lower, upper, PETSC_FALSE));
14579566063dSJacob Faibussowitsch     PetscCall(DMPlexGenerate(boundary, NULL, interpolate, &vol));
1458c22d3578SMatthew G. Knepley     PetscCall(DMGetLabel(vol, "marker", &bdlabel));
1459c22d3578SMatthew G. Knepley     if (bdlabel) PetscCall(DMPlexLabelComplete(vol, bdlabel));
14605de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_TRUE, PETSC_FALSE, vol));
146169d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &vol));
1462d698cf03SStefano Zampini     PetscCall(DMDestroy(&boundary));
1463d698cf03SStefano Zampini   }
14644054ae39SJames Wright   if (interpolate) {
14654054ae39SJames Wright     PetscCall(DMPlexInterpolateInPlace_Internal(dm));
1466d7d2d1d2SJames Wright     PetscCall(DMPlexSetBoxLabel_Internal(dm, periodicity));
14674054ae39SJames Wright   }
14683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1469d6218766SMatthew G. Knepley }
1470d6218766SMatthew G. Knepley 
1471d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCubeMesh_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY, DMBoundaryType bdZ)
1472d71ae5a4SJacob Faibussowitsch {
1473ed0e4b50SMatthew G. Knepley   DMLabel     cutLabel  = NULL;
1474f4eb4c5dSMatthew G. Knepley   PetscInt    markerTop = 1, faceMarkerTop = 1;
1475f4eb4c5dSMatthew G. Knepley   PetscInt    markerBottom = 1, faceMarkerBottom = 1;
1476f4eb4c5dSMatthew G. Knepley   PetscInt    markerFront = 1, faceMarkerFront = 1;
1477f4eb4c5dSMatthew G. Knepley   PetscInt    markerBack = 1, faceMarkerBack = 1;
1478f4eb4c5dSMatthew G. Knepley   PetscInt    markerRight = 1, faceMarkerRight = 1;
1479f4eb4c5dSMatthew G. Knepley   PetscInt    markerLeft = 1, faceMarkerLeft = 1;
14803dfda0b1SToby Isaac   PetscInt    dim;
1481d8211ee3SMatthew G. Knepley   PetscBool   markerSeparate = PETSC_FALSE, cutMarker = PETSC_FALSE;
14823dfda0b1SToby Isaac   PetscMPIInt rank;
14833dfda0b1SToby Isaac 
14843dfda0b1SToby Isaac   PetscFunctionBegin;
14859566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
14869566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
14879566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
14889566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
14899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
14909371c9d4SSatish 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) {
14919371c9d4SSatish Balay     if (cutMarker) {
14929371c9d4SSatish Balay       PetscCall(DMCreateLabel(dm, "periodic_cut"));
14939371c9d4SSatish Balay       PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
14949371c9d4SSatish Balay     }
1495d8211ee3SMatthew G. Knepley   }
14963dfda0b1SToby Isaac   switch (dim) {
14973dfda0b1SToby Isaac   case 2:
1498f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 3;
1499f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1500f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 2;
1501f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 4;
15023dfda0b1SToby Isaac     break;
15033dfda0b1SToby Isaac   case 3:
1504f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
1505f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 2;
1506f4eb4c5dSMatthew G. Knepley     faceMarkerFront  = 3;
1507f4eb4c5dSMatthew G. Knepley     faceMarkerBack   = 4;
1508f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 5;
1509f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 6;
15103dfda0b1SToby Isaac     break;
1511d71ae5a4SJacob Faibussowitsch   default:
1512d71ae5a4SJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Dimension %" PetscInt_FMT " not supported", dim);
15133dfda0b1SToby Isaac   }
15149566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL));
1515f4eb4c5dSMatthew G. Knepley   if (markerSeparate) {
1516f4eb4c5dSMatthew G. Knepley     markerBottom = faceMarkerBottom;
1517f4eb4c5dSMatthew G. Knepley     markerTop    = faceMarkerTop;
1518f4eb4c5dSMatthew G. Knepley     markerFront  = faceMarkerFront;
1519f4eb4c5dSMatthew G. Knepley     markerBack   = faceMarkerBack;
1520f4eb4c5dSMatthew G. Knepley     markerRight  = faceMarkerRight;
1521f4eb4c5dSMatthew G. Knepley     markerLeft   = faceMarkerLeft;
15223dfda0b1SToby Isaac   }
15233dfda0b1SToby Isaac   {
1524dd400576SPatrick Sanan     const PetscInt numXEdges    = rank == 0 ? edges[0] : 0;
1525dd400576SPatrick Sanan     const PetscInt numYEdges    = rank == 0 ? edges[1] : 0;
1526dd400576SPatrick Sanan     const PetscInt numZEdges    = rank == 0 ? edges[2] : 0;
1527dd400576SPatrick Sanan     const PetscInt numXVertices = rank == 0 ? (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST ? edges[0] : edges[0] + 1) : 0;
1528dd400576SPatrick Sanan     const PetscInt numYVertices = rank == 0 ? (bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[1] : edges[1] + 1) : 0;
1529dd400576SPatrick Sanan     const PetscInt numZVertices = rank == 0 ? (bdZ == DM_BOUNDARY_PERIODIC || bdZ == DM_BOUNDARY_TWIST ? edges[2] : edges[2] + 1) : 0;
15303dfda0b1SToby Isaac     const PetscInt numCells     = numXEdges * numYEdges * numZEdges;
15313dfda0b1SToby Isaac     const PetscInt numXFaces    = numYEdges * numZEdges;
15323dfda0b1SToby Isaac     const PetscInt numYFaces    = numXEdges * numZEdges;
15333dfda0b1SToby Isaac     const PetscInt numZFaces    = numXEdges * numYEdges;
15343dfda0b1SToby Isaac     const PetscInt numTotXFaces = numXVertices * numXFaces;
15353dfda0b1SToby Isaac     const PetscInt numTotYFaces = numYVertices * numYFaces;
15363dfda0b1SToby Isaac     const PetscInt numTotZFaces = numZVertices * numZFaces;
15373dfda0b1SToby Isaac     const PetscInt numFaces     = numTotXFaces + numTotYFaces + numTotZFaces;
15383dfda0b1SToby Isaac     const PetscInt numTotXEdges = numXEdges * numYVertices * numZVertices;
15393dfda0b1SToby Isaac     const PetscInt numTotYEdges = numYEdges * numXVertices * numZVertices;
15403dfda0b1SToby Isaac     const PetscInt numTotZEdges = numZEdges * numXVertices * numYVertices;
15413dfda0b1SToby Isaac     const PetscInt numVertices  = numXVertices * numYVertices * numZVertices;
15423dfda0b1SToby Isaac     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numTotZEdges;
15433dfda0b1SToby Isaac     const PetscInt firstVertex  = (dim == 2) ? numFaces : numCells;
15443dfda0b1SToby Isaac     const PetscInt firstXFace   = (dim == 2) ? 0 : numCells + numVertices;
15453dfda0b1SToby Isaac     const PetscInt firstYFace   = firstXFace + numTotXFaces;
15463dfda0b1SToby Isaac     const PetscInt firstZFace   = firstYFace + numTotYFaces;
15473dfda0b1SToby Isaac     const PetscInt firstXEdge   = numCells + numFaces + numVertices;
15483dfda0b1SToby Isaac     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
15493dfda0b1SToby Isaac     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
15503dfda0b1SToby Isaac     Vec            coordinates;
15513dfda0b1SToby Isaac     PetscSection   coordSection;
15523dfda0b1SToby Isaac     PetscScalar   *coords;
15533dfda0b1SToby Isaac     PetscInt       coordSize;
15543dfda0b1SToby Isaac     PetscInt       v, vx, vy, vz;
15553dfda0b1SToby Isaac     PetscInt       c, f, fx, fy, fz, e, ex, ey, ez;
15563dfda0b1SToby Isaac 
15579566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numFaces + numEdges + numVertices));
155848a46eb9SPierre Jolivet     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
155948a46eb9SPierre Jolivet     for (f = firstXFace; f < firstXFace + numFaces; ++f) PetscCall(DMPlexSetConeSize(dm, f, 4));
156048a46eb9SPierre Jolivet     for (e = firstXEdge; e < firstXEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
15619566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm)); /* Allocate space for cones */
15623dfda0b1SToby Isaac     /* Build cells */
15633dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
15643dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
15653dfda0b1SToby Isaac         for (fx = 0; fx < numXEdges; ++fx) {
15663dfda0b1SToby Isaac           PetscInt cell  = (fz * numYEdges + fy) * numXEdges + fx;
15673dfda0b1SToby Isaac           PetscInt faceB = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
15683dfda0b1SToby Isaac           PetscInt faceT = firstZFace + (fy * numXEdges + fx) * numZVertices + ((fz + 1) % numZVertices);
15693dfda0b1SToby Isaac           PetscInt faceF = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
15703dfda0b1SToby Isaac           PetscInt faceK = firstYFace + (fz * numXEdges + fx) * numYVertices + ((fy + 1) % numYVertices);
15713dfda0b1SToby Isaac           PetscInt faceL = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
15723dfda0b1SToby Isaac           PetscInt faceR = firstXFace + (fz * numYEdges + fy) * numXVertices + ((fx + 1) % numXVertices);
15733dfda0b1SToby Isaac           /* B,  T,  F,  K,  R,  L */
1574b5a892a1SMatthew G. Knepley           PetscInt ornt[6] = {-2, 0, 0, -3, 0, -2}; /* ??? */
157542206facSLisandro Dalcin           PetscInt cone[6];
15763dfda0b1SToby Isaac 
15773dfda0b1SToby Isaac           /* no boundary twisting in 3D */
15789371c9d4SSatish Balay           cone[0] = faceB;
15799371c9d4SSatish Balay           cone[1] = faceT;
15809371c9d4SSatish Balay           cone[2] = faceF;
15819371c9d4SSatish Balay           cone[3] = faceK;
15829371c9d4SSatish Balay           cone[4] = faceR;
15839371c9d4SSatish Balay           cone[5] = faceL;
15849566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, cell, cone));
15859566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, cell, ornt));
15869566063dSJacob Faibussowitsch           if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
15879566063dSJacob Faibussowitsch           if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
15889566063dSJacob Faibussowitsch           if (bdZ != DM_BOUNDARY_NONE && fz == numZEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, cell, 2));
15893dfda0b1SToby Isaac         }
15903dfda0b1SToby Isaac       }
15913dfda0b1SToby Isaac     }
15923dfda0b1SToby Isaac     /* Build x faces */
15933dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
15943dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
15953dfda0b1SToby Isaac         for (fx = 0; fx < numXVertices; ++fx) {
15963dfda0b1SToby Isaac           PetscInt face    = firstXFace + (fz * numYEdges + fy) * numXVertices + fx;
15973dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
15983dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (((fy + 1) % numYVertices) * numXVertices + fx) * numZEdges + fz;
15993dfda0b1SToby Isaac           PetscInt edgeB   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
16003dfda0b1SToby Isaac           PetscInt edgeT   = firstYEdge + (((fz + 1) % numZVertices) * numXVertices + fx) * numYEdges + fy;
1601b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
16023dfda0b1SToby Isaac           PetscInt cone[4];
16033dfda0b1SToby Isaac 
16043dfda0b1SToby Isaac           if (dim == 3) {
16053dfda0b1SToby Isaac             /* markers */
16063dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
16073dfda0b1SToby Isaac               if (fx == numXVertices - 1) {
16089566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerRight));
16099566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerRight));
16109371c9d4SSatish Balay               } else if (fx == 0) {
16119566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerLeft));
16129566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerLeft));
16133dfda0b1SToby Isaac               }
16143dfda0b1SToby Isaac             }
16153dfda0b1SToby Isaac           }
16169371c9d4SSatish Balay           cone[0] = edgeB;
16179371c9d4SSatish Balay           cone[1] = edgeR;
16189371c9d4SSatish Balay           cone[2] = edgeT;
16199371c9d4SSatish Balay           cone[3] = edgeL;
16209566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
16219566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
16223dfda0b1SToby Isaac         }
16233dfda0b1SToby Isaac       }
16243dfda0b1SToby Isaac     }
16253dfda0b1SToby Isaac     /* Build y faces */
16263dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
162742206facSLisandro Dalcin       for (fx = 0; fx < numXEdges; ++fx) {
16283dfda0b1SToby Isaac         for (fy = 0; fy < numYVertices; ++fy) {
16293dfda0b1SToby Isaac           PetscInt face    = firstYFace + (fz * numXEdges + fx) * numYVertices + fy;
16303dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy * numXVertices + fx) * numZEdges + fz;
16313dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (fy * numXVertices + ((fx + 1) % numXVertices)) * numZEdges + fz;
16323dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
16333dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (((fz + 1) % numZVertices) * numYVertices + fy) * numXEdges + fx;
1634b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
16353dfda0b1SToby Isaac           PetscInt cone[4];
16363dfda0b1SToby Isaac 
16373dfda0b1SToby Isaac           if (dim == 3) {
16383dfda0b1SToby Isaac             /* markers */
16393dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
16403dfda0b1SToby Isaac               if (fy == numYVertices - 1) {
16419566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBack));
16429566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBack));
16439371c9d4SSatish Balay               } else if (fy == 0) {
16449566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerFront));
16459566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerFront));
16463dfda0b1SToby Isaac               }
16473dfda0b1SToby Isaac             }
16483dfda0b1SToby Isaac           }
16499371c9d4SSatish Balay           cone[0] = edgeB;
16509371c9d4SSatish Balay           cone[1] = edgeR;
16519371c9d4SSatish Balay           cone[2] = edgeT;
16529371c9d4SSatish Balay           cone[3] = edgeL;
16539566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
16549566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
16553dfda0b1SToby Isaac         }
16563dfda0b1SToby Isaac       }
16573dfda0b1SToby Isaac     }
16583dfda0b1SToby Isaac     /* Build z faces */
16593dfda0b1SToby Isaac     for (fy = 0; fy < numYEdges; ++fy) {
16603dfda0b1SToby Isaac       for (fx = 0; fx < numXEdges; ++fx) {
16613dfda0b1SToby Isaac         for (fz = 0; fz < numZVertices; fz++) {
16623dfda0b1SToby Isaac           PetscInt face    = firstZFace + (fy * numXEdges + fx) * numZVertices + fz;
16633dfda0b1SToby Isaac           PetscInt edgeL   = firstYEdge + (fz * numXVertices + fx) * numYEdges + fy;
16643dfda0b1SToby Isaac           PetscInt edgeR   = firstYEdge + (fz * numXVertices + ((fx + 1) % numXVertices)) * numYEdges + fy;
16653dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz * numYVertices + fy) * numXEdges + fx;
16663dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (fz * numYVertices + ((fy + 1) % numYVertices)) * numXEdges + fx;
1667b5a892a1SMatthew G. Knepley           PetscInt ornt[4] = {0, 0, -1, -1};
16683dfda0b1SToby Isaac           PetscInt cone[4];
16693dfda0b1SToby Isaac 
16703dfda0b1SToby Isaac           if (dim == 2) {
16719371c9d4SSatish Balay             if (bdX == DM_BOUNDARY_TWIST && fx == numXEdges - 1) {
16729371c9d4SSatish Balay               edgeR += numYEdges - 1 - 2 * fy;
16739371c9d4SSatish Balay               ornt[1] = -1;
16749371c9d4SSatish Balay             }
16759371c9d4SSatish Balay             if (bdY == DM_BOUNDARY_TWIST && fy == numYEdges - 1) {
16769371c9d4SSatish Balay               edgeT += numXEdges - 1 - 2 * fx;
16779371c9d4SSatish Balay               ornt[2] = 0;
16789371c9d4SSatish Balay             }
16799566063dSJacob Faibussowitsch             if (bdX != DM_BOUNDARY_NONE && fx == numXEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
16809566063dSJacob Faibussowitsch             if (bdY != DM_BOUNDARY_NONE && fy == numYEdges - 1 && cutLabel) PetscCall(DMLabelSetValue(cutLabel, face, 2));
1681d1c88043SMatthew G. Knepley           } else {
16823dfda0b1SToby Isaac             /* markers */
16833dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
16843dfda0b1SToby Isaac               if (fz == numZVertices - 1) {
16859566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerTop));
16869566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerTop));
16879371c9d4SSatish Balay               } else if (fz == 0) {
16889566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", face, faceMarkerBottom));
16899566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", face, markerBottom));
16903dfda0b1SToby Isaac               }
16913dfda0b1SToby Isaac             }
16923dfda0b1SToby Isaac           }
16939371c9d4SSatish Balay           cone[0] = edgeB;
16949371c9d4SSatish Balay           cone[1] = edgeR;
16959371c9d4SSatish Balay           cone[2] = edgeT;
16969371c9d4SSatish Balay           cone[3] = edgeL;
16979566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, face, cone));
16989566063dSJacob Faibussowitsch           PetscCall(DMPlexSetConeOrientation(dm, face, ornt));
16993dfda0b1SToby Isaac         }
17003dfda0b1SToby Isaac       }
17013dfda0b1SToby Isaac     }
17023dfda0b1SToby Isaac     /* Build Z edges*/
17033dfda0b1SToby Isaac     for (vy = 0; vy < numYVertices; vy++) {
17043dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
17053dfda0b1SToby Isaac         for (ez = 0; ez < numZEdges; ez++) {
17063dfda0b1SToby Isaac           const PetscInt edge    = firstZEdge + (vy * numXVertices + vx) * numZEdges + ez;
17073dfda0b1SToby Isaac           const PetscInt vertexB = firstVertex + (ez * numYVertices + vy) * numXVertices + vx;
17083dfda0b1SToby Isaac           const PetscInt vertexT = firstVertex + (((ez + 1) % numZVertices) * numYVertices + vy) * numXVertices + vx;
17093dfda0b1SToby Isaac           PetscInt       cone[2];
17103dfda0b1SToby Isaac 
17119371c9d4SSatish Balay           cone[0] = vertexB;
17129371c9d4SSatish Balay           cone[1] = vertexT;
1713c2df9bbfSMatthew G. Knepley           PetscCall(DMPlexSetCone(dm, edge, cone));
17143dfda0b1SToby Isaac           if (dim == 3) {
17153dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
17163dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
17179566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1718c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1719c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1720c2df9bbfSMatthew G. Knepley               } else if (vx == 0) {
17219566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1722c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1723c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
17243dfda0b1SToby Isaac               }
17253dfda0b1SToby Isaac             }
17263dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
17273dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
17289566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1729c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1730c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1731c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
17329566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1733c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1734c2df9bbfSMatthew G. Knepley                 if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
17353dfda0b1SToby Isaac               }
17363dfda0b1SToby Isaac             }
17373dfda0b1SToby Isaac           }
17383dfda0b1SToby Isaac         }
17393dfda0b1SToby Isaac       }
17403dfda0b1SToby Isaac     }
17413dfda0b1SToby Isaac     /* Build Y edges*/
17423dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
17433dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
17443dfda0b1SToby Isaac         for (ey = 0; ey < numYEdges; ey++) {
17453dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdY == DM_BOUNDARY_TWIST && ey == numYEdges - 1) ? (numXVertices - vx - 1) : (vz * numYVertices + ((ey + 1) % numYVertices)) * numXVertices + vx;
17463dfda0b1SToby Isaac           const PetscInt edge    = firstYEdge + (vz * numXVertices + vx) * numYEdges + ey;
17473dfda0b1SToby Isaac           const PetscInt vertexF = firstVertex + (vz * numYVertices + ey) * numXVertices + vx;
17483dfda0b1SToby Isaac           const PetscInt vertexK = firstVertex + nextv;
17493dfda0b1SToby Isaac           PetscInt       cone[2];
17503dfda0b1SToby Isaac 
17519371c9d4SSatish Balay           cone[0] = vertexF;
17529371c9d4SSatish Balay           cone[1] = vertexK;
17539566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
17543dfda0b1SToby Isaac           if (dim == 2) {
17553dfda0b1SToby Isaac             if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
17563dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
17579566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight));
17589566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
17599566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1760c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1761d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
17629566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft));
17639566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
17649566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1765c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
17663dfda0b1SToby Isaac               }
1767d8211ee3SMatthew G. Knepley             } else {
17684c67ea77SStefano Zampini               if (vx == 0 && cutLabel) {
17699566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
17709566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1771c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
17723dfda0b1SToby Isaac               }
1773d8211ee3SMatthew G. Knepley             }
1774d8211ee3SMatthew G. Knepley           } else {
17753dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
17763dfda0b1SToby Isaac               if (vx == numXVertices - 1) {
17779566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerRight));
1778c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerRight));
1779c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight));
1780d8211ee3SMatthew G. Knepley               } else if (vx == 0) {
17819566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerLeft));
1782c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerLeft));
1783c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft));
17843dfda0b1SToby Isaac               }
17853dfda0b1SToby Isaac             }
17863dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
17873dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
17889566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1789c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1790c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1791d8211ee3SMatthew G. Knepley               } else if (vz == 0) {
17929566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1793c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1794c2df9bbfSMatthew G. Knepley                 if (ey == numYEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
17953dfda0b1SToby Isaac               }
17963dfda0b1SToby Isaac             }
17973dfda0b1SToby Isaac           }
17983dfda0b1SToby Isaac         }
17993dfda0b1SToby Isaac       }
18003dfda0b1SToby Isaac     }
18013dfda0b1SToby Isaac     /* Build X edges*/
18023dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
18033dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; vy++) {
18043dfda0b1SToby Isaac         for (ex = 0; ex < numXEdges; ex++) {
18053dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdX == DM_BOUNDARY_TWIST && ex == numXEdges - 1) ? (numYVertices - vy - 1) * numXVertices : (vz * numYVertices + vy) * numXVertices + (ex + 1) % numXVertices;
18063dfda0b1SToby Isaac           const PetscInt edge    = firstXEdge + (vz * numYVertices + vy) * numXEdges + ex;
18073dfda0b1SToby Isaac           const PetscInt vertexL = firstVertex + (vz * numYVertices + vy) * numXVertices + ex;
18083dfda0b1SToby Isaac           const PetscInt vertexR = firstVertex + nextv;
18093dfda0b1SToby Isaac           PetscInt       cone[2];
18103dfda0b1SToby Isaac 
18119371c9d4SSatish Balay           cone[0] = vertexL;
18129371c9d4SSatish Balay           cone[1] = vertexR;
18139566063dSJacob Faibussowitsch           PetscCall(DMPlexSetCone(dm, edge, cone));
18143dfda0b1SToby Isaac           if (dim == 2) {
18153dfda0b1SToby Isaac             if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
18163dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
18179566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop));
18189566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
18199566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1820c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1821d8211ee3SMatthew G. Knepley               } else if (vy == 0) {
18229566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom));
18239566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
18249566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1825c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
18263dfda0b1SToby Isaac               }
1827d8211ee3SMatthew G. Knepley             } else {
18284c67ea77SStefano Zampini               if (vy == 0 && cutLabel) {
18299566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, edge, 1));
18309566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(cutLabel, cone[0], 1));
1831c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMLabelSetValue(cutLabel, cone[1], 1));
18323dfda0b1SToby Isaac               }
1833d8211ee3SMatthew G. Knepley             }
1834d8211ee3SMatthew G. Knepley           } else {
18353dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
18363dfda0b1SToby Isaac               if (vy == numYVertices - 1) {
18379566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBack));
1838c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBack));
1839c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBack));
1840c2df9bbfSMatthew G. Knepley               } else if (vy == 0) {
18419566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerFront));
1842c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerFront));
1843c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerFront));
18443dfda0b1SToby Isaac               }
18453dfda0b1SToby Isaac             }
18463dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
18473dfda0b1SToby Isaac               if (vz == numZVertices - 1) {
18489566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerTop));
1849c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerTop));
1850c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerTop));
1851c2df9bbfSMatthew G. Knepley               } else if (vz == 0) {
18529566063dSJacob Faibussowitsch                 PetscCall(DMSetLabelValue(dm, "marker", edge, markerBottom));
1853c2df9bbfSMatthew G. Knepley                 PetscCall(DMSetLabelValue(dm, "marker", cone[0], markerBottom));
1854c2df9bbfSMatthew G. Knepley                 if (ex == numXEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerBottom));
18553dfda0b1SToby Isaac               }
18563dfda0b1SToby Isaac             }
18573dfda0b1SToby Isaac           }
18583dfda0b1SToby Isaac         }
18593dfda0b1SToby Isaac       }
18603dfda0b1SToby Isaac     }
18619566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
18629566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
18633dfda0b1SToby Isaac     /* Build coordinates */
18649566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
18659566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
18669566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
18679566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVertices));
18683dfda0b1SToby Isaac     for (v = firstVertex; v < firstVertex + numVertices; ++v) {
18699566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
18709566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
18713dfda0b1SToby Isaac     }
18729566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
18739566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
18749566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
18759566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
18769566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
18779566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
18789566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
18799566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
18803dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; ++vz) {
18813dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; ++vy) {
18823dfda0b1SToby Isaac         for (vx = 0; vx < numXVertices; ++vx) {
18833dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 0] = lower[0] + ((upper[0] - lower[0]) / numXEdges) * vx;
18843dfda0b1SToby Isaac           coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 1] = lower[1] + ((upper[1] - lower[1]) / numYEdges) * vy;
1885ad540459SPierre Jolivet           if (dim == 3) coords[((vz * numYVertices + vy) * numXVertices + vx) * dim + 2] = lower[2] + ((upper[2] - lower[2]) / numZEdges) * vz;
18863dfda0b1SToby Isaac         }
18873dfda0b1SToby Isaac       }
18883dfda0b1SToby Isaac     }
18899566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
18909566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
18919566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
18923dfda0b1SToby Isaac   }
18933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
18943dfda0b1SToby Isaac }
18953dfda0b1SToby Isaac 
1896d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoxMesh_Tensor_Internal(DM dm, PetscInt dim, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
1897d71ae5a4SJacob Faibussowitsch {
18989318fe57SMatthew G. Knepley   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
18999318fe57SMatthew G. Knepley   PetscInt       fac[3] = {0, 0, 0}, d;
1900552f7358SJed Brown 
1901552f7358SJed Brown   PetscFunctionBegin;
19024f572ea9SToby Isaac   PetscAssertPointer(dm, 1);
19039318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, dim, 2);
19049566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
19059371c9d4SSatish Balay   for (d = 0; d < dim; ++d) {
19069371c9d4SSatish Balay     fac[d] = faces[d];
19079371c9d4SSatish Balay     bdt[d] = periodicity[d];
19089371c9d4SSatish Balay   }
19099566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateCubeMesh_Internal(dm, lower, upper, fac, bdt[0], bdt[1], bdt[2]));
19109371c9d4SSatish 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))) {
19116858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
19126858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
1913552f7358SJed Brown 
19149318fe57SMatthew G. Knepley     for (d = 0; d < dim; ++d) {
19156858538eSMatthew G. Knepley       if (periodicity[d] != DM_BOUNDARY_NONE) {
19169318fe57SMatthew G. Knepley         L[d]       = upper[d] - lower[d];
19179318fe57SMatthew G. Knepley         maxCell[d] = 1.1 * (L[d] / PetscMax(1, faces[d]));
1918768d5fceSMatthew G. Knepley       }
19196858538eSMatthew G. Knepley     }
19204fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
1921768d5fceSMatthew G. Knepley   }
19229566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
19233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
19249318fe57SMatthew G. Knepley }
19259318fe57SMatthew G. Knepley 
19265dca41c3SJed 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)
1927d71ae5a4SJacob Faibussowitsch {
19289318fe57SMatthew G. Knepley   PetscFunctionBegin;
192946139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
19305dca41c3SJed Brown   if (shape == DM_SHAPE_ZBOX) PetscCall(DMPlexCreateBoxMesh_Tensor_SFC_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
19316725e60dSJed Brown   else if (dim == 1) PetscCall(DMPlexCreateLineMesh_Internal(dm, faces[0], lower[0], upper[0], periodicity[0]));
19329566063dSJacob Faibussowitsch   else if (simplex) PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(dm, dim, faces, lower, upper, periodicity, interpolate));
19339566063dSJacob Faibussowitsch   else PetscCall(DMPlexCreateBoxMesh_Tensor_Internal(dm, dim, faces, lower, upper, periodicity));
19349318fe57SMatthew G. Knepley   if (!interpolate && dim > 1 && !simplex) {
1935768d5fceSMatthew G. Knepley     DM udm;
1936768d5fceSMatthew G. Knepley 
19379566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(dm, &udm));
19389566063dSJacob Faibussowitsch     PetscCall(DMPlexCopyCoordinates(dm, udm));
193969d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &udm));
1940768d5fceSMatthew G. Knepley   }
194146139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
19423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1943c8c68bd8SToby Isaac }
1944c8c68bd8SToby Isaac 
19455d83a8b1SBarry Smith /*@
1946768d5fceSMatthew G. Knepley   DMPlexCreateBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using simplices or tensor cells (hexahedra).
1947768d5fceSMatthew G. Knepley 
1948d083f849SBarry Smith   Collective
1949768d5fceSMatthew G. Knepley 
1950768d5fceSMatthew G. Knepley   Input Parameters:
1951a1cb98faSBarry Smith + comm               - The communicator for the `DM` object
1952768d5fceSMatthew G. Knepley . dim                - The spatial dimension
1953a1cb98faSBarry Smith . simplex            - `PETSC_TRUE` for simplices, `PETSC_FALSE` for tensor cells
195420f4b53cSBarry Smith . faces              - Number of faces per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
195520f4b53cSBarry Smith . lower              - The lower left corner, or `NULL` for (0, 0, 0)
195620f4b53cSBarry Smith . upper              - The upper right corner, or `NULL` for (1, 1, 1)
195720f4b53cSBarry Smith . periodicity        - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
195842108689Sksagiyam . interpolate        - Flag to create intermediate mesh pieces (edges, faces)
195942108689Sksagiyam . localizationHeight - Flag to localize edges and faces in addition to cells; only significant for periodic meshes
196042108689Sksagiyam - sparseLocalize     - Flag to localize coordinates only for cells near the periodic boundary; only significant for periodic meshes
1961768d5fceSMatthew G. Knepley 
1962768d5fceSMatthew G. Knepley   Output Parameter:
1963a1cb98faSBarry Smith . dm - The `DM` object
1964768d5fceSMatthew G. Knepley 
1965768d5fceSMatthew G. Knepley   Level: beginner
1966768d5fceSMatthew G. Knepley 
1967a1cb98faSBarry Smith   Note:
1968a1cb98faSBarry Smith   To customize this mesh using options, use
1969a1cb98faSBarry Smith .vb
1970a1cb98faSBarry Smith   DMCreate(comm, &dm);
1971a1cb98faSBarry Smith   DMSetType(dm, DMPLEX);
1972a1cb98faSBarry Smith   DMSetFromOptions(dm);
1973a1cb98faSBarry Smith .ve
1974a1cb98faSBarry Smith   and use the options in `DMSetFromOptions()`.
1975a1cb98faSBarry Smith 
1976a4e35b19SJacob Faibussowitsch   Here is the numbering returned for 2 faces in each direction for tensor cells\:
1977a1cb98faSBarry Smith .vb
1978a1cb98faSBarry Smith  10---17---11---18----12
1979a1cb98faSBarry Smith   |         |         |
1980a1cb98faSBarry Smith   |         |         |
1981a1cb98faSBarry Smith  20    2   22    3    24
1982a1cb98faSBarry Smith   |         |         |
1983a1cb98faSBarry Smith   |         |         |
1984a1cb98faSBarry Smith   7---15----8---16----9
1985a1cb98faSBarry Smith   |         |         |
1986a1cb98faSBarry Smith   |         |         |
1987a1cb98faSBarry Smith  19    0   21    1   23
1988a1cb98faSBarry Smith   |         |         |
1989a1cb98faSBarry Smith   |         |         |
1990a1cb98faSBarry Smith   4---13----5---14----6
1991a1cb98faSBarry Smith .ve
1992a1cb98faSBarry Smith   and for simplicial cells
1993a1cb98faSBarry Smith .vb
1994a1cb98faSBarry Smith  14----8---15----9----16
1995a1cb98faSBarry Smith   |\     5  |\      7 |
1996a1cb98faSBarry Smith   | \       | \       |
1997a1cb98faSBarry Smith  13   2    14    3    15
1998a1cb98faSBarry Smith   | 4   \   | 6   \   |
1999a1cb98faSBarry Smith   |       \ |       \ |
2000a1cb98faSBarry Smith  11----6---12----7----13
2001a1cb98faSBarry Smith   |\        |\        |
2002a1cb98faSBarry Smith   | \    1  | \     3 |
2003a1cb98faSBarry Smith  10   0    11    1    12
2004a1cb98faSBarry Smith   | 0   \   | 2   \   |
2005a1cb98faSBarry Smith   |       \ |       \ |
2006a1cb98faSBarry Smith   8----4----9----5----10
2007a1cb98faSBarry Smith .ve
2008a1cb98faSBarry Smith 
20091cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
2010768d5fceSMatthew G. Knepley @*/
201142108689Sksagiyam 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)
2012d71ae5a4SJacob Faibussowitsch {
20139318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
2014fdbf62faSLisandro Dalcin   PetscReal      low[3] = {0, 0, 0};
2015fdbf62faSLisandro Dalcin   PetscReal      upp[3] = {1, 1, 1};
2016fdbf62faSLisandro Dalcin   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
2017552f7358SJed Brown 
2018768d5fceSMatthew G. Knepley   PetscFunctionBegin;
20199566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
20209566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
20215dca41c3SJed Brown   PetscCall(DMPlexCreateBoxMesh_Internal(*dm, DM_SHAPE_BOX, dim, simplex, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt, interpolate));
202242108689Sksagiyam   if (periodicity) {
202342108689Sksagiyam     DM cdm;
202442108689Sksagiyam 
202542108689Sksagiyam     PetscCall(DMGetCoordinateDM(*dm, &cdm));
202642108689Sksagiyam     PetscCall(DMPlexSetMaxProjectionHeight(cdm, localizationHeight));
202742108689Sksagiyam     PetscCall(DMSetSparseLocalize(*dm, sparseLocalize));
202842108689Sksagiyam     PetscCall(DMLocalizeCoordinates(*dm));
202942108689Sksagiyam   }
20303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
20319318fe57SMatthew G. Knepley }
2032fdbf62faSLisandro Dalcin 
2033d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeBoxMesh_Internal(DM dm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[])
2034d71ae5a4SJacob Faibussowitsch {
20359318fe57SMatthew G. Knepley   DM       bdm, vol;
20369318fe57SMatthew G. Knepley   PetscInt i;
20379318fe57SMatthew G. Knepley 
20389318fe57SMatthew G. Knepley   PetscFunctionBegin;
20391fcf445aSMatthew G. Knepley   // TODO Now we can support periodicity
204008401ef6SPierre Jolivet   for (i = 0; i < 3; ++i) PetscCheck(periodicity[i] == DM_BOUNDARY_NONE, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Periodicity not yet supported");
20419566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &bdm));
20429566063dSJacob Faibussowitsch   PetscCall(DMSetType(bdm, DMPLEX));
20439566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(bdm, 2));
204446139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, bdm, 0, 0, 0));
20459566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBoxMesh_Simplex_Internal(bdm, 2, faces, lower, upper, periodicity, PETSC_TRUE));
20461fcf445aSMatthew G. Knepley   PetscCall(DMPlexExtrude(bdm, faces[2], upper[2] - lower[2], PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, NULL, NULL, &vol));
204746139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, bdm, 0, 0, 0));
20489566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&bdm));
204969d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
20509318fe57SMatthew G. Knepley   if (lower[2] != 0.0) {
20519318fe57SMatthew G. Knepley     Vec          v;
20529318fe57SMatthew G. Knepley     PetscScalar *x;
20539318fe57SMatthew G. Knepley     PetscInt     cDim, n;
20549318fe57SMatthew G. Knepley 
20559566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &v));
20569566063dSJacob Faibussowitsch     PetscCall(VecGetBlockSize(v, &cDim));
20579566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(v, &n));
20589566063dSJacob Faibussowitsch     PetscCall(VecGetArray(v, &x));
20599318fe57SMatthew G. Knepley     x += cDim;
20609318fe57SMatthew G. Knepley     for (i = 0; i < n; i += cDim) x[i] += lower[2];
20619566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(v, &x));
20629566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, v));
20639318fe57SMatthew G. Knepley   }
20643ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2065552f7358SJed Brown }
2066552f7358SJed Brown 
206700dabe28SStefano Zampini /*@
206839f4f5dbSPierre Jolivet   DMPlexCreateWedgeBoxMesh - Creates a 3-D mesh tessellating the (x,y) plane and extruding in the third direction using wedge cells.
206900dabe28SStefano Zampini 
2070d083f849SBarry Smith   Collective
207100dabe28SStefano Zampini 
207200dabe28SStefano Zampini   Input Parameters:
2073a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
207420f4b53cSBarry Smith . faces       - Number of faces per dimension, or `NULL` for (1, 1, 1)
207520f4b53cSBarry Smith . lower       - The lower left corner, or `NULL` for (0, 0, 0)
207620f4b53cSBarry Smith . upper       - The upper right corner, or `NULL` for (1, 1, 1)
207720f4b53cSBarry Smith . periodicity - The boundary type for the X,Y,Z direction, or `NULL` for `DM_BOUNDARY_NONE`
2078a1cb98faSBarry Smith . orderHeight - If `PETSC_TRUE`, orders the extruded cells in the height first. Otherwise, orders the cell on the layers first
207900dabe28SStefano Zampini - interpolate - Flag to create intermediate mesh pieces (edges, faces)
208000dabe28SStefano Zampini 
208100dabe28SStefano Zampini   Output Parameter:
2082a1cb98faSBarry Smith . dm - The `DM` object
208300dabe28SStefano Zampini 
208400dabe28SStefano Zampini   Level: beginner
208500dabe28SStefano Zampini 
20861cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateWedgeCylinderMesh()`, `DMExtrude()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
208700dabe28SStefano Zampini @*/
2088d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeBoxMesh(MPI_Comm comm, const PetscInt faces[], const PetscReal lower[], const PetscReal upper[], const DMBoundaryType periodicity[], PetscBool orderHeight, PetscBool interpolate, DM *dm)
2089d71ae5a4SJacob Faibussowitsch {
20909318fe57SMatthew G. Knepley   PetscInt       fac[3] = {1, 1, 1};
209100dabe28SStefano Zampini   PetscReal      low[3] = {0, 0, 0};
209200dabe28SStefano Zampini   PetscReal      upp[3] = {1, 1, 1};
209300dabe28SStefano Zampini   DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
209400dabe28SStefano Zampini 
209500dabe28SStefano Zampini   PetscFunctionBegin;
20969566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
20979566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
20989566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeBoxMesh_Internal(*dm, faces ? faces : fac, lower ? lower : low, upper ? upper : upp, periodicity ? periodicity : bdt));
2099d410b0cfSMatthew G. Knepley   if (!interpolate) {
2100d410b0cfSMatthew G. Knepley     DM udm;
210100dabe28SStefano Zampini 
21029566063dSJacob Faibussowitsch     PetscCall(DMPlexUninterpolate(*dm, &udm));
210369d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(*dm, &udm));
210400dabe28SStefano Zampini   }
21057ff04441SMatthew G. Knepley   if (periodicity) PetscCall(DMLocalizeCoordinates(*dm));
21063ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
210700dabe28SStefano Zampini }
210800dabe28SStefano Zampini 
2109cfb853baSMatthew G. Knepley /*
2110cfb853baSMatthew 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.
2111cfb853baSMatthew G. Knepley 
2112cfb853baSMatthew G. Knepley   Input Parameters:
2113cfb853baSMatthew G. Knepley + len - The length of the tuple
2114cfb853baSMatthew G. Knepley . max - The maximum for each dimension, so values are in [0, max)
2115cfb853baSMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition
2116cfb853baSMatthew G. Knepley 
2117cfb853baSMatthew G. Knepley   Output Parameter:
211820f4b53cSBarry Smith . tup - A tuple of `len` integers whose entries are at most `max`
2119cfb853baSMatthew G. Knepley 
2120cfb853baSMatthew G. Knepley   Level: developer
2121cfb853baSMatthew G. Knepley 
212220f4b53cSBarry Smith   Note:
212320f4b53cSBarry Smith   Ordering is lexicographic with lowest index as least significant in ordering.
212420f4b53cSBarry 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}.
212520f4b53cSBarry Smith 
2126cfb853baSMatthew G. Knepley .seealso: PetscDualSpaceTensorPointLexicographic_Internal(), PetscDualSpaceLatticePointLexicographic_Internal()
2127cfb853baSMatthew G. Knepley */
2128cfb853baSMatthew G. Knepley static PetscErrorCode DMPlexTensorPointLexicographic_Private(PetscInt len, const PetscInt max[], PetscInt tup[])
2129cfb853baSMatthew G. Knepley {
2130cfb853baSMatthew G. Knepley   PetscInt i;
2131cfb853baSMatthew G. Knepley 
2132cfb853baSMatthew G. Knepley   PetscFunctionBegin;
2133cfb853baSMatthew G. Knepley   for (i = 0; i < len; ++i) {
2134cfb853baSMatthew G. Knepley     if (tup[i] < max[i] - 1) {
2135cfb853baSMatthew G. Knepley       break;
2136cfb853baSMatthew G. Knepley     } else {
2137cfb853baSMatthew G. Knepley       tup[i] = 0;
2138cfb853baSMatthew G. Knepley     }
2139cfb853baSMatthew G. Knepley   }
2140cfb853baSMatthew G. Knepley   if (i == len) tup[i - 1] = max[i - 1];
2141cfb853baSMatthew G. Knepley   else ++tup[i];
21423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2143cfb853baSMatthew G. Knepley }
2144cfb853baSMatthew G. Knepley 
2145cfb853baSMatthew G. Knepley static PetscInt TupleToIndex_Private(PetscInt len, const PetscInt max[], const PetscInt tup[])
2146cfb853baSMatthew G. Knepley {
21478d2ec52aSSatish Balay   PetscInt idx = tup[len - 1];
2148cfb853baSMatthew G. Knepley 
21498d2ec52aSSatish Balay   for (PetscInt i = len - 2; i >= 0; --i) {
2150cfb853baSMatthew G. Knepley     idx *= max[i];
2151cfb853baSMatthew G. Knepley     idx += tup[i];
2152cfb853baSMatthew G. Knepley   }
2153cfb853baSMatthew G. Knepley   return idx;
2154cfb853baSMatthew G. Knepley }
2155cfb853baSMatthew G. Knepley 
21568d2ec52aSSatish Balay static void IndexToTuple_Private(PetscInt len, const PetscInt max[], PetscInt idx, PetscInt tup[])
2157cfb853baSMatthew G. Knepley {
21588d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
21598d2ec52aSSatish Balay     tup[i] = idx % max[i];
21608d2ec52aSSatish Balay     idx    = (idx - tup[i]) / max[i];
21618d2ec52aSSatish Balay   }
21628d2ec52aSSatish Balay }
21638d2ec52aSSatish Balay 
21648d2ec52aSSatish Balay static void TupleToRanks_Private(PetscInt len, const PetscInt max[], const PetscInt procs[], const PetscInt tup[], PetscInt ranks[])
21658d2ec52aSSatish Balay {
21668d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
21678d2ec52aSSatish Balay     const PetscInt div = max[i] / procs[i];
21688d2ec52aSSatish Balay     const PetscInt rem = max[i] % procs[i];
21698d2ec52aSSatish Balay     const PetscInt idx = (tup[i] < 0 ? max[i] + tup[i] : tup[i]) % max[i];
21708d2ec52aSSatish Balay 
21718d2ec52aSSatish Balay     if (idx < rem * (div + 1)) ranks[i] = idx / (div + 1);
21728d2ec52aSSatish Balay     else ranks[i] = rem + (idx - rem * (div + 1)) / div;
21738d2ec52aSSatish Balay   }
21748d2ec52aSSatish Balay }
21758d2ec52aSSatish Balay 
21768d2ec52aSSatish Balay static void RanksToSizes_Private(PetscInt len, const PetscInt max[], const PetscInt procs[], const PetscInt ranks[], PetscInt sizes[])
21778d2ec52aSSatish Balay {
21788d2ec52aSSatish Balay   for (PetscInt i = 0; i < len; ++i) {
21798d2ec52aSSatish Balay     const PetscInt div = max[i] / procs[i];
21808d2ec52aSSatish Balay     const PetscInt rem = max[i] % procs[i];
21818d2ec52aSSatish Balay 
21828d2ec52aSSatish Balay     sizes[i] = ranks[i] < rem ? div + 1 : div;
21838d2ec52aSSatish Balay   }
21848d2ec52aSSatish Balay }
21858d2ec52aSSatish Balay 
21868d2ec52aSSatish Balay /*
21878d2ec52aSSatish Balay   In serial, the mesh is completely periodic. In parallel, we will include a layer of ghost vertices around the patch, so our edges will not wrap around in parallel. This will instead be handled by the SF.
21888d2ec52aSSatish Balay 
21898d2ec52aSSatish Balay   The cone for all edges will always be complete. Even in the presence of boundaries, we will keep all support sizes constant. When an edge would not exist in the support, we will create one to wrap back periodically. This allows for uniform stencils, and that edge will not be used for computation.
21908d2ec52aSSatish Balay 
21918d2ec52aSSatish Balay   All point which do not attain the vertex lower or upper bound in any dimension are owned, the rest are leaves owned by another process and present in the SF.
2192dfe9cfe5SMatthew Knepley 
2193dfe9cfe5SMatthew Knepley   Parallel Layout:
2194dfe9cfe5SMatthew Knepley 
2195dfe9cfe5SMatthew Knepley   We create a cubic process grid of dimension P^{1/d} on each side. The IndexToTuple_Private() function maps the global rank to the local rank in each dimension. We divide edges using the PETSc distribution rule in each dimension, and then add overlap. TupleToRanks_Private() returns the local rank in ech dimension for a tuple. RanksToSizes_Private() gives the size in each dimension for the domain with those local ranks.
21968d2ec52aSSatish Balay */
21978d2ec52aSSatish Balay static PetscErrorCode DMPlexCreateHypercubicMesh_Internal(DM dm, PetscInt dim, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], PetscInt overlap, const DMBoundaryType bd[])
21988d2ec52aSSatish Balay {
21998d2ec52aSSatish Balay   const PetscInt debug = ((DM_Plex *)dm->data)->printAdj;
22008d2ec52aSSatish Balay   PetscSF        sf;
2201cfb853baSMatthew G. Knepley   Vec            coordinates;
2202cfb853baSMatthew G. Knepley   PetscSection   coordSection;
2203cfb853baSMatthew G. Knepley   DMLabel        cutLabel    = NULL;
2204cfb853baSMatthew G. Knepley   PetscBool      cutMarker   = PETSC_FALSE;
2205cfb853baSMatthew G. Knepley   PetscBool      periodic    = PETSC_FALSE;
22068d2ec52aSSatish Balay   PetscInt       numCells    = 1;
22078d2ec52aSSatish Balay   PetscInt       numVertices = 1;
22088d2ec52aSSatish Balay   PetscSFNode   *remotes;
2209cfb853baSMatthew G. Knepley   PetscScalar   *coords;
22108d2ec52aSSatish Balay   PetscInt      *procs;     // The number of processes along each dimension
22118d2ec52aSSatish Balay   PetscInt      *lrank;     // Rank in each dimension, lrank[d] \in [0, procs[d])
22128d2ec52aSSatish Balay   PetscInt      *ledges;    // The number of edges along each dimension for this process
22138d2ec52aSSatish Balay   PetscInt      *vstart;    // The first vertex along each dimension on this processes
22148d2ec52aSSatish Balay   PetscInt      *vertices;  // The number of vertices along each dimension on this process
22158d2ec52aSSatish Balay   PetscInt      *rvert;     // The global (not local) vertex number along each dimension
22168d2ec52aSSatish Balay   PetscInt      *rrank;     // The rank along each dimension for the process owning rvert[]
22178d2ec52aSSatish Balay   PetscInt      *rvertices; // The number of vertices along each dimension for the process rrank[]
22188d2ec52aSSatish Balay   PetscInt      *vert, *vtmp, *supp, cone[2], *leaves;
22198d2ec52aSSatish Balay   PetscInt       cell = 0, coordSize, Nl = 0, Nl2 = 0;
22208d2ec52aSSatish Balay   PetscMPIInt    rank, size;
22218d2ec52aSSatish Balay   MPI_Comm       comm;
2222cfb853baSMatthew G. Knepley 
2223cfb853baSMatthew G. Knepley   PetscFunctionBegin;
22248d2ec52aSSatish Balay   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
22258d2ec52aSSatish Balay   PetscCallMPI(MPI_Comm_rank(comm, &rank));
22268d2ec52aSSatish Balay   PetscCallMPI(MPI_Comm_size(comm, &size));
2227cfb853baSMatthew G. Knepley   PetscCall(DMSetDimension(dm, dim));
22288d2ec52aSSatish Balay   PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
22298d2ec52aSSatish Balay   PetscCall(PetscCalloc4(dim, &procs, dim, &lrank, dim, &rrank, 2 * dim, &supp));
22308d2ec52aSSatish Balay   PetscCall(PetscCalloc7(dim, &ledges, dim, &vertices, dim, &rvertices, dim, &vert, dim, &rvert, dim, &vstart, dim, &vtmp));
2231cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
2232cfb853baSMatthew G. Knepley   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_periodic_cut", &cutMarker, NULL));
22338d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) periodic = (periodic || bd[d] == DM_BOUNDARY_PERIODIC) ? PETSC_TRUE : PETSC_FALSE;
2234cfb853baSMatthew G. Knepley   if (periodic && cutMarker) {
2235cfb853baSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, "periodic_cut"));
2236cfb853baSMatthew G. Knepley     PetscCall(DMGetLabel(dm, "periodic_cut", &cutLabel));
2237cfb853baSMatthew G. Knepley   }
22388d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) PetscCheck(bd[d] == DM_BOUNDARY_PERIODIC, comm, PETSC_ERR_SUP, "Hypercubic mesh must be periodic now");
22398d2ec52aSSatish Balay   overlap = overlap == PETSC_DETERMINE ? 1 : overlap;
22408d2ec52aSSatish Balay   PetscCheck(overlap >= 1, comm, PETSC_ERR_SUP, "Overlap %" PetscInt_FMT " must be greater than 0", overlap);
22418d2ec52aSSatish Balay   if (size > 1) {
22428d2ec52aSSatish Balay     PetscInt Npr = 1;
22438d2ec52aSSatish Balay 
22448d2ec52aSSatish Balay     // Make process grid
22458d2ec52aSSatish Balay     if (debug) PetscCall(PetscPrintf(comm, "Process grid:"));
22468d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
22478d2ec52aSSatish Balay       procs[d] = PetscRintReal(PetscPowReal(size, 1. / dim));
22488d2ec52aSSatish Balay       Npr *= procs[d];
22498d2ec52aSSatish Balay       if (debug) PetscCall(PetscPrintf(comm, " %" PetscInt_FMT, procs[d]));
22508d2ec52aSSatish Balay     }
22518d2ec52aSSatish Balay     if (debug) PetscCall(PetscPrintf(comm, "\n"));
22528d2ec52aSSatish Balay     PetscCheck(Npr == size, comm, PETSC_ERR_PLIB, "Process grid size %" PetscInt_FMT " != %d comm size", Npr, size);
22538d2ec52aSSatish Balay     IndexToTuple_Private(dim, procs, rank, lrank);
22548d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
22558d2ec52aSSatish Balay       ledges[d] = edges[d] / procs[d] + (edges[d] % procs[d] > lrank[d] ? 1 : 0);
22568d2ec52aSSatish Balay       vstart[d] = 0;
22578d2ec52aSSatish Balay       for (PetscInt r = 0; r < lrank[d]; ++r) vstart[d] += edges[d] / procs[d] + (edges[d] % procs[d] > r ? 1 : 0);
22588d2ec52aSSatish Balay       vstart[d] -= overlap; // For halo
22598d2ec52aSSatish Balay     }
22608d2ec52aSSatish Balay   } else {
22618d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
22628d2ec52aSSatish Balay       procs[d]  = 1;
22638d2ec52aSSatish Balay       ledges[d] = edges[d];
22648d2ec52aSSatish Balay     }
22658d2ec52aSSatish Balay   }
22668d2ec52aSSatish Balay   // Calculate local patch size
22678d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) {
22688d2ec52aSSatish Balay     vertices[d] = ledges[d] + (procs[d] > 1 ? 2 * overlap : 0);
2269cfb853baSMatthew G. Knepley     numVertices *= vertices[d];
2270cfb853baSMatthew G. Knepley   }
2271cfb853baSMatthew G. Knepley   numCells = numVertices * dim;
2272cfb853baSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
22738d2ec52aSSatish Balay   for (PetscInt c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, 2));
22748d2ec52aSSatish Balay   for (PetscInt v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetSupportSize(dm, v, 2 * dim));
2275cfb853baSMatthew G. Knepley   PetscCall(DMSetUp(dm)); /* Allocate space for cones and supports */
2276cfb853baSMatthew G. Knepley   /* Build cell cones and vertex supports */
2277cfb853baSMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "celltype"));
22788d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Topology for rank %d:\n", rank));
2279cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
2280cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert) + numCells;
2281cfb853baSMatthew G. Knepley     PetscInt       s      = 0;
22828d2ec52aSSatish Balay     PetscBool      leaf   = PETSC_FALSE;
2283cfb853baSMatthew G. Knepley 
22848d2ec52aSSatish Balay     if (debug) {
22858d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "Vertex %" PetscInt_FMT ":", vertex));
22868d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vert[d]));
22878d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "\n"));
22888d2ec52aSSatish Balay     }
2289cfb853baSMatthew G. Knepley     PetscCall(DMPlexSetCellType(dm, vertex, DM_POLYTOPE_POINT));
22908d2ec52aSSatish Balay     // Define edge cones
22918d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
22928d2ec52aSSatish Balay       for (PetscInt e = 0; e < dim; ++e) vtmp[e] = vert[e];
2293cfb853baSMatthew G. Knepley       vtmp[d] = (vert[d] + 1) % vertices[d];
2294cfb853baSMatthew G. Knepley       cone[0] = vertex;
2295cfb853baSMatthew G. Knepley       cone[1] = TupleToIndex_Private(dim, vertices, vtmp) + numCells;
22968d2ec52aSSatish Balay       if (debug) {
22978d2ec52aSSatish Balay         PetscCall(PetscSynchronizedPrintf(comm, "  Vertex %" PetscInt_FMT ":", cone[1]));
22988d2ec52aSSatish Balay         for (PetscInt e = 0; e < dim; ++e) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vtmp[e]));
22998d2ec52aSSatish Balay         PetscCall(PetscSynchronizedPrintf(comm, "\n"));
23008d2ec52aSSatish Balay       }
2301cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, cell, cone));
2302cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetCellType(dm, cell, DM_POLYTOPE_SEGMENT));
23038d2ec52aSSatish Balay       if (debug) PetscCall(PetscSynchronizedPrintf(comm, "  Edge %" PetscInt_FMT " (%" PetscInt_FMT " %" PetscInt_FMT ")\n", cell, cone[0], cone[1]));
2304cfb853baSMatthew G. Knepley       ++cell;
23058d2ec52aSSatish Balay       // Shared vertices are any in the first or last overlap layers
23068d2ec52aSSatish Balay       if (vert[d] < overlap || vert[d] >= vertices[d] - overlap) leaf = PETSC_TRUE;
2307cfb853baSMatthew G. Knepley     }
23088d2ec52aSSatish Balay     if (size > 1 && leaf) ++Nl;
23098d2ec52aSSatish Balay     // Define vertex supports
23108d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
23118d2ec52aSSatish Balay       for (PetscInt e = 0; e < dim; ++e) vtmp[e] = vert[e];
2312cfb853baSMatthew G. Knepley       vtmp[d]   = (vert[d] + vertices[d] - 1) % vertices[d];
2313cfb853baSMatthew G. Knepley       supp[s++] = TupleToIndex_Private(dim, vertices, vtmp) * dim + d;
2314cfb853baSMatthew G. Knepley       supp[s++] = (vertex - numCells) * dim + d;
2315cfb853baSMatthew G. Knepley       PetscCall(DMPlexSetSupport(dm, vertex, supp));
2316cfb853baSMatthew G. Knepley     }
2317cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
2318cfb853baSMatthew G. Knepley   }
23198d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedFlush(comm, NULL));
2320cfb853baSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
23218d2ec52aSSatish Balay   // Allocate for SF
23228d2ec52aSSatish Balay   PetscCall(PetscMalloc1(Nl, &leaves));
23238d2ec52aSSatish Balay   PetscCall(PetscMalloc1(Nl, &remotes));
23248d2ec52aSSatish Balay   // Build coordinates
2325cfb853baSMatthew G. Knepley   PetscCall(DMGetCoordinateSection(dm, &coordSection));
2326cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetNumFields(coordSection, 1));
2327cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
2328cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
23298d2ec52aSSatish Balay   for (PetscInt v = numCells; v < numCells + numVertices; ++v) {
2330cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetDof(coordSection, v, dim));
2331cfb853baSMatthew G. Knepley     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
2332cfb853baSMatthew G. Knepley   }
2333cfb853baSMatthew G. Knepley   PetscCall(PetscSectionSetUp(coordSection));
2334cfb853baSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
23358d2ec52aSSatish Balay   PetscCall(VecCreate(comm, &coordinates));
2336cfb853baSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
2337cfb853baSMatthew G. Knepley   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
2338cfb853baSMatthew G. Knepley   PetscCall(VecSetBlockSize(coordinates, dim));
2339cfb853baSMatthew G. Knepley   PetscCall(VecSetType(coordinates, VECSTANDARD));
2340cfb853baSMatthew G. Knepley   PetscCall(VecGetArray(coordinates, &coords));
23418d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Geometry for rank %d:\n", rank));
23428d2ec52aSSatish Balay   for (PetscInt d = 0; d < dim; ++d) vert[d] = 0;
2343cfb853baSMatthew G. Knepley   while (vert[dim - 1] < vertices[dim - 1]) {
2344cfb853baSMatthew G. Knepley     const PetscInt vertex = TupleToIndex_Private(dim, vertices, vert);
23458d2ec52aSSatish Balay     PetscBool      leaf   = PETSC_FALSE;
2346cfb853baSMatthew G. Knepley 
23478d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
23488d2ec52aSSatish Balay       coords[vertex * dim + d] = lower[d] + ((upper[d] - lower[d]) / edges[d]) * (vert[d] + vstart[d]);
23498d2ec52aSSatish Balay       if (vert[d] < overlap || vert[d] >= vertices[d] - overlap) leaf = PETSC_TRUE;
23508d2ec52aSSatish Balay     }
23518d2ec52aSSatish Balay     if (size > 1 && leaf) {
23528d2ec52aSSatish Balay       PetscInt rnumCells = 1;
23538d2ec52aSSatish Balay 
23548d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) rvert[d] = vert[d] + vstart[d];
23558d2ec52aSSatish Balay       TupleToRanks_Private(dim, edges, procs, rvert, rrank);
23568d2ec52aSSatish Balay       leaves[Nl2]       = vertex + numCells;
23578d2ec52aSSatish Balay       remotes[Nl2].rank = TupleToIndex_Private(dim, procs, rrank);
23588d2ec52aSSatish Balay       RanksToSizes_Private(dim, edges, procs, rrank, rvertices);
23598d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) {
23608d2ec52aSSatish Balay         rvertices[d] += 2 * overlap; // Add halo
23618d2ec52aSSatish Balay         rnumCells *= rvertices[d];
23628d2ec52aSSatish Balay       }
23638d2ec52aSSatish Balay       rnumCells *= dim;
23648d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) {
23658d2ec52aSSatish Balay         const PetscInt diff = rrank[d] - lrank[d];
23668d2ec52aSSatish Balay 
23678d2ec52aSSatish Balay         if (!diff) rvert[d] = vert[d];                                     // Vertex is local
23688d2ec52aSSatish Balay         else if (rvert[d] < 0) rvert[d] = rvertices[d] - 1 + rvert[d];     // Wrap around at the bottom
23698d2ec52aSSatish Balay         else if (rvert[d] >= edges[d]) rvert[d] = rvert[d] - edges[d] + 1; // Wrap around at the top
23708d2ec52aSSatish Balay         else if (diff == -1) rvert[d] = rvertices[d] - 1 + (vert[d] - overlap);
23718d2ec52aSSatish Balay         else if (diff == 1) rvert[d] = (vertices[d] - vert[d] - 1) + overlap;
23728d2ec52aSSatish Balay         else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Process distance %" PetscInt_FMT " in direction %" PetscInt_FMT " should not be possible", diff, d);
23738d2ec52aSSatish Balay       }
23748d2ec52aSSatish Balay       remotes[Nl2].index = TupleToIndex_Private(dim, rvertices, rvert) + rnumCells;
23758d2ec52aSSatish Balay       if (debug) PetscCall(PetscSynchronizedPrintf(comm, "Shared Vertex %" PetscInt_FMT " (%" PetscInt_FMT ", %" PetscInt_FMT ")\n", leaves[Nl2], remotes[Nl2].rank, remotes[Nl2].index));
23768d2ec52aSSatish Balay       ++Nl2;
23778d2ec52aSSatish Balay     }
23788d2ec52aSSatish Balay     if (debug) {
23798d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "Vertex %" PetscInt_FMT ":", vertex));
23808d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %" PetscInt_FMT, vert[d] + vstart[d]));
23818d2ec52aSSatish Balay       for (PetscInt d = 0; d < dim; ++d) PetscCall(PetscSynchronizedPrintf(comm, " %g", (double)PetscRealPart(coords[vertex * dim + d])));
23828d2ec52aSSatish Balay       PetscCall(PetscSynchronizedPrintf(comm, "\n"));
23838d2ec52aSSatish Balay     }
2384cfb853baSMatthew G. Knepley     PetscCall(DMPlexTensorPointLexicographic_Private(dim, vertices, vert));
2385cfb853baSMatthew G. Knepley   }
23868d2ec52aSSatish Balay   if (debug) PetscCall(PetscSynchronizedFlush(comm, NULL));
2387cfb853baSMatthew G. Knepley   PetscCall(VecRestoreArray(coordinates, &coords));
2388cfb853baSMatthew G. Knepley   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
2389cfb853baSMatthew G. Knepley   PetscCall(VecDestroy(&coordinates));
23908d2ec52aSSatish Balay   // Build SF
23918d2ec52aSSatish Balay   PetscCheck(Nl == Nl2, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Initial number of leaves %" PetscInt_FMT " != %" PetscInt_FMT " final number", Nl, Nl2);
23928d2ec52aSSatish Balay   PetscCall(DMGetPointSF(dm, &sf));
23938d2ec52aSSatish Balay   PetscCall(PetscSFSetGraph(sf, numCells + numVertices, Nl, leaves, PETSC_OWN_POINTER, remotes, PETSC_OWN_POINTER));
23948d2ec52aSSatish Balay   if (debug) PetscCall(PetscSFView(sf, PETSC_VIEWER_STDOUT_WORLD));
2395cfb853baSMatthew G. Knepley   //PetscCall(DMSetPeriodicity(dm, NULL, lower, upper));
2396cfb853baSMatthew G. Knepley   // Attach the extent
2397cfb853baSMatthew G. Knepley   {
2398cfb853baSMatthew G. Knepley     PetscContainer c;
23998d2ec52aSSatish Balay     PetscInt      *extent, *lextent;
2400cfb853baSMatthew G. Knepley 
2401cfb853baSMatthew G. Knepley     PetscCall(PetscMalloc1(dim, &extent));
24028d2ec52aSSatish Balay     PetscCall(PetscMalloc1(dim, &lextent));
24038d2ec52aSSatish Balay     for (PetscInt d = 0; d < dim; ++d) {
24048d2ec52aSSatish Balay       extent[d]  = edges[d];
24058d2ec52aSSatish Balay       lextent[d] = ledges[d];
24068d2ec52aSSatish Balay     }
2407cfb853baSMatthew G. Knepley     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
240849abdd8aSBarry Smith     PetscCall(PetscContainerSetCtxDestroy(c, PetscCtxDestroyDefault));
2409cfb853baSMatthew G. Knepley     PetscCall(PetscContainerSetPointer(c, extent));
2410cfb853baSMatthew G. Knepley     PetscCall(PetscObjectCompose((PetscObject)dm, "_extent", (PetscObject)c));
2411cfb853baSMatthew G. Knepley     PetscCall(PetscContainerDestroy(&c));
24128d2ec52aSSatish Balay     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &c));
24138d2ec52aSSatish Balay     PetscCall(PetscContainerSetCtxDestroy(c, PetscCtxDestroyDefault));
24148d2ec52aSSatish Balay     PetscCall(PetscContainerSetPointer(c, lextent));
24158d2ec52aSSatish Balay     PetscCall(PetscObjectCompose((PetscObject)dm, "_lextent", (PetscObject)c));
24168d2ec52aSSatish Balay     PetscCall(PetscContainerDestroy(&c));
2417cfb853baSMatthew G. Knepley   }
24188d2ec52aSSatish Balay   PetscCall(PetscFree4(procs, lrank, rrank, supp));
24198d2ec52aSSatish Balay   PetscCall(PetscFree7(ledges, vertices, rvertices, vert, rvert, vstart, vtmp));
24203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2421cfb853baSMatthew G. Knepley }
2422cfb853baSMatthew G. Knepley 
2423cfb853baSMatthew G. Knepley /*@C
2424aaa8cc7dSPierre Jolivet   DMPlexCreateHypercubicMesh - Creates a periodic mesh on the tensor product of unit intervals using only vertices and edges.
2425cfb853baSMatthew G. Knepley 
2426cfb853baSMatthew G. Knepley   Collective
2427cfb853baSMatthew G. Knepley 
2428cfb853baSMatthew G. Knepley   Input Parameters:
24298d2ec52aSSatish Balay + comm    - The communicator for the `DM` object
2430cfb853baSMatthew G. Knepley . dim     - The spatial dimension
243120f4b53cSBarry Smith . edges   - Number of edges per dimension, or `NULL` for (1,) in 1D and (2, 2) in 2D and (1, 1, 1) in 3D
243220f4b53cSBarry Smith . lower   - The lower left corner, or `NULL` for (0, 0, 0)
24338d2ec52aSSatish Balay . upper   - The upper right corner, or `NULL` for (1, 1, 1)
24348d2ec52aSSatish Balay - overlap - The number of vertices in each direction to include in the overlap (default is 1)
2435cfb853baSMatthew G. Knepley 
2436cfb853baSMatthew G. Knepley   Output Parameter:
2437cfb853baSMatthew G. Knepley . dm - The DM object
2438cfb853baSMatthew G. Knepley 
243920f4b53cSBarry Smith   Level: beginner
244020f4b53cSBarry Smith 
244120f4b53cSBarry Smith   Note:
244220f4b53cSBarry Smith   If you want to customize this mesh using options, you just need to
244320f4b53cSBarry Smith .vb
244420f4b53cSBarry Smith   DMCreate(comm, &dm);
244520f4b53cSBarry Smith   DMSetType(dm, DMPLEX);
244620f4b53cSBarry Smith   DMSetFromOptions(dm);
244720f4b53cSBarry Smith .ve
244820f4b53cSBarry Smith   and use the options on the `DMSetFromOptions()` page.
2449cfb853baSMatthew G. Knepley 
2450cfb853baSMatthew G. Knepley   The vertices are numbered is lexicographic order, and the dim edges exiting a vertex in the positive orthant are number consecutively,
245120f4b53cSBarry Smith .vb
245220f4b53cSBarry Smith  18--0-19--2-20--4-18
245320f4b53cSBarry Smith   |     |     |     |
245420f4b53cSBarry Smith  13    15    17    13
245520f4b53cSBarry Smith   |     |     |     |
245620f4b53cSBarry Smith  24-12-25-14-26-16-24
245720f4b53cSBarry Smith   |     |     |     |
245820f4b53cSBarry Smith   7     9    11     7
245920f4b53cSBarry Smith   |     |     |     |
246020f4b53cSBarry Smith  21--6-22--8-23-10-21
246120f4b53cSBarry Smith   |     |     |     |
246220f4b53cSBarry Smith   1     3     5     1
246320f4b53cSBarry Smith   |     |     |     |
246420f4b53cSBarry Smith  18--0-19--2-20--4-18
246520f4b53cSBarry Smith .ve
2466cfb853baSMatthew G. Knepley 
246776fbde31SPierre Jolivet .seealso: `DMSetFromOptions()`, `DMPlexCreateFromFile()`, `DMPlexCreateHexCylinderMesh()`, `DMSetType()`, `DMCreate()`
2468cfb853baSMatthew G. Knepley @*/
24698d2ec52aSSatish Balay PetscErrorCode DMPlexCreateHypercubicMesh(MPI_Comm comm, PetscInt dim, const PetscInt edges[], const PetscReal lower[], const PetscReal upper[], PetscInt overlap, DM *dm)
2470cfb853baSMatthew G. Knepley {
2471cfb853baSMatthew G. Knepley   PetscInt       *edg;
2472cfb853baSMatthew G. Knepley   PetscReal      *low, *upp;
2473cfb853baSMatthew G. Knepley   DMBoundaryType *bdt;
2474cfb853baSMatthew G. Knepley   PetscInt        d;
2475cfb853baSMatthew G. Knepley 
2476cfb853baSMatthew G. Knepley   PetscFunctionBegin;
2477cfb853baSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
2478cfb853baSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
2479cfb853baSMatthew G. Knepley   PetscCall(PetscMalloc4(dim, &edg, dim, &low, dim, &upp, dim, &bdt));
2480cfb853baSMatthew G. Knepley   for (d = 0; d < dim; ++d) {
2481cfb853baSMatthew G. Knepley     edg[d] = edges ? edges[d] : 1;
2482cfb853baSMatthew G. Knepley     low[d] = lower ? lower[d] : 0.;
2483cfb853baSMatthew G. Knepley     upp[d] = upper ? upper[d] : 1.;
2484cfb853baSMatthew G. Knepley     bdt[d] = DM_BOUNDARY_PERIODIC;
2485cfb853baSMatthew G. Knepley   }
24868d2ec52aSSatish Balay   PetscCall(DMPlexCreateHypercubicMesh_Internal(*dm, dim, low, upp, edg, overlap, bdt));
2487cfb853baSMatthew G. Knepley   PetscCall(PetscFree4(edg, low, upp, bdt));
24883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2489cfb853baSMatthew G. Knepley }
2490cfb853baSMatthew G. Knepley 
2491cc4c1da9SBarry Smith /*@
2492a1cb98faSBarry Smith   DMPlexSetOptionsPrefix - Sets the prefix used for searching for all `DM` options in the database.
2493a9074c1eSMatthew G. Knepley 
249420f4b53cSBarry Smith   Logically Collective
2495a9074c1eSMatthew G. Knepley 
2496a9074c1eSMatthew G. Knepley   Input Parameters:
249720f4b53cSBarry Smith + dm     - the `DM` context
2498a9074c1eSMatthew G. Knepley - prefix - the prefix to prepend to all option names
2499a9074c1eSMatthew G. Knepley 
2500a1cb98faSBarry Smith   Level: advanced
2501a1cb98faSBarry Smith 
2502a1cb98faSBarry Smith   Note:
2503a9074c1eSMatthew G. Knepley   A hyphen (-) must NOT be given at the beginning of the prefix name.
2504a9074c1eSMatthew G. Knepley   The first character of all runtime options is AUTOMATICALLY the hyphen.
2505a9074c1eSMatthew G. Knepley 
25061cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `SNESSetFromOptions()`
2507a9074c1eSMatthew G. Knepley @*/
2508d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexSetOptionsPrefix(DM dm, const char prefix[])
2509d71ae5a4SJacob Faibussowitsch {
2510a9074c1eSMatthew G. Knepley   DM_Plex *mesh = (DM_Plex *)dm->data;
2511a9074c1eSMatthew G. Knepley 
2512a9074c1eSMatthew G. Knepley   PetscFunctionBegin;
2513a9074c1eSMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
25149566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, prefix));
25159566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)mesh->partitioner, prefix));
25163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2517a9074c1eSMatthew G. Knepley }
2518a9074c1eSMatthew G. Knepley 
25199318fe57SMatthew G. Knepley /* Remap geometry to cylinder
252061a622f3SMatthew G. Knepley    TODO: This only works for a single refinement, then it is broken
252161a622f3SMatthew G. Knepley 
25229318fe57SMatthew G. Knepley      Interior square: Linear interpolation is correct
25239318fe57SMatthew G. Knepley      The other cells all have vertices on rays from the origin. We want to uniformly expand the spacing
25249318fe57SMatthew G. Knepley      such that the last vertex is on the unit circle. So the closest and farthest vertices are at distance
25250510c589SMatthew G. Knepley 
25269318fe57SMatthew G. Knepley        phi     = arctan(y/x)
25279318fe57SMatthew G. Knepley        d_close = sqrt(1/8 + 1/4 sin^2(phi))
25289318fe57SMatthew G. Knepley        d_far   = sqrt(1/2 + sin^2(phi))
25290510c589SMatthew G. Knepley 
25309318fe57SMatthew G. Knepley      so we remap them using
25310510c589SMatthew G. Knepley 
25329318fe57SMatthew G. Knepley        x_new = x_close + (x - x_close) (1 - d_close) / (d_far - d_close)
25339318fe57SMatthew G. Knepley        y_new = y_close + (y - y_close) (1 - d_close) / (d_far - d_close)
25340510c589SMatthew G. Knepley 
25359318fe57SMatthew G. Knepley      If pi/4 < phi < 3pi/4 or -3pi/4 < phi < -pi/4, then we switch x and y.
25369318fe57SMatthew G. Knepley */
2537d71ae5a4SJacob 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[])
2538d71ae5a4SJacob Faibussowitsch {
25399318fe57SMatthew G. Knepley   const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
25409318fe57SMatthew G. Knepley   const PetscReal ds2 = 0.5 * dis;
254122cc497dSMatthew G. Knepley 
25429318fe57SMatthew G. Knepley   if ((PetscAbsScalar(u[0]) <= ds2) && (PetscAbsScalar(u[1]) <= ds2)) {
25439318fe57SMatthew G. Knepley     f0[0] = u[0];
25449318fe57SMatthew G. Knepley     f0[1] = u[1];
25459318fe57SMatthew G. Knepley   } else {
25469318fe57SMatthew G. Knepley     PetscReal phi, sinp, cosp, dc, df, x, y, xc, yc;
25470510c589SMatthew G. Knepley 
25489318fe57SMatthew G. Knepley     x    = PetscRealPart(u[0]);
25499318fe57SMatthew G. Knepley     y    = PetscRealPart(u[1]);
25509318fe57SMatthew G. Knepley     phi  = PetscAtan2Real(y, x);
25519318fe57SMatthew G. Knepley     sinp = PetscSinReal(phi);
25529318fe57SMatthew G. Knepley     cosp = PetscCosReal(phi);
25539318fe57SMatthew G. Knepley     if ((PetscAbsReal(phi) > PETSC_PI / 4.0) && (PetscAbsReal(phi) < 3.0 * PETSC_PI / 4.0)) {
25549318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / sinp);
25559318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / sinp);
25569318fe57SMatthew G. Knepley       xc = ds2 * x / PetscAbsReal(y);
25579318fe57SMatthew G. Knepley       yc = ds2 * PetscSignReal(y);
25589318fe57SMatthew G. Knepley     } else {
25599318fe57SMatthew G. Knepley       dc = PetscAbsReal(ds2 / cosp);
25609318fe57SMatthew G. Knepley       df = PetscAbsReal(dis / cosp);
25619318fe57SMatthew G. Knepley       xc = ds2 * PetscSignReal(x);
25629318fe57SMatthew G. Knepley       yc = ds2 * y / PetscAbsReal(x);
25639318fe57SMatthew G. Knepley     }
25649318fe57SMatthew G. Knepley     f0[0] = xc + (u[0] - xc) * (1.0 - dc) / (df - dc);
25659318fe57SMatthew G. Knepley     f0[1] = yc + (u[1] - yc) * (1.0 - dc) / (df - dc);
25669318fe57SMatthew G. Knepley   }
25679318fe57SMatthew G. Knepley   f0[2] = u[2];
25689318fe57SMatthew G. Knepley }
25690510c589SMatthew G. Knepley 
257049704ca5SMatthew G. Knepley static PetscErrorCode DMPlexCreateHexCylinderMesh_Internal(DM dm, DMBoundaryType periodicZ, PetscInt Nr)
2571d71ae5a4SJacob Faibussowitsch {
25720510c589SMatthew G. Knepley   const PetscInt dim = 3;
25739318fe57SMatthew G. Knepley   PetscInt       numCells, numVertices;
2574d8c47e87SMatthew G. Knepley   PetscMPIInt    rank;
25750510c589SMatthew G. Knepley 
25760510c589SMatthew G. Knepley   PetscFunctionBegin;
257746139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
25789566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
25799566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
25800510c589SMatthew G. Knepley   /* Create topology */
25810510c589SMatthew G. Knepley   {
25820510c589SMatthew G. Knepley     PetscInt cone[8], c;
25830510c589SMatthew G. Knepley 
2584dd400576SPatrick Sanan     numCells    = rank == 0 ? 5 : 0;
2585dd400576SPatrick Sanan     numVertices = rank == 0 ? 16 : 0;
2586006a8963SMatthew G. Knepley     if (periodicZ == DM_BOUNDARY_PERIODIC) {
2587ae8bcbbbSMatthew G. Knepley       numCells *= 3;
2588dd400576SPatrick Sanan       numVertices = rank == 0 ? 24 : 0;
2589006a8963SMatthew G. Knepley     }
25909566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
25919566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 8));
25929566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
2593dd400576SPatrick Sanan     if (rank == 0) {
2594006a8963SMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
25959371c9d4SSatish Balay         cone[0] = 15;
25969371c9d4SSatish Balay         cone[1] = 18;
25979371c9d4SSatish Balay         cone[2] = 17;
25989371c9d4SSatish Balay         cone[3] = 16;
25999371c9d4SSatish Balay         cone[4] = 31;
26009371c9d4SSatish Balay         cone[5] = 32;
26019371c9d4SSatish Balay         cone[6] = 33;
26029371c9d4SSatish Balay         cone[7] = 34;
26039566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
26049371c9d4SSatish Balay         cone[0] = 16;
26059371c9d4SSatish Balay         cone[1] = 17;
26069371c9d4SSatish Balay         cone[2] = 24;
26079371c9d4SSatish Balay         cone[3] = 23;
26089371c9d4SSatish Balay         cone[4] = 32;
26099371c9d4SSatish Balay         cone[5] = 36;
26109371c9d4SSatish Balay         cone[6] = 37;
26119371c9d4SSatish Balay         cone[7] = 33; /* 22 25 26 21 */
26129566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
26139371c9d4SSatish Balay         cone[0] = 18;
26149371c9d4SSatish Balay         cone[1] = 27;
26159371c9d4SSatish Balay         cone[2] = 24;
26169371c9d4SSatish Balay         cone[3] = 17;
26179371c9d4SSatish Balay         cone[4] = 34;
26189371c9d4SSatish Balay         cone[5] = 33;
26199371c9d4SSatish Balay         cone[6] = 37;
26209371c9d4SSatish Balay         cone[7] = 38;
26219566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
26229371c9d4SSatish Balay         cone[0] = 29;
26239371c9d4SSatish Balay         cone[1] = 27;
26249371c9d4SSatish Balay         cone[2] = 18;
26259371c9d4SSatish Balay         cone[3] = 15;
26269371c9d4SSatish Balay         cone[4] = 35;
26279371c9d4SSatish Balay         cone[5] = 31;
26289371c9d4SSatish Balay         cone[6] = 34;
26299371c9d4SSatish Balay         cone[7] = 38;
26309566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
26319371c9d4SSatish Balay         cone[0] = 29;
26329371c9d4SSatish Balay         cone[1] = 15;
26339371c9d4SSatish Balay         cone[2] = 16;
26349371c9d4SSatish Balay         cone[3] = 23;
26359371c9d4SSatish Balay         cone[4] = 35;
26369371c9d4SSatish Balay         cone[5] = 36;
26379371c9d4SSatish Balay         cone[6] = 32;
26389371c9d4SSatish Balay         cone[7] = 31;
26399566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2640006a8963SMatthew G. Knepley 
26419371c9d4SSatish Balay         cone[0] = 31;
26429371c9d4SSatish Balay         cone[1] = 34;
26439371c9d4SSatish Balay         cone[2] = 33;
26449371c9d4SSatish Balay         cone[3] = 32;
26459371c9d4SSatish Balay         cone[4] = 19;
26469371c9d4SSatish Balay         cone[5] = 22;
26479371c9d4SSatish Balay         cone[6] = 21;
26489371c9d4SSatish Balay         cone[7] = 20;
26499566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
26509371c9d4SSatish Balay         cone[0] = 32;
26519371c9d4SSatish Balay         cone[1] = 33;
26529371c9d4SSatish Balay         cone[2] = 37;
26539371c9d4SSatish Balay         cone[3] = 36;
26549371c9d4SSatish Balay         cone[4] = 22;
26559371c9d4SSatish Balay         cone[5] = 25;
26569371c9d4SSatish Balay         cone[6] = 26;
26579371c9d4SSatish Balay         cone[7] = 21;
26589566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 6, cone));
26599371c9d4SSatish Balay         cone[0] = 34;
26609371c9d4SSatish Balay         cone[1] = 38;
26619371c9d4SSatish Balay         cone[2] = 37;
26629371c9d4SSatish Balay         cone[3] = 33;
26639371c9d4SSatish Balay         cone[4] = 20;
26649371c9d4SSatish Balay         cone[5] = 21;
26659371c9d4SSatish Balay         cone[6] = 26;
26669371c9d4SSatish Balay         cone[7] = 28;
26679566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 7, cone));
26689371c9d4SSatish Balay         cone[0] = 35;
26699371c9d4SSatish Balay         cone[1] = 38;
26709371c9d4SSatish Balay         cone[2] = 34;
26719371c9d4SSatish Balay         cone[3] = 31;
26729371c9d4SSatish Balay         cone[4] = 30;
26739371c9d4SSatish Balay         cone[5] = 19;
26749371c9d4SSatish Balay         cone[6] = 20;
26759371c9d4SSatish Balay         cone[7] = 28;
26769566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 8, cone));
26779371c9d4SSatish Balay         cone[0] = 35;
26789371c9d4SSatish Balay         cone[1] = 31;
26799371c9d4SSatish Balay         cone[2] = 32;
26809371c9d4SSatish Balay         cone[3] = 36;
26819371c9d4SSatish Balay         cone[4] = 30;
26829371c9d4SSatish Balay         cone[5] = 25;
26839371c9d4SSatish Balay         cone[6] = 22;
26849371c9d4SSatish Balay         cone[7] = 19;
26859566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 9, cone));
2686ae8bcbbbSMatthew G. Knepley 
26879371c9d4SSatish Balay         cone[0] = 19;
26889371c9d4SSatish Balay         cone[1] = 20;
26899371c9d4SSatish Balay         cone[2] = 21;
26909371c9d4SSatish Balay         cone[3] = 22;
26919371c9d4SSatish Balay         cone[4] = 15;
26929371c9d4SSatish Balay         cone[5] = 16;
26939371c9d4SSatish Balay         cone[6] = 17;
26949371c9d4SSatish Balay         cone[7] = 18;
26959566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 10, cone));
26969371c9d4SSatish Balay         cone[0] = 22;
26979371c9d4SSatish Balay         cone[1] = 21;
26989371c9d4SSatish Balay         cone[2] = 26;
26999371c9d4SSatish Balay         cone[3] = 25;
27009371c9d4SSatish Balay         cone[4] = 16;
27019371c9d4SSatish Balay         cone[5] = 23;
27029371c9d4SSatish Balay         cone[6] = 24;
27039371c9d4SSatish Balay         cone[7] = 17;
27049566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 11, cone));
27059371c9d4SSatish Balay         cone[0] = 20;
27069371c9d4SSatish Balay         cone[1] = 28;
27079371c9d4SSatish Balay         cone[2] = 26;
27089371c9d4SSatish Balay         cone[3] = 21;
27099371c9d4SSatish Balay         cone[4] = 18;
27109371c9d4SSatish Balay         cone[5] = 17;
27119371c9d4SSatish Balay         cone[6] = 24;
27129371c9d4SSatish Balay         cone[7] = 27;
27139566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 12, cone));
27149371c9d4SSatish Balay         cone[0] = 30;
27159371c9d4SSatish Balay         cone[1] = 28;
27169371c9d4SSatish Balay         cone[2] = 20;
27179371c9d4SSatish Balay         cone[3] = 19;
27189371c9d4SSatish Balay         cone[4] = 29;
27199371c9d4SSatish Balay         cone[5] = 15;
27209371c9d4SSatish Balay         cone[6] = 18;
27219371c9d4SSatish Balay         cone[7] = 27;
27229566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 13, cone));
27239371c9d4SSatish Balay         cone[0] = 30;
27249371c9d4SSatish Balay         cone[1] = 19;
27259371c9d4SSatish Balay         cone[2] = 22;
27269371c9d4SSatish Balay         cone[3] = 25;
27279371c9d4SSatish Balay         cone[4] = 29;
27289371c9d4SSatish Balay         cone[5] = 23;
27299371c9d4SSatish Balay         cone[6] = 16;
27309371c9d4SSatish Balay         cone[7] = 15;
27319566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
2732006a8963SMatthew G. Knepley       } else {
27339371c9d4SSatish Balay         cone[0] = 5;
27349371c9d4SSatish Balay         cone[1] = 8;
27359371c9d4SSatish Balay         cone[2] = 7;
27369371c9d4SSatish Balay         cone[3] = 6;
27379371c9d4SSatish Balay         cone[4] = 9;
27389371c9d4SSatish Balay         cone[5] = 12;
27399371c9d4SSatish Balay         cone[6] = 11;
27409371c9d4SSatish Balay         cone[7] = 10;
27419566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
27429371c9d4SSatish Balay         cone[0] = 6;
27439371c9d4SSatish Balay         cone[1] = 7;
27449371c9d4SSatish Balay         cone[2] = 14;
27459371c9d4SSatish Balay         cone[3] = 13;
27469371c9d4SSatish Balay         cone[4] = 12;
27479371c9d4SSatish Balay         cone[5] = 15;
27489371c9d4SSatish Balay         cone[6] = 16;
27499371c9d4SSatish Balay         cone[7] = 11;
27509566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
27519371c9d4SSatish Balay         cone[0] = 8;
27529371c9d4SSatish Balay         cone[1] = 17;
27539371c9d4SSatish Balay         cone[2] = 14;
27549371c9d4SSatish Balay         cone[3] = 7;
27559371c9d4SSatish Balay         cone[4] = 10;
27569371c9d4SSatish Balay         cone[5] = 11;
27579371c9d4SSatish Balay         cone[6] = 16;
27589371c9d4SSatish Balay         cone[7] = 18;
27599566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
27609371c9d4SSatish Balay         cone[0] = 19;
27619371c9d4SSatish Balay         cone[1] = 17;
27629371c9d4SSatish Balay         cone[2] = 8;
27639371c9d4SSatish Balay         cone[3] = 5;
27649371c9d4SSatish Balay         cone[4] = 20;
27659371c9d4SSatish Balay         cone[5] = 9;
27669371c9d4SSatish Balay         cone[6] = 10;
27679371c9d4SSatish Balay         cone[7] = 18;
27689566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
27699371c9d4SSatish Balay         cone[0] = 19;
27709371c9d4SSatish Balay         cone[1] = 5;
27719371c9d4SSatish Balay         cone[2] = 6;
27729371c9d4SSatish Balay         cone[3] = 13;
27739371c9d4SSatish Balay         cone[4] = 20;
27749371c9d4SSatish Balay         cone[5] = 15;
27759371c9d4SSatish Balay         cone[6] = 12;
27769371c9d4SSatish Balay         cone[7] = 9;
27779566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
2778006a8963SMatthew G. Knepley       }
2779d8c47e87SMatthew G. Knepley     }
27809566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
27819566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
27820510c589SMatthew G. Knepley   }
2783dbc1dc17SMatthew G. Knepley   /* Create cube geometry */
27840510c589SMatthew G. Knepley   {
27850510c589SMatthew G. Knepley     Vec             coordinates;
27860510c589SMatthew G. Knepley     PetscSection    coordSection;
27870510c589SMatthew G. Knepley     PetscScalar    *coords;
27880510c589SMatthew G. Knepley     PetscInt        coordSize, v;
27890510c589SMatthew G. Knepley     const PetscReal dis = 1.0 / PetscSqrtReal(2.0);
27900510c589SMatthew G. Knepley     const PetscReal ds2 = dis / 2.0;
27910510c589SMatthew G. Knepley 
27920510c589SMatthew G. Knepley     /* Build coordinates */
27939566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
27949566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
27959566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
27969566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
27970510c589SMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
27989566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
27999566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
28000510c589SMatthew G. Knepley     }
28019566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
28029566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
28039566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
28049566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
28059566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
28069566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
28079566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
28089566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
2809dd400576SPatrick Sanan     if (rank == 0) {
28109371c9d4SSatish Balay       coords[0 * dim + 0]  = -ds2;
28119371c9d4SSatish Balay       coords[0 * dim + 1]  = -ds2;
28129371c9d4SSatish Balay       coords[0 * dim + 2]  = 0.0;
28139371c9d4SSatish Balay       coords[1 * dim + 0]  = ds2;
28149371c9d4SSatish Balay       coords[1 * dim + 1]  = -ds2;
28159371c9d4SSatish Balay       coords[1 * dim + 2]  = 0.0;
28169371c9d4SSatish Balay       coords[2 * dim + 0]  = ds2;
28179371c9d4SSatish Balay       coords[2 * dim + 1]  = ds2;
28189371c9d4SSatish Balay       coords[2 * dim + 2]  = 0.0;
28199371c9d4SSatish Balay       coords[3 * dim + 0]  = -ds2;
28209371c9d4SSatish Balay       coords[3 * dim + 1]  = ds2;
28219371c9d4SSatish Balay       coords[3 * dim + 2]  = 0.0;
28229371c9d4SSatish Balay       coords[4 * dim + 0]  = -ds2;
28239371c9d4SSatish Balay       coords[4 * dim + 1]  = -ds2;
28249371c9d4SSatish Balay       coords[4 * dim + 2]  = 1.0;
28259371c9d4SSatish Balay       coords[5 * dim + 0]  = -ds2;
28269371c9d4SSatish Balay       coords[5 * dim + 1]  = ds2;
28279371c9d4SSatish Balay       coords[5 * dim + 2]  = 1.0;
28289371c9d4SSatish Balay       coords[6 * dim + 0]  = ds2;
28299371c9d4SSatish Balay       coords[6 * dim + 1]  = ds2;
28309371c9d4SSatish Balay       coords[6 * dim + 2]  = 1.0;
28319371c9d4SSatish Balay       coords[7 * dim + 0]  = ds2;
28329371c9d4SSatish Balay       coords[7 * dim + 1]  = -ds2;
28339371c9d4SSatish Balay       coords[7 * dim + 2]  = 1.0;
28349371c9d4SSatish Balay       coords[8 * dim + 0]  = dis;
28359371c9d4SSatish Balay       coords[8 * dim + 1]  = -dis;
28369371c9d4SSatish Balay       coords[8 * dim + 2]  = 0.0;
28379371c9d4SSatish Balay       coords[9 * dim + 0]  = dis;
28389371c9d4SSatish Balay       coords[9 * dim + 1]  = dis;
28399371c9d4SSatish Balay       coords[9 * dim + 2]  = 0.0;
28409371c9d4SSatish Balay       coords[10 * dim + 0] = dis;
28419371c9d4SSatish Balay       coords[10 * dim + 1] = -dis;
28429371c9d4SSatish Balay       coords[10 * dim + 2] = 1.0;
28439371c9d4SSatish Balay       coords[11 * dim + 0] = dis;
28449371c9d4SSatish Balay       coords[11 * dim + 1] = dis;
28459371c9d4SSatish Balay       coords[11 * dim + 2] = 1.0;
28469371c9d4SSatish Balay       coords[12 * dim + 0] = -dis;
28479371c9d4SSatish Balay       coords[12 * dim + 1] = dis;
28489371c9d4SSatish Balay       coords[12 * dim + 2] = 0.0;
28499371c9d4SSatish Balay       coords[13 * dim + 0] = -dis;
28509371c9d4SSatish Balay       coords[13 * dim + 1] = dis;
28519371c9d4SSatish Balay       coords[13 * dim + 2] = 1.0;
28529371c9d4SSatish Balay       coords[14 * dim + 0] = -dis;
28539371c9d4SSatish Balay       coords[14 * dim + 1] = -dis;
28549371c9d4SSatish Balay       coords[14 * dim + 2] = 0.0;
28559371c9d4SSatish Balay       coords[15 * dim + 0] = -dis;
28569371c9d4SSatish Balay       coords[15 * dim + 1] = -dis;
28579371c9d4SSatish Balay       coords[15 * dim + 2] = 1.0;
2858ae8bcbbbSMatthew G. Knepley       if (periodicZ == DM_BOUNDARY_PERIODIC) {
28599371c9d4SSatish Balay         /* 15 31 19 */ coords[16 * dim + 0] = -ds2;
28609371c9d4SSatish Balay         coords[16 * dim + 1]                = -ds2;
28619371c9d4SSatish Balay         coords[16 * dim + 2]                = 0.5;
28629371c9d4SSatish Balay         /* 16 32 22 */ coords[17 * dim + 0] = ds2;
28639371c9d4SSatish Balay         coords[17 * dim + 1]                = -ds2;
28649371c9d4SSatish Balay         coords[17 * dim + 2]                = 0.5;
28659371c9d4SSatish Balay         /* 17 33 21 */ coords[18 * dim + 0] = ds2;
28669371c9d4SSatish Balay         coords[18 * dim + 1]                = ds2;
28679371c9d4SSatish Balay         coords[18 * dim + 2]                = 0.5;
28689371c9d4SSatish Balay         /* 18 34 20 */ coords[19 * dim + 0] = -ds2;
28699371c9d4SSatish Balay         coords[19 * dim + 1]                = ds2;
28709371c9d4SSatish Balay         coords[19 * dim + 2]                = 0.5;
28719371c9d4SSatish Balay         /* 29 35 30 */ coords[20 * dim + 0] = -dis;
28729371c9d4SSatish Balay         coords[20 * dim + 1]                = -dis;
28739371c9d4SSatish Balay         coords[20 * dim + 2]                = 0.5;
28749371c9d4SSatish Balay         /* 23 36 25 */ coords[21 * dim + 0] = dis;
28759371c9d4SSatish Balay         coords[21 * dim + 1]                = -dis;
28769371c9d4SSatish Balay         coords[21 * dim + 2]                = 0.5;
28779371c9d4SSatish Balay         /* 24 37 26 */ coords[22 * dim + 0] = dis;
28789371c9d4SSatish Balay         coords[22 * dim + 1]                = dis;
28799371c9d4SSatish Balay         coords[22 * dim + 2]                = 0.5;
28809371c9d4SSatish Balay         /* 27 38 28 */ coords[23 * dim + 0] = -dis;
28819371c9d4SSatish Balay         coords[23 * dim + 1]                = dis;
28829371c9d4SSatish Balay         coords[23 * dim + 2]                = 0.5;
2883ae8bcbbbSMatthew G. Knepley       }
2884d8c47e87SMatthew G. Knepley     }
28859566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
28869566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
28879566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
28880510c589SMatthew G. Knepley   }
2889006a8963SMatthew G. Knepley   /* Create periodicity */
2890006a8963SMatthew G. Knepley   if (periodicZ == DM_BOUNDARY_PERIODIC || periodicZ == DM_BOUNDARY_TWIST) {
28916858538eSMatthew G. Knepley     PetscReal L[3]       = {-1., -1., 0.};
28926858538eSMatthew G. Knepley     PetscReal maxCell[3] = {-1., -1., 0.};
2893006a8963SMatthew G. Knepley     PetscReal lower[3]   = {0.0, 0.0, 0.0};
2894ae8bcbbbSMatthew G. Knepley     PetscReal upper[3]   = {1.0, 1.0, 1.5};
28956858538eSMatthew G. Knepley     PetscInt  numZCells  = 3;
2896006a8963SMatthew G. Knepley 
28976858538eSMatthew G. Knepley     L[2]       = upper[2] - lower[2];
28986858538eSMatthew G. Knepley     maxCell[2] = 1.1 * (L[2] / numZCells);
28994fb89dddSMatthew G. Knepley     PetscCall(DMSetPeriodicity(dm, maxCell, lower, L));
2900006a8963SMatthew G. Knepley   }
2901dbc1dc17SMatthew G. Knepley   {
29029318fe57SMatthew G. Knepley     DM          cdm;
29039318fe57SMatthew G. Knepley     PetscDS     cds;
29049318fe57SMatthew G. Knepley     PetscScalar c[2] = {1.0, 1.0};
2905dbc1dc17SMatthew G. Knepley 
290649704ca5SMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
29079566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
29089566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
29099566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 2, c));
2910dbc1dc17SMatthew G. Knepley   }
291146139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
291246139095SJed Brown 
29139318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
29149566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(dm));
291549704ca5SMatthew G. Knepley 
291649704ca5SMatthew G. Knepley   char        oldprefix[PETSC_MAX_PATH_LEN];
291749704ca5SMatthew G. Knepley   const char *prefix;
291849704ca5SMatthew G. Knepley 
291949704ca5SMatthew G. Knepley   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
292049704ca5SMatthew G. Knepley   PetscCall(PetscStrncpy(oldprefix, prefix, PETSC_MAX_PATH_LEN));
292149704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, "petsc_cyl_ref_"));
292249704ca5SMatthew G. Knepley   for (PetscInt r = 0; r < PetscMax(0, Nr); ++r) {
292349704ca5SMatthew G. Knepley     DM rdm;
292449704ca5SMatthew G. Knepley 
292549704ca5SMatthew G. Knepley     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
292649704ca5SMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &rdm));
292749704ca5SMatthew G. Knepley   }
292849704ca5SMatthew G. Knepley   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldprefix));
292949704ca5SMatthew G. Knepley   PetscCall(DMPlexRemapGeometry(dm, 0.0, snapToCylinder));
293049704ca5SMatthew G. Knepley 
293149704ca5SMatthew G. Knepley   DMLabel         bdlabel, edgelabel;
293249704ca5SMatthew G. Knepley   IS              faceIS;
293349704ca5SMatthew G. Knepley   const PetscInt *faces;
293449704ca5SMatthew G. Knepley   PetscInt        Nf;
293549704ca5SMatthew G. Knepley 
293649704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "marker"));
293749704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
293849704ca5SMatthew G. Knepley   PetscCall(DMCreateLabel(dm, "generatrix"));
293949704ca5SMatthew G. Knepley   PetscCall(DMGetLabel(dm, "generatrix", &edgelabel));
294049704ca5SMatthew G. Knepley   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
294149704ca5SMatthew G. Knepley   // Remove faces on top and bottom
294249704ca5SMatthew G. Knepley   PetscCall(DMLabelGetStratumIS(bdlabel, 1, &faceIS));
2943ba1b3593SJeremy L Thompson   if (faceIS) {
294449704ca5SMatthew G. Knepley     PetscCall(ISGetLocalSize(faceIS, &Nf));
294549704ca5SMatthew G. Knepley     PetscCall(ISGetIndices(faceIS, &faces));
294649704ca5SMatthew G. Knepley     for (PetscInt f = 0; f < Nf; ++f) {
294749704ca5SMatthew G. Knepley       PetscReal vol, normal[3];
294849704ca5SMatthew G. Knepley 
294949704ca5SMatthew G. Knepley       PetscCall(DMPlexComputeCellGeometryFVM(dm, faces[f], &vol, NULL, normal));
295049704ca5SMatthew G. Knepley       if (PetscAbsReal(normal[2]) < PETSC_SMALL) PetscCall(DMLabelSetValue(edgelabel, faces[f], 1));
295149704ca5SMatthew G. Knepley     }
295249704ca5SMatthew G. Knepley     PetscCall(ISRestoreIndices(faceIS, &faces));
295349704ca5SMatthew G. Knepley     PetscCall(ISDestroy(&faceIS));
2954ba1b3593SJeremy L Thompson   }
295549704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, bdlabel));
295649704ca5SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(dm, edgelabel));
29573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
29580510c589SMatthew G. Knepley }
29590510c589SMatthew G. Knepley 
296024119c2aSMatthew G. Knepley /*@
29619318fe57SMatthew G. Knepley   DMPlexCreateHexCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using hexahedra.
296224119c2aSMatthew G. Knepley 
2963d083f849SBarry Smith   Collective
296424119c2aSMatthew G. Knepley 
296524119c2aSMatthew G. Knepley   Input Parameters:
2966a1cb98faSBarry Smith + comm      - The communicator for the `DM` object
296749704ca5SMatthew G. Knepley . periodicZ - The boundary type for the Z direction
296849704ca5SMatthew G. Knepley - Nr        - The number of refinements to carry out
296924119c2aSMatthew G. Knepley 
297024119c2aSMatthew G. Knepley   Output Parameter:
297120f4b53cSBarry Smith . dm - The `DM` object
297224119c2aSMatthew G. Knepley 
297324119c2aSMatthew G. Knepley   Level: beginner
297424119c2aSMatthew G. Knepley 
2975a1cb98faSBarry Smith   Note:
2976a4e35b19SJacob Faibussowitsch   Here is the output numbering looking from the bottom of the cylinder\:
2977a1cb98faSBarry Smith .vb
2978a1cb98faSBarry Smith        17-----14
2979a1cb98faSBarry Smith         |     |
2980a1cb98faSBarry Smith         |  2  |
2981a1cb98faSBarry Smith         |     |
2982a1cb98faSBarry Smith  17-----8-----7-----14
2983a1cb98faSBarry Smith   |     |     |     |
2984a1cb98faSBarry Smith   |  3  |  0  |  1  |
2985a1cb98faSBarry Smith   |     |     |     |
2986a1cb98faSBarry Smith  19-----5-----6-----13
2987a1cb98faSBarry Smith         |     |
2988a1cb98faSBarry Smith         |  4  |
2989a1cb98faSBarry Smith         |     |
2990a1cb98faSBarry Smith        19-----13
2991a1cb98faSBarry Smith 
2992a1cb98faSBarry Smith  and up through the top
2993a1cb98faSBarry Smith 
2994a1cb98faSBarry Smith        18-----16
2995a1cb98faSBarry Smith         |     |
2996a1cb98faSBarry Smith         |  2  |
2997a1cb98faSBarry Smith         |     |
2998a1cb98faSBarry Smith  18----10----11-----16
2999a1cb98faSBarry Smith   |     |     |     |
3000a1cb98faSBarry Smith   |  3  |  0  |  1  |
3001a1cb98faSBarry Smith   |     |     |     |
3002a1cb98faSBarry Smith  20-----9----12-----15
3003a1cb98faSBarry Smith         |     |
3004a1cb98faSBarry Smith         |  4  |
3005a1cb98faSBarry Smith         |     |
3006a1cb98faSBarry Smith        20-----15
3007a1cb98faSBarry Smith .ve
3008a1cb98faSBarry Smith 
30091cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
301024119c2aSMatthew G. Knepley @*/
301149704ca5SMatthew G. Knepley PetscErrorCode DMPlexCreateHexCylinderMesh(MPI_Comm comm, DMBoundaryType periodicZ, PetscInt Nr, DM *dm)
3012d71ae5a4SJacob Faibussowitsch {
30139318fe57SMatthew G. Knepley   PetscFunctionBegin;
301449704ca5SMatthew G. Knepley   PetscAssertPointer(dm, 4);
30159566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
30169566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
301749704ca5SMatthew G. Knepley   PetscCall(DMPlexCreateHexCylinderMesh_Internal(*dm, periodicZ, Nr));
30183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30199318fe57SMatthew G. Knepley }
30209318fe57SMatthew G. Knepley 
3021d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateWedgeCylinderMesh_Internal(DM dm, PetscInt n, PetscBool interpolate)
3022d71ae5a4SJacob Faibussowitsch {
302324119c2aSMatthew G. Knepley   const PetscInt dim = 3;
3024412e9a14SMatthew G. Knepley   PetscInt       numCells, numVertices, v;
30259fe9f049SMatthew G. Knepley   PetscMPIInt    rank;
302624119c2aSMatthew G. Knepley 
302724119c2aSMatthew G. Knepley   PetscFunctionBegin;
302863a3b9bcSJacob Faibussowitsch   PetscCheck(n >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Number of wedges %" PetscInt_FMT " cannot be negative", n);
302946139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
30309566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
30319566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
3032412e9a14SMatthew G. Knepley   /* Must create the celltype label here so that we do not automatically try to compute the types */
30339566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "celltype"));
303424119c2aSMatthew G. Knepley   /* Create topology */
303524119c2aSMatthew G. Knepley   {
303624119c2aSMatthew G. Knepley     PetscInt cone[6], c;
303724119c2aSMatthew G. Knepley 
3038dd400576SPatrick Sanan     numCells    = rank == 0 ? n : 0;
3039dd400576SPatrick Sanan     numVertices = rank == 0 ? 2 * (n + 1) : 0;
30409566063dSJacob Faibussowitsch     PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
30419566063dSJacob Faibussowitsch     for (c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 6));
30429566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dm));
304324119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
30449371c9d4SSatish Balay       cone[0] = c + n * 1;
30459371c9d4SSatish Balay       cone[1] = (c + 1) % n + n * 1;
30469371c9d4SSatish Balay       cone[2] = 0 + 3 * n;
30479371c9d4SSatish Balay       cone[3] = c + n * 2;
30489371c9d4SSatish Balay       cone[4] = (c + 1) % n + n * 2;
30499371c9d4SSatish Balay       cone[5] = 1 + 3 * n;
30509566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(dm, c, cone));
30519566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCellType(dm, c, DM_POLYTOPE_TRI_PRISM_TENSOR));
305224119c2aSMatthew G. Knepley     }
30539566063dSJacob Faibussowitsch     PetscCall(DMPlexSymmetrize(dm));
30549566063dSJacob Faibussowitsch     PetscCall(DMPlexStratify(dm));
305524119c2aSMatthew G. Knepley   }
305648a46eb9SPierre Jolivet   for (v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(dm, v, DM_POLYTOPE_POINT));
305724119c2aSMatthew G. Knepley   /* Create cylinder geometry */
305824119c2aSMatthew G. Knepley   {
305924119c2aSMatthew G. Knepley     Vec          coordinates;
306024119c2aSMatthew G. Knepley     PetscSection coordSection;
306124119c2aSMatthew G. Knepley     PetscScalar *coords;
3062412e9a14SMatthew G. Knepley     PetscInt     coordSize, c;
306324119c2aSMatthew G. Knepley 
306424119c2aSMatthew G. Knepley     /* Build coordinates */
30659566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(dm, &coordSection));
30669566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetNumFields(coordSection, 1));
30679566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
30689566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
306924119c2aSMatthew G. Knepley     for (v = numCells; v < numCells + numVertices; ++v) {
30709566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(coordSection, v, dim));
30719566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
307224119c2aSMatthew G. Knepley     }
30739566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(coordSection));
30749566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
30759566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
30769566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
30779566063dSJacob Faibussowitsch     PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
30789566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(coordinates, dim));
30799566063dSJacob Faibussowitsch     PetscCall(VecSetType(coordinates, VECSTANDARD));
30809566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
308124119c2aSMatthew G. Knepley     for (c = 0; c < numCells; c++) {
30829371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
30839371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
30849371c9d4SSatish Balay       coords[(c + 0 * n) * dim + 2] = 1.0;
30859371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 0] = PetscCosReal(2.0 * c * PETSC_PI / n);
30869371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 1] = PetscSinReal(2.0 * c * PETSC_PI / n);
30879371c9d4SSatish Balay       coords[(c + 1 * n) * dim + 2] = 0.0;
308824119c2aSMatthew G. Knepley     }
3089dd400576SPatrick Sanan     if (rank == 0) {
30909371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 0] = 0.0;
30919371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 1] = 0.0;
30929371c9d4SSatish Balay       coords[(2 * n + 0) * dim + 2] = 1.0;
30939371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 0] = 0.0;
30949371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 1] = 0.0;
30959371c9d4SSatish Balay       coords[(2 * n + 1) * dim + 2] = 0.0;
30969fe9f049SMatthew G. Knepley     }
30979566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(coordinates, &coords));
30989566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinatesLocal(dm, coordinates));
30999566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&coordinates));
310024119c2aSMatthew G. Knepley   }
310146139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
31029318fe57SMatthew G. Knepley   /* Interpolate */
31039566063dSJacob Faibussowitsch   if (interpolate) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
31043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
31059318fe57SMatthew G. Knepley }
31069318fe57SMatthew G. Knepley 
31079318fe57SMatthew G. Knepley /*@
31089318fe57SMatthew G. Knepley   DMPlexCreateWedgeCylinderMesh - Creates a mesh on the tensor product of the unit interval with the circle (cylinder) using wedges.
31099318fe57SMatthew G. Knepley 
31109318fe57SMatthew G. Knepley   Collective
31119318fe57SMatthew G. Knepley 
31129318fe57SMatthew G. Knepley   Input Parameters:
3113a1cb98faSBarry Smith + comm        - The communicator for the `DM` object
31149318fe57SMatthew G. Knepley . n           - The number of wedges around the origin
31159318fe57SMatthew G. Knepley - interpolate - Create edges and faces
31169318fe57SMatthew G. Knepley 
31179318fe57SMatthew G. Knepley   Output Parameter:
3118a1cb98faSBarry Smith . dm - The `DM` object
31199318fe57SMatthew G. Knepley 
31209318fe57SMatthew G. Knepley   Level: beginner
31219318fe57SMatthew G. Knepley 
31221cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateHexCylinderMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
31239318fe57SMatthew G. Knepley @*/
3124d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateWedgeCylinderMesh(MPI_Comm comm, PetscInt n, PetscBool interpolate, DM *dm)
3125d71ae5a4SJacob Faibussowitsch {
31269318fe57SMatthew G. Knepley   PetscFunctionBegin;
31274f572ea9SToby Isaac   PetscAssertPointer(dm, 4);
31289566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
31299566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
31309566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(*dm, n, interpolate));
31313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
313224119c2aSMatthew G. Knepley }
313324119c2aSMatthew G. Knepley 
3134d71ae5a4SJacob Faibussowitsch static inline PetscReal DiffNormReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
3135d71ae5a4SJacob Faibussowitsch {
313665a81367SMatthew G. Knepley   PetscReal prod = 0.0;
313765a81367SMatthew G. Knepley   PetscInt  i;
313865a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += PetscSqr(x[i] - y[i]);
313965a81367SMatthew G. Knepley   return PetscSqrtReal(prod);
314065a81367SMatthew G. Knepley }
3141dd2b43ebSStefano Zampini 
3142d71ae5a4SJacob Faibussowitsch static inline PetscReal DotReal(PetscInt dim, const PetscReal x[], const PetscReal y[])
3143d71ae5a4SJacob Faibussowitsch {
314465a81367SMatthew G. Knepley   PetscReal prod = 0.0;
314565a81367SMatthew G. Knepley   PetscInt  i;
314665a81367SMatthew G. Knepley   for (i = 0; i < dim; ++i) prod += x[i] * y[i];
314765a81367SMatthew G. Knepley   return prod;
314865a81367SMatthew G. Knepley }
314965a81367SMatthew G. Knepley 
315051a74b61SMatthew G. Knepley /* The first constant is the sphere radius */
3151d71ae5a4SJacob 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[])
3152d71ae5a4SJacob Faibussowitsch {
315351a74b61SMatthew G. Knepley   PetscReal r     = PetscRealPart(constants[0]);
315451a74b61SMatthew G. Knepley   PetscReal norm2 = 0.0, fac;
315551a74b61SMatthew G. Knepley   PetscInt  n     = uOff[1] - uOff[0], d;
315651a74b61SMatthew G. Knepley 
315751a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) norm2 += PetscSqr(PetscRealPart(u[d]));
315851a74b61SMatthew G. Knepley   fac = r / PetscSqrtReal(norm2);
315951a74b61SMatthew G. Knepley   for (d = 0; d < n; ++d) f0[d] = u[d] * fac;
316051a74b61SMatthew G. Knepley }
316151a74b61SMatthew G. Knepley 
3162d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateSphereMesh_Internal(DM dm, PetscInt dim, PetscBool simplex, PetscReal R)
3163d71ae5a4SJacob Faibussowitsch {
316465a81367SMatthew G. Knepley   const PetscInt embedDim = dim + 1;
316565a81367SMatthew G. Knepley   PetscSection   coordSection;
316665a81367SMatthew G. Knepley   Vec            coordinates;
316765a81367SMatthew G. Knepley   PetscScalar   *coords;
316865a81367SMatthew G. Knepley   PetscReal     *coordsIn;
316907c565c5SJose E. Roman   PetscInt       numCells, numEdges, numVerts = 0, firstVertex = 0, v, firstEdge, coordSize, d, e;
317065a81367SMatthew G. Knepley   PetscMPIInt    rank;
317165a81367SMatthew G. Knepley 
317265a81367SMatthew G. Knepley   PetscFunctionBegin;
31739318fe57SMatthew G. Knepley   PetscValidLogicalCollectiveBool(dm, simplex, 3);
317446139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
31759566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, dim));
31769566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, dim + 1));
31779566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
317865a81367SMatthew G. Knepley   switch (dim) {
31795c344501SMatthew G. Knepley   case 1:
31805c344501SMatthew G. Knepley     numCells = 16;
31815c344501SMatthew G. Knepley     numVerts = numCells;
31825c344501SMatthew G. Knepley 
31835c344501SMatthew G. Knepley     // Build Topology
31845c344501SMatthew G. Knepley     PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
31855c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
31865c344501SMatthew G. Knepley     PetscCall(DMSetUp(dm));
31875c344501SMatthew G. Knepley     for (PetscInt c = 0; c < numCells; ++c) {
31885c344501SMatthew G. Knepley       PetscInt cone[2];
31895c344501SMatthew G. Knepley 
31905c344501SMatthew G. Knepley       cone[0] = c + numCells;
31915c344501SMatthew G. Knepley       cone[1] = (c + 1) % numVerts + numCells;
31925c344501SMatthew G. Knepley       PetscCall(DMPlexSetCone(dm, c, cone));
31935c344501SMatthew G. Knepley     }
31945c344501SMatthew G. Knepley     PetscCall(DMPlexSymmetrize(dm));
31955c344501SMatthew G. Knepley     PetscCall(DMPlexStratify(dm));
31965c344501SMatthew G. Knepley     PetscCall(PetscMalloc1(numVerts * embedDim, &coordsIn));
31975c344501SMatthew G. Knepley     for (PetscInt v = 0; v < numVerts; ++v) {
31985c344501SMatthew G. Knepley       const PetscReal rad = 2. * PETSC_PI * v / numVerts;
31995c344501SMatthew G. Knepley 
32005c344501SMatthew G. Knepley       coordsIn[v * embedDim + 0] = PetscCosReal(rad);
32015c344501SMatthew G. Knepley       coordsIn[v * embedDim + 1] = PetscSinReal(rad);
32025c344501SMatthew G. Knepley     }
32035c344501SMatthew G. Knepley     break;
320465a81367SMatthew G. Knepley   case 2:
320565a81367SMatthew G. Knepley     if (simplex) {
320651a74b61SMatthew G. Knepley       const PetscReal radius    = PetscSqrtReal(1 + PETSC_PHI * PETSC_PHI) / (1.0 + PETSC_PHI);
320751a74b61SMatthew G. Knepley       const PetscReal edgeLen   = 2.0 / (1.0 + PETSC_PHI) * (R / radius);
320865a81367SMatthew G. Knepley       const PetscInt  degree    = 5;
320951a74b61SMatthew G. Knepley       PetscReal       vertex[3] = {0.0, 1.0 / (1.0 + PETSC_PHI), PETSC_PHI / (1.0 + PETSC_PHI)};
321065a81367SMatthew G. Knepley       PetscInt        s[3]      = {1, 1, 1};
321165a81367SMatthew G. Knepley       PetscInt        cone[3];
321207c565c5SJose E. Roman       PetscInt       *graph;
321365a81367SMatthew G. Knepley 
32149371c9d4SSatish Balay       vertex[0] *= R / radius;
32159371c9d4SSatish Balay       vertex[1] *= R / radius;
32169371c9d4SSatish Balay       vertex[2] *= R / radius;
3217dd400576SPatrick Sanan       numCells    = rank == 0 ? 20 : 0;
3218dd400576SPatrick Sanan       numVerts    = rank == 0 ? 12 : 0;
321965a81367SMatthew G. Knepley       firstVertex = numCells;
322051a74b61SMatthew G. Knepley       /* Use icosahedron, which for a R-sphere has coordinates which are all cyclic permutations of
322165a81367SMatthew G. Knepley 
322265a81367SMatthew G. Knepley            (0, \pm 1/\phi+1, \pm \phi/\phi+1)
322365a81367SMatthew G. Knepley 
322465a81367SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
322551a74b61SMatthew G. Knepley          length is then given by 2/(1+\phi) = 2 * 0.38197 = 0.76393.
322665a81367SMatthew G. Knepley       */
322765a81367SMatthew G. Knepley       /* Construct vertices */
32289566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
3229dd400576SPatrick Sanan       if (rank == 0) {
323007c565c5SJose E. Roman         for (PetscInt p = 0, i = 0; p < embedDim; ++p) {
323165a81367SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
323265a81367SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
323365a81367SMatthew G. Knepley               for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertex[(d + p) % embedDim];
323465a81367SMatthew G. Knepley               ++i;
323565a81367SMatthew G. Knepley             }
323665a81367SMatthew G. Knepley           }
323765a81367SMatthew G. Knepley         }
323845da822fSValeria Barra       }
323965a81367SMatthew G. Knepley       /* Construct graph */
32409566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
324107c565c5SJose E. Roman       for (PetscInt i = 0; i < numVerts; ++i) {
324207c565c5SJose E. Roman         PetscInt k = 0;
324307c565c5SJose E. Roman         for (PetscInt j = 0; j < numVerts; ++j) {
32449371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
32459371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
32469371c9d4SSatish Balay             ++k;
32479371c9d4SSatish Balay           }
324865a81367SMatthew G. Knepley         }
324963a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid icosahedron, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
325065a81367SMatthew G. Knepley       }
325165a81367SMatthew G. Knepley       /* Build Topology */
32529566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
325307c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
32549566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
325565a81367SMatthew G. Knepley       /* Cells */
325607c565c5SJose E. Roman       for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
325707c565c5SJose E. Roman         for (PetscInt j = 0; j < i; ++j) {
325807c565c5SJose E. Roman           for (PetscInt k = 0; k < j; ++k) {
325965a81367SMatthew G. Knepley             if (graph[i * numVerts + j] && graph[j * numVerts + k] && graph[k * numVerts + i]) {
32609371c9d4SSatish Balay               cone[0] = firstVertex + i;
32619371c9d4SSatish Balay               cone[1] = firstVertex + j;
32629371c9d4SSatish Balay               cone[2] = firstVertex + k;
326365a81367SMatthew G. Knepley               /* Check orientation */
326465a81367SMatthew G. Knepley               {
32659371c9d4SSatish Balay                 const PetscInt epsilon[3][3][3] = {
32669371c9d4SSatish Balay                   {{0, 0, 0},  {0, 0, 1},  {0, -1, 0}},
32679371c9d4SSatish Balay                   {{0, 0, -1}, {0, 0, 0},  {1, 0, 0} },
32689371c9d4SSatish Balay                   {{0, 1, 0},  {-1, 0, 0}, {0, 0, 0} }
32699371c9d4SSatish Balay                 };
327065a81367SMatthew G. Knepley                 PetscReal normal[3];
327165a81367SMatthew G. Knepley                 PetscInt  e, f;
327265a81367SMatthew G. Knepley 
327365a81367SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) {
327465a81367SMatthew G. Knepley                   normal[d] = 0.0;
327565a81367SMatthew G. Knepley                   for (e = 0; e < embedDim; ++e) {
3276ad540459SPierre 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]);
327765a81367SMatthew G. Knepley                   }
327865a81367SMatthew G. Knepley                 }
32799371c9d4SSatish Balay                 if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
32809371c9d4SSatish Balay                   PetscInt tmp = cone[1];
32819371c9d4SSatish Balay                   cone[1]      = cone[2];
32829371c9d4SSatish Balay                   cone[2]      = tmp;
328365a81367SMatthew G. Knepley                 }
328465a81367SMatthew G. Knepley               }
32859566063dSJacob Faibussowitsch               PetscCall(DMPlexSetCone(dm, c++, cone));
328665a81367SMatthew G. Knepley             }
328765a81367SMatthew G. Knepley           }
328865a81367SMatthew G. Knepley         }
328965a81367SMatthew G. Knepley       }
32909566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
32919566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
32929566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
329365a81367SMatthew G. Knepley     } else {
32942829fed8SMatthew G. Knepley       /*
32952829fed8SMatthew G. Knepley         12-21--13
32962829fed8SMatthew G. Knepley          |     |
32972829fed8SMatthew G. Knepley         25  4  24
32982829fed8SMatthew G. Knepley          |     |
32992829fed8SMatthew G. Knepley   12-25--9-16--8-24--13
33002829fed8SMatthew G. Knepley    |     |     |     |
33012829fed8SMatthew G. Knepley   23  5 17  0 15  3  22
33022829fed8SMatthew G. Knepley    |     |     |     |
33032829fed8SMatthew G. Knepley   10-20--6-14--7-19--11
33042829fed8SMatthew G. Knepley          |     |
33052829fed8SMatthew G. Knepley         20  1  19
33062829fed8SMatthew G. Knepley          |     |
33072829fed8SMatthew G. Knepley         10-18--11
33082829fed8SMatthew G. Knepley          |     |
33092829fed8SMatthew G. Knepley         23  2  22
33102829fed8SMatthew G. Knepley          |     |
33112829fed8SMatthew G. Knepley         12-21--13
33122829fed8SMatthew G. Knepley        */
33132829fed8SMatthew G. Knepley       PetscInt cone[4], ornt[4];
33142829fed8SMatthew G. Knepley 
3315dd400576SPatrick Sanan       numCells    = rank == 0 ? 6 : 0;
3316dd400576SPatrick Sanan       numEdges    = rank == 0 ? 12 : 0;
3317dd400576SPatrick Sanan       numVerts    = rank == 0 ? 8 : 0;
331865a81367SMatthew G. Knepley       firstVertex = numCells;
331965a81367SMatthew G. Knepley       firstEdge   = numCells + numVerts;
33202829fed8SMatthew G. Knepley       /* Build Topology */
33219566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numEdges + numVerts));
332207c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, 4));
332348a46eb9SPierre Jolivet       for (e = firstEdge; e < firstEdge + numEdges; ++e) PetscCall(DMPlexSetConeSize(dm, e, 2));
33249566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
3325dd400576SPatrick Sanan       if (rank == 0) {
33262829fed8SMatthew G. Knepley         /* Cell 0 */
33279371c9d4SSatish Balay         cone[0] = 14;
33289371c9d4SSatish Balay         cone[1] = 15;
33299371c9d4SSatish Balay         cone[2] = 16;
33309371c9d4SSatish Balay         cone[3] = 17;
33319566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 0, cone));
33329371c9d4SSatish Balay         ornt[0] = 0;
33339371c9d4SSatish Balay         ornt[1] = 0;
33349371c9d4SSatish Balay         ornt[2] = 0;
33359371c9d4SSatish Balay         ornt[3] = 0;
33369566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 0, ornt));
33372829fed8SMatthew G. Knepley         /* Cell 1 */
33389371c9d4SSatish Balay         cone[0] = 18;
33399371c9d4SSatish Balay         cone[1] = 19;
33409371c9d4SSatish Balay         cone[2] = 14;
33419371c9d4SSatish Balay         cone[3] = 20;
33429566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 1, cone));
33439371c9d4SSatish Balay         ornt[0] = 0;
33449371c9d4SSatish Balay         ornt[1] = 0;
33459371c9d4SSatish Balay         ornt[2] = -1;
33469371c9d4SSatish Balay         ornt[3] = 0;
33479566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 1, ornt));
33482829fed8SMatthew G. Knepley         /* Cell 2 */
33499371c9d4SSatish Balay         cone[0] = 21;
33509371c9d4SSatish Balay         cone[1] = 22;
33519371c9d4SSatish Balay         cone[2] = 18;
33529371c9d4SSatish Balay         cone[3] = 23;
33539566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 2, cone));
33549371c9d4SSatish Balay         ornt[0] = 0;
33559371c9d4SSatish Balay         ornt[1] = 0;
33569371c9d4SSatish Balay         ornt[2] = -1;
33579371c9d4SSatish Balay         ornt[3] = 0;
33589566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 2, ornt));
33592829fed8SMatthew G. Knepley         /* Cell 3 */
33609371c9d4SSatish Balay         cone[0] = 19;
33619371c9d4SSatish Balay         cone[1] = 22;
33629371c9d4SSatish Balay         cone[2] = 24;
33639371c9d4SSatish Balay         cone[3] = 15;
33649566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 3, cone));
33659371c9d4SSatish Balay         ornt[0] = -1;
33669371c9d4SSatish Balay         ornt[1] = -1;
33679371c9d4SSatish Balay         ornt[2] = 0;
33689371c9d4SSatish Balay         ornt[3] = -1;
33699566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 3, ornt));
33702829fed8SMatthew G. Knepley         /* Cell 4 */
33719371c9d4SSatish Balay         cone[0] = 16;
33729371c9d4SSatish Balay         cone[1] = 24;
33739371c9d4SSatish Balay         cone[2] = 21;
33749371c9d4SSatish Balay         cone[3] = 25;
33759566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 4, cone));
33769371c9d4SSatish Balay         ornt[0] = -1;
33779371c9d4SSatish Balay         ornt[1] = -1;
33789371c9d4SSatish Balay         ornt[2] = -1;
33799371c9d4SSatish Balay         ornt[3] = 0;
33809566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 4, ornt));
33812829fed8SMatthew G. Knepley         /* Cell 5 */
33829371c9d4SSatish Balay         cone[0] = 20;
33839371c9d4SSatish Balay         cone[1] = 17;
33849371c9d4SSatish Balay         cone[2] = 25;
33859371c9d4SSatish Balay         cone[3] = 23;
33869566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 5, cone));
33879371c9d4SSatish Balay         ornt[0] = -1;
33889371c9d4SSatish Balay         ornt[1] = -1;
33899371c9d4SSatish Balay         ornt[2] = -1;
33909371c9d4SSatish Balay         ornt[3] = -1;
33919566063dSJacob Faibussowitsch         PetscCall(DMPlexSetConeOrientation(dm, 5, ornt));
33922829fed8SMatthew G. Knepley         /* Edges */
33939371c9d4SSatish Balay         cone[0] = 6;
33949371c9d4SSatish Balay         cone[1] = 7;
33959566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 14, cone));
33969371c9d4SSatish Balay         cone[0] = 7;
33979371c9d4SSatish Balay         cone[1] = 8;
33989566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 15, cone));
33999371c9d4SSatish Balay         cone[0] = 8;
34009371c9d4SSatish Balay         cone[1] = 9;
34019566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 16, cone));
34029371c9d4SSatish Balay         cone[0] = 9;
34039371c9d4SSatish Balay         cone[1] = 6;
34049566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 17, cone));
34059371c9d4SSatish Balay         cone[0] = 10;
34069371c9d4SSatish Balay         cone[1] = 11;
34079566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 18, cone));
34089371c9d4SSatish Balay         cone[0] = 11;
34099371c9d4SSatish Balay         cone[1] = 7;
34109566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 19, cone));
34119371c9d4SSatish Balay         cone[0] = 6;
34129371c9d4SSatish Balay         cone[1] = 10;
34139566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 20, cone));
34149371c9d4SSatish Balay         cone[0] = 12;
34159371c9d4SSatish Balay         cone[1] = 13;
34169566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 21, cone));
34179371c9d4SSatish Balay         cone[0] = 13;
34189371c9d4SSatish Balay         cone[1] = 11;
34199566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 22, cone));
34209371c9d4SSatish Balay         cone[0] = 10;
34219371c9d4SSatish Balay         cone[1] = 12;
34229566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 23, cone));
34239371c9d4SSatish Balay         cone[0] = 13;
34249371c9d4SSatish Balay         cone[1] = 8;
34259566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 24, cone));
34269371c9d4SSatish Balay         cone[0] = 12;
34279371c9d4SSatish Balay         cone[1] = 9;
34289566063dSJacob Faibussowitsch         PetscCall(DMPlexSetCone(dm, 25, cone));
342945da822fSValeria Barra       }
34309566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
34319566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
34322829fed8SMatthew G. Knepley       /* Build coordinates */
34339566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
3434dd400576SPatrick Sanan       if (rank == 0) {
34359371c9d4SSatish Balay         coordsIn[0 * embedDim + 0] = -R;
34369371c9d4SSatish Balay         coordsIn[0 * embedDim + 1] = R;
34379371c9d4SSatish Balay         coordsIn[0 * embedDim + 2] = -R;
34389371c9d4SSatish Balay         coordsIn[1 * embedDim + 0] = R;
34399371c9d4SSatish Balay         coordsIn[1 * embedDim + 1] = R;
34409371c9d4SSatish Balay         coordsIn[1 * embedDim + 2] = -R;
34419371c9d4SSatish Balay         coordsIn[2 * embedDim + 0] = R;
34429371c9d4SSatish Balay         coordsIn[2 * embedDim + 1] = -R;
34439371c9d4SSatish Balay         coordsIn[2 * embedDim + 2] = -R;
34449371c9d4SSatish Balay         coordsIn[3 * embedDim + 0] = -R;
34459371c9d4SSatish Balay         coordsIn[3 * embedDim + 1] = -R;
34469371c9d4SSatish Balay         coordsIn[3 * embedDim + 2] = -R;
34479371c9d4SSatish Balay         coordsIn[4 * embedDim + 0] = -R;
34489371c9d4SSatish Balay         coordsIn[4 * embedDim + 1] = R;
34499371c9d4SSatish Balay         coordsIn[4 * embedDim + 2] = R;
34509371c9d4SSatish Balay         coordsIn[5 * embedDim + 0] = R;
34519371c9d4SSatish Balay         coordsIn[5 * embedDim + 1] = R;
34529371c9d4SSatish Balay         coordsIn[5 * embedDim + 2] = R;
34539371c9d4SSatish Balay         coordsIn[6 * embedDim + 0] = -R;
34549371c9d4SSatish Balay         coordsIn[6 * embedDim + 1] = -R;
34559371c9d4SSatish Balay         coordsIn[6 * embedDim + 2] = R;
34569371c9d4SSatish Balay         coordsIn[7 * embedDim + 0] = R;
34579371c9d4SSatish Balay         coordsIn[7 * embedDim + 1] = -R;
34589371c9d4SSatish Balay         coordsIn[7 * embedDim + 2] = R;
345965a81367SMatthew G. Knepley       }
346045da822fSValeria Barra     }
346165a81367SMatthew G. Knepley     break;
346265a81367SMatthew G. Knepley   case 3:
3463116ded15SMatthew G. Knepley     if (simplex) {
3464116ded15SMatthew G. Knepley       const PetscReal edgeLen         = 1.0 / PETSC_PHI;
346551a74b61SMatthew G. Knepley       PetscReal       vertexA[4]      = {0.5, 0.5, 0.5, 0.5};
346651a74b61SMatthew G. Knepley       PetscReal       vertexB[4]      = {1.0, 0.0, 0.0, 0.0};
346751a74b61SMatthew G. Knepley       PetscReal       vertexC[4]      = {0.5, 0.5 * PETSC_PHI, 0.5 / PETSC_PHI, 0.0};
3468116ded15SMatthew G. Knepley       const PetscInt  degree          = 12;
3469116ded15SMatthew G. Knepley       PetscInt        s[4]            = {1, 1, 1};
34709371c9d4SSatish Balay       PetscInt        evenPerm[12][4] = {
34719371c9d4SSatish Balay         {0, 1, 2, 3},
34729371c9d4SSatish Balay         {0, 2, 3, 1},
34739371c9d4SSatish Balay         {0, 3, 1, 2},
34749371c9d4SSatish Balay         {1, 0, 3, 2},
34759371c9d4SSatish Balay         {1, 2, 0, 3},
34769371c9d4SSatish Balay         {1, 3, 2, 0},
34779371c9d4SSatish Balay         {2, 0, 1, 3},
34789371c9d4SSatish Balay         {2, 1, 3, 0},
34799371c9d4SSatish Balay         {2, 3, 0, 1},
34809371c9d4SSatish Balay         {3, 0, 2, 1},
34819371c9d4SSatish Balay         {3, 1, 0, 2},
34829371c9d4SSatish Balay         {3, 2, 1, 0}
34839371c9d4SSatish Balay       };
3484116ded15SMatthew G. Knepley       PetscInt  cone[4];
3485116ded15SMatthew G. Knepley       PetscInt *graph, p, i, j, k, l;
3486116ded15SMatthew G. Knepley 
34879371c9d4SSatish Balay       vertexA[0] *= R;
34889371c9d4SSatish Balay       vertexA[1] *= R;
34899371c9d4SSatish Balay       vertexA[2] *= R;
34909371c9d4SSatish Balay       vertexA[3] *= R;
34919371c9d4SSatish Balay       vertexB[0] *= R;
34929371c9d4SSatish Balay       vertexB[1] *= R;
34939371c9d4SSatish Balay       vertexB[2] *= R;
34949371c9d4SSatish Balay       vertexB[3] *= R;
34959371c9d4SSatish Balay       vertexC[0] *= R;
34969371c9d4SSatish Balay       vertexC[1] *= R;
34979371c9d4SSatish Balay       vertexC[2] *= R;
34989371c9d4SSatish Balay       vertexC[3] *= R;
3499dd400576SPatrick Sanan       numCells    = rank == 0 ? 600 : 0;
3500dd400576SPatrick Sanan       numVerts    = rank == 0 ? 120 : 0;
3501116ded15SMatthew G. Knepley       firstVertex = numCells;
3502116ded15SMatthew G. Knepley       /* Use the 600-cell, which for a unit sphere has coordinates which are
3503116ded15SMatthew G. Knepley 
3504116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm 1,    \pm 1, \pm 1)                          16
3505116ded15SMatthew G. Knepley                (\pm 1,    0,       0,      0)  all cyclic permutations   8
3506116ded15SMatthew G. Knepley            1/2 (\pm 1, \pm phi, \pm 1/phi, 0)  all even permutations    96
3507116ded15SMatthew G. Knepley 
3508116ded15SMatthew G. Knepley          where \phi^2 - \phi - 1 = 0, meaning \phi is the golden ratio \frac{1 + \sqrt{5}}{2}. The edge
35096333ae4fSvaleriabarra          length is then given by 1/\phi = 0.61803.
3510116ded15SMatthew G. Knepley 
3511116ded15SMatthew G. Knepley          http://buzzard.pugetsound.edu/sage-practice/ch03s03.html
3512116ded15SMatthew G. Knepley          http://mathworld.wolfram.com/600-Cell.html
3513116ded15SMatthew G. Knepley       */
3514116ded15SMatthew G. Knepley       /* Construct vertices */
35159566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * embedDim, &coordsIn));
3516116ded15SMatthew G. Knepley       i = 0;
3517dd400576SPatrick Sanan       if (rank == 0) {
3518116ded15SMatthew G. Knepley         for (s[0] = -1; s[0] < 2; s[0] += 2) {
3519116ded15SMatthew G. Knepley           for (s[1] = -1; s[1] < 2; s[1] += 2) {
3520116ded15SMatthew G. Knepley             for (s[2] = -1; s[2] < 2; s[2] += 2) {
3521116ded15SMatthew G. Knepley               for (s[3] = -1; s[3] < 2; s[3] += 2) {
3522116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[d] * vertexA[d];
3523116ded15SMatthew G. Knepley                 ++i;
3524116ded15SMatthew G. Knepley               }
3525116ded15SMatthew G. Knepley             }
3526116ded15SMatthew G. Knepley           }
3527116ded15SMatthew G. Knepley         }
3528116ded15SMatthew G. Knepley         for (p = 0; p < embedDim; ++p) {
3529116ded15SMatthew G. Knepley           s[1] = s[2] = s[3] = 1;
3530116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
3531116ded15SMatthew G. Knepley             for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[(d + p) % embedDim] * vertexB[(d + p) % embedDim];
3532116ded15SMatthew G. Knepley             ++i;
3533116ded15SMatthew G. Knepley           }
3534116ded15SMatthew G. Knepley         }
3535116ded15SMatthew G. Knepley         for (p = 0; p < 12; ++p) {
3536116ded15SMatthew G. Knepley           s[3] = 1;
3537116ded15SMatthew G. Knepley           for (s[0] = -1; s[0] < 2; s[0] += 2) {
3538116ded15SMatthew G. Knepley             for (s[1] = -1; s[1] < 2; s[1] += 2) {
3539116ded15SMatthew G. Knepley               for (s[2] = -1; s[2] < 2; s[2] += 2) {
3540116ded15SMatthew G. Knepley                 for (d = 0; d < embedDim; ++d) coordsIn[i * embedDim + d] = s[evenPerm[p][d]] * vertexC[evenPerm[p][d]];
3541116ded15SMatthew G. Knepley                 ++i;
3542116ded15SMatthew G. Knepley               }
3543116ded15SMatthew G. Knepley             }
3544116ded15SMatthew G. Knepley           }
3545116ded15SMatthew G. Knepley         }
354645da822fSValeria Barra       }
354763a3b9bcSJacob Faibussowitsch       PetscCheck(i == numVerts, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertices %" PetscInt_FMT " != %" PetscInt_FMT, i, numVerts);
3548116ded15SMatthew G. Knepley       /* Construct graph */
35499566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numVerts * numVerts, &graph));
3550116ded15SMatthew G. Knepley       for (i = 0; i < numVerts; ++i) {
3551116ded15SMatthew G. Knepley         for (j = 0, k = 0; j < numVerts; ++j) {
35529371c9d4SSatish Balay           if (PetscAbsReal(DiffNormReal(embedDim, &coordsIn[i * embedDim], &coordsIn[j * embedDim]) - edgeLen) < PETSC_SMALL) {
35539371c9d4SSatish Balay             graph[i * numVerts + j] = 1;
35549371c9d4SSatish Balay             ++k;
35559371c9d4SSatish Balay           }
3556116ded15SMatthew G. Knepley         }
355763a3b9bcSJacob Faibussowitsch         PetscCheck(k == degree, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid 600-cell, vertex %" PetscInt_FMT " degree %" PetscInt_FMT " != %" PetscInt_FMT, i, k, degree);
3558116ded15SMatthew G. Knepley       }
3559116ded15SMatthew G. Knepley       /* Build Topology */
35609566063dSJacob Faibussowitsch       PetscCall(DMPlexSetChart(dm, 0, numCells + numVerts));
356107c565c5SJose E. Roman       for (PetscInt c = 0; c < numCells; c++) PetscCall(DMPlexSetConeSize(dm, c, embedDim));
35629566063dSJacob Faibussowitsch       PetscCall(DMSetUp(dm)); /* Allocate space for cones */
3563116ded15SMatthew G. Knepley       /* Cells */
3564dd400576SPatrick Sanan       if (rank == 0) {
356507c565c5SJose E. Roman         for (PetscInt i = 0, c = 0; i < numVerts; ++i) {
3566116ded15SMatthew G. Knepley           for (j = 0; j < i; ++j) {
3567116ded15SMatthew G. Knepley             for (k = 0; k < j; ++k) {
3568116ded15SMatthew G. Knepley               for (l = 0; l < k; ++l) {
35699371c9d4SSatish 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]) {
35709371c9d4SSatish Balay                   cone[0] = firstVertex + i;
35719371c9d4SSatish Balay                   cone[1] = firstVertex + j;
35729371c9d4SSatish Balay                   cone[2] = firstVertex + k;
35739371c9d4SSatish Balay                   cone[3] = firstVertex + l;
3574116ded15SMatthew G. Knepley                   /* Check orientation: https://ef.gy/linear-algebra:normal-vectors-in-higher-dimensional-spaces */
3575116ded15SMatthew G. Knepley                   {
35769371c9d4SSatish Balay                     const PetscInt epsilon[4][4][4][4] = {
35779371c9d4SSatish 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}}},
3578116ded15SMatthew G. Knepley 
35799371c9d4SSatish 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}}},
3580116ded15SMatthew G. Knepley 
35819371c9d4SSatish 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}}},
3582116ded15SMatthew G. Knepley 
35839371c9d4SSatish 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}} }
35849371c9d4SSatish Balay                     };
3585116ded15SMatthew G. Knepley                     PetscReal normal[4];
3586116ded15SMatthew G. Knepley                     PetscInt  e, f, g;
3587116ded15SMatthew G. Knepley 
3588116ded15SMatthew G. Knepley                     for (d = 0; d < embedDim; ++d) {
3589116ded15SMatthew G. Knepley                       normal[d] = 0.0;
3590116ded15SMatthew G. Knepley                       for (e = 0; e < embedDim; ++e) {
3591116ded15SMatthew G. Knepley                         for (f = 0; f < embedDim; ++f) {
3592116ded15SMatthew G. Knepley                           for (g = 0; g < embedDim; ++g) {
3593116ded15SMatthew 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]);
3594116ded15SMatthew G. Knepley                           }
3595116ded15SMatthew G. Knepley                         }
3596116ded15SMatthew G. Knepley                       }
3597116ded15SMatthew G. Knepley                     }
35989371c9d4SSatish Balay                     if (DotReal(embedDim, normal, &coordsIn[i * embedDim]) < 0) {
35999371c9d4SSatish Balay                       PetscInt tmp = cone[1];
36009371c9d4SSatish Balay                       cone[1]      = cone[2];
36019371c9d4SSatish Balay                       cone[2]      = tmp;
36029371c9d4SSatish Balay                     }
3603116ded15SMatthew G. Knepley                   }
36049566063dSJacob Faibussowitsch                   PetscCall(DMPlexSetCone(dm, c++, cone));
3605116ded15SMatthew G. Knepley                 }
3606116ded15SMatthew G. Knepley               }
3607116ded15SMatthew G. Knepley             }
3608116ded15SMatthew G. Knepley           }
3609116ded15SMatthew G. Knepley         }
361045da822fSValeria Barra       }
36119566063dSJacob Faibussowitsch       PetscCall(DMPlexSymmetrize(dm));
36129566063dSJacob Faibussowitsch       PetscCall(DMPlexStratify(dm));
36139566063dSJacob Faibussowitsch       PetscCall(PetscFree(graph));
3614116ded15SMatthew G. Knepley     }
3615f4d061e9SPierre Jolivet     break;
3616d71ae5a4SJacob Faibussowitsch   default:
3617d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension for sphere: %" PetscInt_FMT, dim);
361865a81367SMatthew G. Knepley   }
361965a81367SMatthew G. Knepley   /* Create coordinates */
36209566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
36219566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
36229566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, embedDim));
36239566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numVerts));
36242829fed8SMatthew G. Knepley   for (v = firstVertex; v < firstVertex + numVerts; ++v) {
36259566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, embedDim));
36269566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, embedDim));
36272829fed8SMatthew G. Knepley   }
36289566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
36299566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
36309566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
36319566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, embedDim));
36329566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
36339566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
36349566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
36359566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
36369371c9d4SSatish Balay   for (v = 0; v < numVerts; ++v)
3637ad540459SPierre Jolivet     for (d = 0; d < embedDim; ++d) coords[v * embedDim + d] = coordsIn[v * embedDim + d];
36389566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
36399566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
36409566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
36419566063dSJacob Faibussowitsch   PetscCall(PetscFree(coordsIn));
364251a74b61SMatthew G. Knepley   {
364351a74b61SMatthew G. Knepley     DM          cdm;
364451a74b61SMatthew G. Knepley     PetscDS     cds;
36459318fe57SMatthew G. Knepley     PetscScalar c = R;
364651a74b61SMatthew G. Knepley 
3647509b31aaSMatthew G. Knepley     PetscCall(DMPlexSetCoordinateMap(dm, snapToSphere));
3648509b31aaSMatthew G. Knepley     PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
36499566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
36509566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
36519566063dSJacob Faibussowitsch     PetscCall(PetscDSSetConstants(cds, 1, &c));
365251a74b61SMatthew G. Knepley   }
365346139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
36549318fe57SMatthew G. Knepley   /* Wait for coordinate creation before doing in-place modification */
36559566063dSJacob Faibussowitsch   if (simplex) PetscCall(DMPlexInterpolateInPlace_Internal(dm));
36563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
36579318fe57SMatthew G. Knepley }
36589318fe57SMatthew G. Knepley 
3659b7f5c055SJed Brown typedef void (*TPSEvaluateFunc)(const PetscReal[], PetscReal *, PetscReal[], PetscReal (*)[3]);
3660b7f5c055SJed Brown 
3661b7f5c055SJed Brown /*
3662b7f5c055SJed Brown  The Schwarz P implicit surface is
3663b7f5c055SJed Brown 
3664b7f5c055SJed Brown      f(x) = cos(x0) + cos(x1) + cos(x2) = 0
3665b7f5c055SJed Brown */
3666d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_SchwarzP(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3667d71ae5a4SJacob Faibussowitsch {
3668b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(y[0] * PETSC_PI), PetscCosReal(y[1] * PETSC_PI), PetscCosReal(y[2] * PETSC_PI)};
3669b7f5c055SJed Brown   PetscReal g[3] = {-PetscSinReal(y[0] * PETSC_PI), -PetscSinReal(y[1] * PETSC_PI), -PetscSinReal(y[2] * PETSC_PI)};
3670b7f5c055SJed Brown   f[0]           = c[0] + c[1] + c[2];
3671b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3672b7f5c055SJed Brown     grad[i] = PETSC_PI * g[i];
3673ad540459SPierre Jolivet     for (PetscInt j = 0; j < 3; j++) hess[i][j] = (i == j) ? -PetscSqr(PETSC_PI) * c[i] : 0.;
3674b7f5c055SJed Brown   }
3675b7f5c055SJed Brown }
3676b7f5c055SJed Brown 
36774663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3678d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_SchwarzP(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3679d71ae5a4SJacob Faibussowitsch {
3680ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) u[i] = -PETSC_PI * PetscSinReal(x[i] * PETSC_PI);
36813ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
36824663dae6SJed Brown }
36834663dae6SJed Brown 
3684b7f5c055SJed Brown /*
3685b7f5c055SJed Brown  The Gyroid implicit surface is
3686b7f5c055SJed Brown 
3687b7f5c055SJed 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)
3688b7f5c055SJed Brown 
3689b7f5c055SJed Brown */
3690d71ae5a4SJacob Faibussowitsch static void TPSEvaluate_Gyroid(const PetscReal y[3], PetscReal *f, PetscReal grad[], PetscReal (*hess)[3])
3691d71ae5a4SJacob Faibussowitsch {
3692b7f5c055SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * y[0]), PetscSinReal(PETSC_PI * (y[1] + .5)), PetscSinReal(PETSC_PI * (y[2] + .25))};
3693b7f5c055SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * y[0]), PetscCosReal(PETSC_PI * (y[1] + .5)), PetscCosReal(PETSC_PI * (y[2] + .25))};
3694b7f5c055SJed Brown   f[0]           = s[0] * c[1] + s[1] * c[2] + s[2] * c[0];
3695b7f5c055SJed Brown   grad[0]        = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
3696b7f5c055SJed Brown   grad[1]        = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
3697b7f5c055SJed Brown   grad[2]        = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
3698b7f5c055SJed Brown   hess[0][0]     = -PetscSqr(PETSC_PI) * (s[0] * c[1] + s[2] * c[0]);
3699b7f5c055SJed Brown   hess[0][1]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3700b7f5c055SJed Brown   hess[0][2]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3701b7f5c055SJed Brown   hess[1][0]     = -PetscSqr(PETSC_PI) * (s[1] * c[2] + s[0] * c[1]);
3702b7f5c055SJed Brown   hess[1][1]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3703b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[0] * s[1]);
3704b7f5c055SJed Brown   hess[2][0]     = -PetscSqr(PETSC_PI) * (s[2] * c[0] + s[1] * c[2]);
3705b7f5c055SJed Brown   hess[2][1]     = -PetscSqr(PETSC_PI) * (c[2] * s[0]);
3706b7f5c055SJed Brown   hess[2][2]     = -PetscSqr(PETSC_PI) * (c[1] * s[2]);
3707b7f5c055SJed Brown }
3708b7f5c055SJed Brown 
37094663dae6SJed Brown // u[] is a tentative normal on input. Replace with the implicit function gradient in the same direction
3710d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSExtrudeNormalFunc_Gyroid(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt r, PetscScalar u[], void *ctx)
3711d71ae5a4SJacob Faibussowitsch {
37124663dae6SJed Brown   PetscReal s[3] = {PetscSinReal(PETSC_PI * x[0]), PetscSinReal(PETSC_PI * (x[1] + .5)), PetscSinReal(PETSC_PI * (x[2] + .25))};
37134663dae6SJed Brown   PetscReal c[3] = {PetscCosReal(PETSC_PI * x[0]), PetscCosReal(PETSC_PI * (x[1] + .5)), PetscCosReal(PETSC_PI * (x[2] + .25))};
37144663dae6SJed Brown   u[0]           = PETSC_PI * (c[0] * c[1] - s[2] * s[0]);
37154663dae6SJed Brown   u[1]           = PETSC_PI * (c[1] * c[2] - s[0] * s[1]);
37164663dae6SJed Brown   u[2]           = PETSC_PI * (c[2] * c[0] - s[1] * s[2]);
37173ba16761SJacob Faibussowitsch   return PETSC_SUCCESS;
37184663dae6SJed Brown }
37194663dae6SJed Brown 
3720b7f5c055SJed Brown /*
3721b7f5c055SJed Brown    We wish to solve
3722b7f5c055SJed Brown 
3723b7f5c055SJed Brown          min_y || y - x ||^2  subject to f(y) = 0
3724b7f5c055SJed Brown 
3725b7f5c055SJed Brown    Let g(y) = grad(f).  The minimization problem is equivalent to asking to satisfy
3726b7f5c055SJed 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
3727b7f5c055SJed Brown    tangent space and ask for both components in the tangent space to be zero.
3728b7f5c055SJed Brown 
3729b7f5c055SJed Brown    Take g to be a column vector and compute the "full QR" factorization Q R = g,
3730b7f5c055SJed Brown    where Q = I - 2 n n^T is a symmetric orthogonal matrix.
3731b7f5c055SJed Brown    The first column of Q is parallel to g so the remaining two columns span the null space.
3732b7f5c055SJed Brown    Let Qn = Q[:,1:] be those remaining columns.  Then Qn Qn^T is an orthogonal projector into the tangent space.
3733da81f932SPierre Jolivet    Since Q is symmetric, this is equivalent to multiplying by Q and taking the last two entries.
3734b7f5c055SJed Brown    In total, we have a system of 3 equations in 3 unknowns:
3735b7f5c055SJed Brown 
3736b7f5c055SJed Brown      f(y) = 0                       1 equation
3737b7f5c055SJed Brown      Qn^T (y - x) = 0               2 equations
3738b7f5c055SJed Brown 
3739b7f5c055SJed Brown    Here, we compute the residual and Jacobian of this system.
3740b7f5c055SJed Brown */
3741d71ae5a4SJacob Faibussowitsch static void TPSNearestPointResJac(TPSEvaluateFunc feval, const PetscScalar x[], const PetscScalar y[], PetscScalar res[], PetscScalar J[])
3742d71ae5a4SJacob Faibussowitsch {
3743b7f5c055SJed Brown   PetscReal yreal[3] = {PetscRealPart(y[0]), PetscRealPart(y[1]), PetscRealPart(y[2])};
3744b7f5c055SJed Brown   PetscReal d[3]     = {PetscRealPart(y[0] - x[0]), PetscRealPart(y[1] - x[1]), PetscRealPart(y[2] - x[2])};
37452f0490c0SSatish Balay   PetscReal f, grad[3], n[3], norm, norm_y[3], nd, nd_y[3], sign;
37469371c9d4SSatish Balay   PetscReal n_y[3][3] = {
37479371c9d4SSatish Balay     {0, 0, 0},
37489371c9d4SSatish Balay     {0, 0, 0},
37499371c9d4SSatish Balay     {0, 0, 0}
37509371c9d4SSatish Balay   };
3751b7f5c055SJed Brown 
3752b7f5c055SJed Brown   feval(yreal, &f, grad, n_y);
3753b7f5c055SJed Brown 
3754b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n[i] = grad[i];
3755b7f5c055SJed Brown   norm = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3756ad540459SPierre Jolivet   for (PetscInt i = 0; i < 3; i++) norm_y[i] = 1. / norm * n[i] * n_y[i][i];
3757b7f5c055SJed Brown 
3758b7f5c055SJed Brown   // Define the Householder reflector
3759b7f5c055SJed Brown   sign = n[0] >= 0 ? 1. : -1.;
3760b7f5c055SJed Brown   n[0] += norm * sign;
3761b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) n_y[0][i] += norm_y[i] * sign;
3762b7f5c055SJed Brown 
3763b7f5c055SJed Brown   norm      = PetscSqrtReal(PetscSqr(n[0]) + PetscSqr(n[1]) + PetscSqr(n[2]));
3764b7f5c055SJed Brown   norm_y[0] = 1. / norm * (n[0] * n_y[0][0]);
3765b7f5c055SJed Brown   norm_y[1] = 1. / norm * (n[0] * n_y[0][1] + n[1] * n_y[1][1]);
3766b7f5c055SJed Brown   norm_y[2] = 1. / norm * (n[0] * n_y[0][2] + n[2] * n_y[2][2]);
3767b7f5c055SJed Brown 
3768b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) {
3769b7f5c055SJed Brown     n[i] /= norm;
3770b7f5c055SJed Brown     for (PetscInt j = 0; j < 3; j++) {
3771b7f5c055SJed Brown       // note that n[i] is n_old[i]/norm when executing the code below
3772b7f5c055SJed Brown       n_y[i][j] = n_y[i][j] / norm - n[i] / norm * norm_y[j];
3773b7f5c055SJed Brown     }
3774b7f5c055SJed Brown   }
3775b7f5c055SJed Brown 
3776b7f5c055SJed Brown   nd = n[0] * d[0] + n[1] * d[1] + n[2] * d[2];
3777b7f5c055SJed 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];
3778b7f5c055SJed Brown 
3779b7f5c055SJed Brown   res[0] = f;
3780b7f5c055SJed Brown   res[1] = d[1] - 2 * n[1] * nd;
3781b7f5c055SJed Brown   res[2] = d[2] - 2 * n[2] * nd;
3782b7f5c055SJed Brown   // J[j][i] is J_{ij} (column major)
3783b7f5c055SJed Brown   for (PetscInt j = 0; j < 3; j++) {
3784b7f5c055SJed Brown     J[0 + j * 3] = grad[j];
3785b7f5c055SJed Brown     J[1 + j * 3] = (j == 1) * 1. - 2 * (n_y[1][j] * nd + n[1] * nd_y[j]);
3786b7f5c055SJed Brown     J[2 + j * 3] = (j == 2) * 1. - 2 * (n_y[2][j] * nd + n[2] * nd_y[j]);
3787b7f5c055SJed Brown   }
3788b7f5c055SJed Brown }
3789b7f5c055SJed Brown 
3790b7f5c055SJed Brown /*
3791b7f5c055SJed Brown    Project x to the nearest point on the implicit surface using Newton's method.
3792b7f5c055SJed Brown */
3793d71ae5a4SJacob Faibussowitsch static PetscErrorCode TPSNearestPoint(TPSEvaluateFunc feval, PetscScalar x[])
3794d71ae5a4SJacob Faibussowitsch {
3795b7f5c055SJed Brown   PetscScalar y[3] = {x[0], x[1], x[2]}; // Initial guess
3796b7f5c055SJed Brown 
3797b7f5c055SJed Brown   PetscFunctionBegin;
3798b7f5c055SJed Brown   for (PetscInt iter = 0; iter < 10; iter++) {
3799b7f5c055SJed Brown     PetscScalar res[3], J[9];
3800b7f5c055SJed Brown     PetscReal   resnorm;
3801b7f5c055SJed Brown     TPSNearestPointResJac(feval, x, y, res, J);
3802b7f5c055SJed Brown     resnorm = PetscSqrtReal(PetscSqr(PetscRealPart(res[0])) + PetscSqr(PetscRealPart(res[1])) + PetscSqr(PetscRealPart(res[2])));
3803b7f5c055SJed Brown     if (0) { // Turn on this monitor if you need to confirm quadratic convergence
380463a3b9bcSJacob 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])));
3805b7f5c055SJed Brown     }
3806b7f5c055SJed Brown     if (resnorm < PETSC_SMALL) break;
3807b7f5c055SJed Brown 
3808b7f5c055SJed Brown     // Take the Newton step
38099566063dSJacob Faibussowitsch     PetscCall(PetscKernel_A_gets_inverse_A_3(J, 0., PETSC_FALSE, NULL));
3810b7f5c055SJed Brown     PetscKernel_v_gets_v_minus_A_times_w_3(y, J, res);
3811b7f5c055SJed Brown   }
3812b7f5c055SJed Brown   for (PetscInt i = 0; i < 3; i++) x[i] = y[i];
38133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3814b7f5c055SJed Brown }
3815b7f5c055SJed Brown 
3816b7f5c055SJed Brown const char *const DMPlexTPSTypes[] = {"SCHWARZ_P", "GYROID", "DMPlexTPSType", "DMPLEX_TPS_", NULL};
3817b7f5c055SJed Brown 
3818d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateTPSMesh_Internal(DM dm, DMPlexTPSType tpstype, const PetscInt extent[], const DMBoundaryType periodic[], PetscBool tps_distribute, PetscInt refinements, PetscInt layers, PetscReal thickness)
3819d71ae5a4SJacob Faibussowitsch {
3820b7f5c055SJed Brown   PetscMPIInt rank;
3821b7f5c055SJed Brown   PetscInt    topoDim = 2, spaceDim = 3, numFaces = 0, numVertices = 0, numEdges = 0;
3822b7f5c055SJed Brown   PetscInt (*edges)[2] = NULL, *edgeSets = NULL;
3823b7f5c055SJed Brown   PetscInt           *cells_flat = NULL;
3824b7f5c055SJed Brown   PetscReal          *vtxCoords  = NULL;
3825b7f5c055SJed Brown   TPSEvaluateFunc     evalFunc   = NULL;
38268434afd1SBarry Smith   PetscSimplePointFn *normalFunc = NULL;
3827b7f5c055SJed Brown   DMLabel             label;
3828b7f5c055SJed Brown 
3829b7f5c055SJed Brown   PetscFunctionBegin;
383046139095SJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_Generate, dm, 0, 0, 0));
38319566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
383263a3b9bcSJacob 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);
3833b7f5c055SJed Brown   switch (tpstype) {
3834b7f5c055SJed Brown   case DMPLEX_TPS_SCHWARZ_P:
3835b7f5c055SJed 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");
3836c5853193SPierre Jolivet     if (rank == 0) {
3837b7f5c055SJed Brown       PetscInt (*cells)[6][4][4] = NULL; // [junction, junction-face, cell, conn]
3838b7f5c055SJed Brown       PetscInt  Njunctions = 0, Ncuts = 0, Npipes[3], vcount;
3839b7f5c055SJed Brown       PetscReal L = 1;
3840b7f5c055SJed Brown 
3841b7f5c055SJed Brown       Npipes[0]   = (extent[0] + 1) * extent[1] * extent[2];
3842b7f5c055SJed Brown       Npipes[1]   = extent[0] * (extent[1] + 1) * extent[2];
3843b7f5c055SJed Brown       Npipes[2]   = extent[0] * extent[1] * (extent[2] + 1);
3844b7f5c055SJed Brown       Njunctions  = extent[0] * extent[1] * extent[2];
3845b7f5c055SJed Brown       Ncuts       = 2 * (extent[0] * extent[1] + extent[1] * extent[2] + extent[2] * extent[0]);
3846b7f5c055SJed Brown       numVertices = 4 * (Npipes[0] + Npipes[1] + Npipes[2]) + 8 * Njunctions;
38479566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(3 * numVertices, &vtxCoords));
38489566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Njunctions, &cells));
38499566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edges));
38509566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(Ncuts * 4, &edgeSets));
3851b7f5c055SJed Brown       // x-normal pipes
3852b7f5c055SJed Brown       vcount = 0;
3853b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0] + 1; i++) {
3854b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3855b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3856b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3857b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * i - 1) * L;
3858b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3859b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3860b7f5c055SJed Brown             }
3861b7f5c055SJed Brown           }
3862b7f5c055SJed Brown         }
3863b7f5c055SJed Brown       }
3864b7f5c055SJed Brown       // y-normal pipes
3865b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3866b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1] + 1; j++) {
3867b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3868b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3869b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3870b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * j - 1) * L;
3871b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * k * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3872b7f5c055SJed Brown             }
3873b7f5c055SJed Brown           }
3874b7f5c055SJed Brown         }
3875b7f5c055SJed Brown       }
3876b7f5c055SJed Brown       // z-normal pipes
3877b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3878b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3879b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2] + 1; k++) {
3880b7f5c055SJed Brown             for (PetscInt l = 0; l < 4; l++) {
3881b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * i * L + PetscCosReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3882b7f5c055SJed Brown               vtxCoords[vcount++] = 2 * j * L + PetscSinReal((2 * l + 1) * PETSC_PI / 4) * L / 2;
3883b7f5c055SJed Brown               vtxCoords[vcount++] = (2 * k - 1) * L;
3884b7f5c055SJed Brown             }
3885b7f5c055SJed Brown           }
3886b7f5c055SJed Brown         }
3887b7f5c055SJed Brown       }
3888b7f5c055SJed Brown       // junctions
3889b7f5c055SJed Brown       for (PetscInt i = 0; i < extent[0]; i++) {
3890b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
3891b7f5c055SJed Brown           for (PetscInt k = 0; k < extent[2]; k++) {
3892b7f5c055SJed Brown             const PetscInt J = (i * extent[1] + j) * extent[2] + k, Jvoff = (Npipes[0] + Npipes[1] + Npipes[2]) * 4 + J * 8;
3893b7f5c055SJed Brown             PetscCheck(vcount / 3 == Jvoff, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected vertex count");
3894b7f5c055SJed Brown             for (PetscInt ii = 0; ii < 2; ii++) {
3895b7f5c055SJed Brown               for (PetscInt jj = 0; jj < 2; jj++) {
3896b7f5c055SJed Brown                 for (PetscInt kk = 0; kk < 2; kk++) {
3897b7f5c055SJed Brown                   double Ls           = (1 - sqrt(2) / 4) * L;
3898b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * i * L + (2 * ii - 1) * Ls;
3899b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * j * L + (2 * jj - 1) * Ls;
3900b7f5c055SJed Brown                   vtxCoords[vcount++] = 2 * k * L + (2 * kk - 1) * Ls;
3901b7f5c055SJed Brown                 }
3902b7f5c055SJed Brown               }
3903b7f5c055SJed Brown             }
3904b7f5c055SJed Brown             const PetscInt jfaces[3][2][4] = {
3905b7f5c055SJed Brown               {{3, 1, 0, 2}, {7, 5, 4, 6}}, // x-aligned
3906b7f5c055SJed Brown               {{5, 4, 0, 1}, {7, 6, 2, 3}}, // y-aligned
3907b7f5c055SJed Brown               {{6, 2, 0, 4}, {7, 3, 1, 5}}  // z-aligned
3908b7f5c055SJed Brown             };
3909b7f5c055SJed Brown             const PetscInt pipe_lo[3] = {// vertex numbers of pipes
39109371c9d4SSatish 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};
3911b7f5c055SJed Brown             const PetscInt pipe_hi[3] = {// vertex numbers of pipes
39129371c9d4SSatish 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};
3913b7f5c055SJed Brown             for (PetscInt dir = 0; dir < 3; dir++) { // x,y,z
3914b7f5c055SJed Brown               const PetscInt ijk[3] = {i, j, k};
3915b7f5c055SJed Brown               for (PetscInt l = 0; l < 4; l++) { // rotations
3916b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][0] = pipe_lo[dir] + l;
3917b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][1] = Jvoff + jfaces[dir][0][l];
3918b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][2] = Jvoff + jfaces[dir][0][(l - 1 + 4) % 4];
3919b7f5c055SJed Brown                 cells[J][dir * 2 + 0][l][3] = pipe_lo[dir] + (l - 1 + 4) % 4;
3920b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][0] = Jvoff + jfaces[dir][1][l];
3921b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][1] = pipe_hi[dir] + l;
3922b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][2] = pipe_hi[dir] + (l - 1 + 4) % 4;
3923b7f5c055SJed Brown                 cells[J][dir * 2 + 1][l][3] = Jvoff + jfaces[dir][1][(l - 1 + 4) % 4];
3924b7f5c055SJed Brown                 if (ijk[dir] == 0) {
3925b7f5c055SJed Brown                   edges[numEdges][0] = pipe_lo[dir] + l;
3926b7f5c055SJed Brown                   edges[numEdges][1] = pipe_lo[dir] + (l + 1) % 4;
3927b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 1;
3928b7f5c055SJed Brown                   numEdges++;
3929b7f5c055SJed Brown                 }
3930b7f5c055SJed Brown                 if (ijk[dir] + 1 == extent[dir]) {
3931b7f5c055SJed Brown                   edges[numEdges][0] = pipe_hi[dir] + l;
3932b7f5c055SJed Brown                   edges[numEdges][1] = pipe_hi[dir] + (l + 1) % 4;
3933b7f5c055SJed Brown                   edgeSets[numEdges] = dir * 2 + 2;
3934b7f5c055SJed Brown                   numEdges++;
3935b7f5c055SJed Brown                 }
3936b7f5c055SJed Brown               }
3937b7f5c055SJed Brown             }
3938b7f5c055SJed Brown           }
3939b7f5c055SJed Brown         }
3940b7f5c055SJed Brown       }
394163a3b9bcSJacob Faibussowitsch       PetscCheck(numEdges == Ncuts * 4, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Edge count %" PetscInt_FMT " incompatible with number of cuts %" PetscInt_FMT, numEdges, Ncuts);
3942b7f5c055SJed Brown       numFaces   = 24 * Njunctions;
3943b7f5c055SJed Brown       cells_flat = cells[0][0][0];
3944b7f5c055SJed Brown     }
3945b7f5c055SJed Brown     evalFunc   = TPSEvaluate_SchwarzP;
39464663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_SchwarzP;
3947b7f5c055SJed Brown     break;
3948b7f5c055SJed Brown   case DMPLEX_TPS_GYROID:
3949c5853193SPierre Jolivet     if (rank == 0) {
3950b7f5c055SJed Brown       // This is a coarse mesh approximation of the gyroid shifted to being the zero of the level set
3951b7f5c055SJed Brown       //
3952b7f5c055SJed 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)
3953b7f5c055SJed Brown       //
3954b7f5c055SJed Brown       // on the cell [0,2]^3.
3955b7f5c055SJed Brown       //
3956b7f5c055SJed Brown       // Think about dividing that cell into four columns, and focus on the column [0,1]x[0,1]x[0,2].
3957b7f5c055SJed Brown       // If you looked at the gyroid in that column at different slices of z you would see that it kind of spins
3958b7f5c055SJed Brown       // like a boomerang:
3959b7f5c055SJed Brown       //
3960b7f5c055SJed Brown       //     z = 0          z = 1/4        z = 1/2        z = 3/4     //
3961b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3962b7f5c055SJed Brown       //                                                              //
3963b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3964b7f5c055SJed Brown       //      \                                   /            \      //
3965b7f5c055SJed Brown       //       \            `-_   _-'            /              }     //
3966b7f5c055SJed Brown       //        *-_            `-'            _-'              /      //
3967b7f5c055SJed Brown       //     +     `-+      +       +      +-'     +      +   /   +   //
3968b7f5c055SJed Brown       //                                                              //
3969b7f5c055SJed Brown       //                                                              //
3970b7f5c055SJed Brown       //     z = 1          z = 5/4        z = 3/2        z = 7/4     //
3971b7f5c055SJed Brown       //     -----          -------        -------        -------     //
3972b7f5c055SJed Brown       //                                                              //
3973b7f5c055SJed Brown       //     +-_     +      +       +      +     _-+      +   /   +   //
3974b7f5c055SJed Brown       //        `-_            _-_            _-`            /        //
3975b7f5c055SJed Brown       //           \        _-'   `-_        /              {         //
3976b7f5c055SJed Brown       //            \                       /                \        //
3977b7f5c055SJed Brown       //     +       +      +       +      +       +      +   \   +   //
3978b7f5c055SJed Brown       //
3979b7f5c055SJed Brown       //
3980b7f5c055SJed Brown       // This course mesh approximates each of these slices by two line segments,
3981b7f5c055SJed Brown       // and then connects the segments in consecutive layers with quadrilateral faces.
3982b7f5c055SJed Brown       // All of the end points of the segments are multiples of 1/4 except for the
3983b7f5c055SJed Brown       // point * in the picture for z = 0 above and the similar points in other layers.
3984b7f5c055SJed Brown       // That point is at (gamma, gamma, 0), where gamma is calculated below.
3985b7f5c055SJed Brown       //
3986b7f5c055SJed Brown       // The column  [1,2]x[1,2]x[0,2] looks the same as this column;
3987b7f5c055SJed Brown       // The columns [1,2]x[0,1]x[0,2] and [0,1]x[1,2]x[0,2] are mirror images.
3988b7f5c055SJed Brown       //
3989b7f5c055SJed Brown       // As for how this method turned into the names given to the vertices:
3990b7f5c055SJed Brown       // that was not systematic, it was just the way it worked out in my handwritten notes.
3991b7f5c055SJed Brown 
3992b7f5c055SJed Brown       PetscInt facesPerBlock = 64;
3993b7f5c055SJed Brown       PetscInt vertsPerBlock = 56;
3994b7f5c055SJed Brown       PetscInt extentPlus[3];
3995b7f5c055SJed Brown       PetscInt numBlocks, numBlocksPlus;
39969371c9d4SSatish 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;
39979371c9d4SSatish Balay       const PetscInt pattern[64][4] = {
39989371c9d4SSatish Balay         /* face to vertex within the coarse discretization of a single gyroid block */
3999b7f5c055SJed Brown         /* layer 0 */
40009371c9d4SSatish Balay         {A,           C,           K,           G          },
40019371c9d4SSatish Balay         {C,           B,           II,          K          },
40029371c9d4SSatish Balay         {D,           A,           H,           L          },
40039371c9d4SSatish Balay         {B + 56 * 1,  D,           L,           J          },
40049371c9d4SSatish Balay         {E,           B + 56 * 1,  J,           N          },
40059371c9d4SSatish Balay         {A + 56 * 2,  E,           N,           H + 56 * 2 },
40069371c9d4SSatish Balay         {F,           A + 56 * 2,  G + 56 * 2,  M          },
40079371c9d4SSatish Balay         {B,           F,           M,           II         },
4008b7f5c055SJed Brown         /* layer 1 */
40099371c9d4SSatish Balay         {G,           K,           Q,           O          },
40109371c9d4SSatish Balay         {K,           II,          P,           Q          },
40119371c9d4SSatish Balay         {L,           H,           O + 56 * 1,  R          },
40129371c9d4SSatish Balay         {J,           L,           R,           P          },
40139371c9d4SSatish Balay         {N,           J,           P,           S          },
40149371c9d4SSatish Balay         {H + 56 * 2,  N,           S,           O + 56 * 3 },
40159371c9d4SSatish Balay         {M,           G + 56 * 2,  O + 56 * 2,  T          },
40169371c9d4SSatish Balay         {II,          M,           T,           P          },
4017b7f5c055SJed Brown         /* layer 2 */
40189371c9d4SSatish Balay         {O,           Q,           Y,           U          },
40199371c9d4SSatish Balay         {Q,           P,           W,           Y          },
40209371c9d4SSatish Balay         {R,           O + 56 * 1,  U + 56 * 1,  Ap         },
40219371c9d4SSatish Balay         {P,           R,           Ap,          W          },
40229371c9d4SSatish Balay         {S,           P,           X,           Bp         },
40239371c9d4SSatish Balay         {O + 56 * 3,  S,           Bp,          V + 56 * 1 },
40249371c9d4SSatish Balay         {T,           O + 56 * 2,  V,           Z          },
40259371c9d4SSatish Balay         {P,           T,           Z,           X          },
4026b7f5c055SJed Brown         /* layer 3 */
40279371c9d4SSatish Balay         {U,           Y,           Ep,          Dp         },
40289371c9d4SSatish Balay         {Y,           W,           Cp,          Ep         },
40299371c9d4SSatish Balay         {Ap,          U + 56 * 1,  Dp + 56 * 1, Gp         },
40309371c9d4SSatish Balay         {W,           Ap,          Gp,          Cp         },
40319371c9d4SSatish Balay         {Bp,          X,           Cp + 56 * 2, Fp         },
40329371c9d4SSatish Balay         {V + 56 * 1,  Bp,          Fp,          Dp + 56 * 1},
40339371c9d4SSatish Balay         {Z,           V,           Dp,          Hp         },
40349371c9d4SSatish Balay         {X,           Z,           Hp,          Cp + 56 * 2},
4035b7f5c055SJed Brown         /* layer 4 */
40369371c9d4SSatish Balay         {Dp,          Ep,          Mp,          Kp         },
40379371c9d4SSatish Balay         {Ep,          Cp,          Ip,          Mp         },
40389371c9d4SSatish Balay         {Gp,          Dp + 56 * 1, Lp,          Np         },
40399371c9d4SSatish Balay         {Cp,          Gp,          Np,          Jp         },
40409371c9d4SSatish Balay         {Fp,          Cp + 56 * 2, Jp + 56 * 2, Pp         },
40419371c9d4SSatish Balay         {Dp + 56 * 1, Fp,          Pp,          Lp         },
40429371c9d4SSatish Balay         {Hp,          Dp,          Kp,          Op         },
40439371c9d4SSatish Balay         {Cp + 56 * 2, Hp,          Op,          Ip + 56 * 2},
4044b7f5c055SJed Brown         /* layer 5 */
40459371c9d4SSatish Balay         {Kp,          Mp,          Sp,          Rp         },
40469371c9d4SSatish Balay         {Mp,          Ip,          Qp,          Sp         },
40479371c9d4SSatish Balay         {Np,          Lp,          Rp,          Tp         },
40489371c9d4SSatish Balay         {Jp,          Np,          Tp,          Qp + 56 * 1},
40499371c9d4SSatish Balay         {Pp,          Jp + 56 * 2, Qp + 56 * 3, Up         },
40509371c9d4SSatish Balay         {Lp,          Pp,          Up,          Rp         },
40519371c9d4SSatish Balay         {Op,          Kp,          Rp,          Vp         },
40529371c9d4SSatish Balay         {Ip + 56 * 2, Op,          Vp,          Qp + 56 * 2},
4053b7f5c055SJed Brown         /* layer 6 */
40549371c9d4SSatish Balay         {Rp,          Sp,          Aq,          Yp         },
40559371c9d4SSatish Balay         {Sp,          Qp,          Wp,          Aq         },
40569371c9d4SSatish Balay         {Tp,          Rp,          Yp,          Cq         },
40579371c9d4SSatish Balay         {Qp + 56 * 1, Tp,          Cq,          Wp + 56 * 1},
40589371c9d4SSatish Balay         {Up,          Qp + 56 * 3, Xp + 56 * 1, Dq         },
40599371c9d4SSatish Balay         {Rp,          Up,          Dq,          Zp         },
40609371c9d4SSatish Balay         {Vp,          Rp,          Zp,          Bq         },
40619371c9d4SSatish Balay         {Qp + 56 * 2, Vp,          Bq,          Xp         },
4062b7f5c055SJed Brown         /* layer 7 (the top is the periodic image of the bottom of layer 0) */
40639371c9d4SSatish Balay         {Yp,          Aq,          C + 56 * 4,  A + 56 * 4 },
40649371c9d4SSatish Balay         {Aq,          Wp,          B + 56 * 4,  C + 56 * 4 },
40659371c9d4SSatish Balay         {Cq,          Yp,          A + 56 * 4,  D + 56 * 4 },
40669371c9d4SSatish Balay         {Wp + 56 * 1, Cq,          D + 56 * 4,  B + 56 * 5 },
40679371c9d4SSatish Balay         {Dq,          Xp + 56 * 1, B + 56 * 5,  E + 56 * 4 },
40689371c9d4SSatish Balay         {Zp,          Dq,          E + 56 * 4,  A + 56 * 6 },
40699371c9d4SSatish Balay         {Bq,          Zp,          A + 56 * 6,  F + 56 * 4 },
40709371c9d4SSatish Balay         {Xp,          Bq,          F + 56 * 4,  B + 56 * 4 }
4071b7f5c055SJed Brown       };
4072b7f5c055SJed Brown       const PetscReal gamma                = PetscAcosReal((PetscSqrtReal(3.) - 1.) / PetscSqrtReal(2.)) / PETSC_PI;
40739371c9d4SSatish Balay       const PetscReal patternCoords[56][3] = {
4074bee3fc89SBarry Smith         {1.,        0.,        0.  }, /* A  */
4075bee3fc89SBarry Smith         {0.,        1.,        0.  }, /* B  */
4076bee3fc89SBarry Smith         {gamma,     gamma,     0.  }, /* C  */
4077bee3fc89SBarry Smith         {1 + gamma, 1 - gamma, 0.  }, /* D  */
4078bee3fc89SBarry Smith         {2 - gamma, 2 - gamma, 0.  }, /* E  */
4079bee3fc89SBarry Smith         {1 - gamma, 1 + gamma, 0.  }, /* F  */
4080b7f5c055SJed Brown 
4081bee3fc89SBarry Smith         {.5,        0,         .25 }, /* G  */
4082bee3fc89SBarry Smith         {1.5,       0.,        .25 }, /* H  */
4083bee3fc89SBarry Smith         {.5,        1.,        .25 }, /* II */
4084bee3fc89SBarry Smith         {1.5,       1.,        .25 }, /* J  */
4085bee3fc89SBarry Smith         {.25,       .5,        .25 }, /* K  */
4086bee3fc89SBarry Smith         {1.25,      .5,        .25 }, /* L  */
4087bee3fc89SBarry Smith         {.75,       1.5,       .25 }, /* M  */
4088bee3fc89SBarry Smith         {1.75,      1.5,       .25 }, /* N  */
4089b7f5c055SJed Brown 
4090bee3fc89SBarry Smith         {0.,        0.,        .5  }, /* O  */
4091bee3fc89SBarry Smith         {1.,        1.,        .5  }, /* P  */
4092bee3fc89SBarry Smith         {gamma,     1 - gamma, .5  }, /* Q  */
4093bee3fc89SBarry Smith         {1 + gamma, gamma,     .5  }, /* R  */
4094bee3fc89SBarry Smith         {2 - gamma, 1 + gamma, .5  }, /* S  */
4095bee3fc89SBarry Smith         {1 - gamma, 2 - gamma, .5  }, /* T  */
4096b7f5c055SJed Brown 
4097bee3fc89SBarry Smith         {0.,        .5,        .75 }, /* U  */
4098bee3fc89SBarry Smith         {0.,        1.5,       .75 }, /* V  */
4099bee3fc89SBarry Smith         {1.,        .5,        .75 }, /* W  */
4100bee3fc89SBarry Smith         {1.,        1.5,       .75 }, /* X  */
4101bee3fc89SBarry Smith         {.5,        .75,       .75 }, /* Y  */
4102bee3fc89SBarry Smith         {.5,        1.75,      .75 }, /* Z  */
4103bee3fc89SBarry Smith         {1.5,       .25,       .75 }, /* Ap */
4104bee3fc89SBarry Smith         {1.5,       1.25,      .75 }, /* Bp */
4105b7f5c055SJed Brown 
4106bee3fc89SBarry Smith         {1.,        0.,        1.  }, /* Cp */
4107bee3fc89SBarry Smith         {0.,        1.,        1.  }, /* Dp */
4108bee3fc89SBarry Smith         {1 - gamma, 1 - gamma, 1.  }, /* Ep */
4109bee3fc89SBarry Smith         {1 + gamma, 1 + gamma, 1.  }, /* Fp */
4110bee3fc89SBarry Smith         {2 - gamma, gamma,     1.  }, /* Gp */
4111bee3fc89SBarry Smith         {gamma,     2 - gamma, 1.  }, /* Hp */
4112b7f5c055SJed Brown 
4113bee3fc89SBarry Smith         {.5,        0.,        1.25}, /* Ip */
4114bee3fc89SBarry Smith         {1.5,       0.,        1.25}, /* Jp */
4115bee3fc89SBarry Smith         {.5,        1.,        1.25}, /* Kp */
4116bee3fc89SBarry Smith         {1.5,       1.,        1.25}, /* Lp */
4117bee3fc89SBarry Smith         {.75,       .5,        1.25}, /* Mp */
4118bee3fc89SBarry Smith         {1.75,      .5,        1.25}, /* Np */
4119bee3fc89SBarry Smith         {.25,       1.5,       1.25}, /* Op */
4120bee3fc89SBarry Smith         {1.25,      1.5,       1.25}, /* Pp */
4121b7f5c055SJed Brown 
4122bee3fc89SBarry Smith         {0.,        0.,        1.5 }, /* Qp */
4123bee3fc89SBarry Smith         {1.,        1.,        1.5 }, /* Rp */
4124bee3fc89SBarry Smith         {1 - gamma, gamma,     1.5 }, /* Sp */
4125bee3fc89SBarry Smith         {2 - gamma, 1 - gamma, 1.5 }, /* Tp */
4126bee3fc89SBarry Smith         {1 + gamma, 2 - gamma, 1.5 }, /* Up */
4127bee3fc89SBarry Smith         {gamma,     1 + gamma, 1.5 }, /* Vp */
4128b7f5c055SJed Brown 
4129bee3fc89SBarry Smith         {0.,        .5,        1.75}, /* Wp */
4130bee3fc89SBarry Smith         {0.,        1.5,       1.75}, /* Xp */
4131bee3fc89SBarry Smith         {1.,        .5,        1.75}, /* Yp */
4132bee3fc89SBarry Smith         {1.,        1.5,       1.75}, /* Zp */
4133bee3fc89SBarry Smith         {.5,        .25,       1.75}, /* Aq */
4134bee3fc89SBarry Smith         {.5,        1.25,      1.75}, /* Bq */
4135bee3fc89SBarry Smith         {1.5,       .75,       1.75}, /* Cq */
4136bee3fc89SBarry Smith         {1.5,       1.75,      1.75}, /* Dq */
4137b7f5c055SJed Brown       };
4138b7f5c055SJed Brown       PetscInt (*cells)[64][4] = NULL;
4139b7f5c055SJed Brown       PetscBool *seen;
4140b7f5c055SJed Brown       PetscInt  *vertToTrueVert;
4141b7f5c055SJed Brown       PetscInt   count;
4142b7f5c055SJed Brown 
4143b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) extentPlus[i] = extent[i] + 1;
4144b7f5c055SJed Brown       numBlocks = 1;
4145b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocks *= extent[i];
4146b7f5c055SJed Brown       numBlocksPlus = 1;
4147b7f5c055SJed Brown       for (PetscInt i = 0; i < 3; i++) numBlocksPlus *= extentPlus[i];
4148b7f5c055SJed Brown       numFaces = numBlocks * facesPerBlock;
41499566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocks, &cells));
41509566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(numBlocksPlus * vertsPerBlock, &seen));
4151b7f5c055SJed Brown       for (PetscInt k = 0; k < extent[2]; k++) {
4152b7f5c055SJed Brown         for (PetscInt j = 0; j < extent[1]; j++) {
4153b7f5c055SJed Brown           for (PetscInt i = 0; i < extent[0]; i++) {
4154b7f5c055SJed Brown             for (PetscInt f = 0; f < facesPerBlock; f++) {
4155b7f5c055SJed Brown               for (PetscInt v = 0; v < 4; v++) {
4156b7f5c055SJed Brown                 PetscInt vertRaw     = pattern[f][v];
4157b7f5c055SJed Brown                 PetscInt blockidx    = vertRaw / 56;
4158b7f5c055SJed Brown                 PetscInt patternvert = vertRaw % 56;
4159b7f5c055SJed Brown                 PetscInt xplus       = (blockidx & 1);
4160b7f5c055SJed Brown                 PetscInt yplus       = (blockidx & 2) >> 1;
4161b7f5c055SJed Brown                 PetscInt zplus       = (blockidx & 4) >> 2;
4162b7f5c055SJed Brown                 PetscInt zcoord      = (periodic && periodic[2] == DM_BOUNDARY_PERIODIC) ? ((k + zplus) % extent[2]) : (k + zplus);
4163b7f5c055SJed Brown                 PetscInt ycoord      = (periodic && periodic[1] == DM_BOUNDARY_PERIODIC) ? ((j + yplus) % extent[1]) : (j + yplus);
4164b7f5c055SJed Brown                 PetscInt xcoord      = (periodic && periodic[0] == DM_BOUNDARY_PERIODIC) ? ((i + xplus) % extent[0]) : (i + xplus);
4165b7f5c055SJed Brown                 PetscInt vert        = ((zcoord * extentPlus[1] + ycoord) * extentPlus[0] + xcoord) * 56 + patternvert;
4166b7f5c055SJed Brown 
4167b7f5c055SJed Brown                 cells[(k * extent[1] + j) * extent[0] + i][f][v] = vert;
4168b7f5c055SJed Brown                 seen[vert]                                       = PETSC_TRUE;
4169b7f5c055SJed Brown               }
4170b7f5c055SJed Brown             }
4171b7f5c055SJed Brown           }
4172b7f5c055SJed Brown         }
4173b7f5c055SJed Brown       }
41749371c9d4SSatish Balay       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++)
41759371c9d4SSatish Balay         if (seen[i]) numVertices++;
4176b7f5c055SJed Brown       count = 0;
41779566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numBlocksPlus * vertsPerBlock, &vertToTrueVert));
41789566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * 3, &vtxCoords));
4179b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocksPlus * vertsPerBlock; i++) vertToTrueVert[i] = -1;
4180b7f5c055SJed Brown       for (PetscInt k = 0; k < extentPlus[2]; k++) {
4181b7f5c055SJed Brown         for (PetscInt j = 0; j < extentPlus[1]; j++) {
4182b7f5c055SJed Brown           for (PetscInt i = 0; i < extentPlus[0]; i++) {
4183b7f5c055SJed Brown             for (PetscInt v = 0; v < vertsPerBlock; v++) {
4184b7f5c055SJed Brown               PetscInt vIdx = ((k * extentPlus[1] + j) * extentPlus[0] + i) * vertsPerBlock + v;
4185b7f5c055SJed Brown 
4186b7f5c055SJed Brown               if (seen[vIdx]) {
4187b7f5c055SJed Brown                 PetscInt thisVert;
4188b7f5c055SJed Brown 
4189b7f5c055SJed Brown                 vertToTrueVert[vIdx] = thisVert = count++;
4190b7f5c055SJed Brown 
4191b7f5c055SJed Brown                 for (PetscInt d = 0; d < 3; d++) vtxCoords[3 * thisVert + d] = patternCoords[v][d];
4192b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 0] += i * 2;
4193b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 1] += j * 2;
4194b7f5c055SJed Brown                 vtxCoords[3 * thisVert + 2] += k * 2;
4195b7f5c055SJed Brown               }
4196b7f5c055SJed Brown             }
4197b7f5c055SJed Brown           }
4198b7f5c055SJed Brown         }
4199b7f5c055SJed Brown       }
4200b7f5c055SJed Brown       for (PetscInt i = 0; i < numBlocks; i++) {
4201b7f5c055SJed Brown         for (PetscInt f = 0; f < facesPerBlock; f++) {
4202ad540459SPierre Jolivet           for (PetscInt v = 0; v < 4; v++) cells[i][f][v] = vertToTrueVert[cells[i][f][v]];
4203b7f5c055SJed Brown         }
4204b7f5c055SJed Brown       }
42059566063dSJacob Faibussowitsch       PetscCall(PetscFree(vertToTrueVert));
42069566063dSJacob Faibussowitsch       PetscCall(PetscFree(seen));
4207b7f5c055SJed Brown       cells_flat = cells[0][0];
4208b7f5c055SJed Brown       numEdges   = 0;
4209b7f5c055SJed Brown       for (PetscInt i = 0; i < numFaces; i++) {
4210b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
4211b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
4212b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
4213b7f5c055SJed Brown 
4214b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
4215b7f5c055SJed Brown             if (!periodic || periodic[0] != DM_BOUNDARY_PERIODIC) {
4216b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) numEdges++;
4217b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) numEdges++;
4218b7f5c055SJed Brown             }
4219b7f5c055SJed Brown           }
4220b7f5c055SJed Brown         }
4221b7f5c055SJed Brown       }
42229566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edges));
42239566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numEdges, &edgeSets));
4224b7f5c055SJed Brown       for (PetscInt edge = 0, i = 0; i < numFaces; i++) {
4225b7f5c055SJed Brown         for (PetscInt e = 0; e < 4; e++) {
4226b7f5c055SJed Brown           PetscInt         ev[]       = {cells_flat[i * 4 + e], cells_flat[i * 4 + ((e + 1) % 4)]};
4227b7f5c055SJed Brown           const PetscReal *evCoords[] = {&vtxCoords[3 * ev[0]], &vtxCoords[3 * ev[1]]};
4228b7f5c055SJed Brown 
4229b7f5c055SJed Brown           for (PetscInt d = 0; d < 3; d++) {
4230b7f5c055SJed Brown             if (!periodic || periodic[d] != DM_BOUNDARY_PERIODIC) {
4231b7f5c055SJed Brown               if (evCoords[0][d] == 0. && evCoords[1][d] == 0.) {
4232b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
4233b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
4234b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d;
4235b7f5c055SJed Brown               }
4236b7f5c055SJed Brown               if (evCoords[0][d] == 2. * extent[d] && evCoords[1][d] == 2. * extent[d]) {
4237b7f5c055SJed Brown                 edges[edge][0]   = ev[0];
4238b7f5c055SJed Brown                 edges[edge][1]   = ev[1];
4239b7f5c055SJed Brown                 edgeSets[edge++] = 2 * d + 1;
4240b7f5c055SJed Brown               }
4241b7f5c055SJed Brown             }
4242b7f5c055SJed Brown           }
4243b7f5c055SJed Brown         }
4244b7f5c055SJed Brown       }
4245b7f5c055SJed Brown     }
4246b7f5c055SJed Brown     evalFunc   = TPSEvaluate_Gyroid;
42474663dae6SJed Brown     normalFunc = TPSExtrudeNormalFunc_Gyroid;
4248b7f5c055SJed Brown     break;
4249b7f5c055SJed Brown   }
4250b7f5c055SJed Brown 
42519566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, topoDim));
4252c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(dm, numFaces, numVertices, 4, cells_flat));
42539566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(dm, 0, 0, 0, NULL));
42549566063dSJacob Faibussowitsch   PetscCall(PetscFree(cells_flat));
4255b7f5c055SJed Brown   {
4256b7f5c055SJed Brown     DM idm;
42579566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(dm, &idm));
425869d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &idm));
4259b7f5c055SJed Brown   }
4260c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, vtxCoords));
42619566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(dm, spaceDim, NULL));
42629566063dSJacob Faibussowitsch   PetscCall(PetscFree(vtxCoords));
4263b7f5c055SJed Brown 
42649566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "Face Sets"));
42659566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
4266b7f5c055SJed Brown   for (PetscInt e = 0; e < numEdges; e++) {
4267b7f5c055SJed Brown     PetscInt        njoin;
4268b7f5c055SJed Brown     const PetscInt *join, verts[] = {numFaces + edges[e][0], numFaces + edges[e][1]};
42699566063dSJacob Faibussowitsch     PetscCall(DMPlexGetJoin(dm, 2, verts, &njoin, &join));
427063a3b9bcSJacob 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]);
42719566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(label, join[0], edgeSets[e]));
42729566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreJoin(dm, 2, verts, &njoin, &join));
4273b7f5c055SJed Brown   }
42749566063dSJacob Faibussowitsch   PetscCall(PetscFree(edges));
42759566063dSJacob Faibussowitsch   PetscCall(PetscFree(edgeSets));
42761436d7faSJed Brown   if (tps_distribute) {
42771436d7faSJed Brown     DM               pdm = NULL;
42781436d7faSJed Brown     PetscPartitioner part;
42791436d7faSJed Brown 
42809566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
42819566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
42829566063dSJacob Faibussowitsch     PetscCall(DMPlexDistribute(dm, 0, NULL, &pdm));
428348a46eb9SPierre Jolivet     if (pdm) PetscCall(DMPlexReplace_Internal(dm, &pdm));
42841436d7faSJed Brown     // Do not auto-distribute again
42859566063dSJacob Faibussowitsch     PetscCall(DMPlexDistributeSetDefault(dm, PETSC_FALSE));
42861436d7faSJed Brown   }
4287b7f5c055SJed Brown 
42889566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
4289b7f5c055SJed Brown   for (PetscInt refine = 0; refine < refinements; refine++) {
4290b7f5c055SJed Brown     PetscInt     m;
4291f1d4225fSZach Atkins     DM           dmf, cdm, cdmf;
4292b7f5c055SJed Brown     Vec          X;
4293b7f5c055SJed Brown     PetscScalar *x;
4294f1d4225fSZach Atkins 
42959566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, MPI_COMM_NULL, &dmf));
4296f1d4225fSZach Atkins     PetscCall(DMGetCoordinateDM(dm, &cdm));
4297f1d4225fSZach Atkins     PetscCall(DMGetCoordinateDM(dmf, &cdmf));
4298f1d4225fSZach Atkins     PetscCall(DMCopyDisc(cdm, cdmf));
429969d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmf));
4300b7f5c055SJed Brown 
43019566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinatesLocal(dm, &X));
43029566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(X, &m));
43039566063dSJacob Faibussowitsch     PetscCall(VecGetArray(X, &x));
430448a46eb9SPierre Jolivet     for (PetscInt i = 0; i < m; i += 3) PetscCall(TPSNearestPoint(evalFunc, &x[i]));
43059566063dSJacob Faibussowitsch     PetscCall(VecRestoreArray(X, &x));
4306b7f5c055SJed Brown   }
4307b7f5c055SJed Brown 
4308b7f5c055SJed Brown   // Face Sets has already been propagated to new vertices during refinement; this propagates to the initial vertices.
43099566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "Face Sets", &label));
43109566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, label));
4311b7f5c055SJed Brown 
431246139095SJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0));
431346139095SJed Brown 
4314b7f5c055SJed Brown   if (thickness > 0) {
43154663dae6SJed Brown     DM              edm, cdm, ecdm;
43164663dae6SJed Brown     DMPlexTransform tr;
43174663dae6SJed Brown     const char     *prefix;
43184663dae6SJed Brown     PetscOptions    options;
43194663dae6SJed Brown     // Code from DMPlexExtrude
43204663dae6SJed Brown     PetscCall(DMPlexTransformCreate(PetscObjectComm((PetscObject)dm), &tr));
43214663dae6SJed Brown     PetscCall(DMPlexTransformSetDM(tr, dm));
4322ce78bad3SBarry Smith     PetscCall(DMPlexTransformSetType(tr, DMPLEXEXTRUDETYPE));
43234663dae6SJed Brown     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
43244663dae6SJed Brown     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tr, prefix));
43254663dae6SJed Brown     PetscCall(PetscObjectGetOptions((PetscObject)dm, &options));
43264663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, options));
43274663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetLayers(tr, layers));
43284663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetThickness(tr, thickness));
43294663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetTensor(tr, PETSC_FALSE));
43304663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetSymmetric(tr, PETSC_TRUE));
43314663dae6SJed Brown     PetscCall(DMPlexTransformExtrudeSetNormalFunction(tr, normalFunc));
43324663dae6SJed Brown     PetscCall(DMPlexTransformSetFromOptions(tr));
43334663dae6SJed Brown     PetscCall(PetscObjectSetOptions((PetscObject)tr, NULL));
43344663dae6SJed Brown     PetscCall(DMPlexTransformSetUp(tr));
43354663dae6SJed Brown     PetscCall(PetscObjectViewFromOptions((PetscObject)tr, NULL, "-dm_plex_tps_transform_view"));
43364663dae6SJed Brown     PetscCall(DMPlexTransformApply(tr, dm, &edm));
43374663dae6SJed Brown     PetscCall(DMCopyDisc(dm, edm));
43384663dae6SJed Brown     PetscCall(DMGetCoordinateDM(dm, &cdm));
43394663dae6SJed Brown     PetscCall(DMGetCoordinateDM(edm, &ecdm));
43404663dae6SJed Brown     PetscCall(DMCopyDisc(cdm, ecdm));
43414663dae6SJed Brown     PetscCall(DMPlexTransformCreateDiscLabels(tr, edm));
43424663dae6SJed Brown     PetscCall(DMPlexTransformDestroy(&tr));
4343a77a5016SMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, edm));
434469d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
4345b7f5c055SJed Brown   }
43463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4347b7f5c055SJed Brown }
4348b7f5c055SJed Brown 
4349b7f5c055SJed Brown /*@
4350b7f5c055SJed Brown   DMPlexCreateTPSMesh - Create a distributed, interpolated mesh of a triply-periodic surface
4351b7f5c055SJed Brown 
4352b7f5c055SJed Brown   Collective
4353b7f5c055SJed Brown 
4354b7f5c055SJed Brown   Input Parameters:
4355a1cb98faSBarry Smith + comm           - The communicator for the `DM` object
4356b7f5c055SJed Brown . tpstype        - Type of triply-periodic surface
4357b7f5c055SJed Brown . extent         - Array of length 3 containing number of periods in each direction
435820f4b53cSBarry Smith . periodic       - array of length 3 with periodicity, or `NULL` for non-periodic
43591436d7faSJed Brown . tps_distribute - Distribute 2D manifold mesh prior to refinement and extrusion (more scalable)
4360817da375SSatish Balay . refinements    - Number of factor-of-2 refinements of 2D manifold mesh
43611436d7faSJed Brown . layers         - Number of cell layers extruded in normal direction
4362817da375SSatish Balay - thickness      - Thickness in normal direction
4363b7f5c055SJed Brown 
4364b7f5c055SJed Brown   Output Parameter:
4365a1cb98faSBarry Smith . dm - The `DM` object
4366a1cb98faSBarry Smith 
4367a1cb98faSBarry Smith   Level: beginner
4368b7f5c055SJed Brown 
4369b7f5c055SJed Brown   Notes:
437015229ffcSPierre Jolivet   This meshes the surface of the Schwarz P or Gyroid surfaces.  Schwarz P is the simplest member of the triply-periodic minimal surfaces.
43711d27aa22SBarry Smith   <https://en.wikipedia.org/wiki/Schwarz_minimal_surface#Schwarz_P_(%22Primitive%22)> and can be cut with "clean" boundaries.
43721d27aa22SBarry 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.
4373b7f5c055SJed Brown   Our implementation creates a very coarse mesh of the surface and refines (by 4-way splitting) as many times as requested.
4374b7f5c055SJed Brown   On each refinement, all vertices are projected to their nearest point on the surface.
4375b7f5c055SJed Brown   This projection could readily be extended to related surfaces.
4376b7f5c055SJed Brown 
43771d27aa22SBarry Smith   See {cite}`maskery2018insights`
43781d27aa22SBarry Smith 
43791d27aa22SBarry Smith   The face (edge) sets for the Schwarz P surface are numbered $1(-x), 2(+x), 3(-y), 4(+y), 5(-z), 6(+z)$.
43801d27aa22SBarry Smith   When the mesh is refined, "Face Sets" contain the new vertices (created during refinement).
43811d27aa22SBarry Smith   Use `DMPlexLabelComplete()` to propagate to coarse-level vertices.
4382b7f5c055SJed Brown 
438360225df5SJacob Faibussowitsch   Developer Notes:
4384b7f5c055SJed Brown   The Gyroid mesh does not currently mark boundary sets.
4385b7f5c055SJed Brown 
43861cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMSetType()`, `DMCreate()`
4387b7f5c055SJed Brown @*/
4388d71ae5a4SJacob 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)
4389d71ae5a4SJacob Faibussowitsch {
4390b7f5c055SJed Brown   PetscFunctionBegin;
43919566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
43929566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
43939566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateTPSMesh_Internal(*dm, tpstype, extent, periodic, tps_distribute, refinements, layers, thickness));
43943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4395b7f5c055SJed Brown }
4396b7f5c055SJed Brown 
43979318fe57SMatthew G. Knepley /*@
43989318fe57SMatthew G. Knepley   DMPlexCreateSphereMesh - Creates a mesh on the d-dimensional sphere, S^d.
43999318fe57SMatthew G. Knepley 
44009318fe57SMatthew G. Knepley   Collective
44019318fe57SMatthew G. Knepley 
44029318fe57SMatthew G. Knepley   Input Parameters:
4403a1cb98faSBarry Smith + comm    - The communicator for the `DM` object
44049318fe57SMatthew G. Knepley . dim     - The dimension
44059318fe57SMatthew G. Knepley . simplex - Use simplices, or tensor product cells
44069318fe57SMatthew G. Knepley - R       - The radius
44079318fe57SMatthew G. Knepley 
44089318fe57SMatthew G. Knepley   Output Parameter:
4409a1cb98faSBarry Smith . dm - The `DM` object
44109318fe57SMatthew G. Knepley 
44119318fe57SMatthew G. Knepley   Level: beginner
44129318fe57SMatthew G. Knepley 
44131cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBallMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
44149318fe57SMatthew G. Knepley @*/
4415d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateSphereMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscReal R, DM *dm)
4416d71ae5a4SJacob Faibussowitsch {
44179318fe57SMatthew G. Knepley   PetscFunctionBegin;
44184f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
44199566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
44209566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
44219566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(*dm, dim, simplex, R));
44223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44239318fe57SMatthew G. Knepley }
44249318fe57SMatthew G. Knepley 
4425d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBallMesh_Internal(DM dm, PetscInt dim, PetscReal R)
4426d71ae5a4SJacob Faibussowitsch {
44279318fe57SMatthew G. Knepley   DM          sdm, vol;
44289318fe57SMatthew G. Knepley   DMLabel     bdlabel;
4429dd2b43ebSStefano Zampini   const char *prefix;
44309318fe57SMatthew G. Knepley 
44319318fe57SMatthew G. Knepley   PetscFunctionBegin;
44329566063dSJacob Faibussowitsch   PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &sdm));
44339566063dSJacob Faibussowitsch   PetscCall(DMSetType(sdm, DMPLEX));
4434dd2b43ebSStefano Zampini   PetscCall(DMGetOptionsPrefix(dm, &prefix));
4435dd2b43ebSStefano Zampini   PetscCall(DMSetOptionsPrefix(sdm, prefix));
4436dd2b43ebSStefano Zampini   PetscCall(DMAppendOptionsPrefix(sdm, "bd_"));
4437dd2b43ebSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(sdm, PETSC_FALSE));
44389566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateSphereMesh_Internal(sdm, dim - 1, PETSC_TRUE, R));
44399566063dSJacob Faibussowitsch   PetscCall(DMSetFromOptions(sdm));
44409566063dSJacob Faibussowitsch   PetscCall(DMViewFromOptions(sdm, NULL, "-dm_view"));
44419566063dSJacob Faibussowitsch   PetscCall(DMPlexGenerate(sdm, NULL, PETSC_TRUE, &vol));
44429566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&sdm));
444369d8a87bSksagiyam   PetscCall(DMPlexReplace_Internal(dm, &vol));
44449566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "marker"));
44459566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "marker", &bdlabel));
44469566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel));
44479566063dSJacob Faibussowitsch   PetscCall(DMPlexLabelComplete(dm, bdlabel));
44483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
444951a74b61SMatthew G. Knepley }
445051a74b61SMatthew G. Knepley 
445151a74b61SMatthew G. Knepley /*@
445251a74b61SMatthew G. Knepley   DMPlexCreateBallMesh - Creates a simplex mesh on the d-dimensional ball, B^d.
445351a74b61SMatthew G. Knepley 
445451a74b61SMatthew G. Knepley   Collective
445551a74b61SMatthew G. Knepley 
445651a74b61SMatthew G. Knepley   Input Parameters:
4457a1cb98faSBarry Smith + comm - The communicator for the `DM` object
445851a74b61SMatthew G. Knepley . dim  - The dimension
445951a74b61SMatthew G. Knepley - R    - The radius
446051a74b61SMatthew G. Knepley 
446151a74b61SMatthew G. Knepley   Output Parameter:
4462a1cb98faSBarry Smith . dm - The `DM` object
446351a74b61SMatthew G. Knepley 
4464a1cb98faSBarry Smith   Options Database Key:
446560225df5SJacob Faibussowitsch . bd_dm_refine - This will refine the surface mesh preserving the sphere geometry
446651a74b61SMatthew G. Knepley 
446751a74b61SMatthew G. Knepley   Level: beginner
446851a74b61SMatthew G. Knepley 
44691cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()`
447051a74b61SMatthew G. Knepley @*/
4471d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBallMesh(MPI_Comm comm, PetscInt dim, PetscReal R, DM *dm)
4472d71ae5a4SJacob Faibussowitsch {
447351a74b61SMatthew G. Knepley   PetscFunctionBegin;
44749566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
44759566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
44769566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateBallMesh_Internal(*dm, dim, R));
44773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44782829fed8SMatthew G. Knepley }
44792829fed8SMatthew G. Knepley 
4480d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateReferenceCell_Internal(DM rdm, DMPolytopeType ct)
4481d71ae5a4SJacob Faibussowitsch {
44820a6ba040SMatthew G. Knepley   PetscFunctionBegin;
44839318fe57SMatthew G. Knepley   switch (ct) {
44849371c9d4SSatish Balay   case DM_POLYTOPE_POINT: {
44859318fe57SMatthew G. Knepley     PetscInt    numPoints[1]        = {1};
44869318fe57SMatthew G. Knepley     PetscInt    coneSize[1]         = {0};
44879318fe57SMatthew G. Knepley     PetscInt    cones[1]            = {0};
44889318fe57SMatthew G. Knepley     PetscInt    coneOrientations[1] = {0};
44899318fe57SMatthew G. Knepley     PetscScalar vertexCoords[1]     = {0.0};
44909318fe57SMatthew G. Knepley 
44919566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 0));
44929566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords));
44939371c9d4SSatish Balay   } break;
44949371c9d4SSatish Balay   case DM_POLYTOPE_SEGMENT: {
44959318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
44969318fe57SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
44979318fe57SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
44989318fe57SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
44999318fe57SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
45009318fe57SMatthew G. Knepley 
45019566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
45029566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45039371c9d4SSatish Balay   } break;
45049371c9d4SSatish Balay   case DM_POLYTOPE_POINT_PRISM_TENSOR: {
4505b5a892a1SMatthew G. Knepley     PetscInt    numPoints[2]        = {2, 1};
4506b5a892a1SMatthew G. Knepley     PetscInt    coneSize[3]         = {2, 0, 0};
4507b5a892a1SMatthew G. Knepley     PetscInt    cones[2]            = {1, 2};
4508b5a892a1SMatthew G. Knepley     PetscInt    coneOrientations[2] = {0, 0};
4509b5a892a1SMatthew G. Knepley     PetscScalar vertexCoords[2]     = {-1.0, 1.0};
4510b5a892a1SMatthew G. Knepley 
45119566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 1));
45129566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45139371c9d4SSatish Balay   } break;
45149371c9d4SSatish Balay   case DM_POLYTOPE_TRIANGLE: {
45159318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {3, 1};
45169318fe57SMatthew G. Knepley     PetscInt    coneSize[4]         = {3, 0, 0, 0};
45179318fe57SMatthew G. Knepley     PetscInt    cones[3]            = {1, 2, 3};
45189318fe57SMatthew G. Knepley     PetscInt    coneOrientations[3] = {0, 0, 0};
45199318fe57SMatthew G. Knepley     PetscScalar vertexCoords[6]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0};
45209318fe57SMatthew G. Knepley 
45219566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
45229566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45239371c9d4SSatish Balay   } break;
45249371c9d4SSatish Balay   case DM_POLYTOPE_QUADRILATERAL: {
45259318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
45269318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
45279318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
45289318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
45299318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0};
45309318fe57SMatthew G. Knepley 
45319566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
45329566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45339371c9d4SSatish Balay   } break;
45349371c9d4SSatish Balay   case DM_POLYTOPE_SEG_PRISM_TENSOR: {
45359318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
45369318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
45379318fe57SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
45389318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
45399318fe57SMatthew G. Knepley     PetscScalar vertexCoords[8]     = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0};
45409318fe57SMatthew G. Knepley 
45419566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 2));
45429566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45439371c9d4SSatish Balay   } break;
45449371c9d4SSatish Balay   case DM_POLYTOPE_TETRAHEDRON: {
45459318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {4, 1};
45469318fe57SMatthew G. Knepley     PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
4547f0edb160SMatthew G. Knepley     PetscInt    cones[4]            = {1, 2, 3, 4};
45489318fe57SMatthew G. Knepley     PetscInt    coneOrientations[4] = {0, 0, 0, 0};
4549f0edb160SMatthew 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};
45509318fe57SMatthew G. Knepley 
45519566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
45529566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45539371c9d4SSatish Balay   } break;
45549371c9d4SSatish Balay   case DM_POLYTOPE_HEXAHEDRON: {
45559318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
45569318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
4557f0edb160SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
45589318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
45599371c9d4SSatish 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};
45609318fe57SMatthew G. Knepley 
45619566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
45629566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45639371c9d4SSatish Balay   } break;
45649371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM: {
45659318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
45669318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
4567f0edb160SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
45689318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
45699371c9d4SSatish 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};
45709318fe57SMatthew G. Knepley 
45719566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
45729566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45739371c9d4SSatish Balay   } break;
45749371c9d4SSatish Balay   case DM_POLYTOPE_TRI_PRISM_TENSOR: {
45759318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {6, 1};
45769318fe57SMatthew G. Knepley     PetscInt    coneSize[7]         = {6, 0, 0, 0, 0, 0, 0};
45779318fe57SMatthew G. Knepley     PetscInt    cones[6]            = {1, 2, 3, 4, 5, 6};
45789318fe57SMatthew G. Knepley     PetscInt    coneOrientations[6] = {0, 0, 0, 0, 0, 0};
45799371c9d4SSatish 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};
45809318fe57SMatthew G. Knepley 
45819566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
45829566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45839371c9d4SSatish Balay   } break;
45849371c9d4SSatish Balay   case DM_POLYTOPE_QUAD_PRISM_TENSOR: {
45859318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {8, 1};
45869318fe57SMatthew G. Knepley     PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
45879318fe57SMatthew G. Knepley     PetscInt    cones[8]            = {1, 2, 3, 4, 5, 6, 7, 8};
45889318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
45899371c9d4SSatish 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};
45909318fe57SMatthew G. Knepley 
45919566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
45929566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
45939371c9d4SSatish Balay   } break;
45949371c9d4SSatish Balay   case DM_POLYTOPE_PYRAMID: {
45959318fe57SMatthew G. Knepley     PetscInt    numPoints[2]        = {5, 1};
45969318fe57SMatthew G. Knepley     PetscInt    coneSize[6]         = {5, 0, 0, 0, 0, 0};
4597f0edb160SMatthew G. Knepley     PetscInt    cones[5]            = {1, 2, 3, 4, 5};
45989318fe57SMatthew G. Knepley     PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
45999371c9d4SSatish 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};
46009318fe57SMatthew G. Knepley 
46019566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(rdm, 3));
46029566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords));
46039371c9d4SSatish Balay   } break;
4604d71ae5a4SJacob Faibussowitsch   default:
4605d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)rdm), PETSC_ERR_ARG_WRONG, "Cannot create reference cell for cell type %s", DMPolytopeTypes[ct]);
46069318fe57SMatthew G. Knepley   }
46079318fe57SMatthew G. Knepley   {
46089318fe57SMatthew G. Knepley     PetscInt Nv, v;
46099318fe57SMatthew G. Knepley 
46109318fe57SMatthew G. Knepley     /* Must create the celltype label here so that we do not automatically try to compute the types */
46119566063dSJacob Faibussowitsch     PetscCall(DMCreateLabel(rdm, "celltype"));
46129566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCellType(rdm, 0, ct));
46139566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(rdm, NULL, &Nv));
46149566063dSJacob Faibussowitsch     for (v = 1; v < Nv; ++v) PetscCall(DMPlexSetCellType(rdm, v, DM_POLYTOPE_POINT));
46159318fe57SMatthew G. Knepley   }
46169566063dSJacob Faibussowitsch   PetscCall(DMPlexInterpolateInPlace_Internal(rdm));
46179566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)rdm, DMPolytopeTypes[ct]));
46183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46190a6ba040SMatthew G. Knepley }
46200a6ba040SMatthew G. Knepley 
46219318fe57SMatthew G. Knepley /*@
4622a1cb98faSBarry Smith   DMPlexCreateReferenceCell - Create a `DMPLEX` with the appropriate FEM reference cell
46239318fe57SMatthew G. Knepley 
46249318fe57SMatthew G. Knepley   Collective
46259318fe57SMatthew G. Knepley 
46269318fe57SMatthew G. Knepley   Input Parameters:
46279318fe57SMatthew G. Knepley + comm - The communicator
46289318fe57SMatthew G. Knepley - ct   - The cell type of the reference cell
46299318fe57SMatthew G. Knepley 
46309318fe57SMatthew G. Knepley   Output Parameter:
46319318fe57SMatthew G. Knepley . refdm - The reference cell
46329318fe57SMatthew G. Knepley 
46339318fe57SMatthew G. Knepley   Level: intermediate
46349318fe57SMatthew G. Knepley 
463542747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()`
46369318fe57SMatthew G. Knepley @*/
4637d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, DMPolytopeType ct, DM *refdm)
4638d71ae5a4SJacob Faibussowitsch {
46390a6ba040SMatthew G. Knepley   PetscFunctionBegin;
46409566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, refdm));
46419566063dSJacob Faibussowitsch   PetscCall(DMSetType(*refdm, DMPLEX));
46429566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceCell_Internal(*refdm, ct));
46433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46449318fe57SMatthew G. Knepley }
464579a015ccSMatthew G. Knepley 
4646d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoundaryLabel_Private(DM dm, const char name[])
4647d71ae5a4SJacob Faibussowitsch {
46489318fe57SMatthew G. Knepley   DM        plex;
46499318fe57SMatthew G. Knepley   DMLabel   label;
46509318fe57SMatthew G. Knepley   PetscBool hasLabel;
46510a6ba040SMatthew G. Knepley 
4652c22d3578SMatthew G. Knepley   PetscFunctionBegin;
46539566063dSJacob Faibussowitsch   PetscCall(DMHasLabel(dm, name, &hasLabel));
46543ba16761SJacob Faibussowitsch   if (hasLabel) PetscFunctionReturn(PETSC_SUCCESS);
46559566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, name));
46569566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, name, &label));
46579566063dSJacob Faibussowitsch   PetscCall(DMConvert(dm, DMPLEX, &plex));
46589566063dSJacob Faibussowitsch   PetscCall(DMPlexMarkBoundaryFaces(plex, 1, label));
46591c8afea9SMatthew G. Knepley   PetscCall(DMPlexLabelComplete(plex, label));
46609566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&plex));
46613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46629318fe57SMatthew G. Knepley }
4663acdc6f61SToby Isaac 
4664669647acSMatthew G. Knepley /*
4665669647acSMatthew 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.
4666669647acSMatthew G. Knepley 
4667669647acSMatthew G. Knepley     (x, y) -> (r, theta) = (x[1], (x[0] - lower[0]) * 2\pi/(upper[0] - lower[0]))
4668669647acSMatthew G. Knepley */
4669d71ae5a4SJacob 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[])
4670d71ae5a4SJacob Faibussowitsch {
4671669647acSMatthew G. Knepley   const PetscReal low = PetscRealPart(constants[0]);
4672669647acSMatthew G. Knepley   const PetscReal upp = PetscRealPart(constants[1]);
4673669647acSMatthew G. Knepley   const PetscReal r   = PetscRealPart(u[1]);
4674669647acSMatthew G. Knepley   const PetscReal th  = 2. * PETSC_PI * (PetscRealPart(u[0]) - low) / (upp - low);
4675669647acSMatthew G. Knepley 
4676669647acSMatthew G. Knepley   f0[0] = r * PetscCosReal(th);
4677669647acSMatthew G. Knepley   f0[1] = r * PetscSinReal(th);
4678669647acSMatthew G. Knepley }
4679669647acSMatthew G. Knepley 
46805390be7dSMatthew G. Knepley // Insert vertices and their joins, marked by depth
46815390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Vertices(DM dm, DMLabel label, DMLabel vlabel, PetscInt val, PetscInt n, const PetscInt vertices[])
46825390be7dSMatthew G. Knepley {
46835390be7dSMatthew G. Knepley   PetscFunctionBegin;
46845390be7dSMatthew G. Knepley   PetscCall(DMPlexMarkSubmesh_Interpolated(dm, vlabel, val, PETSC_FALSE, PETSC_FALSE, label, NULL));
46855390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
46865390be7dSMatthew G. Knepley }
46875390be7dSMatthew G. Knepley 
46885390be7dSMatthew G. Knepley // Insert faces and their closures, marked by depth
46895390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Faces(DM dm, DMLabel label, PetscInt n, const PetscInt faces[])
46905390be7dSMatthew G. Knepley {
46915390be7dSMatthew G. Knepley   PetscFunctionBegin;
46925390be7dSMatthew G. Knepley   for (PetscInt p = 0; p < n; ++p) {
46935390be7dSMatthew G. Knepley     const PetscInt point   = faces[p];
46945390be7dSMatthew G. Knepley     PetscInt      *closure = NULL;
46955390be7dSMatthew G. Knepley     PetscInt       clSize, pdepth;
46965390be7dSMatthew G. Knepley 
46975390be7dSMatthew G. Knepley     PetscCall(DMPlexGetPointDepth(dm, point, &pdepth));
46985390be7dSMatthew G. Knepley     PetscCall(DMLabelSetValue(label, point, pdepth));
46995390be7dSMatthew G. Knepley     PetscCall(DMPlexGetTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
47005390be7dSMatthew G. Knepley     for (PetscInt cl = 0; cl < clSize * 2; cl += 2) {
47015390be7dSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, closure[cl], &pdepth));
47025390be7dSMatthew G. Knepley       PetscCall(DMLabelSetValue(label, closure[cl], pdepth));
47035390be7dSMatthew G. Knepley     }
47045390be7dSMatthew G. Knepley     PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure));
47055390be7dSMatthew G. Knepley   }
47065390be7dSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
47075390be7dSMatthew G. Knepley }
47085390be7dSMatthew G. Knepley 
47094e22dd4cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions, const char pre[], const char name[], const char *option[], const char *value[], PetscBool *flg);
47104e22dd4cSMatthew G. Knepley 
47115dca41c3SJed Brown const char *const DMPlexShapes[] = {"box", "box_surface", "ball", "sphere", "cylinder", "schwarz_p", "gyroid", "doublet", "annulus", "hypercubic", "zbox", "unknown", "DMPlexShape", "DM_SHAPE_", NULL};
47129318fe57SMatthew G. Knepley 
4713ce78bad3SBarry Smith static PetscErrorCode DMPlexCreateFromOptions_Internal(PetscOptionItems PetscOptionsObject, PetscBool *useCoordSpace, DM dm)
4714d71ae5a4SJacob Faibussowitsch {
47159318fe57SMatthew G. Knepley   DMPlexShape    shape   = DM_SHAPE_BOX;
47169318fe57SMatthew G. Knepley   DMPolytopeType cell    = DM_POLYTOPE_TRIANGLE;
47179318fe57SMatthew G. Knepley   PetscInt       dim     = 2;
4718b9da1bb3SMatthew G. Knepley   PetscBool      simplex = PETSC_TRUE, interpolate = PETSC_TRUE, orient = PETSC_FALSE, adjCone = PETSC_FALSE, adjClosure = PETSC_TRUE, refDomain = PETSC_FALSE;
4719d0812dedSMatthew G. Knepley   PetscBool      flg, flg2, fflg, strflg, bdfflg, nameflg;
47209318fe57SMatthew G. Knepley   MPI_Comm       comm;
4721ed5e4e85SVaclav Hapla   char           filename[PETSC_MAX_PATH_LEN]   = "<unspecified>";
4722ed5e4e85SVaclav Hapla   char           bdFilename[PETSC_MAX_PATH_LEN] = "<unspecified>";
4723ed5e4e85SVaclav Hapla   char           plexname[PETSC_MAX_PATH_LEN]   = "";
47244e22dd4cSMatthew G. Knepley   const char    *option;
47259318fe57SMatthew G. Knepley 
47269318fe57SMatthew G. Knepley   PetscFunctionBegin;
4727708be2fdSJed Brown   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
47289566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
47299318fe57SMatthew G. Knepley   /* TODO Turn this into a registration interface */
47309566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_filename", "File containing a mesh", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &fflg));
4731d0812dedSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_file_contents", "Contents of a file format in a string", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &strflg));
47329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_filename", "File containing a mesh boundary", "DMPlexCreateFromFile", bdFilename, bdFilename, sizeof(bdFilename), &bdfflg));
47339566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_name", "Name of the mesh in the file", "DMPlexCreateFromFile", plexname, plexname, sizeof(plexname), &nameflg));
47349566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_cell", "Cell shape", "", DMPolytopeTypes, (PetscEnum)cell, (PetscEnum *)&cell, NULL));
47359566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_reference_cell_domain", "Use a reference cell domain", "", refDomain, &refDomain, NULL));
47369566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-dm_plex_shape", "Shape for built-in mesh", "", DMPlexShapes, (PetscEnum)shape, (PetscEnum *)&shape, &flg));
47379566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_dim", "Topological dimension of the mesh", "DMGetDimension", dim, &dim, &flg, 0));
47389566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_simplex", "Mesh cell shape", "", simplex, &simplex, &flg));
47399566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate", "Flag to create edges and faces automatically", "", interpolate, &interpolate, &flg));
4740b9da1bb3SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_plex_orient", "Orient the constructed mesh", "DMPlexOrient", orient, &orient, &flg));
47419566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_cone", "Set adjacency direction", "DMSetBasicAdjacency", adjCone, &adjCone, &flg));
47429566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_adj_closure", "Set adjacency size", "DMSetBasicAdjacency", adjClosure, &adjClosure, &flg2));
47439566063dSJacob Faibussowitsch   if (flg || flg2) PetscCall(DMSetBasicAdjacency(dm, adjCone, adjClosure));
47443f3e541fSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_adj", "Debug output level all adjacency computations", "", 0, &((DM_Plex *)dm->data)->printAdj, NULL, 0));
47459318fe57SMatthew G. Knepley 
474661a622f3SMatthew G. Knepley   switch (cell) {
474761a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT:
474861a622f3SMatthew G. Knepley   case DM_POLYTOPE_SEGMENT:
474961a622f3SMatthew G. Knepley   case DM_POLYTOPE_POINT_PRISM_TENSOR:
475061a622f3SMatthew G. Knepley   case DM_POLYTOPE_TRIANGLE:
475161a622f3SMatthew G. Knepley   case DM_POLYTOPE_QUADRILATERAL:
475261a622f3SMatthew G. Knepley   case DM_POLYTOPE_TETRAHEDRON:
4753d71ae5a4SJacob Faibussowitsch   case DM_POLYTOPE_HEXAHEDRON:
4754d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_TRUE;
4755d71ae5a4SJacob Faibussowitsch     break;
4756d71ae5a4SJacob Faibussowitsch   default:
4757d71ae5a4SJacob Faibussowitsch     *useCoordSpace = PETSC_FALSE;
4758d71ae5a4SJacob Faibussowitsch     break;
475961a622f3SMatthew G. Knepley   }
476061a622f3SMatthew G. Knepley 
47619318fe57SMatthew G. Knepley   if (fflg) {
47629318fe57SMatthew G. Knepley     DM          dmnew;
47631e4a82c4SMatthew G. Knepley     const char *name;
47649318fe57SMatthew G. Knepley 
47651e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
47661e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), filename, nameflg ? plexname : name, interpolate, &dmnew));
47675de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
476869d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
47699318fe57SMatthew G. Knepley   } else if (refDomain) {
47709566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateReferenceCell_Internal(dm, cell));
47719318fe57SMatthew G. Knepley   } else if (bdfflg) {
47729318fe57SMatthew G. Knepley     DM          bdm, dmnew;
47731e4a82c4SMatthew G. Knepley     const char *name;
47749318fe57SMatthew G. Knepley 
47751e4a82c4SMatthew G. Knepley     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
47761e4a82c4SMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), bdFilename, nameflg ? plexname : name, interpolate, &bdm));
47779566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)bdm, "bd_"));
47789566063dSJacob Faibussowitsch     PetscCall(DMSetFromOptions(bdm));
47799566063dSJacob Faibussowitsch     PetscCall(DMPlexGenerate(bdm, NULL, interpolate, &dmnew));
47809566063dSJacob Faibussowitsch     PetscCall(DMDestroy(&bdm));
47815de52c6dSVaclav Hapla     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
478269d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
4783d0812dedSMatthew G. Knepley   } else if (strflg) {
4784d0812dedSMatthew G. Knepley     DM          dmnew;
4785d0812dedSMatthew G. Knepley     PetscViewer viewer;
4786d0812dedSMatthew G. Knepley     const char *contents;
4787d0812dedSMatthew G. Knepley     char       *strname;
4788d0812dedSMatthew G. Knepley     char        tmpdir[PETSC_MAX_PATH_LEN];
4789d0812dedSMatthew G. Knepley     char        tmpfilename[PETSC_MAX_PATH_LEN];
4790d0812dedSMatthew G. Knepley     char        name[PETSC_MAX_PATH_LEN];
4791d0812dedSMatthew G. Knepley     MPI_Comm    comm;
4792d0812dedSMatthew G. Knepley     PetscMPIInt rank;
4793d0812dedSMatthew G. Knepley 
4794d0812dedSMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
4795d0812dedSMatthew G. Knepley     PetscCallMPI(MPI_Comm_rank(comm, &rank));
4796d0812dedSMatthew G. Knepley     PetscCall(PetscStrchr(filename, ':', &strname));
4797d0812dedSMatthew G. Knepley     PetscCheck(strname, comm, PETSC_ERR_ARG_WRONG, "File contents must have the form \"ext:string_name\", not %s", filename);
4798d0812dedSMatthew G. Knepley     strname[0] = '\0';
4799d0812dedSMatthew G. Knepley     ++strname;
4800d0812dedSMatthew G. Knepley     PetscCall(PetscDLSym(NULL, strname, (void **)&contents));
4801d0812dedSMatthew G. Knepley     PetscCheck(contents, comm, PETSC_ERR_ARG_WRONG, "Could not locate mesh string %s", strname);
4802d0812dedSMatthew G. Knepley     PetscCall(PetscGetTmp(comm, tmpdir, PETSC_MAX_PATH_LEN));
4803ed32af8cSMatthew G. Knepley     PetscCall(PetscStrlcat(tmpdir, "/meshXXXXXX", PETSC_MAX_PATH_LEN));
4804ed32af8cSMatthew G. Knepley     PetscCall(PetscMkdtemp(tmpdir));
4805ed32af8cSMatthew G. Knepley     PetscCall(PetscSNPrintf(tmpfilename, PETSC_MAX_PATH_LEN, "%s/mesh.%s", tmpdir, filename));
4806d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIOpen(comm, tmpfilename, &viewer));
4807d0812dedSMatthew G. Knepley     PetscCall(PetscViewerASCIIPrintf(viewer, "%s\n", contents));
4808d0812dedSMatthew G. Knepley     PetscCall(PetscViewerDestroy(&viewer));
4809d0812dedSMatthew G. Knepley     PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), tmpfilename, plexname, interpolate, &dmnew));
4810ed32af8cSMatthew G. Knepley     PetscCall(PetscRMTree(tmpdir));
4811d0812dedSMatthew G. Knepley     PetscCall(PetscSNPrintf(name, PETSC_MAX_PATH_LEN, "%s Mesh", strname));
4812d0812dedSMatthew G. Knepley     PetscCall(PetscObjectSetName((PetscObject)dm, name));
4813d0812dedSMatthew G. Knepley     PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
4814d0812dedSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &dmnew));
48159318fe57SMatthew G. Knepley   } else {
48169566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)dm, DMPlexShapes[shape]));
48179318fe57SMatthew G. Knepley     switch (shape) {
4818669647acSMatthew G. Knepley     case DM_SHAPE_BOX:
48195dca41c3SJed Brown     case DM_SHAPE_ZBOX:
4820669647acSMatthew G. Knepley     case DM_SHAPE_ANNULUS: {
48219318fe57SMatthew G. Knepley       PetscInt       faces[3]  = {0, 0, 0};
48229318fe57SMatthew G. Knepley       PetscReal      lower[3]  = {0, 0, 0};
48239318fe57SMatthew G. Knepley       PetscReal      upper[3]  = {1, 1, 1};
48249318fe57SMatthew G. Knepley       DMBoundaryType bdt[3]    = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4825669647acSMatthew G. Knepley       PetscBool      isAnnular = shape == DM_SHAPE_ANNULUS ? PETSC_TRUE : PETSC_FALSE;
48269318fe57SMatthew G. Knepley       PetscInt       i, n;
48279318fe57SMatthew G. Knepley 
48289318fe57SMatthew G. Knepley       n = dim;
48299318fe57SMatthew G. Knepley       for (i = 0; i < dim; ++i) faces[i] = (dim == 1 ? 1 : 4 - dim);
48309566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
48319318fe57SMatthew G. Knepley       n = 3;
48329566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
483363a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
48349318fe57SMatthew G. Knepley       n = 3;
48359566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
483663a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
48379318fe57SMatthew G. Knepley       n = 3;
48389566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
483963a3b9bcSJacob Faibussowitsch       PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
4840669647acSMatthew G. Knepley 
4841669647acSMatthew G. Knepley       PetscCheck(!isAnnular || dim == 2, comm, PETSC_ERR_ARG_OUTOFRANGE, "Only two dimensional annuli have been implemented");
4842669647acSMatthew G. Knepley       if (isAnnular)
4843669647acSMatthew G. Knepley         for (i = 0; i < dim - 1; ++i) bdt[i] = DM_BOUNDARY_PERIODIC;
4844669647acSMatthew G. Knepley 
48459318fe57SMatthew G. Knepley       switch (cell) {
484661a622f3SMatthew G. Knepley       case DM_POLYTOPE_TRI_PRISM_TENSOR:
48479566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeBoxMesh_Internal(dm, faces, lower, upper, bdt));
4848d410b0cfSMatthew G. Knepley         if (!interpolate) {
4849d410b0cfSMatthew G. Knepley           DM udm;
4850d410b0cfSMatthew G. Knepley 
48519566063dSJacob Faibussowitsch           PetscCall(DMPlexUninterpolate(dm, &udm));
485269d8a87bSksagiyam           PetscCall(DMPlexReplace_Internal(dm, &udm));
4853d410b0cfSMatthew G. Knepley         }
48549318fe57SMatthew G. Knepley         break;
4855d71ae5a4SJacob Faibussowitsch       default:
48565dca41c3SJed Brown         PetscCall(DMPlexCreateBoxMesh_Internal(dm, shape, dim, simplex, faces, lower, upper, bdt, interpolate));
4857d71ae5a4SJacob Faibussowitsch         break;
48589318fe57SMatthew G. Knepley       }
4859669647acSMatthew G. Knepley       if (isAnnular) {
4860669647acSMatthew G. Knepley         DM          cdm;
4861669647acSMatthew G. Knepley         PetscDS     cds;
4862669647acSMatthew G. Knepley         PetscScalar bounds[2] = {lower[0], upper[0]};
4863669647acSMatthew G. Knepley 
4864669647acSMatthew G. Knepley         // Fix coordinates for annular region
4865669647acSMatthew G. Knepley         PetscCall(DMSetPeriodicity(dm, NULL, NULL, NULL));
4866669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinatesLocal(dm, NULL));
4867669647acSMatthew G. Knepley         PetscCall(DMSetCellCoordinates(dm, NULL));
4868e44f6aebSMatthew G. Knepley         PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL));
4869669647acSMatthew G. Knepley         PetscCall(DMGetCoordinateDM(dm, &cdm));
4870669647acSMatthew G. Knepley         PetscCall(DMGetDS(cdm, &cds));
4871669647acSMatthew G. Knepley         PetscCall(PetscDSSetConstants(cds, 2, bounds));
4872669647acSMatthew G. Knepley         PetscCall(DMPlexRemapGeometry(dm, 0.0, boxToAnnulus));
4873669647acSMatthew G. Knepley       }
48749371c9d4SSatish Balay     } break;
48759371c9d4SSatish Balay     case DM_SHAPE_BOX_SURFACE: {
48769318fe57SMatthew G. Knepley       PetscInt  faces[3] = {0, 0, 0};
48779318fe57SMatthew G. Knepley       PetscReal lower[3] = {0, 0, 0};
48789318fe57SMatthew G. Knepley       PetscReal upper[3] = {1, 1, 1};
48799318fe57SMatthew G. Knepley       PetscInt  i, n;
48809318fe57SMatthew G. Knepley 
48819318fe57SMatthew G. Knepley       n = dim + 1;
48829318fe57SMatthew G. Knepley       for (i = 0; i < dim + 1; ++i) faces[i] = (dim + 1 == 1 ? 1 : 4 - (dim + 1));
48839566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg));
48849318fe57SMatthew G. Knepley       n = 3;
48859566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
488663a3b9bcSJacob 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);
48879318fe57SMatthew G. Knepley       n = 3;
48889566063dSJacob Faibussowitsch       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
488963a3b9bcSJacob 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);
48909566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(dm, dim + 1, faces, lower, upper, interpolate));
48919371c9d4SSatish Balay     } break;
48929371c9d4SSatish Balay     case DM_SHAPE_SPHERE: {
48939318fe57SMatthew G. Knepley       PetscReal R = 1.0;
48949318fe57SMatthew G. Knepley 
48959566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_sphere_radius", "Radius of the sphere", "", R, &R, &flg));
48969566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateSphereMesh_Internal(dm, dim, simplex, R));
48979371c9d4SSatish Balay     } break;
48989371c9d4SSatish Balay     case DM_SHAPE_BALL: {
48999318fe57SMatthew G. Knepley       PetscReal R = 1.0;
49009318fe57SMatthew G. Knepley 
49019566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_ball_radius", "Radius of the ball", "", R, &R, &flg));
49029566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateBallMesh_Internal(dm, dim, R));
49039371c9d4SSatish Balay     } break;
49049371c9d4SSatish Balay     case DM_SHAPE_CYLINDER: {
49059318fe57SMatthew G. Knepley       DMBoundaryType bdt = DM_BOUNDARY_NONE;
49069318fe57SMatthew G. Knepley       PetscInt       Nw  = 6;
490749704ca5SMatthew G. Knepley       PetscInt       Nr  = 0;
49089318fe57SMatthew G. Knepley 
49099566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnum("-dm_plex_cylinder_bd", "Boundary type in the z direction", "", DMBoundaryTypes, (PetscEnum)bdt, (PetscEnum *)&bdt, NULL));
49109566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_wedges", "Number of wedges around the cylinder", "", Nw, &Nw, NULL));
491149704ca5SMatthew G. Knepley       PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_refine", "Number of refinements before projection", "", Nr, &Nr, NULL));
49129318fe57SMatthew G. Knepley       switch (cell) {
4913d71ae5a4SJacob Faibussowitsch       case DM_POLYTOPE_TRI_PRISM_TENSOR:
4914d71ae5a4SJacob Faibussowitsch         PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(dm, Nw, interpolate));
4915d71ae5a4SJacob Faibussowitsch         break;
4916d71ae5a4SJacob Faibussowitsch       default:
491749704ca5SMatthew G. Knepley         PetscCall(DMPlexCreateHexCylinderMesh_Internal(dm, bdt, Nr));
4918d71ae5a4SJacob Faibussowitsch         break;
49199318fe57SMatthew G. Knepley       }
49209371c9d4SSatish Balay     } break;
4921b7f5c055SJed Brown     case DM_SHAPE_SCHWARZ_P: // fallthrough
49229371c9d4SSatish Balay     case DM_SHAPE_GYROID: {
4923b7f5c055SJed Brown       PetscInt       extent[3] = {1, 1, 1}, refine = 0, layers = 0, three;
4924b7f5c055SJed Brown       PetscReal      thickness   = 0.;
4925b7f5c055SJed Brown       DMBoundaryType periodic[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
4926b7f5c055SJed Brown       DMPlexTPSType  tps_type    = shape == DM_SHAPE_SCHWARZ_P ? DMPLEX_TPS_SCHWARZ_P : DMPLEX_TPS_GYROID;
49271436d7faSJed Brown       PetscBool      tps_distribute;
49289566063dSJacob Faibussowitsch       PetscCall(PetscOptionsIntArray("-dm_plex_tps_extent", "Number of replicas for each of three dimensions", NULL, extent, (three = 3, &three), NULL));
49299566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_refine", "Number of refinements", NULL, refine, &refine, NULL));
49309566063dSJacob Faibussowitsch       PetscCall(PetscOptionsEnumArray("-dm_plex_tps_periodic", "Periodicity in each of three dimensions", NULL, DMBoundaryTypes, (PetscEnum *)periodic, (three = 3, &three), NULL));
49319566063dSJacob Faibussowitsch       PetscCall(PetscOptionsInt("-dm_plex_tps_layers", "Number of layers in volumetric extrusion (or zero to not extrude)", NULL, layers, &layers, NULL));
49329566063dSJacob Faibussowitsch       PetscCall(PetscOptionsReal("-dm_plex_tps_thickness", "Thickness of volumetric extrusion", NULL, thickness, &thickness, NULL));
49339566063dSJacob Faibussowitsch       PetscCall(DMPlexDistributeGetDefault(dm, &tps_distribute));
49349566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_tps_distribute", "Distribute the 2D mesh prior to refinement and extrusion", NULL, tps_distribute, &tps_distribute, NULL));
49359566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateTPSMesh_Internal(dm, tps_type, extent, periodic, tps_distribute, refine, layers, thickness));
49369371c9d4SSatish Balay     } break;
49379371c9d4SSatish Balay     case DM_SHAPE_DOUBLET: {
493805bd46c0SStefano Zampini       DM        dmnew;
493905bd46c0SStefano Zampini       PetscReal rl = 0.0;
494005bd46c0SStefano Zampini 
494105bd46c0SStefano Zampini       PetscCall(PetscOptionsReal("-dm_plex_doublet_refinementlimit", "Refinement limit", NULL, rl, &rl, NULL));
494205bd46c0SStefano Zampini       PetscCall(DMPlexCreateDoublet(PetscObjectComm((PetscObject)dm), dim, simplex, interpolate, rl, &dmnew));
49435de52c6dSVaclav Hapla       PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew));
494469d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &dmnew));
49459371c9d4SSatish Balay     } break;
4946cfb853baSMatthew G. Knepley     case DM_SHAPE_HYPERCUBIC: {
49478d2ec52aSSatish Balay       PetscInt       *edges, overlap = 1;
4948cfb853baSMatthew G. Knepley       PetscReal      *lower, *upper;
4949cfb853baSMatthew G. Knepley       DMBoundaryType *bdt;
4950cfb853baSMatthew G. Knepley       PetscInt        n, d;
4951cfb853baSMatthew G. Knepley 
4952cfb853baSMatthew G. Knepley       *useCoordSpace = PETSC_FALSE;
4953cfb853baSMatthew G. Knepley       PetscCall(PetscMalloc4(dim, &edges, dim, &lower, dim, &upper, dim, &bdt));
4954cfb853baSMatthew G. Knepley       for (d = 0; d < dim; ++d) {
4955cfb853baSMatthew G. Knepley         edges[d] = 1;
4956cfb853baSMatthew G. Knepley         lower[d] = 0.;
4957cfb853baSMatthew G. Knepley         upper[d] = 1.;
4958cfb853baSMatthew G. Knepley         bdt[d]   = DM_BOUNDARY_PERIODIC;
4959cfb853baSMatthew G. Knepley       }
4960cfb853baSMatthew G. Knepley       n = dim;
4961cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", edges, &n, &flg));
4962cfb853baSMatthew G. Knepley       n = dim;
4963cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg));
4964cfb853baSMatthew 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);
4965cfb853baSMatthew G. Knepley       n = dim;
4966cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg));
4967cfb853baSMatthew 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);
4968cfb853baSMatthew G. Knepley       n = dim;
4969cfb853baSMatthew G. Knepley       PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
4970cfb853baSMatthew 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);
49718d2ec52aSSatish Balay       PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", overlap, &overlap, NULL, 0));
49728d2ec52aSSatish Balay       PetscCall(DMPlexCreateHypercubicMesh_Internal(dm, dim, lower, upper, edges, overlap, bdt));
4973cfb853baSMatthew G. Knepley       PetscCall(PetscFree4(edges, lower, upper, bdt));
4974cfb853baSMatthew G. Knepley     } break;
4975d71ae5a4SJacob Faibussowitsch     default:
4976d71ae5a4SJacob Faibussowitsch       SETERRQ(comm, PETSC_ERR_SUP, "Domain shape %s is unsupported", DMPlexShapes[shape]);
49779318fe57SMatthew G. Knepley     }
49789318fe57SMatthew G. Knepley   }
49799566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
498048a46eb9SPierre Jolivet   if (!((PetscObject)dm)->name && nameflg) PetscCall(PetscObjectSetName((PetscObject)dm, plexname));
4981b9da1bb3SMatthew G. Knepley   if (orient) PetscCall(DMPlexOrient(dm));
49824e22dd4cSMatthew G. Knepley   // Allow label creation
49834e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_label_", &option, NULL, &flg));
49844e22dd4cSMatthew G. Knepley   if (flg) {
49854e22dd4cSMatthew G. Knepley     DMLabel     label;
49864e22dd4cSMatthew G. Knepley     PetscInt    points[1024], n = 1024;
49874e22dd4cSMatthew G. Knepley     char        fulloption[PETSC_MAX_PATH_LEN];
49884e22dd4cSMatthew G. Knepley     const char *name = &option[14];
49894e22dd4cSMatthew G. Knepley 
49904e22dd4cSMatthew G. Knepley     PetscCall(DMCreateLabel(dm, name));
49914e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, name, &label));
49924e22dd4cSMatthew G. Knepley     fulloption[0] = '-';
49934e22dd4cSMatthew G. Knepley     fulloption[1] = 0;
49944e22dd4cSMatthew G. Knepley     PetscCall(PetscStrlcat(fulloption, option, PETSC_MAX_PATH_LEN));
49954e22dd4cSMatthew G. Knepley     PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, NULL));
49964e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < n; ++p) PetscCall(DMLabelSetValue(label, points[p], 1));
49974e22dd4cSMatthew G. Knepley   }
4998dd0eeac9SMatthew G. Knepley   // Allow cohesive label creation
4999dd0eeac9SMatthew G. Knepley   //   Faces are input, completed, and all points are marked with their depth
5000dd0eeac9SMatthew G. Knepley   PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_cohesive_label_", &option, NULL, &flg));
5001dd0eeac9SMatthew G. Knepley   if (flg) {
5002dd0eeac9SMatthew G. Knepley     DMLabel   label;
5003dd0eeac9SMatthew G. Knepley     PetscInt  points[1024], n, pStart, pEnd, Nl = 1;
50045390be7dSMatthew G. Knepley     PetscBool noCreate = PETSC_FALSE;
5005dd0eeac9SMatthew G. Knepley     char      fulloption[PETSC_MAX_PATH_LEN];
5006dd0eeac9SMatthew G. Knepley     char      name[PETSC_MAX_PATH_LEN];
5007dd0eeac9SMatthew G. Knepley     size_t    len;
5008dd0eeac9SMatthew G. Knepley 
5009dd0eeac9SMatthew G. Knepley     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
5010dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrncpy(name, &option[23], PETSC_MAX_PATH_LEN));
5011dd0eeac9SMatthew G. Knepley     PetscCall(PetscStrlen(name, &len));
5012dd0eeac9SMatthew G. Knepley     if (name[len - 1] == '0') Nl = 10;
5013dd0eeac9SMatthew G. Knepley     for (PetscInt l = 0; l < Nl; ++l) {
50146497c311SBarry Smith       if (l > 0) name[len - 1] = (char)('0' + l);
5015dd0eeac9SMatthew G. Knepley       fulloption[0] = 0;
5016dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, "-dm_plex_cohesive_label_", 32));
5017dd0eeac9SMatthew G. Knepley       PetscCall(PetscStrlcat(fulloption, name, PETSC_MAX_PATH_LEN - 32));
5018dd0eeac9SMatthew G. Knepley       n = 1024;
5019dd0eeac9SMatthew G. Knepley       PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, &flg));
5020dd0eeac9SMatthew G. Knepley       if (!flg) break;
50215390be7dSMatthew G. Knepley       PetscCall(DMHasLabel(dm, name, &noCreate));
50225390be7dSMatthew G. Knepley       if (noCreate) {
50235390be7dSMatthew G. Knepley         DMLabel         inlabel;
50245390be7dSMatthew G. Knepley         IS              pointIS;
50255390be7dSMatthew G. Knepley         const PetscInt *lpoints;
50265390be7dSMatthew G. Knepley         PetscInt        pdep, ln, inval = points[0];
50275390be7dSMatthew G. Knepley         char            newname[PETSC_MAX_PATH_LEN];
50285390be7dSMatthew G. Knepley 
50295390be7dSMatthew G. Knepley         PetscCheck(n == 1, comm, PETSC_ERR_ARG_WRONG, "Must specify a label value with this option");
50305390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &inlabel));
50315390be7dSMatthew G. Knepley         PetscCall(DMLabelGetStratumIS(inlabel, inval, &pointIS));
50325390be7dSMatthew G. Knepley         PetscCall(ISGetLocalSize(pointIS, &ln));
50335390be7dSMatthew G. Knepley         PetscCall(ISGetIndices(pointIS, &lpoints));
50345390be7dSMatthew G. Knepley         PetscCall(DMPlexGetPointDepth(dm, lpoints[0], &pdep));
50355390be7dSMatthew G. Knepley         PetscCall(PetscSNPrintf(newname, PETSC_MAX_PATH_LEN, "%s%" PetscInt_FMT, name, points[0]));
50365390be7dSMatthew G. Knepley         PetscCall(DMCreateLabel(dm, newname));
50375390be7dSMatthew G. Knepley         PetscCall(DMGetLabel(dm, newname, &label));
50385390be7dSMatthew G. Knepley         if (!pdep) PetscCall(ProcessCohesiveLabel_Vertices(dm, label, inlabel, inval, ln, lpoints));
50395390be7dSMatthew G. Knepley         else PetscCall(ProcessCohesiveLabel_Faces(dm, label, ln, lpoints));
50405390be7dSMatthew G. Knepley         PetscCall(ISRestoreIndices(pointIS, &lpoints));
50415390be7dSMatthew G. Knepley         PetscCall(ISDestroy(&pointIS));
50425390be7dSMatthew G. Knepley       } else {
5043dd0eeac9SMatthew G. Knepley         PetscCall(DMCreateLabel(dm, name));
5044dd0eeac9SMatthew G. Knepley         PetscCall(DMGetLabel(dm, name, &label));
5045dd0eeac9SMatthew G. Knepley         if (pStart >= pEnd) n = 0;
50465390be7dSMatthew G. Knepley         PetscCall(ProcessCohesiveLabel_Faces(dm, label, n, points));
5047dd0eeac9SMatthew G. Knepley       }
5048dd0eeac9SMatthew G. Knepley       PetscCall(DMPlexOrientLabel(dm, label));
50490542aa8cSMatthew G. Knepley       PetscCall(DMPlexLabelCohesiveComplete(dm, label, NULL, 1, PETSC_FALSE, PETSC_FALSE, NULL));
5050dd0eeac9SMatthew G. Knepley     }
5051dd0eeac9SMatthew G. Knepley   }
50525390be7dSMatthew G. Knepley   PetscCall(DMViewFromOptions(dm, NULL, "-created_dm_view"));
5053708be2fdSJed Brown   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromOptions, dm, 0, 0, 0));
50543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
50550a6ba040SMatthew G. Knepley }
50560a6ba040SMatthew G. Knepley 
5057ce78bad3SBarry Smith PetscErrorCode DMSetFromOptions_NonRefinement_Plex(DM dm, PetscOptionItems PetscOptionsObject)
5058d71ae5a4SJacob Faibussowitsch {
50590a6ba040SMatthew G. Knepley   DM_Plex  *mesh = (DM_Plex *)dm->data;
50607f9d8d6cSVaclav Hapla   PetscBool flg, flg2;
50619318fe57SMatthew G. Knepley   char      bdLabel[PETSC_MAX_PATH_LEN];
5062adc21957SMatthew G. Knepley   char      method[PETSC_MAX_PATH_LEN];
50630a6ba040SMatthew G. Knepley 
50640a6ba040SMatthew G. Knepley   PetscFunctionBegin;
50650a6ba040SMatthew G. Knepley   /* Handle viewing */
50669566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMPlexMatSetClosure", PETSC_FALSE, &mesh->printSetValues, NULL));
50675962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fem", "Debug output level for all fem computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printFEM, NULL, 0));
50685962854dSMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fvm", "Debug output level for all fvm computations", "DMPlexSNESComputeResidualFVM", 0, &mesh->printFVM, NULL, 0));
50699566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMPlexSNESComputeResidualFEM", mesh->printTol, &mesh->printTol, NULL));
50709566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_l2", "Debug output level all L2 diff computations", "DMComputeL2Diff", 0, &mesh->printL2, NULL, 0));
5071f5867de0SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_locate", "Debug output level all point location computations", "DMLocatePoints", 0, &mesh->printLocate, NULL, 0));
5072a77a5016SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_plex_print_project", "Debug output level all projection computations", "DMPlexProject", 0, &mesh->printProject, NULL, 0));
50739566063dSJacob Faibussowitsch   PetscCall(DMMonitorSetFromOptions(dm, "-dm_plex_monitor_throughput", "Monitor the simulation throughput", "DMPlexMonitorThroughput", DMPlexMonitorThroughput, NULL, &flg));
50749566063dSJacob Faibussowitsch   if (flg) PetscCall(PetscLogDefaultBegin());
50755e2c5519SMatthew G. Knepley   // Interpolation
50765e2c5519SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_plex_interpolate_prefer_tensor", "When different orderings exist, prefer the tensor order", "DMPlexSetInterpolationPreferTensor", mesh->interpolatePreferTensor, &mesh->interpolatePreferTensor, NULL));
50779318fe57SMatthew G. Knepley   /* Labeling */
50789566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_plex_boundary_label", "Label to mark the mesh boundary", "", bdLabel, bdLabel, sizeof(bdLabel), &flg));
50799566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexCreateBoundaryLabel_Private(dm, bdLabel));
5080953fc75cSMatthew G. Knepley   /* Point Location */
50819566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMInterpolate", PETSC_FALSE, &mesh->useHashLocation, NULL));
50820848f4b5SMatthew G. Knepley   /* Partitioning and distribution */
50839566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", PETSC_FALSE, &mesh->partitionBalance, NULL));
5084d02c7345SMatthew G. Knepley   /* Reordering */
5085adc21957SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_reorder_section", "Compute point permutation for local section", "DMReorderSectionSetDefault", PETSC_FALSE, &flg2, &flg));
5086adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetDefault(dm, flg2 ? DM_REORDER_DEFAULT_TRUE : DM_REORDER_DEFAULT_FALSE));
5087adc21957SMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_reorder_section_type", "Reordering method for local section", "DMReorderSectionSetType", method, method, PETSC_MAX_PATH_LEN, &flg));
5088adc21957SMatthew G. Knepley   if (flg) PetscCall(DMReorderSectionSetType(dm, method));
50892e62ab5aSMatthew G. Knepley   /* Generation and remeshing */
50909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_remesh_bd", "Allow changes to the boundary on remeshing", "DMAdapt", PETSC_FALSE, &mesh->remeshBd, NULL));
509161f058f9SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_plex_save_transform", "Save the transform producing this mesh", "DMAdapt", PETSC_FALSE, &mesh->saveTransform, NULL));
5092b29cfa1cSToby Isaac   /* Projection behavior */
5093d5b43468SJose E. Roman   PetscCall(PetscOptionsBoundedInt("-dm_plex_max_projection_height", "Maximum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL, 0));
50949566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_regular_refinement", "Use special nested projection algorithm for regular refinement", "DMPlexSetRegularRefinement", mesh->regularRefinement, &mesh->regularRefinement, NULL));
5095f12cf164SMatthew G. Knepley   /* Checking structure */
5096f12cf164SMatthew G. Knepley   {
50977f9d8d6cSVaclav Hapla     PetscBool all = PETSC_FALSE;
5098f12cf164SMatthew G. Knepley 
50997f9d8d6cSVaclav Hapla     PetscCall(PetscOptionsBool("-dm_plex_check_all", "Perform all basic checks", "DMPlexCheck", PETSC_FALSE, &all, NULL));
51007f9d8d6cSVaclav Hapla     if (all) {
51017f9d8d6cSVaclav Hapla       PetscCall(DMPlexCheck(dm));
51027f9d8d6cSVaclav Hapla     } else {
51039566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_symmetry", "Check that the adjacency information in the mesh is symmetric", "DMPlexCheckSymmetry", PETSC_FALSE, &flg, &flg2));
51047f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSymmetry(dm));
51059566063dSJacob 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));
51067f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckSkeleton(dm, 0));
51079566063dSJacob 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));
51087f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckFaces(dm, 0));
51099566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_geometry", "Check that cells have positive volume", "DMPlexCheckGeometry", PETSC_FALSE, &flg, &flg2));
51107f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckGeometry(dm));
51119566063dSJacob Faibussowitsch       PetscCall(PetscOptionsBool("-dm_plex_check_pointsf", "Check some necessary conditions for PointSF", "DMPlexCheckPointSF", PETSC_FALSE, &flg, &flg2));
5112d7d32a9aSMatthew G. Knepley       if (flg && flg2) PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE));
51139566063dSJacob 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));
51147f9d8d6cSVaclav Hapla       if (flg && flg2) PetscCall(DMPlexCheckInterfaceCones(dm));
51157f9d8d6cSVaclav Hapla     }
51169566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-dm_plex_check_cell_shape", "Check cell shape", "DMPlexCheckCellShape", PETSC_FALSE, &flg, &flg2));
51179566063dSJacob Faibussowitsch     if (flg && flg2) PetscCall(DMPlexCheckCellShape(dm, PETSC_TRUE, PETSC_DETERMINE));
5118f12cf164SMatthew G. Knepley   }
51199318fe57SMatthew G. Knepley   {
51209318fe57SMatthew G. Knepley     PetscReal scale = 1.0;
51214f3833eaSMatthew G. Knepley 
51229566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-dm_plex_scale", "Scale factor for mesh coordinates", "DMPlexScale", scale, &scale, &flg));
51239318fe57SMatthew G. Knepley     if (flg) {
51249318fe57SMatthew G. Knepley       Vec coordinates, coordinatesLocal;
51259318fe57SMatthew G. Knepley 
51269566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(dm, &coordinates));
51279566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal));
51289566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinates, scale));
51299566063dSJacob Faibussowitsch       PetscCall(VecScale(coordinatesLocal, scale));
51309318fe57SMatthew G. Knepley     }
51319318fe57SMatthew G. Knepley   }
51329566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerSetFromOptions(mesh->partitioner));
51333ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
513468d4fef7SMatthew G. Knepley }
513568d4fef7SMatthew G. Knepley 
5136ce78bad3SBarry Smith PetscErrorCode DMSetFromOptions_Overlap_Plex(DM dm, PetscOptionItems PetscOptionsObject, PetscInt *overlap)
5137d71ae5a4SJacob Faibussowitsch {
5138c506a872SMatthew G. Knepley   PetscInt  numOvLabels = 16, numOvExLabels = 16;
5139c506a872SMatthew G. Knepley   char     *ovLabelNames[16], *ovExLabelNames[16];
5140c506a872SMatthew G. Knepley   PetscInt  numOvValues = 16, numOvExValues = 16, l;
5141c506a872SMatthew G. Knepley   PetscBool flg;
5142c506a872SMatthew G. Knepley 
5143c506a872SMatthew G. Knepley   PetscFunctionBegin;
5144c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", *overlap, overlap, NULL, 0));
5145c506a872SMatthew G. Knepley   PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_labels", "List of overlap label names", "DMPlexDistribute", ovLabelNames, &numOvLabels, &flg));
5146c506a872SMatthew G. Knepley   if (!flg) numOvLabels = 0;
5147c506a872SMatthew G. Knepley   if (numOvLabels) {
5148c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvLabels = numOvLabels;
5149c506a872SMatthew G. Knepley     for (l = 0; l < numOvLabels; ++l) {
5150c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovLabelNames[l], &((DM_Plex *)dm->data)->ovLabels[l]));
5151c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovLabelNames[l]);
5152c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovLabelNames[l]));
5153c506a872SMatthew G. Knepley     }
5154c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_values", "List of overlap label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovValues, &numOvValues, &flg));
5155c506a872SMatthew G. Knepley     if (!flg) numOvValues = 0;
5156c506a872SMatthew 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);
5157c506a872SMatthew G. Knepley 
5158c506a872SMatthew G. Knepley     PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_exclude_labels", "List of overlap exclude label names", "DMPlexDistribute", ovExLabelNames, &numOvExLabels, &flg));
5159c506a872SMatthew G. Knepley     if (!flg) numOvExLabels = 0;
5160c506a872SMatthew G. Knepley     ((DM_Plex *)dm->data)->numOvExLabels = numOvExLabels;
5161c506a872SMatthew G. Knepley     for (l = 0; l < numOvExLabels; ++l) {
5162c506a872SMatthew G. Knepley       PetscCall(DMGetLabel(dm, ovExLabelNames[l], &((DM_Plex *)dm->data)->ovExLabels[l]));
5163c506a872SMatthew G. Knepley       PetscCheck(((DM_Plex *)dm->data)->ovExLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovExLabelNames[l]);
5164c506a872SMatthew G. Knepley       PetscCall(PetscFree(ovExLabelNames[l]));
5165c506a872SMatthew G. Knepley     }
5166c506a872SMatthew G. Knepley     PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_exclude_values", "List of overlap exclude label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovExValues, &numOvExValues, &flg));
5167c506a872SMatthew G. Knepley     if (!flg) numOvExValues = 0;
5168c506a872SMatthew 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);
5169c506a872SMatthew G. Knepley   }
51703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5171c506a872SMatthew G. Knepley }
5172c506a872SMatthew G. Knepley 
5173ce78bad3SBarry Smith static PetscErrorCode DMSetFromOptions_Plex(DM dm, PetscOptionItems PetscOptionsObject)
5174d71ae5a4SJacob Faibussowitsch {
5175bdf63967SMatthew G. Knepley   PetscFunctionList    ordlist;
5176bdf63967SMatthew G. Knepley   char                 oname[256];
51774e22dd4cSMatthew G. Knepley   char                 sublabelname[PETSC_MAX_PATH_LEN] = "";
5178adc21957SMatthew G. Knepley   DMReorderDefaultFlag reorder;
5179d410b0cfSMatthew G. Knepley   PetscReal            volume    = -1.0;
51809318fe57SMatthew G. Knepley   PetscInt             prerefine = 0, refine = 0, r, coarsen = 0, overlap = 0, extLayers = 0, dim;
5181530e699aSMatthew G. Knepley   PetscBool            uniformOrig = PETSC_FALSE, uniform = PETSC_TRUE, distribute, saveSF = PETSC_FALSE, interpolate = PETSC_TRUE, coordSpace = PETSC_FALSE, remap = PETSC_TRUE, ghostCells = PETSC_FALSE, isHierarchy, flg;
518268d4fef7SMatthew G. Knepley 
518368d4fef7SMatthew G. Knepley   PetscFunctionBegin;
5184d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "DMPlex Options");
5185dd4c3f67SMatthew G. Knepley   if (dm->cloneOpts) goto non_refine;
51869318fe57SMatthew G. Knepley   /* Handle automatic creation */
51879566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
5188530e699aSMatthew G. Knepley   if (dim < 0) PetscCall(DMPlexCreateFromOptions_Internal(PetscOptionsObject, &coordSpace, dm));
51896bc1bd01Sksagiyam   PetscCall(DMGetDimension(dm, &dim));
5190d89e6e46SMatthew G. Knepley   /* Handle interpolation before distribution */
51919566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_interpolate_pre", "Flag to interpolate mesh before distribution", "", interpolate, &interpolate, &flg));
5192d89e6e46SMatthew G. Knepley   if (flg) {
5193d89e6e46SMatthew G. Knepley     DMPlexInterpolatedFlag interpolated;
5194d89e6e46SMatthew G. Knepley 
51959566063dSJacob Faibussowitsch     PetscCall(DMPlexIsInterpolated(dm, &interpolated));
5196d89e6e46SMatthew G. Knepley     if (interpolated == DMPLEX_INTERPOLATED_FULL && !interpolate) {
5197d89e6e46SMatthew G. Knepley       DM udm;
5198d89e6e46SMatthew G. Knepley 
51999566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(dm, &udm));
520069d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &udm));
5201d89e6e46SMatthew G. Knepley     } else if (interpolated != DMPLEX_INTERPOLATED_FULL && interpolate) {
5202d89e6e46SMatthew G. Knepley       DM idm;
5203d89e6e46SMatthew G. Knepley 
52049566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(dm, &idm));
520569d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &idm));
5206d89e6e46SMatthew G. Knepley     }
5207d89e6e46SMatthew G. Knepley   }
52084e22dd4cSMatthew G. Knepley   // Handle submesh selection before distribution
52094e22dd4cSMatthew G. Knepley   PetscCall(PetscOptionsString("-dm_plex_submesh", "Label to use for submesh selection", "", sublabelname, sublabelname, PETSC_MAX_PATH_LEN, &flg));
52104e22dd4cSMatthew G. Knepley   if (flg) {
52114e22dd4cSMatthew G. Knepley     DM              subdm;
52124e22dd4cSMatthew G. Knepley     DMLabel         label;
52134e22dd4cSMatthew G. Knepley     IS              valueIS, pointIS;
52144e22dd4cSMatthew G. Knepley     const PetscInt *values, *points;
52154e22dd4cSMatthew G. Knepley     PetscBool       markedFaces = PETSC_FALSE;
52164e22dd4cSMatthew G. Knepley     PetscInt        Nv, value, Np;
52174e22dd4cSMatthew G. Knepley 
52184e22dd4cSMatthew G. Knepley     PetscCall(DMGetLabel(dm, sublabelname, &label));
52194e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetNumValues(label, &Nv));
52204e22dd4cSMatthew 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);
52214e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetValueIS(label, &valueIS));
52224e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(valueIS, &values));
52234e22dd4cSMatthew G. Knepley     value = values[0];
52244e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(valueIS, &values));
52254e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&valueIS));
52264e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumSize(label, value, &Np));
52274e22dd4cSMatthew G. Knepley     PetscCall(DMLabelGetStratumIS(label, value, &pointIS));
52284e22dd4cSMatthew G. Knepley     PetscCall(ISGetIndices(pointIS, &points));
52294e22dd4cSMatthew G. Knepley     for (PetscInt p = 0; p < Np; ++p) {
52304e22dd4cSMatthew G. Knepley       PetscInt pdepth;
52314e22dd4cSMatthew G. Knepley 
52324e22dd4cSMatthew G. Knepley       PetscCall(DMPlexGetPointDepth(dm, points[p], &pdepth));
52334e22dd4cSMatthew G. Knepley       if (pdepth) {
52344e22dd4cSMatthew G. Knepley         markedFaces = PETSC_TRUE;
52354e22dd4cSMatthew G. Knepley         break;
52364e22dd4cSMatthew G. Knepley       }
52374e22dd4cSMatthew G. Knepley     }
52384e22dd4cSMatthew G. Knepley     PetscCall(ISRestoreIndices(pointIS, &points));
52394e22dd4cSMatthew G. Knepley     PetscCall(ISDestroy(&pointIS));
52404e22dd4cSMatthew G. Knepley     PetscCall(DMPlexCreateSubmesh(dm, label, value, markedFaces, &subdm));
52414e22dd4cSMatthew G. Knepley     PetscCall(DMPlexReplace_Internal(dm, &subdm));
52424e22dd4cSMatthew G. Knepley     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
52434e22dd4cSMatthew G. Knepley   }
52449b44eab4SMatthew G. Knepley   /* Handle DMPlex refinement before distribution */
52459566063dSJacob Faibussowitsch   PetscCall(DMPlexGetRefinementUniform(dm, &uniformOrig));
52469566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_pre", "The number of refinements before distribution", "DMCreate", prerefine, &prerefine, NULL, 0));
52479566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap_pre", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
52489566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_uniform_pre", "Flag for uniform refinement before distribution", "DMCreate", uniform, &uniform, &flg));
52499566063dSJacob Faibussowitsch   if (flg) PetscCall(DMPlexSetRefinementUniform(dm, uniform));
52509566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-dm_refine_volume_limit_pre", "The maximum cell volume after refinement before distribution", "DMCreate", volume, &volume, &flg));
52519318fe57SMatthew G. Knepley   if (flg) {
52529566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementUniform(dm, PETSC_FALSE));
52539566063dSJacob Faibussowitsch     PetscCall(DMPlexSetRefinementLimit(dm, volume));
52549318fe57SMatthew G. Knepley     prerefine = PetscMax(prerefine, 1);
52559318fe57SMatthew G. Knepley   }
5256b23db253SStefano Zampini   if (prerefine) PetscCall(DMLocalizeCoordinates(dm));
52579b44eab4SMatthew G. Knepley   for (r = 0; r < prerefine; ++r) {
52589b44eab4SMatthew G. Knepley     DM             rdm;
5259509b31aaSMatthew G. Knepley     PetscPointFunc coordFunc;
52609b44eab4SMatthew G. Knepley 
5261509b31aaSMatthew G. Knepley     PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc));
5262dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
52639566063dSJacob Faibussowitsch     PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
526469d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &rdm));
5265dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
526661a622f3SMatthew G. Knepley     if (coordFunc && remap) {
52679566063dSJacob Faibussowitsch       PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
5268509b31aaSMatthew G. Knepley       PetscCall(DMPlexSetCoordinateMap(dm, coordFunc));
52699b44eab4SMatthew G. Knepley     }
52709b44eab4SMatthew G. Knepley   }
52719566063dSJacob Faibussowitsch   PetscCall(DMPlexSetRefinementUniform(dm, uniformOrig));
52729318fe57SMatthew G. Knepley   /* Handle DMPlex extrusion before distribution */
52739566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_extrude", "The number of layers to extrude", "", extLayers, &extLayers, NULL, 0));
52749318fe57SMatthew G. Knepley   if (extLayers) {
52759318fe57SMatthew G. Knepley     DM edm;
52769318fe57SMatthew G. Knepley 
52779566063dSJacob Faibussowitsch     PetscCall(DMExtrude(dm, extLayers, &edm));
527869d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &edm));
5279509b31aaSMatthew G. Knepley     PetscCall(DMPlexSetCoordinateMap(dm, NULL));
5280dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
5281d410b0cfSMatthew G. Knepley     extLayers = 0;
52825e17fc22SAidan Hamilton     PetscCall(DMGetDimension(dm, &dim));
52839318fe57SMatthew G. Knepley   }
5284bdf63967SMatthew G. Knepley   /* Handle DMPlex reordering before distribution */
52856bc1bd01Sksagiyam   PetscCall(DMPlexReorderGetDefault(dm, &reorder));
52869566063dSJacob Faibussowitsch   PetscCall(MatGetOrderingList(&ordlist));
52876bc1bd01Sksagiyam   PetscCall(PetscStrncpy(oname, MATORDERINGNATURAL, sizeof(oname)));
52889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-dm_plex_reorder", "Set mesh reordering type", "DMPlexGetOrdering", ordlist, MATORDERINGNATURAL, oname, sizeof(oname), &flg));
5289adc21957SMatthew G. Knepley   if (reorder == DM_REORDER_DEFAULT_TRUE || flg) {
5290bdf63967SMatthew G. Knepley     DM pdm;
5291bdf63967SMatthew G. Knepley     IS perm;
5292bdf63967SMatthew G. Knepley 
52939566063dSJacob Faibussowitsch     PetscCall(DMPlexGetOrdering(dm, oname, NULL, &perm));
52949566063dSJacob Faibussowitsch     PetscCall(DMPlexPermute(dm, perm, &pdm));
52959566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&perm));
529669d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &pdm));
5297dbbe0bcdSBarry Smith     PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
5298bdf63967SMatthew G. Knepley   }
52999b44eab4SMatthew G. Knepley   /* Handle DMPlex distribution */
53009566063dSJacob Faibussowitsch   PetscCall(DMPlexDistributeGetDefault(dm, &distribute));
5301c506a872SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute", "Flag to redistribute a mesh among processes", "DMPlexDistribute", distribute, &distribute, NULL));
5302a286e215SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_distribute_save_sf", "Flag to save the migration SF", "DMPlexSetMigrationSF", saveSF, &saveSF, NULL));
5303dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_Overlap_Plex(dm, PetscOptionsObject, &overlap));
53049b44eab4SMatthew G. Knepley   if (distribute) {
53059b44eab4SMatthew G. Knepley     DM               pdm = NULL;
53069b44eab4SMatthew G. Knepley     PetscPartitioner part;
5307a286e215SMatthew G. Knepley     PetscSF          sfMigration;
53089566ead2SJames Wright     PetscBool        use_partition_balance;
53099b44eab4SMatthew G. Knepley 
53109566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPartitioner(dm, &part));
53119566063dSJacob Faibussowitsch     PetscCall(PetscPartitionerSetFromOptions(part));
53129566ead2SJames Wright     PetscCall(DMPlexGetPartitionBalance(dm, &use_partition_balance));
53139566ead2SJames Wright     PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", use_partition_balance, &use_partition_balance, &flg));
53149566ead2SJames Wright     if (flg) PetscCall(DMPlexSetPartitionBalance(dm, use_partition_balance));
5315a286e215SMatthew G. Knepley     PetscCall(DMPlexDistribute(dm, overlap, &sfMigration, &pdm));
53165d2873a6SJames Wright     if (pdm) {
53175d2873a6SJames Wright       // Delete the local section to force the existing one to be rebuilt with the distributed DM
53185d2873a6SJames Wright       PetscCall(DMSetLocalSection(dm, pdm->localSection));
53195d2873a6SJames Wright       PetscCall(DMPlexReplace_Internal(dm, &pdm));
53205d2873a6SJames Wright     }
5321a286e215SMatthew G. Knepley     if (saveSF) PetscCall(DMPlexSetMigrationSF(dm, sfMigration));
5322a286e215SMatthew G. Knepley     PetscCall(PetscSFDestroy(&sfMigration));
53239b44eab4SMatthew G. Knepley   }
53244054ae39SJames Wright 
53254054ae39SJames Wright   {
53264054ae39SJames Wright     PetscBool useBoxLabel = PETSC_FALSE;
53274054ae39SJames Wright     PetscCall(PetscOptionsBool("-dm_plex_box_label", "Create 'Face Sets' assuming boundary faces align with cartesian directions", "DMCreate", useBoxLabel, &useBoxLabel, NULL));
5328d7d2d1d2SJames Wright     if (useBoxLabel) {
5329d7d2d1d2SJames Wright       PetscInt       n      = 3;
5330d7d2d1d2SJames Wright       DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE};
5331d7d2d1d2SJames Wright 
5332d7d2d1d2SJames Wright       PetscCall(PetscOptionsEnumArray("-dm_plex_box_label_bd", "Boundary type for each dimension when using -dm_plex_box_label", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg));
5333d7d2d1d2SJames Wright       PetscCheck(!flg || !(n != dim), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim);
5334d7d2d1d2SJames Wright       PetscCall(DMPlexSetBoxLabel_Internal(dm, bdt));
5335d7d2d1d2SJames Wright     }
53364054ae39SJames Wright   }
5337d2b2dc1eSMatthew G. Knepley   /* Must check CEED options before creating function space for coordinates */
5338d2b2dc1eSMatthew G. Knepley   {
5339d2b2dc1eSMatthew G. Knepley     PetscBool useCeed = PETSC_FALSE, flg;
5340d2b2dc1eSMatthew G. Knepley 
5341d2b2dc1eSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_plex_use_ceed", "Use LibCEED as the FEM backend", "DMPlexSetUseCeed", useCeed, &useCeed, &flg));
5342d2b2dc1eSMatthew G. Knepley     if (flg) PetscCall(DMPlexSetUseCeed(dm, useCeed));
5343d2b2dc1eSMatthew G. Knepley   }
53449318fe57SMatthew G. Knepley   /* Create coordinate space */
5345530e699aSMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_coord_space", "Use an FEM space for coordinates", "", coordSpace, &coordSpace, NULL));
5346530e699aSMatthew G. Knepley   if (coordSpace) {
5347e44f6aebSMatthew G. Knepley     PetscInt  degree = 1, deg;
53485515ebd3SMatthew G. Knepley     PetscInt  height = 0;
53495515ebd3SMatthew G. Knepley     DM        cdm;
5350530e699aSMatthew G. Knepley     PetscBool localize = PETSC_TRUE, sparseLocalize = PETSC_TRUE;
53519318fe57SMatthew G. Knepley 
53529566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-dm_coord_petscspace_degree", "FEM degree for coordinate space", "", degree, &degree, NULL));
5353e44f6aebSMatthew G. Knepley     PetscCall(DMGetCoordinateDegree_Internal(dm, &deg));
5354509b31aaSMatthew G. Knepley     if (coordSpace && deg <= 1) PetscCall(DMPlexCreateCoordinateSpace(dm, degree, PETSC_TRUE, NULL));
53555515ebd3SMatthew G. Knepley     PetscCall(DMGetCoordinateDM(dm, &cdm));
5356530e699aSMatthew G. Knepley     if (!coordSpace) {
535761a622f3SMatthew G. Knepley       PetscDS      cds;
535861a622f3SMatthew G. Knepley       PetscObject  obj;
535961a622f3SMatthew G. Knepley       PetscClassId id;
536061a622f3SMatthew G. Knepley 
53619566063dSJacob Faibussowitsch       PetscCall(DMGetDS(cdm, &cds));
53629566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
53639566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
536461a622f3SMatthew G. Knepley       if (id == PETSCFE_CLASSID) {
536561a622f3SMatthew G. Knepley         PetscContainer dummy;
536661a622f3SMatthew G. Knepley 
53679566063dSJacob Faibussowitsch         PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &dummy));
53689566063dSJacob Faibussowitsch         PetscCall(PetscObjectSetName((PetscObject)dummy, "coordinates"));
53699566063dSJacob Faibussowitsch         PetscCall(DMSetField(cdm, 0, NULL, (PetscObject)dummy));
53709566063dSJacob Faibussowitsch         PetscCall(PetscContainerDestroy(&dummy));
53719566063dSJacob Faibussowitsch         PetscCall(DMClearDS(cdm));
537261a622f3SMatthew G. Knepley       }
5373509b31aaSMatthew G. Knepley       PetscCall(DMPlexSetCoordinateMap(dm, NULL));
537461a622f3SMatthew G. Knepley     }
5375c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_localize", "Localize mesh coordinates", "", localize, &localize, NULL));
5376c3db174cSMatthew G. Knepley     PetscCall(PetscOptionsBool("-dm_sparse_localize", "Localize only necessary cells", "DMSetSparseLocalize", sparseLocalize, &sparseLocalize, &flg));
5377c3db174cSMatthew G. Knepley     if (flg) PetscCall(DMSetSparseLocalize(dm, sparseLocalize));
53785515ebd3SMatthew G. Knepley     PetscCall(PetscOptionsInt("-dm_localize_height", "Localize edges and faces in addition to cells", "", height, &height, &flg));
53795515ebd3SMatthew G. Knepley     if (flg) PetscCall(DMPlexSetMaxProjectionHeight(cdm, height));
5380c3db174cSMatthew G. Knepley     if (localize) PetscCall(DMLocalizeCoordinates(dm));
53819318fe57SMatthew G. Knepley   }
538268d4fef7SMatthew G. Knepley   /* Handle DMPlex refinement */
538361a622f3SMatthew G. Knepley   remap = PETSC_TRUE;
53849566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL, 0));
53859566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_refine_remap", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL));
53869566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy, 0));
53879566063dSJacob Faibussowitsch   if (refine) PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE));
538868d4fef7SMatthew G. Knepley   if (refine && isHierarchy) {
5389acdc6f61SToby Isaac     DM *dms, coarseDM;
539068d4fef7SMatthew G. Knepley 
53919566063dSJacob Faibussowitsch     PetscCall(DMGetCoarseDM(dm, &coarseDM));
53929566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)coarseDM));
53939566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(refine, &dms));
53949566063dSJacob Faibussowitsch     PetscCall(DMRefineHierarchy(dm, refine, dms));
539568d4fef7SMatthew G. Knepley     /* Total hack since we do not pass in a pointer */
53969566063dSJacob Faibussowitsch     PetscCall(DMPlexSwap_Static(dm, dms[refine - 1]));
539768d4fef7SMatthew G. Knepley     if (refine == 1) {
53989566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[0]));
53999566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
540068d4fef7SMatthew G. Knepley     } else {
54019566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, dms[refine - 2]));
54029566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE));
54039566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dms[0], dms[refine - 1]));
54049566063dSJacob Faibussowitsch       PetscCall(DMPlexSetRegularRefinement(dms[0], PETSC_TRUE));
540568d4fef7SMatthew G. Knepley     }
54069566063dSJacob Faibussowitsch     PetscCall(DMSetCoarseDM(dms[refine - 1], coarseDM));
54079566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)coarseDM));
540868d4fef7SMatthew G. Knepley     /* Free DMs */
540968d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
5410dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
54119566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
541268d4fef7SMatthew G. Knepley     }
54139566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
541468d4fef7SMatthew G. Knepley   } else {
541568d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
54169318fe57SMatthew G. Knepley       DM             rdm;
5417509b31aaSMatthew G. Knepley       PetscPointFunc coordFunc;
541868d4fef7SMatthew G. Knepley 
5419509b31aaSMatthew G. Knepley       PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc));
5420dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
54219566063dSJacob Faibussowitsch       PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm));
542268d4fef7SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
542369d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
5424dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
542561a622f3SMatthew G. Knepley       if (coordFunc && remap) {
54269566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
5427509b31aaSMatthew G. Knepley         PetscCall(DMPlexSetCoordinateMap(dm, coordFunc));
542851a74b61SMatthew G. Knepley       }
542968d4fef7SMatthew G. Knepley     }
543068d4fef7SMatthew G. Knepley   }
54313cf6fe12SMatthew G. Knepley   /* Handle DMPlex coarsening */
54329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL, 0));
54339566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBoundedInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy, 0));
5434b653a561SMatthew G. Knepley   if (coarsen && isHierarchy) {
5435b653a561SMatthew G. Knepley     DM *dms;
5436b653a561SMatthew G. Knepley 
54379566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(coarsen, &dms));
54389566063dSJacob Faibussowitsch     PetscCall(DMCoarsenHierarchy(dm, coarsen, dms));
5439b653a561SMatthew G. Knepley     /* Free DMs */
5440b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
5441dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject));
54429566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dms[r]));
5443b653a561SMatthew G. Knepley     }
54449566063dSJacob Faibussowitsch     PetscCall(PetscFree(dms));
5445b653a561SMatthew G. Knepley   } else {
5446b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
54479318fe57SMatthew G. Knepley       DM             cdm;
5448509b31aaSMatthew G. Knepley       PetscPointFunc coordFunc;
54493cf6fe12SMatthew G. Knepley 
5450509b31aaSMatthew G. Knepley       PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc));
5451dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
54529566063dSJacob Faibussowitsch       PetscCall(DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &cdm));
54533cf6fe12SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
545469d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &cdm));
5455dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
54569318fe57SMatthew G. Knepley       if (coordFunc) {
54579566063dSJacob Faibussowitsch         PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc));
5458509b31aaSMatthew G. Knepley         PetscCall(DMPlexSetCoordinateMap(dm, coordFunc));
54599318fe57SMatthew G. Knepley       }
54603cf6fe12SMatthew G. Knepley     }
5461b653a561SMatthew G. Knepley   }
5462be664eb1SMatthew G. Knepley   // Handle coordinate remapping
5463be664eb1SMatthew G. Knepley   remap = PETSC_FALSE;
5464be664eb1SMatthew G. Knepley   PetscCall(PetscOptionsBool("-dm_coord_remap", "Flag to control coordinate remapping", "", remap, &remap, NULL));
5465be664eb1SMatthew G. Knepley   if (remap) {
5466be664eb1SMatthew G. Knepley     DMPlexCoordMap map     = DM_COORD_MAP_NONE;
5467be664eb1SMatthew G. Knepley     PetscPointFunc mapFunc = NULL;
5468be664eb1SMatthew G. Knepley     PetscScalar    params[16];
5469f45b553cSPierre Jolivet     PetscInt       Np = PETSC_STATIC_ARRAY_LENGTH(params), cdim;
5470be664eb1SMatthew G. Knepley     MPI_Comm       comm;
5471be664eb1SMatthew G. Knepley 
5472be664eb1SMatthew G. Knepley     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
5473be664eb1SMatthew G. Knepley     PetscCall(DMGetCoordinateDim(dm, &cdim));
5474be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsScalarArray("-dm_coord_map_params", "Parameters for the coordinate remapping", "", params, &Np, &flg));
5475be664eb1SMatthew G. Knepley     if (!flg) Np = 0;
5476be664eb1SMatthew G. Knepley     // TODO Allow user to pass a map function by name
5477be664eb1SMatthew G. Knepley     PetscCall(PetscOptionsEnum("-dm_coord_map", "Coordinate mapping for built-in mesh", "", DMPlexCoordMaps, (PetscEnum)map, (PetscEnum *)&map, &flg));
5478be664eb1SMatthew G. Knepley     if (flg) {
5479be664eb1SMatthew G. Knepley       switch (map) {
5480be664eb1SMatthew G. Knepley       case DM_COORD_MAP_NONE:
5481be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
5482be664eb1SMatthew G. Knepley         break;
5483be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHEAR:
5484be664eb1SMatthew G. Knepley         mapFunc = coordMap_shear;
5485be664eb1SMatthew G. Knepley         if (!Np) {
5486be664eb1SMatthew G. Knepley           Np        = cdim + 1;
5487be664eb1SMatthew G. Knepley           params[0] = 0;
5488be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
5489be664eb1SMatthew G. Knepley         }
5490be664eb1SMatthew 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);
5491be664eb1SMatthew G. Knepley         break;
5492be664eb1SMatthew G. Knepley       case DM_COORD_MAP_FLARE:
5493be664eb1SMatthew G. Knepley         mapFunc = coordMap_flare;
5494be664eb1SMatthew G. Knepley         if (!Np) {
5495be664eb1SMatthew G. Knepley           Np        = cdim + 1;
5496be664eb1SMatthew G. Knepley           params[0] = 0;
5497be664eb1SMatthew G. Knepley           for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0;
5498be664eb1SMatthew G. Knepley         }
5499be664eb1SMatthew 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);
5500be664eb1SMatthew G. Knepley         break;
5501be664eb1SMatthew G. Knepley       case DM_COORD_MAP_ANNULUS:
5502be664eb1SMatthew G. Knepley         mapFunc = coordMap_annulus;
5503be664eb1SMatthew G. Knepley         if (!Np) {
5504be664eb1SMatthew G. Knepley           Np        = 2;
5505be664eb1SMatthew G. Knepley           params[0] = 1.;
5506be664eb1SMatthew G. Knepley           params[1] = 2.;
5507be664eb1SMatthew G. Knepley         }
5508be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The annulus coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
5509be664eb1SMatthew G. Knepley         break;
5510be664eb1SMatthew G. Knepley       case DM_COORD_MAP_SHELL:
5511be664eb1SMatthew G. Knepley         mapFunc = coordMap_shell;
5512be664eb1SMatthew G. Knepley         if (!Np) {
5513be664eb1SMatthew G. Knepley           Np        = 2;
5514be664eb1SMatthew G. Knepley           params[0] = 1.;
5515be664eb1SMatthew G. Knepley           params[1] = 2.;
5516be664eb1SMatthew G. Knepley         }
5517be664eb1SMatthew G. Knepley         PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The spherical shell coordinate map must have 2 parameters, not %" PetscInt_FMT, Np);
5518be664eb1SMatthew G. Knepley         break;
5519530e699aSMatthew G. Knepley       case DM_COORD_MAP_SINUSOID:
5520530e699aSMatthew G. Knepley         mapFunc = coordMap_sinusoid;
5521530e699aSMatthew G. Knepley         if (!Np) {
5522530e699aSMatthew G. Knepley           Np        = 3;
5523530e699aSMatthew G. Knepley           params[0] = 1.;
5524530e699aSMatthew G. Knepley           params[1] = 1.;
5525530e699aSMatthew G. Knepley           params[2] = 1.;
5526530e699aSMatthew G. Knepley         }
5527530e699aSMatthew G. Knepley         PetscCheck(Np == 3, comm, PETSC_ERR_ARG_WRONG, "The sinusoidal coordinate map must have 3 parameters, not %" PetscInt_FMT, Np);
5528530e699aSMatthew G. Knepley         break;
5529be664eb1SMatthew G. Knepley       default:
5530be664eb1SMatthew G. Knepley         mapFunc = coordMap_identity;
5531be664eb1SMatthew G. Knepley       }
5532be664eb1SMatthew G. Knepley     }
5533be664eb1SMatthew G. Knepley     if (Np) {
5534be664eb1SMatthew G. Knepley       DM      cdm;
5535be664eb1SMatthew G. Knepley       PetscDS cds;
5536be664eb1SMatthew G. Knepley 
5537be664eb1SMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
5538be664eb1SMatthew G. Knepley       PetscCall(DMGetDS(cdm, &cds));
5539be664eb1SMatthew G. Knepley       PetscCall(PetscDSSetConstants(cds, Np, params));
5540be664eb1SMatthew G. Knepley     }
5541be664eb1SMatthew G. Knepley     PetscCall(DMPlexRemapGeometry(dm, 0.0, mapFunc));
5542be664eb1SMatthew G. Knepley   }
5543909dfd52SMatthew G. Knepley   /* Handle ghost cells */
55449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_plex_create_fv_ghost_cells", "Flag to create finite volume ghost cells on the boundary", "DMCreate", ghostCells, &ghostCells, NULL));
5545909dfd52SMatthew G. Knepley   if (ghostCells) {
5546909dfd52SMatthew G. Knepley     DM   gdm;
5547909dfd52SMatthew G. Knepley     char lname[PETSC_MAX_PATH_LEN];
5548909dfd52SMatthew G. Knepley 
5549909dfd52SMatthew G. Knepley     lname[0] = '\0';
55509566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-dm_plex_fv_ghost_cells_label", "Label name for ghost cells boundary", "DMCreate", lname, lname, sizeof(lname), &flg));
55519566063dSJacob Faibussowitsch     PetscCall(DMPlexConstructGhostCells(dm, flg ? lname : NULL, NULL, &gdm));
555269d8a87bSksagiyam     PetscCall(DMPlexReplace_Internal(dm, &gdm));
5553909dfd52SMatthew G. Knepley   }
55546913077dSMatthew G. Knepley   /* Handle 1D order */
5555adc21957SMatthew G. Knepley   if (reorder != DM_REORDER_DEFAULT_FALSE && dim == 1) {
55566913077dSMatthew G. Knepley     DM           cdm, rdm;
55576913077dSMatthew G. Knepley     PetscDS      cds;
55586913077dSMatthew G. Knepley     PetscObject  obj;
55596913077dSMatthew G. Knepley     PetscClassId id = PETSC_OBJECT_CLASSID;
55606913077dSMatthew G. Knepley     IS           perm;
55616bc1bd01Sksagiyam     PetscInt     Nf;
55626913077dSMatthew G. Knepley     PetscBool    distributed;
55636913077dSMatthew G. Knepley 
55649566063dSJacob Faibussowitsch     PetscCall(DMPlexIsDistributed(dm, &distributed));
55659566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm, &cdm));
55669566063dSJacob Faibussowitsch     PetscCall(DMGetDS(cdm, &cds));
55679566063dSJacob Faibussowitsch     PetscCall(PetscDSGetNumFields(cds, &Nf));
55686913077dSMatthew G. Knepley     if (Nf) {
55699566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(cds, 0, &obj));
55709566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetClassId(obj, &id));
55716913077dSMatthew G. Knepley     }
55726bc1bd01Sksagiyam     if (!distributed && id != PETSCFE_CLASSID) {
55739566063dSJacob Faibussowitsch       PetscCall(DMPlexGetOrdering1D(dm, &perm));
55749566063dSJacob Faibussowitsch       PetscCall(DMPlexPermute(dm, perm, &rdm));
557569d8a87bSksagiyam       PetscCall(DMPlexReplace_Internal(dm, &rdm));
55769566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&perm));
55776913077dSMatthew G. Knepley     }
55786913077dSMatthew G. Knepley   }
55793cf6fe12SMatthew G. Knepley /* Handle */
5580dd4c3f67SMatthew G. Knepley non_refine:
5581dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject));
558222d6dc08SStefano Zampini   char    *phases[16];
558322d6dc08SStefano Zampini   PetscInt Nphases = 16;
558422d6dc08SStefano Zampini   PetscCall(PetscOptionsStringArray("-dm_plex_option_phases", "Option phase prefixes", "DMSetFromOptions", phases, &Nphases, &flg));
5585d0609cedSBarry Smith   PetscOptionsHeadEnd();
558622d6dc08SStefano Zampini 
558722d6dc08SStefano Zampini   // Phases
558822d6dc08SStefano Zampini   if (flg) {
5589530e699aSMatthew G. Knepley     DM          cdm;
5590530e699aSMatthew G. Knepley     char       *oldPrefix, *oldCoordPrefix;
5591530e699aSMatthew G. Knepley     const char *tmp;
559222d6dc08SStefano Zampini 
5593530e699aSMatthew G. Knepley     PetscCall(DMGetCoordinateDM(dm, &cdm));
5594530e699aSMatthew G. Knepley     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &tmp));
5595530e699aSMatthew G. Knepley     PetscCall(PetscStrallocpy(tmp, &oldPrefix));
5596530e699aSMatthew G. Knepley     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)cdm, &tmp));
5597530e699aSMatthew G. Knepley     PetscCall(PetscStrallocpy(tmp, &oldCoordPrefix));
559822d6dc08SStefano Zampini     for (PetscInt ph = 0; ph < Nphases; ++ph) {
559922d6dc08SStefano Zampini       PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)dm, phases[ph]));
5600530e699aSMatthew G. Knepley       PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)cdm, phases[ph]));
560122d6dc08SStefano Zampini       PetscCall(PetscInfo(dm, "Options phase %s for DM %s\n", phases[ph], dm->hdr.name));
560222d6dc08SStefano Zampini       PetscCall(DMSetFromOptions(dm));
560322d6dc08SStefano Zampini       PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldPrefix));
5604530e699aSMatthew G. Knepley       PetscCall(DMGetCoordinateDM(dm, &cdm));
5605530e699aSMatthew G. Knepley       PetscCall(PetscObjectSetOptionsPrefix((PetscObject)cdm, oldCoordPrefix));
560622d6dc08SStefano Zampini       PetscCall(PetscFree(phases[ph]));
560722d6dc08SStefano Zampini     }
5608530e699aSMatthew G. Knepley     PetscCall(PetscFree(oldPrefix));
5609530e699aSMatthew G. Knepley     PetscCall(PetscFree(oldCoordPrefix));
561022d6dc08SStefano Zampini   }
56113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
56120a6ba040SMatthew G. Knepley }
56130a6ba040SMatthew G. Knepley 
5614d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_Plex(DM dm, Vec *vec)
5615d71ae5a4SJacob Faibussowitsch {
5616552f7358SJed Brown   PetscFunctionBegin;
56179566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector_Section_Private(dm, vec));
56189566063dSJacob Faibussowitsch   /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */
56199566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex));
56209566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_Plex_Native));
56219566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex));
56229566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_Plex_Native));
56233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5624552f7358SJed Brown }
5625552f7358SJed Brown 
5626d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_Plex(DM dm, Vec *vec)
5627d71ae5a4SJacob Faibussowitsch {
5628552f7358SJed Brown   PetscFunctionBegin;
56299566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector_Section_Private(dm, vec));
56309566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex_Local));
56319566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex_Local));
56323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5633552f7358SJed Brown }
5634552f7358SJed Brown 
5635d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
5636d71ae5a4SJacob Faibussowitsch {
5637793f3fe5SMatthew G. Knepley   PetscInt depth, d;
5638793f3fe5SMatthew G. Knepley 
5639793f3fe5SMatthew G. Knepley   PetscFunctionBegin;
56409566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm, &depth));
5641793f3fe5SMatthew G. Knepley   if (depth == 1) {
56429566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &d));
56439566063dSJacob Faibussowitsch     if (dim == 0) PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
56449566063dSJacob Faibussowitsch     else if (dim == d) PetscCall(DMPlexGetDepthStratum(dm, 1, pStart, pEnd));
56459371c9d4SSatish Balay     else {
56469371c9d4SSatish Balay       *pStart = 0;
56479371c9d4SSatish Balay       *pEnd   = 0;
56489371c9d4SSatish Balay     }
5649793f3fe5SMatthew G. Knepley   } else {
56509566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd));
5651793f3fe5SMatthew G. Knepley   }
56523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5653793f3fe5SMatthew G. Knepley }
5654793f3fe5SMatthew G. Knepley 
5655d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetNeighbors_Plex(DM dm, PetscInt *nranks, const PetscMPIInt *ranks[])
5656d71ae5a4SJacob Faibussowitsch {
5657502a2867SDave May   PetscSF            sf;
56586497c311SBarry Smith   PetscMPIInt        niranks, njranks;
56596497c311SBarry Smith   PetscInt           n;
56600a19bb7dSprj-   const PetscMPIInt *iranks, *jranks;
56610a19bb7dSprj-   DM_Plex           *data = (DM_Plex *)dm->data;
5662502a2867SDave May 
56632f356facSMatthew G. Knepley   PetscFunctionBegin;
56649566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dm, &sf));
56650a19bb7dSprj-   if (!data->neighbors) {
56669566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
56679566063dSJacob Faibussowitsch     PetscCall(PetscSFGetRootRanks(sf, &njranks, &jranks, NULL, NULL, NULL));
56689566063dSJacob Faibussowitsch     PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL));
56699566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors));
56709566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + 1, jranks, njranks));
56719566063dSJacob Faibussowitsch     PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks));
56720a19bb7dSprj-     n = njranks + niranks;
56739566063dSJacob Faibussowitsch     PetscCall(PetscSortRemoveDupsMPIInt(&n, data->neighbors + 1));
56740a19bb7dSprj-     /* The following cast should never fail: can't have more neighbors than PETSC_MPI_INT_MAX */
56759566063dSJacob Faibussowitsch     PetscCall(PetscMPIIntCast(n, data->neighbors));
56760a19bb7dSprj-   }
56770a19bb7dSprj-   if (nranks) *nranks = data->neighbors[0];
56780a19bb7dSprj-   if (ranks) {
56790a19bb7dSprj-     if (data->neighbors[0]) *ranks = data->neighbors + 1;
56800a19bb7dSprj-     else *ranks = NULL;
56810a19bb7dSprj-   }
56823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5683502a2867SDave May }
5684502a2867SDave May 
56851eb70e55SToby Isaac PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM, DM, Mat, Vec, Vec);
56861eb70e55SToby Isaac 
5687d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_Plex(DM dm)
5688d71ae5a4SJacob Faibussowitsch {
5689552f7358SJed Brown   PetscFunctionBegin;
5690552f7358SJed Brown   dm->ops->view                      = DMView_Plex;
56912c40f234SMatthew G. Knepley   dm->ops->load                      = DMLoad_Plex;
5692552f7358SJed Brown   dm->ops->setfromoptions            = DMSetFromOptions_Plex;
569338221697SMatthew G. Knepley   dm->ops->clone                     = DMClone_Plex;
5694552f7358SJed Brown   dm->ops->setup                     = DMSetUp_Plex;
56951bb6d2a8SBarry Smith   dm->ops->createlocalsection        = DMCreateLocalSection_Plex;
5696adc21957SMatthew G. Knepley   dm->ops->createsectionpermutation  = DMCreateSectionPermutation_Plex;
569766ad2231SToby Isaac   dm->ops->createdefaultconstraints  = DMCreateDefaultConstraints_Plex;
5698552f7358SJed Brown   dm->ops->createglobalvector        = DMCreateGlobalVector_Plex;
5699552f7358SJed Brown   dm->ops->createlocalvector         = DMCreateLocalVector_Plex;
5700184d77edSJed Brown   dm->ops->getlocaltoglobalmapping   = NULL;
57010298fd71SBarry Smith   dm->ops->createfieldis             = NULL;
5702552f7358SJed Brown   dm->ops->createcoordinatedm        = DMCreateCoordinateDM_Plex;
5703f19dbd58SToby Isaac   dm->ops->createcoordinatefield     = DMCreateCoordinateField_Plex;
57040a6ba040SMatthew G. Knepley   dm->ops->getcoloring               = NULL;
5705552f7358SJed Brown   dm->ops->creatematrix              = DMCreateMatrix_Plex;
5706bceba477SMatthew G. Knepley   dm->ops->createinterpolation       = DMCreateInterpolation_Plex;
5707bd041c0cSMatthew G. Knepley   dm->ops->createmassmatrix          = DMCreateMassMatrix_Plex;
5708b4937a87SMatthew G. Knepley   dm->ops->createmassmatrixlumped    = DMCreateMassMatrixLumped_Plex;
57095a84ad33SLisandro Dalcin   dm->ops->createinjection           = DMCreateInjection_Plex;
5710552f7358SJed Brown   dm->ops->refine                    = DMRefine_Plex;
57110a6ba040SMatthew G. Knepley   dm->ops->coarsen                   = DMCoarsen_Plex;
57120a6ba040SMatthew G. Knepley   dm->ops->refinehierarchy           = DMRefineHierarchy_Plex;
5713b653a561SMatthew G. Knepley   dm->ops->coarsenhierarchy          = DMCoarsenHierarchy_Plex;
5714d410b0cfSMatthew G. Knepley   dm->ops->extrude                   = DMExtrude_Plex;
57150298fd71SBarry Smith   dm->ops->globaltolocalbegin        = NULL;
57160298fd71SBarry Smith   dm->ops->globaltolocalend          = NULL;
57170298fd71SBarry Smith   dm->ops->localtoglobalbegin        = NULL;
57180298fd71SBarry Smith   dm->ops->localtoglobalend          = NULL;
5719552f7358SJed Brown   dm->ops->destroy                   = DMDestroy_Plex;
5720552f7358SJed Brown   dm->ops->createsubdm               = DMCreateSubDM_Plex;
57212adcc780SMatthew G. Knepley   dm->ops->createsuperdm             = DMCreateSuperDM_Plex;
5722793f3fe5SMatthew G. Knepley   dm->ops->getdimpoints              = DMGetDimPoints_Plex;
5723552f7358SJed Brown   dm->ops->locatepoints              = DMLocatePoints_Plex;
57240709b2feSToby Isaac   dm->ops->projectfunctionlocal      = DMProjectFunctionLocal_Plex;
57250709b2feSToby Isaac   dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_Plex;
5726bfc4295aSToby Isaac   dm->ops->projectfieldlocal         = DMProjectFieldLocal_Plex;
57278c6c5593SMatthew G. Knepley   dm->ops->projectfieldlabellocal    = DMProjectFieldLabelLocal_Plex;
5728ece3a9fcSMatthew G. Knepley   dm->ops->projectbdfieldlabellocal  = DMProjectBdFieldLabelLocal_Plex;
57290709b2feSToby Isaac   dm->ops->computel2diff             = DMComputeL2Diff_Plex;
5730b698f381SToby Isaac   dm->ops->computel2gradientdiff     = DMComputeL2GradientDiff_Plex;
57312a16baeaSToby Isaac   dm->ops->computel2fielddiff        = DMComputeL2FieldDiff_Plex;
573228d58a37SPierre Jolivet   dm->ops->getneighbors              = DMGetNeighbors_Plex;
57336c6a6b79SMatthew G. Knepley   dm->ops->getlocalboundingbox       = DMGetLocalBoundingBox_Coordinates;
5734907a3e9cSStefano Zampini   dm->ops->createdomaindecomposition = DMCreateDomainDecomposition_Plex;
5735907a3e9cSStefano Zampini   dm->ops->createddscatters          = DMCreateDomainDecompositionScatters_Plex;
57369566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_Plex));
57376c51210dSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", DMPlexInsertTimeDerivativeBoundaryValues_Plex));
5738*01468941SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBounds_C", DMPlexInsertBounds_Plex));
57399566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", DMSetUpGLVisViewer_Plex));
57409566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_Plex));
57419566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", DMPlexDistributeGetDefault_Plex));
57429566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", DMPlexDistributeSetDefault_Plex));
57436bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", DMPlexReorderGetDefault_Plex));
57446bc1bd01Sksagiyam   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", DMPlexReorderSetDefault_Plex));
5745adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetDefault_C", DMReorderSectionGetDefault_Plex));
5746adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetDefault_C", DMReorderSectionSetDefault_Plex));
5747adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetType_C", DMReorderSectionGetType_Plex));
5748adc21957SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetType_C", DMReorderSectionSetType_Plex));
57499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", DMInterpolateSolution_Plex));
5750c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMPlexGetOverlap_Plex));
5751c506a872SMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", DMPlexSetOverlap_Plex));
5752d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", DMPlexGetUseCeed_Plex));
5753d2b2dc1eSMatthew G. Knepley   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", DMPlexSetUseCeed_Plex));
57543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5755552f7358SJed Brown }
5756552f7358SJed Brown 
5757d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode DMClone_Plex(DM dm, DM *newdm)
5758d71ae5a4SJacob Faibussowitsch {
575963a16f15SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex *)dm->data;
57601fca310dSJames Wright   const PetscSF *face_sfs;
57611fca310dSJames Wright   PetscInt       num_face_sfs;
576263a16f15SMatthew G. Knepley 
576363a16f15SMatthew G. Knepley   PetscFunctionBegin;
576463a16f15SMatthew G. Knepley   mesh->refct++;
576563a16f15SMatthew G. Knepley   (*newdm)->data = mesh;
57661fca310dSJames Wright   PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &face_sfs));
57671fca310dSJames Wright   PetscCall(DMPlexSetIsoperiodicFaceSF(*newdm, num_face_sfs, (PetscSF *)face_sfs));
57689566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)*newdm, DMPLEX));
57699566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(*newdm));
57703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
577163a16f15SMatthew G. Knepley }
577263a16f15SMatthew G. Knepley 
57738818961aSMatthew G Knepley /*MC
57740b4b7b1cSBarry Smith   DMPLEX = "plex" - A `DM` object that encapsulates an unstructured mesh (or grid), or CW Complex {cite}`cwcomplex`,
57750b4b7b1cSBarry Smith            which can be expressed using a Hasse Diagram {cite}`hassediagram`.
577620f4b53cSBarry Smith            In the local representation, `Vec`s contain all unknowns in the interior and shared boundary. This is
57770b4b7b1cSBarry Smith            specified by a `PetscSection` object. Ownership in the global representation is determined by
5778a1cb98faSBarry Smith            ownership of the underlying `DMPLEX` points. This is specified by another `PetscSection` object.
57798818961aSMatthew G Knepley 
5780e5893cccSMatthew G. Knepley   Options Database Keys:
5781250712c9SMatthew G. Knepley + -dm_refine_pre                     - Refine mesh before distribution
5782250712c9SMatthew G. Knepley + -dm_refine_uniform_pre             - Choose uniform or generator-based refinement
5783250712c9SMatthew G. Knepley + -dm_refine_volume_limit_pre        - Cell volume limit after pre-refinement using generator
5784250712c9SMatthew G. Knepley . -dm_distribute                     - Distribute mesh across processes
5785250712c9SMatthew G. Knepley . -dm_distribute_overlap             - Number of cells to overlap for distribution
5786250712c9SMatthew G. Knepley . -dm_refine                         - Refine mesh after distribution
5787c3db174cSMatthew G. Knepley . -dm_localize <bool>                - Whether to localize coordinates for periodic meshes
5788c3db174cSMatthew G. Knepley . -dm_sparse_localize <bool>         - Whether to only localize cells on the periodic boundary
5789250712c9SMatthew G. Knepley . -dm_plex_hash_location             - Use grid hashing for point location
5790ddce0771SMatthew G. Knepley . -dm_plex_hash_box_faces <n,m,p>    - The number of divisions in each direction of the grid hash
5791f12cf164SMatthew G. Knepley . -dm_plex_partition_balance         - Attempt to evenly divide points on partition boundary between processes
5792f12cf164SMatthew G. Knepley . -dm_plex_remesh_bd                 - Allow changes to the boundary on remeshing
5793d5b43468SJose E. Roman . -dm_plex_max_projection_height     - Maximum mesh point height used to project locally
5794f12cf164SMatthew G. Knepley . -dm_plex_regular_refinement        - Use special nested projection algorithm for regular refinement
5795d02c7345SMatthew G. Knepley . -dm_plex_reorder_section           - Use specialized blocking if available
5796aaa8cc7dSPierre Jolivet . -dm_plex_check_all                 - Perform all checks below
5797f12cf164SMatthew G. Knepley . -dm_plex_check_symmetry            - Check that the adjacency information in the mesh is symmetric
5798f12cf164SMatthew G. Knepley . -dm_plex_check_skeleton <celltype> - Check that each cell has the correct number of vertices
5799f12cf164SMatthew 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
5800f12cf164SMatthew G. Knepley . -dm_plex_check_geometry            - Check that cells have positive volume
5801f12cf164SMatthew G. Knepley . -dm_view :mesh.tex:ascii_latex     - View the mesh in LaTeX/TikZ
5802e5893cccSMatthew G. Knepley . -dm_plex_view_scale <num>          - Scale the TikZ
58035962854dSMatthew G. Knepley . -dm_plex_print_fem <num>           - View FEM assembly information, such as element vectors and matrices
58045962854dSMatthew G. Knepley - -dm_plex_print_fvm <num>           - View FVM assembly information, such as flux updates
5805e5893cccSMatthew G. Knepley 
58068818961aSMatthew G Knepley   Level: intermediate
58078818961aSMatthew G Knepley 
58081cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMPlexCreate()`, `DMCreate()`, `DMSetType()`, `PetscSection`
58098818961aSMatthew G Knepley M*/
58108818961aSMatthew G Knepley 
5811d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm)
5812d71ae5a4SJacob Faibussowitsch {
5813552f7358SJed Brown   DM_Plex *mesh;
5814412e9a14SMatthew G. Knepley   PetscInt unit;
5815552f7358SJed Brown 
5816552f7358SJed Brown   PetscFunctionBegin;
5817f39ec787SMatthew G. Knepley   PetscCall(PetscCitationsRegister(PlexCitation, &Plexcite));
5818552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
58194dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mesh));
5820adc21957SMatthew G. Knepley   dm->reorderSection = DM_REORDER_DEFAULT_NOTSET;
5821552f7358SJed Brown   dm->data           = mesh;
5822552f7358SJed Brown 
5823552f7358SJed Brown   mesh->refct = 1;
58249566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection));
58259566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection));
5826552f7358SJed Brown   mesh->refinementUniform       = PETSC_TRUE;
5827552f7358SJed Brown   mesh->refinementLimit         = -1.0;
5828e600fa54SMatthew G. Knepley   mesh->distDefault             = PETSC_TRUE;
5829adc21957SMatthew G. Knepley   mesh->reorderDefault          = DM_REORDER_DEFAULT_NOTSET;
58301d1f2f2aSksagiyam   mesh->distributionName        = NULL;
58317d0f5628SVaclav Hapla   mesh->interpolated            = DMPLEX_INTERPOLATED_INVALID;
58327d0f5628SVaclav Hapla   mesh->interpolatedCollective  = DMPLEX_INTERPOLATED_INVALID;
58335e2c5519SMatthew G. Knepley   mesh->interpolatePreferTensor = PETSC_TRUE;
5834552f7358SJed Brown 
58359566063dSJacob Faibussowitsch   PetscCall(PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner));
58362e62ab5aSMatthew G. Knepley   mesh->remeshBd = PETSC_FALSE;
5837d9deefdfSMatthew G. Knepley 
58388865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
5839552f7358SJed Brown 
5840df0420ecSMatthew G. Knepley   mesh->depthState    = -1;
5841ba2698f1SMatthew G. Knepley   mesh->celltypeState = -1;
58426113b454SMatthew G. Knepley   mesh->printTol      = 1.0e-10;
5843c29ce622SStefano Zampini   mesh->nonempty_comm = MPI_COMM_SELF;
5844552f7358SJed Brown 
58459566063dSJacob Faibussowitsch   PetscCall(DMInitialize_Plex(dm));
58463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5847552f7358SJed Brown }
5848552f7358SJed Brown 
5849552f7358SJed Brown /*@
5850a1cb98faSBarry Smith   DMPlexCreate - Creates a `DMPLEX` object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
5851552f7358SJed Brown 
5852d083f849SBarry Smith   Collective
5853552f7358SJed Brown 
5854552f7358SJed Brown   Input Parameter:
5855a1cb98faSBarry Smith . comm - The communicator for the `DMPLEX` object
5856552f7358SJed Brown 
5857552f7358SJed Brown   Output Parameter:
5858a1cb98faSBarry Smith . mesh - The `DMPLEX` object
5859552f7358SJed Brown 
5860552f7358SJed Brown   Level: beginner
5861552f7358SJed Brown 
586242747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMCreate()`, `DMSetType()`
5863552f7358SJed Brown @*/
5864d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
5865d71ae5a4SJacob Faibussowitsch {
5866552f7358SJed Brown   PetscFunctionBegin;
58674f572ea9SToby Isaac   PetscAssertPointer(mesh, 2);
58689566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, mesh));
58699566063dSJacob Faibussowitsch   PetscCall(DMSetType(*mesh, DMPLEX));
58703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5871552f7358SJed Brown }
5872552f7358SJed Brown 
5873b09969d6SVaclav Hapla /*@C
5874b0fe842aSMatthew G. Knepley   DMPlexBuildFromCellListParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) where all cells have the same celltype
5875a1cb98faSBarry Smith 
587620f4b53cSBarry Smith   Collective; No Fortran Support
5877b09969d6SVaclav Hapla 
5878b09969d6SVaclav Hapla   Input Parameters:
5879a1cb98faSBarry Smith + dm          - The `DM`
5880b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
5881a1cb98faSBarry Smith . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
5882a1cb98faSBarry Smith . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
5883b09969d6SVaclav Hapla . numCorners  - The number of vertices for each cell
58845e488331SVaclav Hapla - cells       - An array of numCells*numCorners numbers, the global vertex numbers for each cell
5885b09969d6SVaclav Hapla 
5886be8c289dSNicolas Barral   Output Parameters:
5887a1cb98faSBarry Smith + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
5888be8c289dSNicolas Barral - verticesAdjSaved - (Optional) vertex adjacency array
5889b09969d6SVaclav Hapla 
5890b09969d6SVaclav Hapla   Level: advanced
5891b09969d6SVaclav Hapla 
5892a1cb98faSBarry Smith   Notes:
5893a1cb98faSBarry Smith   Two triangles sharing a face
5894a1cb98faSBarry Smith .vb
5895a1cb98faSBarry Smith 
5896a1cb98faSBarry Smith         2
5897a1cb98faSBarry Smith       / | \
5898a1cb98faSBarry Smith      /  |  \
5899a1cb98faSBarry Smith     /   |   \
5900a1cb98faSBarry Smith    0  0 | 1  3
5901a1cb98faSBarry Smith     \   |   /
5902a1cb98faSBarry Smith      \  |  /
5903a1cb98faSBarry Smith       \ | /
5904a1cb98faSBarry Smith         1
5905a1cb98faSBarry Smith .ve
5906a1cb98faSBarry Smith   would have input
5907a1cb98faSBarry Smith .vb
5908a1cb98faSBarry Smith   numCells = 2, numVertices = 4
5909a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
5910a1cb98faSBarry Smith .ve
5911a1cb98faSBarry Smith   which would result in the `DMPLEX`
5912a1cb98faSBarry Smith .vb
5913a1cb98faSBarry Smith 
5914a1cb98faSBarry Smith         4
5915a1cb98faSBarry Smith       / | \
5916a1cb98faSBarry Smith      /  |  \
5917a1cb98faSBarry Smith     /   |   \
5918a1cb98faSBarry Smith    2  0 | 1  5
5919a1cb98faSBarry Smith     \   |   /
5920a1cb98faSBarry Smith      \  |  /
5921a1cb98faSBarry Smith       \ | /
5922a1cb98faSBarry Smith         3
5923a1cb98faSBarry Smith .ve
5924a1cb98faSBarry Smith 
5925a1cb98faSBarry Smith   Vertices are implicitly numbered consecutively 0,...,NVertices.
5926a1cb98faSBarry Smith   Each rank owns a chunk of numVertices consecutive vertices.
5927a1cb98faSBarry Smith   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
5928a1cb98faSBarry Smith   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
5929a1cb98faSBarry Smith   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
5930a1cb98faSBarry Smith 
5931a1cb98faSBarry Smith   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
5932a1cb98faSBarry Smith 
59331cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
5934a1cb98faSBarry Smith           `PetscSF`
5935b09969d6SVaclav Hapla @*/
5936ce78bad3SBarry Smith PetscErrorCode DMPlexBuildFromCellListParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, const PetscInt cells[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdjSaved[])
5937d71ae5a4SJacob Faibussowitsch {
59382464107aSksagiyam   PetscSF     sfPoint;
59392464107aSksagiyam   PetscLayout layout;
594082fb893eSVaclav Hapla   PetscInt    numVerticesAdj, *verticesAdj, *cones, c, p;
5941a47d0d45SMatthew G. Knepley 
5942a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
594325b6865aSVaclav Hapla   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
59449566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
594525b6865aSVaclav Hapla   /* Get/check global number of vertices */
594625b6865aSVaclav Hapla   {
594725b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
594825b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
594925b6865aSVaclav Hapla 
595025b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
59511690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
59529371c9d4SSatish Balay     for (i = 0; i < len; i++)
59539371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
595425b6865aSVaclav Hapla     ++NVerticesInCells;
5955462c564dSBarry Smith     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
595625b6865aSVaclav Hapla 
595725b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
59589371c9d4SSatish Balay     else
59599371c9d4SSatish 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);
596025b6865aSVaclav Hapla   }
59619079aca8SVaclav Hapla   /* Count locally unique vertices */
59629079aca8SVaclav Hapla   {
59639079aca8SVaclav Hapla     PetscHSetI vhash;
59649079aca8SVaclav Hapla     PetscInt   off = 0;
59659079aca8SVaclav Hapla 
59669566063dSJacob Faibussowitsch     PetscCall(PetscHSetICreate(&vhash));
5967a47d0d45SMatthew G. Knepley     for (c = 0; c < numCells; ++c) {
596848a46eb9SPierre Jolivet       for (p = 0; p < numCorners; ++p) PetscCall(PetscHSetIAdd(vhash, cells[c * numCorners + p]));
5969a47d0d45SMatthew G. Knepley     }
59709566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
59719566063dSJacob Faibussowitsch     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
5972ad540459SPierre Jolivet     else verticesAdj = *verticesAdjSaved;
59739566063dSJacob Faibussowitsch     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
59749566063dSJacob Faibussowitsch     PetscCall(PetscHSetIDestroy(&vhash));
597563a3b9bcSJacob Faibussowitsch     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
5976a47d0d45SMatthew G. Knepley   }
59779566063dSJacob Faibussowitsch   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
5978a47d0d45SMatthew G. Knepley   /* Create cones */
59799566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
59809566063dSJacob Faibussowitsch   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
59819566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
59829566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
5983a47d0d45SMatthew G. Knepley   for (c = 0; c < numCells; ++c) {
5984a47d0d45SMatthew G. Knepley     for (p = 0; p < numCorners; ++p) {
5985a47d0d45SMatthew G. Knepley       const PetscInt gv = cells[c * numCorners + p];
5986a47d0d45SMatthew G. Knepley       PetscInt       lv;
5987a47d0d45SMatthew G. Knepley 
59889079aca8SVaclav Hapla       /* Positions within verticesAdj form 0-based local vertex numbering;
59899079aca8SVaclav Hapla          we need to shift it by numCells to get correct DAG points (cells go first) */
59909566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
599163a3b9bcSJacob Faibussowitsch       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
5992961cfab0SVaclav Hapla       cones[c * numCorners + p] = lv + numCells;
5993a47d0d45SMatthew G. Knepley     }
5994a47d0d45SMatthew G. Knepley   }
59952464107aSksagiyam   /* Build point sf */
59969566063dSJacob Faibussowitsch   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
59979566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetSize(layout, NVertices));
59989566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
59999566063dSJacob Faibussowitsch   PetscCall(PetscLayoutSetBlockSize(layout, 1));
60009566063dSJacob Faibussowitsch   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
60019566063dSJacob Faibussowitsch   PetscCall(PetscLayoutDestroy(&layout));
60029566063dSJacob Faibussowitsch   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
60039566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
60042464107aSksagiyam   if (dm->sf) {
60052464107aSksagiyam     const char *prefix;
60062464107aSksagiyam 
60079566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
60089566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
60092464107aSksagiyam   }
60109566063dSJacob Faibussowitsch   PetscCall(DMSetPointSF(dm, sfPoint));
60119566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&sfPoint));
6012f4f49eeaSPierre Jolivet   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
6013a47d0d45SMatthew G. Knepley   /* Fill in the rest of the topology structure */
60149566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
60159566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
60169566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
60173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6018a47d0d45SMatthew G. Knepley }
6019a47d0d45SMatthew G. Knepley 
6020b0fe842aSMatthew G. Knepley /*@C
6021b0fe842aSMatthew G. Knepley   DMPlexBuildFromCellSectionParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) allowing multiple celltypes
6022b0fe842aSMatthew G. Knepley 
6023b0fe842aSMatthew G. Knepley   Collective; No Fortran Support
6024b0fe842aSMatthew G. Knepley 
6025b0fe842aSMatthew G. Knepley   Input Parameters:
6026b0fe842aSMatthew G. Knepley + dm          - The `DM`
6027b0fe842aSMatthew G. Knepley . numCells    - The number of cells owned by this process
6028b0fe842aSMatthew G. Knepley . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE`
6029b0fe842aSMatthew G. Knepley . NVertices   - The global number of vertices, or `PETSC_DETERMINE`
6030b0fe842aSMatthew G. Knepley . cellSection - The `PetscSection` giving the number of vertices for each cell (layout of cells)
6031b0fe842aSMatthew G. Knepley - cells       - An array of the global vertex numbers for each cell
6032b0fe842aSMatthew G. Knepley 
6033b0fe842aSMatthew G. Knepley   Output Parameters:
6034b0fe842aSMatthew G. Knepley + vertexSF         - (Optional) `PetscSF` describing complete vertex ownership
6035b0fe842aSMatthew G. Knepley - verticesAdjSaved - (Optional) vertex adjacency array
6036b0fe842aSMatthew G. Knepley 
6037b0fe842aSMatthew G. Knepley   Level: advanced
6038b0fe842aSMatthew G. Knepley 
6039b0fe842aSMatthew G. Knepley   Notes:
6040b0fe842aSMatthew G. Knepley   A triangle and quadrilateral sharing a face
6041b0fe842aSMatthew G. Knepley .vb
6042b0fe842aSMatthew G. Knepley         2----------3
6043b0fe842aSMatthew G. Knepley       / |          |
6044b0fe842aSMatthew G. Knepley      /  |          |
6045b0fe842aSMatthew G. Knepley     /   |          |
6046b0fe842aSMatthew G. Knepley    0  0 |     1    |
6047b0fe842aSMatthew G. Knepley     \   |          |
6048b0fe842aSMatthew G. Knepley      \  |          |
6049b0fe842aSMatthew G. Knepley       \ |          |
6050b0fe842aSMatthew G. Knepley         1----------4
6051b0fe842aSMatthew G. Knepley .ve
6052b0fe842aSMatthew G. Knepley   would have input
6053b0fe842aSMatthew G. Knepley .vb
6054b0fe842aSMatthew G. Knepley   numCells = 2, numVertices = 5
6055b0fe842aSMatthew G. Knepley   cells = [0 1 2  1 4 3 2]
6056b0fe842aSMatthew G. Knepley .ve
6057b0fe842aSMatthew G. Knepley   which would result in the `DMPLEX`
6058b0fe842aSMatthew G. Knepley .vb
6059b0fe842aSMatthew G. Knepley         4----------5
6060b0fe842aSMatthew G. Knepley       / |          |
6061b0fe842aSMatthew G. Knepley      /  |          |
6062b0fe842aSMatthew G. Knepley     /   |          |
6063b0fe842aSMatthew G. Knepley    2  0 |     1    |
6064b0fe842aSMatthew G. Knepley     \   |          |
6065b0fe842aSMatthew G. Knepley      \  |          |
6066b0fe842aSMatthew G. Knepley       \ |          |
6067b0fe842aSMatthew G. Knepley         3----------6
6068b0fe842aSMatthew G. Knepley .ve
6069b0fe842aSMatthew G. Knepley 
6070b0fe842aSMatthew G. Knepley   Vertices are implicitly numbered consecutively 0,...,NVertices.
6071b0fe842aSMatthew G. Knepley   Each rank owns a chunk of numVertices consecutive vertices.
6072b0fe842aSMatthew G. Knepley   If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout.
6073b0fe842aSMatthew G. Knepley   If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1.
6074b0fe842aSMatthew G. Knepley   If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks.
6075b0fe842aSMatthew G. Knepley 
6076b0fe842aSMatthew G. Knepley   The cell distribution is arbitrary non-overlapping, independent of the vertex distribution.
6077b0fe842aSMatthew G. Knepley 
6078b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexCreateFromCellSectionParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`,
6079b0fe842aSMatthew G. Knepley           `PetscSF`
6080b0fe842aSMatthew G. Knepley @*/
6081ce78bad3SBarry Smith PetscErrorCode DMPlexBuildFromCellSectionParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, const PetscInt cells[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdjSaved[])
6082b0fe842aSMatthew G. Knepley {
6083b0fe842aSMatthew G. Knepley   PetscSF     sfPoint;
6084b0fe842aSMatthew G. Knepley   PetscLayout layout;
6085b0fe842aSMatthew G. Knepley   PetscInt    numVerticesAdj, *verticesAdj, *cones, cStart, cEnd, len;
6086b0fe842aSMatthew G. Knepley 
6087b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
6088b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(dm, NVertices, 4);
6089b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
6090b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetChart(cellSection, &cStart, &cEnd));
6091b0fe842aSMatthew G. Knepley   PetscCall(PetscSectionGetStorageSize(cellSection, &len));
6092b0fe842aSMatthew G. Knepley   PetscCheck(cStart == 0 && cEnd == numCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Section chart [%" PetscInt_FMT ", %" PetscInt_FMT ") should be [0, %" PetscInt_FMT ")", cStart, cEnd, numCells);
6093b0fe842aSMatthew G. Knepley   /* Get/check global number of vertices */
6094b0fe842aSMatthew G. Knepley   {
6095b0fe842aSMatthew G. Knepley     PetscInt NVerticesInCells;
6096b0fe842aSMatthew G. Knepley 
6097b0fe842aSMatthew G. Knepley     /* NVerticesInCells = max(cells) + 1 */
6098b0fe842aSMatthew G. Knepley     NVerticesInCells = PETSC_MIN_INT;
6099b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++)
6100b0fe842aSMatthew G. Knepley       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
6101b0fe842aSMatthew G. Knepley     ++NVerticesInCells;
6102b0fe842aSMatthew G. Knepley     PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
6103b0fe842aSMatthew G. Knepley 
6104b0fe842aSMatthew G. Knepley     if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells;
6105b0fe842aSMatthew G. Knepley     else
6106b0fe842aSMatthew G. Knepley       PetscCheck(NVertices == PETSC_DECIDE || NVertices >= NVerticesInCells, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Specified global number of vertices %" PetscInt_FMT " must be greater than or equal to the number of vertices in cells %" PetscInt_FMT, NVertices, NVerticesInCells);
6107b0fe842aSMatthew G. Knepley   }
6108b0fe842aSMatthew G. Knepley   /* Count locally unique vertices */
6109b0fe842aSMatthew G. Knepley   {
6110b0fe842aSMatthew G. Knepley     PetscHSetI vhash;
6111b0fe842aSMatthew G. Knepley     PetscInt   off = 0;
6112b0fe842aSMatthew G. Knepley 
6113b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetICreate(&vhash));
6114b0fe842aSMatthew G. Knepley     for (PetscInt i = 0; i < len; i++) PetscCall(PetscHSetIAdd(vhash, cells[i]));
6115b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj));
6116b0fe842aSMatthew G. Knepley     if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj));
6117b0fe842aSMatthew G. Knepley     else verticesAdj = *verticesAdjSaved;
6118b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj));
6119b0fe842aSMatthew G. Knepley     PetscCall(PetscHSetIDestroy(&vhash));
6120b0fe842aSMatthew G. Knepley     PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj);
6121b0fe842aSMatthew G. Knepley   }
6122b0fe842aSMatthew G. Knepley   PetscCall(PetscSortInt(numVerticesAdj, verticesAdj));
6123b0fe842aSMatthew G. Knepley   /* Create cones */
6124b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj));
6125b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
6126b0fe842aSMatthew G. Knepley     PetscInt dof;
6127b0fe842aSMatthew G. Knepley 
6128b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
6129b0fe842aSMatthew G. Knepley     PetscCall(DMPlexSetConeSize(dm, c, dof));
6130b0fe842aSMatthew G. Knepley   }
6131b0fe842aSMatthew G. Knepley   PetscCall(DMSetUp(dm));
6132b0fe842aSMatthew G. Knepley   PetscCall(DMPlexGetCones(dm, &cones));
6133b0fe842aSMatthew G. Knepley   for (PetscInt c = 0; c < numCells; ++c) {
6134b0fe842aSMatthew G. Knepley     PetscInt dof, off;
6135b0fe842aSMatthew G. Knepley 
6136b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetDof(cellSection, c, &dof));
6137b0fe842aSMatthew G. Knepley     PetscCall(PetscSectionGetOffset(cellSection, c, &off));
6138b0fe842aSMatthew G. Knepley     for (PetscInt p = off; p < off + dof; ++p) {
6139b0fe842aSMatthew G. Knepley       const PetscInt gv = cells[p];
6140b0fe842aSMatthew G. Knepley       PetscInt       lv;
6141b0fe842aSMatthew G. Knepley 
6142b0fe842aSMatthew G. Knepley       /* Positions within verticesAdj form 0-based local vertex numbering;
6143b0fe842aSMatthew G. Knepley          we need to shift it by numCells to get correct DAG points (cells go first) */
6144b0fe842aSMatthew G. Knepley       PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv));
6145b0fe842aSMatthew G. Knepley       PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv);
6146b0fe842aSMatthew G. Knepley       cones[p] = lv + numCells;
6147b0fe842aSMatthew G. Knepley     }
6148b0fe842aSMatthew G. Knepley   }
6149b0fe842aSMatthew G. Knepley   /* Build point sf */
6150b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout));
6151b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetSize(layout, NVertices));
6152b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetLocalSize(layout, numVertices));
6153b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutSetBlockSize(layout, 1));
6154b0fe842aSMatthew G. Knepley   PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint));
6155b0fe842aSMatthew G. Knepley   PetscCall(PetscLayoutDestroy(&layout));
6156b0fe842aSMatthew G. Knepley   if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj));
6157b0fe842aSMatthew G. Knepley   PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF"));
6158b0fe842aSMatthew G. Knepley   if (dm->sf) {
6159b0fe842aSMatthew G. Knepley     const char *prefix;
6160b0fe842aSMatthew G. Knepley 
6161b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix));
6162b0fe842aSMatthew G. Knepley     PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix));
6163b0fe842aSMatthew G. Knepley   }
6164b0fe842aSMatthew G. Knepley   PetscCall(DMSetPointSF(dm, sfPoint));
6165b0fe842aSMatthew G. Knepley   PetscCall(PetscSFDestroy(&sfPoint));
6166b0fe842aSMatthew G. Knepley   if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF"));
6167b0fe842aSMatthew G. Knepley   /* Fill in the rest of the topology structure */
6168b0fe842aSMatthew G. Knepley   PetscCall(DMPlexSymmetrize(dm));
6169b0fe842aSMatthew G. Knepley   PetscCall(DMPlexStratify(dm));
6170b0fe842aSMatthew G. Knepley   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
6171b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
6172b0fe842aSMatthew G. Knepley }
6173b0fe842aSMatthew G. Knepley 
6174cc4c1da9SBarry Smith /*@
6175a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellListParallel - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
6176a1cb98faSBarry Smith 
617720f4b53cSBarry Smith   Collective; No Fortran Support
6178b09969d6SVaclav Hapla 
6179b09969d6SVaclav Hapla   Input Parameters:
6180a1cb98faSBarry Smith + dm           - The `DM`
6181b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
6182a1cb98faSBarry Smith . sfVert       - `PetscSF` describing complete vertex ownership
6183b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
6184b09969d6SVaclav Hapla 
6185b09969d6SVaclav Hapla   Level: advanced
6186b09969d6SVaclav Hapla 
61871cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellListParallel()`
6188b09969d6SVaclav Hapla @*/
6189d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellListParallel(DM dm, PetscInt spaceDim, PetscSF sfVert, const PetscReal vertexCoords[])
6190d71ae5a4SJacob Faibussowitsch {
6191a47d0d45SMatthew G. Knepley   PetscSection coordSection;
6192a47d0d45SMatthew G. Knepley   Vec          coordinates;
6193a47d0d45SMatthew G. Knepley   PetscScalar *coords;
61941edcf0b2SVaclav Hapla   PetscInt     numVertices, numVerticesAdj, coordSize, v, vStart, vEnd;
6195835f2295SStefano Zampini   PetscMPIInt  spaceDimi;
6196a47d0d45SMatthew G. Knepley 
6197a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
61989566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
61999566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
62001dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
62019566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
62029566063dSJacob Faibussowitsch   PetscCall(PetscSFGetGraph(sfVert, &numVertices, &numVerticesAdj, NULL, NULL));
62031dca8a05SBarry 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);
62049566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
62059566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
62069566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
62079566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
62081edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
62099566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
62109566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
6211a47d0d45SMatthew G. Knepley   }
62129566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
62139566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
62149566063dSJacob Faibussowitsch   PetscCall(VecCreate(PetscObjectComm((PetscObject)dm), &coordinates));
62159566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
62169566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
62179566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
62189566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
62199566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
6220a47d0d45SMatthew G. Knepley   {
6221a47d0d45SMatthew G. Knepley     MPI_Datatype coordtype;
6222a47d0d45SMatthew G. Knepley 
6223a47d0d45SMatthew G. Knepley     /* Need a temp buffer for coords if we have complex/single */
6224835f2295SStefano Zampini     PetscCall(PetscMPIIntCast(spaceDim, &spaceDimi));
6225835f2295SStefano Zampini     PetscCallMPI(MPI_Type_contiguous(spaceDimi, MPIU_SCALAR, &coordtype));
62269566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_commit(&coordtype));
622721016a8bSBarry Smith #if defined(PETSC_USE_COMPLEX)
622821016a8bSBarry Smith     {
622921016a8bSBarry Smith       PetscScalar *svertexCoords;
623021016a8bSBarry Smith       PetscInt     i;
62319566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numVertices * spaceDim, &svertexCoords));
62323612f820SVaclav Hapla       for (i = 0; i < numVertices * spaceDim; i++) svertexCoords[i] = vertexCoords[i];
62339566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
62349566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE));
62359566063dSJacob Faibussowitsch       PetscCall(PetscFree(svertexCoords));
623621016a8bSBarry Smith     }
623721016a8bSBarry Smith #else
62389566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastBegin(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
62399566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastEnd(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE));
624021016a8bSBarry Smith #endif
62419566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Type_free(&coordtype));
6242a47d0d45SMatthew G. Knepley   }
62439566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
62449566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
62459566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
62469566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
62473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6248a47d0d45SMatthew G. Knepley }
6249a47d0d45SMatthew G. Knepley 
6250c3edce3dSSatish Balay /*@
6251b0fe842aSMatthew G. Knepley   DMPlexCreateFromCellListParallelPetsc - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) where all cells have the same celltype
6252a1cb98faSBarry Smith 
6253a1cb98faSBarry Smith   Collective
6254a47d0d45SMatthew G. Knepley 
6255a47d0d45SMatthew G. Knepley   Input Parameters:
6256a47d0d45SMatthew G. Knepley + comm         - The communicator
6257a47d0d45SMatthew G. Knepley . dim          - The topological dimension of the mesh
6258a47d0d45SMatthew G. Knepley . numCells     - The number of cells owned by this process
6259a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
6260a1cb98faSBarry Smith . NVertices    - The global number of vertices, or `PETSC_DECIDE`
6261a47d0d45SMatthew G. Knepley . numCorners   - The number of vertices for each cell
6262a47d0d45SMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
6263a47d0d45SMatthew G. Knepley . cells        - An array of numCells*numCorners numbers, the global vertex numbers for each cell
6264a47d0d45SMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
6265a47d0d45SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
6266a47d0d45SMatthew G. Knepley 
6267d8d19677SJose E. Roman   Output Parameters:
6268a1cb98faSBarry Smith + dm          - The `DM`
6269a1cb98faSBarry Smith . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
627060225df5SJacob Faibussowitsch - verticesAdj - (Optional) vertex adjacency array
6271a47d0d45SMatthew G. Knepley 
6272b09969d6SVaclav Hapla   Level: intermediate
6273a47d0d45SMatthew G. Knepley 
6274a1cb98faSBarry Smith   Notes:
6275a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
6276a1cb98faSBarry Smith   `DMPlexBuildFromCellListParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
6277a1cb98faSBarry Smith 
6278a1cb98faSBarry Smith   See `DMPlexBuildFromCellListParallel()` for an example and details about the topology-related parameters.
6279a1cb98faSBarry Smith 
6280a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
6281a1cb98faSBarry Smith 
62821cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
6283a47d0d45SMatthew G. Knepley @*/
6284ce78bad3SBarry Smith 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[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdj[], DM *dm)
6285d71ae5a4SJacob Faibussowitsch {
6286a47d0d45SMatthew G. Knepley   PetscSF sfVert;
6287a47d0d45SMatthew G. Knepley 
6288a47d0d45SMatthew G. Knepley   PetscFunctionBegin;
62899566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
62909566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
6291a47d0d45SMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
6292064a246eSJacob Faibussowitsch   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
62939566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
62949566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildFromCellListParallel(*dm, numCells, numVertices, NVertices, numCorners, cells, &sfVert, verticesAdj));
6295a47d0d45SMatthew G. Knepley   if (interpolate) {
62965fd9971aSMatthew G. Knepley     DM idm;
6297a47d0d45SMatthew G. Knepley 
62989566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
62999566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
6300a47d0d45SMatthew G. Knepley     *dm = idm;
6301a47d0d45SMatthew G. Knepley   }
63029566063dSJacob Faibussowitsch   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
630318d54ad4SMichael Lange   if (vertexSF) *vertexSF = sfVert;
63049566063dSJacob Faibussowitsch   else PetscCall(PetscSFDestroy(&sfVert));
63053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6306a47d0d45SMatthew G. Knepley }
6307a47d0d45SMatthew G. Knepley 
6308cc4c1da9SBarry Smith /*@
6309b0fe842aSMatthew G. Knepley   DMPlexCreateFromCellSectionParallel - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) and supports multiple celltypes
6310b0fe842aSMatthew G. Knepley 
6311b0fe842aSMatthew G. Knepley   Collective
6312b0fe842aSMatthew G. Knepley 
6313b0fe842aSMatthew G. Knepley   Input Parameters:
6314b0fe842aSMatthew G. Knepley + comm         - The communicator
6315b0fe842aSMatthew G. Knepley . dim          - The topological dimension of the mesh
6316b0fe842aSMatthew G. Knepley . numCells     - The number of cells owned by this process
6317b0fe842aSMatthew G. Knepley . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`
6318b0fe842aSMatthew G. Knepley . NVertices    - The global number of vertices, or `PETSC_DECIDE`
6319b0fe842aSMatthew G. Knepley . cellSection  - The `PetscSection` giving the number of vertices for each cell (layout of cells)
6320b0fe842aSMatthew G. Knepley . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
6321b0fe842aSMatthew G. Knepley . cells        - An array of the global vertex numbers for each cell
6322b0fe842aSMatthew G. Knepley . spaceDim     - The spatial dimension used for coordinates
6323b0fe842aSMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
6324b0fe842aSMatthew G. Knepley 
6325b0fe842aSMatthew G. Knepley   Output Parameters:
6326b0fe842aSMatthew G. Knepley + dm          - The `DM`
6327b0fe842aSMatthew G. Knepley . vertexSF    - (Optional) `PetscSF` describing complete vertex ownership
6328b0fe842aSMatthew G. Knepley - verticesAdj - (Optional) vertex adjacency array
6329b0fe842aSMatthew G. Knepley 
6330b0fe842aSMatthew G. Knepley   Level: intermediate
6331b0fe842aSMatthew G. Knepley 
6332b0fe842aSMatthew G. Knepley   Notes:
6333b0fe842aSMatthew G. Knepley   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`,
6334b0fe842aSMatthew G. Knepley   `DMPlexBuildFromCellSectionParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()`
6335b0fe842aSMatthew G. Knepley 
6336b0fe842aSMatthew G. Knepley   See `DMPlexBuildFromCellSectionParallel()` for an example and details about the topology-related parameters.
6337b0fe842aSMatthew G. Knepley 
6338b0fe842aSMatthew G. Knepley   See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters.
6339b0fe842aSMatthew G. Knepley 
6340b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
6341b0fe842aSMatthew G. Knepley @*/
6342ce78bad3SBarry Smith PetscErrorCode DMPlexCreateFromCellSectionParallel(MPI_Comm comm, PetscInt dim, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, PetscBool interpolate, const PetscInt cells[], PetscInt spaceDim, const PetscReal vertexCoords[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdj[], DM *dm)
6343b0fe842aSMatthew G. Knepley {
6344b0fe842aSMatthew G. Knepley   PetscSF sfVert;
6345b0fe842aSMatthew G. Knepley 
6346b0fe842aSMatthew G. Knepley   PetscFunctionBegin;
6347b0fe842aSMatthew G. Knepley   PetscCall(DMCreate(comm, dm));
6348b0fe842aSMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
6349b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, dim, 2);
6350b0fe842aSMatthew G. Knepley   PetscValidLogicalCollectiveInt(*dm, spaceDim, 9);
6351b0fe842aSMatthew G. Knepley   PetscCall(DMSetDimension(*dm, dim));
6352b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildFromCellSectionParallel(*dm, numCells, numVertices, NVertices, cellSection, cells, &sfVert, verticesAdj));
6353b0fe842aSMatthew G. Knepley   if (interpolate) {
6354b0fe842aSMatthew G. Knepley     DM idm;
6355b0fe842aSMatthew G. Knepley 
6356b0fe842aSMatthew G. Knepley     PetscCall(DMPlexInterpolate(*dm, &idm));
6357b0fe842aSMatthew G. Knepley     PetscCall(DMDestroy(dm));
6358b0fe842aSMatthew G. Knepley     *dm = idm;
6359b0fe842aSMatthew G. Knepley   }
6360b0fe842aSMatthew G. Knepley   PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords));
6361b0fe842aSMatthew G. Knepley   if (vertexSF) *vertexSF = sfVert;
6362b0fe842aSMatthew G. Knepley   else PetscCall(PetscSFDestroy(&sfVert));
6363b0fe842aSMatthew G. Knepley   PetscFunctionReturn(PETSC_SUCCESS);
6364b0fe842aSMatthew G. Knepley }
6365b0fe842aSMatthew G. Knepley 
6366b0fe842aSMatthew G. Knepley /*@
6367a1cb98faSBarry Smith   DMPlexBuildFromCellList - Build `DMPLEX` topology from a list of vertices for each cell (common mesh generator output)
6368a1cb98faSBarry Smith 
636920f4b53cSBarry Smith   Collective; No Fortran Support
63709298eaa6SMatthew G Knepley 
63719298eaa6SMatthew G Knepley   Input Parameters:
6372a1cb98faSBarry Smith + dm          - The `DM`
6373b09969d6SVaclav Hapla . numCells    - The number of cells owned by this process
6374a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DETERMINE`
63759298eaa6SMatthew G Knepley . numCorners  - The number of vertices for each cell
6376a3b724e8SBarry Smith - cells       - An array of `numCells` x `numCorners` numbers, the global vertex numbers for each cell
63779298eaa6SMatthew G Knepley 
6378b09969d6SVaclav Hapla   Level: advanced
63799298eaa6SMatthew G Knepley 
6380b09969d6SVaclav Hapla   Notes:
6381b09969d6SVaclav Hapla   Two triangles sharing a face
6382a1cb98faSBarry Smith .vb
63839298eaa6SMatthew G Knepley 
6384a1cb98faSBarry Smith         2
6385a1cb98faSBarry Smith       / | \
6386a1cb98faSBarry Smith      /  |  \
6387a1cb98faSBarry Smith     /   |   \
6388a1cb98faSBarry Smith    0  0 | 1  3
6389a1cb98faSBarry Smith     \   |   /
6390a1cb98faSBarry Smith      \  |  /
6391a1cb98faSBarry Smith       \ | /
6392a1cb98faSBarry Smith         1
6393a1cb98faSBarry Smith .ve
6394a1cb98faSBarry Smith   would have input
6395a1cb98faSBarry Smith .vb
6396a1cb98faSBarry Smith   numCells = 2, numVertices = 4
6397a1cb98faSBarry Smith   cells = [0 1 2  1 3 2]
6398a1cb98faSBarry Smith .ve
6399a1cb98faSBarry Smith   which would result in the `DMPLEX`
6400a1cb98faSBarry Smith .vb
6401a1cb98faSBarry Smith 
6402a1cb98faSBarry Smith         4
6403a1cb98faSBarry Smith       / | \
6404a1cb98faSBarry Smith      /  |  \
6405a1cb98faSBarry Smith     /   |   \
6406a1cb98faSBarry Smith    2  0 | 1  5
6407a1cb98faSBarry Smith     \   |   /
6408a1cb98faSBarry Smith      \  |  /
6409a1cb98faSBarry Smith       \ | /
6410a1cb98faSBarry Smith         3
6411a1cb98faSBarry Smith .ve
6412a1cb98faSBarry Smith 
6413a1cb98faSBarry Smith   If numVertices is `PETSC_DETERMINE`, it is computed by PETSc as the maximum vertex index in cells + 1.
641425b6865aSVaclav Hapla 
64151cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListPetsc()`
6416b09969d6SVaclav Hapla @*/
6417d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellList(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const PetscInt cells[])
6418d71ae5a4SJacob Faibussowitsch {
6419961cfab0SVaclav Hapla   PetscInt *cones, c, p, dim;
6420b09969d6SVaclav Hapla 
6421b09969d6SVaclav Hapla   PetscFunctionBegin;
64229566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
64239566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
642425b6865aSVaclav Hapla   /* Get/check global number of vertices */
642525b6865aSVaclav Hapla   {
642625b6865aSVaclav Hapla     PetscInt       NVerticesInCells, i;
642725b6865aSVaclav Hapla     const PetscInt len = numCells * numCorners;
642825b6865aSVaclav Hapla 
642925b6865aSVaclav Hapla     /* NVerticesInCells = max(cells) + 1 */
64301690c2aeSBarry Smith     NVerticesInCells = PETSC_INT_MIN;
64319371c9d4SSatish Balay     for (i = 0; i < len; i++)
64329371c9d4SSatish Balay       if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i];
643325b6865aSVaclav Hapla     ++NVerticesInCells;
643425b6865aSVaclav Hapla 
643525b6865aSVaclav Hapla     if (numVertices == PETSC_DECIDE) numVertices = NVerticesInCells;
64369371c9d4SSatish Balay     else
64379371c9d4SSatish 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);
643825b6865aSVaclav Hapla   }
64399566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices));
644048a46eb9SPierre Jolivet   for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners));
64419566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
64429566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCones(dm, &cones));
6443b09969d6SVaclav Hapla   for (c = 0; c < numCells; ++c) {
6444ad540459SPierre Jolivet     for (p = 0; p < numCorners; ++p) cones[c * numCorners + p] = cells[c * numCorners + p] + numCells;
6445b09969d6SVaclav Hapla   }
64469566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
64479566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
64489566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0));
64493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6450b09969d6SVaclav Hapla }
6451b09969d6SVaclav Hapla 
6452cc4c1da9SBarry Smith /*@
6453a1cb98faSBarry Smith   DMPlexBuildCoordinatesFromCellList - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output)
6454a1cb98faSBarry Smith 
6455cc4c1da9SBarry Smith   Collective
6456b09969d6SVaclav Hapla 
6457b09969d6SVaclav Hapla   Input Parameters:
6458a1cb98faSBarry Smith + dm           - The `DM`
6459b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
6460b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
6461b09969d6SVaclav Hapla 
6462b09969d6SVaclav Hapla   Level: advanced
6463b09969d6SVaclav Hapla 
64641cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellList()`
6465b09969d6SVaclav Hapla @*/
6466d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellList(DM dm, PetscInt spaceDim, const PetscReal vertexCoords[])
6467d71ae5a4SJacob Faibussowitsch {
6468b09969d6SVaclav Hapla   PetscSection coordSection;
6469b09969d6SVaclav Hapla   Vec          coordinates;
6470b09969d6SVaclav Hapla   DM           cdm;
6471b09969d6SVaclav Hapla   PetscScalar *coords;
64721edcf0b2SVaclav Hapla   PetscInt     v, vStart, vEnd, d;
6473b09969d6SVaclav Hapla 
6474b09969d6SVaclav Hapla   PetscFunctionBegin;
64759566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
64769566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
64771dca8a05SBarry Smith   PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first.");
64789566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(dm, spaceDim));
64799566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
64809566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
64819566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim));
64829566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd));
64831edcf0b2SVaclav Hapla   for (v = vStart; v < vEnd; ++v) {
64849566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, spaceDim));
64859566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim));
6486b09969d6SVaclav Hapla   }
64879566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
6488b09969d6SVaclav Hapla 
64899566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
64909566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(cdm, &coordinates));
64919566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, spaceDim));
64929566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
64939566063dSJacob Faibussowitsch   PetscCall(VecGetArrayWrite(coordinates, &coords));
64941edcf0b2SVaclav Hapla   for (v = 0; v < vEnd - vStart; ++v) {
6495ad540459SPierre Jolivet     for (d = 0; d < spaceDim; ++d) coords[v * spaceDim + d] = vertexCoords[v * spaceDim + d];
6496b09969d6SVaclav Hapla   }
64979566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayWrite(coordinates, &coords));
64989566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
64999566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
65009566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0));
65013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6502b09969d6SVaclav Hapla }
6503b09969d6SVaclav Hapla 
6504b09969d6SVaclav Hapla /*@
6505a1cb98faSBarry Smith   DMPlexCreateFromCellListPetsc - Create `DMPLEX` from a list of vertices for each cell (common mesh generator output), but only process 0 takes in the input
65063df08285SMatthew G. Knepley 
6507a1cb98faSBarry Smith   Collective
6508b09969d6SVaclav Hapla 
6509b09969d6SVaclav Hapla   Input Parameters:
6510b09969d6SVaclav Hapla + comm         - The communicator
6511b09969d6SVaclav Hapla . dim          - The topological dimension of the mesh
65123df08285SMatthew G. Knepley . numCells     - The number of cells, only on process 0
6513a1cb98faSBarry Smith . numVertices  - The number of vertices owned by this process, or `PETSC_DECIDE`, only on process 0
65143df08285SMatthew G. Knepley . numCorners   - The number of vertices for each cell, only on process 0
6515b09969d6SVaclav Hapla . interpolate  - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
6516ce78bad3SBarry Smith . cells        - An array of $ numCells \times numCorners$ numbers, the vertices for each cell, only on process 0
6517b09969d6SVaclav Hapla . spaceDim     - The spatial dimension used for coordinates
6518ce78bad3SBarry Smith - vertexCoords - An array of $ numVertices \times spaceDim$ numbers, the coordinates of each vertex, only on process 0
6519b09969d6SVaclav Hapla 
6520b09969d6SVaclav Hapla   Output Parameter:
6521a1cb98faSBarry Smith . dm - The `DM`, which only has points on process 0
652225b6865aSVaclav Hapla 
6523b09969d6SVaclav Hapla   Level: intermediate
6524b09969d6SVaclav Hapla 
6525a1cb98faSBarry Smith   Notes:
6526a1cb98faSBarry Smith   This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, `DMPlexBuildFromCellList()`,
6527a1cb98faSBarry Smith   `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellList()`
6528a1cb98faSBarry Smith 
6529a1cb98faSBarry Smith   See `DMPlexBuildFromCellList()` for an example and details about the topology-related parameters.
6530a1cb98faSBarry Smith   See `DMPlexBuildCoordinatesFromCellList()` for details about the geometry-related parameters.
6531a1cb98faSBarry Smith   See `DMPlexCreateFromCellListParallelPetsc()` for parallel input
6532a1cb98faSBarry Smith 
65331cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellList()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()`
65349298eaa6SMatthew G Knepley @*/
6535d71ae5a4SJacob 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)
6536d71ae5a4SJacob Faibussowitsch {
65373df08285SMatthew G. Knepley   PetscMPIInt rank;
65389298eaa6SMatthew G Knepley 
65399298eaa6SMatthew G Knepley   PetscFunctionBegin;
654028b400f6SJacob 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.");
65419566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
65429566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
65439566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
65449566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(*dm, dim));
6545c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildFromCellList(*dm, numCells, numVertices, numCorners, cells));
65469566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildFromCellList(*dm, 0, 0, 0, NULL));
65479298eaa6SMatthew G Knepley   if (interpolate) {
65485fd9971aSMatthew G. Knepley     DM idm;
65499298eaa6SMatthew G Knepley 
65509566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
65519566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
65529298eaa6SMatthew G Knepley     *dm = idm;
65539298eaa6SMatthew G Knepley   }
6554c5853193SPierre Jolivet   if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, vertexCoords));
65559566063dSJacob Faibussowitsch   else PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, NULL));
65563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
65579298eaa6SMatthew G Knepley }
65589298eaa6SMatthew G Knepley 
6559939f6067SMatthew G. Knepley /*@
656020f4b53cSBarry Smith   DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a `DM`
6561939f6067SMatthew G. Knepley 
6562939f6067SMatthew G. Knepley   Input Parameters:
656320f4b53cSBarry Smith + dm               - The empty `DM` object, usually from `DMCreate()` and `DMSetDimension()`
6564939f6067SMatthew G. Knepley . depth            - The depth of the DAG
6565ce78bad3SBarry Smith . numPoints        - Array of size $ depth + 1 $ containing the number of points at each `depth`
6566939f6067SMatthew G. Knepley . coneSize         - The cone size of each point
6567939f6067SMatthew G. Knepley . cones            - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point
6568939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point
6569ce78bad3SBarry Smith - vertexCoords     - An array of $ numPoints[0] \times spacedim $ numbers representing the coordinates of each vertex, with `spacedim` the value set via `DMSetCoordinateDim()`
6570939f6067SMatthew G. Knepley 
6571939f6067SMatthew G. Knepley   Output Parameter:
657220f4b53cSBarry Smith . dm - The `DM`
657320f4b53cSBarry Smith 
657420f4b53cSBarry Smith   Level: advanced
6575939f6067SMatthew G. Knepley 
6576a1cb98faSBarry Smith   Note:
6577a1cb98faSBarry Smith   Two triangles sharing a face would have input
6578a1cb98faSBarry Smith .vb
6579a1cb98faSBarry Smith   depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0]
6580a1cb98faSBarry Smith   cones = [2 3 4  3 5 4], coneOrientations = [0 0 0  0 0 0]
6581a1cb98faSBarry Smith  vertexCoords = [-1.0 0.0  0.0 -1.0  0.0 1.0  1.0 0.0]
6582a1cb98faSBarry Smith .ve
6583939f6067SMatthew G. Knepley   which would result in the DMPlex
6584a1cb98faSBarry Smith .vb
6585a1cb98faSBarry Smith         4
6586a1cb98faSBarry Smith       / | \
6587a1cb98faSBarry Smith      /  |  \
6588a1cb98faSBarry Smith     /   |   \
6589a1cb98faSBarry Smith    2  0 | 1  5
6590a1cb98faSBarry Smith     \   |   /
6591a1cb98faSBarry Smith      \  |  /
6592a1cb98faSBarry Smith       \ | /
6593a1cb98faSBarry Smith         3
6594a1cb98faSBarry Smith .ve
6595a1cb98faSBarry Smith   Notice that all points are numbered consecutively, unlike `DMPlexCreateFromCellListPetsc()`
6596939f6067SMatthew G. Knepley 
6597ce78bad3SBarry Smith   Developer Note:
6598ce78bad3SBarry Smith   This does not create anything so should not have create in the name.
6599ce78bad3SBarry Smith 
66001cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
6601939f6067SMatthew G. Knepley @*/
6602d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
6603d71ae5a4SJacob Faibussowitsch {
66049298eaa6SMatthew G Knepley   Vec          coordinates;
66059298eaa6SMatthew G Knepley   PetscSection coordSection;
66069298eaa6SMatthew G Knepley   PetscScalar *coords;
6607811e8653SToby Isaac   PetscInt     coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off;
66089298eaa6SMatthew G Knepley 
66099298eaa6SMatthew G Knepley   PetscFunctionBegin;
66109566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
66119566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dimEmbed));
661263a3b9bcSJacob Faibussowitsch   PetscCheck(dimEmbed >= dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Embedding dimension %" PetscInt_FMT " cannot be less than intrinsic dimension %" PetscInt_FMT, dimEmbed, dim);
66139298eaa6SMatthew G Knepley   for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
66149566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(dm, pStart, pEnd));
66159298eaa6SMatthew G Knepley   for (p = pStart; p < pEnd; ++p) {
66169566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeSize(dm, p, coneSize[p - pStart]));
6617ad540459SPierre Jolivet     if (firstVertex < 0 && !coneSize[p - pStart]) firstVertex = p - pStart;
661897e052ccSToby Isaac   }
66191dca8a05SBarry Smith   PetscCheck(firstVertex >= 0 || !numPoints[0], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Expected %" PetscInt_FMT " vertices but could not find any", numPoints[0]);
66209566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm)); /* Allocate space for cones */
66219298eaa6SMatthew G Knepley   for (p = pStart, off = 0; p < pEnd; off += coneSize[p - pStart], ++p) {
66229566063dSJacob Faibussowitsch     PetscCall(DMPlexSetCone(dm, p, &cones[off]));
66239566063dSJacob Faibussowitsch     PetscCall(DMPlexSetConeOrientation(dm, p, &coneOrientations[off]));
66249298eaa6SMatthew G Knepley   }
66259566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(dm));
66269566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(dm));
66279298eaa6SMatthew G Knepley   /* Build coordinates */
66289566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm, &coordSection));
66299566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
66309566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed));
66319566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numPoints[0]));
66329298eaa6SMatthew G Knepley   for (v = firstVertex; v < firstVertex + numPoints[0]; ++v) {
66339566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed));
66349566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed));
66359298eaa6SMatthew G Knepley   }
66369566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
66379566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
66389566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
66399566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
66409566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
664158b7e2c1SStefano Zampini   PetscCall(VecSetBlockSize(coordinates, PetscMax(dimEmbed, 1)));
66429566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
66439318fe57SMatthew G. Knepley   if (vertexCoords) {
66449566063dSJacob Faibussowitsch     PetscCall(VecGetArray(coordinates, &coords));
66459298eaa6SMatthew G Knepley     for (v = 0; v < numPoints[0]; ++v) {
66469298eaa6SMatthew G Knepley       PetscInt off;
66479298eaa6SMatthew G Knepley 
66489566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSection, v + firstVertex, &off));
6649ad540459SPierre Jolivet       for (d = 0; d < dimEmbed; ++d) coords[off + d] = vertexCoords[v * dimEmbed + d];
66509298eaa6SMatthew G Knepley     }
66519318fe57SMatthew G. Knepley   }
66529566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
66539566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(dm, coordinates));
66549566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
66553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
66569298eaa6SMatthew G Knepley }
66578415267dSToby Isaac 
6658a4e35b19SJacob Faibussowitsch /*
6659a1cb98faSBarry Smith   DMPlexCreateCellVertexFromFile - Create a `DMPLEX` mesh from a simple cell-vertex file.
6660a1cb98faSBarry Smith 
6661a1cb98faSBarry Smith   Collective
66628ca92349SMatthew G. Knepley 
66638ca92349SMatthew G. Knepley + comm        - The MPI communicator
66648ca92349SMatthew G. Knepley . filename    - Name of the .dat file
66658ca92349SMatthew G. Knepley - interpolate - Create faces and edges in the mesh
66668ca92349SMatthew G. Knepley 
66678ca92349SMatthew G. Knepley   Output Parameter:
6668a1cb98faSBarry Smith . dm  - The `DM` object representing the mesh
66698ca92349SMatthew G. Knepley 
66708ca92349SMatthew G. Knepley   Level: beginner
66718ca92349SMatthew G. Knepley 
6672a1cb98faSBarry Smith   Note:
6673a1cb98faSBarry Smith   The format is the simplest possible:
6674a1cb98faSBarry Smith .vb
6675d0812dedSMatthew G. Knepley   dim Ne Nv Nc Nl
6676d0812dedSMatthew G. Knepley   v_1 v_2 ... v_Nc
6677d0812dedSMatthew G. Knepley   ...
6678d0812dedSMatthew G. Knepley   x y z marker_1 ... marker_Nl
6679a1cb98faSBarry Smith .ve
6680a1cb98faSBarry Smith 
6681a1cb98faSBarry Smith   Developer Note:
6682a1cb98faSBarry Smith   Should use a `PetscViewer` not a filename
6683a1cb98faSBarry Smith 
66846afe31f6SMartin Diehl .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateGmsh()`, `DMPlexCreate()`
6685a4e35b19SJacob Faibussowitsch */
6686ff6a9541SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCellVertexFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
6687d71ae5a4SJacob Faibussowitsch {
66888ca92349SMatthew G. Knepley   DMLabel      marker;
66898ca92349SMatthew G. Knepley   PetscViewer  viewer;
66908ca92349SMatthew G. Knepley   Vec          coordinates;
66918ca92349SMatthew G. Knepley   PetscSection coordSection;
66928ca92349SMatthew G. Knepley   PetscScalar *coords;
66938ca92349SMatthew G. Knepley   char         line[PETSC_MAX_PATH_LEN];
6694d0812dedSMatthew G. Knepley   PetscInt     cdim, coordSize, v, c, d;
66958ca92349SMatthew G. Knepley   PetscMPIInt  rank;
6696d0812dedSMatthew G. Knepley   int          snum, dim, Nv, Nc, Ncn, Nl;
66978ca92349SMatthew G. Knepley 
66988ca92349SMatthew G. Knepley   PetscFunctionBegin;
66999566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
67009566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(comm, &viewer));
67019566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII));
67029566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
67039566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileSetName(viewer, filename));
6704dd400576SPatrick Sanan   if (rank == 0) {
6705d0812dedSMatthew G. Knepley     PetscCall(PetscViewerRead(viewer, line, 5, NULL, PETSC_STRING));
6706d0812dedSMatthew G. Knepley     snum = sscanf(line, "%d %d %d %d %d", &dim, &Nc, &Nv, &Ncn, &Nl);
6707d0812dedSMatthew G. Knepley     PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
670825ce1634SJed Brown   } else {
6709f8d5e320SMatthew G. Knepley     Nc = Nv = Ncn = Nl = 0;
67108ca92349SMatthew G. Knepley   }
6711d0812dedSMatthew G. Knepley   PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm));
6712835f2295SStefano Zampini   cdim = dim;
67139566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, dm));
67149566063dSJacob Faibussowitsch   PetscCall(DMSetType(*dm, DMPLEX));
67159566063dSJacob Faibussowitsch   PetscCall(DMPlexSetChart(*dm, 0, Nc + Nv));
6716835f2295SStefano Zampini   PetscCall(DMSetDimension(*dm, dim));
67179566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinateDim(*dm, cdim));
67188ca92349SMatthew G. Knepley   /* Read topology */
6719dd400576SPatrick Sanan   if (rank == 0) {
6720f8d5e320SMatthew G. Knepley     char     format[PETSC_MAX_PATH_LEN];
6721f8d5e320SMatthew G. Knepley     PetscInt cone[8];
67228ca92349SMatthew G. Knepley     int      vbuf[8], v;
67238ca92349SMatthew G. Knepley 
67249371c9d4SSatish Balay     for (c = 0; c < Ncn; ++c) {
67259371c9d4SSatish Balay       format[c * 3 + 0] = '%';
67269371c9d4SSatish Balay       format[c * 3 + 1] = 'd';
67279371c9d4SSatish Balay       format[c * 3 + 2] = ' ';
67289371c9d4SSatish Balay     }
6729f8d5e320SMatthew G. Knepley     format[Ncn * 3 - 1] = '\0';
67309566063dSJacob Faibussowitsch     for (c = 0; c < Nc; ++c) PetscCall(DMPlexSetConeSize(*dm, c, Ncn));
67319566063dSJacob Faibussowitsch     PetscCall(DMSetUp(*dm));
67328ca92349SMatthew G. Knepley     for (c = 0; c < Nc; ++c) {
67339566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, Ncn, NULL, PETSC_STRING));
6734f8d5e320SMatthew G. Knepley       switch (Ncn) {
6735d71ae5a4SJacob Faibussowitsch       case 2:
6736d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1]);
6737d71ae5a4SJacob Faibussowitsch         break;
6738d71ae5a4SJacob Faibussowitsch       case 3:
6739d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2]);
6740d71ae5a4SJacob Faibussowitsch         break;
6741d71ae5a4SJacob Faibussowitsch       case 4:
6742d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3]);
6743d71ae5a4SJacob Faibussowitsch         break;
6744d71ae5a4SJacob Faibussowitsch       case 6:
6745d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5]);
6746d71ae5a4SJacob Faibussowitsch         break;
6747d71ae5a4SJacob Faibussowitsch       case 8:
6748d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5], &vbuf[6], &vbuf[7]);
6749d71ae5a4SJacob Faibussowitsch         break;
6750d71ae5a4SJacob Faibussowitsch       default:
6751d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No cell shape with %d vertices", Ncn);
6752f8d5e320SMatthew G. Knepley       }
675308401ef6SPierre Jolivet       PetscCheck(snum == Ncn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6754f8d5e320SMatthew G. Knepley       for (v = 0; v < Ncn; ++v) cone[v] = vbuf[v] + Nc;
67558ca92349SMatthew G. Knepley       /* Hexahedra are inverted */
6756f8d5e320SMatthew G. Knepley       if (Ncn == 8) {
67578ca92349SMatthew G. Knepley         PetscInt tmp = cone[1];
67588ca92349SMatthew G. Knepley         cone[1]      = cone[3];
67598ca92349SMatthew G. Knepley         cone[3]      = tmp;
67608ca92349SMatthew G. Knepley       }
67619566063dSJacob Faibussowitsch       PetscCall(DMPlexSetCone(*dm, c, cone));
67628ca92349SMatthew G. Knepley     }
67638ca92349SMatthew G. Knepley   }
67649566063dSJacob Faibussowitsch   PetscCall(DMPlexSymmetrize(*dm));
67659566063dSJacob Faibussowitsch   PetscCall(DMPlexStratify(*dm));
67668ca92349SMatthew G. Knepley   /* Read coordinates */
67679566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(*dm, &coordSection));
67689566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(coordSection, 1));
67699566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim));
67709566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(coordSection, Nc, Nc + Nv));
67718ca92349SMatthew G. Knepley   for (v = Nc; v < Nc + Nv; ++v) {
67729566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(coordSection, v, cdim));
67739566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim));
67748ca92349SMatthew G. Knepley   }
67759566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(coordSection));
67769566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
67779566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
67789566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
67799566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
67809566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(coordinates, cdim));
67819566063dSJacob Faibussowitsch   PetscCall(VecSetType(coordinates, VECSTANDARD));
67829566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordinates, &coords));
6783dd400576SPatrick Sanan   if (rank == 0) {
6784f8d5e320SMatthew G. Knepley     char   format[PETSC_MAX_PATH_LEN];
67858ca92349SMatthew G. Knepley     double x[3];
6786f8d5e320SMatthew G. Knepley     int    l, val[3];
67878ca92349SMatthew G. Knepley 
6788f8d5e320SMatthew G. Knepley     if (Nl) {
67899371c9d4SSatish Balay       for (l = 0; l < Nl; ++l) {
67909371c9d4SSatish Balay         format[l * 3 + 0] = '%';
67919371c9d4SSatish Balay         format[l * 3 + 1] = 'd';
67929371c9d4SSatish Balay         format[l * 3 + 2] = ' ';
67939371c9d4SSatish Balay       }
6794f8d5e320SMatthew G. Knepley       format[Nl * 3 - 1] = '\0';
67959566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
67969566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &marker));
6797f8d5e320SMatthew G. Knepley     }
67988ca92349SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
67999566063dSJacob Faibussowitsch       PetscCall(PetscViewerRead(viewer, line, 3 + Nl, NULL, PETSC_STRING));
6800f8d5e320SMatthew G. Knepley       snum = sscanf(line, "%lg %lg %lg", &x[0], &x[1], &x[2]);
680108401ef6SPierre Jolivet       PetscCheck(snum == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
6802f8d5e320SMatthew G. Knepley       switch (Nl) {
6803d71ae5a4SJacob Faibussowitsch       case 0:
6804d71ae5a4SJacob Faibussowitsch         snum = 0;
6805d71ae5a4SJacob Faibussowitsch         break;
6806d71ae5a4SJacob Faibussowitsch       case 1:
6807d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0]);
6808d71ae5a4SJacob Faibussowitsch         break;
6809d71ae5a4SJacob Faibussowitsch       case 2:
6810d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1]);
6811d71ae5a4SJacob Faibussowitsch         break;
6812d71ae5a4SJacob Faibussowitsch       case 3:
6813d71ae5a4SJacob Faibussowitsch         snum = sscanf(line, format, &val[0], &val[1], &val[2]);
6814d71ae5a4SJacob Faibussowitsch         break;
6815d71ae5a4SJacob Faibussowitsch       default:
6816d71ae5a4SJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Request support for %d labels", Nl);
6817f8d5e320SMatthew G. Knepley       }
681808401ef6SPierre Jolivet       PetscCheck(snum == Nl, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line);
68198ca92349SMatthew G. Knepley       for (d = 0; d < cdim; ++d) coords[v * cdim + d] = x[d];
68209566063dSJacob Faibussowitsch       for (l = 0; l < Nl; ++l) PetscCall(DMLabelSetValue(marker, v + Nc, val[l]));
68218ca92349SMatthew G. Knepley     }
68228ca92349SMatthew G. Knepley   }
68239566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordinates, &coords));
68249566063dSJacob Faibussowitsch   PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
68259566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&coordinates));
68269566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&viewer));
68278ca92349SMatthew G. Knepley   if (interpolate) {
68288ca92349SMatthew G. Knepley     DM      idm;
68298ca92349SMatthew G. Knepley     DMLabel bdlabel;
68308ca92349SMatthew G. Knepley 
68319566063dSJacob Faibussowitsch     PetscCall(DMPlexInterpolate(*dm, &idm));
68329566063dSJacob Faibussowitsch     PetscCall(DMDestroy(dm));
68338ca92349SMatthew G. Knepley     *dm = idm;
68348ca92349SMatthew G. Knepley 
6835f8d5e320SMatthew G. Knepley     if (!Nl) {
68369566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(*dm, "marker"));
68379566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(*dm, "marker", &bdlabel));
68389566063dSJacob Faibussowitsch       PetscCall(DMPlexMarkBoundaryFaces(*dm, PETSC_DETERMINE, bdlabel));
68399566063dSJacob Faibussowitsch       PetscCall(DMPlexLabelComplete(*dm, bdlabel));
68408ca92349SMatthew G. Knepley     }
6841f8d5e320SMatthew G. Knepley   }
68423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
68438ca92349SMatthew G. Knepley }
68448ca92349SMatthew G. Knepley 
6845cc4c1da9SBarry Smith /*@
6846a1cb98faSBarry Smith   DMPlexCreateFromFile - This takes a filename and produces a `DM`
6847a1cb98faSBarry Smith 
6848a1cb98faSBarry Smith   Collective
6849ca522641SMatthew G. Knepley 
6850ca522641SMatthew G. Knepley   Input Parameters:
6851ca522641SMatthew G. Knepley + comm        - The communicator
6852ca522641SMatthew G. Knepley . filename    - A file name
6853a1cb98faSBarry Smith . plexname    - The object name of the resulting `DM`, also used for intra-datafile lookup by some formats
6854ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
6855ca522641SMatthew G. Knepley 
6856ca522641SMatthew G. Knepley   Output Parameter:
6857a1cb98faSBarry Smith . dm - The `DM`
6858ca522641SMatthew G. Knepley 
6859a1cb98faSBarry Smith   Options Database Key:
6860a1cb98faSBarry Smith . -dm_plex_create_from_hdf5_xdmf - use the `PETSC_VIEWER_HDF5_XDMF` format for reading HDF5
686102ef0d99SVaclav Hapla 
6862b44f4de4SBarry Smith   Use `-dm_plex_create_ prefix` to pass options to the internal `PetscViewer`, e.g. `-dm_plex_create_viewer_hdf5_collective`
6863bca97951SVaclav Hapla 
6864ca522641SMatthew G. Knepley   Level: beginner
6865ca522641SMatthew G. Knepley 
6866a1cb98faSBarry Smith   Notes:
6867a1cb98faSBarry Smith   Using `PETSCVIEWERHDF5` type with `PETSC_VIEWER_HDF5_PETSC` format, one can save multiple `DMPLEX`
6868a1cb98faSBarry Smith   meshes in a single HDF5 file. This in turn requires one to name the `DMPLEX` object with `PetscObjectSetName()`
6869a1cb98faSBarry Smith   before saving it with `DMView()` and before loading it with `DMLoad()` for identification of the mesh object.
6870a1cb98faSBarry Smith   The input parameter name is thus used to name the `DMPLEX` object when `DMPlexCreateFromFile()` internally
6871a1cb98faSBarry Smith   calls `DMLoad()`. Currently, name is ignored for other viewer types and/or formats.
6872a1cb98faSBarry Smith 
68731cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`, `PetscObjectSetName()`, `DMView()`, `DMLoad()`
6874ca522641SMatthew G. Knepley @*/
6875d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], const char plexname[], PetscBool interpolate, DM *dm)
6876d71ae5a4SJacob Faibussowitsch {
6877ef3a5affSJacob Faibussowitsch   const char  extGmsh[]      = ".msh";
6878ef3a5affSJacob Faibussowitsch   const char  extGmsh2[]     = ".msh2";
6879ef3a5affSJacob Faibussowitsch   const char  extGmsh4[]     = ".msh4";
6880ef3a5affSJacob Faibussowitsch   const char  extCGNS[]      = ".cgns";
6881ef3a5affSJacob Faibussowitsch   const char  extExodus[]    = ".exo";
6882ef3a5affSJacob Faibussowitsch   const char  extExodus_e[]  = ".e";
6883ef3a5affSJacob Faibussowitsch   const char  extGenesis[]   = ".gen";
6884ef3a5affSJacob Faibussowitsch   const char  extFluent[]    = ".cas";
6885ef3a5affSJacob Faibussowitsch   const char  extHDF5[]      = ".h5";
68866f2c871aSStefano Zampini   const char  extXDMFHDF5[]  = ".xdmf.h5";
6887ef3a5affSJacob Faibussowitsch   const char  extPLY[]       = ".ply";
68885552b385SBrandon   const char  extEGADSlite[] = ".egadslite";
6889ef3a5affSJacob Faibussowitsch   const char  extEGADS[]     = ".egads";
6890ef3a5affSJacob Faibussowitsch   const char  extIGES[]      = ".igs";
68915552b385SBrandon   const char  extIGES2[]     = ".iges";
6892ef3a5affSJacob Faibussowitsch   const char  extSTEP[]      = ".stp";
68935552b385SBrandon   const char  extSTEP2[]     = ".step";
68945552b385SBrandon   const char  extBREP[]      = ".brep";
6895ef3a5affSJacob Faibussowitsch   const char  extCV[]        = ".dat";
6896ca522641SMatthew G. Knepley   size_t      len;
68975552b385SBrandon   PetscBool   isGmsh, isGmsh2, isGmsh4, isCGNS, isExodus, isGenesis, isFluent, isHDF5, isPLY, isEGADSlite, isEGADS, isIGES, isIGES2, isSTEP, isSTEP2, isBREP, isCV, isXDMFHDF5;
6898ca522641SMatthew G. Knepley   PetscMPIInt rank;
6899ca522641SMatthew G. Knepley 
6900ca522641SMatthew G. Knepley   PetscFunctionBegin;
69014f572ea9SToby Isaac   PetscAssertPointer(filename, 2);
69024f572ea9SToby Isaac   if (plexname) PetscAssertPointer(plexname, 3);
69034f572ea9SToby Isaac   PetscAssertPointer(dm, 5);
69049566063dSJacob Faibussowitsch   PetscCall(DMInitializePackage());
69059566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(DMPLEX_CreateFromFile, 0, 0, 0, 0));
69069566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
69079566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(filename, &len));
690828b400f6SJacob Faibussowitsch   PetscCheck(len, comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path");
6909ef3a5affSJacob Faibussowitsch 
69109371c9d4SSatish Balay #define CheckExtension(extension__, is_extension__) \
69119371c9d4SSatish Balay   do { \
6912274aaeaaSJacob Faibussowitsch     PetscAssert(sizeof(extension__), comm, PETSC_ERR_PLIB, "Zero-size extension: %s", extension__); \
6913274aaeaaSJacob Faibussowitsch     /* don't count the null-terminator at the end */ \
6914274aaeaaSJacob Faibussowitsch     const size_t ext_len = sizeof(extension__) - 1; \
6915274aaeaaSJacob Faibussowitsch     if (len < ext_len) { \
6916ef3a5affSJacob Faibussowitsch       is_extension__ = PETSC_FALSE; \
6917ef3a5affSJacob Faibussowitsch     } else { \
6918274aaeaaSJacob Faibussowitsch       PetscCall(PetscStrncmp(filename + len - ext_len, extension__, ext_len, &is_extension__)); \
6919ef3a5affSJacob Faibussowitsch     } \
6920ef3a5affSJacob Faibussowitsch   } while (0)
6921ef3a5affSJacob Faibussowitsch 
6922ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh, isGmsh);
6923ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh2, isGmsh2);
6924ef3a5affSJacob Faibussowitsch   CheckExtension(extGmsh4, isGmsh4);
6925ef3a5affSJacob Faibussowitsch   CheckExtension(extCGNS, isCGNS);
6926ef3a5affSJacob Faibussowitsch   CheckExtension(extExodus, isExodus);
6927ef3a5affSJacob Faibussowitsch   if (!isExodus) CheckExtension(extExodus_e, isExodus);
6928ef3a5affSJacob Faibussowitsch   CheckExtension(extGenesis, isGenesis);
6929ef3a5affSJacob Faibussowitsch   CheckExtension(extFluent, isFluent);
6930ef3a5affSJacob Faibussowitsch   CheckExtension(extHDF5, isHDF5);
6931ef3a5affSJacob Faibussowitsch   CheckExtension(extPLY, isPLY);
69325552b385SBrandon   CheckExtension(extEGADSlite, isEGADSlite);
6933ef3a5affSJacob Faibussowitsch   CheckExtension(extEGADS, isEGADS);
6934ef3a5affSJacob Faibussowitsch   CheckExtension(extIGES, isIGES);
69355552b385SBrandon   CheckExtension(extIGES2, isIGES2);
6936ef3a5affSJacob Faibussowitsch   CheckExtension(extSTEP, isSTEP);
69375552b385SBrandon   CheckExtension(extSTEP2, isSTEP2);
69385552b385SBrandon   CheckExtension(extBREP, isBREP);
6939ef3a5affSJacob Faibussowitsch   CheckExtension(extCV, isCV);
69406f2c871aSStefano Zampini   CheckExtension(extXDMFHDF5, isXDMFHDF5);
6941ef3a5affSJacob Faibussowitsch 
6942ef3a5affSJacob Faibussowitsch #undef CheckExtension
6943ef3a5affSJacob Faibussowitsch 
6944de78e4feSLisandro Dalcin   if (isGmsh || isGmsh2 || isGmsh4) {
69459566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateGmshFromFile(comm, filename, interpolate, dm));
6946ca522641SMatthew G. Knepley   } else if (isCGNS) {
69479566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm));
694890c68965SMatthew G. Knepley   } else if (isExodus || isGenesis) {
69499566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateExodusFromFile(comm, filename, interpolate, dm));
69502f0bd6dcSMichael Lange   } else if (isFluent) {
69519566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFluentFromFile(comm, filename, interpolate, dm));
6952cc2f8f65SMatthew G. Knepley   } else if (isHDF5) {
6953cc2f8f65SMatthew G. Knepley     PetscViewer viewer;
6954cc2f8f65SMatthew G. Knepley 
695543b242b4SVaclav 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 */
69566f2c871aSStefano Zampini     PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_create_from_hdf5_xdmf", &isXDMFHDF5, NULL));
69579566063dSJacob Faibussowitsch     PetscCall(PetscViewerCreate(comm, &viewer));
69589566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetType(viewer, PETSCVIEWERHDF5));
69599566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetOptionsPrefix(viewer, "dm_plex_create_"));
69609566063dSJacob Faibussowitsch     PetscCall(PetscViewerSetFromOptions(viewer));
69619566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
69629566063dSJacob Faibussowitsch     PetscCall(PetscViewerFileSetName(viewer, filename));
6963cd7e8a5eSksagiyam 
69649566063dSJacob Faibussowitsch     PetscCall(DMCreate(comm, dm));
6965f4f49eeaSPierre Jolivet     PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
69669566063dSJacob Faibussowitsch     PetscCall(DMSetType(*dm, DMPLEX));
69676f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPushFormat(viewer, PETSC_VIEWER_HDF5_XDMF));
69689566063dSJacob Faibussowitsch     PetscCall(DMLoad(*dm, viewer));
69696f2c871aSStefano Zampini     if (isXDMFHDF5) PetscCall(PetscViewerPopFormat(viewer));
69709566063dSJacob Faibussowitsch     PetscCall(PetscViewerDestroy(&viewer));
69715fd9971aSMatthew G. Knepley 
69725fd9971aSMatthew G. Knepley     if (interpolate) {
69735fd9971aSMatthew G. Knepley       DM idm;
69745fd9971aSMatthew G. Knepley 
69759566063dSJacob Faibussowitsch       PetscCall(DMPlexInterpolate(*dm, &idm));
69769566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
69775fd9971aSMatthew G. Knepley       *dm = idm;
69785fd9971aSMatthew G. Knepley     }
6979f2801cd6SMatthew G. Knepley   } else if (isPLY) {
69809566063dSJacob Faibussowitsch     PetscCall(DMPlexCreatePLYFromFile(comm, filename, interpolate, dm));
69815552b385SBrandon   } else if (isEGADSlite || isEGADS || isIGES || isIGES2 || isSTEP || isSTEP2 || isBREP) {
69825552b385SBrandon     PetscCall(DMPlexCreateGeomFromFile(comm, filename, dm, isEGADSlite));
69835552b385SBrandon 
69847bee2925SMatthew Knepley     if (!interpolate) {
69857bee2925SMatthew Knepley       DM udm;
69867bee2925SMatthew Knepley 
69879566063dSJacob Faibussowitsch       PetscCall(DMPlexUninterpolate(*dm, &udm));
69889566063dSJacob Faibussowitsch       PetscCall(DMDestroy(dm));
69897bee2925SMatthew Knepley       *dm = udm;
69907bee2925SMatthew Knepley     }
69918ca92349SMatthew G. Knepley   } else if (isCV) {
69929566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCellVertexFromFile(comm, filename, interpolate, dm));
699398921bdaSJacob Faibussowitsch   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename);
69949566063dSJacob Faibussowitsch   PetscCall(PetscStrlen(plexname, &len));
6995f4f49eeaSPierre Jolivet   if (len) PetscCall(PetscObjectSetName((PetscObject)*dm, plexname));
69969566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(DMPLEX_CreateFromFile, 0, 0, 0, 0));
69973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6998ca522641SMatthew G. Knepley }
699912b8a6daSStefano Zampini 
7000cc4c1da9SBarry Smith /*@
70019f6c5813SMatthew 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.
70029f6c5813SMatthew G. Knepley 
70030528010dSStefano Zampini   Input Parameters:
70040528010dSStefano Zampini + tr     - The `DMPlexTransform`
70050528010dSStefano Zampini - prefix - An options prefix, or NULL
70069f6c5813SMatthew G. Knepley 
70079f6c5813SMatthew G. Knepley   Output Parameter:
70089f6c5813SMatthew G. Knepley . dm - The `DM`
70099f6c5813SMatthew G. Knepley 
70109f6c5813SMatthew G. Knepley   Level: beginner
70119f6c5813SMatthew G. Knepley 
701220f4b53cSBarry Smith   Notes:
701320f4b53cSBarry 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.
701420f4b53cSBarry Smith 
70159f6c5813SMatthew G. Knepley .seealso: `DMPlexCreateFromFile`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`
70169f6c5813SMatthew G. Knepley @*/
70170528010dSStefano Zampini PetscErrorCode DMPlexCreateEphemeral(DMPlexTransform tr, const char prefix[], DM *dm)
70189f6c5813SMatthew G. Knepley {
70190528010dSStefano Zampini   DM           bdm, bcdm, cdm;
70200528010dSStefano Zampini   Vec          coordinates, coordinatesNew;
70210528010dSStefano Zampini   PetscSection cs;
7022817b2c36SMatthew G. Knepley   PetscInt     cdim, Nl;
70239f6c5813SMatthew G. Knepley 
70249f6c5813SMatthew G. Knepley   PetscFunctionBegin;
70259f6c5813SMatthew G. Knepley   PetscCall(DMCreate(PetscObjectComm((PetscObject)tr), dm));
70269f6c5813SMatthew G. Knepley   PetscCall(DMSetType(*dm, DMPLEX));
70270528010dSStefano Zampini   ((DM_Plex *)(*dm)->data)->interpolated = DMPLEX_INTERPOLATED_FULL;
70280528010dSStefano Zampini   // Handle coordinates
70290528010dSStefano Zampini   PetscCall(DMPlexTransformGetDM(tr, &bdm));
7030817b2c36SMatthew G. Knepley   PetscCall(DMPlexTransformSetDimensions(tr, bdm, *dm));
7031817b2c36SMatthew G. Knepley   PetscCall(DMGetCoordinateDim(*dm, &cdim));
70320528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(bdm, &bcdm));
70330528010dSStefano Zampini   PetscCall(DMGetCoordinateDM(*dm, &cdm));
70340528010dSStefano Zampini   PetscCall(DMCopyDisc(bcdm, cdm));
70350528010dSStefano Zampini   PetscCall(DMGetLocalSection(cdm, &cs));
70360528010dSStefano Zampini   PetscCall(PetscSectionSetNumFields(cs, 1));
70370528010dSStefano Zampini   PetscCall(PetscSectionSetFieldComponents(cs, 0, cdim));
70380528010dSStefano Zampini   PetscCall(DMGetCoordinatesLocal(bdm, &coordinates));
70390528010dSStefano Zampini   PetscCall(VecDuplicate(coordinates, &coordinatesNew));
70400528010dSStefano Zampini   PetscCall(VecCopy(coordinates, coordinatesNew));
70410528010dSStefano Zampini   PetscCall(DMSetCoordinatesLocal(*dm, coordinatesNew));
70420528010dSStefano Zampini   PetscCall(VecDestroy(&coordinatesNew));
70439f6c5813SMatthew G. Knepley 
70449f6c5813SMatthew G. Knepley   PetscCall(PetscObjectReference((PetscObject)tr));
70459f6c5813SMatthew G. Knepley   PetscCall(DMPlexTransformDestroy(&((DM_Plex *)(*dm)->data)->tr));
70469f6c5813SMatthew G. Knepley   ((DM_Plex *)(*dm)->data)->tr = tr;
70470528010dSStefano Zampini   PetscCall(DMPlexDistributeSetDefault(*dm, PETSC_FALSE));
70480528010dSStefano Zampini   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*dm, prefix));
70490528010dSStefano Zampini   PetscCall(DMSetFromOptions(*dm));
70509f6c5813SMatthew G. Knepley 
70519f6c5813SMatthew G. Knepley   PetscCall(DMGetNumLabels(bdm, &Nl));
70529f6c5813SMatthew G. Knepley   for (PetscInt l = 0; l < Nl; ++l) {
70539f6c5813SMatthew G. Knepley     DMLabel     label, labelNew;
70549f6c5813SMatthew G. Knepley     const char *lname;
70559f6c5813SMatthew G. Knepley     PetscBool   isDepth, isCellType;
70569f6c5813SMatthew G. Knepley 
70579f6c5813SMatthew G. Knepley     PetscCall(DMGetLabelName(bdm, l, &lname));
70589f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "depth", &isDepth));
70599f6c5813SMatthew G. Knepley     if (isDepth) continue;
70609f6c5813SMatthew G. Knepley     PetscCall(PetscStrcmp(lname, "celltype", &isCellType));
70619f6c5813SMatthew G. Knepley     if (isCellType) continue;
70629f6c5813SMatthew G. Knepley     PetscCall(DMCreateLabel(*dm, lname));
70639f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(bdm, lname, &label));
70649f6c5813SMatthew G. Knepley     PetscCall(DMGetLabel(*dm, lname, &labelNew));
70659f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetType(labelNew, DMLABELEPHEMERAL));
70669f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetLabel(labelNew, label));
70679f6c5813SMatthew G. Knepley     PetscCall(DMLabelEphemeralSetTransform(labelNew, tr));
70689f6c5813SMatthew G. Knepley     PetscCall(DMLabelSetUp(labelNew));
70699f6c5813SMatthew G. Knepley   }
70703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
70719f6c5813SMatthew G. Knepley }
7072