xref: /petsc/src/dm/impls/plex/plexcreate.c (revision da80777bb9afa786145217fc25b54e33c3ebf541)
1552f7358SJed Brown #define PETSCDM_DLL
2552f7358SJed Brown #include <petsc-private/pleximpl.h>    /*I   "petscdmplex.h"   I*/
3552f7358SJed Brown #include <petscdmda.h>
4552f7358SJed Brown 
5552f7358SJed Brown #undef __FUNCT__
6552f7358SJed Brown #define __FUNCT__ "DMSetFromOptions_Plex"
7552f7358SJed Brown PetscErrorCode  DMSetFromOptions_Plex(DM dm)
8552f7358SJed Brown {
9552f7358SJed Brown   DM_Plex    *mesh = (DM_Plex *) dm->data;
10552f7358SJed Brown   PetscErrorCode ierr;
11552f7358SJed Brown 
12552f7358SJed Brown   PetscFunctionBegin;
13552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
14552f7358SJed Brown   ierr = PetscOptionsHead("DMPlex Options");CHKERRQ(ierr);
15552f7358SJed Brown     /* Handle DMPlex refinement */
16552f7358SJed Brown     /* Handle associated vectors */
17552f7358SJed Brown     /* Handle viewing */
18552f7358SJed Brown     ierr = PetscOptionsBool("-dm_plex_print_set_values", "Output all set values info", "DMView", PETSC_FALSE, &mesh->printSetValues, PETSC_NULL);CHKERRQ(ierr);
19552f7358SJed Brown     ierr = PetscOptionsInt("-dm_plex_print_fem", "Debug output level all fem computations", "DMView", 0, &mesh->printFEM, PETSC_NULL);CHKERRQ(ierr);
20552f7358SJed Brown   ierr = PetscOptionsTail();CHKERRQ(ierr);
21552f7358SJed Brown   PetscFunctionReturn(0);
22552f7358SJed Brown }
23552f7358SJed Brown 
24552f7358SJed Brown #undef __FUNCT__
25552f7358SJed Brown #define __FUNCT__ "DMPlexCreateSquareBoundary"
26552f7358SJed Brown /*
27552f7358SJed Brown  Simple square boundary:
28552f7358SJed Brown 
29552f7358SJed Brown  18--5-17--4--16
30552f7358SJed Brown   |     |     |
31552f7358SJed Brown   6    10     3
32552f7358SJed Brown   |     |     |
33552f7358SJed Brown  19-11-20--9--15
34552f7358SJed Brown   |     |     |
35552f7358SJed Brown   7     8     2
36552f7358SJed Brown   |     |     |
37552f7358SJed Brown  12--0-13--1--14
38552f7358SJed Brown */
39552f7358SJed Brown PetscErrorCode DMPlexCreateSquareBoundary(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
40552f7358SJed Brown {
41552f7358SJed Brown   PetscInt       numVertices = (edges[0]+1)*(edges[1]+1);
42552f7358SJed Brown   PetscInt       numEdges    = edges[0]*(edges[1]+1) + (edges[0]+1)*edges[1];
43552f7358SJed Brown   PetscInt       markerTop      = 1;
44552f7358SJed Brown   PetscInt       markerBottom   = 1;
45552f7358SJed Brown   PetscInt       markerRight    = 1;
46552f7358SJed Brown   PetscInt       markerLeft     = 1;
47552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
48552f7358SJed Brown   Vec            coordinates;
49552f7358SJed Brown   PetscSection   coordSection;
50552f7358SJed Brown   PetscScalar   *coords;
51552f7358SJed Brown   PetscInt       coordSize;
52552f7358SJed Brown   PetscMPIInt    rank;
53552f7358SJed Brown   PetscInt       v, vx, vy;
54552f7358SJed Brown   PetscErrorCode ierr;
55552f7358SJed Brown 
56552f7358SJed Brown   PetscFunctionBegin;
57552f7358SJed Brown   ierr = PetscOptionsGetBool(((PetscObject) dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, PETSC_NULL);CHKERRQ(ierr);
58552f7358SJed Brown   if (markerSeparate) {
59552f7358SJed Brown     markerTop    = 1;
60552f7358SJed Brown     markerBottom = 0;
61552f7358SJed Brown     markerRight  = 0;
62552f7358SJed Brown     markerLeft   = 0;
63552f7358SJed Brown   }
64552f7358SJed Brown   ierr = MPI_Comm_rank(((PetscObject) dm)->comm, &rank);CHKERRQ(ierr);
65552f7358SJed Brown   if (!rank) {
66552f7358SJed Brown     PetscInt e, ex, ey;
67552f7358SJed Brown 
68552f7358SJed Brown     ierr = DMPlexSetChart(dm, 0, numEdges+numVertices);CHKERRQ(ierr);
69552f7358SJed Brown     for (e = 0; e < numEdges; ++e) {
70552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
71552f7358SJed Brown     }
72552f7358SJed Brown     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
73552f7358SJed Brown     for (vx = 0; vx <= edges[0]; vx++) {
74552f7358SJed Brown       for (ey = 0; ey < edges[1]; ey++) {
75552f7358SJed Brown         PetscInt edge    = vx*edges[1] + ey + edges[0]*(edges[1]+1);
76552f7358SJed Brown         PetscInt vertex  = ey*(edges[0]+1) + vx + numEdges;
77*da80777bSKarl Rupp         PetscInt cone[2];
78552f7358SJed Brown 
79*da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+edges[0]+1;
80552f7358SJed Brown         ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
81552f7358SJed Brown         if (vx == edges[0]) {
82552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerRight);CHKERRQ(ierr);
83552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerRight);CHKERRQ(ierr);
84552f7358SJed Brown           if (ey == edges[1]-1) {
85552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerRight);CHKERRQ(ierr);
86552f7358SJed Brown           }
87552f7358SJed Brown         } else if (vx == 0) {
88552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerLeft);CHKERRQ(ierr);
89552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerLeft);CHKERRQ(ierr);
90552f7358SJed Brown           if (ey == edges[1]-1) {
91552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerLeft);CHKERRQ(ierr);
92552f7358SJed Brown           }
93552f7358SJed Brown         }
94552f7358SJed Brown       }
95552f7358SJed Brown     }
96552f7358SJed Brown     for (vy = 0; vy <= edges[1]; vy++) {
97552f7358SJed Brown       for (ex = 0; ex < edges[0]; ex++) {
98552f7358SJed Brown         PetscInt edge    = vy*edges[0]     + ex;
99552f7358SJed Brown         PetscInt vertex  = vy*(edges[0]+1) + ex + numEdges;
100*da80777bSKarl Rupp         PetscInt cone[2];
101552f7358SJed Brown 
102*da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+1;
103552f7358SJed Brown         ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
104552f7358SJed Brown         if (vy == edges[1]) {
105552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerTop);CHKERRQ(ierr);
106552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerTop);CHKERRQ(ierr);
107552f7358SJed Brown           if (ex == edges[0]-1) {
108552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerTop);CHKERRQ(ierr);
109552f7358SJed Brown           }
110552f7358SJed Brown         } else if (vy == 0) {
111552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerBottom);CHKERRQ(ierr);
112552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerBottom);CHKERRQ(ierr);
113552f7358SJed Brown           if (ex == edges[0]-1) {
114552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerBottom);CHKERRQ(ierr);
115552f7358SJed Brown           }
116552f7358SJed Brown         }
117552f7358SJed Brown       }
118552f7358SJed Brown     }
119552f7358SJed Brown   }
120552f7358SJed Brown   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
121552f7358SJed Brown   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
122552f7358SJed Brown   /* Build coordinates */
123552f7358SJed Brown   ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
124552f7358SJed Brown   ierr = PetscSectionSetChart(coordSection, numEdges, numEdges + numVertices);CHKERRQ(ierr);
125552f7358SJed Brown   for (v = numEdges; v < numEdges+numVertices; ++v) {
126552f7358SJed Brown     ierr = PetscSectionSetDof(coordSection, v, 2);CHKERRQ(ierr);
127552f7358SJed Brown   }
128552f7358SJed Brown   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
129552f7358SJed Brown   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
130552f7358SJed Brown   ierr = VecCreate(((PetscObject) dm)->comm, &coordinates);CHKERRQ(ierr);
131552f7358SJed Brown   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
132552f7358SJed Brown   ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
133552f7358SJed Brown   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
134552f7358SJed Brown   for (vy = 0; vy <= edges[1]; ++vy) {
135552f7358SJed Brown     for (vx = 0; vx <= edges[0]; ++vx) {
136552f7358SJed Brown       coords[(vy*(edges[0]+1)+vx)*2+0] = lower[0] + ((upper[0] - lower[0])/edges[0])*vx;
137552f7358SJed Brown       coords[(vy*(edges[0]+1)+vx)*2+1] = lower[1] + ((upper[1] - lower[1])/edges[1])*vy;
138552f7358SJed Brown     }
139552f7358SJed Brown   }
140552f7358SJed Brown   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
141552f7358SJed Brown   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
142552f7358SJed Brown   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
143552f7358SJed Brown   PetscFunctionReturn(0);
144552f7358SJed Brown }
145552f7358SJed Brown 
146552f7358SJed Brown #undef __FUNCT__
147552f7358SJed Brown #define __FUNCT__ "DMPlexCreateCubeBoundary"
148552f7358SJed Brown /*
149552f7358SJed Brown  Simple cubic boundary:
150552f7358SJed Brown 
151552f7358SJed Brown      2-------3
152552f7358SJed Brown     /|      /|
153552f7358SJed Brown    6-------7 |
154552f7358SJed Brown    | |     | |
155552f7358SJed Brown    | 0-----|-1
156552f7358SJed Brown    |/      |/
157552f7358SJed Brown    4-------5
158552f7358SJed Brown */
159552f7358SJed Brown PetscErrorCode DMPlexCreateCubeBoundary(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt faces[])
160552f7358SJed Brown {
161552f7358SJed Brown   PetscInt       numVertices = (faces[0]+1)*(faces[1]+1)*(faces[2]+1);
162552f7358SJed Brown   PetscInt       numFaces    = 6;
163552f7358SJed Brown   Vec            coordinates;
164552f7358SJed Brown   PetscSection   coordSection;
165552f7358SJed Brown   PetscScalar   *coords;
166552f7358SJed Brown   PetscInt       coordSize;
167552f7358SJed Brown   PetscMPIInt    rank;
168552f7358SJed Brown   PetscInt       v, vx, vy, vz;
169552f7358SJed Brown   PetscErrorCode ierr;
170552f7358SJed Brown 
171552f7358SJed Brown   PetscFunctionBegin;
172552f7358SJed Brown   if ((faces[0] < 1) || (faces[1] < 1) || (faces[2] < 1)) SETERRQ(((PetscObject) dm)->comm, PETSC_ERR_SUP, "Must have at least 1 face per side");
173552f7358SJed Brown   if ((faces[0] > 1) || (faces[1] > 1) || (faces[2] > 1)) SETERRQ(((PetscObject) dm)->comm, PETSC_ERR_SUP, "Currently can't handle more than 1 face per side");
174552f7358SJed Brown   ierr = PetscMalloc(numVertices*2 * sizeof(PetscReal), &coords);CHKERRQ(ierr);
175552f7358SJed Brown   ierr = MPI_Comm_rank(((PetscObject) dm)->comm, &rank);CHKERRQ(ierr);
176552f7358SJed Brown   if (!rank) {
177552f7358SJed Brown     PetscInt f;
178552f7358SJed Brown 
179552f7358SJed Brown     ierr = DMPlexSetChart(dm, 0, numFaces+numVertices);CHKERRQ(ierr);
180552f7358SJed Brown     for (f = 0; f < numFaces; ++f) {
181552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, f, 4);CHKERRQ(ierr);
182552f7358SJed Brown     }
183552f7358SJed Brown     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
184552f7358SJed Brown     for (v = 0; v < numFaces+numVertices; ++v) {
185552f7358SJed Brown       ierr = DMPlexSetLabelValue(dm, "marker", v, 1);CHKERRQ(ierr);
186552f7358SJed Brown     }
187552f7358SJed Brown     { /* Side 0 (Front) */
188*da80777bSKarl Rupp       PetscInt cone[4];
189*da80777bSKarl Rupp       cone[0] = numFaces+4; cone[1] = numFaces+5; cone[2] = numFaces+7; cone[3] = numFaces+6;
190552f7358SJed Brown       ierr = DMPlexSetCone(dm, 0, cone);CHKERRQ(ierr);
191552f7358SJed Brown     }
192552f7358SJed Brown     { /* Side 1 (Back) */
193*da80777bSKarl Rupp       PetscInt cone[4];
194*da80777bSKarl Rupp       cone[0] = numFaces+1; cone[1] = numFaces+0; cone[2] = numFaces+2; cone[3] = numFaces+3;
195552f7358SJed Brown       ierr = DMPlexSetCone(dm, 1, cone);CHKERRQ(ierr);
196552f7358SJed Brown     }
197552f7358SJed Brown     { /* Side 2 (Bottom) */
198*da80777bSKarl Rupp       PetscInt cone[4];
199*da80777bSKarl Rupp       cone[0] = numFaces+0; cone[1] = numFaces+1; cone[2] = numFaces+5; cone[3] = numFaces+4;
200552f7358SJed Brown       ierr = DMPlexSetCone(dm, 2, cone);CHKERRQ(ierr);
201552f7358SJed Brown     }
202552f7358SJed Brown     { /* Side 3 (Top) */
203*da80777bSKarl Rupp       PetscInt cone[4];
204*da80777bSKarl Rupp       cone[0] = numFaces+6; cone[1] = numFaces+7; cone[2] = numFaces+3; cone[3] = numFaces+2;
205552f7358SJed Brown       ierr = DMPlexSetCone(dm, 3, cone);CHKERRQ(ierr);
206552f7358SJed Brown     }
207552f7358SJed Brown     { /* Side 4 (Left) */
208*da80777bSKarl Rupp       PetscInt cone[4];
209*da80777bSKarl Rupp       cone[0] = numFaces+0; cone[1] = numFaces+4; cone[2] = numFaces+6; cone[3] = numFaces+2;
210552f7358SJed Brown       ierr = DMPlexSetCone(dm, 4, cone);CHKERRQ(ierr);
211552f7358SJed Brown     }
212552f7358SJed Brown     { /* Side 5 (Right) */
213*da80777bSKarl Rupp       PetscInt cone[4];
214*da80777bSKarl Rupp       cone[0] = numFaces+5; cone[1] = numFaces+1; cone[2] = numFaces+3; cone[3] = numFaces+7;
215552f7358SJed Brown       ierr = DMPlexSetCone(dm, 5, cone);CHKERRQ(ierr);
216552f7358SJed Brown     }
217552f7358SJed Brown   }
218552f7358SJed Brown   ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
219552f7358SJed Brown   ierr = DMPlexStratify(dm);CHKERRQ(ierr);
220552f7358SJed Brown   /* Build coordinates */
221552f7358SJed Brown   ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
222552f7358SJed Brown   ierr = PetscSectionSetChart(coordSection, numFaces, numFaces + numVertices);CHKERRQ(ierr);
223552f7358SJed Brown   for (v = numFaces; v < numFaces+numVertices; ++v) {
224552f7358SJed Brown     ierr = PetscSectionSetDof(coordSection, v, 3);CHKERRQ(ierr);
225552f7358SJed Brown   }
226552f7358SJed Brown   ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
227552f7358SJed Brown   ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
228552f7358SJed Brown   ierr = VecCreate(((PetscObject) dm)->comm, &coordinates);CHKERRQ(ierr);
229552f7358SJed Brown   ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
230552f7358SJed Brown   ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
231552f7358SJed Brown   ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
232552f7358SJed Brown   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
233552f7358SJed Brown   for (vz = 0; vz <= faces[2]; ++vz) {
234552f7358SJed Brown     for (vy = 0; vy <= faces[1]; ++vy) {
235552f7358SJed Brown       for (vx = 0; vx <= faces[0]; ++vx) {
236552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+0] = lower[0] + ((upper[0] - lower[0])/faces[0])*vx;
237552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+1] = lower[1] + ((upper[1] - lower[1])/faces[1])*vy;
238552f7358SJed Brown         coords[((vz*(faces[1]+1)+vy)*(faces[0]+1)+vx)*3+2] = lower[2] + ((upper[2] - lower[2])/faces[2])*vz;
239552f7358SJed Brown       }
240552f7358SJed Brown     }
241552f7358SJed Brown   }
242552f7358SJed Brown   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
243552f7358SJed Brown   ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
244552f7358SJed Brown   ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
245552f7358SJed Brown   PetscFunctionReturn(0);
246552f7358SJed Brown }
247552f7358SJed Brown 
248552f7358SJed Brown #undef __FUNCT__
249552f7358SJed Brown #define __FUNCT__ "DMPlexCreateSquareMesh"
250552f7358SJed Brown /*
251552f7358SJed Brown  Simple square mesh:
252552f7358SJed Brown 
253552f7358SJed Brown  22--8-23--9--24
254552f7358SJed Brown   |     |     |
255552f7358SJed Brown  13  2 14  3  15
256552f7358SJed Brown   |     |     |
257552f7358SJed Brown  19--6-20--7--21
258552f7358SJed Brown   |     |     |
259552f7358SJed Brown  10  0 11  1 12
260552f7358SJed Brown   |     |     |
261552f7358SJed Brown  16--4-17--5--18
262552f7358SJed Brown */
263552f7358SJed Brown PetscErrorCode DMPlexCreateSquareMesh(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
264552f7358SJed Brown {
265552f7358SJed Brown   PetscInt       markerTop      = 1;
266552f7358SJed Brown   PetscInt       markerBottom   = 1;
267552f7358SJed Brown   PetscInt       markerRight    = 1;
268552f7358SJed Brown   PetscInt       markerLeft     = 1;
269552f7358SJed Brown   PetscBool      markerSeparate = PETSC_FALSE;
270552f7358SJed Brown   PetscMPIInt    rank;
271552f7358SJed Brown   PetscErrorCode ierr;
272552f7358SJed Brown 
273552f7358SJed Brown   PetscFunctionBegin;
274552f7358SJed Brown   ierr = MPI_Comm_rank(((PetscObject) dm)->comm, &rank);CHKERRQ(ierr);
275552f7358SJed Brown   ierr = PetscOptionsGetBool(((PetscObject) dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, PETSC_NULL);CHKERRQ(ierr);
276552f7358SJed Brown   if (markerSeparate) {
277552f7358SJed Brown     markerTop    = 3;
278552f7358SJed Brown     markerBottom = 1;
279552f7358SJed Brown     markerRight  = 2;
280552f7358SJed Brown     markerLeft   = 4;
281552f7358SJed Brown   }
282552f7358SJed Brown   {
283552f7358SJed Brown     const PetscInt numXEdges    = !rank ? edges[0]   : 0;
284552f7358SJed Brown     const PetscInt numYEdges    = !rank ? edges[1]   : 0;
285552f7358SJed Brown     const PetscInt numXVertices = !rank ? edges[0]+1 : 0;
286552f7358SJed Brown     const PetscInt numYVertices = !rank ? edges[1]+1 : 0;
287552f7358SJed Brown     const PetscInt numTotXEdges = numXEdges*numYVertices;
288552f7358SJed Brown     const PetscInt numTotYEdges = numYEdges*numXVertices;
289552f7358SJed Brown     const PetscInt numVertices  = numXVertices*numYVertices;
290552f7358SJed Brown     const PetscInt numEdges     = numTotXEdges + numTotYEdges;
291552f7358SJed Brown     const PetscInt numFaces     = numXEdges*numYEdges;
292552f7358SJed Brown     const PetscInt firstVertex  = numFaces;
293552f7358SJed Brown     const PetscInt firstXEdge   = numFaces + numVertices;
294552f7358SJed Brown     const PetscInt firstYEdge   = numFaces + numVertices + numTotXEdges;
295552f7358SJed Brown     Vec            coordinates;
296552f7358SJed Brown     PetscSection   coordSection;
297552f7358SJed Brown     PetscScalar   *coords;
298552f7358SJed Brown     PetscInt       coordSize;
299552f7358SJed Brown     PetscInt       v, vx, vy;
300552f7358SJed Brown     PetscInt       f, fx, fy, e, ex, ey;
301552f7358SJed Brown 
302552f7358SJed Brown     ierr = DMPlexSetChart(dm, 0, numFaces+numEdges+numVertices);CHKERRQ(ierr);
303552f7358SJed Brown     for (f = 0; f < numFaces; ++f) {
304552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, f, 4);CHKERRQ(ierr);
305552f7358SJed Brown     }
306552f7358SJed Brown     for (e = firstXEdge; e < firstXEdge+numEdges; ++e) {
307552f7358SJed Brown       ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
308552f7358SJed Brown     }
309552f7358SJed Brown     ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
310552f7358SJed Brown     /* Build faces */
311552f7358SJed Brown     for (fy = 0; fy < numYEdges; fy++) {
312552f7358SJed Brown       for (fx = 0; fx < numXEdges; fx++) {
313552f7358SJed Brown         const PetscInt face    = fy*numXEdges + fx;
314552f7358SJed Brown         const PetscInt edgeL   = firstYEdge + fx*numYEdges + fy;
315552f7358SJed Brown         const PetscInt edgeB   = firstXEdge + fy*numXEdges + fx;
316552f7358SJed Brown         const PetscInt ornt[4] = {0,     0,               -2,              -2};
317*da80777bSKarl Rupp         PetscInt cone[4];
318552f7358SJed Brown 
319*da80777bSKarl Rupp         cone[0] = edgeB; cone[1] = edgeL+numYEdges; cone[2] = edgeB+numXEdges; cone[3] = edgeL;
320552f7358SJed Brown         ierr = DMPlexSetCone(dm, face, cone);CHKERRQ(ierr);
321552f7358SJed Brown         ierr = DMPlexSetConeOrientation(dm, face, ornt);CHKERRQ(ierr);
322552f7358SJed Brown       }
323552f7358SJed Brown     }
324552f7358SJed Brown     /* Build Y edges*/
325552f7358SJed Brown     for (vx = 0; vx < numXVertices; vx++) {
326552f7358SJed Brown       for (ey = 0; ey < numYEdges; ey++) {
327552f7358SJed Brown         const PetscInt edge    = firstYEdge  + vx*numYEdges + ey;
328552f7358SJed Brown         const PetscInt vertex  = firstVertex + ey*numXVertices + vx;
329*da80777bSKarl Rupp         PetscInt cone[2];
330552f7358SJed Brown 
331*da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+numXVertices;
332552f7358SJed Brown         ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
333552f7358SJed Brown         if (vx == numXVertices-1) {
334552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerRight);CHKERRQ(ierr);
335552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerRight);CHKERRQ(ierr);
336552f7358SJed Brown           if (ey == numYEdges-1) {
337552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerRight);CHKERRQ(ierr);
338552f7358SJed Brown           }
339552f7358SJed Brown         } else if (vx == 0) {
340552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerLeft);CHKERRQ(ierr);
341552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerLeft);CHKERRQ(ierr);
342552f7358SJed Brown           if (ey == numYEdges-1) {
343552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerLeft);CHKERRQ(ierr);
344552f7358SJed Brown           }
345552f7358SJed Brown         }
346552f7358SJed Brown       }
347552f7358SJed Brown     }
348552f7358SJed Brown     /* Build X edges*/
349552f7358SJed Brown     for (vy = 0; vy < numYVertices; vy++) {
350552f7358SJed Brown       for (ex = 0; ex < numXEdges; ex++) {
351552f7358SJed Brown         const PetscInt edge    = firstXEdge  + vy*numXEdges + ex;
352552f7358SJed Brown         const PetscInt vertex  = firstVertex + vy*numXVertices + ex;
353*da80777bSKarl Rupp         PetscInt cone[2];
354552f7358SJed Brown 
355*da80777bSKarl Rupp         cone[0] = vertex; cone[1] = vertex+1;
356552f7358SJed Brown         ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
357552f7358SJed Brown         if (vy == numYVertices-1) {
358552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerTop);CHKERRQ(ierr);
359552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerTop);CHKERRQ(ierr);
360552f7358SJed Brown           if (ex == numXEdges-1) {
361552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerTop);CHKERRQ(ierr);
362552f7358SJed Brown           }
363552f7358SJed Brown         } else if (vy == 0) {
364552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", edge,    markerBottom);CHKERRQ(ierr);
365552f7358SJed Brown           ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerBottom);CHKERRQ(ierr);
366552f7358SJed Brown           if (ex == numXEdges-1) {
367552f7358SJed Brown             ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerBottom);CHKERRQ(ierr);
368552f7358SJed Brown           }
369552f7358SJed Brown         }
370552f7358SJed Brown       }
371552f7358SJed Brown     }
372552f7358SJed Brown     ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
373552f7358SJed Brown     ierr = DMPlexStratify(dm);CHKERRQ(ierr);
374552f7358SJed Brown     /* Build coordinates */
375552f7358SJed Brown     ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
376552f7358SJed Brown     ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numVertices);CHKERRQ(ierr);
377552f7358SJed Brown     for (v = firstVertex; v < firstVertex+numVertices; ++v) {
378552f7358SJed Brown       ierr = PetscSectionSetDof(coordSection, v, 2);CHKERRQ(ierr);
379552f7358SJed Brown     }
380552f7358SJed Brown     ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
381552f7358SJed Brown     ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
382552f7358SJed Brown     ierr = VecCreate(((PetscObject) dm)->comm, &coordinates);CHKERRQ(ierr);
383552f7358SJed Brown     ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
384552f7358SJed Brown     ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
385552f7358SJed Brown     ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
386552f7358SJed Brown     for (vy = 0; vy < numYVertices; ++vy) {
387552f7358SJed Brown       for (vx = 0; vx < numXVertices; ++vx) {
388552f7358SJed Brown         coords[(vy*numXVertices+vx)*2+0] = lower[0] + ((upper[0] - lower[0])/numXEdges)*vx;
389552f7358SJed Brown         coords[(vy*numXVertices+vx)*2+1] = lower[1] + ((upper[1] - lower[1])/numYEdges)*vy;
390552f7358SJed Brown       }
391552f7358SJed Brown     }
392552f7358SJed Brown     ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
393552f7358SJed Brown     ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
394552f7358SJed Brown     ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
395552f7358SJed Brown   }
396552f7358SJed Brown   PetscFunctionReturn(0);
397552f7358SJed Brown }
398552f7358SJed Brown 
399552f7358SJed Brown #undef __FUNCT__
400552f7358SJed Brown #define __FUNCT__ "DMPlexCreateBoxMesh"
4010adebc6cSBarry Smith PetscErrorCode DMPlexCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool interpolate, DM *dm)
4020adebc6cSBarry Smith {
403552f7358SJed Brown   DM             boundary;
404552f7358SJed Brown   PetscErrorCode ierr;
405552f7358SJed Brown 
406552f7358SJed Brown   PetscFunctionBegin;
407552f7358SJed Brown   PetscValidPointer(dm, 4);
408552f7358SJed Brown   ierr = DMCreate(comm, &boundary);CHKERRQ(ierr);
409552f7358SJed Brown   PetscValidLogicalCollectiveInt(boundary,dim,2);
410552f7358SJed Brown   ierr = DMSetType(boundary, DMPLEX);CHKERRQ(ierr);
411552f7358SJed Brown   ierr = DMPlexSetDimension(boundary, dim-1);CHKERRQ(ierr);
412552f7358SJed Brown   switch (dim) {
413552f7358SJed Brown   case 2:
414552f7358SJed Brown   {
415552f7358SJed Brown     PetscReal lower[2] = {0.0, 0.0};
416552f7358SJed Brown     PetscReal upper[2] = {1.0, 1.0};
417552f7358SJed Brown     PetscInt  edges[2] = {2, 2};
418552f7358SJed Brown 
419552f7358SJed Brown     ierr = DMPlexCreateSquareBoundary(boundary, lower, upper, edges);CHKERRQ(ierr);
420552f7358SJed Brown     break;
421552f7358SJed Brown   }
422552f7358SJed Brown   case 3:
423552f7358SJed Brown   {
424552f7358SJed Brown     PetscReal lower[3] = {0.0, 0.0, 0.0};
425552f7358SJed Brown     PetscReal upper[3] = {1.0, 1.0, 1.0};
426552f7358SJed Brown     PetscInt  faces[3] = {1, 1, 1};
427552f7358SJed Brown 
428552f7358SJed Brown     ierr = DMPlexCreateCubeBoundary(boundary, lower, upper, faces);CHKERRQ(ierr);
429552f7358SJed Brown     break;
430552f7358SJed Brown   }
431552f7358SJed Brown   default:
432552f7358SJed Brown     SETERRQ1(comm, PETSC_ERR_SUP, "Dimension not supported: %d", dim);
433552f7358SJed Brown   }
434552f7358SJed Brown   ierr = DMPlexGenerate(boundary, PETSC_NULL, interpolate, dm);CHKERRQ(ierr);
435552f7358SJed Brown   ierr = DMDestroy(&boundary);CHKERRQ(ierr);
436552f7358SJed Brown   PetscFunctionReturn(0);
437552f7358SJed Brown }
438552f7358SJed Brown 
439552f7358SJed Brown #undef __FUNCT__
440552f7358SJed Brown #define __FUNCT__ "DMPlexCreateHexBoxMesh"
441a6dfd86eSKarl Rupp PetscErrorCode DMPlexCreateHexBoxMesh(MPI_Comm comm, PetscInt dim, const PetscInt cells[], DM *dm)
442a6dfd86eSKarl Rupp {
443552f7358SJed Brown   PetscErrorCode ierr;
444552f7358SJed Brown 
445552f7358SJed Brown   PetscFunctionBegin;
446552f7358SJed Brown   PetscValidPointer(dm, 4);
447552f7358SJed Brown   ierr = DMCreate(comm, dm);CHKERRQ(ierr);
448552f7358SJed Brown   PetscValidLogicalCollectiveInt(*dm,dim,2);
449552f7358SJed Brown   ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
450552f7358SJed Brown   ierr = DMPlexSetDimension(*dm, dim);CHKERRQ(ierr);
451552f7358SJed Brown   switch (dim) {
452552f7358SJed Brown   case 2:
453552f7358SJed Brown   {
454552f7358SJed Brown     PetscReal lower[2] = {0.0, 0.0};
455552f7358SJed Brown     PetscReal upper[2] = {1.0, 1.0};
456552f7358SJed Brown 
457552f7358SJed Brown     ierr = DMPlexCreateSquareMesh(*dm, lower, upper, cells);CHKERRQ(ierr);
458552f7358SJed Brown     break;
459552f7358SJed Brown   }
460552f7358SJed Brown #if 0
461552f7358SJed Brown   case 3:
462552f7358SJed Brown   {
463552f7358SJed Brown     PetscReal lower[3] = {0.0, 0.0, 0.0};
464552f7358SJed Brown     PetscReal upper[3] = {1.0, 1.0, 1.0};
465552f7358SJed Brown 
466552f7358SJed Brown     ierr = DMPlexCreateCubeMesh(boundary, lower, upper, cells);CHKERRQ(ierr);
467552f7358SJed Brown     break;
468552f7358SJed Brown   }
469552f7358SJed Brown #endif
470552f7358SJed Brown   default:
471552f7358SJed Brown     SETERRQ1(comm, PETSC_ERR_SUP, "Dimension not supported: %d", dim);
472552f7358SJed Brown   }
473552f7358SJed Brown   PetscFunctionReturn(0);
474552f7358SJed Brown }
475552f7358SJed Brown 
476552f7358SJed Brown /* External function declarations here */
477552f7358SJed Brown extern PetscErrorCode DMCreateInterpolation_Plex(DM dmCoarse, DM dmFine, Mat *interpolation, Vec *scaling);
478552f7358SJed Brown extern PetscErrorCode DMCreateMatrix_Plex(DM dm, MatType mtype, Mat *J);
479552f7358SJed Brown extern PetscErrorCode DMCreateCoordinateDM_Plex(DM dm, DM *cdm);
480552f7358SJed Brown extern PetscErrorCode DMRefine_Plex(DM dm, MPI_Comm comm, DM *dmRefined);
481552f7358SJed Brown extern PetscErrorCode DMSetUp_Plex(DM dm);
482552f7358SJed Brown extern PetscErrorCode DMDestroy_Plex(DM dm);
483552f7358SJed Brown extern PetscErrorCode DMView_Plex(DM dm, PetscViewer viewer);
484552f7358SJed Brown extern PetscErrorCode DMCreateSubDM_Plex(DM dm, PetscInt numFields, PetscInt fields[], IS *is, DM *subdm);
485552f7358SJed Brown extern PetscErrorCode DMLocatePoints_Plex(DM dm, Vec v, IS *cellIS);
486552f7358SJed Brown 
487552f7358SJed Brown #undef __FUNCT__
488552f7358SJed Brown #define __FUNCT__ "DMCreateGlobalVector_Plex"
489552f7358SJed Brown static PetscErrorCode DMCreateGlobalVector_Plex(DM dm,Vec *vec)
490552f7358SJed Brown {
491552f7358SJed Brown   PetscErrorCode ierr;
492552f7358SJed Brown 
493552f7358SJed Brown   PetscFunctionBegin;
494552f7358SJed Brown   ierr = DMCreateGlobalVector_Section_Private(dm,vec);CHKERRQ(ierr);
495552f7358SJed Brown   /* ierr = VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM);CHKERRQ(ierr); */
496552f7358SJed Brown   ierr = VecSetOperation(*vec, VECOP_VIEW, (void(*)(void)) VecView_Plex);CHKERRQ(ierr);
497552f7358SJed Brown   PetscFunctionReturn(0);
498552f7358SJed Brown }
499552f7358SJed Brown 
500552f7358SJed Brown #undef __FUNCT__
501552f7358SJed Brown #define __FUNCT__ "DMCreateLocalVector_Plex"
502552f7358SJed Brown static PetscErrorCode DMCreateLocalVector_Plex(DM dm,Vec *vec)
503552f7358SJed Brown {
504552f7358SJed Brown   PetscErrorCode ierr;
505552f7358SJed Brown 
506552f7358SJed Brown   PetscFunctionBegin;
507552f7358SJed Brown   ierr = DMCreateLocalVector_Section_Private(dm,vec);CHKERRQ(ierr);
508552f7358SJed Brown   ierr = VecSetOperation(*vec, VECOP_VIEW, (void(*)(void)) VecView_Plex_Local);CHKERRQ(ierr);
509552f7358SJed Brown   PetscFunctionReturn(0);
510552f7358SJed Brown }
511552f7358SJed Brown 
512552f7358SJed Brown 
513552f7358SJed Brown #undef __FUNCT__
514552f7358SJed Brown #define __FUNCT__ "DMInitialize_Plex"
515552f7358SJed Brown PetscErrorCode DMInitialize_Plex(DM dm)
516552f7358SJed Brown {
517552f7358SJed Brown   PetscErrorCode ierr;
518552f7358SJed Brown 
519552f7358SJed Brown   PetscFunctionBegin;
520552f7358SJed Brown   ierr = PetscStrallocpy(VECSTANDARD, (char**)&dm->vectype);CHKERRQ(ierr);
521552f7358SJed Brown   dm->ops->view               = DMView_Plex;
522552f7358SJed Brown   dm->ops->setfromoptions     = DMSetFromOptions_Plex;
523552f7358SJed Brown   dm->ops->setup              = DMSetUp_Plex;
524552f7358SJed Brown   dm->ops->createglobalvector = DMCreateGlobalVector_Plex;
525552f7358SJed Brown   dm->ops->createlocalvector  = DMCreateLocalVector_Plex;
526552f7358SJed Brown   dm->ops->createlocaltoglobalmapping      = PETSC_NULL;
527552f7358SJed Brown   dm->ops->createlocaltoglobalmappingblock = PETSC_NULL;
528552f7358SJed Brown   dm->ops->createfieldis      = PETSC_NULL;
529552f7358SJed Brown   dm->ops->createcoordinatedm = DMCreateCoordinateDM_Plex;
530552f7358SJed Brown   dm->ops->getcoloring        = 0;
531552f7358SJed Brown   dm->ops->creatematrix       = DMCreateMatrix_Plex;
532552f7358SJed Brown   dm->ops->createinterpolation= 0;
533552f7358SJed Brown   dm->ops->getaggregates      = 0;
534552f7358SJed Brown   dm->ops->getinjection       = 0;
535552f7358SJed Brown   dm->ops->refine             = DMRefine_Plex;
536552f7358SJed Brown   dm->ops->coarsen            = 0;
537552f7358SJed Brown   dm->ops->refinehierarchy    = 0;
538552f7358SJed Brown   dm->ops->coarsenhierarchy   = 0;
539552f7358SJed Brown   dm->ops->globaltolocalbegin = PETSC_NULL;
540552f7358SJed Brown   dm->ops->globaltolocalend   = PETSC_NULL;
541552f7358SJed Brown   dm->ops->localtoglobalbegin = PETSC_NULL;
542552f7358SJed Brown   dm->ops->localtoglobalend   = PETSC_NULL;
543552f7358SJed Brown   dm->ops->destroy            = DMDestroy_Plex;
544552f7358SJed Brown   dm->ops->createsubdm        = DMCreateSubDM_Plex;
545552f7358SJed Brown   dm->ops->locatepoints       = DMLocatePoints_Plex;
546552f7358SJed Brown   PetscFunctionReturn(0);
547552f7358SJed Brown }
548552f7358SJed Brown 
549552f7358SJed Brown EXTERN_C_BEGIN
550552f7358SJed Brown #undef __FUNCT__
551552f7358SJed Brown #define __FUNCT__ "DMCreate_Plex"
552552f7358SJed Brown PetscErrorCode DMCreate_Plex(DM dm)
553552f7358SJed Brown {
554552f7358SJed Brown   DM_Plex       *mesh;
555770b213bSMatthew G Knepley   PetscInt       unit, d;
556552f7358SJed Brown   PetscErrorCode ierr;
557552f7358SJed Brown 
558552f7358SJed Brown   PetscFunctionBegin;
559552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
560552f7358SJed Brown   ierr = PetscNewLog(dm, DM_Plex, &mesh);CHKERRQ(ierr);
561552f7358SJed Brown   dm->data = mesh;
562552f7358SJed Brown 
563552f7358SJed Brown   mesh->refct             = 1;
564552f7358SJed Brown   mesh->dim               = 0;
565552f7358SJed Brown   ierr = PetscSectionCreate(((PetscObject) dm)->comm, &mesh->coneSection);CHKERRQ(ierr);
566552f7358SJed Brown   mesh->maxConeSize       = 0;
567552f7358SJed Brown   mesh->cones             = PETSC_NULL;
568552f7358SJed Brown   mesh->coneOrientations  = PETSC_NULL;
569552f7358SJed Brown   ierr = PetscSectionCreate(((PetscObject) dm)->comm, &mesh->supportSection);CHKERRQ(ierr);
570552f7358SJed Brown   mesh->maxSupportSize    = 0;
571552f7358SJed Brown   mesh->supports          = PETSC_NULL;
572552f7358SJed Brown   mesh->refinementUniform = PETSC_TRUE;
573552f7358SJed Brown   mesh->refinementLimit   = -1.0;
574552f7358SJed Brown 
575552f7358SJed Brown   mesh->facesTmp         = PETSC_NULL;
576552f7358SJed Brown 
577552f7358SJed Brown   mesh->subpointMap      = PETSC_NULL;
578552f7358SJed Brown 
579552f7358SJed Brown   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) {
580552f7358SJed Brown     mesh->scale[unit]    = 1.0;
581552f7358SJed Brown   }
582552f7358SJed Brown 
583552f7358SJed Brown   mesh->labels               = PETSC_NULL;
584552f7358SJed Brown   mesh->globalVertexNumbers  = PETSC_NULL;
585552f7358SJed Brown   mesh->globalCellNumbers    = PETSC_NULL;
586770b213bSMatthew G Knepley   for (d = 0; d < 8; ++d) {
587770b213bSMatthew G Knepley     mesh->hybridPointMax[d]  = PETSC_DETERMINE;
588770b213bSMatthew G Knepley   }
589552f7358SJed Brown   mesh->vtkCellHeight        = 0;
590552f7358SJed Brown   mesh->preallocCenterDim    = -1;
591552f7358SJed Brown 
592552f7358SJed Brown   mesh->integrateResidualFEM       = PETSC_NULL;
593552f7358SJed Brown   mesh->integrateJacobianActionFEM = PETSC_NULL;
594552f7358SJed Brown   mesh->integrateJacobianFEM       = PETSC_NULL;
595552f7358SJed Brown 
596552f7358SJed Brown   mesh->printSetValues       = PETSC_FALSE;
597552f7358SJed Brown   mesh->printFEM             = 0;
598552f7358SJed Brown 
599552f7358SJed Brown   ierr = DMInitialize_Plex(dm);CHKERRQ(ierr);
600552f7358SJed Brown   PetscFunctionReturn(0);
601552f7358SJed Brown }
602552f7358SJed Brown EXTERN_C_END
603552f7358SJed Brown 
604552f7358SJed Brown #undef __FUNCT__
605552f7358SJed Brown #define __FUNCT__ "DMPlexCreate"
606552f7358SJed Brown /*@
607552f7358SJed Brown   DMPlexCreate - Creates a DMPlex object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
608552f7358SJed Brown 
609552f7358SJed Brown   Collective on MPI_Comm
610552f7358SJed Brown 
611552f7358SJed Brown   Input Parameter:
612552f7358SJed Brown . comm - The communicator for the DMPlex object
613552f7358SJed Brown 
614552f7358SJed Brown   Output Parameter:
615552f7358SJed Brown . mesh  - The DMPlex object
616552f7358SJed Brown 
617552f7358SJed Brown   Level: beginner
618552f7358SJed Brown 
619552f7358SJed Brown .keywords: DMPlex, create
620552f7358SJed Brown @*/
621552f7358SJed Brown PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
622552f7358SJed Brown {
623552f7358SJed Brown   PetscErrorCode ierr;
624552f7358SJed Brown 
625552f7358SJed Brown   PetscFunctionBegin;
626552f7358SJed Brown   PetscValidPointer(mesh,2);
627552f7358SJed Brown   ierr = DMCreate(comm, mesh);CHKERRQ(ierr);
628552f7358SJed Brown   ierr = DMSetType(*mesh, DMPLEX);CHKERRQ(ierr);
629552f7358SJed Brown   PetscFunctionReturn(0);
630552f7358SJed Brown }
631552f7358SJed Brown 
632552f7358SJed Brown #undef __FUNCT__
633552f7358SJed Brown #define __FUNCT__ "DMPlexClone"
634552f7358SJed Brown /*@
635552f7358SJed Brown   DMPlexClone - Creates a DMPlex object with the same mesh as the original.
636552f7358SJed Brown 
637552f7358SJed Brown   Collective on MPI_Comm
638552f7358SJed Brown 
639552f7358SJed Brown   Input Parameter:
640552f7358SJed Brown . dm - The original DMPlex object
641552f7358SJed Brown 
642552f7358SJed Brown   Output Parameter:
643552f7358SJed Brown . newdm  - The new DMPlex object
644552f7358SJed Brown 
645552f7358SJed Brown   Level: beginner
646552f7358SJed Brown 
647552f7358SJed Brown .keywords: DMPlex, create
648552f7358SJed Brown @*/
649552f7358SJed Brown PetscErrorCode DMPlexClone(DM dm, DM *newdm)
650552f7358SJed Brown {
651552f7358SJed Brown   DM_Plex    *mesh;
652552f7358SJed Brown   void          *ctx;
653552f7358SJed Brown   PetscErrorCode ierr;
654552f7358SJed Brown 
655552f7358SJed Brown   PetscFunctionBegin;
656552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
657552f7358SJed Brown   PetscValidPointer(newdm,2);
658552f7358SJed Brown   ierr = DMCreate(((PetscObject) dm)->comm, newdm);CHKERRQ(ierr);
659552f7358SJed Brown   ierr = PetscSFDestroy(&(*newdm)->sf);CHKERRQ(ierr);
660552f7358SJed Brown   ierr = PetscObjectReference((PetscObject) dm->sf);CHKERRQ(ierr);
661552f7358SJed Brown   (*newdm)->sf = dm->sf;
662552f7358SJed Brown   mesh = (DM_Plex *) dm->data;
663552f7358SJed Brown   mesh->refct++;
664552f7358SJed Brown   (*newdm)->data = mesh;
665552f7358SJed Brown   ierr = PetscObjectChangeTypeName((PetscObject) *newdm, DMPLEX);CHKERRQ(ierr);
666552f7358SJed Brown   ierr = DMInitialize_Plex(*newdm);CHKERRQ(ierr);
667552f7358SJed Brown   ierr = DMGetApplicationContext(dm, &ctx);CHKERRQ(ierr);
668552f7358SJed Brown   ierr = DMSetApplicationContext(*newdm, ctx);CHKERRQ(ierr);
669552f7358SJed Brown   PetscFunctionReturn(0);
670552f7358SJed Brown }
671