xref: /petsc/src/dm/impls/plex/plexcreate.c (revision 8865f1ea4ea560cd84ab8db62e98b7095cdff96f)
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;
77da80777bSKarl Rupp         PetscInt cone[2];
78552f7358SJed Brown 
79da80777bSKarl 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;
100da80777bSKarl Rupp         PetscInt cone[2];
101552f7358SJed Brown 
102da80777bSKarl 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) */
188da80777bSKarl Rupp       PetscInt cone[4];
189da80777bSKarl 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) */
193da80777bSKarl Rupp       PetscInt cone[4];
194da80777bSKarl 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) */
198da80777bSKarl Rupp       PetscInt cone[4];
199da80777bSKarl 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) */
203da80777bSKarl Rupp       PetscInt cone[4];
204da80777bSKarl 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) */
208da80777bSKarl Rupp       PetscInt cone[4];
209da80777bSKarl 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) */
213da80777bSKarl Rupp       PetscInt cone[4];
214da80777bSKarl 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};
317da80777bSKarl Rupp         PetscInt       cone[4];
318552f7358SJed Brown 
319da80777bSKarl 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;
329da80777bSKarl Rupp         PetscInt       cone[2];
330552f7358SJed Brown 
331da80777bSKarl 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;
353da80777bSKarl Rupp         PetscInt       cone[2];
354552f7358SJed Brown 
355da80777bSKarl 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);
521*8865f1eaSKarl Rupp 
522552f7358SJed Brown   dm->ops->view                            = DMView_Plex;
523552f7358SJed Brown   dm->ops->setfromoptions                  = DMSetFromOptions_Plex;
524552f7358SJed Brown   dm->ops->setup                           = DMSetUp_Plex;
525552f7358SJed Brown   dm->ops->createglobalvector              = DMCreateGlobalVector_Plex;
526552f7358SJed Brown   dm->ops->createlocalvector               = DMCreateLocalVector_Plex;
527552f7358SJed Brown   dm->ops->createlocaltoglobalmapping      = PETSC_NULL;
528552f7358SJed Brown   dm->ops->createlocaltoglobalmappingblock = PETSC_NULL;
529552f7358SJed Brown   dm->ops->createfieldis                   = PETSC_NULL;
530552f7358SJed Brown   dm->ops->createcoordinatedm              = DMCreateCoordinateDM_Plex;
531552f7358SJed Brown   dm->ops->getcoloring                     = 0;
532552f7358SJed Brown   dm->ops->creatematrix                    = DMCreateMatrix_Plex;
533552f7358SJed Brown   dm->ops->createinterpolation             = 0;
534552f7358SJed Brown   dm->ops->getaggregates                   = 0;
535552f7358SJed Brown   dm->ops->getinjection                    = 0;
536552f7358SJed Brown   dm->ops->refine                          = DMRefine_Plex;
537552f7358SJed Brown   dm->ops->coarsen                         = 0;
538552f7358SJed Brown   dm->ops->refinehierarchy                 = 0;
539552f7358SJed Brown   dm->ops->coarsenhierarchy                = 0;
540552f7358SJed Brown   dm->ops->globaltolocalbegin              = PETSC_NULL;
541552f7358SJed Brown   dm->ops->globaltolocalend                = PETSC_NULL;
542552f7358SJed Brown   dm->ops->localtoglobalbegin              = PETSC_NULL;
543552f7358SJed Brown   dm->ops->localtoglobalend                = PETSC_NULL;
544552f7358SJed Brown   dm->ops->destroy                         = DMDestroy_Plex;
545552f7358SJed Brown   dm->ops->createsubdm                     = DMCreateSubDM_Plex;
546552f7358SJed Brown   dm->ops->locatepoints                    = DMLocatePoints_Plex;
547552f7358SJed Brown   PetscFunctionReturn(0);
548552f7358SJed Brown }
549552f7358SJed Brown 
550552f7358SJed Brown EXTERN_C_BEGIN
551552f7358SJed Brown #undef __FUNCT__
552552f7358SJed Brown #define __FUNCT__ "DMCreate_Plex"
553552f7358SJed Brown PetscErrorCode DMCreate_Plex(DM dm)
554552f7358SJed Brown {
555552f7358SJed Brown   DM_Plex        *mesh;
556770b213bSMatthew G Knepley   PetscInt       unit, d;
557552f7358SJed Brown   PetscErrorCode ierr;
558552f7358SJed Brown 
559552f7358SJed Brown   PetscFunctionBegin;
560552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
561552f7358SJed Brown   ierr     = PetscNewLog(dm, DM_Plex, &mesh);CHKERRQ(ierr);
562552f7358SJed Brown   dm->data = mesh;
563552f7358SJed Brown 
564552f7358SJed Brown   mesh->refct             = 1;
565552f7358SJed Brown   mesh->dim               = 0;
566552f7358SJed Brown   ierr                    = PetscSectionCreate(((PetscObject) dm)->comm, &mesh->coneSection);CHKERRQ(ierr);
567552f7358SJed Brown   mesh->maxConeSize       = 0;
568552f7358SJed Brown   mesh->cones             = PETSC_NULL;
569552f7358SJed Brown   mesh->coneOrientations  = PETSC_NULL;
570552f7358SJed Brown   ierr                    = PetscSectionCreate(((PetscObject) dm)->comm, &mesh->supportSection);CHKERRQ(ierr);
571552f7358SJed Brown   mesh->maxSupportSize    = 0;
572552f7358SJed Brown   mesh->supports          = PETSC_NULL;
573552f7358SJed Brown   mesh->refinementUniform = PETSC_TRUE;
574552f7358SJed Brown   mesh->refinementLimit   = -1.0;
575552f7358SJed Brown 
576552f7358SJed Brown   mesh->facesTmp = PETSC_NULL;
577552f7358SJed Brown 
578552f7358SJed Brown   mesh->subpointMap = PETSC_NULL;
579552f7358SJed Brown 
580*8865f1eaSKarl Rupp   for (unit = 0; unit < NUM_PETSC_UNITS; ++unit) mesh->scale[unit] = 1.0;
581552f7358SJed Brown 
582552f7358SJed Brown   mesh->labels              = PETSC_NULL;
583552f7358SJed Brown   mesh->globalVertexNumbers = PETSC_NULL;
584552f7358SJed Brown   mesh->globalCellNumbers   = PETSC_NULL;
585*8865f1eaSKarl Rupp   for (d = 0; d < 8; ++d) mesh->hybridPointMax[d] = PETSC_DETERMINE;
586552f7358SJed Brown   mesh->vtkCellHeight     = 0;
587552f7358SJed Brown   mesh->preallocCenterDim = -1;
588552f7358SJed Brown 
589552f7358SJed Brown   mesh->integrateResidualFEM       = PETSC_NULL;
590552f7358SJed Brown   mesh->integrateJacobianActionFEM = PETSC_NULL;
591552f7358SJed Brown   mesh->integrateJacobianFEM       = PETSC_NULL;
592552f7358SJed Brown 
593552f7358SJed Brown   mesh->printSetValues = PETSC_FALSE;
594552f7358SJed Brown   mesh->printFEM       = 0;
595552f7358SJed Brown 
596552f7358SJed Brown   ierr = DMInitialize_Plex(dm);CHKERRQ(ierr);
597552f7358SJed Brown   PetscFunctionReturn(0);
598552f7358SJed Brown }
599552f7358SJed Brown EXTERN_C_END
600552f7358SJed Brown 
601552f7358SJed Brown #undef __FUNCT__
602552f7358SJed Brown #define __FUNCT__ "DMPlexCreate"
603552f7358SJed Brown /*@
604552f7358SJed Brown   DMPlexCreate - Creates a DMPlex object, which encapsulates an unstructured mesh, or CW complex, which can be expressed using a Hasse Diagram.
605552f7358SJed Brown 
606552f7358SJed Brown   Collective on MPI_Comm
607552f7358SJed Brown 
608552f7358SJed Brown   Input Parameter:
609552f7358SJed Brown . comm - The communicator for the DMPlex object
610552f7358SJed Brown 
611552f7358SJed Brown   Output Parameter:
612552f7358SJed Brown . mesh  - The DMPlex object
613552f7358SJed Brown 
614552f7358SJed Brown   Level: beginner
615552f7358SJed Brown 
616552f7358SJed Brown .keywords: DMPlex, create
617552f7358SJed Brown @*/
618552f7358SJed Brown PetscErrorCode DMPlexCreate(MPI_Comm comm, DM *mesh)
619552f7358SJed Brown {
620552f7358SJed Brown   PetscErrorCode ierr;
621552f7358SJed Brown 
622552f7358SJed Brown   PetscFunctionBegin;
623552f7358SJed Brown   PetscValidPointer(mesh,2);
624552f7358SJed Brown   ierr = DMCreate(comm, mesh);CHKERRQ(ierr);
625552f7358SJed Brown   ierr = DMSetType(*mesh, DMPLEX);CHKERRQ(ierr);
626552f7358SJed Brown   PetscFunctionReturn(0);
627552f7358SJed Brown }
628552f7358SJed Brown 
629552f7358SJed Brown #undef __FUNCT__
630552f7358SJed Brown #define __FUNCT__ "DMPlexClone"
631552f7358SJed Brown /*@
632552f7358SJed Brown   DMPlexClone - Creates a DMPlex object with the same mesh as the original.
633552f7358SJed Brown 
634552f7358SJed Brown   Collective on MPI_Comm
635552f7358SJed Brown 
636552f7358SJed Brown   Input Parameter:
637552f7358SJed Brown . dm - The original DMPlex object
638552f7358SJed Brown 
639552f7358SJed Brown   Output Parameter:
640552f7358SJed Brown . newdm  - The new DMPlex object
641552f7358SJed Brown 
642552f7358SJed Brown   Level: beginner
643552f7358SJed Brown 
644552f7358SJed Brown .keywords: DMPlex, create
645552f7358SJed Brown @*/
646552f7358SJed Brown PetscErrorCode DMPlexClone(DM dm, DM *newdm)
647552f7358SJed Brown {
648552f7358SJed Brown   DM_Plex        *mesh;
649552f7358SJed Brown   void           *ctx;
650552f7358SJed Brown   PetscErrorCode ierr;
651552f7358SJed Brown 
652552f7358SJed Brown   PetscFunctionBegin;
653552f7358SJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
654552f7358SJed Brown   PetscValidPointer(newdm,2);
655552f7358SJed Brown   ierr         = DMCreate(((PetscObject) dm)->comm, newdm);CHKERRQ(ierr);
656552f7358SJed Brown   ierr         = PetscSFDestroy(&(*newdm)->sf);CHKERRQ(ierr);
657552f7358SJed Brown   ierr         = PetscObjectReference((PetscObject) dm->sf);CHKERRQ(ierr);
658552f7358SJed Brown   (*newdm)->sf = dm->sf;
659552f7358SJed Brown   mesh         = (DM_Plex*) dm->data;
660552f7358SJed Brown   mesh->refct++;
661552f7358SJed Brown   (*newdm)->data = mesh;
662552f7358SJed Brown   ierr           = PetscObjectChangeTypeName((PetscObject) *newdm, DMPLEX);CHKERRQ(ierr);
663552f7358SJed Brown   ierr           = DMInitialize_Plex(*newdm);CHKERRQ(ierr);
664552f7358SJed Brown   ierr           = DMGetApplicationContext(dm, &ctx);CHKERRQ(ierr);
665552f7358SJed Brown   ierr           = DMSetApplicationContext(*newdm, ctx);CHKERRQ(ierr);
666552f7358SJed Brown   PetscFunctionReturn(0);
667552f7358SJed Brown }
668