xref: /petsc/src/dm/impls/plex/plexcreate.c (revision a8fb8f2974ee66bb003afa65900fab794a3643ec)
1552f7358SJed Brown #define PETSCDM_DLL
2af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h>    /*I   "petscdmplex.h"   I*/
3552f7358SJed Brown #include <petscdmda.h>
40c312b8eSJed Brown #include <petscsf.h>
5552f7358SJed Brown 
6552f7358SJed Brown #undef __FUNCT__
71df5d5c5SMatthew G. Knepley #define __FUNCT__ "DMPlexCreateDoublet"
81df5d5c5SMatthew G. Knepley /*@
91df5d5c5SMatthew G. Knepley   DMPlexCreateDoublet - Creates a mesh of two cells of the specified type, optionally with later refinement.
101df5d5c5SMatthew G. Knepley 
111df5d5c5SMatthew G. Knepley   Collective on MPI_Comm
121df5d5c5SMatthew G. Knepley 
131df5d5c5SMatthew G. Knepley   Input Parameters:
141df5d5c5SMatthew G. Knepley + comm - The communicator for the DM object
151df5d5c5SMatthew G. Knepley . dim - The spatial dimension
161df5d5c5SMatthew G. Knepley . simplex - Flag for simplicial cells, otherwise they are tensor product cells
171df5d5c5SMatthew G. Knepley . interpolate - Flag to create intermediate mesh pieces (edges, faces)
181df5d5c5SMatthew G. Knepley . refinementUniform - Flag for uniform parallel refinement
191df5d5c5SMatthew G. Knepley - refinementLimit - A nonzero number indicates the largest admissible volume for a refined cell
201df5d5c5SMatthew G. Knepley 
211df5d5c5SMatthew G. Knepley   Output Parameter:
221df5d5c5SMatthew G. Knepley . dm  - The DM object
231df5d5c5SMatthew G. Knepley 
241df5d5c5SMatthew G. Knepley   Level: beginner
251df5d5c5SMatthew G. Knepley 
261df5d5c5SMatthew G. Knepley .keywords: DM, create
271df5d5c5SMatthew G. Knepley .seealso: DMSetType(), DMCreate()
281df5d5c5SMatthew G. Knepley @*/
291df5d5c5SMatthew G. Knepley PetscErrorCode DMPlexCreateDoublet(MPI_Comm comm, PetscInt dim, PetscBool simplex, PetscBool interpolate, PetscBool refinementUniform, PetscReal refinementLimit, DM *newdm)
301df5d5c5SMatthew G. Knepley {
311df5d5c5SMatthew G. Knepley   DM             dm;
321df5d5c5SMatthew G. Knepley   PetscInt       p;
331df5d5c5SMatthew G. Knepley   PetscMPIInt    rank;
341df5d5c5SMatthew G. Knepley   PetscErrorCode ierr;
351df5d5c5SMatthew G. Knepley 
361df5d5c5SMatthew G. Knepley   PetscFunctionBegin;
371df5d5c5SMatthew G. Knepley   ierr = DMCreate(comm, &dm);CHKERRQ(ierr);
381df5d5c5SMatthew G. Knepley   ierr = DMSetType(dm, DMPLEX);CHKERRQ(ierr);
39c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(dm, dim);CHKERRQ(ierr);
401df5d5c5SMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
41ce78fa2fSMatthew G. Knepley   switch (dim) {
42ce78fa2fSMatthew G. Knepley   case 2:
43ce78fa2fSMatthew G. Knepley     if (simplex) {ierr = PetscObjectSetName((PetscObject) dm, "triangular");CHKERRQ(ierr);}
44ce78fa2fSMatthew G. Knepley     else         {ierr = PetscObjectSetName((PetscObject) dm, "quadrilateral");CHKERRQ(ierr);}
45ce78fa2fSMatthew G. Knepley     break;
46ce78fa2fSMatthew G. Knepley   case 3:
47ce78fa2fSMatthew G. Knepley     if (simplex) {ierr = PetscObjectSetName((PetscObject) dm, "tetrahedral");CHKERRQ(ierr);}
48ce78fa2fSMatthew G. Knepley     else         {ierr = PetscObjectSetName((PetscObject) dm, "hexahedral");CHKERRQ(ierr);}
49ce78fa2fSMatthew G. Knepley     break;
50ce78fa2fSMatthew G. Knepley   default:
51ce78fa2fSMatthew G. Knepley     SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %d", dim);
52ce78fa2fSMatthew G. Knepley   }
531df5d5c5SMatthew G. Knepley   if (rank) {
541df5d5c5SMatthew G. Knepley     PetscInt numPoints[2] = {0, 0};
551df5d5c5SMatthew G. Knepley     ierr = DMPlexCreateFromDAG(dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
561df5d5c5SMatthew G. Knepley   } else {
571df5d5c5SMatthew G. Knepley     switch (dim) {
581df5d5c5SMatthew G. Knepley     case 2:
591df5d5c5SMatthew G. Knepley       if (simplex) {
601df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {4, 2};
611df5d5c5SMatthew G. Knepley         PetscInt    coneSize[6]         = {3, 3, 0, 0, 0, 0};
621df5d5c5SMatthew G. Knepley         PetscInt    cones[6]            = {2, 3, 4,  5, 4, 3};
631df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[6] = {0, 0, 0,  0, 0, 0};
641df5d5c5SMatthew G. Knepley         PetscScalar vertexCoords[8]     = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
651df5d5c5SMatthew G. Knepley         PetscInt    markerPoints[8]     = {2, 1, 3, 1, 4, 1, 5, 1};
661df5d5c5SMatthew G. Knepley 
671df5d5c5SMatthew G. Knepley         ierr = DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
68c58f1c22SToby Isaac         for (p = 0; p < 4; ++p) {ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr);}
691df5d5c5SMatthew G. Knepley       } else {
701df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {6, 2};
711df5d5c5SMatthew G. Knepley         PetscInt    coneSize[8]         = {4, 4, 0, 0, 0, 0, 0, 0};
721df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {2, 3, 4, 5,  3, 6, 7, 4};
731df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0,  0, 0, 0, 0};
741df5d5c5SMatthew 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};
751df5d5c5SMatthew G. Knepley 
761df5d5c5SMatthew G. Knepley         ierr = DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
771df5d5c5SMatthew G. Knepley       }
781df5d5c5SMatthew G. Knepley       break;
791df5d5c5SMatthew G. Knepley     case 3:
801df5d5c5SMatthew G. Knepley       if (simplex) {
811df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]        = {5, 2};
821df5d5c5SMatthew G. Knepley         PetscInt    coneSize[7]         = {4, 4, 0, 0, 0, 0, 0};
831df5d5c5SMatthew G. Knepley         PetscInt    cones[8]            = {4, 3, 5, 2,  5, 3, 4, 6};
841df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[8] = {0, 0, 0, 0,  0, 0, 0, 0};
851df5d5c5SMatthew 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};
861df5d5c5SMatthew G. Knepley         PetscInt    markerPoints[10]    = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1};
871df5d5c5SMatthew G. Knepley 
881df5d5c5SMatthew G. Knepley         ierr = DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
89c58f1c22SToby Isaac         for (p = 0; p < 5; ++p) {ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr);}
901df5d5c5SMatthew G. Knepley       } else {
911df5d5c5SMatthew G. Knepley         PetscInt    numPoints[2]         = {12, 2};
921df5d5c5SMatthew G. Knepley         PetscInt    coneSize[14]         = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
931df5d5c5SMatthew G. Knepley         PetscInt    cones[16]            = {2, 3, 4, 5, 6, 7, 8, 9,  5, 4, 10, 11, 7, 12, 13, 8};
941df5d5c5SMatthew G. Knepley         PetscInt    coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0,  0, 0,  0,  0, 0,  0,  0, 0};
951df5d5c5SMatthew G. Knepley         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,
961df5d5c5SMatthew G. Knepley                                             -1.0, -0.5,  0.5,   0.0, -0.5,  0.5,  0.0,  0.5,  0.5,  -1.0,  0.5,  0.5,
971df5d5c5SMatthew G. Knepley                                              1.0,  0.5, -0.5,   1.0, -0.5, -0.5,  1.0, -0.5,  0.5,   1.0,  0.5,  0.5};
981df5d5c5SMatthew G. Knepley 
991df5d5c5SMatthew G. Knepley         ierr = DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
1001df5d5c5SMatthew G. Knepley       }
1011df5d5c5SMatthew G. Knepley       break;
1021df5d5c5SMatthew G. Knepley     default:
1031df5d5c5SMatthew G. Knepley       SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %d", dim);
1041df5d5c5SMatthew G. Knepley     }
1051df5d5c5SMatthew G. Knepley   }
1061df5d5c5SMatthew G. Knepley   *newdm = dm;
1071df5d5c5SMatthew G. Knepley   if (refinementLimit > 0.0) {
1081df5d5c5SMatthew G. Knepley     DM rdm;
1091df5d5c5SMatthew G. Knepley     const char *name;
1101df5d5c5SMatthew G. Knepley 
1111df5d5c5SMatthew G. Knepley     ierr = DMPlexSetRefinementUniform(*newdm, PETSC_FALSE);CHKERRQ(ierr);
1121df5d5c5SMatthew G. Knepley     ierr = DMPlexSetRefinementLimit(*newdm, refinementLimit);CHKERRQ(ierr);
1131df5d5c5SMatthew G. Knepley     ierr = DMRefine(*newdm, comm, &rdm);CHKERRQ(ierr);
1141df5d5c5SMatthew G. Knepley     ierr = PetscObjectGetName((PetscObject) *newdm, &name);CHKERRQ(ierr);
1151df5d5c5SMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject)    rdm,  name);CHKERRQ(ierr);
1161df5d5c5SMatthew G. Knepley     ierr = DMDestroy(newdm);CHKERRQ(ierr);
1171df5d5c5SMatthew G. Knepley     *newdm = rdm;
1181df5d5c5SMatthew G. Knepley   }
1191df5d5c5SMatthew G. Knepley   if (interpolate) {
120e56d480eSMatthew G. Knepley     DM idm = NULL;
1211df5d5c5SMatthew G. Knepley     const char *name;
1221df5d5c5SMatthew G. Knepley 
1231df5d5c5SMatthew G. Knepley     ierr = DMPlexInterpolate(*newdm, &idm);CHKERRQ(ierr);
1241df5d5c5SMatthew G. Knepley     ierr = PetscObjectGetName((PetscObject) *newdm, &name);CHKERRQ(ierr);
1251df5d5c5SMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject)    idm,  name);CHKERRQ(ierr);
1261df5d5c5SMatthew G. Knepley     ierr = DMPlexCopyCoordinates(*newdm, idm);CHKERRQ(ierr);
127c58f1c22SToby Isaac     ierr = DMCopyLabels(*newdm, idm);CHKERRQ(ierr);
1281df5d5c5SMatthew G. Knepley     ierr = DMDestroy(newdm);CHKERRQ(ierr);
1291df5d5c5SMatthew G. Knepley     *newdm = idm;
1301df5d5c5SMatthew G. Knepley   }
1311df5d5c5SMatthew G. Knepley   {
1321df5d5c5SMatthew G. Knepley     DM refinedMesh     = NULL;
1331df5d5c5SMatthew G. Knepley     DM distributedMesh = NULL;
1341df5d5c5SMatthew G. Knepley 
1351df5d5c5SMatthew G. Knepley     /* Distribute mesh over processes */
13680cf41d5SMatthew G. Knepley     ierr = DMPlexDistribute(*newdm, 0, NULL, &distributedMesh);CHKERRQ(ierr);
1371df5d5c5SMatthew G. Knepley     if (distributedMesh) {
1381df5d5c5SMatthew G. Knepley       ierr = DMDestroy(newdm);CHKERRQ(ierr);
1391df5d5c5SMatthew G. Knepley       *newdm = distributedMesh;
1401df5d5c5SMatthew G. Knepley     }
1411df5d5c5SMatthew G. Knepley     if (refinementUniform) {
1421df5d5c5SMatthew G. Knepley       ierr = DMPlexSetRefinementUniform(*newdm, refinementUniform);CHKERRQ(ierr);
1431df5d5c5SMatthew G. Knepley       ierr = DMRefine(*newdm, comm, &refinedMesh);CHKERRQ(ierr);
1441df5d5c5SMatthew G. Knepley       if (refinedMesh) {
1451df5d5c5SMatthew G. Knepley         ierr = DMDestroy(newdm);CHKERRQ(ierr);
1461df5d5c5SMatthew G. Knepley         *newdm = refinedMesh;
1471df5d5c5SMatthew G. Knepley       }
1481df5d5c5SMatthew G. Knepley     }
1491df5d5c5SMatthew G. Knepley   }
1501df5d5c5SMatthew G. Knepley   PetscFunctionReturn(0);
1511df5d5c5SMatthew G. Knepley }
1521df5d5c5SMatthew G. Knepley 
1531df5d5c5SMatthew G. Knepley #undef __FUNCT__
154552f7358SJed Brown #define __FUNCT__ "DMPlexCreateSquareBoundary"
15526492d91SMatthew G. Knepley /*@
15626492d91SMatthew G. Knepley   DMPlexCreateSquareBoundary - Creates a 1D mesh the is the boundary of a square lattice.
157552f7358SJed Brown 
15826492d91SMatthew G. Knepley   Collective on MPI_Comm
15926492d91SMatthew G. Knepley 
16026492d91SMatthew G. Knepley   Input Parameters:
16126492d91SMatthew G. Knepley + comm  - The communicator for the DM object
16226492d91SMatthew G. Knepley . lower - The lower left corner coordinates
16326492d91SMatthew G. Knepley . upper - The upper right corner coordinates
16426492d91SMatthew G. Knepley - edges - The number of cells in each direction
16526492d91SMatthew G. Knepley 
16626492d91SMatthew G. Knepley   Output Parameter:
16726492d91SMatthew G. Knepley . dm  - The DM object
16826492d91SMatthew G. Knepley 
16926492d91SMatthew G. Knepley   Note: Here is the numbering returned for 2 cells in each direction:
17026492d91SMatthew G. Knepley $ 18--5-17--4--16
17126492d91SMatthew G. Knepley $  |     |     |
17226492d91SMatthew G. Knepley $  6    10     3
17326492d91SMatthew G. Knepley $  |     |     |
17426492d91SMatthew G. Knepley $ 19-11-20--9--15
17526492d91SMatthew G. Knepley $  |     |     |
17626492d91SMatthew G. Knepley $  7     8     2
17726492d91SMatthew G. Knepley $  |     |     |
17826492d91SMatthew G. Knepley $ 12--0-13--1--14
17926492d91SMatthew G. Knepley 
18026492d91SMatthew G. Knepley   Level: beginner
18126492d91SMatthew G. Knepley 
18226492d91SMatthew G. Knepley .keywords: DM, create
18326492d91SMatthew G. Knepley .seealso: DMPlexCreateBoxMesh(), DMPlexCreateCubeBoundary(), DMSetType(), DMCreate()
18426492d91SMatthew G. Knepley @*/
185552f7358SJed Brown PetscErrorCode DMPlexCreateSquareBoundary(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
186552f7358SJed Brown {
1871df21d24SMatthew G. Knepley   const PetscInt numVertices    = (edges[0]+1)*(edges[1]+1);
1881df21d24SMatthew G. Knepley   const PetscInt numEdges       = edges[0]*(edges[1]+1) + (edges[0]+1)*edges[1];
189552f7358SJed Brown   PetscInt       markerTop      = 1;
190552f7358SJed Brown   PetscInt       markerBottom   = 1;
191552f7358SJed Brown   PetscInt       markerRight    = 1;
192552f7358SJed Brown   PetscInt       markerLeft     = 1;
193552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
194552f7358SJed Brown   Vec            coordinates;
195552f7358SJed Brown   PetscSection   coordSection;
196552f7358SJed Brown   PetscScalar    *coords;
197552f7358SJed Brown   PetscInt       coordSize;
198552f7358SJed Brown   PetscMPIInt    rank;
199552f7358SJed Brown   PetscInt       v, vx, vy;
200552f7358SJed Brown   PetscErrorCode ierr;
201552f7358SJed Brown 
202552f7358SJed Brown   PetscFunctionBegin;
2030298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject) dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL);CHKERRQ(ierr);
204552f7358SJed Brown   if (markerSeparate) {
2051df21d24SMatthew G. Knepley     markerTop    = 3;
2061df21d24SMatthew G. Knepley     markerBottom = 1;
2071df21d24SMatthew G. Knepley     markerRight  = 2;
2081df21d24SMatthew G. Knepley     markerLeft   = 4;
209552f7358SJed Brown   }
21082f516ccSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank);CHKERRQ(ierr);
211552f7358SJed Brown   if (!rank) {
212552f7358SJed Brown     PetscInt e, ex, ey;
213552f7358SJed Brown 
214552f7358SJed Brown     ierr = DMPlexSetChart(dm, 0, numEdges+numVertices);CHKERRQ(ierr);
215552f7358SJed Brown     for (e = 0; e < numEdges; ++e) {
216552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
217552f7358SJed Brown     }
218552f7358SJed Brown     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
219552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
220552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
221552f7358SJed Brown         PetscInt edge   = vx*edges[1] + ey + edges[0]*(edges[1]+1);
222552f7358SJed Brown         PetscInt vertex = ey*(edges[0]+1) + vx + numEdges;
223da80777bSKarl Rupp         PetscInt cone[2];
224552f7358SJed Brown 
225da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+edges[0]+1;
226552f7358SJed Brown         ierr    = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
227552f7358SJed Brown         if (vx == edges[0]) {
228c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", edge,    markerRight);CHKERRQ(ierr);
229c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", cone[0], markerRight);CHKERRQ(ierr);
230552f7358SJed Brown           if (ey == edges[1]-1) {
231c58f1c22SToby Isaac             ierr = DMSetLabelValue(dm, "marker", cone[1], markerRight);CHKERRQ(ierr);
232552f7358SJed Brown           }
233552f7358SJed Brown         } else if (vx == 0) {
234c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", edge,    markerLeft);CHKERRQ(ierr);
235c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", cone[0], markerLeft);CHKERRQ(ierr);
236552f7358SJed Brown           if (ey == edges[1]-1) {
237c58f1c22SToby Isaac             ierr = DMSetLabelValue(dm, "marker", cone[1], markerLeft);CHKERRQ(ierr);
238552f7358SJed Brown           }
239552f7358SJed Brown         }
240552f7358SJed Brown       }
241552f7358SJed Brown     }
242552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
243552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
244552f7358SJed Brown         PetscInt edge   = vy*edges[0]     + ex;
245552f7358SJed Brown         PetscInt vertex = vy*(edges[0]+1) + ex + numEdges;
246da80777bSKarl Rupp         PetscInt cone[2];
247552f7358SJed Brown 
248da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+1;
249552f7358SJed Brown         ierr    = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
250552f7358SJed Brown         if (vy == edges[1]) {
251c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", edge,    markerTop);CHKERRQ(ierr);
252c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", cone[0], markerTop);CHKERRQ(ierr);
253552f7358SJed Brown           if (ex == edges[0]-1) {
254c58f1c22SToby Isaac             ierr = DMSetLabelValue(dm, "marker", cone[1], markerTop);CHKERRQ(ierr);
255552f7358SJed Brown           }
256552f7358SJed Brown         } else if (vy == 0) {
257c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", edge,    markerBottom);CHKERRQ(ierr);
258c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm, "marker", cone[0], markerBottom);CHKERRQ(ierr);
259552f7358SJed Brown           if (ex == edges[0]-1) {
260c58f1c22SToby Isaac             ierr = DMSetLabelValue(dm, "marker", cone[1], markerBottom);CHKERRQ(ierr);
261552f7358SJed Brown           }
262552f7358SJed Brown         }
263552f7358SJed Brown       }
264552f7358SJed Brown     }
265552f7358SJed Brown   }
266552f7358SJed Brown   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
267552f7358SJed Brown   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
268552f7358SJed Brown   /* Build coordinates */
269c2166f76SMatthew G. Knepley   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
270552f7358SJed Brown   ierr = PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices);CHKERRQ(ierr);
271552f7358SJed Brown   for (v = numEdges; v < numEdges+numVertices; ++v) {
272552f7358SJed Brown     ierr = PetscSectionSetDof(coordSection, v, 2);CHKERRQ(ierr);
273552f7358SJed Brown   }
274552f7358SJed Brown   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
275552f7358SJed Brown   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
27682f516ccSBarry Smith   ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
2774b66c01dSMatthew G. Knepley   ierr = VecSetBlockSize(coordinates, 2);CHKERRQ(ierr);
278da16285aSMichael Lange   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
279552f7358SJed Brown   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
2802eb5907fSJed Brown   ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
281552f7358SJed Brown   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
282552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
283552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
284552f7358SJed Brown       coords[(vy*(edges[0]+1)+vx)*2+0] = lower[0] + ((upper[0] - lower[0])/edges[0])*vx;
285552f7358SJed Brown       coords[(vy*(edges[0]+1)+vx)*2+1] = lower[1] + ((upper[1] - lower[1])/edges[1])*vy;
286552f7358SJed Brown     }
287552f7358SJed Brown   }
288552f7358SJed Brown   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
289552f7358SJed Brown   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
290552f7358SJed Brown   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
291552f7358SJed Brown   PetscFunctionReturn(0);
292552f7358SJed Brown }
293552f7358SJed Brown 
294552f7358SJed Brown #undef __FUNCT__
295552f7358SJed Brown #define __FUNCT__ "DMPlexCreateCubeBoundary"
29626492d91SMatthew G. Knepley /*@
29726492d91SMatthew G. Knepley   DMPlexCreateCubeBoundary - Creates a 2D mesh the is the boundary of a cubic lattice.
298552f7358SJed Brown 
29926492d91SMatthew G. Knepley   Collective on MPI_Comm
30026492d91SMatthew G. Knepley 
30126492d91SMatthew G. Knepley   Input Parameters:
30226492d91SMatthew G. Knepley + comm  - The communicator for the DM object
30326492d91SMatthew G. Knepley . lower - The lower left front corner coordinates
30426492d91SMatthew G. Knepley . upper - The upper right back corner coordinates
30526492d91SMatthew G. Knepley - edges - The number of cells in each direction
30626492d91SMatthew G. Knepley 
30726492d91SMatthew G. Knepley   Output Parameter:
30826492d91SMatthew G. Knepley . dm  - The DM object
30926492d91SMatthew G. Knepley 
31026492d91SMatthew G. Knepley   Level: beginner
31126492d91SMatthew G. Knepley 
31226492d91SMatthew G. Knepley .keywords: DM, create
31326492d91SMatthew G. Knepley .seealso: DMPlexCreateBoxMesh(), DMPlexCreateSquareBoundary(), DMSetType(), DMCreate()
31426492d91SMatthew G. Knepley @*/
315552f7358SJed Brown PetscErrorCode DMPlexCreateCubeBoundary(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
316552f7358SJed Brown {
3179e8abbc3SMichael Lange   PetscInt       vertices[3], numVertices;
3187b59f5a9SMichael Lange   PetscInt       numFaces    = 2*faces[0]*faces[1] + 2*faces[1]*faces[2] + 2*faces[0]*faces[2];
319552f7358SJed Brown   Vec            coordinates;
320552f7358SJed Brown   PetscSection   coordSection;
321552f7358SJed Brown   PetscScalar    *coords;
322552f7358SJed Brown   PetscInt       coordSize;
323552f7358SJed Brown   PetscMPIInt    rank;
324552f7358SJed Brown   PetscInt       v, vx, vy, vz;
3257b59f5a9SMichael Lange   PetscInt       voffset, iface=0, cone[4];
326552f7358SJed Brown   PetscErrorCode ierr;
327552f7358SJed Brown 
328552f7358SJed Brown   PetscFunctionBegin;
32982f516ccSBarry Smith   if ((faces[0] < 1) || (faces[1] < 1) || (faces[2] < 1)) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Must have at least 1 face per side");
33082f516ccSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank);CHKERRQ(ierr);
3319e8abbc3SMichael Lange   vertices[0] = faces[0]+1; vertices[1] = faces[1]+1; vertices[2] = faces[2]+1;
3329e8abbc3SMichael Lange   numVertices = vertices[0]*vertices[1]*vertices[2];
333552f7358SJed Brown   if (!rank) {
334552f7358SJed Brown     PetscInt f;
335552f7358SJed Brown 
336552f7358SJed Brown     ierr = DMPlexSetChart(dm, 0, numFaces+numVertices);CHKERRQ(ierr);
337552f7358SJed Brown     for (f = 0; f < numFaces; ++f) {
338552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, f, 4);CHKERRQ(ierr);
339552f7358SJed Brown     }
340552f7358SJed Brown     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
341552f7358SJed Brown     for (v = 0; v < numFaces+numVertices; ++v) {
342c58f1c22SToby Isaac       ierr = DMSetLabelValue(dm, "marker", v, 1);CHKERRQ(ierr);
343552f7358SJed Brown     }
3447b59f5a9SMichael Lange 
3457b59f5a9SMichael Lange     /* Side 0 (Top) */
3467b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
3477b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
3487b59f5a9SMichael Lange         voffset = numFaces + vertices[0]*vertices[1]*(vertices[2]-1) + vy*vertices[0] + vx;
3497b59f5a9SMichael Lange         cone[0] = voffset; cone[1] = voffset+1; cone[2] = voffset+vertices[0]+1; cone[3] = voffset+vertices[0];
3507b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
3517b59f5a9SMichael Lange         iface++;
352552f7358SJed Brown       }
353552f7358SJed Brown     }
3547b59f5a9SMichael Lange 
3557b59f5a9SMichael Lange     /* Side 1 (Bottom) */
3567b59f5a9SMichael Lange     for (vy = 0; vy < faces[1]; vy++) {
3577b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
3587b59f5a9SMichael Lange         voffset = numFaces + vy*(faces[0]+1) + vx;
3597b59f5a9SMichael Lange         cone[0] = voffset+1; cone[1] = voffset; cone[2] = voffset+vertices[0]; cone[3] = voffset+vertices[0]+1;
3607b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
3617b59f5a9SMichael Lange         iface++;
362552f7358SJed Brown       }
363552f7358SJed Brown     }
3647b59f5a9SMichael Lange 
3657b59f5a9SMichael Lange     /* Side 2 (Front) */
3667b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
3677b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
3687b59f5a9SMichael Lange         voffset = numFaces + vz*vertices[0]*vertices[1] + vx;
3697b59f5a9SMichael Lange         cone[0] = voffset; cone[1] = voffset+1; cone[2] = voffset+vertices[0]*vertices[1]+1; cone[3] = voffset+vertices[0]*vertices[1];
3707b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
3717b59f5a9SMichael Lange         iface++;
372552f7358SJed Brown       }
3737b59f5a9SMichael Lange     }
3747b59f5a9SMichael Lange 
3757b59f5a9SMichael Lange     /* Side 3 (Back) */
3767b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
3777b59f5a9SMichael Lange       for (vx = 0; vx < faces[0]; vx++) {
3787b59f5a9SMichael Lange         voffset = numFaces + vz*vertices[0]*vertices[1] + vertices[0]*(vertices[1]-1) + vx;
3797b59f5a9SMichael Lange         cone[0] = voffset+vertices[0]*vertices[1]; cone[1] = voffset+vertices[0]*vertices[1]+1;
3807b59f5a9SMichael Lange         cone[2] = voffset+1; cone[3] = voffset;
3817b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
3827b59f5a9SMichael Lange         iface++;
3837b59f5a9SMichael Lange       }
3847b59f5a9SMichael Lange     }
3857b59f5a9SMichael Lange 
3867b59f5a9SMichael Lange     /* Side 4 (Left) */
3877b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
3887b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
3897b59f5a9SMichael Lange         voffset = numFaces + vz*vertices[0]*vertices[1] + vy*vertices[0];
3907b59f5a9SMichael Lange         cone[0] = voffset; cone[1] = voffset+vertices[0]*vertices[1];
3917b59f5a9SMichael Lange         cone[2] = voffset+vertices[0]*vertices[1]+vertices[0]; cone[3] = voffset+vertices[0];
3927b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
3937b59f5a9SMichael Lange         iface++;
3947b59f5a9SMichael Lange       }
3957b59f5a9SMichael Lange     }
3967b59f5a9SMichael Lange 
3977b59f5a9SMichael Lange     /* Side 5 (Right) */
3987b59f5a9SMichael Lange     for (vz = 0; vz < faces[2]; vz++) {
3997b59f5a9SMichael Lange       for (vy = 0; vy < faces[1]; vy++) {
4007b59f5a9SMichael Lange         voffset = numFaces + vz*vertices[0]*vertices[1] + vy*vertices[0] + vx;
4017b59f5a9SMichael Lange         cone[0] = voffset+vertices[0]*vertices[1]; cone[1] = voffset;
4027b59f5a9SMichael Lange         cone[2] = voffset+vertices[0]; cone[3] = voffset+vertices[0]*vertices[1]+vertices[0];
4037b59f5a9SMichael Lange         ierr    = DMPlexSetCone(dm, iface, cone);CHKERRQ(ierr);
4047b59f5a9SMichael Lange         iface++;
4057b59f5a9SMichael Lange       }
406552f7358SJed Brown     }
407552f7358SJed Brown   }
408552f7358SJed Brown   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
409552f7358SJed Brown   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
410552f7358SJed Brown   /* Build coordinates */
411c2166f76SMatthew G. Knepley   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
412552f7358SJed Brown   ierr = PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices);CHKERRQ(ierr);
413552f7358SJed Brown   for (v = numFaces; v < numFaces+numVertices; ++v) {
414552f7358SJed Brown     ierr = PetscSectionSetDof(coordSection, v, 3);CHKERRQ(ierr);
415552f7358SJed Brown   }
416552f7358SJed Brown   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
417552f7358SJed Brown   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
41882f516ccSBarry Smith   ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
4194b66c01dSMatthew G. Knepley   ierr = VecSetBlockSize(coordinates, 3);CHKERRQ(ierr);
420552f7358SJed Brown   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
421552f7358SJed Brown   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
4222eb5907fSJed Brown   ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
423552f7358SJed Brown   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
424552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
425552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
426552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
427552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+0] = lower[0] + ((upper[0] - lower[0])/faces[0])*vx;
428552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+1] = lower[1] + ((upper[1] - lower[1])/faces[1])*vy;
429552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+2] = lower[2] + ((upper[2] - lower[2])/faces[2])*vz;
430552f7358SJed Brown       }
431552f7358SJed Brown     }
432552f7358SJed Brown   }
433552f7358SJed Brown   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
434552f7358SJed Brown   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
435552f7358SJed Brown   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
436552f7358SJed Brown   PetscFunctionReturn(0);
437552f7358SJed Brown }
438552f7358SJed Brown 
439552f7358SJed Brown #undef __FUNCT__
4403dfda0b1SToby Isaac #define __FUNCT__ "DMPlexCreateCubeMesh_Internal"
4413dfda0b1SToby Isaac static PetscErrorCode DMPlexCreateCubeMesh_Internal(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY, DMBoundaryType bdZ)
4423dfda0b1SToby Isaac {
443f4eb4c5dSMatthew G. Knepley   PetscInt       markerTop      = 1, faceMarkerTop      = 1;
444f4eb4c5dSMatthew G. Knepley   PetscInt       markerBottom   = 1, faceMarkerBottom   = 1;
445f4eb4c5dSMatthew G. Knepley   PetscInt       markerFront    = 1, faceMarkerFront    = 1;
446f4eb4c5dSMatthew G. Knepley   PetscInt       markerBack     = 1, faceMarkerBack     = 1;
447f4eb4c5dSMatthew G. Knepley   PetscInt       markerRight    = 1, faceMarkerRight    = 1;
448f4eb4c5dSMatthew G. Knepley   PetscInt       markerLeft     = 1, faceMarkerLeft     = 1;
4493dfda0b1SToby Isaac   PetscInt       dim;
4503dfda0b1SToby Isaac   PetscBool      markerSeparate = PETSC_FALSE;
4513dfda0b1SToby Isaac   PetscMPIInt    rank;
4523dfda0b1SToby Isaac   PetscErrorCode ierr;
4533dfda0b1SToby Isaac 
4543dfda0b1SToby Isaac   PetscFunctionBegin;
455f0226e14SMatthew G. Knepley   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
4563dfda0b1SToby Isaac   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank);CHKERRQ(ierr);
4573dfda0b1SToby Isaac   switch (dim) {
4583dfda0b1SToby Isaac   case 2:
459f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 3;
460f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
461f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 2;
462f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 4;
4633dfda0b1SToby Isaac     break;
4643dfda0b1SToby Isaac   case 3:
465f4eb4c5dSMatthew G. Knepley     faceMarkerBottom = 1;
466f4eb4c5dSMatthew G. Knepley     faceMarkerTop    = 2;
467f4eb4c5dSMatthew G. Knepley     faceMarkerFront  = 3;
468f4eb4c5dSMatthew G. Knepley     faceMarkerBack   = 4;
469f4eb4c5dSMatthew G. Knepley     faceMarkerRight  = 5;
470f4eb4c5dSMatthew G. Knepley     faceMarkerLeft   = 6;
4713dfda0b1SToby Isaac     break;
4723dfda0b1SToby Isaac   default:
4733dfda0b1SToby Isaac     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Dimension %d not supported",dim);
4743dfda0b1SToby Isaac     break;
4753dfda0b1SToby Isaac   }
476f4eb4c5dSMatthew G. Knepley   ierr = PetscOptionsGetBool(((PetscObject) dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL);CHKERRQ(ierr);
477f4eb4c5dSMatthew G. Knepley   if (markerSeparate) {
478f4eb4c5dSMatthew G. Knepley     markerBottom = faceMarkerBottom;
479f4eb4c5dSMatthew G. Knepley     markerTop    = faceMarkerTop;
480f4eb4c5dSMatthew G. Knepley     markerFront  = faceMarkerFront;
481f4eb4c5dSMatthew G. Knepley     markerBack   = faceMarkerBack;
482f4eb4c5dSMatthew G. Knepley     markerRight  = faceMarkerRight;
483f4eb4c5dSMatthew G. Knepley     markerLeft   = faceMarkerLeft;
4843dfda0b1SToby Isaac   }
4853dfda0b1SToby Isaac   {
4863dfda0b1SToby Isaac     const PetscInt numXEdges    = !rank ? edges[0]   : 0;
4873dfda0b1SToby Isaac     const PetscInt numYEdges    = !rank ? edges[1]   : 0;
4883dfda0b1SToby Isaac     const PetscInt numZEdges    = !rank ? edges[2]   : 0;
4893dfda0b1SToby Isaac     const PetscInt numXVertices = !rank ? (bdX == DM_BOUNDARY_PERIODIC || bdX == DM_BOUNDARY_TWIST ? edges[0] : edges[0]+1) : 0;
4903dfda0b1SToby Isaac     const PetscInt numYVertices = !rank ? (bdY == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[1] : edges[1]+1) : 0;
4913dfda0b1SToby Isaac     const PetscInt numZVertices = !rank ? (bdZ == DM_BOUNDARY_PERIODIC || bdY == DM_BOUNDARY_TWIST ? edges[2] : edges[2]+1) : 0;
4923dfda0b1SToby Isaac     const PetscInt numCells     = numXEdges*numYEdges*numZEdges;
4933dfda0b1SToby Isaac     const PetscInt numXFaces    = numYEdges*numZEdges;
4943dfda0b1SToby Isaac     const PetscInt numYFaces    = numXEdges*numZEdges;
4953dfda0b1SToby Isaac     const PetscInt numZFaces    = numXEdges*numYEdges;
4963dfda0b1SToby Isaac     const PetscInt numTotXFaces = numXVertices*numXFaces;
4973dfda0b1SToby Isaac     const PetscInt numTotYFaces = numYVertices*numYFaces;
4983dfda0b1SToby Isaac     const PetscInt numTotZFaces = numZVertices*numZFaces;
4993dfda0b1SToby Isaac     const PetscInt numFaces     = numTotXFaces + numTotYFaces + numTotZFaces;
5003dfda0b1SToby Isaac     const PetscInt numTotXEdges = numXEdges*numYVertices*numZVertices;
5013dfda0b1SToby Isaac     const PetscInt numTotYEdges = numYEdges*numXVertices*numZVertices;
5023dfda0b1SToby Isaac     const PetscInt numTotZEdges = numZEdges*numXVertices*numYVertices;
5033dfda0b1SToby Isaac     const PetscInt numVertices  = numXVertices*numYVertices*numZVertices;
5043dfda0b1SToby Isaac     const PetscInt numEdges     = numTotXEdges + numTotYEdges + numTotZEdges;
5053dfda0b1SToby Isaac     const PetscInt firstVertex  = (dim == 2) ? numFaces : numCells;
5063dfda0b1SToby Isaac     const PetscInt firstXFace   = (dim == 2) ? 0 : numCells + numVertices;
5073dfda0b1SToby Isaac     const PetscInt firstYFace   = firstXFace + numTotXFaces;
5083dfda0b1SToby Isaac     const PetscInt firstZFace   = firstYFace + numTotYFaces;
5093dfda0b1SToby Isaac     const PetscInt firstXEdge   = numCells + numFaces + numVertices;
5103dfda0b1SToby Isaac     const PetscInt firstYEdge   = firstXEdge + numTotXEdges;
5113dfda0b1SToby Isaac     const PetscInt firstZEdge   = firstYEdge + numTotYEdges;
5123dfda0b1SToby Isaac     Vec            coordinates;
5133dfda0b1SToby Isaac     PetscSection   coordSection;
5143dfda0b1SToby Isaac     PetscScalar   *coords;
5153dfda0b1SToby Isaac     PetscInt       coordSize;
5163dfda0b1SToby Isaac     PetscInt       v, vx, vy, vz;
5173dfda0b1SToby Isaac     PetscInt       c, f, fx, fy, fz, e, ex, ey, ez;
5183dfda0b1SToby Isaac 
5193dfda0b1SToby Isaac     ierr = DMPlexSetChart(dm, 0, numCells+numFaces+numEdges+numVertices);CHKERRQ(ierr);
5203dfda0b1SToby Isaac     for (c = 0; c < numCells; c++) {
5213dfda0b1SToby Isaac       ierr = DMPlexSetConeSize(dm, c, 6);CHKERRQ(ierr);
5223dfda0b1SToby Isaac     }
5233dfda0b1SToby Isaac     for (f = firstXFace; f < firstXFace+numFaces; ++f) {
5243dfda0b1SToby Isaac       ierr = DMPlexSetConeSize(dm, f, 4);CHKERRQ(ierr);
5253dfda0b1SToby Isaac     }
5263dfda0b1SToby Isaac     for (e = firstXEdge; e < firstXEdge+numEdges; ++e) {
5273dfda0b1SToby Isaac       ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
5283dfda0b1SToby Isaac     }
5293dfda0b1SToby Isaac     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
5303dfda0b1SToby Isaac     /* Build cells */
5313dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
5323dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
5333dfda0b1SToby Isaac         for (fx = 0; fx < numXEdges; ++fx) {
5343dfda0b1SToby Isaac           PetscInt cell    = (fz*numYEdges + fy)*numXEdges + fx;
5353dfda0b1SToby Isaac           PetscInt faceB   = firstZFace + (fy*numXEdges+fx)*numZVertices +   fz;
5363dfda0b1SToby Isaac           PetscInt faceT   = firstZFace + (fy*numXEdges+fx)*numZVertices + ((fz+1)%numZVertices);
5373dfda0b1SToby Isaac           PetscInt faceF   = firstYFace + (fz*numXEdges+fx)*numYVertices +   fy;
5383dfda0b1SToby Isaac           PetscInt faceK   = firstYFace + (fz*numXEdges+fx)*numYVertices + ((fy+1)%numYVertices);
5393dfda0b1SToby Isaac           PetscInt faceL   = firstXFace + (fz*numYEdges+fy)*numXVertices +   fx;
5403dfda0b1SToby Isaac           PetscInt faceR   = firstXFace + (fz*numYEdges+fy)*numXVertices + ((fx+1)%numXVertices);
5413dfda0b1SToby Isaac                             /* B,  T,  F,  K,  R,  L */
5423dfda0b1SToby Isaac           PetscInt ornt[8] = {-4,  0,  0, -1,  0, -4}; /* ??? */
5433dfda0b1SToby Isaac           PetscInt cone[8];
5443dfda0b1SToby Isaac 
5453dfda0b1SToby Isaac           /* no boundary twisting in 3D */
5463dfda0b1SToby Isaac           cone[0] = faceB; cone[1] = faceT; cone[2] = faceF; cone[3] = faceK; cone[4] = faceR; cone[5] = faceL;
5473dfda0b1SToby Isaac           ierr    = DMPlexSetCone(dm, cell, cone);CHKERRQ(ierr);
5483dfda0b1SToby Isaac           ierr    = DMPlexSetConeOrientation(dm, cell, ornt);CHKERRQ(ierr);
5493dfda0b1SToby Isaac         }
5503dfda0b1SToby Isaac       }
5513dfda0b1SToby Isaac     }
5523dfda0b1SToby Isaac     /* Build x faces */
5533dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
5543dfda0b1SToby Isaac       for (fy = 0; fy < numYEdges; ++fy) {
5553dfda0b1SToby Isaac         for (fx = 0; fx < numXVertices; ++fx) {
5563dfda0b1SToby Isaac           PetscInt face    = firstXFace + (fz*numYEdges+fy)*numXVertices + fx;
5573dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (  fy*                 numXVertices+fx)*numZEdges + fz;
5583dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (((fy+1)%numYVertices)*numXVertices+fx)*numZEdges + fz;
5593dfda0b1SToby Isaac           PetscInt edgeB   = firstYEdge + (  fz*                 numXVertices+fx)*numYEdges + fy;
5603dfda0b1SToby Isaac           PetscInt edgeT   = firstYEdge + (((fz+1)%numZVertices)*numXVertices+fx)*numYEdges + fy;
5613dfda0b1SToby Isaac           PetscInt ornt[4] = {0, 0, -2, -2};
5623dfda0b1SToby Isaac           PetscInt cone[4];
5633dfda0b1SToby Isaac 
5643dfda0b1SToby Isaac           if (dim == 3) {
5653dfda0b1SToby Isaac             /* markers */
5663dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
5673dfda0b1SToby Isaac               if (fx == numXVertices-1) {
568c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerRight);CHKERRQ(ierr);
569c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerRight);CHKERRQ(ierr);
5703dfda0b1SToby Isaac               }
5713dfda0b1SToby Isaac               else if (fx == 0) {
572c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerLeft);CHKERRQ(ierr);
573c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerLeft);CHKERRQ(ierr);
5743dfda0b1SToby Isaac               }
5753dfda0b1SToby Isaac             }
5763dfda0b1SToby Isaac           }
5773dfda0b1SToby Isaac           cone[0] = edgeB; cone[1] = edgeR; cone[2] = edgeT; cone[3] = edgeL;
5783dfda0b1SToby Isaac           ierr    = DMPlexSetCone(dm, face, cone);CHKERRQ(ierr);
5793dfda0b1SToby Isaac           ierr    = DMPlexSetConeOrientation(dm, face, ornt);CHKERRQ(ierr);
5803dfda0b1SToby Isaac         }
5813dfda0b1SToby Isaac       }
5823dfda0b1SToby Isaac     }
5833dfda0b1SToby Isaac     /* Build y faces */
5843dfda0b1SToby Isaac     for (fz = 0; fz < numZEdges; ++fz) {
5853dfda0b1SToby Isaac       for (fx = 0; fx < numYEdges; ++fx) {
5863dfda0b1SToby Isaac         for (fy = 0; fy < numYVertices; ++fy) {
5873dfda0b1SToby Isaac           PetscInt face    = firstYFace + (fz*numXEdges+fx)*numYVertices + fy;
5883dfda0b1SToby Isaac           PetscInt edgeL   = firstZEdge + (fy*numXVertices+  fx                 )*numZEdges + fz;
5893dfda0b1SToby Isaac           PetscInt edgeR   = firstZEdge + (fy*numXVertices+((fx+1)%numXVertices))*numZEdges + fz;
5903dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (  fz                 *numYVertices+fy)*numXEdges + fx;
5913dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (((fz+1)%numZVertices)*numYVertices+fy)*numXEdges + fx;
5923dfda0b1SToby Isaac           PetscInt ornt[4] = {0, 0, -2, -2};
5933dfda0b1SToby Isaac           PetscInt cone[4];
5943dfda0b1SToby Isaac 
5953dfda0b1SToby Isaac           if (dim == 3) {
5963dfda0b1SToby Isaac             /* markers */
5973dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
5983dfda0b1SToby Isaac               if (fy == numYVertices-1) {
599c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerBack);CHKERRQ(ierr);
600c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerBack);CHKERRQ(ierr);
6013dfda0b1SToby Isaac               }
6023dfda0b1SToby Isaac               else if (fy == 0) {
603c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerFront);CHKERRQ(ierr);
604c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerFront);CHKERRQ(ierr);
6053dfda0b1SToby Isaac               }
6063dfda0b1SToby Isaac             }
6073dfda0b1SToby Isaac           }
6083dfda0b1SToby Isaac           cone[0] = edgeB; cone[1] = edgeR; cone[2] = edgeT; cone[3] = edgeL;
6093dfda0b1SToby Isaac           ierr    = DMPlexSetCone(dm, face, cone);CHKERRQ(ierr);
6103dfda0b1SToby Isaac           ierr    = DMPlexSetConeOrientation(dm, face, ornt);CHKERRQ(ierr);
6113dfda0b1SToby Isaac         }
6123dfda0b1SToby Isaac       }
6133dfda0b1SToby Isaac     }
6143dfda0b1SToby Isaac     /* Build z faces */
6153dfda0b1SToby Isaac     for (fy = 0; fy < numYEdges; ++fy) {
6163dfda0b1SToby Isaac       for (fx = 0; fx < numXEdges; ++fx) {
6173dfda0b1SToby Isaac         for (fz = 0; fz < numZVertices; fz++) {
6183dfda0b1SToby Isaac           PetscInt face    = firstZFace + (fy*numXEdges+fx)*numZVertices + fz;
6193dfda0b1SToby Isaac           PetscInt edgeL   = firstYEdge + (fz*numXVertices+  fx                 )*numYEdges + fy;
6203dfda0b1SToby Isaac           PetscInt edgeR   = firstYEdge + (fz*numXVertices+((fx+1)%numXVertices))*numYEdges + fy;
6213dfda0b1SToby Isaac           PetscInt edgeB   = firstXEdge + (fz*numYVertices+  fy                 )*numXEdges + fx;
6223dfda0b1SToby Isaac           PetscInt edgeT   = firstXEdge + (fz*numYVertices+((fy+1)%numYVertices))*numXEdges + fx;
6233dfda0b1SToby Isaac           PetscInt ornt[4] = {0, 0, -2, -2};
6243dfda0b1SToby Isaac           PetscInt cone[4];
6253dfda0b1SToby Isaac 
6263dfda0b1SToby Isaac           if (dim == 2) {
6273dfda0b1SToby Isaac             if (bdX == DM_BOUNDARY_TWIST && fx == numXEdges-1) {edgeR += numYEdges-1-2*fy; ornt[1] = -2;}
6283dfda0b1SToby Isaac             if (bdY == DM_BOUNDARY_TWIST && fy == numYEdges-1) {edgeT += numXEdges-1-2*fx; ornt[2] =  0;}
6293dfda0b1SToby Isaac           }
6303dfda0b1SToby Isaac           else {
6313dfda0b1SToby Isaac             /* markers */
6323dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
6333dfda0b1SToby Isaac               if (fz == numZVertices-1) {
634c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerTop);CHKERRQ(ierr);
635c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerTop);CHKERRQ(ierr);
6363dfda0b1SToby Isaac               }
6373dfda0b1SToby Isaac               else if (fz == 0) {
638c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", face, faceMarkerBottom);CHKERRQ(ierr);
639c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", face, markerBottom);CHKERRQ(ierr);
6403dfda0b1SToby Isaac               }
6413dfda0b1SToby Isaac             }
6423dfda0b1SToby Isaac           }
6433dfda0b1SToby Isaac           cone[0] = edgeB; cone[1] = edgeR; cone[2] = edgeT; cone[3] = edgeL;
6443dfda0b1SToby Isaac           ierr    = DMPlexSetCone(dm, face, cone);CHKERRQ(ierr);
6453dfda0b1SToby Isaac           ierr    = DMPlexSetConeOrientation(dm, face, ornt);CHKERRQ(ierr);
6463dfda0b1SToby Isaac         }
6473dfda0b1SToby Isaac       }
6483dfda0b1SToby Isaac     }
6493dfda0b1SToby Isaac     /* Build Z edges*/
6503dfda0b1SToby Isaac     for (vy = 0; vy < numYVertices; vy++) {
6513dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
6523dfda0b1SToby Isaac         for (ez = 0; ez < numZEdges; ez++) {
6533dfda0b1SToby Isaac           const PetscInt edge    = firstZEdge  + (vy*numXVertices+vx)*numZEdges + ez;
6543dfda0b1SToby Isaac           const PetscInt vertexB = firstVertex + (  ez                 *numYVertices+vy)*numXVertices + vx;
6553dfda0b1SToby Isaac           const PetscInt vertexT = firstVertex + (((ez+1)%numZVertices)*numYVertices+vy)*numXVertices + vx;
6563dfda0b1SToby Isaac           PetscInt       cone[2];
6573dfda0b1SToby Isaac 
6583dfda0b1SToby Isaac           if (dim == 3) {
6593dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
6603dfda0b1SToby Isaac               if (vx == numXVertices-1) {
661c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerRight);CHKERRQ(ierr);
6623dfda0b1SToby Isaac               }
6633dfda0b1SToby Isaac               else if (vx == 0) {
664c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerLeft);CHKERRQ(ierr);
6653dfda0b1SToby Isaac               }
6663dfda0b1SToby Isaac             }
6673dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
6683dfda0b1SToby Isaac               if (vy == numYVertices-1) {
669c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerBack);CHKERRQ(ierr);
6703dfda0b1SToby Isaac               }
6713dfda0b1SToby Isaac               else if (vy == 0) {
672c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerFront);CHKERRQ(ierr);
6733dfda0b1SToby Isaac               }
6743dfda0b1SToby Isaac             }
6753dfda0b1SToby Isaac           }
6763dfda0b1SToby Isaac           cone[0] = vertexB; cone[1] = vertexT;
6773dfda0b1SToby Isaac           ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
6783dfda0b1SToby Isaac         }
6793dfda0b1SToby Isaac       }
6803dfda0b1SToby Isaac     }
6813dfda0b1SToby Isaac     /* Build Y edges*/
6823dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
6833dfda0b1SToby Isaac       for (vx = 0; vx < numXVertices; vx++) {
6843dfda0b1SToby Isaac         for (ey = 0; ey < numYEdges; ey++) {
6853dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdY == DM_BOUNDARY_TWIST && ey == numYEdges-1) ? (numXVertices-vx-1) : (vz*numYVertices+((ey+1)%numYVertices))*numXVertices + vx;
6863dfda0b1SToby Isaac           const PetscInt edge    = firstYEdge  + (vz*numXVertices+vx)*numYEdges + ey;
6873dfda0b1SToby Isaac           const PetscInt vertexF = firstVertex + (vz*numYVertices+ey)*numXVertices + vx;
6883dfda0b1SToby Isaac           const PetscInt vertexK = firstVertex + nextv;
6893dfda0b1SToby Isaac           PetscInt       cone[2];
6903dfda0b1SToby Isaac 
6913dfda0b1SToby Isaac           cone[0] = vertexF; cone[1] = vertexK;
6923dfda0b1SToby Isaac           ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
6933dfda0b1SToby Isaac           if (dim == 2) {
6943dfda0b1SToby Isaac             if ((bdX != DM_BOUNDARY_PERIODIC) && (bdX != DM_BOUNDARY_TWIST)) {
6953dfda0b1SToby Isaac               if (vx == numXVertices-1) {
696c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", edge, faceMarkerRight);CHKERRQ(ierr);
697c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge,    markerRight);CHKERRQ(ierr);
698c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", cone[0], markerRight);CHKERRQ(ierr);
6993dfda0b1SToby Isaac                 if (ey == numYEdges-1) {
700c58f1c22SToby Isaac                   ierr = DMSetLabelValue(dm, "marker", cone[1], markerRight);CHKERRQ(ierr);
7013dfda0b1SToby Isaac                 }
7023dfda0b1SToby Isaac               }
7033dfda0b1SToby Isaac               else if (vx == 0) {
704c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", edge, faceMarkerLeft);CHKERRQ(ierr);
705c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge,    markerLeft);CHKERRQ(ierr);
706c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", cone[0], markerLeft);CHKERRQ(ierr);
7073dfda0b1SToby Isaac                 if (ey == numYEdges-1) {
708c58f1c22SToby Isaac                   ierr = DMSetLabelValue(dm, "marker", cone[1], markerLeft);CHKERRQ(ierr);
7093dfda0b1SToby Isaac                 }
7103dfda0b1SToby Isaac               }
7113dfda0b1SToby Isaac             }
7123dfda0b1SToby Isaac           }
7133dfda0b1SToby Isaac           else {
7143dfda0b1SToby Isaac             if (bdX != DM_BOUNDARY_PERIODIC) {
7153dfda0b1SToby Isaac               if (vx == numXVertices-1) {
716c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerRight);CHKERRQ(ierr);
7173dfda0b1SToby Isaac               }
7183dfda0b1SToby Isaac               else if (vx == 0) {
719c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerLeft);CHKERRQ(ierr);
7203dfda0b1SToby Isaac               }
7213dfda0b1SToby Isaac             }
7223dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
7233dfda0b1SToby Isaac               if (vz == numZVertices-1) {
724c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerTop);CHKERRQ(ierr);
7253dfda0b1SToby Isaac               }
7263dfda0b1SToby Isaac               else if (vz == 0) {
727c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerBottom);CHKERRQ(ierr);
7283dfda0b1SToby Isaac               }
7293dfda0b1SToby Isaac             }
7303dfda0b1SToby Isaac           }
7313dfda0b1SToby Isaac         }
7323dfda0b1SToby Isaac       }
7333dfda0b1SToby Isaac     }
7343dfda0b1SToby Isaac     /* Build X edges*/
7353dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; vz++) {
7363dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; vy++) {
7373dfda0b1SToby Isaac         for (ex = 0; ex < numXEdges; ex++) {
7383dfda0b1SToby Isaac           const PetscInt nextv   = (dim == 2 && bdX == DM_BOUNDARY_TWIST && ex == numXEdges-1) ? (numYVertices-vy-1)*numXVertices : (vz*numYVertices+vy)*numXVertices + (ex+1)%numXVertices;
7393dfda0b1SToby Isaac           const PetscInt edge    = firstXEdge  + (vz*numYVertices+vy)*numXEdges + ex;
7403dfda0b1SToby Isaac           const PetscInt vertexL = firstVertex + (vz*numYVertices+vy)*numXVertices + ex;
7413dfda0b1SToby Isaac           const PetscInt vertexR = firstVertex + nextv;
7423dfda0b1SToby Isaac           PetscInt       cone[2];
7433dfda0b1SToby Isaac 
7443dfda0b1SToby Isaac           cone[0] = vertexL; cone[1] = vertexR;
7453dfda0b1SToby Isaac           ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
7463dfda0b1SToby Isaac           if (dim == 2) {
7473dfda0b1SToby Isaac             if ((bdY != DM_BOUNDARY_PERIODIC) && (bdY != DM_BOUNDARY_TWIST)) {
7483dfda0b1SToby Isaac               if (vy == numYVertices-1) {
749c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", edge, faceMarkerTop);CHKERRQ(ierr);
750c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge,    markerTop);CHKERRQ(ierr);
751c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", cone[0], markerTop);CHKERRQ(ierr);
7523dfda0b1SToby Isaac                 if (ex == numXEdges-1) {
753c58f1c22SToby Isaac                   ierr = DMSetLabelValue(dm, "marker", cone[1], markerTop);CHKERRQ(ierr);
7543dfda0b1SToby Isaac                 }
7553dfda0b1SToby Isaac               }
7563dfda0b1SToby Isaac               else if (vy == 0) {
757c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "Face Sets", edge, faceMarkerBottom);CHKERRQ(ierr);
758c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge,    markerBottom);CHKERRQ(ierr);
759c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", cone[0], markerBottom);CHKERRQ(ierr);
7603dfda0b1SToby Isaac                 if (ex == numXEdges-1) {
761c58f1c22SToby Isaac                   ierr = DMSetLabelValue(dm, "marker", cone[1], markerBottom);CHKERRQ(ierr);
7623dfda0b1SToby Isaac                 }
7633dfda0b1SToby Isaac               }
7643dfda0b1SToby Isaac             }
7653dfda0b1SToby Isaac           }
7663dfda0b1SToby Isaac           else {
7673dfda0b1SToby Isaac             if (bdY != DM_BOUNDARY_PERIODIC) {
7683dfda0b1SToby Isaac               if (vy == numYVertices-1) {
769c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerBack);CHKERRQ(ierr);
7703dfda0b1SToby Isaac               }
7713dfda0b1SToby Isaac               else if (vy == 0) {
772c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerFront);CHKERRQ(ierr);
7733dfda0b1SToby Isaac               }
7743dfda0b1SToby Isaac             }
7753dfda0b1SToby Isaac             if (bdZ != DM_BOUNDARY_PERIODIC) {
7763dfda0b1SToby Isaac               if (vz == numZVertices-1) {
777c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerTop);CHKERRQ(ierr);
7783dfda0b1SToby Isaac               }
7793dfda0b1SToby Isaac               else if (vz == 0) {
780c58f1c22SToby Isaac                 ierr = DMSetLabelValue(dm, "marker", edge, markerBottom);CHKERRQ(ierr);
7813dfda0b1SToby Isaac               }
7823dfda0b1SToby Isaac             }
7833dfda0b1SToby Isaac           }
7843dfda0b1SToby Isaac         }
7853dfda0b1SToby Isaac       }
7863dfda0b1SToby Isaac     }
7873dfda0b1SToby Isaac     ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
7883dfda0b1SToby Isaac     ierr = DMPlexStratify(dm);CHKERRQ(ierr);
7893dfda0b1SToby Isaac     /* Build coordinates */
7903dfda0b1SToby Isaac     ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
7913dfda0b1SToby Isaac     ierr = PetscSectionSetNumFields(coordSection, 1);CHKERRQ(ierr);
7923dfda0b1SToby Isaac     ierr = PetscSectionSetFieldComponents(coordSection, 0, dim);CHKERRQ(ierr);
7933dfda0b1SToby Isaac     ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numVertices);CHKERRQ(ierr);
7943dfda0b1SToby Isaac     for (v = firstVertex; v < firstVertex+numVertices; ++v) {
7953dfda0b1SToby Isaac       ierr = PetscSectionSetDof(coordSection, v, dim);CHKERRQ(ierr);
7963dfda0b1SToby Isaac       ierr = PetscSectionSetFieldDof(coordSection, v, 0, dim);CHKERRQ(ierr);
7973dfda0b1SToby Isaac     }
7983dfda0b1SToby Isaac     ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
7993dfda0b1SToby Isaac     ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
8003dfda0b1SToby Isaac     ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
8013dfda0b1SToby Isaac     ierr = VecSetBlockSize(coordinates, dim);CHKERRQ(ierr);
802da16285aSMichael Lange     ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
8033dfda0b1SToby Isaac     ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
8043dfda0b1SToby Isaac     ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
8053dfda0b1SToby Isaac     ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
8063dfda0b1SToby Isaac     for (vz = 0; vz < numZVertices; ++vz) {
8073dfda0b1SToby Isaac       for (vy = 0; vy < numYVertices; ++vy) {
8083dfda0b1SToby Isaac         for (vx = 0; vx < numXVertices; ++vx) {
8093dfda0b1SToby Isaac           coords[((vz*numYVertices+vy)*numXVertices+vx)*dim+0] = lower[0] + ((upper[0] - lower[0])/numXEdges)*vx;
8103dfda0b1SToby Isaac           coords[((vz*numYVertices+vy)*numXVertices+vx)*dim+1] = lower[1] + ((upper[1] - lower[1])/numYEdges)*vy;
8113dfda0b1SToby Isaac           if (dim == 3) {
8123dfda0b1SToby Isaac             coords[((vz*numYVertices+vy)*numXVertices+vx)*dim+2] = lower[2] + ((upper[2] - lower[2])/numZEdges)*vz;
8133dfda0b1SToby Isaac           }
8143dfda0b1SToby Isaac         }
8153dfda0b1SToby Isaac       }
8163dfda0b1SToby Isaac     }
8173dfda0b1SToby Isaac     ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
8183dfda0b1SToby Isaac     ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
8193dfda0b1SToby Isaac     ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
8203dfda0b1SToby Isaac   }
8213dfda0b1SToby Isaac   PetscFunctionReturn(0);
8223dfda0b1SToby Isaac }
8233dfda0b1SToby Isaac 
8243dfda0b1SToby Isaac #undef __FUNCT__
825552f7358SJed Brown #define __FUNCT__ "DMPlexCreateSquareMesh"
82626492d91SMatthew G. Knepley /*@
82726492d91SMatthew G. Knepley   DMPlexCreateSquareMesh - Creates a 2D mesh for a square lattice.
828552f7358SJed Brown 
82926492d91SMatthew G. Knepley   Collective on MPI_Comm
83026492d91SMatthew G. Knepley 
83126492d91SMatthew G. Knepley   Input Parameters:
83226492d91SMatthew G. Knepley + comm  - The communicator for the DM object
83326492d91SMatthew G. Knepley . lower - The lower left corner coordinates
83426492d91SMatthew G. Knepley . upper - The upper right corner coordinates
835fbf5b483SMatthew G. Knepley . edges - The number of cells in each direction
836fbf5b483SMatthew G. Knepley . bdX   - The boundary type for the X direction
837fbf5b483SMatthew G. Knepley - bdY   - The boundary type for the Y direction
83826492d91SMatthew G. Knepley 
83926492d91SMatthew G. Knepley   Output Parameter:
84026492d91SMatthew G. Knepley . dm  - The DM object
84126492d91SMatthew G. Knepley 
84226492d91SMatthew G. Knepley   Note: Here is the numbering returned for 2 cells in each direction:
84326492d91SMatthew G. Knepley $ 22--8-23--9--24
84426492d91SMatthew G. Knepley $  |     |     |
84526492d91SMatthew G. Knepley $ 13  2 14  3  15
84626492d91SMatthew G. Knepley $  |     |     |
84726492d91SMatthew G. Knepley $ 19--6-20--7--21
84826492d91SMatthew G. Knepley $  |     |     |
84926492d91SMatthew G. Knepley $ 10  0 11  1 12
85026492d91SMatthew G. Knepley $  |     |     |
85126492d91SMatthew G. Knepley $ 16--4-17--5--18
85226492d91SMatthew G. Knepley 
85326492d91SMatthew G. Knepley   Level: beginner
85426492d91SMatthew G. Knepley 
85526492d91SMatthew G. Knepley .keywords: DM, create
85626492d91SMatthew G. Knepley .seealso: DMPlexCreateBoxMesh(), DMPlexCreateSquareBoundary(), DMPlexCreateCubeBoundary(), DMSetType(), DMCreate()
85726492d91SMatthew G. Knepley @*/
858bff4a2f0SMatthew G. Knepley PetscErrorCode DMPlexCreateSquareMesh(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[], DMBoundaryType bdX, DMBoundaryType bdY)
859552f7358SJed Brown {
8603dfda0b1SToby Isaac   PetscReal      lower3[3], upper3[3];
8613dfda0b1SToby Isaac   PetscInt       edges3[3];
862552f7358SJed Brown   PetscErrorCode ierr;
863552f7358SJed Brown 
864552f7358SJed Brown   PetscFunctionBegin;
8653dfda0b1SToby Isaac   lower3[0] = lower[0]; lower3[1] = lower[1]; lower3[2] = 0.;
8663dfda0b1SToby Isaac   upper3[0] = upper[0]; upper3[1] = upper[1]; upper3[2] = 0.;
8673dfda0b1SToby Isaac   edges3[0] = edges[0]; edges3[1] = edges[1]; edges3[2] = 0;
8683dfda0b1SToby Isaac   ierr = DMPlexCreateCubeMesh_Internal(dm, lower3, upper3, edges3, bdX, bdY, DM_BOUNDARY_NONE);CHKERRQ(ierr);
869552f7358SJed Brown   PetscFunctionReturn(0);
870552f7358SJed Brown }
871552f7358SJed Brown 
872552f7358SJed Brown #undef __FUNCT__
873552f7358SJed Brown #define __FUNCT__ "DMPlexCreateBoxMesh"
8746b75479cSMatthew G Knepley /*@
87526492d91SMatthew G. Knepley   DMPlexCreateBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using simplices.
8766b75479cSMatthew G Knepley 
8776b75479cSMatthew G Knepley   Collective on MPI_Comm
8786b75479cSMatthew G Knepley 
8796b75479cSMatthew G Knepley   Input Parameters:
8806b75479cSMatthew G Knepley + comm - The communicator for the DM object
8816b75479cSMatthew G Knepley . dim - The spatial dimension
8826b75479cSMatthew G Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
8836b75479cSMatthew G Knepley 
8846b75479cSMatthew G Knepley   Output Parameter:
8856b75479cSMatthew G Knepley . dm  - The DM object
8866b75479cSMatthew G Knepley 
8876b75479cSMatthew G Knepley   Level: beginner
8886b75479cSMatthew G Knepley 
8896b75479cSMatthew G Knepley .keywords: DM, create
89026492d91SMatthew G. Knepley .seealso: DMPlexCreateHexBoxMesh(), DMSetType(), DMCreate()
8916b75479cSMatthew G Knepley @*/
8920adebc6cSBarry Smith PetscErrorCode DMPlexCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool interpolate, DM *dm)
8930adebc6cSBarry Smith {
894552f7358SJed Brown   DM             boundary;
895552f7358SJed Brown   PetscErrorCode ierr;
896552f7358SJed Brown 
897552f7358SJed Brown   PetscFunctionBegin;
898552f7358SJed Brown   PetscValidPointer(dm, 4);
899552f7358SJed Brown   ierr = DMCreate(comm, &boundary);CHKERRQ(ierr);
900552f7358SJed Brown   PetscValidLogicalCollectiveInt(boundary,dim,2);
901552f7358SJed Brown   ierr = DMSetType(boundary, DMPLEX);CHKERRQ(ierr);
902c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(boundary, dim-1);CHKERRQ(ierr);
903552f7358SJed Brown   switch (dim) {
904552f7358SJed Brown   case 2:
905552f7358SJed Brown   {
906552f7358SJed Brown     PetscReal lower[2] = {0.0, 0.0};
907552f7358SJed Brown     PetscReal upper[2] = {1.0, 1.0};
908552f7358SJed Brown     PetscInt  edges[2] = {2, 2};
909552f7358SJed Brown 
910552f7358SJed Brown     ierr = DMPlexCreateSquareBoundary(boundary, lower, upper, edges);CHKERRQ(ierr);
911552f7358SJed Brown     break;
912552f7358SJed Brown   }
913552f7358SJed Brown   case 3:
914552f7358SJed Brown   {
915552f7358SJed Brown     PetscReal lower[3] = {0.0, 0.0, 0.0};
916552f7358SJed Brown     PetscReal upper[3] = {1.0, 1.0, 1.0};
917552f7358SJed Brown     PetscInt  faces[3] = {1, 1, 1};
918552f7358SJed Brown 
919552f7358SJed Brown     ierr = DMPlexCreateCubeBoundary(boundary, lower, upper, faces);CHKERRQ(ierr);
920552f7358SJed Brown     break;
921552f7358SJed Brown   }
922552f7358SJed Brown   default:
923552f7358SJed Brown     SETERRQ1(comm, PETSC_ERR_SUP, "Dimension not supported: %d", dim);
924552f7358SJed Brown   }
9250298fd71SBarry Smith   ierr = DMPlexGenerate(boundary, NULL, interpolate, dm);CHKERRQ(ierr);
926552f7358SJed Brown   ierr = DMDestroy(&boundary);CHKERRQ(ierr);
927552f7358SJed Brown   PetscFunctionReturn(0);
928552f7358SJed Brown }
929552f7358SJed Brown 
930552f7358SJed Brown #undef __FUNCT__
931552f7358SJed Brown #define __FUNCT__ "DMPlexCreateHexBoxMesh"
93226492d91SMatthew G. Knepley /*@
93326492d91SMatthew G. Knepley   DMPlexCreateHexBoxMesh - Creates a mesh on the tensor product of unit intervals (box) using hexahedra.
93426492d91SMatthew G. Knepley 
93526492d91SMatthew G. Knepley   Collective on MPI_Comm
93626492d91SMatthew G. Knepley 
93726492d91SMatthew G. Knepley   Input Parameters:
93826492d91SMatthew G. Knepley + comm  - The communicator for the DM object
93926492d91SMatthew G. Knepley . dim   - The spatial dimension
940fbf5b483SMatthew G. Knepley . periodicX - The boundary type for the X direction
941fbf5b483SMatthew G. Knepley . periodicY - The boundary type for the Y direction
942fbf5b483SMatthew G. Knepley . periodicZ - The boundary type for the Z direction
94326492d91SMatthew G. Knepley - cells - The number of cells in each direction
94426492d91SMatthew G. Knepley 
94526492d91SMatthew G. Knepley   Output Parameter:
94626492d91SMatthew G. Knepley . dm  - The DM object
94726492d91SMatthew G. Knepley 
94826492d91SMatthew G. Knepley   Level: beginner
94926492d91SMatthew G. Knepley 
95026492d91SMatthew G. Knepley .keywords: DM, create
95126492d91SMatthew G. Knepley .seealso: DMPlexCreateBoxMesh(), DMSetType(), DMCreate()
95226492d91SMatthew G. Knepley @*/
953bff4a2f0SMatthew G. Knepley PetscErrorCode DMPlexCreateHexBoxMesh(MPI_Comm comm, PetscInt dim, const PetscInt cells[], DMBoundaryType periodicX, DMBoundaryType periodicY, DMBoundaryType periodicZ, DM *dm)
954a6dfd86eSKarl Rupp {
955552f7358SJed Brown   PetscErrorCode ierr;
956552f7358SJed Brown 
957552f7358SJed Brown   PetscFunctionBegin;
958552f7358SJed Brown   PetscValidPointer(dm, 4);
959552f7358SJed Brown   ierr = DMCreate(comm, dm);CHKERRQ(ierr);
960552f7358SJed Brown   PetscValidLogicalCollectiveInt(*dm,dim,2);
961552f7358SJed Brown   ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
962c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(*dm, dim);CHKERRQ(ierr);
963552f7358SJed Brown   switch (dim) {
964552f7358SJed Brown   case 2:
965552f7358SJed Brown   {
966552f7358SJed Brown     PetscReal lower[2] = {0.0, 0.0};
967552f7358SJed Brown     PetscReal upper[2] = {1.0, 1.0};
968552f7358SJed Brown 
969becd5721SMatthew G. Knepley     ierr = DMPlexCreateSquareMesh(*dm, lower, upper, cells, periodicX, periodicY);CHKERRQ(ierr);
970552f7358SJed Brown     break;
971552f7358SJed Brown   }
972552f7358SJed Brown   case 3:
973552f7358SJed Brown   {
974552f7358SJed Brown     PetscReal lower[3] = {0.0, 0.0, 0.0};
975552f7358SJed Brown     PetscReal upper[3] = {1.0, 1.0, 1.0};
976552f7358SJed Brown 
9773dfda0b1SToby Isaac     ierr = DMPlexCreateCubeMesh_Internal(*dm, lower, upper, cells, periodicX, periodicY, periodicZ);CHKERRQ(ierr);
978552f7358SJed Brown     break;
979552f7358SJed Brown   }
980552f7358SJed Brown   default:
981552f7358SJed Brown     SETERRQ1(comm, PETSC_ERR_SUP, "Dimension not supported: %d", dim);
982552f7358SJed Brown   }
983552f7358SJed Brown   PetscFunctionReturn(0);
984552f7358SJed Brown }
985552f7358SJed Brown 
986552f7358SJed Brown /* External function declarations here */
987552f7358SJed Brown extern PetscErrorCode DMCreateInterpolation_Plex(DM dmCoarse, DM dmFine, Mat *interpolation, Vec *scaling);
9886dbf9973SLawrence Mitchell extern PetscErrorCode DMCreateInjection_Plex(DM dmCoarse, DM dmFine, Mat *mat);
989fd59a867SMatthew G. Knepley extern PetscErrorCode DMCreateDefaultSection_Plex(DM dm);
99066ad2231SToby Isaac extern PetscErrorCode DMCreateDefaultConstraints_Plex(DM dm);
991b412c318SBarry Smith extern PetscErrorCode DMCreateMatrix_Plex(DM dm,  Mat *J);
992552f7358SJed Brown extern PetscErrorCode DMCreateCoordinateDM_Plex(DM dm, DM *cdm);
993552f7358SJed Brown extern PetscErrorCode DMRefine_Plex(DM dm, MPI_Comm comm, DM *dmRefined);
9940a6ba040SMatthew G. Knepley extern PetscErrorCode DMCoarsen_Plex(DM dm, MPI_Comm comm, DM *dmCoarsened);
9950a6ba040SMatthew G. Knepley extern PetscErrorCode DMRefineHierarchy_Plex(DM dm, PetscInt nlevels, DM dmRefined[]);
996b653a561SMatthew G. Knepley extern PetscErrorCode DMCoarsenHierarchy_Plex(DM dm, PetscInt nlevels, DM dmCoarsened[]);
9979a922efaSMatthew G. Knepley extern PetscErrorCode DMClone_Plex(DM dm, DM *newdm);
998552f7358SJed Brown extern PetscErrorCode DMSetUp_Plex(DM dm);
999552f7358SJed Brown extern PetscErrorCode DMDestroy_Plex(DM dm);
1000552f7358SJed Brown extern PetscErrorCode DMView_Plex(DM dm, PetscViewer viewer);
10012c40f234SMatthew G. Knepley extern PetscErrorCode DMLoad_Plex(DM dm, PetscViewer viewer);
1002552f7358SJed Brown extern PetscErrorCode DMCreateSubDM_Plex(DM dm, PetscInt numFields, PetscInt fields[], IS *is, DM *subdm);
1003552f7358SJed Brown extern PetscErrorCode DMLocatePoints_Plex(DM dm, Vec v, IS *cellIS);
1004552f7358SJed Brown 
1005552f7358SJed Brown #undef __FUNCT__
10060a6ba040SMatthew G. Knepley #define __FUNCT__ "DMPlexReplace_Static"
10070a6ba040SMatthew G. Knepley /* Replace dm with the contents of dmNew
10080a6ba040SMatthew G. Knepley    - Share the DM_Plex structure
10090a6ba040SMatthew G. Knepley    - Share the coordinates
1010d7973521SMatthew G. Knepley    - Share the SF
10110a6ba040SMatthew G. Knepley */
10120a6ba040SMatthew G. Knepley static PetscErrorCode DMPlexReplace_Static(DM dm, DM dmNew)
10130a6ba040SMatthew G. Knepley {
1014d7973521SMatthew G. Knepley   PetscSF          sf;
101555fbe3e3SMatthew G. Knepley   DM               coordDM;
10160a6ba040SMatthew G. Knepley   Vec              coords;
101755fbe3e3SMatthew G. Knepley   const PetscReal *maxCell, *L;
10185dc8c3f7SMatthew G. Knepley   const DMBoundaryType *bd;
10190a6ba040SMatthew G. Knepley   PetscErrorCode   ierr;
10200a6ba040SMatthew G. Knepley 
10210a6ba040SMatthew G. Knepley   PetscFunctionBegin;
1022d7973521SMatthew G. Knepley   ierr = DMGetPointSF(dmNew, &sf);CHKERRQ(ierr);
1023d7973521SMatthew G. Knepley   ierr = DMSetPointSF(dm, sf);CHKERRQ(ierr);
102455fbe3e3SMatthew G. Knepley   ierr = DMGetCoordinateDM(dmNew, &coordDM);CHKERRQ(ierr);
10250a6ba040SMatthew G. Knepley   ierr = DMGetCoordinatesLocal(dmNew, &coords);CHKERRQ(ierr);
102655fbe3e3SMatthew G. Knepley   ierr = DMSetCoordinateDM(dm, coordDM);CHKERRQ(ierr);
10270a6ba040SMatthew G. Knepley   ierr = DMSetCoordinatesLocal(dm, coords);CHKERRQ(ierr);
10285dc8c3f7SMatthew G. Knepley   ierr = DMGetPeriodicity(dm, &maxCell, &L, &bd);CHKERRQ(ierr);
10295dc8c3f7SMatthew G. Knepley   if (L) {ierr = DMSetPeriodicity(dmNew, maxCell, L, bd);CHKERRQ(ierr);}
10300a6ba040SMatthew G. Knepley   ierr = DMDestroy_Plex(dm);CHKERRQ(ierr);
10310a6ba040SMatthew G. Knepley   dm->data = dmNew->data;
10320a6ba040SMatthew G. Knepley   ((DM_Plex *) dmNew->data)->refct++;
1033c58f1c22SToby Isaac   dmNew->labels->refct++;
1034c58f1c22SToby Isaac   if (!--(dm->labels->refct)) {
1035c58f1c22SToby Isaac     DMLabelLink next = dm->labels->next;
1036c58f1c22SToby Isaac 
1037c58f1c22SToby Isaac     /* destroy the labels */
1038c58f1c22SToby Isaac     while (next) {
1039c58f1c22SToby Isaac       DMLabelLink tmp = next->next;
1040c58f1c22SToby Isaac 
1041c58f1c22SToby Isaac       ierr = DMLabelDestroy(&next->label);CHKERRQ(ierr);
1042c58f1c22SToby Isaac       ierr = PetscFree(next);CHKERRQ(ierr);
1043c58f1c22SToby Isaac       next = tmp;
1044c58f1c22SToby Isaac     }
1045c58f1c22SToby Isaac     ierr = PetscFree(dm->labels);CHKERRQ(ierr);
1046c58f1c22SToby Isaac   }
1047c58f1c22SToby Isaac   dm->labels = dmNew->labels;
1048c58f1c22SToby Isaac   dm->depthLabel = dmNew->depthLabel;
10490a6ba040SMatthew G. Knepley   PetscFunctionReturn(0);
10500a6ba040SMatthew G. Knepley }
10510a6ba040SMatthew G. Knepley 
10520a6ba040SMatthew G. Knepley #undef __FUNCT__
10530a6ba040SMatthew G. Knepley #define __FUNCT__ "DMPlexSwap_Static"
10540a6ba040SMatthew G. Knepley /* Swap dm with the contents of dmNew
10550a6ba040SMatthew G. Knepley    - Swap the DM_Plex structure
10560a6ba040SMatthew G. Knepley    - Swap the coordinates
10570a6ba040SMatthew G. Knepley */
10580a6ba040SMatthew G. Knepley static PetscErrorCode DMPlexSwap_Static(DM dmA, DM dmB)
10590a6ba040SMatthew G. Knepley {
10600a6ba040SMatthew G. Knepley   DM              coordDMA, coordDMB;
10610a6ba040SMatthew G. Knepley   Vec             coordsA,  coordsB;
10620a6ba040SMatthew G. Knepley   void            *tmp;
1063c58f1c22SToby Isaac   DMLabelLinkList listTmp;
1064c58f1c22SToby Isaac   DMLabel         depthTmp;
10650a6ba040SMatthew G. Knepley   PetscErrorCode  ierr;
10660a6ba040SMatthew G. Knepley 
10670a6ba040SMatthew G. Knepley   PetscFunctionBegin;
10680a6ba040SMatthew G. Knepley   ierr = DMGetCoordinateDM(dmA, &coordDMA);CHKERRQ(ierr);
10690a6ba040SMatthew G. Knepley   ierr = DMGetCoordinateDM(dmB, &coordDMB);CHKERRQ(ierr);
10700a6ba040SMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) coordDMA);CHKERRQ(ierr);
10710a6ba040SMatthew G. Knepley   ierr = DMSetCoordinateDM(dmA, coordDMB);CHKERRQ(ierr);
10720a6ba040SMatthew G. Knepley   ierr = DMSetCoordinateDM(dmB, coordDMA);CHKERRQ(ierr);
10730a6ba040SMatthew G. Knepley   ierr = PetscObjectDereference((PetscObject) coordDMA);CHKERRQ(ierr);
10740a6ba040SMatthew G. Knepley 
10750a6ba040SMatthew G. Knepley   ierr = DMGetCoordinatesLocal(dmA, &coordsA);CHKERRQ(ierr);
10760a6ba040SMatthew G. Knepley   ierr = DMGetCoordinatesLocal(dmB, &coordsB);CHKERRQ(ierr);
10770a6ba040SMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) coordsA);CHKERRQ(ierr);
10780a6ba040SMatthew G. Knepley   ierr = DMSetCoordinatesLocal(dmA, coordsB);CHKERRQ(ierr);
10790a6ba040SMatthew G. Knepley   ierr = DMSetCoordinatesLocal(dmB, coordsA);CHKERRQ(ierr);
10800a6ba040SMatthew G. Knepley   ierr = PetscObjectDereference((PetscObject) coordsA);CHKERRQ(ierr);
10810a6ba040SMatthew G. Knepley   tmp       = dmA->data;
10820a6ba040SMatthew G. Knepley   dmA->data = dmB->data;
10830a6ba040SMatthew G. Knepley   dmB->data = tmp;
1084c58f1c22SToby Isaac   listTmp   = dmA->labels;
1085c58f1c22SToby Isaac   dmA->labels = dmB->labels;
1086c58f1c22SToby Isaac   dmB->labels = listTmp;
1087c58f1c22SToby Isaac   depthTmp  = dmA->depthLabel;
1088c58f1c22SToby Isaac   dmA->depthLabel = dmB->depthLabel;
1089c58f1c22SToby Isaac   dmB->depthLabel = depthTmp;
10900a6ba040SMatthew G. Knepley   PetscFunctionReturn(0);
10910a6ba040SMatthew G. Knepley }
10920a6ba040SMatthew G. Knepley 
10930a6ba040SMatthew G. Knepley #undef __FUNCT__
109468d4fef7SMatthew G. Knepley #define __FUNCT__ "DMSetFromOptions_NonRefinement_Plex"
10958c34d3f5SBarry Smith PetscErrorCode  DMSetFromOptions_NonRefinement_Plex(PetscOptions *PetscOptionsObject,DM dm)
10960a6ba040SMatthew G. Knepley {
10970a6ba040SMatthew G. Knepley   DM_Plex       *mesh = (DM_Plex*) dm->data;
10986c73c22cSMatthew G. Knepley   DMBoundary     b;
10990a6ba040SMatthew G. Knepley   PetscErrorCode ierr;
11000a6ba040SMatthew G. Knepley 
11010a6ba040SMatthew G. Knepley   PetscFunctionBegin;
11026c73c22cSMatthew G. Knepley   /* Handle boundary conditions */
11036c73c22cSMatthew G. Knepley   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject) dm), NULL, "Boundary condition options", "");CHKERRQ(ierr);
11046c73c22cSMatthew G. Knepley   for (b = mesh->boundary; b; b = b->next) {
11056c73c22cSMatthew G. Knepley     char      optname[1024];
11066c73c22cSMatthew G. Knepley     PetscInt  ids[1024], len = 1024, i;
11076c73c22cSMatthew G. Knepley     PetscBool flg;
11086c73c22cSMatthew G. Knepley 
11096c73c22cSMatthew G. Knepley     ierr = PetscSNPrintf(optname, sizeof(optname), "-bc_%s", b->name);CHKERRQ(ierr);
11106c73c22cSMatthew G. Knepley     ierr = PetscMemzero(ids, sizeof(ids));CHKERRQ(ierr);
11116c73c22cSMatthew G. Knepley     ierr = PetscOptionsIntArray(optname, "List of boundary IDs", "", ids, &len, &flg);CHKERRQ(ierr);
11126c73c22cSMatthew G. Knepley     if (flg) {
11136c73c22cSMatthew G. Knepley       DMLabel label;
11146c73c22cSMatthew G. Knepley 
1115c58f1c22SToby Isaac       ierr = DMGetLabel(dm, b->labelname, &label);CHKERRQ(ierr);
11166c73c22cSMatthew G. Knepley       for (i = 0; i < len; ++i) {
11176c73c22cSMatthew G. Knepley         PetscBool has;
11186c73c22cSMatthew G. Knepley 
11196c73c22cSMatthew G. Knepley         ierr = DMLabelHasValue(label, ids[i], &has);CHKERRQ(ierr);
11206c73c22cSMatthew G. Knepley         if (!has) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Boundary id %D is not present in the label %s", ids[i], b->name);
11216c73c22cSMatthew G. Knepley       }
11226c73c22cSMatthew G. Knepley       b->numids = len;
11236c73c22cSMatthew G. Knepley       ierr = PetscFree(b->ids);CHKERRQ(ierr);
11246c73c22cSMatthew G. Knepley       ierr = PetscMalloc1(len, &b->ids);CHKERRQ(ierr);
11256c73c22cSMatthew G. Knepley       ierr = PetscMemcpy(b->ids, ids, len*sizeof(PetscInt));CHKERRQ(ierr);
11266c73c22cSMatthew G. Knepley     }
11270e0f25f2SMatthew G. Knepley     ierr = PetscSNPrintf(optname, sizeof(optname), "-bc_%s_comp", b->name);CHKERRQ(ierr);
11280e0f25f2SMatthew G. Knepley     ierr = PetscMemzero(ids, sizeof(ids));CHKERRQ(ierr);
11290e0f25f2SMatthew G. Knepley     ierr = PetscOptionsIntArray(optname, "List of boundary field components", "", ids, &len, &flg);CHKERRQ(ierr);
11300e0f25f2SMatthew G. Knepley     if (flg) {
11310e0f25f2SMatthew G. Knepley       b->numcomps = len;
11320e0f25f2SMatthew G. Knepley       ierr = PetscFree(b->comps);CHKERRQ(ierr);
11330e0f25f2SMatthew G. Knepley       ierr = PetscMalloc1(len, &b->comps);CHKERRQ(ierr);
11340e0f25f2SMatthew G. Knepley       ierr = PetscMemcpy(b->comps, ids, len*sizeof(PetscInt));CHKERRQ(ierr);
11350e0f25f2SMatthew G. Knepley     }
11366c73c22cSMatthew G. Knepley   }
11376c73c22cSMatthew G. Knepley   ierr = PetscOptionsEnd();CHKERRQ(ierr);
11380a6ba040SMatthew G. Knepley   /* Handle viewing */
11390a6ba040SMatthew G. Knepley   ierr = PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMView", PETSC_FALSE, &mesh->printSetValues, NULL);CHKERRQ(ierr);
11400a6ba040SMatthew G. Knepley   ierr = PetscOptionsInt("-dm_plex_print_fem", "Debug output level all fem computations", "DMView", 0, &mesh->printFEM, NULL);CHKERRQ(ierr);
114194ae4db5SBarry Smith   ierr = PetscOptionsReal("-dm_plex_print_tol", "Tolerance for FEM output", "DMView", mesh->printTol, &mesh->printTol, NULL);CHKERRQ(ierr);
1142953fc75cSMatthew G. Knepley   /* Point Location */
1143953fc75cSMatthew G. Knepley   ierr = PetscOptionsBool("-dm_plex_hash_location", "Use grid hashing for point location", "DMView", PETSC_FALSE, &mesh->useHashLocation, NULL);CHKERRQ(ierr);
1144b29cfa1cSToby Isaac   /* Projection behavior */
1145b29cfa1cSToby Isaac   ierr = PetscOptionsInt("-dm_plex_max_projection_height", "Maxmimum mesh point height used to project locally", "DMPlexSetMaxProjectionHeight", 0, &mesh->maxProjectionHeight, NULL);CHKERRQ(ierr);
114668d4fef7SMatthew G. Knepley   PetscFunctionReturn(0);
114768d4fef7SMatthew G. Knepley }
114868d4fef7SMatthew G. Knepley 
114968d4fef7SMatthew G. Knepley #undef __FUNCT__
115068d4fef7SMatthew G. Knepley #define __FUNCT__ "DMSetFromOptions_Plex"
11518c34d3f5SBarry Smith PetscErrorCode  DMSetFromOptions_Plex(PetscOptions *PetscOptionsObject,DM dm)
115268d4fef7SMatthew G. Knepley {
1153b653a561SMatthew G. Knepley   PetscInt       refine = 0, coarsen = 0, r;
1154b653a561SMatthew G. Knepley   PetscBool      isHierarchy;
115568d4fef7SMatthew G. Knepley   PetscErrorCode ierr;
115668d4fef7SMatthew G. Knepley 
115768d4fef7SMatthew G. Knepley   PetscFunctionBegin;
115868d4fef7SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
11591a1499c8SBarry Smith   ierr = PetscOptionsHead(PetscOptionsObject,"DMPlex Options");CHKERRQ(ierr);
116068d4fef7SMatthew G. Knepley   /* Handle DMPlex refinement */
116168d4fef7SMatthew G. Knepley   ierr = PetscOptionsInt("-dm_refine", "The number of uniform refinements", "DMCreate", refine, &refine, NULL);CHKERRQ(ierr);
116268d4fef7SMatthew G. Knepley   ierr = PetscOptionsInt("-dm_refine_hierarchy", "The number of uniform refinements", "DMCreate", refine, &refine, &isHierarchy);CHKERRQ(ierr);
1163b6a0289aSMatthew G. Knepley   if (refine) {ierr = DMPlexSetRefinementUniform(dm, PETSC_TRUE);CHKERRQ(ierr);}
116468d4fef7SMatthew G. Knepley   if (refine && isHierarchy) {
116568d4fef7SMatthew G. Knepley     DM *dms;
116668d4fef7SMatthew G. Knepley 
116768d4fef7SMatthew G. Knepley     ierr = PetscMalloc1(refine,&dms);CHKERRQ(ierr);
116868d4fef7SMatthew G. Knepley     ierr = DMRefineHierarchy(dm, refine, dms);CHKERRQ(ierr);
116968d4fef7SMatthew G. Knepley     /* Total hack since we do not pass in a pointer */
117068d4fef7SMatthew G. Knepley     ierr = DMPlexSwap_Static(dm, dms[refine-1]);CHKERRQ(ierr);
117168d4fef7SMatthew G. Knepley     if (refine == 1) {
1172*a8fb8f29SToby Isaac       ierr = DMSetCoarseDM(dm, dms[0]);CHKERRQ(ierr);
11730aef6b92SMatthew G. Knepley       ierr = DMPlexSetRegularRefinement(dm, PETSC_TRUE);CHKERRQ(ierr);
117468d4fef7SMatthew G. Knepley     } else {
1175*a8fb8f29SToby Isaac       ierr = DMSetCoarseDM(dm, dms[refine-2]);CHKERRQ(ierr);
11760aef6b92SMatthew G. Knepley       ierr = DMPlexSetRegularRefinement(dm, PETSC_TRUE);CHKERRQ(ierr);
1177*a8fb8f29SToby Isaac       ierr = DMSetCoarseDM(dms[0], dms[refine-1]);CHKERRQ(ierr);
11780aef6b92SMatthew G. Knepley       ierr = DMPlexSetRegularRefinement(dms[0], PETSC_TRUE);CHKERRQ(ierr);
117968d4fef7SMatthew G. Knepley     }
118068d4fef7SMatthew G. Knepley     /* Free DMs */
118168d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
11821a1499c8SBarry Smith       ierr = DMSetFromOptions_NonRefinement_Plex(PetscOptionsObject, dm);CHKERRQ(ierr);
118368d4fef7SMatthew G. Knepley       ierr = DMDestroy(&dms[r]);CHKERRQ(ierr);
118468d4fef7SMatthew G. Knepley     }
118568d4fef7SMatthew G. Knepley     ierr = PetscFree(dms);CHKERRQ(ierr);
118668d4fef7SMatthew G. Knepley   } else {
118768d4fef7SMatthew G. Knepley     for (r = 0; r < refine; ++r) {
118868d4fef7SMatthew G. Knepley       DM refinedMesh;
118968d4fef7SMatthew G. Knepley 
11901a1499c8SBarry Smith       ierr = DMSetFromOptions_NonRefinement_Plex(PetscOptionsObject, dm);CHKERRQ(ierr);
119168d4fef7SMatthew G. Knepley       ierr = DMRefine(dm, PetscObjectComm((PetscObject) dm), &refinedMesh);CHKERRQ(ierr);
119268d4fef7SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
119368d4fef7SMatthew G. Knepley       ierr = DMPlexReplace_Static(dm, refinedMesh);CHKERRQ(ierr);
119468d4fef7SMatthew G. Knepley       ierr = DMDestroy(&refinedMesh);CHKERRQ(ierr);
119568d4fef7SMatthew G. Knepley     }
119668d4fef7SMatthew G. Knepley   }
11973cf6fe12SMatthew G. Knepley   /* Handle DMPlex coarsening */
1198b653a561SMatthew G. Knepley   ierr = PetscOptionsInt("-dm_coarsen", "Coarsen the mesh", "DMCreate", coarsen, &coarsen, NULL);CHKERRQ(ierr);
1199b653a561SMatthew G. Knepley   ierr = PetscOptionsInt("-dm_coarsen_hierarchy", "The number of coarsenings", "DMCreate", coarsen, &coarsen, &isHierarchy);CHKERRQ(ierr);
1200b653a561SMatthew G. Knepley   if (coarsen && isHierarchy) {
1201b653a561SMatthew G. Knepley     DM *dms;
1202b653a561SMatthew G. Knepley 
1203b653a561SMatthew G. Knepley     ierr = PetscMalloc1(coarsen, &dms);CHKERRQ(ierr);
1204b653a561SMatthew G. Knepley     ierr = DMCoarsenHierarchy(dm, coarsen, dms);CHKERRQ(ierr);
1205b653a561SMatthew G. Knepley     /* Free DMs */
1206b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
1207b653a561SMatthew G. Knepley       ierr = DMSetFromOptions_NonRefinement_Plex(PetscOptionsObject, dm);CHKERRQ(ierr);
1208b653a561SMatthew G. Knepley       ierr = DMDestroy(&dms[r]);CHKERRQ(ierr);
1209b653a561SMatthew G. Knepley     }
1210b653a561SMatthew G. Knepley     ierr = PetscFree(dms);CHKERRQ(ierr);
1211b653a561SMatthew G. Knepley   } else {
1212b653a561SMatthew G. Knepley     for (r = 0; r < coarsen; ++r) {
12133cf6fe12SMatthew G. Knepley       DM coarseMesh;
12143cf6fe12SMatthew G. Knepley 
12153cf6fe12SMatthew G. Knepley       ierr = DMSetFromOptions_NonRefinement_Plex(PetscOptionsObject, dm);CHKERRQ(ierr);
12163cf6fe12SMatthew G. Knepley       ierr = DMCoarsen(dm, PetscObjectComm((PetscObject) dm), &coarseMesh);CHKERRQ(ierr);
12173cf6fe12SMatthew G. Knepley       /* Total hack since we do not pass in a pointer */
12183cf6fe12SMatthew G. Knepley       ierr = DMPlexReplace_Static(dm, coarseMesh);CHKERRQ(ierr);
12193cf6fe12SMatthew G. Knepley       ierr = DMDestroy(&coarseMesh);CHKERRQ(ierr);
12203cf6fe12SMatthew G. Knepley     }
1221b653a561SMatthew G. Knepley   }
12223cf6fe12SMatthew G. Knepley   /* Handle */
12231a1499c8SBarry Smith   ierr = DMSetFromOptions_NonRefinement_Plex(PetscOptionsObject, dm);CHKERRQ(ierr);
12240a6ba040SMatthew G. Knepley   ierr = PetscOptionsTail();CHKERRQ(ierr);
12250a6ba040SMatthew G. Knepley   PetscFunctionReturn(0);
12260a6ba040SMatthew G. Knepley }
12270a6ba040SMatthew G. Knepley 
12280a6ba040SMatthew G. Knepley #undef __FUNCT__
1229552f7358SJed Brown #define __FUNCT__ "DMCreateGlobalVector_Plex"
1230552f7358SJed Brown static PetscErrorCode DMCreateGlobalVector_Plex(DM dm,Vec *vec)
1231552f7358SJed Brown {
1232552f7358SJed Brown   PetscErrorCode ierr;
1233552f7358SJed Brown 
1234552f7358SJed Brown   PetscFunctionBegin;
1235552f7358SJed Brown   ierr = DMCreateGlobalVector_Section_Private(dm,vec);CHKERRQ(ierr);
1236552f7358SJed Brown   /* ierr = VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM);CHKERRQ(ierr); */
1237552f7358SJed Brown   ierr = VecSetOperation(*vec, VECOP_VIEW, (void (*)(void)) VecView_Plex);CHKERRQ(ierr);
1238d930f514SMatthew G. Knepley   ierr = VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void)) VecView_Plex_Native);CHKERRQ(ierr);
12392c40f234SMatthew G. Knepley   ierr = VecSetOperation(*vec, VECOP_LOAD, (void (*)(void)) VecLoad_Plex);CHKERRQ(ierr);
1240d930f514SMatthew G. Knepley   ierr = VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void)) VecLoad_Plex_Native);CHKERRQ(ierr);
1241552f7358SJed Brown   PetscFunctionReturn(0);
1242552f7358SJed Brown }
1243552f7358SJed Brown 
1244552f7358SJed Brown #undef __FUNCT__
1245552f7358SJed Brown #define __FUNCT__ "DMCreateLocalVector_Plex"
1246552f7358SJed Brown static PetscErrorCode DMCreateLocalVector_Plex(DM dm,Vec *vec)
1247552f7358SJed Brown {
1248552f7358SJed Brown   PetscErrorCode ierr;
1249552f7358SJed Brown 
1250552f7358SJed Brown   PetscFunctionBegin;
1251552f7358SJed Brown   ierr = DMCreateLocalVector_Section_Private(dm,vec);CHKERRQ(ierr);
1252552f7358SJed Brown   ierr = VecSetOperation(*vec, VECOP_VIEW, (void (*)(void)) VecView_Plex_Local);CHKERRQ(ierr);
12532c40f234SMatthew G. Knepley   ierr = VecSetOperation(*vec, VECOP_LOAD, (void (*)(void)) VecLoad_Plex_Local);CHKERRQ(ierr);
1254552f7358SJed Brown   PetscFunctionReturn(0);
1255552f7358SJed Brown }
1256552f7358SJed Brown 
125738221697SMatthew G. Knepley #undef __FUNCT__
1258793f3fe5SMatthew G. Knepley #define __FUNCT__ "DMGetDimPoints_Plex"
1259793f3fe5SMatthew G. Knepley static PetscErrorCode DMGetDimPoints_Plex(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
1260793f3fe5SMatthew G. Knepley {
1261793f3fe5SMatthew G. Knepley   PetscInt       depth, d;
1262793f3fe5SMatthew G. Knepley   PetscErrorCode ierr;
1263793f3fe5SMatthew G. Knepley 
1264793f3fe5SMatthew G. Knepley   PetscFunctionBegin;
1265793f3fe5SMatthew G. Knepley   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
1266793f3fe5SMatthew G. Knepley   if (depth == 1) {
1267793f3fe5SMatthew G. Knepley     ierr = DMGetDimension(dm, &d);CHKERRQ(ierr);
1268793f3fe5SMatthew G. Knepley     if (dim == 0)      {ierr = DMPlexGetDepthStratum(dm, dim, pStart, pEnd);CHKERRQ(ierr);}
1269793f3fe5SMatthew G. Knepley     else if (dim == d) {ierr = DMPlexGetDepthStratum(dm, 1, pStart, pEnd);CHKERRQ(ierr);}
1270793f3fe5SMatthew G. Knepley     else               {*pStart = 0; *pEnd = 0;}
1271793f3fe5SMatthew G. Knepley   } else {
1272793f3fe5SMatthew G. Knepley     ierr = DMPlexGetDepthStratum(dm, dim, pStart, pEnd);CHKERRQ(ierr);
1273793f3fe5SMatthew G. Knepley   }
1274793f3fe5SMatthew G. Knepley   PetscFunctionReturn(0);
1275793f3fe5SMatthew G. Knepley }
1276793f3fe5SMatthew G. Knepley 
1277793f3fe5SMatthew G. Knepley #undef __FUNCT__
1278552f7358SJed Brown #define __FUNCT__ "DMInitialize_Plex"
1279552f7358SJed Brown PetscErrorCode DMInitialize_Plex(DM dm)
1280552f7358SJed Brown {
1281552f7358SJed Brown   PetscFunctionBegin;
1282552f7358SJed Brown   dm->ops->view                            = DMView_Plex;
12832c40f234SMatthew G. Knepley   dm->ops->load                            = DMLoad_Plex;
1284552f7358SJed Brown   dm->ops->setfromoptions                  = DMSetFromOptions_Plex;
128538221697SMatthew G. Knepley   dm->ops->clone                           = DMClone_Plex;
1286552f7358SJed Brown   dm->ops->setup                           = DMSetUp_Plex;
1287fd59a867SMatthew G. Knepley   dm->ops->createdefaultsection            = DMCreateDefaultSection_Plex;
128866ad2231SToby Isaac   dm->ops->createdefaultconstraints        = DMCreateDefaultConstraints_Plex;
1289552f7358SJed Brown   dm->ops->createglobalvector              = DMCreateGlobalVector_Plex;
1290552f7358SJed Brown   dm->ops->createlocalvector               = DMCreateLocalVector_Plex;
1291184d77edSJed Brown   dm->ops->getlocaltoglobalmapping         = NULL;
12920298fd71SBarry Smith   dm->ops->createfieldis                   = NULL;
1293552f7358SJed Brown   dm->ops->createcoordinatedm              = DMCreateCoordinateDM_Plex;
12940a6ba040SMatthew G. Knepley   dm->ops->getcoloring                     = NULL;
1295552f7358SJed Brown   dm->ops->creatematrix                    = DMCreateMatrix_Plex;
1296bceba477SMatthew G. Knepley   dm->ops->createinterpolation             = DMCreateInterpolation_Plex;
1297bceba477SMatthew G. Knepley   dm->ops->getaggregates                   = NULL;
1298bceba477SMatthew G. Knepley   dm->ops->getinjection                    = DMCreateInjection_Plex;
1299552f7358SJed Brown   dm->ops->refine                          = DMRefine_Plex;
13000a6ba040SMatthew G. Knepley   dm->ops->coarsen                         = DMCoarsen_Plex;
13010a6ba040SMatthew G. Knepley   dm->ops->refinehierarchy                 = DMRefineHierarchy_Plex;
1302b653a561SMatthew G. Knepley   dm->ops->coarsenhierarchy                = DMCoarsenHierarchy_Plex;
13030298fd71SBarry Smith   dm->ops->globaltolocalbegin              = NULL;
13040298fd71SBarry Smith   dm->ops->globaltolocalend                = NULL;
13050298fd71SBarry Smith   dm->ops->localtoglobalbegin              = NULL;
13060298fd71SBarry Smith   dm->ops->localtoglobalend                = NULL;
1307552f7358SJed Brown   dm->ops->destroy                         = DMDestroy_Plex;
1308552f7358SJed Brown   dm->ops->createsubdm                     = DMCreateSubDM_Plex;
1309793f3fe5SMatthew G. Knepley   dm->ops->getdimpoints                    = DMGetDimPoints_Plex;
1310552f7358SJed Brown   dm->ops->locatepoints                    = DMLocatePoints_Plex;
1311552f7358SJed Brown   PetscFunctionReturn(0);
1312552f7358SJed Brown }
1313552f7358SJed Brown 
131463a16f15SMatthew G. Knepley #undef __FUNCT__
131563a16f15SMatthew G. Knepley #define __FUNCT__ "DMClone_Plex"
131663a16f15SMatthew G. Knepley PetscErrorCode DMClone_Plex(DM dm, DM *newdm)
131763a16f15SMatthew G. Knepley {
131863a16f15SMatthew G. Knepley   DM_Plex        *mesh = (DM_Plex *) dm->data;
131963a16f15SMatthew G. Knepley   PetscErrorCode ierr;
132063a16f15SMatthew G. Knepley 
132163a16f15SMatthew G. Knepley   PetscFunctionBegin;
132263a16f15SMatthew G. Knepley   mesh->refct++;
132363a16f15SMatthew G. Knepley   (*newdm)->data = mesh;
132463a16f15SMatthew G. Knepley   ierr = PetscObjectChangeTypeName((PetscObject) *newdm, DMPLEX);CHKERRQ(ierr);
132563a16f15SMatthew G. Knepley   ierr = DMInitialize_Plex(*newdm);CHKERRQ(ierr);
132663a16f15SMatthew G. Knepley   PetscFunctionReturn(0);
132763a16f15SMatthew G. Knepley }
132863a16f15SMatthew G. Knepley 
13298818961aSMatthew G Knepley /*MC
13308818961aSMatthew G Knepley   DMPLEX = "plex" - A DM object that encapsulates an unstructured mesh, or CW Complex, which can be expressed using a Hasse Diagram.
13318818961aSMatthew G Knepley                     In the local representation, Vecs contain all unknowns in the interior and shared boundary. This is
13328818961aSMatthew G Knepley                     specified by a PetscSection object. Ownership in the global representation is determined by
13338818961aSMatthew G Knepley                     ownership of the underlying DMPlex points. This is specified by another PetscSection object.
13348818961aSMatthew G Knepley 
13358818961aSMatthew G Knepley   Level: intermediate
13368818961aSMatthew G Knepley 
13378818961aSMatthew G Knepley .seealso: DMType, DMPlexCreate(), DMCreate(), DMSetType()
13388818961aSMatthew G Knepley M*/
13398818961aSMatthew G Knepley 
1340552f7358SJed Brown #undef __FUNCT__
1341552f7358SJed Brown #define __FUNCT__ "DMCreate_Plex"
13428cc058d9SJed Brown PETSC_EXTERN PetscErrorCode DMCreate_Plex(DM dm)
1343552f7358SJed Brown {
1344552f7358SJed Brown   DM_Plex        *mesh;
1345770b213bSMatthew G Knepley   PetscInt       unit, d;
1346552f7358SJed Brown   PetscErrorCode ierr;
1347552f7358SJed Brown 
1348552f7358SJed Brown   PetscFunctionBegin;
1349552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1350b00a9115SJed Brown   ierr     = PetscNewLog(dm,&mesh);CHKERRQ(ierr);
1351c73cfb54SMatthew G. Knepley   dm->dim  = 0;
1352552f7358SJed Brown   dm->data = mesh;
1353552f7358SJed Brown 
1354552f7358SJed Brown   mesh->refct             = 1;
135582f516ccSBarry Smith   ierr                    = PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->coneSection);CHKERRQ(ierr);
1356552f7358SJed Brown   mesh->maxConeSize       = 0;
13570298fd71SBarry Smith   mesh->cones             = NULL;
13580298fd71SBarry Smith   mesh->coneOrientations  = NULL;
135982f516ccSBarry Smith   ierr                    = PetscSectionCreate(PetscObjectComm((PetscObject)dm), &mesh->supportSection);CHKERRQ(ierr);
1360552f7358SJed Brown   mesh->maxSupportSize    = 0;
13610298fd71SBarry Smith   mesh->supports          = NULL;
1362552f7358SJed Brown   mesh->refinementUniform = PETSC_TRUE;
1363552f7358SJed Brown   mesh->refinementLimit   = -1.0;
1364552f7358SJed Brown 
13650298fd71SBarry Smith   mesh->facesTmp = NULL;
1366552f7358SJed Brown 
1367d9deefdfSMatthew G. Knepley   mesh->tetgenOpts   = NULL;
1368d9deefdfSMatthew G. Knepley   mesh->triangleOpts = NULL;
136977623264SMatthew G. Knepley   ierr = PetscPartitionerCreate(PetscObjectComm((PetscObject)dm), &mesh->partitioner);CHKERRQ(ierr);
137077623264SMatthew G. Knepley   ierr = PetscPartitionerSetTypeFromOptions_Internal(mesh->partitioner);CHKERRQ(ierr);
1371d9deefdfSMatthew G. Knepley 
13720298fd71SBarry Smith   mesh->subpointMap = NULL;
1373552f7358SJed Brown 
13748865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
1375552f7358SJed Brown 
13760aef6b92SMatthew G. Knepley   mesh->regularRefinement   = PETSC_FALSE;
1377df0420ecSMatthew G. Knepley   mesh->depthState          = -1;
13780298fd71SBarry Smith   mesh->globalVertexNumbers = NULL;
13790298fd71SBarry Smith   mesh->globalCellNumbers   = NULL;
1380a68b90caSToby Isaac   mesh->anchorSection       = NULL;
1381a68b90caSToby Isaac   mesh->anchorIS            = NULL;
138241e6d900SToby Isaac   mesh->createanchors       = NULL;
1383fa73a4e1SToby Isaac   mesh->computeanchormatrix = NULL;
1384d961a43aSToby Isaac   mesh->parentSection       = NULL;
1385d961a43aSToby Isaac   mesh->parents             = NULL;
1386d961a43aSToby Isaac   mesh->childIDs            = NULL;
1387d961a43aSToby Isaac   mesh->childSection        = NULL;
1388d961a43aSToby Isaac   mesh->children            = NULL;
1389d6a7ad0dSToby Isaac   mesh->referenceTree       = NULL;
1390dcbd3bf7SToby Isaac   mesh->getchildsymmetry    = NULL;
13918865f1eaSKarl Rupp   for (d = 0; d < 8; ++d) mesh->hybridPointMax[d] = PETSC_DETERMINE;
1392552f7358SJed Brown   mesh->vtkCellHeight       = 0;
139370034214SMatthew G. Knepley   mesh->useCone             = PETSC_FALSE;
139470034214SMatthew G. Knepley   mesh->useClosure          = PETSC_TRUE;
1395e228b242SToby Isaac   mesh->useAnchors          = PETSC_FALSE;
1396552f7358SJed Brown 
1397b29cfa1cSToby Isaac   mesh->maxProjectionHeight = 0;
1398b29cfa1cSToby Isaac 
1399552f7358SJed Brown   mesh->printSetValues = PETSC_FALSE;
1400552f7358SJed Brown   mesh->printFEM       = 0;
14016113b454SMatthew G. Knepley   mesh->printTol       = 1.0e-10;
1402552f7358SJed Brown 
1403552f7358SJed Brown   ierr = DMInitialize_Plex(dm);CHKERRQ(ierr);
1404552f7358SJed Brown   PetscFunctionReturn(0);
1405552f7358SJed Brown }
1406552f7358SJed Brown 
1407552f7358SJed Brown #undef __FUNCT__
1408552f7358SJed Brown #define __FUNCT__ "DMPlexCreate"
1409552f7358SJed Brown /*@
1410552f7358SJed Brown   DMPlexCreate - Creates a DMPlex object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
1411552f7358SJed Brown 
1412552f7358SJed Brown   Collective on MPI_Comm
1413552f7358SJed Brown 
1414552f7358SJed Brown   Input Parameter:
1415552f7358SJed Brown . comm - The communicator for the DMPlex object
1416552f7358SJed Brown 
1417552f7358SJed Brown   Output Parameter:
1418552f7358SJed Brown . mesh  - The DMPlex object
1419552f7358SJed Brown 
1420552f7358SJed Brown   Level: beginner
1421552f7358SJed Brown 
1422552f7358SJed Brown .keywords: DMPlex, create
1423552f7358SJed Brown @*/
1424552f7358SJed Brown PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
1425552f7358SJed Brown {
1426552f7358SJed Brown   PetscErrorCode ierr;
1427552f7358SJed Brown 
1428552f7358SJed Brown   PetscFunctionBegin;
1429552f7358SJed Brown   PetscValidPointer(mesh,2);
1430552f7358SJed Brown   ierr = DMCreate(comm, mesh);CHKERRQ(ierr);
1431552f7358SJed Brown   ierr = DMSetType(*mesh, DMPLEX);CHKERRQ(ierr);
1432552f7358SJed Brown   PetscFunctionReturn(0);
1433552f7358SJed Brown }
1434552f7358SJed Brown 
1435552f7358SJed Brown #undef __FUNCT__
14369298eaa6SMatthew G Knepley #define __FUNCT__ "DMPlexBuildFromCellList_Private"
14379298eaa6SMatthew G Knepley /*
14389298eaa6SMatthew G Knepley   This takes as input the common mesh generator output, a list of the vertices for each cell
14399298eaa6SMatthew G Knepley */
14409298eaa6SMatthew G Knepley PetscErrorCode DMPlexBuildFromCellList_Private(DM dm, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, const int cells[])
14419298eaa6SMatthew G Knepley {
14429298eaa6SMatthew G Knepley   PetscInt      *cone, c, p;
14439298eaa6SMatthew G Knepley   PetscErrorCode ierr;
14449298eaa6SMatthew G Knepley 
14459298eaa6SMatthew G Knepley   PetscFunctionBegin;
14469298eaa6SMatthew G Knepley   ierr = DMPlexSetChart(dm, 0, numCells+numVertices);CHKERRQ(ierr);
14479298eaa6SMatthew G Knepley   for (c = 0; c < numCells; ++c) {
14489298eaa6SMatthew G Knepley     ierr = DMPlexSetConeSize(dm, c, numCorners);CHKERRQ(ierr);
14499298eaa6SMatthew G Knepley   }
14509298eaa6SMatthew G Knepley   ierr = DMSetUp(dm);CHKERRQ(ierr);
14519298eaa6SMatthew G Knepley   ierr = DMGetWorkArray(dm, numCorners, PETSC_INT, &cone);CHKERRQ(ierr);
14529298eaa6SMatthew G Knepley   for (c = 0; c < numCells; ++c) {
14539298eaa6SMatthew G Knepley     for (p = 0; p < numCorners; ++p) {
14549298eaa6SMatthew G Knepley       cone[p] = cells[c*numCorners+p]+numCells;
14559298eaa6SMatthew G Knepley     }
14569298eaa6SMatthew G Knepley     ierr = DMPlexSetCone(dm, c, cone);CHKERRQ(ierr);
14579298eaa6SMatthew G Knepley   }
14589298eaa6SMatthew G Knepley   ierr = DMRestoreWorkArray(dm, numCorners, PETSC_INT, &cone);CHKERRQ(ierr);
14599298eaa6SMatthew G Knepley   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
14609298eaa6SMatthew G Knepley   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
14619298eaa6SMatthew G Knepley   PetscFunctionReturn(0);
14629298eaa6SMatthew G Knepley }
14639298eaa6SMatthew G Knepley 
14649298eaa6SMatthew G Knepley #undef __FUNCT__
14659298eaa6SMatthew G Knepley #define __FUNCT__ "DMPlexBuildCoordinates_Private"
14669298eaa6SMatthew G Knepley /*
14679298eaa6SMatthew G Knepley   This takes as input the coordinates for each vertex
14689298eaa6SMatthew G Knepley */
14699298eaa6SMatthew G Knepley PetscErrorCode DMPlexBuildCoordinates_Private(DM dm, PetscInt spaceDim, PetscInt numCells, PetscInt numVertices, const double vertexCoords[])
14709298eaa6SMatthew G Knepley {
14719298eaa6SMatthew G Knepley   PetscSection   coordSection;
14729298eaa6SMatthew G Knepley   Vec            coordinates;
14739298eaa6SMatthew G Knepley   PetscScalar   *coords;
14749298eaa6SMatthew G Knepley   PetscInt       coordSize, v, d;
14759298eaa6SMatthew G Knepley   PetscErrorCode ierr;
14769298eaa6SMatthew G Knepley 
14779298eaa6SMatthew G Knepley   PetscFunctionBegin;
1478c2166f76SMatthew G. Knepley   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
14799298eaa6SMatthew G Knepley   ierr = PetscSectionSetNumFields(coordSection, 1);CHKERRQ(ierr);
14809298eaa6SMatthew G Knepley   ierr = PetscSectionSetFieldComponents(coordSection, 0, spaceDim);CHKERRQ(ierr);
14819298eaa6SMatthew G Knepley   ierr = PetscSectionSetChart(coordSection, numCells, numCells + numVertices);CHKERRQ(ierr);
14829298eaa6SMatthew G Knepley   for (v = numCells; v < numCells+numVertices; ++v) {
14839298eaa6SMatthew G Knepley     ierr = PetscSectionSetDof(coordSection, v, spaceDim);CHKERRQ(ierr);
14849298eaa6SMatthew G Knepley     ierr = PetscSectionSetFieldDof(coordSection, v, 0, spaceDim);CHKERRQ(ierr);
14859298eaa6SMatthew G Knepley   }
14869298eaa6SMatthew G Knepley   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
14879298eaa6SMatthew G Knepley   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
14889298eaa6SMatthew G Knepley   ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
14894b66c01dSMatthew G. Knepley   ierr = VecSetBlockSize(coordinates, spaceDim);CHKERRQ(ierr);
14909298eaa6SMatthew G Knepley   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
14919298eaa6SMatthew G Knepley   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
14922eb5907fSJed Brown   ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
14939298eaa6SMatthew G Knepley   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
14949298eaa6SMatthew G Knepley   for (v = 0; v < numVertices; ++v) {
14959298eaa6SMatthew G Knepley     for (d = 0; d < spaceDim; ++d) {
14969298eaa6SMatthew G Knepley       coords[v*spaceDim+d] = vertexCoords[v*spaceDim+d];
14979298eaa6SMatthew G Knepley     }
14989298eaa6SMatthew G Knepley   }
14999298eaa6SMatthew G Knepley   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
15009298eaa6SMatthew G Knepley   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
15019298eaa6SMatthew G Knepley   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
15029298eaa6SMatthew G Knepley   PetscFunctionReturn(0);
15039298eaa6SMatthew G Knepley }
15049298eaa6SMatthew G Knepley 
15059298eaa6SMatthew G Knepley #undef __FUNCT__
15069298eaa6SMatthew G Knepley #define __FUNCT__ "DMPlexCreateFromCellList"
15079298eaa6SMatthew G Knepley /*@C
15089298eaa6SMatthew G Knepley   DMPlexCreateFromCellList - This takes as input common mesh generator output, a list of the vertices for each cell, and produces a DM
15099298eaa6SMatthew G Knepley 
15109298eaa6SMatthew G Knepley   Input Parameters:
15119298eaa6SMatthew G Knepley + comm - The communicator
15129298eaa6SMatthew G Knepley . dim - The topological dimension of the mesh
15139298eaa6SMatthew G Knepley . numCells - The number of cells
15149298eaa6SMatthew G Knepley . numVertices - The number of vertices
15159298eaa6SMatthew G Knepley . numCorners - The number of vertices for each cell
15169298eaa6SMatthew G Knepley . interpolate - Flag indicating that intermediate mesh entities (faces, edges) should be created automatically
15179298eaa6SMatthew G Knepley . cells - An array of numCells*numCorners numbers, the vertices for each cell
15189298eaa6SMatthew G Knepley . spaceDim - The spatial dimension used for coordinates
15199298eaa6SMatthew G Knepley - vertexCoords - An array of numVertices*spaceDim numbers, the coordinates of each vertex
15209298eaa6SMatthew G Knepley 
15219298eaa6SMatthew G Knepley   Output Parameter:
15229298eaa6SMatthew G Knepley . dm - The DM
15239298eaa6SMatthew G Knepley 
15249298eaa6SMatthew G Knepley   Note: Two triangles sharing a face
15259298eaa6SMatthew G Knepley $
15269298eaa6SMatthew G Knepley $        2
15279298eaa6SMatthew G Knepley $      / | \
15289298eaa6SMatthew G Knepley $     /  |  \
15299298eaa6SMatthew G Knepley $    /   |   \
15309298eaa6SMatthew G Knepley $   0  0 | 1  3
15319298eaa6SMatthew G Knepley $    \   |   /
15329298eaa6SMatthew G Knepley $     \  |  /
15339298eaa6SMatthew G Knepley $      \ | /
15349298eaa6SMatthew G Knepley $        1
15359298eaa6SMatthew G Knepley would have input
15369298eaa6SMatthew G Knepley $  numCells = 2, numVertices = 4
15379298eaa6SMatthew G Knepley $  cells = [0 1 2  1 3 2]
15389298eaa6SMatthew G Knepley $
15399298eaa6SMatthew G Knepley which would result in the DMPlex
15409298eaa6SMatthew G Knepley $
15419298eaa6SMatthew G Knepley $        4
15429298eaa6SMatthew G Knepley $      / | \
15439298eaa6SMatthew G Knepley $     /  |  \
15449298eaa6SMatthew G Knepley $    /   |   \
15459298eaa6SMatthew G Knepley $   2  0 | 1  5
15469298eaa6SMatthew G Knepley $    \   |   /
15479298eaa6SMatthew G Knepley $     \  |  /
15489298eaa6SMatthew G Knepley $      \ | /
15499298eaa6SMatthew G Knepley $        3
15509298eaa6SMatthew G Knepley 
15519298eaa6SMatthew G Knepley   Level: beginner
15529298eaa6SMatthew G Knepley 
1553939f6067SMatthew G. Knepley .seealso: DMPlexCreateFromDAG(), DMPlexCreate()
15549298eaa6SMatthew G Knepley @*/
15559298eaa6SMatthew G Knepley PetscErrorCode DMPlexCreateFromCellList(MPI_Comm comm, PetscInt dim, PetscInt numCells, PetscInt numVertices, PetscInt numCorners, PetscBool interpolate, const int cells[], PetscInt spaceDim, const double vertexCoords[], DM *dm)
15569298eaa6SMatthew G Knepley {
15579298eaa6SMatthew G Knepley   PetscErrorCode ierr;
15589298eaa6SMatthew G Knepley 
15599298eaa6SMatthew G Knepley   PetscFunctionBegin;
15609298eaa6SMatthew G Knepley   ierr = DMCreate(comm, dm);CHKERRQ(ierr);
15619298eaa6SMatthew G Knepley   ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
1562c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(*dm, dim);CHKERRQ(ierr);
15639298eaa6SMatthew G Knepley   ierr = DMPlexBuildFromCellList_Private(*dm, numCells, numVertices, numCorners, cells);CHKERRQ(ierr);
15649298eaa6SMatthew G Knepley   if (interpolate) {
1565e56d480eSMatthew G. Knepley     DM idm = NULL;
15669298eaa6SMatthew G Knepley 
15679298eaa6SMatthew G Knepley     ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
15689298eaa6SMatthew G Knepley     ierr = DMDestroy(dm);CHKERRQ(ierr);
15699298eaa6SMatthew G Knepley     *dm  = idm;
15709298eaa6SMatthew G Knepley   }
15719298eaa6SMatthew G Knepley   ierr = DMPlexBuildCoordinates_Private(*dm, spaceDim, numCells, numVertices, vertexCoords);CHKERRQ(ierr);
15729298eaa6SMatthew G Knepley   PetscFunctionReturn(0);
15739298eaa6SMatthew G Knepley }
15749298eaa6SMatthew G Knepley 
15759298eaa6SMatthew G Knepley #undef __FUNCT__
15769298eaa6SMatthew G Knepley #define __FUNCT__ "DMPlexCreateFromDAG"
1577939f6067SMatthew G. Knepley /*@
1578939f6067SMatthew G. Knepley   DMPlexCreateFromDAG - This takes as input the adjacency-list representation of the Directed Acyclic Graph (Hasse Diagram) encoding a mesh, and produces a DM
1579939f6067SMatthew G. Knepley 
1580939f6067SMatthew G. Knepley   Input Parameters:
1581c73cfb54SMatthew G. Knepley + dm - The empty DM object, usually from DMCreate() and DMSetDimension()
1582939f6067SMatthew G. Knepley . depth - The depth of the DAG
1583939f6067SMatthew G. Knepley . numPoints - The number of points at each depth
1584939f6067SMatthew G. Knepley . coneSize - The cone size of each point
1585939f6067SMatthew G. Knepley . cones - The concatenation of the cone points for each point, the cone list must be oriented correctly for each point
1586939f6067SMatthew G. Knepley . coneOrientations - The orientation of each cone point
1587939f6067SMatthew G. Knepley - vertexCoords - An array of numVertices*dim numbers, the coordinates of each vertex
1588939f6067SMatthew G. Knepley 
1589939f6067SMatthew G. Knepley   Output Parameter:
1590939f6067SMatthew G. Knepley . dm - The DM
1591939f6067SMatthew G. Knepley 
1592939f6067SMatthew G. Knepley   Note: Two triangles sharing a face would have input
1593939f6067SMatthew G. Knepley $  depth = 1, numPoints = [4 2], coneSize = [3 3 0 0 0 0]
1594939f6067SMatthew G. Knepley $  cones = [2 3 4  3 5 4], coneOrientations = [0 0 0  0 0 0]
1595939f6067SMatthew G. Knepley $ vertexCoords = [-1.0 0.0  0.0 -1.0  0.0 1.0  1.0 0.0]
1596939f6067SMatthew G. Knepley $
1597939f6067SMatthew G. Knepley which would result in the DMPlex
1598939f6067SMatthew G. Knepley $
1599939f6067SMatthew G. Knepley $        4
1600939f6067SMatthew G. Knepley $      / | \
1601939f6067SMatthew G. Knepley $     /  |  \
1602939f6067SMatthew G. Knepley $    /   |   \
1603939f6067SMatthew G. Knepley $   2  0 | 1  5
1604939f6067SMatthew G. Knepley $    \   |   /
1605939f6067SMatthew G. Knepley $     \  |  /
1606939f6067SMatthew G. Knepley $      \ | /
1607939f6067SMatthew G. Knepley $        3
1608939f6067SMatthew G. Knepley $
1609939f6067SMatthew G. Knepley $ Notice that all points are numbered consecutively, unlikely DMPlexCreateFromCellList()
1610939f6067SMatthew G. Knepley 
1611939f6067SMatthew G. Knepley   Level: advanced
1612939f6067SMatthew G. Knepley 
1613939f6067SMatthew G. Knepley .seealso: DMPlexCreateFromCellList(), DMPlexCreate()
1614939f6067SMatthew G. Knepley @*/
16159298eaa6SMatthew G Knepley PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
16169298eaa6SMatthew G Knepley {
16179298eaa6SMatthew G Knepley   Vec            coordinates;
16189298eaa6SMatthew G Knepley   PetscSection   coordSection;
16199298eaa6SMatthew G Knepley   PetscScalar    *coords;
1620811e8653SToby Isaac   PetscInt       coordSize, firstVertex = -1, pStart = 0, pEnd = 0, p, v, dim, dimEmbed, d, off;
16219298eaa6SMatthew G Knepley   PetscErrorCode ierr;
16229298eaa6SMatthew G Knepley 
16239298eaa6SMatthew G Knepley   PetscFunctionBegin;
1624c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1625811e8653SToby Isaac   ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr);
1626811e8653SToby Isaac   if (dimEmbed < dim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Embedding dimension %d cannot be less than intrinsic dimension %d",dimEmbed,dim);
16279298eaa6SMatthew G Knepley   for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
16289298eaa6SMatthew G Knepley   ierr = DMPlexSetChart(dm, pStart, pEnd);CHKERRQ(ierr);
16299298eaa6SMatthew G Knepley   for (p = pStart; p < pEnd; ++p) {
16309298eaa6SMatthew G Knepley     ierr = DMPlexSetConeSize(dm, p, coneSize[p-pStart]);CHKERRQ(ierr);
163197e052ccSToby Isaac     if (firstVertex < 0 && !coneSize[p - pStart]) {
163297e052ccSToby Isaac       firstVertex = p - pStart;
16339298eaa6SMatthew G Knepley     }
163497e052ccSToby Isaac   }
163597e052ccSToby Isaac   if (firstVertex < 0 && numPoints[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Expected %d vertices but could not find any", numPoints[0]);
16369298eaa6SMatthew G Knepley   ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
16379298eaa6SMatthew G Knepley   for (p = pStart, off = 0; p < pEnd; off += coneSize[p-pStart], ++p) {
16389298eaa6SMatthew G Knepley     ierr = DMPlexSetCone(dm, p, &cones[off]);CHKERRQ(ierr);
16399298eaa6SMatthew G Knepley     ierr = DMPlexSetConeOrientation(dm, p, &coneOrientations[off]);CHKERRQ(ierr);
16409298eaa6SMatthew G Knepley   }
16419298eaa6SMatthew G Knepley   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
16429298eaa6SMatthew G Knepley   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
16439298eaa6SMatthew G Knepley   /* Build coordinates */
1644c2166f76SMatthew G. Knepley   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
16459298eaa6SMatthew G Knepley   ierr = PetscSectionSetNumFields(coordSection, 1);CHKERRQ(ierr);
1646811e8653SToby Isaac   ierr = PetscSectionSetFieldComponents(coordSection, 0, dimEmbed);CHKERRQ(ierr);
16479298eaa6SMatthew G Knepley   ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numPoints[0]);CHKERRQ(ierr);
16489298eaa6SMatthew G Knepley   for (v = firstVertex; v < firstVertex+numPoints[0]; ++v) {
1649811e8653SToby Isaac     ierr = PetscSectionSetDof(coordSection, v, dimEmbed);CHKERRQ(ierr);
1650811e8653SToby Isaac     ierr = PetscSectionSetFieldDof(coordSection, v, 0, dimEmbed);CHKERRQ(ierr);
16519298eaa6SMatthew G Knepley   }
16529298eaa6SMatthew G Knepley   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
16539298eaa6SMatthew G Knepley   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
16549298eaa6SMatthew G Knepley   ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
1655811e8653SToby Isaac   ierr = VecSetBlockSize(coordinates, dimEmbed);CHKERRQ(ierr);
16566f8cbbeeSMatthew G Knepley   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
16579298eaa6SMatthew G Knepley   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
16582eb5907fSJed Brown   ierr = VecSetType(coordinates,VECSTANDARD);CHKERRQ(ierr);
16599298eaa6SMatthew G Knepley   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
16609298eaa6SMatthew G Knepley   for (v = 0; v < numPoints[0]; ++v) {
16619298eaa6SMatthew G Knepley     PetscInt off;
16629298eaa6SMatthew G Knepley 
16639298eaa6SMatthew G Knepley     ierr = PetscSectionGetOffset(coordSection, v+firstVertex, &off);CHKERRQ(ierr);
1664811e8653SToby Isaac     for (d = 0; d < dimEmbed; ++d) {
1665811e8653SToby Isaac       coords[off+d] = vertexCoords[v*dimEmbed+d];
16669298eaa6SMatthew G Knepley     }
16679298eaa6SMatthew G Knepley   }
16689298eaa6SMatthew G Knepley   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
16699298eaa6SMatthew G Knepley   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
16709298eaa6SMatthew G Knepley   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
16719298eaa6SMatthew G Knepley   PetscFunctionReturn(0);
16729298eaa6SMatthew G Knepley }
16738415267dSToby Isaac 
16748415267dSToby Isaac #undef __FUNCT__
1675ca522641SMatthew G. Knepley #define __FUNCT__ "DMPlexCreateFromFile"
1676ca522641SMatthew G. Knepley /*@C
1677ca522641SMatthew G. Knepley   DMPlexCreateFromFile - This takes a filename and produces a DM
1678ca522641SMatthew G. Knepley 
1679ca522641SMatthew G. Knepley   Input Parameters:
1680ca522641SMatthew G. Knepley + comm - The communicator
1681ca522641SMatthew G. Knepley . filename - A file name
1682ca522641SMatthew G. Knepley - interpolate - Flag to create intermediate mesh pieces (edges, faces)
1683ca522641SMatthew G. Knepley 
1684ca522641SMatthew G. Knepley   Output Parameter:
1685ca522641SMatthew G. Knepley . dm - The DM
1686ca522641SMatthew G. Knepley 
1687ca522641SMatthew G. Knepley   Level: beginner
1688ca522641SMatthew G. Knepley 
1689ca522641SMatthew G. Knepley .seealso: DMPlexCreateFromDAG(), DMPlexCreateFromCellList(), DMPlexCreate()
1690ca522641SMatthew G. Knepley @*/
1691ca522641SMatthew G. Knepley PetscErrorCode DMPlexCreateFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
1692ca522641SMatthew G. Knepley {
1693ca522641SMatthew G. Knepley   const char    *extGmsh   = ".msh";
1694ca522641SMatthew G. Knepley   const char    *extCGNS   = ".cgns";
1695ca522641SMatthew G. Knepley   const char    *extExodus = ".exo";
16962f0bd6dcSMichael Lange   const char    *extFluent = ".cas";
1697cc2f8f65SMatthew G. Knepley   const char    *extHDF5   = ".h5";
1698ca522641SMatthew G. Knepley   size_t         len;
1699cc2f8f65SMatthew G. Knepley   PetscBool      isGmsh, isCGNS, isExodus, isFluent, isHDF5;
1700ca522641SMatthew G. Knepley   PetscMPIInt    rank;
1701ca522641SMatthew G. Knepley   PetscErrorCode ierr;
1702ca522641SMatthew G. Knepley 
1703ca522641SMatthew G. Knepley   PetscFunctionBegin;
1704ca522641SMatthew G. Knepley   PetscValidPointer(filename, 2);
1705ca522641SMatthew G. Knepley   PetscValidPointer(dm, 4);
1706ca522641SMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
1707ca522641SMatthew G. Knepley   ierr = PetscStrlen(filename, &len);CHKERRQ(ierr);
1708ca522641SMatthew G. Knepley   if (!len) SETERRQ(comm, PETSC_ERR_ARG_WRONG, "Filename must be a valid path");
1709ca522641SMatthew G. Knepley   ierr = PetscStrncmp(&filename[PetscMax(0,len-4)], extGmsh,   4, &isGmsh);CHKERRQ(ierr);
1710ca522641SMatthew G. Knepley   ierr = PetscStrncmp(&filename[PetscMax(0,len-5)], extCGNS,   5, &isCGNS);CHKERRQ(ierr);
1711ca522641SMatthew G. Knepley   ierr = PetscStrncmp(&filename[PetscMax(0,len-4)], extExodus, 4, &isExodus);CHKERRQ(ierr);
17122f0bd6dcSMichael Lange   ierr = PetscStrncmp(&filename[PetscMax(0,len-4)], extFluent, 4, &isFluent);CHKERRQ(ierr);
1713cc2f8f65SMatthew G. Knepley   ierr = PetscStrncmp(&filename[PetscMax(0,len-3)], extHDF5,   3, &isHDF5);CHKERRQ(ierr);
1714ca522641SMatthew G. Knepley   if (isGmsh) {
17157d282ae0SMichael Lange     ierr = DMPlexCreateGmshFromFile(comm, filename, interpolate, dm);CHKERRQ(ierr);
1716ca522641SMatthew G. Knepley   } else if (isCGNS) {
1717ca522641SMatthew G. Knepley     ierr = DMPlexCreateCGNSFromFile(comm, filename, interpolate, dm);CHKERRQ(ierr);
1718ca522641SMatthew G. Knepley   } else if (isExodus) {
1719ca522641SMatthew G. Knepley     ierr = DMPlexCreateExodusFromFile(comm, filename, interpolate, dm);CHKERRQ(ierr);
17202f0bd6dcSMichael Lange   } else if (isFluent) {
17212f0bd6dcSMichael Lange     ierr = DMPlexCreateFluentFromFile(comm, filename, interpolate, dm);CHKERRQ(ierr);
1722cc2f8f65SMatthew G. Knepley   } else if (isHDF5) {
1723cc2f8f65SMatthew G. Knepley     PetscViewer viewer;
1724cc2f8f65SMatthew G. Knepley 
1725cc2f8f65SMatthew G. Knepley     ierr = PetscViewerCreate(comm, &viewer);CHKERRQ(ierr);
1726cc2f8f65SMatthew G. Knepley     ierr = PetscViewerSetType(viewer, PETSCVIEWERHDF5);CHKERRQ(ierr);
1727cc2f8f65SMatthew G. Knepley     ierr = PetscViewerFileSetMode(viewer, FILE_MODE_READ);CHKERRQ(ierr);
1728cc2f8f65SMatthew G. Knepley     ierr = PetscViewerFileSetName(viewer, filename);CHKERRQ(ierr);
1729cc2f8f65SMatthew G. Knepley     ierr = DMCreate(comm, dm);CHKERRQ(ierr);
1730cc2f8f65SMatthew G. Knepley     ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
1731cc2f8f65SMatthew G. Knepley     ierr = DMLoad(*dm, viewer);CHKERRQ(ierr);
1732cc2f8f65SMatthew G. Knepley     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
1733ca522641SMatthew G. Knepley   } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot load file %s: unrecognized extension", filename);
1734ca522641SMatthew G. Knepley   PetscFunctionReturn(0);
1735ca522641SMatthew G. Knepley }
1736ca522641SMatthew G. Knepley 
1737ca522641SMatthew G. Knepley #undef __FUNCT__
17388415267dSToby Isaac #define __FUNCT__ "DMPlexCreateReferenceCell"
17398415267dSToby Isaac /*@
17408415267dSToby Isaac   DMPlexCreateReferenceCell - Create a DMPLEX with the appropriate FEM reference cell
17418415267dSToby Isaac 
17428415267dSToby Isaac   Collective on comm
17438415267dSToby Isaac 
17448415267dSToby Isaac   Input Parameters:
17458415267dSToby Isaac + comm    - The communicator
17468415267dSToby Isaac . dim     - The spatial dimension
17478415267dSToby Isaac - simplex - Flag for simplex, otherwise use a tensor-product cell
17488415267dSToby Isaac 
17498415267dSToby Isaac   Output Parameter:
17508415267dSToby Isaac . refdm - The reference cell
17518415267dSToby Isaac 
17528415267dSToby Isaac   Level: intermediate
17538415267dSToby Isaac 
17548415267dSToby Isaac .keywords: reference cell
17558415267dSToby Isaac .seealso:
17568415267dSToby Isaac @*/
17578415267dSToby Isaac PetscErrorCode DMPlexCreateReferenceCell(MPI_Comm comm, PetscInt dim, PetscBool simplex, DM *refdm)
17588415267dSToby Isaac {
17598415267dSToby Isaac   DM             rdm;
17608415267dSToby Isaac   PetscErrorCode ierr;
17618415267dSToby Isaac 
17628415267dSToby Isaac   PetscFunctionBeginUser;
17638415267dSToby Isaac   ierr = DMCreate(comm, &rdm);CHKERRQ(ierr);
17648415267dSToby Isaac   ierr = DMSetType(rdm, DMPLEX);CHKERRQ(ierr);
17658415267dSToby Isaac   ierr = DMSetDimension(rdm, dim);CHKERRQ(ierr);
17668415267dSToby Isaac   switch (dim) {
17678415267dSToby Isaac   case 0:
17688415267dSToby Isaac   {
17698415267dSToby Isaac     PetscInt    numPoints[1]        = {1};
17708415267dSToby Isaac     PetscInt    coneSize[1]         = {0};
17718415267dSToby Isaac     PetscInt    cones[1]            = {0};
17728415267dSToby Isaac     PetscInt    coneOrientations[1] = {0};
17738415267dSToby Isaac     PetscScalar vertexCoords[1]     = {0.0};
17748415267dSToby Isaac 
17758415267dSToby Isaac     ierr = DMPlexCreateFromDAG(rdm, 0, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
17768415267dSToby Isaac   }
17778415267dSToby Isaac   break;
17788415267dSToby Isaac   case 1:
17798415267dSToby Isaac   {
17808415267dSToby Isaac     PetscInt    numPoints[2]        = {2, 1};
17818415267dSToby Isaac     PetscInt    coneSize[3]         = {2, 0, 0};
17828415267dSToby Isaac     PetscInt    cones[2]            = {1, 2};
17838415267dSToby Isaac     PetscInt    coneOrientations[2] = {0, 0};
17848415267dSToby Isaac     PetscScalar vertexCoords[2]     = {-1.0,  1.0};
17858415267dSToby Isaac 
17868415267dSToby Isaac     ierr = DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
17878415267dSToby Isaac   }
17888415267dSToby Isaac   break;
17898415267dSToby Isaac   case 2:
17908415267dSToby Isaac     if (simplex) {
17918415267dSToby Isaac       PetscInt    numPoints[2]        = {3, 1};
17928415267dSToby Isaac       PetscInt    coneSize[4]         = {3, 0, 0, 0};
17938415267dSToby Isaac       PetscInt    cones[3]            = {1, 2, 3};
17948415267dSToby Isaac       PetscInt    coneOrientations[3] = {0, 0, 0};
17958415267dSToby Isaac       PetscScalar vertexCoords[6]     = {-1.0, -1.0,  1.0, -1.0,  -1.0, 1.0};
17968415267dSToby Isaac 
17978415267dSToby Isaac       ierr = DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
17988415267dSToby Isaac     } else {
17998415267dSToby Isaac       PetscInt    numPoints[2]        = {4, 1};
18008415267dSToby Isaac       PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
18018415267dSToby Isaac       PetscInt    cones[4]            = {1, 2, 3, 4};
18028415267dSToby Isaac       PetscInt    coneOrientations[4] = {0, 0, 0, 0};
18038415267dSToby Isaac       PetscScalar vertexCoords[8]     = {-1.0, -1.0,  1.0, -1.0,  1.0, 1.0,  -1.0, 1.0};
18048415267dSToby Isaac 
18058415267dSToby Isaac       ierr = DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
18068415267dSToby Isaac     }
18078415267dSToby Isaac   break;
18088415267dSToby Isaac   case 3:
18098415267dSToby Isaac     if (simplex) {
18108415267dSToby Isaac       PetscInt    numPoints[2]        = {4, 1};
18118415267dSToby Isaac       PetscInt    coneSize[5]         = {4, 0, 0, 0, 0};
18128415267dSToby Isaac       PetscInt    cones[4]            = {1, 3, 2, 4};
18138415267dSToby Isaac       PetscInt    coneOrientations[4] = {0, 0, 0, 0};
18148415267dSToby Isaac       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};
18158415267dSToby Isaac 
18168415267dSToby Isaac       ierr = DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
18178415267dSToby Isaac     } else {
18188415267dSToby Isaac       PetscInt    numPoints[2]        = {8, 1};
18198415267dSToby Isaac       PetscInt    coneSize[9]         = {8, 0, 0, 0, 0, 0, 0, 0, 0};
18208415267dSToby Isaac       PetscInt    cones[8]            = {1, 4, 3, 2, 5, 6, 7, 8};
18218415267dSToby Isaac       PetscInt    coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
18228415267dSToby Isaac       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,
18238415267dSToby Isaac                                          -1.0, -1.0,  1.0,  1.0, -1.0,  1.0,  1.0, 1.0,  1.0,  -1.0, 1.0,  1.0};
18248415267dSToby Isaac 
18258415267dSToby Isaac       ierr = DMPlexCreateFromDAG(rdm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
18268415267dSToby Isaac     }
18278415267dSToby Isaac   break;
18288415267dSToby Isaac   default:
18298415267dSToby Isaac     SETERRQ1(comm, PETSC_ERR_ARG_WRONG, "Cannot create reference cell for dimension %d", dim);
18308415267dSToby Isaac   }
18318415267dSToby Isaac   *refdm = NULL;
18328415267dSToby Isaac   ierr = DMPlexInterpolate(rdm, refdm);CHKERRQ(ierr);
18338415267dSToby Isaac   ierr = DMPlexCopyCoordinates(rdm, *refdm);CHKERRQ(ierr);
18348415267dSToby Isaac   ierr = DMDestroy(&rdm);CHKERRQ(ierr);
18358415267dSToby Isaac   PetscFunctionReturn(0);
18368415267dSToby Isaac }
1837