xref: /petsc/src/dm/impls/plex/plex.c (revision cac3c07dbc4e95423e22cb699bb64807a71d0bfe)
1 #include <petsc/private/dmpleximpl.h> /*I      "petscdmplex.h"   I*/
2 #include <petsc/private/dmlabelimpl.h>
3 #include <petsc/private/isimpl.h>
4 #include <petsc/private/vecimpl.h>
5 #include <petsc/private/glvisvecimpl.h>
6 #include <petscsf.h>
7 #include <petscds.h>
8 #include <petscdraw.h>
9 #include <petscdmfield.h>
10 #include <petscdmplextransform.h>
11 
12 /* Logging support */
13 PetscLogEvent DMPLEX_Interpolate, DMPLEX_Partition, DMPLEX_Distribute, DMPLEX_DistributeCones, DMPLEX_DistributeLabels, DMPLEX_DistributeSF, DMPLEX_DistributeOverlap, DMPLEX_DistributeField, DMPLEX_DistributeData, DMPLEX_Migrate, DMPLEX_InterpolateSF, DMPLEX_GlobalToNaturalBegin, DMPLEX_GlobalToNaturalEnd, DMPLEX_NaturalToGlobalBegin, DMPLEX_NaturalToGlobalEnd, DMPLEX_Stratify, DMPLEX_Symmetrize, DMPLEX_Preallocate, DMPLEX_ResidualFEM, DMPLEX_JacobianFEM, DMPLEX_InterpolatorFEM, DMPLEX_InjectorFEM, DMPLEX_IntegralFEM, DMPLEX_CreateGmsh, DMPLEX_RebalanceSharedPoints, DMPLEX_PartSelf, DMPLEX_PartLabelInvert, DMPLEX_PartLabelCreateSF, DMPLEX_PartStratSF, DMPLEX_CreatePointSF, DMPLEX_LocatePoints, DMPLEX_TopologyView, DMPLEX_LabelsView, DMPLEX_CoordinatesView, DMPLEX_SectionView, DMPLEX_GlobalVectorView, DMPLEX_LocalVectorView, DMPLEX_TopologyLoad, DMPLEX_LabelsLoad, DMPLEX_CoordinatesLoad, DMPLEX_SectionLoad, DMPLEX_GlobalVectorLoad, DMPLEX_LocalVectorLoad;
14 PetscLogEvent DMPLEX_RebalBuildGraph, DMPLEX_RebalRewriteSF, DMPLEX_RebalGatherGraph, DMPLEX_RebalPartition, DMPLEX_RebalScatterPart, DMPLEX_Generate, DMPLEX_Transform, DMPLEX_GetLocalOffsets, DMPLEX_Uninterpolate;
15 
16 PetscBool  Plexcite       = PETSC_FALSE;
17 const char PlexCitation[] = "@article{LangeMitchellKnepleyGorman2015,\n"
18                             "title     = {Efficient mesh management in {Firedrake} using {PETSc-DMPlex}},\n"
19                             "author    = {Michael Lange and Lawrence Mitchell and Matthew G. Knepley and Gerard J. Gorman},\n"
20                             "journal   = {SIAM Journal on Scientific Computing},\n"
21                             "volume    = {38},\n"
22                             "number    = {5},\n"
23                             "pages     = {S143--S155},\n"
24                             "eprint    = {http://arxiv.org/abs/1506.07749},\n"
25                             "doi       = {10.1137/15M1026092},\n"
26                             "year      = {2016},\n"
27                             "petsc_uses={DMPlex},\n}\n";
28 
29 PETSC_EXTERN PetscErrorCode VecView_MPI(Vec, PetscViewer);
30 
31 /*@
32   DMPlexIsSimplex - Is the first cell in this mesh a simplex?
33 
34   Input Parameter:
35 . dm - The `DMPLEX` object
36 
37   Output Parameter:
38 . simplex - Flag checking for a simplex
39 
40   Level: intermediate
41 
42   Note:
43   This just gives the first range of cells found. If the mesh has several cell types, it will only give the first.
44   If the mesh has no cells, this returns `PETSC_FALSE`.
45 
46 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetSimplexOrBoxCells()`, `DMPlexGetCellType()`, `DMPlexGetHeightStratum()`, `DMPolytopeTypeGetNumVertices()`
47 @*/
48 PetscErrorCode DMPlexIsSimplex(DM dm, PetscBool *simplex)
49 {
50   DMPolytopeType ct;
51   PetscInt       cStart, cEnd;
52 
53   PetscFunctionBegin;
54   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
55   if (cEnd <= cStart) {
56     *simplex = PETSC_FALSE;
57     PetscFunctionReturn(PETSC_SUCCESS);
58   }
59   PetscCall(DMPlexGetCellType(dm, cStart, &ct));
60   *simplex = DMPolytopeTypeGetNumVertices(ct) == DMPolytopeTypeGetDim(ct) + 1 ? PETSC_TRUE : PETSC_FALSE;
61   PetscFunctionReturn(PETSC_SUCCESS);
62 }
63 
64 /*@
65   DMPlexGetSimplexOrBoxCells - Get the range of cells which are neither prisms nor ghost FV cells
66 
67   Input Parameters:
68 + dm     - The `DMPLEX` object
69 - height - The cell height in the Plex, 0 is the default
70 
71   Output Parameters:
72 + cStart - The first "normal" cell
73 - cEnd   - The upper bound on "normal" cells
74 
75   Level: developer
76 
77   Note:
78   This function requires that tensor cells are ordered last.
79 
80 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexConstructGhostCells()`, `DMPlexGetCellTypeStratum()`
81 @*/
82 PetscErrorCode DMPlexGetSimplexOrBoxCells(DM dm, PetscInt height, PetscInt *cStart, PetscInt *cEnd)
83 {
84   DMLabel         ctLabel;
85   IS              valueIS;
86   const PetscInt *ctypes;
87   PetscInt        Nct, cS = PETSC_MAX_INT, cE = 0;
88 
89   PetscFunctionBegin;
90   PetscCall(DMPlexGetCellTypeLabel(dm, &ctLabel));
91   PetscCall(DMLabelGetValueIS(ctLabel, &valueIS));
92   PetscCall(ISGetLocalSize(valueIS, &Nct));
93   PetscCall(ISGetIndices(valueIS, &ctypes));
94   if (!Nct) cS = cE = 0;
95   for (PetscInt t = 0; t < Nct; ++t) {
96     const DMPolytopeType ct = (DMPolytopeType)ctypes[t];
97     PetscInt             ctS, ctE, ht;
98 
99     if (ct == DM_POLYTOPE_UNKNOWN) {
100       // If any cells are not typed, just use all cells
101       PetscCall(DMPlexGetHeightStratum(dm, PetscMax(height, 0), cStart, cEnd));
102       break;
103     }
104     if (DMPolytopeTypeIsHybrid(ct) || ct == DM_POLYTOPE_FV_GHOST) continue;
105     PetscCall(DMLabelGetStratumBounds(ctLabel, ct, &ctS, &ctE));
106     if (ctS >= ctE) continue;
107     // Check that a point has the right height
108     PetscCall(DMPlexGetPointHeight(dm, ctS, &ht));
109     if (ht != height) continue;
110     cS = PetscMin(cS, ctS);
111     cE = PetscMax(cE, ctE);
112   }
113   PetscCall(ISDestroy(&valueIS));
114   // Reset label for fast lookup
115   PetscCall(DMLabelMakeAllInvalid_Internal(ctLabel));
116   if (cStart) *cStart = cS;
117   if (cEnd) *cEnd = cE;
118   PetscFunctionReturn(PETSC_SUCCESS);
119 }
120 
121 PetscErrorCode DMPlexGetFieldTypes_Internal(DM dm, PetscSection section, PetscInt field, PetscInt *types, PetscInt **ssStart, PetscInt **ssEnd, PetscViewerVTKFieldType **sft)
122 {
123   PetscInt                 cdim, pStart, pEnd, vStart, vEnd, cStart, cEnd, c, depth, cellHeight, t;
124   PetscInt                *sStart, *sEnd;
125   PetscViewerVTKFieldType *ft;
126   PetscInt                 vcdof[DM_NUM_POLYTOPES + 1], globalvcdof[DM_NUM_POLYTOPES + 1];
127   DMLabel                  depthLabel, ctLabel;
128 
129   PetscFunctionBegin;
130 
131   /* the vcdof and globalvcdof are sized to allow every polytope type and simple vertex at DM_NUM_POLYTOPES */
132   PetscCall(PetscArrayzero(vcdof, DM_NUM_POLYTOPES + 1));
133   PetscCall(DMGetCoordinateDim(dm, &cdim));
134   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
135   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
136   if (field >= 0) {
137     if ((vStart >= pStart) && (vStart < pEnd)) PetscCall(PetscSectionGetFieldDof(section, vStart, field, &vcdof[DM_NUM_POLYTOPES]));
138   } else {
139     if ((vStart >= pStart) && (vStart < pEnd)) PetscCall(PetscSectionGetDof(section, vStart, &vcdof[DM_NUM_POLYTOPES]));
140   }
141 
142   PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
143   PetscCall(DMPlexGetDepth(dm, &depth));
144   PetscCall(DMPlexGetDepthLabel(dm, &depthLabel));
145   PetscCall(DMPlexGetCellTypeLabel(dm, &ctLabel));
146   for (c = 0; c < DM_NUM_POLYTOPES; ++c) {
147     const DMPolytopeType ict = (DMPolytopeType)c;
148     PetscInt             dep;
149 
150     if (ict == DM_POLYTOPE_FV_GHOST) continue;
151     PetscCall(DMLabelGetStratumBounds(ctLabel, ict, &cStart, &cEnd));
152     if (pStart >= 0) {
153       PetscCall(DMLabelGetValue(depthLabel, cStart, &dep));
154       if (dep != depth - cellHeight) continue;
155     }
156     if (field >= 0) {
157       if ((cStart >= pStart) && (cStart < pEnd)) PetscCall(PetscSectionGetFieldDof(section, cStart, field, &vcdof[c]));
158     } else {
159       if ((cStart >= pStart) && (cStart < pEnd)) PetscCall(PetscSectionGetDof(section, cStart, &vcdof[c]));
160     }
161     PetscCall(MPIU_Allreduce(vcdof, globalvcdof, 2, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
162   }
163 
164   PetscCall(MPIU_Allreduce(vcdof, globalvcdof, DM_NUM_POLYTOPES + 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
165   *types = 0;
166 
167   for (c = 0; c < DM_NUM_POLYTOPES + 1; ++c) {
168     if (globalvcdof[c]) ++(*types);
169   }
170 
171   PetscCall(PetscMalloc3(*types, &sStart, *types, &sEnd, *types, &ft));
172   t = 0;
173   if (globalvcdof[DM_NUM_POLYTOPES]) {
174     sStart[t] = vStart;
175     sEnd[t]   = vEnd;
176     ft[t]     = (globalvcdof[t] == cdim) ? PETSC_VTK_POINT_VECTOR_FIELD : PETSC_VTK_POINT_FIELD;
177     ++t;
178   }
179 
180   for (c = 0; c < DM_NUM_POLYTOPES; ++c) {
181     if (globalvcdof[c]) {
182       const DMPolytopeType ict = (DMPolytopeType)c;
183 
184       PetscCall(DMLabelGetStratumBounds(ctLabel, ict, &cStart, &cEnd));
185       sStart[t] = cStart;
186       sEnd[t]   = cEnd;
187       ft[t]     = (globalvcdof[c] == cdim) ? PETSC_VTK_CELL_VECTOR_FIELD : PETSC_VTK_CELL_FIELD;
188       ++t;
189     }
190   }
191 
192   if (!(*types)) {
193     if (field >= 0) {
194       const char *fieldname;
195 
196       PetscCall(PetscSectionGetFieldName(section, field, &fieldname));
197       PetscCall(PetscInfo((PetscObject)dm, "Could not classify VTK output type of section field %" PetscInt_FMT " \"%s\"\n", field, fieldname));
198     } else {
199       PetscCall(PetscInfo((PetscObject)dm, "Could not classify VTK output type of section\n"));
200     }
201   }
202 
203   *ssStart = sStart;
204   *ssEnd   = sEnd;
205   *sft     = ft;
206   PetscFunctionReturn(PETSC_SUCCESS);
207 }
208 
209 PetscErrorCode DMPlexRestoreFieldTypes_Internal(DM dm, PetscSection section, PetscInt field, PetscInt *types, PetscInt **sStart, PetscInt **sEnd, PetscViewerVTKFieldType **ft)
210 {
211   PetscFunctionBegin;
212   PetscCall(PetscFree3(*sStart, *sEnd, *ft));
213   PetscFunctionReturn(PETSC_SUCCESS);
214 }
215 
216 PetscErrorCode DMPlexGetFieldType_Internal(DM dm, PetscSection section, PetscInt field, PetscInt *sStart, PetscInt *sEnd, PetscViewerVTKFieldType *ft)
217 {
218   PetscInt cdim, pStart, pEnd, vStart, vEnd, cStart, cEnd;
219   PetscInt vcdof[2] = {0, 0}, globalvcdof[2];
220 
221   PetscFunctionBegin;
222   *ft = PETSC_VTK_INVALID;
223   PetscCall(DMGetCoordinateDim(dm, &cdim));
224   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
225   PetscCall(DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd));
226   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
227   if (field >= 0) {
228     if ((vStart >= pStart) && (vStart < pEnd)) PetscCall(PetscSectionGetFieldDof(section, vStart, field, &vcdof[0]));
229     if ((cStart >= pStart) && (cStart < pEnd)) PetscCall(PetscSectionGetFieldDof(section, cStart, field, &vcdof[1]));
230   } else {
231     if ((vStart >= pStart) && (vStart < pEnd)) PetscCall(PetscSectionGetDof(section, vStart, &vcdof[0]));
232     if ((cStart >= pStart) && (cStart < pEnd)) PetscCall(PetscSectionGetDof(section, cStart, &vcdof[1]));
233   }
234   PetscCall(MPIU_Allreduce(vcdof, globalvcdof, 2, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
235   if (globalvcdof[0]) {
236     *sStart = vStart;
237     *sEnd   = vEnd;
238     if (globalvcdof[0] == cdim) *ft = PETSC_VTK_POINT_VECTOR_FIELD;
239     else *ft = PETSC_VTK_POINT_FIELD;
240   } else if (globalvcdof[1]) {
241     *sStart = cStart;
242     *sEnd   = cEnd;
243     if (globalvcdof[1] == cdim) *ft = PETSC_VTK_CELL_VECTOR_FIELD;
244     else *ft = PETSC_VTK_CELL_FIELD;
245   } else {
246     if (field >= 0) {
247       const char *fieldname;
248 
249       PetscCall(PetscSectionGetFieldName(section, field, &fieldname));
250       PetscCall(PetscInfo((PetscObject)dm, "Could not classify VTK output type of section field %" PetscInt_FMT " \"%s\"\n", field, fieldname));
251     } else {
252       PetscCall(PetscInfo((PetscObject)dm, "Could not classify VTK output type of section\n"));
253     }
254   }
255   PetscFunctionReturn(PETSC_SUCCESS);
256 }
257 
258 /*@
259   DMPlexVecView1D - Plot many 1D solutions on the same line graph
260 
261   Collective
262 
263   Input Parameters:
264 + dm     - The `DMPLEX` object
265 . n      - The number of vectors
266 . u      - The array of local vectors
267 - viewer - The `PetscViewer`
268 
269   Level: advanced
270 
271 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `VecViewFromOptions()`, `VecView()`
272 @*/
273 PetscErrorCode DMPlexVecView1D(DM dm, PetscInt n, Vec u[], PetscViewer viewer)
274 {
275   PetscDS            ds;
276   PetscDraw          draw = NULL;
277   PetscDrawLG        lg;
278   Vec                coordinates;
279   const PetscScalar *coords, **sol;
280   PetscReal         *vals;
281   PetscInt          *Nc;
282   PetscInt           Nf, f, c, Nl, l, i, vStart, vEnd, v;
283   char             **names;
284 
285   PetscFunctionBegin;
286   PetscCall(DMGetDS(dm, &ds));
287   PetscCall(PetscDSGetNumFields(ds, &Nf));
288   PetscCall(PetscDSGetTotalComponents(ds, &Nl));
289   PetscCall(PetscDSGetComponents(ds, &Nc));
290 
291   PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
292   if (!draw) PetscFunctionReturn(PETSC_SUCCESS);
293   PetscCall(PetscDrawLGCreate(draw, n * Nl, &lg));
294 
295   PetscCall(PetscMalloc3(n, &sol, n * Nl, &names, n * Nl, &vals));
296   for (i = 0, l = 0; i < n; ++i) {
297     const char *vname;
298 
299     PetscCall(PetscObjectGetName((PetscObject)u[i], &vname));
300     for (f = 0; f < Nf; ++f) {
301       PetscObject disc;
302       const char *fname;
303       char        tmpname[PETSC_MAX_PATH_LEN];
304 
305       PetscCall(PetscDSGetDiscretization(ds, f, &disc));
306       /* TODO Create names for components */
307       for (c = 0; c < Nc[f]; ++c, ++l) {
308         PetscCall(PetscObjectGetName(disc, &fname));
309         PetscCall(PetscStrncpy(tmpname, vname, sizeof(tmpname)));
310         PetscCall(PetscStrlcat(tmpname, ":", sizeof(tmpname)));
311         PetscCall(PetscStrlcat(tmpname, fname, sizeof(tmpname)));
312         PetscCall(PetscStrallocpy(tmpname, &names[l]));
313       }
314     }
315   }
316   PetscCall(PetscDrawLGSetLegend(lg, (const char *const *)names));
317   /* Just add P_1 support for now */
318   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
319   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
320   PetscCall(VecGetArrayRead(coordinates, &coords));
321   for (i = 0; i < n; ++i) PetscCall(VecGetArrayRead(u[i], &sol[i]));
322   for (v = vStart; v < vEnd; ++v) {
323     PetscScalar *x, *svals;
324 
325     PetscCall(DMPlexPointLocalRead(dm, v, coords, &x));
326     for (i = 0; i < n; ++i) {
327       PetscCall(DMPlexPointLocalRead(dm, v, sol[i], &svals));
328       for (l = 0; l < Nl; ++l) vals[i * Nl + l] = PetscRealPart(svals[l]);
329     }
330     PetscCall(PetscDrawLGAddCommonPoint(lg, PetscRealPart(x[0]), vals));
331   }
332   PetscCall(VecRestoreArrayRead(coordinates, &coords));
333   for (i = 0; i < n; ++i) PetscCall(VecRestoreArrayRead(u[i], &sol[i]));
334   for (l = 0; l < n * Nl; ++l) PetscCall(PetscFree(names[l]));
335   PetscCall(PetscFree3(sol, names, vals));
336 
337   PetscCall(PetscDrawLGDraw(lg));
338   PetscCall(PetscDrawLGDestroy(&lg));
339   PetscFunctionReturn(PETSC_SUCCESS);
340 }
341 
342 static PetscErrorCode VecView_Plex_Local_Draw_1D(Vec u, PetscViewer viewer)
343 {
344   DM dm;
345 
346   PetscFunctionBegin;
347   PetscCall(VecGetDM(u, &dm));
348   PetscCall(DMPlexVecView1D(dm, 1, &u, viewer));
349   PetscFunctionReturn(PETSC_SUCCESS);
350 }
351 
352 static PetscErrorCode VecView_Plex_Local_Draw_2D(Vec v, PetscViewer viewer)
353 {
354   DM                 dm;
355   PetscSection       s;
356   PetscDraw          draw, popup;
357   DM                 cdm;
358   PetscSection       coordSection;
359   Vec                coordinates;
360   const PetscScalar *array;
361   PetscReal          lbound[3], ubound[3];
362   PetscReal          vbound[2], time;
363   PetscBool          flg;
364   PetscInt           dim, Nf, f, Nc, comp, vStart, vEnd, cStart, cEnd, c, N, level, step, w = 0;
365   const char        *name;
366   char               title[PETSC_MAX_PATH_LEN];
367 
368   PetscFunctionBegin;
369   PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
370   PetscCall(VecGetDM(v, &dm));
371   PetscCall(DMGetCoordinateDim(dm, &dim));
372   PetscCall(DMGetLocalSection(dm, &s));
373   PetscCall(PetscSectionGetNumFields(s, &Nf));
374   PetscCall(DMGetCoarsenLevel(dm, &level));
375   PetscCall(DMGetCoordinateDM(dm, &cdm));
376   PetscCall(DMGetLocalSection(cdm, &coordSection));
377   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
378   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
379   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
380 
381   PetscCall(PetscObjectGetName((PetscObject)v, &name));
382   PetscCall(DMGetOutputSequenceNumber(dm, &step, &time));
383 
384   PetscCall(VecGetLocalSize(coordinates, &N));
385   PetscCall(DMGetBoundingBox(dm, lbound, ubound));
386   PetscCall(PetscDrawClear(draw));
387 
388   /* Could implement something like DMDASelectFields() */
389   for (f = 0; f < Nf; ++f) {
390     DM          fdm = dm;
391     Vec         fv  = v;
392     IS          fis;
393     char        prefix[PETSC_MAX_PATH_LEN];
394     const char *fname;
395 
396     PetscCall(PetscSectionGetFieldComponents(s, f, &Nc));
397     PetscCall(PetscSectionGetFieldName(s, f, &fname));
398 
399     if (v->hdr.prefix) PetscCall(PetscStrncpy(prefix, v->hdr.prefix, sizeof(prefix)));
400     else prefix[0] = '\0';
401     if (Nf > 1) {
402       PetscCall(DMCreateSubDM(dm, 1, &f, &fis, &fdm));
403       PetscCall(VecGetSubVector(v, fis, &fv));
404       PetscCall(PetscStrlcat(prefix, fname, sizeof(prefix)));
405       PetscCall(PetscStrlcat(prefix, "_", sizeof(prefix)));
406     }
407     for (comp = 0; comp < Nc; ++comp, ++w) {
408       PetscInt nmax = 2;
409 
410       PetscCall(PetscViewerDrawGetDraw(viewer, w, &draw));
411       if (Nc > 1) PetscCall(PetscSNPrintf(title, sizeof(title), "%s:%s_%" PetscInt_FMT " Step: %" PetscInt_FMT " Time: %.4g", name, fname, comp, step, (double)time));
412       else PetscCall(PetscSNPrintf(title, sizeof(title), "%s:%s Step: %" PetscInt_FMT " Time: %.4g", name, fname, step, (double)time));
413       PetscCall(PetscDrawSetTitle(draw, title));
414 
415       /* TODO Get max and min only for this component */
416       PetscCall(PetscOptionsGetRealArray(NULL, prefix, "-vec_view_bounds", vbound, &nmax, &flg));
417       if (!flg) {
418         PetscCall(VecMin(fv, NULL, &vbound[0]));
419         PetscCall(VecMax(fv, NULL, &vbound[1]));
420         if (vbound[1] <= vbound[0]) vbound[1] = vbound[0] + 1.0;
421       }
422 
423       PetscCall(PetscDrawGetPopup(draw, &popup));
424       PetscCall(PetscDrawScalePopup(popup, vbound[0], vbound[1]));
425       PetscCall(PetscDrawSetCoordinates(draw, lbound[0], lbound[1], ubound[0], ubound[1]));
426       PetscCall(VecGetArrayRead(fv, &array));
427       for (c = cStart; c < cEnd; ++c) {
428         PetscScalar       *coords = NULL, *a = NULL;
429         const PetscScalar *coords_arr;
430         PetscBool          isDG;
431         PetscInt           numCoords, color[4] = {-1, -1, -1, -1};
432 
433         PetscCall(DMPlexPointLocalRead(fdm, c, array, &a));
434         if (a) {
435           color[0] = PetscDrawRealToColor(PetscRealPart(a[comp]), vbound[0], vbound[1]);
436           color[1] = color[2] = color[3] = color[0];
437         } else {
438           PetscScalar *vals = NULL;
439           PetscInt     numVals, va;
440 
441           PetscCall(DMPlexVecGetClosure(fdm, NULL, fv, c, &numVals, &vals));
442           PetscCheck(numVals % Nc == 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of components %" PetscInt_FMT " does not divide the number of values in the closure %" PetscInt_FMT, Nc, numVals);
443           switch (numVals / Nc) {
444           case 3: /* P1 Triangle */
445           case 4: /* P1 Quadrangle */
446             for (va = 0; va < numVals / Nc; ++va) color[va] = PetscDrawRealToColor(PetscRealPart(vals[va * Nc + comp]), vbound[0], vbound[1]);
447             break;
448           case 6: /* P2 Triangle */
449           case 8: /* P2 Quadrangle */
450             for (va = 0; va < numVals / (Nc * 2); ++va) color[va] = PetscDrawRealToColor(PetscRealPart(vals[va * Nc + comp + numVals / (Nc * 2)]), vbound[0], vbound[1]);
451             break;
452           default:
453             SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of values for cell closure %" PetscInt_FMT " cannot be handled", numVals / Nc);
454           }
455           PetscCall(DMPlexVecRestoreClosure(fdm, NULL, fv, c, &numVals, &vals));
456         }
457         PetscCall(DMPlexGetCellCoordinates(dm, c, &isDG, &numCoords, &coords_arr, &coords));
458         switch (numCoords) {
459         case 6:
460         case 12: /* Localized triangle */
461           PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), color[0], color[1], color[2]));
462           break;
463         case 8:
464         case 16: /* Localized quadrilateral */
465           PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), color[0], color[1], color[2]));
466           PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[4]), PetscRealPart(coords[5]), PetscRealPart(coords[6]), PetscRealPart(coords[7]), PetscRealPart(coords[0]), PetscRealPart(coords[1]), color[2], color[3], color[0]));
467           break;
468         default:
469           SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot draw cells with %" PetscInt_FMT " coordinates", numCoords);
470         }
471         PetscCall(DMPlexRestoreCellCoordinates(dm, c, &isDG, &numCoords, &coords_arr, &coords));
472       }
473       PetscCall(VecRestoreArrayRead(fv, &array));
474       PetscCall(PetscDrawFlush(draw));
475       PetscCall(PetscDrawPause(draw));
476       PetscCall(PetscDrawSave(draw));
477     }
478     if (Nf > 1) {
479       PetscCall(VecRestoreSubVector(v, fis, &fv));
480       PetscCall(ISDestroy(&fis));
481       PetscCall(DMDestroy(&fdm));
482     }
483   }
484   PetscFunctionReturn(PETSC_SUCCESS);
485 }
486 
487 static PetscErrorCode VecView_Plex_Local_Draw(Vec v, PetscViewer viewer)
488 {
489   DM        dm;
490   PetscDraw draw;
491   PetscInt  dim;
492   PetscBool isnull;
493 
494   PetscFunctionBegin;
495   PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
496   PetscCall(PetscDrawIsNull(draw, &isnull));
497   if (isnull) PetscFunctionReturn(PETSC_SUCCESS);
498 
499   PetscCall(VecGetDM(v, &dm));
500   PetscCall(DMGetCoordinateDim(dm, &dim));
501   switch (dim) {
502   case 1:
503     PetscCall(VecView_Plex_Local_Draw_1D(v, viewer));
504     break;
505   case 2:
506     PetscCall(VecView_Plex_Local_Draw_2D(v, viewer));
507     break;
508   default:
509     SETERRQ(PetscObjectComm((PetscObject)v), PETSC_ERR_SUP, "Cannot draw meshes of dimension %" PetscInt_FMT ". Try PETSCVIEWERGLVIS", dim);
510   }
511   PetscFunctionReturn(PETSC_SUCCESS);
512 }
513 
514 static PetscErrorCode VecView_Plex_Local_VTK(Vec v, PetscViewer viewer)
515 {
516   DM                      dm;
517   Vec                     locv;
518   const char             *name;
519   PetscSection            section;
520   PetscInt                pStart, pEnd;
521   PetscInt                numFields;
522   PetscViewerVTKFieldType ft;
523 
524   PetscFunctionBegin;
525   PetscCall(VecGetDM(v, &dm));
526   PetscCall(DMCreateLocalVector(dm, &locv)); /* VTK viewer requires exclusive ownership of the vector */
527   PetscCall(PetscObjectGetName((PetscObject)v, &name));
528   PetscCall(PetscObjectSetName((PetscObject)locv, name));
529   PetscCall(VecCopy(v, locv));
530   PetscCall(DMGetLocalSection(dm, &section));
531   PetscCall(PetscSectionGetNumFields(section, &numFields));
532   if (!numFields) {
533     PetscCall(DMPlexGetFieldType_Internal(dm, section, PETSC_DETERMINE, &pStart, &pEnd, &ft));
534     PetscCall(PetscViewerVTKAddField(viewer, (PetscObject)dm, DMPlexVTKWriteAll, PETSC_DEFAULT, ft, PETSC_TRUE, (PetscObject)locv));
535   } else {
536     PetscInt f;
537 
538     for (f = 0; f < numFields; f++) {
539       PetscCall(DMPlexGetFieldType_Internal(dm, section, f, &pStart, &pEnd, &ft));
540       if (ft == PETSC_VTK_INVALID) continue;
541       PetscCall(PetscObjectReference((PetscObject)locv));
542       PetscCall(PetscViewerVTKAddField(viewer, (PetscObject)dm, DMPlexVTKWriteAll, f, ft, PETSC_TRUE, (PetscObject)locv));
543     }
544     PetscCall(VecDestroy(&locv));
545   }
546   PetscFunctionReturn(PETSC_SUCCESS);
547 }
548 
549 PetscErrorCode VecView_Plex_Local(Vec v, PetscViewer viewer)
550 {
551   DM        dm;
552   PetscBool isvtk, ishdf5, isdraw, isglvis, iscgns;
553 
554   PetscFunctionBegin;
555   PetscCall(VecGetDM(v, &dm));
556   PetscCheck(dm, PetscObjectComm((PetscObject)v), PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
557   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
558   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
559   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw));
560   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERGLVIS, &isglvis));
561   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERCGNS, &iscgns));
562   if (isvtk || ishdf5 || isdraw || isglvis || iscgns) {
563     PetscInt    i, numFields;
564     PetscObject fe;
565     PetscBool   fem  = PETSC_FALSE;
566     Vec         locv = v;
567     const char *name;
568     PetscInt    step;
569     PetscReal   time;
570 
571     PetscCall(DMGetNumFields(dm, &numFields));
572     for (i = 0; i < numFields; i++) {
573       PetscCall(DMGetField(dm, i, NULL, &fe));
574       if (fe->classid == PETSCFE_CLASSID) {
575         fem = PETSC_TRUE;
576         break;
577       }
578     }
579     if (fem) {
580       PetscObject isZero;
581 
582       PetscCall(DMGetLocalVector(dm, &locv));
583       PetscCall(PetscObjectGetName((PetscObject)v, &name));
584       PetscCall(PetscObjectSetName((PetscObject)locv, name));
585       PetscCall(PetscObjectQuery((PetscObject)v, "__Vec_bc_zero__", &isZero));
586       PetscCall(PetscObjectCompose((PetscObject)locv, "__Vec_bc_zero__", isZero));
587       PetscCall(VecCopy(v, locv));
588       PetscCall(DMGetOutputSequenceNumber(dm, NULL, &time));
589       PetscCall(DMPlexInsertBoundaryValues(dm, PETSC_TRUE, locv, time, NULL, NULL, NULL));
590     }
591     if (isvtk) {
592       PetscCall(VecView_Plex_Local_VTK(locv, viewer));
593     } else if (ishdf5) {
594 #if defined(PETSC_HAVE_HDF5)
595       PetscCall(VecView_Plex_Local_HDF5_Internal(locv, viewer));
596 #else
597       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
598 #endif
599     } else if (isdraw) {
600       PetscCall(VecView_Plex_Local_Draw(locv, viewer));
601     } else if (isglvis) {
602       PetscCall(DMGetOutputSequenceNumber(dm, &step, NULL));
603       PetscCall(PetscViewerGLVisSetSnapId(viewer, step));
604       PetscCall(VecView_GLVis(locv, viewer));
605     } else if (iscgns) {
606 #if defined(PETSC_HAVE_CGNS)
607       PetscCall(VecView_Plex_Local_CGNS(locv, viewer));
608 #else
609       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "CGNS not supported in this build.\nPlease reconfigure using --download-cgns");
610 #endif
611     }
612     if (fem) {
613       PetscCall(PetscObjectCompose((PetscObject)locv, "__Vec_bc_zero__", NULL));
614       PetscCall(DMRestoreLocalVector(dm, &locv));
615     }
616   } else {
617     PetscBool isseq;
618 
619     PetscCall(PetscObjectTypeCompare((PetscObject)v, VECSEQ, &isseq));
620     if (isseq) PetscCall(VecView_Seq(v, viewer));
621     else PetscCall(VecView_MPI(v, viewer));
622   }
623   PetscFunctionReturn(PETSC_SUCCESS);
624 }
625 
626 PetscErrorCode VecView_Plex(Vec v, PetscViewer viewer)
627 {
628   DM        dm;
629   PetscBool isvtk, ishdf5, isdraw, isglvis, isexodusii, iscgns;
630 
631   PetscFunctionBegin;
632   PetscCall(VecGetDM(v, &dm));
633   PetscCheck(dm, PetscObjectComm((PetscObject)v), PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
634   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
635   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
636   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw));
637   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERGLVIS, &isglvis));
638   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERCGNS, &iscgns));
639   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWEREXODUSII, &isexodusii));
640   if (isvtk || isdraw || isglvis || iscgns) {
641     Vec         locv;
642     PetscObject isZero;
643     const char *name;
644 
645     PetscCall(DMGetLocalVector(dm, &locv));
646     PetscCall(PetscObjectGetName((PetscObject)v, &name));
647     PetscCall(PetscObjectSetName((PetscObject)locv, name));
648     PetscCall(DMGlobalToLocalBegin(dm, v, INSERT_VALUES, locv));
649     PetscCall(DMGlobalToLocalEnd(dm, v, INSERT_VALUES, locv));
650     PetscCall(PetscObjectQuery((PetscObject)v, "__Vec_bc_zero__", &isZero));
651     PetscCall(PetscObjectCompose((PetscObject)locv, "__Vec_bc_zero__", isZero));
652     PetscCall(VecView_Plex_Local(locv, viewer));
653     PetscCall(PetscObjectCompose((PetscObject)locv, "__Vec_bc_zero__", NULL));
654     PetscCall(DMRestoreLocalVector(dm, &locv));
655   } else if (ishdf5) {
656 #if defined(PETSC_HAVE_HDF5)
657     PetscCall(VecView_Plex_HDF5_Internal(v, viewer));
658 #else
659     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
660 #endif
661   } else if (isexodusii) {
662 #if defined(PETSC_HAVE_EXODUSII)
663     PetscCall(VecView_PlexExodusII_Internal(v, viewer));
664 #else
665     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "ExodusII not supported in this build.\nPlease reconfigure using --download-exodusii");
666 #endif
667   } else {
668     PetscBool isseq;
669 
670     PetscCall(PetscObjectTypeCompare((PetscObject)v, VECSEQ, &isseq));
671     if (isseq) PetscCall(VecView_Seq(v, viewer));
672     else PetscCall(VecView_MPI(v, viewer));
673   }
674   PetscFunctionReturn(PETSC_SUCCESS);
675 }
676 
677 PetscErrorCode VecView_Plex_Native(Vec originalv, PetscViewer viewer)
678 {
679   DM                dm;
680   MPI_Comm          comm;
681   PetscViewerFormat format;
682   Vec               v;
683   PetscBool         isvtk, ishdf5;
684 
685   PetscFunctionBegin;
686   PetscCall(VecGetDM(originalv, &dm));
687   PetscCall(PetscObjectGetComm((PetscObject)originalv, &comm));
688   PetscCheck(dm, comm, PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
689   PetscCall(PetscViewerGetFormat(viewer, &format));
690   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
691   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
692   if (format == PETSC_VIEWER_NATIVE) {
693     /* Natural ordering is the common case for DMDA, NATIVE means plain vector, for PLEX is the opposite */
694     /* this need a better fix */
695     if (dm->useNatural) {
696       if (dm->sfNatural) {
697         const char *vecname;
698         PetscInt    n, nroots;
699 
700         PetscCall(VecGetLocalSize(originalv, &n));
701         PetscCall(PetscSFGetGraph(dm->sfNatural, &nroots, NULL, NULL, NULL));
702         if (n == nroots) {
703           PetscCall(DMPlexCreateNaturalVector(dm, &v));
704           PetscCall(DMPlexGlobalToNaturalBegin(dm, originalv, v));
705           PetscCall(DMPlexGlobalToNaturalEnd(dm, originalv, v));
706           PetscCall(PetscObjectGetName((PetscObject)originalv, &vecname));
707           PetscCall(PetscObjectSetName((PetscObject)v, vecname));
708         } else SETERRQ(comm, PETSC_ERR_ARG_WRONG, "DM global to natural SF only handles global vectors");
709       } else SETERRQ(comm, PETSC_ERR_ARG_WRONGSTATE, "DM global to natural SF was not created");
710     } else v = originalv;
711   } else v = originalv;
712 
713   if (ishdf5) {
714 #if defined(PETSC_HAVE_HDF5)
715     PetscCall(VecView_Plex_HDF5_Native_Internal(v, viewer));
716 #else
717     SETERRQ(comm, PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
718 #endif
719   } else if (isvtk) {
720     SETERRQ(comm, PETSC_ERR_SUP, "VTK format does not support viewing in natural order. Please switch to HDF5.");
721   } else {
722     PetscBool isseq;
723 
724     PetscCall(PetscObjectTypeCompare((PetscObject)v, VECSEQ, &isseq));
725     if (isseq) PetscCall(VecView_Seq(v, viewer));
726     else PetscCall(VecView_MPI(v, viewer));
727   }
728   if (v != originalv) PetscCall(VecDestroy(&v));
729   PetscFunctionReturn(PETSC_SUCCESS);
730 }
731 
732 PetscErrorCode VecLoad_Plex_Local(Vec v, PetscViewer viewer)
733 {
734   DM        dm;
735   PetscBool ishdf5;
736 
737   PetscFunctionBegin;
738   PetscCall(VecGetDM(v, &dm));
739   PetscCheck(dm, PetscObjectComm((PetscObject)v), PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
740   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
741   if (ishdf5) {
742     DM          dmBC;
743     Vec         gv;
744     const char *name;
745 
746     PetscCall(DMGetOutputDM(dm, &dmBC));
747     PetscCall(DMGetGlobalVector(dmBC, &gv));
748     PetscCall(PetscObjectGetName((PetscObject)v, &name));
749     PetscCall(PetscObjectSetName((PetscObject)gv, name));
750     PetscCall(VecLoad_Default(gv, viewer));
751     PetscCall(DMGlobalToLocalBegin(dmBC, gv, INSERT_VALUES, v));
752     PetscCall(DMGlobalToLocalEnd(dmBC, gv, INSERT_VALUES, v));
753     PetscCall(DMRestoreGlobalVector(dmBC, &gv));
754   } else PetscCall(VecLoad_Default(v, viewer));
755   PetscFunctionReturn(PETSC_SUCCESS);
756 }
757 
758 PetscErrorCode VecLoad_Plex(Vec v, PetscViewer viewer)
759 {
760   DM        dm;
761   PetscBool ishdf5, isexodusii;
762 
763   PetscFunctionBegin;
764   PetscCall(VecGetDM(v, &dm));
765   PetscCheck(dm, PetscObjectComm((PetscObject)v), PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
766   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
767   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWEREXODUSII, &isexodusii));
768   if (ishdf5) {
769 #if defined(PETSC_HAVE_HDF5)
770     PetscCall(VecLoad_Plex_HDF5_Internal(v, viewer));
771 #else
772     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
773 #endif
774   } else if (isexodusii) {
775 #if defined(PETSC_HAVE_EXODUSII)
776     PetscCall(VecLoad_PlexExodusII_Internal(v, viewer));
777 #else
778     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "ExodusII not supported in this build.\nPlease reconfigure using --download-exodusii");
779 #endif
780   } else PetscCall(VecLoad_Default(v, viewer));
781   PetscFunctionReturn(PETSC_SUCCESS);
782 }
783 
784 PetscErrorCode VecLoad_Plex_Native(Vec originalv, PetscViewer viewer)
785 {
786   DM                dm;
787   PetscViewerFormat format;
788   PetscBool         ishdf5;
789 
790   PetscFunctionBegin;
791   PetscCall(VecGetDM(originalv, &dm));
792   PetscCheck(dm, PetscObjectComm((PetscObject)originalv), PETSC_ERR_ARG_WRONG, "Vector not generated from a DM");
793   PetscCall(PetscViewerGetFormat(viewer, &format));
794   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
795   if (format == PETSC_VIEWER_NATIVE) {
796     if (dm->useNatural) {
797       if (dm->sfNatural) {
798         if (ishdf5) {
799 #if defined(PETSC_HAVE_HDF5)
800           Vec         v;
801           const char *vecname;
802 
803           PetscCall(DMPlexCreateNaturalVector(dm, &v));
804           PetscCall(PetscObjectGetName((PetscObject)originalv, &vecname));
805           PetscCall(PetscObjectSetName((PetscObject)v, vecname));
806           PetscCall(VecLoad_Plex_HDF5_Native_Internal(v, viewer));
807           PetscCall(DMPlexNaturalToGlobalBegin(dm, v, originalv));
808           PetscCall(DMPlexNaturalToGlobalEnd(dm, v, originalv));
809           PetscCall(VecDestroy(&v));
810 #else
811           SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
812 #endif
813         } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Reading in natural order is not supported for anything but HDF5.");
814       }
815     } else PetscCall(VecLoad_Default(originalv, viewer));
816   }
817   PetscFunctionReturn(PETSC_SUCCESS);
818 }
819 
820 PETSC_UNUSED static PetscErrorCode DMPlexView_Ascii_Geometry(DM dm, PetscViewer viewer)
821 {
822   PetscSection       coordSection;
823   Vec                coordinates;
824   DMLabel            depthLabel, celltypeLabel;
825   const char        *name[4];
826   const PetscScalar *a;
827   PetscInt           dim, pStart, pEnd, cStart, cEnd, c;
828 
829   PetscFunctionBegin;
830   PetscCall(DMGetDimension(dm, &dim));
831   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
832   PetscCall(DMGetCoordinateSection(dm, &coordSection));
833   PetscCall(DMPlexGetDepthLabel(dm, &depthLabel));
834   PetscCall(DMPlexGetCellTypeLabel(dm, &celltypeLabel));
835   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
836   PetscCall(PetscSectionGetChart(coordSection, &pStart, &pEnd));
837   PetscCall(VecGetArrayRead(coordinates, &a));
838   name[0]       = "vertex";
839   name[1]       = "edge";
840   name[dim - 1] = "face";
841   name[dim]     = "cell";
842   for (c = cStart; c < cEnd; ++c) {
843     PetscInt *closure = NULL;
844     PetscInt  closureSize, cl, ct;
845 
846     PetscCall(DMLabelGetValue(celltypeLabel, c, &ct));
847     PetscCall(PetscViewerASCIIPrintf(viewer, "Geometry for cell %" PetscInt_FMT " polytope type %s:\n", c, DMPolytopeTypes[ct]));
848     PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
849     PetscCall(PetscViewerASCIIPushTab(viewer));
850     for (cl = 0; cl < closureSize * 2; cl += 2) {
851       PetscInt point = closure[cl], depth, dof, off, d, p;
852 
853       if ((point < pStart) || (point >= pEnd)) continue;
854       PetscCall(PetscSectionGetDof(coordSection, point, &dof));
855       if (!dof) continue;
856       PetscCall(DMLabelGetValue(depthLabel, point, &depth));
857       PetscCall(PetscSectionGetOffset(coordSection, point, &off));
858       PetscCall(PetscViewerASCIIPrintf(viewer, "%s %" PetscInt_FMT " coords:", name[depth], point));
859       for (p = 0; p < dof / dim; ++p) {
860         PetscCall(PetscViewerASCIIPrintf(viewer, " ("));
861         for (d = 0; d < dim; ++d) {
862           if (d > 0) PetscCall(PetscViewerASCIIPrintf(viewer, ", "));
863           PetscCall(PetscViewerASCIIPrintf(viewer, "%g", (double)PetscRealPart(a[off + p * dim + d])));
864         }
865         PetscCall(PetscViewerASCIIPrintf(viewer, ")"));
866       }
867       PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
868     }
869     PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
870     PetscCall(PetscViewerASCIIPopTab(viewer));
871   }
872   PetscCall(VecRestoreArrayRead(coordinates, &a));
873   PetscFunctionReturn(PETSC_SUCCESS);
874 }
875 
876 typedef enum {
877   CS_CARTESIAN,
878   CS_POLAR,
879   CS_CYLINDRICAL,
880   CS_SPHERICAL
881 } CoordSystem;
882 const char *CoordSystems[] = {"cartesian", "polar", "cylindrical", "spherical", "CoordSystem", "CS_", NULL};
883 
884 static PetscErrorCode DMPlexView_Ascii_Coordinates(PetscViewer viewer, CoordSystem cs, PetscInt dim, const PetscScalar x[])
885 {
886   PetscInt i;
887 
888   PetscFunctionBegin;
889   if (dim > 3) {
890     for (i = 0; i < dim; ++i) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, " %g", (double)PetscRealPart(x[i])));
891   } else {
892     PetscReal coords[3], trcoords[3] = {0., 0., 0.};
893 
894     for (i = 0; i < dim; ++i) coords[i] = PetscRealPart(x[i]);
895     switch (cs) {
896     case CS_CARTESIAN:
897       for (i = 0; i < dim; ++i) trcoords[i] = coords[i];
898       break;
899     case CS_POLAR:
900       PetscCheck(dim == 2, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Polar coordinates are for 2 dimension, not %" PetscInt_FMT, dim);
901       trcoords[0] = PetscSqrtReal(PetscSqr(coords[0]) + PetscSqr(coords[1]));
902       trcoords[1] = PetscAtan2Real(coords[1], coords[0]);
903       break;
904     case CS_CYLINDRICAL:
905       PetscCheck(dim == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cylindrical coordinates are for 3 dimension, not %" PetscInt_FMT, dim);
906       trcoords[0] = PetscSqrtReal(PetscSqr(coords[0]) + PetscSqr(coords[1]));
907       trcoords[1] = PetscAtan2Real(coords[1], coords[0]);
908       trcoords[2] = coords[2];
909       break;
910     case CS_SPHERICAL:
911       PetscCheck(dim == 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Spherical coordinates are for 3 dimension, not %" PetscInt_FMT, dim);
912       trcoords[0] = PetscSqrtReal(PetscSqr(coords[0]) + PetscSqr(coords[1]) + PetscSqr(coords[2]));
913       trcoords[1] = PetscAtan2Real(PetscSqrtReal(PetscSqr(coords[0]) + PetscSqr(coords[1])), coords[2]);
914       trcoords[2] = PetscAtan2Real(coords[1], coords[0]);
915       break;
916     }
917     for (i = 0; i < dim; ++i) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, " %g", (double)trcoords[i]));
918   }
919   PetscFunctionReturn(PETSC_SUCCESS);
920 }
921 
922 static PetscErrorCode DMPlexView_Ascii(DM dm, PetscViewer viewer)
923 {
924   DM_Plex          *mesh = (DM_Plex *)dm->data;
925   DM                cdm, cdmCell;
926   PetscSection      coordSection, coordSectionCell;
927   Vec               coordinates, coordinatesCell;
928   PetscViewerFormat format;
929 
930   PetscFunctionBegin;
931   PetscCall(PetscViewerGetFormat(viewer, &format));
932   if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
933     const char *name;
934     PetscInt    dim, cellHeight, maxConeSize, maxSupportSize;
935     PetscInt    pStart, pEnd, p, numLabels, l;
936     PetscMPIInt rank, size;
937 
938     PetscCall(DMGetCoordinateDM(dm, &cdm));
939     PetscCall(DMGetCoordinateSection(dm, &coordSection));
940     PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
941     PetscCall(DMGetCellCoordinateDM(dm, &cdmCell));
942     PetscCall(DMGetCellCoordinateSection(dm, &coordSectionCell));
943     PetscCall(DMGetCellCoordinatesLocal(dm, &coordinatesCell));
944     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
945     PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm), &size));
946     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
947     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
948     PetscCall(DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize));
949     PetscCall(DMGetDimension(dm, &dim));
950     PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
951     if (name) PetscCall(PetscViewerASCIIPrintf(viewer, "%s in %" PetscInt_FMT " dimension%s:\n", name, dim, dim == 1 ? "" : "s"));
952     else PetscCall(PetscViewerASCIIPrintf(viewer, "Mesh in %" PetscInt_FMT " dimension%s:\n", dim, dim == 1 ? "" : "s"));
953     if (cellHeight) PetscCall(PetscViewerASCIIPrintf(viewer, "  Cells are at height %" PetscInt_FMT "\n", cellHeight));
954     PetscCall(PetscViewerASCIIPrintf(viewer, "Supports:\n"));
955     PetscCall(PetscViewerASCIIPushSynchronized(viewer));
956     PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "[%d] Max support size: %" PetscInt_FMT "\n", rank, maxSupportSize));
957     for (p = pStart; p < pEnd; ++p) {
958       PetscInt dof, off, s;
959 
960       PetscCall(PetscSectionGetDof(mesh->supportSection, p, &dof));
961       PetscCall(PetscSectionGetOffset(mesh->supportSection, p, &off));
962       for (s = off; s < off + dof; ++s) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "[%d]: %" PetscInt_FMT " ----> %" PetscInt_FMT "\n", rank, p, mesh->supports[s]));
963     }
964     PetscCall(PetscViewerFlush(viewer));
965     PetscCall(PetscViewerASCIIPrintf(viewer, "Cones:\n"));
966     PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "[%d] Max cone size: %" PetscInt_FMT "\n", rank, maxConeSize));
967     for (p = pStart; p < pEnd; ++p) {
968       PetscInt dof, off, c;
969 
970       PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
971       PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
972       for (c = off; c < off + dof; ++c) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "[%d]: %" PetscInt_FMT " <---- %" PetscInt_FMT " (%" PetscInt_FMT ")\n", rank, p, mesh->cones[c], mesh->coneOrientations[c]));
973     }
974     PetscCall(PetscViewerFlush(viewer));
975     PetscCall(PetscViewerASCIIPopSynchronized(viewer));
976     if (coordSection && coordinates) {
977       CoordSystem        cs = CS_CARTESIAN;
978       const PetscScalar *array, *arrayCell = NULL;
979       PetscInt           Nf, Nc, pvStart, pvEnd, pcStart = PETSC_MAX_INT, pcEnd = PETSC_MIN_INT, pStart, pEnd, p;
980       PetscMPIInt        rank;
981       const char        *name;
982 
983       PetscCall(PetscOptionsGetEnum(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_coord_system", CoordSystems, (PetscEnum *)&cs, NULL));
984       PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
985       PetscCall(PetscSectionGetNumFields(coordSection, &Nf));
986       PetscCheck(Nf == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Coordinate section should have 1 field, not %" PetscInt_FMT, Nf);
987       PetscCall(PetscSectionGetFieldComponents(coordSection, 0, &Nc));
988       PetscCall(PetscSectionGetChart(coordSection, &pvStart, &pvEnd));
989       if (coordSectionCell) PetscCall(PetscSectionGetChart(coordSectionCell, &pcStart, &pcEnd));
990       pStart = PetscMin(pvStart, pcStart);
991       pEnd   = PetscMax(pvEnd, pcEnd);
992       PetscCall(PetscObjectGetName((PetscObject)coordinates, &name));
993       PetscCall(PetscViewerASCIIPrintf(viewer, "%s with %" PetscInt_FMT " fields\n", name, Nf));
994       PetscCall(PetscViewerASCIIPrintf(viewer, "  field 0 with %" PetscInt_FMT " components\n", Nc));
995       if (cs != CS_CARTESIAN) PetscCall(PetscViewerASCIIPrintf(viewer, "  output coordinate system: %s\n", CoordSystems[cs]));
996 
997       PetscCall(VecGetArrayRead(coordinates, &array));
998       if (coordinatesCell) PetscCall(VecGetArrayRead(coordinatesCell, &arrayCell));
999       PetscCall(PetscViewerASCIIPushSynchronized(viewer));
1000       PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "Process %d:\n", rank));
1001       for (p = pStart; p < pEnd; ++p) {
1002         PetscInt dof, off;
1003 
1004         if (p >= pvStart && p < pvEnd) {
1005           PetscCall(PetscSectionGetDof(coordSection, p, &dof));
1006           PetscCall(PetscSectionGetOffset(coordSection, p, &off));
1007           if (dof) {
1008             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "  (%4" PetscInt_FMT ") dim %2" PetscInt_FMT " offset %3" PetscInt_FMT, p, dof, off));
1009             PetscCall(DMPlexView_Ascii_Coordinates(viewer, cs, dof, &array[off]));
1010             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\n"));
1011           }
1012         }
1013         if (cdmCell && p >= pcStart && p < pcEnd) {
1014           PetscCall(PetscSectionGetDof(coordSectionCell, p, &dof));
1015           PetscCall(PetscSectionGetOffset(coordSectionCell, p, &off));
1016           if (dof) {
1017             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "  (%4" PetscInt_FMT ") dim %2" PetscInt_FMT " offset %3" PetscInt_FMT, p, dof, off));
1018             PetscCall(DMPlexView_Ascii_Coordinates(viewer, cs, dof, &arrayCell[off]));
1019             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\n"));
1020           }
1021         }
1022       }
1023       PetscCall(PetscViewerFlush(viewer));
1024       PetscCall(PetscViewerASCIIPopSynchronized(viewer));
1025       PetscCall(VecRestoreArrayRead(coordinates, &array));
1026       if (coordinatesCell) PetscCall(VecRestoreArrayRead(coordinatesCell, &arrayCell));
1027     }
1028     PetscCall(DMGetNumLabels(dm, &numLabels));
1029     if (numLabels) PetscCall(PetscViewerASCIIPrintf(viewer, "Labels:\n"));
1030     for (l = 0; l < numLabels; ++l) {
1031       DMLabel     label;
1032       PetscBool   isdepth;
1033       const char *name;
1034 
1035       PetscCall(DMGetLabelName(dm, l, &name));
1036       PetscCall(PetscStrcmp(name, "depth", &isdepth));
1037       if (isdepth) continue;
1038       PetscCall(DMGetLabel(dm, name, &label));
1039       PetscCall(DMLabelView(label, viewer));
1040     }
1041     if (size > 1) {
1042       PetscSF sf;
1043 
1044       PetscCall(DMGetPointSF(dm, &sf));
1045       PetscCall(PetscSFView(sf, viewer));
1046     }
1047     if (mesh->periodic.face_sf) PetscCall(PetscSFView(mesh->periodic.face_sf, viewer));
1048     PetscCall(PetscViewerFlush(viewer));
1049   } else if (format == PETSC_VIEWER_ASCII_LATEX) {
1050     const char  *name, *color;
1051     const char  *defcolors[3]  = {"gray", "orange", "green"};
1052     const char  *deflcolors[4] = {"blue", "cyan", "red", "magenta"};
1053     char         lname[PETSC_MAX_PATH_LEN];
1054     PetscReal    scale      = 2.0;
1055     PetscReal    tikzscale  = 1.0;
1056     PetscBool    useNumbers = PETSC_TRUE, drawNumbers[4], drawColors[4], useLabels, useColors, plotEdges, drawHasse = PETSC_FALSE;
1057     double       tcoords[3];
1058     PetscScalar *coords;
1059     PetscInt     numLabels, l, numColors, numLColors, dim, d, depth, cStart, cEnd, c, vStart, vEnd, v, eStart = 0, eEnd = 0, e, p, n;
1060     PetscMPIInt  rank, size;
1061     char       **names, **colors, **lcolors;
1062     PetscBool    flg, lflg;
1063     PetscBT      wp = NULL;
1064     PetscInt     pEnd, pStart;
1065 
1066     PetscCall(DMGetCoordinateDM(dm, &cdm));
1067     PetscCall(DMGetCoordinateSection(dm, &coordSection));
1068     PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
1069     PetscCall(DMGetCellCoordinateDM(dm, &cdmCell));
1070     PetscCall(DMGetCellCoordinateSection(dm, &coordSectionCell));
1071     PetscCall(DMGetCellCoordinatesLocal(dm, &coordinatesCell));
1072     PetscCall(DMGetDimension(dm, &dim));
1073     PetscCall(DMPlexGetDepth(dm, &depth));
1074     PetscCall(DMGetNumLabels(dm, &numLabels));
1075     numLabels  = PetscMax(numLabels, 10);
1076     numColors  = 10;
1077     numLColors = 10;
1078     PetscCall(PetscCalloc3(numLabels, &names, numColors, &colors, numLColors, &lcolors));
1079     PetscCall(PetscOptionsGetReal(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_scale", &scale, NULL));
1080     PetscCall(PetscOptionsGetReal(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_tikzscale", &tikzscale, NULL));
1081     PetscCall(PetscOptionsGetBool(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_numbers", &useNumbers, NULL));
1082     for (d = 0; d < 4; ++d) drawNumbers[d] = useNumbers;
1083     for (d = 0; d < 4; ++d) drawColors[d] = PETSC_TRUE;
1084     n = 4;
1085     PetscCall(PetscOptionsGetBoolArray(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_numbers_depth", drawNumbers, &n, &flg));
1086     PetscCheck(!flg || n == dim + 1, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Number of flags %" PetscInt_FMT " != %" PetscInt_FMT " dim+1", n, dim + 1);
1087     n = 4;
1088     PetscCall(PetscOptionsGetBoolArray(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_colors_depth", drawColors, &n, &flg));
1089     PetscCheck(!flg || n == dim + 1, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Number of flags %" PetscInt_FMT " != %" PetscInt_FMT " dim+1", n, dim + 1);
1090     PetscCall(PetscOptionsGetStringArray(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_labels", names, &numLabels, &useLabels));
1091     if (!useLabels) numLabels = 0;
1092     PetscCall(PetscOptionsGetStringArray(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_colors", colors, &numColors, &useColors));
1093     if (!useColors) {
1094       numColors = 3;
1095       for (c = 0; c < numColors; ++c) PetscCall(PetscStrallocpy(defcolors[c], &colors[c]));
1096     }
1097     PetscCall(PetscOptionsGetStringArray(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_lcolors", lcolors, &numLColors, &useColors));
1098     if (!useColors) {
1099       numLColors = 4;
1100       for (c = 0; c < numLColors; ++c) PetscCall(PetscStrallocpy(deflcolors[c], &lcolors[c]));
1101     }
1102     PetscCall(PetscOptionsGetString(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_label_filter", lname, sizeof(lname), &lflg));
1103     plotEdges = (PetscBool)(depth > 1 && drawNumbers[1] && dim < 3);
1104     PetscCall(PetscOptionsGetBool(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_edges", &plotEdges, &flg));
1105     PetscCheck(!flg || !plotEdges || depth >= dim, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Mesh must be interpolated");
1106     if (depth < dim) plotEdges = PETSC_FALSE;
1107     PetscCall(PetscOptionsGetBool(((PetscObject)viewer)->options, ((PetscObject)viewer)->prefix, "-dm_plex_view_hasse", &drawHasse, NULL));
1108 
1109     /* filter points with labelvalue != labeldefaultvalue */
1110     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
1111     PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
1112     PetscCall(DMPlexGetDepthStratum(dm, 1, &eStart, &eEnd));
1113     PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
1114     if (lflg) {
1115       DMLabel lbl;
1116 
1117       PetscCall(DMGetLabel(dm, lname, &lbl));
1118       if (lbl) {
1119         PetscInt val, defval;
1120 
1121         PetscCall(DMLabelGetDefaultValue(lbl, &defval));
1122         PetscCall(PetscBTCreate(pEnd - pStart, &wp));
1123         for (c = pStart; c < pEnd; c++) {
1124           PetscInt *closure = NULL;
1125           PetscInt  closureSize;
1126 
1127           PetscCall(DMLabelGetValue(lbl, c, &val));
1128           if (val == defval) continue;
1129 
1130           PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
1131           for (p = 0; p < closureSize * 2; p += 2) PetscCall(PetscBTSet(wp, closure[p] - pStart));
1132           PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
1133         }
1134       }
1135     }
1136 
1137     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1138     PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm), &size));
1139     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
1140     PetscCall(PetscViewerASCIIPrintf(viewer, "\
1141 \\documentclass[tikz]{standalone}\n\n\
1142 \\usepackage{pgflibraryshapes}\n\
1143 \\usetikzlibrary{backgrounds}\n\
1144 \\usetikzlibrary{arrows}\n\
1145 \\begin{document}\n"));
1146     if (size > 1) {
1147       PetscCall(PetscViewerASCIIPrintf(viewer, "%s for process ", name));
1148       for (p = 0; p < size; ++p) {
1149         if (p) PetscCall(PetscViewerASCIIPrintf(viewer, (p == size - 1) ? ", and " : ", "));
1150         PetscCall(PetscViewerASCIIPrintf(viewer, "{\\textcolor{%s}%" PetscInt_FMT "}", colors[p % numColors], p));
1151       }
1152       PetscCall(PetscViewerASCIIPrintf(viewer, ".\n\n\n"));
1153     }
1154     if (drawHasse) {
1155       PetscInt maxStratum = PetscMax(vEnd - vStart, PetscMax(eEnd - eStart, cEnd - cStart));
1156 
1157       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\vStart}{%" PetscInt_FMT "}\n", vStart));
1158       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\vEnd}{%" PetscInt_FMT "}\n", vEnd - 1));
1159       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\numVertices}{%" PetscInt_FMT "}\n", vEnd - vStart));
1160       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\vShift}{%.2f}\n", 3 + (maxStratum - (vEnd - vStart)) / 2.));
1161       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\eStart}{%" PetscInt_FMT "}\n", eStart));
1162       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\eEnd}{%" PetscInt_FMT "}\n", eEnd - 1));
1163       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\eShift}{%.2f}\n", 3 + (maxStratum - (eEnd - eStart)) / 2.));
1164       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\numEdges}{%" PetscInt_FMT "}\n", eEnd - eStart));
1165       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\cStart}{%" PetscInt_FMT "}\n", cStart));
1166       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\cEnd}{%" PetscInt_FMT "}\n", cEnd - 1));
1167       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\numCells}{%" PetscInt_FMT "}\n", cEnd - cStart));
1168       PetscCall(PetscViewerASCIIPrintf(viewer, "\\newcommand{\\cShift}{%.2f}\n", 3 + (maxStratum - (cEnd - cStart)) / 2.));
1169     }
1170     PetscCall(PetscViewerASCIIPrintf(viewer, "\\begin{tikzpicture}[scale = %g,font=\\fontsize{8}{8}\\selectfont]\n", (double)tikzscale));
1171 
1172     /* Plot vertices */
1173     PetscCall(VecGetArray(coordinates, &coords));
1174     PetscCall(PetscViewerASCIIPushSynchronized(viewer));
1175     for (v = vStart; v < vEnd; ++v) {
1176       PetscInt  off, dof, d;
1177       PetscBool isLabeled = PETSC_FALSE;
1178 
1179       if (wp && !PetscBTLookup(wp, v - pStart)) continue;
1180       PetscCall(PetscSectionGetDof(coordSection, v, &dof));
1181       PetscCall(PetscSectionGetOffset(coordSection, v, &off));
1182       PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\\path ("));
1183       PetscCheck(dof <= 3, PETSC_COMM_SELF, PETSC_ERR_PLIB, "coordSection vertex %" PetscInt_FMT " has dof %" PetscInt_FMT " > 3", v, dof);
1184       for (d = 0; d < dof; ++d) {
1185         tcoords[d] = (double)(scale * PetscRealPart(coords[off + d]));
1186         tcoords[d] = PetscAbs(tcoords[d]) < 1e-10 ? 0.0 : tcoords[d];
1187       }
1188       /* Rotate coordinates since PGF makes z point out of the page instead of up */
1189       if (dim == 3) {
1190         PetscReal tmp = tcoords[1];
1191         tcoords[1]    = tcoords[2];
1192         tcoords[2]    = -tmp;
1193       }
1194       for (d = 0; d < dof; ++d) {
1195         if (d > 0) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ","));
1196         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "%g", (double)tcoords[d]));
1197       }
1198       if (drawHasse) color = colors[0 % numColors];
1199       else color = colors[rank % numColors];
1200       for (l = 0; l < numLabels; ++l) {
1201         PetscInt val;
1202         PetscCall(DMGetLabelValue(dm, names[l], v, &val));
1203         if (val >= 0) {
1204           color     = lcolors[l % numLColors];
1205           isLabeled = PETSC_TRUE;
1206           break;
1207         }
1208       }
1209       if (drawNumbers[0]) {
1210         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [draw,shape=circle,color=%s] {%" PetscInt_FMT "};\n", v, rank, color, v));
1211       } else if (drawColors[0]) {
1212         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [fill,inner sep=%dpt,shape=circle,color=%s] {};\n", v, rank, !isLabeled ? 1 : 2, color));
1213       } else PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [] {};\n", v, rank));
1214     }
1215     PetscCall(VecRestoreArray(coordinates, &coords));
1216     PetscCall(PetscViewerFlush(viewer));
1217     /* Plot edges */
1218     if (plotEdges) {
1219       PetscCall(VecGetArray(coordinates, &coords));
1220       PetscCall(PetscViewerASCIIPrintf(viewer, "\\path\n"));
1221       for (e = eStart; e < eEnd; ++e) {
1222         const PetscInt *cone;
1223         PetscInt        coneSize, offA, offB, dof, d;
1224 
1225         if (wp && !PetscBTLookup(wp, e - pStart)) continue;
1226         PetscCall(DMPlexGetConeSize(dm, e, &coneSize));
1227         PetscCheck(coneSize == 2, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Edge %" PetscInt_FMT " cone should have two vertices, not %" PetscInt_FMT, e, coneSize);
1228         PetscCall(DMPlexGetCone(dm, e, &cone));
1229         PetscCall(PetscSectionGetDof(coordSection, cone[0], &dof));
1230         PetscCall(PetscSectionGetOffset(coordSection, cone[0], &offA));
1231         PetscCall(PetscSectionGetOffset(coordSection, cone[1], &offB));
1232         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "("));
1233         for (d = 0; d < dof; ++d) {
1234           tcoords[d] = (double)(0.5 * scale * PetscRealPart(coords[offA + d] + coords[offB + d]));
1235           tcoords[d] = PetscAbs(tcoords[d]) < 1e-10 ? 0.0 : tcoords[d];
1236         }
1237         /* Rotate coordinates since PGF makes z point out of the page instead of up */
1238         if (dim == 3) {
1239           PetscReal tmp = tcoords[1];
1240           tcoords[1]    = tcoords[2];
1241           tcoords[2]    = -tmp;
1242         }
1243         for (d = 0; d < dof; ++d) {
1244           if (d > 0) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ","));
1245           PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "%g", (double)tcoords[d]));
1246         }
1247         if (drawHasse) color = colors[1 % numColors];
1248         else color = colors[rank % numColors];
1249         for (l = 0; l < numLabels; ++l) {
1250           PetscInt val;
1251           PetscCall(DMGetLabelValue(dm, names[l], e, &val));
1252           if (val >= 0) {
1253             color = lcolors[l % numLColors];
1254             break;
1255           }
1256         }
1257         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [draw,shape=circle,color=%s] {%" PetscInt_FMT "} --\n", e, rank, color, e));
1258       }
1259       PetscCall(VecRestoreArray(coordinates, &coords));
1260       PetscCall(PetscViewerFlush(viewer));
1261       PetscCall(PetscViewerASCIIPrintf(viewer, "(0,0);\n"));
1262     }
1263     /* Plot cells */
1264     if (dim == 3 || !drawNumbers[1]) {
1265       for (e = eStart; e < eEnd; ++e) {
1266         const PetscInt *cone;
1267 
1268         if (wp && !PetscBTLookup(wp, e - pStart)) continue;
1269         color = colors[rank % numColors];
1270         for (l = 0; l < numLabels; ++l) {
1271           PetscInt val;
1272           PetscCall(DMGetLabelValue(dm, names[l], e, &val));
1273           if (val >= 0) {
1274             color = lcolors[l % numLColors];
1275             break;
1276           }
1277         }
1278         PetscCall(DMPlexGetCone(dm, e, &cone));
1279         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\\draw[color=%s] (%" PetscInt_FMT "_%d) -- (%" PetscInt_FMT "_%d);\n", color, cone[0], rank, cone[1], rank));
1280       }
1281     } else {
1282       DMPolytopeType ct;
1283 
1284       /* Drawing a 2D polygon */
1285       for (c = cStart; c < cEnd; ++c) {
1286         if (wp && !PetscBTLookup(wp, c - pStart)) continue;
1287         PetscCall(DMPlexGetCellType(dm, c, &ct));
1288         if (DMPolytopeTypeIsHybrid(ct)) {
1289           const PetscInt *cone;
1290           PetscInt        coneSize, e;
1291 
1292           PetscCall(DMPlexGetCone(dm, c, &cone));
1293           PetscCall(DMPlexGetConeSize(dm, c, &coneSize));
1294           for (e = 0; e < coneSize; ++e) {
1295             const PetscInt *econe;
1296 
1297             PetscCall(DMPlexGetCone(dm, cone[e], &econe));
1298             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\\draw[color=%s] (%" PetscInt_FMT "_%d) -- (%" PetscInt_FMT "_%d) -- (%" PetscInt_FMT "_%d);\n", colors[rank % numColors], econe[0], rank, cone[e], rank, econe[1], rank));
1299           }
1300         } else {
1301           PetscInt *closure = NULL;
1302           PetscInt  closureSize, Nv = 0, v;
1303 
1304           PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
1305           for (p = 0; p < closureSize * 2; p += 2) {
1306             const PetscInt point = closure[p];
1307 
1308             if ((point >= vStart) && (point < vEnd)) closure[Nv++] = point;
1309           }
1310           PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\\draw[color=%s] ", colors[rank % numColors]));
1311           for (v = 0; v <= Nv; ++v) {
1312             const PetscInt vertex = closure[v % Nv];
1313 
1314             if (v > 0) {
1315               if (plotEdges) {
1316                 const PetscInt *edge;
1317                 PetscInt        endpoints[2], ne;
1318 
1319                 endpoints[0] = closure[v - 1];
1320                 endpoints[1] = vertex;
1321                 PetscCall(DMPlexGetJoin(dm, 2, endpoints, &ne, &edge));
1322                 PetscCheck(ne == 1, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not find edge for vertices %" PetscInt_FMT ", %" PetscInt_FMT, endpoints[0], endpoints[1]);
1323                 PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, " -- (%" PetscInt_FMT "_%d) -- ", edge[0], rank));
1324                 PetscCall(DMPlexRestoreJoin(dm, 2, endpoints, &ne, &edge));
1325               } else PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, " -- "));
1326             }
1327             PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "(%" PetscInt_FMT "_%d)", vertex, rank));
1328           }
1329           PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ";\n"));
1330           PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
1331         }
1332       }
1333     }
1334     for (c = cStart; c < cEnd; ++c) {
1335       double             ccoords[3] = {0.0, 0.0, 0.0};
1336       PetscBool          isLabeled  = PETSC_FALSE;
1337       PetscScalar       *cellCoords = NULL;
1338       const PetscScalar *array;
1339       PetscInt           numCoords, cdim, d;
1340       PetscBool          isDG;
1341 
1342       if (wp && !PetscBTLookup(wp, c - pStart)) continue;
1343       PetscCall(DMGetCoordinateDim(dm, &cdim));
1344       PetscCall(DMPlexGetCellCoordinates(dm, c, &isDG, &numCoords, &array, &cellCoords));
1345       PetscCheck(!(numCoords % cdim), PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "coordinate dim %" PetscInt_FMT " does not divide numCoords %" PetscInt_FMT, cdim, numCoords);
1346       PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "\\path ("));
1347       for (p = 0; p < numCoords / cdim; ++p) {
1348         for (d = 0; d < cdim; ++d) {
1349           tcoords[d] = (double)(scale * PetscRealPart(cellCoords[p * cdim + d]));
1350           tcoords[d] = PetscAbs(tcoords[d]) < 1e-10 ? 0.0 : tcoords[d];
1351         }
1352         /* Rotate coordinates since PGF makes z point out of the page instead of up */
1353         if (cdim == 3) {
1354           PetscReal tmp = tcoords[1];
1355           tcoords[1]    = tcoords[2];
1356           tcoords[2]    = -tmp;
1357         }
1358         for (d = 0; d < dim; ++d) ccoords[d] += tcoords[d];
1359       }
1360       for (d = 0; d < cdim; ++d) ccoords[d] /= (numCoords / cdim);
1361       PetscCall(DMPlexRestoreCellCoordinates(dm, c, &isDG, &numCoords, &array, &cellCoords));
1362       for (d = 0; d < cdim; ++d) {
1363         if (d > 0) PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ","));
1364         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "%g", (double)ccoords[d]));
1365       }
1366       if (drawHasse) color = colors[depth % numColors];
1367       else color = colors[rank % numColors];
1368       for (l = 0; l < numLabels; ++l) {
1369         PetscInt val;
1370         PetscCall(DMGetLabelValue(dm, names[l], c, &val));
1371         if (val >= 0) {
1372           color     = lcolors[l % numLColors];
1373           isLabeled = PETSC_TRUE;
1374           break;
1375         }
1376       }
1377       if (drawNumbers[dim]) {
1378         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [draw,shape=circle,color=%s] {%" PetscInt_FMT "};\n", c, rank, color, c));
1379       } else if (drawColors[dim]) {
1380         PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [fill,inner sep=%dpt,shape=circle,color=%s] {};\n", c, rank, !isLabeled ? 1 : 2, color));
1381       } else PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, ") node(%" PetscInt_FMT "_%d) [] {};\n", c, rank));
1382     }
1383     if (drawHasse) {
1384       color = colors[depth % numColors];
1385       PetscCall(PetscViewerASCIIPrintf(viewer, "%% Cells\n"));
1386       PetscCall(PetscViewerASCIIPrintf(viewer, "\\foreach \\c in {\\cStart,...,\\cEnd}\n"));
1387       PetscCall(PetscViewerASCIIPrintf(viewer, "{\n"));
1388       PetscCall(PetscViewerASCIIPrintf(viewer, "  \\node(\\c_%d) [draw,shape=circle,color=%s,minimum size = 6mm] at (\\cShift+\\c-\\cStart,0) {\\c};\n", rank, color));
1389       PetscCall(PetscViewerASCIIPrintf(viewer, "}\n"));
1390 
1391       color = colors[1 % numColors];
1392       PetscCall(PetscViewerASCIIPrintf(viewer, "%% Edges\n"));
1393       PetscCall(PetscViewerASCIIPrintf(viewer, "\\foreach \\e in {\\eStart,...,\\eEnd}\n"));
1394       PetscCall(PetscViewerASCIIPrintf(viewer, "{\n"));
1395       PetscCall(PetscViewerASCIIPrintf(viewer, "  \\node(\\e_%d) [draw,shape=circle,color=%s,minimum size = 6mm] at (\\eShift+\\e-\\eStart,1) {\\e};\n", rank, color));
1396       PetscCall(PetscViewerASCIIPrintf(viewer, "}\n"));
1397 
1398       color = colors[0 % numColors];
1399       PetscCall(PetscViewerASCIIPrintf(viewer, "%% Vertices\n"));
1400       PetscCall(PetscViewerASCIIPrintf(viewer, "\\foreach \\v in {\\vStart,...,\\vEnd}\n"));
1401       PetscCall(PetscViewerASCIIPrintf(viewer, "{\n"));
1402       PetscCall(PetscViewerASCIIPrintf(viewer, "  \\node(\\v_%d) [draw,shape=circle,color=%s,minimum size = 6mm] at (\\vShift+\\v-\\vStart,2) {\\v};\n", rank, color));
1403       PetscCall(PetscViewerASCIIPrintf(viewer, "}\n"));
1404 
1405       for (p = pStart; p < pEnd; ++p) {
1406         const PetscInt *cone;
1407         PetscInt        coneSize, cp;
1408 
1409         PetscCall(DMPlexGetCone(dm, p, &cone));
1410         PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
1411         for (cp = 0; cp < coneSize; ++cp) PetscCall(PetscViewerASCIIPrintf(viewer, "\\draw[->, shorten >=1pt] (%" PetscInt_FMT "_%d) -- (%" PetscInt_FMT "_%d);\n", cone[cp], rank, p, rank));
1412       }
1413     }
1414     PetscCall(PetscViewerFlush(viewer));
1415     PetscCall(PetscViewerASCIIPopSynchronized(viewer));
1416     PetscCall(PetscViewerASCIIPrintf(viewer, "\\end{tikzpicture}\n"));
1417     PetscCall(PetscViewerASCIIPrintf(viewer, "\\end{document}\n"));
1418     for (l = 0; l < numLabels; ++l) PetscCall(PetscFree(names[l]));
1419     for (c = 0; c < numColors; ++c) PetscCall(PetscFree(colors[c]));
1420     for (c = 0; c < numLColors; ++c) PetscCall(PetscFree(lcolors[c]));
1421     PetscCall(PetscFree3(names, colors, lcolors));
1422     PetscCall(PetscBTDestroy(&wp));
1423   } else if (format == PETSC_VIEWER_LOAD_BALANCE) {
1424     Vec                    cown, acown;
1425     VecScatter             sct;
1426     ISLocalToGlobalMapping g2l;
1427     IS                     gid, acis;
1428     MPI_Comm               comm, ncomm = MPI_COMM_NULL;
1429     MPI_Group              ggroup, ngroup;
1430     PetscScalar           *array, nid;
1431     const PetscInt        *idxs;
1432     PetscInt              *idxs2, *start, *adjacency, *work;
1433     PetscInt64             lm[3], gm[3];
1434     PetscInt               i, c, cStart, cEnd, cum, numVertices, ect, ectn, cellHeight;
1435     PetscMPIInt            d1, d2, rank;
1436 
1437     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
1438     PetscCallMPI(MPI_Comm_rank(comm, &rank));
1439 #if defined(PETSC_HAVE_MPI_PROCESS_SHARED_MEMORY)
1440     PetscCallMPI(MPI_Comm_split_type(comm, MPI_COMM_TYPE_SHARED, rank, MPI_INFO_NULL, &ncomm));
1441 #endif
1442     if (ncomm != MPI_COMM_NULL) {
1443       PetscCallMPI(MPI_Comm_group(comm, &ggroup));
1444       PetscCallMPI(MPI_Comm_group(ncomm, &ngroup));
1445       d1 = 0;
1446       PetscCallMPI(MPI_Group_translate_ranks(ngroup, 1, &d1, ggroup, &d2));
1447       nid = d2;
1448       PetscCallMPI(MPI_Group_free(&ggroup));
1449       PetscCallMPI(MPI_Group_free(&ngroup));
1450       PetscCallMPI(MPI_Comm_free(&ncomm));
1451     } else nid = 0.0;
1452 
1453     /* Get connectivity */
1454     PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
1455     PetscCall(DMPlexCreatePartitionerGraph(dm, cellHeight, &numVertices, &start, &adjacency, &gid));
1456 
1457     /* filter overlapped local cells */
1458     PetscCall(DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd));
1459     PetscCall(ISGetIndices(gid, &idxs));
1460     PetscCall(ISGetLocalSize(gid, &cum));
1461     PetscCall(PetscMalloc1(cum, &idxs2));
1462     for (c = cStart, cum = 0; c < cEnd; c++) {
1463       if (idxs[c - cStart] < 0) continue;
1464       idxs2[cum++] = idxs[c - cStart];
1465     }
1466     PetscCall(ISRestoreIndices(gid, &idxs));
1467     PetscCheck(numVertices == cum, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected %" PetscInt_FMT " != %" PetscInt_FMT, numVertices, cum);
1468     PetscCall(ISDestroy(&gid));
1469     PetscCall(ISCreateGeneral(comm, numVertices, idxs2, PETSC_OWN_POINTER, &gid));
1470 
1471     /* support for node-aware cell locality */
1472     PetscCall(ISCreateGeneral(comm, start[numVertices], adjacency, PETSC_USE_POINTER, &acis));
1473     PetscCall(VecCreateSeq(PETSC_COMM_SELF, start[numVertices], &acown));
1474     PetscCall(VecCreateMPI(comm, numVertices, PETSC_DECIDE, &cown));
1475     PetscCall(VecGetArray(cown, &array));
1476     for (c = 0; c < numVertices; c++) array[c] = nid;
1477     PetscCall(VecRestoreArray(cown, &array));
1478     PetscCall(VecScatterCreate(cown, acis, acown, NULL, &sct));
1479     PetscCall(VecScatterBegin(sct, cown, acown, INSERT_VALUES, SCATTER_FORWARD));
1480     PetscCall(VecScatterEnd(sct, cown, acown, INSERT_VALUES, SCATTER_FORWARD));
1481     PetscCall(ISDestroy(&acis));
1482     PetscCall(VecScatterDestroy(&sct));
1483     PetscCall(VecDestroy(&cown));
1484 
1485     /* compute edgeCut */
1486     for (c = 0, cum = 0; c < numVertices; c++) cum = PetscMax(cum, start[c + 1] - start[c]);
1487     PetscCall(PetscMalloc1(cum, &work));
1488     PetscCall(ISLocalToGlobalMappingCreateIS(gid, &g2l));
1489     PetscCall(ISLocalToGlobalMappingSetType(g2l, ISLOCALTOGLOBALMAPPINGHASH));
1490     PetscCall(ISDestroy(&gid));
1491     PetscCall(VecGetArray(acown, &array));
1492     for (c = 0, ect = 0, ectn = 0; c < numVertices; c++) {
1493       PetscInt totl;
1494 
1495       totl = start[c + 1] - start[c];
1496       PetscCall(ISGlobalToLocalMappingApply(g2l, IS_GTOLM_MASK, totl, adjacency + start[c], NULL, work));
1497       for (i = 0; i < totl; i++) {
1498         if (work[i] < 0) {
1499           ect += 1;
1500           ectn += (array[i + start[c]] != nid) ? 0 : 1;
1501         }
1502       }
1503     }
1504     PetscCall(PetscFree(work));
1505     PetscCall(VecRestoreArray(acown, &array));
1506     lm[0] = numVertices > 0 ? numVertices : PETSC_MAX_INT;
1507     lm[1] = -numVertices;
1508     PetscCall(MPIU_Allreduce(lm, gm, 2, MPIU_INT64, MPI_MIN, comm));
1509     PetscCall(PetscViewerASCIIPrintf(viewer, "  Cell balance: %.2f (max %" PetscInt_FMT ", min %" PetscInt_FMT, -((double)gm[1]) / ((double)gm[0]), -(PetscInt)gm[1], (PetscInt)gm[0]));
1510     lm[0] = ect;                     /* edgeCut */
1511     lm[1] = ectn;                    /* node-aware edgeCut */
1512     lm[2] = numVertices > 0 ? 0 : 1; /* empty processes */
1513     PetscCall(MPIU_Allreduce(lm, gm, 3, MPIU_INT64, MPI_SUM, comm));
1514     PetscCall(PetscViewerASCIIPrintf(viewer, ", empty %" PetscInt_FMT ")\n", (PetscInt)gm[2]));
1515 #if defined(PETSC_HAVE_MPI_PROCESS_SHARED_MEMORY)
1516     PetscCall(PetscViewerASCIIPrintf(viewer, "  Edge Cut: %" PetscInt_FMT " (on node %.3f)\n", (PetscInt)(gm[0] / 2), gm[0] ? ((double)(gm[1])) / ((double)gm[0]) : 1.));
1517 #else
1518     PetscCall(PetscViewerASCIIPrintf(viewer, "  Edge Cut: %" PetscInt_FMT " (on node %.3f)\n", (PetscInt)(gm[0] / 2), 0.0));
1519 #endif
1520     PetscCall(ISLocalToGlobalMappingDestroy(&g2l));
1521     PetscCall(PetscFree(start));
1522     PetscCall(PetscFree(adjacency));
1523     PetscCall(VecDestroy(&acown));
1524   } else {
1525     const char    *name;
1526     PetscInt      *sizes, *hybsizes, *ghostsizes;
1527     PetscInt       locDepth, depth, cellHeight, dim, d;
1528     PetscInt       pStart, pEnd, p, gcStart, gcEnd, gcNum;
1529     PetscInt       numLabels, l, maxSize = 17;
1530     DMPolytopeType ct0 = DM_POLYTOPE_UNKNOWN;
1531     MPI_Comm       comm;
1532     PetscMPIInt    size, rank;
1533 
1534     PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
1535     PetscCallMPI(MPI_Comm_size(comm, &size));
1536     PetscCallMPI(MPI_Comm_rank(comm, &rank));
1537     PetscCall(DMGetDimension(dm, &dim));
1538     PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
1539     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
1540     if (name) PetscCall(PetscViewerASCIIPrintf(viewer, "%s in %" PetscInt_FMT " dimension%s:\n", name, dim, dim == 1 ? "" : "s"));
1541     else PetscCall(PetscViewerASCIIPrintf(viewer, "Mesh in %" PetscInt_FMT " dimension%s:\n", dim, dim == 1 ? "" : "s"));
1542     if (cellHeight) PetscCall(PetscViewerASCIIPrintf(viewer, "  Cells are at height %" PetscInt_FMT "\n", cellHeight));
1543     PetscCall(DMPlexGetDepth(dm, &locDepth));
1544     PetscCall(MPIU_Allreduce(&locDepth, &depth, 1, MPIU_INT, MPI_MAX, comm));
1545     PetscCall(DMPlexGetCellTypeStratum(dm, DM_POLYTOPE_FV_GHOST, &gcStart, &gcEnd));
1546     gcNum = gcEnd - gcStart;
1547     if (size < maxSize) PetscCall(PetscCalloc3(size, &sizes, size, &hybsizes, size, &ghostsizes));
1548     else PetscCall(PetscCalloc3(3, &sizes, 3, &hybsizes, 3, &ghostsizes));
1549     for (d = 0; d <= depth; d++) {
1550       PetscInt Nc[2] = {0, 0}, ict;
1551 
1552       PetscCall(DMPlexGetDepthStratum(dm, d, &pStart, &pEnd));
1553       if (pStart < pEnd) PetscCall(DMPlexGetCellType(dm, pStart, &ct0));
1554       ict = ct0;
1555       PetscCallMPI(MPI_Bcast(&ict, 1, MPIU_INT, 0, comm));
1556       ct0 = (DMPolytopeType)ict;
1557       for (p = pStart; p < pEnd; ++p) {
1558         DMPolytopeType ct;
1559 
1560         PetscCall(DMPlexGetCellType(dm, p, &ct));
1561         if (ct == ct0) ++Nc[0];
1562         else ++Nc[1];
1563       }
1564       if (size < maxSize) {
1565         PetscCallMPI(MPI_Gather(&Nc[0], 1, MPIU_INT, sizes, 1, MPIU_INT, 0, comm));
1566         PetscCallMPI(MPI_Gather(&Nc[1], 1, MPIU_INT, hybsizes, 1, MPIU_INT, 0, comm));
1567         if (d == depth) PetscCallMPI(MPI_Gather(&gcNum, 1, MPIU_INT, ghostsizes, 1, MPIU_INT, 0, comm));
1568         PetscCall(PetscViewerASCIIPrintf(viewer, "  Number of %" PetscInt_FMT "-cells per rank:", (depth == 1) && d ? dim : d));
1569         for (p = 0; p < size; ++p) {
1570           if (rank == 0) {
1571             PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT, sizes[p] + hybsizes[p]));
1572             if (hybsizes[p] > 0) PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT ")", hybsizes[p]));
1573             if (ghostsizes[p] > 0) PetscCall(PetscViewerASCIIPrintf(viewer, " [%" PetscInt_FMT "]", ghostsizes[p]));
1574           }
1575         }
1576       } else {
1577         PetscInt locMinMax[2];
1578 
1579         locMinMax[0] = Nc[0] + Nc[1];
1580         locMinMax[1] = Nc[0] + Nc[1];
1581         PetscCall(PetscGlobalMinMaxInt(comm, locMinMax, sizes));
1582         locMinMax[0] = Nc[1];
1583         locMinMax[1] = Nc[1];
1584         PetscCall(PetscGlobalMinMaxInt(comm, locMinMax, hybsizes));
1585         if (d == depth) {
1586           locMinMax[0] = gcNum;
1587           locMinMax[1] = gcNum;
1588           PetscCall(PetscGlobalMinMaxInt(comm, locMinMax, ghostsizes));
1589         }
1590         PetscCall(PetscViewerASCIIPrintf(viewer, "  Min/Max of %" PetscInt_FMT "-cells per rank:", (depth == 1) && d ? dim : d));
1591         PetscCall(PetscViewerASCIIPrintf(viewer, " %" PetscInt_FMT "/%" PetscInt_FMT, sizes[0], sizes[1]));
1592         if (hybsizes[0] > 0) PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT "/%" PetscInt_FMT ")", hybsizes[0], hybsizes[1]));
1593         if (ghostsizes[0] > 0) PetscCall(PetscViewerASCIIPrintf(viewer, " [%" PetscInt_FMT "/%" PetscInt_FMT "]", ghostsizes[0], ghostsizes[1]));
1594       }
1595       PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
1596     }
1597     PetscCall(PetscFree3(sizes, hybsizes, ghostsizes));
1598     {
1599       const PetscReal *maxCell;
1600       const PetscReal *L;
1601       PetscBool        localized;
1602 
1603       PetscCall(DMGetPeriodicity(dm, &maxCell, NULL, &L));
1604       PetscCall(DMGetCoordinatesLocalized(dm, &localized));
1605       if (L || localized) {
1606         PetscCall(PetscViewerASCIIPrintf(viewer, "Periodic mesh"));
1607         PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_FALSE));
1608         if (L) {
1609           PetscCall(PetscViewerASCIIPrintf(viewer, " ("));
1610           for (d = 0; d < dim; ++d) {
1611             if (d > 0) PetscCall(PetscViewerASCIIPrintf(viewer, ", "));
1612             PetscCall(PetscViewerASCIIPrintf(viewer, "%s", L[d] > 0.0 ? "PERIODIC" : "NONE"));
1613           }
1614           PetscCall(PetscViewerASCIIPrintf(viewer, ")"));
1615         }
1616         PetscCall(PetscViewerASCIIPrintf(viewer, " coordinates %s\n", localized ? "localized" : "not localized"));
1617         PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_TRUE));
1618       }
1619     }
1620     PetscCall(DMGetNumLabels(dm, &numLabels));
1621     if (numLabels) PetscCall(PetscViewerASCIIPrintf(viewer, "Labels:\n"));
1622     for (l = 0; l < numLabels; ++l) {
1623       DMLabel         label;
1624       const char     *name;
1625       IS              valueIS;
1626       const PetscInt *values;
1627       PetscInt        numValues, v;
1628 
1629       PetscCall(DMGetLabelName(dm, l, &name));
1630       PetscCall(DMGetLabel(dm, name, &label));
1631       PetscCall(DMLabelGetNumValues(label, &numValues));
1632       PetscCall(PetscViewerASCIIPrintf(viewer, "  %s: %" PetscInt_FMT " strata with value/size (", name, numValues));
1633       PetscCall(DMLabelGetValueIS(label, &valueIS));
1634       PetscCall(ISGetIndices(valueIS, &values));
1635       PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_FALSE));
1636       for (v = 0; v < numValues; ++v) {
1637         PetscInt size;
1638 
1639         PetscCall(DMLabelGetStratumSize(label, values[v], &size));
1640         if (v > 0) PetscCall(PetscViewerASCIIPrintf(viewer, ", "));
1641         PetscCall(PetscViewerASCIIPrintf(viewer, "%" PetscInt_FMT " (%" PetscInt_FMT ")", values[v], size));
1642       }
1643       PetscCall(PetscViewerASCIIPrintf(viewer, ")\n"));
1644       PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_TRUE));
1645       PetscCall(ISRestoreIndices(valueIS, &values));
1646       PetscCall(ISDestroy(&valueIS));
1647     }
1648     {
1649       char    **labelNames;
1650       PetscInt  Nl = numLabels;
1651       PetscBool flg;
1652 
1653       PetscCall(PetscMalloc1(Nl, &labelNames));
1654       PetscCall(PetscOptionsGetStringArray(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_view_labels", labelNames, &Nl, &flg));
1655       for (l = 0; l < Nl; ++l) {
1656         DMLabel label;
1657 
1658         PetscCall(DMHasLabel(dm, labelNames[l], &flg));
1659         if (flg) {
1660           PetscCall(DMGetLabel(dm, labelNames[l], &label));
1661           PetscCall(DMLabelView(label, viewer));
1662         }
1663         PetscCall(PetscFree(labelNames[l]));
1664       }
1665       PetscCall(PetscFree(labelNames));
1666     }
1667     /* If no fields are specified, people do not want to see adjacency */
1668     if (dm->Nf) {
1669       PetscInt f;
1670 
1671       for (f = 0; f < dm->Nf; ++f) {
1672         const char *name;
1673 
1674         PetscCall(PetscObjectGetName(dm->fields[f].disc, &name));
1675         if (numLabels) PetscCall(PetscViewerASCIIPrintf(viewer, "Field %s:\n", name));
1676         PetscCall(PetscViewerASCIIPushTab(viewer));
1677         if (dm->fields[f].label) PetscCall(DMLabelView(dm->fields[f].label, viewer));
1678         if (dm->fields[f].adjacency[0]) {
1679           if (dm->fields[f].adjacency[1]) PetscCall(PetscViewerASCIIPrintf(viewer, "adjacency FVM++\n"));
1680           else PetscCall(PetscViewerASCIIPrintf(viewer, "adjacency FVM\n"));
1681         } else {
1682           if (dm->fields[f].adjacency[1]) PetscCall(PetscViewerASCIIPrintf(viewer, "adjacency FEM\n"));
1683           else PetscCall(PetscViewerASCIIPrintf(viewer, "adjacency FUNKY\n"));
1684         }
1685         PetscCall(PetscViewerASCIIPopTab(viewer));
1686       }
1687     }
1688     PetscCall(DMGetCoarseDM(dm, &cdm));
1689     if (cdm) {
1690       PetscCall(PetscViewerASCIIPushTab(viewer));
1691       PetscCall(PetscViewerASCIIPrintf(viewer, "Defined by transform from:\n"));
1692       PetscCall(DMPlexView_Ascii(cdm, viewer));
1693       PetscCall(PetscViewerASCIIPopTab(viewer));
1694     }
1695   }
1696   PetscFunctionReturn(PETSC_SUCCESS);
1697 }
1698 
1699 static PetscErrorCode DMPlexDrawCell(DM dm, PetscDraw draw, PetscInt cell, const PetscScalar coords[])
1700 {
1701   DMPolytopeType ct;
1702   PetscMPIInt    rank;
1703   PetscInt       cdim;
1704 
1705   PetscFunctionBegin;
1706   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1707   PetscCall(DMPlexGetCellType(dm, cell, &ct));
1708   PetscCall(DMGetCoordinateDim(dm, &cdim));
1709   switch (ct) {
1710   case DM_POLYTOPE_SEGMENT:
1711   case DM_POLYTOPE_POINT_PRISM_TENSOR:
1712     switch (cdim) {
1713     case 1: {
1714       const PetscReal y  = 0.5;  /* TODO Put it in the middle of the viewport */
1715       const PetscReal dy = 0.05; /* TODO Make it a fraction of the total length */
1716 
1717       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), y, PetscRealPart(coords[1]), y, PETSC_DRAW_BLACK));
1718       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), y + dy, PetscRealPart(coords[0]), y - dy, PETSC_DRAW_BLACK));
1719       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[1]), y + dy, PetscRealPart(coords[1]), y - dy, PETSC_DRAW_BLACK));
1720     } break;
1721     case 2: {
1722       const PetscReal dx = (PetscRealPart(coords[3]) - PetscRealPart(coords[1]));
1723       const PetscReal dy = (PetscRealPart(coords[2]) - PetscRealPart(coords[0]));
1724       const PetscReal l  = 0.1 / PetscSqrtReal(dx * dx + dy * dy);
1725 
1726       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PETSC_DRAW_BLACK));
1727       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]) + l * dx, PetscRealPart(coords[1]) + l * dy, PetscRealPart(coords[0]) - l * dx, PetscRealPart(coords[1]) - l * dy, PETSC_DRAW_BLACK));
1728       PetscCall(PetscDrawLine(draw, PetscRealPart(coords[2]) + l * dx, PetscRealPart(coords[3]) + l * dy, PetscRealPart(coords[2]) - l * dx, PetscRealPart(coords[3]) - l * dy, PETSC_DRAW_BLACK));
1729     } break;
1730     default:
1731       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot draw cells of dimension %" PetscInt_FMT, cdim);
1732     }
1733     break;
1734   case DM_POLYTOPE_TRIANGLE:
1735     PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2));
1736     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PETSC_DRAW_BLACK));
1737     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_BLACK));
1738     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[4]), PetscRealPart(coords[5]), PetscRealPart(coords[0]), PetscRealPart(coords[1]), PETSC_DRAW_BLACK));
1739     break;
1740   case DM_POLYTOPE_QUADRILATERAL:
1741     PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2));
1742     PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PetscRealPart(coords[6]), PetscRealPart(coords[7]), PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2));
1743     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PETSC_DRAW_BLACK));
1744     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_BLACK));
1745     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[4]), PetscRealPart(coords[5]), PetscRealPart(coords[6]), PetscRealPart(coords[7]), PETSC_DRAW_BLACK));
1746     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[6]), PetscRealPart(coords[7]), PetscRealPart(coords[0]), PetscRealPart(coords[1]), PETSC_DRAW_BLACK));
1747     break;
1748   case DM_POLYTOPE_SEG_PRISM_TENSOR:
1749     PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2));
1750     PetscCall(PetscDrawTriangle(draw, PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[6]), PetscRealPart(coords[7]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2, PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2));
1751     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[0]), PetscRealPart(coords[1]), PetscRealPart(coords[2]), PetscRealPart(coords[3]), PETSC_DRAW_BLACK));
1752     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[2]), PetscRealPart(coords[3]), PetscRealPart(coords[6]), PetscRealPart(coords[7]), PETSC_DRAW_BLACK));
1753     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[6]), PetscRealPart(coords[7]), PetscRealPart(coords[4]), PetscRealPart(coords[5]), PETSC_DRAW_BLACK));
1754     PetscCall(PetscDrawLine(draw, PetscRealPart(coords[4]), PetscRealPart(coords[5]), PetscRealPart(coords[0]), PetscRealPart(coords[1]), PETSC_DRAW_BLACK));
1755     break;
1756   case DM_POLYTOPE_FV_GHOST:
1757     break;
1758   default:
1759     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot draw cells of type %s", DMPolytopeTypes[ct]);
1760   }
1761   PetscFunctionReturn(PETSC_SUCCESS);
1762 }
1763 
1764 static PetscErrorCode DrawPolygon_Private(DM dm, PetscDraw draw, PetscInt cell, PetscInt Nv, const PetscReal refVertices[], const PetscScalar coords[], PetscInt edgeDiv, PetscReal refCoords[], PetscReal edgeCoords[])
1765 {
1766   PetscReal   centroid[2] = {0., 0.};
1767   PetscMPIInt rank;
1768   PetscInt    fillColor;
1769 
1770   PetscFunctionBegin;
1771   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
1772   fillColor = PETSC_DRAW_WHITE + rank % (PETSC_DRAW_BASIC_COLORS - 2) + 2;
1773   for (PetscInt v = 0; v < Nv; ++v) {
1774     centroid[0] += PetscRealPart(coords[v * 2 + 0]) / Nv;
1775     centroid[1] += PetscRealPart(coords[v * 2 + 1]) / Nv;
1776   }
1777   for (PetscInt e = 0; e < Nv; ++e) {
1778     refCoords[0] = refVertices[e * 2 + 0];
1779     refCoords[1] = refVertices[e * 2 + 1];
1780     for (PetscInt d = 1; d <= edgeDiv; ++d) {
1781       refCoords[d * 2 + 0] = refCoords[0] + (refVertices[(e + 1) % Nv * 2 + 0] - refCoords[0]) * d / edgeDiv;
1782       refCoords[d * 2 + 1] = refCoords[1] + (refVertices[(e + 1) % Nv * 2 + 1] - refCoords[1]) * d / edgeDiv;
1783     }
1784     PetscCall(DMPlexReferenceToCoordinates(dm, cell, edgeDiv + 1, refCoords, edgeCoords));
1785     for (PetscInt d = 0; d < edgeDiv; ++d) {
1786       PetscCall(PetscDrawTriangle(draw, centroid[0], centroid[1], edgeCoords[d * 2 + 0], edgeCoords[d * 2 + 1], edgeCoords[(d + 1) * 2 + 0], edgeCoords[(d + 1) * 2 + 1], fillColor, fillColor, fillColor));
1787       PetscCall(PetscDrawLine(draw, edgeCoords[d * 2 + 0], edgeCoords[d * 2 + 1], edgeCoords[(d + 1) * 2 + 0], edgeCoords[(d + 1) * 2 + 1], PETSC_DRAW_BLACK));
1788     }
1789   }
1790   PetscFunctionReturn(PETSC_SUCCESS);
1791 }
1792 
1793 static PetscErrorCode DMPlexDrawCellHighOrder(DM dm, PetscDraw draw, PetscInt cell, const PetscScalar coords[], PetscInt edgeDiv, PetscReal refCoords[], PetscReal edgeCoords[])
1794 {
1795   DMPolytopeType ct;
1796 
1797   PetscFunctionBegin;
1798   PetscCall(DMPlexGetCellType(dm, cell, &ct));
1799   switch (ct) {
1800   case DM_POLYTOPE_TRIANGLE: {
1801     PetscReal refVertices[6] = {-1., -1., 1., -1., -1., 1.};
1802 
1803     PetscCall(DrawPolygon_Private(dm, draw, cell, 3, refVertices, coords, edgeDiv, refCoords, edgeCoords));
1804   } break;
1805   case DM_POLYTOPE_QUADRILATERAL: {
1806     PetscReal refVertices[8] = {-1., -1., 1., -1., 1., 1., -1., 1.};
1807 
1808     PetscCall(DrawPolygon_Private(dm, draw, cell, 4, refVertices, coords, edgeDiv, refCoords, edgeCoords));
1809   } break;
1810   default:
1811     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot draw cells of type %s", DMPolytopeTypes[ct]);
1812   }
1813   PetscFunctionReturn(PETSC_SUCCESS);
1814 }
1815 
1816 static PetscErrorCode DMPlexView_Draw(DM dm, PetscViewer viewer)
1817 {
1818   PetscDraw    draw;
1819   DM           cdm;
1820   PetscSection coordSection;
1821   Vec          coordinates;
1822   PetscReal    xyl[3], xyr[3];
1823   PetscReal   *refCoords, *edgeCoords;
1824   PetscBool    isnull, drawAffine;
1825   PetscInt     dim, vStart, vEnd, cStart, cEnd, c, cDegree, edgeDiv;
1826 
1827   PetscFunctionBegin;
1828   PetscCall(DMGetCoordinateDim(dm, &dim));
1829   PetscCheck(dim <= 2, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Cannot draw meshes of dimension %" PetscInt_FMT, dim);
1830   PetscCall(DMGetCoordinateDegree_Internal(dm, &cDegree));
1831   drawAffine = cDegree > 1 ? PETSC_FALSE : PETSC_TRUE;
1832   edgeDiv    = cDegree + 1;
1833   PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_view_draw_affine", &drawAffine, NULL));
1834   if (!drawAffine) PetscCall(PetscMalloc2((edgeDiv + 1) * dim, &refCoords, (edgeDiv + 1) * dim, &edgeCoords));
1835   PetscCall(DMGetCoordinateDM(dm, &cdm));
1836   PetscCall(DMGetLocalSection(cdm, &coordSection));
1837   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
1838   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
1839   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
1840 
1841   PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
1842   PetscCall(PetscDrawIsNull(draw, &isnull));
1843   if (isnull) PetscFunctionReturn(PETSC_SUCCESS);
1844   PetscCall(PetscDrawSetTitle(draw, "Mesh"));
1845 
1846   PetscCall(DMGetBoundingBox(dm, xyl, xyr));
1847   PetscCall(PetscDrawSetCoordinates(draw, xyl[0], xyl[1], xyr[0], xyr[1]));
1848   PetscCall(PetscDrawClear(draw));
1849 
1850   for (c = cStart; c < cEnd; ++c) {
1851     PetscScalar       *coords = NULL;
1852     const PetscScalar *coords_arr;
1853     PetscInt           numCoords;
1854     PetscBool          isDG;
1855 
1856     PetscCall(DMPlexGetCellCoordinates(dm, c, &isDG, &numCoords, &coords_arr, &coords));
1857     if (drawAffine) PetscCall(DMPlexDrawCell(dm, draw, c, coords));
1858     else PetscCall(DMPlexDrawCellHighOrder(dm, draw, c, coords, edgeDiv, refCoords, edgeCoords));
1859     PetscCall(DMPlexRestoreCellCoordinates(dm, c, &isDG, &numCoords, &coords_arr, &coords));
1860   }
1861   if (!drawAffine) PetscCall(PetscFree2(refCoords, edgeCoords));
1862   PetscCall(PetscDrawFlush(draw));
1863   PetscCall(PetscDrawPause(draw));
1864   PetscCall(PetscDrawSave(draw));
1865   PetscFunctionReturn(PETSC_SUCCESS);
1866 }
1867 
1868 static PetscErrorCode DMPlexCreateHighOrderSurrogate_Internal(DM dm, DM *hdm)
1869 {
1870   DM           odm = dm, rdm = dm, cdm;
1871   PetscFE      fe;
1872   PetscSpace   sp;
1873   PetscClassId id;
1874   PetscInt     degree;
1875   PetscBool    hoView = PETSC_TRUE;
1876 
1877   PetscFunctionBegin;
1878   PetscObjectOptionsBegin((PetscObject)dm);
1879   PetscCall(PetscOptionsBool("-dm_plex_high_order_view", "Subsample to view meshes with high order coordinates", "DMPlexCreateHighOrderSurrogate_Internal", hoView, &hoView, NULL));
1880   PetscOptionsEnd();
1881   PetscCall(PetscObjectReference((PetscObject)dm));
1882   *hdm = dm;
1883   if (!hoView) PetscFunctionReturn(PETSC_SUCCESS);
1884   PetscCall(DMGetCoordinateDM(dm, &cdm));
1885   PetscCall(DMGetField(cdm, 0, NULL, (PetscObject *)&fe));
1886   PetscCall(PetscObjectGetClassId((PetscObject)fe, &id));
1887   if (id != PETSCFE_CLASSID) PetscFunctionReturn(PETSC_SUCCESS);
1888   PetscCall(PetscFEGetBasisSpace(fe, &sp));
1889   PetscCall(PetscSpaceGetDegree(sp, &degree, NULL));
1890   for (PetscInt r = 0, rd = PetscCeilReal(((PetscReal)degree) / 2.); r < (PetscInt)PetscCeilReal(PetscLog2Real(degree)); ++r, rd = PetscCeilReal(((PetscReal)rd) / 2.)) {
1891     DM  cdm, rcdm;
1892     Mat In;
1893     Vec cl, rcl;
1894 
1895     PetscCall(DMRefine(odm, PetscObjectComm((PetscObject)odm), &rdm));
1896     PetscCall(DMPlexCreateCoordinateSpace(rdm, rd, PETSC_FALSE, NULL));
1897     PetscCall(PetscObjectSetName((PetscObject)rdm, "Refined Mesh with Linear Coordinates"));
1898     PetscCall(DMGetCoordinateDM(odm, &cdm));
1899     PetscCall(DMGetCoordinateDM(rdm, &rcdm));
1900     PetscCall(DMGetCoordinatesLocal(odm, &cl));
1901     PetscCall(DMGetCoordinatesLocal(rdm, &rcl));
1902     PetscCall(DMSetCoarseDM(rcdm, cdm));
1903     PetscCall(DMCreateInterpolation(cdm, rcdm, &In, NULL));
1904     PetscCall(MatMult(In, cl, rcl));
1905     PetscCall(MatDestroy(&In));
1906     PetscCall(DMSetCoordinatesLocal(rdm, rcl));
1907     PetscCall(DMDestroy(&odm));
1908     odm = rdm;
1909   }
1910   *hdm = rdm;
1911   PetscFunctionReturn(PETSC_SUCCESS);
1912 }
1913 
1914 #if defined(PETSC_HAVE_EXODUSII)
1915   #include <exodusII.h>
1916   #include <petscviewerexodusii.h>
1917 #endif
1918 
1919 PetscErrorCode DMView_Plex(DM dm, PetscViewer viewer)
1920 {
1921   PetscBool iascii, ishdf5, isvtk, isdraw, flg, isglvis, isexodus, iscgns;
1922   char      name[PETSC_MAX_PATH_LEN];
1923 
1924   PetscFunctionBegin;
1925   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1926   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1927   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
1928   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
1929   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
1930   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw));
1931   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERGLVIS, &isglvis));
1932   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWEREXODUSII, &isexodus));
1933   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERCGNS, &iscgns));
1934   if (iascii) {
1935     PetscViewerFormat format;
1936     PetscCall(PetscViewerGetFormat(viewer, &format));
1937     if (format == PETSC_VIEWER_ASCII_GLVIS) PetscCall(DMPlexView_GLVis(dm, viewer));
1938     else PetscCall(DMPlexView_Ascii(dm, viewer));
1939   } else if (ishdf5) {
1940 #if defined(PETSC_HAVE_HDF5)
1941     PetscCall(DMPlexView_HDF5_Internal(dm, viewer));
1942 #else
1943     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
1944 #endif
1945   } else if (isvtk) {
1946     PetscCall(DMPlexVTKWriteAll((PetscObject)dm, viewer));
1947   } else if (isdraw) {
1948     DM hdm;
1949 
1950     PetscCall(DMPlexCreateHighOrderSurrogate_Internal(dm, &hdm));
1951     PetscCall(DMPlexView_Draw(hdm, viewer));
1952     PetscCall(DMDestroy(&hdm));
1953   } else if (isglvis) {
1954     PetscCall(DMPlexView_GLVis(dm, viewer));
1955 #if defined(PETSC_HAVE_EXODUSII)
1956   } else if (isexodus) {
1957     /*
1958       exodusII requires that all sets be part of exactly one cell set.
1959       If the dm does not have a "Cell Sets" label defined, we create one
1960       with ID 1, containing all cells.
1961       Note that if the Cell Sets label is defined but does not cover all cells,
1962       we may still have a problem. This should probably be checked here or in the viewer;
1963     */
1964     PetscInt numCS;
1965     PetscCall(DMGetLabelSize(dm, "Cell Sets", &numCS));
1966     if (!numCS) {
1967       PetscInt cStart, cEnd, c;
1968       PetscCall(DMCreateLabel(dm, "Cell Sets"));
1969       PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
1970       for (c = cStart; c < cEnd; ++c) PetscCall(DMSetLabelValue(dm, "Cell Sets", c, 1));
1971     }
1972     PetscCall(DMView_PlexExodusII(dm, viewer));
1973 #endif
1974 #if defined(PETSC_HAVE_CGNS)
1975   } else if (iscgns) {
1976     PetscCall(DMView_PlexCGNS(dm, viewer));
1977 #endif
1978   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Viewer type %s not yet supported for DMPlex writing", ((PetscObject)viewer)->type_name);
1979   /* Optionally view the partition */
1980   PetscCall(PetscOptionsHasName(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_partition_view", &flg));
1981   if (flg) {
1982     Vec ranks;
1983     PetscCall(DMPlexCreateRankField(dm, &ranks));
1984     PetscCall(VecView(ranks, viewer));
1985     PetscCall(VecDestroy(&ranks));
1986   }
1987   /* Optionally view a label */
1988   PetscCall(PetscOptionsGetString(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_label_view", name, sizeof(name), &flg));
1989   if (flg) {
1990     DMLabel label;
1991     Vec     val;
1992 
1993     PetscCall(DMGetLabel(dm, name, &label));
1994     PetscCheck(label, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Label %s provided to -dm_label_view does not exist in this DM", name);
1995     PetscCall(DMPlexCreateLabelField(dm, label, &val));
1996     PetscCall(VecView(val, viewer));
1997     PetscCall(VecDestroy(&val));
1998   }
1999   PetscFunctionReturn(PETSC_SUCCESS);
2000 }
2001 
2002 /*@
2003   DMPlexTopologyView - Saves a `DMPLEX` topology into a file
2004 
2005   Collective
2006 
2007   Input Parameters:
2008 + dm     - The `DM` whose topology is to be saved
2009 - viewer - The `PetscViewer` to save it in
2010 
2011   Level: advanced
2012 
2013 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`, `DMPlexCoordinatesView()`, `DMPlexLabelsView()`, `DMPlexTopologyLoad()`, `PetscViewer`
2014 @*/
2015 PetscErrorCode DMPlexTopologyView(DM dm, PetscViewer viewer)
2016 {
2017   PetscBool ishdf5;
2018 
2019   PetscFunctionBegin;
2020   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2021   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2022   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2023   PetscCall(PetscLogEventBegin(DMPLEX_TopologyView, viewer, 0, 0, 0));
2024   if (ishdf5) {
2025 #if defined(PETSC_HAVE_HDF5)
2026     PetscViewerFormat format;
2027     PetscCall(PetscViewerGetFormat(viewer, &format));
2028     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2029       IS globalPointNumbering;
2030 
2031       PetscCall(DMPlexCreatePointNumbering(dm, &globalPointNumbering));
2032       PetscCall(DMPlexTopologyView_HDF5_Internal(dm, globalPointNumbering, viewer));
2033       PetscCall(ISDestroy(&globalPointNumbering));
2034     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 output.", PetscViewerFormats[format]);
2035 #else
2036     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2037 #endif
2038   }
2039   PetscCall(PetscLogEventEnd(DMPLEX_TopologyView, viewer, 0, 0, 0));
2040   PetscFunctionReturn(PETSC_SUCCESS);
2041 }
2042 
2043 /*@
2044   DMPlexCoordinatesView - Saves `DMPLEX` coordinates into a file
2045 
2046   Collective
2047 
2048   Input Parameters:
2049 + dm     - The `DM` whose coordinates are to be saved
2050 - viewer - The `PetscViewer` for saving
2051 
2052   Level: advanced
2053 
2054 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`, `DMPlexTopologyView()`, `DMPlexLabelsView()`, `DMPlexCoordinatesLoad()`, `PetscViewer`
2055 @*/
2056 PetscErrorCode DMPlexCoordinatesView(DM dm, PetscViewer viewer)
2057 {
2058   PetscBool ishdf5;
2059 
2060   PetscFunctionBegin;
2061   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2062   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2063   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2064   PetscCall(PetscLogEventBegin(DMPLEX_CoordinatesView, viewer, 0, 0, 0));
2065   if (ishdf5) {
2066 #if defined(PETSC_HAVE_HDF5)
2067     PetscViewerFormat format;
2068     PetscCall(PetscViewerGetFormat(viewer, &format));
2069     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2070       PetscCall(DMPlexCoordinatesView_HDF5_Internal(dm, viewer));
2071     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 output.", PetscViewerFormats[format]);
2072 #else
2073     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2074 #endif
2075   }
2076   PetscCall(PetscLogEventEnd(DMPLEX_CoordinatesView, viewer, 0, 0, 0));
2077   PetscFunctionReturn(PETSC_SUCCESS);
2078 }
2079 
2080 /*@
2081   DMPlexLabelsView - Saves `DMPLEX` labels into a file
2082 
2083   Collective
2084 
2085   Input Parameters:
2086 + dm     - The `DM` whose labels are to be saved
2087 - viewer - The `PetscViewer` for saving
2088 
2089   Level: advanced
2090 
2091 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`, `DMPlexTopologyView()`, `DMPlexCoordinatesView()`, `DMPlexLabelsLoad()`, `PetscViewer`
2092 @*/
2093 PetscErrorCode DMPlexLabelsView(DM dm, PetscViewer viewer)
2094 {
2095   PetscBool ishdf5;
2096 
2097   PetscFunctionBegin;
2098   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2099   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2100   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2101   PetscCall(PetscLogEventBegin(DMPLEX_LabelsView, viewer, 0, 0, 0));
2102   if (ishdf5) {
2103 #if defined(PETSC_HAVE_HDF5)
2104     IS                globalPointNumbering;
2105     PetscViewerFormat format;
2106 
2107     PetscCall(PetscViewerGetFormat(viewer, &format));
2108     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2109       PetscCall(DMPlexCreatePointNumbering(dm, &globalPointNumbering));
2110       PetscCall(DMPlexLabelsView_HDF5_Internal(dm, globalPointNumbering, viewer));
2111       PetscCall(ISDestroy(&globalPointNumbering));
2112     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 input.", PetscViewerFormats[format]);
2113 #else
2114     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2115 #endif
2116   }
2117   PetscCall(PetscLogEventEnd(DMPLEX_LabelsView, viewer, 0, 0, 0));
2118   PetscFunctionReturn(PETSC_SUCCESS);
2119 }
2120 
2121 /*@
2122   DMPlexSectionView - Saves a section associated with a `DMPLEX`
2123 
2124   Collective
2125 
2126   Input Parameters:
2127 + dm        - The `DM` that contains the topology on which the section to be saved is defined
2128 . viewer    - The `PetscViewer` for saving
2129 - sectiondm - The `DM` that contains the section to be saved, can be `NULL`
2130 
2131   Level: advanced
2132 
2133   Notes:
2134   This function is a wrapper around `PetscSectionView()`; in addition to the raw section, it saves information that associates the section points to the topology (`dm`) points. When the topology (`dm`) and the section are later loaded with `DMPlexTopologyLoad()` and `DMPlexSectionLoad()`, respectively, this information is used to match section points with topology points.
2135 
2136   In general `dm` and `sectiondm` are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2137 
2138 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`, `DMPlexTopologyView()`, `DMPlexCoordinatesView()`, `DMPlexLabelsView()`, `DMPlexGlobalVectorView()`, `DMPlexLocalVectorView()`, `PetscSectionView()`, `DMPlexSectionLoad()`, `PetscViewer`
2139 @*/
2140 PetscErrorCode DMPlexSectionView(DM dm, PetscViewer viewer, DM sectiondm)
2141 {
2142   PetscBool ishdf5;
2143 
2144   PetscFunctionBegin;
2145   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2146   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2147   if (!sectiondm) sectiondm = dm;
2148   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2149   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2150   PetscCall(PetscLogEventBegin(DMPLEX_SectionView, viewer, 0, 0, 0));
2151   if (ishdf5) {
2152 #if defined(PETSC_HAVE_HDF5)
2153     PetscCall(DMPlexSectionView_HDF5_Internal(dm, viewer, sectiondm));
2154 #else
2155     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2156 #endif
2157   }
2158   PetscCall(PetscLogEventEnd(DMPLEX_SectionView, viewer, 0, 0, 0));
2159   PetscFunctionReturn(PETSC_SUCCESS);
2160 }
2161 
2162 /*@
2163   DMPlexGlobalVectorView - Saves a global vector
2164 
2165   Collective
2166 
2167   Input Parameters:
2168 + dm        - The `DM` that represents the topology
2169 . viewer    - The `PetscViewer` to save data with
2170 . sectiondm - The `DM` that contains the global section on which vec is defined, can be `NULL`
2171 - vec       - The global vector to be saved
2172 
2173   Level: advanced
2174 
2175   Notes:
2176   In general `dm` and `sectiondm` are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2177 
2178   Calling sequence:
2179 .vb
2180        DMCreate(PETSC_COMM_WORLD, &dm);
2181        DMSetType(dm, DMPLEX);
2182        PetscObjectSetName((PetscObject)dm, "topologydm_name");
2183        DMClone(dm, &sectiondm);
2184        PetscObjectSetName((PetscObject)sectiondm, "sectiondm_name");
2185        PetscSectionCreate(PETSC_COMM_WORLD, &section);
2186        DMPlexGetChart(sectiondm, &pStart, &pEnd);
2187        PetscSectionSetChart(section, pStart, pEnd);
2188        PetscSectionSetUp(section);
2189        DMSetLocalSection(sectiondm, section);
2190        PetscSectionDestroy(&section);
2191        DMGetGlobalVector(sectiondm, &vec);
2192        PetscObjectSetName((PetscObject)vec, "vec_name");
2193        DMPlexTopologyView(dm, viewer);
2194        DMPlexSectionView(dm, viewer, sectiondm);
2195        DMPlexGlobalVectorView(dm, viewer, sectiondm, vec);
2196        DMRestoreGlobalVector(sectiondm, &vec);
2197        DMDestroy(&sectiondm);
2198        DMDestroy(&dm);
2199 .ve
2200 
2201 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTopologyView()`, `DMPlexSectionView()`, `DMPlexLocalVectorView()`, `DMPlexGlobalVectorLoad()`, `DMPlexLocalVectorLoad()`
2202 @*/
2203 PetscErrorCode DMPlexGlobalVectorView(DM dm, PetscViewer viewer, DM sectiondm, Vec vec)
2204 {
2205   PetscBool ishdf5;
2206 
2207   PetscFunctionBegin;
2208   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2209   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2210   if (!sectiondm) sectiondm = dm;
2211   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2212   PetscValidHeaderSpecific(vec, VEC_CLASSID, 4);
2213   /* Check consistency */
2214   {
2215     PetscSection section;
2216     PetscBool    includesConstraints;
2217     PetscInt     m, m1;
2218 
2219     PetscCall(VecGetLocalSize(vec, &m1));
2220     PetscCall(DMGetGlobalSection(sectiondm, &section));
2221     PetscCall(PetscSectionGetIncludesConstraints(section, &includesConstraints));
2222     if (includesConstraints) PetscCall(PetscSectionGetStorageSize(section, &m));
2223     else PetscCall(PetscSectionGetConstrainedStorageSize(section, &m));
2224     PetscCheck(m1 == m, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Global vector size (%" PetscInt_FMT ") != global section storage size (%" PetscInt_FMT ")", m1, m);
2225   }
2226   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2227   PetscCall(PetscLogEventBegin(DMPLEX_GlobalVectorView, viewer, 0, 0, 0));
2228   if (ishdf5) {
2229 #if defined(PETSC_HAVE_HDF5)
2230     PetscCall(DMPlexGlobalVectorView_HDF5_Internal(dm, viewer, sectiondm, vec));
2231 #else
2232     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2233 #endif
2234   }
2235   PetscCall(PetscLogEventEnd(DMPLEX_GlobalVectorView, viewer, 0, 0, 0));
2236   PetscFunctionReturn(PETSC_SUCCESS);
2237 }
2238 
2239 /*@
2240   DMPlexLocalVectorView - Saves a local vector
2241 
2242   Collective
2243 
2244   Input Parameters:
2245 + dm        - The `DM` that represents the topology
2246 . viewer    - The `PetscViewer` to save data with
2247 . sectiondm - The `DM` that contains the local section on which `vec` is defined, can be `NULL`
2248 - vec       - The local vector to be saved
2249 
2250   Level: advanced
2251 
2252   Note:
2253   In general `dm` and `sectiondm` are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2254 
2255   Calling sequence:
2256 .vb
2257        DMCreate(PETSC_COMM_WORLD, &dm);
2258        DMSetType(dm, DMPLEX);
2259        PetscObjectSetName((PetscObject)dm, "topologydm_name");
2260        DMClone(dm, &sectiondm);
2261        PetscObjectSetName((PetscObject)sectiondm, "sectiondm_name");
2262        PetscSectionCreate(PETSC_COMM_WORLD, &section);
2263        DMPlexGetChart(sectiondm, &pStart, &pEnd);
2264        PetscSectionSetChart(section, pStart, pEnd);
2265        PetscSectionSetUp(section);
2266        DMSetLocalSection(sectiondm, section);
2267        DMGetLocalVector(sectiondm, &vec);
2268        PetscObjectSetName((PetscObject)vec, "vec_name");
2269        DMPlexTopologyView(dm, viewer);
2270        DMPlexSectionView(dm, viewer, sectiondm);
2271        DMPlexLocalVectorView(dm, viewer, sectiondm, vec);
2272        DMRestoreLocalVector(sectiondm, &vec);
2273        DMDestroy(&sectiondm);
2274        DMDestroy(&dm);
2275 .ve
2276 
2277 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTopologyView()`, `DMPlexSectionView()`, `DMPlexGlobalVectorView()`, `DMPlexGlobalVectorLoad()`, `DMPlexLocalVectorLoad()`
2278 @*/
2279 PetscErrorCode DMPlexLocalVectorView(DM dm, PetscViewer viewer, DM sectiondm, Vec vec)
2280 {
2281   PetscBool ishdf5;
2282 
2283   PetscFunctionBegin;
2284   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2285   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2286   if (!sectiondm) sectiondm = dm;
2287   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2288   PetscValidHeaderSpecific(vec, VEC_CLASSID, 4);
2289   /* Check consistency */
2290   {
2291     PetscSection section;
2292     PetscBool    includesConstraints;
2293     PetscInt     m, m1;
2294 
2295     PetscCall(VecGetLocalSize(vec, &m1));
2296     PetscCall(DMGetLocalSection(sectiondm, &section));
2297     PetscCall(PetscSectionGetIncludesConstraints(section, &includesConstraints));
2298     if (includesConstraints) PetscCall(PetscSectionGetStorageSize(section, &m));
2299     else PetscCall(PetscSectionGetConstrainedStorageSize(section, &m));
2300     PetscCheck(m1 == m, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Local vector size (%" PetscInt_FMT ") != local section storage size (%" PetscInt_FMT ")", m1, m);
2301   }
2302   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2303   PetscCall(PetscLogEventBegin(DMPLEX_LocalVectorView, viewer, 0, 0, 0));
2304   if (ishdf5) {
2305 #if defined(PETSC_HAVE_HDF5)
2306     PetscCall(DMPlexLocalVectorView_HDF5_Internal(dm, viewer, sectiondm, vec));
2307 #else
2308     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2309 #endif
2310   }
2311   PetscCall(PetscLogEventEnd(DMPLEX_LocalVectorView, viewer, 0, 0, 0));
2312   PetscFunctionReturn(PETSC_SUCCESS);
2313 }
2314 
2315 PetscErrorCode DMLoad_Plex(DM dm, PetscViewer viewer)
2316 {
2317   PetscBool ishdf5;
2318 
2319   PetscFunctionBegin;
2320   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2321   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2322   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2323   if (ishdf5) {
2324 #if defined(PETSC_HAVE_HDF5)
2325     PetscViewerFormat format;
2326     PetscCall(PetscViewerGetFormat(viewer, &format));
2327     if (format == PETSC_VIEWER_HDF5_XDMF || format == PETSC_VIEWER_HDF5_VIZ) {
2328       PetscCall(DMPlexLoad_HDF5_Xdmf_Internal(dm, viewer));
2329     } else if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2330       PetscCall(DMPlexLoad_HDF5_Internal(dm, viewer));
2331     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 input.", PetscViewerFormats[format]);
2332     PetscFunctionReturn(PETSC_SUCCESS);
2333 #else
2334     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2335 #endif
2336   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Viewer type %s not yet supported for DMPlex loading", ((PetscObject)viewer)->type_name);
2337 }
2338 
2339 /*@
2340   DMPlexTopologyLoad - Loads a topology into a `DMPLEX`
2341 
2342   Collective
2343 
2344   Input Parameters:
2345 + dm     - The `DM` into which the topology is loaded
2346 - viewer - The `PetscViewer` for the saved topology
2347 
2348   Output Parameter:
2349 . globalToLocalPointSF - The `PetscSF` that pushes points in [0, N) to the associated points in the loaded `DMPLEX`, where N is the global number of points; `NULL` if unneeded
2350 
2351   Level: advanced
2352 
2353 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMLoad()`, `DMPlexCoordinatesLoad()`, `DMPlexLabelsLoad()`, `DMView()`, `PetscViewerHDF5Open()`, `PetscViewerPushFormat()`,
2354           `PetscViewer`, `PetscSF`
2355 @*/
2356 PetscErrorCode DMPlexTopologyLoad(DM dm, PetscViewer viewer, PetscSF *globalToLocalPointSF)
2357 {
2358   PetscBool ishdf5;
2359 
2360   PetscFunctionBegin;
2361   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2362   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2363   if (globalToLocalPointSF) PetscAssertPointer(globalToLocalPointSF, 3);
2364   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2365   PetscCall(PetscLogEventBegin(DMPLEX_TopologyLoad, viewer, 0, 0, 0));
2366   if (ishdf5) {
2367 #if defined(PETSC_HAVE_HDF5)
2368     PetscViewerFormat format;
2369     PetscCall(PetscViewerGetFormat(viewer, &format));
2370     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2371       PetscCall(DMPlexTopologyLoad_HDF5_Internal(dm, viewer, globalToLocalPointSF));
2372     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 input.", PetscViewerFormats[format]);
2373 #else
2374     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2375 #endif
2376   }
2377   PetscCall(PetscLogEventEnd(DMPLEX_TopologyLoad, viewer, 0, 0, 0));
2378   PetscFunctionReturn(PETSC_SUCCESS);
2379 }
2380 
2381 /*@
2382   DMPlexCoordinatesLoad - Loads coordinates into a `DMPLEX`
2383 
2384   Collective
2385 
2386   Input Parameters:
2387 + dm                   - The `DM` into which the coordinates are loaded
2388 . viewer               - The `PetscViewer` for the saved coordinates
2389 - globalToLocalPointSF - The `PetscSF` returned by `DMPlexTopologyLoad()` when loading dm from viewer
2390 
2391   Level: advanced
2392 
2393 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMLoad()`, `DMPlexTopologyLoad()`, `DMPlexLabelsLoad()`, `DMView()`, `PetscViewerHDF5Open()`, `PetscViewerPushFormat()`,
2394           `PetscSF`, `PetscViewer`
2395 @*/
2396 PetscErrorCode DMPlexCoordinatesLoad(DM dm, PetscViewer viewer, PetscSF globalToLocalPointSF)
2397 {
2398   PetscBool ishdf5;
2399 
2400   PetscFunctionBegin;
2401   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2402   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2403   PetscValidHeaderSpecific(globalToLocalPointSF, PETSCSF_CLASSID, 3);
2404   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2405   PetscCall(PetscLogEventBegin(DMPLEX_CoordinatesLoad, viewer, 0, 0, 0));
2406   if (ishdf5) {
2407 #if defined(PETSC_HAVE_HDF5)
2408     PetscViewerFormat format;
2409     PetscCall(PetscViewerGetFormat(viewer, &format));
2410     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2411       PetscCall(DMPlexCoordinatesLoad_HDF5_Internal(dm, viewer, globalToLocalPointSF));
2412     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 input.", PetscViewerFormats[format]);
2413 #else
2414     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2415 #endif
2416   }
2417   PetscCall(PetscLogEventEnd(DMPLEX_CoordinatesLoad, viewer, 0, 0, 0));
2418   PetscFunctionReturn(PETSC_SUCCESS);
2419 }
2420 
2421 /*@
2422   DMPlexLabelsLoad - Loads labels into a `DMPLEX`
2423 
2424   Collective
2425 
2426   Input Parameters:
2427 + dm                   - The `DM` into which the labels are loaded
2428 . viewer               - The `PetscViewer` for the saved labels
2429 - globalToLocalPointSF - The `PetscSF` returned by `DMPlexTopologyLoad()` when loading `dm` from viewer
2430 
2431   Level: advanced
2432 
2433   Note:
2434   The `PetscSF` argument must not be NULL if the `DM` is distributed, otherwise an error occurs.
2435 
2436 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMLoad()`, `DMPlexTopologyLoad()`, `DMPlexCoordinatesLoad()`, `DMView()`, `PetscViewerHDF5Open()`, `PetscViewerPushFormat()`,
2437           `PetscSF`, `PetscViewer`
2438 @*/
2439 PetscErrorCode DMPlexLabelsLoad(DM dm, PetscViewer viewer, PetscSF globalToLocalPointSF)
2440 {
2441   PetscBool ishdf5;
2442 
2443   PetscFunctionBegin;
2444   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2445   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2446   if (globalToLocalPointSF) PetscValidHeaderSpecific(globalToLocalPointSF, PETSCSF_CLASSID, 3);
2447   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2448   PetscCall(PetscLogEventBegin(DMPLEX_LabelsLoad, viewer, 0, 0, 0));
2449   if (ishdf5) {
2450 #if defined(PETSC_HAVE_HDF5)
2451     PetscViewerFormat format;
2452 
2453     PetscCall(PetscViewerGetFormat(viewer, &format));
2454     if (format == PETSC_VIEWER_HDF5_PETSC || format == PETSC_VIEWER_DEFAULT || format == PETSC_VIEWER_NATIVE) {
2455       PetscCall(DMPlexLabelsLoad_HDF5_Internal(dm, viewer, globalToLocalPointSF));
2456     } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "PetscViewerFormat %s not supported for HDF5 input.", PetscViewerFormats[format]);
2457 #else
2458     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2459 #endif
2460   }
2461   PetscCall(PetscLogEventEnd(DMPLEX_LabelsLoad, viewer, 0, 0, 0));
2462   PetscFunctionReturn(PETSC_SUCCESS);
2463 }
2464 
2465 /*@
2466   DMPlexSectionLoad - Loads section into a `DMPLEX`
2467 
2468   Collective
2469 
2470   Input Parameters:
2471 + dm                   - The `DM` that represents the topology
2472 . viewer               - The `PetscViewer` that represents the on-disk section (sectionA)
2473 . sectiondm            - The `DM` into which the on-disk section (sectionA) is migrated, can be `NULL`
2474 - globalToLocalPointSF - The `PetscSF` returned by `DMPlexTopologyLoad(`) when loading dm from viewer
2475 
2476   Output Parameters:
2477 + globalDofSF - The `PetscSF` that migrates any on-disk `Vec` data associated with sectionA into a global `Vec` associated with the `sectiondm`'s global section (`NULL` if not needed)
2478 - localDofSF  - The `PetscSF` that migrates any on-disk `Vec` data associated with sectionA into a local `Vec` associated with the `sectiondm`'s local section (`NULL` if not needed)
2479 
2480   Level: advanced
2481 
2482   Notes:
2483   This function is a wrapper around `PetscSectionLoad()`; it loads, in addition to the raw section, a list of global point numbers that associates each on-disk section point with a global point number in [0, NX), where NX is the number of topology points in `dm`. Noting that globalToLocalPointSF associates each topology point in dm with a global number in [0, NX), one can readily establish an association of the on-disk section points with the topology points.
2484 
2485   In general `dm` and `sectiondm` are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2486 
2487   The output parameter, `globalDofSF` (`localDofSF`), can later be used with `DMPlexGlobalVectorLoad()` (`DMPlexLocalVectorLoad()`) to load on-disk vectors into global (local) vectors associated with sectiondm's global (local) section.
2488 
2489   Example using 2 processes:
2490 .vb
2491   NX (number of points on dm): 4
2492   sectionA                   : the on-disk section
2493   vecA                       : a vector associated with sectionA
2494   sectionB                   : sectiondm's local section constructed in this function
2495   vecB (local)               : a vector associated with sectiondm's local section
2496   vecB (global)              : a vector associated with sectiondm's global section
2497 
2498                                      rank 0    rank 1
2499   vecA (global)                  : [.0 .4 .1 | .2 .3]        <- to be loaded in DMPlexGlobalVectorLoad() or DMPlexLocalVectorLoad()
2500   sectionA->atlasOff             :       0 2 | 1             <- loaded in PetscSectionLoad()
2501   sectionA->atlasDof             :       1 3 | 1             <- loaded in PetscSectionLoad()
2502   sectionA's global point numbers:       0 2 | 3             <- loaded in DMPlexSectionLoad()
2503   [0, NX)                        :       0 1 | 2 3           <- conceptual partition used in globalToLocalPointSF
2504   sectionB's global point numbers:     0 1 3 | 3 2           <- associated with [0, NX) by globalToLocalPointSF
2505   sectionB->atlasDof             :     1 0 1 | 1 3
2506   sectionB->atlasOff (no perm)   :     0 1 1 | 0 1
2507   vecB (local)                   :   [.0 .4] | [.4 .1 .2 .3] <- to be constructed by calling DMPlexLocalVectorLoad() with localDofSF
2508   vecB (global)                  :    [.0 .4 | .1 .2 .3]     <- to be constructed by calling DMPlexGlobalVectorLoad() with globalDofSF
2509 .ve
2510   where "|" represents a partition of loaded data, and global point 3 is assumed to be owned by rank 0.
2511 
2512 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMLoad()`, `DMPlexTopologyLoad()`, `DMPlexCoordinatesLoad()`, `DMPlexLabelsLoad()`, `DMPlexGlobalVectorLoad()`, `DMPlexLocalVectorLoad()`, `PetscSectionLoad()`, `DMPlexSectionView()`, `PetscSF`, `PetscViewer`
2513 @*/
2514 PetscErrorCode DMPlexSectionLoad(DM dm, PetscViewer viewer, DM sectiondm, PetscSF globalToLocalPointSF, PetscSF *globalDofSF, PetscSF *localDofSF)
2515 {
2516   PetscBool ishdf5;
2517 
2518   PetscFunctionBegin;
2519   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2520   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2521   if (!sectiondm) sectiondm = dm;
2522   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2523   PetscValidHeaderSpecific(globalToLocalPointSF, PETSCSF_CLASSID, 4);
2524   if (globalDofSF) PetscAssertPointer(globalDofSF, 5);
2525   if (localDofSF) PetscAssertPointer(localDofSF, 6);
2526   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2527   PetscCall(PetscLogEventBegin(DMPLEX_SectionLoad, viewer, 0, 0, 0));
2528   if (ishdf5) {
2529 #if defined(PETSC_HAVE_HDF5)
2530     PetscCall(DMPlexSectionLoad_HDF5_Internal(dm, viewer, sectiondm, globalToLocalPointSF, globalDofSF, localDofSF));
2531 #else
2532     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2533 #endif
2534   }
2535   PetscCall(PetscLogEventEnd(DMPLEX_SectionLoad, viewer, 0, 0, 0));
2536   PetscFunctionReturn(PETSC_SUCCESS);
2537 }
2538 
2539 /*@
2540   DMPlexGlobalVectorLoad - Loads on-disk vector data into a global vector
2541 
2542   Collective
2543 
2544   Input Parameters:
2545 + dm        - The `DM` that represents the topology
2546 . viewer    - The `PetscViewer` that represents the on-disk vector data
2547 . sectiondm - The `DM` that contains the global section on which vec is defined, can be `NULL`
2548 . sf        - The `PetscSF` that migrates the on-disk vector data into vec
2549 - vec       - The global vector to set values of
2550 
2551   Level: advanced
2552 
2553   Notes:
2554   In general dm and sectiondm are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2555 
2556   Calling sequence:
2557 .vb
2558        DMCreate(PETSC_COMM_WORLD, &dm);
2559        DMSetType(dm, DMPLEX);
2560        PetscObjectSetName((PetscObject)dm, "topologydm_name");
2561        DMPlexTopologyLoad(dm, viewer, &sfX);
2562        DMClone(dm, &sectiondm);
2563        PetscObjectSetName((PetscObject)sectiondm, "sectiondm_name");
2564        DMPlexSectionLoad(dm, viewer, sectiondm, sfX, &gsf, NULL);
2565        DMGetGlobalVector(sectiondm, &vec);
2566        PetscObjectSetName((PetscObject)vec, "vec_name");
2567        DMPlexGlobalVectorLoad(dm, viewer, sectiondm, gsf, vec);
2568        DMRestoreGlobalVector(sectiondm, &vec);
2569        PetscSFDestroy(&gsf);
2570        PetscSFDestroy(&sfX);
2571        DMDestroy(&sectiondm);
2572        DMDestroy(&dm);
2573 .ve
2574 
2575 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTopologyLoad()`, `DMPlexSectionLoad()`, `DMPlexLocalVectorLoad()`, `DMPlexGlobalVectorView()`, `DMPlexLocalVectorView()`,
2576           `PetscSF`, `PetscViewer`
2577 @*/
2578 PetscErrorCode DMPlexGlobalVectorLoad(DM dm, PetscViewer viewer, DM sectiondm, PetscSF sf, Vec vec)
2579 {
2580   PetscBool ishdf5;
2581 
2582   PetscFunctionBegin;
2583   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2584   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2585   if (!sectiondm) sectiondm = dm;
2586   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2587   PetscValidHeaderSpecific(sf, PETSCSF_CLASSID, 4);
2588   PetscValidHeaderSpecific(vec, VEC_CLASSID, 5);
2589   /* Check consistency */
2590   {
2591     PetscSection section;
2592     PetscBool    includesConstraints;
2593     PetscInt     m, m1;
2594 
2595     PetscCall(VecGetLocalSize(vec, &m1));
2596     PetscCall(DMGetGlobalSection(sectiondm, &section));
2597     PetscCall(PetscSectionGetIncludesConstraints(section, &includesConstraints));
2598     if (includesConstraints) PetscCall(PetscSectionGetStorageSize(section, &m));
2599     else PetscCall(PetscSectionGetConstrainedStorageSize(section, &m));
2600     PetscCheck(m1 == m, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Global vector size (%" PetscInt_FMT ") != global section storage size (%" PetscInt_FMT ")", m1, m);
2601   }
2602   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2603   PetscCall(PetscLogEventBegin(DMPLEX_GlobalVectorLoad, viewer, 0, 0, 0));
2604   if (ishdf5) {
2605 #if defined(PETSC_HAVE_HDF5)
2606     PetscCall(DMPlexVecLoad_HDF5_Internal(dm, viewer, sectiondm, sf, vec));
2607 #else
2608     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2609 #endif
2610   }
2611   PetscCall(PetscLogEventEnd(DMPLEX_GlobalVectorLoad, viewer, 0, 0, 0));
2612   PetscFunctionReturn(PETSC_SUCCESS);
2613 }
2614 
2615 /*@
2616   DMPlexLocalVectorLoad - Loads on-disk vector data into a local vector
2617 
2618   Collective
2619 
2620   Input Parameters:
2621 + dm        - The `DM` that represents the topology
2622 . viewer    - The `PetscViewer` that represents the on-disk vector data
2623 . sectiondm - The `DM` that contains the local section on which vec is defined, can be `NULL`
2624 . sf        - The `PetscSF` that migrates the on-disk vector data into vec
2625 - vec       - The local vector to set values of
2626 
2627   Level: advanced
2628 
2629   Notes:
2630   In general `dm` and `sectiondm` are two different objects, the former carrying the topology and the latter carrying the section, and have been given a topology name and a section name, respectively, with `PetscObjectSetName()`. In practice, however, they can be the same object (or in case `sectiondm` is `NULL`) if it carries both topology and section; in that case the name of the object is used as both the topology name and the section name.
2631 
2632   Calling sequence:
2633 .vb
2634        DMCreate(PETSC_COMM_WORLD, &dm);
2635        DMSetType(dm, DMPLEX);
2636        PetscObjectSetName((PetscObject)dm, "topologydm_name");
2637        DMPlexTopologyLoad(dm, viewer, &sfX);
2638        DMClone(dm, &sectiondm);
2639        PetscObjectSetName((PetscObject)sectiondm, "sectiondm_name");
2640        DMPlexSectionLoad(dm, viewer, sectiondm, sfX, NULL, &lsf);
2641        DMGetLocalVector(sectiondm, &vec);
2642        PetscObjectSetName((PetscObject)vec, "vec_name");
2643        DMPlexLocalVectorLoad(dm, viewer, sectiondm, lsf, vec);
2644        DMRestoreLocalVector(sectiondm, &vec);
2645        PetscSFDestroy(&lsf);
2646        PetscSFDestroy(&sfX);
2647        DMDestroy(&sectiondm);
2648        DMDestroy(&dm);
2649 .ve
2650 
2651 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexTopologyLoad()`, `DMPlexSectionLoad()`, `DMPlexGlobalVectorLoad()`, `DMPlexGlobalVectorView()`, `DMPlexLocalVectorView()`,
2652           `PetscSF`, `PetscViewer`
2653 @*/
2654 PetscErrorCode DMPlexLocalVectorLoad(DM dm, PetscViewer viewer, DM sectiondm, PetscSF sf, Vec vec)
2655 {
2656   PetscBool ishdf5;
2657 
2658   PetscFunctionBegin;
2659   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2660   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
2661   if (!sectiondm) sectiondm = dm;
2662   PetscValidHeaderSpecific(sectiondm, DM_CLASSID, 3);
2663   PetscValidHeaderSpecific(sf, PETSCSF_CLASSID, 4);
2664   PetscValidHeaderSpecific(vec, VEC_CLASSID, 5);
2665   /* Check consistency */
2666   {
2667     PetscSection section;
2668     PetscBool    includesConstraints;
2669     PetscInt     m, m1;
2670 
2671     PetscCall(VecGetLocalSize(vec, &m1));
2672     PetscCall(DMGetLocalSection(sectiondm, &section));
2673     PetscCall(PetscSectionGetIncludesConstraints(section, &includesConstraints));
2674     if (includesConstraints) PetscCall(PetscSectionGetStorageSize(section, &m));
2675     else PetscCall(PetscSectionGetConstrainedStorageSize(section, &m));
2676     PetscCheck(m1 == m, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Local vector size (%" PetscInt_FMT ") != local section storage size (%" PetscInt_FMT ")", m1, m);
2677   }
2678   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
2679   PetscCall(PetscLogEventBegin(DMPLEX_LocalVectorLoad, viewer, 0, 0, 0));
2680   if (ishdf5) {
2681 #if defined(PETSC_HAVE_HDF5)
2682     PetscCall(DMPlexVecLoad_HDF5_Internal(dm, viewer, sectiondm, sf, vec));
2683 #else
2684     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5");
2685 #endif
2686   }
2687   PetscCall(PetscLogEventEnd(DMPLEX_LocalVectorLoad, viewer, 0, 0, 0));
2688   PetscFunctionReturn(PETSC_SUCCESS);
2689 }
2690 
2691 PetscErrorCode DMDestroy_Plex(DM dm)
2692 {
2693   DM_Plex *mesh = (DM_Plex *)dm->data;
2694 
2695   PetscFunctionBegin;
2696   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", NULL));
2697   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", NULL));
2698   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", NULL));
2699   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMInterpolateSolution_C", NULL));
2700   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertTimeDerivativeBoundaryValues_C", NULL));
2701   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", NULL));
2702   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeGetDefault_C", NULL));
2703   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexDistributeSetDefault_C", NULL));
2704   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "MatComputeNeumannOverlap_C", NULL));
2705   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderGetDefault_C", NULL));
2706   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSetDefault_C", NULL));
2707   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSectionGetDefault_C", NULL));
2708   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexReorderSectionSetDefault_C", NULL));
2709   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", NULL));
2710   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetOverlap_C", NULL));
2711   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetUseCeed_C", NULL));
2712   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexSetUseCeed_C", NULL));
2713   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMGetIsoperiodicPointSF_C", NULL));
2714   if (--mesh->refct > 0) PetscFunctionReturn(PETSC_SUCCESS);
2715   PetscCall(PetscSectionDestroy(&mesh->coneSection));
2716   PetscCall(PetscFree(mesh->cones));
2717   PetscCall(PetscFree(mesh->coneOrientations));
2718   PetscCall(PetscSectionDestroy(&mesh->supportSection));
2719   PetscCall(PetscSectionDestroy(&mesh->subdomainSection));
2720   PetscCall(PetscFree(mesh->supports));
2721   PetscCall(PetscFree(mesh->cellTypes));
2722   PetscCall(DMPlexTransformDestroy(&mesh->tr));
2723   PetscCall(PetscFree(mesh->tetgenOpts));
2724   PetscCall(PetscFree(mesh->triangleOpts));
2725   PetscCall(PetscFree(mesh->transformType));
2726   PetscCall(PetscFree(mesh->distributionName));
2727   PetscCall(PetscPartitionerDestroy(&mesh->partitioner));
2728   PetscCall(DMLabelDestroy(&mesh->subpointMap));
2729   PetscCall(ISDestroy(&mesh->subpointIS));
2730   PetscCall(ISDestroy(&mesh->globalVertexNumbers));
2731   PetscCall(ISDestroy(&mesh->globalCellNumbers));
2732   PetscCall(PetscSFDestroy(&mesh->periodic.face_sf));
2733   PetscCall(PetscSFDestroy(&mesh->periodic.composed_sf));
2734   PetscCall(ISDestroy(&mesh->periodic.periodic_points));
2735   PetscCall(PetscSectionDestroy(&mesh->anchorSection));
2736   PetscCall(ISDestroy(&mesh->anchorIS));
2737   PetscCall(PetscSectionDestroy(&mesh->parentSection));
2738   PetscCall(PetscFree(mesh->parents));
2739   PetscCall(PetscFree(mesh->childIDs));
2740   PetscCall(PetscSectionDestroy(&mesh->childSection));
2741   PetscCall(PetscFree(mesh->children));
2742   PetscCall(DMDestroy(&mesh->referenceTree));
2743   PetscCall(PetscGridHashDestroy(&mesh->lbox));
2744   PetscCall(PetscFree(mesh->neighbors));
2745   if (mesh->metricCtx) PetscCall(PetscFree(mesh->metricCtx));
2746   /* This was originally freed in DMDestroy(), but that prevents reference counting of backend objects */
2747   PetscCall(PetscFree(mesh));
2748   PetscFunctionReturn(PETSC_SUCCESS);
2749 }
2750 
2751 PetscErrorCode DMCreateMatrix_Plex(DM dm, Mat *J)
2752 {
2753   PetscSection           sectionGlobal, sectionLocal;
2754   PetscInt               bs = -1, mbs;
2755   PetscInt               localSize, localStart = 0;
2756   PetscBool              isShell, isBlock, isSeqBlock, isMPIBlock, isSymBlock, isSymSeqBlock, isSymMPIBlock, isMatIS;
2757   MatType                mtype;
2758   ISLocalToGlobalMapping ltog;
2759 
2760   PetscFunctionBegin;
2761   PetscCall(MatInitializePackage());
2762   mtype = dm->mattype;
2763   PetscCall(DMGetLocalSection(dm, &sectionLocal));
2764   PetscCall(DMGetGlobalSection(dm, &sectionGlobal));
2765   /* PetscCall(PetscSectionGetStorageSize(sectionGlobal, &localSize)); */
2766   PetscCall(PetscSectionGetConstrainedStorageSize(sectionGlobal, &localSize));
2767   PetscCallMPI(MPI_Exscan(&localSize, &localStart, 1, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject)dm)));
2768   PetscCall(MatCreate(PetscObjectComm((PetscObject)dm), J));
2769   PetscCall(MatSetSizes(*J, localSize, localSize, PETSC_DETERMINE, PETSC_DETERMINE));
2770   PetscCall(MatSetType(*J, mtype));
2771   PetscCall(MatSetFromOptions(*J));
2772   PetscCall(MatGetBlockSize(*J, &mbs));
2773   if (mbs > 1) bs = mbs;
2774   PetscCall(PetscStrcmp(mtype, MATSHELL, &isShell));
2775   PetscCall(PetscStrcmp(mtype, MATBAIJ, &isBlock));
2776   PetscCall(PetscStrcmp(mtype, MATSEQBAIJ, &isSeqBlock));
2777   PetscCall(PetscStrcmp(mtype, MATMPIBAIJ, &isMPIBlock));
2778   PetscCall(PetscStrcmp(mtype, MATSBAIJ, &isSymBlock));
2779   PetscCall(PetscStrcmp(mtype, MATSEQSBAIJ, &isSymSeqBlock));
2780   PetscCall(PetscStrcmp(mtype, MATMPISBAIJ, &isSymMPIBlock));
2781   PetscCall(PetscStrcmp(mtype, MATIS, &isMatIS));
2782   if (!isShell) {
2783     PetscBool fillMatrix = (PetscBool)(!dm->prealloc_only && !isMatIS);
2784     PetscInt *dnz, *onz, *dnzu, *onzu, bsLocal[2], bsMinMax[2], *pblocks;
2785     PetscInt  pStart, pEnd, p, dof, cdof, num_fields;
2786 
2787     PetscCall(DMGetLocalToGlobalMapping(dm, &ltog));
2788 
2789     PetscCall(PetscCalloc1(localSize, &pblocks));
2790     PetscCall(PetscSectionGetChart(sectionGlobal, &pStart, &pEnd));
2791     PetscCall(PetscSectionGetNumFields(sectionGlobal, &num_fields));
2792     for (p = pStart; p < pEnd; ++p) {
2793       switch (dm->blocking_type) {
2794       case DM_BLOCKING_TOPOLOGICAL_POINT: { // One block per topological point
2795         PetscInt bdof, offset;
2796 
2797         PetscCall(PetscSectionGetDof(sectionGlobal, p, &dof));
2798         PetscCall(PetscSectionGetOffset(sectionGlobal, p, &offset));
2799         PetscCall(PetscSectionGetConstraintDof(sectionGlobal, p, &cdof));
2800         for (PetscInt i = 0; i < dof - cdof; ++i) pblocks[offset - localStart + i] = dof - cdof;
2801         // Signal block concatenation
2802         if (dof - cdof && sectionLocal->blockStarts && !PetscBTLookup(sectionLocal->blockStarts, p)) pblocks[offset - localStart] = -(dof - cdof);
2803         dof  = dof < 0 ? -(dof + 1) : dof;
2804         bdof = cdof && (dof - cdof) ? 1 : dof;
2805         if (dof) {
2806           if (bs < 0) {
2807             bs = bdof;
2808           } else if (bs != bdof) {
2809             bs = 1;
2810           }
2811         }
2812       } break;
2813       case DM_BLOCKING_FIELD_NODE: {
2814         for (PetscInt field = 0; field < num_fields; field++) {
2815           PetscInt num_comp, bdof, offset;
2816           PetscCall(PetscSectionGetFieldComponents(sectionGlobal, field, &num_comp));
2817           PetscCall(PetscSectionGetFieldDof(sectionGlobal, p, field, &dof));
2818           if (dof < 0) continue;
2819           PetscCall(PetscSectionGetFieldOffset(sectionGlobal, p, field, &offset));
2820           PetscCall(PetscSectionGetFieldConstraintDof(sectionGlobal, p, field, &cdof));
2821           PetscAssert(dof % num_comp == 0, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Point %" PetscInt_FMT " field %" PetscInt_FMT " has %" PetscInt_FMT " dof, not divisible by %" PetscInt_FMT " component ", p, field, dof, num_comp);
2822           PetscInt num_nodes = dof / num_comp;
2823           for (PetscInt i = 0; i < dof - cdof; i++) pblocks[offset - localStart + i] = (dof - cdof) / num_nodes;
2824           // Handle possibly constant block size (unlikely)
2825           bdof = cdof && (dof - cdof) ? 1 : dof;
2826           if (dof) {
2827             if (bs < 0) {
2828               bs = bdof;
2829             } else if (bs != bdof) {
2830               bs = 1;
2831             }
2832           }
2833         }
2834       } break;
2835       }
2836     }
2837     /* Must have same blocksize on all procs (some might have no points) */
2838     bsLocal[0] = bs < 0 ? PETSC_MAX_INT : bs;
2839     bsLocal[1] = bs;
2840     PetscCall(PetscGlobalMinMaxInt(PetscObjectComm((PetscObject)dm), bsLocal, bsMinMax));
2841     if (bsMinMax[0] != bsMinMax[1]) bs = 1;
2842     else bs = bsMinMax[0];
2843     bs = PetscMax(1, bs);
2844     PetscCall(MatSetLocalToGlobalMapping(*J, ltog, ltog));
2845     if (dm->prealloc_skip) { // User will likely use MatSetPreallocationCOO(), but still set structural parameters
2846       PetscCall(MatSetBlockSize(*J, bs));
2847       PetscCall(MatSetUp(*J));
2848     } else {
2849       PetscCall(PetscCalloc4(localSize / bs, &dnz, localSize / bs, &onz, localSize / bs, &dnzu, localSize / bs, &onzu));
2850       PetscCall(DMPlexPreallocateOperator(dm, bs, dnz, onz, dnzu, onzu, *J, fillMatrix));
2851       PetscCall(PetscFree4(dnz, onz, dnzu, onzu));
2852     }
2853     { // Consolidate blocks
2854       PetscInt nblocks = 0;
2855       for (PetscInt i = 0; i < localSize; i += PetscMax(1, pblocks[i])) {
2856         if (pblocks[i] == 0) continue;
2857         // Negative block size indicates the blocks should be concatenated
2858         if (pblocks[i] < 0) {
2859           pblocks[i] = -pblocks[i];
2860           pblocks[nblocks - 1] += pblocks[i];
2861         } else {
2862           pblocks[nblocks++] = pblocks[i]; // nblocks always <= i
2863         }
2864         for (PetscInt j = 1; j < pblocks[i]; j++) PetscCheck(pblocks[i + j] == pblocks[i], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Block of size %" PetscInt_FMT " mismatches entry %" PetscInt_FMT, pblocks[i], pblocks[i + j]);
2865       }
2866       PetscCall(MatSetVariableBlockSizes(*J, nblocks, pblocks));
2867     }
2868     PetscCall(PetscFree(pblocks));
2869   }
2870   PetscCall(MatSetDM(*J, dm));
2871   PetscFunctionReturn(PETSC_SUCCESS);
2872 }
2873 
2874 /*@
2875   DMPlexGetSubdomainSection - Returns the section associated with the subdomain
2876 
2877   Not Collective
2878 
2879   Input Parameter:
2880 . dm - The `DMPLEX`
2881 
2882   Output Parameter:
2883 . subsection - The subdomain section
2884 
2885   Level: developer
2886 
2887 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `PetscSection`
2888 @*/
2889 PetscErrorCode DMPlexGetSubdomainSection(DM dm, PetscSection *subsection)
2890 {
2891   DM_Plex *mesh = (DM_Plex *)dm->data;
2892 
2893   PetscFunctionBegin;
2894   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2895   if (!mesh->subdomainSection) {
2896     PetscSection section;
2897     PetscSF      sf;
2898 
2899     PetscCall(PetscSFCreate(PETSC_COMM_SELF, &sf));
2900     PetscCall(DMGetLocalSection(dm, &section));
2901     PetscCall(PetscSectionCreateGlobalSection(section, sf, PETSC_FALSE, PETSC_TRUE, &mesh->subdomainSection));
2902     PetscCall(PetscSFDestroy(&sf));
2903   }
2904   *subsection = mesh->subdomainSection;
2905   PetscFunctionReturn(PETSC_SUCCESS);
2906 }
2907 
2908 /*@
2909   DMPlexGetChart - Return the interval for all mesh points [`pStart`, `pEnd`)
2910 
2911   Not Collective
2912 
2913   Input Parameter:
2914 . dm - The `DMPLEX`
2915 
2916   Output Parameters:
2917 + pStart - The first mesh point
2918 - pEnd   - The upper bound for mesh points
2919 
2920   Level: beginner
2921 
2922 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSetChart()`
2923 @*/
2924 PetscErrorCode DMPlexGetChart(DM dm, PetscInt *pStart, PetscInt *pEnd)
2925 {
2926   DM_Plex *mesh = (DM_Plex *)dm->data;
2927 
2928   PetscFunctionBegin;
2929   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2930   if (mesh->tr) PetscCall(DMPlexTransformGetChart(mesh->tr, pStart, pEnd));
2931   else PetscCall(PetscSectionGetChart(mesh->coneSection, pStart, pEnd));
2932   PetscFunctionReturn(PETSC_SUCCESS);
2933 }
2934 
2935 /*@
2936   DMPlexSetChart - Set the interval for all mesh points [`pStart`, `pEnd`)
2937 
2938   Not Collective
2939 
2940   Input Parameters:
2941 + dm     - The `DMPLEX`
2942 . pStart - The first mesh point
2943 - pEnd   - The upper bound for mesh points
2944 
2945   Level: beginner
2946 
2947 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetChart()`
2948 @*/
2949 PetscErrorCode DMPlexSetChart(DM dm, PetscInt pStart, PetscInt pEnd)
2950 {
2951   DM_Plex *mesh = (DM_Plex *)dm->data;
2952 
2953   PetscFunctionBegin;
2954   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2955   PetscCall(PetscSectionSetChart(mesh->coneSection, pStart, pEnd));
2956   PetscCall(PetscSectionSetChart(mesh->supportSection, pStart, pEnd));
2957   PetscCall(PetscFree(mesh->cellTypes));
2958   PetscFunctionReturn(PETSC_SUCCESS);
2959 }
2960 
2961 /*@
2962   DMPlexGetConeSize - Return the number of in-edges for this point in the DAG
2963 
2964   Not Collective
2965 
2966   Input Parameters:
2967 + dm - The `DMPLEX`
2968 - p  - The point, which must lie in the chart set with `DMPlexSetChart()`
2969 
2970   Output Parameter:
2971 . size - The cone size for point `p`
2972 
2973   Level: beginner
2974 
2975 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSetConeSize()`, `DMPlexSetChart()`
2976 @*/
2977 PetscErrorCode DMPlexGetConeSize(DM dm, PetscInt p, PetscInt *size)
2978 {
2979   DM_Plex *mesh = (DM_Plex *)dm->data;
2980 
2981   PetscFunctionBegin;
2982   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2983   PetscAssertPointer(size, 3);
2984   if (mesh->tr) PetscCall(DMPlexTransformGetConeSize(mesh->tr, p, size));
2985   else PetscCall(PetscSectionGetDof(mesh->coneSection, p, size));
2986   PetscFunctionReturn(PETSC_SUCCESS);
2987 }
2988 
2989 /*@
2990   DMPlexSetConeSize - Set the number of in-edges for this point in the DAG
2991 
2992   Not Collective
2993 
2994   Input Parameters:
2995 + dm   - The `DMPLEX`
2996 . p    - The point, which must lie in the chart set with `DMPlexSetChart()`
2997 - size - The cone size for point `p`
2998 
2999   Level: beginner
3000 
3001   Note:
3002   This should be called after `DMPlexSetChart()`.
3003 
3004 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetConeSize()`, `DMPlexSetChart()`
3005 @*/
3006 PetscErrorCode DMPlexSetConeSize(DM dm, PetscInt p, PetscInt size)
3007 {
3008   DM_Plex *mesh = (DM_Plex *)dm->data;
3009 
3010   PetscFunctionBegin;
3011   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3012   PetscCheck(!mesh->tr, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Cannot call DMPlexSetConeSize() on a mesh with a transform defined.");
3013   PetscCall(PetscSectionSetDof(mesh->coneSection, p, size));
3014   PetscFunctionReturn(PETSC_SUCCESS);
3015 }
3016 
3017 /*@C
3018   DMPlexGetCone - Return the points on the in-edges for this point in the DAG
3019 
3020   Not Collective
3021 
3022   Input Parameters:
3023 + dm - The `DMPLEX`
3024 - p  - The point, which must lie in the chart set with `DMPlexSetChart()`
3025 
3026   Output Parameter:
3027 . cone - An array of points which are on the in-edges for point `p`
3028 
3029   Level: beginner
3030 
3031   Fortran Notes:
3032   You must also call `DMPlexRestoreCone()` after you finish using the returned array.
3033   `DMPlexRestoreCone()` is not needed/available in C.
3034 
3035 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetConeSize()`, `DMPlexSetCone()`, `DMPlexGetConeTuple()`, `DMPlexSetChart()`, `DMPlexRestoreCone()`
3036 @*/
3037 PetscErrorCode DMPlexGetCone(DM dm, PetscInt p, const PetscInt *cone[])
3038 {
3039   DM_Plex *mesh = (DM_Plex *)dm->data;
3040   PetscInt off;
3041 
3042   PetscFunctionBegin;
3043   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3044   PetscAssertPointer(cone, 3);
3045   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3046   *cone = PetscSafePointerPlusOffset(mesh->cones, off);
3047   PetscFunctionReturn(PETSC_SUCCESS);
3048 }
3049 
3050 /*@C
3051   DMPlexGetConeTuple - Return the points on the in-edges of several points in the DAG
3052 
3053   Not Collective
3054 
3055   Input Parameters:
3056 + dm - The `DMPLEX`
3057 - p  - The `IS` of points, which must lie in the chart set with `DMPlexSetChart()`
3058 
3059   Output Parameters:
3060 + pConesSection - `PetscSection` describing the layout of `pCones`
3061 - pCones        - An array of points which are on the in-edges for the point set `p`
3062 
3063   Level: intermediate
3064 
3065 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexGetConeRecursive()`, `DMPlexSetChart()`, `PetscSection`, `IS`
3066 @*/
3067 PetscErrorCode DMPlexGetConeTuple(DM dm, IS p, PetscSection *pConesSection, IS *pCones)
3068 {
3069   PetscSection cs, newcs;
3070   PetscInt    *cones;
3071   PetscInt    *newarr = NULL;
3072   PetscInt     n;
3073 
3074   PetscFunctionBegin;
3075   PetscCall(DMPlexGetCones(dm, &cones));
3076   PetscCall(DMPlexGetConeSection(dm, &cs));
3077   PetscCall(PetscSectionExtractDofsFromArray(cs, MPIU_INT, cones, p, &newcs, pCones ? ((void **)&newarr) : NULL));
3078   if (pConesSection) *pConesSection = newcs;
3079   if (pCones) {
3080     PetscCall(PetscSectionGetStorageSize(newcs, &n));
3081     PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)p), n, newarr, PETSC_OWN_POINTER, pCones));
3082   }
3083   PetscFunctionReturn(PETSC_SUCCESS);
3084 }
3085 
3086 /*@
3087   DMPlexGetConeRecursiveVertices - Expand each given point into its cone points and do that recursively until we end up just with vertices.
3088 
3089   Not Collective
3090 
3091   Input Parameters:
3092 + dm     - The `DMPLEX`
3093 - points - The `IS` of points, which must lie in the chart set with `DMPlexSetChart()`
3094 
3095   Output Parameter:
3096 . expandedPoints - An array of vertices recursively expanded from input points
3097 
3098   Level: advanced
3099 
3100   Notes:
3101   Like `DMPlexGetConeRecursive()` but returns only the 0-depth `IS` (i.e. vertices only) and no sections.
3102 
3103   There is no corresponding Restore function, just call `ISDestroy()` on the returned `IS` to deallocate.
3104 
3105 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexGetConeTuple()`, `DMPlexGetConeRecursive()`, `DMPlexRestoreConeRecursive()`,
3106           `DMPlexGetDepth()`, `IS`
3107 @*/
3108 PetscErrorCode DMPlexGetConeRecursiveVertices(DM dm, IS points, IS *expandedPoints)
3109 {
3110   IS      *expandedPointsAll;
3111   PetscInt depth;
3112 
3113   PetscFunctionBegin;
3114   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3115   PetscValidHeaderSpecific(points, IS_CLASSID, 2);
3116   PetscAssertPointer(expandedPoints, 3);
3117   PetscCall(DMPlexGetConeRecursive(dm, points, &depth, &expandedPointsAll, NULL));
3118   *expandedPoints = expandedPointsAll[0];
3119   PetscCall(PetscObjectReference((PetscObject)expandedPointsAll[0]));
3120   PetscCall(DMPlexRestoreConeRecursive(dm, points, &depth, &expandedPointsAll, NULL));
3121   PetscFunctionReturn(PETSC_SUCCESS);
3122 }
3123 
3124 /*@
3125   DMPlexGetConeRecursive - Expand each given point into its cone points and do that recursively until we end up just with vertices (DAG points of depth 0, i.e. without cones).
3126 
3127   Not Collective
3128 
3129   Input Parameters:
3130 + dm     - The `DMPLEX`
3131 - points - The `IS` of points, which must lie in the chart set with `DMPlexSetChart()`
3132 
3133   Output Parameters:
3134 + depth          - (optional) Size of the output arrays, equal to `DMPLEX` depth, returned by `DMPlexGetDepth()`
3135 . expandedPoints - (optional) An array of index sets with recursively expanded cones
3136 - sections       - (optional) An array of sections which describe mappings from points to their cone points
3137 
3138   Level: advanced
3139 
3140   Notes:
3141   Like `DMPlexGetConeTuple()` but recursive.
3142 
3143   Array `expandedPoints` has size equal to `depth`. Each `expandedPoints`[d] contains DAG points with maximum depth d, recursively cone-wise expanded from the input points.
3144   For example, for d=0 it contains only vertices, for d=1 it can contain vertices and edges, etc.
3145 
3146   Array section has size equal to `depth`.  Each `PetscSection` `sections`[d] realizes mapping from `expandedPoints`[d+1] (section points) to `expandedPoints`[d] (section dofs) as follows\:
3147   (1) DAG points in `expandedPoints`[d+1] with `depth` d+1 to their cone points in `expandedPoints`[d];
3148   (2) DAG points in `expandedPoints`[d+1] with `depth` in [0,d] to the same points in `expandedPoints`[d].
3149 
3150 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexGetConeTuple()`, `DMPlexRestoreConeRecursive()`, `DMPlexGetConeRecursiveVertices()`,
3151           `DMPlexGetDepth()`, `PetscSection`, `IS`
3152 @*/
3153 PetscErrorCode DMPlexGetConeRecursive(DM dm, IS points, PetscInt *depth, IS *expandedPoints[], PetscSection *sections[])
3154 {
3155   const PetscInt *arr0 = NULL, *cone = NULL;
3156   PetscInt       *arr = NULL, *newarr = NULL;
3157   PetscInt        d, depth_, i, n, newn, cn, co, start, end;
3158   IS             *expandedPoints_;
3159   PetscSection   *sections_;
3160 
3161   PetscFunctionBegin;
3162   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3163   PetscValidHeaderSpecific(points, IS_CLASSID, 2);
3164   if (depth) PetscAssertPointer(depth, 3);
3165   if (expandedPoints) PetscAssertPointer(expandedPoints, 4);
3166   if (sections) PetscAssertPointer(sections, 5);
3167   PetscCall(ISGetLocalSize(points, &n));
3168   PetscCall(ISGetIndices(points, &arr0));
3169   PetscCall(DMPlexGetDepth(dm, &depth_));
3170   PetscCall(PetscCalloc1(depth_, &expandedPoints_));
3171   PetscCall(PetscCalloc1(depth_, &sections_));
3172   arr = (PetscInt *)arr0; /* this is ok because first generation of arr is not modified */
3173   for (d = depth_ - 1; d >= 0; d--) {
3174     PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &sections_[d]));
3175     PetscCall(PetscSectionSetChart(sections_[d], 0, n));
3176     for (i = 0; i < n; i++) {
3177       PetscCall(DMPlexGetDepthStratum(dm, d + 1, &start, &end));
3178       if (arr[i] >= start && arr[i] < end) {
3179         PetscCall(DMPlexGetConeSize(dm, arr[i], &cn));
3180         PetscCall(PetscSectionSetDof(sections_[d], i, cn));
3181       } else {
3182         PetscCall(PetscSectionSetDof(sections_[d], i, 1));
3183       }
3184     }
3185     PetscCall(PetscSectionSetUp(sections_[d]));
3186     PetscCall(PetscSectionGetStorageSize(sections_[d], &newn));
3187     PetscCall(PetscMalloc1(newn, &newarr));
3188     for (i = 0; i < n; i++) {
3189       PetscCall(PetscSectionGetDof(sections_[d], i, &cn));
3190       PetscCall(PetscSectionGetOffset(sections_[d], i, &co));
3191       if (cn > 1) {
3192         PetscCall(DMPlexGetCone(dm, arr[i], &cone));
3193         PetscCall(PetscMemcpy(&newarr[co], cone, cn * sizeof(PetscInt)));
3194       } else {
3195         newarr[co] = arr[i];
3196       }
3197     }
3198     PetscCall(ISCreateGeneral(PETSC_COMM_SELF, newn, newarr, PETSC_OWN_POINTER, &expandedPoints_[d]));
3199     arr = newarr;
3200     n   = newn;
3201   }
3202   PetscCall(ISRestoreIndices(points, &arr0));
3203   *depth = depth_;
3204   if (expandedPoints) *expandedPoints = expandedPoints_;
3205   else {
3206     for (d = 0; d < depth_; d++) PetscCall(ISDestroy(&expandedPoints_[d]));
3207     PetscCall(PetscFree(expandedPoints_));
3208   }
3209   if (sections) *sections = sections_;
3210   else {
3211     for (d = 0; d < depth_; d++) PetscCall(PetscSectionDestroy(&sections_[d]));
3212     PetscCall(PetscFree(sections_));
3213   }
3214   PetscFunctionReturn(PETSC_SUCCESS);
3215 }
3216 
3217 /*@
3218   DMPlexRestoreConeRecursive - Deallocates arrays created by `DMPlexGetConeRecursive()`
3219 
3220   Not Collective
3221 
3222   Input Parameters:
3223 + dm     - The `DMPLEX`
3224 - points - The `IS` of points, which must lie in the chart set with `DMPlexSetChart()`
3225 
3226   Output Parameters:
3227 + depth          - (optional) Size of the output arrays, equal to `DMPLEX` depth, returned by `DMPlexGetDepth()`
3228 . expandedPoints - (optional) An array of recursively expanded cones
3229 - sections       - (optional) An array of sections which describe mappings from points to their cone points
3230 
3231   Level: advanced
3232 
3233   Note:
3234   See `DMPlexGetConeRecursive()`
3235 
3236 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexGetConeTuple()`, `DMPlexGetConeRecursive()`, `DMPlexGetConeRecursiveVertices()`,
3237           `DMPlexGetDepth()`, `IS`, `PetscSection`
3238 @*/
3239 PetscErrorCode DMPlexRestoreConeRecursive(DM dm, IS points, PetscInt *depth, IS *expandedPoints[], PetscSection *sections[])
3240 {
3241   PetscInt d, depth_;
3242 
3243   PetscFunctionBegin;
3244   PetscCall(DMPlexGetDepth(dm, &depth_));
3245   PetscCheck(!depth || *depth == depth_, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "depth changed since last call to DMPlexGetConeRecursive");
3246   if (depth) *depth = 0;
3247   if (expandedPoints) {
3248     for (d = 0; d < depth_; d++) PetscCall(ISDestroy(&((*expandedPoints)[d])));
3249     PetscCall(PetscFree(*expandedPoints));
3250   }
3251   if (sections) {
3252     for (d = 0; d < depth_; d++) PetscCall(PetscSectionDestroy(&((*sections)[d])));
3253     PetscCall(PetscFree(*sections));
3254   }
3255   PetscFunctionReturn(PETSC_SUCCESS);
3256 }
3257 
3258 /*@
3259   DMPlexSetCone - Set the points on the in-edges for this point in the DAG; that is these are the points that cover the specific point
3260 
3261   Not Collective
3262 
3263   Input Parameters:
3264 + dm   - The `DMPLEX`
3265 . p    - The point, which must lie in the chart set with `DMPlexSetChart()`
3266 - cone - An array of points which are on the in-edges for point `p`
3267 
3268   Level: beginner
3269 
3270   Note:
3271   This should be called after all calls to `DMPlexSetConeSize()` and `DMSetUp()`.
3272 
3273 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMSetUp()`, `DMPlexSetSupport()`, `DMPlexSetSupportSize()`
3274 @*/
3275 PetscErrorCode DMPlexSetCone(DM dm, PetscInt p, const PetscInt cone[])
3276 {
3277   DM_Plex *mesh = (DM_Plex *)dm->data;
3278   PetscInt dof, off, c;
3279 
3280   PetscFunctionBegin;
3281   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3282   PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3283   if (dof) PetscAssertPointer(cone, 3);
3284   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3285   if (PetscDefined(USE_DEBUG)) {
3286     PetscInt pStart, pEnd;
3287     PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, &pEnd));
3288     PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3289     for (c = 0; c < dof; ++c) {
3290       PetscCheck(!(cone[c] < pStart) && !(cone[c] >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", cone[c], pStart, pEnd);
3291       mesh->cones[off + c] = cone[c];
3292     }
3293   } else {
3294     for (c = 0; c < dof; ++c) mesh->cones[off + c] = cone[c];
3295   }
3296   PetscFunctionReturn(PETSC_SUCCESS);
3297 }
3298 
3299 /*@C
3300   DMPlexGetConeOrientation - Return the orientations on the in-edges for this point in the DAG
3301 
3302   Not Collective
3303 
3304   Input Parameters:
3305 + dm - The `DMPLEX`
3306 - p  - The point, which must lie in the chart set with `DMPlexSetChart()`
3307 
3308   Output Parameter:
3309 . coneOrientation - An array of orientations which are on the in-edges for point `p`. An orientation is an
3310                     integer giving the prescription for cone traversal.
3311 
3312   Level: beginner
3313 
3314   Note:
3315   The number indexes the symmetry transformations for the cell type (see manual). Orientation 0 is always
3316   the identity transformation. Negative orientation indicates reflection so that -(o+1) is the reflection
3317   of o, however it is not necessarily the inverse. To get the inverse, use `DMPolytopeTypeComposeOrientationInv()`
3318   with the identity.
3319 
3320   Fortran Notes:
3321   You must also call `DMPlexRestoreConeOrientation()` after you finish using the returned array.
3322   `DMPlexRestoreConeOrientation()` is not needed/available in C.
3323 
3324 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPolytopeTypeComposeOrientation()`, `DMPolytopeTypeComposeOrientationInv()`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexSetCone()`, `DMPlexSetChart()`
3325 @*/
3326 PetscErrorCode DMPlexGetConeOrientation(DM dm, PetscInt p, const PetscInt *coneOrientation[])
3327 {
3328   DM_Plex *mesh = (DM_Plex *)dm->data;
3329   PetscInt off;
3330 
3331   PetscFunctionBegin;
3332   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3333   if (PetscDefined(USE_DEBUG)) {
3334     PetscInt dof;
3335     PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3336     if (dof) PetscAssertPointer(coneOrientation, 3);
3337   }
3338   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3339 
3340   *coneOrientation = &mesh->coneOrientations[off];
3341   PetscFunctionReturn(PETSC_SUCCESS);
3342 }
3343 
3344 /*@
3345   DMPlexSetConeOrientation - Set the orientations on the in-edges for this point in the DAG
3346 
3347   Not Collective
3348 
3349   Input Parameters:
3350 + dm              - The `DMPLEX`
3351 . p               - The point, which must lie in the chart set with `DMPlexSetChart()`
3352 - coneOrientation - An array of orientations
3353 
3354   Level: beginner
3355 
3356   Notes:
3357   This should be called after all calls to `DMPlexSetConeSize()` and `DMSetUp()`.
3358 
3359   The meaning of coneOrientation is detailed in `DMPlexGetConeOrientation()`.
3360 
3361 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetConeOrientation()`, `DMPlexSetCone()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMSetUp()`
3362 @*/
3363 PetscErrorCode DMPlexSetConeOrientation(DM dm, PetscInt p, const PetscInt coneOrientation[])
3364 {
3365   DM_Plex *mesh = (DM_Plex *)dm->data;
3366   PetscInt pStart, pEnd;
3367   PetscInt dof, off, c;
3368 
3369   PetscFunctionBegin;
3370   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3371   PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3372   if (dof) PetscAssertPointer(coneOrientation, 3);
3373   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3374   if (PetscDefined(USE_DEBUG)) {
3375     PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, &pEnd));
3376     PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3377     for (c = 0; c < dof; ++c) {
3378       PetscInt cdof, o = coneOrientation[c];
3379 
3380       PetscCall(PetscSectionGetDof(mesh->coneSection, mesh->cones[off + c], &cdof));
3381       PetscCheck(!o || (o >= -(cdof + 1) && o < cdof), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone orientation %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ". %" PetscInt_FMT ")", o, -(cdof + 1), cdof);
3382       mesh->coneOrientations[off + c] = o;
3383     }
3384   } else {
3385     for (c = 0; c < dof; ++c) mesh->coneOrientations[off + c] = coneOrientation[c];
3386   }
3387   PetscFunctionReturn(PETSC_SUCCESS);
3388 }
3389 
3390 /*@
3391   DMPlexInsertCone - Insert a point into the in-edges for the point p in the DAG
3392 
3393   Not Collective
3394 
3395   Input Parameters:
3396 + dm        - The `DMPLEX`
3397 . p         - The point, which must lie in the chart set with `DMPlexSetChart()`
3398 . conePos   - The local index in the cone where the point should be put
3399 - conePoint - The mesh point to insert
3400 
3401   Level: beginner
3402 
3403 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMSetUp()`
3404 @*/
3405 PetscErrorCode DMPlexInsertCone(DM dm, PetscInt p, PetscInt conePos, PetscInt conePoint)
3406 {
3407   DM_Plex *mesh = (DM_Plex *)dm->data;
3408   PetscInt pStart, pEnd;
3409   PetscInt dof, off;
3410 
3411   PetscFunctionBegin;
3412   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3413   if (PetscDefined(USE_DEBUG)) {
3414     PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, &pEnd));
3415     PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3416     PetscCheck(!(conePoint < pStart) && !(conePoint >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", conePoint, pStart, pEnd);
3417     PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3418     PetscCheck(!(conePos < 0) && !(conePos >= dof), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone position %" PetscInt_FMT " of point %" PetscInt_FMT " is not in the valid range [0, %" PetscInt_FMT ")", conePos, p, dof);
3419   }
3420   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3421   mesh->cones[off + conePos] = conePoint;
3422   PetscFunctionReturn(PETSC_SUCCESS);
3423 }
3424 
3425 /*@
3426   DMPlexInsertConeOrientation - Insert a point orientation for the in-edge for the point p in the DAG
3427 
3428   Not Collective
3429 
3430   Input Parameters:
3431 + dm              - The `DMPLEX`
3432 . p               - The point, which must lie in the chart set with `DMPlexSetChart()`
3433 . conePos         - The local index in the cone where the point should be put
3434 - coneOrientation - The point orientation to insert
3435 
3436   Level: beginner
3437 
3438   Note:
3439   The meaning of coneOrientation values is detailed in `DMPlexGetConeOrientation()`.
3440 
3441 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMSetUp()`
3442 @*/
3443 PetscErrorCode DMPlexInsertConeOrientation(DM dm, PetscInt p, PetscInt conePos, PetscInt coneOrientation)
3444 {
3445   DM_Plex *mesh = (DM_Plex *)dm->data;
3446   PetscInt pStart, pEnd;
3447   PetscInt dof, off;
3448 
3449   PetscFunctionBegin;
3450   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3451   if (PetscDefined(USE_DEBUG)) {
3452     PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, &pEnd));
3453     PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3454     PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3455     PetscCheck(!(conePos < 0) && !(conePos >= dof), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cone position %" PetscInt_FMT " of point %" PetscInt_FMT " is not in the valid range [0, %" PetscInt_FMT ")", conePos, p, dof);
3456   }
3457   PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3458   mesh->coneOrientations[off + conePos] = coneOrientation;
3459   PetscFunctionReturn(PETSC_SUCCESS);
3460 }
3461 
3462 /*@C
3463   DMPlexGetOrientedCone - Return the points and orientations on the in-edges for this point in the DAG
3464 
3465   Not collective
3466 
3467   Input Parameters:
3468 + dm - The DMPlex
3469 - p  - The point, which must lie in the chart set with DMPlexSetChart()
3470 
3471   Output Parameters:
3472 + cone - An array of points which are on the in-edges for point `p`
3473 - ornt - An array of orientations which are on the in-edges for point `p`. An orientation is an
3474         integer giving the prescription for cone traversal.
3475 
3476   Level: beginner
3477 
3478   Notes:
3479   The number indexes the symmetry transformations for the cell type (see manual). Orientation 0 is always
3480   the identity transformation. Negative orientation indicates reflection so that -(o+1) is the reflection
3481   of o, however it is not necessarily the inverse. To get the inverse, use `DMPolytopeTypeComposeOrientationInv()`
3482   with the identity.
3483 
3484   Fortran Notes:
3485   You must also call `DMPlexRestoreCone()` after you finish using the returned array.
3486   `DMPlexRestoreCone()` is not needed/available in C.
3487 
3488 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexRestoreOrientedCone()`, `DMPlexGetConeSize()`, `DMPlexGetCone()`, `DMPlexGetChart()`
3489 @*/
3490 PetscErrorCode DMPlexGetOrientedCone(DM dm, PetscInt p, const PetscInt *cone[], const PetscInt *ornt[])
3491 {
3492   DM_Plex *mesh = (DM_Plex *)dm->data;
3493 
3494   PetscFunctionBegin;
3495   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3496   if (mesh->tr) {
3497     PetscCall(DMPlexTransformGetCone(mesh->tr, p, cone, ornt));
3498   } else {
3499     PetscInt off;
3500     if (PetscDefined(USE_DEBUG)) {
3501       PetscInt dof;
3502       PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
3503       if (dof) {
3504         if (cone) PetscAssertPointer(cone, 3);
3505         if (ornt) PetscAssertPointer(ornt, 4);
3506       }
3507     }
3508     PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
3509     if (cone) *cone = PetscSafePointerPlusOffset(mesh->cones, off);
3510     if (ornt) *ornt = PetscSafePointerPlusOffset(mesh->coneOrientations, off);
3511   }
3512   PetscFunctionReturn(PETSC_SUCCESS);
3513 }
3514 
3515 /*@C
3516   DMPlexRestoreOrientedCone - Restore the points and orientations on the in-edges for this point in the DAG
3517 
3518   Not Collective
3519 
3520   Input Parameters:
3521 + dm   - The DMPlex
3522 . p    - The point, which must lie in the chart set with `DMPlexSetChart()`
3523 . cone - An array of points which are on the in-edges for point p
3524 - ornt - An array of orientations which are on the in-edges for point `p`. An orientation is an
3525         integer giving the prescription for cone traversal.
3526 
3527   Level: beginner
3528 
3529   Notes:
3530   The number indexes the symmetry transformations for the cell type (see manual). Orientation 0 is always
3531   the identity transformation. Negative orientation indicates reflection so that -(o+1) is the reflection
3532   of o, however it is not necessarily the inverse. To get the inverse, use `DMPolytopeTypeComposeOrientationInv()`
3533   with the identity.
3534 
3535   Fortran Notes:
3536   You must also call `DMPlexRestoreCone()` after you finish using the returned array.
3537   `DMPlexRestoreCone()` is not needed/available in C.
3538 
3539 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetOrientedCone()`, `DMPlexGetConeSize()`, `DMPlexGetCone()`, `DMPlexGetChart()`
3540 @*/
3541 PetscErrorCode DMPlexRestoreOrientedCone(DM dm, PetscInt p, const PetscInt *cone[], const PetscInt *ornt[])
3542 {
3543   DM_Plex *mesh = (DM_Plex *)dm->data;
3544 
3545   PetscFunctionBegin;
3546   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3547   if (mesh->tr) PetscCall(DMPlexTransformRestoreCone(mesh->tr, p, cone, ornt));
3548   PetscFunctionReturn(PETSC_SUCCESS);
3549 }
3550 
3551 /*@
3552   DMPlexGetSupportSize - Return the number of out-edges for this point in the DAG
3553 
3554   Not Collective
3555 
3556   Input Parameters:
3557 + dm - The `DMPLEX`
3558 - p  - The point, which must lie in the chart set with `DMPlexSetChart()`
3559 
3560   Output Parameter:
3561 . size - The support size for point `p`
3562 
3563   Level: beginner
3564 
3565 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSetConeSize()`, `DMPlexSetChart()`, `DMPlexGetConeSize()`
3566 @*/
3567 PetscErrorCode DMPlexGetSupportSize(DM dm, PetscInt p, PetscInt *size)
3568 {
3569   DM_Plex *mesh = (DM_Plex *)dm->data;
3570 
3571   PetscFunctionBegin;
3572   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3573   PetscAssertPointer(size, 3);
3574   PetscCall(PetscSectionGetDof(mesh->supportSection, p, size));
3575   PetscFunctionReturn(PETSC_SUCCESS);
3576 }
3577 
3578 /*@
3579   DMPlexSetSupportSize - Set the number of out-edges for this point in the DAG
3580 
3581   Not Collective
3582 
3583   Input Parameters:
3584 + dm   - The `DMPLEX`
3585 . p    - The point, which must lie in the chart set with `DMPlexSetChart()`
3586 - size - The support size for point `p`
3587 
3588   Level: beginner
3589 
3590   Note:
3591   This should be called after `DMPlexSetChart()`.
3592 
3593 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetSupportSize()`, `DMPlexSetChart()`
3594 @*/
3595 PetscErrorCode DMPlexSetSupportSize(DM dm, PetscInt p, PetscInt size)
3596 {
3597   DM_Plex *mesh = (DM_Plex *)dm->data;
3598 
3599   PetscFunctionBegin;
3600   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3601   PetscCall(PetscSectionSetDof(mesh->supportSection, p, size));
3602   PetscFunctionReturn(PETSC_SUCCESS);
3603 }
3604 
3605 /*@C
3606   DMPlexGetSupport - Return the points on the out-edges for this point in the DAG
3607 
3608   Not Collective
3609 
3610   Input Parameters:
3611 + dm - The `DMPLEX`
3612 - p  - The point, which must lie in the chart set with `DMPlexSetChart()`
3613 
3614   Output Parameter:
3615 . support - An array of points which are on the out-edges for point `p`
3616 
3617   Level: beginner
3618 
3619   Fortran Notes:
3620   You must also call `DMPlexRestoreSupport()` after you finish using the returned array.
3621   `DMPlexRestoreSupport()` is not needed/available in C.
3622 
3623 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetSupportSize()`, `DMPlexSetSupport()`, `DMPlexGetCone()`, `DMPlexSetChart()`
3624 @*/
3625 PetscErrorCode DMPlexGetSupport(DM dm, PetscInt p, const PetscInt *support[])
3626 {
3627   DM_Plex *mesh = (DM_Plex *)dm->data;
3628   PetscInt off;
3629 
3630   PetscFunctionBegin;
3631   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3632   PetscAssertPointer(support, 3);
3633   PetscCall(PetscSectionGetOffset(mesh->supportSection, p, &off));
3634   *support = PetscSafePointerPlusOffset(mesh->supports, off);
3635   PetscFunctionReturn(PETSC_SUCCESS);
3636 }
3637 
3638 /*@
3639   DMPlexSetSupport - Set the points on the out-edges for this point in the DAG, that is the list of points that this point covers
3640 
3641   Not Collective
3642 
3643   Input Parameters:
3644 + dm      - The `DMPLEX`
3645 . p       - The point, which must lie in the chart set with `DMPlexSetChart()`
3646 - support - An array of points which are on the out-edges for point `p`
3647 
3648   Level: beginner
3649 
3650   Note:
3651   This should be called after all calls to `DMPlexSetSupportSize()` and `DMSetUp()`.
3652 
3653 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexSetCone()`, `DMPlexSetConeSize()`, `DMPlexCreate()`, `DMPlexGetSupport()`, `DMPlexSetChart()`, `DMPlexSetSupportSize()`, `DMSetUp()`
3654 @*/
3655 PetscErrorCode DMPlexSetSupport(DM dm, PetscInt p, const PetscInt support[])
3656 {
3657   DM_Plex *mesh = (DM_Plex *)dm->data;
3658   PetscInt pStart, pEnd;
3659   PetscInt dof, off, c;
3660 
3661   PetscFunctionBegin;
3662   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3663   PetscCall(PetscSectionGetChart(mesh->supportSection, &pStart, &pEnd));
3664   PetscCall(PetscSectionGetDof(mesh->supportSection, p, &dof));
3665   if (dof) PetscAssertPointer(support, 3);
3666   PetscCall(PetscSectionGetOffset(mesh->supportSection, p, &off));
3667   PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3668   for (c = 0; c < dof; ++c) {
3669     PetscCheck(!(support[c] < pStart) && !(support[c] >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Support point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", support[c], pStart, pEnd);
3670     mesh->supports[off + c] = support[c];
3671   }
3672   PetscFunctionReturn(PETSC_SUCCESS);
3673 }
3674 
3675 /*@
3676   DMPlexInsertSupport - Insert a point into the out-edges for the point p in the DAG
3677 
3678   Not Collective
3679 
3680   Input Parameters:
3681 + dm           - The `DMPLEX`
3682 . p            - The point, which must lie in the chart set with `DMPlexSetChart()`
3683 . supportPos   - The local index in the cone where the point should be put
3684 - supportPoint - The mesh point to insert
3685 
3686   Level: beginner
3687 
3688 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexGetCone()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMSetUp()`
3689 @*/
3690 PetscErrorCode DMPlexInsertSupport(DM dm, PetscInt p, PetscInt supportPos, PetscInt supportPoint)
3691 {
3692   DM_Plex *mesh = (DM_Plex *)dm->data;
3693   PetscInt pStart, pEnd;
3694   PetscInt dof, off;
3695 
3696   PetscFunctionBegin;
3697   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3698   PetscCall(PetscSectionGetChart(mesh->supportSection, &pStart, &pEnd));
3699   PetscCall(PetscSectionGetDof(mesh->supportSection, p, &dof));
3700   PetscCall(PetscSectionGetOffset(mesh->supportSection, p, &off));
3701   PetscCheck(!(p < pStart) && !(p >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Mesh point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, pStart, pEnd);
3702   PetscCheck(!(supportPoint < pStart) && !(supportPoint >= pEnd), PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Support point %" PetscInt_FMT " is not in the valid range [%" PetscInt_FMT ", %" PetscInt_FMT ")", supportPoint, pStart, pEnd);
3703   PetscCheck(supportPos < dof, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Support position %" PetscInt_FMT " of point %" PetscInt_FMT " is not in the valid range [0, %" PetscInt_FMT ")", supportPos, p, dof);
3704   mesh->supports[off + supportPos] = supportPoint;
3705   PetscFunctionReturn(PETSC_SUCCESS);
3706 }
3707 
3708 /* Converts an orientation o in the current numbering to the previous scheme used in Plex */
3709 PetscInt DMPolytopeConvertNewOrientation_Internal(DMPolytopeType ct, PetscInt o)
3710 {
3711   switch (ct) {
3712   case DM_POLYTOPE_SEGMENT:
3713     if (o == -1) return -2;
3714     break;
3715   case DM_POLYTOPE_TRIANGLE:
3716     if (o == -3) return -1;
3717     if (o == -2) return -3;
3718     if (o == -1) return -2;
3719     break;
3720   case DM_POLYTOPE_QUADRILATERAL:
3721     if (o == -4) return -2;
3722     if (o == -3) return -1;
3723     if (o == -2) return -4;
3724     if (o == -1) return -3;
3725     break;
3726   default:
3727     return o;
3728   }
3729   return o;
3730 }
3731 
3732 /* Converts an orientation o in the previous scheme used in Plex to the current numbering */
3733 PetscInt DMPolytopeConvertOldOrientation_Internal(DMPolytopeType ct, PetscInt o)
3734 {
3735   switch (ct) {
3736   case DM_POLYTOPE_SEGMENT:
3737     if ((o == -2) || (o == 1)) return -1;
3738     if (o == -1) return 0;
3739     break;
3740   case DM_POLYTOPE_TRIANGLE:
3741     if (o == -3) return -2;
3742     if (o == -2) return -1;
3743     if (o == -1) return -3;
3744     break;
3745   case DM_POLYTOPE_QUADRILATERAL:
3746     if (o == -4) return -2;
3747     if (o == -3) return -1;
3748     if (o == -2) return -4;
3749     if (o == -1) return -3;
3750     break;
3751   default:
3752     return o;
3753   }
3754   return o;
3755 }
3756 
3757 /* Takes in a mesh whose orientations are in the previous scheme and converts them all to the current numbering */
3758 PetscErrorCode DMPlexConvertOldOrientations_Internal(DM dm)
3759 {
3760   PetscInt pStart, pEnd, p;
3761 
3762   PetscFunctionBegin;
3763   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
3764   for (p = pStart; p < pEnd; ++p) {
3765     const PetscInt *cone, *ornt;
3766     PetscInt        coneSize, c;
3767 
3768     PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
3769     PetscCall(DMPlexGetCone(dm, p, &cone));
3770     PetscCall(DMPlexGetConeOrientation(dm, p, &ornt));
3771     for (c = 0; c < coneSize; ++c) {
3772       DMPolytopeType ct;
3773       const PetscInt o = ornt[c];
3774 
3775       PetscCall(DMPlexGetCellType(dm, cone[c], &ct));
3776       switch (ct) {
3777       case DM_POLYTOPE_SEGMENT:
3778         if ((o == -2) || (o == 1)) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -1));
3779         if (o == -1) PetscCall(DMPlexInsertConeOrientation(dm, p, c, 0));
3780         break;
3781       case DM_POLYTOPE_TRIANGLE:
3782         if (o == -3) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -2));
3783         if (o == -2) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -1));
3784         if (o == -1) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -3));
3785         break;
3786       case DM_POLYTOPE_QUADRILATERAL:
3787         if (o == -4) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -2));
3788         if (o == -3) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -1));
3789         if (o == -2) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -4));
3790         if (o == -1) PetscCall(DMPlexInsertConeOrientation(dm, p, c, -3));
3791         break;
3792       default:
3793         break;
3794       }
3795     }
3796   }
3797   PetscFunctionReturn(PETSC_SUCCESS);
3798 }
3799 
3800 static inline PetscErrorCode DMPlexGetTransitiveClosure_Hot_Private(DM dm, PetscInt p, PetscBool useCone, PetscInt *size, const PetscInt *arr[], const PetscInt *ornt[])
3801 {
3802   DM_Plex *mesh = (DM_Plex *)dm->data;
3803 
3804   PetscFunctionBeginHot;
3805   if (PetscDefined(USE_DEBUG) || mesh->tr) {
3806     if (useCone) {
3807       PetscCall(DMPlexGetConeSize(dm, p, size));
3808       PetscCall(DMPlexGetOrientedCone(dm, p, arr, ornt));
3809     } else {
3810       PetscCall(DMPlexGetSupportSize(dm, p, size));
3811       PetscCall(DMPlexGetSupport(dm, p, arr));
3812     }
3813   } else {
3814     if (useCone) {
3815       const PetscSection s   = mesh->coneSection;
3816       const PetscInt     ps  = p - s->pStart;
3817       const PetscInt     off = s->atlasOff[ps];
3818 
3819       *size = s->atlasDof[ps];
3820       *arr  = mesh->cones + off;
3821       *ornt = mesh->coneOrientations + off;
3822     } else {
3823       const PetscSection s   = mesh->supportSection;
3824       const PetscInt     ps  = p - s->pStart;
3825       const PetscInt     off = s->atlasOff[ps];
3826 
3827       *size = s->atlasDof[ps];
3828       *arr  = mesh->supports + off;
3829     }
3830   }
3831   PetscFunctionReturn(PETSC_SUCCESS);
3832 }
3833 
3834 static inline PetscErrorCode DMPlexRestoreTransitiveClosure_Hot_Private(DM dm, PetscInt p, PetscBool useCone, PetscInt *size, const PetscInt *arr[], const PetscInt *ornt[])
3835 {
3836   DM_Plex *mesh = (DM_Plex *)dm->data;
3837 
3838   PetscFunctionBeginHot;
3839   if (PetscDefined(USE_DEBUG) || mesh->tr) {
3840     if (useCone) PetscCall(DMPlexRestoreOrientedCone(dm, p, arr, ornt));
3841   }
3842   PetscFunctionReturn(PETSC_SUCCESS);
3843 }
3844 
3845 static PetscErrorCode DMPlexGetTransitiveClosure_Depth1_Private(DM dm, PetscInt p, PetscInt ornt, PetscBool useCone, PetscInt *numPoints, PetscInt *points[])
3846 {
3847   DMPolytopeType  ct = DM_POLYTOPE_UNKNOWN;
3848   PetscInt       *closure;
3849   const PetscInt *tmp = NULL, *tmpO = NULL;
3850   PetscInt        off = 0, tmpSize, t;
3851 
3852   PetscFunctionBeginHot;
3853   if (ornt) {
3854     PetscCall(DMPlexGetCellType(dm, p, &ct));
3855     if (ct == DM_POLYTOPE_FV_GHOST || ct == DM_POLYTOPE_INTERIOR_GHOST || ct == DM_POLYTOPE_UNKNOWN || ct == DM_POLYTOPE_UNKNOWN_CELL || ct == DM_POLYTOPE_UNKNOWN_FACE) ct = DM_POLYTOPE_UNKNOWN;
3856   }
3857   if (*points) {
3858     closure = *points;
3859   } else {
3860     PetscInt maxConeSize, maxSupportSize;
3861     PetscCall(DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize));
3862     PetscCall(DMGetWorkArray(dm, 2 * (PetscMax(maxConeSize, maxSupportSize) + 1), MPIU_INT, &closure));
3863   }
3864   PetscCall(DMPlexGetTransitiveClosure_Hot_Private(dm, p, useCone, &tmpSize, &tmp, &tmpO));
3865   if (ct == DM_POLYTOPE_UNKNOWN) {
3866     closure[off++] = p;
3867     closure[off++] = 0;
3868     for (t = 0; t < tmpSize; ++t) {
3869       closure[off++] = tmp[t];
3870       closure[off++] = tmpO ? tmpO[t] : 0;
3871     }
3872   } else {
3873     const PetscInt *arr = DMPolytopeTypeGetArrangement(ct, ornt);
3874 
3875     /* We assume that cells with a valid type have faces with a valid type */
3876     closure[off++] = p;
3877     closure[off++] = ornt;
3878     for (t = 0; t < tmpSize; ++t) {
3879       DMPolytopeType ft;
3880 
3881       PetscCall(DMPlexGetCellType(dm, tmp[t], &ft));
3882       closure[off++] = tmp[arr[t]];
3883       closure[off++] = tmpO ? DMPolytopeTypeComposeOrientation(ft, ornt, tmpO[t]) : 0;
3884     }
3885   }
3886   PetscCall(DMPlexRestoreTransitiveClosure_Hot_Private(dm, p, useCone, &tmpSize, &tmp, &tmpO));
3887   if (numPoints) *numPoints = tmpSize + 1;
3888   if (points) *points = closure;
3889   PetscFunctionReturn(PETSC_SUCCESS);
3890 }
3891 
3892 /* We need a special tensor version because we want to allow duplicate points in the endcaps for hybrid cells */
3893 static PetscErrorCode DMPlexTransitiveClosure_Tensor_Internal(DM dm, PetscInt point, DMPolytopeType ct, PetscInt o, PetscBool useCone, PetscInt *numPoints, PetscInt **points)
3894 {
3895   const PetscInt *arr = DMPolytopeTypeGetArrangement(ct, o);
3896   const PetscInt *cone, *ornt;
3897   PetscInt       *pts, *closure = NULL;
3898   DMPolytopeType  ft;
3899   PetscInt        maxConeSize, maxSupportSize, coneSeries, supportSeries, maxSize;
3900   PetscInt        dim, coneSize, c, d, clSize, cl;
3901 
3902   PetscFunctionBeginHot;
3903   PetscCall(DMGetDimension(dm, &dim));
3904   PetscCall(DMPlexGetTransitiveClosure_Hot_Private(dm, point, PETSC_TRUE, &coneSize, &cone, &ornt));
3905   PetscCall(DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize));
3906   coneSeries    = (maxConeSize > 1) ? ((PetscPowInt(maxConeSize, dim + 1) - 1) / (maxConeSize - 1)) : dim + 1;
3907   supportSeries = (maxSupportSize > 1) ? ((PetscPowInt(maxSupportSize, dim + 1) - 1) / (maxSupportSize - 1)) : dim + 1;
3908   maxSize       = PetscMax(coneSeries, supportSeries);
3909   if (*points) {
3910     pts = *points;
3911   } else PetscCall(DMGetWorkArray(dm, 2 * maxSize, MPIU_INT, &pts));
3912   c        = 0;
3913   pts[c++] = point;
3914   pts[c++] = o;
3915   PetscCall(DMPlexGetCellType(dm, cone[arr[0 * 2 + 0]], &ft));
3916   PetscCall(DMPlexGetTransitiveClosure_Internal(dm, cone[arr[0 * 2 + 0]], DMPolytopeTypeComposeOrientation(ft, arr[0 * 2 + 1], ornt[0]), useCone, &clSize, &closure));
3917   for (cl = 0; cl < clSize * 2; cl += 2) {
3918     pts[c++] = closure[cl];
3919     pts[c++] = closure[cl + 1];
3920   }
3921   PetscCall(DMPlexGetTransitiveClosure_Internal(dm, cone[arr[1 * 2 + 0]], DMPolytopeTypeComposeOrientation(ft, arr[1 * 2 + 1], ornt[1]), useCone, &clSize, &closure));
3922   for (cl = 0; cl < clSize * 2; cl += 2) {
3923     pts[c++] = closure[cl];
3924     pts[c++] = closure[cl + 1];
3925   }
3926   PetscCall(DMPlexRestoreTransitiveClosure(dm, cone[0], useCone, &clSize, &closure));
3927   for (d = 2; d < coneSize; ++d) {
3928     PetscCall(DMPlexGetCellType(dm, cone[arr[d * 2 + 0]], &ft));
3929     pts[c++] = cone[arr[d * 2 + 0]];
3930     pts[c++] = DMPolytopeTypeComposeOrientation(ft, arr[d * 2 + 1], ornt[d]);
3931   }
3932   PetscCall(DMPlexRestoreTransitiveClosure_Hot_Private(dm, point, PETSC_TRUE, &coneSize, &cone, &ornt));
3933   if (dim >= 3) {
3934     for (d = 2; d < coneSize; ++d) {
3935       const PetscInt  fpoint = cone[arr[d * 2 + 0]];
3936       const PetscInt *fcone, *fornt;
3937       PetscInt        fconeSize, fc, i;
3938 
3939       PetscCall(DMPlexGetCellType(dm, fpoint, &ft));
3940       const PetscInt *farr = DMPolytopeTypeGetArrangement(ft, DMPolytopeTypeComposeOrientation(ft, arr[d * 2 + 1], ornt[d]));
3941       PetscCall(DMPlexGetTransitiveClosure_Hot_Private(dm, fpoint, PETSC_TRUE, &fconeSize, &fcone, &fornt));
3942       for (fc = 0; fc < fconeSize; ++fc) {
3943         const PetscInt cp = fcone[farr[fc * 2 + 0]];
3944         const PetscInt co = farr[fc * 2 + 1];
3945 
3946         for (i = 0; i < c; i += 2)
3947           if (pts[i] == cp) break;
3948         if (i == c) {
3949           PetscCall(DMPlexGetCellType(dm, cp, &ft));
3950           pts[c++] = cp;
3951           pts[c++] = DMPolytopeTypeComposeOrientation(ft, co, fornt[farr[fc * 2 + 0]]);
3952         }
3953       }
3954       PetscCall(DMPlexRestoreTransitiveClosure_Hot_Private(dm, fpoint, PETSC_TRUE, &fconeSize, &fcone, &fornt));
3955     }
3956   }
3957   *numPoints = c / 2;
3958   *points    = pts;
3959   PetscFunctionReturn(PETSC_SUCCESS);
3960 }
3961 
3962 PetscErrorCode DMPlexGetTransitiveClosure_Internal(DM dm, PetscInt p, PetscInt ornt, PetscBool useCone, PetscInt *numPoints, PetscInt *points[])
3963 {
3964   DMPolytopeType ct;
3965   PetscInt      *closure, *fifo;
3966   PetscInt       closureSize = 0, fifoStart = 0, fifoSize = 0;
3967   PetscInt       maxConeSize, maxSupportSize, coneSeries, supportSeries;
3968   PetscInt       depth, maxSize;
3969 
3970   PetscFunctionBeginHot;
3971   PetscCall(DMPlexGetDepth(dm, &depth));
3972   if (depth == 1) {
3973     PetscCall(DMPlexGetTransitiveClosure_Depth1_Private(dm, p, ornt, useCone, numPoints, points));
3974     PetscFunctionReturn(PETSC_SUCCESS);
3975   }
3976   PetscCall(DMPlexGetCellType(dm, p, &ct));
3977   if (ct == DM_POLYTOPE_FV_GHOST || ct == DM_POLYTOPE_INTERIOR_GHOST || ct == DM_POLYTOPE_UNKNOWN || ct == DM_POLYTOPE_UNKNOWN_CELL || ct == DM_POLYTOPE_UNKNOWN_FACE) ct = DM_POLYTOPE_UNKNOWN;
3978   if (DMPolytopeTypeIsHybrid(ct) && ct != DM_POLYTOPE_POINT_PRISM_TENSOR) {
3979     PetscCall(DMPlexTransitiveClosure_Tensor_Internal(dm, p, ct, ornt, useCone, numPoints, points));
3980     PetscFunctionReturn(PETSC_SUCCESS);
3981   }
3982   PetscCall(DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize));
3983   coneSeries    = (maxConeSize > 1) ? ((PetscPowInt(maxConeSize, depth + 1) - 1) / (maxConeSize - 1)) : depth + 1;
3984   supportSeries = (maxSupportSize > 1) ? ((PetscPowInt(maxSupportSize, depth + 1) - 1) / (maxSupportSize - 1)) : depth + 1;
3985   maxSize       = PetscMax(coneSeries, supportSeries);
3986   PetscCall(DMGetWorkArray(dm, 3 * maxSize, MPIU_INT, &fifo));
3987   if (*points) {
3988     closure = *points;
3989   } else PetscCall(DMGetWorkArray(dm, 2 * maxSize, MPIU_INT, &closure));
3990   closure[closureSize++] = p;
3991   closure[closureSize++] = ornt;
3992   fifo[fifoSize++]       = p;
3993   fifo[fifoSize++]       = ornt;
3994   fifo[fifoSize++]       = ct;
3995   /* Should kick out early when depth is reached, rather than checking all vertices for empty cones */
3996   while (fifoSize - fifoStart) {
3997     const PetscInt       q    = fifo[fifoStart++];
3998     const PetscInt       o    = fifo[fifoStart++];
3999     const DMPolytopeType qt   = (DMPolytopeType)fifo[fifoStart++];
4000     const PetscInt      *qarr = DMPolytopeTypeGetArrangement(qt, o);
4001     const PetscInt      *tmp, *tmpO = NULL;
4002     PetscInt             tmpSize, t;
4003 
4004     if (PetscDefined(USE_DEBUG)) {
4005       PetscInt nO = DMPolytopeTypeGetNumArrangements(qt) / 2;
4006       PetscCheck(!o || !(o >= nO || o < -nO), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid orientation %" PetscInt_FMT " not in [%" PetscInt_FMT ",%" PetscInt_FMT ") for %s %" PetscInt_FMT, o, -nO, nO, DMPolytopeTypes[qt], q);
4007     }
4008     PetscCall(DMPlexGetTransitiveClosure_Hot_Private(dm, q, useCone, &tmpSize, &tmp, &tmpO));
4009     for (t = 0; t < tmpSize; ++t) {
4010       const PetscInt ip = useCone && qarr ? qarr[t * 2] : t;
4011       const PetscInt io = useCone && qarr ? qarr[t * 2 + 1] : 0;
4012       const PetscInt cp = tmp[ip];
4013       PetscCall(DMPlexGetCellType(dm, cp, &ct));
4014       const PetscInt co = tmpO ? DMPolytopeTypeComposeOrientation(ct, io, tmpO[ip]) : 0;
4015       PetscInt       c;
4016 
4017       /* Check for duplicate */
4018       for (c = 0; c < closureSize; c += 2) {
4019         if (closure[c] == cp) break;
4020       }
4021       if (c == closureSize) {
4022         closure[closureSize++] = cp;
4023         closure[closureSize++] = co;
4024         fifo[fifoSize++]       = cp;
4025         fifo[fifoSize++]       = co;
4026         fifo[fifoSize++]       = ct;
4027       }
4028     }
4029     PetscCall(DMPlexRestoreTransitiveClosure_Hot_Private(dm, q, useCone, &tmpSize, &tmp, &tmpO));
4030   }
4031   PetscCall(DMRestoreWorkArray(dm, 3 * maxSize, MPIU_INT, &fifo));
4032   if (numPoints) *numPoints = closureSize / 2;
4033   if (points) *points = closure;
4034   PetscFunctionReturn(PETSC_SUCCESS);
4035 }
4036 
4037 /*@C
4038   DMPlexGetTransitiveClosure - Return the points on the transitive closure of the in-edges or out-edges for this point in the DAG
4039 
4040   Not Collective
4041 
4042   Input Parameters:
4043 + dm      - The `DMPLEX`
4044 . p       - The mesh point
4045 - useCone - `PETSC_TRUE` for the closure, otherwise return the star
4046 
4047   Input/Output Parameter:
4048 . points - The points and point orientations, interleaved as pairs [p0, o0, p1, o1, ...];
4049            if `NULL` on input, internal storage will be returned, otherwise the provided array is used
4050 
4051   Output Parameter:
4052 . numPoints - The number of points in the closure, so points[] is of size 2*`numPoints`
4053 
4054   Level: beginner
4055 
4056   Note:
4057   If using internal storage (points is `NULL` on input), each call overwrites the last output.
4058 
4059   Fortran Notes:
4060   The `numPoints` argument is not present in the Fortran binding since it is internal to the array.
4061 
4062 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexRestoreTransitiveClosure()`, `DMPlexCreate()`, `DMPlexSetCone()`, `DMPlexSetChart()`, `DMPlexGetCone()`
4063 @*/
4064 PetscErrorCode DMPlexGetTransitiveClosure(DM dm, PetscInt p, PetscBool useCone, PetscInt *numPoints, PetscInt *points[])
4065 {
4066   PetscFunctionBeginHot;
4067   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4068   if (numPoints) PetscAssertPointer(numPoints, 4);
4069   if (points) PetscAssertPointer(points, 5);
4070   PetscCall(DMPlexGetTransitiveClosure_Internal(dm, p, 0, useCone, numPoints, points));
4071   PetscFunctionReturn(PETSC_SUCCESS);
4072 }
4073 
4074 /*@C
4075   DMPlexRestoreTransitiveClosure - Restore the array of points on the transitive closure of the in-edges or out-edges for this point in the DAG
4076 
4077   Not Collective
4078 
4079   Input Parameters:
4080 + dm        - The `DMPLEX`
4081 . p         - The mesh point
4082 . useCone   - `PETSC_TRUE` for the closure, otherwise return the star
4083 . numPoints - The number of points in the closure, so points[] is of size 2*`numPoints`
4084 - points    - The points and point orientations, interleaved as pairs [p0, o0, p1, o1, ...]
4085 
4086   Level: beginner
4087 
4088   Note:
4089   If not using internal storage (points is not `NULL` on input), this call is unnecessary
4090 
4091 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetTransitiveClosure()`, `DMPlexCreate()`, `DMPlexSetCone()`, `DMPlexSetChart()`, `DMPlexGetCone()`
4092 @*/
4093 PetscErrorCode DMPlexRestoreTransitiveClosure(DM dm, PetscInt p, PetscBool useCone, PetscInt *numPoints, PetscInt *points[])
4094 {
4095   PetscFunctionBeginHot;
4096   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4097   if (numPoints) *numPoints = 0;
4098   PetscCall(DMRestoreWorkArray(dm, 0, MPIU_INT, points));
4099   PetscFunctionReturn(PETSC_SUCCESS);
4100 }
4101 
4102 /*@
4103   DMPlexGetMaxSizes - Return the maximum number of in-edges (cone) and out-edges (support) for any point in the DAG
4104 
4105   Not Collective
4106 
4107   Input Parameter:
4108 . dm - The `DMPLEX`
4109 
4110   Output Parameters:
4111 + maxConeSize    - The maximum number of in-edges
4112 - maxSupportSize - The maximum number of out-edges
4113 
4114   Level: beginner
4115 
4116 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSetConeSize()`, `DMPlexSetChart()`
4117 @*/
4118 PetscErrorCode DMPlexGetMaxSizes(DM dm, PetscInt *maxConeSize, PetscInt *maxSupportSize)
4119 {
4120   DM_Plex *mesh = (DM_Plex *)dm->data;
4121 
4122   PetscFunctionBegin;
4123   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4124   if (maxConeSize) PetscCall(PetscSectionGetMaxDof(mesh->coneSection, maxConeSize));
4125   if (maxSupportSize) PetscCall(PetscSectionGetMaxDof(mesh->supportSection, maxSupportSize));
4126   PetscFunctionReturn(PETSC_SUCCESS);
4127 }
4128 
4129 PetscErrorCode DMSetUp_Plex(DM dm)
4130 {
4131   DM_Plex *mesh = (DM_Plex *)dm->data;
4132   PetscInt size, maxSupportSize;
4133 
4134   PetscFunctionBegin;
4135   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4136   PetscCall(PetscSectionSetUp(mesh->coneSection));
4137   PetscCall(PetscSectionGetStorageSize(mesh->coneSection, &size));
4138   PetscCall(PetscMalloc1(size, &mesh->cones));
4139   PetscCall(PetscCalloc1(size, &mesh->coneOrientations));
4140   PetscCall(PetscSectionGetMaxDof(mesh->supportSection, &maxSupportSize));
4141   if (maxSupportSize) {
4142     PetscCall(PetscSectionSetUp(mesh->supportSection));
4143     PetscCall(PetscSectionGetStorageSize(mesh->supportSection, &size));
4144     PetscCall(PetscMalloc1(size, &mesh->supports));
4145   }
4146   PetscFunctionReturn(PETSC_SUCCESS);
4147 }
4148 
4149 PetscErrorCode DMCreateSubDM_Plex(DM dm, PetscInt numFields, const PetscInt fields[], IS *is, DM *subdm)
4150 {
4151   PetscFunctionBegin;
4152   if (subdm) PetscCall(DMClone(dm, subdm));
4153   PetscCall(DMCreateSectionSubDM(dm, numFields, fields, is, subdm));
4154   if (subdm) (*subdm)->useNatural = dm->useNatural;
4155   if (dm->useNatural && dm->sfMigration) {
4156     PetscSF sfNatural;
4157 
4158     (*subdm)->sfMigration = dm->sfMigration;
4159     PetscCall(PetscObjectReference((PetscObject)dm->sfMigration));
4160     PetscCall(DMPlexCreateGlobalToNaturalSF(*subdm, NULL, (*subdm)->sfMigration, &sfNatural));
4161     (*subdm)->sfNatural = sfNatural;
4162   }
4163   PetscFunctionReturn(PETSC_SUCCESS);
4164 }
4165 
4166 PetscErrorCode DMCreateSuperDM_Plex(DM dms[], PetscInt len, IS **is, DM *superdm)
4167 {
4168   PetscInt i = 0;
4169 
4170   PetscFunctionBegin;
4171   PetscCall(DMClone(dms[0], superdm));
4172   PetscCall(DMCreateSectionSuperDM(dms, len, is, superdm));
4173   (*superdm)->useNatural = PETSC_FALSE;
4174   for (i = 0; i < len; i++) {
4175     if (dms[i]->useNatural && dms[i]->sfMigration) {
4176       PetscSF sfNatural;
4177 
4178       (*superdm)->sfMigration = dms[i]->sfMigration;
4179       PetscCall(PetscObjectReference((PetscObject)dms[i]->sfMigration));
4180       (*superdm)->useNatural = PETSC_TRUE;
4181       PetscCall(DMPlexCreateGlobalToNaturalSF(*superdm, NULL, (*superdm)->sfMigration, &sfNatural));
4182       (*superdm)->sfNatural = sfNatural;
4183       break;
4184     }
4185   }
4186   PetscFunctionReturn(PETSC_SUCCESS);
4187 }
4188 
4189 /*@
4190   DMPlexSymmetrize - Create support (out-edge) information from cone (in-edge) information
4191 
4192   Not Collective
4193 
4194   Input Parameter:
4195 . dm - The `DMPLEX`
4196 
4197   Level: beginner
4198 
4199   Note:
4200   This should be called after all calls to `DMPlexSetCone()`
4201 
4202 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSetChart()`, `DMPlexSetConeSize()`, `DMPlexSetCone()`
4203 @*/
4204 PetscErrorCode DMPlexSymmetrize(DM dm)
4205 {
4206   DM_Plex  *mesh = (DM_Plex *)dm->data;
4207   PetscInt *offsets;
4208   PetscInt  supportSize;
4209   PetscInt  pStart, pEnd, p;
4210 
4211   PetscFunctionBegin;
4212   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4213   PetscCheck(!mesh->supports, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "Supports were already setup in this DMPlex");
4214   PetscCall(PetscLogEventBegin(DMPLEX_Symmetrize, dm, 0, 0, 0));
4215   /* Calculate support sizes */
4216   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4217   for (p = pStart; p < pEnd; ++p) {
4218     PetscInt dof, off, c;
4219 
4220     PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
4221     PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
4222     for (c = off; c < off + dof; ++c) PetscCall(PetscSectionAddDof(mesh->supportSection, mesh->cones[c], 1));
4223   }
4224   PetscCall(PetscSectionSetUp(mesh->supportSection));
4225   /* Calculate supports */
4226   PetscCall(PetscSectionGetStorageSize(mesh->supportSection, &supportSize));
4227   PetscCall(PetscMalloc1(supportSize, &mesh->supports));
4228   PetscCall(PetscCalloc1(pEnd - pStart, &offsets));
4229   for (p = pStart; p < pEnd; ++p) {
4230     PetscInt dof, off, c;
4231 
4232     PetscCall(PetscSectionGetDof(mesh->coneSection, p, &dof));
4233     PetscCall(PetscSectionGetOffset(mesh->coneSection, p, &off));
4234     for (c = off; c < off + dof; ++c) {
4235       const PetscInt q = mesh->cones[c];
4236       PetscInt       offS;
4237 
4238       PetscCall(PetscSectionGetOffset(mesh->supportSection, q, &offS));
4239 
4240       mesh->supports[offS + offsets[q]] = p;
4241       ++offsets[q];
4242     }
4243   }
4244   PetscCall(PetscFree(offsets));
4245   PetscCall(PetscLogEventEnd(DMPLEX_Symmetrize, dm, 0, 0, 0));
4246   PetscFunctionReturn(PETSC_SUCCESS);
4247 }
4248 
4249 static PetscErrorCode DMPlexCreateDepthStratum(DM dm, DMLabel label, PetscInt depth, PetscInt pStart, PetscInt pEnd)
4250 {
4251   IS stratumIS;
4252 
4253   PetscFunctionBegin;
4254   if (pStart >= pEnd) PetscFunctionReturn(PETSC_SUCCESS);
4255   if (PetscDefined(USE_DEBUG)) {
4256     PetscInt  qStart, qEnd, numLevels, level;
4257     PetscBool overlap = PETSC_FALSE;
4258     PetscCall(DMLabelGetNumValues(label, &numLevels));
4259     for (level = 0; level < numLevels; level++) {
4260       PetscCall(DMLabelGetStratumBounds(label, level, &qStart, &qEnd));
4261       if ((pStart >= qStart && pStart < qEnd) || (pEnd > qStart && pEnd <= qEnd)) {
4262         overlap = PETSC_TRUE;
4263         break;
4264       }
4265     }
4266     PetscCheck(!overlap, PETSC_COMM_SELF, PETSC_ERR_PLIB, "New depth %" PetscInt_FMT " range [%" PetscInt_FMT ",%" PetscInt_FMT ") overlaps with depth %" PetscInt_FMT " range [%" PetscInt_FMT ",%" PetscInt_FMT ")", depth, pStart, pEnd, level, qStart, qEnd);
4267   }
4268   PetscCall(ISCreateStride(PETSC_COMM_SELF, pEnd - pStart, pStart, 1, &stratumIS));
4269   PetscCall(DMLabelSetStratumIS(label, depth, stratumIS));
4270   PetscCall(ISDestroy(&stratumIS));
4271   PetscFunctionReturn(PETSC_SUCCESS);
4272 }
4273 
4274 static PetscErrorCode DMPlexStratify_CellType_Private(DM dm, DMLabel label)
4275 {
4276   PetscInt *pMin, *pMax;
4277   PetscInt  pStart, pEnd;
4278   PetscInt  dmin = PETSC_MAX_INT, dmax = PETSC_MIN_INT;
4279 
4280   PetscFunctionBegin;
4281   {
4282     DMLabel label2;
4283 
4284     PetscCall(DMPlexGetCellTypeLabel(dm, &label2));
4285     PetscCall(PetscObjectViewFromOptions((PetscObject)label2, NULL, "-ct_view"));
4286   }
4287   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4288   for (PetscInt p = pStart; p < pEnd; ++p) {
4289     DMPolytopeType ct;
4290 
4291     PetscCall(DMPlexGetCellType(dm, p, &ct));
4292     dmin = PetscMin(DMPolytopeTypeGetDim(ct), dmin);
4293     dmax = PetscMax(DMPolytopeTypeGetDim(ct), dmax);
4294   }
4295   PetscCall(PetscMalloc2(dmax + 1, &pMin, dmax + 1, &pMax));
4296   for (PetscInt d = dmin; d <= dmax; ++d) {
4297     pMin[d] = PETSC_MAX_INT;
4298     pMax[d] = PETSC_MIN_INT;
4299   }
4300   for (PetscInt p = pStart; p < pEnd; ++p) {
4301     DMPolytopeType ct;
4302     PetscInt       d;
4303 
4304     PetscCall(DMPlexGetCellType(dm, p, &ct));
4305     d       = DMPolytopeTypeGetDim(ct);
4306     pMin[d] = PetscMin(p, pMin[d]);
4307     pMax[d] = PetscMax(p, pMax[d]);
4308   }
4309   for (PetscInt d = dmin; d <= dmax; ++d) {
4310     if (pMin[d] > pMax[d]) continue;
4311     PetscCall(DMPlexCreateDepthStratum(dm, label, d, pMin[d], pMax[d] + 1));
4312   }
4313   PetscCall(PetscFree2(pMin, pMax));
4314   PetscFunctionReturn(PETSC_SUCCESS);
4315 }
4316 
4317 static PetscErrorCode DMPlexStratify_Topological_Private(DM dm, DMLabel label)
4318 {
4319   PetscInt pStart, pEnd;
4320   PetscInt numRoots = 0, numLeaves = 0;
4321 
4322   PetscFunctionBegin;
4323   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4324   {
4325     /* Initialize roots and count leaves */
4326     PetscInt sMin = PETSC_MAX_INT;
4327     PetscInt sMax = PETSC_MIN_INT;
4328     PetscInt coneSize, supportSize;
4329 
4330     for (PetscInt p = pStart; p < pEnd; ++p) {
4331       PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
4332       PetscCall(DMPlexGetSupportSize(dm, p, &supportSize));
4333       if (!coneSize && supportSize) {
4334         sMin = PetscMin(p, sMin);
4335         sMax = PetscMax(p, sMax);
4336         ++numRoots;
4337       } else if (!supportSize && coneSize) {
4338         ++numLeaves;
4339       } else if (!supportSize && !coneSize) {
4340         /* Isolated points */
4341         sMin = PetscMin(p, sMin);
4342         sMax = PetscMax(p, sMax);
4343       }
4344     }
4345     PetscCall(DMPlexCreateDepthStratum(dm, label, 0, sMin, sMax + 1));
4346   }
4347 
4348   if (numRoots + numLeaves == (pEnd - pStart)) {
4349     PetscInt sMin = PETSC_MAX_INT;
4350     PetscInt sMax = PETSC_MIN_INT;
4351     PetscInt coneSize, supportSize;
4352 
4353     for (PetscInt p = pStart; p < pEnd; ++p) {
4354       PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
4355       PetscCall(DMPlexGetSupportSize(dm, p, &supportSize));
4356       if (!supportSize && coneSize) {
4357         sMin = PetscMin(p, sMin);
4358         sMax = PetscMax(p, sMax);
4359       }
4360     }
4361     PetscCall(DMPlexCreateDepthStratum(dm, label, 1, sMin, sMax + 1));
4362   } else {
4363     PetscInt level = 0;
4364     PetscInt qStart, qEnd;
4365 
4366     PetscCall(DMLabelGetStratumBounds(label, level, &qStart, &qEnd));
4367     while (qEnd > qStart) {
4368       PetscInt sMin = PETSC_MAX_INT;
4369       PetscInt sMax = PETSC_MIN_INT;
4370 
4371       for (PetscInt q = qStart; q < qEnd; ++q) {
4372         const PetscInt *support;
4373         PetscInt        supportSize;
4374 
4375         PetscCall(DMPlexGetSupportSize(dm, q, &supportSize));
4376         PetscCall(DMPlexGetSupport(dm, q, &support));
4377         for (PetscInt s = 0; s < supportSize; ++s) {
4378           sMin = PetscMin(support[s], sMin);
4379           sMax = PetscMax(support[s], sMax);
4380         }
4381       }
4382       PetscCall(DMLabelGetNumValues(label, &level));
4383       PetscCall(DMPlexCreateDepthStratum(dm, label, level, sMin, sMax + 1));
4384       PetscCall(DMLabelGetStratumBounds(label, level, &qStart, &qEnd));
4385     }
4386   }
4387   PetscFunctionReturn(PETSC_SUCCESS);
4388 }
4389 
4390 /*@
4391   DMPlexStratify - Computes the strata for all points in the `DMPLEX`
4392 
4393   Collective
4394 
4395   Input Parameter:
4396 . dm - The `DMPLEX`
4397 
4398   Level: beginner
4399 
4400   Notes:
4401   The strata group all points of the same grade, and this function calculates the strata. This
4402   grade can be seen as the height (or depth) of the point in the DAG.
4403 
4404   The DAG for most topologies is a graded poset (https://en.wikipedia.org/wiki/Graded_poset), and
4405   can be illustrated by a Hasse Diagram (https://en.wikipedia.org/wiki/Hasse_diagram).
4406   Concretely, `DMPlexStratify()` creates a new label named "depth" containing the depth in the DAG of each point. For cell-vertex
4407   meshes, vertices are depth 0 and cells are depth 1. For fully interpolated meshes, depth 0 for vertices, 1 for edges, and so on
4408   until cells have depth equal to the dimension of the mesh. The depth label can be accessed through `DMPlexGetDepthLabel()` or `DMPlexGetDepthStratum()`, or
4409   manually via `DMGetLabel()`.  The height is defined implicitly by height = maxDimension - depth, and can be accessed
4410   via `DMPlexGetHeightStratum()`.  For example, cells have height 0 and faces have height 1.
4411 
4412   The depth of a point is calculated by executing a breadth-first search (BFS) on the DAG. This could produce surprising results
4413   if run on a partially interpolated mesh, meaning one that had some edges and faces, but not others. For example, suppose that
4414   we had a mesh consisting of one triangle (c0) and three vertices (v0, v1, v2), and only one edge is on the boundary so we choose
4415   to interpolate only that one (e0), so that
4416 .vb
4417   cone(c0) = {e0, v2}
4418   cone(e0) = {v0, v1}
4419 .ve
4420   If `DMPlexStratify()` is run on this mesh, it will give depths
4421 .vb
4422    depth 0 = {v0, v1, v2}
4423    depth 1 = {e0, c0}
4424 .ve
4425   where the triangle has been given depth 1, instead of 2, because it is reachable from vertex v2.
4426 
4427   `DMPlexStratify()` should be called after all calls to `DMPlexSymmetrize()`
4428 
4429 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSymmetrize()`, `DMPlexComputeCellTypes()`
4430 @*/
4431 PetscErrorCode DMPlexStratify(DM dm)
4432 {
4433   DM_Plex  *mesh = (DM_Plex *)dm->data;
4434   DMLabel   label;
4435   PetscBool flg = PETSC_FALSE;
4436 
4437   PetscFunctionBegin;
4438   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4439   PetscCall(PetscLogEventBegin(DMPLEX_Stratify, dm, 0, 0, 0));
4440 
4441   // Create depth label
4442   PetscCall(DMCreateLabel(dm, "depth"));
4443   PetscCall(DMPlexGetDepthLabel(dm, &label));
4444 
4445   PetscCall(PetscOptionsGetBool(NULL, dm->hdr.prefix, "-dm_plex_stratify_celltype", &flg, NULL));
4446   if (flg) PetscCall(DMPlexStratify_CellType_Private(dm, label));
4447   else PetscCall(DMPlexStratify_Topological_Private(dm, label));
4448 
4449   { /* just in case there is an empty process */
4450     PetscInt numValues, maxValues = 0, v;
4451 
4452     PetscCall(DMLabelGetNumValues(label, &numValues));
4453     PetscCall(MPIU_Allreduce(&numValues, &maxValues, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
4454     for (v = numValues; v < maxValues; v++) PetscCall(DMLabelAddStratum(label, v));
4455   }
4456   PetscCall(PetscObjectStateGet((PetscObject)label, &mesh->depthState));
4457   PetscCall(PetscLogEventEnd(DMPLEX_Stratify, dm, 0, 0, 0));
4458   PetscFunctionReturn(PETSC_SUCCESS);
4459 }
4460 
4461 PetscErrorCode DMPlexComputeCellType_Internal(DM dm, PetscInt p, PetscInt pdepth, DMPolytopeType *pt)
4462 {
4463   DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
4464   PetscInt       dim, depth, pheight, coneSize;
4465 
4466   PetscFunctionBeginHot;
4467   PetscCall(DMGetDimension(dm, &dim));
4468   PetscCall(DMPlexGetDepth(dm, &depth));
4469   PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
4470   pheight = depth - pdepth;
4471   if (depth <= 1) {
4472     switch (pdepth) {
4473     case 0:
4474       ct = DM_POLYTOPE_POINT;
4475       break;
4476     case 1:
4477       switch (coneSize) {
4478       case 2:
4479         ct = DM_POLYTOPE_SEGMENT;
4480         break;
4481       case 3:
4482         ct = DM_POLYTOPE_TRIANGLE;
4483         break;
4484       case 4:
4485         switch (dim) {
4486         case 2:
4487           ct = DM_POLYTOPE_QUADRILATERAL;
4488           break;
4489         case 3:
4490           ct = DM_POLYTOPE_TETRAHEDRON;
4491           break;
4492         default:
4493           break;
4494         }
4495         break;
4496       case 5:
4497         ct = DM_POLYTOPE_PYRAMID;
4498         break;
4499       case 6:
4500         ct = DM_POLYTOPE_TRI_PRISM_TENSOR;
4501         break;
4502       case 8:
4503         ct = DM_POLYTOPE_HEXAHEDRON;
4504         break;
4505       default:
4506         break;
4507       }
4508     }
4509   } else {
4510     if (pdepth == 0) {
4511       ct = DM_POLYTOPE_POINT;
4512     } else if (pheight == 0) {
4513       switch (dim) {
4514       case 1:
4515         switch (coneSize) {
4516         case 2:
4517           ct = DM_POLYTOPE_SEGMENT;
4518           break;
4519         default:
4520           break;
4521         }
4522         break;
4523       case 2:
4524         switch (coneSize) {
4525         case 3:
4526           ct = DM_POLYTOPE_TRIANGLE;
4527           break;
4528         case 4:
4529           ct = DM_POLYTOPE_QUADRILATERAL;
4530           break;
4531         default:
4532           break;
4533         }
4534         break;
4535       case 3:
4536         switch (coneSize) {
4537         case 4:
4538           ct = DM_POLYTOPE_TETRAHEDRON;
4539           break;
4540         case 5: {
4541           const PetscInt *cone;
4542           PetscInt        faceConeSize;
4543 
4544           PetscCall(DMPlexGetCone(dm, p, &cone));
4545           PetscCall(DMPlexGetConeSize(dm, cone[0], &faceConeSize));
4546           switch (faceConeSize) {
4547           case 3:
4548             ct = DM_POLYTOPE_TRI_PRISM_TENSOR;
4549             break;
4550           case 4:
4551             ct = DM_POLYTOPE_PYRAMID;
4552             break;
4553           }
4554         } break;
4555         case 6:
4556           ct = DM_POLYTOPE_HEXAHEDRON;
4557           break;
4558         default:
4559           break;
4560         }
4561         break;
4562       default:
4563         break;
4564       }
4565     } else if (pheight > 0) {
4566       switch (coneSize) {
4567       case 2:
4568         ct = DM_POLYTOPE_SEGMENT;
4569         break;
4570       case 3:
4571         ct = DM_POLYTOPE_TRIANGLE;
4572         break;
4573       case 4:
4574         ct = DM_POLYTOPE_QUADRILATERAL;
4575         break;
4576       default:
4577         break;
4578       }
4579     }
4580   }
4581   *pt = ct;
4582   PetscFunctionReturn(PETSC_SUCCESS);
4583 }
4584 
4585 /*@
4586   DMPlexComputeCellTypes - Infer the polytope type of every cell using its dimension and cone size.
4587 
4588   Collective
4589 
4590   Input Parameter:
4591 . dm - The `DMPLEX`
4592 
4593   Level: developer
4594 
4595   Note:
4596   This function is normally called automatically when a cell type is requested. It creates an
4597   internal `DMLabel` named "celltype" which can be directly accessed using `DMGetLabel()`. A user may disable
4598   automatic creation by creating the label manually, using `DMCreateLabel`(dm, "celltype").
4599 
4600   `DMPlexComputeCellTypes()` should be called after all calls to `DMPlexSymmetrize()` and `DMPlexStratify()`
4601 
4602 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreate()`, `DMPlexSymmetrize()`, `DMPlexStratify()`, `DMGetLabel()`, `DMCreateLabel()`
4603 @*/
4604 PetscErrorCode DMPlexComputeCellTypes(DM dm)
4605 {
4606   DM_Plex *mesh;
4607   DMLabel  ctLabel;
4608   PetscInt pStart, pEnd, p;
4609 
4610   PetscFunctionBegin;
4611   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4612   mesh = (DM_Plex *)dm->data;
4613   PetscCall(DMCreateLabel(dm, "celltype"));
4614   PetscCall(DMPlexGetCellTypeLabel(dm, &ctLabel));
4615   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
4616   PetscCall(PetscFree(mesh->cellTypes));
4617   PetscCall(PetscMalloc1(pEnd - pStart, &mesh->cellTypes));
4618   for (p = pStart; p < pEnd; ++p) {
4619     DMPolytopeType ct = DM_POLYTOPE_UNKNOWN;
4620     PetscInt       pdepth;
4621 
4622     PetscCall(DMPlexGetPointDepth(dm, p, &pdepth));
4623     PetscCall(DMPlexComputeCellType_Internal(dm, p, pdepth, &ct));
4624     PetscCheck(ct != DM_POLYTOPE_UNKNOWN && ct != DM_POLYTOPE_UNKNOWN_CELL && ct != DM_POLYTOPE_UNKNOWN_FACE, PETSC_COMM_SELF, PETSC_ERR_SUP, "Point %" PetscInt_FMT " is screwed up", p);
4625     PetscCall(DMLabelSetValue(ctLabel, p, ct));
4626     mesh->cellTypes[p - pStart].value_as_uint8 = ct;
4627   }
4628   PetscCall(PetscObjectStateGet((PetscObject)ctLabel, &mesh->celltypeState));
4629   PetscCall(PetscObjectViewFromOptions((PetscObject)ctLabel, NULL, "-dm_plex_celltypes_view"));
4630   PetscFunctionReturn(PETSC_SUCCESS);
4631 }
4632 
4633 /*@C
4634   DMPlexGetJoin - Get an array for the join of the set of points
4635 
4636   Not Collective
4637 
4638   Input Parameters:
4639 + dm        - The `DMPLEX` object
4640 . numPoints - The number of input points for the join
4641 - points    - The input points
4642 
4643   Output Parameters:
4644 + numCoveredPoints - The number of points in the join
4645 - coveredPoints    - The points in the join
4646 
4647   Level: intermediate
4648 
4649   Note:
4650   Currently, this is restricted to a single level join
4651 
4652   Fortran Notes:
4653   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4654 
4655 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexRestoreJoin()`, `DMPlexGetMeet()`
4656 @*/
4657 PetscErrorCode DMPlexGetJoin(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveredPoints, const PetscInt **coveredPoints)
4658 {
4659   DM_Plex  *mesh = (DM_Plex *)dm->data;
4660   PetscInt *join[2];
4661   PetscInt  joinSize, i = 0;
4662   PetscInt  dof, off, p, c, m;
4663   PetscInt  maxSupportSize;
4664 
4665   PetscFunctionBegin;
4666   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4667   PetscAssertPointer(points, 3);
4668   PetscAssertPointer(numCoveredPoints, 4);
4669   PetscAssertPointer(coveredPoints, 5);
4670   PetscCall(PetscSectionGetMaxDof(mesh->supportSection, &maxSupportSize));
4671   PetscCall(DMGetWorkArray(dm, maxSupportSize, MPIU_INT, &join[0]));
4672   PetscCall(DMGetWorkArray(dm, maxSupportSize, MPIU_INT, &join[1]));
4673   /* Copy in support of first point */
4674   PetscCall(PetscSectionGetDof(mesh->supportSection, points[0], &dof));
4675   PetscCall(PetscSectionGetOffset(mesh->supportSection, points[0], &off));
4676   for (joinSize = 0; joinSize < dof; ++joinSize) join[i][joinSize] = mesh->supports[off + joinSize];
4677   /* Check each successive support */
4678   for (p = 1; p < numPoints; ++p) {
4679     PetscInt newJoinSize = 0;
4680 
4681     PetscCall(PetscSectionGetDof(mesh->supportSection, points[p], &dof));
4682     PetscCall(PetscSectionGetOffset(mesh->supportSection, points[p], &off));
4683     for (c = 0; c < dof; ++c) {
4684       const PetscInt point = mesh->supports[off + c];
4685 
4686       for (m = 0; m < joinSize; ++m) {
4687         if (point == join[i][m]) {
4688           join[1 - i][newJoinSize++] = point;
4689           break;
4690         }
4691       }
4692     }
4693     joinSize = newJoinSize;
4694     i        = 1 - i;
4695   }
4696   *numCoveredPoints = joinSize;
4697   *coveredPoints    = join[i];
4698   PetscCall(DMRestoreWorkArray(dm, maxSupportSize, MPIU_INT, &join[1 - i]));
4699   PetscFunctionReturn(PETSC_SUCCESS);
4700 }
4701 
4702 /*@C
4703   DMPlexRestoreJoin - Restore an array for the join of the set of points
4704 
4705   Not Collective
4706 
4707   Input Parameters:
4708 + dm        - The `DMPLEX` object
4709 . numPoints - The number of input points for the join
4710 - points    - The input points
4711 
4712   Output Parameters:
4713 + numCoveredPoints - The number of points in the join
4714 - coveredPoints    - The points in the join
4715 
4716   Level: intermediate
4717 
4718   Fortran Notes:
4719   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4720 
4721 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetJoin()`, `DMPlexGetFullJoin()`, `DMPlexGetMeet()`
4722 @*/
4723 PetscErrorCode DMPlexRestoreJoin(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveredPoints, const PetscInt **coveredPoints)
4724 {
4725   PetscFunctionBegin;
4726   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4727   if (points) PetscAssertPointer(points, 3);
4728   if (numCoveredPoints) PetscAssertPointer(numCoveredPoints, 4);
4729   PetscAssertPointer(coveredPoints, 5);
4730   PetscCall(DMRestoreWorkArray(dm, 0, MPIU_INT, (void *)coveredPoints));
4731   if (numCoveredPoints) *numCoveredPoints = 0;
4732   PetscFunctionReturn(PETSC_SUCCESS);
4733 }
4734 
4735 /*@C
4736   DMPlexGetFullJoin - Get an array for the join of the set of points
4737 
4738   Not Collective
4739 
4740   Input Parameters:
4741 + dm        - The `DMPLEX` object
4742 . numPoints - The number of input points for the join
4743 - points    - The input points
4744 
4745   Output Parameters:
4746 + numCoveredPoints - The number of points in the join
4747 - coveredPoints    - The points in the join
4748 
4749   Level: intermediate
4750 
4751   Fortran Notes:
4752   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4753 
4754 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetJoin()`, `DMPlexRestoreJoin()`, `DMPlexGetMeet()`
4755 @*/
4756 PetscErrorCode DMPlexGetFullJoin(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveredPoints, const PetscInt **coveredPoints)
4757 {
4758   PetscInt *offsets, **closures;
4759   PetscInt *join[2];
4760   PetscInt  depth = 0, maxSize, joinSize = 0, i = 0;
4761   PetscInt  p, d, c, m, ms;
4762 
4763   PetscFunctionBegin;
4764   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4765   PetscAssertPointer(points, 3);
4766   PetscAssertPointer(numCoveredPoints, 4);
4767   PetscAssertPointer(coveredPoints, 5);
4768 
4769   PetscCall(DMPlexGetDepth(dm, &depth));
4770   PetscCall(PetscCalloc1(numPoints, &closures));
4771   PetscCall(DMGetWorkArray(dm, numPoints * (depth + 2), MPIU_INT, &offsets));
4772   PetscCall(DMPlexGetMaxSizes(dm, NULL, &ms));
4773   maxSize = (ms > 1) ? ((PetscPowInt(ms, depth + 1) - 1) / (ms - 1)) : depth + 1;
4774   PetscCall(DMGetWorkArray(dm, maxSize, MPIU_INT, &join[0]));
4775   PetscCall(DMGetWorkArray(dm, maxSize, MPIU_INT, &join[1]));
4776 
4777   for (p = 0; p < numPoints; ++p) {
4778     PetscInt closureSize;
4779 
4780     PetscCall(DMPlexGetTransitiveClosure(dm, points[p], PETSC_FALSE, &closureSize, &closures[p]));
4781 
4782     offsets[p * (depth + 2) + 0] = 0;
4783     for (d = 0; d < depth + 1; ++d) {
4784       PetscInt pStart, pEnd, i;
4785 
4786       PetscCall(DMPlexGetDepthStratum(dm, d, &pStart, &pEnd));
4787       for (i = offsets[p * (depth + 2) + d]; i < closureSize; ++i) {
4788         if ((pStart > closures[p][i * 2]) || (pEnd <= closures[p][i * 2])) {
4789           offsets[p * (depth + 2) + d + 1] = i;
4790           break;
4791         }
4792       }
4793       if (i == closureSize) offsets[p * (depth + 2) + d + 1] = i;
4794     }
4795     PetscCheck(offsets[p * (depth + 2) + depth + 1] == closureSize, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Total size of closure %" PetscInt_FMT " should be %" PetscInt_FMT, offsets[p * (depth + 2) + depth + 1], closureSize);
4796   }
4797   for (d = 0; d < depth + 1; ++d) {
4798     PetscInt dof;
4799 
4800     /* Copy in support of first point */
4801     dof = offsets[d + 1] - offsets[d];
4802     for (joinSize = 0; joinSize < dof; ++joinSize) join[i][joinSize] = closures[0][(offsets[d] + joinSize) * 2];
4803     /* Check each successive cone */
4804     for (p = 1; p < numPoints && joinSize; ++p) {
4805       PetscInt newJoinSize = 0;
4806 
4807       dof = offsets[p * (depth + 2) + d + 1] - offsets[p * (depth + 2) + d];
4808       for (c = 0; c < dof; ++c) {
4809         const PetscInt point = closures[p][(offsets[p * (depth + 2) + d] + c) * 2];
4810 
4811         for (m = 0; m < joinSize; ++m) {
4812           if (point == join[i][m]) {
4813             join[1 - i][newJoinSize++] = point;
4814             break;
4815           }
4816         }
4817       }
4818       joinSize = newJoinSize;
4819       i        = 1 - i;
4820     }
4821     if (joinSize) break;
4822   }
4823   *numCoveredPoints = joinSize;
4824   *coveredPoints    = join[i];
4825   for (p = 0; p < numPoints; ++p) PetscCall(DMPlexRestoreTransitiveClosure(dm, points[p], PETSC_FALSE, NULL, &closures[p]));
4826   PetscCall(PetscFree(closures));
4827   PetscCall(DMRestoreWorkArray(dm, numPoints * (depth + 2), MPIU_INT, &offsets));
4828   PetscCall(DMRestoreWorkArray(dm, ms, MPIU_INT, &join[1 - i]));
4829   PetscFunctionReturn(PETSC_SUCCESS);
4830 }
4831 
4832 /*@C
4833   DMPlexGetMeet - Get an array for the meet of the set of points
4834 
4835   Not Collective
4836 
4837   Input Parameters:
4838 + dm        - The `DMPLEX` object
4839 . numPoints - The number of input points for the meet
4840 - points    - The input points
4841 
4842   Output Parameters:
4843 + numCoveringPoints - The number of points in the meet
4844 - coveringPoints    - The points in the meet
4845 
4846   Level: intermediate
4847 
4848   Note:
4849   Currently, this is restricted to a single level meet
4850 
4851   Fortran Notes:
4852   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4853 
4854 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexRestoreMeet()`, `DMPlexGetJoin()`
4855 @*/
4856 PetscErrorCode DMPlexGetMeet(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveringPoints, const PetscInt **coveringPoints)
4857 {
4858   DM_Plex  *mesh = (DM_Plex *)dm->data;
4859   PetscInt *meet[2];
4860   PetscInt  meetSize, i = 0;
4861   PetscInt  dof, off, p, c, m;
4862   PetscInt  maxConeSize;
4863 
4864   PetscFunctionBegin;
4865   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4866   PetscAssertPointer(points, 3);
4867   PetscAssertPointer(numCoveringPoints, 4);
4868   PetscAssertPointer(coveringPoints, 5);
4869   PetscCall(PetscSectionGetMaxDof(mesh->coneSection, &maxConeSize));
4870   PetscCall(DMGetWorkArray(dm, maxConeSize, MPIU_INT, &meet[0]));
4871   PetscCall(DMGetWorkArray(dm, maxConeSize, MPIU_INT, &meet[1]));
4872   /* Copy in cone of first point */
4873   PetscCall(PetscSectionGetDof(mesh->coneSection, points[0], &dof));
4874   PetscCall(PetscSectionGetOffset(mesh->coneSection, points[0], &off));
4875   for (meetSize = 0; meetSize < dof; ++meetSize) meet[i][meetSize] = mesh->cones[off + meetSize];
4876   /* Check each successive cone */
4877   for (p = 1; p < numPoints; ++p) {
4878     PetscInt newMeetSize = 0;
4879 
4880     PetscCall(PetscSectionGetDof(mesh->coneSection, points[p], &dof));
4881     PetscCall(PetscSectionGetOffset(mesh->coneSection, points[p], &off));
4882     for (c = 0; c < dof; ++c) {
4883       const PetscInt point = mesh->cones[off + c];
4884 
4885       for (m = 0; m < meetSize; ++m) {
4886         if (point == meet[i][m]) {
4887           meet[1 - i][newMeetSize++] = point;
4888           break;
4889         }
4890       }
4891     }
4892     meetSize = newMeetSize;
4893     i        = 1 - i;
4894   }
4895   *numCoveringPoints = meetSize;
4896   *coveringPoints    = meet[i];
4897   PetscCall(DMRestoreWorkArray(dm, maxConeSize, MPIU_INT, &meet[1 - i]));
4898   PetscFunctionReturn(PETSC_SUCCESS);
4899 }
4900 
4901 /*@C
4902   DMPlexRestoreMeet - Restore an array for the meet of the set of points
4903 
4904   Not Collective
4905 
4906   Input Parameters:
4907 + dm        - The `DMPLEX` object
4908 . numPoints - The number of input points for the meet
4909 - points    - The input points
4910 
4911   Output Parameters:
4912 + numCoveredPoints - The number of points in the meet
4913 - coveredPoints    - The points in the meet
4914 
4915   Level: intermediate
4916 
4917   Fortran Notes:
4918   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4919 
4920 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetMeet()`, `DMPlexGetFullMeet()`, `DMPlexGetJoin()`
4921 @*/
4922 PetscErrorCode DMPlexRestoreMeet(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveredPoints, const PetscInt **coveredPoints)
4923 {
4924   PetscFunctionBegin;
4925   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4926   if (points) PetscAssertPointer(points, 3);
4927   if (numCoveredPoints) PetscAssertPointer(numCoveredPoints, 4);
4928   PetscAssertPointer(coveredPoints, 5);
4929   PetscCall(DMRestoreWorkArray(dm, 0, MPIU_INT, (void *)coveredPoints));
4930   if (numCoveredPoints) *numCoveredPoints = 0;
4931   PetscFunctionReturn(PETSC_SUCCESS);
4932 }
4933 
4934 /*@C
4935   DMPlexGetFullMeet - Get an array for the meet of the set of points
4936 
4937   Not Collective
4938 
4939   Input Parameters:
4940 + dm        - The `DMPLEX` object
4941 . numPoints - The number of input points for the meet
4942 - points    - The input points
4943 
4944   Output Parameters:
4945 + numCoveredPoints - The number of points in the meet
4946 - coveredPoints    - The points in the meet
4947 
4948   Level: intermediate
4949 
4950   Fortran Notes:
4951   The `numCoveredPoints` argument is not present in the Fortran binding since it is internal to the array.
4952 
4953 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetMeet()`, `DMPlexRestoreMeet()`, `DMPlexGetJoin()`
4954 @*/
4955 PetscErrorCode DMPlexGetFullMeet(DM dm, PetscInt numPoints, const PetscInt points[], PetscInt *numCoveredPoints, const PetscInt **coveredPoints)
4956 {
4957   PetscInt *offsets, **closures;
4958   PetscInt *meet[2];
4959   PetscInt  height = 0, maxSize, meetSize = 0, i = 0;
4960   PetscInt  p, h, c, m, mc;
4961 
4962   PetscFunctionBegin;
4963   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4964   PetscAssertPointer(points, 3);
4965   PetscAssertPointer(numCoveredPoints, 4);
4966   PetscAssertPointer(coveredPoints, 5);
4967 
4968   PetscCall(DMPlexGetDepth(dm, &height));
4969   PetscCall(PetscMalloc1(numPoints, &closures));
4970   PetscCall(DMGetWorkArray(dm, numPoints * (height + 2), MPIU_INT, &offsets));
4971   PetscCall(DMPlexGetMaxSizes(dm, &mc, NULL));
4972   maxSize = (mc > 1) ? ((PetscPowInt(mc, height + 1) - 1) / (mc - 1)) : height + 1;
4973   PetscCall(DMGetWorkArray(dm, maxSize, MPIU_INT, &meet[0]));
4974   PetscCall(DMGetWorkArray(dm, maxSize, MPIU_INT, &meet[1]));
4975 
4976   for (p = 0; p < numPoints; ++p) {
4977     PetscInt closureSize;
4978 
4979     PetscCall(DMPlexGetTransitiveClosure(dm, points[p], PETSC_TRUE, &closureSize, &closures[p]));
4980 
4981     offsets[p * (height + 2) + 0] = 0;
4982     for (h = 0; h < height + 1; ++h) {
4983       PetscInt pStart, pEnd, i;
4984 
4985       PetscCall(DMPlexGetHeightStratum(dm, h, &pStart, &pEnd));
4986       for (i = offsets[p * (height + 2) + h]; i < closureSize; ++i) {
4987         if ((pStart > closures[p][i * 2]) || (pEnd <= closures[p][i * 2])) {
4988           offsets[p * (height + 2) + h + 1] = i;
4989           break;
4990         }
4991       }
4992       if (i == closureSize) offsets[p * (height + 2) + h + 1] = i;
4993     }
4994     PetscCheck(offsets[p * (height + 2) + height + 1] == closureSize, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Total size of closure %" PetscInt_FMT " should be %" PetscInt_FMT, offsets[p * (height + 2) + height + 1], closureSize);
4995   }
4996   for (h = 0; h < height + 1; ++h) {
4997     PetscInt dof;
4998 
4999     /* Copy in cone of first point */
5000     dof = offsets[h + 1] - offsets[h];
5001     for (meetSize = 0; meetSize < dof; ++meetSize) meet[i][meetSize] = closures[0][(offsets[h] + meetSize) * 2];
5002     /* Check each successive cone */
5003     for (p = 1; p < numPoints && meetSize; ++p) {
5004       PetscInt newMeetSize = 0;
5005 
5006       dof = offsets[p * (height + 2) + h + 1] - offsets[p * (height + 2) + h];
5007       for (c = 0; c < dof; ++c) {
5008         const PetscInt point = closures[p][(offsets[p * (height + 2) + h] + c) * 2];
5009 
5010         for (m = 0; m < meetSize; ++m) {
5011           if (point == meet[i][m]) {
5012             meet[1 - i][newMeetSize++] = point;
5013             break;
5014           }
5015         }
5016       }
5017       meetSize = newMeetSize;
5018       i        = 1 - i;
5019     }
5020     if (meetSize) break;
5021   }
5022   *numCoveredPoints = meetSize;
5023   *coveredPoints    = meet[i];
5024   for (p = 0; p < numPoints; ++p) PetscCall(DMPlexRestoreTransitiveClosure(dm, points[p], PETSC_TRUE, NULL, &closures[p]));
5025   PetscCall(PetscFree(closures));
5026   PetscCall(DMRestoreWorkArray(dm, numPoints * (height + 2), MPIU_INT, &offsets));
5027   PetscCall(DMRestoreWorkArray(dm, mc, MPIU_INT, &meet[1 - i]));
5028   PetscFunctionReturn(PETSC_SUCCESS);
5029 }
5030 
5031 /*@C
5032   DMPlexEqual - Determine if two `DM` have the same topology
5033 
5034   Not Collective
5035 
5036   Input Parameters:
5037 + dmA - A `DMPLEX` object
5038 - dmB - A `DMPLEX` object
5039 
5040   Output Parameter:
5041 . equal - `PETSC_TRUE` if the topologies are identical
5042 
5043   Level: intermediate
5044 
5045   Note:
5046   We are not solving graph isomorphism, so we do not permute.
5047 
5048 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCone()`
5049 @*/
5050 PetscErrorCode DMPlexEqual(DM dmA, DM dmB, PetscBool *equal)
5051 {
5052   PetscInt depth, depthB, pStart, pEnd, pStartB, pEndB, p;
5053 
5054   PetscFunctionBegin;
5055   PetscValidHeaderSpecific(dmA, DM_CLASSID, 1);
5056   PetscValidHeaderSpecific(dmB, DM_CLASSID, 2);
5057   PetscAssertPointer(equal, 3);
5058 
5059   *equal = PETSC_FALSE;
5060   PetscCall(DMPlexGetDepth(dmA, &depth));
5061   PetscCall(DMPlexGetDepth(dmB, &depthB));
5062   if (depth != depthB) PetscFunctionReturn(PETSC_SUCCESS);
5063   PetscCall(DMPlexGetChart(dmA, &pStart, &pEnd));
5064   PetscCall(DMPlexGetChart(dmB, &pStartB, &pEndB));
5065   if ((pStart != pStartB) || (pEnd != pEndB)) PetscFunctionReturn(PETSC_SUCCESS);
5066   for (p = pStart; p < pEnd; ++p) {
5067     const PetscInt *cone, *coneB, *ornt, *orntB, *support, *supportB;
5068     PetscInt        coneSize, coneSizeB, c, supportSize, supportSizeB, s;
5069 
5070     PetscCall(DMPlexGetConeSize(dmA, p, &coneSize));
5071     PetscCall(DMPlexGetCone(dmA, p, &cone));
5072     PetscCall(DMPlexGetConeOrientation(dmA, p, &ornt));
5073     PetscCall(DMPlexGetConeSize(dmB, p, &coneSizeB));
5074     PetscCall(DMPlexGetCone(dmB, p, &coneB));
5075     PetscCall(DMPlexGetConeOrientation(dmB, p, &orntB));
5076     if (coneSize != coneSizeB) PetscFunctionReturn(PETSC_SUCCESS);
5077     for (c = 0; c < coneSize; ++c) {
5078       if (cone[c] != coneB[c]) PetscFunctionReturn(PETSC_SUCCESS);
5079       if (ornt[c] != orntB[c]) PetscFunctionReturn(PETSC_SUCCESS);
5080     }
5081     PetscCall(DMPlexGetSupportSize(dmA, p, &supportSize));
5082     PetscCall(DMPlexGetSupport(dmA, p, &support));
5083     PetscCall(DMPlexGetSupportSize(dmB, p, &supportSizeB));
5084     PetscCall(DMPlexGetSupport(dmB, p, &supportB));
5085     if (supportSize != supportSizeB) PetscFunctionReturn(PETSC_SUCCESS);
5086     for (s = 0; s < supportSize; ++s) {
5087       if (support[s] != supportB[s]) PetscFunctionReturn(PETSC_SUCCESS);
5088     }
5089   }
5090   *equal = PETSC_TRUE;
5091   PetscFunctionReturn(PETSC_SUCCESS);
5092 }
5093 
5094 /*@C
5095   DMPlexGetNumFaceVertices - Returns the number of vertices on a face
5096 
5097   Not Collective
5098 
5099   Input Parameters:
5100 + dm         - The `DMPLEX`
5101 . cellDim    - The cell dimension
5102 - numCorners - The number of vertices on a cell
5103 
5104   Output Parameter:
5105 . numFaceVertices - The number of vertices on a face
5106 
5107   Level: developer
5108 
5109   Note:
5110   Of course this can only work for a restricted set of symmetric shapes
5111 
5112 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCone()`
5113 @*/
5114 PetscErrorCode DMPlexGetNumFaceVertices(DM dm, PetscInt cellDim, PetscInt numCorners, PetscInt *numFaceVertices)
5115 {
5116   MPI_Comm comm;
5117 
5118   PetscFunctionBegin;
5119   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
5120   PetscAssertPointer(numFaceVertices, 4);
5121   switch (cellDim) {
5122   case 0:
5123     *numFaceVertices = 0;
5124     break;
5125   case 1:
5126     *numFaceVertices = 1;
5127     break;
5128   case 2:
5129     switch (numCorners) {
5130     case 3:                 /* triangle */
5131       *numFaceVertices = 2; /* Edge has 2 vertices */
5132       break;
5133     case 4:                 /* quadrilateral */
5134       *numFaceVertices = 2; /* Edge has 2 vertices */
5135       break;
5136     case 6:                 /* quadratic triangle, tri and quad cohesive Lagrange cells */
5137       *numFaceVertices = 3; /* Edge has 3 vertices */
5138       break;
5139     case 9:                 /* quadratic quadrilateral, quadratic quad cohesive Lagrange cells */
5140       *numFaceVertices = 3; /* Edge has 3 vertices */
5141       break;
5142     default:
5143       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Invalid number of face corners %" PetscInt_FMT " for dimension %" PetscInt_FMT, numCorners, cellDim);
5144     }
5145     break;
5146   case 3:
5147     switch (numCorners) {
5148     case 4:                 /* tetradehdron */
5149       *numFaceVertices = 3; /* Face has 3 vertices */
5150       break;
5151     case 6:                 /* tet cohesive cells */
5152       *numFaceVertices = 4; /* Face has 4 vertices */
5153       break;
5154     case 8:                 /* hexahedron */
5155       *numFaceVertices = 4; /* Face has 4 vertices */
5156       break;
5157     case 9:                 /* tet cohesive Lagrange cells */
5158       *numFaceVertices = 6; /* Face has 6 vertices */
5159       break;
5160     case 10:                /* quadratic tetrahedron */
5161       *numFaceVertices = 6; /* Face has 6 vertices */
5162       break;
5163     case 12:                /* hex cohesive Lagrange cells */
5164       *numFaceVertices = 6; /* Face has 6 vertices */
5165       break;
5166     case 18:                /* quadratic tet cohesive Lagrange cells */
5167       *numFaceVertices = 6; /* Face has 6 vertices */
5168       break;
5169     case 27:                /* quadratic hexahedron, quadratic hex cohesive Lagrange cells */
5170       *numFaceVertices = 9; /* Face has 9 vertices */
5171       break;
5172     default:
5173       SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Invalid number of face corners %" PetscInt_FMT " for dimension %" PetscInt_FMT, numCorners, cellDim);
5174     }
5175     break;
5176   default:
5177     SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Invalid cell dimension %" PetscInt_FMT, cellDim);
5178   }
5179   PetscFunctionReturn(PETSC_SUCCESS);
5180 }
5181 
5182 /*@
5183   DMPlexGetDepthLabel - Get the `DMLabel` recording the depth of each point
5184 
5185   Not Collective
5186 
5187   Input Parameter:
5188 . dm - The `DMPLEX` object
5189 
5190   Output Parameter:
5191 . depthLabel - The `DMLabel` recording point depth
5192 
5193   Level: developer
5194 
5195 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetDepth()`, `DMPlexGetHeightStratum()`, `DMPlexGetDepthStratum()`, `DMPlexGetPointDepth()`,
5196 @*/
5197 PetscErrorCode DMPlexGetDepthLabel(DM dm, DMLabel *depthLabel)
5198 {
5199   PetscFunctionBegin;
5200   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5201   PetscAssertPointer(depthLabel, 2);
5202   *depthLabel = dm->depthLabel;
5203   PetscFunctionReturn(PETSC_SUCCESS);
5204 }
5205 
5206 /*@
5207   DMPlexGetDepth - Get the depth of the DAG representing this mesh
5208 
5209   Not Collective
5210 
5211   Input Parameter:
5212 . dm - The `DMPLEX` object
5213 
5214   Output Parameter:
5215 . depth - The number of strata (breadth first levels) in the DAG
5216 
5217   Level: developer
5218 
5219   Notes:
5220   This returns maximum of point depths over all points, i.e. maximum value of the label returned by `DMPlexGetDepthLabel()`.
5221 
5222   The point depth is described more in detail in `DMPlexGetDepthStratum()`.
5223 
5224   An empty mesh gives -1.
5225 
5226 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetDepthLabel()`, `DMPlexGetDepthStratum()`, `DMPlexGetPointDepth()`, `DMPlexSymmetrize()`
5227 @*/
5228 PetscErrorCode DMPlexGetDepth(DM dm, PetscInt *depth)
5229 {
5230   DM_Plex *mesh = (DM_Plex *)dm->data;
5231   DMLabel  label;
5232   PetscInt d = 0;
5233 
5234   PetscFunctionBegin;
5235   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5236   PetscAssertPointer(depth, 2);
5237   if (mesh->tr) {
5238     PetscCall(DMPlexTransformGetDepth(mesh->tr, depth));
5239   } else {
5240     PetscCall(DMPlexGetDepthLabel(dm, &label));
5241     if (label) PetscCall(DMLabelGetNumValues(label, &d));
5242     *depth = d - 1;
5243   }
5244   PetscFunctionReturn(PETSC_SUCCESS);
5245 }
5246 
5247 /*@
5248   DMPlexGetDepthStratum - Get the bounds [`start`, `end`) for all points at a certain depth.
5249 
5250   Not Collective
5251 
5252   Input Parameters:
5253 + dm    - The `DMPLEX` object
5254 - depth - The requested depth
5255 
5256   Output Parameters:
5257 + start - The first point at this `depth`
5258 - end   - One beyond the last point at this `depth`
5259 
5260   Level: developer
5261 
5262   Notes:
5263   Depth indexing is related to topological dimension.  Depth stratum 0 contains the lowest topological dimension points,
5264   often "vertices".  If the mesh is "interpolated" (see `DMPlexInterpolate()`), then depth stratum 1 contains the next
5265   higher dimension, e.g., "edges".
5266 
5267 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetHeightStratum()`, `DMPlexGetCellTypeStratum()`, `DMPlexGetDepth()`, `DMPlexGetDepthLabel()`, `DMPlexGetPointDepth()`, `DMPlexSymmetrize()`, `DMPlexInterpolate()`
5268 @*/
5269 PetscErrorCode DMPlexGetDepthStratum(DM dm, PetscInt depth, PetscInt *start, PetscInt *end)
5270 {
5271   DM_Plex *mesh = (DM_Plex *)dm->data;
5272   DMLabel  label;
5273   PetscInt pStart, pEnd;
5274 
5275   PetscFunctionBegin;
5276   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5277   if (start) {
5278     PetscAssertPointer(start, 3);
5279     *start = 0;
5280   }
5281   if (end) {
5282     PetscAssertPointer(end, 4);
5283     *end = 0;
5284   }
5285   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
5286   if (pStart == pEnd) PetscFunctionReturn(PETSC_SUCCESS);
5287   if (depth < 0) {
5288     if (start) *start = pStart;
5289     if (end) *end = pEnd;
5290     PetscFunctionReturn(PETSC_SUCCESS);
5291   }
5292   if (mesh->tr) {
5293     PetscCall(DMPlexTransformGetDepthStratum(mesh->tr, depth, start, end));
5294   } else {
5295     PetscCall(DMPlexGetDepthLabel(dm, &label));
5296     PetscCheck(label, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "No label named depth was found");
5297     PetscCall(DMLabelGetStratumBounds(label, depth, start, end));
5298   }
5299   PetscFunctionReturn(PETSC_SUCCESS);
5300 }
5301 
5302 /*@
5303   DMPlexGetHeightStratum - Get the bounds [`start`, `end`) for all points at a certain height.
5304 
5305   Not Collective
5306 
5307   Input Parameters:
5308 + dm     - The `DMPLEX` object
5309 - height - The requested height
5310 
5311   Output Parameters:
5312 + start - The first point at this `height`
5313 - end   - One beyond the last point at this `height`
5314 
5315   Level: developer
5316 
5317   Notes:
5318   Height indexing is related to topological codimension.  Height stratum 0 contains the highest topological dimension
5319   points, often called "cells" or "elements".  If the mesh is "interpolated" (see `DMPlexInterpolate()`), then height
5320   stratum 1 contains the boundary of these "cells", often called "faces" or "facets".
5321 
5322 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetDepthStratum()`, `DMPlexGetCellTypeStratum()`, `DMPlexGetDepth()`, `DMPlexGetPointHeight()`
5323 @*/
5324 PetscErrorCode DMPlexGetHeightStratum(DM dm, PetscInt height, PetscInt *start, PetscInt *end)
5325 {
5326   DMLabel  label;
5327   PetscInt depth, pStart, pEnd;
5328 
5329   PetscFunctionBegin;
5330   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5331   if (start) {
5332     PetscAssertPointer(start, 3);
5333     *start = 0;
5334   }
5335   if (end) {
5336     PetscAssertPointer(end, 4);
5337     *end = 0;
5338   }
5339   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
5340   if (pStart == pEnd) PetscFunctionReturn(PETSC_SUCCESS);
5341   if (height < 0) {
5342     if (start) *start = pStart;
5343     if (end) *end = pEnd;
5344     PetscFunctionReturn(PETSC_SUCCESS);
5345   }
5346   PetscCall(DMPlexGetDepthLabel(dm, &label));
5347   if (label) PetscCall(DMLabelGetNumValues(label, &depth));
5348   else PetscCall(DMGetDimension(dm, &depth));
5349   PetscCheck(depth >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Depth not yet computed");
5350   PetscCall(DMPlexGetDepthStratum(dm, depth - 1 - height, start, end));
5351   PetscFunctionReturn(PETSC_SUCCESS);
5352 }
5353 
5354 /*@
5355   DMPlexGetPointDepth - Get the `depth` of a given point
5356 
5357   Not Collective
5358 
5359   Input Parameters:
5360 + dm    - The `DMPLEX` object
5361 - point - The point
5362 
5363   Output Parameter:
5364 . depth - The depth of the `point`
5365 
5366   Level: intermediate
5367 
5368 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellType()`, `DMPlexGetDepthLabel()`, `DMPlexGetDepth()`, `DMPlexGetPointHeight()`
5369 @*/
5370 PetscErrorCode DMPlexGetPointDepth(DM dm, PetscInt point, PetscInt *depth)
5371 {
5372   PetscFunctionBegin;
5373   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5374   PetscAssertPointer(depth, 3);
5375   PetscCall(DMLabelGetValue(dm->depthLabel, point, depth));
5376   PetscFunctionReturn(PETSC_SUCCESS);
5377 }
5378 
5379 /*@
5380   DMPlexGetPointHeight - Get the `height` of a given point
5381 
5382   Not Collective
5383 
5384   Input Parameters:
5385 + dm    - The `DMPLEX` object
5386 - point - The point
5387 
5388   Output Parameter:
5389 . height - The height of the `point`
5390 
5391   Level: intermediate
5392 
5393 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellType()`, `DMPlexGetDepthLabel()`, `DMPlexGetDepth()`, `DMPlexGetPointDepth()`
5394 @*/
5395 PetscErrorCode DMPlexGetPointHeight(DM dm, PetscInt point, PetscInt *height)
5396 {
5397   PetscInt n, pDepth;
5398 
5399   PetscFunctionBegin;
5400   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5401   PetscAssertPointer(height, 3);
5402   PetscCall(DMLabelGetNumValues(dm->depthLabel, &n));
5403   PetscCall(DMLabelGetValue(dm->depthLabel, point, &pDepth));
5404   *height = n - 1 - pDepth; /* DAG depth is n-1 */
5405   PetscFunctionReturn(PETSC_SUCCESS);
5406 }
5407 
5408 /*@
5409   DMPlexGetCellTypeLabel - Get the `DMLabel` recording the polytope type of each cell
5410 
5411   Not Collective
5412 
5413   Input Parameter:
5414 . dm - The `DMPLEX` object
5415 
5416   Output Parameter:
5417 . celltypeLabel - The `DMLabel` recording cell polytope type
5418 
5419   Level: developer
5420 
5421   Note:
5422   This function will trigger automatica computation of cell types. This can be disabled by calling
5423   `DMCreateLabel`(dm, "celltype") beforehand.
5424 
5425 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellType()`, `DMPlexGetDepthLabel()`, `DMCreateLabel()`
5426 @*/
5427 PetscErrorCode DMPlexGetCellTypeLabel(DM dm, DMLabel *celltypeLabel)
5428 {
5429   PetscFunctionBegin;
5430   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5431   PetscAssertPointer(celltypeLabel, 2);
5432   if (!dm->celltypeLabel) PetscCall(DMPlexComputeCellTypes(dm));
5433   *celltypeLabel = dm->celltypeLabel;
5434   PetscFunctionReturn(PETSC_SUCCESS);
5435 }
5436 
5437 /*@
5438   DMPlexGetCellType - Get the polytope type of a given cell
5439 
5440   Not Collective
5441 
5442   Input Parameters:
5443 + dm   - The `DMPLEX` object
5444 - cell - The cell
5445 
5446   Output Parameter:
5447 . celltype - The polytope type of the cell
5448 
5449   Level: intermediate
5450 
5451 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPolytopeType`, `DMPlexGetCellTypeLabel()`, `DMPlexGetDepthLabel()`, `DMPlexGetDepth()`
5452 @*/
5453 PetscErrorCode DMPlexGetCellType(DM dm, PetscInt cell, DMPolytopeType *celltype)
5454 {
5455   DM_Plex *mesh = (DM_Plex *)dm->data;
5456   DMLabel  label;
5457   PetscInt ct;
5458 
5459   PetscFunctionBegin;
5460   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5461   PetscAssertPointer(celltype, 3);
5462   if (mesh->tr) {
5463     PetscCall(DMPlexTransformGetCellType(mesh->tr, cell, celltype));
5464   } else {
5465     PetscInt pStart, pEnd;
5466 
5467     PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, NULL));
5468     if (!mesh->cellTypes) { /* XXX remove? optimize? */
5469       PetscCall(PetscSectionGetChart(mesh->coneSection, NULL, &pEnd));
5470       PetscCall(PetscMalloc1(pEnd - pStart, &mesh->cellTypes));
5471       PetscCall(DMPlexGetCellTypeLabel(dm, &label));
5472       for (PetscInt p = pStart; p < pEnd; p++) {
5473         PetscCall(DMLabelGetValue(label, p, &ct));
5474         mesh->cellTypes[p - pStart].value_as_uint8 = (DMPolytopeType)ct;
5475       }
5476     }
5477     *celltype = (DMPolytopeType)mesh->cellTypes[cell - pStart].value_as_uint8;
5478     if (PetscDefined(USE_DEBUG)) {
5479       PetscCall(DMPlexGetCellTypeLabel(dm, &label));
5480       PetscCall(DMLabelGetValue(label, cell, &ct));
5481       PetscCheck(ct >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cell %" PetscInt_FMT " has not been assigned a cell type", cell);
5482       PetscCheck(ct == (PetscInt)*celltype, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid cellType for %" PetscInt_FMT ": %d != %" PetscInt_FMT, cell, (int)*celltype, ct);
5483     }
5484   }
5485   PetscFunctionReturn(PETSC_SUCCESS);
5486 }
5487 
5488 /*@
5489   DMPlexSetCellType - Set the polytope type of a given cell
5490 
5491   Not Collective
5492 
5493   Input Parameters:
5494 + dm       - The `DMPLEX` object
5495 . cell     - The cell
5496 - celltype - The polytope type of the cell
5497 
5498   Level: advanced
5499 
5500   Note:
5501   By default, cell types will be automatically computed using `DMPlexComputeCellTypes()` before this function
5502   is executed. This function will override the computed type. However, if automatic classification will not succeed
5503   and a user wants to manually specify all types, the classification must be disabled by calling
5504   DMCreateLabel(dm, "celltype") before getting or setting any cell types.
5505 
5506 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellTypeLabel()`, `DMPlexGetDepthLabel()`, `DMPlexGetDepth()`, `DMPlexComputeCellTypes()`, `DMCreateLabel()`
5507 @*/
5508 PetscErrorCode DMPlexSetCellType(DM dm, PetscInt cell, DMPolytopeType celltype)
5509 {
5510   DM_Plex *mesh = (DM_Plex *)dm->data;
5511   DMLabel  label;
5512   PetscInt pStart, pEnd;
5513 
5514   PetscFunctionBegin;
5515   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5516   PetscCall(PetscSectionGetChart(mesh->coneSection, &pStart, &pEnd));
5517   PetscCall(DMPlexGetCellTypeLabel(dm, &label));
5518   PetscCall(DMLabelSetValue(label, cell, celltype));
5519   if (!mesh->cellTypes) PetscCall(PetscMalloc1(pEnd - pStart, &mesh->cellTypes));
5520   mesh->cellTypes[cell - pStart].value_as_uint8 = celltype;
5521   PetscFunctionReturn(PETSC_SUCCESS);
5522 }
5523 
5524 PetscErrorCode DMCreateCoordinateDM_Plex(DM dm, DM *cdm)
5525 {
5526   PetscSection section, s;
5527   Mat          m;
5528   PetscInt     maxHeight;
5529   const char  *prefix;
5530 
5531   PetscFunctionBegin;
5532   PetscCall(DMClone(dm, cdm));
5533   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
5534   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)*cdm, prefix));
5535   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)*cdm, "cdm_"));
5536   PetscCall(DMPlexGetMaxProjectionHeight(dm, &maxHeight));
5537   PetscCall(DMPlexSetMaxProjectionHeight(*cdm, maxHeight));
5538   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &section));
5539   PetscCall(DMSetLocalSection(*cdm, section));
5540   PetscCall(PetscSectionDestroy(&section));
5541   PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &s));
5542   PetscCall(MatCreate(PETSC_COMM_SELF, &m));
5543   PetscCall(DMSetDefaultConstraints(*cdm, s, m, NULL));
5544   PetscCall(PetscSectionDestroy(&s));
5545   PetscCall(MatDestroy(&m));
5546 
5547   PetscCall(DMSetNumFields(*cdm, 1));
5548   PetscCall(DMCreateDS(*cdm));
5549   (*cdm)->cloneOpts = PETSC_TRUE;
5550   if (dm->setfromoptionscalled) PetscCall(DMSetFromOptions(*cdm));
5551   PetscFunctionReturn(PETSC_SUCCESS);
5552 }
5553 
5554 PetscErrorCode DMCreateCoordinateField_Plex(DM dm, DMField *field)
5555 {
5556   Vec coordsLocal, cellCoordsLocal;
5557   DM  coordsDM, cellCoordsDM;
5558 
5559   PetscFunctionBegin;
5560   *field = NULL;
5561   PetscCall(DMGetCoordinatesLocal(dm, &coordsLocal));
5562   PetscCall(DMGetCoordinateDM(dm, &coordsDM));
5563   PetscCall(DMGetCellCoordinatesLocal(dm, &cellCoordsLocal));
5564   PetscCall(DMGetCellCoordinateDM(dm, &cellCoordsDM));
5565   if (coordsLocal && coordsDM) {
5566     if (cellCoordsLocal && cellCoordsDM) PetscCall(DMFieldCreateDSWithDG(coordsDM, cellCoordsDM, 0, coordsLocal, cellCoordsLocal, field));
5567     else PetscCall(DMFieldCreateDS(coordsDM, 0, coordsLocal, field));
5568   }
5569   PetscFunctionReturn(PETSC_SUCCESS);
5570 }
5571 
5572 /*@C
5573   DMPlexGetConeSection - Return a section which describes the layout of cone data
5574 
5575   Not Collective
5576 
5577   Input Parameter:
5578 . dm - The `DMPLEX` object
5579 
5580   Output Parameter:
5581 . section - The `PetscSection` object
5582 
5583   Level: developer
5584 
5585 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetSupportSection()`, `DMPlexGetCones()`, `DMPlexGetConeOrientations()`, `PetscSection`
5586 @*/
5587 PetscErrorCode DMPlexGetConeSection(DM dm, PetscSection *section)
5588 {
5589   DM_Plex *mesh = (DM_Plex *)dm->data;
5590 
5591   PetscFunctionBegin;
5592   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5593   if (section) *section = mesh->coneSection;
5594   PetscFunctionReturn(PETSC_SUCCESS);
5595 }
5596 
5597 /*@C
5598   DMPlexGetSupportSection - Return a section which describes the layout of support data
5599 
5600   Not Collective
5601 
5602   Input Parameter:
5603 . dm - The `DMPLEX` object
5604 
5605   Output Parameter:
5606 . section - The `PetscSection` object
5607 
5608   Level: developer
5609 
5610 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetConeSection()`, `PetscSection`
5611 @*/
5612 PetscErrorCode DMPlexGetSupportSection(DM dm, PetscSection *section)
5613 {
5614   DM_Plex *mesh = (DM_Plex *)dm->data;
5615 
5616   PetscFunctionBegin;
5617   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5618   if (section) *section = mesh->supportSection;
5619   PetscFunctionReturn(PETSC_SUCCESS);
5620 }
5621 
5622 /*@C
5623   DMPlexGetCones - Return cone data
5624 
5625   Not Collective
5626 
5627   Input Parameter:
5628 . dm - The `DMPLEX` object
5629 
5630   Output Parameter:
5631 . cones - The cone for each point
5632 
5633   Level: developer
5634 
5635 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetConeSection()`
5636 @*/
5637 PetscErrorCode DMPlexGetCones(DM dm, PetscInt *cones[])
5638 {
5639   DM_Plex *mesh = (DM_Plex *)dm->data;
5640 
5641   PetscFunctionBegin;
5642   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5643   if (cones) *cones = mesh->cones;
5644   PetscFunctionReturn(PETSC_SUCCESS);
5645 }
5646 
5647 /*@C
5648   DMPlexGetConeOrientations - Return cone orientation data
5649 
5650   Not Collective
5651 
5652   Input Parameter:
5653 . dm - The `DMPLEX` object
5654 
5655   Output Parameter:
5656 . coneOrientations - The array of cone orientations for all points
5657 
5658   Level: developer
5659 
5660   Notes:
5661   The `PetscSection` returned by `DMPlexGetConeSection()` partitions coneOrientations into cone orientations of particular points as returned by `DMPlexGetConeOrientation()`.
5662 
5663   The meaning of coneOrientations values is detailed in `DMPlexGetConeOrientation()`.
5664 
5665 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetConeSection()`, `DMPlexGetConeOrientation()`, `PetscSection`
5666 @*/
5667 PetscErrorCode DMPlexGetConeOrientations(DM dm, PetscInt *coneOrientations[])
5668 {
5669   DM_Plex *mesh = (DM_Plex *)dm->data;
5670 
5671   PetscFunctionBegin;
5672   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5673   if (coneOrientations) *coneOrientations = mesh->coneOrientations;
5674   PetscFunctionReturn(PETSC_SUCCESS);
5675 }
5676 
5677 /******************************** FEM Support **********************************/
5678 
5679 PetscErrorCode DMPlexGetAllCells_Internal(DM plex, IS *cellIS)
5680 {
5681   PetscInt depth;
5682 
5683   PetscFunctionBegin;
5684   PetscCall(DMPlexGetDepth(plex, &depth));
5685   PetscCall(DMGetStratumIS(plex, "dim", depth, cellIS));
5686   if (!*cellIS) PetscCall(DMGetStratumIS(plex, "depth", depth, cellIS));
5687   PetscFunctionReturn(PETSC_SUCCESS);
5688 }
5689 
5690 PetscErrorCode DMPlexGetAllFaces_Internal(DM plex, IS *faceIS)
5691 {
5692   PetscInt depth;
5693 
5694   PetscFunctionBegin;
5695   PetscCall(DMPlexGetDepth(plex, &depth));
5696   PetscCall(DMGetStratumIS(plex, "dim", depth - 1, faceIS));
5697   if (!*faceIS) PetscCall(DMGetStratumIS(plex, "depth", depth - 1, faceIS));
5698   PetscFunctionReturn(PETSC_SUCCESS);
5699 }
5700 
5701 /*
5702  Returns number of components and tensor degree for the field.  For interpolated meshes, line should be a point
5703  representing a line in the section.
5704 */
5705 static PetscErrorCode PetscSectionFieldGetTensorDegree_Private(DM dm, PetscSection section, PetscInt field, PetscInt line, PetscInt *Nc, PetscInt *k, PetscBool *continuous, PetscBool *tensor)
5706 {
5707   PetscObject  obj;
5708   PetscClassId id;
5709   PetscFE      fe = NULL;
5710 
5711   PetscFunctionBeginHot;
5712   PetscCall(PetscSectionGetFieldComponents(section, field, Nc));
5713   PetscCall(DMGetField(dm, field, NULL, &obj));
5714   PetscCall(PetscObjectGetClassId(obj, &id));
5715   if (id == PETSCFE_CLASSID) fe = (PetscFE)obj;
5716 
5717   if (!fe) {
5718     /* Assume the full interpolated mesh is in the chart; lines in particular */
5719     /* An order k SEM disc has k-1 dofs on an edge */
5720     PetscCall(PetscSectionGetFieldDof(section, line, field, k));
5721     *k = *k / *Nc + 1;
5722   } else {
5723     PetscInt       dual_space_size, dim;
5724     PetscDualSpace dsp;
5725 
5726     PetscCall(DMGetDimension(dm, &dim));
5727     PetscCall(PetscFEGetDualSpace(fe, &dsp));
5728     PetscCall(PetscDualSpaceGetDimension(dsp, &dual_space_size));
5729     *k = (PetscInt)PetscCeilReal(PetscPowReal(dual_space_size / *Nc, 1.0 / dim)) - 1;
5730     PetscCall(PetscDualSpaceLagrangeGetContinuity(dsp, continuous));
5731     PetscCall(PetscDualSpaceLagrangeGetTensor(dsp, tensor));
5732   }
5733   PetscFunctionReturn(PETSC_SUCCESS);
5734 }
5735 
5736 static PetscErrorCode GetFieldSize_Private(PetscInt dim, PetscInt k, PetscBool tensor, PetscInt *dof)
5737 {
5738   PetscFunctionBeginHot;
5739   if (tensor) {
5740     *dof = PetscPowInt(k + 1, dim);
5741   } else {
5742     switch (dim) {
5743     case 1:
5744       *dof = k + 1;
5745       break;
5746     case 2:
5747       *dof = ((k + 1) * (k + 2)) / 2;
5748       break;
5749     case 3:
5750       *dof = ((k + 1) * (k + 2) * (k + 3)) / 6;
5751       break;
5752     default:
5753       *dof = 0;
5754     }
5755   }
5756   PetscFunctionReturn(PETSC_SUCCESS);
5757 }
5758 
5759 /*@
5760 
5761   DMPlexSetClosurePermutationTensor - Create a permutation from the default (BFS) point ordering in the closure, to a
5762   lexicographic ordering over the tensor product cell (i.e., line, quad, hex, etc.), and set this permutation in the
5763   section provided (or the section of the `DM`).
5764 
5765   Input Parameters:
5766 + dm      - The `DM`
5767 . point   - Either a cell (highest dim point) or an edge (dim 1 point), or `PETSC_DETERMINE`
5768 - section - The `PetscSection` to reorder, or `NULL` for the default section
5769 
5770   Example:
5771   A typical interpolated single-quad mesh might order points as
5772 .vb
5773   [c0, v1, v2, v3, v4, e5, e6, e7, e8]
5774 
5775   v4 -- e6 -- v3
5776   |           |
5777   e7    c0    e8
5778   |           |
5779   v1 -- e5 -- v2
5780 .ve
5781 
5782   (There is no significance to the ordering described here.)  The default section for a Q3 quad might typically assign
5783   dofs in the order of points, e.g.,
5784 .vb
5785     c0 -> [0,1,2,3]
5786     v1 -> [4]
5787     ...
5788     e5 -> [8, 9]
5789 .ve
5790 
5791   which corresponds to the dofs
5792 .vb
5793     6   10  11  7
5794     13  2   3   15
5795     12  0   1   14
5796     4   8   9   5
5797 .ve
5798 
5799   The closure in BFS ordering works through height strata (cells, edges, vertices) to produce the ordering
5800 .vb
5801   0 1 2 3 8 9 14 15 11 10 13 12 4 5 7 6
5802 .ve
5803 
5804   After calling DMPlexSetClosurePermutationTensor(), the closure will be ordered lexicographically,
5805 .vb
5806    4 8 9 5 12 0 1 14 13 2 3 15 6 10 11 7
5807 .ve
5808 
5809   Level: developer
5810 
5811   Notes:
5812   The point is used to determine the number of dofs/field on an edge. For SEM, this is related to the polynomial
5813   degree of the basis.
5814 
5815   This is required to run with libCEED.
5816 
5817 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMGetLocalSection()`, `PetscSectionSetClosurePermutation()`, `DMSetGlobalSection()`
5818 @*/
5819 PetscErrorCode DMPlexSetClosurePermutationTensor(DM dm, PetscInt point, PetscSection section)
5820 {
5821   DMLabel   label;
5822   PetscInt  dim, depth = -1, eStart = -1, Nf;
5823   PetscBool continuous = PETSC_TRUE, tensor = PETSC_TRUE;
5824 
5825   PetscFunctionBegin;
5826   PetscCall(DMGetDimension(dm, &dim));
5827   if (dim < 1) PetscFunctionReturn(PETSC_SUCCESS);
5828   if (point < 0) {
5829     PetscInt sStart, sEnd;
5830 
5831     PetscCall(DMPlexGetDepthStratum(dm, 1, &sStart, &sEnd));
5832     point = sEnd - sStart ? sStart : point;
5833   }
5834   PetscCall(DMPlexGetDepthLabel(dm, &label));
5835   if (point >= 0) PetscCall(DMLabelGetValue(label, point, &depth));
5836   if (!section) PetscCall(DMGetLocalSection(dm, &section));
5837   if (depth == 1) {
5838     eStart = point;
5839   } else if (depth == dim) {
5840     const PetscInt *cone;
5841 
5842     PetscCall(DMPlexGetCone(dm, point, &cone));
5843     if (dim == 2) eStart = cone[0];
5844     else if (dim == 3) {
5845       const PetscInt *cone2;
5846       PetscCall(DMPlexGetCone(dm, cone[0], &cone2));
5847       eStart = cone2[0];
5848     } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Point %" PetscInt_FMT " of depth %" PetscInt_FMT " cannot be used to bootstrap spectral ordering for dim %" PetscInt_FMT, point, depth, dim);
5849   } else PetscCheck(depth < 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Point %" PetscInt_FMT " of depth %" PetscInt_FMT " cannot be used to bootstrap spectral ordering for dim %" PetscInt_FMT, point, depth, dim);
5850 
5851   PetscCall(PetscSectionGetNumFields(section, &Nf));
5852   for (PetscInt d = 1; d <= dim; d++) {
5853     PetscInt  k, f, Nc, c, i, j, size = 0, offset = 0, foffset = 0;
5854     PetscInt *perm;
5855 
5856     for (f = 0; f < Nf; ++f) {
5857       PetscInt dof;
5858 
5859       PetscCall(PetscSectionFieldGetTensorDegree_Private(dm, section, f, eStart, &Nc, &k, &continuous, &tensor));
5860       PetscCheck(dim == 1 || tensor || !continuous, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Continuous field %" PetscInt_FMT " must have a tensor product discretization", f);
5861       if (!continuous && d < dim) continue;
5862       PetscCall(GetFieldSize_Private(d, k, tensor, &dof));
5863       size += dof * Nc;
5864     }
5865     PetscCall(PetscMalloc1(size, &perm));
5866     for (f = 0; f < Nf; ++f) {
5867       switch (d) {
5868       case 1:
5869         PetscCall(PetscSectionFieldGetTensorDegree_Private(dm, section, f, eStart, &Nc, &k, &continuous, &tensor));
5870         if (!continuous && d < dim) continue;
5871         /*
5872          Original ordering is [ edge of length k-1; vtx0; vtx1 ]
5873          We want              [ vtx0; edge of length k-1; vtx1 ]
5874          */
5875         if (continuous) {
5876           for (c = 0; c < Nc; c++, offset++) perm[offset] = (k - 1) * Nc + c + foffset;
5877           for (i = 0; i < k - 1; i++)
5878             for (c = 0; c < Nc; c++, offset++) perm[offset] = i * Nc + c + foffset;
5879           for (c = 0; c < Nc; c++, offset++) perm[offset] = k * Nc + c + foffset;
5880           foffset = offset;
5881         } else {
5882           PetscInt dof;
5883 
5884           PetscCall(GetFieldSize_Private(d, k, tensor, &dof));
5885           for (i = 0; i < dof * Nc; ++i, ++offset) perm[offset] = i + foffset;
5886           foffset = offset;
5887         }
5888         break;
5889       case 2:
5890         /* The original quad closure is oriented clockwise, {f, e_b, e_r, e_t, e_l, v_lb, v_rb, v_tr, v_tl} */
5891         PetscCall(PetscSectionFieldGetTensorDegree_Private(dm, section, f, eStart, &Nc, &k, &continuous, &tensor));
5892         if (!continuous && d < dim) continue;
5893         /* The SEM order is
5894 
5895          v_lb, {e_b}, v_rb,
5896          e^{(k-1)-i}_l, {f^{i*(k-1)}}, e^i_r,
5897          v_lt, reverse {e_t}, v_rt
5898          */
5899         if (continuous) {
5900           const PetscInt of   = 0;
5901           const PetscInt oeb  = of + PetscSqr(k - 1);
5902           const PetscInt oer  = oeb + (k - 1);
5903           const PetscInt oet  = oer + (k - 1);
5904           const PetscInt oel  = oet + (k - 1);
5905           const PetscInt ovlb = oel + (k - 1);
5906           const PetscInt ovrb = ovlb + 1;
5907           const PetscInt ovrt = ovrb + 1;
5908           const PetscInt ovlt = ovrt + 1;
5909           PetscInt       o;
5910 
5911           /* bottom */
5912           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovlb * Nc + c + foffset;
5913           for (o = oeb; o < oer; ++o)
5914             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
5915           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovrb * Nc + c + foffset;
5916           /* middle */
5917           for (i = 0; i < k - 1; ++i) {
5918             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oel + (k - 2) - i) * Nc + c + foffset;
5919             for (o = of + (k - 1) * i; o < of + (k - 1) * (i + 1); ++o)
5920               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
5921             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oer + i) * Nc + c + foffset;
5922           }
5923           /* top */
5924           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovlt * Nc + c + foffset;
5925           for (o = oel - 1; o >= oet; --o)
5926             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
5927           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovrt * Nc + c + foffset;
5928           foffset = offset;
5929         } else {
5930           PetscInt dof;
5931 
5932           PetscCall(GetFieldSize_Private(d, k, tensor, &dof));
5933           for (i = 0; i < dof * Nc; ++i, ++offset) perm[offset] = i + foffset;
5934           foffset = offset;
5935         }
5936         break;
5937       case 3:
5938         /* The original hex closure is
5939 
5940          {c,
5941          f_b, f_t, f_f, f_b, f_r, f_l,
5942          e_bl, e_bb, e_br, e_bf,  e_tf, e_tr, e_tb, e_tl,  e_rf, e_lf, e_lb, e_rb,
5943          v_blf, v_blb, v_brb, v_brf, v_tlf, v_trf, v_trb, v_tlb}
5944          */
5945         PetscCall(PetscSectionFieldGetTensorDegree_Private(dm, section, f, eStart, &Nc, &k, &continuous, &tensor));
5946         if (!continuous && d < dim) continue;
5947         /* The SEM order is
5948          Bottom Slice
5949          v_blf, {e^{(k-1)-n}_bf}, v_brf,
5950          e^{i}_bl, f^{n*(k-1)+(k-1)-i}_b, e^{(k-1)-i}_br,
5951          v_blb, {e_bb}, v_brb,
5952 
5953          Middle Slice (j)
5954          {e^{(k-1)-j}_lf}, {f^{j*(k-1)+n}_f}, e^j_rf,
5955          f^{i*(k-1)+j}_l, {c^{(j*(k-1) + i)*(k-1)+n}_t}, f^{j*(k-1)+i}_r,
5956          e^j_lb, {f^{j*(k-1)+(k-1)-n}_b}, e^{(k-1)-j}_rb,
5957 
5958          Top Slice
5959          v_tlf, {e_tf}, v_trf,
5960          e^{(k-1)-i}_tl, {f^{i*(k-1)}_t}, e^{i}_tr,
5961          v_tlb, {e^{(k-1)-n}_tb}, v_trb,
5962          */
5963         if (continuous) {
5964           const PetscInt oc    = 0;
5965           const PetscInt ofb   = oc + PetscSqr(k - 1) * (k - 1);
5966           const PetscInt oft   = ofb + PetscSqr(k - 1);
5967           const PetscInt off   = oft + PetscSqr(k - 1);
5968           const PetscInt ofk   = off + PetscSqr(k - 1);
5969           const PetscInt ofr   = ofk + PetscSqr(k - 1);
5970           const PetscInt ofl   = ofr + PetscSqr(k - 1);
5971           const PetscInt oebl  = ofl + PetscSqr(k - 1);
5972           const PetscInt oebb  = oebl + (k - 1);
5973           const PetscInt oebr  = oebb + (k - 1);
5974           const PetscInt oebf  = oebr + (k - 1);
5975           const PetscInt oetf  = oebf + (k - 1);
5976           const PetscInt oetr  = oetf + (k - 1);
5977           const PetscInt oetb  = oetr + (k - 1);
5978           const PetscInt oetl  = oetb + (k - 1);
5979           const PetscInt oerf  = oetl + (k - 1);
5980           const PetscInt oelf  = oerf + (k - 1);
5981           const PetscInt oelb  = oelf + (k - 1);
5982           const PetscInt oerb  = oelb + (k - 1);
5983           const PetscInt ovblf = oerb + (k - 1);
5984           const PetscInt ovblb = ovblf + 1;
5985           const PetscInt ovbrb = ovblb + 1;
5986           const PetscInt ovbrf = ovbrb + 1;
5987           const PetscInt ovtlf = ovbrf + 1;
5988           const PetscInt ovtrf = ovtlf + 1;
5989           const PetscInt ovtrb = ovtrf + 1;
5990           const PetscInt ovtlb = ovtrb + 1;
5991           PetscInt       o, n;
5992 
5993           /* Bottom Slice */
5994           /*   bottom */
5995           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovblf * Nc + c + foffset;
5996           for (o = oetf - 1; o >= oebf; --o)
5997             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
5998           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovbrf * Nc + c + foffset;
5999           /*   middle */
6000           for (i = 0; i < k - 1; ++i) {
6001             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oebl + i) * Nc + c + foffset;
6002             for (n = 0; n < k - 1; ++n) {
6003               o = ofb + n * (k - 1) + i;
6004               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6005             }
6006             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oebr + (k - 2) - i) * Nc + c + foffset;
6007           }
6008           /*   top */
6009           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovblb * Nc + c + foffset;
6010           for (o = oebb; o < oebr; ++o)
6011             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6012           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovbrb * Nc + c + foffset;
6013 
6014           /* Middle Slice */
6015           for (j = 0; j < k - 1; ++j) {
6016             /*   bottom */
6017             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oelf + (k - 2) - j) * Nc + c + foffset;
6018             for (o = off + j * (k - 1); o < off + (j + 1) * (k - 1); ++o)
6019               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6020             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oerf + j) * Nc + c + foffset;
6021             /*   middle */
6022             for (i = 0; i < k - 1; ++i) {
6023               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (ofl + i * (k - 1) + j) * Nc + c + foffset;
6024               for (n = 0; n < k - 1; ++n)
6025                 for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oc + (j * (k - 1) + i) * (k - 1) + n) * Nc + c + foffset;
6026               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (ofr + j * (k - 1) + i) * Nc + c + foffset;
6027             }
6028             /*   top */
6029             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oelb + j) * Nc + c + foffset;
6030             for (o = ofk + j * (k - 1) + (k - 2); o >= ofk + j * (k - 1); --o)
6031               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6032             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oerb + (k - 2) - j) * Nc + c + foffset;
6033           }
6034 
6035           /* Top Slice */
6036           /*   bottom */
6037           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovtlf * Nc + c + foffset;
6038           for (o = oetf; o < oetr; ++o)
6039             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6040           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovtrf * Nc + c + foffset;
6041           /*   middle */
6042           for (i = 0; i < k - 1; ++i) {
6043             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oetl + (k - 2) - i) * Nc + c + foffset;
6044             for (n = 0; n < k - 1; ++n)
6045               for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oft + i * (k - 1) + n) * Nc + c + foffset;
6046             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = (oetr + i) * Nc + c + foffset;
6047           }
6048           /*   top */
6049           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovtlb * Nc + c + foffset;
6050           for (o = oetl - 1; o >= oetb; --o)
6051             for (c = 0; c < Nc; ++c, ++offset) perm[offset] = o * Nc + c + foffset;
6052           for (c = 0; c < Nc; ++c, ++offset) perm[offset] = ovtrb * Nc + c + foffset;
6053 
6054           foffset = offset;
6055         } else {
6056           PetscInt dof;
6057 
6058           PetscCall(GetFieldSize_Private(d, k, tensor, &dof));
6059           for (i = 0; i < dof * Nc; ++i, ++offset) perm[offset] = i + foffset;
6060           foffset = offset;
6061         }
6062         break;
6063       default:
6064         SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No spectral ordering for dimension %" PetscInt_FMT, d);
6065       }
6066     }
6067     PetscCheck(offset == size, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Number of permutation entries %" PetscInt_FMT " != %" PetscInt_FMT, offset, size);
6068     /* Check permutation */
6069     {
6070       PetscInt *check;
6071 
6072       PetscCall(PetscMalloc1(size, &check));
6073       for (i = 0; i < size; ++i) {
6074         check[i] = -1;
6075         PetscCheck(perm[i] >= 0 && perm[i] < size, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid permutation index p[%" PetscInt_FMT "] = %" PetscInt_FMT, i, perm[i]);
6076       }
6077       for (i = 0; i < size; ++i) check[perm[i]] = i;
6078       for (i = 0; i < size; ++i) PetscCheck(check[i] >= 0, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Missing permutation index %" PetscInt_FMT, i);
6079       PetscCall(PetscFree(check));
6080     }
6081     PetscCall(PetscSectionSetClosurePermutation_Internal(section, (PetscObject)dm, d, size, PETSC_OWN_POINTER, perm));
6082     if (d == dim) { // Add permutation for localized (in case this is a coordinate DM)
6083       PetscInt *loc_perm;
6084       PetscCall(PetscMalloc1(size * 2, &loc_perm));
6085       for (PetscInt i = 0; i < size; i++) {
6086         loc_perm[i]        = perm[i];
6087         loc_perm[size + i] = size + perm[i];
6088       }
6089       PetscCall(PetscSectionSetClosurePermutation_Internal(section, (PetscObject)dm, d, size * 2, PETSC_OWN_POINTER, loc_perm));
6090     }
6091   }
6092   PetscFunctionReturn(PETSC_SUCCESS);
6093 }
6094 
6095 PetscErrorCode DMPlexGetPointDualSpaceFEM(DM dm, PetscInt point, PetscInt field, PetscDualSpace *dspace)
6096 {
6097   PetscDS  prob;
6098   PetscInt depth, Nf, h;
6099   DMLabel  label;
6100 
6101   PetscFunctionBeginHot;
6102   PetscCall(DMGetDS(dm, &prob));
6103   Nf      = prob->Nf;
6104   label   = dm->depthLabel;
6105   *dspace = NULL;
6106   if (field < Nf) {
6107     PetscObject disc = prob->disc[field];
6108 
6109     if (disc->classid == PETSCFE_CLASSID) {
6110       PetscDualSpace dsp;
6111 
6112       PetscCall(PetscFEGetDualSpace((PetscFE)disc, &dsp));
6113       PetscCall(DMLabelGetNumValues(label, &depth));
6114       PetscCall(DMLabelGetValue(label, point, &h));
6115       h = depth - 1 - h;
6116       if (h) {
6117         PetscCall(PetscDualSpaceGetHeightSubspace(dsp, h, dspace));
6118       } else {
6119         *dspace = dsp;
6120       }
6121     }
6122   }
6123   PetscFunctionReturn(PETSC_SUCCESS);
6124 }
6125 
6126 static inline PetscErrorCode DMPlexVecGetClosure_Depth1_Static(DM dm, PetscSection section, Vec v, PetscInt point, PetscInt *csize, PetscScalar *values[])
6127 {
6128   PetscScalar       *array;
6129   const PetscScalar *vArray;
6130   const PetscInt    *cone, *coneO;
6131   PetscInt           pStart, pEnd, p, numPoints, size = 0, offset = 0;
6132 
6133   PetscFunctionBeginHot;
6134   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
6135   PetscCall(DMPlexGetConeSize(dm, point, &numPoints));
6136   PetscCall(DMPlexGetCone(dm, point, &cone));
6137   PetscCall(DMPlexGetConeOrientation(dm, point, &coneO));
6138   if (!values || !*values) {
6139     if ((point >= pStart) && (point < pEnd)) {
6140       PetscInt dof;
6141 
6142       PetscCall(PetscSectionGetDof(section, point, &dof));
6143       size += dof;
6144     }
6145     for (p = 0; p < numPoints; ++p) {
6146       const PetscInt cp = cone[p];
6147       PetscInt       dof;
6148 
6149       if ((cp < pStart) || (cp >= pEnd)) continue;
6150       PetscCall(PetscSectionGetDof(section, cp, &dof));
6151       size += dof;
6152     }
6153     if (!values) {
6154       if (csize) *csize = size;
6155       PetscFunctionReturn(PETSC_SUCCESS);
6156     }
6157     PetscCall(DMGetWorkArray(dm, size, MPIU_SCALAR, &array));
6158   } else {
6159     array = *values;
6160   }
6161   size = 0;
6162   PetscCall(VecGetArrayRead(v, &vArray));
6163   if ((point >= pStart) && (point < pEnd)) {
6164     PetscInt           dof, off, d;
6165     const PetscScalar *varr;
6166 
6167     PetscCall(PetscSectionGetDof(section, point, &dof));
6168     PetscCall(PetscSectionGetOffset(section, point, &off));
6169     varr = PetscSafePointerPlusOffset(vArray, off);
6170     for (d = 0; d < dof; ++d, ++offset) array[offset] = varr[d];
6171     size += dof;
6172   }
6173   for (p = 0; p < numPoints; ++p) {
6174     const PetscInt     cp = cone[p];
6175     PetscInt           o  = coneO[p];
6176     PetscInt           dof, off, d;
6177     const PetscScalar *varr;
6178 
6179     if ((cp < pStart) || (cp >= pEnd)) continue;
6180     PetscCall(PetscSectionGetDof(section, cp, &dof));
6181     PetscCall(PetscSectionGetOffset(section, cp, &off));
6182     varr = PetscSafePointerPlusOffset(vArray, off);
6183     if (o >= 0) {
6184       for (d = 0; d < dof; ++d, ++offset) array[offset] = varr[d];
6185     } else {
6186       for (d = dof - 1; d >= 0; --d, ++offset) array[offset] = varr[d];
6187     }
6188     size += dof;
6189   }
6190   PetscCall(VecRestoreArrayRead(v, &vArray));
6191   if (!*values) {
6192     if (csize) *csize = size;
6193     *values = array;
6194   } else {
6195     PetscCheck(size <= *csize, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Size of input array %" PetscInt_FMT " < actual size %" PetscInt_FMT, *csize, size);
6196     *csize = size;
6197   }
6198   PetscFunctionReturn(PETSC_SUCCESS);
6199 }
6200 
6201 /* Compress out points not in the section */
6202 static inline PetscErrorCode CompressPoints_Private(PetscSection section, PetscInt *numPoints, PetscInt points[])
6203 {
6204   const PetscInt np = *numPoints;
6205   PetscInt       pStart, pEnd, p, q;
6206 
6207   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
6208   for (p = 0, q = 0; p < np; ++p) {
6209     const PetscInt r = points[p * 2];
6210     if ((r >= pStart) && (r < pEnd)) {
6211       points[q * 2]     = r;
6212       points[q * 2 + 1] = points[p * 2 + 1];
6213       ++q;
6214     }
6215   }
6216   *numPoints = q;
6217   return PETSC_SUCCESS;
6218 }
6219 
6220 /* Compressed closure does not apply closure permutation */
6221 PetscErrorCode DMPlexGetCompressedClosure(DM dm, PetscSection section, PetscInt point, PetscInt ornt, PetscInt *numPoints, PetscInt **points, PetscSection *clSec, IS *clPoints, const PetscInt **clp)
6222 {
6223   const PetscInt *cla = NULL;
6224   PetscInt        np, *pts = NULL;
6225 
6226   PetscFunctionBeginHot;
6227   PetscCall(PetscSectionGetClosureIndex(section, (PetscObject)dm, clSec, clPoints));
6228   if (!ornt && *clPoints) {
6229     PetscInt dof, off;
6230 
6231     PetscCall(PetscSectionGetDof(*clSec, point, &dof));
6232     PetscCall(PetscSectionGetOffset(*clSec, point, &off));
6233     PetscCall(ISGetIndices(*clPoints, &cla));
6234     np  = dof / 2;
6235     pts = PetscSafePointerPlusOffset((PetscInt *)cla, off);
6236   } else {
6237     PetscCall(DMPlexGetTransitiveClosure_Internal(dm, point, ornt, PETSC_TRUE, &np, &pts));
6238     PetscCall(CompressPoints_Private(section, &np, pts));
6239   }
6240   *numPoints = np;
6241   *points    = pts;
6242   *clp       = cla;
6243   PetscFunctionReturn(PETSC_SUCCESS);
6244 }
6245 
6246 PetscErrorCode DMPlexRestoreCompressedClosure(DM dm, PetscSection section, PetscInt point, PetscInt *numPoints, PetscInt **points, PetscSection *clSec, IS *clPoints, const PetscInt **clp)
6247 {
6248   PetscFunctionBeginHot;
6249   if (!*clPoints) {
6250     PetscCall(DMPlexRestoreTransitiveClosure(dm, point, PETSC_TRUE, numPoints, points));
6251   } else {
6252     PetscCall(ISRestoreIndices(*clPoints, clp));
6253   }
6254   *numPoints = 0;
6255   *points    = NULL;
6256   *clSec     = NULL;
6257   *clPoints  = NULL;
6258   *clp       = NULL;
6259   PetscFunctionReturn(PETSC_SUCCESS);
6260 }
6261 
6262 static inline PetscErrorCode DMPlexVecGetClosure_Static(DM dm, PetscSection section, PetscInt numPoints, const PetscInt points[], const PetscInt clperm[], const PetscScalar vArray[], PetscInt *size, PetscScalar array[])
6263 {
6264   PetscInt            offset = 0, p;
6265   const PetscInt    **perms  = NULL;
6266   const PetscScalar **flips  = NULL;
6267 
6268   PetscFunctionBeginHot;
6269   *size = 0;
6270   PetscCall(PetscSectionGetPointSyms(section, numPoints, points, &perms, &flips));
6271   for (p = 0; p < numPoints; p++) {
6272     const PetscInt     point = points[2 * p];
6273     const PetscInt    *perm  = perms ? perms[p] : NULL;
6274     const PetscScalar *flip  = flips ? flips[p] : NULL;
6275     PetscInt           dof, off, d;
6276     const PetscScalar *varr;
6277 
6278     PetscCall(PetscSectionGetDof(section, point, &dof));
6279     PetscCall(PetscSectionGetOffset(section, point, &off));
6280     varr = PetscSafePointerPlusOffset(vArray, off);
6281     if (clperm) {
6282       if (perm) {
6283         for (d = 0; d < dof; d++) array[clperm[offset + perm[d]]] = varr[d];
6284       } else {
6285         for (d = 0; d < dof; d++) array[clperm[offset + d]] = varr[d];
6286       }
6287       if (flip) {
6288         for (d = 0; d < dof; d++) array[clperm[offset + d]] *= flip[d];
6289       }
6290     } else {
6291       if (perm) {
6292         for (d = 0; d < dof; d++) array[offset + perm[d]] = varr[d];
6293       } else {
6294         for (d = 0; d < dof; d++) array[offset + d] = varr[d];
6295       }
6296       if (flip) {
6297         for (d = 0; d < dof; d++) array[offset + d] *= flip[d];
6298       }
6299     }
6300     offset += dof;
6301   }
6302   PetscCall(PetscSectionRestorePointSyms(section, numPoints, points, &perms, &flips));
6303   *size = offset;
6304   PetscFunctionReturn(PETSC_SUCCESS);
6305 }
6306 
6307 static inline PetscErrorCode DMPlexVecGetClosure_Fields_Static(DM dm, PetscSection section, PetscInt numPoints, const PetscInt points[], PetscInt numFields, const PetscInt clperm[], const PetscScalar vArray[], PetscInt *size, PetscScalar array[])
6308 {
6309   PetscInt offset = 0, f;
6310 
6311   PetscFunctionBeginHot;
6312   *size = 0;
6313   for (f = 0; f < numFields; ++f) {
6314     PetscInt            p;
6315     const PetscInt    **perms = NULL;
6316     const PetscScalar **flips = NULL;
6317 
6318     PetscCall(PetscSectionGetFieldPointSyms(section, f, numPoints, points, &perms, &flips));
6319     for (p = 0; p < numPoints; p++) {
6320       const PetscInt     point = points[2 * p];
6321       PetscInt           fdof, foff, b;
6322       const PetscScalar *varr;
6323       const PetscInt    *perm = perms ? perms[p] : NULL;
6324       const PetscScalar *flip = flips ? flips[p] : NULL;
6325 
6326       PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
6327       PetscCall(PetscSectionGetFieldOffset(section, point, f, &foff));
6328       varr = &vArray[foff];
6329       if (clperm) {
6330         if (perm) {
6331           for (b = 0; b < fdof; b++) array[clperm[offset + perm[b]]] = varr[b];
6332         } else {
6333           for (b = 0; b < fdof; b++) array[clperm[offset + b]] = varr[b];
6334         }
6335         if (flip) {
6336           for (b = 0; b < fdof; b++) array[clperm[offset + b]] *= flip[b];
6337         }
6338       } else {
6339         if (perm) {
6340           for (b = 0; b < fdof; b++) array[offset + perm[b]] = varr[b];
6341         } else {
6342           for (b = 0; b < fdof; b++) array[offset + b] = varr[b];
6343         }
6344         if (flip) {
6345           for (b = 0; b < fdof; b++) array[offset + b] *= flip[b];
6346         }
6347       }
6348       offset += fdof;
6349     }
6350     PetscCall(PetscSectionRestoreFieldPointSyms(section, f, numPoints, points, &perms, &flips));
6351   }
6352   *size = offset;
6353   PetscFunctionReturn(PETSC_SUCCESS);
6354 }
6355 
6356 PetscErrorCode DMPlexVecGetOrientedClosure_Internal(DM dm, PetscSection section, PetscBool useClPerm, Vec v, PetscInt point, PetscInt ornt, PetscInt *csize, PetscScalar *values[])
6357 {
6358   PetscSection    clSection;
6359   IS              clPoints;
6360   PetscInt       *points = NULL;
6361   const PetscInt *clp, *perm = NULL;
6362   PetscInt        depth, numFields, numPoints, asize;
6363 
6364   PetscFunctionBeginHot;
6365   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6366   if (!section) PetscCall(DMGetLocalSection(dm, &section));
6367   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
6368   PetscValidHeaderSpecific(v, VEC_CLASSID, 4);
6369   PetscCall(DMPlexGetDepth(dm, &depth));
6370   PetscCall(PetscSectionGetNumFields(section, &numFields));
6371   if (depth == 1 && numFields < 2) {
6372     PetscCall(DMPlexVecGetClosure_Depth1_Static(dm, section, v, point, csize, values));
6373     PetscFunctionReturn(PETSC_SUCCESS);
6374   }
6375   /* Get points */
6376   PetscCall(DMPlexGetCompressedClosure(dm, section, point, ornt, &numPoints, &points, &clSection, &clPoints, &clp));
6377   /* Get sizes */
6378   asize = 0;
6379   for (PetscInt p = 0; p < numPoints * 2; p += 2) {
6380     PetscInt dof;
6381     PetscCall(PetscSectionGetDof(section, points[p], &dof));
6382     asize += dof;
6383   }
6384   if (values) {
6385     const PetscScalar *vArray;
6386     PetscInt           size;
6387 
6388     if (*values) {
6389       PetscCheck(*csize >= asize, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Provided array size %" PetscInt_FMT " not sufficient to hold closure size %" PetscInt_FMT, *csize, asize);
6390     } else PetscCall(DMGetWorkArray(dm, asize, MPIU_SCALAR, values));
6391     if (useClPerm) PetscCall(PetscSectionGetClosureInversePermutation_Internal(section, (PetscObject)dm, depth, asize, &perm));
6392     PetscCall(VecGetArrayRead(v, &vArray));
6393     /* Get values */
6394     if (numFields > 0) PetscCall(DMPlexVecGetClosure_Fields_Static(dm, section, numPoints, points, numFields, perm, vArray, &size, *values));
6395     else PetscCall(DMPlexVecGetClosure_Static(dm, section, numPoints, points, perm, vArray, &size, *values));
6396     PetscCheck(asize == size, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Section size %" PetscInt_FMT " does not match Vec closure size %" PetscInt_FMT, asize, size);
6397     /* Cleanup array */
6398     PetscCall(VecRestoreArrayRead(v, &vArray));
6399   }
6400   if (csize) *csize = asize;
6401   /* Cleanup points */
6402   PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &numPoints, &points, &clSection, &clPoints, &clp));
6403   PetscFunctionReturn(PETSC_SUCCESS);
6404 }
6405 
6406 /*@C
6407   DMPlexVecGetClosure - Get an array of the values on the closure of 'point'
6408 
6409   Not collective
6410 
6411   Input Parameters:
6412 + dm      - The `DM`
6413 . section - The section describing the layout in `v`, or `NULL` to use the default section
6414 . v       - The local vector
6415 - point   - The point in the `DM`
6416 
6417   Input/Output Parameters:
6418 + csize  - The size of the input values array, or `NULL`; on output the number of values in the closure
6419 - values - An array to use for the values, or `NULL` to have it allocated automatically;
6420            if the user provided `NULL`, it is a borrowed array and should not be freed
6421 
6422   Level: intermediate
6423 
6424   Notes:
6425   `DMPlexVecGetClosure()`/`DMPlexVecRestoreClosure()` only allocates the values array if it set to `NULL` in the
6426   calling function. This is because `DMPlexVecGetClosure()` is typically called in the inner loop of a `Vec` or `Mat`
6427   assembly function, and a user may already have allocated storage for this operation.
6428 
6429   A typical use could be
6430 .vb
6431    values = NULL;
6432    PetscCall(DMPlexVecGetClosure(dm, NULL, v, p, &clSize, &values));
6433    for (cl = 0; cl < clSize; ++cl) {
6434      <Compute on closure>
6435    }
6436    PetscCall(DMPlexVecRestoreClosure(dm, NULL, v, p, &clSize, &values));
6437 .ve
6438   or
6439 .vb
6440    PetscMalloc1(clMaxSize, &values);
6441    for (p = pStart; p < pEnd; ++p) {
6442      clSize = clMaxSize;
6443      PetscCall(DMPlexVecGetClosure(dm, NULL, v, p, &clSize, &values));
6444      for (cl = 0; cl < clSize; ++cl) {
6445        <Compute on closure>
6446      }
6447    }
6448    PetscFree(values);
6449 .ve
6450 
6451   Fortran Notes:
6452   The `csize` argument is not present in the Fortran binding since it is internal to the array.
6453 
6454 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexVecRestoreClosure()`, `DMPlexVecSetClosure()`, `DMPlexMatSetClosure()`
6455 @*/
6456 PetscErrorCode DMPlexVecGetClosure(DM dm, PetscSection section, Vec v, PetscInt point, PetscInt *csize, PetscScalar *values[])
6457 {
6458   PetscFunctionBeginHot;
6459   PetscCall(DMPlexVecGetOrientedClosure_Internal(dm, section, PETSC_TRUE, v, point, 0, csize, values));
6460   PetscFunctionReturn(PETSC_SUCCESS);
6461 }
6462 
6463 PetscErrorCode DMPlexVecGetClosureAtDepth_Internal(DM dm, PetscSection section, Vec v, PetscInt point, PetscInt depth, PetscInt *csize, PetscScalar *values[])
6464 {
6465   DMLabel            depthLabel;
6466   PetscSection       clSection;
6467   IS                 clPoints;
6468   PetscScalar       *array;
6469   const PetscScalar *vArray;
6470   PetscInt          *points = NULL;
6471   const PetscInt    *clp, *perm = NULL;
6472   PetscInt           mdepth, numFields, numPoints, Np = 0, p, clsize, size;
6473 
6474   PetscFunctionBeginHot;
6475   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6476   if (!section) PetscCall(DMGetLocalSection(dm, &section));
6477   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
6478   PetscValidHeaderSpecific(v, VEC_CLASSID, 3);
6479   PetscCall(DMPlexGetDepth(dm, &mdepth));
6480   PetscCall(DMPlexGetDepthLabel(dm, &depthLabel));
6481   PetscCall(PetscSectionGetNumFields(section, &numFields));
6482   if (mdepth == 1 && numFields < 2) {
6483     PetscCall(DMPlexVecGetClosure_Depth1_Static(dm, section, v, point, csize, values));
6484     PetscFunctionReturn(PETSC_SUCCESS);
6485   }
6486   /* Get points */
6487   PetscCall(DMPlexGetCompressedClosure(dm, section, point, 0, &numPoints, &points, &clSection, &clPoints, &clp));
6488   for (clsize = 0, p = 0; p < Np; p++) {
6489     PetscInt dof;
6490     PetscCall(PetscSectionGetDof(section, points[2 * p], &dof));
6491     clsize += dof;
6492   }
6493   PetscCall(PetscSectionGetClosureInversePermutation_Internal(section, (PetscObject)dm, depth, clsize, &perm));
6494   /* Filter points */
6495   for (p = 0; p < numPoints * 2; p += 2) {
6496     PetscInt dep;
6497 
6498     PetscCall(DMLabelGetValue(depthLabel, points[p], &dep));
6499     if (dep != depth) continue;
6500     points[Np * 2 + 0] = points[p];
6501     points[Np * 2 + 1] = points[p + 1];
6502     ++Np;
6503   }
6504   /* Get array */
6505   if (!values || !*values) {
6506     PetscInt asize = 0, dof;
6507 
6508     for (p = 0; p < Np * 2; p += 2) {
6509       PetscCall(PetscSectionGetDof(section, points[p], &dof));
6510       asize += dof;
6511     }
6512     if (!values) {
6513       PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &numPoints, &points, &clSection, &clPoints, &clp));
6514       if (csize) *csize = asize;
6515       PetscFunctionReturn(PETSC_SUCCESS);
6516     }
6517     PetscCall(DMGetWorkArray(dm, asize, MPIU_SCALAR, &array));
6518   } else {
6519     array = *values;
6520   }
6521   PetscCall(VecGetArrayRead(v, &vArray));
6522   /* Get values */
6523   if (numFields > 0) PetscCall(DMPlexVecGetClosure_Fields_Static(dm, section, Np, points, numFields, perm, vArray, &size, array));
6524   else PetscCall(DMPlexVecGetClosure_Static(dm, section, Np, points, perm, vArray, &size, array));
6525   /* Cleanup points */
6526   PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &numPoints, &points, &clSection, &clPoints, &clp));
6527   /* Cleanup array */
6528   PetscCall(VecRestoreArrayRead(v, &vArray));
6529   if (!*values) {
6530     if (csize) *csize = size;
6531     *values = array;
6532   } else {
6533     PetscCheck(size <= *csize, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Size of input array %" PetscInt_FMT " < actual size %" PetscInt_FMT, *csize, size);
6534     *csize = size;
6535   }
6536   PetscFunctionReturn(PETSC_SUCCESS);
6537 }
6538 
6539 /*@C
6540   DMPlexVecRestoreClosure - Restore the array of the values on the closure of 'point'
6541 
6542   Not collective
6543 
6544   Input Parameters:
6545 + dm      - The `DM`
6546 . section - The section describing the layout in `v`, or `NULL` to use the default section
6547 . v       - The local vector
6548 . point   - The point in the `DM`
6549 . csize   - The number of values in the closure, or `NULL`
6550 - values  - The array of values, which is a borrowed array and should not be freed
6551 
6552   Level: intermediate
6553 
6554   Note:
6555   The array values are discarded and not copied back into `v`. In order to copy values back to `v`, use `DMPlexVecSetClosure()`
6556 
6557   Fortran Notes:
6558   The `csize` argument is not present in the Fortran binding since it is internal to the array.
6559 
6560 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexVecGetClosure()`, `DMPlexVecSetClosure()`, `DMPlexMatSetClosure()`
6561 @*/
6562 PetscErrorCode DMPlexVecRestoreClosure(DM dm, PetscSection section, Vec v, PetscInt point, PetscInt *csize, PetscScalar *values[])
6563 {
6564   PetscInt size = 0;
6565 
6566   PetscFunctionBegin;
6567   /* Should work without recalculating size */
6568   PetscCall(DMRestoreWorkArray(dm, size, MPIU_SCALAR, (void *)values));
6569   *values = NULL;
6570   PetscFunctionReturn(PETSC_SUCCESS);
6571 }
6572 
6573 static inline void add(PetscScalar *x, PetscScalar y)
6574 {
6575   *x += y;
6576 }
6577 static inline void insert(PetscScalar *x, PetscScalar y)
6578 {
6579   *x = y;
6580 }
6581 
6582 static inline PetscErrorCode updatePoint_private(PetscSection section, PetscInt point, PetscInt dof, void (*fuse)(PetscScalar *, PetscScalar), PetscBool setBC, const PetscInt perm[], const PetscScalar flip[], const PetscInt clperm[], const PetscScalar values[], PetscInt offset, PetscScalar array[])
6583 {
6584   PetscInt        cdof;  /* The number of constraints on this point */
6585   const PetscInt *cdofs; /* The indices of the constrained dofs on this point */
6586   PetscScalar    *a;
6587   PetscInt        off, cind = 0, k;
6588 
6589   PetscFunctionBegin;
6590   PetscCall(PetscSectionGetConstraintDof(section, point, &cdof));
6591   PetscCall(PetscSectionGetOffset(section, point, &off));
6592   a = &array[off];
6593   if (!cdof || setBC) {
6594     if (clperm) {
6595       if (perm) {
6596         for (k = 0; k < dof; ++k) fuse(&a[k], values[clperm[offset + perm[k]]] * (flip ? flip[perm[k]] : 1.));
6597       } else {
6598         for (k = 0; k < dof; ++k) fuse(&a[k], values[clperm[offset + k]] * (flip ? flip[k] : 1.));
6599       }
6600     } else {
6601       if (perm) {
6602         for (k = 0; k < dof; ++k) fuse(&a[k], values[offset + perm[k]] * (flip ? flip[perm[k]] : 1.));
6603       } else {
6604         for (k = 0; k < dof; ++k) fuse(&a[k], values[offset + k] * (flip ? flip[k] : 1.));
6605       }
6606     }
6607   } else {
6608     PetscCall(PetscSectionGetConstraintIndices(section, point, &cdofs));
6609     if (clperm) {
6610       if (perm) {
6611         for (k = 0; k < dof; ++k) {
6612           if ((cind < cdof) && (k == cdofs[cind])) {
6613             ++cind;
6614             continue;
6615           }
6616           fuse(&a[k], values[clperm[offset + perm[k]]] * (flip ? flip[perm[k]] : 1.));
6617         }
6618       } else {
6619         for (k = 0; k < dof; ++k) {
6620           if ((cind < cdof) && (k == cdofs[cind])) {
6621             ++cind;
6622             continue;
6623           }
6624           fuse(&a[k], values[clperm[offset + k]] * (flip ? flip[k] : 1.));
6625         }
6626       }
6627     } else {
6628       if (perm) {
6629         for (k = 0; k < dof; ++k) {
6630           if ((cind < cdof) && (k == cdofs[cind])) {
6631             ++cind;
6632             continue;
6633           }
6634           fuse(&a[k], values[offset + perm[k]] * (flip ? flip[perm[k]] : 1.));
6635         }
6636       } else {
6637         for (k = 0; k < dof; ++k) {
6638           if ((cind < cdof) && (k == cdofs[cind])) {
6639             ++cind;
6640             continue;
6641           }
6642           fuse(&a[k], values[offset + k] * (flip ? flip[k] : 1.));
6643         }
6644       }
6645     }
6646   }
6647   PetscFunctionReturn(PETSC_SUCCESS);
6648 }
6649 
6650 static inline PetscErrorCode updatePointBC_private(PetscSection section, PetscInt point, PetscInt dof, void (*fuse)(PetscScalar *, PetscScalar), const PetscInt perm[], const PetscScalar flip[], const PetscInt clperm[], const PetscScalar values[], PetscInt offset, PetscScalar array[])
6651 {
6652   PetscInt        cdof;  /* The number of constraints on this point */
6653   const PetscInt *cdofs; /* The indices of the constrained dofs on this point */
6654   PetscScalar    *a;
6655   PetscInt        off, cind = 0, k;
6656 
6657   PetscFunctionBegin;
6658   PetscCall(PetscSectionGetConstraintDof(section, point, &cdof));
6659   PetscCall(PetscSectionGetOffset(section, point, &off));
6660   a = &array[off];
6661   if (cdof) {
6662     PetscCall(PetscSectionGetConstraintIndices(section, point, &cdofs));
6663     if (clperm) {
6664       if (perm) {
6665         for (k = 0; k < dof; ++k) {
6666           if ((cind < cdof) && (k == cdofs[cind])) {
6667             fuse(&a[k], values[clperm[offset + perm[k]]] * (flip ? flip[perm[k]] : 1.));
6668             cind++;
6669           }
6670         }
6671       } else {
6672         for (k = 0; k < dof; ++k) {
6673           if ((cind < cdof) && (k == cdofs[cind])) {
6674             fuse(&a[k], values[clperm[offset + k]] * (flip ? flip[k] : 1.));
6675             cind++;
6676           }
6677         }
6678       }
6679     } else {
6680       if (perm) {
6681         for (k = 0; k < dof; ++k) {
6682           if ((cind < cdof) && (k == cdofs[cind])) {
6683             fuse(&a[k], values[offset + perm[k]] * (flip ? flip[perm[k]] : 1.));
6684             cind++;
6685           }
6686         }
6687       } else {
6688         for (k = 0; k < dof; ++k) {
6689           if ((cind < cdof) && (k == cdofs[cind])) {
6690             fuse(&a[k], values[offset + k] * (flip ? flip[k] : 1.));
6691             cind++;
6692           }
6693         }
6694       }
6695     }
6696   }
6697   PetscFunctionReturn(PETSC_SUCCESS);
6698 }
6699 
6700 static inline PetscErrorCode updatePointFields_private(PetscSection section, PetscInt point, const PetscInt *perm, const PetscScalar *flip, PetscInt f, void (*fuse)(PetscScalar *, PetscScalar), PetscBool setBC, const PetscInt clperm[], const PetscScalar values[], PetscInt *offset, PetscScalar array[])
6701 {
6702   PetscScalar    *a;
6703   PetscInt        fdof, foff, fcdof, foffset = *offset;
6704   const PetscInt *fcdofs; /* The indices of the constrained dofs for field f on this point */
6705   PetscInt        cind = 0, b;
6706 
6707   PetscFunctionBegin;
6708   PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
6709   PetscCall(PetscSectionGetFieldConstraintDof(section, point, f, &fcdof));
6710   PetscCall(PetscSectionGetFieldOffset(section, point, f, &foff));
6711   a = &array[foff];
6712   if (!fcdof || setBC) {
6713     if (clperm) {
6714       if (perm) {
6715         for (b = 0; b < fdof; b++) fuse(&a[b], values[clperm[foffset + perm[b]]] * (flip ? flip[perm[b]] : 1.));
6716       } else {
6717         for (b = 0; b < fdof; b++) fuse(&a[b], values[clperm[foffset + b]] * (flip ? flip[b] : 1.));
6718       }
6719     } else {
6720       if (perm) {
6721         for (b = 0; b < fdof; b++) fuse(&a[b], values[foffset + perm[b]] * (flip ? flip[perm[b]] : 1.));
6722       } else {
6723         for (b = 0; b < fdof; b++) fuse(&a[b], values[foffset + b] * (flip ? flip[b] : 1.));
6724       }
6725     }
6726   } else {
6727     PetscCall(PetscSectionGetFieldConstraintIndices(section, point, f, &fcdofs));
6728     if (clperm) {
6729       if (perm) {
6730         for (b = 0; b < fdof; b++) {
6731           if ((cind < fcdof) && (b == fcdofs[cind])) {
6732             ++cind;
6733             continue;
6734           }
6735           fuse(&a[b], values[clperm[foffset + perm[b]]] * (flip ? flip[perm[b]] : 1.));
6736         }
6737       } else {
6738         for (b = 0; b < fdof; b++) {
6739           if ((cind < fcdof) && (b == fcdofs[cind])) {
6740             ++cind;
6741             continue;
6742           }
6743           fuse(&a[b], values[clperm[foffset + b]] * (flip ? flip[b] : 1.));
6744         }
6745       }
6746     } else {
6747       if (perm) {
6748         for (b = 0; b < fdof; b++) {
6749           if ((cind < fcdof) && (b == fcdofs[cind])) {
6750             ++cind;
6751             continue;
6752           }
6753           fuse(&a[b], values[foffset + perm[b]] * (flip ? flip[perm[b]] : 1.));
6754         }
6755       } else {
6756         for (b = 0; b < fdof; b++) {
6757           if ((cind < fcdof) && (b == fcdofs[cind])) {
6758             ++cind;
6759             continue;
6760           }
6761           fuse(&a[b], values[foffset + b] * (flip ? flip[b] : 1.));
6762         }
6763       }
6764     }
6765   }
6766   *offset += fdof;
6767   PetscFunctionReturn(PETSC_SUCCESS);
6768 }
6769 
6770 static inline PetscErrorCode updatePointFieldsBC_private(PetscSection section, PetscInt point, const PetscInt perm[], const PetscScalar flip[], PetscInt f, PetscInt Ncc, const PetscInt comps[], void (*fuse)(PetscScalar *, PetscScalar), const PetscInt clperm[], const PetscScalar values[], PetscInt *offset, PetscScalar array[])
6771 {
6772   PetscScalar    *a;
6773   PetscInt        fdof, foff, fcdof, foffset = *offset;
6774   const PetscInt *fcdofs; /* The indices of the constrained dofs for field f on this point */
6775   PetscInt        Nc, cind = 0, ncind = 0, b;
6776   PetscBool       ncSet, fcSet;
6777 
6778   PetscFunctionBegin;
6779   PetscCall(PetscSectionGetFieldComponents(section, f, &Nc));
6780   PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
6781   PetscCall(PetscSectionGetFieldConstraintDof(section, point, f, &fcdof));
6782   PetscCall(PetscSectionGetFieldOffset(section, point, f, &foff));
6783   a = &array[foff];
6784   if (fcdof) {
6785     /* We just override fcdof and fcdofs with Ncc and comps */
6786     PetscCall(PetscSectionGetFieldConstraintIndices(section, point, f, &fcdofs));
6787     if (clperm) {
6788       if (perm) {
6789         if (comps) {
6790           for (b = 0; b < fdof; b++) {
6791             ncSet = fcSet = PETSC_FALSE;
6792             if (b % Nc == comps[ncind]) {
6793               ncind = (ncind + 1) % Ncc;
6794               ncSet = PETSC_TRUE;
6795             }
6796             if ((cind < fcdof) && (b == fcdofs[cind])) {
6797               ++cind;
6798               fcSet = PETSC_TRUE;
6799             }
6800             if (ncSet && fcSet) fuse(&a[b], values[clperm[foffset + perm[b]]] * (flip ? flip[perm[b]] : 1.));
6801           }
6802         } else {
6803           for (b = 0; b < fdof; b++) {
6804             if ((cind < fcdof) && (b == fcdofs[cind])) {
6805               fuse(&a[b], values[clperm[foffset + perm[b]]] * (flip ? flip[perm[b]] : 1.));
6806               ++cind;
6807             }
6808           }
6809         }
6810       } else {
6811         if (comps) {
6812           for (b = 0; b < fdof; b++) {
6813             ncSet = fcSet = PETSC_FALSE;
6814             if (b % Nc == comps[ncind]) {
6815               ncind = (ncind + 1) % Ncc;
6816               ncSet = PETSC_TRUE;
6817             }
6818             if ((cind < fcdof) && (b == fcdofs[cind])) {
6819               ++cind;
6820               fcSet = PETSC_TRUE;
6821             }
6822             if (ncSet && fcSet) fuse(&a[b], values[clperm[foffset + b]] * (flip ? flip[b] : 1.));
6823           }
6824         } else {
6825           for (b = 0; b < fdof; b++) {
6826             if ((cind < fcdof) && (b == fcdofs[cind])) {
6827               fuse(&a[b], values[clperm[foffset + b]] * (flip ? flip[b] : 1.));
6828               ++cind;
6829             }
6830           }
6831         }
6832       }
6833     } else {
6834       if (perm) {
6835         if (comps) {
6836           for (b = 0; b < fdof; b++) {
6837             ncSet = fcSet = PETSC_FALSE;
6838             if (b % Nc == comps[ncind]) {
6839               ncind = (ncind + 1) % Ncc;
6840               ncSet = PETSC_TRUE;
6841             }
6842             if ((cind < fcdof) && (b == fcdofs[cind])) {
6843               ++cind;
6844               fcSet = PETSC_TRUE;
6845             }
6846             if (ncSet && fcSet) fuse(&a[b], values[foffset + perm[b]] * (flip ? flip[perm[b]] : 1.));
6847           }
6848         } else {
6849           for (b = 0; b < fdof; b++) {
6850             if ((cind < fcdof) && (b == fcdofs[cind])) {
6851               fuse(&a[b], values[foffset + perm[b]] * (flip ? flip[perm[b]] : 1.));
6852               ++cind;
6853             }
6854           }
6855         }
6856       } else {
6857         if (comps) {
6858           for (b = 0; b < fdof; b++) {
6859             ncSet = fcSet = PETSC_FALSE;
6860             if (b % Nc == comps[ncind]) {
6861               ncind = (ncind + 1) % Ncc;
6862               ncSet = PETSC_TRUE;
6863             }
6864             if ((cind < fcdof) && (b == fcdofs[cind])) {
6865               ++cind;
6866               fcSet = PETSC_TRUE;
6867             }
6868             if (ncSet && fcSet) fuse(&a[b], values[foffset + b] * (flip ? flip[b] : 1.));
6869           }
6870         } else {
6871           for (b = 0; b < fdof; b++) {
6872             if ((cind < fcdof) && (b == fcdofs[cind])) {
6873               fuse(&a[b], values[foffset + b] * (flip ? flip[b] : 1.));
6874               ++cind;
6875             }
6876           }
6877         }
6878       }
6879     }
6880   }
6881   *offset += fdof;
6882   PetscFunctionReturn(PETSC_SUCCESS);
6883 }
6884 
6885 static inline PetscErrorCode DMPlexVecSetClosure_Depth1_Static(DM dm, PetscSection section, Vec v, PetscInt point, const PetscScalar values[], InsertMode mode)
6886 {
6887   PetscScalar    *array;
6888   const PetscInt *cone, *coneO;
6889   PetscInt        pStart, pEnd, p, numPoints, off, dof;
6890 
6891   PetscFunctionBeginHot;
6892   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
6893   PetscCall(DMPlexGetConeSize(dm, point, &numPoints));
6894   PetscCall(DMPlexGetCone(dm, point, &cone));
6895   PetscCall(DMPlexGetConeOrientation(dm, point, &coneO));
6896   PetscCall(VecGetArray(v, &array));
6897   for (p = 0, off = 0; p <= numPoints; ++p, off += dof) {
6898     const PetscInt cp = !p ? point : cone[p - 1];
6899     const PetscInt o  = !p ? 0 : coneO[p - 1];
6900 
6901     if ((cp < pStart) || (cp >= pEnd)) {
6902       dof = 0;
6903       continue;
6904     }
6905     PetscCall(PetscSectionGetDof(section, cp, &dof));
6906     /* ADD_VALUES */
6907     {
6908       const PetscInt *cdofs; /* The indices of the constrained dofs on this point */
6909       PetscScalar    *a;
6910       PetscInt        cdof, coff, cind = 0, k;
6911 
6912       PetscCall(PetscSectionGetConstraintDof(section, cp, &cdof));
6913       PetscCall(PetscSectionGetOffset(section, cp, &coff));
6914       a = &array[coff];
6915       if (!cdof) {
6916         if (o >= 0) {
6917           for (k = 0; k < dof; ++k) a[k] += values[off + k];
6918         } else {
6919           for (k = 0; k < dof; ++k) a[k] += values[off + dof - k - 1];
6920         }
6921       } else {
6922         PetscCall(PetscSectionGetConstraintIndices(section, cp, &cdofs));
6923         if (o >= 0) {
6924           for (k = 0; k < dof; ++k) {
6925             if ((cind < cdof) && (k == cdofs[cind])) {
6926               ++cind;
6927               continue;
6928             }
6929             a[k] += values[off + k];
6930           }
6931         } else {
6932           for (k = 0; k < dof; ++k) {
6933             if ((cind < cdof) && (k == cdofs[cind])) {
6934               ++cind;
6935               continue;
6936             }
6937             a[k] += values[off + dof - k - 1];
6938           }
6939         }
6940       }
6941     }
6942   }
6943   PetscCall(VecRestoreArray(v, &array));
6944   PetscFunctionReturn(PETSC_SUCCESS);
6945 }
6946 
6947 /*@C
6948   DMPlexVecSetClosure - Set an array of the values on the closure of `point`
6949 
6950   Not collective
6951 
6952   Input Parameters:
6953 + dm      - The `DM`
6954 . section - The section describing the layout in `v`, or `NULL` to use the default section
6955 . v       - The local vector
6956 . point   - The point in the `DM`
6957 . values  - The array of values
6958 - mode    - The insert mode. One of `INSERT_ALL_VALUES`, `ADD_ALL_VALUES`, `INSERT_VALUES`, `ADD_VALUES`, `INSERT_BC_VALUES`, and `ADD_BC_VALUES`,
6959          where `INSERT_ALL_VALUES` and `ADD_ALL_VALUES` also overwrite boundary conditions.
6960 
6961   Level: intermediate
6962 
6963 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexVecGetClosure()`, `DMPlexMatSetClosure()`
6964 @*/
6965 PetscErrorCode DMPlexVecSetClosure(DM dm, PetscSection section, Vec v, PetscInt point, const PetscScalar values[], InsertMode mode)
6966 {
6967   PetscSection    clSection;
6968   IS              clPoints;
6969   PetscScalar    *array;
6970   PetscInt       *points = NULL;
6971   const PetscInt *clp, *clperm = NULL;
6972   PetscInt        depth, numFields, numPoints, p, clsize;
6973 
6974   PetscFunctionBeginHot;
6975   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6976   if (!section) PetscCall(DMGetLocalSection(dm, &section));
6977   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
6978   PetscValidHeaderSpecific(v, VEC_CLASSID, 3);
6979   PetscCall(DMPlexGetDepth(dm, &depth));
6980   PetscCall(PetscSectionGetNumFields(section, &numFields));
6981   if (depth == 1 && numFields < 2 && mode == ADD_VALUES) {
6982     PetscCall(DMPlexVecSetClosure_Depth1_Static(dm, section, v, point, values, mode));
6983     PetscFunctionReturn(PETSC_SUCCESS);
6984   }
6985   /* Get points */
6986   PetscCall(DMPlexGetCompressedClosure(dm, section, point, 0, &numPoints, &points, &clSection, &clPoints, &clp));
6987   for (clsize = 0, p = 0; p < numPoints; p++) {
6988     PetscInt dof;
6989     PetscCall(PetscSectionGetDof(section, points[2 * p], &dof));
6990     clsize += dof;
6991   }
6992   PetscCall(PetscSectionGetClosureInversePermutation_Internal(section, (PetscObject)dm, depth, clsize, &clperm));
6993   /* Get array */
6994   PetscCall(VecGetArray(v, &array));
6995   /* Get values */
6996   if (numFields > 0) {
6997     PetscInt offset = 0, f;
6998     for (f = 0; f < numFields; ++f) {
6999       const PetscInt    **perms = NULL;
7000       const PetscScalar **flips = NULL;
7001 
7002       PetscCall(PetscSectionGetFieldPointSyms(section, f, numPoints, points, &perms, &flips));
7003       switch (mode) {
7004       case INSERT_VALUES:
7005         for (p = 0; p < numPoints; p++) {
7006           const PetscInt     point = points[2 * p];
7007           const PetscInt    *perm  = perms ? perms[p] : NULL;
7008           const PetscScalar *flip  = flips ? flips[p] : NULL;
7009           PetscCall(updatePointFields_private(section, point, perm, flip, f, insert, PETSC_FALSE, clperm, values, &offset, array));
7010         }
7011         break;
7012       case INSERT_ALL_VALUES:
7013         for (p = 0; p < numPoints; p++) {
7014           const PetscInt     point = points[2 * p];
7015           const PetscInt    *perm  = perms ? perms[p] : NULL;
7016           const PetscScalar *flip  = flips ? flips[p] : NULL;
7017           PetscCall(updatePointFields_private(section, point, perm, flip, f, insert, PETSC_TRUE, clperm, values, &offset, array));
7018         }
7019         break;
7020       case INSERT_BC_VALUES:
7021         for (p = 0; p < numPoints; p++) {
7022           const PetscInt     point = points[2 * p];
7023           const PetscInt    *perm  = perms ? perms[p] : NULL;
7024           const PetscScalar *flip  = flips ? flips[p] : NULL;
7025           PetscCall(updatePointFieldsBC_private(section, point, perm, flip, f, -1, NULL, insert, clperm, values, &offset, array));
7026         }
7027         break;
7028       case ADD_VALUES:
7029         for (p = 0; p < numPoints; p++) {
7030           const PetscInt     point = points[2 * p];
7031           const PetscInt    *perm  = perms ? perms[p] : NULL;
7032           const PetscScalar *flip  = flips ? flips[p] : NULL;
7033           PetscCall(updatePointFields_private(section, point, perm, flip, f, add, PETSC_FALSE, clperm, values, &offset, array));
7034         }
7035         break;
7036       case ADD_ALL_VALUES:
7037         for (p = 0; p < numPoints; p++) {
7038           const PetscInt     point = points[2 * p];
7039           const PetscInt    *perm  = perms ? perms[p] : NULL;
7040           const PetscScalar *flip  = flips ? flips[p] : NULL;
7041           PetscCall(updatePointFields_private(section, point, perm, flip, f, add, PETSC_TRUE, clperm, values, &offset, array));
7042         }
7043         break;
7044       case ADD_BC_VALUES:
7045         for (p = 0; p < numPoints; p++) {
7046           const PetscInt     point = points[2 * p];
7047           const PetscInt    *perm  = perms ? perms[p] : NULL;
7048           const PetscScalar *flip  = flips ? flips[p] : NULL;
7049           PetscCall(updatePointFieldsBC_private(section, point, perm, flip, f, -1, NULL, add, clperm, values, &offset, array));
7050         }
7051         break;
7052       default:
7053         SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insert mode %d", mode);
7054       }
7055       PetscCall(PetscSectionRestoreFieldPointSyms(section, f, numPoints, points, &perms, &flips));
7056     }
7057   } else {
7058     PetscInt            dof, off;
7059     const PetscInt    **perms = NULL;
7060     const PetscScalar **flips = NULL;
7061 
7062     PetscCall(PetscSectionGetPointSyms(section, numPoints, points, &perms, &flips));
7063     switch (mode) {
7064     case INSERT_VALUES:
7065       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7066         const PetscInt     point = points[2 * p];
7067         const PetscInt    *perm  = perms ? perms[p] : NULL;
7068         const PetscScalar *flip  = flips ? flips[p] : NULL;
7069         PetscCall(PetscSectionGetDof(section, point, &dof));
7070         PetscCall(updatePoint_private(section, point, dof, insert, PETSC_FALSE, perm, flip, clperm, values, off, array));
7071       }
7072       break;
7073     case INSERT_ALL_VALUES:
7074       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7075         const PetscInt     point = points[2 * p];
7076         const PetscInt    *perm  = perms ? perms[p] : NULL;
7077         const PetscScalar *flip  = flips ? flips[p] : NULL;
7078         PetscCall(PetscSectionGetDof(section, point, &dof));
7079         PetscCall(updatePoint_private(section, point, dof, insert, PETSC_TRUE, perm, flip, clperm, values, off, array));
7080       }
7081       break;
7082     case INSERT_BC_VALUES:
7083       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7084         const PetscInt     point = points[2 * p];
7085         const PetscInt    *perm  = perms ? perms[p] : NULL;
7086         const PetscScalar *flip  = flips ? flips[p] : NULL;
7087         PetscCall(PetscSectionGetDof(section, point, &dof));
7088         PetscCall(updatePointBC_private(section, point, dof, insert, perm, flip, clperm, values, off, array));
7089       }
7090       break;
7091     case ADD_VALUES:
7092       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7093         const PetscInt     point = points[2 * p];
7094         const PetscInt    *perm  = perms ? perms[p] : NULL;
7095         const PetscScalar *flip  = flips ? flips[p] : NULL;
7096         PetscCall(PetscSectionGetDof(section, point, &dof));
7097         PetscCall(updatePoint_private(section, point, dof, add, PETSC_FALSE, perm, flip, clperm, values, off, array));
7098       }
7099       break;
7100     case ADD_ALL_VALUES:
7101       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7102         const PetscInt     point = points[2 * p];
7103         const PetscInt    *perm  = perms ? perms[p] : NULL;
7104         const PetscScalar *flip  = flips ? flips[p] : NULL;
7105         PetscCall(PetscSectionGetDof(section, point, &dof));
7106         PetscCall(updatePoint_private(section, point, dof, add, PETSC_TRUE, perm, flip, clperm, values, off, array));
7107       }
7108       break;
7109     case ADD_BC_VALUES:
7110       for (p = 0, off = 0; p < numPoints; p++, off += dof) {
7111         const PetscInt     point = points[2 * p];
7112         const PetscInt    *perm  = perms ? perms[p] : NULL;
7113         const PetscScalar *flip  = flips ? flips[p] : NULL;
7114         PetscCall(PetscSectionGetDof(section, point, &dof));
7115         PetscCall(updatePointBC_private(section, point, dof, add, perm, flip, clperm, values, off, array));
7116       }
7117       break;
7118     default:
7119       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insert mode %d", mode);
7120     }
7121     PetscCall(PetscSectionRestorePointSyms(section, numPoints, points, &perms, &flips));
7122   }
7123   /* Cleanup points */
7124   PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &numPoints, &points, &clSection, &clPoints, &clp));
7125   /* Cleanup array */
7126   PetscCall(VecRestoreArray(v, &array));
7127   PetscFunctionReturn(PETSC_SUCCESS);
7128 }
7129 
7130 /* Check whether the given point is in the label. If not, update the offset to skip this point */
7131 static inline PetscErrorCode CheckPoint_Private(DMLabel label, PetscInt labelId, PetscSection section, PetscInt point, PetscInt f, PetscInt *offset, PetscBool *contains)
7132 {
7133   PetscFunctionBegin;
7134   *contains = PETSC_TRUE;
7135   if (label) {
7136     PetscInt fdof;
7137 
7138     PetscCall(DMLabelStratumHasPoint(label, labelId, point, contains));
7139     if (!*contains) {
7140       PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
7141       *offset += fdof;
7142       PetscFunctionReturn(PETSC_SUCCESS);
7143     }
7144   }
7145   PetscFunctionReturn(PETSC_SUCCESS);
7146 }
7147 
7148 /* Unlike DMPlexVecSetClosure(), this uses plex-native closure permutation, not a user-specified permutation such as DMPlexSetClosurePermutationTensor(). */
7149 PetscErrorCode DMPlexVecSetFieldClosure_Internal(DM dm, PetscSection section, Vec v, PetscBool fieldActive[], PetscInt point, PetscInt Ncc, const PetscInt comps[], DMLabel label, PetscInt labelId, const PetscScalar values[], InsertMode mode)
7150 {
7151   PetscSection    clSection;
7152   IS              clPoints;
7153   PetscScalar    *array;
7154   PetscInt       *points = NULL;
7155   const PetscInt *clp;
7156   PetscInt        numFields, numPoints, p;
7157   PetscInt        offset = 0, f;
7158 
7159   PetscFunctionBeginHot;
7160   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7161   if (!section) PetscCall(DMGetLocalSection(dm, &section));
7162   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
7163   PetscValidHeaderSpecific(v, VEC_CLASSID, 3);
7164   PetscCall(PetscSectionGetNumFields(section, &numFields));
7165   /* Get points */
7166   PetscCall(DMPlexGetCompressedClosure(dm, section, point, 0, &numPoints, &points, &clSection, &clPoints, &clp));
7167   /* Get array */
7168   PetscCall(VecGetArray(v, &array));
7169   /* Get values */
7170   for (f = 0; f < numFields; ++f) {
7171     const PetscInt    **perms = NULL;
7172     const PetscScalar **flips = NULL;
7173     PetscBool           contains;
7174 
7175     if (!fieldActive[f]) {
7176       for (p = 0; p < numPoints * 2; p += 2) {
7177         PetscInt fdof;
7178         PetscCall(PetscSectionGetFieldDof(section, points[p], f, &fdof));
7179         offset += fdof;
7180       }
7181       continue;
7182     }
7183     PetscCall(PetscSectionGetFieldPointSyms(section, f, numPoints, points, &perms, &flips));
7184     switch (mode) {
7185     case INSERT_VALUES:
7186       for (p = 0; p < numPoints; p++) {
7187         const PetscInt     point = points[2 * p];
7188         const PetscInt    *perm  = perms ? perms[p] : NULL;
7189         const PetscScalar *flip  = flips ? flips[p] : NULL;
7190         PetscCall(CheckPoint_Private(label, labelId, section, point, f, &offset, &contains));
7191         if (!contains) continue;
7192         PetscCall(updatePointFields_private(section, point, perm, flip, f, insert, PETSC_FALSE, NULL, values, &offset, array));
7193       }
7194       break;
7195     case INSERT_ALL_VALUES:
7196       for (p = 0; p < numPoints; p++) {
7197         const PetscInt     point = points[2 * p];
7198         const PetscInt    *perm  = perms ? perms[p] : NULL;
7199         const PetscScalar *flip  = flips ? flips[p] : NULL;
7200         PetscCall(CheckPoint_Private(label, labelId, section, point, f, &offset, &contains));
7201         if (!contains) continue;
7202         PetscCall(updatePointFields_private(section, point, perm, flip, f, insert, PETSC_TRUE, NULL, values, &offset, array));
7203       }
7204       break;
7205     case INSERT_BC_VALUES:
7206       for (p = 0; p < numPoints; p++) {
7207         const PetscInt     point = points[2 * p];
7208         const PetscInt    *perm  = perms ? perms[p] : NULL;
7209         const PetscScalar *flip  = flips ? flips[p] : NULL;
7210         PetscCall(CheckPoint_Private(label, labelId, section, point, f, &offset, &contains));
7211         if (!contains) continue;
7212         PetscCall(updatePointFieldsBC_private(section, point, perm, flip, f, Ncc, comps, insert, NULL, values, &offset, array));
7213       }
7214       break;
7215     case ADD_VALUES:
7216       for (p = 0; p < numPoints; p++) {
7217         const PetscInt     point = points[2 * p];
7218         const PetscInt    *perm  = perms ? perms[p] : NULL;
7219         const PetscScalar *flip  = flips ? flips[p] : NULL;
7220         PetscCall(CheckPoint_Private(label, labelId, section, point, f, &offset, &contains));
7221         if (!contains) continue;
7222         PetscCall(updatePointFields_private(section, point, perm, flip, f, add, PETSC_FALSE, NULL, values, &offset, array));
7223       }
7224       break;
7225     case ADD_ALL_VALUES:
7226       for (p = 0; p < numPoints; p++) {
7227         const PetscInt     point = points[2 * p];
7228         const PetscInt    *perm  = perms ? perms[p] : NULL;
7229         const PetscScalar *flip  = flips ? flips[p] : NULL;
7230         PetscCall(CheckPoint_Private(label, labelId, section, point, f, &offset, &contains));
7231         if (!contains) continue;
7232         PetscCall(updatePointFields_private(section, point, perm, flip, f, add, PETSC_TRUE, NULL, values, &offset, array));
7233       }
7234       break;
7235     default:
7236       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insert mode %d", mode);
7237     }
7238     PetscCall(PetscSectionRestoreFieldPointSyms(section, f, numPoints, points, &perms, &flips));
7239   }
7240   /* Cleanup points */
7241   PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &numPoints, &points, &clSection, &clPoints, &clp));
7242   /* Cleanup array */
7243   PetscCall(VecRestoreArray(v, &array));
7244   PetscFunctionReturn(PETSC_SUCCESS);
7245 }
7246 
7247 static PetscErrorCode DMPlexPrintMatSetValues(PetscViewer viewer, Mat A, PetscInt point, PetscInt numRIndices, const PetscInt rindices[], PetscInt numCIndices, const PetscInt cindices[], const PetscScalar values[])
7248 {
7249   PetscMPIInt rank;
7250   PetscInt    i, j;
7251 
7252   PetscFunctionBegin;
7253   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)A), &rank));
7254   PetscCall(PetscViewerASCIIPrintf(viewer, "[%d]mat for point %" PetscInt_FMT "\n", rank, point));
7255   for (i = 0; i < numRIndices; i++) PetscCall(PetscViewerASCIIPrintf(viewer, "[%d]mat row indices[%" PetscInt_FMT "] = %" PetscInt_FMT "\n", rank, i, rindices[i]));
7256   for (i = 0; i < numCIndices; i++) PetscCall(PetscViewerASCIIPrintf(viewer, "[%d]mat col indices[%" PetscInt_FMT "] = %" PetscInt_FMT "\n", rank, i, cindices[i]));
7257   numCIndices = numCIndices ? numCIndices : numRIndices;
7258   if (!values) PetscFunctionReturn(PETSC_SUCCESS);
7259   for (i = 0; i < numRIndices; i++) {
7260     PetscCall(PetscViewerASCIIPrintf(viewer, "[%d]", rank));
7261     for (j = 0; j < numCIndices; j++) {
7262 #if defined(PETSC_USE_COMPLEX)
7263       PetscCall(PetscViewerASCIIPrintf(viewer, " (%g,%g)", (double)PetscRealPart(values[i * numCIndices + j]), (double)PetscImaginaryPart(values[i * numCIndices + j])));
7264 #else
7265       PetscCall(PetscViewerASCIIPrintf(viewer, " %g", (double)values[i * numCIndices + j]));
7266 #endif
7267     }
7268     PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
7269   }
7270   PetscFunctionReturn(PETSC_SUCCESS);
7271 }
7272 
7273 /*
7274   DMPlexGetIndicesPoint_Internal - Add the indices for dofs on a point to an index array
7275 
7276   Input Parameters:
7277 + section - The section for this data layout
7278 . islocal - Is the section (and thus indices being requested) local or global?
7279 . point   - The point contributing dofs with these indices
7280 . off     - The global offset of this point
7281 . loff    - The local offset of each field
7282 . setBC   - The flag determining whether to include indices of boundary values
7283 . perm    - A permutation of the dofs on this point, or NULL
7284 - indperm - A permutation of the entire indices array, or NULL
7285 
7286   Output Parameter:
7287 . indices - Indices for dofs on this point
7288 
7289   Level: developer
7290 
7291   Note: The indices could be local or global, depending on the value of 'off'.
7292 */
7293 PetscErrorCode DMPlexGetIndicesPoint_Internal(PetscSection section, PetscBool islocal, PetscInt point, PetscInt off, PetscInt *loff, PetscBool setBC, const PetscInt perm[], const PetscInt indperm[], PetscInt indices[])
7294 {
7295   PetscInt        dof;   /* The number of unknowns on this point */
7296   PetscInt        cdof;  /* The number of constraints on this point */
7297   const PetscInt *cdofs; /* The indices of the constrained dofs on this point */
7298   PetscInt        cind = 0, k;
7299 
7300   PetscFunctionBegin;
7301   PetscCheck(islocal || !setBC, PetscObjectComm((PetscObject)section), PETSC_ERR_ARG_INCOMP, "setBC incompatible with global indices; use a local section or disable setBC");
7302   PetscCall(PetscSectionGetDof(section, point, &dof));
7303   PetscCall(PetscSectionGetConstraintDof(section, point, &cdof));
7304   if (!cdof || setBC) {
7305     for (k = 0; k < dof; ++k) {
7306       const PetscInt preind = perm ? *loff + perm[k] : *loff + k;
7307       const PetscInt ind    = indperm ? indperm[preind] : preind;
7308 
7309       indices[ind] = off + k;
7310     }
7311   } else {
7312     PetscCall(PetscSectionGetConstraintIndices(section, point, &cdofs));
7313     for (k = 0; k < dof; ++k) {
7314       const PetscInt preind = perm ? *loff + perm[k] : *loff + k;
7315       const PetscInt ind    = indperm ? indperm[preind] : preind;
7316 
7317       if ((cind < cdof) && (k == cdofs[cind])) {
7318         /* Insert check for returning constrained indices */
7319         indices[ind] = -(off + k + 1);
7320         ++cind;
7321       } else {
7322         indices[ind] = off + k - (islocal ? 0 : cind);
7323       }
7324     }
7325   }
7326   *loff += dof;
7327   PetscFunctionReturn(PETSC_SUCCESS);
7328 }
7329 
7330 /*
7331  DMPlexGetIndicesPointFields_Internal - gets section indices for a point in its canonical ordering.
7332 
7333  Input Parameters:
7334 + section - a section (global or local)
7335 - islocal - `PETSC_TRUE` if requesting local indices (i.e., section is local); `PETSC_FALSE` for global
7336 . point - point within section
7337 . off - The offset of this point in the (local or global) indexed space - should match islocal and (usually) the section
7338 . foffs - array of length numFields containing the offset in canonical point ordering (the location in indices) of each field
7339 . setBC - identify constrained (boundary condition) points via involution.
7340 . perms - perms[f][permsoff][:] is a permutation of dofs within each field
7341 . permsoff - offset
7342 - indperm - index permutation
7343 
7344  Output Parameter:
7345 . foffs - each entry is incremented by the number of (unconstrained if setBC=FALSE) dofs in that field
7346 . indices - array to hold indices (as defined by section) of each dof associated with point
7347 
7348  Notes:
7349  If section is local and setBC=true, there is no distinction between constrained and unconstrained dofs.
7350  If section is local and setBC=false, the indices for constrained points are the involution -(i+1) of their position
7351  in the local vector.
7352 
7353  If section is global and setBC=false, the indices for constrained points are negative (and their value is not
7354  significant).  It is invalid to call with a global section and setBC=true.
7355 
7356  Developer Note:
7357  The section is only used for field layout, so islocal is technically a statement about the offset (off).  At some point
7358  in the future, global sections may have fields set, in which case we could pass the global section and obtain the
7359  offset could be obtained from the section instead of passing it explicitly as we do now.
7360 
7361  Example:
7362  Suppose a point contains one field with three components, and for which the unconstrained indices are {10, 11, 12}.
7363  When the middle component is constrained, we get the array {10, -12, 12} for (islocal=TRUE, setBC=FALSE).
7364  Note that -12 is the involution of 11, so the user can involute negative indices to recover local indices.
7365  The global vector does not store constrained dofs, so when this function returns global indices, say {110, -112, 111}, the value of -112 is an arbitrary flag that should not be interpreted beyond its sign.
7366 
7367  Level: developer
7368 */
7369 PetscErrorCode DMPlexGetIndicesPointFields_Internal(PetscSection section, PetscBool islocal, PetscInt point, PetscInt off, PetscInt foffs[], PetscBool setBC, const PetscInt ***perms, PetscInt permsoff, const PetscInt indperm[], PetscInt indices[])
7370 {
7371   PetscInt numFields, foff, f;
7372 
7373   PetscFunctionBegin;
7374   PetscCheck(islocal || !setBC, PetscObjectComm((PetscObject)section), PETSC_ERR_ARG_INCOMP, "setBC incompatible with global indices; use a local section or disable setBC");
7375   PetscCall(PetscSectionGetNumFields(section, &numFields));
7376   for (f = 0, foff = 0; f < numFields; ++f) {
7377     PetscInt        fdof, cfdof;
7378     const PetscInt *fcdofs; /* The indices of the constrained dofs for field f on this point */
7379     PetscInt        cind = 0, b;
7380     const PetscInt *perm = (perms && perms[f]) ? perms[f][permsoff] : NULL;
7381 
7382     PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
7383     PetscCall(PetscSectionGetFieldConstraintDof(section, point, f, &cfdof));
7384     if (!cfdof || setBC) {
7385       for (b = 0; b < fdof; ++b) {
7386         const PetscInt preind = perm ? foffs[f] + perm[b] : foffs[f] + b;
7387         const PetscInt ind    = indperm ? indperm[preind] : preind;
7388 
7389         indices[ind] = off + foff + b;
7390       }
7391     } else {
7392       PetscCall(PetscSectionGetFieldConstraintIndices(section, point, f, &fcdofs));
7393       for (b = 0; b < fdof; ++b) {
7394         const PetscInt preind = perm ? foffs[f] + perm[b] : foffs[f] + b;
7395         const PetscInt ind    = indperm ? indperm[preind] : preind;
7396 
7397         if ((cind < cfdof) && (b == fcdofs[cind])) {
7398           indices[ind] = -(off + foff + b + 1);
7399           ++cind;
7400         } else {
7401           indices[ind] = off + foff + b - (islocal ? 0 : cind);
7402         }
7403       }
7404     }
7405     foff += (setBC || islocal ? fdof : (fdof - cfdof));
7406     foffs[f] += fdof;
7407   }
7408   PetscFunctionReturn(PETSC_SUCCESS);
7409 }
7410 
7411 /*
7412   This version believes the globalSection offsets for each field, rather than just the point offset
7413 
7414  . foffs - The offset into 'indices' for each field, since it is segregated by field
7415 
7416  Notes:
7417  The semantics of this function relate to that of setBC=FALSE in DMPlexGetIndicesPointFields_Internal.
7418  Since this function uses global indices, setBC=TRUE would be invalid, so no such argument exists.
7419 */
7420 static PetscErrorCode DMPlexGetIndicesPointFieldsSplit_Internal(PetscSection section, PetscSection globalSection, PetscInt point, PetscInt foffs[], const PetscInt ***perms, PetscInt permsoff, const PetscInt indperm[], PetscInt indices[])
7421 {
7422   PetscInt numFields, foff, f;
7423 
7424   PetscFunctionBegin;
7425   PetscCall(PetscSectionGetNumFields(section, &numFields));
7426   for (f = 0; f < numFields; ++f) {
7427     PetscInt        fdof, cfdof;
7428     const PetscInt *fcdofs; /* The indices of the constrained dofs for field f on this point */
7429     PetscInt        cind = 0, b;
7430     const PetscInt *perm = (perms && perms[f]) ? perms[f][permsoff] : NULL;
7431 
7432     PetscCall(PetscSectionGetFieldDof(section, point, f, &fdof));
7433     PetscCall(PetscSectionGetFieldConstraintDof(section, point, f, &cfdof));
7434     PetscCall(PetscSectionGetFieldOffset(globalSection, point, f, &foff));
7435     if (!cfdof) {
7436       for (b = 0; b < fdof; ++b) {
7437         const PetscInt preind = perm ? foffs[f] + perm[b] : foffs[f] + b;
7438         const PetscInt ind    = indperm ? indperm[preind] : preind;
7439 
7440         indices[ind] = foff + b;
7441       }
7442     } else {
7443       PetscCall(PetscSectionGetFieldConstraintIndices(section, point, f, &fcdofs));
7444       for (b = 0; b < fdof; ++b) {
7445         const PetscInt preind = perm ? foffs[f] + perm[b] : foffs[f] + b;
7446         const PetscInt ind    = indperm ? indperm[preind] : preind;
7447 
7448         if ((cind < cfdof) && (b == fcdofs[cind])) {
7449           indices[ind] = -(foff + b + 1);
7450           ++cind;
7451         } else {
7452           indices[ind] = foff + b - cind;
7453         }
7454       }
7455     }
7456     foffs[f] += fdof;
7457   }
7458   PetscFunctionReturn(PETSC_SUCCESS);
7459 }
7460 
7461 PetscErrorCode DMPlexAnchorsModifyMat(DM dm, PetscSection section, PetscInt numPoints, PetscInt numIndices, const PetscInt points[], const PetscInt ***perms, const PetscScalar values[], PetscInt *outNumPoints, PetscInt *outNumIndices, PetscInt *outPoints[], PetscScalar *outValues[], PetscInt offsets[], PetscBool multiplyLeft)
7462 {
7463   Mat             cMat;
7464   PetscSection    aSec, cSec;
7465   IS              aIS;
7466   PetscInt        aStart = -1, aEnd = -1;
7467   const PetscInt *anchors;
7468   PetscInt        numFields, f, p, q, newP = 0;
7469   PetscInt        newNumPoints = 0, newNumIndices = 0;
7470   PetscInt       *newPoints, *indices, *newIndices;
7471   PetscInt        maxAnchor, maxDof;
7472   PetscInt        newOffsets[32];
7473   PetscInt       *pointMatOffsets[32];
7474   PetscInt       *newPointOffsets[32];
7475   PetscScalar    *pointMat[32];
7476   PetscScalar    *newValues      = NULL, *tmpValues;
7477   PetscBool       anyConstrained = PETSC_FALSE;
7478 
7479   PetscFunctionBegin;
7480   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7481   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
7482   PetscCall(PetscSectionGetNumFields(section, &numFields));
7483 
7484   PetscCall(DMPlexGetAnchors(dm, &aSec, &aIS));
7485   /* if there are point-to-point constraints */
7486   if (aSec) {
7487     PetscCall(PetscArrayzero(newOffsets, 32));
7488     PetscCall(ISGetIndices(aIS, &anchors));
7489     PetscCall(PetscSectionGetChart(aSec, &aStart, &aEnd));
7490     /* figure out how many points are going to be in the new element matrix
7491      * (we allow double counting, because it's all just going to be summed
7492      * into the global matrix anyway) */
7493     for (p = 0; p < 2 * numPoints; p += 2) {
7494       PetscInt b    = points[p];
7495       PetscInt bDof = 0, bSecDof;
7496 
7497       PetscCall(PetscSectionGetDof(section, b, &bSecDof));
7498       if (!bSecDof) continue;
7499       if (b >= aStart && b < aEnd) PetscCall(PetscSectionGetDof(aSec, b, &bDof));
7500       if (bDof) {
7501         /* this point is constrained */
7502         /* it is going to be replaced by its anchors */
7503         PetscInt bOff, q;
7504 
7505         anyConstrained = PETSC_TRUE;
7506         newNumPoints += bDof;
7507         PetscCall(PetscSectionGetOffset(aSec, b, &bOff));
7508         for (q = 0; q < bDof; q++) {
7509           PetscInt a = anchors[bOff + q];
7510           PetscInt aDof;
7511 
7512           PetscCall(PetscSectionGetDof(section, a, &aDof));
7513           newNumIndices += aDof;
7514           for (f = 0; f < numFields; ++f) {
7515             PetscInt fDof;
7516 
7517             PetscCall(PetscSectionGetFieldDof(section, a, f, &fDof));
7518             newOffsets[f + 1] += fDof;
7519           }
7520         }
7521       } else {
7522         /* this point is not constrained */
7523         newNumPoints++;
7524         newNumIndices += bSecDof;
7525         for (f = 0; f < numFields; ++f) {
7526           PetscInt fDof;
7527 
7528           PetscCall(PetscSectionGetFieldDof(section, b, f, &fDof));
7529           newOffsets[f + 1] += fDof;
7530         }
7531       }
7532     }
7533   }
7534   if (!anyConstrained) {
7535     if (outNumPoints) *outNumPoints = 0;
7536     if (outNumIndices) *outNumIndices = 0;
7537     if (outPoints) *outPoints = NULL;
7538     if (outValues) *outValues = NULL;
7539     if (aSec) PetscCall(ISRestoreIndices(aIS, &anchors));
7540     PetscFunctionReturn(PETSC_SUCCESS);
7541   }
7542 
7543   if (outNumPoints) *outNumPoints = newNumPoints;
7544   if (outNumIndices) *outNumIndices = newNumIndices;
7545 
7546   for (f = 0; f < numFields; ++f) newOffsets[f + 1] += newOffsets[f];
7547 
7548   if (!outPoints && !outValues) {
7549     if (offsets) {
7550       for (f = 0; f <= numFields; f++) offsets[f] = newOffsets[f];
7551     }
7552     if (aSec) PetscCall(ISRestoreIndices(aIS, &anchors));
7553     PetscFunctionReturn(PETSC_SUCCESS);
7554   }
7555 
7556   PetscCheck(!numFields || newOffsets[numFields] == newNumIndices, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, newOffsets[numFields], newNumIndices);
7557 
7558   PetscCall(DMGetDefaultConstraints(dm, &cSec, &cMat, NULL));
7559 
7560   /* workspaces */
7561   if (numFields) {
7562     for (f = 0; f < numFields; f++) {
7563       PetscCall(DMGetWorkArray(dm, numPoints + 1, MPIU_INT, &pointMatOffsets[f]));
7564       PetscCall(DMGetWorkArray(dm, numPoints + 1, MPIU_INT, &newPointOffsets[f]));
7565     }
7566   } else {
7567     PetscCall(DMGetWorkArray(dm, numPoints + 1, MPIU_INT, &pointMatOffsets[0]));
7568     PetscCall(DMGetWorkArray(dm, numPoints, MPIU_INT, &newPointOffsets[0]));
7569   }
7570 
7571   /* get workspaces for the point-to-point matrices */
7572   if (numFields) {
7573     PetscInt totalOffset, totalMatOffset;
7574 
7575     for (p = 0; p < numPoints; p++) {
7576       PetscInt b    = points[2 * p];
7577       PetscInt bDof = 0, bSecDof;
7578 
7579       PetscCall(PetscSectionGetDof(section, b, &bSecDof));
7580       if (!bSecDof) {
7581         for (f = 0; f < numFields; f++) {
7582           newPointOffsets[f][p + 1] = 0;
7583           pointMatOffsets[f][p + 1] = 0;
7584         }
7585         continue;
7586       }
7587       if (b >= aStart && b < aEnd) PetscCall(PetscSectionGetDof(aSec, b, &bDof));
7588       if (bDof) {
7589         for (f = 0; f < numFields; f++) {
7590           PetscInt fDof, q, bOff, allFDof = 0;
7591 
7592           PetscCall(PetscSectionGetFieldDof(section, b, f, &fDof));
7593           PetscCall(PetscSectionGetOffset(aSec, b, &bOff));
7594           for (q = 0; q < bDof; q++) {
7595             PetscInt a = anchors[bOff + q];
7596             PetscInt aFDof;
7597 
7598             PetscCall(PetscSectionGetFieldDof(section, a, f, &aFDof));
7599             allFDof += aFDof;
7600           }
7601           newPointOffsets[f][p + 1] = allFDof;
7602           pointMatOffsets[f][p + 1] = fDof * allFDof;
7603         }
7604       } else {
7605         for (f = 0; f < numFields; f++) {
7606           PetscInt fDof;
7607 
7608           PetscCall(PetscSectionGetFieldDof(section, b, f, &fDof));
7609           newPointOffsets[f][p + 1] = fDof;
7610           pointMatOffsets[f][p + 1] = 0;
7611         }
7612       }
7613     }
7614     for (f = 0, totalOffset = 0, totalMatOffset = 0; f < numFields; f++) {
7615       newPointOffsets[f][0] = totalOffset;
7616       pointMatOffsets[f][0] = totalMatOffset;
7617       for (p = 0; p < numPoints; p++) {
7618         newPointOffsets[f][p + 1] += newPointOffsets[f][p];
7619         pointMatOffsets[f][p + 1] += pointMatOffsets[f][p];
7620       }
7621       totalOffset    = newPointOffsets[f][numPoints];
7622       totalMatOffset = pointMatOffsets[f][numPoints];
7623       PetscCall(DMGetWorkArray(dm, pointMatOffsets[f][numPoints], MPIU_SCALAR, &pointMat[f]));
7624     }
7625   } else {
7626     for (p = 0; p < numPoints; p++) {
7627       PetscInt b    = points[2 * p];
7628       PetscInt bDof = 0, bSecDof;
7629 
7630       PetscCall(PetscSectionGetDof(section, b, &bSecDof));
7631       if (!bSecDof) {
7632         newPointOffsets[0][p + 1] = 0;
7633         pointMatOffsets[0][p + 1] = 0;
7634         continue;
7635       }
7636       if (b >= aStart && b < aEnd) PetscCall(PetscSectionGetDof(aSec, b, &bDof));
7637       if (bDof) {
7638         PetscInt bOff, q, allDof = 0;
7639 
7640         PetscCall(PetscSectionGetOffset(aSec, b, &bOff));
7641         for (q = 0; q < bDof; q++) {
7642           PetscInt a = anchors[bOff + q], aDof;
7643 
7644           PetscCall(PetscSectionGetDof(section, a, &aDof));
7645           allDof += aDof;
7646         }
7647         newPointOffsets[0][p + 1] = allDof;
7648         pointMatOffsets[0][p + 1] = bSecDof * allDof;
7649       } else {
7650         newPointOffsets[0][p + 1] = bSecDof;
7651         pointMatOffsets[0][p + 1] = 0;
7652       }
7653     }
7654     newPointOffsets[0][0] = 0;
7655     pointMatOffsets[0][0] = 0;
7656     for (p = 0; p < numPoints; p++) {
7657       newPointOffsets[0][p + 1] += newPointOffsets[0][p];
7658       pointMatOffsets[0][p + 1] += pointMatOffsets[0][p];
7659     }
7660     PetscCall(DMGetWorkArray(dm, pointMatOffsets[0][numPoints], MPIU_SCALAR, &pointMat[0]));
7661   }
7662 
7663   /* output arrays */
7664   PetscCall(DMGetWorkArray(dm, 2 * newNumPoints, MPIU_INT, &newPoints));
7665 
7666   /* get the point-to-point matrices; construct newPoints */
7667   PetscCall(PetscSectionGetMaxDof(aSec, &maxAnchor));
7668   PetscCall(PetscSectionGetMaxDof(section, &maxDof));
7669   PetscCall(DMGetWorkArray(dm, maxDof, MPIU_INT, &indices));
7670   PetscCall(DMGetWorkArray(dm, maxAnchor * maxDof, MPIU_INT, &newIndices));
7671   if (numFields) {
7672     for (p = 0, newP = 0; p < numPoints; p++) {
7673       PetscInt b    = points[2 * p];
7674       PetscInt o    = points[2 * p + 1];
7675       PetscInt bDof = 0, bSecDof;
7676 
7677       PetscCall(PetscSectionGetDof(section, b, &bSecDof));
7678       if (!bSecDof) continue;
7679       if (b >= aStart && b < aEnd) PetscCall(PetscSectionGetDof(aSec, b, &bDof));
7680       if (bDof) {
7681         PetscInt fStart[32], fEnd[32], fAnchorStart[32], fAnchorEnd[32], bOff, q;
7682 
7683         fStart[0] = 0;
7684         fEnd[0]   = 0;
7685         for (f = 0; f < numFields; f++) {
7686           PetscInt fDof;
7687 
7688           PetscCall(PetscSectionGetFieldDof(cSec, b, f, &fDof));
7689           fStart[f + 1] = fStart[f] + fDof;
7690           fEnd[f + 1]   = fStart[f + 1];
7691         }
7692         PetscCall(PetscSectionGetOffset(cSec, b, &bOff));
7693         PetscCall(DMPlexGetIndicesPointFields_Internal(cSec, PETSC_TRUE, b, bOff, fEnd, PETSC_TRUE, perms, p, NULL, indices));
7694 
7695         fAnchorStart[0] = 0;
7696         fAnchorEnd[0]   = 0;
7697         for (f = 0; f < numFields; f++) {
7698           PetscInt fDof = newPointOffsets[f][p + 1] - newPointOffsets[f][p];
7699 
7700           fAnchorStart[f + 1] = fAnchorStart[f] + fDof;
7701           fAnchorEnd[f + 1]   = fAnchorStart[f + 1];
7702         }
7703         PetscCall(PetscSectionGetOffset(aSec, b, &bOff));
7704         for (q = 0; q < bDof; q++) {
7705           PetscInt a = anchors[bOff + q], aOff;
7706 
7707           /* we take the orientation of ap into account in the order that we constructed the indices above: the newly added points have no orientation */
7708           newPoints[2 * (newP + q)]     = a;
7709           newPoints[2 * (newP + q) + 1] = 0;
7710           PetscCall(PetscSectionGetOffset(section, a, &aOff));
7711           PetscCall(DMPlexGetIndicesPointFields_Internal(section, PETSC_TRUE, a, aOff, fAnchorEnd, PETSC_TRUE, NULL, -1, NULL, newIndices));
7712         }
7713         newP += bDof;
7714 
7715         if (outValues) {
7716           /* get the point-to-point submatrix */
7717           for (f = 0; f < numFields; f++) PetscCall(MatGetValues(cMat, fEnd[f] - fStart[f], indices + fStart[f], fAnchorEnd[f] - fAnchorStart[f], newIndices + fAnchorStart[f], pointMat[f] + pointMatOffsets[f][p]));
7718         }
7719       } else {
7720         newPoints[2 * newP]     = b;
7721         newPoints[2 * newP + 1] = o;
7722         newP++;
7723       }
7724     }
7725   } else {
7726     for (p = 0; p < numPoints; p++) {
7727       PetscInt b    = points[2 * p];
7728       PetscInt o    = points[2 * p + 1];
7729       PetscInt bDof = 0, bSecDof;
7730 
7731       PetscCall(PetscSectionGetDof(section, b, &bSecDof));
7732       if (!bSecDof) continue;
7733       if (b >= aStart && b < aEnd) PetscCall(PetscSectionGetDof(aSec, b, &bDof));
7734       if (bDof) {
7735         PetscInt bEnd = 0, bAnchorEnd = 0, bOff;
7736 
7737         PetscCall(PetscSectionGetOffset(cSec, b, &bOff));
7738         PetscCall(DMPlexGetIndicesPoint_Internal(cSec, PETSC_TRUE, b, bOff, &bEnd, PETSC_TRUE, (perms && perms[0]) ? perms[0][p] : NULL, NULL, indices));
7739 
7740         PetscCall(PetscSectionGetOffset(aSec, b, &bOff));
7741         for (q = 0; q < bDof; q++) {
7742           PetscInt a = anchors[bOff + q], aOff;
7743 
7744           /* we take the orientation of ap into account in the order that we constructed the indices above: the newly added points have no orientation */
7745 
7746           newPoints[2 * (newP + q)]     = a;
7747           newPoints[2 * (newP + q) + 1] = 0;
7748           PetscCall(PetscSectionGetOffset(section, a, &aOff));
7749           PetscCall(DMPlexGetIndicesPoint_Internal(section, PETSC_TRUE, a, aOff, &bAnchorEnd, PETSC_TRUE, NULL, NULL, newIndices));
7750         }
7751         newP += bDof;
7752 
7753         /* get the point-to-point submatrix */
7754         if (outValues) PetscCall(MatGetValues(cMat, bEnd, indices, bAnchorEnd, newIndices, pointMat[0] + pointMatOffsets[0][p]));
7755       } else {
7756         newPoints[2 * newP]     = b;
7757         newPoints[2 * newP + 1] = o;
7758         newP++;
7759       }
7760     }
7761   }
7762 
7763   if (outValues) {
7764     PetscCall(DMGetWorkArray(dm, newNumIndices * numIndices, MPIU_SCALAR, &tmpValues));
7765     PetscCall(PetscArrayzero(tmpValues, newNumIndices * numIndices));
7766     /* multiply constraints on the right */
7767     if (numFields) {
7768       for (f = 0; f < numFields; f++) {
7769         PetscInt oldOff = offsets[f];
7770 
7771         for (p = 0; p < numPoints; p++) {
7772           PetscInt cStart = newPointOffsets[f][p];
7773           PetscInt b      = points[2 * p];
7774           PetscInt c, r, k;
7775           PetscInt dof;
7776 
7777           PetscCall(PetscSectionGetFieldDof(section, b, f, &dof));
7778           if (!dof) continue;
7779           if (pointMatOffsets[f][p] < pointMatOffsets[f][p + 1]) {
7780             PetscInt           nCols = newPointOffsets[f][p + 1] - cStart;
7781             const PetscScalar *mat   = pointMat[f] + pointMatOffsets[f][p];
7782 
7783             for (r = 0; r < numIndices; r++) {
7784               for (c = 0; c < nCols; c++) {
7785                 for (k = 0; k < dof; k++) tmpValues[r * newNumIndices + cStart + c] += values[r * numIndices + oldOff + k] * mat[k * nCols + c];
7786               }
7787             }
7788           } else {
7789             /* copy this column as is */
7790             for (r = 0; r < numIndices; r++) {
7791               for (c = 0; c < dof; c++) tmpValues[r * newNumIndices + cStart + c] = values[r * numIndices + oldOff + c];
7792             }
7793           }
7794           oldOff += dof;
7795         }
7796       }
7797     } else {
7798       PetscInt oldOff = 0;
7799       for (p = 0; p < numPoints; p++) {
7800         PetscInt cStart = newPointOffsets[0][p];
7801         PetscInt b      = points[2 * p];
7802         PetscInt c, r, k;
7803         PetscInt dof;
7804 
7805         PetscCall(PetscSectionGetDof(section, b, &dof));
7806         if (!dof) continue;
7807         if (pointMatOffsets[0][p] < pointMatOffsets[0][p + 1]) {
7808           PetscInt           nCols = newPointOffsets[0][p + 1] - cStart;
7809           const PetscScalar *mat   = pointMat[0] + pointMatOffsets[0][p];
7810 
7811           for (r = 0; r < numIndices; r++) {
7812             for (c = 0; c < nCols; c++) {
7813               for (k = 0; k < dof; k++) tmpValues[r * newNumIndices + cStart + c] += mat[k * nCols + c] * values[r * numIndices + oldOff + k];
7814             }
7815           }
7816         } else {
7817           /* copy this column as is */
7818           for (r = 0; r < numIndices; r++) {
7819             for (c = 0; c < dof; c++) tmpValues[r * newNumIndices + cStart + c] = values[r * numIndices + oldOff + c];
7820           }
7821         }
7822         oldOff += dof;
7823       }
7824     }
7825 
7826     if (multiplyLeft) {
7827       PetscCall(DMGetWorkArray(dm, newNumIndices * newNumIndices, MPIU_SCALAR, &newValues));
7828       PetscCall(PetscArrayzero(newValues, newNumIndices * newNumIndices));
7829       /* multiply constraints transpose on the left */
7830       if (numFields) {
7831         for (f = 0; f < numFields; f++) {
7832           PetscInt oldOff = offsets[f];
7833 
7834           for (p = 0; p < numPoints; p++) {
7835             PetscInt rStart = newPointOffsets[f][p];
7836             PetscInt b      = points[2 * p];
7837             PetscInt c, r, k;
7838             PetscInt dof;
7839 
7840             PetscCall(PetscSectionGetFieldDof(section, b, f, &dof));
7841             if (pointMatOffsets[f][p] < pointMatOffsets[f][p + 1]) {
7842               PetscInt                          nRows = newPointOffsets[f][p + 1] - rStart;
7843               const PetscScalar *PETSC_RESTRICT mat   = pointMat[f] + pointMatOffsets[f][p];
7844 
7845               for (r = 0; r < nRows; r++) {
7846                 for (c = 0; c < newNumIndices; c++) {
7847                   for (k = 0; k < dof; k++) newValues[(rStart + r) * newNumIndices + c] += mat[k * nRows + r] * tmpValues[(oldOff + k) * newNumIndices + c];
7848                 }
7849               }
7850             } else {
7851               /* copy this row as is */
7852               for (r = 0; r < dof; r++) {
7853                 for (c = 0; c < newNumIndices; c++) newValues[(rStart + r) * newNumIndices + c] = tmpValues[(oldOff + r) * newNumIndices + c];
7854               }
7855             }
7856             oldOff += dof;
7857           }
7858         }
7859       } else {
7860         PetscInt oldOff = 0;
7861 
7862         for (p = 0; p < numPoints; p++) {
7863           PetscInt rStart = newPointOffsets[0][p];
7864           PetscInt b      = points[2 * p];
7865           PetscInt c, r, k;
7866           PetscInt dof;
7867 
7868           PetscCall(PetscSectionGetDof(section, b, &dof));
7869           if (pointMatOffsets[0][p] < pointMatOffsets[0][p + 1]) {
7870             PetscInt                          nRows = newPointOffsets[0][p + 1] - rStart;
7871             const PetscScalar *PETSC_RESTRICT mat   = pointMat[0] + pointMatOffsets[0][p];
7872 
7873             for (r = 0; r < nRows; r++) {
7874               for (c = 0; c < newNumIndices; c++) {
7875                 for (k = 0; k < dof; k++) newValues[(rStart + r) * newNumIndices + c] += mat[k * nRows + r] * tmpValues[(oldOff + k) * newNumIndices + c];
7876               }
7877             }
7878           } else {
7879             /* copy this row as is */
7880             for (r = 0; r < dof; r++) {
7881               for (c = 0; c < newNumIndices; c++) newValues[(rStart + r) * newNumIndices + c] = tmpValues[(oldOff + r) * newNumIndices + c];
7882             }
7883           }
7884           oldOff += dof;
7885         }
7886       }
7887 
7888       PetscCall(DMRestoreWorkArray(dm, newNumIndices * numIndices, MPIU_SCALAR, &tmpValues));
7889     } else {
7890       newValues = tmpValues;
7891     }
7892   }
7893 
7894   /* clean up */
7895   PetscCall(DMRestoreWorkArray(dm, maxDof, MPIU_INT, &indices));
7896   PetscCall(DMRestoreWorkArray(dm, maxAnchor * maxDof, MPIU_INT, &newIndices));
7897 
7898   if (numFields) {
7899     for (f = 0; f < numFields; f++) {
7900       PetscCall(DMRestoreWorkArray(dm, pointMatOffsets[f][numPoints], MPIU_SCALAR, &pointMat[f]));
7901       PetscCall(DMRestoreWorkArray(dm, numPoints + 1, MPIU_INT, &pointMatOffsets[f]));
7902       PetscCall(DMRestoreWorkArray(dm, numPoints + 1, MPIU_INT, &newPointOffsets[f]));
7903     }
7904   } else {
7905     PetscCall(DMRestoreWorkArray(dm, pointMatOffsets[0][numPoints], MPIU_SCALAR, &pointMat[0]));
7906     PetscCall(DMRestoreWorkArray(dm, numPoints + 1, MPIU_INT, &pointMatOffsets[0]));
7907     PetscCall(DMRestoreWorkArray(dm, numPoints + 1, MPIU_INT, &newPointOffsets[0]));
7908   }
7909   PetscCall(ISRestoreIndices(aIS, &anchors));
7910 
7911   /* output */
7912   if (outPoints) {
7913     *outPoints = newPoints;
7914   } else {
7915     PetscCall(DMRestoreWorkArray(dm, 2 * newNumPoints, MPIU_INT, &newPoints));
7916   }
7917   if (outValues) *outValues = newValues;
7918   for (f = 0; f <= numFields; f++) offsets[f] = newOffsets[f];
7919   PetscFunctionReturn(PETSC_SUCCESS);
7920 }
7921 
7922 /*@C
7923   DMPlexGetClosureIndices - Gets the global dof indices associated with the closure of the given point within the provided sections.
7924 
7925   Not collective
7926 
7927   Input Parameters:
7928 + dm         - The `DM`
7929 . section    - The `PetscSection` describing the points (a local section)
7930 . idxSection - The `PetscSection` from which to obtain indices (may be local or global)
7931 . point      - The point defining the closure
7932 - useClPerm  - Use the closure point permutation if available
7933 
7934   Output Parameters:
7935 + numIndices - The number of dof indices in the closure of point with the input sections
7936 . indices    - The dof indices
7937 . outOffsets - Array to write the field offsets into, or `NULL`
7938 - values     - The input values, which may be modified if sign flips are induced by the point symmetries, or `NULL`
7939 
7940   Level: advanced
7941 
7942   Notes:
7943   Must call `DMPlexRestoreClosureIndices()` to free allocated memory
7944 
7945   If `idxSection` is global, any constrained dofs (see `DMAddBoundary()`, for example) will get negative indices.  The value
7946   of those indices is not significant.  If `idxSection` is local, the constrained dofs will yield the involution -(idx+1)
7947   of their index in a local vector.  A caller who does not wish to distinguish those points may recover the nonnegative
7948   indices via involution, -(-(idx+1)+1)==idx.  Local indices are provided when `idxSection` == section, otherwise global
7949   indices (with the above semantics) are implied.
7950 
7951 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexRestoreClosureIndices()`, `DMPlexVecGetClosure()`, `DMPlexMatSetClosure()`, `DMGetLocalSection()`,
7952           `PetscSection`, `DMGetGlobalSection()`
7953 @*/
7954 PetscErrorCode DMPlexGetClosureIndices(DM dm, PetscSection section, PetscSection idxSection, PetscInt point, PetscBool useClPerm, PetscInt *numIndices, PetscInt *indices[], PetscInt outOffsets[], PetscScalar *values[])
7955 {
7956   /* Closure ordering */
7957   PetscSection    clSection;
7958   IS              clPoints;
7959   const PetscInt *clp;
7960   PetscInt       *points;
7961   const PetscInt *clperm = NULL;
7962   /* Dof permutation and sign flips */
7963   const PetscInt    **perms[32] = {NULL};
7964   const PetscScalar **flips[32] = {NULL};
7965   PetscScalar        *valCopy   = NULL;
7966   /* Hanging node constraints */
7967   PetscInt    *pointsC = NULL;
7968   PetscScalar *valuesC = NULL;
7969   PetscInt     NclC, NiC;
7970 
7971   PetscInt *idx;
7972   PetscInt  Nf, Ncl, Ni = 0, offsets[32], p, f;
7973   PetscBool isLocal = (section == idxSection) ? PETSC_TRUE : PETSC_FALSE;
7974 
7975   PetscFunctionBeginHot;
7976   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7977   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
7978   PetscValidHeaderSpecific(idxSection, PETSC_SECTION_CLASSID, 3);
7979   if (numIndices) PetscAssertPointer(numIndices, 6);
7980   if (indices) PetscAssertPointer(indices, 7);
7981   if (outOffsets) PetscAssertPointer(outOffsets, 8);
7982   if (values) PetscAssertPointer(values, 9);
7983   PetscCall(PetscSectionGetNumFields(section, &Nf));
7984   PetscCheck(Nf <= 31, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Number of fields %" PetscInt_FMT " limited to 31", Nf);
7985   PetscCall(PetscArrayzero(offsets, 32));
7986   /* 1) Get points in closure */
7987   PetscCall(DMPlexGetCompressedClosure(dm, section, point, 0, &Ncl, &points, &clSection, &clPoints, &clp));
7988   if (useClPerm) {
7989     PetscInt depth, clsize;
7990     PetscCall(DMPlexGetPointDepth(dm, point, &depth));
7991     for (clsize = 0, p = 0; p < Ncl; p++) {
7992       PetscInt dof;
7993       PetscCall(PetscSectionGetDof(section, points[2 * p], &dof));
7994       clsize += dof;
7995     }
7996     PetscCall(PetscSectionGetClosureInversePermutation_Internal(section, (PetscObject)dm, depth, clsize, &clperm));
7997   }
7998   /* 2) Get number of indices on these points and field offsets from section */
7999   for (p = 0; p < Ncl * 2; p += 2) {
8000     PetscInt dof, fdof;
8001 
8002     PetscCall(PetscSectionGetDof(section, points[p], &dof));
8003     for (f = 0; f < Nf; ++f) {
8004       PetscCall(PetscSectionGetFieldDof(section, points[p], f, &fdof));
8005       offsets[f + 1] += fdof;
8006     }
8007     Ni += dof;
8008   }
8009   for (f = 1; f < Nf; ++f) offsets[f + 1] += offsets[f];
8010   PetscCheck(!Nf || offsets[Nf] == Ni, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, offsets[Nf], Ni);
8011   /* 3) Get symmetries and sign flips. Apply sign flips to values if passed in (only works for square values matrix) */
8012   for (f = 0; f < PetscMax(1, Nf); ++f) {
8013     if (Nf) PetscCall(PetscSectionGetFieldPointSyms(section, f, Ncl, points, &perms[f], &flips[f]));
8014     else PetscCall(PetscSectionGetPointSyms(section, Ncl, points, &perms[f], &flips[f]));
8015     /* may need to apply sign changes to the element matrix */
8016     if (values && flips[f]) {
8017       PetscInt foffset = offsets[f];
8018 
8019       for (p = 0; p < Ncl; ++p) {
8020         PetscInt           pnt  = points[2 * p], fdof;
8021         const PetscScalar *flip = flips[f] ? flips[f][p] : NULL;
8022 
8023         if (!Nf) PetscCall(PetscSectionGetDof(section, pnt, &fdof));
8024         else PetscCall(PetscSectionGetFieldDof(section, pnt, f, &fdof));
8025         if (flip) {
8026           PetscInt i, j, k;
8027 
8028           if (!valCopy) {
8029             PetscCall(DMGetWorkArray(dm, Ni * Ni, MPIU_SCALAR, &valCopy));
8030             for (j = 0; j < Ni * Ni; ++j) valCopy[j] = (*values)[j];
8031             *values = valCopy;
8032           }
8033           for (i = 0; i < fdof; ++i) {
8034             PetscScalar fval = flip[i];
8035 
8036             for (k = 0; k < Ni; ++k) {
8037               valCopy[Ni * (foffset + i) + k] *= fval;
8038               valCopy[Ni * k + (foffset + i)] *= fval;
8039             }
8040           }
8041         }
8042         foffset += fdof;
8043       }
8044     }
8045   }
8046   /* 4) Apply hanging node constraints. Get new symmetries and replace all storage with constrained storage */
8047   PetscCall(DMPlexAnchorsModifyMat(dm, section, Ncl, Ni, points, perms, values ? *values : NULL, &NclC, &NiC, &pointsC, values ? &valuesC : NULL, offsets, PETSC_TRUE));
8048   if (NclC) {
8049     if (valCopy) PetscCall(DMRestoreWorkArray(dm, Ni * Ni, MPIU_SCALAR, &valCopy));
8050     for (f = 0; f < PetscMax(1, Nf); ++f) {
8051       if (Nf) PetscCall(PetscSectionRestoreFieldPointSyms(section, f, Ncl, points, &perms[f], &flips[f]));
8052       else PetscCall(PetscSectionRestorePointSyms(section, Ncl, points, &perms[f], &flips[f]));
8053     }
8054     for (f = 0; f < PetscMax(1, Nf); ++f) {
8055       if (Nf) PetscCall(PetscSectionGetFieldPointSyms(section, f, NclC, pointsC, &perms[f], &flips[f]));
8056       else PetscCall(PetscSectionGetPointSyms(section, NclC, pointsC, &perms[f], &flips[f]));
8057     }
8058     PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &Ncl, &points, &clSection, &clPoints, &clp));
8059     Ncl    = NclC;
8060     Ni     = NiC;
8061     points = pointsC;
8062     if (values) *values = valuesC;
8063   }
8064   /* 5) Calculate indices */
8065   PetscCall(DMGetWorkArray(dm, Ni, MPIU_INT, &idx));
8066   if (Nf) {
8067     PetscInt  idxOff;
8068     PetscBool useFieldOffsets;
8069 
8070     if (outOffsets) {
8071       for (f = 0; f <= Nf; f++) outOffsets[f] = offsets[f];
8072     }
8073     PetscCall(PetscSectionGetUseFieldOffsets(idxSection, &useFieldOffsets));
8074     if (useFieldOffsets) {
8075       for (p = 0; p < Ncl; ++p) {
8076         const PetscInt pnt = points[p * 2];
8077 
8078         PetscCall(DMPlexGetIndicesPointFieldsSplit_Internal(section, idxSection, pnt, offsets, perms, p, clperm, idx));
8079       }
8080     } else {
8081       for (p = 0; p < Ncl; ++p) {
8082         const PetscInt pnt = points[p * 2];
8083 
8084         PetscCall(PetscSectionGetOffset(idxSection, pnt, &idxOff));
8085         /* Note that we pass a local section even though we're using global offsets.  This is because global sections do
8086          * not (at the time of this writing) have fields set. They probably should, in which case we would pass the
8087          * global section. */
8088         PetscCall(DMPlexGetIndicesPointFields_Internal(section, isLocal, pnt, idxOff < 0 ? -(idxOff + 1) : idxOff, offsets, PETSC_FALSE, perms, p, clperm, idx));
8089       }
8090     }
8091   } else {
8092     PetscInt off = 0, idxOff;
8093 
8094     for (p = 0; p < Ncl; ++p) {
8095       const PetscInt  pnt  = points[p * 2];
8096       const PetscInt *perm = perms[0] ? perms[0][p] : NULL;
8097 
8098       PetscCall(PetscSectionGetOffset(idxSection, pnt, &idxOff));
8099       /* Note that we pass a local section even though we're using global offsets.  This is because global sections do
8100        * not (at the time of this writing) have fields set. They probably should, in which case we would pass the global section. */
8101       PetscCall(DMPlexGetIndicesPoint_Internal(section, isLocal, pnt, idxOff < 0 ? -(idxOff + 1) : idxOff, &off, PETSC_FALSE, perm, clperm, idx));
8102     }
8103   }
8104   /* 6) Cleanup */
8105   for (f = 0; f < PetscMax(1, Nf); ++f) {
8106     if (Nf) PetscCall(PetscSectionRestoreFieldPointSyms(section, f, Ncl, points, &perms[f], &flips[f]));
8107     else PetscCall(PetscSectionRestorePointSyms(section, Ncl, points, &perms[f], &flips[f]));
8108   }
8109   if (NclC) {
8110     PetscCall(DMRestoreWorkArray(dm, NclC * 2, MPIU_INT, &pointsC));
8111   } else {
8112     PetscCall(DMPlexRestoreCompressedClosure(dm, section, point, &Ncl, &points, &clSection, &clPoints, &clp));
8113   }
8114 
8115   if (numIndices) *numIndices = Ni;
8116   if (indices) *indices = idx;
8117   PetscFunctionReturn(PETSC_SUCCESS);
8118 }
8119 
8120 /*@C
8121   DMPlexRestoreClosureIndices - Restores the global dof indices associated with the closure of the given point within the provided sections.
8122 
8123   Not collective
8124 
8125   Input Parameters:
8126 + dm         - The `DM`
8127 . section    - The `PetscSection` describing the points (a local section)
8128 . idxSection - The `PetscSection` from which to obtain indices (may be local or global)
8129 . point      - The point defining the closure
8130 - useClPerm  - Use the closure point permutation if available
8131 
8132   Output Parameters:
8133 + numIndices - The number of dof indices in the closure of point with the input sections
8134 . indices    - The dof indices
8135 . outOffsets - Array to write the field offsets into, or `NULL`
8136 - values     - The input values, which may be modified if sign flips are induced by the point symmetries, or `NULL`
8137 
8138   Level: advanced
8139 
8140   Notes:
8141   If values were modified, the user is responsible for calling `DMRestoreWorkArray`(dm, 0, `MPIU_SCALAR`, &values).
8142 
8143   If idxSection is global, any constrained dofs (see `DMAddBoundary()`, for example) will get negative indices.  The value
8144   of those indices is not significant.  If idxSection is local, the constrained dofs will yield the involution -(idx+1)
8145   of their index in a local vector.  A caller who does not wish to distinguish those points may recover the nonnegative
8146   indices via involution, -(-(idx+1)+1)==idx.  Local indices are provided when idxSection == section, otherwise global
8147   indices (with the above semantics) are implied.
8148 
8149 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetClosureIndices()`, `DMPlexVecGetClosure()`, `DMPlexMatSetClosure()`, `DMGetLocalSection()`, `DMGetGlobalSection()`
8150 @*/
8151 PetscErrorCode DMPlexRestoreClosureIndices(DM dm, PetscSection section, PetscSection idxSection, PetscInt point, PetscBool useClPerm, PetscInt *numIndices, PetscInt *indices[], PetscInt outOffsets[], PetscScalar *values[])
8152 {
8153   PetscFunctionBegin;
8154   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8155   PetscAssertPointer(indices, 7);
8156   PetscCall(DMRestoreWorkArray(dm, 0, MPIU_INT, indices));
8157   PetscFunctionReturn(PETSC_SUCCESS);
8158 }
8159 
8160 PetscErrorCode DMPlexMatSetClosure_Internal(DM dm, PetscSection section, PetscSection globalSection, PetscBool useClPerm, Mat A, PetscInt point, const PetscScalar values[], InsertMode mode)
8161 {
8162   DM_Plex           *mesh = (DM_Plex *)dm->data;
8163   PetscInt          *indices;
8164   PetscInt           numIndices;
8165   const PetscScalar *valuesOrig = values;
8166   PetscErrorCode     ierr;
8167 
8168   PetscFunctionBegin;
8169   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8170   if (!section) PetscCall(DMGetLocalSection(dm, &section));
8171   PetscValidHeaderSpecific(section, PETSC_SECTION_CLASSID, 2);
8172   if (!globalSection) PetscCall(DMGetGlobalSection(dm, &globalSection));
8173   PetscValidHeaderSpecific(globalSection, PETSC_SECTION_CLASSID, 3);
8174   PetscValidHeaderSpecific(A, MAT_CLASSID, 5);
8175 
8176   PetscCall(DMPlexGetClosureIndices(dm, section, globalSection, point, useClPerm, &numIndices, &indices, NULL, (PetscScalar **)&values));
8177 
8178   if (mesh->printSetValues) PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDOUT_SELF, A, point, numIndices, indices, 0, NULL, values));
8179   /* TODO: fix this code to not use error codes as handle-able exceptions! */
8180   ierr = MatSetValues(A, numIndices, indices, numIndices, indices, values, mode);
8181   if (ierr) {
8182     PetscMPIInt rank;
8183 
8184     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)A), &rank));
8185     PetscCall((*PetscErrorPrintf)("[%d]ERROR in DMPlexMatSetClosure\n", rank));
8186     PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDERR_SELF, A, point, numIndices, indices, 0, NULL, values));
8187     PetscCall(DMPlexRestoreClosureIndices(dm, section, globalSection, point, PETSC_TRUE, &numIndices, &indices, NULL, (PetscScalar **)&values));
8188     if (values != valuesOrig) PetscCall(DMRestoreWorkArray(dm, 0, MPIU_SCALAR, &values));
8189     SETERRQ(PetscObjectComm((PetscObject)dm), ierr, "Not possible to set matrix values");
8190   }
8191   if (mesh->printFEM > 1) {
8192     PetscInt i;
8193     PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Indices:"));
8194     for (i = 0; i < numIndices; ++i) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, indices[i]));
8195     PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
8196   }
8197 
8198   PetscCall(DMPlexRestoreClosureIndices(dm, section, globalSection, point, PETSC_TRUE, &numIndices, &indices, NULL, (PetscScalar **)&values));
8199   if (values != valuesOrig) PetscCall(DMRestoreWorkArray(dm, 0, MPIU_SCALAR, &values));
8200   PetscFunctionReturn(PETSC_SUCCESS);
8201 }
8202 
8203 /*@C
8204   DMPlexMatSetClosure - Set an array of the values on the closure of 'point'
8205 
8206   Not collective
8207 
8208   Input Parameters:
8209 + dm            - The `DM`
8210 . section       - The section describing the layout in `v`, or `NULL` to use the default section
8211 . globalSection - The section describing the layout in `v`, or `NULL` to use the default global section
8212 . A             - The matrix
8213 . point         - The point in the `DM`
8214 . values        - The array of values
8215 - mode          - The insert mode, where `INSERT_ALL_VALUES` and `ADD_ALL_VALUES` also overwrite boundary conditions
8216 
8217   Level: intermediate
8218 
8219 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexMatSetClosureGeneral()`, `DMPlexVecGetClosure()`, `DMPlexVecSetClosure()`
8220 @*/
8221 PetscErrorCode DMPlexMatSetClosure(DM dm, PetscSection section, PetscSection globalSection, Mat A, PetscInt point, const PetscScalar values[], InsertMode mode)
8222 {
8223   PetscFunctionBegin;
8224   PetscCall(DMPlexMatSetClosure_Internal(dm, section, globalSection, PETSC_TRUE, A, point, values, mode));
8225   PetscFunctionReturn(PETSC_SUCCESS);
8226 }
8227 
8228 /*@C
8229   DMPlexMatSetClosureGeneral - Set an array of the values on the closure of 'point' using a different row and column section
8230 
8231   Not collective
8232 
8233   Input Parameters:
8234 + dmRow            - The `DM` for the row fields
8235 . sectionRow       - The section describing the layout, or `NULL` to use the default section in `dmRow`
8236 . useRowPerm       - The flag to use the closure permutation of the `dmRow` if available
8237 . globalSectionRow - The section describing the layout, or `NULL` to use the default global section in `dmRow`
8238 . dmCol            - The `DM` for the column fields
8239 . sectionCol       - The section describing the layout, or `NULL` to use the default section in `dmCol`
8240 . useColPerm       - The flag to use the closure permutation of the `dmCol` if available
8241 . globalSectionCol - The section describing the layout, or `NULL` to use the default global section in `dmCol`
8242 . A                - The matrix
8243 . point            - The point in the `DM`
8244 . values           - The array of values
8245 - mode             - The insert mode, where `INSERT_ALL_VALUES` and `ADD_ALL_VALUES` also overwrite boundary conditions
8246 
8247   Level: intermediate
8248 
8249 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexMatSetClosure()`, `DMPlexVecGetClosure()`, `DMPlexVecSetClosure()`
8250 @*/
8251 PetscErrorCode DMPlexMatSetClosureGeneral(DM dmRow, PetscSection sectionRow, PetscSection globalSectionRow, PetscBool useRowPerm, DM dmCol, PetscSection sectionCol, PetscSection globalSectionCol, PetscBool useColPerm, Mat A, PetscInt point, const PetscScalar values[], InsertMode mode)
8252 {
8253   DM_Plex           *mesh = (DM_Plex *)dmRow->data;
8254   PetscInt          *indicesRow, *indicesCol;
8255   PetscInt           numIndicesRow, numIndicesCol;
8256   const PetscScalar *valuesOrig = values;
8257   PetscErrorCode     ierr;
8258 
8259   PetscFunctionBegin;
8260   PetscValidHeaderSpecific(dmRow, DM_CLASSID, 1);
8261   if (!sectionRow) PetscCall(DMGetLocalSection(dmRow, &sectionRow));
8262   PetscValidHeaderSpecific(sectionRow, PETSC_SECTION_CLASSID, 2);
8263   if (!globalSectionRow) PetscCall(DMGetGlobalSection(dmRow, &globalSectionRow));
8264   PetscValidHeaderSpecific(globalSectionRow, PETSC_SECTION_CLASSID, 3);
8265   PetscValidHeaderSpecific(dmCol, DM_CLASSID, 5);
8266   if (!sectionCol) PetscCall(DMGetLocalSection(dmCol, &sectionCol));
8267   PetscValidHeaderSpecific(sectionCol, PETSC_SECTION_CLASSID, 6);
8268   if (!globalSectionCol) PetscCall(DMGetGlobalSection(dmCol, &globalSectionCol));
8269   PetscValidHeaderSpecific(globalSectionCol, PETSC_SECTION_CLASSID, 7);
8270   PetscValidHeaderSpecific(A, MAT_CLASSID, 9);
8271 
8272   PetscCall(DMPlexGetClosureIndices(dmRow, sectionRow, globalSectionRow, point, useRowPerm, &numIndicesRow, &indicesRow, NULL, (PetscScalar **)&values));
8273   PetscCall(DMPlexGetClosureIndices(dmCol, sectionCol, globalSectionCol, point, useColPerm, &numIndicesCol, &indicesCol, NULL, (PetscScalar **)&values));
8274 
8275   if (mesh->printSetValues) PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDOUT_SELF, A, point, numIndicesRow, indicesRow, numIndicesCol, indicesCol, values));
8276   /* TODO: fix this code to not use error codes as handle-able exceptions! */
8277   ierr = MatSetValues(A, numIndicesRow, indicesRow, numIndicesCol, indicesCol, values, mode);
8278   if (ierr) {
8279     PetscMPIInt rank;
8280 
8281     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)A), &rank));
8282     PetscCall((*PetscErrorPrintf)("[%d]ERROR in DMPlexMatSetClosure\n", rank));
8283     PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDERR_SELF, A, point, numIndicesRow, indicesRow, numIndicesCol, indicesCol, values));
8284     PetscCall(DMPlexRestoreClosureIndices(dmRow, sectionRow, globalSectionRow, point, PETSC_TRUE, &numIndicesRow, &indicesRow, NULL, (PetscScalar **)&values));
8285     PetscCall(DMPlexRestoreClosureIndices(dmCol, sectionCol, globalSectionCol, point, PETSC_TRUE, &numIndicesCol, &indicesRow, NULL, (PetscScalar **)&values));
8286     if (values != valuesOrig) PetscCall(DMRestoreWorkArray(dmRow, 0, MPIU_SCALAR, &values));
8287   }
8288 
8289   PetscCall(DMPlexRestoreClosureIndices(dmRow, sectionRow, globalSectionRow, point, useRowPerm, &numIndicesRow, &indicesRow, NULL, (PetscScalar **)&values));
8290   PetscCall(DMPlexRestoreClosureIndices(dmCol, sectionCol, globalSectionCol, point, useColPerm, &numIndicesCol, &indicesCol, NULL, (PetscScalar **)&values));
8291   if (values != valuesOrig) PetscCall(DMRestoreWorkArray(dmRow, 0, MPIU_SCALAR, &values));
8292   PetscFunctionReturn(PETSC_SUCCESS);
8293 }
8294 
8295 PetscErrorCode DMPlexMatSetClosureRefined(DM dmf, PetscSection fsection, PetscSection globalFSection, DM dmc, PetscSection csection, PetscSection globalCSection, Mat A, PetscInt point, const PetscScalar values[], InsertMode mode)
8296 {
8297   DM_Plex        *mesh    = (DM_Plex *)dmf->data;
8298   PetscInt       *fpoints = NULL, *ftotpoints = NULL;
8299   PetscInt       *cpoints = NULL;
8300   PetscInt       *findices, *cindices;
8301   const PetscInt *fclperm = NULL, *cclperm = NULL; /* Closure permutations cannot work here */
8302   PetscInt        foffsets[32], coffsets[32];
8303   DMPolytopeType  ct;
8304   PetscInt        numFields, numSubcells, maxFPoints, numFPoints, numCPoints, numFIndices, numCIndices, dof, off, globalOff, pStart, pEnd, p, q, r, s, f;
8305   PetscErrorCode  ierr;
8306 
8307   PetscFunctionBegin;
8308   PetscValidHeaderSpecific(dmf, DM_CLASSID, 1);
8309   PetscValidHeaderSpecific(dmc, DM_CLASSID, 4);
8310   if (!fsection) PetscCall(DMGetLocalSection(dmf, &fsection));
8311   PetscValidHeaderSpecific(fsection, PETSC_SECTION_CLASSID, 2);
8312   if (!csection) PetscCall(DMGetLocalSection(dmc, &csection));
8313   PetscValidHeaderSpecific(csection, PETSC_SECTION_CLASSID, 5);
8314   if (!globalFSection) PetscCall(DMGetGlobalSection(dmf, &globalFSection));
8315   PetscValidHeaderSpecific(globalFSection, PETSC_SECTION_CLASSID, 3);
8316   if (!globalCSection) PetscCall(DMGetGlobalSection(dmc, &globalCSection));
8317   PetscValidHeaderSpecific(globalCSection, PETSC_SECTION_CLASSID, 6);
8318   PetscValidHeaderSpecific(A, MAT_CLASSID, 7);
8319   PetscCall(PetscSectionGetNumFields(fsection, &numFields));
8320   PetscCheck(numFields <= 31, PetscObjectComm((PetscObject)dmf), PETSC_ERR_ARG_OUTOFRANGE, "Number of fields %" PetscInt_FMT " limited to 31", numFields);
8321   PetscCall(PetscArrayzero(foffsets, 32));
8322   PetscCall(PetscArrayzero(coffsets, 32));
8323   /* Column indices */
8324   PetscCall(DMPlexGetTransitiveClosure(dmc, point, PETSC_TRUE, &numCPoints, &cpoints));
8325   maxFPoints = numCPoints;
8326   /* Compress out points not in the section */
8327   /*   TODO: Squeeze out points with 0 dof as well */
8328   PetscCall(PetscSectionGetChart(csection, &pStart, &pEnd));
8329   for (p = 0, q = 0; p < numCPoints * 2; p += 2) {
8330     if ((cpoints[p] >= pStart) && (cpoints[p] < pEnd)) {
8331       cpoints[q * 2]     = cpoints[p];
8332       cpoints[q * 2 + 1] = cpoints[p + 1];
8333       ++q;
8334     }
8335   }
8336   numCPoints = q;
8337   for (p = 0, numCIndices = 0; p < numCPoints * 2; p += 2) {
8338     PetscInt fdof;
8339 
8340     PetscCall(PetscSectionGetDof(csection, cpoints[p], &dof));
8341     if (!dof) continue;
8342     for (f = 0; f < numFields; ++f) {
8343       PetscCall(PetscSectionGetFieldDof(csection, cpoints[p], f, &fdof));
8344       coffsets[f + 1] += fdof;
8345     }
8346     numCIndices += dof;
8347   }
8348   for (f = 1; f < numFields; ++f) coffsets[f + 1] += coffsets[f];
8349   /* Row indices */
8350   PetscCall(DMPlexGetCellType(dmc, point, &ct));
8351   {
8352     DMPlexTransform tr;
8353     DMPolytopeType *rct;
8354     PetscInt       *rsize, *rcone, *rornt, Nt;
8355 
8356     PetscCall(DMPlexTransformCreate(PETSC_COMM_SELF, &tr));
8357     PetscCall(DMPlexTransformSetType(tr, DMPLEXREFINEREGULAR));
8358     PetscCall(DMPlexTransformCellTransform(tr, ct, point, NULL, &Nt, &rct, &rsize, &rcone, &rornt));
8359     numSubcells = rsize[Nt - 1];
8360     PetscCall(DMPlexTransformDestroy(&tr));
8361   }
8362   PetscCall(DMGetWorkArray(dmf, maxFPoints * 2 * numSubcells, MPIU_INT, &ftotpoints));
8363   for (r = 0, q = 0; r < numSubcells; ++r) {
8364     /* TODO Map from coarse to fine cells */
8365     PetscCall(DMPlexGetTransitiveClosure(dmf, point * numSubcells + r, PETSC_TRUE, &numFPoints, &fpoints));
8366     /* Compress out points not in the section */
8367     PetscCall(PetscSectionGetChart(fsection, &pStart, &pEnd));
8368     for (p = 0; p < numFPoints * 2; p += 2) {
8369       if ((fpoints[p] >= pStart) && (fpoints[p] < pEnd)) {
8370         PetscCall(PetscSectionGetDof(fsection, fpoints[p], &dof));
8371         if (!dof) continue;
8372         for (s = 0; s < q; ++s)
8373           if (fpoints[p] == ftotpoints[s * 2]) break;
8374         if (s < q) continue;
8375         ftotpoints[q * 2]     = fpoints[p];
8376         ftotpoints[q * 2 + 1] = fpoints[p + 1];
8377         ++q;
8378       }
8379     }
8380     PetscCall(DMPlexRestoreTransitiveClosure(dmf, point, PETSC_TRUE, &numFPoints, &fpoints));
8381   }
8382   numFPoints = q;
8383   for (p = 0, numFIndices = 0; p < numFPoints * 2; p += 2) {
8384     PetscInt fdof;
8385 
8386     PetscCall(PetscSectionGetDof(fsection, ftotpoints[p], &dof));
8387     if (!dof) continue;
8388     for (f = 0; f < numFields; ++f) {
8389       PetscCall(PetscSectionGetFieldDof(fsection, ftotpoints[p], f, &fdof));
8390       foffsets[f + 1] += fdof;
8391     }
8392     numFIndices += dof;
8393   }
8394   for (f = 1; f < numFields; ++f) foffsets[f + 1] += foffsets[f];
8395 
8396   PetscCheck(!numFields || foffsets[numFields] == numFIndices, PetscObjectComm((PetscObject)dmf), PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, foffsets[numFields], numFIndices);
8397   PetscCheck(!numFields || coffsets[numFields] == numCIndices, PetscObjectComm((PetscObject)dmc), PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, coffsets[numFields], numCIndices);
8398   PetscCall(DMGetWorkArray(dmf, numFIndices, MPIU_INT, &findices));
8399   PetscCall(DMGetWorkArray(dmc, numCIndices, MPIU_INT, &cindices));
8400   if (numFields) {
8401     const PetscInt **permsF[32] = {NULL};
8402     const PetscInt **permsC[32] = {NULL};
8403 
8404     for (f = 0; f < numFields; f++) {
8405       PetscCall(PetscSectionGetFieldPointSyms(fsection, f, numFPoints, ftotpoints, &permsF[f], NULL));
8406       PetscCall(PetscSectionGetFieldPointSyms(csection, f, numCPoints, cpoints, &permsC[f], NULL));
8407     }
8408     for (p = 0; p < numFPoints; p++) {
8409       PetscCall(PetscSectionGetOffset(globalFSection, ftotpoints[2 * p], &globalOff));
8410       PetscCall(DMPlexGetIndicesPointFields_Internal(fsection, PETSC_FALSE, ftotpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, foffsets, PETSC_FALSE, permsF, p, fclperm, findices));
8411     }
8412     for (p = 0; p < numCPoints; p++) {
8413       PetscCall(PetscSectionGetOffset(globalCSection, cpoints[2 * p], &globalOff));
8414       PetscCall(DMPlexGetIndicesPointFields_Internal(csection, PETSC_FALSE, cpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, coffsets, PETSC_FALSE, permsC, p, cclperm, cindices));
8415     }
8416     for (f = 0; f < numFields; f++) {
8417       PetscCall(PetscSectionRestoreFieldPointSyms(fsection, f, numFPoints, ftotpoints, &permsF[f], NULL));
8418       PetscCall(PetscSectionRestoreFieldPointSyms(csection, f, numCPoints, cpoints, &permsC[f], NULL));
8419     }
8420   } else {
8421     const PetscInt **permsF = NULL;
8422     const PetscInt **permsC = NULL;
8423 
8424     PetscCall(PetscSectionGetPointSyms(fsection, numFPoints, ftotpoints, &permsF, NULL));
8425     PetscCall(PetscSectionGetPointSyms(csection, numCPoints, cpoints, &permsC, NULL));
8426     for (p = 0, off = 0; p < numFPoints; p++) {
8427       const PetscInt *perm = permsF ? permsF[p] : NULL;
8428 
8429       PetscCall(PetscSectionGetOffset(globalFSection, ftotpoints[2 * p], &globalOff));
8430       PetscCall(DMPlexGetIndicesPoint_Internal(fsection, PETSC_FALSE, ftotpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, &off, PETSC_FALSE, perm, fclperm, findices));
8431     }
8432     for (p = 0, off = 0; p < numCPoints; p++) {
8433       const PetscInt *perm = permsC ? permsC[p] : NULL;
8434 
8435       PetscCall(PetscSectionGetOffset(globalCSection, cpoints[2 * p], &globalOff));
8436       PetscCall(DMPlexGetIndicesPoint_Internal(csection, PETSC_FALSE, cpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, &off, PETSC_FALSE, perm, cclperm, cindices));
8437     }
8438     PetscCall(PetscSectionRestorePointSyms(fsection, numFPoints, ftotpoints, &permsF, NULL));
8439     PetscCall(PetscSectionRestorePointSyms(csection, numCPoints, cpoints, &permsC, NULL));
8440   }
8441   if (mesh->printSetValues) PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDOUT_SELF, A, point, numFIndices, findices, numCIndices, cindices, values));
8442   /* TODO: flips */
8443   /* TODO: fix this code to not use error codes as handle-able exceptions! */
8444   ierr = MatSetValues(A, numFIndices, findices, numCIndices, cindices, values, mode);
8445   if (ierr) {
8446     PetscMPIInt rank;
8447 
8448     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)A), &rank));
8449     PetscCall((*PetscErrorPrintf)("[%d]ERROR in DMPlexMatSetClosure\n", rank));
8450     PetscCall(DMPlexPrintMatSetValues(PETSC_VIEWER_STDERR_SELF, A, point, numFIndices, findices, numCIndices, cindices, values));
8451     PetscCall(DMRestoreWorkArray(dmf, numFIndices, MPIU_INT, &findices));
8452     PetscCall(DMRestoreWorkArray(dmc, numCIndices, MPIU_INT, &cindices));
8453   }
8454   PetscCall(DMRestoreWorkArray(dmf, numCPoints * 2 * 4, MPIU_INT, &ftotpoints));
8455   PetscCall(DMPlexRestoreTransitiveClosure(dmc, point, PETSC_TRUE, &numCPoints, &cpoints));
8456   PetscCall(DMRestoreWorkArray(dmf, numFIndices, MPIU_INT, &findices));
8457   PetscCall(DMRestoreWorkArray(dmc, numCIndices, MPIU_INT, &cindices));
8458   PetscFunctionReturn(PETSC_SUCCESS);
8459 }
8460 
8461 PetscErrorCode DMPlexMatGetClosureIndicesRefined(DM dmf, PetscSection fsection, PetscSection globalFSection, DM dmc, PetscSection csection, PetscSection globalCSection, PetscInt point, PetscInt cindices[], PetscInt findices[])
8462 {
8463   PetscInt       *fpoints = NULL, *ftotpoints = NULL;
8464   PetscInt       *cpoints      = NULL;
8465   PetscInt        foffsets[32] = {0}, coffsets[32] = {0};
8466   const PetscInt *fclperm = NULL, *cclperm = NULL; /* Closure permutations cannot work here */
8467   DMPolytopeType  ct;
8468   PetscInt        numFields, numSubcells, maxFPoints, numFPoints, numCPoints, numFIndices, numCIndices, dof, off, globalOff, pStart, pEnd, p, q, r, s, f;
8469 
8470   PetscFunctionBegin;
8471   PetscValidHeaderSpecific(dmf, DM_CLASSID, 1);
8472   PetscValidHeaderSpecific(dmc, DM_CLASSID, 4);
8473   if (!fsection) PetscCall(DMGetLocalSection(dmf, &fsection));
8474   PetscValidHeaderSpecific(fsection, PETSC_SECTION_CLASSID, 2);
8475   if (!csection) PetscCall(DMGetLocalSection(dmc, &csection));
8476   PetscValidHeaderSpecific(csection, PETSC_SECTION_CLASSID, 5);
8477   if (!globalFSection) PetscCall(DMGetGlobalSection(dmf, &globalFSection));
8478   PetscValidHeaderSpecific(globalFSection, PETSC_SECTION_CLASSID, 3);
8479   if (!globalCSection) PetscCall(DMGetGlobalSection(dmc, &globalCSection));
8480   PetscValidHeaderSpecific(globalCSection, PETSC_SECTION_CLASSID, 6);
8481   PetscCall(PetscSectionGetNumFields(fsection, &numFields));
8482   PetscCheck(numFields <= 31, PetscObjectComm((PetscObject)dmf), PETSC_ERR_ARG_OUTOFRANGE, "Number of fields %" PetscInt_FMT " limited to 31", numFields);
8483   /* Column indices */
8484   PetscCall(DMPlexGetTransitiveClosure(dmc, point, PETSC_TRUE, &numCPoints, &cpoints));
8485   maxFPoints = numCPoints;
8486   /* Compress out points not in the section */
8487   /*   TODO: Squeeze out points with 0 dof as well */
8488   PetscCall(PetscSectionGetChart(csection, &pStart, &pEnd));
8489   for (p = 0, q = 0; p < numCPoints * 2; p += 2) {
8490     if ((cpoints[p] >= pStart) && (cpoints[p] < pEnd)) {
8491       cpoints[q * 2]     = cpoints[p];
8492       cpoints[q * 2 + 1] = cpoints[p + 1];
8493       ++q;
8494     }
8495   }
8496   numCPoints = q;
8497   for (p = 0, numCIndices = 0; p < numCPoints * 2; p += 2) {
8498     PetscInt fdof;
8499 
8500     PetscCall(PetscSectionGetDof(csection, cpoints[p], &dof));
8501     if (!dof) continue;
8502     for (f = 0; f < numFields; ++f) {
8503       PetscCall(PetscSectionGetFieldDof(csection, cpoints[p], f, &fdof));
8504       coffsets[f + 1] += fdof;
8505     }
8506     numCIndices += dof;
8507   }
8508   for (f = 1; f < numFields; ++f) coffsets[f + 1] += coffsets[f];
8509   /* Row indices */
8510   PetscCall(DMPlexGetCellType(dmc, point, &ct));
8511   {
8512     DMPlexTransform tr;
8513     DMPolytopeType *rct;
8514     PetscInt       *rsize, *rcone, *rornt, Nt;
8515 
8516     PetscCall(DMPlexTransformCreate(PETSC_COMM_SELF, &tr));
8517     PetscCall(DMPlexTransformSetType(tr, DMPLEXREFINEREGULAR));
8518     PetscCall(DMPlexTransformCellTransform(tr, ct, point, NULL, &Nt, &rct, &rsize, &rcone, &rornt));
8519     numSubcells = rsize[Nt - 1];
8520     PetscCall(DMPlexTransformDestroy(&tr));
8521   }
8522   PetscCall(DMGetWorkArray(dmf, maxFPoints * 2 * numSubcells, MPIU_INT, &ftotpoints));
8523   for (r = 0, q = 0; r < numSubcells; ++r) {
8524     /* TODO Map from coarse to fine cells */
8525     PetscCall(DMPlexGetTransitiveClosure(dmf, point * numSubcells + r, PETSC_TRUE, &numFPoints, &fpoints));
8526     /* Compress out points not in the section */
8527     PetscCall(PetscSectionGetChart(fsection, &pStart, &pEnd));
8528     for (p = 0; p < numFPoints * 2; p += 2) {
8529       if ((fpoints[p] >= pStart) && (fpoints[p] < pEnd)) {
8530         PetscCall(PetscSectionGetDof(fsection, fpoints[p], &dof));
8531         if (!dof) continue;
8532         for (s = 0; s < q; ++s)
8533           if (fpoints[p] == ftotpoints[s * 2]) break;
8534         if (s < q) continue;
8535         ftotpoints[q * 2]     = fpoints[p];
8536         ftotpoints[q * 2 + 1] = fpoints[p + 1];
8537         ++q;
8538       }
8539     }
8540     PetscCall(DMPlexRestoreTransitiveClosure(dmf, point, PETSC_TRUE, &numFPoints, &fpoints));
8541   }
8542   numFPoints = q;
8543   for (p = 0, numFIndices = 0; p < numFPoints * 2; p += 2) {
8544     PetscInt fdof;
8545 
8546     PetscCall(PetscSectionGetDof(fsection, ftotpoints[p], &dof));
8547     if (!dof) continue;
8548     for (f = 0; f < numFields; ++f) {
8549       PetscCall(PetscSectionGetFieldDof(fsection, ftotpoints[p], f, &fdof));
8550       foffsets[f + 1] += fdof;
8551     }
8552     numFIndices += dof;
8553   }
8554   for (f = 1; f < numFields; ++f) foffsets[f + 1] += foffsets[f];
8555 
8556   PetscCheck(!numFields || foffsets[numFields] == numFIndices, PetscObjectComm((PetscObject)dmf), PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, foffsets[numFields], numFIndices);
8557   PetscCheck(!numFields || coffsets[numFields] == numCIndices, PetscObjectComm((PetscObject)dmc), PETSC_ERR_PLIB, "Invalid size for closure %" PetscInt_FMT " should be %" PetscInt_FMT, coffsets[numFields], numCIndices);
8558   if (numFields) {
8559     const PetscInt **permsF[32] = {NULL};
8560     const PetscInt **permsC[32] = {NULL};
8561 
8562     for (f = 0; f < numFields; f++) {
8563       PetscCall(PetscSectionGetFieldPointSyms(fsection, f, numFPoints, ftotpoints, &permsF[f], NULL));
8564       PetscCall(PetscSectionGetFieldPointSyms(csection, f, numCPoints, cpoints, &permsC[f], NULL));
8565     }
8566     for (p = 0; p < numFPoints; p++) {
8567       PetscCall(PetscSectionGetOffset(globalFSection, ftotpoints[2 * p], &globalOff));
8568       PetscCall(DMPlexGetIndicesPointFields_Internal(fsection, PETSC_FALSE, ftotpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, foffsets, PETSC_FALSE, permsF, p, fclperm, findices));
8569     }
8570     for (p = 0; p < numCPoints; p++) {
8571       PetscCall(PetscSectionGetOffset(globalCSection, cpoints[2 * p], &globalOff));
8572       PetscCall(DMPlexGetIndicesPointFields_Internal(csection, PETSC_FALSE, cpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, coffsets, PETSC_FALSE, permsC, p, cclperm, cindices));
8573     }
8574     for (f = 0; f < numFields; f++) {
8575       PetscCall(PetscSectionRestoreFieldPointSyms(fsection, f, numFPoints, ftotpoints, &permsF[f], NULL));
8576       PetscCall(PetscSectionRestoreFieldPointSyms(csection, f, numCPoints, cpoints, &permsC[f], NULL));
8577     }
8578   } else {
8579     const PetscInt **permsF = NULL;
8580     const PetscInt **permsC = NULL;
8581 
8582     PetscCall(PetscSectionGetPointSyms(fsection, numFPoints, ftotpoints, &permsF, NULL));
8583     PetscCall(PetscSectionGetPointSyms(csection, numCPoints, cpoints, &permsC, NULL));
8584     for (p = 0, off = 0; p < numFPoints; p++) {
8585       const PetscInt *perm = permsF ? permsF[p] : NULL;
8586 
8587       PetscCall(PetscSectionGetOffset(globalFSection, ftotpoints[2 * p], &globalOff));
8588       PetscCall(DMPlexGetIndicesPoint_Internal(fsection, PETSC_FALSE, ftotpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, &off, PETSC_FALSE, perm, fclperm, findices));
8589     }
8590     for (p = 0, off = 0; p < numCPoints; p++) {
8591       const PetscInt *perm = permsC ? permsC[p] : NULL;
8592 
8593       PetscCall(PetscSectionGetOffset(globalCSection, cpoints[2 * p], &globalOff));
8594       PetscCall(DMPlexGetIndicesPoint_Internal(csection, PETSC_FALSE, cpoints[2 * p], globalOff < 0 ? -(globalOff + 1) : globalOff, &off, PETSC_FALSE, perm, cclperm, cindices));
8595     }
8596     PetscCall(PetscSectionRestorePointSyms(fsection, numFPoints, ftotpoints, &permsF, NULL));
8597     PetscCall(PetscSectionRestorePointSyms(csection, numCPoints, cpoints, &permsC, NULL));
8598   }
8599   PetscCall(DMRestoreWorkArray(dmf, numCPoints * 2 * 4, MPIU_INT, &ftotpoints));
8600   PetscCall(DMPlexRestoreTransitiveClosure(dmc, point, PETSC_TRUE, &numCPoints, &cpoints));
8601   PetscFunctionReturn(PETSC_SUCCESS);
8602 }
8603 
8604 /*@C
8605   DMPlexGetVTKCellHeight - Returns the height in the DAG used to determine which points are cells (normally 0)
8606 
8607   Input Parameter:
8608 . dm - The `DMPLEX` object
8609 
8610   Output Parameter:
8611 . cellHeight - The height of a cell
8612 
8613   Level: developer
8614 
8615 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexSetVTKCellHeight()`
8616 @*/
8617 PetscErrorCode DMPlexGetVTKCellHeight(DM dm, PetscInt *cellHeight)
8618 {
8619   DM_Plex *mesh = (DM_Plex *)dm->data;
8620 
8621   PetscFunctionBegin;
8622   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8623   PetscAssertPointer(cellHeight, 2);
8624   *cellHeight = mesh->vtkCellHeight;
8625   PetscFunctionReturn(PETSC_SUCCESS);
8626 }
8627 
8628 /*@C
8629   DMPlexSetVTKCellHeight - Sets the height in the DAG used to determine which points are cells (normally 0)
8630 
8631   Input Parameters:
8632 + dm         - The `DMPLEX` object
8633 - cellHeight - The height of a cell
8634 
8635   Level: developer
8636 
8637 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetVTKCellHeight()`
8638 @*/
8639 PetscErrorCode DMPlexSetVTKCellHeight(DM dm, PetscInt cellHeight)
8640 {
8641   DM_Plex *mesh = (DM_Plex *)dm->data;
8642 
8643   PetscFunctionBegin;
8644   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8645   mesh->vtkCellHeight = cellHeight;
8646   PetscFunctionReturn(PETSC_SUCCESS);
8647 }
8648 
8649 /*@
8650   DMPlexGetCellTypeStratum - Get the range of cells of a given celltype
8651 
8652   Input Parameters:
8653 + dm - The `DMPLEX` object
8654 - ct - The `DMPolytopeType` of the cell
8655 
8656   Output Parameters:
8657 + start - The first cell of this type, or `NULL`
8658 - end   - The upper bound on this celltype, or `NULL`
8659 
8660   Level: advanced
8661 
8662 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexConstructGhostCells()`, `DMPlexGetDepthStratum()`, `DMPlexGetHeightStratum()`
8663 @*/
8664 PetscErrorCode DMPlexGetCellTypeStratum(DM dm, DMPolytopeType ct, PetscInt *start, PetscInt *end)
8665 {
8666   DM_Plex *mesh = (DM_Plex *)dm->data;
8667   DMLabel  label;
8668   PetscInt pStart, pEnd;
8669 
8670   PetscFunctionBegin;
8671   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8672   if (start) {
8673     PetscAssertPointer(start, 3);
8674     *start = 0;
8675   }
8676   if (end) {
8677     PetscAssertPointer(end, 4);
8678     *end = 0;
8679   }
8680   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
8681   if (pStart == pEnd) PetscFunctionReturn(PETSC_SUCCESS);
8682   if (mesh->tr) {
8683     PetscCall(DMPlexTransformGetCellTypeStratum(mesh->tr, ct, start, end));
8684   } else {
8685     PetscCall(DMPlexGetCellTypeLabel(dm, &label));
8686     PetscCheck(label, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "No label named celltype was found");
8687     PetscCall(DMLabelGetStratumBounds(label, ct, start, end));
8688   }
8689   PetscFunctionReturn(PETSC_SUCCESS);
8690 }
8691 
8692 PetscErrorCode DMPlexCreateNumbering_Plex(DM dm, PetscInt pStart, PetscInt pEnd, PetscInt shift, PetscInt *globalSize, PetscSF sf, IS *numbering)
8693 {
8694   PetscSection section, globalSection;
8695   PetscInt    *numbers, p;
8696 
8697   PetscFunctionBegin;
8698   if (PetscDefined(USE_DEBUG)) PetscCall(DMPlexCheckPointSF(dm, sf, PETSC_TRUE));
8699   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &section));
8700   PetscCall(PetscSectionSetChart(section, pStart, pEnd));
8701   for (p = pStart; p < pEnd; ++p) PetscCall(PetscSectionSetDof(section, p, 1));
8702   PetscCall(PetscSectionSetUp(section));
8703   PetscCall(PetscSectionCreateGlobalSection(section, sf, PETSC_FALSE, PETSC_FALSE, &globalSection));
8704   PetscCall(PetscMalloc1(pEnd - pStart, &numbers));
8705   for (p = pStart; p < pEnd; ++p) {
8706     PetscCall(PetscSectionGetOffset(globalSection, p, &numbers[p - pStart]));
8707     if (numbers[p - pStart] < 0) numbers[p - pStart] -= shift;
8708     else numbers[p - pStart] += shift;
8709   }
8710   PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), pEnd - pStart, numbers, PETSC_OWN_POINTER, numbering));
8711   if (globalSize) {
8712     PetscLayout layout;
8713     PetscCall(PetscSectionGetPointLayout(PetscObjectComm((PetscObject)dm), globalSection, &layout));
8714     PetscCall(PetscLayoutGetSize(layout, globalSize));
8715     PetscCall(PetscLayoutDestroy(&layout));
8716   }
8717   PetscCall(PetscSectionDestroy(&section));
8718   PetscCall(PetscSectionDestroy(&globalSection));
8719   PetscFunctionReturn(PETSC_SUCCESS);
8720 }
8721 
8722 PetscErrorCode DMPlexCreateCellNumbering_Internal(DM dm, PetscBool includeHybrid, IS *globalCellNumbers)
8723 {
8724   PetscInt cellHeight, cStart, cEnd;
8725 
8726   PetscFunctionBegin;
8727   PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
8728   if (includeHybrid) PetscCall(DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd));
8729   else PetscCall(DMPlexGetSimplexOrBoxCells(dm, cellHeight, &cStart, &cEnd));
8730   PetscCall(DMPlexCreateNumbering_Plex(dm, cStart, cEnd, 0, NULL, dm->sf, globalCellNumbers));
8731   PetscFunctionReturn(PETSC_SUCCESS);
8732 }
8733 
8734 /*@
8735   DMPlexGetCellNumbering - Get a global cell numbering for all cells on this process
8736 
8737   Input Parameter:
8738 . dm - The `DMPLEX` object
8739 
8740   Output Parameter:
8741 . globalCellNumbers - Global cell numbers for all cells on this process
8742 
8743   Level: developer
8744 
8745 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetVertexNumbering()`
8746 @*/
8747 PetscErrorCode DMPlexGetCellNumbering(DM dm, IS *globalCellNumbers)
8748 {
8749   DM_Plex *mesh = (DM_Plex *)dm->data;
8750 
8751   PetscFunctionBegin;
8752   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8753   if (!mesh->globalCellNumbers) PetscCall(DMPlexCreateCellNumbering_Internal(dm, PETSC_FALSE, &mesh->globalCellNumbers));
8754   *globalCellNumbers = mesh->globalCellNumbers;
8755   PetscFunctionReturn(PETSC_SUCCESS);
8756 }
8757 
8758 PetscErrorCode DMPlexCreateVertexNumbering_Internal(DM dm, PetscBool includeHybrid, IS *globalVertexNumbers)
8759 {
8760   PetscInt vStart, vEnd;
8761 
8762   PetscFunctionBegin;
8763   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8764   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
8765   PetscCall(DMPlexCreateNumbering_Plex(dm, vStart, vEnd, 0, NULL, dm->sf, globalVertexNumbers));
8766   PetscFunctionReturn(PETSC_SUCCESS);
8767 }
8768 
8769 /*@
8770   DMPlexGetVertexNumbering - Get a global vertex numbering for all vertices on this process
8771 
8772   Input Parameter:
8773 . dm - The `DMPLEX` object
8774 
8775   Output Parameter:
8776 . globalVertexNumbers - Global vertex numbers for all vertices on this process
8777 
8778   Level: developer
8779 
8780 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellNumbering()`
8781 @*/
8782 PetscErrorCode DMPlexGetVertexNumbering(DM dm, IS *globalVertexNumbers)
8783 {
8784   DM_Plex *mesh = (DM_Plex *)dm->data;
8785 
8786   PetscFunctionBegin;
8787   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8788   if (!mesh->globalVertexNumbers) PetscCall(DMPlexCreateVertexNumbering_Internal(dm, PETSC_FALSE, &mesh->globalVertexNumbers));
8789   *globalVertexNumbers = mesh->globalVertexNumbers;
8790   PetscFunctionReturn(PETSC_SUCCESS);
8791 }
8792 
8793 /*@
8794   DMPlexCreatePointNumbering - Create a global numbering for all points.
8795 
8796   Collective
8797 
8798   Input Parameter:
8799 . dm - The `DMPLEX` object
8800 
8801   Output Parameter:
8802 . globalPointNumbers - Global numbers for all points on this process
8803 
8804   Level: developer
8805 
8806   Notes:
8807   The point numbering `IS` is parallel, with local portion indexed by local points (see `DMGetLocalSection()`). The global
8808   points are taken as stratified, with each MPI rank owning a contiguous subset of each stratum. In the IS, owned points
8809   will have their non-negative value while points owned by different ranks will be involuted -(idx+1). As an example,
8810   consider a parallel mesh in which the first two elements and first two vertices are owned by rank 0.
8811 
8812   The partitioned mesh is
8813   ```
8814   (2)--0--(3)--1--(4)    (1)--0--(2)
8815   ```
8816   and its global numbering is
8817   ```
8818   (3)--0--(4)--1--(5)--2--(6)
8819   ```
8820   Then the global numbering is provided as
8821   ```
8822   [0] Number of indices in set 5
8823   [0] 0 0
8824   [0] 1 1
8825   [0] 2 3
8826   [0] 3 4
8827   [0] 4 -6
8828   [1] Number of indices in set 3
8829   [1] 0 2
8830   [1] 1 5
8831   [1] 2 6
8832   ```
8833 
8834 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetCellNumbering()`
8835 @*/
8836 PetscErrorCode DMPlexCreatePointNumbering(DM dm, IS *globalPointNumbers)
8837 {
8838   IS        nums[4];
8839   PetscInt  depths[4], gdepths[4], starts[4];
8840   PetscInt  depth, d, shift = 0;
8841   PetscBool empty = PETSC_FALSE;
8842 
8843   PetscFunctionBegin;
8844   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8845   PetscCall(DMPlexGetDepth(dm, &depth));
8846   // For unstratified meshes use dim instead of depth
8847   if (depth < 0) PetscCall(DMGetDimension(dm, &depth));
8848   // If any stratum is empty, we must mark all empty
8849   for (d = 0; d <= depth; ++d) {
8850     PetscInt end;
8851 
8852     depths[d] = depth - d;
8853     PetscCall(DMPlexGetDepthStratum(dm, depths[d], &starts[d], &end));
8854     if (!(starts[d] - end)) empty = PETSC_TRUE;
8855   }
8856   if (empty)
8857     for (d = 0; d <= depth; ++d) {
8858       depths[d] = -1;
8859       starts[d] = -1;
8860     }
8861   else PetscCall(PetscSortIntWithArray(depth + 1, starts, depths));
8862   PetscCall(MPIU_Allreduce(depths, gdepths, depth + 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
8863   for (d = 0; d <= depth; ++d) PetscCheck(starts[d] < 0 || depths[d] == gdepths[d], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected depth %" PetscInt_FMT ", found %" PetscInt_FMT, depths[d], gdepths[d]);
8864   // Note here that 'shift' is collective, so that the numbering is stratified by depth
8865   for (d = 0; d <= depth; ++d) {
8866     PetscInt pStart, pEnd, gsize;
8867 
8868     PetscCall(DMPlexGetDepthStratum(dm, gdepths[d], &pStart, &pEnd));
8869     PetscCall(DMPlexCreateNumbering_Plex(dm, pStart, pEnd, shift, &gsize, dm->sf, &nums[d]));
8870     shift += gsize;
8871   }
8872   PetscCall(ISConcatenate(PETSC_COMM_SELF, depth + 1, nums, globalPointNumbers));
8873   for (d = 0; d <= depth; ++d) PetscCall(ISDestroy(&nums[d]));
8874   PetscFunctionReturn(PETSC_SUCCESS);
8875 }
8876 
8877 /*@
8878   DMPlexCreateRankField - Create a cell field whose value is the rank of the owner
8879 
8880   Input Parameter:
8881 . dm - The `DMPLEX` object
8882 
8883   Output Parameter:
8884 . ranks - The rank field
8885 
8886   Options Database Key:
8887 . -dm_partition_view - Adds the rank field into the `DM` output from `-dm_view` using the same viewer
8888 
8889   Level: intermediate
8890 
8891 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`
8892 @*/
8893 PetscErrorCode DMPlexCreateRankField(DM dm, Vec *ranks)
8894 {
8895   DM             rdm;
8896   PetscFE        fe;
8897   PetscScalar   *r;
8898   PetscMPIInt    rank;
8899   DMPolytopeType ct;
8900   PetscInt       dim, cStart, cEnd, c;
8901   PetscBool      simplex;
8902 
8903   PetscFunctionBeginUser;
8904   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8905   PetscAssertPointer(ranks, 2);
8906   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
8907   PetscCall(DMClone(dm, &rdm));
8908   PetscCall(DMGetDimension(rdm, &dim));
8909   PetscCall(DMPlexGetHeightStratum(rdm, 0, &cStart, &cEnd));
8910   PetscCall(DMPlexGetCellType(dm, cStart, &ct));
8911   simplex = DMPolytopeTypeGetNumVertices(ct) == DMPolytopeTypeGetDim(ct) + 1 ? PETSC_TRUE : PETSC_FALSE;
8912   PetscCall(PetscFECreateDefault(PETSC_COMM_SELF, dim, 1, simplex, "PETSc___rank_", -1, &fe));
8913   PetscCall(PetscObjectSetName((PetscObject)fe, "rank"));
8914   PetscCall(DMSetField(rdm, 0, NULL, (PetscObject)fe));
8915   PetscCall(PetscFEDestroy(&fe));
8916   PetscCall(DMCreateDS(rdm));
8917   PetscCall(DMCreateGlobalVector(rdm, ranks));
8918   PetscCall(PetscObjectSetName((PetscObject)*ranks, "partition"));
8919   PetscCall(VecGetArray(*ranks, &r));
8920   for (c = cStart; c < cEnd; ++c) {
8921     PetscScalar *lr;
8922 
8923     PetscCall(DMPlexPointGlobalRef(rdm, c, r, &lr));
8924     if (lr) *lr = rank;
8925   }
8926   PetscCall(VecRestoreArray(*ranks, &r));
8927   PetscCall(DMDestroy(&rdm));
8928   PetscFunctionReturn(PETSC_SUCCESS);
8929 }
8930 
8931 /*@
8932   DMPlexCreateLabelField - Create a field whose value is the label value for that point
8933 
8934   Input Parameters:
8935 + dm    - The `DMPLEX`
8936 - label - The `DMLabel`
8937 
8938   Output Parameter:
8939 . val - The label value field
8940 
8941   Options Database Key:
8942 . -dm_label_view - Adds the label value field into the `DM` output from `-dm_view` using the same viewer
8943 
8944   Level: intermediate
8945 
8946 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMView()`
8947 @*/
8948 PetscErrorCode DMPlexCreateLabelField(DM dm, DMLabel label, Vec *val)
8949 {
8950   DM             rdm, plex;
8951   Vec            lval;
8952   PetscSection   section;
8953   PetscFE        fe;
8954   PetscScalar   *v;
8955   PetscInt       dim, pStart, pEnd, p, cStart;
8956   DMPolytopeType ct;
8957   char           name[PETSC_MAX_PATH_LEN];
8958   const char    *lname, *prefix;
8959 
8960   PetscFunctionBeginUser;
8961   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
8962   PetscAssertPointer(label, 2);
8963   PetscAssertPointer(val, 3);
8964   PetscCall(DMClone(dm, &rdm));
8965   PetscCall(DMConvert(rdm, DMPLEX, &plex));
8966   PetscCall(DMPlexGetHeightStratum(plex, 0, &cStart, NULL));
8967   PetscCall(DMPlexGetCellType(plex, cStart, &ct));
8968   PetscCall(DMDestroy(&plex));
8969   PetscCall(DMGetDimension(rdm, &dim));
8970   PetscCall(DMGetOptionsPrefix(dm, &prefix));
8971   PetscCall(PetscObjectGetName((PetscObject)label, &lname));
8972   PetscCall(PetscSNPrintf(name, sizeof(name), "%s%s_", prefix ? prefix : "", lname));
8973   PetscCall(PetscFECreateByCell(PETSC_COMM_SELF, dim, 1, ct, name, -1, &fe));
8974   PetscCall(PetscObjectSetName((PetscObject)fe, ""));
8975   PetscCall(DMSetField(rdm, 0, NULL, (PetscObject)fe));
8976   PetscCall(PetscFEDestroy(&fe));
8977   PetscCall(DMCreateDS(rdm));
8978   PetscCall(DMCreateGlobalVector(rdm, val));
8979   PetscCall(DMCreateLocalVector(rdm, &lval));
8980   PetscCall(PetscObjectSetName((PetscObject)*val, lname));
8981   PetscCall(DMGetLocalSection(rdm, &section));
8982   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
8983   PetscCall(VecGetArray(lval, &v));
8984   for (p = pStart; p < pEnd; ++p) {
8985     PetscInt cval, dof, off;
8986 
8987     PetscCall(PetscSectionGetDof(section, p, &dof));
8988     if (!dof) continue;
8989     PetscCall(DMLabelGetValue(label, p, &cval));
8990     PetscCall(PetscSectionGetOffset(section, p, &off));
8991     for (PetscInt d = 0; d < dof; d++) v[off + d] = cval;
8992   }
8993   PetscCall(VecRestoreArray(lval, &v));
8994   PetscCall(DMLocalToGlobal(rdm, lval, INSERT_VALUES, *val));
8995   PetscCall(VecDestroy(&lval));
8996   PetscCall(DMDestroy(&rdm));
8997   PetscFunctionReturn(PETSC_SUCCESS);
8998 }
8999 
9000 /*@
9001   DMPlexCheckSymmetry - Check that the adjacency information in the mesh is symmetric.
9002 
9003   Input Parameter:
9004 . dm - The `DMPLEX` object
9005 
9006   Level: developer
9007 
9008   Notes:
9009   This is a useful diagnostic when creating meshes programmatically.
9010 
9011   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9012 
9013 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`, `DMSetFromOptions()`
9014 @*/
9015 PetscErrorCode DMPlexCheckSymmetry(DM dm)
9016 {
9017   PetscSection    coneSection, supportSection;
9018   const PetscInt *cone, *support;
9019   PetscInt        coneSize, c, supportSize, s;
9020   PetscInt        pStart, pEnd, p, pp, csize, ssize;
9021   PetscBool       storagecheck = PETSC_TRUE;
9022 
9023   PetscFunctionBegin;
9024   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9025   PetscCall(DMViewFromOptions(dm, NULL, "-sym_dm_view"));
9026   PetscCall(DMPlexGetConeSection(dm, &coneSection));
9027   PetscCall(DMPlexGetSupportSection(dm, &supportSection));
9028   /* Check that point p is found in the support of its cone points, and vice versa */
9029   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
9030   for (p = pStart; p < pEnd; ++p) {
9031     PetscCall(DMPlexGetConeSize(dm, p, &coneSize));
9032     PetscCall(DMPlexGetCone(dm, p, &cone));
9033     for (c = 0; c < coneSize; ++c) {
9034       PetscBool dup = PETSC_FALSE;
9035       PetscInt  d;
9036       for (d = c - 1; d >= 0; --d) {
9037         if (cone[c] == cone[d]) {
9038           dup = PETSC_TRUE;
9039           break;
9040         }
9041       }
9042       PetscCall(DMPlexGetSupportSize(dm, cone[c], &supportSize));
9043       PetscCall(DMPlexGetSupport(dm, cone[c], &support));
9044       for (s = 0; s < supportSize; ++s) {
9045         if (support[s] == p) break;
9046       }
9047       if ((s >= supportSize) || (dup && (support[s + 1] != p))) {
9048         PetscCall(PetscPrintf(PETSC_COMM_SELF, "p: %" PetscInt_FMT " cone: ", p));
9049         for (s = 0; s < coneSize; ++s) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%" PetscInt_FMT ", ", cone[s]));
9050         PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
9051         PetscCall(PetscPrintf(PETSC_COMM_SELF, "p: %" PetscInt_FMT " support: ", cone[c]));
9052         for (s = 0; s < supportSize; ++s) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%" PetscInt_FMT ", ", support[s]));
9053         PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
9054         PetscCheck(!dup, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point %" PetscInt_FMT " not repeatedly found in support of repeated cone point %" PetscInt_FMT, p, cone[c]);
9055         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point %" PetscInt_FMT " not found in support of cone point %" PetscInt_FMT, p, cone[c]);
9056       }
9057     }
9058     PetscCall(DMPlexGetTreeParent(dm, p, &pp, NULL));
9059     if (p != pp) {
9060       storagecheck = PETSC_FALSE;
9061       continue;
9062     }
9063     PetscCall(DMPlexGetSupportSize(dm, p, &supportSize));
9064     PetscCall(DMPlexGetSupport(dm, p, &support));
9065     for (s = 0; s < supportSize; ++s) {
9066       PetscCall(DMPlexGetConeSize(dm, support[s], &coneSize));
9067       PetscCall(DMPlexGetCone(dm, support[s], &cone));
9068       for (c = 0; c < coneSize; ++c) {
9069         PetscCall(DMPlexGetTreeParent(dm, cone[c], &pp, NULL));
9070         if (cone[c] != pp) {
9071           c = 0;
9072           break;
9073         }
9074         if (cone[c] == p) break;
9075       }
9076       if (c >= coneSize) {
9077         PetscCall(PetscPrintf(PETSC_COMM_SELF, "p: %" PetscInt_FMT " support: ", p));
9078         for (c = 0; c < supportSize; ++c) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%" PetscInt_FMT ", ", support[c]));
9079         PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
9080         PetscCall(PetscPrintf(PETSC_COMM_SELF, "p: %" PetscInt_FMT " cone: ", support[s]));
9081         for (c = 0; c < coneSize; ++c) PetscCall(PetscPrintf(PETSC_COMM_SELF, "%" PetscInt_FMT ", ", cone[c]));
9082         PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
9083         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point %" PetscInt_FMT " not found in cone of support point %" PetscInt_FMT, p, support[s]);
9084       }
9085     }
9086   }
9087   if (storagecheck) {
9088     PetscCall(PetscSectionGetStorageSize(coneSection, &csize));
9089     PetscCall(PetscSectionGetStorageSize(supportSection, &ssize));
9090     PetscCheck(csize == ssize, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Total cone size %" PetscInt_FMT " != Total support size %" PetscInt_FMT, csize, ssize);
9091   }
9092   PetscFunctionReturn(PETSC_SUCCESS);
9093 }
9094 
9095 /*
9096   For submeshes with cohesive cells (see DMPlexConstructCohesiveCells()), we allow a special case where some of the boundary of a face (edges and vertices) are not duplicated. We call these special boundary points "unsplit", since the same edge or vertex appears in both copies of the face. These unsplit points throw off our counting, so we have to explicitly account for them here.
9097 */
9098 static PetscErrorCode DMPlexCellUnsplitVertices_Private(DM dm, PetscInt c, DMPolytopeType ct, PetscInt *unsplit)
9099 {
9100   DMPolytopeType  cct;
9101   PetscInt        ptpoints[4];
9102   const PetscInt *cone, *ccone, *ptcone;
9103   PetscInt        coneSize, cp, cconeSize, ccp, npt = 0, pt;
9104 
9105   PetscFunctionBegin;
9106   *unsplit = 0;
9107   switch (ct) {
9108   case DM_POLYTOPE_POINT_PRISM_TENSOR:
9109     ptpoints[npt++] = c;
9110     break;
9111   case DM_POLYTOPE_SEG_PRISM_TENSOR:
9112     PetscCall(DMPlexGetCone(dm, c, &cone));
9113     PetscCall(DMPlexGetConeSize(dm, c, &coneSize));
9114     for (cp = 0; cp < coneSize; ++cp) {
9115       PetscCall(DMPlexGetCellType(dm, cone[cp], &cct));
9116       if (cct == DM_POLYTOPE_POINT_PRISM_TENSOR) ptpoints[npt++] = cone[cp];
9117     }
9118     break;
9119   case DM_POLYTOPE_TRI_PRISM_TENSOR:
9120   case DM_POLYTOPE_QUAD_PRISM_TENSOR:
9121     PetscCall(DMPlexGetCone(dm, c, &cone));
9122     PetscCall(DMPlexGetConeSize(dm, c, &coneSize));
9123     for (cp = 0; cp < coneSize; ++cp) {
9124       PetscCall(DMPlexGetCone(dm, cone[cp], &ccone));
9125       PetscCall(DMPlexGetConeSize(dm, cone[cp], &cconeSize));
9126       for (ccp = 0; ccp < cconeSize; ++ccp) {
9127         PetscCall(DMPlexGetCellType(dm, ccone[ccp], &cct));
9128         if (cct == DM_POLYTOPE_POINT_PRISM_TENSOR) {
9129           PetscInt p;
9130           for (p = 0; p < npt; ++p)
9131             if (ptpoints[p] == ccone[ccp]) break;
9132           if (p == npt) ptpoints[npt++] = ccone[ccp];
9133         }
9134       }
9135     }
9136     break;
9137   default:
9138     break;
9139   }
9140   for (pt = 0; pt < npt; ++pt) {
9141     PetscCall(DMPlexGetCone(dm, ptpoints[pt], &ptcone));
9142     if (ptcone[0] == ptcone[1]) ++(*unsplit);
9143   }
9144   PetscFunctionReturn(PETSC_SUCCESS);
9145 }
9146 
9147 /*@
9148   DMPlexCheckSkeleton - Check that each cell has the correct number of vertices
9149 
9150   Input Parameters:
9151 + dm         - The `DMPLEX` object
9152 - cellHeight - Normally 0
9153 
9154   Level: developer
9155 
9156   Notes:
9157   This is a useful diagnostic when creating meshes programmatically.
9158   Currently applicable only to homogeneous simplex or tensor meshes.
9159 
9160   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9161 
9162 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`, `DMSetFromOptions()`
9163 @*/
9164 PetscErrorCode DMPlexCheckSkeleton(DM dm, PetscInt cellHeight)
9165 {
9166   DMPlexInterpolatedFlag interp;
9167   DMPolytopeType         ct;
9168   PetscInt               vStart, vEnd, cStart, cEnd, c;
9169 
9170   PetscFunctionBegin;
9171   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9172   PetscCall(DMPlexIsInterpolated(dm, &interp));
9173   PetscCall(DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd));
9174   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
9175   for (c = cStart; c < cEnd; ++c) {
9176     PetscInt *closure = NULL;
9177     PetscInt  coneSize, closureSize, cl, Nv = 0;
9178 
9179     PetscCall(DMPlexGetCellType(dm, c, &ct));
9180     PetscCheck((PetscInt)ct >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " has no cell type", c);
9181     if (ct == DM_POLYTOPE_UNKNOWN) continue;
9182     if (interp == DMPLEX_INTERPOLATED_FULL) {
9183       PetscCall(DMPlexGetConeSize(dm, c, &coneSize));
9184       PetscCheck(coneSize == DMPolytopeTypeGetConeSize(ct), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " of type %s has cone size %" PetscInt_FMT " != %" PetscInt_FMT, c, DMPolytopeTypes[ct], coneSize, DMPolytopeTypeGetConeSize(ct));
9185     }
9186     PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
9187     for (cl = 0; cl < closureSize * 2; cl += 2) {
9188       const PetscInt p = closure[cl];
9189       if ((p >= vStart) && (p < vEnd)) ++Nv;
9190     }
9191     PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
9192     /* Special Case: Tensor faces with identified vertices */
9193     if (Nv < DMPolytopeTypeGetNumVertices(ct)) {
9194       PetscInt unsplit;
9195 
9196       PetscCall(DMPlexCellUnsplitVertices_Private(dm, c, ct, &unsplit));
9197       if (Nv + unsplit == DMPolytopeTypeGetNumVertices(ct)) continue;
9198     }
9199     PetscCheck(Nv == DMPolytopeTypeGetNumVertices(ct), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " of type %s has %" PetscInt_FMT " vertices != %" PetscInt_FMT, c, DMPolytopeTypes[ct], Nv, DMPolytopeTypeGetNumVertices(ct));
9200   }
9201   PetscFunctionReturn(PETSC_SUCCESS);
9202 }
9203 
9204 /*@
9205   DMPlexCheckFaces - Check that the faces of each cell give a vertex order this is consistent with what we expect from the cell type
9206 
9207   Collective
9208 
9209   Input Parameters:
9210 + dm         - The `DMPLEX` object
9211 - cellHeight - Normally 0
9212 
9213   Level: developer
9214 
9215   Notes:
9216   This is a useful diagnostic when creating meshes programmatically.
9217   This routine is only relevant for meshes that are fully interpolated across all ranks.
9218   It will error out if a partially interpolated mesh is given on some rank.
9219   It will do nothing for locally uninterpolated mesh (as there is nothing to check).
9220 
9221   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9222 
9223 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`, `DMPlexGetVTKCellHeight()`, `DMSetFromOptions()`
9224 @*/
9225 PetscErrorCode DMPlexCheckFaces(DM dm, PetscInt cellHeight)
9226 {
9227   PetscInt               dim, depth, vStart, vEnd, cStart, cEnd, c, h;
9228   DMPlexInterpolatedFlag interpEnum;
9229 
9230   PetscFunctionBegin;
9231   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9232   PetscCall(DMPlexIsInterpolatedCollective(dm, &interpEnum));
9233   if (interpEnum == DMPLEX_INTERPOLATED_NONE) PetscFunctionReturn(PETSC_SUCCESS);
9234   if (interpEnum != DMPLEX_INTERPOLATED_FULL) {
9235     PetscCall(PetscPrintf(PetscObjectComm((PetscObject)dm), "DMPlexCheckFaces() warning: Mesh is only partially interpolated, this is currently not supported"));
9236     PetscFunctionReturn(PETSC_SUCCESS);
9237   }
9238 
9239   PetscCall(DMGetDimension(dm, &dim));
9240   PetscCall(DMPlexGetDepth(dm, &depth));
9241   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
9242   for (h = cellHeight; h < PetscMin(depth, dim); ++h) {
9243     PetscCall(DMPlexGetHeightStratum(dm, h, &cStart, &cEnd));
9244     for (c = cStart; c < cEnd; ++c) {
9245       const PetscInt       *cone, *ornt, *faceSizes, *faces;
9246       const DMPolytopeType *faceTypes;
9247       DMPolytopeType        ct;
9248       PetscInt              numFaces, coneSize, f;
9249       PetscInt             *closure = NULL, closureSize, cl, numCorners = 0, fOff = 0, unsplit;
9250 
9251       PetscCall(DMPlexGetCellType(dm, c, &ct));
9252       PetscCall(DMPlexCellUnsplitVertices_Private(dm, c, ct, &unsplit));
9253       if (unsplit) continue;
9254       PetscCall(DMPlexGetConeSize(dm, c, &coneSize));
9255       PetscCall(DMPlexGetCone(dm, c, &cone));
9256       PetscCall(DMPlexGetConeOrientation(dm, c, &ornt));
9257       PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
9258       for (cl = 0; cl < closureSize * 2; cl += 2) {
9259         const PetscInt p = closure[cl];
9260         if ((p >= vStart) && (p < vEnd)) closure[numCorners++] = p;
9261       }
9262       PetscCall(DMPlexGetRawFaces_Internal(dm, ct, closure, &numFaces, &faceTypes, &faceSizes, &faces));
9263       PetscCheck(coneSize == numFaces, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " of type %s has %" PetscInt_FMT " faces but should have %" PetscInt_FMT, c, DMPolytopeTypes[ct], coneSize, numFaces);
9264       for (f = 0; f < numFaces; ++f) {
9265         DMPolytopeType fct;
9266         PetscInt      *fclosure = NULL, fclosureSize, cl, fnumCorners = 0, v;
9267 
9268         PetscCall(DMPlexGetCellType(dm, cone[f], &fct));
9269         PetscCall(DMPlexGetTransitiveClosure_Internal(dm, cone[f], ornt[f], PETSC_TRUE, &fclosureSize, &fclosure));
9270         for (cl = 0; cl < fclosureSize * 2; cl += 2) {
9271           const PetscInt p = fclosure[cl];
9272           if ((p >= vStart) && (p < vEnd)) fclosure[fnumCorners++] = p;
9273         }
9274         PetscCheck(fnumCorners == faceSizes[f], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %" PetscInt_FMT " of type %s (cone idx %" PetscInt_FMT ") of cell %" PetscInt_FMT " of type %s has %" PetscInt_FMT " vertices but should have %" PetscInt_FMT, cone[f], DMPolytopeTypes[fct], f, c, DMPolytopeTypes[ct], fnumCorners, faceSizes[f]);
9275         for (v = 0; v < fnumCorners; ++v) {
9276           if (fclosure[v] != faces[fOff + v]) {
9277             PetscInt v1;
9278 
9279             PetscCall(PetscPrintf(PETSC_COMM_SELF, "face closure:"));
9280             for (v1 = 0; v1 < fnumCorners; ++v1) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, fclosure[v1]));
9281             PetscCall(PetscPrintf(PETSC_COMM_SELF, "\ncell face:"));
9282             for (v1 = 0; v1 < fnumCorners; ++v1) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %" PetscInt_FMT, faces[fOff + v1]));
9283             PetscCall(PetscPrintf(PETSC_COMM_SELF, "\n"));
9284             SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %" PetscInt_FMT " of type %s (cone idx %" PetscInt_FMT ", ornt %" PetscInt_FMT ") of cell %" PetscInt_FMT " of type %s vertex %" PetscInt_FMT ", %" PetscInt_FMT " != %" PetscInt_FMT, cone[f], DMPolytopeTypes[fct], f, ornt[f], c, DMPolytopeTypes[ct], v, fclosure[v], faces[fOff + v]);
9285           }
9286         }
9287         PetscCall(DMPlexRestoreTransitiveClosure(dm, cone[f], PETSC_TRUE, &fclosureSize, &fclosure));
9288         fOff += faceSizes[f];
9289       }
9290       PetscCall(DMPlexRestoreRawFaces_Internal(dm, ct, closure, &numFaces, &faceTypes, &faceSizes, &faces));
9291       PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure));
9292     }
9293   }
9294   PetscFunctionReturn(PETSC_SUCCESS);
9295 }
9296 
9297 /*@
9298   DMPlexCheckGeometry - Check the geometry of mesh cells
9299 
9300   Input Parameter:
9301 . dm - The `DMPLEX` object
9302 
9303   Level: developer
9304 
9305   Notes:
9306   This is a useful diagnostic when creating meshes programmatically.
9307 
9308   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9309 
9310 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`, `DMSetFromOptions()`
9311 @*/
9312 PetscErrorCode DMPlexCheckGeometry(DM dm)
9313 {
9314   Vec       coordinates;
9315   PetscReal detJ, J[9], refVol = 1.0;
9316   PetscReal vol;
9317   PetscInt  dim, depth, dE, d, cStart, cEnd, c;
9318 
9319   PetscFunctionBegin;
9320   PetscCall(DMGetDimension(dm, &dim));
9321   PetscCall(DMGetCoordinateDim(dm, &dE));
9322   if (dim != dE) PetscFunctionReturn(PETSC_SUCCESS);
9323   PetscCall(DMPlexGetDepth(dm, &depth));
9324   for (d = 0; d < dim; ++d) refVol *= 2.0;
9325   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
9326   /* Make sure local coordinates are created, because that step is collective */
9327   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
9328   if (!coordinates) PetscFunctionReturn(PETSC_SUCCESS);
9329   for (c = cStart; c < cEnd; ++c) {
9330     DMPolytopeType ct;
9331     PetscInt       unsplit;
9332     PetscBool      ignoreZeroVol = PETSC_FALSE;
9333 
9334     PetscCall(DMPlexGetCellType(dm, c, &ct));
9335     switch (ct) {
9336     case DM_POLYTOPE_SEG_PRISM_TENSOR:
9337     case DM_POLYTOPE_TRI_PRISM_TENSOR:
9338     case DM_POLYTOPE_QUAD_PRISM_TENSOR:
9339       ignoreZeroVol = PETSC_TRUE;
9340       break;
9341     default:
9342       break;
9343     }
9344     switch (ct) {
9345     case DM_POLYTOPE_TRI_PRISM:
9346     case DM_POLYTOPE_TRI_PRISM_TENSOR:
9347     case DM_POLYTOPE_QUAD_PRISM_TENSOR:
9348     case DM_POLYTOPE_PYRAMID:
9349       continue;
9350     default:
9351       break;
9352     }
9353     PetscCall(DMPlexCellUnsplitVertices_Private(dm, c, ct, &unsplit));
9354     if (unsplit) continue;
9355     PetscCall(DMPlexComputeCellGeometryFEM(dm, c, NULL, NULL, J, NULL, &detJ));
9356     PetscCheck(detJ >= -PETSC_SMALL && (detJ > 0.0 || ignoreZeroVol), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh cell %" PetscInt_FMT " of type %s is inverted, |J| = %g", c, DMPolytopeTypes[ct], (double)detJ);
9357     PetscCall(PetscInfo(dm, "Cell %" PetscInt_FMT " FEM Volume %g\n", c, (double)(detJ * refVol)));
9358     /* This should work with periodicity since DG coordinates should be used */
9359     if (depth > 1) {
9360       PetscCall(DMPlexComputeCellGeometryFVM(dm, c, &vol, NULL, NULL));
9361       PetscCheck(vol >= -PETSC_SMALL && (vol > 0.0 || ignoreZeroVol), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh cell %" PetscInt_FMT " of type %s is inverted, vol = %g", c, DMPolytopeTypes[ct], (double)vol);
9362       PetscCall(PetscInfo(dm, "Cell %" PetscInt_FMT " FVM Volume %g\n", c, (double)vol));
9363     }
9364   }
9365   PetscFunctionReturn(PETSC_SUCCESS);
9366 }
9367 
9368 /*@
9369   DMPlexCheckPointSF - Check that several necessary conditions are met for the point `PetscSF` of this plex.
9370 
9371   Collective
9372 
9373   Input Parameters:
9374 + dm              - The `DMPLEX` object
9375 . pointSF         - The `PetscSF`, or `NULL` for `PointSF` attached to `DM`
9376 - allowExtraRoots - Flag to allow extra points not present in the `DM`
9377 
9378   Level: developer
9379 
9380   Notes:
9381   This is mainly intended for debugging/testing purposes.
9382 
9383   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9384 
9385   Extra roots can come from periodic cuts, where additional points appear on the boundary
9386 
9387 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMGetPointSF()`, `DMSetFromOptions()`
9388 @*/
9389 PetscErrorCode DMPlexCheckPointSF(DM dm, PetscSF pointSF, PetscBool allowExtraRoots)
9390 {
9391   PetscInt           l, nleaves, nroots, overlap;
9392   const PetscInt    *locals;
9393   const PetscSFNode *remotes;
9394   PetscBool          distributed;
9395   MPI_Comm           comm;
9396   PetscMPIInt        rank;
9397 
9398   PetscFunctionBegin;
9399   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9400   if (pointSF) PetscValidHeaderSpecific(pointSF, PETSCSF_CLASSID, 2);
9401   else pointSF = dm->sf;
9402   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
9403   PetscCheck(pointSF, comm, PETSC_ERR_ARG_WRONGSTATE, "DMPlex must have Point SF attached");
9404   PetscCallMPI(MPI_Comm_rank(comm, &rank));
9405   {
9406     PetscMPIInt mpiFlag;
9407 
9408     PetscCallMPI(MPI_Comm_compare(comm, PetscObjectComm((PetscObject)pointSF), &mpiFlag));
9409     PetscCheck(mpiFlag == MPI_CONGRUENT || mpiFlag == MPI_IDENT, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMECOMM, "DM and Point SF have different communicators (flag %d)", mpiFlag);
9410   }
9411   PetscCall(PetscSFGetGraph(pointSF, &nroots, &nleaves, &locals, &remotes));
9412   PetscCall(DMPlexIsDistributed(dm, &distributed));
9413   if (!distributed) {
9414     PetscCheck(nroots < 0 || nleaves == 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Undistributed DMPlex cannot have non-empty PointSF (has %" PetscInt_FMT " roots, %" PetscInt_FMT " leaves)", nroots, nleaves);
9415     PetscFunctionReturn(PETSC_SUCCESS);
9416   }
9417   PetscCheck(nroots >= 0, comm, PETSC_ERR_ARG_WRONGSTATE, "This DMPlex is distributed but its PointSF has no graph set (has %" PetscInt_FMT " roots, %" PetscInt_FMT " leaves)", nroots, nleaves);
9418   PetscCall(DMPlexGetOverlap(dm, &overlap));
9419 
9420   /* Check SF graph is compatible with DMPlex chart */
9421   {
9422     PetscInt pStart, pEnd, maxLeaf;
9423 
9424     PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
9425     PetscCall(PetscSFGetLeafRange(pointSF, NULL, &maxLeaf));
9426     PetscCheck(allowExtraRoots || pEnd - pStart == nroots, PETSC_COMM_SELF, PETSC_ERR_PLIB, "pEnd - pStart = %" PetscInt_FMT " != nroots = %" PetscInt_FMT, pEnd - pStart, nroots);
9427     PetscCheck(maxLeaf < pEnd, PETSC_COMM_SELF, PETSC_ERR_PLIB, "maxLeaf = %" PetscInt_FMT " >= pEnd = %" PetscInt_FMT, maxLeaf, pEnd);
9428   }
9429 
9430   /* Check Point SF has no local points referenced */
9431   for (l = 0; l < nleaves; l++) {
9432     PetscAssert(remotes[l].rank != (PetscInt)rank, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point SF contains local point %" PetscInt_FMT " <- (%" PetscInt_FMT ",%" PetscInt_FMT ")", locals ? locals[l] : l, remotes[l].rank, remotes[l].index);
9433   }
9434 
9435   /* Check there are no cells in interface */
9436   if (!overlap) {
9437     PetscInt cellHeight, cStart, cEnd;
9438 
9439     PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
9440     PetscCall(DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd));
9441     for (l = 0; l < nleaves; ++l) {
9442       const PetscInt point = locals ? locals[l] : l;
9443 
9444       PetscCheck(point < cStart || point >= cEnd, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point SF contains %" PetscInt_FMT " which is a cell", point);
9445     }
9446   }
9447 
9448   /* If some point is in interface, then all its cone points must be also in interface (either as leaves or roots) */
9449   {
9450     const PetscInt *rootdegree;
9451 
9452     PetscCall(PetscSFComputeDegreeBegin(pointSF, &rootdegree));
9453     PetscCall(PetscSFComputeDegreeEnd(pointSF, &rootdegree));
9454     for (l = 0; l < nleaves; ++l) {
9455       const PetscInt  point = locals ? locals[l] : l;
9456       const PetscInt *cone;
9457       PetscInt        coneSize, c, idx;
9458 
9459       PetscCall(DMPlexGetConeSize(dm, point, &coneSize));
9460       PetscCall(DMPlexGetCone(dm, point, &cone));
9461       for (c = 0; c < coneSize; ++c) {
9462         if (!rootdegree[cone[c]]) {
9463           if (locals) {
9464             PetscCall(PetscFindInt(cone[c], nleaves, locals, &idx));
9465           } else {
9466             idx = (cone[c] < nleaves) ? cone[c] : -1;
9467           }
9468           PetscCheck(idx >= 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point SF contains %" PetscInt_FMT " but not %" PetscInt_FMT " from its cone", point, cone[c]);
9469         }
9470       }
9471     }
9472   }
9473   PetscFunctionReturn(PETSC_SUCCESS);
9474 }
9475 
9476 /*@
9477   DMPlexCheck - Perform various checks of `DMPLEX` sanity
9478 
9479   Input Parameter:
9480 . dm - The `DMPLEX` object
9481 
9482   Level: developer
9483 
9484   Notes:
9485   This is a useful diagnostic when creating meshes programmatically.
9486 
9487   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9488 
9489   Currently does not include `DMPlexCheckCellShape()`.
9490 
9491 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`, `DMSetFromOptions()`
9492 @*/
9493 PetscErrorCode DMPlexCheck(DM dm)
9494 {
9495   PetscInt cellHeight;
9496 
9497   PetscFunctionBegin;
9498   PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
9499   PetscCall(DMPlexCheckSymmetry(dm));
9500   PetscCall(DMPlexCheckSkeleton(dm, cellHeight));
9501   PetscCall(DMPlexCheckFaces(dm, cellHeight));
9502   PetscCall(DMPlexCheckGeometry(dm));
9503   PetscCall(DMPlexCheckPointSF(dm, NULL, PETSC_FALSE));
9504   PetscCall(DMPlexCheckInterfaceCones(dm));
9505   PetscFunctionReturn(PETSC_SUCCESS);
9506 }
9507 
9508 typedef struct cell_stats {
9509   PetscReal min, max, sum, squaresum;
9510   PetscInt  count;
9511 } cell_stats_t;
9512 
9513 static void MPIAPI cell_stats_reduce(void *a, void *b, int *len, MPI_Datatype *datatype)
9514 {
9515   PetscInt i, N = *len;
9516 
9517   for (i = 0; i < N; i++) {
9518     cell_stats_t *A = (cell_stats_t *)a;
9519     cell_stats_t *B = (cell_stats_t *)b;
9520 
9521     B->min = PetscMin(A->min, B->min);
9522     B->max = PetscMax(A->max, B->max);
9523     B->sum += A->sum;
9524     B->squaresum += A->squaresum;
9525     B->count += A->count;
9526   }
9527 }
9528 
9529 /*@
9530   DMPlexCheckCellShape - Checks the Jacobian of the mapping from reference to real cells and computes some minimal statistics.
9531 
9532   Collective
9533 
9534   Input Parameters:
9535 + dm        - The `DMPLEX` object
9536 . output    - If true, statistics will be displayed on `stdout`
9537 - condLimit - Display all cells above this condition number, or `PETSC_DETERMINE` for no cell output
9538 
9539   Level: developer
9540 
9541   Notes:
9542   This is mainly intended for debugging/testing purposes.
9543 
9544   For the complete list of DMPlexCheck* functions, see `DMSetFromOptions()`.
9545 
9546 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexComputeOrthogonalQuality()`
9547 @*/
9548 PetscErrorCode DMPlexCheckCellShape(DM dm, PetscBool output, PetscReal condLimit)
9549 {
9550   DM           dmCoarse;
9551   cell_stats_t stats, globalStats;
9552   MPI_Comm     comm = PetscObjectComm((PetscObject)dm);
9553   PetscReal   *J, *invJ, min = 0, max = 0, mean = 0, stdev = 0;
9554   PetscReal    limit = condLimit > 0 ? condLimit : PETSC_MAX_REAL;
9555   PetscInt     cdim, cStart, cEnd, c, eStart, eEnd, count = 0;
9556   PetscMPIInt  rank, size;
9557 
9558   PetscFunctionBegin;
9559   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9560   stats.min = PETSC_MAX_REAL;
9561   stats.max = PETSC_MIN_REAL;
9562   stats.sum = stats.squaresum = 0.;
9563   stats.count                 = 0;
9564 
9565   PetscCallMPI(MPI_Comm_size(comm, &size));
9566   PetscCallMPI(MPI_Comm_rank(comm, &rank));
9567   PetscCall(DMGetCoordinateDim(dm, &cdim));
9568   PetscCall(PetscMalloc2(PetscSqr(cdim), &J, PetscSqr(cdim), &invJ));
9569   PetscCall(DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd));
9570   PetscCall(DMPlexGetDepthStratum(dm, 1, &eStart, &eEnd));
9571   for (c = cStart; c < cEnd; c++) {
9572     PetscInt  i;
9573     PetscReal frobJ = 0., frobInvJ = 0., cond2, cond, detJ;
9574 
9575     PetscCall(DMPlexComputeCellGeometryAffineFEM(dm, c, NULL, J, invJ, &detJ));
9576     PetscCheck(detJ >= 0.0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh cell %" PetscInt_FMT " is inverted", c);
9577     for (i = 0; i < PetscSqr(cdim); ++i) {
9578       frobJ += J[i] * J[i];
9579       frobInvJ += invJ[i] * invJ[i];
9580     }
9581     cond2 = frobJ * frobInvJ;
9582     cond  = PetscSqrtReal(cond2);
9583 
9584     stats.min = PetscMin(stats.min, cond);
9585     stats.max = PetscMax(stats.max, cond);
9586     stats.sum += cond;
9587     stats.squaresum += cond2;
9588     stats.count++;
9589     if (output && cond > limit) {
9590       PetscSection coordSection;
9591       Vec          coordsLocal;
9592       PetscScalar *coords = NULL;
9593       PetscInt     Nv, d, clSize, cl, *closure = NULL;
9594 
9595       PetscCall(DMGetCoordinatesLocal(dm, &coordsLocal));
9596       PetscCall(DMGetCoordinateSection(dm, &coordSection));
9597       PetscCall(DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, &Nv, &coords));
9598       PetscCall(PetscSynchronizedPrintf(comm, "[%d] Cell %" PetscInt_FMT " cond %g\n", rank, c, (double)cond));
9599       for (i = 0; i < Nv / cdim; ++i) {
9600         PetscCall(PetscSynchronizedPrintf(comm, "  Vertex %" PetscInt_FMT ": (", i));
9601         for (d = 0; d < cdim; ++d) {
9602           if (d > 0) PetscCall(PetscSynchronizedPrintf(comm, ", "));
9603           PetscCall(PetscSynchronizedPrintf(comm, "%g", (double)PetscRealPart(coords[i * cdim + d])));
9604         }
9605         PetscCall(PetscSynchronizedPrintf(comm, ")\n"));
9606       }
9607       PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &clSize, &closure));
9608       for (cl = 0; cl < clSize * 2; cl += 2) {
9609         const PetscInt edge = closure[cl];
9610 
9611         if ((edge >= eStart) && (edge < eEnd)) {
9612           PetscReal len;
9613 
9614           PetscCall(DMPlexComputeCellGeometryFVM(dm, edge, &len, NULL, NULL));
9615           PetscCall(PetscSynchronizedPrintf(comm, "  Edge %" PetscInt_FMT ": length %g\n", edge, (double)len));
9616         }
9617       }
9618       PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &clSize, &closure));
9619       PetscCall(DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, &Nv, &coords));
9620     }
9621   }
9622   if (output) PetscCall(PetscSynchronizedFlush(comm, NULL));
9623 
9624   if (size > 1) {
9625     PetscMPIInt  blockLengths[2] = {4, 1};
9626     MPI_Aint     blockOffsets[2] = {offsetof(cell_stats_t, min), offsetof(cell_stats_t, count)};
9627     MPI_Datatype blockTypes[2]   = {MPIU_REAL, MPIU_INT}, statType;
9628     MPI_Op       statReduce;
9629 
9630     PetscCallMPI(MPI_Type_create_struct(2, blockLengths, blockOffsets, blockTypes, &statType));
9631     PetscCallMPI(MPI_Type_commit(&statType));
9632     PetscCallMPI(MPI_Op_create(cell_stats_reduce, PETSC_TRUE, &statReduce));
9633     PetscCallMPI(MPI_Reduce(&stats, &globalStats, 1, statType, statReduce, 0, comm));
9634     PetscCallMPI(MPI_Op_free(&statReduce));
9635     PetscCallMPI(MPI_Type_free(&statType));
9636   } else {
9637     PetscCall(PetscArraycpy(&globalStats, &stats, 1));
9638   }
9639   if (rank == 0) {
9640     count = globalStats.count;
9641     min   = globalStats.min;
9642     max   = globalStats.max;
9643     mean  = globalStats.sum / globalStats.count;
9644     stdev = globalStats.count > 1 ? PetscSqrtReal(PetscMax((globalStats.squaresum - globalStats.count * mean * mean) / (globalStats.count - 1), 0)) : 0.0;
9645   }
9646 
9647   if (output) PetscCall(PetscPrintf(comm, "Mesh with %" PetscInt_FMT " cells, shape condition numbers: min = %g, max = %g, mean = %g, stddev = %g\n", count, (double)min, (double)max, (double)mean, (double)stdev));
9648   PetscCall(PetscFree2(J, invJ));
9649 
9650   PetscCall(DMGetCoarseDM(dm, &dmCoarse));
9651   if (dmCoarse) {
9652     PetscBool isplex;
9653 
9654     PetscCall(PetscObjectTypeCompare((PetscObject)dmCoarse, DMPLEX, &isplex));
9655     if (isplex) PetscCall(DMPlexCheckCellShape(dmCoarse, output, condLimit));
9656   }
9657   PetscFunctionReturn(PETSC_SUCCESS);
9658 }
9659 
9660 /*@
9661   DMPlexComputeOrthogonalQuality - Compute cell-wise orthogonal quality mesh statistic. Optionally tags all cells with
9662   orthogonal quality below given tolerance.
9663 
9664   Collective
9665 
9666   Input Parameters:
9667 + dm   - The `DMPLEX` object
9668 . fv   - Optional `PetscFV` object for pre-computed cell/face centroid information
9669 - atol - [0, 1] Absolute tolerance for tagging cells.
9670 
9671   Output Parameters:
9672 + OrthQual      - `Vec` containing orthogonal quality per cell
9673 - OrthQualLabel - `DMLabel` tagging cells below atol with `DM_ADAPT_REFINE`
9674 
9675   Options Database Keys:
9676 + -dm_plex_orthogonal_quality_label_view - view OrthQualLabel if label is requested. Currently only `PETSCVIEWERASCII` is supported.
9677 - -dm_plex_orthogonal_quality_vec_view   - view OrthQual vector.
9678 
9679   Level: intermediate
9680 
9681   Notes:
9682   Orthogonal quality is given by the following formula\:
9683 
9684   $ \min \left[ \frac{A_i \cdot f_i}{\|A_i\| \|f_i\|} , \frac{A_i \cdot c_i}{\|A_i\| \|c_i\|} \right]$
9685 
9686   Where A_i is the i'th face-normal vector, f_i is the vector from the cell centroid to the i'th face centroid, and c_i
9687   is the vector from the current cells centroid to the centroid of its i'th neighbor (which shares a face with the
9688   current cell). This computes the vector similarity between each cell face and its corresponding neighbor centroid by
9689   calculating the cosine of the angle between these vectors.
9690 
9691   Orthogonal quality ranges from 1 (best) to 0 (worst).
9692 
9693   This routine is mainly useful for FVM, however is not restricted to only FVM. The `PetscFV` object is optionally used to check for
9694   pre-computed FVM cell data, but if it is not passed in then this data will be computed.
9695 
9696   Cells are tagged if they have an orthogonal quality less than or equal to the absolute tolerance.
9697 
9698 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCheckCellShape()`, `DMCreateLabel()`, `PetscFV`, `DMLabel`, `Vec`
9699 @*/
9700 PetscErrorCode DMPlexComputeOrthogonalQuality(DM dm, PetscFV fv, PetscReal atol, Vec *OrthQual, DMLabel *OrthQualLabel)
9701 {
9702   PetscInt               nc, cellHeight, cStart, cEnd, cell, cellIter = 0;
9703   PetscInt              *idx;
9704   PetscScalar           *oqVals;
9705   const PetscScalar     *cellGeomArr, *faceGeomArr;
9706   PetscReal             *ci, *fi, *Ai;
9707   MPI_Comm               comm;
9708   Vec                    cellgeom, facegeom;
9709   DM                     dmFace, dmCell;
9710   IS                     glob;
9711   ISLocalToGlobalMapping ltog;
9712   PetscViewer            vwr;
9713 
9714   PetscFunctionBegin;
9715   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9716   if (fv) PetscValidHeaderSpecific(fv, PETSCFV_CLASSID, 2);
9717   PetscAssertPointer(OrthQual, 4);
9718   PetscCheck(atol >= 0.0 && atol <= 1.0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Absolute tolerance %g not in [0,1]", (double)atol);
9719   PetscCall(PetscObjectGetComm((PetscObject)dm, &comm));
9720   PetscCall(DMGetDimension(dm, &nc));
9721   PetscCheck(nc >= 2, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "DM must have dimension >= 2 (current %" PetscInt_FMT ")", nc);
9722   {
9723     DMPlexInterpolatedFlag interpFlag;
9724 
9725     PetscCall(DMPlexIsInterpolated(dm, &interpFlag));
9726     if (interpFlag != DMPLEX_INTERPOLATED_FULL) {
9727       PetscMPIInt rank;
9728 
9729       PetscCallMPI(MPI_Comm_rank(comm, &rank));
9730       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "DM must be fully interpolated, DM on rank %d is not fully interpolated", rank);
9731     }
9732   }
9733   if (OrthQualLabel) {
9734     PetscAssertPointer(OrthQualLabel, 5);
9735     PetscCall(DMCreateLabel(dm, "Orthogonal_Quality"));
9736     PetscCall(DMGetLabel(dm, "Orthogonal_Quality", OrthQualLabel));
9737   } else {
9738     *OrthQualLabel = NULL;
9739   }
9740   PetscCall(DMPlexGetVTKCellHeight(dm, &cellHeight));
9741   PetscCall(DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd));
9742   PetscCall(DMPlexCreateCellNumbering_Internal(dm, PETSC_TRUE, &glob));
9743   PetscCall(ISLocalToGlobalMappingCreateIS(glob, &ltog));
9744   PetscCall(ISLocalToGlobalMappingSetType(ltog, ISLOCALTOGLOBALMAPPINGHASH));
9745   PetscCall(VecCreate(comm, OrthQual));
9746   PetscCall(VecSetType(*OrthQual, VECSTANDARD));
9747   PetscCall(VecSetSizes(*OrthQual, cEnd - cStart, PETSC_DETERMINE));
9748   PetscCall(VecSetLocalToGlobalMapping(*OrthQual, ltog));
9749   PetscCall(VecSetUp(*OrthQual));
9750   PetscCall(ISDestroy(&glob));
9751   PetscCall(ISLocalToGlobalMappingDestroy(&ltog));
9752   PetscCall(DMPlexGetDataFVM(dm, fv, &cellgeom, &facegeom, NULL));
9753   PetscCall(VecGetArrayRead(cellgeom, &cellGeomArr));
9754   PetscCall(VecGetArrayRead(facegeom, &faceGeomArr));
9755   PetscCall(VecGetDM(cellgeom, &dmCell));
9756   PetscCall(VecGetDM(facegeom, &dmFace));
9757   PetscCall(PetscMalloc5(cEnd - cStart, &idx, cEnd - cStart, &oqVals, nc, &ci, nc, &fi, nc, &Ai));
9758   for (cell = cStart; cell < cEnd; cellIter++, cell++) {
9759     PetscInt         cellneigh, cellneighiter = 0, adjSize = PETSC_DETERMINE;
9760     PetscInt         cellarr[2], *adj = NULL;
9761     PetscScalar     *cArr, *fArr;
9762     PetscReal        minvalc = 1.0, minvalf = 1.0;
9763     PetscFVCellGeom *cg;
9764 
9765     idx[cellIter] = cell - cStart;
9766     cellarr[0]    = cell;
9767     /* Make indexing into cellGeom easier */
9768     PetscCall(DMPlexPointLocalRead(dmCell, cell, cellGeomArr, &cg));
9769     PetscCall(DMPlexGetAdjacency_Internal(dm, cell, PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, &adjSize, &adj));
9770     /* Technically 1 too big, but easier than fiddling with empty adjacency array */
9771     PetscCall(PetscCalloc2(adjSize, &cArr, adjSize, &fArr));
9772     for (cellneigh = 0; cellneigh < adjSize; cellneighiter++, cellneigh++) {
9773       PetscInt         i;
9774       const PetscInt   neigh  = adj[cellneigh];
9775       PetscReal        normci = 0, normfi = 0, normai = 0;
9776       PetscFVCellGeom *cgneigh;
9777       PetscFVFaceGeom *fg;
9778 
9779       /* Don't count ourselves in the neighbor list */
9780       if (neigh == cell) continue;
9781       PetscCall(DMPlexPointLocalRead(dmCell, neigh, cellGeomArr, &cgneigh));
9782       cellarr[1] = neigh;
9783       {
9784         PetscInt        numcovpts;
9785         const PetscInt *covpts;
9786 
9787         PetscCall(DMPlexGetMeet(dm, 2, cellarr, &numcovpts, &covpts));
9788         PetscCall(DMPlexPointLocalRead(dmFace, covpts[0], faceGeomArr, &fg));
9789         PetscCall(DMPlexRestoreMeet(dm, 2, cellarr, &numcovpts, &covpts));
9790       }
9791 
9792       /* Compute c_i, f_i and their norms */
9793       for (i = 0; i < nc; i++) {
9794         ci[i] = cgneigh->centroid[i] - cg->centroid[i];
9795         fi[i] = fg->centroid[i] - cg->centroid[i];
9796         Ai[i] = fg->normal[i];
9797         normci += PetscPowReal(ci[i], 2);
9798         normfi += PetscPowReal(fi[i], 2);
9799         normai += PetscPowReal(Ai[i], 2);
9800       }
9801       normci = PetscSqrtReal(normci);
9802       normfi = PetscSqrtReal(normfi);
9803       normai = PetscSqrtReal(normai);
9804 
9805       /* Normalize and compute for each face-cell-normal pair */
9806       for (i = 0; i < nc; i++) {
9807         ci[i] = ci[i] / normci;
9808         fi[i] = fi[i] / normfi;
9809         Ai[i] = Ai[i] / normai;
9810         /* PetscAbs because I don't know if normals are guaranteed to point out */
9811         cArr[cellneighiter] += PetscAbs(Ai[i] * ci[i]);
9812         fArr[cellneighiter] += PetscAbs(Ai[i] * fi[i]);
9813       }
9814       if (PetscRealPart(cArr[cellneighiter]) < minvalc) minvalc = PetscRealPart(cArr[cellneighiter]);
9815       if (PetscRealPart(fArr[cellneighiter]) < minvalf) minvalf = PetscRealPart(fArr[cellneighiter]);
9816     }
9817     PetscCall(PetscFree(adj));
9818     PetscCall(PetscFree2(cArr, fArr));
9819     /* Defer to cell if they're equal */
9820     oqVals[cellIter] = PetscMin(minvalf, minvalc);
9821     if (OrthQualLabel) {
9822       if (PetscRealPart(oqVals[cellIter]) <= atol) PetscCall(DMLabelSetValue(*OrthQualLabel, cell, DM_ADAPT_REFINE));
9823     }
9824   }
9825   PetscCall(VecSetValuesLocal(*OrthQual, cEnd - cStart, idx, oqVals, INSERT_VALUES));
9826   PetscCall(VecAssemblyBegin(*OrthQual));
9827   PetscCall(VecAssemblyEnd(*OrthQual));
9828   PetscCall(VecRestoreArrayRead(cellgeom, &cellGeomArr));
9829   PetscCall(VecRestoreArrayRead(facegeom, &faceGeomArr));
9830   PetscCall(PetscOptionsGetViewer(comm, NULL, NULL, "-dm_plex_orthogonal_quality_label_view", &vwr, NULL, NULL));
9831   if (OrthQualLabel) {
9832     if (vwr) PetscCall(DMLabelView(*OrthQualLabel, vwr));
9833   }
9834   PetscCall(PetscFree5(idx, oqVals, ci, fi, Ai));
9835   PetscCall(PetscOptionsRestoreViewer(&vwr));
9836   PetscCall(VecViewFromOptions(*OrthQual, NULL, "-dm_plex_orthogonal_quality_vec_view"));
9837   PetscFunctionReturn(PETSC_SUCCESS);
9838 }
9839 
9840 /* this is here instead of DMGetOutputDM because output DM still has constraints in the local indices that affect
9841  * interpolator construction */
9842 static PetscErrorCode DMGetFullDM(DM dm, DM *odm)
9843 {
9844   PetscSection section, newSection, gsection;
9845   PetscSF      sf;
9846   PetscBool    hasConstraints, ghasConstraints;
9847 
9848   PetscFunctionBegin;
9849   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
9850   PetscAssertPointer(odm, 2);
9851   PetscCall(DMGetLocalSection(dm, &section));
9852   PetscCall(PetscSectionHasConstraints(section, &hasConstraints));
9853   PetscCall(MPIU_Allreduce(&hasConstraints, &ghasConstraints, 1, MPIU_BOOL, MPI_LOR, PetscObjectComm((PetscObject)dm)));
9854   if (!ghasConstraints) {
9855     PetscCall(PetscObjectReference((PetscObject)dm));
9856     *odm = dm;
9857     PetscFunctionReturn(PETSC_SUCCESS);
9858   }
9859   PetscCall(DMClone(dm, odm));
9860   PetscCall(DMCopyFields(dm, *odm));
9861   PetscCall(DMGetLocalSection(*odm, &newSection));
9862   PetscCall(DMGetPointSF(*odm, &sf));
9863   PetscCall(PetscSectionCreateGlobalSection(newSection, sf, PETSC_TRUE, PETSC_FALSE, &gsection));
9864   PetscCall(DMSetGlobalSection(*odm, gsection));
9865   PetscCall(PetscSectionDestroy(&gsection));
9866   PetscFunctionReturn(PETSC_SUCCESS);
9867 }
9868 
9869 static PetscErrorCode DMCreateAffineInterpolationCorrection_Plex(DM dmc, DM dmf, Vec *shift)
9870 {
9871   DM        dmco, dmfo;
9872   Mat       interpo;
9873   Vec       rscale;
9874   Vec       cglobalo, clocal;
9875   Vec       fglobal, fglobalo, flocal;
9876   PetscBool regular;
9877 
9878   PetscFunctionBegin;
9879   PetscCall(DMGetFullDM(dmc, &dmco));
9880   PetscCall(DMGetFullDM(dmf, &dmfo));
9881   PetscCall(DMSetCoarseDM(dmfo, dmco));
9882   PetscCall(DMPlexGetRegularRefinement(dmf, &regular));
9883   PetscCall(DMPlexSetRegularRefinement(dmfo, regular));
9884   PetscCall(DMCreateInterpolation(dmco, dmfo, &interpo, &rscale));
9885   PetscCall(DMCreateGlobalVector(dmco, &cglobalo));
9886   PetscCall(DMCreateLocalVector(dmc, &clocal));
9887   PetscCall(VecSet(cglobalo, 0.));
9888   PetscCall(VecSet(clocal, 0.));
9889   PetscCall(DMCreateGlobalVector(dmf, &fglobal));
9890   PetscCall(DMCreateGlobalVector(dmfo, &fglobalo));
9891   PetscCall(DMCreateLocalVector(dmf, &flocal));
9892   PetscCall(VecSet(fglobal, 0.));
9893   PetscCall(VecSet(fglobalo, 0.));
9894   PetscCall(VecSet(flocal, 0.));
9895   PetscCall(DMPlexInsertBoundaryValues(dmc, PETSC_TRUE, clocal, 0., NULL, NULL, NULL));
9896   PetscCall(DMLocalToGlobalBegin(dmco, clocal, INSERT_VALUES, cglobalo));
9897   PetscCall(DMLocalToGlobalEnd(dmco, clocal, INSERT_VALUES, cglobalo));
9898   PetscCall(MatMult(interpo, cglobalo, fglobalo));
9899   PetscCall(DMGlobalToLocalBegin(dmfo, fglobalo, INSERT_VALUES, flocal));
9900   PetscCall(DMGlobalToLocalEnd(dmfo, fglobalo, INSERT_VALUES, flocal));
9901   PetscCall(DMLocalToGlobalBegin(dmf, flocal, INSERT_VALUES, fglobal));
9902   PetscCall(DMLocalToGlobalEnd(dmf, flocal, INSERT_VALUES, fglobal));
9903   *shift = fglobal;
9904   PetscCall(VecDestroy(&flocal));
9905   PetscCall(VecDestroy(&fglobalo));
9906   PetscCall(VecDestroy(&clocal));
9907   PetscCall(VecDestroy(&cglobalo));
9908   PetscCall(VecDestroy(&rscale));
9909   PetscCall(MatDestroy(&interpo));
9910   PetscCall(DMDestroy(&dmfo));
9911   PetscCall(DMDestroy(&dmco));
9912   PetscFunctionReturn(PETSC_SUCCESS);
9913 }
9914 
9915 PETSC_INTERN PetscErrorCode DMInterpolateSolution_Plex(DM coarse, DM fine, Mat interp, Vec coarseSol, Vec fineSol)
9916 {
9917   PetscObject shifto;
9918   Vec         shift;
9919 
9920   PetscFunctionBegin;
9921   if (!interp) {
9922     Vec rscale;
9923 
9924     PetscCall(DMCreateInterpolation(coarse, fine, &interp, &rscale));
9925     PetscCall(VecDestroy(&rscale));
9926   } else {
9927     PetscCall(PetscObjectReference((PetscObject)interp));
9928   }
9929   PetscCall(PetscObjectQuery((PetscObject)interp, "_DMInterpolateSolution_Plex_Vec", &shifto));
9930   if (!shifto) {
9931     PetscCall(DMCreateAffineInterpolationCorrection_Plex(coarse, fine, &shift));
9932     PetscCall(PetscObjectCompose((PetscObject)interp, "_DMInterpolateSolution_Plex_Vec", (PetscObject)shift));
9933     shifto = (PetscObject)shift;
9934     PetscCall(VecDestroy(&shift));
9935   }
9936   shift = (Vec)shifto;
9937   PetscCall(MatInterpolate(interp, coarseSol, fineSol));
9938   PetscCall(VecAXPY(fineSol, 1.0, shift));
9939   PetscCall(MatDestroy(&interp));
9940   PetscFunctionReturn(PETSC_SUCCESS);
9941 }
9942 
9943 /* Pointwise interpolation
9944      Just code FEM for now
9945      u^f = I u^c
9946      sum_k u^f_k phi^f_k = I sum_j u^c_j phi^c_j
9947      u^f_i = sum_j psi^f_i I phi^c_j u^c_j
9948      I_{ij} = psi^f_i phi^c_j
9949 */
9950 PetscErrorCode DMCreateInterpolation_Plex(DM dmCoarse, DM dmFine, Mat *interpolation, Vec *scaling)
9951 {
9952   PetscSection gsc, gsf;
9953   PetscInt     m, n;
9954   void        *ctx;
9955   DM           cdm;
9956   PetscBool    regular, ismatis, isRefined = dmCoarse->data == dmFine->data ? PETSC_FALSE : PETSC_TRUE;
9957 
9958   PetscFunctionBegin;
9959   PetscCall(DMGetGlobalSection(dmFine, &gsf));
9960   PetscCall(PetscSectionGetConstrainedStorageSize(gsf, &m));
9961   PetscCall(DMGetGlobalSection(dmCoarse, &gsc));
9962   PetscCall(PetscSectionGetConstrainedStorageSize(gsc, &n));
9963 
9964   PetscCall(PetscStrcmp(dmCoarse->mattype, MATIS, &ismatis));
9965   PetscCall(MatCreate(PetscObjectComm((PetscObject)dmCoarse), interpolation));
9966   PetscCall(MatSetSizes(*interpolation, m, n, PETSC_DETERMINE, PETSC_DETERMINE));
9967   PetscCall(MatSetType(*interpolation, ismatis ? MATAIJ : dmCoarse->mattype));
9968   PetscCall(DMGetApplicationContext(dmFine, &ctx));
9969 
9970   PetscCall(DMGetCoarseDM(dmFine, &cdm));
9971   PetscCall(DMPlexGetRegularRefinement(dmFine, &regular));
9972   if (!isRefined || (regular && cdm == dmCoarse)) PetscCall(DMPlexComputeInterpolatorNested(dmCoarse, dmFine, isRefined, *interpolation, ctx));
9973   else PetscCall(DMPlexComputeInterpolatorGeneral(dmCoarse, dmFine, *interpolation, ctx));
9974   PetscCall(MatViewFromOptions(*interpolation, NULL, "-interp_mat_view"));
9975   if (scaling) {
9976     /* Use naive scaling */
9977     PetscCall(DMCreateInterpolationScale(dmCoarse, dmFine, *interpolation, scaling));
9978   }
9979   PetscFunctionReturn(PETSC_SUCCESS);
9980 }
9981 
9982 PetscErrorCode DMCreateInjection_Plex(DM dmCoarse, DM dmFine, Mat *mat)
9983 {
9984   VecScatter ctx;
9985 
9986   PetscFunctionBegin;
9987   PetscCall(DMPlexComputeInjectorFEM(dmCoarse, dmFine, &ctx, NULL));
9988   PetscCall(MatCreateScatter(PetscObjectComm((PetscObject)ctx), ctx, mat));
9989   PetscCall(VecScatterDestroy(&ctx));
9990   PetscFunctionReturn(PETSC_SUCCESS);
9991 }
9992 
9993 static void g0_identity_private(PetscInt dim, PetscInt Nf, PetscInt NfAux, const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[])
9994 {
9995   const PetscInt Nc = uOff[1] - uOff[0];
9996   PetscInt       c;
9997   for (c = 0; c < Nc; ++c) g0[c * Nc + c] = 1.0;
9998 }
9999 
10000 PetscErrorCode DMCreateMassMatrixLumped_Plex(DM dm, Vec *mass)
10001 {
10002   DM           dmc;
10003   PetscDS      ds;
10004   Vec          ones, locmass;
10005   IS           cellIS;
10006   PetscFormKey key;
10007   PetscInt     depth;
10008 
10009   PetscFunctionBegin;
10010   PetscCall(DMClone(dm, &dmc));
10011   PetscCall(DMCopyDisc(dm, dmc));
10012   PetscCall(DMGetDS(dmc, &ds));
10013   PetscCall(PetscDSSetJacobian(ds, 0, 0, g0_identity_private, NULL, NULL, NULL));
10014   PetscCall(DMCreateGlobalVector(dmc, mass));
10015   PetscCall(DMGetLocalVector(dmc, &ones));
10016   PetscCall(DMGetLocalVector(dmc, &locmass));
10017   PetscCall(DMPlexGetDepth(dmc, &depth));
10018   PetscCall(DMGetStratumIS(dmc, "depth", depth, &cellIS));
10019   PetscCall(VecSet(locmass, 0.0));
10020   PetscCall(VecSet(ones, 1.0));
10021   key.label = NULL;
10022   key.value = 0;
10023   key.field = 0;
10024   key.part  = 0;
10025   PetscCall(DMPlexComputeJacobian_Action_Internal(dmc, key, cellIS, 0.0, 0.0, ones, NULL, ones, locmass, NULL));
10026   PetscCall(ISDestroy(&cellIS));
10027   PetscCall(VecSet(*mass, 0.0));
10028   PetscCall(DMLocalToGlobalBegin(dmc, locmass, ADD_VALUES, *mass));
10029   PetscCall(DMLocalToGlobalEnd(dmc, locmass, ADD_VALUES, *mass));
10030   PetscCall(DMRestoreLocalVector(dmc, &ones));
10031   PetscCall(DMRestoreLocalVector(dmc, &locmass));
10032   PetscCall(DMDestroy(&dmc));
10033   PetscFunctionReturn(PETSC_SUCCESS);
10034 }
10035 
10036 PetscErrorCode DMCreateMassMatrix_Plex(DM dmCoarse, DM dmFine, Mat *mass)
10037 {
10038   PetscSection gsc, gsf;
10039   PetscInt     m, n;
10040   void        *ctx;
10041   DM           cdm;
10042   PetscBool    regular;
10043 
10044   PetscFunctionBegin;
10045   if (dmFine == dmCoarse) {
10046     DM            dmc;
10047     PetscDS       ds;
10048     PetscWeakForm wf;
10049     Vec           u;
10050     IS            cellIS;
10051     PetscFormKey  key;
10052     PetscInt      depth;
10053 
10054     PetscCall(DMClone(dmFine, &dmc));
10055     PetscCall(DMCopyDisc(dmFine, dmc));
10056     PetscCall(DMGetDS(dmc, &ds));
10057     PetscCall(PetscDSGetWeakForm(ds, &wf));
10058     PetscCall(PetscWeakFormClear(wf));
10059     PetscCall(PetscDSSetJacobian(ds, 0, 0, g0_identity_private, NULL, NULL, NULL));
10060     PetscCall(DMCreateMatrix(dmc, mass));
10061     PetscCall(DMGetLocalVector(dmc, &u));
10062     PetscCall(DMPlexGetDepth(dmc, &depth));
10063     PetscCall(DMGetStratumIS(dmc, "depth", depth, &cellIS));
10064     PetscCall(MatZeroEntries(*mass));
10065     key.label = NULL;
10066     key.value = 0;
10067     key.field = 0;
10068     key.part  = 0;
10069     PetscCall(DMPlexComputeJacobian_Internal(dmc, key, cellIS, 0.0, 0.0, u, NULL, *mass, *mass, NULL));
10070     PetscCall(ISDestroy(&cellIS));
10071     PetscCall(DMRestoreLocalVector(dmc, &u));
10072     PetscCall(DMDestroy(&dmc));
10073   } else {
10074     PetscCall(DMGetGlobalSection(dmFine, &gsf));
10075     PetscCall(PetscSectionGetConstrainedStorageSize(gsf, &m));
10076     PetscCall(DMGetGlobalSection(dmCoarse, &gsc));
10077     PetscCall(PetscSectionGetConstrainedStorageSize(gsc, &n));
10078 
10079     PetscCall(MatCreate(PetscObjectComm((PetscObject)dmCoarse), mass));
10080     PetscCall(MatSetSizes(*mass, m, n, PETSC_DETERMINE, PETSC_DETERMINE));
10081     PetscCall(MatSetType(*mass, dmCoarse->mattype));
10082     PetscCall(DMGetApplicationContext(dmFine, &ctx));
10083 
10084     PetscCall(DMGetCoarseDM(dmFine, &cdm));
10085     PetscCall(DMPlexGetRegularRefinement(dmFine, &regular));
10086     if (regular && cdm == dmCoarse) PetscCall(DMPlexComputeMassMatrixNested(dmCoarse, dmFine, *mass, ctx));
10087     else PetscCall(DMPlexComputeMassMatrixGeneral(dmCoarse, dmFine, *mass, ctx));
10088   }
10089   PetscCall(MatViewFromOptions(*mass, NULL, "-mass_mat_view"));
10090   PetscFunctionReturn(PETSC_SUCCESS);
10091 }
10092 
10093 /*@
10094   DMPlexGetRegularRefinement - Get the flag indicating that this mesh was obtained by regular refinement from its coarse mesh
10095 
10096   Input Parameter:
10097 . dm - The `DMPLEX` object
10098 
10099   Output Parameter:
10100 . regular - The flag
10101 
10102   Level: intermediate
10103 
10104 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexSetRegularRefinement()`
10105 @*/
10106 PetscErrorCode DMPlexGetRegularRefinement(DM dm, PetscBool *regular)
10107 {
10108   PetscFunctionBegin;
10109   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10110   PetscAssertPointer(regular, 2);
10111   *regular = ((DM_Plex *)dm->data)->regularRefinement;
10112   PetscFunctionReturn(PETSC_SUCCESS);
10113 }
10114 
10115 /*@
10116   DMPlexSetRegularRefinement - Set the flag indicating that this mesh was obtained by regular refinement from its coarse mesh
10117 
10118   Input Parameters:
10119 + dm      - The `DMPLEX` object
10120 - regular - The flag
10121 
10122   Level: intermediate
10123 
10124 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetRegularRefinement()`
10125 @*/
10126 PetscErrorCode DMPlexSetRegularRefinement(DM dm, PetscBool regular)
10127 {
10128   PetscFunctionBegin;
10129   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10130   ((DM_Plex *)dm->data)->regularRefinement = regular;
10131   PetscFunctionReturn(PETSC_SUCCESS);
10132 }
10133 
10134 /*@
10135   DMPlexGetAnchors - Get the layout of the anchor (point-to-point) constraints.  Typically, the user will not have to
10136   call DMPlexGetAnchors() directly: if there are anchors, then `DMPlexGetAnchors()` is called during `DMGetDefaultConstraints()`.
10137 
10138   Not Collective
10139 
10140   Input Parameter:
10141 . dm - The `DMPLEX` object
10142 
10143   Output Parameters:
10144 + anchorSection - If not `NULL`, set to the section describing which points anchor the constrained points.
10145 - anchorIS      - If not `NULL`, set to the list of anchors indexed by `anchorSection`
10146 
10147   Level: intermediate
10148 
10149 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexSetAnchors()`, `DMGetDefaultConstraints()`, `DMSetDefaultConstraints()`, `IS`, `PetscSection`
10150 @*/
10151 PetscErrorCode DMPlexGetAnchors(DM dm, PetscSection *anchorSection, IS *anchorIS)
10152 {
10153   DM_Plex *plex = (DM_Plex *)dm->data;
10154 
10155   PetscFunctionBegin;
10156   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10157   if (!plex->anchorSection && !plex->anchorIS && plex->createanchors) PetscCall((*plex->createanchors)(dm));
10158   if (anchorSection) *anchorSection = plex->anchorSection;
10159   if (anchorIS) *anchorIS = plex->anchorIS;
10160   PetscFunctionReturn(PETSC_SUCCESS);
10161 }
10162 
10163 /*@
10164   DMPlexSetAnchors - Set the layout of the local anchor (point-to-point) constraints.
10165 
10166   Collective
10167 
10168   Input Parameters:
10169 + dm            - The `DMPLEX` object
10170 . anchorSection - The section that describes the mapping from constrained points to the anchor points listed in anchorIS.
10171                   Must have a local communicator (`PETSC_COMM_SELF` or derivative).
10172 - anchorIS      - The list of all anchor points.  Must have a local communicator (`PETSC_COMM_SELF` or derivative).
10173 
10174   Level: intermediate
10175 
10176   Notes:
10177   Unlike boundary conditions, when a point's degrees of freedom in a section are constrained to
10178   an outside value, the anchor constraints set a point's degrees of freedom to be a linear
10179   combination of other points' degrees of freedom.
10180 
10181   After specifying the layout of constraints with `DMPlexSetAnchors()`, one specifies the constraints by calling
10182   `DMGetDefaultConstraints()` and filling in the entries in the constraint matrix.
10183 
10184   The reference counts of `anchorSection` and `anchorIS` are incremented.
10185 
10186 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexGetAnchors()`, `DMGetDefaultConstraints()`, `DMSetDefaultConstraints()`
10187 @*/
10188 PetscErrorCode DMPlexSetAnchors(DM dm, PetscSection anchorSection, IS anchorIS)
10189 {
10190   DM_Plex    *plex = (DM_Plex *)dm->data;
10191   PetscMPIInt result;
10192 
10193   PetscFunctionBegin;
10194   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10195   if (anchorSection) {
10196     PetscValidHeaderSpecific(anchorSection, PETSC_SECTION_CLASSID, 2);
10197     PetscCallMPI(MPI_Comm_compare(PETSC_COMM_SELF, PetscObjectComm((PetscObject)anchorSection), &result));
10198     PetscCheck(result == MPI_CONGRUENT || result == MPI_IDENT, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMECOMM, "anchor section must have local communicator");
10199   }
10200   if (anchorIS) {
10201     PetscValidHeaderSpecific(anchorIS, IS_CLASSID, 3);
10202     PetscCallMPI(MPI_Comm_compare(PETSC_COMM_SELF, PetscObjectComm((PetscObject)anchorIS), &result));
10203     PetscCheck(result == MPI_CONGRUENT || result == MPI_IDENT, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMECOMM, "anchor IS must have local communicator");
10204   }
10205 
10206   PetscCall(PetscObjectReference((PetscObject)anchorSection));
10207   PetscCall(PetscSectionDestroy(&plex->anchorSection));
10208   plex->anchorSection = anchorSection;
10209 
10210   PetscCall(PetscObjectReference((PetscObject)anchorIS));
10211   PetscCall(ISDestroy(&plex->anchorIS));
10212   plex->anchorIS = anchorIS;
10213 
10214   if (PetscUnlikelyDebug(anchorIS && anchorSection)) {
10215     PetscInt        size, a, pStart, pEnd;
10216     const PetscInt *anchors;
10217 
10218     PetscCall(PetscSectionGetChart(anchorSection, &pStart, &pEnd));
10219     PetscCall(ISGetLocalSize(anchorIS, &size));
10220     PetscCall(ISGetIndices(anchorIS, &anchors));
10221     for (a = 0; a < size; a++) {
10222       PetscInt p;
10223 
10224       p = anchors[a];
10225       if (p >= pStart && p < pEnd) {
10226         PetscInt dof;
10227 
10228         PetscCall(PetscSectionGetDof(anchorSection, p, &dof));
10229         if (dof) {
10230           PetscCall(ISRestoreIndices(anchorIS, &anchors));
10231           SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Point %" PetscInt_FMT " cannot be constrained and an anchor", p);
10232         }
10233       }
10234     }
10235     PetscCall(ISRestoreIndices(anchorIS, &anchors));
10236   }
10237   /* reset the generic constraints */
10238   PetscCall(DMSetDefaultConstraints(dm, NULL, NULL, NULL));
10239   PetscFunctionReturn(PETSC_SUCCESS);
10240 }
10241 
10242 static PetscErrorCode DMPlexCreateConstraintSection_Anchors(DM dm, PetscSection section, PetscSection *cSec)
10243 {
10244   PetscSection anchorSection;
10245   PetscInt     pStart, pEnd, sStart, sEnd, p, dof, numFields, f;
10246 
10247   PetscFunctionBegin;
10248   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10249   PetscCall(DMPlexGetAnchors(dm, &anchorSection, NULL));
10250   PetscCall(PetscSectionCreate(PETSC_COMM_SELF, cSec));
10251   PetscCall(PetscSectionGetNumFields(section, &numFields));
10252   if (numFields) {
10253     PetscInt f;
10254     PetscCall(PetscSectionSetNumFields(*cSec, numFields));
10255 
10256     for (f = 0; f < numFields; f++) {
10257       PetscInt numComp;
10258 
10259       PetscCall(PetscSectionGetFieldComponents(section, f, &numComp));
10260       PetscCall(PetscSectionSetFieldComponents(*cSec, f, numComp));
10261     }
10262   }
10263   PetscCall(PetscSectionGetChart(anchorSection, &pStart, &pEnd));
10264   PetscCall(PetscSectionGetChart(section, &sStart, &sEnd));
10265   pStart = PetscMax(pStart, sStart);
10266   pEnd   = PetscMin(pEnd, sEnd);
10267   pEnd   = PetscMax(pStart, pEnd);
10268   PetscCall(PetscSectionSetChart(*cSec, pStart, pEnd));
10269   for (p = pStart; p < pEnd; p++) {
10270     PetscCall(PetscSectionGetDof(anchorSection, p, &dof));
10271     if (dof) {
10272       PetscCall(PetscSectionGetDof(section, p, &dof));
10273       PetscCall(PetscSectionSetDof(*cSec, p, dof));
10274       for (f = 0; f < numFields; f++) {
10275         PetscCall(PetscSectionGetFieldDof(section, p, f, &dof));
10276         PetscCall(PetscSectionSetFieldDof(*cSec, p, f, dof));
10277       }
10278     }
10279   }
10280   PetscCall(PetscSectionSetUp(*cSec));
10281   PetscCall(PetscObjectSetName((PetscObject)*cSec, "Constraint Section"));
10282   PetscFunctionReturn(PETSC_SUCCESS);
10283 }
10284 
10285 static PetscErrorCode DMPlexCreateConstraintMatrix_Anchors(DM dm, PetscSection section, PetscSection cSec, Mat *cMat)
10286 {
10287   PetscSection    aSec;
10288   PetscInt        pStart, pEnd, p, sStart, sEnd, dof, aDof, aOff, off, nnz, annz, m, n, q, a, offset, *i, *j;
10289   const PetscInt *anchors;
10290   PetscInt        numFields, f;
10291   IS              aIS;
10292   MatType         mtype;
10293   PetscBool       iscuda, iskokkos;
10294 
10295   PetscFunctionBegin;
10296   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10297   PetscCall(PetscSectionGetStorageSize(cSec, &m));
10298   PetscCall(PetscSectionGetStorageSize(section, &n));
10299   PetscCall(MatCreate(PETSC_COMM_SELF, cMat));
10300   PetscCall(MatSetSizes(*cMat, m, n, m, n));
10301   PetscCall(PetscStrcmp(dm->mattype, MATSEQAIJCUSPARSE, &iscuda));
10302   if (!iscuda) PetscCall(PetscStrcmp(dm->mattype, MATMPIAIJCUSPARSE, &iscuda));
10303   PetscCall(PetscStrcmp(dm->mattype, MATSEQAIJKOKKOS, &iskokkos));
10304   if (!iskokkos) PetscCall(PetscStrcmp(dm->mattype, MATMPIAIJKOKKOS, &iskokkos));
10305   if (iscuda) mtype = MATSEQAIJCUSPARSE;
10306   else if (iskokkos) mtype = MATSEQAIJKOKKOS;
10307   else mtype = MATSEQAIJ;
10308   PetscCall(MatSetType(*cMat, mtype));
10309   PetscCall(DMPlexGetAnchors(dm, &aSec, &aIS));
10310   PetscCall(ISGetIndices(aIS, &anchors));
10311   /* cSec will be a subset of aSec and section */
10312   PetscCall(PetscSectionGetChart(cSec, &pStart, &pEnd));
10313   PetscCall(PetscSectionGetChart(section, &sStart, &sEnd));
10314   PetscCall(PetscMalloc1(m + 1, &i));
10315   i[0] = 0;
10316   PetscCall(PetscSectionGetNumFields(section, &numFields));
10317   for (p = pStart; p < pEnd; p++) {
10318     PetscInt rDof, rOff, r;
10319 
10320     PetscCall(PetscSectionGetDof(aSec, p, &rDof));
10321     if (!rDof) continue;
10322     PetscCall(PetscSectionGetOffset(aSec, p, &rOff));
10323     if (numFields) {
10324       for (f = 0; f < numFields; f++) {
10325         annz = 0;
10326         for (r = 0; r < rDof; r++) {
10327           a = anchors[rOff + r];
10328           if (a < sStart || a >= sEnd) continue;
10329           PetscCall(PetscSectionGetFieldDof(section, a, f, &aDof));
10330           annz += aDof;
10331         }
10332         PetscCall(PetscSectionGetFieldDof(cSec, p, f, &dof));
10333         PetscCall(PetscSectionGetFieldOffset(cSec, p, f, &off));
10334         for (q = 0; q < dof; q++) i[off + q + 1] = i[off + q] + annz;
10335       }
10336     } else {
10337       annz = 0;
10338       PetscCall(PetscSectionGetDof(cSec, p, &dof));
10339       for (q = 0; q < dof; q++) {
10340         a = anchors[rOff + q];
10341         if (a < sStart || a >= sEnd) continue;
10342         PetscCall(PetscSectionGetDof(section, a, &aDof));
10343         annz += aDof;
10344       }
10345       PetscCall(PetscSectionGetDof(cSec, p, &dof));
10346       PetscCall(PetscSectionGetOffset(cSec, p, &off));
10347       for (q = 0; q < dof; q++) i[off + q + 1] = i[off + q] + annz;
10348     }
10349   }
10350   nnz = i[m];
10351   PetscCall(PetscMalloc1(nnz, &j));
10352   offset = 0;
10353   for (p = pStart; p < pEnd; p++) {
10354     if (numFields) {
10355       for (f = 0; f < numFields; f++) {
10356         PetscCall(PetscSectionGetFieldDof(cSec, p, f, &dof));
10357         for (q = 0; q < dof; q++) {
10358           PetscInt rDof, rOff, r;
10359           PetscCall(PetscSectionGetDof(aSec, p, &rDof));
10360           PetscCall(PetscSectionGetOffset(aSec, p, &rOff));
10361           for (r = 0; r < rDof; r++) {
10362             PetscInt s;
10363 
10364             a = anchors[rOff + r];
10365             if (a < sStart || a >= sEnd) continue;
10366             PetscCall(PetscSectionGetFieldDof(section, a, f, &aDof));
10367             PetscCall(PetscSectionGetFieldOffset(section, a, f, &aOff));
10368             for (s = 0; s < aDof; s++) j[offset++] = aOff + s;
10369           }
10370         }
10371       }
10372     } else {
10373       PetscCall(PetscSectionGetDof(cSec, p, &dof));
10374       for (q = 0; q < dof; q++) {
10375         PetscInt rDof, rOff, r;
10376         PetscCall(PetscSectionGetDof(aSec, p, &rDof));
10377         PetscCall(PetscSectionGetOffset(aSec, p, &rOff));
10378         for (r = 0; r < rDof; r++) {
10379           PetscInt s;
10380 
10381           a = anchors[rOff + r];
10382           if (a < sStart || a >= sEnd) continue;
10383           PetscCall(PetscSectionGetDof(section, a, &aDof));
10384           PetscCall(PetscSectionGetOffset(section, a, &aOff));
10385           for (s = 0; s < aDof; s++) j[offset++] = aOff + s;
10386         }
10387       }
10388     }
10389   }
10390   PetscCall(MatSeqAIJSetPreallocationCSR(*cMat, i, j, NULL));
10391   PetscCall(PetscFree(i));
10392   PetscCall(PetscFree(j));
10393   PetscCall(ISRestoreIndices(aIS, &anchors));
10394   PetscFunctionReturn(PETSC_SUCCESS);
10395 }
10396 
10397 PetscErrorCode DMCreateDefaultConstraints_Plex(DM dm)
10398 {
10399   DM_Plex     *plex = (DM_Plex *)dm->data;
10400   PetscSection anchorSection, section, cSec;
10401   Mat          cMat;
10402 
10403   PetscFunctionBegin;
10404   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10405   PetscCall(DMPlexGetAnchors(dm, &anchorSection, NULL));
10406   if (anchorSection) {
10407     PetscInt Nf;
10408 
10409     PetscCall(DMGetLocalSection(dm, &section));
10410     PetscCall(DMPlexCreateConstraintSection_Anchors(dm, section, &cSec));
10411     PetscCall(DMPlexCreateConstraintMatrix_Anchors(dm, section, cSec, &cMat));
10412     PetscCall(DMGetNumFields(dm, &Nf));
10413     if (Nf && plex->computeanchormatrix) PetscCall((*plex->computeanchormatrix)(dm, section, cSec, cMat));
10414     PetscCall(DMSetDefaultConstraints(dm, cSec, cMat, NULL));
10415     PetscCall(PetscSectionDestroy(&cSec));
10416     PetscCall(MatDestroy(&cMat));
10417   }
10418   PetscFunctionReturn(PETSC_SUCCESS);
10419 }
10420 
10421 PetscErrorCode DMCreateSubDomainDM_Plex(DM dm, DMLabel label, PetscInt value, IS *is, DM *subdm)
10422 {
10423   IS           subis;
10424   PetscSection section, subsection;
10425 
10426   PetscFunctionBegin;
10427   PetscCall(DMGetLocalSection(dm, &section));
10428   PetscCheck(section, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Must set default section for DM before splitting subdomain");
10429   PetscCheck(subdm, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Must set output subDM for splitting subdomain");
10430   /* Create subdomain */
10431   PetscCall(DMPlexFilter(dm, label, value, subdm));
10432   /* Create submodel */
10433   PetscCall(DMPlexGetSubpointIS(*subdm, &subis));
10434   PetscCall(PetscSectionCreateSubmeshSection(section, subis, &subsection));
10435   PetscCall(DMSetLocalSection(*subdm, subsection));
10436   PetscCall(PetscSectionDestroy(&subsection));
10437   PetscCall(DMCopyDisc(dm, *subdm));
10438   /* Create map from submodel to global model */
10439   if (is) {
10440     PetscSection    sectionGlobal, subsectionGlobal;
10441     IS              spIS;
10442     const PetscInt *spmap;
10443     PetscInt       *subIndices;
10444     PetscInt        subSize = 0, subOff = 0, pStart, pEnd, p;
10445     PetscInt        Nf, f, bs = -1, bsLocal[2], bsMinMax[2];
10446 
10447     PetscCall(DMPlexGetSubpointIS(*subdm, &spIS));
10448     PetscCall(ISGetIndices(spIS, &spmap));
10449     PetscCall(PetscSectionGetNumFields(section, &Nf));
10450     PetscCall(DMGetGlobalSection(dm, &sectionGlobal));
10451     PetscCall(DMGetGlobalSection(*subdm, &subsectionGlobal));
10452     PetscCall(PetscSectionGetChart(subsection, &pStart, &pEnd));
10453     for (p = pStart; p < pEnd; ++p) {
10454       PetscInt gdof, pSubSize = 0;
10455 
10456       PetscCall(PetscSectionGetDof(sectionGlobal, p, &gdof));
10457       if (gdof > 0) {
10458         for (f = 0; f < Nf; ++f) {
10459           PetscInt fdof, fcdof;
10460 
10461           PetscCall(PetscSectionGetFieldDof(subsection, p, f, &fdof));
10462           PetscCall(PetscSectionGetFieldConstraintDof(subsection, p, f, &fcdof));
10463           pSubSize += fdof - fcdof;
10464         }
10465         subSize += pSubSize;
10466         if (pSubSize) {
10467           if (bs < 0) {
10468             bs = pSubSize;
10469           } else if (bs != pSubSize) {
10470             /* Layout does not admit a pointwise block size */
10471             bs = 1;
10472           }
10473         }
10474       }
10475     }
10476     /* Must have same blocksize on all procs (some might have no points) */
10477     bsLocal[0] = bs < 0 ? PETSC_MAX_INT : bs;
10478     bsLocal[1] = bs;
10479     PetscCall(PetscGlobalMinMaxInt(PetscObjectComm((PetscObject)dm), bsLocal, bsMinMax));
10480     if (bsMinMax[0] != bsMinMax[1]) {
10481       bs = 1;
10482     } else {
10483       bs = bsMinMax[0];
10484     }
10485     PetscCall(PetscMalloc1(subSize, &subIndices));
10486     for (p = pStart; p < pEnd; ++p) {
10487       PetscInt gdof, goff;
10488 
10489       PetscCall(PetscSectionGetDof(subsectionGlobal, p, &gdof));
10490       if (gdof > 0) {
10491         const PetscInt point = spmap[p];
10492 
10493         PetscCall(PetscSectionGetOffset(sectionGlobal, point, &goff));
10494         for (f = 0; f < Nf; ++f) {
10495           PetscInt fdof, fcdof, fc, f2, poff = 0;
10496 
10497           /* Can get rid of this loop by storing field information in the global section */
10498           for (f2 = 0; f2 < f; ++f2) {
10499             PetscCall(PetscSectionGetFieldDof(section, p, f2, &fdof));
10500             PetscCall(PetscSectionGetFieldConstraintDof(section, p, f2, &fcdof));
10501             poff += fdof - fcdof;
10502           }
10503           PetscCall(PetscSectionGetFieldDof(section, p, f, &fdof));
10504           PetscCall(PetscSectionGetFieldConstraintDof(section, p, f, &fcdof));
10505           for (fc = 0; fc < fdof - fcdof; ++fc, ++subOff) subIndices[subOff] = goff + poff + fc;
10506         }
10507       }
10508     }
10509     PetscCall(ISRestoreIndices(spIS, &spmap));
10510     PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), subSize, subIndices, PETSC_OWN_POINTER, is));
10511     if (bs > 1) {
10512       /* We need to check that the block size does not come from non-contiguous fields */
10513       PetscInt i, j, set = 1;
10514       for (i = 0; i < subSize; i += bs) {
10515         for (j = 0; j < bs; ++j) {
10516           if (subIndices[i + j] != subIndices[i] + j) {
10517             set = 0;
10518             break;
10519           }
10520         }
10521       }
10522       if (set) PetscCall(ISSetBlockSize(*is, bs));
10523     }
10524     /* Attach nullspace */
10525     for (f = 0; f < Nf; ++f) {
10526       (*subdm)->nullspaceConstructors[f] = dm->nullspaceConstructors[f];
10527       if ((*subdm)->nullspaceConstructors[f]) break;
10528     }
10529     if (f < Nf) {
10530       MatNullSpace nullSpace;
10531       PetscCall((*(*subdm)->nullspaceConstructors[f])(*subdm, f, f, &nullSpace));
10532 
10533       PetscCall(PetscObjectCompose((PetscObject)*is, "nullspace", (PetscObject)nullSpace));
10534       PetscCall(MatNullSpaceDestroy(&nullSpace));
10535     }
10536   }
10537   PetscFunctionReturn(PETSC_SUCCESS);
10538 }
10539 
10540 /*@
10541   DMPlexMonitorThroughput - Report the cell throughput of FE integration
10542 
10543   Input Parameters:
10544 + dm    - The `DM`
10545 - dummy - unused argument
10546 
10547   Options Database Key:
10548 . -dm_plex_monitor_throughput - Activate the monitor
10549 
10550   Level: developer
10551 
10552 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMSetFromOptions()`, `DMPlexCreate()`
10553 @*/
10554 PetscErrorCode DMPlexMonitorThroughput(DM dm, void *dummy)
10555 {
10556   PetscLogHandler default_handler;
10557 
10558   PetscFunctionBegin;
10559   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
10560   PetscCall(PetscLogGetDefaultHandler(&default_handler));
10561   if (default_handler) {
10562     PetscLogEvent      event;
10563     PetscEventPerfInfo eventInfo;
10564     PetscReal          cellRate, flopRate;
10565     PetscInt           cStart, cEnd, Nf, N;
10566     const char        *name;
10567 
10568     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
10569     PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
10570     PetscCall(DMGetNumFields(dm, &Nf));
10571     PetscCall(PetscLogEventGetId("DMPlexResidualFE", &event));
10572     PetscCall(PetscLogEventGetPerfInfo(PETSC_DEFAULT, event, &eventInfo));
10573     N        = (cEnd - cStart) * Nf * eventInfo.count;
10574     flopRate = eventInfo.flops / eventInfo.time;
10575     cellRate = N / eventInfo.time;
10576     PetscCall(PetscPrintf(PetscObjectComm((PetscObject)dm), "DM (%s) FE Residual Integration: %" PetscInt_FMT " integrals %d reps\n  Cell rate: %.2g/s flop rate: %.2g MF/s\n", name ? name : "unknown", N, eventInfo.count, (double)cellRate, (double)(flopRate / 1.e6)));
10577   } else {
10578     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Plex Throughput Monitor is not supported if logging is turned off or the default log handler is not running. Reconfigure using --with-log and run with -log_view.");
10579   }
10580   PetscFunctionReturn(PETSC_SUCCESS);
10581 }
10582