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; 127*509b31aaSMatthew 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; 161*509b31aaSMatthew G. Knepley PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc)); 162*509b31aaSMatthew 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)); 305*509b31aaSMatthew 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), ¢roidsf)); 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(¢roidsf)); 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 3647*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, snapToSphere)); 3648*509b31aaSMatthew 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; 4291b7f5c055SJed Brown DM dmf; 4292b7f5c055SJed Brown Vec X; 4293b7f5c055SJed Brown PetscScalar *x; 42949566063dSJacob Faibussowitsch PetscCall(DMRefine(dm, MPI_COMM_NULL, &dmf)); 429569d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &dmf)); 4296b7f5c055SJed Brown 42979566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &X)); 42989566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(X, &m)); 42999566063dSJacob Faibussowitsch PetscCall(VecGetArray(X, &x)); 430048a46eb9SPierre Jolivet for (PetscInt i = 0; i < m; i += 3) PetscCall(TPSNearestPoint(evalFunc, &x[i])); 43019566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(X, &x)); 4302b7f5c055SJed Brown } 4303b7f5c055SJed Brown 4304b7f5c055SJed Brown // Face Sets has already been propagated to new vertices during refinement; this propagates to the initial vertices. 43059566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "Face Sets", &label)); 43069566063dSJacob Faibussowitsch PetscCall(DMPlexLabelComplete(dm, label)); 4307b7f5c055SJed Brown 430846139095SJed Brown PetscCall(PetscLogEventEnd(DMPLEX_Generate, dm, 0, 0, 0)); 430946139095SJed Brown 4310b7f5c055SJed Brown if (thickness > 0) { 43114663dae6SJed Brown DM edm, cdm, ecdm; 43124663dae6SJed Brown DMPlexTransform tr; 43134663dae6SJed Brown const char *prefix; 43144663dae6SJed Brown PetscOptions options; 43154663dae6SJed Brown // Code from DMPlexExtrude 43164663dae6SJed Brown PetscCall(DMPlexTransformCreate(PetscObjectComm((PetscObject)dm), &tr)); 43174663dae6SJed Brown PetscCall(DMPlexTransformSetDM(tr, dm)); 4318ce78bad3SBarry Smith PetscCall(DMPlexTransformSetType(tr, DMPLEXEXTRUDETYPE)); 43194663dae6SJed Brown PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix)); 43204663dae6SJed Brown PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tr, prefix)); 43214663dae6SJed Brown PetscCall(PetscObjectGetOptions((PetscObject)dm, &options)); 43224663dae6SJed Brown PetscCall(PetscObjectSetOptions((PetscObject)tr, options)); 43234663dae6SJed Brown PetscCall(DMPlexTransformExtrudeSetLayers(tr, layers)); 43244663dae6SJed Brown PetscCall(DMPlexTransformExtrudeSetThickness(tr, thickness)); 43254663dae6SJed Brown PetscCall(DMPlexTransformExtrudeSetTensor(tr, PETSC_FALSE)); 43264663dae6SJed Brown PetscCall(DMPlexTransformExtrudeSetSymmetric(tr, PETSC_TRUE)); 43274663dae6SJed Brown PetscCall(DMPlexTransformExtrudeSetNormalFunction(tr, normalFunc)); 43284663dae6SJed Brown PetscCall(DMPlexTransformSetFromOptions(tr)); 43294663dae6SJed Brown PetscCall(PetscObjectSetOptions((PetscObject)tr, NULL)); 43304663dae6SJed Brown PetscCall(DMPlexTransformSetUp(tr)); 43314663dae6SJed Brown PetscCall(PetscObjectViewFromOptions((PetscObject)tr, NULL, "-dm_plex_tps_transform_view")); 43324663dae6SJed Brown PetscCall(DMPlexTransformApply(tr, dm, &edm)); 43334663dae6SJed Brown PetscCall(DMCopyDisc(dm, edm)); 43344663dae6SJed Brown PetscCall(DMGetCoordinateDM(dm, &cdm)); 43354663dae6SJed Brown PetscCall(DMGetCoordinateDM(edm, &ecdm)); 43364663dae6SJed Brown PetscCall(DMCopyDisc(cdm, ecdm)); 43374663dae6SJed Brown PetscCall(DMPlexTransformCreateDiscLabels(tr, edm)); 43384663dae6SJed Brown PetscCall(DMPlexTransformDestroy(&tr)); 4339a77a5016SMatthew G. Knepley PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, edm)); 434069d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &edm)); 4341b7f5c055SJed Brown } 43423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4343b7f5c055SJed Brown } 4344b7f5c055SJed Brown 4345b7f5c055SJed Brown /*@ 4346b7f5c055SJed Brown DMPlexCreateTPSMesh - Create a distributed, interpolated mesh of a triply-periodic surface 4347b7f5c055SJed Brown 4348b7f5c055SJed Brown Collective 4349b7f5c055SJed Brown 4350b7f5c055SJed Brown Input Parameters: 4351a1cb98faSBarry Smith + comm - The communicator for the `DM` object 4352b7f5c055SJed Brown . tpstype - Type of triply-periodic surface 4353b7f5c055SJed Brown . extent - Array of length 3 containing number of periods in each direction 435420f4b53cSBarry Smith . periodic - array of length 3 with periodicity, or `NULL` for non-periodic 43551436d7faSJed Brown . tps_distribute - Distribute 2D manifold mesh prior to refinement and extrusion (more scalable) 4356817da375SSatish Balay . refinements - Number of factor-of-2 refinements of 2D manifold mesh 43571436d7faSJed Brown . layers - Number of cell layers extruded in normal direction 4358817da375SSatish Balay - thickness - Thickness in normal direction 4359b7f5c055SJed Brown 4360b7f5c055SJed Brown Output Parameter: 4361a1cb98faSBarry Smith . dm - The `DM` object 4362a1cb98faSBarry Smith 4363a1cb98faSBarry Smith Level: beginner 4364b7f5c055SJed Brown 4365b7f5c055SJed Brown Notes: 436615229ffcSPierre Jolivet This meshes the surface of the Schwarz P or Gyroid surfaces. Schwarz P is the simplest member of the triply-periodic minimal surfaces. 43671d27aa22SBarry Smith <https://en.wikipedia.org/wiki/Schwarz_minimal_surface#Schwarz_P_(%22Primitive%22)> and can be cut with "clean" boundaries. 43681d27aa22SBarry 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. 4369b7f5c055SJed Brown Our implementation creates a very coarse mesh of the surface and refines (by 4-way splitting) as many times as requested. 4370b7f5c055SJed Brown On each refinement, all vertices are projected to their nearest point on the surface. 4371b7f5c055SJed Brown This projection could readily be extended to related surfaces. 4372b7f5c055SJed Brown 43731d27aa22SBarry Smith See {cite}`maskery2018insights` 43741d27aa22SBarry Smith 43751d27aa22SBarry Smith The face (edge) sets for the Schwarz P surface are numbered $1(-x), 2(+x), 3(-y), 4(+y), 5(-z), 6(+z)$. 43761d27aa22SBarry Smith When the mesh is refined, "Face Sets" contain the new vertices (created during refinement). 43771d27aa22SBarry Smith Use `DMPlexLabelComplete()` to propagate to coarse-level vertices. 4378b7f5c055SJed Brown 437960225df5SJacob Faibussowitsch Developer Notes: 4380b7f5c055SJed Brown The Gyroid mesh does not currently mark boundary sets. 4381b7f5c055SJed Brown 43821cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMSetType()`, `DMCreate()` 4383b7f5c055SJed Brown @*/ 4384d71ae5a4SJacob 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) 4385d71ae5a4SJacob Faibussowitsch { 4386b7f5c055SJed Brown PetscFunctionBegin; 43879566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 43889566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 43899566063dSJacob Faibussowitsch PetscCall(DMPlexCreateTPSMesh_Internal(*dm, tpstype, extent, periodic, tps_distribute, refinements, layers, thickness)); 43903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 4391b7f5c055SJed Brown } 4392b7f5c055SJed Brown 43939318fe57SMatthew G. Knepley /*@ 43949318fe57SMatthew G. Knepley DMPlexCreateSphereMesh - Creates a mesh on the d-dimensional sphere, S^d. 43959318fe57SMatthew G. Knepley 43969318fe57SMatthew G. Knepley Collective 43979318fe57SMatthew G. Knepley 43989318fe57SMatthew G. Knepley Input Parameters: 4399a1cb98faSBarry Smith + comm - The communicator for the `DM` object 44009318fe57SMatthew G. Knepley . dim - The dimension 44019318fe57SMatthew G. Knepley . simplex - Use simplices, or tensor product cells 44029318fe57SMatthew G. Knepley - R - The radius 44039318fe57SMatthew G. Knepley 44049318fe57SMatthew G. Knepley Output Parameter: 4405a1cb98faSBarry Smith . dm - The `DM` object 44069318fe57SMatthew G. Knepley 44079318fe57SMatthew G. Knepley Level: beginner 44089318fe57SMatthew G. Knepley 44091cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBallMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()` 44109318fe57SMatthew G. Knepley @*/ 4411d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateSphereMesh(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscReal R, DM *dm) 4412d71ae5a4SJacob Faibussowitsch { 44139318fe57SMatthew G. Knepley PetscFunctionBegin; 44144f572ea9SToby Isaac PetscAssertPointer(dm, 5); 44159566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 44169566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 44179566063dSJacob Faibussowitsch PetscCall(DMPlexCreateSphereMesh_Internal(*dm, dim, simplex, R)); 44183ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 44199318fe57SMatthew G. Knepley } 44209318fe57SMatthew G. Knepley 4421d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBallMesh_Internal(DM dm, PetscInt dim, PetscReal R) 4422d71ae5a4SJacob Faibussowitsch { 44239318fe57SMatthew G. Knepley DM sdm, vol; 44249318fe57SMatthew G. Knepley DMLabel bdlabel; 4425dd2b43ebSStefano Zampini const char *prefix; 44269318fe57SMatthew G. Knepley 44279318fe57SMatthew G. Knepley PetscFunctionBegin; 44289566063dSJacob Faibussowitsch PetscCall(DMCreate(PetscObjectComm((PetscObject)dm), &sdm)); 44299566063dSJacob Faibussowitsch PetscCall(DMSetType(sdm, DMPLEX)); 4430dd2b43ebSStefano Zampini PetscCall(DMGetOptionsPrefix(dm, &prefix)); 4431dd2b43ebSStefano Zampini PetscCall(DMSetOptionsPrefix(sdm, prefix)); 4432dd2b43ebSStefano Zampini PetscCall(DMAppendOptionsPrefix(sdm, "bd_")); 4433dd2b43ebSStefano Zampini PetscCall(DMPlexDistributeSetDefault(sdm, PETSC_FALSE)); 44349566063dSJacob Faibussowitsch PetscCall(DMPlexCreateSphereMesh_Internal(sdm, dim - 1, PETSC_TRUE, R)); 44359566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(sdm)); 44369566063dSJacob Faibussowitsch PetscCall(DMViewFromOptions(sdm, NULL, "-dm_view")); 44379566063dSJacob Faibussowitsch PetscCall(DMPlexGenerate(sdm, NULL, PETSC_TRUE, &vol)); 44389566063dSJacob Faibussowitsch PetscCall(DMDestroy(&sdm)); 443969d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &vol)); 44409566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(dm, "marker")); 44419566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "marker", &bdlabel)); 44429566063dSJacob Faibussowitsch PetscCall(DMPlexMarkBoundaryFaces(dm, PETSC_DETERMINE, bdlabel)); 44439566063dSJacob Faibussowitsch PetscCall(DMPlexLabelComplete(dm, bdlabel)); 44443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 444551a74b61SMatthew G. Knepley } 444651a74b61SMatthew G. Knepley 444751a74b61SMatthew G. Knepley /*@ 444851a74b61SMatthew G. Knepley DMPlexCreateBallMesh - Creates a simplex mesh on the d-dimensional ball, B^d. 444951a74b61SMatthew G. Knepley 445051a74b61SMatthew G. Knepley Collective 445151a74b61SMatthew G. Knepley 445251a74b61SMatthew G. Knepley Input Parameters: 4453a1cb98faSBarry Smith + comm - The communicator for the `DM` object 445451a74b61SMatthew G. Knepley . dim - The dimension 445551a74b61SMatthew G. Knepley - R - The radius 445651a74b61SMatthew G. Knepley 445751a74b61SMatthew G. Knepley Output Parameter: 4458a1cb98faSBarry Smith . dm - The `DM` object 445951a74b61SMatthew G. Knepley 4460a1cb98faSBarry Smith Options Database Key: 446160225df5SJacob Faibussowitsch . bd_dm_refine - This will refine the surface mesh preserving the sphere geometry 446251a74b61SMatthew G. Knepley 446351a74b61SMatthew G. Knepley Level: beginner 446451a74b61SMatthew G. Knepley 44651cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateSphereMesh()`, `DMPlexCreateBoxMesh()`, `DMSetType()`, `DMCreate()` 446651a74b61SMatthew G. Knepley @*/ 4467d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateBallMesh(MPI_Comm comm, PetscInt dim, PetscReal R, DM *dm) 4468d71ae5a4SJacob Faibussowitsch { 446951a74b61SMatthew G. Knepley PetscFunctionBegin; 44709566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 44719566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 44729566063dSJacob Faibussowitsch PetscCall(DMPlexCreateBallMesh_Internal(*dm, dim, R)); 44733ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 44742829fed8SMatthew G. Knepley } 44752829fed8SMatthew G. Knepley 4476d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateReferenceCell_Internal(DM rdm, DMPolytopeType ct) 4477d71ae5a4SJacob Faibussowitsch { 44780a6ba040SMatthew G. Knepley PetscFunctionBegin; 44799318fe57SMatthew G. Knepley switch (ct) { 44809371c9d4SSatish Balay case DM_POLYTOPE_POINT: { 44819318fe57SMatthew G. Knepley PetscInt numPoints[1] = {1}; 44829318fe57SMatthew G. Knepley PetscInt coneSize[1] = {0}; 44839318fe57SMatthew G. Knepley PetscInt cones[1] = {0}; 44849318fe57SMatthew G. Knepley PetscInt coneOrientations[1] = {0}; 44859318fe57SMatthew G. Knepley PetscScalar vertexCoords[1] = {0.0}; 44869318fe57SMatthew G. Knepley 44879566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 0)); 44889566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 44899371c9d4SSatish Balay } break; 44909371c9d4SSatish Balay case DM_POLYTOPE_SEGMENT: { 44919318fe57SMatthew G. Knepley PetscInt numPoints[2] = {2, 1}; 44929318fe57SMatthew G. Knepley PetscInt coneSize[3] = {2, 0, 0}; 44939318fe57SMatthew G. Knepley PetscInt cones[2] = {1, 2}; 44949318fe57SMatthew G. Knepley PetscInt coneOrientations[2] = {0, 0}; 44959318fe57SMatthew G. Knepley PetscScalar vertexCoords[2] = {-1.0, 1.0}; 44969318fe57SMatthew G. Knepley 44979566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 1)); 44989566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 44999371c9d4SSatish Balay } break; 45009371c9d4SSatish Balay case DM_POLYTOPE_POINT_PRISM_TENSOR: { 4501b5a892a1SMatthew G. Knepley PetscInt numPoints[2] = {2, 1}; 4502b5a892a1SMatthew G. Knepley PetscInt coneSize[3] = {2, 0, 0}; 4503b5a892a1SMatthew G. Knepley PetscInt cones[2] = {1, 2}; 4504b5a892a1SMatthew G. Knepley PetscInt coneOrientations[2] = {0, 0}; 4505b5a892a1SMatthew G. Knepley PetscScalar vertexCoords[2] = {-1.0, 1.0}; 4506b5a892a1SMatthew G. Knepley 45079566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 1)); 45089566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45099371c9d4SSatish Balay } break; 45109371c9d4SSatish Balay case DM_POLYTOPE_TRIANGLE: { 45119318fe57SMatthew G. Knepley PetscInt numPoints[2] = {3, 1}; 45129318fe57SMatthew G. Knepley PetscInt coneSize[4] = {3, 0, 0, 0}; 45139318fe57SMatthew G. Knepley PetscInt cones[3] = {1, 2, 3}; 45149318fe57SMatthew G. Knepley PetscInt coneOrientations[3] = {0, 0, 0}; 45159318fe57SMatthew G. Knepley PetscScalar vertexCoords[6] = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0}; 45169318fe57SMatthew G. Knepley 45179566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 2)); 45189566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45199371c9d4SSatish Balay } break; 45209371c9d4SSatish Balay case DM_POLYTOPE_QUADRILATERAL: { 45219318fe57SMatthew G. Knepley PetscInt numPoints[2] = {4, 1}; 45229318fe57SMatthew G. Knepley PetscInt coneSize[5] = {4, 0, 0, 0, 0}; 45239318fe57SMatthew G. Knepley PetscInt cones[4] = {1, 2, 3, 4}; 45249318fe57SMatthew G. Knepley PetscInt coneOrientations[4] = {0, 0, 0, 0}; 45259318fe57SMatthew G. Knepley PetscScalar vertexCoords[8] = {-1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0}; 45269318fe57SMatthew G. Knepley 45279566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 2)); 45289566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45299371c9d4SSatish Balay } break; 45309371c9d4SSatish Balay case DM_POLYTOPE_SEG_PRISM_TENSOR: { 45319318fe57SMatthew G. Knepley PetscInt numPoints[2] = {4, 1}; 45329318fe57SMatthew G. Knepley PetscInt coneSize[5] = {4, 0, 0, 0, 0}; 45339318fe57SMatthew G. Knepley PetscInt cones[4] = {1, 2, 3, 4}; 45349318fe57SMatthew G. Knepley PetscInt coneOrientations[4] = {0, 0, 0, 0}; 45359318fe57SMatthew G. Knepley PetscScalar vertexCoords[8] = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0}; 45369318fe57SMatthew G. Knepley 45379566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 2)); 45389566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45399371c9d4SSatish Balay } break; 45409371c9d4SSatish Balay case DM_POLYTOPE_TETRAHEDRON: { 45419318fe57SMatthew G. Knepley PetscInt numPoints[2] = {4, 1}; 45429318fe57SMatthew G. Knepley PetscInt coneSize[5] = {4, 0, 0, 0, 0}; 4543f0edb160SMatthew G. Knepley PetscInt cones[4] = {1, 2, 3, 4}; 45449318fe57SMatthew G. Knepley PetscInt coneOrientations[4] = {0, 0, 0, 0}; 4545f0edb160SMatthew 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}; 45469318fe57SMatthew G. Knepley 45479566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45489566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45499371c9d4SSatish Balay } break; 45509371c9d4SSatish Balay case DM_POLYTOPE_HEXAHEDRON: { 45519318fe57SMatthew G. Knepley PetscInt numPoints[2] = {8, 1}; 45529318fe57SMatthew G. Knepley PetscInt coneSize[9] = {8, 0, 0, 0, 0, 0, 0, 0, 0}; 4553f0edb160SMatthew G. Knepley PetscInt cones[8] = {1, 2, 3, 4, 5, 6, 7, 8}; 45549318fe57SMatthew G. Knepley PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 45559371c9d4SSatish 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}; 45569318fe57SMatthew G. Knepley 45579566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45589566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45599371c9d4SSatish Balay } break; 45609371c9d4SSatish Balay case DM_POLYTOPE_TRI_PRISM: { 45619318fe57SMatthew G. Knepley PetscInt numPoints[2] = {6, 1}; 45629318fe57SMatthew G. Knepley PetscInt coneSize[7] = {6, 0, 0, 0, 0, 0, 0}; 4563f0edb160SMatthew G. Knepley PetscInt cones[6] = {1, 2, 3, 4, 5, 6}; 45649318fe57SMatthew G. Knepley PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0}; 45659371c9d4SSatish 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}; 45669318fe57SMatthew G. Knepley 45679566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45689566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45699371c9d4SSatish Balay } break; 45709371c9d4SSatish Balay case DM_POLYTOPE_TRI_PRISM_TENSOR: { 45719318fe57SMatthew G. Knepley PetscInt numPoints[2] = {6, 1}; 45729318fe57SMatthew G. Knepley PetscInt coneSize[7] = {6, 0, 0, 0, 0, 0, 0}; 45739318fe57SMatthew G. Knepley PetscInt cones[6] = {1, 2, 3, 4, 5, 6}; 45749318fe57SMatthew G. Knepley PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0}; 45759371c9d4SSatish 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}; 45769318fe57SMatthew G. Knepley 45779566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45789566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45799371c9d4SSatish Balay } break; 45809371c9d4SSatish Balay case DM_POLYTOPE_QUAD_PRISM_TENSOR: { 45819318fe57SMatthew G. Knepley PetscInt numPoints[2] = {8, 1}; 45829318fe57SMatthew G. Knepley PetscInt coneSize[9] = {8, 0, 0, 0, 0, 0, 0, 0, 0}; 45839318fe57SMatthew G. Knepley PetscInt cones[8] = {1, 2, 3, 4, 5, 6, 7, 8}; 45849318fe57SMatthew G. Knepley PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 45859371c9d4SSatish 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}; 45869318fe57SMatthew G. Knepley 45879566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45889566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45899371c9d4SSatish Balay } break; 45909371c9d4SSatish Balay case DM_POLYTOPE_PYRAMID: { 45919318fe57SMatthew G. Knepley PetscInt numPoints[2] = {5, 1}; 45929318fe57SMatthew G. Knepley PetscInt coneSize[6] = {5, 0, 0, 0, 0, 0}; 4593f0edb160SMatthew G. Knepley PetscInt cones[5] = {1, 2, 3, 4, 5}; 45949318fe57SMatthew G. Knepley PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 45959371c9d4SSatish 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}; 45969318fe57SMatthew G. Knepley 45979566063dSJacob Faibussowitsch PetscCall(DMSetDimension(rdm, 3)); 45989566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords)); 45999371c9d4SSatish Balay } break; 4600d71ae5a4SJacob Faibussowitsch default: 4601d71ae5a4SJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)rdm), PETSC_ERR_ARG_WRONG, "Cannot create reference cell for cell type %s", DMPolytopeTypes[ct]); 46029318fe57SMatthew G. Knepley } 46039318fe57SMatthew G. Knepley { 46049318fe57SMatthew G. Knepley PetscInt Nv, v; 46059318fe57SMatthew G. Knepley 46069318fe57SMatthew G. Knepley /* Must create the celltype label here so that we do not automatically try to compute the types */ 46079566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(rdm, "celltype")); 46089566063dSJacob Faibussowitsch PetscCall(DMPlexSetCellType(rdm, 0, ct)); 46099566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(rdm, NULL, &Nv)); 46109566063dSJacob Faibussowitsch for (v = 1; v < Nv; ++v) PetscCall(DMPlexSetCellType(rdm, v, DM_POLYTOPE_POINT)); 46119318fe57SMatthew G. Knepley } 46129566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolateInPlace_Internal(rdm)); 46139566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)rdm, DMPolytopeTypes[ct])); 46143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 46150a6ba040SMatthew G. Knepley } 46160a6ba040SMatthew G. Knepley 46179318fe57SMatthew G. Knepley /*@ 4618a1cb98faSBarry Smith DMPlexCreateReferenceCell - Create a `DMPLEX` with the appropriate FEM reference cell 46199318fe57SMatthew G. Knepley 46209318fe57SMatthew G. Knepley Collective 46219318fe57SMatthew G. Knepley 46229318fe57SMatthew G. Knepley Input Parameters: 46239318fe57SMatthew G. Knepley + comm - The communicator 46249318fe57SMatthew G. Knepley - ct - The cell type of the reference cell 46259318fe57SMatthew G. Knepley 46269318fe57SMatthew G. Knepley Output Parameter: 46279318fe57SMatthew G. Knepley . refdm - The reference cell 46289318fe57SMatthew G. Knepley 46299318fe57SMatthew G. Knepley Level: intermediate 46309318fe57SMatthew G. Knepley 463142747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateBoxMesh()` 46329318fe57SMatthew G. Knepley @*/ 4633d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, DMPolytopeType ct, DM *refdm) 4634d71ae5a4SJacob Faibussowitsch { 46350a6ba040SMatthew G. Knepley PetscFunctionBegin; 46369566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, refdm)); 46379566063dSJacob Faibussowitsch PetscCall(DMSetType(*refdm, DMPLEX)); 46389566063dSJacob Faibussowitsch PetscCall(DMPlexCreateReferenceCell_Internal(*refdm, ct)); 46393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 46409318fe57SMatthew G. Knepley } 464179a015ccSMatthew G. Knepley 4642d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateBoundaryLabel_Private(DM dm, const char name[]) 4643d71ae5a4SJacob Faibussowitsch { 46449318fe57SMatthew G. Knepley DM plex; 46459318fe57SMatthew G. Knepley DMLabel label; 46469318fe57SMatthew G. Knepley PetscBool hasLabel; 46470a6ba040SMatthew G. Knepley 4648c22d3578SMatthew G. Knepley PetscFunctionBegin; 46499566063dSJacob Faibussowitsch PetscCall(DMHasLabel(dm, name, &hasLabel)); 46503ba16761SJacob Faibussowitsch if (hasLabel) PetscFunctionReturn(PETSC_SUCCESS); 46519566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(dm, name)); 46529566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, name, &label)); 46539566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 46549566063dSJacob Faibussowitsch PetscCall(DMPlexMarkBoundaryFaces(plex, 1, label)); 46551c8afea9SMatthew G. Knepley PetscCall(DMPlexLabelComplete(plex, label)); 46569566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 46573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 46589318fe57SMatthew G. Knepley } 4659acdc6f61SToby Isaac 4660669647acSMatthew G. Knepley /* 4661669647acSMatthew 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. 4662669647acSMatthew G. Knepley 4663669647acSMatthew G. Knepley (x, y) -> (r, theta) = (x[1], (x[0] - lower[0]) * 2\pi/(upper[0] - lower[0])) 4664669647acSMatthew G. Knepley */ 4665d71ae5a4SJacob 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[]) 4666d71ae5a4SJacob Faibussowitsch { 4667669647acSMatthew G. Knepley const PetscReal low = PetscRealPart(constants[0]); 4668669647acSMatthew G. Knepley const PetscReal upp = PetscRealPart(constants[1]); 4669669647acSMatthew G. Knepley const PetscReal r = PetscRealPart(u[1]); 4670669647acSMatthew G. Knepley const PetscReal th = 2. * PETSC_PI * (PetscRealPart(u[0]) - low) / (upp - low); 4671669647acSMatthew G. Knepley 4672669647acSMatthew G. Knepley f0[0] = r * PetscCosReal(th); 4673669647acSMatthew G. Knepley f0[1] = r * PetscSinReal(th); 4674669647acSMatthew G. Knepley } 4675669647acSMatthew G. Knepley 46765390be7dSMatthew G. Knepley // Insert vertices and their joins, marked by depth 46775390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Vertices(DM dm, DMLabel label, DMLabel vlabel, PetscInt val, PetscInt n, const PetscInt vertices[]) 46785390be7dSMatthew G. Knepley { 46795390be7dSMatthew G. Knepley PetscFunctionBegin; 46805390be7dSMatthew G. Knepley PetscCall(DMPlexMarkSubmesh_Interpolated(dm, vlabel, val, PETSC_FALSE, PETSC_FALSE, label, NULL)); 46815390be7dSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 46825390be7dSMatthew G. Knepley } 46835390be7dSMatthew G. Knepley 46845390be7dSMatthew G. Knepley // Insert faces and their closures, marked by depth 46855390be7dSMatthew G. Knepley static PetscErrorCode ProcessCohesiveLabel_Faces(DM dm, DMLabel label, PetscInt n, const PetscInt faces[]) 46865390be7dSMatthew G. Knepley { 46875390be7dSMatthew G. Knepley PetscFunctionBegin; 46885390be7dSMatthew G. Knepley for (PetscInt p = 0; p < n; ++p) { 46895390be7dSMatthew G. Knepley const PetscInt point = faces[p]; 46905390be7dSMatthew G. Knepley PetscInt *closure = NULL; 46915390be7dSMatthew G. Knepley PetscInt clSize, pdepth; 46925390be7dSMatthew G. Knepley 46935390be7dSMatthew G. Knepley PetscCall(DMPlexGetPointDepth(dm, point, &pdepth)); 46945390be7dSMatthew G. Knepley PetscCall(DMLabelSetValue(label, point, pdepth)); 46955390be7dSMatthew G. Knepley PetscCall(DMPlexGetTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure)); 46965390be7dSMatthew G. Knepley for (PetscInt cl = 0; cl < clSize * 2; cl += 2) { 46975390be7dSMatthew G. Knepley PetscCall(DMPlexGetPointDepth(dm, closure[cl], &pdepth)); 46985390be7dSMatthew G. Knepley PetscCall(DMLabelSetValue(label, closure[cl], pdepth)); 46995390be7dSMatthew G. Knepley } 47005390be7dSMatthew G. Knepley PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, &clSize, &closure)); 47015390be7dSMatthew G. Knepley } 47025390be7dSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 47035390be7dSMatthew G. Knepley } 47045390be7dSMatthew G. Knepley 47054e22dd4cSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscOptionsFindPairPrefix_Private(PetscOptions, const char pre[], const char name[], const char *option[], const char *value[], PetscBool *flg); 47064e22dd4cSMatthew G. Knepley 47075dca41c3SJed Brown const char *const DMPlexShapes[] = {"box", "box_surface", "ball", "sphere", "cylinder", "schwarz_p", "gyroid", "doublet", "annulus", "hypercubic", "zbox", "unknown", "DMPlexShape", "DM_SHAPE_", NULL}; 47089318fe57SMatthew G. Knepley 4709ce78bad3SBarry Smith static PetscErrorCode DMPlexCreateFromOptions_Internal(PetscOptionItems PetscOptionsObject, PetscBool *useCoordSpace, DM dm) 4710d71ae5a4SJacob Faibussowitsch { 47119318fe57SMatthew G. Knepley DMPlexShape shape = DM_SHAPE_BOX; 47129318fe57SMatthew G. Knepley DMPolytopeType cell = DM_POLYTOPE_TRIANGLE; 47139318fe57SMatthew G. Knepley PetscInt dim = 2; 4714b9da1bb3SMatthew G. Knepley PetscBool simplex = PETSC_TRUE, interpolate = PETSC_TRUE, orient = PETSC_FALSE, adjCone = PETSC_FALSE, adjClosure = PETSC_TRUE, refDomain = PETSC_FALSE; 4715d0812dedSMatthew G. Knepley PetscBool flg, flg2, fflg, strflg, bdfflg, nameflg; 47169318fe57SMatthew G. Knepley MPI_Comm comm; 4717ed5e4e85SVaclav Hapla char filename[PETSC_MAX_PATH_LEN] = "<unspecified>"; 4718ed5e4e85SVaclav Hapla char bdFilename[PETSC_MAX_PATH_LEN] = "<unspecified>"; 4719ed5e4e85SVaclav Hapla char plexname[PETSC_MAX_PATH_LEN] = ""; 47204e22dd4cSMatthew G. Knepley const char *option; 47219318fe57SMatthew G. Knepley 47229318fe57SMatthew G. Knepley PetscFunctionBegin; 4723708be2fdSJed Brown PetscCall(PetscLogEventBegin(DMPLEX_CreateFromOptions, dm, 0, 0, 0)); 47249566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 47259318fe57SMatthew G. Knepley /* TODO Turn this into a registration interface */ 47269566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-dm_plex_filename", "File containing a mesh", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &fflg)); 4727d0812dedSMatthew G. Knepley PetscCall(PetscOptionsString("-dm_plex_file_contents", "Contents of a file format in a string", "DMPlexCreateFromFile", filename, filename, sizeof(filename), &strflg)); 47289566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-dm_plex_boundary_filename", "File containing a mesh boundary", "DMPlexCreateFromFile", bdFilename, bdFilename, sizeof(bdFilename), &bdfflg)); 47299566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-dm_plex_name", "Name of the mesh in the file", "DMPlexCreateFromFile", plexname, plexname, sizeof(plexname), &nameflg)); 47309566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-dm_plex_cell", "Cell shape", "", DMPolytopeTypes, (PetscEnum)cell, (PetscEnum *)&cell, NULL)); 47319566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_reference_cell_domain", "Use a reference cell domain", "", refDomain, &refDomain, NULL)); 47329566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-dm_plex_shape", "Shape for built-in mesh", "", DMPlexShapes, (PetscEnum)shape, (PetscEnum *)&shape, &flg)); 47339566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_plex_dim", "Topological dimension of the mesh", "DMGetDimension", dim, &dim, &flg, 0)); 47349566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_simplex", "Mesh cell shape", "", simplex, &simplex, &flg)); 47359566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_interpolate", "Flag to create edges and faces automatically", "", interpolate, &interpolate, &flg)); 4736b9da1bb3SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_plex_orient", "Orient the constructed mesh", "DMPlexOrient", orient, &orient, &flg)); 47379566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_adj_cone", "Set adjacency direction", "DMSetBasicAdjacency", adjCone, &adjCone, &flg)); 47389566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_adj_closure", "Set adjacency size", "DMSetBasicAdjacency", adjClosure, &adjClosure, &flg2)); 47399566063dSJacob Faibussowitsch if (flg || flg2) PetscCall(DMSetBasicAdjacency(dm, adjCone, adjClosure)); 47403f3e541fSMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_plex_print_adj", "Debug output level all adjacency computations", "", 0, &((DM_Plex *)dm->data)->printAdj, NULL, 0)); 47419318fe57SMatthew G. Knepley 474261a622f3SMatthew G. Knepley switch (cell) { 474361a622f3SMatthew G. Knepley case DM_POLYTOPE_POINT: 474461a622f3SMatthew G. Knepley case DM_POLYTOPE_SEGMENT: 474561a622f3SMatthew G. Knepley case DM_POLYTOPE_POINT_PRISM_TENSOR: 474661a622f3SMatthew G. Knepley case DM_POLYTOPE_TRIANGLE: 474761a622f3SMatthew G. Knepley case DM_POLYTOPE_QUADRILATERAL: 474861a622f3SMatthew G. Knepley case DM_POLYTOPE_TETRAHEDRON: 4749d71ae5a4SJacob Faibussowitsch case DM_POLYTOPE_HEXAHEDRON: 4750d71ae5a4SJacob Faibussowitsch *useCoordSpace = PETSC_TRUE; 4751d71ae5a4SJacob Faibussowitsch break; 4752d71ae5a4SJacob Faibussowitsch default: 4753d71ae5a4SJacob Faibussowitsch *useCoordSpace = PETSC_FALSE; 4754d71ae5a4SJacob Faibussowitsch break; 475561a622f3SMatthew G. Knepley } 475661a622f3SMatthew G. Knepley 47579318fe57SMatthew G. Knepley if (fflg) { 47589318fe57SMatthew G. Knepley DM dmnew; 47591e4a82c4SMatthew G. Knepley const char *name; 47609318fe57SMatthew G. Knepley 47611e4a82c4SMatthew G. Knepley PetscCall(PetscObjectGetName((PetscObject)dm, &name)); 47621e4a82c4SMatthew G. Knepley PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), filename, nameflg ? plexname : name, interpolate, &dmnew)); 47635de52c6dSVaclav Hapla PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew)); 476469d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &dmnew)); 47659318fe57SMatthew G. Knepley } else if (refDomain) { 47669566063dSJacob Faibussowitsch PetscCall(DMPlexCreateReferenceCell_Internal(dm, cell)); 47679318fe57SMatthew G. Knepley } else if (bdfflg) { 47689318fe57SMatthew G. Knepley DM bdm, dmnew; 47691e4a82c4SMatthew G. Knepley const char *name; 47709318fe57SMatthew G. Knepley 47711e4a82c4SMatthew G. Knepley PetscCall(PetscObjectGetName((PetscObject)dm, &name)); 47721e4a82c4SMatthew G. Knepley PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), bdFilename, nameflg ? plexname : name, interpolate, &bdm)); 47739566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)bdm, "bd_")); 47749566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(bdm)); 47759566063dSJacob Faibussowitsch PetscCall(DMPlexGenerate(bdm, NULL, interpolate, &dmnew)); 47769566063dSJacob Faibussowitsch PetscCall(DMDestroy(&bdm)); 47775de52c6dSVaclav Hapla PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew)); 477869d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &dmnew)); 4779d0812dedSMatthew G. Knepley } else if (strflg) { 4780d0812dedSMatthew G. Knepley DM dmnew; 4781d0812dedSMatthew G. Knepley PetscViewer viewer; 4782d0812dedSMatthew G. Knepley const char *contents; 4783d0812dedSMatthew G. Knepley char *strname; 4784d0812dedSMatthew G. Knepley char tmpdir[PETSC_MAX_PATH_LEN]; 4785d0812dedSMatthew G. Knepley char tmpfilename[PETSC_MAX_PATH_LEN]; 4786d0812dedSMatthew G. Knepley char name[PETSC_MAX_PATH_LEN]; 4787d0812dedSMatthew G. Knepley MPI_Comm comm; 4788d0812dedSMatthew G. Knepley PetscMPIInt rank; 4789d0812dedSMatthew G. Knepley 4790d0812dedSMatthew G. Knepley PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 4791d0812dedSMatthew G. Knepley PetscCallMPI(MPI_Comm_rank(comm, &rank)); 4792d0812dedSMatthew G. Knepley PetscCall(PetscStrchr(filename, ':', &strname)); 4793d0812dedSMatthew G. Knepley PetscCheck(strname, comm, PETSC_ERR_ARG_WRONG, "File contents must have the form \"ext:string_name\", not %s", filename); 4794d0812dedSMatthew G. Knepley strname[0] = '\0'; 4795d0812dedSMatthew G. Knepley ++strname; 4796d0812dedSMatthew G. Knepley PetscCall(PetscDLSym(NULL, strname, (void **)&contents)); 4797d0812dedSMatthew G. Knepley PetscCheck(contents, comm, PETSC_ERR_ARG_WRONG, "Could not locate mesh string %s", strname); 4798d0812dedSMatthew G. Knepley PetscCall(PetscGetTmp(comm, tmpdir, PETSC_MAX_PATH_LEN)); 4799ed32af8cSMatthew G. Knepley PetscCall(PetscStrlcat(tmpdir, "/meshXXXXXX", PETSC_MAX_PATH_LEN)); 4800ed32af8cSMatthew G. Knepley PetscCall(PetscMkdtemp(tmpdir)); 4801ed32af8cSMatthew G. Knepley PetscCall(PetscSNPrintf(tmpfilename, PETSC_MAX_PATH_LEN, "%s/mesh.%s", tmpdir, filename)); 4802d0812dedSMatthew G. Knepley PetscCall(PetscViewerASCIIOpen(comm, tmpfilename, &viewer)); 4803d0812dedSMatthew G. Knepley PetscCall(PetscViewerASCIIPrintf(viewer, "%s\n", contents)); 4804d0812dedSMatthew G. Knepley PetscCall(PetscViewerDestroy(&viewer)); 4805d0812dedSMatthew G. Knepley PetscCall(DMPlexCreateFromFile(PetscObjectComm((PetscObject)dm), tmpfilename, plexname, interpolate, &dmnew)); 4806ed32af8cSMatthew G. Knepley PetscCall(PetscRMTree(tmpdir)); 4807d0812dedSMatthew G. Knepley PetscCall(PetscSNPrintf(name, PETSC_MAX_PATH_LEN, "%s Mesh", strname)); 4808d0812dedSMatthew G. Knepley PetscCall(PetscObjectSetName((PetscObject)dm, name)); 4809d0812dedSMatthew G. Knepley PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew)); 4810d0812dedSMatthew G. Knepley PetscCall(DMPlexReplace_Internal(dm, &dmnew)); 48119318fe57SMatthew G. Knepley } else { 48129566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dm, DMPlexShapes[shape])); 48139318fe57SMatthew G. Knepley switch (shape) { 4814669647acSMatthew G. Knepley case DM_SHAPE_BOX: 48155dca41c3SJed Brown case DM_SHAPE_ZBOX: 4816669647acSMatthew G. Knepley case DM_SHAPE_ANNULUS: { 48179318fe57SMatthew G. Knepley PetscInt faces[3] = {0, 0, 0}; 48189318fe57SMatthew G. Knepley PetscReal lower[3] = {0, 0, 0}; 48199318fe57SMatthew G. Knepley PetscReal upper[3] = {1, 1, 1}; 48209318fe57SMatthew G. Knepley DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE}; 4821669647acSMatthew G. Knepley PetscBool isAnnular = shape == DM_SHAPE_ANNULUS ? PETSC_TRUE : PETSC_FALSE; 48229318fe57SMatthew G. Knepley PetscInt i, n; 48239318fe57SMatthew G. Knepley 48249318fe57SMatthew G. Knepley n = dim; 48259318fe57SMatthew G. Knepley for (i = 0; i < dim; ++i) faces[i] = (dim == 1 ? 1 : 4 - dim); 48269566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg)); 48279318fe57SMatthew G. Knepley n = 3; 48289566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg)); 482963a3b9bcSJacob Faibussowitsch PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Lower box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim); 48309318fe57SMatthew G. Knepley n = 3; 48319566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg)); 483263a3b9bcSJacob Faibussowitsch PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Upper box point had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim); 48339318fe57SMatthew G. Knepley n = 3; 48349566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg)); 483563a3b9bcSJacob Faibussowitsch PetscCheck(!flg || !(n != dim), comm, PETSC_ERR_ARG_SIZ, "Box boundary types had %" PetscInt_FMT " values, should have been %" PetscInt_FMT, n, dim); 4836669647acSMatthew G. Knepley 4837669647acSMatthew G. Knepley PetscCheck(!isAnnular || dim == 2, comm, PETSC_ERR_ARG_OUTOFRANGE, "Only two dimensional annuli have been implemented"); 4838669647acSMatthew G. Knepley if (isAnnular) 4839669647acSMatthew G. Knepley for (i = 0; i < dim - 1; ++i) bdt[i] = DM_BOUNDARY_PERIODIC; 4840669647acSMatthew G. Knepley 48419318fe57SMatthew G. Knepley switch (cell) { 484261a622f3SMatthew G. Knepley case DM_POLYTOPE_TRI_PRISM_TENSOR: 48439566063dSJacob Faibussowitsch PetscCall(DMPlexCreateWedgeBoxMesh_Internal(dm, faces, lower, upper, bdt)); 4844d410b0cfSMatthew G. Knepley if (!interpolate) { 4845d410b0cfSMatthew G. Knepley DM udm; 4846d410b0cfSMatthew G. Knepley 48479566063dSJacob Faibussowitsch PetscCall(DMPlexUninterpolate(dm, &udm)); 484869d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &udm)); 4849d410b0cfSMatthew G. Knepley } 48509318fe57SMatthew G. Knepley break; 4851d71ae5a4SJacob Faibussowitsch default: 48525dca41c3SJed Brown PetscCall(DMPlexCreateBoxMesh_Internal(dm, shape, dim, simplex, faces, lower, upper, bdt, interpolate)); 4853d71ae5a4SJacob Faibussowitsch break; 48549318fe57SMatthew G. Knepley } 4855669647acSMatthew G. Knepley if (isAnnular) { 4856669647acSMatthew G. Knepley DM cdm; 4857669647acSMatthew G. Knepley PetscDS cds; 4858669647acSMatthew G. Knepley PetscScalar bounds[2] = {lower[0], upper[0]}; 4859669647acSMatthew G. Knepley 4860669647acSMatthew G. Knepley // Fix coordinates for annular region 4861669647acSMatthew G. Knepley PetscCall(DMSetPeriodicity(dm, NULL, NULL, NULL)); 4862669647acSMatthew G. Knepley PetscCall(DMSetCellCoordinatesLocal(dm, NULL)); 4863669647acSMatthew G. Knepley PetscCall(DMSetCellCoordinates(dm, NULL)); 4864e44f6aebSMatthew G. Knepley PetscCall(DMPlexCreateCoordinateSpace(dm, 1, PETSC_TRUE, NULL)); 4865669647acSMatthew G. Knepley PetscCall(DMGetCoordinateDM(dm, &cdm)); 4866669647acSMatthew G. Knepley PetscCall(DMGetDS(cdm, &cds)); 4867669647acSMatthew G. Knepley PetscCall(PetscDSSetConstants(cds, 2, bounds)); 4868669647acSMatthew G. Knepley PetscCall(DMPlexRemapGeometry(dm, 0.0, boxToAnnulus)); 4869669647acSMatthew G. Knepley } 48709371c9d4SSatish Balay } break; 48719371c9d4SSatish Balay case DM_SHAPE_BOX_SURFACE: { 48729318fe57SMatthew G. Knepley PetscInt faces[3] = {0, 0, 0}; 48739318fe57SMatthew G. Knepley PetscReal lower[3] = {0, 0, 0}; 48749318fe57SMatthew G. Knepley PetscReal upper[3] = {1, 1, 1}; 48759318fe57SMatthew G. Knepley PetscInt i, n; 48769318fe57SMatthew G. Knepley 48779318fe57SMatthew G. Knepley n = dim + 1; 48789318fe57SMatthew G. Knepley for (i = 0; i < dim + 1; ++i) faces[i] = (dim + 1 == 1 ? 1 : 4 - (dim + 1)); 48799566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", faces, &n, &flg)); 48809318fe57SMatthew G. Knepley n = 3; 48819566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg)); 488263a3b9bcSJacob 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); 48839318fe57SMatthew G. Knepley n = 3; 48849566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg)); 488563a3b9bcSJacob 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); 48869566063dSJacob Faibussowitsch PetscCall(DMPlexCreateBoxSurfaceMesh_Internal(dm, dim + 1, faces, lower, upper, interpolate)); 48879371c9d4SSatish Balay } break; 48889371c9d4SSatish Balay case DM_SHAPE_SPHERE: { 48899318fe57SMatthew G. Knepley PetscReal R = 1.0; 48909318fe57SMatthew G. Knepley 48919566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_plex_sphere_radius", "Radius of the sphere", "", R, &R, &flg)); 48929566063dSJacob Faibussowitsch PetscCall(DMPlexCreateSphereMesh_Internal(dm, dim, simplex, R)); 48939371c9d4SSatish Balay } break; 48949371c9d4SSatish Balay case DM_SHAPE_BALL: { 48959318fe57SMatthew G. Knepley PetscReal R = 1.0; 48969318fe57SMatthew G. Knepley 48979566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_plex_ball_radius", "Radius of the ball", "", R, &R, &flg)); 48989566063dSJacob Faibussowitsch PetscCall(DMPlexCreateBallMesh_Internal(dm, dim, R)); 48999371c9d4SSatish Balay } break; 49009371c9d4SSatish Balay case DM_SHAPE_CYLINDER: { 49019318fe57SMatthew G. Knepley DMBoundaryType bdt = DM_BOUNDARY_NONE; 49029318fe57SMatthew G. Knepley PetscInt Nw = 6; 490349704ca5SMatthew G. Knepley PetscInt Nr = 0; 49049318fe57SMatthew G. Knepley 49059566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-dm_plex_cylinder_bd", "Boundary type in the z direction", "", DMBoundaryTypes, (PetscEnum)bdt, (PetscEnum *)&bdt, NULL)); 49069566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_wedges", "Number of wedges around the cylinder", "", Nw, &Nw, NULL)); 490749704ca5SMatthew G. Knepley PetscCall(PetscOptionsInt("-dm_plex_cylinder_num_refine", "Number of refinements before projection", "", Nr, &Nr, NULL)); 49089318fe57SMatthew G. Knepley switch (cell) { 4909d71ae5a4SJacob Faibussowitsch case DM_POLYTOPE_TRI_PRISM_TENSOR: 4910d71ae5a4SJacob Faibussowitsch PetscCall(DMPlexCreateWedgeCylinderMesh_Internal(dm, Nw, interpolate)); 4911d71ae5a4SJacob Faibussowitsch break; 4912d71ae5a4SJacob Faibussowitsch default: 491349704ca5SMatthew G. Knepley PetscCall(DMPlexCreateHexCylinderMesh_Internal(dm, bdt, Nr)); 4914d71ae5a4SJacob Faibussowitsch break; 49159318fe57SMatthew G. Knepley } 49169371c9d4SSatish Balay } break; 4917b7f5c055SJed Brown case DM_SHAPE_SCHWARZ_P: // fallthrough 49189371c9d4SSatish Balay case DM_SHAPE_GYROID: { 4919b7f5c055SJed Brown PetscInt extent[3] = {1, 1, 1}, refine = 0, layers = 0, three; 4920b7f5c055SJed Brown PetscReal thickness = 0.; 4921b7f5c055SJed Brown DMBoundaryType periodic[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE}; 4922b7f5c055SJed Brown DMPlexTPSType tps_type = shape == DM_SHAPE_SCHWARZ_P ? DMPLEX_TPS_SCHWARZ_P : DMPLEX_TPS_GYROID; 49231436d7faSJed Brown PetscBool tps_distribute; 49249566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-dm_plex_tps_extent", "Number of replicas for each of three dimensions", NULL, extent, (three = 3, &three), NULL)); 49259566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-dm_plex_tps_refine", "Number of refinements", NULL, refine, &refine, NULL)); 49269566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnumArray("-dm_plex_tps_periodic", "Periodicity in each of three dimensions", NULL, DMBoundaryTypes, (PetscEnum *)periodic, (three = 3, &three), NULL)); 49279566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-dm_plex_tps_layers", "Number of layers in volumetric extrusion (or zero to not extrude)", NULL, layers, &layers, NULL)); 49289566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_plex_tps_thickness", "Thickness of volumetric extrusion", NULL, thickness, &thickness, NULL)); 49299566063dSJacob Faibussowitsch PetscCall(DMPlexDistributeGetDefault(dm, &tps_distribute)); 49309566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_tps_distribute", "Distribute the 2D mesh prior to refinement and extrusion", NULL, tps_distribute, &tps_distribute, NULL)); 49319566063dSJacob Faibussowitsch PetscCall(DMPlexCreateTPSMesh_Internal(dm, tps_type, extent, periodic, tps_distribute, refine, layers, thickness)); 49329371c9d4SSatish Balay } break; 49339371c9d4SSatish Balay case DM_SHAPE_DOUBLET: { 493405bd46c0SStefano Zampini DM dmnew; 493505bd46c0SStefano Zampini PetscReal rl = 0.0; 493605bd46c0SStefano Zampini 493705bd46c0SStefano Zampini PetscCall(PetscOptionsReal("-dm_plex_doublet_refinementlimit", "Refinement limit", NULL, rl, &rl, NULL)); 493805bd46c0SStefano Zampini PetscCall(DMPlexCreateDoublet(PetscObjectComm((PetscObject)dm), dim, simplex, interpolate, rl, &dmnew)); 49395de52c6dSVaclav Hapla PetscCall(DMPlexCopy_Internal(dm, PETSC_FALSE, PETSC_FALSE, dmnew)); 494069d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &dmnew)); 49419371c9d4SSatish Balay } break; 4942cfb853baSMatthew G. Knepley case DM_SHAPE_HYPERCUBIC: { 49438d2ec52aSSatish Balay PetscInt *edges, overlap = 1; 4944cfb853baSMatthew G. Knepley PetscReal *lower, *upper; 4945cfb853baSMatthew G. Knepley DMBoundaryType *bdt; 4946cfb853baSMatthew G. Knepley PetscInt n, d; 4947cfb853baSMatthew G. Knepley 4948cfb853baSMatthew G. Knepley *useCoordSpace = PETSC_FALSE; 4949cfb853baSMatthew G. Knepley PetscCall(PetscMalloc4(dim, &edges, dim, &lower, dim, &upper, dim, &bdt)); 4950cfb853baSMatthew G. Knepley for (d = 0; d < dim; ++d) { 4951cfb853baSMatthew G. Knepley edges[d] = 1; 4952cfb853baSMatthew G. Knepley lower[d] = 0.; 4953cfb853baSMatthew G. Knepley upper[d] = 1.; 4954cfb853baSMatthew G. Knepley bdt[d] = DM_BOUNDARY_PERIODIC; 4955cfb853baSMatthew G. Knepley } 4956cfb853baSMatthew G. Knepley n = dim; 4957cfb853baSMatthew G. Knepley PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", edges, &n, &flg)); 4958cfb853baSMatthew G. Knepley n = dim; 4959cfb853baSMatthew G. Knepley PetscCall(PetscOptionsRealArray("-dm_plex_box_lower", "Lower left corner of box", "", lower, &n, &flg)); 4960cfb853baSMatthew 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); 4961cfb853baSMatthew G. Knepley n = dim; 4962cfb853baSMatthew G. Knepley PetscCall(PetscOptionsRealArray("-dm_plex_box_upper", "Upper right corner of box", "", upper, &n, &flg)); 4963cfb853baSMatthew 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); 4964cfb853baSMatthew G. Knepley n = dim; 4965cfb853baSMatthew G. Knepley PetscCall(PetscOptionsEnumArray("-dm_plex_box_bd", "Boundary type for each dimension", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg)); 4966cfb853baSMatthew 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); 49678d2ec52aSSatish Balay PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", overlap, &overlap, NULL, 0)); 49688d2ec52aSSatish Balay PetscCall(DMPlexCreateHypercubicMesh_Internal(dm, dim, lower, upper, edges, overlap, bdt)); 4969cfb853baSMatthew G. Knepley PetscCall(PetscFree4(edges, lower, upper, bdt)); 4970cfb853baSMatthew G. Knepley } break; 4971d71ae5a4SJacob Faibussowitsch default: 4972d71ae5a4SJacob Faibussowitsch SETERRQ(comm, PETSC_ERR_SUP, "Domain shape %s is unsupported", DMPlexShapes[shape]); 49739318fe57SMatthew G. Knepley } 49749318fe57SMatthew G. Knepley } 49759566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE)); 497648a46eb9SPierre Jolivet if (!((PetscObject)dm)->name && nameflg) PetscCall(PetscObjectSetName((PetscObject)dm, plexname)); 4977b9da1bb3SMatthew G. Knepley if (orient) PetscCall(DMPlexOrient(dm)); 49784e22dd4cSMatthew G. Knepley // Allow label creation 49794e22dd4cSMatthew G. Knepley PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_label_", &option, NULL, &flg)); 49804e22dd4cSMatthew G. Knepley if (flg) { 49814e22dd4cSMatthew G. Knepley DMLabel label; 49824e22dd4cSMatthew G. Knepley PetscInt points[1024], n = 1024; 49834e22dd4cSMatthew G. Knepley char fulloption[PETSC_MAX_PATH_LEN]; 49844e22dd4cSMatthew G. Knepley const char *name = &option[14]; 49854e22dd4cSMatthew G. Knepley 49864e22dd4cSMatthew G. Knepley PetscCall(DMCreateLabel(dm, name)); 49874e22dd4cSMatthew G. Knepley PetscCall(DMGetLabel(dm, name, &label)); 49884e22dd4cSMatthew G. Knepley fulloption[0] = '-'; 49894e22dd4cSMatthew G. Knepley fulloption[1] = 0; 49904e22dd4cSMatthew G. Knepley PetscCall(PetscStrlcat(fulloption, option, PETSC_MAX_PATH_LEN)); 49914e22dd4cSMatthew G. Knepley PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, NULL)); 49924e22dd4cSMatthew G. Knepley for (PetscInt p = 0; p < n; ++p) PetscCall(DMLabelSetValue(label, points[p], 1)); 49934e22dd4cSMatthew G. Knepley } 4994dd0eeac9SMatthew G. Knepley // Allow cohesive label creation 4995dd0eeac9SMatthew G. Knepley // Faces are input, completed, and all points are marked with their depth 4996dd0eeac9SMatthew G. Knepley PetscCall(PetscOptionsFindPairPrefix_Private(NULL, ((PetscObject)dm)->prefix, "-dm_plex_cohesive_label_", &option, NULL, &flg)); 4997dd0eeac9SMatthew G. Knepley if (flg) { 4998dd0eeac9SMatthew G. Knepley DMLabel label; 4999dd0eeac9SMatthew G. Knepley PetscInt points[1024], n, pStart, pEnd, Nl = 1; 50005390be7dSMatthew G. Knepley PetscBool noCreate = PETSC_FALSE; 5001dd0eeac9SMatthew G. Knepley char fulloption[PETSC_MAX_PATH_LEN]; 5002dd0eeac9SMatthew G. Knepley char name[PETSC_MAX_PATH_LEN]; 5003dd0eeac9SMatthew G. Knepley size_t len; 5004dd0eeac9SMatthew G. Knepley 5005dd0eeac9SMatthew G. Knepley PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 5006dd0eeac9SMatthew G. Knepley PetscCall(PetscStrncpy(name, &option[23], PETSC_MAX_PATH_LEN)); 5007dd0eeac9SMatthew G. Knepley PetscCall(PetscStrlen(name, &len)); 5008dd0eeac9SMatthew G. Knepley if (name[len - 1] == '0') Nl = 10; 5009dd0eeac9SMatthew G. Knepley for (PetscInt l = 0; l < Nl; ++l) { 50106497c311SBarry Smith if (l > 0) name[len - 1] = (char)('0' + l); 5011dd0eeac9SMatthew G. Knepley fulloption[0] = 0; 5012dd0eeac9SMatthew G. Knepley PetscCall(PetscStrlcat(fulloption, "-dm_plex_cohesive_label_", 32)); 5013dd0eeac9SMatthew G. Knepley PetscCall(PetscStrlcat(fulloption, name, PETSC_MAX_PATH_LEN - 32)); 5014dd0eeac9SMatthew G. Knepley n = 1024; 5015dd0eeac9SMatthew G. Knepley PetscCall(PetscOptionsGetIntArray(NULL, ((PetscObject)dm)->prefix, fulloption, points, &n, &flg)); 5016dd0eeac9SMatthew G. Knepley if (!flg) break; 50175390be7dSMatthew G. Knepley PetscCall(DMHasLabel(dm, name, &noCreate)); 50185390be7dSMatthew G. Knepley if (noCreate) { 50195390be7dSMatthew G. Knepley DMLabel inlabel; 50205390be7dSMatthew G. Knepley IS pointIS; 50215390be7dSMatthew G. Knepley const PetscInt *lpoints; 50225390be7dSMatthew G. Knepley PetscInt pdep, ln, inval = points[0]; 50235390be7dSMatthew G. Knepley char newname[PETSC_MAX_PATH_LEN]; 50245390be7dSMatthew G. Knepley 50255390be7dSMatthew G. Knepley PetscCheck(n == 1, comm, PETSC_ERR_ARG_WRONG, "Must specify a label value with this option"); 50265390be7dSMatthew G. Knepley PetscCall(DMGetLabel(dm, name, &inlabel)); 50275390be7dSMatthew G. Knepley PetscCall(DMLabelGetStratumIS(inlabel, inval, &pointIS)); 50285390be7dSMatthew G. Knepley PetscCall(ISGetLocalSize(pointIS, &ln)); 50295390be7dSMatthew G. Knepley PetscCall(ISGetIndices(pointIS, &lpoints)); 50305390be7dSMatthew G. Knepley PetscCall(DMPlexGetPointDepth(dm, lpoints[0], &pdep)); 50315390be7dSMatthew G. Knepley PetscCall(PetscSNPrintf(newname, PETSC_MAX_PATH_LEN, "%s%" PetscInt_FMT, name, points[0])); 50325390be7dSMatthew G. Knepley PetscCall(DMCreateLabel(dm, newname)); 50335390be7dSMatthew G. Knepley PetscCall(DMGetLabel(dm, newname, &label)); 50345390be7dSMatthew G. Knepley if (!pdep) PetscCall(ProcessCohesiveLabel_Vertices(dm, label, inlabel, inval, ln, lpoints)); 50355390be7dSMatthew G. Knepley else PetscCall(ProcessCohesiveLabel_Faces(dm, label, ln, lpoints)); 50365390be7dSMatthew G. Knepley PetscCall(ISRestoreIndices(pointIS, &lpoints)); 50375390be7dSMatthew G. Knepley PetscCall(ISDestroy(&pointIS)); 50385390be7dSMatthew G. Knepley } else { 5039dd0eeac9SMatthew G. Knepley PetscCall(DMCreateLabel(dm, name)); 5040dd0eeac9SMatthew G. Knepley PetscCall(DMGetLabel(dm, name, &label)); 5041dd0eeac9SMatthew G. Knepley if (pStart >= pEnd) n = 0; 50425390be7dSMatthew G. Knepley PetscCall(ProcessCohesiveLabel_Faces(dm, label, n, points)); 5043dd0eeac9SMatthew G. Knepley } 5044dd0eeac9SMatthew G. Knepley PetscCall(DMPlexOrientLabel(dm, label)); 50450542aa8cSMatthew G. Knepley PetscCall(DMPlexLabelCohesiveComplete(dm, label, NULL, 1, PETSC_FALSE, PETSC_FALSE, NULL)); 5046dd0eeac9SMatthew G. Knepley } 5047dd0eeac9SMatthew G. Knepley } 50485390be7dSMatthew G. Knepley PetscCall(DMViewFromOptions(dm, NULL, "-created_dm_view")); 5049708be2fdSJed Brown PetscCall(PetscLogEventEnd(DMPLEX_CreateFromOptions, dm, 0, 0, 0)); 50503ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 50510a6ba040SMatthew G. Knepley } 50520a6ba040SMatthew G. Knepley 5053ce78bad3SBarry Smith PetscErrorCode DMSetFromOptions_NonRefinement_Plex(DM dm, PetscOptionItems PetscOptionsObject) 5054d71ae5a4SJacob Faibussowitsch { 50550a6ba040SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *)dm->data; 50567f9d8d6cSVaclav Hapla PetscBool flg, flg2; 50579318fe57SMatthew G. Knepley char bdLabel[PETSC_MAX_PATH_LEN]; 5058adc21957SMatthew G. Knepley char method[PETSC_MAX_PATH_LEN]; 50590a6ba040SMatthew G. Knepley 50600a6ba040SMatthew G. Knepley PetscFunctionBegin; 50610a6ba040SMatthew G. Knepley /* Handle viewing */ 50629566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMPlexMatSetClosure", PETSC_FALSE, &mesh->printSetValues, NULL)); 50635962854dSMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fem", "Debug output level for all fem computations", "DMPlexSNESComputeResidualFEM", 0, &mesh->printFEM, NULL, 0)); 50645962854dSMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_plex_print_fvm", "Debug output level for all fvm computations", "DMPlexSNESComputeResidualFVM", 0, &mesh->printFVM, NULL, 0)); 50659566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMPlexSNESComputeResidualFEM", mesh->printTol, &mesh->printTol, NULL)); 50669566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_plex_print_l2", "Debug output level all L2 diff computations", "DMComputeL2Diff", 0, &mesh->printL2, NULL, 0)); 5067f5867de0SMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_plex_print_locate", "Debug output level all point location computations", "DMLocatePoints", 0, &mesh->printLocate, NULL, 0)); 5068a77a5016SMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_plex_print_project", "Debug output level all projection computations", "DMPlexProject", 0, &mesh->printProject, NULL, 0)); 50699566063dSJacob Faibussowitsch PetscCall(DMMonitorSetFromOptions(dm, "-dm_plex_monitor_throughput", "Monitor the simulation throughput", "DMPlexMonitorThroughput", DMPlexMonitorThroughput, NULL, &flg)); 50709566063dSJacob Faibussowitsch if (flg) PetscCall(PetscLogDefaultBegin()); 50715e2c5519SMatthew G. Knepley // Interpolation 50725e2c5519SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_plex_interpolate_prefer_tensor", "When different orderings exist, prefer the tensor order", "DMPlexSetInterpolationPreferTensor", mesh->interpolatePreferTensor, &mesh->interpolatePreferTensor, NULL)); 50739318fe57SMatthew G. Knepley /* Labeling */ 50749566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-dm_plex_boundary_label", "Label to mark the mesh boundary", "", bdLabel, bdLabel, sizeof(bdLabel), &flg)); 50759566063dSJacob Faibussowitsch if (flg) PetscCall(DMPlexCreateBoundaryLabel_Private(dm, bdLabel)); 5076953fc75cSMatthew G. Knepley /* Point Location */ 50779566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMInterpolate", PETSC_FALSE, &mesh->useHashLocation, NULL)); 50780848f4b5SMatthew G. Knepley /* Partitioning and distribution */ 50799566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_partition_balance", "Attempt to evenly divide points on partition boundary between processes", "DMPlexSetPartitionBalance", PETSC_FALSE, &mesh->partitionBalance, NULL)); 5080d02c7345SMatthew G. Knepley /* Reordering */ 5081adc21957SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_reorder_section", "Compute point permutation for local section", "DMReorderSectionSetDefault", PETSC_FALSE, &flg2, &flg)); 5082adc21957SMatthew G. Knepley if (flg) PetscCall(DMReorderSectionSetDefault(dm, flg2 ? DM_REORDER_DEFAULT_TRUE : DM_REORDER_DEFAULT_FALSE)); 5083adc21957SMatthew G. Knepley PetscCall(PetscOptionsString("-dm_reorder_section_type", "Reordering method for local section", "DMReorderSectionSetType", method, method, PETSC_MAX_PATH_LEN, &flg)); 5084adc21957SMatthew G. Knepley if (flg) PetscCall(DMReorderSectionSetType(dm, method)); 50852e62ab5aSMatthew G. Knepley /* Generation and remeshing */ 50869566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_remesh_bd", "Allow changes to the boundary on remeshing", "DMAdapt", PETSC_FALSE, &mesh->remeshBd, NULL)); 508761f058f9SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_plex_save_transform", "Save the transform producing this mesh", "DMAdapt", PETSC_FALSE, &mesh->saveTransform, NULL)); 5088b29cfa1cSToby Isaac /* Projection behavior */ 5089d5b43468SJose E. Roman PetscCall(PetscOptionsBoundedInt("-dm_plex_max_projection_height", "Maximum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL, 0)); 50909566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_regular_refinement", "Use special nested projection algorithm for regular refinement", "DMPlexSetRegularRefinement", mesh->regularRefinement, &mesh->regularRefinement, NULL)); 5091f12cf164SMatthew G. Knepley /* Checking structure */ 5092f12cf164SMatthew G. Knepley { 50937f9d8d6cSVaclav Hapla PetscBool all = PETSC_FALSE; 5094f12cf164SMatthew G. Knepley 50957f9d8d6cSVaclav Hapla PetscCall(PetscOptionsBool("-dm_plex_check_all", "Perform all basic checks", "DMPlexCheck", PETSC_FALSE, &all, NULL)); 50967f9d8d6cSVaclav Hapla if (all) { 50977f9d8d6cSVaclav Hapla PetscCall(DMPlexCheck(dm)); 50987f9d8d6cSVaclav Hapla } else { 50999566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_check_symmetry", "Check that the adjacency information in the mesh is symmetric", "DMPlexCheckSymmetry", PETSC_FALSE, &flg, &flg2)); 51007f9d8d6cSVaclav Hapla if (flg && flg2) PetscCall(DMPlexCheckSymmetry(dm)); 51019566063dSJacob 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)); 51027f9d8d6cSVaclav Hapla if (flg && flg2) PetscCall(DMPlexCheckSkeleton(dm, 0)); 51039566063dSJacob 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)); 51047f9d8d6cSVaclav Hapla if (flg && flg2) PetscCall(DMPlexCheckFaces(dm, 0)); 51059566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_check_geometry", "Check that cells have positive volume", "DMPlexCheckGeometry", PETSC_FALSE, &flg, &flg2)); 51067f9d8d6cSVaclav Hapla if (flg && flg2) PetscCall(DMPlexCheckGeometry(dm)); 51079566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_check_pointsf", "Check some necessary conditions for PointSF", "DMPlexCheckPointSF", PETSC_FALSE, &flg, &flg2)); 5108d7d32a9aSMatthew G. Knepley if (flg && flg2) PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE)); 51099566063dSJacob 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)); 51107f9d8d6cSVaclav Hapla if (flg && flg2) PetscCall(DMPlexCheckInterfaceCones(dm)); 51117f9d8d6cSVaclav Hapla } 51129566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_check_cell_shape", "Check cell shape", "DMPlexCheckCellShape", PETSC_FALSE, &flg, &flg2)); 51139566063dSJacob Faibussowitsch if (flg && flg2) PetscCall(DMPlexCheckCellShape(dm, PETSC_TRUE, PETSC_DETERMINE)); 5114f12cf164SMatthew G. Knepley } 51159318fe57SMatthew G. Knepley { 51169318fe57SMatthew G. Knepley PetscReal scale = 1.0; 51174f3833eaSMatthew G. Knepley 51189566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_plex_scale", "Scale factor for mesh coordinates", "DMPlexScale", scale, &scale, &flg)); 51199318fe57SMatthew G. Knepley if (flg) { 51209318fe57SMatthew G. Knepley Vec coordinates, coordinatesLocal; 51219318fe57SMatthew G. Knepley 51229566063dSJacob Faibussowitsch PetscCall(DMGetCoordinates(dm, &coordinates)); 51239566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal)); 51249566063dSJacob Faibussowitsch PetscCall(VecScale(coordinates, scale)); 51259566063dSJacob Faibussowitsch PetscCall(VecScale(coordinatesLocal, scale)); 51269318fe57SMatthew G. Knepley } 51279318fe57SMatthew G. Knepley } 51289566063dSJacob Faibussowitsch PetscCall(PetscPartitionerSetFromOptions(mesh->partitioner)); 51293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 513068d4fef7SMatthew G. Knepley } 513168d4fef7SMatthew G. Knepley 5132ce78bad3SBarry Smith PetscErrorCode DMSetFromOptions_Overlap_Plex(DM dm, PetscOptionItems PetscOptionsObject, PetscInt *overlap) 5133d71ae5a4SJacob Faibussowitsch { 5134c506a872SMatthew G. Knepley PetscInt numOvLabels = 16, numOvExLabels = 16; 5135c506a872SMatthew G. Knepley char *ovLabelNames[16], *ovExLabelNames[16]; 5136c506a872SMatthew G. Knepley PetscInt numOvValues = 16, numOvExValues = 16, l; 5137c506a872SMatthew G. Knepley PetscBool flg; 5138c506a872SMatthew G. Knepley 5139c506a872SMatthew G. Knepley PetscFunctionBegin; 5140c506a872SMatthew G. Knepley PetscCall(PetscOptionsBoundedInt("-dm_distribute_overlap", "The size of the overlap halo", "DMPlexDistribute", *overlap, overlap, NULL, 0)); 5141c506a872SMatthew G. Knepley PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_labels", "List of overlap label names", "DMPlexDistribute", ovLabelNames, &numOvLabels, &flg)); 5142c506a872SMatthew G. Knepley if (!flg) numOvLabels = 0; 5143c506a872SMatthew G. Knepley if (numOvLabels) { 5144c506a872SMatthew G. Knepley ((DM_Plex *)dm->data)->numOvLabels = numOvLabels; 5145c506a872SMatthew G. Knepley for (l = 0; l < numOvLabels; ++l) { 5146c506a872SMatthew G. Knepley PetscCall(DMGetLabel(dm, ovLabelNames[l], &((DM_Plex *)dm->data)->ovLabels[l])); 5147c506a872SMatthew G. Knepley PetscCheck(((DM_Plex *)dm->data)->ovLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovLabelNames[l]); 5148c506a872SMatthew G. Knepley PetscCall(PetscFree(ovLabelNames[l])); 5149c506a872SMatthew G. Knepley } 5150c506a872SMatthew G. Knepley PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_values", "List of overlap label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovValues, &numOvValues, &flg)); 5151c506a872SMatthew G. Knepley if (!flg) numOvValues = 0; 5152c506a872SMatthew 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); 5153c506a872SMatthew G. Knepley 5154c506a872SMatthew G. Knepley PetscCall(PetscOptionsStringArray("-dm_distribute_overlap_exclude_labels", "List of overlap exclude label names", "DMPlexDistribute", ovExLabelNames, &numOvExLabels, &flg)); 5155c506a872SMatthew G. Knepley if (!flg) numOvExLabels = 0; 5156c506a872SMatthew G. Knepley ((DM_Plex *)dm->data)->numOvExLabels = numOvExLabels; 5157c506a872SMatthew G. Knepley for (l = 0; l < numOvExLabels; ++l) { 5158c506a872SMatthew G. Knepley PetscCall(DMGetLabel(dm, ovExLabelNames[l], &((DM_Plex *)dm->data)->ovExLabels[l])); 5159c506a872SMatthew G. Knepley PetscCheck(((DM_Plex *)dm->data)->ovExLabels[l], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid label name %s", ovExLabelNames[l]); 5160c506a872SMatthew G. Knepley PetscCall(PetscFree(ovExLabelNames[l])); 5161c506a872SMatthew G. Knepley } 5162c506a872SMatthew G. Knepley PetscCall(PetscOptionsIntArray("-dm_distribute_overlap_exclude_values", "List of overlap exclude label values", "DMPlexDistribute", ((DM_Plex *)dm->data)->ovExValues, &numOvExValues, &flg)); 5163c506a872SMatthew G. Knepley if (!flg) numOvExValues = 0; 5164c506a872SMatthew 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); 5165c506a872SMatthew G. Knepley } 51663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5167c506a872SMatthew G. Knepley } 5168c506a872SMatthew G. Knepley 5169ce78bad3SBarry Smith static PetscErrorCode DMSetFromOptions_Plex(DM dm, PetscOptionItems PetscOptionsObject) 5170d71ae5a4SJacob Faibussowitsch { 5171bdf63967SMatthew G. Knepley PetscFunctionList ordlist; 5172bdf63967SMatthew G. Knepley char oname[256]; 51734e22dd4cSMatthew G. Knepley char sublabelname[PETSC_MAX_PATH_LEN] = ""; 5174adc21957SMatthew G. Knepley DMReorderDefaultFlag reorder; 5175d410b0cfSMatthew G. Knepley PetscReal volume = -1.0; 51769318fe57SMatthew G. Knepley PetscInt prerefine = 0, refine = 0, r, coarsen = 0, overlap = 0, extLayers = 0, dim; 5177530e699aSMatthew 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; 517868d4fef7SMatthew G. Knepley 517968d4fef7SMatthew G. Knepley PetscFunctionBegin; 5180d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "DMPlex Options"); 5181dd4c3f67SMatthew G. Knepley if (dm->cloneOpts) goto non_refine; 51829318fe57SMatthew G. Knepley /* Handle automatic creation */ 51839566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 5184530e699aSMatthew G. Knepley if (dim < 0) PetscCall(DMPlexCreateFromOptions_Internal(PetscOptionsObject, &coordSpace, dm)); 51856bc1bd01Sksagiyam PetscCall(DMGetDimension(dm, &dim)); 5186d89e6e46SMatthew G. Knepley /* Handle interpolation before distribution */ 51879566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_interpolate_pre", "Flag to interpolate mesh before distribution", "", interpolate, &interpolate, &flg)); 5188d89e6e46SMatthew G. Knepley if (flg) { 5189d89e6e46SMatthew G. Knepley DMPlexInterpolatedFlag interpolated; 5190d89e6e46SMatthew G. Knepley 51919566063dSJacob Faibussowitsch PetscCall(DMPlexIsInterpolated(dm, &interpolated)); 5192d89e6e46SMatthew G. Knepley if (interpolated == DMPLEX_INTERPOLATED_FULL && !interpolate) { 5193d89e6e46SMatthew G. Knepley DM udm; 5194d89e6e46SMatthew G. Knepley 51959566063dSJacob Faibussowitsch PetscCall(DMPlexUninterpolate(dm, &udm)); 519669d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &udm)); 5197d89e6e46SMatthew G. Knepley } else if (interpolated != DMPLEX_INTERPOLATED_FULL && interpolate) { 5198d89e6e46SMatthew G. Knepley DM idm; 5199d89e6e46SMatthew G. Knepley 52009566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(dm, &idm)); 520169d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &idm)); 5202d89e6e46SMatthew G. Knepley } 5203d89e6e46SMatthew G. Knepley } 52044e22dd4cSMatthew G. Knepley // Handle submesh selection before distribution 52054e22dd4cSMatthew G. Knepley PetscCall(PetscOptionsString("-dm_plex_submesh", "Label to use for submesh selection", "", sublabelname, sublabelname, PETSC_MAX_PATH_LEN, &flg)); 52064e22dd4cSMatthew G. Knepley if (flg) { 52074e22dd4cSMatthew G. Knepley DM subdm; 52084e22dd4cSMatthew G. Knepley DMLabel label; 52094e22dd4cSMatthew G. Knepley IS valueIS, pointIS; 52104e22dd4cSMatthew G. Knepley const PetscInt *values, *points; 52114e22dd4cSMatthew G. Knepley PetscBool markedFaces = PETSC_FALSE; 52124e22dd4cSMatthew G. Knepley PetscInt Nv, value, Np; 52134e22dd4cSMatthew G. Knepley 52144e22dd4cSMatthew G. Knepley PetscCall(DMGetLabel(dm, sublabelname, &label)); 52154e22dd4cSMatthew G. Knepley PetscCall(DMLabelGetNumValues(label, &Nv)); 52164e22dd4cSMatthew 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); 52174e22dd4cSMatthew G. Knepley PetscCall(DMLabelGetValueIS(label, &valueIS)); 52184e22dd4cSMatthew G. Knepley PetscCall(ISGetIndices(valueIS, &values)); 52194e22dd4cSMatthew G. Knepley value = values[0]; 52204e22dd4cSMatthew G. Knepley PetscCall(ISRestoreIndices(valueIS, &values)); 52214e22dd4cSMatthew G. Knepley PetscCall(ISDestroy(&valueIS)); 52224e22dd4cSMatthew G. Knepley PetscCall(DMLabelGetStratumSize(label, value, &Np)); 52234e22dd4cSMatthew G. Knepley PetscCall(DMLabelGetStratumIS(label, value, &pointIS)); 52244e22dd4cSMatthew G. Knepley PetscCall(ISGetIndices(pointIS, &points)); 52254e22dd4cSMatthew G. Knepley for (PetscInt p = 0; p < Np; ++p) { 52264e22dd4cSMatthew G. Knepley PetscInt pdepth; 52274e22dd4cSMatthew G. Knepley 52284e22dd4cSMatthew G. Knepley PetscCall(DMPlexGetPointDepth(dm, points[p], &pdepth)); 52294e22dd4cSMatthew G. Knepley if (pdepth) { 52304e22dd4cSMatthew G. Knepley markedFaces = PETSC_TRUE; 52314e22dd4cSMatthew G. Knepley break; 52324e22dd4cSMatthew G. Knepley } 52334e22dd4cSMatthew G. Knepley } 52344e22dd4cSMatthew G. Knepley PetscCall(ISRestoreIndices(pointIS, &points)); 52354e22dd4cSMatthew G. Knepley PetscCall(ISDestroy(&pointIS)); 52364e22dd4cSMatthew G. Knepley PetscCall(DMPlexCreateSubmesh(dm, label, value, markedFaces, &subdm)); 52374e22dd4cSMatthew G. Knepley PetscCall(DMPlexReplace_Internal(dm, &subdm)); 52384e22dd4cSMatthew G. Knepley PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 52394e22dd4cSMatthew G. Knepley } 52409b44eab4SMatthew G. Knepley /* Handle DMPlex refinement before distribution */ 52419566063dSJacob Faibussowitsch PetscCall(DMPlexGetRefinementUniform(dm, &uniformOrig)); 52429566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_refine_pre", "The number of refinements before distribution", "DMCreate", prerefine, &prerefine, NULL, 0)); 52439566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_refine_remap_pre", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL)); 52449566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_refine_uniform_pre", "Flag for uniform refinement before distribution", "DMCreate", uniform, &uniform, &flg)); 52459566063dSJacob Faibussowitsch if (flg) PetscCall(DMPlexSetRefinementUniform(dm, uniform)); 52469566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-dm_refine_volume_limit_pre", "The maximum cell volume after refinement before distribution", "DMCreate", volume, &volume, &flg)); 52479318fe57SMatthew G. Knepley if (flg) { 52489566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(dm, PETSC_FALSE)); 52499566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementLimit(dm, volume)); 52509318fe57SMatthew G. Knepley prerefine = PetscMax(prerefine, 1); 52519318fe57SMatthew G. Knepley } 5252b23db253SStefano Zampini if (prerefine) PetscCall(DMLocalizeCoordinates(dm)); 52539b44eab4SMatthew G. Knepley for (r = 0; r < prerefine; ++r) { 52549b44eab4SMatthew G. Knepley DM rdm; 5255*509b31aaSMatthew G. Knepley PetscPointFunc coordFunc; 52569b44eab4SMatthew G. Knepley 5257*509b31aaSMatthew G. Knepley PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc)); 5258dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 52599566063dSJacob Faibussowitsch PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm)); 526069d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &rdm)); 5261dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 526261a622f3SMatthew G. Knepley if (coordFunc && remap) { 52639566063dSJacob Faibussowitsch PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc)); 5264*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, coordFunc)); 52659b44eab4SMatthew G. Knepley } 52669b44eab4SMatthew G. Knepley } 52679566063dSJacob Faibussowitsch PetscCall(DMPlexSetRefinementUniform(dm, uniformOrig)); 52689318fe57SMatthew G. Knepley /* Handle DMPlex extrusion before distribution */ 52699566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_extrude", "The number of layers to extrude", "", extLayers, &extLayers, NULL, 0)); 52709318fe57SMatthew G. Knepley if (extLayers) { 52719318fe57SMatthew G. Knepley DM edm; 52729318fe57SMatthew G. Knepley 52739566063dSJacob Faibussowitsch PetscCall(DMExtrude(dm, extLayers, &edm)); 527469d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &edm)); 5275*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, NULL)); 5276dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 5277d410b0cfSMatthew G. Knepley extLayers = 0; 52785e17fc22SAidan Hamilton PetscCall(DMGetDimension(dm, &dim)); 52799318fe57SMatthew G. Knepley } 5280bdf63967SMatthew G. Knepley /* Handle DMPlex reordering before distribution */ 52816bc1bd01Sksagiyam PetscCall(DMPlexReorderGetDefault(dm, &reorder)); 52829566063dSJacob Faibussowitsch PetscCall(MatGetOrderingList(&ordlist)); 52836bc1bd01Sksagiyam PetscCall(PetscStrncpy(oname, MATORDERINGNATURAL, sizeof(oname))); 52849566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-dm_plex_reorder", "Set mesh reordering type", "DMPlexGetOrdering", ordlist, MATORDERINGNATURAL, oname, sizeof(oname), &flg)); 5285adc21957SMatthew G. Knepley if (reorder == DM_REORDER_DEFAULT_TRUE || flg) { 5286bdf63967SMatthew G. Knepley DM pdm; 5287bdf63967SMatthew G. Knepley IS perm; 5288bdf63967SMatthew G. Knepley 52899566063dSJacob Faibussowitsch PetscCall(DMPlexGetOrdering(dm, oname, NULL, &perm)); 52909566063dSJacob Faibussowitsch PetscCall(DMPlexPermute(dm, perm, &pdm)); 52919566063dSJacob Faibussowitsch PetscCall(ISDestroy(&perm)); 529269d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &pdm)); 5293dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 5294bdf63967SMatthew G. Knepley } 52959b44eab4SMatthew G. Knepley /* Handle DMPlex distribution */ 52969566063dSJacob Faibussowitsch PetscCall(DMPlexDistributeGetDefault(dm, &distribute)); 5297c506a872SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_distribute", "Flag to redistribute a mesh among processes", "DMPlexDistribute", distribute, &distribute, NULL)); 5298a286e215SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_distribute_save_sf", "Flag to save the migration SF", "DMPlexSetMigrationSF", saveSF, &saveSF, NULL)); 5299dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_Overlap_Plex(dm, PetscOptionsObject, &overlap)); 53009b44eab4SMatthew G. Knepley if (distribute) { 53019b44eab4SMatthew G. Knepley DM pdm = NULL; 53029b44eab4SMatthew G. Knepley PetscPartitioner part; 5303a286e215SMatthew G. Knepley PetscSF sfMigration; 53049b44eab4SMatthew G. Knepley 53059566063dSJacob Faibussowitsch PetscCall(DMPlexGetPartitioner(dm, &part)); 53069566063dSJacob Faibussowitsch PetscCall(PetscPartitionerSetFromOptions(part)); 5307a286e215SMatthew G. Knepley PetscCall(DMPlexDistribute(dm, overlap, &sfMigration, &pdm)); 53085d2873a6SJames Wright if (pdm) { 53095d2873a6SJames Wright // Delete the local section to force the existing one to be rebuilt with the distributed DM 53105d2873a6SJames Wright PetscCall(DMSetLocalSection(dm, pdm->localSection)); 53115d2873a6SJames Wright PetscCall(DMPlexReplace_Internal(dm, &pdm)); 53125d2873a6SJames Wright } 5313a286e215SMatthew G. Knepley if (saveSF) PetscCall(DMPlexSetMigrationSF(dm, sfMigration)); 5314a286e215SMatthew G. Knepley PetscCall(PetscSFDestroy(&sfMigration)); 53159b44eab4SMatthew G. Knepley } 53164054ae39SJames Wright 53174054ae39SJames Wright { 53184054ae39SJames Wright PetscBool useBoxLabel = PETSC_FALSE; 53194054ae39SJames Wright PetscCall(PetscOptionsBool("-dm_plex_box_label", "Create 'Face Sets' assuming boundary faces align with cartesian directions", "DMCreate", useBoxLabel, &useBoxLabel, NULL)); 5320d7d2d1d2SJames Wright if (useBoxLabel) { 5321d7d2d1d2SJames Wright PetscInt n = 3; 5322d7d2d1d2SJames Wright DMBoundaryType bdt[3] = {DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE}; 5323d7d2d1d2SJames Wright 5324d7d2d1d2SJames Wright PetscCall(PetscOptionsEnumArray("-dm_plex_box_label_bd", "Boundary type for each dimension when using -dm_plex_box_label", "", DMBoundaryTypes, (PetscEnum *)bdt, &n, &flg)); 5325d7d2d1d2SJames 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); 5326d7d2d1d2SJames Wright PetscCall(DMPlexSetBoxLabel_Internal(dm, bdt)); 5327d7d2d1d2SJames Wright } 53284054ae39SJames Wright } 5329d2b2dc1eSMatthew G. Knepley /* Must check CEED options before creating function space for coordinates */ 5330d2b2dc1eSMatthew G. Knepley { 5331d2b2dc1eSMatthew G. Knepley PetscBool useCeed = PETSC_FALSE, flg; 5332d2b2dc1eSMatthew G. Knepley 5333d2b2dc1eSMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_plex_use_ceed", "Use LibCEED as the FEM backend", "DMPlexSetUseCeed", useCeed, &useCeed, &flg)); 5334d2b2dc1eSMatthew G. Knepley if (flg) PetscCall(DMPlexSetUseCeed(dm, useCeed)); 5335d2b2dc1eSMatthew G. Knepley } 53369318fe57SMatthew G. Knepley /* Create coordinate space */ 5337530e699aSMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_coord_space", "Use an FEM space for coordinates", "", coordSpace, &coordSpace, NULL)); 5338530e699aSMatthew G. Knepley if (coordSpace) { 5339e44f6aebSMatthew G. Knepley PetscInt degree = 1, deg; 53405515ebd3SMatthew G. Knepley PetscInt height = 0; 53415515ebd3SMatthew G. Knepley DM cdm; 5342530e699aSMatthew G. Knepley PetscBool localize = PETSC_TRUE, sparseLocalize = PETSC_TRUE; 53439318fe57SMatthew G. Knepley 53449566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-dm_coord_petscspace_degree", "FEM degree for coordinate space", "", degree, °ree, NULL)); 5345e44f6aebSMatthew G. Knepley PetscCall(DMGetCoordinateDegree_Internal(dm, °)); 5346*509b31aaSMatthew G. Knepley if (coordSpace && deg <= 1) PetscCall(DMPlexCreateCoordinateSpace(dm, degree, PETSC_TRUE, NULL)); 53475515ebd3SMatthew G. Knepley PetscCall(DMGetCoordinateDM(dm, &cdm)); 5348530e699aSMatthew G. Knepley if (!coordSpace) { 534961a622f3SMatthew G. Knepley PetscDS cds; 535061a622f3SMatthew G. Knepley PetscObject obj; 535161a622f3SMatthew G. Knepley PetscClassId id; 535261a622f3SMatthew G. Knepley 53539566063dSJacob Faibussowitsch PetscCall(DMGetDS(cdm, &cds)); 53549566063dSJacob Faibussowitsch PetscCall(PetscDSGetDiscretization(cds, 0, &obj)); 53559566063dSJacob Faibussowitsch PetscCall(PetscObjectGetClassId(obj, &id)); 535661a622f3SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 535761a622f3SMatthew G. Knepley PetscContainer dummy; 535861a622f3SMatthew G. Knepley 53599566063dSJacob Faibussowitsch PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &dummy)); 53609566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)dummy, "coordinates")); 53619566063dSJacob Faibussowitsch PetscCall(DMSetField(cdm, 0, NULL, (PetscObject)dummy)); 53629566063dSJacob Faibussowitsch PetscCall(PetscContainerDestroy(&dummy)); 53639566063dSJacob Faibussowitsch PetscCall(DMClearDS(cdm)); 536461a622f3SMatthew G. Knepley } 5365*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, NULL)); 536661a622f3SMatthew G. Knepley } 5367c3db174cSMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_localize", "Localize mesh coordinates", "", localize, &localize, NULL)); 5368c3db174cSMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_sparse_localize", "Localize only necessary cells", "DMSetSparseLocalize", sparseLocalize, &sparseLocalize, &flg)); 5369c3db174cSMatthew G. Knepley if (flg) PetscCall(DMSetSparseLocalize(dm, sparseLocalize)); 53705515ebd3SMatthew G. Knepley PetscCall(PetscOptionsInt("-dm_localize_height", "Localize edges and faces in addition to cells", "", height, &height, &flg)); 53715515ebd3SMatthew G. Knepley if (flg) PetscCall(DMPlexSetMaxProjectionHeight(cdm, height)); 5372c3db174cSMatthew G. Knepley if (localize) PetscCall(DMLocalizeCoordinates(dm)); 53739318fe57SMatthew G. Knepley } 537468d4fef7SMatthew G. Knepley /* Handle DMPlex refinement */ 537561a622f3SMatthew G. Knepley remap = PETSC_TRUE; 53769566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL, 0)); 53779566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_refine_remap", "Flag to control coordinate remapping", "DMCreate", remap, &remap, NULL)); 53789566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy, 0)); 53799566063dSJacob Faibussowitsch if (refine) PetscCall(DMPlexSetRefinementUniform(dm, PETSC_TRUE)); 538068d4fef7SMatthew G. Knepley if (refine && isHierarchy) { 5381acdc6f61SToby Isaac DM *dms, coarseDM; 538268d4fef7SMatthew G. Knepley 53839566063dSJacob Faibussowitsch PetscCall(DMGetCoarseDM(dm, &coarseDM)); 53849566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)coarseDM)); 53859566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(refine, &dms)); 53869566063dSJacob Faibussowitsch PetscCall(DMRefineHierarchy(dm, refine, dms)); 538768d4fef7SMatthew G. Knepley /* Total hack since we do not pass in a pointer */ 53889566063dSJacob Faibussowitsch PetscCall(DMPlexSwap_Static(dm, dms[refine - 1])); 538968d4fef7SMatthew G. Knepley if (refine == 1) { 53909566063dSJacob Faibussowitsch PetscCall(DMSetCoarseDM(dm, dms[0])); 53919566063dSJacob Faibussowitsch PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE)); 539268d4fef7SMatthew G. Knepley } else { 53939566063dSJacob Faibussowitsch PetscCall(DMSetCoarseDM(dm, dms[refine - 2])); 53949566063dSJacob Faibussowitsch PetscCall(DMPlexSetRegularRefinement(dm, PETSC_TRUE)); 53959566063dSJacob Faibussowitsch PetscCall(DMSetCoarseDM(dms[0], dms[refine - 1])); 53969566063dSJacob Faibussowitsch PetscCall(DMPlexSetRegularRefinement(dms[0], PETSC_TRUE)); 539768d4fef7SMatthew G. Knepley } 53989566063dSJacob Faibussowitsch PetscCall(DMSetCoarseDM(dms[refine - 1], coarseDM)); 53999566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)coarseDM)); 540068d4fef7SMatthew G. Knepley /* Free DMs */ 540168d4fef7SMatthew G. Knepley for (r = 0; r < refine; ++r) { 5402dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject)); 54039566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dms[r])); 540468d4fef7SMatthew G. Knepley } 54059566063dSJacob Faibussowitsch PetscCall(PetscFree(dms)); 540668d4fef7SMatthew G. Knepley } else { 540768d4fef7SMatthew G. Knepley for (r = 0; r < refine; ++r) { 54089318fe57SMatthew G. Knepley DM rdm; 5409*509b31aaSMatthew G. Knepley PetscPointFunc coordFunc; 541068d4fef7SMatthew G. Knepley 5411*509b31aaSMatthew G. Knepley PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc)); 5412dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 54139566063dSJacob Faibussowitsch PetscCall(DMRefine(dm, PetscObjectComm((PetscObject)dm), &rdm)); 541468d4fef7SMatthew G. Knepley /* Total hack since we do not pass in a pointer */ 541569d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &rdm)); 5416dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 541761a622f3SMatthew G. Knepley if (coordFunc && remap) { 54189566063dSJacob Faibussowitsch PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc)); 5419*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, coordFunc)); 542051a74b61SMatthew G. Knepley } 542168d4fef7SMatthew G. Knepley } 542268d4fef7SMatthew G. Knepley } 54233cf6fe12SMatthew G. Knepley /* Handle DMPlex coarsening */ 54249566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL, 0)); 54259566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy, 0)); 5426b653a561SMatthew G. Knepley if (coarsen && isHierarchy) { 5427b653a561SMatthew G. Knepley DM *dms; 5428b653a561SMatthew G. Knepley 54299566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(coarsen, &dms)); 54309566063dSJacob Faibussowitsch PetscCall(DMCoarsenHierarchy(dm, coarsen, dms)); 5431b653a561SMatthew G. Knepley /* Free DMs */ 5432b653a561SMatthew G. Knepley for (r = 0; r < coarsen; ++r) { 5433dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dms[r], PetscOptionsObject)); 54349566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dms[r])); 5435b653a561SMatthew G. Knepley } 54369566063dSJacob Faibussowitsch PetscCall(PetscFree(dms)); 5437b653a561SMatthew G. Knepley } else { 5438b653a561SMatthew G. Knepley for (r = 0; r < coarsen; ++r) { 54399318fe57SMatthew G. Knepley DM cdm; 5440*509b31aaSMatthew G. Knepley PetscPointFunc coordFunc; 54413cf6fe12SMatthew G. Knepley 5442*509b31aaSMatthew G. Knepley PetscCall(DMPlexGetCoordinateMap(dm, &coordFunc)); 5443dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 54449566063dSJacob Faibussowitsch PetscCall(DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &cdm)); 54453cf6fe12SMatthew G. Knepley /* Total hack since we do not pass in a pointer */ 544669d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &cdm)); 5447dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 54489318fe57SMatthew G. Knepley if (coordFunc) { 54499566063dSJacob Faibussowitsch PetscCall(DMPlexRemapGeometry(dm, 0.0, coordFunc)); 5450*509b31aaSMatthew G. Knepley PetscCall(DMPlexSetCoordinateMap(dm, coordFunc)); 54519318fe57SMatthew G. Knepley } 54523cf6fe12SMatthew G. Knepley } 5453b653a561SMatthew G. Knepley } 5454be664eb1SMatthew G. Knepley // Handle coordinate remapping 5455be664eb1SMatthew G. Knepley remap = PETSC_FALSE; 5456be664eb1SMatthew G. Knepley PetscCall(PetscOptionsBool("-dm_coord_remap", "Flag to control coordinate remapping", "", remap, &remap, NULL)); 5457be664eb1SMatthew G. Knepley if (remap) { 5458be664eb1SMatthew G. Knepley DMPlexCoordMap map = DM_COORD_MAP_NONE; 5459be664eb1SMatthew G. Knepley PetscPointFunc mapFunc = NULL; 5460be664eb1SMatthew G. Knepley PetscScalar params[16]; 5461f45b553cSPierre Jolivet PetscInt Np = PETSC_STATIC_ARRAY_LENGTH(params), cdim; 5462be664eb1SMatthew G. Knepley MPI_Comm comm; 5463be664eb1SMatthew G. Knepley 5464be664eb1SMatthew G. Knepley PetscCall(PetscObjectGetComm((PetscObject)dm, &comm)); 5465be664eb1SMatthew G. Knepley PetscCall(DMGetCoordinateDim(dm, &cdim)); 5466be664eb1SMatthew G. Knepley PetscCall(PetscOptionsScalarArray("-dm_coord_map_params", "Parameters for the coordinate remapping", "", params, &Np, &flg)); 5467be664eb1SMatthew G. Knepley if (!flg) Np = 0; 5468be664eb1SMatthew G. Knepley // TODO Allow user to pass a map function by name 5469be664eb1SMatthew G. Knepley PetscCall(PetscOptionsEnum("-dm_coord_map", "Coordinate mapping for built-in mesh", "", DMPlexCoordMaps, (PetscEnum)map, (PetscEnum *)&map, &flg)); 5470be664eb1SMatthew G. Knepley if (flg) { 5471be664eb1SMatthew G. Knepley switch (map) { 5472be664eb1SMatthew G. Knepley case DM_COORD_MAP_NONE: 5473be664eb1SMatthew G. Knepley mapFunc = coordMap_identity; 5474be664eb1SMatthew G. Knepley break; 5475be664eb1SMatthew G. Knepley case DM_COORD_MAP_SHEAR: 5476be664eb1SMatthew G. Knepley mapFunc = coordMap_shear; 5477be664eb1SMatthew G. Knepley if (!Np) { 5478be664eb1SMatthew G. Knepley Np = cdim + 1; 5479be664eb1SMatthew G. Knepley params[0] = 0; 5480be664eb1SMatthew G. Knepley for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0; 5481be664eb1SMatthew G. Knepley } 5482be664eb1SMatthew 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); 5483be664eb1SMatthew G. Knepley break; 5484be664eb1SMatthew G. Knepley case DM_COORD_MAP_FLARE: 5485be664eb1SMatthew G. Knepley mapFunc = coordMap_flare; 5486be664eb1SMatthew G. Knepley if (!Np) { 5487be664eb1SMatthew G. Knepley Np = cdim + 1; 5488be664eb1SMatthew G. Knepley params[0] = 0; 5489be664eb1SMatthew G. Knepley for (PetscInt d = 1; d <= cdim; ++d) params[d] = 1.0; 5490be664eb1SMatthew G. Knepley } 5491be664eb1SMatthew 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); 5492be664eb1SMatthew G. Knepley break; 5493be664eb1SMatthew G. Knepley case DM_COORD_MAP_ANNULUS: 5494be664eb1SMatthew G. Knepley mapFunc = coordMap_annulus; 5495be664eb1SMatthew G. Knepley if (!Np) { 5496be664eb1SMatthew G. Knepley Np = 2; 5497be664eb1SMatthew G. Knepley params[0] = 1.; 5498be664eb1SMatthew G. Knepley params[1] = 2.; 5499be664eb1SMatthew G. Knepley } 5500be664eb1SMatthew G. Knepley PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The annulus coordinate map must have 2 parameters, not %" PetscInt_FMT, Np); 5501be664eb1SMatthew G. Knepley break; 5502be664eb1SMatthew G. Knepley case DM_COORD_MAP_SHELL: 5503be664eb1SMatthew G. Knepley mapFunc = coordMap_shell; 5504be664eb1SMatthew G. Knepley if (!Np) { 5505be664eb1SMatthew G. Knepley Np = 2; 5506be664eb1SMatthew G. Knepley params[0] = 1.; 5507be664eb1SMatthew G. Knepley params[1] = 2.; 5508be664eb1SMatthew G. Knepley } 5509be664eb1SMatthew G. Knepley PetscCheck(Np == 2, comm, PETSC_ERR_ARG_WRONG, "The spherical shell coordinate map must have 2 parameters, not %" PetscInt_FMT, Np); 5510be664eb1SMatthew G. Knepley break; 5511530e699aSMatthew G. Knepley case DM_COORD_MAP_SINUSOID: 5512530e699aSMatthew G. Knepley mapFunc = coordMap_sinusoid; 5513530e699aSMatthew G. Knepley if (!Np) { 5514530e699aSMatthew G. Knepley Np = 3; 5515530e699aSMatthew G. Knepley params[0] = 1.; 5516530e699aSMatthew G. Knepley params[1] = 1.; 5517530e699aSMatthew G. Knepley params[2] = 1.; 5518530e699aSMatthew G. Knepley } 5519530e699aSMatthew G. Knepley PetscCheck(Np == 3, comm, PETSC_ERR_ARG_WRONG, "The sinusoidal coordinate map must have 3 parameters, not %" PetscInt_FMT, Np); 5520530e699aSMatthew G. Knepley break; 5521be664eb1SMatthew G. Knepley default: 5522be664eb1SMatthew G. Knepley mapFunc = coordMap_identity; 5523be664eb1SMatthew G. Knepley } 5524be664eb1SMatthew G. Knepley } 5525be664eb1SMatthew G. Knepley if (Np) { 5526be664eb1SMatthew G. Knepley DM cdm; 5527be664eb1SMatthew G. Knepley PetscDS cds; 5528be664eb1SMatthew G. Knepley 5529be664eb1SMatthew G. Knepley PetscCall(DMGetCoordinateDM(dm, &cdm)); 5530be664eb1SMatthew G. Knepley PetscCall(DMGetDS(cdm, &cds)); 5531be664eb1SMatthew G. Knepley PetscCall(PetscDSSetConstants(cds, Np, params)); 5532be664eb1SMatthew G. Knepley } 5533be664eb1SMatthew G. Knepley PetscCall(DMPlexRemapGeometry(dm, 0.0, mapFunc)); 5534be664eb1SMatthew G. Knepley } 5535909dfd52SMatthew G. Knepley /* Handle ghost cells */ 55369566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-dm_plex_create_fv_ghost_cells", "Flag to create finite volume ghost cells on the boundary", "DMCreate", ghostCells, &ghostCells, NULL)); 5537909dfd52SMatthew G. Knepley if (ghostCells) { 5538909dfd52SMatthew G. Knepley DM gdm; 5539909dfd52SMatthew G. Knepley char lname[PETSC_MAX_PATH_LEN]; 5540909dfd52SMatthew G. Knepley 5541909dfd52SMatthew G. Knepley lname[0] = '\0'; 55429566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-dm_plex_fv_ghost_cells_label", "Label name for ghost cells boundary", "DMCreate", lname, lname, sizeof(lname), &flg)); 55439566063dSJacob Faibussowitsch PetscCall(DMPlexConstructGhostCells(dm, flg ? lname : NULL, NULL, &gdm)); 554469d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &gdm)); 5545909dfd52SMatthew G. Knepley } 55466913077dSMatthew G. Knepley /* Handle 1D order */ 5547adc21957SMatthew G. Knepley if (reorder != DM_REORDER_DEFAULT_FALSE && dim == 1) { 55486913077dSMatthew G. Knepley DM cdm, rdm; 55496913077dSMatthew G. Knepley PetscDS cds; 55506913077dSMatthew G. Knepley PetscObject obj; 55516913077dSMatthew G. Knepley PetscClassId id = PETSC_OBJECT_CLASSID; 55526913077dSMatthew G. Knepley IS perm; 55536bc1bd01Sksagiyam PetscInt Nf; 55546913077dSMatthew G. Knepley PetscBool distributed; 55556913077dSMatthew G. Knepley 55569566063dSJacob Faibussowitsch PetscCall(DMPlexIsDistributed(dm, &distributed)); 55579566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDM(dm, &cdm)); 55589566063dSJacob Faibussowitsch PetscCall(DMGetDS(cdm, &cds)); 55599566063dSJacob Faibussowitsch PetscCall(PetscDSGetNumFields(cds, &Nf)); 55606913077dSMatthew G. Knepley if (Nf) { 55619566063dSJacob Faibussowitsch PetscCall(PetscDSGetDiscretization(cds, 0, &obj)); 55629566063dSJacob Faibussowitsch PetscCall(PetscObjectGetClassId(obj, &id)); 55636913077dSMatthew G. Knepley } 55646bc1bd01Sksagiyam if (!distributed && id != PETSCFE_CLASSID) { 55659566063dSJacob Faibussowitsch PetscCall(DMPlexGetOrdering1D(dm, &perm)); 55669566063dSJacob Faibussowitsch PetscCall(DMPlexPermute(dm, perm, &rdm)); 556769d8a87bSksagiyam PetscCall(DMPlexReplace_Internal(dm, &rdm)); 55689566063dSJacob Faibussowitsch PetscCall(ISDestroy(&perm)); 55696913077dSMatthew G. Knepley } 55706913077dSMatthew G. Knepley } 55713cf6fe12SMatthew G. Knepley /* Handle */ 5572dd4c3f67SMatthew G. Knepley non_refine: 5573dbbe0bcdSBarry Smith PetscCall(DMSetFromOptions_NonRefinement_Plex(dm, PetscOptionsObject)); 557422d6dc08SStefano Zampini char *phases[16]; 557522d6dc08SStefano Zampini PetscInt Nphases = 16; 557622d6dc08SStefano Zampini PetscCall(PetscOptionsStringArray("-dm_plex_option_phases", "Option phase prefixes", "DMSetFromOptions", phases, &Nphases, &flg)); 5577d0609cedSBarry Smith PetscOptionsHeadEnd(); 557822d6dc08SStefano Zampini 557922d6dc08SStefano Zampini // Phases 558022d6dc08SStefano Zampini if (flg) { 5581530e699aSMatthew G. Knepley DM cdm; 5582530e699aSMatthew G. Knepley char *oldPrefix, *oldCoordPrefix; 5583530e699aSMatthew G. Knepley const char *tmp; 558422d6dc08SStefano Zampini 5585530e699aSMatthew G. Knepley PetscCall(DMGetCoordinateDM(dm, &cdm)); 5586530e699aSMatthew G. Knepley PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &tmp)); 5587530e699aSMatthew G. Knepley PetscCall(PetscStrallocpy(tmp, &oldPrefix)); 5588530e699aSMatthew G. Knepley PetscCall(PetscObjectGetOptionsPrefix((PetscObject)cdm, &tmp)); 5589530e699aSMatthew G. Knepley PetscCall(PetscStrallocpy(tmp, &oldCoordPrefix)); 559022d6dc08SStefano Zampini for (PetscInt ph = 0; ph < Nphases; ++ph) { 559122d6dc08SStefano Zampini PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)dm, phases[ph])); 5592530e699aSMatthew G. Knepley PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)cdm, phases[ph])); 559322d6dc08SStefano Zampini PetscCall(PetscInfo(dm, "Options phase %s for DM %s\n", phases[ph], dm->hdr.name)); 559422d6dc08SStefano Zampini PetscCall(DMSetFromOptions(dm)); 559522d6dc08SStefano Zampini PetscCall(PetscObjectSetOptionsPrefix((PetscObject)dm, oldPrefix)); 5596530e699aSMatthew G. Knepley PetscCall(DMGetCoordinateDM(dm, &cdm)); 5597530e699aSMatthew G. Knepley PetscCall(PetscObjectSetOptionsPrefix((PetscObject)cdm, oldCoordPrefix)); 559822d6dc08SStefano Zampini PetscCall(PetscFree(phases[ph])); 559922d6dc08SStefano Zampini } 5600530e699aSMatthew G. Knepley PetscCall(PetscFree(oldPrefix)); 5601530e699aSMatthew G. Knepley PetscCall(PetscFree(oldCoordPrefix)); 560222d6dc08SStefano Zampini } 56033ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 56040a6ba040SMatthew G. Knepley } 56050a6ba040SMatthew G. Knepley 5606d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_Plex(DM dm, Vec *vec) 5607d71ae5a4SJacob Faibussowitsch { 5608552f7358SJed Brown PetscFunctionBegin; 56099566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector_Section_Private(dm, vec)); 56109566063dSJacob Faibussowitsch /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */ 56119566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex)); 56129566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_Plex_Native)); 56139566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex)); 56149566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_Plex_Native)); 56153ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5616552f7358SJed Brown } 5617552f7358SJed Brown 5618d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_Plex(DM dm, Vec *vec) 5619d71ae5a4SJacob Faibussowitsch { 5620552f7358SJed Brown PetscFunctionBegin; 56219566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector_Section_Private(dm, vec)); 56229566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_Plex_Local)); 56239566063dSJacob Faibussowitsch PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_Plex_Local)); 56243ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5625552f7358SJed Brown } 5626552f7358SJed Brown 5627d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd) 5628d71ae5a4SJacob Faibussowitsch { 5629793f3fe5SMatthew G. Knepley PetscInt depth, d; 5630793f3fe5SMatthew G. Knepley 5631793f3fe5SMatthew G. Knepley PetscFunctionBegin; 56329566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm, &depth)); 5633793f3fe5SMatthew G. Knepley if (depth == 1) { 56349566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &d)); 56359566063dSJacob Faibussowitsch if (dim == 0) PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd)); 56369566063dSJacob Faibussowitsch else if (dim == d) PetscCall(DMPlexGetDepthStratum(dm, 1, pStart, pEnd)); 56379371c9d4SSatish Balay else { 56389371c9d4SSatish Balay *pStart = 0; 56399371c9d4SSatish Balay *pEnd = 0; 56409371c9d4SSatish Balay } 5641793f3fe5SMatthew G. Knepley } else { 56429566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, dim, pStart, pEnd)); 5643793f3fe5SMatthew G. Knepley } 56443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5645793f3fe5SMatthew G. Knepley } 5646793f3fe5SMatthew G. Knepley 5647d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetNeighbors_Plex(DM dm, PetscInt *nranks, const PetscMPIInt *ranks[]) 5648d71ae5a4SJacob Faibussowitsch { 5649502a2867SDave May PetscSF sf; 56506497c311SBarry Smith PetscMPIInt niranks, njranks; 56516497c311SBarry Smith PetscInt n; 56520a19bb7dSprj- const PetscMPIInt *iranks, *jranks; 56530a19bb7dSprj- DM_Plex *data = (DM_Plex *)dm->data; 5654502a2867SDave May 56552f356facSMatthew G. Knepley PetscFunctionBegin; 56569566063dSJacob Faibussowitsch PetscCall(DMGetPointSF(dm, &sf)); 56570a19bb7dSprj- if (!data->neighbors) { 56589566063dSJacob Faibussowitsch PetscCall(PetscSFSetUp(sf)); 56599566063dSJacob Faibussowitsch PetscCall(PetscSFGetRootRanks(sf, &njranks, &jranks, NULL, NULL, NULL)); 56609566063dSJacob Faibussowitsch PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL)); 56619566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors)); 56629566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(data->neighbors + 1, jranks, njranks)); 56639566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks)); 56640a19bb7dSprj- n = njranks + niranks; 56659566063dSJacob Faibussowitsch PetscCall(PetscSortRemoveDupsMPIInt(&n, data->neighbors + 1)); 56660a19bb7dSprj- /* The following cast should never fail: can't have more neighbors than PETSC_MPI_INT_MAX */ 56679566063dSJacob Faibussowitsch PetscCall(PetscMPIIntCast(n, data->neighbors)); 56680a19bb7dSprj- } 56690a19bb7dSprj- if (nranks) *nranks = data->neighbors[0]; 56700a19bb7dSprj- if (ranks) { 56710a19bb7dSprj- if (data->neighbors[0]) *ranks = data->neighbors + 1; 56720a19bb7dSprj- else *ranks = NULL; 56730a19bb7dSprj- } 56743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5675502a2867SDave May } 5676502a2867SDave May 56771eb70e55SToby Isaac PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM, DM, Mat, Vec, Vec); 56781eb70e55SToby Isaac 5679d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_Plex(DM dm) 5680d71ae5a4SJacob Faibussowitsch { 5681552f7358SJed Brown PetscFunctionBegin; 5682552f7358SJed Brown dm->ops->view = DMView_Plex; 56832c40f234SMatthew G. Knepley dm->ops->load = DMLoad_Plex; 5684552f7358SJed Brown dm->ops->setfromoptions = DMSetFromOptions_Plex; 568538221697SMatthew G. Knepley dm->ops->clone = DMClone_Plex; 5686552f7358SJed Brown dm->ops->setup = DMSetUp_Plex; 56871bb6d2a8SBarry Smith dm->ops->createlocalsection = DMCreateLocalSection_Plex; 5688adc21957SMatthew G. Knepley dm->ops->createsectionpermutation = DMCreateSectionPermutation_Plex; 568966ad2231SToby Isaac dm->ops->createdefaultconstraints = DMCreateDefaultConstraints_Plex; 5690552f7358SJed Brown dm->ops->createglobalvector = DMCreateGlobalVector_Plex; 5691552f7358SJed Brown dm->ops->createlocalvector = DMCreateLocalVector_Plex; 5692184d77edSJed Brown dm->ops->getlocaltoglobalmapping = NULL; 56930298fd71SBarry Smith dm->ops->createfieldis = NULL; 5694552f7358SJed Brown dm->ops->createcoordinatedm = DMCreateCoordinateDM_Plex; 5695f19dbd58SToby Isaac dm->ops->createcoordinatefield = DMCreateCoordinateField_Plex; 56960a6ba040SMatthew G. Knepley dm->ops->getcoloring = NULL; 5697552f7358SJed Brown dm->ops->creatematrix = DMCreateMatrix_Plex; 5698bceba477SMatthew G. Knepley dm->ops->createinterpolation = DMCreateInterpolation_Plex; 5699bd041c0cSMatthew G. Knepley dm->ops->createmassmatrix = DMCreateMassMatrix_Plex; 5700b4937a87SMatthew G. Knepley dm->ops->createmassmatrixlumped = DMCreateMassMatrixLumped_Plex; 57015a84ad33SLisandro Dalcin dm->ops->createinjection = DMCreateInjection_Plex; 5702552f7358SJed Brown dm->ops->refine = DMRefine_Plex; 57030a6ba040SMatthew G. Knepley dm->ops->coarsen = DMCoarsen_Plex; 57040a6ba040SMatthew G. Knepley dm->ops->refinehierarchy = DMRefineHierarchy_Plex; 5705b653a561SMatthew G. Knepley dm->ops->coarsenhierarchy = DMCoarsenHierarchy_Plex; 5706d410b0cfSMatthew G. Knepley dm->ops->extrude = DMExtrude_Plex; 57070298fd71SBarry Smith dm->ops->globaltolocalbegin = NULL; 57080298fd71SBarry Smith dm->ops->globaltolocalend = NULL; 57090298fd71SBarry Smith dm->ops->localtoglobalbegin = NULL; 57100298fd71SBarry Smith dm->ops->localtoglobalend = NULL; 5711552f7358SJed Brown dm->ops->destroy = DMDestroy_Plex; 5712552f7358SJed Brown dm->ops->createsubdm = DMCreateSubDM_Plex; 57132adcc780SMatthew G. Knepley dm->ops->createsuperdm = DMCreateSuperDM_Plex; 5714793f3fe5SMatthew G. Knepley dm->ops->getdimpoints = DMGetDimPoints_Plex; 5715552f7358SJed Brown dm->ops->locatepoints = DMLocatePoints_Plex; 57160709b2feSToby Isaac dm->ops->projectfunctionlocal = DMProjectFunctionLocal_Plex; 57170709b2feSToby Isaac dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_Plex; 5718bfc4295aSToby Isaac dm->ops->projectfieldlocal = DMProjectFieldLocal_Plex; 57198c6c5593SMatthew G. Knepley dm->ops->projectfieldlabellocal = DMProjectFieldLabelLocal_Plex; 5720ece3a9fcSMatthew G. Knepley dm->ops->projectbdfieldlabellocal = DMProjectBdFieldLabelLocal_Plex; 57210709b2feSToby Isaac dm->ops->computel2diff = DMComputeL2Diff_Plex; 5722b698f381SToby Isaac dm->ops->computel2gradientdiff = DMComputeL2GradientDiff_Plex; 57232a16baeaSToby Isaac dm->ops->computel2fielddiff = DMComputeL2FieldDiff_Plex; 572428d58a37SPierre Jolivet dm->ops->getneighbors = DMGetNeighbors_Plex; 57256c6a6b79SMatthew G. Knepley dm->ops->getlocalboundingbox = DMGetLocalBoundingBox_Coordinates; 5726907a3e9cSStefano Zampini dm->ops->createdomaindecomposition = DMCreateDomainDecomposition_Plex; 5727907a3e9cSStefano Zampini dm->ops->createddscatters = DMCreateDomainDecompositionScatters_Plex; 57289566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_Plex)); 57296c51210dSStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", DMPlexInsertTimeDerivativeBoundaryValues_Plex)); 57309566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", DMSetUpGLVisViewer_Plex)); 57319566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_Plex)); 57329566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", DMPlexDistributeGetDefault_Plex)); 57339566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", DMPlexDistributeSetDefault_Plex)); 57346bc1bd01Sksagiyam PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", DMPlexReorderGetDefault_Plex)); 57356bc1bd01Sksagiyam PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", DMPlexReorderSetDefault_Plex)); 5736adc21957SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetDefault_C", DMReorderSectionGetDefault_Plex)); 5737adc21957SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetDefault_C", DMReorderSectionSetDefault_Plex)); 5738adc21957SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionGetType_C", DMReorderSectionGetType_Plex)); 5739adc21957SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMReorderSectionSetType_C", DMReorderSectionSetType_Plex)); 57409566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", DMInterpolateSolution_Plex)); 5741c506a872SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMPlexGetOverlap_Plex)); 5742c506a872SMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", DMPlexSetOverlap_Plex)); 5743d2b2dc1eSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", DMPlexGetUseCeed_Plex)); 5744d2b2dc1eSMatthew G. Knepley PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", DMPlexSetUseCeed_Plex)); 57453ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5746552f7358SJed Brown } 5747552f7358SJed Brown 5748d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode DMClone_Plex(DM dm, DM *newdm) 5749d71ae5a4SJacob Faibussowitsch { 575063a16f15SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *)dm->data; 57511fca310dSJames Wright const PetscSF *face_sfs; 57521fca310dSJames Wright PetscInt num_face_sfs; 575363a16f15SMatthew G. Knepley 575463a16f15SMatthew G. Knepley PetscFunctionBegin; 575563a16f15SMatthew G. Knepley mesh->refct++; 575663a16f15SMatthew G. Knepley (*newdm)->data = mesh; 57571fca310dSJames Wright PetscCall(DMPlexGetIsoperiodicFaceSF(dm, &num_face_sfs, &face_sfs)); 57581fca310dSJames Wright PetscCall(DMPlexSetIsoperiodicFaceSF(*newdm, num_face_sfs, (PetscSF *)face_sfs)); 57599566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)*newdm, DMPLEX)); 57609566063dSJacob Faibussowitsch PetscCall(DMInitialize_Plex(*newdm)); 57613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 576263a16f15SMatthew G. Knepley } 576363a16f15SMatthew G. Knepley 57648818961aSMatthew G Knepley /*MC 57650b4b7b1cSBarry Smith DMPLEX = "plex" - A `DM` object that encapsulates an unstructured mesh (or grid), or CW Complex {cite}`cwcomplex`, 57660b4b7b1cSBarry Smith which can be expressed using a Hasse Diagram {cite}`hassediagram`. 576720f4b53cSBarry Smith In the local representation, `Vec`s contain all unknowns in the interior and shared boundary. This is 57680b4b7b1cSBarry Smith specified by a `PetscSection` object. Ownership in the global representation is determined by 5769a1cb98faSBarry Smith ownership of the underlying `DMPLEX` points. This is specified by another `PetscSection` object. 57708818961aSMatthew G Knepley 5771e5893cccSMatthew G. Knepley Options Database Keys: 5772250712c9SMatthew G. Knepley + -dm_refine_pre - Refine mesh before distribution 5773250712c9SMatthew G. Knepley + -dm_refine_uniform_pre - Choose uniform or generator-based refinement 5774250712c9SMatthew G. Knepley + -dm_refine_volume_limit_pre - Cell volume limit after pre-refinement using generator 5775250712c9SMatthew G. Knepley . -dm_distribute - Distribute mesh across processes 5776250712c9SMatthew G. Knepley . -dm_distribute_overlap - Number of cells to overlap for distribution 5777250712c9SMatthew G. Knepley . -dm_refine - Refine mesh after distribution 5778c3db174cSMatthew G. Knepley . -dm_localize <bool> - Whether to localize coordinates for periodic meshes 5779c3db174cSMatthew G. Knepley . -dm_sparse_localize <bool> - Whether to only localize cells on the periodic boundary 5780250712c9SMatthew G. Knepley . -dm_plex_hash_location - Use grid hashing for point location 5781ddce0771SMatthew G. Knepley . -dm_plex_hash_box_faces <n,m,p> - The number of divisions in each direction of the grid hash 5782f12cf164SMatthew G. Knepley . -dm_plex_partition_balance - Attempt to evenly divide points on partition boundary between processes 5783f12cf164SMatthew G. Knepley . -dm_plex_remesh_bd - Allow changes to the boundary on remeshing 5784d5b43468SJose E. Roman . -dm_plex_max_projection_height - Maximum mesh point height used to project locally 5785f12cf164SMatthew G. Knepley . -dm_plex_regular_refinement - Use special nested projection algorithm for regular refinement 5786d02c7345SMatthew G. Knepley . -dm_plex_reorder_section - Use specialized blocking if available 5787aaa8cc7dSPierre Jolivet . -dm_plex_check_all - Perform all checks below 5788f12cf164SMatthew G. Knepley . -dm_plex_check_symmetry - Check that the adjacency information in the mesh is symmetric 5789f12cf164SMatthew G. Knepley . -dm_plex_check_skeleton <celltype> - Check that each cell has the correct number of vertices 5790f12cf164SMatthew 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 5791f12cf164SMatthew G. Knepley . -dm_plex_check_geometry - Check that cells have positive volume 5792f12cf164SMatthew G. Knepley . -dm_view :mesh.tex:ascii_latex - View the mesh in LaTeX/TikZ 5793e5893cccSMatthew G. Knepley . -dm_plex_view_scale <num> - Scale the TikZ 57945962854dSMatthew G. Knepley . -dm_plex_print_fem <num> - View FEM assembly information, such as element vectors and matrices 57955962854dSMatthew G. Knepley - -dm_plex_print_fvm <num> - View FVM assembly information, such as flux updates 5796e5893cccSMatthew G. Knepley 57978818961aSMatthew G Knepley Level: intermediate 57988818961aSMatthew G Knepley 57991cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMPlexCreate()`, `DMCreate()`, `DMSetType()`, `PetscSection` 58008818961aSMatthew G Knepley M*/ 58018818961aSMatthew G Knepley 5802d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm) 5803d71ae5a4SJacob Faibussowitsch { 5804552f7358SJed Brown DM_Plex *mesh; 5805412e9a14SMatthew G. Knepley PetscInt unit; 5806552f7358SJed Brown 5807552f7358SJed Brown PetscFunctionBegin; 5808f39ec787SMatthew G. Knepley PetscCall(PetscCitationsRegister(PlexCitation, &Plexcite)); 5809552f7358SJed Brown PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 58104dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&mesh)); 5811adc21957SMatthew G. Knepley dm->reorderSection = DM_REORDER_DEFAULT_NOTSET; 5812552f7358SJed Brown dm->data = mesh; 5813552f7358SJed Brown 5814552f7358SJed Brown mesh->refct = 1; 58159566063dSJacob Faibussowitsch PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection)); 58169566063dSJacob Faibussowitsch PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection)); 5817552f7358SJed Brown mesh->refinementUniform = PETSC_TRUE; 5818552f7358SJed Brown mesh->refinementLimit = -1.0; 5819e600fa54SMatthew G. Knepley mesh->distDefault = PETSC_TRUE; 5820adc21957SMatthew G. Knepley mesh->reorderDefault = DM_REORDER_DEFAULT_NOTSET; 58211d1f2f2aSksagiyam mesh->distributionName = NULL; 58227d0f5628SVaclav Hapla mesh->interpolated = DMPLEX_INTERPOLATED_INVALID; 58237d0f5628SVaclav Hapla mesh->interpolatedCollective = DMPLEX_INTERPOLATED_INVALID; 58245e2c5519SMatthew G. Knepley mesh->interpolatePreferTensor = PETSC_TRUE; 5825552f7358SJed Brown 58269566063dSJacob Faibussowitsch PetscCall(PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner)); 58272e62ab5aSMatthew G. Knepley mesh->remeshBd = PETSC_FALSE; 5828d9deefdfSMatthew G. Knepley 58298865f1eaSKarl Rupp for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0; 5830552f7358SJed Brown 5831df0420ecSMatthew G. Knepley mesh->depthState = -1; 5832ba2698f1SMatthew G. Knepley mesh->celltypeState = -1; 58336113b454SMatthew G. Knepley mesh->printTol = 1.0e-10; 5834c29ce622SStefano Zampini mesh->nonempty_comm = MPI_COMM_SELF; 5835552f7358SJed Brown 58369566063dSJacob Faibussowitsch PetscCall(DMInitialize_Plex(dm)); 58373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5838552f7358SJed Brown } 5839552f7358SJed Brown 5840552f7358SJed Brown /*@ 5841a1cb98faSBarry Smith DMPlexCreate - Creates a `DMPLEX` object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram. 5842552f7358SJed Brown 5843d083f849SBarry Smith Collective 5844552f7358SJed Brown 5845552f7358SJed Brown Input Parameter: 5846a1cb98faSBarry Smith . comm - The communicator for the `DMPLEX` object 5847552f7358SJed Brown 5848552f7358SJed Brown Output Parameter: 5849a1cb98faSBarry Smith . mesh - The `DMPLEX` object 5850552f7358SJed Brown 5851552f7358SJed Brown Level: beginner 5852552f7358SJed Brown 585342747ad1SJacob Faibussowitsch .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMType`, `DMCreate()`, `DMSetType()` 5854552f7358SJed Brown @*/ 5855d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh) 5856d71ae5a4SJacob Faibussowitsch { 5857552f7358SJed Brown PetscFunctionBegin; 58584f572ea9SToby Isaac PetscAssertPointer(mesh, 2); 58599566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, mesh)); 58609566063dSJacob Faibussowitsch PetscCall(DMSetType(*mesh, DMPLEX)); 58613ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 5862552f7358SJed Brown } 5863552f7358SJed Brown 5864b09969d6SVaclav Hapla /*@C 5865b0fe842aSMatthew 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 5866a1cb98faSBarry Smith 586720f4b53cSBarry Smith Collective; No Fortran Support 5868b09969d6SVaclav Hapla 5869b09969d6SVaclav Hapla Input Parameters: 5870a1cb98faSBarry Smith + dm - The `DM` 5871b09969d6SVaclav Hapla . numCells - The number of cells owned by this process 5872a1cb98faSBarry Smith . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE` 5873a1cb98faSBarry Smith . NVertices - The global number of vertices, or `PETSC_DETERMINE` 5874b09969d6SVaclav Hapla . numCorners - The number of vertices for each cell 58755e488331SVaclav Hapla - cells - An array of numCells*numCorners numbers, the global vertex numbers for each cell 5876b09969d6SVaclav Hapla 5877be8c289dSNicolas Barral Output Parameters: 5878a1cb98faSBarry Smith + vertexSF - (Optional) `PetscSF` describing complete vertex ownership 5879be8c289dSNicolas Barral - verticesAdjSaved - (Optional) vertex adjacency array 5880b09969d6SVaclav Hapla 5881b09969d6SVaclav Hapla Level: advanced 5882b09969d6SVaclav Hapla 5883a1cb98faSBarry Smith Notes: 5884a1cb98faSBarry Smith Two triangles sharing a face 5885a1cb98faSBarry Smith .vb 5886a1cb98faSBarry Smith 5887a1cb98faSBarry Smith 2 5888a1cb98faSBarry Smith / | \ 5889a1cb98faSBarry Smith / | \ 5890a1cb98faSBarry Smith / | \ 5891a1cb98faSBarry Smith 0 0 | 1 3 5892a1cb98faSBarry Smith \ | / 5893a1cb98faSBarry Smith \ | / 5894a1cb98faSBarry Smith \ | / 5895a1cb98faSBarry Smith 1 5896a1cb98faSBarry Smith .ve 5897a1cb98faSBarry Smith would have input 5898a1cb98faSBarry Smith .vb 5899a1cb98faSBarry Smith numCells = 2, numVertices = 4 5900a1cb98faSBarry Smith cells = [0 1 2 1 3 2] 5901a1cb98faSBarry Smith .ve 5902a1cb98faSBarry Smith which would result in the `DMPLEX` 5903a1cb98faSBarry Smith .vb 5904a1cb98faSBarry Smith 5905a1cb98faSBarry Smith 4 5906a1cb98faSBarry Smith / | \ 5907a1cb98faSBarry Smith / | \ 5908a1cb98faSBarry Smith / | \ 5909a1cb98faSBarry Smith 2 0 | 1 5 5910a1cb98faSBarry Smith \ | / 5911a1cb98faSBarry Smith \ | / 5912a1cb98faSBarry Smith \ | / 5913a1cb98faSBarry Smith 3 5914a1cb98faSBarry Smith .ve 5915a1cb98faSBarry Smith 5916a1cb98faSBarry Smith Vertices are implicitly numbered consecutively 0,...,NVertices. 5917a1cb98faSBarry Smith Each rank owns a chunk of numVertices consecutive vertices. 5918a1cb98faSBarry Smith If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout. 5919a1cb98faSBarry Smith If NVertices is `PETSC_DETERMINE` and numVertices is PETSC_DECIDE, NVertices is computed by PETSc as the maximum vertex index in cells + 1. 5920a1cb98faSBarry Smith If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks. 5921a1cb98faSBarry Smith 5922a1cb98faSBarry Smith The cell distribution is arbitrary non-overlapping, independent of the vertex distribution. 5923a1cb98faSBarry Smith 59241cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildCoordinatesFromCellListParallel()`, 5925a1cb98faSBarry Smith `PetscSF` 5926b09969d6SVaclav Hapla @*/ 5927ce78bad3SBarry Smith PetscErrorCode DMPlexBuildFromCellListParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscInt numCorners, const PetscInt cells[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdjSaved[]) 5928d71ae5a4SJacob Faibussowitsch { 59292464107aSksagiyam PetscSF sfPoint; 59302464107aSksagiyam PetscLayout layout; 593182fb893eSVaclav Hapla PetscInt numVerticesAdj, *verticesAdj, *cones, c, p; 5932a47d0d45SMatthew G. Knepley 5933a47d0d45SMatthew G. Knepley PetscFunctionBegin; 593425b6865aSVaclav Hapla PetscValidLogicalCollectiveInt(dm, NVertices, 4); 59359566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 593625b6865aSVaclav Hapla /* Get/check global number of vertices */ 593725b6865aSVaclav Hapla { 593825b6865aSVaclav Hapla PetscInt NVerticesInCells, i; 593925b6865aSVaclav Hapla const PetscInt len = numCells * numCorners; 594025b6865aSVaclav Hapla 594125b6865aSVaclav Hapla /* NVerticesInCells = max(cells) + 1 */ 59421690c2aeSBarry Smith NVerticesInCells = PETSC_INT_MIN; 59439371c9d4SSatish Balay for (i = 0; i < len; i++) 59449371c9d4SSatish Balay if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i]; 594525b6865aSVaclav Hapla ++NVerticesInCells; 5946462c564dSBarry Smith PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm))); 594725b6865aSVaclav Hapla 594825b6865aSVaclav Hapla if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells; 59499371c9d4SSatish Balay else 59509371c9d4SSatish 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); 595125b6865aSVaclav Hapla } 59529079aca8SVaclav Hapla /* Count locally unique vertices */ 59539079aca8SVaclav Hapla { 59549079aca8SVaclav Hapla PetscHSetI vhash; 59559079aca8SVaclav Hapla PetscInt off = 0; 59569079aca8SVaclav Hapla 59579566063dSJacob Faibussowitsch PetscCall(PetscHSetICreate(&vhash)); 5958a47d0d45SMatthew G. Knepley for (c = 0; c < numCells; ++c) { 595948a46eb9SPierre Jolivet for (p = 0; p < numCorners; ++p) PetscCall(PetscHSetIAdd(vhash, cells[c * numCorners + p])); 5960a47d0d45SMatthew G. Knepley } 59619566063dSJacob Faibussowitsch PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj)); 59629566063dSJacob Faibussowitsch if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj)); 5963ad540459SPierre Jolivet else verticesAdj = *verticesAdjSaved; 59649566063dSJacob Faibussowitsch PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj)); 59659566063dSJacob Faibussowitsch PetscCall(PetscHSetIDestroy(&vhash)); 596663a3b9bcSJacob Faibussowitsch PetscCheck(off == numVerticesAdj, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid number of local vertices %" PetscInt_FMT " should be %" PetscInt_FMT, off, numVerticesAdj); 5967a47d0d45SMatthew G. Knepley } 59689566063dSJacob Faibussowitsch PetscCall(PetscSortInt(numVerticesAdj, verticesAdj)); 5969a47d0d45SMatthew G. Knepley /* Create cones */ 59709566063dSJacob Faibussowitsch PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj)); 59719566063dSJacob Faibussowitsch for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners)); 59729566063dSJacob Faibussowitsch PetscCall(DMSetUp(dm)); 59739566063dSJacob Faibussowitsch PetscCall(DMPlexGetCones(dm, &cones)); 5974a47d0d45SMatthew G. Knepley for (c = 0; c < numCells; ++c) { 5975a47d0d45SMatthew G. Knepley for (p = 0; p < numCorners; ++p) { 5976a47d0d45SMatthew G. Knepley const PetscInt gv = cells[c * numCorners + p]; 5977a47d0d45SMatthew G. Knepley PetscInt lv; 5978a47d0d45SMatthew G. Knepley 59799079aca8SVaclav Hapla /* Positions within verticesAdj form 0-based local vertex numbering; 59809079aca8SVaclav Hapla we need to shift it by numCells to get correct DAG points (cells go first) */ 59819566063dSJacob Faibussowitsch PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv)); 598263a3b9bcSJacob Faibussowitsch PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv); 5983961cfab0SVaclav Hapla cones[c * numCorners + p] = lv + numCells; 5984a47d0d45SMatthew G. Knepley } 5985a47d0d45SMatthew G. Knepley } 59862464107aSksagiyam /* Build point sf */ 59879566063dSJacob Faibussowitsch PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout)); 59889566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetSize(layout, NVertices)); 59899566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetLocalSize(layout, numVertices)); 59909566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetBlockSize(layout, 1)); 59919566063dSJacob Faibussowitsch PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint)); 59929566063dSJacob Faibussowitsch PetscCall(PetscLayoutDestroy(&layout)); 59939566063dSJacob Faibussowitsch if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj)); 59949566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF")); 59952464107aSksagiyam if (dm->sf) { 59962464107aSksagiyam const char *prefix; 59972464107aSksagiyam 59989566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix)); 59999566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix)); 60002464107aSksagiyam } 60019566063dSJacob Faibussowitsch PetscCall(DMSetPointSF(dm, sfPoint)); 60029566063dSJacob Faibussowitsch PetscCall(PetscSFDestroy(&sfPoint)); 6003f4f49eeaSPierre Jolivet if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF")); 6004a47d0d45SMatthew G. Knepley /* Fill in the rest of the topology structure */ 60059566063dSJacob Faibussowitsch PetscCall(DMPlexSymmetrize(dm)); 60069566063dSJacob Faibussowitsch PetscCall(DMPlexStratify(dm)); 60079566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 60083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6009a47d0d45SMatthew G. Knepley } 6010a47d0d45SMatthew G. Knepley 6011b0fe842aSMatthew G. Knepley /*@C 6012b0fe842aSMatthew G. Knepley DMPlexBuildFromCellSectionParallel - Build distributed `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) allowing multiple celltypes 6013b0fe842aSMatthew G. Knepley 6014b0fe842aSMatthew G. Knepley Collective; No Fortran Support 6015b0fe842aSMatthew G. Knepley 6016b0fe842aSMatthew G. Knepley Input Parameters: 6017b0fe842aSMatthew G. Knepley + dm - The `DM` 6018b0fe842aSMatthew G. Knepley . numCells - The number of cells owned by this process 6019b0fe842aSMatthew G. Knepley . numVertices - The number of vertices to be owned by this process, or `PETSC_DECIDE` 6020b0fe842aSMatthew G. Knepley . NVertices - The global number of vertices, or `PETSC_DETERMINE` 6021b0fe842aSMatthew G. Knepley . cellSection - The `PetscSection` giving the number of vertices for each cell (layout of cells) 6022b0fe842aSMatthew G. Knepley - cells - An array of the global vertex numbers for each cell 6023b0fe842aSMatthew G. Knepley 6024b0fe842aSMatthew G. Knepley Output Parameters: 6025b0fe842aSMatthew G. Knepley + vertexSF - (Optional) `PetscSF` describing complete vertex ownership 6026b0fe842aSMatthew G. Knepley - verticesAdjSaved - (Optional) vertex adjacency array 6027b0fe842aSMatthew G. Knepley 6028b0fe842aSMatthew G. Knepley Level: advanced 6029b0fe842aSMatthew G. Knepley 6030b0fe842aSMatthew G. Knepley Notes: 6031b0fe842aSMatthew G. Knepley A triangle and quadrilateral sharing a face 6032b0fe842aSMatthew G. Knepley .vb 6033b0fe842aSMatthew G. Knepley 2----------3 6034b0fe842aSMatthew G. Knepley / | | 6035b0fe842aSMatthew G. Knepley / | | 6036b0fe842aSMatthew G. Knepley / | | 6037b0fe842aSMatthew G. Knepley 0 0 | 1 | 6038b0fe842aSMatthew G. Knepley \ | | 6039b0fe842aSMatthew G. Knepley \ | | 6040b0fe842aSMatthew G. Knepley \ | | 6041b0fe842aSMatthew G. Knepley 1----------4 6042b0fe842aSMatthew G. Knepley .ve 6043b0fe842aSMatthew G. Knepley would have input 6044b0fe842aSMatthew G. Knepley .vb 6045b0fe842aSMatthew G. Knepley numCells = 2, numVertices = 5 6046b0fe842aSMatthew G. Knepley cells = [0 1 2 1 4 3 2] 6047b0fe842aSMatthew G. Knepley .ve 6048b0fe842aSMatthew G. Knepley which would result in the `DMPLEX` 6049b0fe842aSMatthew G. Knepley .vb 6050b0fe842aSMatthew G. Knepley 4----------5 6051b0fe842aSMatthew G. Knepley / | | 6052b0fe842aSMatthew G. Knepley / | | 6053b0fe842aSMatthew G. Knepley / | | 6054b0fe842aSMatthew G. Knepley 2 0 | 1 | 6055b0fe842aSMatthew G. Knepley \ | | 6056b0fe842aSMatthew G. Knepley \ | | 6057b0fe842aSMatthew G. Knepley \ | | 6058b0fe842aSMatthew G. Knepley 3----------6 6059b0fe842aSMatthew G. Knepley .ve 6060b0fe842aSMatthew G. Knepley 6061b0fe842aSMatthew G. Knepley Vertices are implicitly numbered consecutively 0,...,NVertices. 6062b0fe842aSMatthew G. Knepley Each rank owns a chunk of numVertices consecutive vertices. 6063b0fe842aSMatthew G. Knepley If numVertices is `PETSC_DECIDE`, PETSc will distribute them as evenly as possible using PetscLayout. 6064b0fe842aSMatthew 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. 6065b0fe842aSMatthew G. Knepley If only NVertices is `PETSC_DETERMINE`, it is computed as the sum of numVertices over all ranks. 6066b0fe842aSMatthew G. Knepley 6067b0fe842aSMatthew G. Knepley The cell distribution is arbitrary non-overlapping, independent of the vertex distribution. 6068b0fe842aSMatthew G. Knepley 6069b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexCreateFromCellSectionParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, 6070b0fe842aSMatthew G. Knepley `PetscSF` 6071b0fe842aSMatthew G. Knepley @*/ 6072ce78bad3SBarry Smith PetscErrorCode DMPlexBuildFromCellSectionParallel(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt NVertices, PetscSection cellSection, const PetscInt cells[], PeOp PetscSF *vertexSF, PeOp PetscInt *verticesAdjSaved[]) 6073b0fe842aSMatthew G. Knepley { 6074b0fe842aSMatthew G. Knepley PetscSF sfPoint; 6075b0fe842aSMatthew G. Knepley PetscLayout layout; 6076b0fe842aSMatthew G. Knepley PetscInt numVerticesAdj, *verticesAdj, *cones, cStart, cEnd, len; 6077b0fe842aSMatthew G. Knepley 6078b0fe842aSMatthew G. Knepley PetscFunctionBegin; 6079b0fe842aSMatthew G. Knepley PetscValidLogicalCollectiveInt(dm, NVertices, 4); 6080b0fe842aSMatthew G. Knepley PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 6081b0fe842aSMatthew G. Knepley PetscCall(PetscSectionGetChart(cellSection, &cStart, &cEnd)); 6082b0fe842aSMatthew G. Knepley PetscCall(PetscSectionGetStorageSize(cellSection, &len)); 6083b0fe842aSMatthew 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); 6084b0fe842aSMatthew G. Knepley /* Get/check global number of vertices */ 6085b0fe842aSMatthew G. Knepley { 6086b0fe842aSMatthew G. Knepley PetscInt NVerticesInCells; 6087b0fe842aSMatthew G. Knepley 6088b0fe842aSMatthew G. Knepley /* NVerticesInCells = max(cells) + 1 */ 6089b0fe842aSMatthew G. Knepley NVerticesInCells = PETSC_MIN_INT; 6090b0fe842aSMatthew G. Knepley for (PetscInt i = 0; i < len; i++) 6091b0fe842aSMatthew G. Knepley if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i]; 6092b0fe842aSMatthew G. Knepley ++NVerticesInCells; 6093b0fe842aSMatthew G. Knepley PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, &NVerticesInCells, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm))); 6094b0fe842aSMatthew G. Knepley 6095b0fe842aSMatthew G. Knepley if (numVertices == PETSC_DECIDE && NVertices == PETSC_DECIDE) NVertices = NVerticesInCells; 6096b0fe842aSMatthew G. Knepley else 6097b0fe842aSMatthew 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); 6098b0fe842aSMatthew G. Knepley } 6099b0fe842aSMatthew G. Knepley /* Count locally unique vertices */ 6100b0fe842aSMatthew G. Knepley { 6101b0fe842aSMatthew G. Knepley PetscHSetI vhash; 6102b0fe842aSMatthew G. Knepley PetscInt off = 0; 6103b0fe842aSMatthew G. Knepley 6104b0fe842aSMatthew G. Knepley PetscCall(PetscHSetICreate(&vhash)); 6105b0fe842aSMatthew G. Knepley for (PetscInt i = 0; i < len; i++) PetscCall(PetscHSetIAdd(vhash, cells[i])); 6106b0fe842aSMatthew G. Knepley PetscCall(PetscHSetIGetSize(vhash, &numVerticesAdj)); 6107b0fe842aSMatthew G. Knepley if (!verticesAdjSaved) PetscCall(PetscMalloc1(numVerticesAdj, &verticesAdj)); 6108b0fe842aSMatthew G. Knepley else verticesAdj = *verticesAdjSaved; 6109b0fe842aSMatthew G. Knepley PetscCall(PetscHSetIGetElems(vhash, &off, verticesAdj)); 6110b0fe842aSMatthew G. Knepley PetscCall(PetscHSetIDestroy(&vhash)); 6111b0fe842aSMatthew 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); 6112b0fe842aSMatthew G. Knepley } 6113b0fe842aSMatthew G. Knepley PetscCall(PetscSortInt(numVerticesAdj, verticesAdj)); 6114b0fe842aSMatthew G. Knepley /* Create cones */ 6115b0fe842aSMatthew G. Knepley PetscCall(DMPlexSetChart(dm, 0, numCells + numVerticesAdj)); 6116b0fe842aSMatthew G. Knepley for (PetscInt c = 0; c < numCells; ++c) { 6117b0fe842aSMatthew G. Knepley PetscInt dof; 6118b0fe842aSMatthew G. Knepley 6119b0fe842aSMatthew G. Knepley PetscCall(PetscSectionGetDof(cellSection, c, &dof)); 6120b0fe842aSMatthew G. Knepley PetscCall(DMPlexSetConeSize(dm, c, dof)); 6121b0fe842aSMatthew G. Knepley } 6122b0fe842aSMatthew G. Knepley PetscCall(DMSetUp(dm)); 6123b0fe842aSMatthew G. Knepley PetscCall(DMPlexGetCones(dm, &cones)); 6124b0fe842aSMatthew G. Knepley for (PetscInt c = 0; c < numCells; ++c) { 6125b0fe842aSMatthew G. Knepley PetscInt dof, off; 6126b0fe842aSMatthew G. Knepley 6127b0fe842aSMatthew G. Knepley PetscCall(PetscSectionGetDof(cellSection, c, &dof)); 6128b0fe842aSMatthew G. Knepley PetscCall(PetscSectionGetOffset(cellSection, c, &off)); 6129b0fe842aSMatthew G. Knepley for (PetscInt p = off; p < off + dof; ++p) { 6130b0fe842aSMatthew G. Knepley const PetscInt gv = cells[p]; 6131b0fe842aSMatthew G. Knepley PetscInt lv; 6132b0fe842aSMatthew G. Knepley 6133b0fe842aSMatthew G. Knepley /* Positions within verticesAdj form 0-based local vertex numbering; 6134b0fe842aSMatthew G. Knepley we need to shift it by numCells to get correct DAG points (cells go first) */ 6135b0fe842aSMatthew G. Knepley PetscCall(PetscFindInt(gv, numVerticesAdj, verticesAdj, &lv)); 6136b0fe842aSMatthew G. Knepley PetscCheck(lv >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Could not find global vertex %" PetscInt_FMT " in local connectivity", gv); 6137b0fe842aSMatthew G. Knepley cones[p] = lv + numCells; 6138b0fe842aSMatthew G. Knepley } 6139b0fe842aSMatthew G. Knepley } 6140b0fe842aSMatthew G. Knepley /* Build point sf */ 6141b0fe842aSMatthew G. Knepley PetscCall(PetscLayoutCreate(PetscObjectComm((PetscObject)dm), &layout)); 6142b0fe842aSMatthew G. Knepley PetscCall(PetscLayoutSetSize(layout, NVertices)); 6143b0fe842aSMatthew G. Knepley PetscCall(PetscLayoutSetLocalSize(layout, numVertices)); 6144b0fe842aSMatthew G. Knepley PetscCall(PetscLayoutSetBlockSize(layout, 1)); 6145b0fe842aSMatthew G. Knepley PetscCall(PetscSFCreateByMatchingIndices(layout, numVerticesAdj, verticesAdj, NULL, numCells, numVerticesAdj, verticesAdj, NULL, numCells, vertexSF, &sfPoint)); 6146b0fe842aSMatthew G. Knepley PetscCall(PetscLayoutDestroy(&layout)); 6147b0fe842aSMatthew G. Knepley if (!verticesAdjSaved) PetscCall(PetscFree(verticesAdj)); 6148b0fe842aSMatthew G. Knepley PetscCall(PetscObjectSetName((PetscObject)sfPoint, "point SF")); 6149b0fe842aSMatthew G. Knepley if (dm->sf) { 6150b0fe842aSMatthew G. Knepley const char *prefix; 6151b0fe842aSMatthew G. Knepley 6152b0fe842aSMatthew G. Knepley PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm->sf, &prefix)); 6153b0fe842aSMatthew G. Knepley PetscCall(PetscObjectSetOptionsPrefix((PetscObject)sfPoint, prefix)); 6154b0fe842aSMatthew G. Knepley } 6155b0fe842aSMatthew G. Knepley PetscCall(DMSetPointSF(dm, sfPoint)); 6156b0fe842aSMatthew G. Knepley PetscCall(PetscSFDestroy(&sfPoint)); 6157b0fe842aSMatthew G. Knepley if (vertexSF) PetscCall(PetscObjectSetName((PetscObject)*vertexSF, "Vertex Ownership SF")); 6158b0fe842aSMatthew G. Knepley /* Fill in the rest of the topology structure */ 6159b0fe842aSMatthew G. Knepley PetscCall(DMPlexSymmetrize(dm)); 6160b0fe842aSMatthew G. Knepley PetscCall(DMPlexStratify(dm)); 6161b0fe842aSMatthew G. Knepley PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 6162b0fe842aSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 6163b0fe842aSMatthew G. Knepley } 6164b0fe842aSMatthew G. Knepley 6165cc4c1da9SBarry Smith /*@ 6166a1cb98faSBarry Smith DMPlexBuildCoordinatesFromCellListParallel - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output) 6167a1cb98faSBarry Smith 616820f4b53cSBarry Smith Collective; No Fortran Support 6169b09969d6SVaclav Hapla 6170b09969d6SVaclav Hapla Input Parameters: 6171a1cb98faSBarry Smith + dm - The `DM` 6172b09969d6SVaclav Hapla . spaceDim - The spatial dimension used for coordinates 6173a1cb98faSBarry Smith . sfVert - `PetscSF` describing complete vertex ownership 6174b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex 6175b09969d6SVaclav Hapla 6176b09969d6SVaclav Hapla Level: advanced 6177b09969d6SVaclav Hapla 61781cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellListParallel()` 6179b09969d6SVaclav Hapla @*/ 6180d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellListParallel(DM dm, PetscInt spaceDim, PetscSF sfVert, const PetscReal vertexCoords[]) 6181d71ae5a4SJacob Faibussowitsch { 6182a47d0d45SMatthew G. Knepley PetscSection coordSection; 6183a47d0d45SMatthew G. Knepley Vec coordinates; 6184a47d0d45SMatthew G. Knepley PetscScalar *coords; 61851edcf0b2SVaclav Hapla PetscInt numVertices, numVerticesAdj, coordSize, v, vStart, vEnd; 6186835f2295SStefano Zampini PetscMPIInt spaceDimi; 6187a47d0d45SMatthew G. Knepley 6188a47d0d45SMatthew G. Knepley PetscFunctionBegin; 61899566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0)); 61909566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd)); 61911dca8a05SBarry Smith PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first."); 61929566063dSJacob Faibussowitsch PetscCall(DMSetCoordinateDim(dm, spaceDim)); 61939566063dSJacob Faibussowitsch PetscCall(PetscSFGetGraph(sfVert, &numVertices, &numVerticesAdj, NULL, NULL)); 61941dca8a05SBarry 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); 61959566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 61969566063dSJacob Faibussowitsch PetscCall(PetscSectionSetNumFields(coordSection, 1)); 61979566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim)); 61989566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd)); 61991edcf0b2SVaclav Hapla for (v = vStart; v < vEnd; ++v) { 62009566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(coordSection, v, spaceDim)); 62019566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim)); 6202a47d0d45SMatthew G. Knepley } 62039566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(coordSection)); 62049566063dSJacob Faibussowitsch PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize)); 62059566063dSJacob Faibussowitsch PetscCall(VecCreate(PetscObjectComm((PetscObject)dm), &coordinates)); 62069566063dSJacob Faibussowitsch PetscCall(VecSetBlockSize(coordinates, spaceDim)); 62079566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates")); 62089566063dSJacob Faibussowitsch PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE)); 62099566063dSJacob Faibussowitsch PetscCall(VecSetType(coordinates, VECSTANDARD)); 62109566063dSJacob Faibussowitsch PetscCall(VecGetArray(coordinates, &coords)); 6211a47d0d45SMatthew G. Knepley { 6212a47d0d45SMatthew G. Knepley MPI_Datatype coordtype; 6213a47d0d45SMatthew G. Knepley 6214a47d0d45SMatthew G. Knepley /* Need a temp buffer for coords if we have complex/single */ 6215835f2295SStefano Zampini PetscCall(PetscMPIIntCast(spaceDim, &spaceDimi)); 6216835f2295SStefano Zampini PetscCallMPI(MPI_Type_contiguous(spaceDimi, MPIU_SCALAR, &coordtype)); 62179566063dSJacob Faibussowitsch PetscCallMPI(MPI_Type_commit(&coordtype)); 621821016a8bSBarry Smith #if defined(PETSC_USE_COMPLEX) 621921016a8bSBarry Smith { 622021016a8bSBarry Smith PetscScalar *svertexCoords; 622121016a8bSBarry Smith PetscInt i; 62229566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(numVertices * spaceDim, &svertexCoords)); 62233612f820SVaclav Hapla for (i = 0; i < numVertices * spaceDim; i++) svertexCoords[i] = vertexCoords[i]; 62249566063dSJacob Faibussowitsch PetscCall(PetscSFBcastBegin(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE)); 62259566063dSJacob Faibussowitsch PetscCall(PetscSFBcastEnd(sfVert, coordtype, svertexCoords, coords, MPI_REPLACE)); 62269566063dSJacob Faibussowitsch PetscCall(PetscFree(svertexCoords)); 622721016a8bSBarry Smith } 622821016a8bSBarry Smith #else 62299566063dSJacob Faibussowitsch PetscCall(PetscSFBcastBegin(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE)); 62309566063dSJacob Faibussowitsch PetscCall(PetscSFBcastEnd(sfVert, coordtype, vertexCoords, coords, MPI_REPLACE)); 623121016a8bSBarry Smith #endif 62329566063dSJacob Faibussowitsch PetscCallMPI(MPI_Type_free(&coordtype)); 6233a47d0d45SMatthew G. Knepley } 62349566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(coordinates, &coords)); 62359566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(dm, coordinates)); 62369566063dSJacob Faibussowitsch PetscCall(VecDestroy(&coordinates)); 62379566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0)); 62383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6239a47d0d45SMatthew G. Knepley } 6240a47d0d45SMatthew G. Knepley 6241c3edce3dSSatish Balay /*@ 6242b0fe842aSMatthew 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 6243a1cb98faSBarry Smith 6244a1cb98faSBarry Smith Collective 6245a47d0d45SMatthew G. Knepley 6246a47d0d45SMatthew G. Knepley Input Parameters: 6247a47d0d45SMatthew G. Knepley + comm - The communicator 6248a47d0d45SMatthew G. Knepley . dim - The topological dimension of the mesh 6249a47d0d45SMatthew G. Knepley . numCells - The number of cells owned by this process 6250a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DECIDE` 6251a1cb98faSBarry Smith . NVertices - The global number of vertices, or `PETSC_DECIDE` 6252a47d0d45SMatthew G. Knepley . numCorners - The number of vertices for each cell 6253a47d0d45SMatthew G. Knepley . interpolate - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically 6254a47d0d45SMatthew G. Knepley . cells - An array of numCells*numCorners numbers, the global vertex numbers for each cell 6255a47d0d45SMatthew G. Knepley . spaceDim - The spatial dimension used for coordinates 6256a47d0d45SMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex 6257a47d0d45SMatthew G. Knepley 6258d8d19677SJose E. Roman Output Parameters: 6259a1cb98faSBarry Smith + dm - The `DM` 6260a1cb98faSBarry Smith . vertexSF - (Optional) `PetscSF` describing complete vertex ownership 626160225df5SJacob Faibussowitsch - verticesAdj - (Optional) vertex adjacency array 6262a47d0d45SMatthew G. Knepley 6263b09969d6SVaclav Hapla Level: intermediate 6264a47d0d45SMatthew G. Knepley 6265a1cb98faSBarry Smith Notes: 6266a1cb98faSBarry Smith This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, 6267a1cb98faSBarry Smith `DMPlexBuildFromCellListParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()` 6268a1cb98faSBarry Smith 6269a1cb98faSBarry Smith See `DMPlexBuildFromCellListParallel()` for an example and details about the topology-related parameters. 6270a1cb98faSBarry Smith 6271a1cb98faSBarry Smith See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters. 6272a1cb98faSBarry Smith 62731cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()` 6274a47d0d45SMatthew G. Knepley @*/ 6275ce78bad3SBarry 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) 6276d71ae5a4SJacob Faibussowitsch { 6277a47d0d45SMatthew G. Knepley PetscSF sfVert; 6278a47d0d45SMatthew G. Knepley 6279a47d0d45SMatthew G. Knepley PetscFunctionBegin; 62809566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 62819566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 6282a47d0d45SMatthew G. Knepley PetscValidLogicalCollectiveInt(*dm, dim, 2); 6283064a246eSJacob Faibussowitsch PetscValidLogicalCollectiveInt(*dm, spaceDim, 9); 62849566063dSJacob Faibussowitsch PetscCall(DMSetDimension(*dm, dim)); 62859566063dSJacob Faibussowitsch PetscCall(DMPlexBuildFromCellListParallel(*dm, numCells, numVertices, NVertices, numCorners, cells, &sfVert, verticesAdj)); 6286a47d0d45SMatthew G. Knepley if (interpolate) { 62875fd9971aSMatthew G. Knepley DM idm; 6288a47d0d45SMatthew G. Knepley 62899566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(*dm, &idm)); 62909566063dSJacob Faibussowitsch PetscCall(DMDestroy(dm)); 6291a47d0d45SMatthew G. Knepley *dm = idm; 6292a47d0d45SMatthew G. Knepley } 62939566063dSJacob Faibussowitsch PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords)); 629418d54ad4SMichael Lange if (vertexSF) *vertexSF = sfVert; 62959566063dSJacob Faibussowitsch else PetscCall(PetscSFDestroy(&sfVert)); 62963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6297a47d0d45SMatthew G. Knepley } 6298a47d0d45SMatthew G. Knepley 6299cc4c1da9SBarry Smith /*@ 6300b0fe842aSMatthew G. Knepley DMPlexCreateFromCellSectionParallel - Create distributed `DMPLEX` from a list of vertices for each cell (common mesh generator output) and supports multiple celltypes 6301b0fe842aSMatthew G. Knepley 6302b0fe842aSMatthew G. Knepley Collective 6303b0fe842aSMatthew G. Knepley 6304b0fe842aSMatthew G. Knepley Input Parameters: 6305b0fe842aSMatthew G. Knepley + comm - The communicator 6306b0fe842aSMatthew G. Knepley . dim - The topological dimension of the mesh 6307b0fe842aSMatthew G. Knepley . numCells - The number of cells owned by this process 6308b0fe842aSMatthew G. Knepley . numVertices - The number of vertices owned by this process, or `PETSC_DECIDE` 6309b0fe842aSMatthew G. Knepley . NVertices - The global number of vertices, or `PETSC_DECIDE` 6310b0fe842aSMatthew G. Knepley . cellSection - The `PetscSection` giving the number of vertices for each cell (layout of cells) 6311b0fe842aSMatthew G. Knepley . interpolate - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically 6312b0fe842aSMatthew G. Knepley . cells - An array of the global vertex numbers for each cell 6313b0fe842aSMatthew G. Knepley . spaceDim - The spatial dimension used for coordinates 6314b0fe842aSMatthew G. Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex 6315b0fe842aSMatthew G. Knepley 6316b0fe842aSMatthew G. Knepley Output Parameters: 6317b0fe842aSMatthew G. Knepley + dm - The `DM` 6318b0fe842aSMatthew G. Knepley . vertexSF - (Optional) `PetscSF` describing complete vertex ownership 6319b0fe842aSMatthew G. Knepley - verticesAdj - (Optional) vertex adjacency array 6320b0fe842aSMatthew G. Knepley 6321b0fe842aSMatthew G. Knepley Level: intermediate 6322b0fe842aSMatthew G. Knepley 6323b0fe842aSMatthew G. Knepley Notes: 6324b0fe842aSMatthew G. Knepley This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, 6325b0fe842aSMatthew G. Knepley `DMPlexBuildFromCellSectionParallel()`, `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellListParallel()` 6326b0fe842aSMatthew G. Knepley 6327b0fe842aSMatthew G. Knepley See `DMPlexBuildFromCellSectionParallel()` for an example and details about the topology-related parameters. 6328b0fe842aSMatthew G. Knepley 6329b0fe842aSMatthew G. Knepley See `DMPlexBuildCoordinatesFromCellListParallel()` for details about the geometry-related parameters. 6330b0fe842aSMatthew G. Knepley 6331b0fe842aSMatthew G. Knepley .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()` 6332b0fe842aSMatthew G. Knepley @*/ 6333ce78bad3SBarry 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) 6334b0fe842aSMatthew G. Knepley { 6335b0fe842aSMatthew G. Knepley PetscSF sfVert; 6336b0fe842aSMatthew G. Knepley 6337b0fe842aSMatthew G. Knepley PetscFunctionBegin; 6338b0fe842aSMatthew G. Knepley PetscCall(DMCreate(comm, dm)); 6339b0fe842aSMatthew G. Knepley PetscCall(DMSetType(*dm, DMPLEX)); 6340b0fe842aSMatthew G. Knepley PetscValidLogicalCollectiveInt(*dm, dim, 2); 6341b0fe842aSMatthew G. Knepley PetscValidLogicalCollectiveInt(*dm, spaceDim, 9); 6342b0fe842aSMatthew G. Knepley PetscCall(DMSetDimension(*dm, dim)); 6343b0fe842aSMatthew G. Knepley PetscCall(DMPlexBuildFromCellSectionParallel(*dm, numCells, numVertices, NVertices, cellSection, cells, &sfVert, verticesAdj)); 6344b0fe842aSMatthew G. Knepley if (interpolate) { 6345b0fe842aSMatthew G. Knepley DM idm; 6346b0fe842aSMatthew G. Knepley 6347b0fe842aSMatthew G. Knepley PetscCall(DMPlexInterpolate(*dm, &idm)); 6348b0fe842aSMatthew G. Knepley PetscCall(DMDestroy(dm)); 6349b0fe842aSMatthew G. Knepley *dm = idm; 6350b0fe842aSMatthew G. Knepley } 6351b0fe842aSMatthew G. Knepley PetscCall(DMPlexBuildCoordinatesFromCellListParallel(*dm, spaceDim, sfVert, vertexCoords)); 6352b0fe842aSMatthew G. Knepley if (vertexSF) *vertexSF = sfVert; 6353b0fe842aSMatthew G. Knepley else PetscCall(PetscSFDestroy(&sfVert)); 6354b0fe842aSMatthew G. Knepley PetscFunctionReturn(PETSC_SUCCESS); 6355b0fe842aSMatthew G. Knepley } 6356b0fe842aSMatthew G. Knepley 6357b0fe842aSMatthew G. Knepley /*@ 6358a1cb98faSBarry Smith DMPlexBuildFromCellList - Build `DMPLEX` topology from a list of vertices for each cell (common mesh generator output) 6359a1cb98faSBarry Smith 636020f4b53cSBarry Smith Collective; No Fortran Support 63619298eaa6SMatthew G Knepley 63629298eaa6SMatthew G Knepley Input Parameters: 6363a1cb98faSBarry Smith + dm - The `DM` 6364b09969d6SVaclav Hapla . numCells - The number of cells owned by this process 6365a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DETERMINE` 63669298eaa6SMatthew G Knepley . numCorners - The number of vertices for each cell 6367a3b724e8SBarry Smith - cells - An array of `numCells` x `numCorners` numbers, the global vertex numbers for each cell 63689298eaa6SMatthew G Knepley 6369b09969d6SVaclav Hapla Level: advanced 63709298eaa6SMatthew G Knepley 6371b09969d6SVaclav Hapla Notes: 6372b09969d6SVaclav Hapla Two triangles sharing a face 6373a1cb98faSBarry Smith .vb 63749298eaa6SMatthew G Knepley 6375a1cb98faSBarry Smith 2 6376a1cb98faSBarry Smith / | \ 6377a1cb98faSBarry Smith / | \ 6378a1cb98faSBarry Smith / | \ 6379a1cb98faSBarry Smith 0 0 | 1 3 6380a1cb98faSBarry Smith \ | / 6381a1cb98faSBarry Smith \ | / 6382a1cb98faSBarry Smith \ | / 6383a1cb98faSBarry Smith 1 6384a1cb98faSBarry Smith .ve 6385a1cb98faSBarry Smith would have input 6386a1cb98faSBarry Smith .vb 6387a1cb98faSBarry Smith numCells = 2, numVertices = 4 6388a1cb98faSBarry Smith cells = [0 1 2 1 3 2] 6389a1cb98faSBarry Smith .ve 6390a1cb98faSBarry Smith which would result in the `DMPLEX` 6391a1cb98faSBarry Smith .vb 6392a1cb98faSBarry Smith 6393a1cb98faSBarry Smith 4 6394a1cb98faSBarry Smith / | \ 6395a1cb98faSBarry Smith / | \ 6396a1cb98faSBarry Smith / | \ 6397a1cb98faSBarry Smith 2 0 | 1 5 6398a1cb98faSBarry Smith \ | / 6399a1cb98faSBarry Smith \ | / 6400a1cb98faSBarry Smith \ | / 6401a1cb98faSBarry Smith 3 6402a1cb98faSBarry Smith .ve 6403a1cb98faSBarry Smith 6404a1cb98faSBarry Smith If numVertices is `PETSC_DETERMINE`, it is computed by PETSc as the maximum vertex index in cells + 1. 640525b6865aSVaclav Hapla 64061cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildFromCellListParallel()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromCellListPetsc()` 6407b09969d6SVaclav Hapla @*/ 6408d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildFromCellList(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const PetscInt cells[]) 6409d71ae5a4SJacob Faibussowitsch { 6410961cfab0SVaclav Hapla PetscInt *cones, c, p, dim; 6411b09969d6SVaclav Hapla 6412b09969d6SVaclav Hapla PetscFunctionBegin; 64139566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 64149566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 641525b6865aSVaclav Hapla /* Get/check global number of vertices */ 641625b6865aSVaclav Hapla { 641725b6865aSVaclav Hapla PetscInt NVerticesInCells, i; 641825b6865aSVaclav Hapla const PetscInt len = numCells * numCorners; 641925b6865aSVaclav Hapla 642025b6865aSVaclav Hapla /* NVerticesInCells = max(cells) + 1 */ 64211690c2aeSBarry Smith NVerticesInCells = PETSC_INT_MIN; 64229371c9d4SSatish Balay for (i = 0; i < len; i++) 64239371c9d4SSatish Balay if (cells[i] > NVerticesInCells) NVerticesInCells = cells[i]; 642425b6865aSVaclav Hapla ++NVerticesInCells; 642525b6865aSVaclav Hapla 642625b6865aSVaclav Hapla if (numVertices == PETSC_DECIDE) numVertices = NVerticesInCells; 64279371c9d4SSatish Balay else 64289371c9d4SSatish 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); 642925b6865aSVaclav Hapla } 64309566063dSJacob Faibussowitsch PetscCall(DMPlexSetChart(dm, 0, numCells + numVertices)); 643148a46eb9SPierre Jolivet for (c = 0; c < numCells; ++c) PetscCall(DMPlexSetConeSize(dm, c, numCorners)); 64329566063dSJacob Faibussowitsch PetscCall(DMSetUp(dm)); 64339566063dSJacob Faibussowitsch PetscCall(DMPlexGetCones(dm, &cones)); 6434b09969d6SVaclav Hapla for (c = 0; c < numCells; ++c) { 6435ad540459SPierre Jolivet for (p = 0; p < numCorners; ++p) cones[c * numCorners + p] = cells[c * numCorners + p] + numCells; 6436b09969d6SVaclav Hapla } 64379566063dSJacob Faibussowitsch PetscCall(DMPlexSymmetrize(dm)); 64389566063dSJacob Faibussowitsch PetscCall(DMPlexStratify(dm)); 64399566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(DMPLEX_BuildFromCellList, dm, 0, 0, 0)); 64403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6441b09969d6SVaclav Hapla } 6442b09969d6SVaclav Hapla 6443cc4c1da9SBarry Smith /*@ 6444a1cb98faSBarry Smith DMPlexBuildCoordinatesFromCellList - Build `DM` coordinates from a list of coordinates for each owned vertex (common mesh generator output) 6445a1cb98faSBarry Smith 6446cc4c1da9SBarry Smith Collective 6447b09969d6SVaclav Hapla 6448b09969d6SVaclav Hapla Input Parameters: 6449a1cb98faSBarry Smith + dm - The `DM` 6450b09969d6SVaclav Hapla . spaceDim - The spatial dimension used for coordinates 6451b09969d6SVaclav Hapla - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex 6452b09969d6SVaclav Hapla 6453b09969d6SVaclav Hapla Level: advanced 6454b09969d6SVaclav Hapla 64551cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexBuildCoordinatesFromCellListParallel()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexBuildFromCellList()` 6456b09969d6SVaclav Hapla @*/ 6457d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexBuildCoordinatesFromCellList(DM dm, PetscInt spaceDim, const PetscReal vertexCoords[]) 6458d71ae5a4SJacob Faibussowitsch { 6459b09969d6SVaclav Hapla PetscSection coordSection; 6460b09969d6SVaclav Hapla Vec coordinates; 6461b09969d6SVaclav Hapla DM cdm; 6462b09969d6SVaclav Hapla PetscScalar *coords; 64631edcf0b2SVaclav Hapla PetscInt v, vStart, vEnd, d; 6464b09969d6SVaclav Hapla 6465b09969d6SVaclav Hapla PetscFunctionBegin; 64669566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0)); 64679566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd)); 64681dca8a05SBarry Smith PetscCheck(vStart >= 0 && vEnd >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "DM is not set up properly. DMPlexBuildFromCellList() should be called first."); 64699566063dSJacob Faibussowitsch PetscCall(DMSetCoordinateDim(dm, spaceDim)); 64709566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 64719566063dSJacob Faibussowitsch PetscCall(PetscSectionSetNumFields(coordSection, 1)); 64729566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldComponents(coordSection, 0, spaceDim)); 64739566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(coordSection, vStart, vEnd)); 64741edcf0b2SVaclav Hapla for (v = vStart; v < vEnd; ++v) { 64759566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(coordSection, v, spaceDim)); 64769566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, spaceDim)); 6477b09969d6SVaclav Hapla } 64789566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(coordSection)); 6479b09969d6SVaclav Hapla 64809566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDM(dm, &cdm)); 64819566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(cdm, &coordinates)); 64829566063dSJacob Faibussowitsch PetscCall(VecSetBlockSize(coordinates, spaceDim)); 64839566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates")); 64849566063dSJacob Faibussowitsch PetscCall(VecGetArrayWrite(coordinates, &coords)); 64851edcf0b2SVaclav Hapla for (v = 0; v < vEnd - vStart; ++v) { 6486ad540459SPierre Jolivet for (d = 0; d < spaceDim; ++d) coords[v * spaceDim + d] = vertexCoords[v * spaceDim + d]; 6487b09969d6SVaclav Hapla } 64889566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayWrite(coordinates, &coords)); 64899566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(dm, coordinates)); 64909566063dSJacob Faibussowitsch PetscCall(VecDestroy(&coordinates)); 64919566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(DMPLEX_BuildCoordinatesFromCellList, dm, 0, 0, 0)); 64923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6493b09969d6SVaclav Hapla } 6494b09969d6SVaclav Hapla 6495b09969d6SVaclav Hapla /*@ 6496a1cb98faSBarry Smith DMPlexCreateFromCellListPetsc - Create `DMPLEX` from a list of vertices for each cell (common mesh generator output), but only process 0 takes in the input 64973df08285SMatthew G. Knepley 6498a1cb98faSBarry Smith Collective 6499b09969d6SVaclav Hapla 6500b09969d6SVaclav Hapla Input Parameters: 6501b09969d6SVaclav Hapla + comm - The communicator 6502b09969d6SVaclav Hapla . dim - The topological dimension of the mesh 65033df08285SMatthew G. Knepley . numCells - The number of cells, only on process 0 6504a1cb98faSBarry Smith . numVertices - The number of vertices owned by this process, or `PETSC_DECIDE`, only on process 0 65053df08285SMatthew G. Knepley . numCorners - The number of vertices for each cell, only on process 0 6506b09969d6SVaclav Hapla . interpolate - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically 6507ce78bad3SBarry Smith . cells - An array of $ numCells \times numCorners$ numbers, the vertices for each cell, only on process 0 6508b09969d6SVaclav Hapla . spaceDim - The spatial dimension used for coordinates 6509ce78bad3SBarry Smith - vertexCoords - An array of $ numVertices \times spaceDim$ numbers, the coordinates of each vertex, only on process 0 6510b09969d6SVaclav Hapla 6511b09969d6SVaclav Hapla Output Parameter: 6512a1cb98faSBarry Smith . dm - The `DM`, which only has points on process 0 651325b6865aSVaclav Hapla 6514b09969d6SVaclav Hapla Level: intermediate 6515b09969d6SVaclav Hapla 6516a1cb98faSBarry Smith Notes: 6517a1cb98faSBarry Smith This function is just a convenient sequence of `DMCreate()`, `DMSetType()`, `DMSetDimension()`, `DMPlexBuildFromCellList()`, 6518a1cb98faSBarry Smith `DMPlexInterpolate()`, `DMPlexBuildCoordinatesFromCellList()` 6519a1cb98faSBarry Smith 6520a1cb98faSBarry Smith See `DMPlexBuildFromCellList()` for an example and details about the topology-related parameters. 6521a1cb98faSBarry Smith See `DMPlexBuildCoordinatesFromCellList()` for details about the geometry-related parameters. 6522a1cb98faSBarry Smith See `DMPlexCreateFromCellListParallelPetsc()` for parallel input 6523a1cb98faSBarry Smith 65241cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListParallelPetsc()`, `DMPlexBuildFromCellList()`, `DMPlexBuildCoordinatesFromCellList()`, `DMPlexCreateFromDAG()`, `DMPlexCreate()` 65259298eaa6SMatthew G Knepley @*/ 6526d71ae5a4SJacob 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) 6527d71ae5a4SJacob Faibussowitsch { 65283df08285SMatthew G. Knepley PetscMPIInt rank; 65299298eaa6SMatthew G Knepley 65309298eaa6SMatthew G Knepley PetscFunctionBegin; 653128b400f6SJacob 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."); 65329566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 65339566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 65349566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 65359566063dSJacob Faibussowitsch PetscCall(DMSetDimension(*dm, dim)); 6536c5853193SPierre Jolivet if (rank == 0) PetscCall(DMPlexBuildFromCellList(*dm, numCells, numVertices, numCorners, cells)); 65379566063dSJacob Faibussowitsch else PetscCall(DMPlexBuildFromCellList(*dm, 0, 0, 0, NULL)); 65389298eaa6SMatthew G Knepley if (interpolate) { 65395fd9971aSMatthew G. Knepley DM idm; 65409298eaa6SMatthew G Knepley 65419566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(*dm, &idm)); 65429566063dSJacob Faibussowitsch PetscCall(DMDestroy(dm)); 65439298eaa6SMatthew G Knepley *dm = idm; 65449298eaa6SMatthew G Knepley } 6545c5853193SPierre Jolivet if (rank == 0) PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, vertexCoords)); 65469566063dSJacob Faibussowitsch else PetscCall(DMPlexBuildCoordinatesFromCellList(*dm, spaceDim, NULL)); 65473ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 65489298eaa6SMatthew G Knepley } 65499298eaa6SMatthew G Knepley 6550939f6067SMatthew G. Knepley /*@ 655120f4b53cSBarry Smith DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a `DM` 6552939f6067SMatthew G. Knepley 6553939f6067SMatthew G. Knepley Input Parameters: 655420f4b53cSBarry Smith + dm - The empty `DM` object, usually from `DMCreate()` and `DMSetDimension()` 6555939f6067SMatthew G. Knepley . depth - The depth of the DAG 6556ce78bad3SBarry Smith . numPoints - Array of size $ depth + 1 $ containing the number of points at each `depth` 6557939f6067SMatthew G. Knepley . coneSize - The cone size of each point 6558939f6067SMatthew G. Knepley . cones - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point 6559939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point 6560ce78bad3SBarry Smith - vertexCoords - An array of $ numPoints[0] \times spacedim $ numbers representing the coordinates of each vertex, with `spacedim` the value set via `DMSetCoordinateDim()` 6561939f6067SMatthew G. Knepley 6562939f6067SMatthew G. Knepley Output Parameter: 656320f4b53cSBarry Smith . dm - The `DM` 656420f4b53cSBarry Smith 656520f4b53cSBarry Smith Level: advanced 6566939f6067SMatthew G. Knepley 6567a1cb98faSBarry Smith Note: 6568a1cb98faSBarry Smith Two triangles sharing a face would have input 6569a1cb98faSBarry Smith .vb 6570a1cb98faSBarry Smith depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0] 6571a1cb98faSBarry Smith cones = [2 3 4 3 5 4], coneOrientations = [0 0 0 0 0 0] 6572a1cb98faSBarry Smith vertexCoords = [-1.0 0.0 0.0 -1.0 0.0 1.0 1.0 0.0] 6573a1cb98faSBarry Smith .ve 6574939f6067SMatthew G. Knepley which would result in the DMPlex 6575a1cb98faSBarry Smith .vb 6576a1cb98faSBarry Smith 4 6577a1cb98faSBarry Smith / | \ 6578a1cb98faSBarry Smith / | \ 6579a1cb98faSBarry Smith / | \ 6580a1cb98faSBarry Smith 2 0 | 1 5 6581a1cb98faSBarry Smith \ | / 6582a1cb98faSBarry Smith \ | / 6583a1cb98faSBarry Smith \ | / 6584a1cb98faSBarry Smith 3 6585a1cb98faSBarry Smith .ve 6586a1cb98faSBarry Smith Notice that all points are numbered consecutively, unlike `DMPlexCreateFromCellListPetsc()` 6587939f6067SMatthew G. Knepley 6588ce78bad3SBarry Smith Developer Note: 6589ce78bad3SBarry Smith This does not create anything so should not have create in the name. 6590ce78bad3SBarry Smith 65911cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()` 6592939f6067SMatthew G. Knepley @*/ 6593d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[]) 6594d71ae5a4SJacob Faibussowitsch { 65959298eaa6SMatthew G Knepley Vec coordinates; 65969298eaa6SMatthew G Knepley PetscSection coordSection; 65979298eaa6SMatthew G Knepley PetscScalar *coords; 6598811e8653SToby Isaac PetscInt coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off; 65999298eaa6SMatthew G Knepley 66009298eaa6SMatthew G Knepley PetscFunctionBegin; 66019566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 66029566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDim(dm, &dimEmbed)); 660363a3b9bcSJacob Faibussowitsch PetscCheck(dimEmbed >= dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Embedding dimension %" PetscInt_FMT " cannot be less than intrinsic dimension %" PetscInt_FMT, dimEmbed, dim); 66049298eaa6SMatthew G Knepley for (d = 0; d <= depth; ++d) pEnd += numPoints[d]; 66059566063dSJacob Faibussowitsch PetscCall(DMPlexSetChart(dm, pStart, pEnd)); 66069298eaa6SMatthew G Knepley for (p = pStart; p < pEnd; ++p) { 66079566063dSJacob Faibussowitsch PetscCall(DMPlexSetConeSize(dm, p, coneSize[p - pStart])); 6608ad540459SPierre Jolivet if (firstVertex < 0 && !coneSize[p - pStart]) firstVertex = p - pStart; 660997e052ccSToby Isaac } 66101dca8a05SBarry Smith PetscCheck(firstVertex >= 0 || !numPoints[0], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Expected %" PetscInt_FMT " vertices but could not find any", numPoints[0]); 66119566063dSJacob Faibussowitsch PetscCall(DMSetUp(dm)); /* Allocate space for cones */ 66129298eaa6SMatthew G Knepley for (p = pStart, off = 0; p < pEnd; off += coneSize[p - pStart], ++p) { 66139566063dSJacob Faibussowitsch PetscCall(DMPlexSetCone(dm, p, &cones[off])); 66149566063dSJacob Faibussowitsch PetscCall(DMPlexSetConeOrientation(dm, p, &coneOrientations[off])); 66159298eaa6SMatthew G Knepley } 66169566063dSJacob Faibussowitsch PetscCall(DMPlexSymmetrize(dm)); 66179566063dSJacob Faibussowitsch PetscCall(DMPlexStratify(dm)); 66189298eaa6SMatthew G Knepley /* Build coordinates */ 66199566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 66209566063dSJacob Faibussowitsch PetscCall(PetscSectionSetNumFields(coordSection, 1)); 66219566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dimEmbed)); 66229566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(coordSection, firstVertex, firstVertex + numPoints[0])); 66239298eaa6SMatthew G Knepley for (v = firstVertex; v < firstVertex + numPoints[0]; ++v) { 66249566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(coordSection, v, dimEmbed)); 66259566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed)); 66269298eaa6SMatthew G Knepley } 66279566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(coordSection)); 66289566063dSJacob Faibussowitsch PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize)); 66299566063dSJacob Faibussowitsch PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates)); 66309566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates")); 66319566063dSJacob Faibussowitsch PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE)); 663258b7e2c1SStefano Zampini PetscCall(VecSetBlockSize(coordinates, PetscMax(dimEmbed, 1))); 66339566063dSJacob Faibussowitsch PetscCall(VecSetType(coordinates, VECSTANDARD)); 66349318fe57SMatthew G. Knepley if (vertexCoords) { 66359566063dSJacob Faibussowitsch PetscCall(VecGetArray(coordinates, &coords)); 66369298eaa6SMatthew G Knepley for (v = 0; v < numPoints[0]; ++v) { 66379298eaa6SMatthew G Knepley PetscInt off; 66389298eaa6SMatthew G Knepley 66399566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(coordSection, v + firstVertex, &off)); 6640ad540459SPierre Jolivet for (d = 0; d < dimEmbed; ++d) coords[off + d] = vertexCoords[v * dimEmbed + d]; 66419298eaa6SMatthew G Knepley } 66429318fe57SMatthew G. Knepley } 66439566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(coordinates, &coords)); 66449566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(dm, coordinates)); 66459566063dSJacob Faibussowitsch PetscCall(VecDestroy(&coordinates)); 66463ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 66479298eaa6SMatthew G Knepley } 66488415267dSToby Isaac 6649a4e35b19SJacob Faibussowitsch /* 6650a1cb98faSBarry Smith DMPlexCreateCellVertexFromFile - Create a `DMPLEX` mesh from a simple cell-vertex file. 6651a1cb98faSBarry Smith 6652a1cb98faSBarry Smith Collective 66538ca92349SMatthew G. Knepley 66548ca92349SMatthew G. Knepley + comm - The MPI communicator 66558ca92349SMatthew G. Knepley . filename - Name of the .dat file 66568ca92349SMatthew G. Knepley - interpolate - Create faces and edges in the mesh 66578ca92349SMatthew G. Knepley 66588ca92349SMatthew G. Knepley Output Parameter: 6659a1cb98faSBarry Smith . dm - The `DM` object representing the mesh 66608ca92349SMatthew G. Knepley 66618ca92349SMatthew G. Knepley Level: beginner 66628ca92349SMatthew G. Knepley 6663a1cb98faSBarry Smith Note: 6664a1cb98faSBarry Smith The format is the simplest possible: 6665a1cb98faSBarry Smith .vb 6666d0812dedSMatthew G. Knepley dim Ne Nv Nc Nl 6667d0812dedSMatthew G. Knepley v_1 v_2 ... v_Nc 6668d0812dedSMatthew G. Knepley ... 6669d0812dedSMatthew G. Knepley x y z marker_1 ... marker_Nl 6670a1cb98faSBarry Smith .ve 6671a1cb98faSBarry Smith 6672a1cb98faSBarry Smith Developer Note: 6673a1cb98faSBarry Smith Should use a `PetscViewer` not a filename 6674a1cb98faSBarry Smith 66756afe31f6SMartin Diehl .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateGmsh()`, `DMPlexCreate()` 6676a4e35b19SJacob Faibussowitsch */ 6677ff6a9541SJacob Faibussowitsch static PetscErrorCode DMPlexCreateCellVertexFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm) 6678d71ae5a4SJacob Faibussowitsch { 66798ca92349SMatthew G. Knepley DMLabel marker; 66808ca92349SMatthew G. Knepley PetscViewer viewer; 66818ca92349SMatthew G. Knepley Vec coordinates; 66828ca92349SMatthew G. Knepley PetscSection coordSection; 66838ca92349SMatthew G. Knepley PetscScalar *coords; 66848ca92349SMatthew G. Knepley char line[PETSC_MAX_PATH_LEN]; 6685d0812dedSMatthew G. Knepley PetscInt cdim, coordSize, v, c, d; 66868ca92349SMatthew G. Knepley PetscMPIInt rank; 6687d0812dedSMatthew G. Knepley int snum, dim, Nv, Nc, Ncn, Nl; 66888ca92349SMatthew G. Knepley 66898ca92349SMatthew G. Knepley PetscFunctionBegin; 66909566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 66919566063dSJacob Faibussowitsch PetscCall(PetscViewerCreate(comm, &viewer)); 66929566063dSJacob Faibussowitsch PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII)); 66939566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ)); 66949566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetName(viewer, filename)); 6695dd400576SPatrick Sanan if (rank == 0) { 6696d0812dedSMatthew G. Knepley PetscCall(PetscViewerRead(viewer, line, 5, NULL, PETSC_STRING)); 6697d0812dedSMatthew G. Knepley snum = sscanf(line, "%d %d %d %d %d", &dim, &Nc, &Nv, &Ncn, &Nl); 6698d0812dedSMatthew G. Knepley PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line); 669925ce1634SJed Brown } else { 6700f8d5e320SMatthew G. Knepley Nc = Nv = Ncn = Nl = 0; 67018ca92349SMatthew G. Knepley } 6702d0812dedSMatthew G. Knepley PetscCallMPI(MPI_Bcast(&dim, 1, MPI_INT, 0, comm)); 6703835f2295SStefano Zampini cdim = dim; 67049566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 67059566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 67069566063dSJacob Faibussowitsch PetscCall(DMPlexSetChart(*dm, 0, Nc + Nv)); 6707835f2295SStefano Zampini PetscCall(DMSetDimension(*dm, dim)); 67089566063dSJacob Faibussowitsch PetscCall(DMSetCoordinateDim(*dm, cdim)); 67098ca92349SMatthew G. Knepley /* Read topology */ 6710dd400576SPatrick Sanan if (rank == 0) { 6711f8d5e320SMatthew G. Knepley char format[PETSC_MAX_PATH_LEN]; 6712f8d5e320SMatthew G. Knepley PetscInt cone[8]; 67138ca92349SMatthew G. Knepley int vbuf[8], v; 67148ca92349SMatthew G. Knepley 67159371c9d4SSatish Balay for (c = 0; c < Ncn; ++c) { 67169371c9d4SSatish Balay format[c * 3 + 0] = '%'; 67179371c9d4SSatish Balay format[c * 3 + 1] = 'd'; 67189371c9d4SSatish Balay format[c * 3 + 2] = ' '; 67199371c9d4SSatish Balay } 6720f8d5e320SMatthew G. Knepley format[Ncn * 3 - 1] = '\0'; 67219566063dSJacob Faibussowitsch for (c = 0; c < Nc; ++c) PetscCall(DMPlexSetConeSize(*dm, c, Ncn)); 67229566063dSJacob Faibussowitsch PetscCall(DMSetUp(*dm)); 67238ca92349SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 67249566063dSJacob Faibussowitsch PetscCall(PetscViewerRead(viewer, line, Ncn, NULL, PETSC_STRING)); 6725f8d5e320SMatthew G. Knepley switch (Ncn) { 6726d71ae5a4SJacob Faibussowitsch case 2: 6727d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &vbuf[0], &vbuf[1]); 6728d71ae5a4SJacob Faibussowitsch break; 6729d71ae5a4SJacob Faibussowitsch case 3: 6730d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2]); 6731d71ae5a4SJacob Faibussowitsch break; 6732d71ae5a4SJacob Faibussowitsch case 4: 6733d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3]); 6734d71ae5a4SJacob Faibussowitsch break; 6735d71ae5a4SJacob Faibussowitsch case 6: 6736d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5]); 6737d71ae5a4SJacob Faibussowitsch break; 6738d71ae5a4SJacob Faibussowitsch case 8: 6739d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &vbuf[0], &vbuf[1], &vbuf[2], &vbuf[3], &vbuf[4], &vbuf[5], &vbuf[6], &vbuf[7]); 6740d71ae5a4SJacob Faibussowitsch break; 6741d71ae5a4SJacob Faibussowitsch default: 6742d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No cell shape with %d vertices", Ncn); 6743f8d5e320SMatthew G. Knepley } 674408401ef6SPierre Jolivet PetscCheck(snum == Ncn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line); 6745f8d5e320SMatthew G. Knepley for (v = 0; v < Ncn; ++v) cone[v] = vbuf[v] + Nc; 67468ca92349SMatthew G. Knepley /* Hexahedra are inverted */ 6747f8d5e320SMatthew G. Knepley if (Ncn == 8) { 67488ca92349SMatthew G. Knepley PetscInt tmp = cone[1]; 67498ca92349SMatthew G. Knepley cone[1] = cone[3]; 67508ca92349SMatthew G. Knepley cone[3] = tmp; 67518ca92349SMatthew G. Knepley } 67529566063dSJacob Faibussowitsch PetscCall(DMPlexSetCone(*dm, c, cone)); 67538ca92349SMatthew G. Knepley } 67548ca92349SMatthew G. Knepley } 67559566063dSJacob Faibussowitsch PetscCall(DMPlexSymmetrize(*dm)); 67569566063dSJacob Faibussowitsch PetscCall(DMPlexStratify(*dm)); 67578ca92349SMatthew G. Knepley /* Read coordinates */ 67589566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(*dm, &coordSection)); 67599566063dSJacob Faibussowitsch PetscCall(PetscSectionSetNumFields(coordSection, 1)); 67609566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldComponents(coordSection, 0, cdim)); 67619566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(coordSection, Nc, Nc + Nv)); 67628ca92349SMatthew G. Knepley for (v = Nc; v < Nc + Nv; ++v) { 67639566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(coordSection, v, cdim)); 67649566063dSJacob Faibussowitsch PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, cdim)); 67658ca92349SMatthew G. Knepley } 67669566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(coordSection)); 67679566063dSJacob Faibussowitsch PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize)); 67689566063dSJacob Faibussowitsch PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates)); 67699566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates")); 67709566063dSJacob Faibussowitsch PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE)); 67719566063dSJacob Faibussowitsch PetscCall(VecSetBlockSize(coordinates, cdim)); 67729566063dSJacob Faibussowitsch PetscCall(VecSetType(coordinates, VECSTANDARD)); 67739566063dSJacob Faibussowitsch PetscCall(VecGetArray(coordinates, &coords)); 6774dd400576SPatrick Sanan if (rank == 0) { 6775f8d5e320SMatthew G. Knepley char format[PETSC_MAX_PATH_LEN]; 67768ca92349SMatthew G. Knepley double x[3]; 6777f8d5e320SMatthew G. Knepley int l, val[3]; 67788ca92349SMatthew G. Knepley 6779f8d5e320SMatthew G. Knepley if (Nl) { 67809371c9d4SSatish Balay for (l = 0; l < Nl; ++l) { 67819371c9d4SSatish Balay format[l * 3 + 0] = '%'; 67829371c9d4SSatish Balay format[l * 3 + 1] = 'd'; 67839371c9d4SSatish Balay format[l * 3 + 2] = ' '; 67849371c9d4SSatish Balay } 6785f8d5e320SMatthew G. Knepley format[Nl * 3 - 1] = '\0'; 67869566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(*dm, "marker")); 67879566063dSJacob Faibussowitsch PetscCall(DMGetLabel(*dm, "marker", &marker)); 6788f8d5e320SMatthew G. Knepley } 67898ca92349SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 67909566063dSJacob Faibussowitsch PetscCall(PetscViewerRead(viewer, line, 3 + Nl, NULL, PETSC_STRING)); 6791f8d5e320SMatthew G. Knepley snum = sscanf(line, "%lg %lg %lg", &x[0], &x[1], &x[2]); 679208401ef6SPierre Jolivet PetscCheck(snum == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line); 6793f8d5e320SMatthew G. Knepley switch (Nl) { 6794d71ae5a4SJacob Faibussowitsch case 0: 6795d71ae5a4SJacob Faibussowitsch snum = 0; 6796d71ae5a4SJacob Faibussowitsch break; 6797d71ae5a4SJacob Faibussowitsch case 1: 6798d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &val[0]); 6799d71ae5a4SJacob Faibussowitsch break; 6800d71ae5a4SJacob Faibussowitsch case 2: 6801d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &val[0], &val[1]); 6802d71ae5a4SJacob Faibussowitsch break; 6803d71ae5a4SJacob Faibussowitsch case 3: 6804d71ae5a4SJacob Faibussowitsch snum = sscanf(line, format, &val[0], &val[1], &val[2]); 6805d71ae5a4SJacob Faibussowitsch break; 6806d71ae5a4SJacob Faibussowitsch default: 6807d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Request support for %d labels", Nl); 6808f8d5e320SMatthew G. Knepley } 680908401ef6SPierre Jolivet PetscCheck(snum == Nl, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unable to parse cell-vertex file: %s", line); 68108ca92349SMatthew G. Knepley for (d = 0; d < cdim; ++d) coords[v * cdim + d] = x[d]; 68119566063dSJacob Faibussowitsch for (l = 0; l < Nl; ++l) PetscCall(DMLabelSetValue(marker, v + Nc, val[l])); 68128ca92349SMatthew G. Knepley } 68138ca92349SMatthew G. Knepley } 68149566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(coordinates, &coords)); 68159566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(*dm, coordinates)); 68169566063dSJacob Faibussowitsch PetscCall(VecDestroy(&coordinates)); 68179566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 68188ca92349SMatthew G. Knepley if (interpolate) { 68198ca92349SMatthew G. Knepley DM idm; 68208ca92349SMatthew G. Knepley DMLabel bdlabel; 68218ca92349SMatthew G. Knepley 68229566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(*dm, &idm)); 68239566063dSJacob Faibussowitsch PetscCall(DMDestroy(dm)); 68248ca92349SMatthew G. Knepley *dm = idm; 68258ca92349SMatthew G. Knepley 6826f8d5e320SMatthew G. Knepley if (!Nl) { 68279566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(*dm, "marker")); 68289566063dSJacob Faibussowitsch PetscCall(DMGetLabel(*dm, "marker", &bdlabel)); 68299566063dSJacob Faibussowitsch PetscCall(DMPlexMarkBoundaryFaces(*dm, PETSC_DETERMINE, bdlabel)); 68309566063dSJacob Faibussowitsch PetscCall(DMPlexLabelComplete(*dm, bdlabel)); 68318ca92349SMatthew G. Knepley } 6832f8d5e320SMatthew G. Knepley } 68333ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 68348ca92349SMatthew G. Knepley } 68358ca92349SMatthew G. Knepley 6836cc4c1da9SBarry Smith /*@ 6837a1cb98faSBarry Smith DMPlexCreateFromFile - This takes a filename and produces a `DM` 6838a1cb98faSBarry Smith 6839a1cb98faSBarry Smith Collective 6840ca522641SMatthew G. Knepley 6841ca522641SMatthew G. Knepley Input Parameters: 6842ca522641SMatthew G. Knepley + comm - The communicator 6843ca522641SMatthew G. Knepley . filename - A file name 6844a1cb98faSBarry Smith . plexname - The object name of the resulting `DM`, also used for intra-datafile lookup by some formats 6845ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces) 6846ca522641SMatthew G. Knepley 6847ca522641SMatthew G. Knepley Output Parameter: 6848a1cb98faSBarry Smith . dm - The `DM` 6849ca522641SMatthew G. Knepley 6850a1cb98faSBarry Smith Options Database Key: 6851a1cb98faSBarry Smith . -dm_plex_create_from_hdf5_xdmf - use the `PETSC_VIEWER_HDF5_XDMF` format for reading HDF5 685202ef0d99SVaclav Hapla 685337fdd005SBarry Smith Use `-dm_plex_create_ prefix` to pass options to the internal `PetscViewer`, e.g. 6854bca97951SVaclav Hapla $ -dm_plex_create_viewer_hdf5_collective 6855bca97951SVaclav Hapla 6856ca522641SMatthew G. Knepley Level: beginner 6857ca522641SMatthew G. Knepley 6858a1cb98faSBarry Smith Notes: 6859a1cb98faSBarry Smith Using `PETSCVIEWERHDF5` type with `PETSC_VIEWER_HDF5_PETSC` format, one can save multiple `DMPLEX` 6860a1cb98faSBarry Smith meshes in a single HDF5 file. This in turn requires one to name the `DMPLEX` object with `PetscObjectSetName()` 6861a1cb98faSBarry Smith before saving it with `DMView()` and before loading it with `DMLoad()` for identification of the mesh object. 6862a1cb98faSBarry Smith The input parameter name is thus used to name the `DMPLEX` object when `DMPlexCreateFromFile()` internally 6863a1cb98faSBarry Smith calls `DMLoad()`. Currently, name is ignored for other viewer types and/or formats. 6864a1cb98faSBarry Smith 68651cc06b55SBarry Smith .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()`, `PetscObjectSetName()`, `DMView()`, `DMLoad()` 6866ca522641SMatthew G. Knepley @*/ 6867d71ae5a4SJacob Faibussowitsch PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], const char plexname[], PetscBool interpolate, DM *dm) 6868d71ae5a4SJacob Faibussowitsch { 6869ef3a5affSJacob Faibussowitsch const char extGmsh[] = ".msh"; 6870ef3a5affSJacob Faibussowitsch const char extGmsh2[] = ".msh2"; 6871ef3a5affSJacob Faibussowitsch const char extGmsh4[] = ".msh4"; 6872ef3a5affSJacob Faibussowitsch const char extCGNS[] = ".cgns"; 6873ef3a5affSJacob Faibussowitsch const char extExodus[] = ".exo"; 6874ef3a5affSJacob Faibussowitsch const char extExodus_e[] = ".e"; 6875ef3a5affSJacob Faibussowitsch const char extGenesis[] = ".gen"; 6876ef3a5affSJacob Faibussowitsch const char extFluent[] = ".cas"; 6877ef3a5affSJacob Faibussowitsch const char extHDF5[] = ".h5"; 68786f2c871aSStefano Zampini const char extXDMFHDF5[] = ".xdmf.h5"; 6879ef3a5affSJacob Faibussowitsch const char extPLY[] = ".ply"; 68805552b385SBrandon const char extEGADSlite[] = ".egadslite"; 6881ef3a5affSJacob Faibussowitsch const char extEGADS[] = ".egads"; 6882ef3a5affSJacob Faibussowitsch const char extIGES[] = ".igs"; 68835552b385SBrandon const char extIGES2[] = ".iges"; 6884ef3a5affSJacob Faibussowitsch const char extSTEP[] = ".stp"; 68855552b385SBrandon const char extSTEP2[] = ".step"; 68865552b385SBrandon const char extBREP[] = ".brep"; 6887ef3a5affSJacob Faibussowitsch const char extCV[] = ".dat"; 6888ca522641SMatthew G. Knepley size_t len; 68895552b385SBrandon PetscBool isGmsh, isGmsh2, isGmsh4, isCGNS, isExodus, isGenesis, isFluent, isHDF5, isPLY, isEGADSlite, isEGADS, isIGES, isIGES2, isSTEP, isSTEP2, isBREP, isCV, isXDMFHDF5; 6890ca522641SMatthew G. Knepley PetscMPIInt rank; 6891ca522641SMatthew G. Knepley 6892ca522641SMatthew G. Knepley PetscFunctionBegin; 68934f572ea9SToby Isaac PetscAssertPointer(filename, 2); 68944f572ea9SToby Isaac if (plexname) PetscAssertPointer(plexname, 3); 68954f572ea9SToby Isaac PetscAssertPointer(dm, 5); 68969566063dSJacob Faibussowitsch PetscCall(DMInitializePackage()); 68979566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(DMPLEX_CreateFromFile, 0, 0, 0, 0)); 68989566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 68999566063dSJacob Faibussowitsch PetscCall(PetscStrlen(filename, &len)); 690028b400f6SJacob Faibussowitsch PetscCheck(len, comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path"); 6901ef3a5affSJacob Faibussowitsch 69029371c9d4SSatish Balay #define CheckExtension(extension__, is_extension__) \ 69039371c9d4SSatish Balay do { \ 6904274aaeaaSJacob Faibussowitsch PetscAssert(sizeof(extension__), comm, PETSC_ERR_PLIB, "Zero-size extension: %s", extension__); \ 6905274aaeaaSJacob Faibussowitsch /* don't count the null-terminator at the end */ \ 6906274aaeaaSJacob Faibussowitsch const size_t ext_len = sizeof(extension__) - 1; \ 6907274aaeaaSJacob Faibussowitsch if (len < ext_len) { \ 6908ef3a5affSJacob Faibussowitsch is_extension__ = PETSC_FALSE; \ 6909ef3a5affSJacob Faibussowitsch } else { \ 6910274aaeaaSJacob Faibussowitsch PetscCall(PetscStrncmp(filename + len - ext_len, extension__, ext_len, &is_extension__)); \ 6911ef3a5affSJacob Faibussowitsch } \ 6912ef3a5affSJacob Faibussowitsch } while (0) 6913ef3a5affSJacob Faibussowitsch 6914ef3a5affSJacob Faibussowitsch CheckExtension(extGmsh, isGmsh); 6915ef3a5affSJacob Faibussowitsch CheckExtension(extGmsh2, isGmsh2); 6916ef3a5affSJacob Faibussowitsch CheckExtension(extGmsh4, isGmsh4); 6917ef3a5affSJacob Faibussowitsch CheckExtension(extCGNS, isCGNS); 6918ef3a5affSJacob Faibussowitsch CheckExtension(extExodus, isExodus); 6919ef3a5affSJacob Faibussowitsch if (!isExodus) CheckExtension(extExodus_e, isExodus); 6920ef3a5affSJacob Faibussowitsch CheckExtension(extGenesis, isGenesis); 6921ef3a5affSJacob Faibussowitsch CheckExtension(extFluent, isFluent); 6922ef3a5affSJacob Faibussowitsch CheckExtension(extHDF5, isHDF5); 6923ef3a5affSJacob Faibussowitsch CheckExtension(extPLY, isPLY); 69245552b385SBrandon CheckExtension(extEGADSlite, isEGADSlite); 6925ef3a5affSJacob Faibussowitsch CheckExtension(extEGADS, isEGADS); 6926ef3a5affSJacob Faibussowitsch CheckExtension(extIGES, isIGES); 69275552b385SBrandon CheckExtension(extIGES2, isIGES2); 6928ef3a5affSJacob Faibussowitsch CheckExtension(extSTEP, isSTEP); 69295552b385SBrandon CheckExtension(extSTEP2, isSTEP2); 69305552b385SBrandon CheckExtension(extBREP, isBREP); 6931ef3a5affSJacob Faibussowitsch CheckExtension(extCV, isCV); 69326f2c871aSStefano Zampini CheckExtension(extXDMFHDF5, isXDMFHDF5); 6933ef3a5affSJacob Faibussowitsch 6934ef3a5affSJacob Faibussowitsch #undef CheckExtension 6935ef3a5affSJacob Faibussowitsch 6936de78e4feSLisandro Dalcin if (isGmsh || isGmsh2 || isGmsh4) { 69379566063dSJacob Faibussowitsch PetscCall(DMPlexCreateGmshFromFile(comm, filename, interpolate, dm)); 6938ca522641SMatthew G. Knepley } else if (isCGNS) { 69399566063dSJacob Faibussowitsch PetscCall(DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm)); 694090c68965SMatthew G. Knepley } else if (isExodus || isGenesis) { 69419566063dSJacob Faibussowitsch PetscCall(DMPlexCreateExodusFromFile(comm, filename, interpolate, dm)); 69422f0bd6dcSMichael Lange } else if (isFluent) { 69439566063dSJacob Faibussowitsch PetscCall(DMPlexCreateFluentFromFile(comm, filename, interpolate, dm)); 6944cc2f8f65SMatthew G. Knepley } else if (isHDF5) { 6945cc2f8f65SMatthew G. Knepley PetscViewer viewer; 6946cc2f8f65SMatthew G. Knepley 694743b242b4SVaclav 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 */ 69486f2c871aSStefano Zampini PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_create_from_hdf5_xdmf", &isXDMFHDF5, NULL)); 69499566063dSJacob Faibussowitsch PetscCall(PetscViewerCreate(comm, &viewer)); 69509566063dSJacob Faibussowitsch PetscCall(PetscViewerSetType(viewer, PETSCVIEWERHDF5)); 69519566063dSJacob Faibussowitsch PetscCall(PetscViewerSetOptionsPrefix(viewer, "dm_plex_create_")); 69529566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions(viewer)); 69539566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ)); 69549566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetName(viewer, filename)); 6955cd7e8a5eSksagiyam 69569566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, dm)); 6957f4f49eeaSPierre Jolivet PetscCall(PetscObjectSetName((PetscObject)*dm, plexname)); 69589566063dSJacob Faibussowitsch PetscCall(DMSetType(*dm, DMPLEX)); 69596f2c871aSStefano Zampini if (isXDMFHDF5) PetscCall(PetscViewerPushFormat(viewer, PETSC_VIEWER_HDF5_XDMF)); 69609566063dSJacob Faibussowitsch PetscCall(DMLoad(*dm, viewer)); 69616f2c871aSStefano Zampini if (isXDMFHDF5) PetscCall(PetscViewerPopFormat(viewer)); 69629566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 69635fd9971aSMatthew G. Knepley 69645fd9971aSMatthew G. Knepley if (interpolate) { 69655fd9971aSMatthew G. Knepley DM idm; 69665fd9971aSMatthew G. Knepley 69679566063dSJacob Faibussowitsch PetscCall(DMPlexInterpolate(*dm, &idm)); 69689566063dSJacob Faibussowitsch PetscCall(DMDestroy(dm)); 69695fd9971aSMatthew G. Knepley *dm = idm; 69705fd9971aSMatthew G. Knepley } 6971f2801cd6SMatthew G. Knepley } else if (isPLY) { 69729566063dSJacob Faibussowitsch PetscCall(DMPlexCreatePLYFromFile(comm, filename, interpolate, dm)); 69735552b385SBrandon } else if (isEGADSlite || isEGADS || isIGES || isIGES2 || isSTEP || isSTEP2 || isBREP) { 69745552b385SBrandon PetscCall(DMPlexCreateGeomFromFile(comm, filename, dm, isEGADSlite)); 69755552b385SBrandon 69767bee2925SMatthew Knepley if (!interpolate) { 69777bee2925SMatthew Knepley DM udm; 69787bee2925SMatthew Knepley 69799566063dSJacob Faibussowitsch PetscCall(DMPlexUninterpolate(*dm, &udm)); 69809566063dSJacob Faibussowitsch PetscCall(DMDestroy(dm)); 69817bee2925SMatthew Knepley *dm = udm; 69827bee2925SMatthew Knepley } 69838ca92349SMatthew G. Knepley } else if (isCV) { 69849566063dSJacob Faibussowitsch PetscCall(DMPlexCreateCellVertexFromFile(comm, filename, interpolate, dm)); 698598921bdaSJacob Faibussowitsch } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename); 69869566063dSJacob Faibussowitsch PetscCall(PetscStrlen(plexname, &len)); 6987f4f49eeaSPierre Jolivet if (len) PetscCall(PetscObjectSetName((PetscObject)*dm, plexname)); 69889566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(DMPLEX_CreateFromFile, 0, 0, 0, 0)); 69893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 6990ca522641SMatthew G. Knepley } 699112b8a6daSStefano Zampini 6992cc4c1da9SBarry Smith /*@ 69939f6c5813SMatthew 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. 69949f6c5813SMatthew G. Knepley 69950528010dSStefano Zampini Input Parameters: 69960528010dSStefano Zampini + tr - The `DMPlexTransform` 69970528010dSStefano Zampini - prefix - An options prefix, or NULL 69989f6c5813SMatthew G. Knepley 69999f6c5813SMatthew G. Knepley Output Parameter: 70009f6c5813SMatthew G. Knepley . dm - The `DM` 70019f6c5813SMatthew G. Knepley 70029f6c5813SMatthew G. Knepley Level: beginner 70039f6c5813SMatthew G. Knepley 700420f4b53cSBarry Smith Notes: 700520f4b53cSBarry 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. 700620f4b53cSBarry Smith 70079f6c5813SMatthew G. Knepley .seealso: `DMPlexCreateFromFile`, `DMPlexCreateFromDAG()`, `DMPlexCreateFromCellListPetsc()`, `DMPlexCreate()` 70089f6c5813SMatthew G. Knepley @*/ 70090528010dSStefano Zampini PetscErrorCode DMPlexCreateEphemeral(DMPlexTransform tr, const char prefix[], DM *dm) 70109f6c5813SMatthew G. Knepley { 70110528010dSStefano Zampini DM bdm, bcdm, cdm; 70120528010dSStefano Zampini Vec coordinates, coordinatesNew; 70130528010dSStefano Zampini PetscSection cs; 7014817b2c36SMatthew G. Knepley PetscInt cdim, Nl; 70159f6c5813SMatthew G. Knepley 70169f6c5813SMatthew G. Knepley PetscFunctionBegin; 70179f6c5813SMatthew G. Knepley PetscCall(DMCreate(PetscObjectComm((PetscObject)tr), dm)); 70189f6c5813SMatthew G. Knepley PetscCall(DMSetType(*dm, DMPLEX)); 70190528010dSStefano Zampini ((DM_Plex *)(*dm)->data)->interpolated = DMPLEX_INTERPOLATED_FULL; 70200528010dSStefano Zampini // Handle coordinates 70210528010dSStefano Zampini PetscCall(DMPlexTransformGetDM(tr, &bdm)); 7022817b2c36SMatthew G. Knepley PetscCall(DMPlexTransformSetDimensions(tr, bdm, *dm)); 7023817b2c36SMatthew G. Knepley PetscCall(DMGetCoordinateDim(*dm, &cdim)); 70240528010dSStefano Zampini PetscCall(DMGetCoordinateDM(bdm, &bcdm)); 70250528010dSStefano Zampini PetscCall(DMGetCoordinateDM(*dm, &cdm)); 70260528010dSStefano Zampini PetscCall(DMCopyDisc(bcdm, cdm)); 70270528010dSStefano Zampini PetscCall(DMGetLocalSection(cdm, &cs)); 70280528010dSStefano Zampini PetscCall(PetscSectionSetNumFields(cs, 1)); 70290528010dSStefano Zampini PetscCall(PetscSectionSetFieldComponents(cs, 0, cdim)); 70300528010dSStefano Zampini PetscCall(DMGetCoordinatesLocal(bdm, &coordinates)); 70310528010dSStefano Zampini PetscCall(VecDuplicate(coordinates, &coordinatesNew)); 70320528010dSStefano Zampini PetscCall(VecCopy(coordinates, coordinatesNew)); 70330528010dSStefano Zampini PetscCall(DMSetCoordinatesLocal(*dm, coordinatesNew)); 70340528010dSStefano Zampini PetscCall(VecDestroy(&coordinatesNew)); 70359f6c5813SMatthew G. Knepley 70369f6c5813SMatthew G. Knepley PetscCall(PetscObjectReference((PetscObject)tr)); 70379f6c5813SMatthew G. Knepley PetscCall(DMPlexTransformDestroy(&((DM_Plex *)(*dm)->data)->tr)); 70389f6c5813SMatthew G. Knepley ((DM_Plex *)(*dm)->data)->tr = tr; 70390528010dSStefano Zampini PetscCall(DMPlexDistributeSetDefault(*dm, PETSC_FALSE)); 70400528010dSStefano Zampini PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*dm, prefix)); 70410528010dSStefano Zampini PetscCall(DMSetFromOptions(*dm)); 70429f6c5813SMatthew G. Knepley 70439f6c5813SMatthew G. Knepley PetscCall(DMGetNumLabels(bdm, &Nl)); 70449f6c5813SMatthew G. Knepley for (PetscInt l = 0; l < Nl; ++l) { 70459f6c5813SMatthew G. Knepley DMLabel label, labelNew; 70469f6c5813SMatthew G. Knepley const char *lname; 70479f6c5813SMatthew G. Knepley PetscBool isDepth, isCellType; 70489f6c5813SMatthew G. Knepley 70499f6c5813SMatthew G. Knepley PetscCall(DMGetLabelName(bdm, l, &lname)); 70509f6c5813SMatthew G. Knepley PetscCall(PetscStrcmp(lname, "depth", &isDepth)); 70519f6c5813SMatthew G. Knepley if (isDepth) continue; 70529f6c5813SMatthew G. Knepley PetscCall(PetscStrcmp(lname, "celltype", &isCellType)); 70539f6c5813SMatthew G. Knepley if (isCellType) continue; 70549f6c5813SMatthew G. Knepley PetscCall(DMCreateLabel(*dm, lname)); 70559f6c5813SMatthew G. Knepley PetscCall(DMGetLabel(bdm, lname, &label)); 70569f6c5813SMatthew G. Knepley PetscCall(DMGetLabel(*dm, lname, &labelNew)); 70579f6c5813SMatthew G. Knepley PetscCall(DMLabelSetType(labelNew, DMLABELEPHEMERAL)); 70589f6c5813SMatthew G. Knepley PetscCall(DMLabelEphemeralSetLabel(labelNew, label)); 70599f6c5813SMatthew G. Knepley PetscCall(DMLabelEphemeralSetTransform(labelNew, tr)); 70609f6c5813SMatthew G. Knepley PetscCall(DMLabelSetUp(labelNew)); 70619f6c5813SMatthew G. Knepley } 70623ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 70639f6c5813SMatthew G. Knepley } 7064