xref: /petsc/src/dm/impls/forest/p4est/pforest.h (revision 2827ebad7ef18d4ed3f6d7633d71fb685bb29d6f)
10a96aa3bSJed Brown #include <petscds.h>
20a96aa3bSJed Brown #include <petsc/private/dmimpl.h>
30a96aa3bSJed Brown #include <petsc/private/dmforestimpl.h>
40a96aa3bSJed Brown #include <petsc/private/dmpleximpl.h>
50a96aa3bSJed Brown #include <petsc/private/dmlabelimpl.h>
60a96aa3bSJed Brown #include <petsc/private/viewerimpl.h>
70a96aa3bSJed Brown #include <../src/sys/classes/viewer/impls/vtk/vtkvimpl.h>
80a96aa3bSJed Brown #include "petsc_p4est_package.h"
90a96aa3bSJed Brown 
100a96aa3bSJed Brown #if defined(PETSC_HAVE_P4EST)
110a96aa3bSJed Brown 
120a96aa3bSJed Brown   #if !defined(P4_TO_P8)
130a96aa3bSJed Brown     #include <p4est.h>
140a96aa3bSJed Brown     #include <p4est_extended.h>
150a96aa3bSJed Brown     #include <p4est_geometry.h>
160a96aa3bSJed Brown     #include <p4est_ghost.h>
170a96aa3bSJed Brown     #include <p4est_lnodes.h>
180a96aa3bSJed Brown     #include <p4est_vtk.h>
190a96aa3bSJed Brown     #include <p4est_plex.h>
200a96aa3bSJed Brown     #include <p4est_bits.h>
210a96aa3bSJed Brown     #include <p4est_algorithms.h>
220a96aa3bSJed Brown   #else
230a96aa3bSJed Brown     #include <p8est.h>
240a96aa3bSJed Brown     #include <p8est_extended.h>
250a96aa3bSJed Brown     #include <p8est_geometry.h>
260a96aa3bSJed Brown     #include <p8est_ghost.h>
270a96aa3bSJed Brown     #include <p8est_lnodes.h>
280a96aa3bSJed Brown     #include <p8est_vtk.h>
290a96aa3bSJed Brown     #include <p8est_plex.h>
300a96aa3bSJed Brown     #include <p8est_bits.h>
310a96aa3bSJed Brown     #include <p8est_algorithms.h>
320a96aa3bSJed Brown   #endif
330a96aa3bSJed Brown 
349371c9d4SSatish Balay typedef enum {
359371c9d4SSatish Balay   PATTERN_HASH,
369371c9d4SSatish Balay   PATTERN_FRACTAL,
379371c9d4SSatish Balay   PATTERN_CORNER,
389371c9d4SSatish Balay   PATTERN_CENTER,
399371c9d4SSatish Balay   PATTERN_COUNT
409371c9d4SSatish Balay } DMRefinePattern;
410a96aa3bSJed Brown static const char *DMRefinePatternName[PATTERN_COUNT] = {"hash", "fractal", "corner", "center"};
420a96aa3bSJed Brown 
439371c9d4SSatish Balay typedef struct _DMRefinePatternCtx {
440a96aa3bSJed Brown   PetscInt       corner;
450a96aa3bSJed Brown   PetscBool      fractal[P4EST_CHILDREN];
460a96aa3bSJed Brown   PetscReal      hashLikelihood;
470a96aa3bSJed Brown   PetscInt       maxLevel;
480a96aa3bSJed Brown   p4est_refine_t refine_fn;
499371c9d4SSatish Balay } DMRefinePatternCtx;
500a96aa3bSJed Brown 
51d71ae5a4SJacob Faibussowitsch static int DMRefinePattern_Corner(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
52d71ae5a4SJacob Faibussowitsch {
530a96aa3bSJed Brown   p4est_quadrant_t    root, rootcorner;
540a96aa3bSJed Brown   DMRefinePatternCtx *ctx;
550a96aa3bSJed Brown 
560a96aa3bSJed Brown   ctx = (DMRefinePatternCtx *)p4est->user_pointer;
570a96aa3bSJed Brown   if (quadrant->level >= ctx->maxLevel) return 0;
580a96aa3bSJed Brown 
590a96aa3bSJed Brown   root.x = root.y = 0;
600a96aa3bSJed Brown   #if defined(P4_TO_P8)
610a96aa3bSJed Brown   root.z = 0;
620a96aa3bSJed Brown   #endif
630a96aa3bSJed Brown   root.level = 0;
640a96aa3bSJed Brown   p4est_quadrant_corner_descendant(&root, &rootcorner, ctx->corner, quadrant->level);
650a96aa3bSJed Brown   if (p4est_quadrant_is_equal(quadrant, &rootcorner)) return 1;
660a96aa3bSJed Brown   return 0;
670a96aa3bSJed Brown }
680a96aa3bSJed Brown 
69d71ae5a4SJacob Faibussowitsch static int DMRefinePattern_Center(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
70d71ae5a4SJacob Faibussowitsch {
710a96aa3bSJed Brown   int                 cid;
720a96aa3bSJed Brown   p4est_quadrant_t    ancestor, ancestorcorner;
730a96aa3bSJed Brown   DMRefinePatternCtx *ctx;
740a96aa3bSJed Brown 
750a96aa3bSJed Brown   ctx = (DMRefinePatternCtx *)p4est->user_pointer;
760a96aa3bSJed Brown   if (quadrant->level >= ctx->maxLevel) return 0;
770a96aa3bSJed Brown   if (quadrant->level <= 1) return 1;
780a96aa3bSJed Brown 
790a96aa3bSJed Brown   p4est_quadrant_ancestor(quadrant, 1, &ancestor);
800a96aa3bSJed Brown   cid = p4est_quadrant_child_id(&ancestor);
810a96aa3bSJed Brown   p4est_quadrant_corner_descendant(&ancestor, &ancestorcorner, P4EST_CHILDREN - 1 - cid, quadrant->level);
820a96aa3bSJed Brown   if (p4est_quadrant_is_equal(quadrant, &ancestorcorner)) return 1;
830a96aa3bSJed Brown   return 0;
840a96aa3bSJed Brown }
850a96aa3bSJed Brown 
86d71ae5a4SJacob Faibussowitsch static int DMRefinePattern_Fractal(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
87d71ae5a4SJacob Faibussowitsch {
880a96aa3bSJed Brown   int                 cid;
890a96aa3bSJed Brown   DMRefinePatternCtx *ctx;
900a96aa3bSJed Brown 
910a96aa3bSJed Brown   ctx = (DMRefinePatternCtx *)p4est->user_pointer;
920a96aa3bSJed Brown   if (quadrant->level >= ctx->maxLevel) return 0;
930a96aa3bSJed Brown   if (!quadrant->level) return 1;
940a96aa3bSJed Brown   cid = p4est_quadrant_child_id(quadrant);
950a96aa3bSJed Brown   if (ctx->fractal[cid ^ ((int)(quadrant->level % P4EST_CHILDREN))]) return 1;
960a96aa3bSJed Brown   return 0;
970a96aa3bSJed Brown }
980a96aa3bSJed Brown 
990a96aa3bSJed Brown   /* simplified from MurmurHash3 by Austin Appleby */
1000a96aa3bSJed Brown   #define DMPROT32(x, y) ((x << y) | (x >> (32 - y)))
101d71ae5a4SJacob Faibussowitsch static uint32_t DMPforestHash(const uint32_t *blocks, uint32_t nblocks)
102d71ae5a4SJacob Faibussowitsch {
1030a96aa3bSJed Brown   uint32_t c1   = 0xcc9e2d51;
1040a96aa3bSJed Brown   uint32_t c2   = 0x1b873593;
1050a96aa3bSJed Brown   uint32_t r1   = 15;
1060a96aa3bSJed Brown   uint32_t r2   = 13;
1070a96aa3bSJed Brown   uint32_t m    = 5;
1080a96aa3bSJed Brown   uint32_t n    = 0xe6546b64;
1090a96aa3bSJed Brown   uint32_t hash = 0;
1100a96aa3bSJed Brown   int      len  = nblocks * 4;
1110a96aa3bSJed Brown   uint32_t i;
1120a96aa3bSJed Brown 
1130a96aa3bSJed Brown   for (i = 0; i < nblocks; i++) {
1140a96aa3bSJed Brown     uint32_t k;
1150a96aa3bSJed Brown 
1160a96aa3bSJed Brown     k = blocks[i];
1170a96aa3bSJed Brown     k *= c1;
1180a96aa3bSJed Brown     k = DMPROT32(k, r1);
1190a96aa3bSJed Brown     k *= c2;
1200a96aa3bSJed Brown 
1210a96aa3bSJed Brown     hash ^= k;
1220a96aa3bSJed Brown     hash = DMPROT32(hash, r2) * m + n;
1230a96aa3bSJed Brown   }
1240a96aa3bSJed Brown 
1250a96aa3bSJed Brown   hash ^= len;
1260a96aa3bSJed Brown   hash ^= (hash >> 16);
1270a96aa3bSJed Brown   hash *= 0x85ebca6b;
1280a96aa3bSJed Brown   hash ^= (hash >> 13);
1290a96aa3bSJed Brown   hash *= 0xc2b2ae35;
1300a96aa3bSJed Brown   hash ^= (hash >> 16);
1310a96aa3bSJed Brown 
1320a96aa3bSJed Brown   return hash;
1330a96aa3bSJed Brown }
1340a96aa3bSJed Brown 
1350a96aa3bSJed Brown   #if defined(UINT32_MAX)
1360a96aa3bSJed Brown     #define DMP4EST_HASH_MAX UINT32_MAX
1370a96aa3bSJed Brown   #else
1380a96aa3bSJed Brown     #define DMP4EST_HASH_MAX ((uint32_t)0xffffffff)
1390a96aa3bSJed Brown   #endif
1400a96aa3bSJed Brown 
141d71ae5a4SJacob Faibussowitsch static int DMRefinePattern_Hash(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
142d71ae5a4SJacob Faibussowitsch {
1430a96aa3bSJed Brown   uint32_t            data[5];
1440a96aa3bSJed Brown   uint32_t            result;
1450a96aa3bSJed Brown   DMRefinePatternCtx *ctx;
1460a96aa3bSJed Brown 
1470a96aa3bSJed Brown   ctx = (DMRefinePatternCtx *)p4est->user_pointer;
1480a96aa3bSJed Brown   if (quadrant->level >= ctx->maxLevel) return 0;
1490a96aa3bSJed Brown   data[0] = ((uint32_t)quadrant->level) << 24;
1500a96aa3bSJed Brown   data[1] = (uint32_t)which_tree;
1510a96aa3bSJed Brown   data[2] = (uint32_t)quadrant->x;
1520a96aa3bSJed Brown   data[3] = (uint32_t)quadrant->y;
1530a96aa3bSJed Brown   #if defined(P4_TO_P8)
1540a96aa3bSJed Brown   data[4] = (uint32_t)quadrant->z;
1550a96aa3bSJed Brown   #endif
1560a96aa3bSJed Brown 
1570a96aa3bSJed Brown   result = DMPforestHash(data, 2 + P4EST_DIM);
1580a96aa3bSJed Brown   if (((double)result / (double)DMP4EST_HASH_MAX) < ctx->hashLikelihood) return 1;
1590a96aa3bSJed Brown   return 0;
1600a96aa3bSJed Brown }
1610a96aa3bSJed Brown 
1620a96aa3bSJed Brown   #define DMConvert_pforest_plex _infix_pforest(DMConvert, _plex)
1630a96aa3bSJed Brown static PetscErrorCode DMConvert_pforest_plex(DM, DMType, DM *);
1640a96aa3bSJed Brown 
1650a96aa3bSJed Brown   #define DMFTopology_pforest _append_pforest(DMFTopology)
1660a96aa3bSJed Brown typedef struct {
1670a96aa3bSJed Brown   PetscInt              refct;
1680a96aa3bSJed Brown   p4est_connectivity_t *conn;
1690a96aa3bSJed Brown   p4est_geometry_t     *geom;
1700a96aa3bSJed Brown   PetscInt             *tree_face_to_uniq; /* p4est does not explicitly enumerate facets, but we must to keep track of labels */
1710a96aa3bSJed Brown } DMFTopology_pforest;
1720a96aa3bSJed Brown 
1730a96aa3bSJed Brown   #define DM_Forest_pforest _append_pforest(DM_Forest)
1740a96aa3bSJed Brown typedef struct {
1750a96aa3bSJed Brown   DMFTopology_pforest *topo;
1760a96aa3bSJed Brown   p4est_t             *forest;
1770a96aa3bSJed Brown   p4est_ghost_t       *ghost;
1780a96aa3bSJed Brown   p4est_lnodes_t      *lnodes;
1790a96aa3bSJed Brown   PetscBool            partition_for_coarsening;
1800a96aa3bSJed Brown   PetscBool            coarsen_hierarchy;
1810a96aa3bSJed Brown   PetscBool            labelsFinalized;
1820a96aa3bSJed Brown   PetscBool            adaptivitySuccess;
1830a96aa3bSJed Brown   PetscInt             cLocalStart;
1840a96aa3bSJed Brown   PetscInt             cLocalEnd;
1850a96aa3bSJed Brown   DM                   plex;
1860a96aa3bSJed Brown   char                *ghostName;
1870a96aa3bSJed Brown   PetscSF              pointAdaptToSelfSF;
1880a96aa3bSJed Brown   PetscSF              pointSelfToAdaptSF;
1890a96aa3bSJed Brown   PetscInt            *pointAdaptToSelfCids;
1900a96aa3bSJed Brown   PetscInt            *pointSelfToAdaptCids;
1910a96aa3bSJed Brown } DM_Forest_pforest;
1920a96aa3bSJed Brown 
1930a96aa3bSJed Brown   #define DM_Forest_geometry_pforest _append_pforest(DM_Forest_geometry)
1940a96aa3bSJed Brown typedef struct {
1950a96aa3bSJed Brown   DM base;
1960a96aa3bSJed Brown   PetscErrorCode (*map)(DM, PetscInt, PetscInt, const PetscReal[], PetscReal[], void *);
1970a96aa3bSJed Brown   void             *mapCtx;
1980a96aa3bSJed Brown   PetscInt          coordDim;
1990a96aa3bSJed Brown   p4est_geometry_t *inner;
2009371c9d4SSatish Balay } DM_Forest_geometry_pforest;
2010a96aa3bSJed Brown 
2020a96aa3bSJed Brown   #define GeometryMapping_pforest _append_pforest(GeometryMapping)
203d71ae5a4SJacob Faibussowitsch static void GeometryMapping_pforest(p4est_geometry_t *geom, p4est_topidx_t which_tree, const double abc[3], double xyz[3])
204d71ae5a4SJacob Faibussowitsch {
2050a96aa3bSJed Brown   DM_Forest_geometry_pforest *geom_pforest = (DM_Forest_geometry_pforest *)geom->user;
2060a96aa3bSJed Brown   PetscReal                   PetscABC[3]  = {0.};
2070a96aa3bSJed Brown   PetscReal                   PetscXYZ[3]  = {0.};
2080a96aa3bSJed Brown   PetscInt                    i, d = PetscMin(3, geom_pforest->coordDim);
2090a96aa3bSJed Brown   double                      ABC[3];
2100a96aa3bSJed Brown   PetscErrorCode              ierr;
2110a96aa3bSJed Brown 
2120a96aa3bSJed Brown   (geom_pforest->inner->X)(geom_pforest->inner, which_tree, abc, ABC);
2130a96aa3bSJed Brown 
2140a96aa3bSJed Brown   for (i = 0; i < d; i++) PetscABC[i] = ABC[i];
2159371c9d4SSatish Balay   ierr = (geom_pforest->map)(geom_pforest->base, (PetscInt)which_tree, geom_pforest->coordDim, PetscABC, PetscXYZ, geom_pforest->mapCtx);
2169371c9d4SSatish Balay   PETSC_P4EST_ASSERT(!ierr);
2170a96aa3bSJed Brown   for (i = 0; i < d; i++) xyz[i] = PetscXYZ[i];
2180a96aa3bSJed Brown }
2190a96aa3bSJed Brown 
2200a96aa3bSJed Brown   #define GeometryDestroy_pforest _append_pforest(GeometryDestroy)
221d71ae5a4SJacob Faibussowitsch static void GeometryDestroy_pforest(p4est_geometry_t *geom)
222d71ae5a4SJacob Faibussowitsch {
2230a96aa3bSJed Brown   DM_Forest_geometry_pforest *geom_pforest = (DM_Forest_geometry_pforest *)geom->user;
2240a96aa3bSJed Brown   PetscErrorCode              ierr;
2250a96aa3bSJed Brown 
2260a96aa3bSJed Brown   p4est_geometry_destroy(geom_pforest->inner);
2279371c9d4SSatish Balay   ierr = PetscFree(geom->user);
2289371c9d4SSatish Balay   PETSC_P4EST_ASSERT(!ierr);
2299371c9d4SSatish Balay   ierr = PetscFree(geom);
2309371c9d4SSatish Balay   PETSC_P4EST_ASSERT(!ierr);
2310a96aa3bSJed Brown }
2320a96aa3bSJed Brown 
2330a96aa3bSJed Brown   #define DMFTopologyDestroy_pforest _append_pforest(DMFTopologyDestroy)
234d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMFTopologyDestroy_pforest(DMFTopology_pforest **topo)
235d71ae5a4SJacob Faibussowitsch {
2360a96aa3bSJed Brown   PetscFunctionBegin;
2373ba16761SJacob Faibussowitsch   if (!(*topo)) PetscFunctionReturn(PETSC_SUCCESS);
2380a96aa3bSJed Brown   if (--((*topo)->refct) > 0) {
2390a96aa3bSJed Brown     *topo = NULL;
2403ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
2410a96aa3bSJed Brown   }
242792fecdfSBarry Smith   if ((*topo)->geom) PetscCallP4est(p4est_geometry_destroy, ((*topo)->geom));
243792fecdfSBarry Smith   PetscCallP4est(p4est_connectivity_destroy, ((*topo)->conn));
2449566063dSJacob Faibussowitsch   PetscCall(PetscFree((*topo)->tree_face_to_uniq));
2459566063dSJacob Faibussowitsch   PetscCall(PetscFree(*topo));
2460a96aa3bSJed Brown   *topo = NULL;
2473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2480a96aa3bSJed Brown }
2490a96aa3bSJed Brown 
2500a96aa3bSJed Brown static PetscErrorCode PforestConnectivityEnumerateFacets(p4est_connectivity_t *, PetscInt **);
2510a96aa3bSJed Brown 
2520a96aa3bSJed Brown   #define DMFTopologyCreateBrick_pforest _append_pforest(DMFTopologyCreateBrick)
253d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMFTopologyCreateBrick_pforest(DM dm, PetscInt N[], PetscInt P[], PetscReal B[], DMFTopology_pforest **topo, PetscBool useMorton)
254d71ae5a4SJacob Faibussowitsch {
2550a96aa3bSJed Brown   double  *vertices;
2560a96aa3bSJed Brown   PetscInt i, numVerts;
2570a96aa3bSJed Brown 
2580a96aa3bSJed Brown   PetscFunctionBegin;
25928b400f6SJacob Faibussowitsch   PetscCheck(useMorton, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Lexicographic ordering not implemented yet");
2604dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(topo));
2610a96aa3bSJed Brown 
2620a96aa3bSJed Brown   (*topo)->refct = 1;
2630a96aa3bSJed Brown   #if !defined(P4_TO_P8)
264792fecdfSBarry Smith   PetscCallP4estReturn((*topo)->conn, p4est_connectivity_new_brick, ((int)N[0], (int)N[1], (P[0] == DM_BOUNDARY_NONE) ? 0 : 1, (P[1] == DM_BOUNDARY_NONE) ? 0 : 1));
2650a96aa3bSJed Brown   #else
266792fecdfSBarry Smith   PetscCallP4estReturn((*topo)->conn, p8est_connectivity_new_brick, ((int)N[0], (int)N[1], (int)N[2], (P[0] == DM_BOUNDARY_NONE) ? 0 : 1, (P[1] == DM_BOUNDARY_NONE) ? 0 : 1, (P[2] == DM_BOUNDARY_NONE) ? 0 : 1));
2670a96aa3bSJed Brown   #endif
2680a96aa3bSJed Brown   numVerts = (*topo)->conn->num_vertices;
2690a96aa3bSJed Brown   vertices = (*topo)->conn->vertices;
2700a96aa3bSJed Brown   for (i = 0; i < 3 * numVerts; i++) {
2710a96aa3bSJed Brown     PetscInt j = i % 3;
2720a96aa3bSJed Brown 
2730a96aa3bSJed Brown     vertices[i] = B[2 * j] + (vertices[i] / N[j]) * (B[2 * j + 1] - B[2 * j]);
2740a96aa3bSJed Brown   }
2750a96aa3bSJed Brown   (*topo)->geom = NULL;
2769566063dSJacob Faibussowitsch   PetscCall(PforestConnectivityEnumerateFacets((*topo)->conn, &(*topo)->tree_face_to_uniq));
2773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2780a96aa3bSJed Brown }
2790a96aa3bSJed Brown 
2800a96aa3bSJed Brown   #define DMFTopologyCreate_pforest _append_pforest(DMFTopologyCreate)
281d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMFTopologyCreate_pforest(DM dm, DMForestTopology topologyName, DMFTopology_pforest **topo)
282d71ae5a4SJacob Faibussowitsch {
2830a96aa3bSJed Brown   const char *name = (const char *)topologyName;
2840a96aa3bSJed Brown   const char *prefix;
2850a96aa3bSJed Brown   PetscBool   isBrick, isShell, isSphere, isMoebius;
2860a96aa3bSJed Brown 
2870a96aa3bSJed Brown   PetscFunctionBegin;
2880a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2890a96aa3bSJed Brown   PetscValidCharPointer(name, 2);
2900a96aa3bSJed Brown   PetscValidPointer(topo, 3);
2919566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(name, "brick", &isBrick));
2929566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(name, "shell", &isShell));
2939566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(name, "sphere", &isSphere));
2949566063dSJacob Faibussowitsch   PetscCall(PetscStrcmp(name, "moebius", &isMoebius));
2959566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
2960a96aa3bSJed Brown   if (isBrick) {
2970a96aa3bSJed Brown     PetscBool flgN, flgP, flgM, flgB, useMorton = PETSC_TRUE, periodic = PETSC_FALSE;
2980a96aa3bSJed Brown     PetscInt  N[3] = {2, 2, 2}, P[3] = {0, 0, 0}, nretN = P4EST_DIM, nretP = P4EST_DIM, nretB = 2 * P4EST_DIM, i;
2994fb89dddSMatthew G. Knepley     PetscReal B[6] = {0.0, 1.0, 0.0, 1.0, 0.0, 1.0}, Lstart[3] = {0., 0., 0.}, L[3] = {-1.0, -1.0, -1.0}, maxCell[3] = {-1.0, -1.0, -1.0};
3000a96aa3bSJed Brown 
3010a96aa3bSJed Brown     if (dm->setfromoptionscalled) {
3029566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetIntArray(((PetscObject)dm)->options, prefix, "-dm_p4est_brick_size", N, &nretN, &flgN));
3039566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetIntArray(((PetscObject)dm)->options, prefix, "-dm_p4est_brick_periodicity", P, &nretP, &flgP));
3049566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetRealArray(((PetscObject)dm)->options, prefix, "-dm_p4est_brick_bounds", B, &nretB, &flgB));
3059566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, prefix, "-dm_p4est_brick_use_morton_curve", &useMorton, &flgM));
3061dca8a05SBarry Smith       PetscCheck(!flgN || nretN == P4EST_DIM, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Need to give %d sizes in -dm_p4est_brick_size, gave %" PetscInt_FMT, P4EST_DIM, nretN);
3071dca8a05SBarry Smith       PetscCheck(!flgP || nretP == P4EST_DIM, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Need to give %d periodicities in -dm_p4est_brick_periodicity, gave %" PetscInt_FMT, P4EST_DIM, nretP);
3081dca8a05SBarry Smith       PetscCheck(!flgB || nretB == 2 * P4EST_DIM, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_SIZ, "Need to give %d bounds in -dm_p4est_brick_bounds, gave %" PetscInt_FMT, P4EST_DIM, nretP);
3090a96aa3bSJed Brown     }
3100a96aa3bSJed Brown     for (i = 0; i < P4EST_DIM; i++) {
3110a96aa3bSJed Brown       P[i]     = (P[i] ? DM_BOUNDARY_PERIODIC : DM_BOUNDARY_NONE);
3120a96aa3bSJed Brown       periodic = (PetscBool)(P[i] || periodic);
3130a96aa3bSJed Brown       if (!flgB) B[2 * i + 1] = N[i];
3146858538eSMatthew G. Knepley       if (P[i]) {
3154fb89dddSMatthew G. Knepley         Lstart[i]  = B[2 * i + 0];
3164fb89dddSMatthew G. Knepley         L[i]       = B[2 * i + 1] - B[2 * i + 0];
3176858538eSMatthew G. Knepley         maxCell[i] = 1.1 * (L[i] / N[i]);
3186858538eSMatthew G. Knepley       }
3190a96aa3bSJed Brown     }
3209566063dSJacob Faibussowitsch     PetscCall(DMFTopologyCreateBrick_pforest(dm, N, P, B, topo, useMorton));
3214fb89dddSMatthew G. Knepley     if (periodic) PetscCall(DMSetPeriodicity(dm, maxCell, Lstart, L));
3220a96aa3bSJed Brown   } else {
3234dfa11a4SJacob Faibussowitsch     PetscCall(PetscNew(topo));
3240a96aa3bSJed Brown 
3250a96aa3bSJed Brown     (*topo)->refct = 1;
326792fecdfSBarry Smith     PetscCallP4estReturn((*topo)->conn, p4est_connectivity_new_byname, (name));
3270a96aa3bSJed Brown     (*topo)->geom = NULL;
3281baa6e33SBarry Smith     if (isMoebius) PetscCall(DMSetCoordinateDim(dm, 3));
3290a96aa3bSJed Brown   #if defined(P4_TO_P8)
3300a96aa3bSJed Brown     if (isShell) {
3310a96aa3bSJed Brown       PetscReal R2 = 1., R1 = .55;
3320a96aa3bSJed Brown 
3330a96aa3bSJed Brown       if (dm->setfromoptionscalled) {
3349566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_shell_outer_radius", &R2, NULL));
3359566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_shell_inner_radius", &R1, NULL));
3360a96aa3bSJed Brown       }
337792fecdfSBarry Smith       PetscCallP4estReturn((*topo)->geom, p8est_geometry_new_shell, ((*topo)->conn, R2, R1));
3380a96aa3bSJed Brown     } else if (isSphere) {
3390a96aa3bSJed Brown       PetscReal R2 = 1., R1 = 0.191728, R0 = 0.039856;
3400a96aa3bSJed Brown 
3410a96aa3bSJed Brown       if (dm->setfromoptionscalled) {
3429566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_sphere_outer_radius", &R2, NULL));
3439566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_sphere_inner_radius", &R1, NULL));
3449566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_sphere_core_radius", &R0, NULL));
3450a96aa3bSJed Brown       }
346792fecdfSBarry Smith       PetscCallP4estReturn((*topo)->geom, p8est_geometry_new_sphere, ((*topo)->conn, R2, R1, R0));
3470a96aa3bSJed Brown     }
3480a96aa3bSJed Brown   #endif
3499566063dSJacob Faibussowitsch     PetscCall(PforestConnectivityEnumerateFacets((*topo)->conn, &(*topo)->tree_face_to_uniq));
3500a96aa3bSJed Brown   }
3513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3520a96aa3bSJed Brown }
3530a96aa3bSJed Brown 
3540a96aa3bSJed Brown   #define DMConvert_plex_pforest _append_pforest(DMConvert_plex)
355d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMConvert_plex_pforest(DM dm, DMType newtype, DM *pforest)
356d71ae5a4SJacob Faibussowitsch {
3570a96aa3bSJed Brown   MPI_Comm  comm;
3580a96aa3bSJed Brown   PetscBool isPlex;
3590a96aa3bSJed Brown   PetscInt  dim;
3600a96aa3bSJed Brown   void     *ctx;
3610a96aa3bSJed Brown 
3620a96aa3bSJed Brown   PetscFunctionBegin;
3630a96aa3bSJed Brown 
3640a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3650a96aa3bSJed Brown   comm = PetscObjectComm((PetscObject)dm);
3669566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)dm, DMPLEX, &isPlex));
36728b400f6SJacob Faibussowitsch   PetscCheck(isPlex, comm, PETSC_ERR_ARG_WRONG, "Expected DM type %s, got %s", DMPLEX, ((PetscObject)dm)->type_name);
3689566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
36963a3b9bcSJacob Faibussowitsch   PetscCheck(dim == P4EST_DIM, comm, PETSC_ERR_ARG_WRONG, "Expected DM dimension %d, got %" PetscInt_FMT, P4EST_DIM, dim);
3709566063dSJacob Faibussowitsch   PetscCall(DMCreate(comm, pforest));
3719566063dSJacob Faibussowitsch   PetscCall(DMSetType(*pforest, DMPFOREST));
3729566063dSJacob Faibussowitsch   PetscCall(DMForestSetBaseDM(*pforest, dm));
3739566063dSJacob Faibussowitsch   PetscCall(DMGetApplicationContext(dm, &ctx));
3749566063dSJacob Faibussowitsch   PetscCall(DMSetApplicationContext(*pforest, ctx));
3759566063dSJacob Faibussowitsch   PetscCall(DMCopyDisc(dm, *pforest));
3763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3770a96aa3bSJed Brown }
3780a96aa3bSJed Brown 
3790a96aa3bSJed Brown   #define DMForestDestroy_pforest _append_pforest(DMForestDestroy)
380d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestDestroy_pforest(DM dm)
381d71ae5a4SJacob Faibussowitsch {
3820a96aa3bSJed Brown   DM_Forest         *forest  = (DM_Forest *)dm->data;
3830a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)forest->data;
3840a96aa3bSJed Brown 
3850a96aa3bSJed Brown   PetscFunctionBegin;
3860a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
387792fecdfSBarry Smith   if (pforest->lnodes) PetscCallP4est(p4est_lnodes_destroy, (pforest->lnodes));
3880a96aa3bSJed Brown   pforest->lnodes = NULL;
389792fecdfSBarry Smith   if (pforest->ghost) PetscCallP4est(p4est_ghost_destroy, (pforest->ghost));
3900a96aa3bSJed Brown   pforest->ghost = NULL;
391792fecdfSBarry Smith   if (pforest->forest) PetscCallP4est(p4est_destroy, (pforest->forest));
3920a96aa3bSJed Brown   pforest->forest = NULL;
3939566063dSJacob Faibussowitsch   PetscCall(DMFTopologyDestroy_pforest(&pforest->topo));
3949566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, PetscStringize(DMConvert_plex_pforest) "_C", NULL));
3959566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, PetscStringize(DMConvert_pforest_plex) "_C", NULL));
3962e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMSetUpGLVisViewer_C", NULL));
3979566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", NULL));
3982e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", NULL));
3992e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "MatComputeNeumannOverlap_C", NULL));
4009566063dSJacob Faibussowitsch   PetscCall(PetscFree(pforest->ghostName));
4019566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&pforest->plex));
4029566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&pforest->pointAdaptToSelfSF));
4039566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&pforest->pointSelfToAdaptSF));
4049566063dSJacob Faibussowitsch   PetscCall(PetscFree(pforest->pointAdaptToSelfCids));
4059566063dSJacob Faibussowitsch   PetscCall(PetscFree(pforest->pointSelfToAdaptCids));
4069566063dSJacob Faibussowitsch   PetscCall(PetscFree(forest->data));
4073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4080a96aa3bSJed Brown }
4090a96aa3bSJed Brown 
4100a96aa3bSJed Brown   #define DMForestTemplate_pforest _append_pforest(DMForestTemplate)
411d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestTemplate_pforest(DM dm, DM tdm)
412d71ae5a4SJacob Faibussowitsch {
4130a96aa3bSJed Brown   DM_Forest_pforest *pforest  = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
4140a96aa3bSJed Brown   DM_Forest_pforest *tpforest = (DM_Forest_pforest *)((DM_Forest *)tdm->data)->data;
4150a96aa3bSJed Brown 
4160a96aa3bSJed Brown   PetscFunctionBegin;
4170a96aa3bSJed Brown   if (pforest->topo) pforest->topo->refct++;
4189566063dSJacob Faibussowitsch   PetscCall(DMFTopologyDestroy_pforest(&(tpforest->topo)));
4190a96aa3bSJed Brown   tpforest->topo = pforest->topo;
4203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4210a96aa3bSJed Brown }
4220a96aa3bSJed Brown 
4230a96aa3bSJed Brown   #define DMPlexCreateConnectivity_pforest _append_pforest(DMPlexCreateConnectivity)
4240a96aa3bSJed Brown static PetscErrorCode DMPlexCreateConnectivity_pforest(DM, p4est_connectivity_t **, PetscInt **);
4250a96aa3bSJed Brown 
4269371c9d4SSatish Balay typedef struct _PforestAdaptCtx {
4270a96aa3bSJed Brown   PetscInt  maxLevel;
4280a96aa3bSJed Brown   PetscInt  minLevel;
4290a96aa3bSJed Brown   PetscInt  currLevel;
4300a96aa3bSJed Brown   PetscBool anyChange;
4319371c9d4SSatish Balay } PforestAdaptCtx;
4320a96aa3bSJed Brown 
433d71ae5a4SJacob Faibussowitsch static int pforest_coarsen_currlevel(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrants[])
434d71ae5a4SJacob Faibussowitsch {
4350a96aa3bSJed Brown   PforestAdaptCtx *ctx       = (PforestAdaptCtx *)p4est->user_pointer;
4360a96aa3bSJed Brown   PetscInt         minLevel  = ctx->minLevel;
4370a96aa3bSJed Brown   PetscInt         currLevel = ctx->currLevel;
4380a96aa3bSJed Brown 
4390a96aa3bSJed Brown   if (quadrants[0]->level <= minLevel) return 0;
4400a96aa3bSJed Brown   return (int)((PetscInt)quadrants[0]->level == currLevel);
4410a96aa3bSJed Brown }
4420a96aa3bSJed Brown 
443d71ae5a4SJacob Faibussowitsch static int pforest_coarsen_uniform(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrants[])
444d71ae5a4SJacob Faibussowitsch {
4450a96aa3bSJed Brown   PforestAdaptCtx *ctx      = (PforestAdaptCtx *)p4est->user_pointer;
4460a96aa3bSJed Brown   PetscInt         minLevel = ctx->minLevel;
4470a96aa3bSJed Brown 
4480a96aa3bSJed Brown   return (int)((PetscInt)quadrants[0]->level > minLevel);
4490a96aa3bSJed Brown }
4500a96aa3bSJed Brown 
451d71ae5a4SJacob Faibussowitsch static int pforest_coarsen_flag_any(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrants[])
452d71ae5a4SJacob Faibussowitsch {
4530a96aa3bSJed Brown   PetscInt         i;
4540a96aa3bSJed Brown   PetscBool        any      = PETSC_FALSE;
4550a96aa3bSJed Brown   PforestAdaptCtx *ctx      = (PforestAdaptCtx *)p4est->user_pointer;
4560a96aa3bSJed Brown   PetscInt         minLevel = ctx->minLevel;
4570a96aa3bSJed Brown 
4580a96aa3bSJed Brown   if (quadrants[0]->level <= minLevel) return 0;
4590a96aa3bSJed Brown   for (i = 0; i < P4EST_CHILDREN; i++) {
4600a96aa3bSJed Brown     if (quadrants[i]->p.user_int == DM_ADAPT_KEEP) {
4610a96aa3bSJed Brown       any = PETSC_FALSE;
4620a96aa3bSJed Brown       break;
4630a96aa3bSJed Brown     }
4640a96aa3bSJed Brown     if (quadrants[i]->p.user_int == DM_ADAPT_COARSEN) {
4650a96aa3bSJed Brown       any = PETSC_TRUE;
4660a96aa3bSJed Brown       break;
4670a96aa3bSJed Brown     }
4680a96aa3bSJed Brown   }
4690a96aa3bSJed Brown   return any ? 1 : 0;
4700a96aa3bSJed Brown }
4710a96aa3bSJed Brown 
472d71ae5a4SJacob Faibussowitsch static int pforest_coarsen_flag_all(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrants[])
473d71ae5a4SJacob Faibussowitsch {
4740a96aa3bSJed Brown   PetscInt         i;
4750a96aa3bSJed Brown   PetscBool        all      = PETSC_TRUE;
4760a96aa3bSJed Brown   PforestAdaptCtx *ctx      = (PforestAdaptCtx *)p4est->user_pointer;
4770a96aa3bSJed Brown   PetscInt         minLevel = ctx->minLevel;
4780a96aa3bSJed Brown 
4790a96aa3bSJed Brown   if (quadrants[0]->level <= minLevel) return 0;
4800a96aa3bSJed Brown   for (i = 0; i < P4EST_CHILDREN; i++) {
4810a96aa3bSJed Brown     if (quadrants[i]->p.user_int != DM_ADAPT_COARSEN) {
4820a96aa3bSJed Brown       all = PETSC_FALSE;
4830a96aa3bSJed Brown       break;
4840a96aa3bSJed Brown     }
4850a96aa3bSJed Brown   }
4860a96aa3bSJed Brown   return all ? 1 : 0;
4870a96aa3bSJed Brown }
4880a96aa3bSJed Brown 
489d71ae5a4SJacob Faibussowitsch static void pforest_init_determine(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
490d71ae5a4SJacob Faibussowitsch {
4910a96aa3bSJed Brown   quadrant->p.user_int = DM_ADAPT_DETERMINE;
4920a96aa3bSJed Brown }
4930a96aa3bSJed Brown 
494d71ae5a4SJacob Faibussowitsch static int pforest_refine_uniform(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
495d71ae5a4SJacob Faibussowitsch {
4960a96aa3bSJed Brown   PforestAdaptCtx *ctx      = (PforestAdaptCtx *)p4est->user_pointer;
4970a96aa3bSJed Brown   PetscInt         maxLevel = ctx->maxLevel;
4980a96aa3bSJed Brown 
4990a96aa3bSJed Brown   return ((PetscInt)quadrant->level < maxLevel);
5000a96aa3bSJed Brown }
5010a96aa3bSJed Brown 
502d71ae5a4SJacob Faibussowitsch static int pforest_refine_flag(p4est_t *p4est, p4est_topidx_t which_tree, p4est_quadrant_t *quadrant)
503d71ae5a4SJacob Faibussowitsch {
5040a96aa3bSJed Brown   PforestAdaptCtx *ctx      = (PforestAdaptCtx *)p4est->user_pointer;
5050a96aa3bSJed Brown   PetscInt         maxLevel = ctx->maxLevel;
5060a96aa3bSJed Brown 
5070a96aa3bSJed Brown   if ((PetscInt)quadrant->level >= maxLevel) return 0;
5080a96aa3bSJed Brown 
5090a96aa3bSJed Brown   return (quadrant->p.user_int == DM_ADAPT_REFINE);
5100a96aa3bSJed Brown }
5110a96aa3bSJed Brown 
512d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestComputeLocalCellTransferSF_loop(p4est_t *p4estFrom, PetscInt FromOffset, p4est_t *p4estTo, PetscInt ToOffset, p4est_topidx_t flt, p4est_topidx_t llt, PetscInt *toFineLeavesCount, PetscInt *toLeaves, PetscSFNode *fromRoots, PetscInt *fromFineLeavesCount, PetscInt *fromLeaves, PetscSFNode *toRoots)
513d71ae5a4SJacob Faibussowitsch {
5140a96aa3bSJed Brown   PetscMPIInt    rank = p4estFrom->mpirank;
5150a96aa3bSJed Brown   p4est_topidx_t t;
5160a96aa3bSJed Brown   PetscInt       toFineLeaves = 0, fromFineLeaves = 0;
5170a96aa3bSJed Brown 
5180a96aa3bSJed Brown   PetscFunctionBegin;
5190a96aa3bSJed Brown   for (t = flt; t <= llt; t++) { /* count roots and leaves */
5200a96aa3bSJed Brown     p4est_tree_t     *treeFrom  = &(((p4est_tree_t *)p4estFrom->trees->array)[t]);
5210a96aa3bSJed Brown     p4est_tree_t     *treeTo    = &(((p4est_tree_t *)p4estTo->trees->array)[t]);
5220a96aa3bSJed Brown     p4est_quadrant_t *firstFrom = &treeFrom->first_desc;
5230a96aa3bSJed Brown     p4est_quadrant_t *firstTo   = &treeTo->first_desc;
5240a96aa3bSJed Brown     PetscInt          numFrom   = (PetscInt)treeFrom->quadrants.elem_count;
5250a96aa3bSJed Brown     PetscInt          numTo     = (PetscInt)treeTo->quadrants.elem_count;
5260a96aa3bSJed Brown     p4est_quadrant_t *quadsFrom = (p4est_quadrant_t *)treeFrom->quadrants.array;
5270a96aa3bSJed Brown     p4est_quadrant_t *quadsTo   = (p4est_quadrant_t *)treeTo->quadrants.array;
5280a96aa3bSJed Brown     PetscInt          currentFrom, currentTo;
5290a96aa3bSJed Brown     PetscInt          treeOffsetFrom = (PetscInt)treeFrom->quadrants_offset;
5300a96aa3bSJed Brown     PetscInt          treeOffsetTo   = (PetscInt)treeTo->quadrants_offset;
5310a96aa3bSJed Brown     int               comp;
5320a96aa3bSJed Brown 
533792fecdfSBarry Smith     PetscCallP4estReturn(comp, p4est_quadrant_is_equal, (firstFrom, firstTo));
53428b400f6SJacob Faibussowitsch     PetscCheck(comp, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "non-matching partitions");
5350a96aa3bSJed Brown 
5360a96aa3bSJed Brown     for (currentFrom = 0, currentTo = 0; currentFrom < numFrom && currentTo < numTo;) {
5370a96aa3bSJed Brown       p4est_quadrant_t *quadFrom = &quadsFrom[currentFrom];
5380a96aa3bSJed Brown       p4est_quadrant_t *quadTo   = &quadsTo[currentTo];
5390a96aa3bSJed Brown 
5400a96aa3bSJed Brown       if (quadFrom->level == quadTo->level) {
5410a96aa3bSJed Brown         if (toLeaves) {
5420a96aa3bSJed Brown           toLeaves[toFineLeaves]        = currentTo + treeOffsetTo + ToOffset;
5430a96aa3bSJed Brown           fromRoots[toFineLeaves].rank  = rank;
5440a96aa3bSJed Brown           fromRoots[toFineLeaves].index = currentFrom + treeOffsetFrom + FromOffset;
5450a96aa3bSJed Brown         }
5460a96aa3bSJed Brown         toFineLeaves++;
5470a96aa3bSJed Brown         currentFrom++;
5480a96aa3bSJed Brown         currentTo++;
5490a96aa3bSJed Brown       } else {
5500a96aa3bSJed Brown         int fromIsAncestor;
5510a96aa3bSJed Brown 
552792fecdfSBarry Smith         PetscCallP4estReturn(fromIsAncestor, p4est_quadrant_is_ancestor, (quadFrom, quadTo));
5530a96aa3bSJed Brown         if (fromIsAncestor) {
5540a96aa3bSJed Brown           p4est_quadrant_t lastDesc;
5550a96aa3bSJed Brown 
5560a96aa3bSJed Brown           if (toLeaves) {
5570a96aa3bSJed Brown             toLeaves[toFineLeaves]        = currentTo + treeOffsetTo + ToOffset;
5580a96aa3bSJed Brown             fromRoots[toFineLeaves].rank  = rank;
5590a96aa3bSJed Brown             fromRoots[toFineLeaves].index = currentFrom + treeOffsetFrom + FromOffset;
5600a96aa3bSJed Brown           }
5610a96aa3bSJed Brown           toFineLeaves++;
5620a96aa3bSJed Brown           currentTo++;
563792fecdfSBarry Smith           PetscCallP4est(p4est_quadrant_last_descendant, (quadFrom, &lastDesc, quadTo->level));
564792fecdfSBarry Smith           PetscCallP4estReturn(comp, p4est_quadrant_is_equal, (quadTo, &lastDesc));
5650a96aa3bSJed Brown           if (comp) currentFrom++;
5660a96aa3bSJed Brown         } else {
5670a96aa3bSJed Brown           p4est_quadrant_t lastDesc;
5680a96aa3bSJed Brown 
5690a96aa3bSJed Brown           if (fromLeaves) {
5700a96aa3bSJed Brown             fromLeaves[fromFineLeaves]    = currentFrom + treeOffsetFrom + FromOffset;
5710a96aa3bSJed Brown             toRoots[fromFineLeaves].rank  = rank;
5720a96aa3bSJed Brown             toRoots[fromFineLeaves].index = currentTo + treeOffsetTo + ToOffset;
5730a96aa3bSJed Brown           }
5740a96aa3bSJed Brown           fromFineLeaves++;
5750a96aa3bSJed Brown           currentFrom++;
576792fecdfSBarry Smith           PetscCallP4est(p4est_quadrant_last_descendant, (quadTo, &lastDesc, quadFrom->level));
577792fecdfSBarry Smith           PetscCallP4estReturn(comp, p4est_quadrant_is_equal, (quadFrom, &lastDesc));
5780a96aa3bSJed Brown           if (comp) currentTo++;
5790a96aa3bSJed Brown         }
5800a96aa3bSJed Brown       }
5810a96aa3bSJed Brown     }
5820a96aa3bSJed Brown   }
5830a96aa3bSJed Brown   *toFineLeavesCount   = toFineLeaves;
5840a96aa3bSJed Brown   *fromFineLeavesCount = fromFineLeaves;
5853ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5860a96aa3bSJed Brown }
5870a96aa3bSJed Brown 
5880a96aa3bSJed Brown /* Compute the maximum level across all the trees */
589d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetRefinementLevel(DM dm, PetscInt *lev)
590d71ae5a4SJacob Faibussowitsch {
5910a96aa3bSJed Brown   p4est_topidx_t     t, flt, llt;
5920a96aa3bSJed Brown   DM_Forest         *forest      = (DM_Forest *)dm->data;
5930a96aa3bSJed Brown   DM_Forest_pforest *pforest     = (DM_Forest_pforest *)forest->data;
5940a96aa3bSJed Brown   PetscInt           maxlevelloc = 0;
5950a96aa3bSJed Brown   p4est_t           *p4est;
5960a96aa3bSJed Brown 
5970a96aa3bSJed Brown   PetscFunctionBegin;
59828b400f6SJacob Faibussowitsch   PetscCheck(pforest, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Missing DM_Forest_pforest");
59928b400f6SJacob Faibussowitsch   PetscCheck(pforest->forest, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Missing p4est_t");
6000a96aa3bSJed Brown   p4est = pforest->forest;
6010a96aa3bSJed Brown   flt   = p4est->first_local_tree;
6020a96aa3bSJed Brown   llt   = p4est->last_local_tree;
6030a96aa3bSJed Brown   for (t = flt; t <= llt; t++) {
6040a96aa3bSJed Brown     p4est_tree_t *tree = &(((p4est_tree_t *)p4est->trees->array)[t]);
6050a96aa3bSJed Brown     maxlevelloc        = PetscMax((PetscInt)tree->maxlevel, maxlevelloc);
6060a96aa3bSJed Brown   }
6071c2dc1cbSBarry Smith   PetscCall(MPIU_Allreduce(&maxlevelloc, lev, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
6083ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6090a96aa3bSJed Brown }
6100a96aa3bSJed Brown 
6110a96aa3bSJed Brown /* Puts identity in coarseToFine */
6120a96aa3bSJed Brown /* assumes a matching partition */
613d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestComputeLocalCellTransferSF(MPI_Comm comm, p4est_t *p4estFrom, PetscInt FromOffset, p4est_t *p4estTo, PetscInt ToOffset, PetscSF *fromCoarseToFine, PetscSF *toCoarseFromFine)
614d71ae5a4SJacob Faibussowitsch {
6150a96aa3bSJed Brown   p4est_topidx_t flt, llt;
6160a96aa3bSJed Brown   PetscSF        fromCoarse, toCoarse;
6170a96aa3bSJed Brown   PetscInt       numRootsFrom, numRootsTo, numLeavesFrom, numLeavesTo;
6180a96aa3bSJed Brown   PetscInt      *fromLeaves = NULL, *toLeaves = NULL;
6190a96aa3bSJed Brown   PetscSFNode   *fromRoots = NULL, *toRoots = NULL;
6200a96aa3bSJed Brown 
6210a96aa3bSJed Brown   PetscFunctionBegin;
6220a96aa3bSJed Brown   flt = p4estFrom->first_local_tree;
6230a96aa3bSJed Brown   llt = p4estFrom->last_local_tree;
6249566063dSJacob Faibussowitsch   PetscCall(PetscSFCreate(comm, &fromCoarse));
62548a46eb9SPierre Jolivet   if (toCoarseFromFine) PetscCall(PetscSFCreate(comm, &toCoarse));
6260a96aa3bSJed Brown   numRootsFrom = p4estFrom->local_num_quadrants + FromOffset;
6270a96aa3bSJed Brown   numRootsTo   = p4estTo->local_num_quadrants + ToOffset;
6289566063dSJacob Faibussowitsch   PetscCall(DMPforestComputeLocalCellTransferSF_loop(p4estFrom, FromOffset, p4estTo, ToOffset, flt, llt, &numLeavesTo, NULL, NULL, &numLeavesFrom, NULL, NULL));
6299566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(numLeavesTo, &toLeaves));
6309566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(numLeavesTo, &fromRoots));
6310a96aa3bSJed Brown   if (toCoarseFromFine) {
6329566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(numLeavesFrom, &fromLeaves));
6339566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(numLeavesFrom, &fromRoots));
6340a96aa3bSJed Brown   }
6359566063dSJacob Faibussowitsch   PetscCall(DMPforestComputeLocalCellTransferSF_loop(p4estFrom, FromOffset, p4estTo, ToOffset, flt, llt, &numLeavesTo, toLeaves, fromRoots, &numLeavesFrom, fromLeaves, toRoots));
6360a96aa3bSJed Brown   if (!ToOffset && (numLeavesTo == numRootsTo)) { /* compress */
6379566063dSJacob Faibussowitsch     PetscCall(PetscFree(toLeaves));
6389566063dSJacob Faibussowitsch     PetscCall(PetscSFSetGraph(fromCoarse, numRootsFrom, numLeavesTo, NULL, PETSC_OWN_POINTER, fromRoots, PETSC_OWN_POINTER));
6391baa6e33SBarry Smith   } else PetscCall(PetscSFSetGraph(fromCoarse, numRootsFrom, numLeavesTo, toLeaves, PETSC_OWN_POINTER, fromRoots, PETSC_OWN_POINTER));
6400a96aa3bSJed Brown   *fromCoarseToFine = fromCoarse;
6410a96aa3bSJed Brown   if (toCoarseFromFine) {
6429566063dSJacob Faibussowitsch     PetscCall(PetscSFSetGraph(toCoarse, numRootsTo, numLeavesFrom, fromLeaves, PETSC_OWN_POINTER, toRoots, PETSC_OWN_POINTER));
6430a96aa3bSJed Brown     *toCoarseFromFine = toCoarse;
6440a96aa3bSJed Brown   }
6453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6460a96aa3bSJed Brown }
6470a96aa3bSJed Brown 
6480a96aa3bSJed Brown /* range of processes whose B sections overlap this ranks A section */
649d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestComputeOverlappingRanks(PetscMPIInt size, PetscMPIInt rank, p4est_t *p4estA, p4est_t *p4estB, PetscInt *startB, PetscInt *endB)
650d71ae5a4SJacob Faibussowitsch {
6510a96aa3bSJed Brown   p4est_quadrant_t *myCoarseStart = &(p4estA->global_first_position[rank]);
6520a96aa3bSJed Brown   p4est_quadrant_t *myCoarseEnd   = &(p4estA->global_first_position[rank + 1]);
6530a96aa3bSJed Brown   p4est_quadrant_t *globalFirstB  = p4estB->global_first_position;
6540a96aa3bSJed Brown 
6550a96aa3bSJed Brown   PetscFunctionBegin;
6560a96aa3bSJed Brown   *startB = -1;
6570a96aa3bSJed Brown   *endB   = -1;
6580a96aa3bSJed Brown   if (p4estA->local_num_quadrants) {
6590a96aa3bSJed Brown     PetscInt lo, hi, guess;
6600a96aa3bSJed Brown     /* binary search to find interval containing myCoarseStart */
6610a96aa3bSJed Brown     lo    = 0;
6620a96aa3bSJed Brown     hi    = size;
6630a96aa3bSJed Brown     guess = rank;
6640a96aa3bSJed Brown     while (1) {
6650a96aa3bSJed Brown       int startCompMy, myCompEnd;
6660a96aa3bSJed Brown 
667792fecdfSBarry Smith       PetscCallP4estReturn(startCompMy, p4est_quadrant_compare_piggy, (&globalFirstB[guess], myCoarseStart));
668792fecdfSBarry Smith       PetscCallP4estReturn(myCompEnd, p4est_quadrant_compare_piggy, (myCoarseStart, &globalFirstB[guess + 1]));
6690a96aa3bSJed Brown       if (startCompMy <= 0 && myCompEnd < 0) {
6700a96aa3bSJed Brown         *startB = guess;
6710a96aa3bSJed Brown         break;
6720a96aa3bSJed Brown       } else if (startCompMy > 0) { /* guess is to high */
6730a96aa3bSJed Brown         hi = guess;
6740a96aa3bSJed Brown       } else { /* guess is to low */
6750a96aa3bSJed Brown         lo = guess + 1;
6760a96aa3bSJed Brown       }
6770a96aa3bSJed Brown       guess = lo + (hi - lo) / 2;
6780a96aa3bSJed Brown     }
6790a96aa3bSJed Brown     /* reset bounds, but not guess */
6800a96aa3bSJed Brown     lo = 0;
6810a96aa3bSJed Brown     hi = size;
6820a96aa3bSJed Brown     while (1) {
6830a96aa3bSJed Brown       int startCompMy, myCompEnd;
6840a96aa3bSJed Brown 
685792fecdfSBarry Smith       PetscCallP4estReturn(startCompMy, p4est_quadrant_compare_piggy, (&globalFirstB[guess], myCoarseEnd));
686792fecdfSBarry Smith       PetscCallP4estReturn(myCompEnd, p4est_quadrant_compare_piggy, (myCoarseEnd, &globalFirstB[guess + 1]));
6870a96aa3bSJed Brown       if (startCompMy < 0 && myCompEnd <= 0) { /* notice that the comparison operators are different from above */
6880a96aa3bSJed Brown         *endB = guess + 1;
6890a96aa3bSJed Brown         break;
6900a96aa3bSJed Brown       } else if (startCompMy >= 0) { /* guess is to high */
6910a96aa3bSJed Brown         hi = guess;
6920a96aa3bSJed Brown       } else { /* guess is to low */
6930a96aa3bSJed Brown         lo = guess + 1;
6940a96aa3bSJed Brown       }
6950a96aa3bSJed Brown       guess = lo + (hi - lo) / 2;
6960a96aa3bSJed Brown     }
6970a96aa3bSJed Brown   }
6983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
6990a96aa3bSJed Brown }
7000a96aa3bSJed Brown 
7010a96aa3bSJed Brown static PetscErrorCode DMPforestGetPlex(DM, DM *);
7020a96aa3bSJed Brown 
7030a96aa3bSJed Brown   #define DMSetUp_pforest _append_pforest(DMSetUp)
704d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMSetUp_pforest(DM dm)
705d71ae5a4SJacob Faibussowitsch {
7060a96aa3bSJed Brown   DM_Forest         *forest  = (DM_Forest *)dm->data;
7070a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)forest->data;
7080a96aa3bSJed Brown   DM                 base, adaptFrom;
7090a96aa3bSJed Brown   DMForestTopology   topoName;
7100a96aa3bSJed Brown   PetscSF            preCoarseToFine = NULL, coarseToPreFine = NULL;
7110a96aa3bSJed Brown   PforestAdaptCtx    ctx;
7120a96aa3bSJed Brown 
7130a96aa3bSJed Brown   PetscFunctionBegin;
7140a96aa3bSJed Brown   ctx.minLevel  = PETSC_MAX_INT;
7150a96aa3bSJed Brown   ctx.maxLevel  = 0;
7160a96aa3bSJed Brown   ctx.currLevel = 0;
7170a96aa3bSJed Brown   ctx.anyChange = PETSC_FALSE;
7180a96aa3bSJed Brown   /* sanity check */
7199566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dm, &adaptFrom));
7209566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseDM(dm, &base));
7219566063dSJacob Faibussowitsch   PetscCall(DMForestGetTopology(dm, &topoName));
7221dca8a05SBarry Smith   PetscCheck(adaptFrom || base || topoName, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "A forest needs a topology, a base DM, or a DM to adapt from");
7230a96aa3bSJed Brown 
7240a96aa3bSJed Brown   /* === Step 1: DMFTopology === */
7250a96aa3bSJed Brown   if (adaptFrom) { /* reference already created topology */
7260a96aa3bSJed Brown     PetscBool          ispforest;
7270a96aa3bSJed Brown     DM_Forest         *aforest  = (DM_Forest *)adaptFrom->data;
7280a96aa3bSJed Brown     DM_Forest_pforest *apforest = (DM_Forest_pforest *)aforest->data;
7290a96aa3bSJed Brown 
7309566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)adaptFrom, DMPFOREST, &ispforest));
73128b400f6SJacob Faibussowitsch     PetscCheck(ispforest, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_NOTSAMETYPE, "Trying to adapt from %s, which is not %s", ((PetscObject)adaptFrom)->type_name, DMPFOREST);
73228b400f6SJacob Faibussowitsch     PetscCheck(apforest->topo, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "The pre-adaptation forest must have a topology");
7339566063dSJacob Faibussowitsch     PetscCall(DMSetUp(adaptFrom));
7349566063dSJacob Faibussowitsch     PetscCall(DMForestGetBaseDM(dm, &base));
7359566063dSJacob Faibussowitsch     PetscCall(DMForestGetTopology(dm, &topoName));
7360a96aa3bSJed Brown   } else if (base) { /* construct a connectivity from base */
7370a96aa3bSJed Brown     PetscBool isPlex, isDA;
7380a96aa3bSJed Brown 
7399566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)base, &topoName));
7409566063dSJacob Faibussowitsch     PetscCall(DMForestSetTopology(dm, topoName));
7419566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)base, DMPLEX, &isPlex));
7429566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)base, DMDA, &isDA));
7430a96aa3bSJed Brown     if (isPlex) {
7440a96aa3bSJed Brown       MPI_Comm              comm = PetscObjectComm((PetscObject)dm);
7450a96aa3bSJed Brown       PetscInt              depth;
7460a96aa3bSJed Brown       PetscMPIInt           size;
7470a96aa3bSJed Brown       p4est_connectivity_t *conn = NULL;
7480a96aa3bSJed Brown       DMFTopology_pforest  *topo;
7490a96aa3bSJed Brown       PetscInt             *tree_face_to_uniq = NULL;
7500a96aa3bSJed Brown 
7519566063dSJacob Faibussowitsch       PetscCall(DMPlexGetDepth(base, &depth));
7520a96aa3bSJed Brown       if (depth == 1) {
7530a96aa3bSJed Brown         DM connDM;
7540a96aa3bSJed Brown 
7559566063dSJacob Faibussowitsch         PetscCall(DMPlexInterpolate(base, &connDM));
7560a96aa3bSJed Brown         base = connDM;
7579566063dSJacob Faibussowitsch         PetscCall(DMForestSetBaseDM(dm, base));
7589566063dSJacob Faibussowitsch         PetscCall(DMDestroy(&connDM));
75963a3b9bcSJacob Faibussowitsch       } else PetscCheck(depth == P4EST_DIM, comm, PETSC_ERR_ARG_WRONG, "Base plex is neither interpolated nor uninterpolated? depth %" PetscInt_FMT ", expected 2 or %d", depth, P4EST_DIM + 1);
7609566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Comm_size(comm, &size));
7610a96aa3bSJed Brown       if (size > 1) {
7620a96aa3bSJed Brown         DM      dmRedundant;
7630a96aa3bSJed Brown         PetscSF sf;
7640a96aa3bSJed Brown 
7659566063dSJacob Faibussowitsch         PetscCall(DMPlexGetRedundantDM(base, &sf, &dmRedundant));
76628b400f6SJacob Faibussowitsch         PetscCheck(dmRedundant, comm, PETSC_ERR_PLIB, "Could not create redundant DM");
7679566063dSJacob Faibussowitsch         PetscCall(PetscObjectCompose((PetscObject)dmRedundant, "_base_migration_sf", (PetscObject)sf));
7689566063dSJacob Faibussowitsch         PetscCall(PetscSFDestroy(&sf));
7690a96aa3bSJed Brown         base = dmRedundant;
7709566063dSJacob Faibussowitsch         PetscCall(DMForestSetBaseDM(dm, base));
7719566063dSJacob Faibussowitsch         PetscCall(DMDestroy(&dmRedundant));
7720a96aa3bSJed Brown       }
7739566063dSJacob Faibussowitsch       PetscCall(DMViewFromOptions(base, NULL, "-dm_p4est_base_view"));
7749566063dSJacob Faibussowitsch       PetscCall(DMPlexCreateConnectivity_pforest(base, &conn, &tree_face_to_uniq));
7754dfa11a4SJacob Faibussowitsch       PetscCall(PetscNew(&topo));
7760a96aa3bSJed Brown       topo->refct = 1;
7770a96aa3bSJed Brown       topo->conn  = conn;
7780a96aa3bSJed Brown       topo->geom  = NULL;
7790a96aa3bSJed Brown       {
7800a96aa3bSJed Brown         PetscErrorCode (*map)(DM, PetscInt, PetscInt, const PetscReal[], PetscReal[], void *);
7810a96aa3bSJed Brown         void *mapCtx;
7820a96aa3bSJed Brown 
7839566063dSJacob Faibussowitsch         PetscCall(DMForestGetBaseCoordinateMapping(dm, &map, &mapCtx));
7840a96aa3bSJed Brown         if (map) {
7850a96aa3bSJed Brown           DM_Forest_geometry_pforest *geom_pforest;
7860a96aa3bSJed Brown           p4est_geometry_t           *geom;
7870a96aa3bSJed Brown 
7889566063dSJacob Faibussowitsch           PetscCall(PetscNew(&geom_pforest));
7899566063dSJacob Faibussowitsch           PetscCall(DMGetCoordinateDim(dm, &geom_pforest->coordDim));
7900a96aa3bSJed Brown           geom_pforest->map    = map;
7910a96aa3bSJed Brown           geom_pforest->mapCtx = mapCtx;
792792fecdfSBarry Smith           PetscCallP4estReturn(geom_pforest->inner, p4est_geometry_new_connectivity, (conn));
7939566063dSJacob Faibussowitsch           PetscCall(PetscNew(&geom));
7940a96aa3bSJed Brown           geom->name    = topoName;
7950a96aa3bSJed Brown           geom->user    = geom_pforest;
7960a96aa3bSJed Brown           geom->X       = GeometryMapping_pforest;
7970a96aa3bSJed Brown           geom->destroy = GeometryDestroy_pforest;
7980a96aa3bSJed Brown           topo->geom    = geom;
7990a96aa3bSJed Brown         }
8000a96aa3bSJed Brown       }
8010a96aa3bSJed Brown       topo->tree_face_to_uniq = tree_face_to_uniq;
8020a96aa3bSJed Brown       pforest->topo           = topo;
80328b400f6SJacob Faibussowitsch     } else PetscCheck(!isDA, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Not implemented yet");
8040a96aa3bSJed Brown   #if 0
8050a96aa3bSJed Brown       PetscInt N[3], P[3];
8060a96aa3bSJed Brown 
8070a96aa3bSJed Brown       /* get the sizes, periodicities */
8080a96aa3bSJed Brown       /* ... */
8090a96aa3bSJed Brown                                                                   /* don't use Morton order */
8109566063dSJacob Faibussowitsch       PetscCall(DMFTopologyCreateBrick_pforest(dm,N,P,&pforest->topo,PETSC_FALSE));
8110a96aa3bSJed Brown   #endif
8120a96aa3bSJed Brown     {
8130a96aa3bSJed Brown       PetscInt numLabels, l;
8140a96aa3bSJed Brown 
8159566063dSJacob Faibussowitsch       PetscCall(DMGetNumLabels(base, &numLabels));
8160a96aa3bSJed Brown       for (l = 0; l < numLabels; l++) {
8170a96aa3bSJed Brown         PetscBool   isDepth, isGhost, isVTK, isDim, isCellType;
8180a96aa3bSJed Brown         DMLabel     label, labelNew;
8190a96aa3bSJed Brown         PetscInt    defVal;
8200a96aa3bSJed Brown         const char *name;
8210a96aa3bSJed Brown 
8229566063dSJacob Faibussowitsch         PetscCall(DMGetLabelName(base, l, &name));
8239566063dSJacob Faibussowitsch         PetscCall(DMGetLabelByNum(base, l, &label));
8249566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "depth", &isDepth));
8250a96aa3bSJed Brown         if (isDepth) continue;
8269566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "dim", &isDim));
8270a96aa3bSJed Brown         if (isDim) continue;
8289566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "celltype", &isCellType));
8290a96aa3bSJed Brown         if (isCellType) continue;
8309566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "ghost", &isGhost));
8310a96aa3bSJed Brown         if (isGhost) continue;
8329566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "vtk", &isVTK));
8330a96aa3bSJed Brown         if (isVTK) continue;
8349566063dSJacob Faibussowitsch         PetscCall(DMCreateLabel(dm, name));
8359566063dSJacob Faibussowitsch         PetscCall(DMGetLabel(dm, name, &labelNew));
8369566063dSJacob Faibussowitsch         PetscCall(DMLabelGetDefaultValue(label, &defVal));
8379566063dSJacob Faibussowitsch         PetscCall(DMLabelSetDefaultValue(labelNew, defVal));
8380a96aa3bSJed Brown       }
8390a96aa3bSJed Brown       /* map dm points (internal plex) to base
8400a96aa3bSJed Brown          we currently create the subpoint_map for the entire hierarchy, starting from the finest forest
8410a96aa3bSJed Brown          and propagating back to the coarsest
8420a96aa3bSJed Brown          This is not an optimal approach, since we need the map only on the coarsest level
8430a96aa3bSJed Brown          during DMForestTransferVecFromBase */
8449566063dSJacob Faibussowitsch       PetscCall(DMForestGetMinimumRefinement(dm, &l));
84548a46eb9SPierre Jolivet       if (!l) PetscCall(DMCreateLabel(dm, "_forest_base_subpoint_map"));
8460a96aa3bSJed Brown     }
8470a96aa3bSJed Brown   } else { /* construct from topology name */
8480a96aa3bSJed Brown     DMFTopology_pforest *topo;
8490a96aa3bSJed Brown 
8509566063dSJacob Faibussowitsch     PetscCall(DMFTopologyCreate_pforest(dm, topoName, &topo));
8510a96aa3bSJed Brown     pforest->topo = topo;
8520a96aa3bSJed Brown     /* TODO: construct base? */
8530a96aa3bSJed Brown   }
8540a96aa3bSJed Brown 
8550a96aa3bSJed Brown   /* === Step 2: get the leaves of the forest === */
8560a96aa3bSJed Brown   if (adaptFrom) { /* start with the old forest */
8570a96aa3bSJed Brown     DMLabel            adaptLabel;
8580a96aa3bSJed Brown     PetscInt           defaultValue;
8590a96aa3bSJed Brown     PetscInt           numValues, numValuesGlobal, cLocalStart, count;
8600a96aa3bSJed Brown     DM_Forest         *aforest  = (DM_Forest *)adaptFrom->data;
8610a96aa3bSJed Brown     DM_Forest_pforest *apforest = (DM_Forest_pforest *)aforest->data;
8620a96aa3bSJed Brown     PetscBool          computeAdaptSF;
8630a96aa3bSJed Brown     p4est_topidx_t     flt, llt, t;
8640a96aa3bSJed Brown 
8650a96aa3bSJed Brown     flt         = apforest->forest->first_local_tree;
8660a96aa3bSJed Brown     llt         = apforest->forest->last_local_tree;
8670a96aa3bSJed Brown     cLocalStart = apforest->cLocalStart;
8689566063dSJacob Faibussowitsch     PetscCall(DMForestGetComputeAdaptivitySF(dm, &computeAdaptSF));
869792fecdfSBarry Smith     PetscCallP4estReturn(pforest->forest, p4est_copy, (apforest->forest, 0)); /* 0 indicates no data copying */
8709566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityLabel(dm, &adaptLabel));
8710a96aa3bSJed Brown     if (adaptLabel) {
8720a96aa3bSJed Brown       /* apply the refinement/coarsening by flags, plus minimum/maximum refinement */
8739566063dSJacob Faibussowitsch       PetscCall(DMLabelGetNumValues(adaptLabel, &numValues));
874712fec58SPierre Jolivet       PetscCall(MPIU_Allreduce(&numValues, &numValuesGlobal, 1, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)adaptFrom)));
8759566063dSJacob Faibussowitsch       PetscCall(DMLabelGetDefaultValue(adaptLabel, &defaultValue));
8760a96aa3bSJed Brown       if (!numValuesGlobal && defaultValue == DM_ADAPT_COARSEN_LAST) { /* uniform coarsen of the last level only (equivalent to DM_ADAPT_COARSEN for conforming grids)  */
8779566063dSJacob Faibussowitsch         PetscCall(DMForestGetMinimumRefinement(dm, &ctx.minLevel));
8789566063dSJacob Faibussowitsch         PetscCall(DMPforestGetRefinementLevel(dm, &ctx.currLevel));
8790a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)&ctx;
880792fecdfSBarry Smith         PetscCallP4est(p4est_coarsen, (pforest->forest, 0, pforest_coarsen_currlevel, NULL));
8810a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)dm;
882792fecdfSBarry Smith         PetscCallP4est(p4est_balance, (pforest->forest, P4EST_CONNECT_FULL, NULL));
8830a96aa3bSJed Brown         /* we will have to change the offset after we compute the overlap */
88448a46eb9SPierre Jolivet         if (computeAdaptSF) PetscCall(DMPforestComputeLocalCellTransferSF(PetscObjectComm((PetscObject)dm), pforest->forest, 0, apforest->forest, apforest->cLocalStart, &coarseToPreFine, NULL));
8850a96aa3bSJed Brown       } else if (!numValuesGlobal && defaultValue == DM_ADAPT_COARSEN) { /* uniform coarsen */
8869566063dSJacob Faibussowitsch         PetscCall(DMForestGetMinimumRefinement(dm, &ctx.minLevel));
8870a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)&ctx;
888792fecdfSBarry Smith         PetscCallP4est(p4est_coarsen, (pforest->forest, 0, pforest_coarsen_uniform, NULL));
8890a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)dm;
890792fecdfSBarry Smith         PetscCallP4est(p4est_balance, (pforest->forest, P4EST_CONNECT_FULL, NULL));
8910a96aa3bSJed Brown         /* we will have to change the offset after we compute the overlap */
89248a46eb9SPierre Jolivet         if (computeAdaptSF) PetscCall(DMPforestComputeLocalCellTransferSF(PetscObjectComm((PetscObject)dm), pforest->forest, 0, apforest->forest, apforest->cLocalStart, &coarseToPreFine, NULL));
8930a96aa3bSJed Brown       } else if (!numValuesGlobal && defaultValue == DM_ADAPT_REFINE) { /* uniform refine */
8949566063dSJacob Faibussowitsch         PetscCall(DMForestGetMaximumRefinement(dm, &ctx.maxLevel));
8950a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)&ctx;
896792fecdfSBarry Smith         PetscCallP4est(p4est_refine, (pforest->forest, 0, pforest_refine_uniform, NULL));
8970a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)dm;
898792fecdfSBarry Smith         PetscCallP4est(p4est_balance, (pforest->forest, P4EST_CONNECT_FULL, NULL));
8990a96aa3bSJed Brown         /* we will have to change the offset after we compute the overlap */
90048a46eb9SPierre Jolivet         if (computeAdaptSF) PetscCall(DMPforestComputeLocalCellTransferSF(PetscObjectComm((PetscObject)dm), apforest->forest, apforest->cLocalStart, pforest->forest, 0, &preCoarseToFine, NULL));
9010a96aa3bSJed Brown       } else if (numValuesGlobal) {
9020a96aa3bSJed Brown         p4est_t                   *p4est = pforest->forest;
9030a96aa3bSJed Brown         PetscInt                  *cellFlags;
9040a96aa3bSJed Brown         DMForestAdaptivityStrategy strategy;
9050a96aa3bSJed Brown         PetscSF                    cellSF;
9060a96aa3bSJed Brown         PetscInt                   c, cStart, cEnd;
9070a96aa3bSJed Brown         PetscBool                  adaptAny;
9080a96aa3bSJed Brown 
9099566063dSJacob Faibussowitsch         PetscCall(DMForestGetMaximumRefinement(dm, &ctx.maxLevel));
9109566063dSJacob Faibussowitsch         PetscCall(DMForestGetMinimumRefinement(dm, &ctx.minLevel));
9119566063dSJacob Faibussowitsch         PetscCall(DMForestGetAdaptivityStrategy(dm, &strategy));
9129566063dSJacob Faibussowitsch         PetscCall(PetscStrncmp(strategy, "any", 3, &adaptAny));
9139566063dSJacob Faibussowitsch         PetscCall(DMForestGetCellChart(adaptFrom, &cStart, &cEnd));
9149566063dSJacob Faibussowitsch         PetscCall(DMForestGetCellSF(adaptFrom, &cellSF));
9159566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(cEnd - cStart, &cellFlags));
9169566063dSJacob Faibussowitsch         for (c = cStart; c < cEnd; c++) PetscCall(DMLabelGetValue(adaptLabel, c, &cellFlags[c - cStart]));
9170a96aa3bSJed Brown         if (cellSF) {
9180a96aa3bSJed Brown           if (adaptAny) {
9199566063dSJacob Faibussowitsch             PetscCall(PetscSFReduceBegin(cellSF, MPIU_INT, cellFlags, cellFlags, MPI_MAX));
9209566063dSJacob Faibussowitsch             PetscCall(PetscSFReduceEnd(cellSF, MPIU_INT, cellFlags, cellFlags, MPI_MAX));
9210a96aa3bSJed Brown           } else {
9229566063dSJacob Faibussowitsch             PetscCall(PetscSFReduceBegin(cellSF, MPIU_INT, cellFlags, cellFlags, MPI_MIN));
9239566063dSJacob Faibussowitsch             PetscCall(PetscSFReduceEnd(cellSF, MPIU_INT, cellFlags, cellFlags, MPI_MIN));
9240a96aa3bSJed Brown           }
9250a96aa3bSJed Brown         }
9260a96aa3bSJed Brown         for (t = flt, count = cLocalStart; t <= llt; t++) {
9270a96aa3bSJed Brown           p4est_tree_t     *tree     = &(((p4est_tree_t *)p4est->trees->array)[t]);
9280a96aa3bSJed Brown           PetscInt          numQuads = (PetscInt)tree->quadrants.elem_count, i;
9290a96aa3bSJed Brown           p4est_quadrant_t *quads    = (p4est_quadrant_t *)tree->quadrants.array;
9300a96aa3bSJed Brown 
9310a96aa3bSJed Brown           for (i = 0; i < numQuads; i++) {
9320a96aa3bSJed Brown             p4est_quadrant_t *q = &quads[i];
9330a96aa3bSJed Brown             q->p.user_int       = cellFlags[count++];
9340a96aa3bSJed Brown           }
9350a96aa3bSJed Brown         }
9369566063dSJacob Faibussowitsch         PetscCall(PetscFree(cellFlags));
9370a96aa3bSJed Brown 
9380a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)&ctx;
939792fecdfSBarry Smith         if (adaptAny) PetscCallP4est(p4est_coarsen, (pforest->forest, 0, pforest_coarsen_flag_any, pforest_init_determine));
940792fecdfSBarry Smith         else PetscCallP4est(p4est_coarsen, (pforest->forest, 0, pforest_coarsen_flag_all, pforest_init_determine));
941792fecdfSBarry Smith         PetscCallP4est(p4est_refine, (pforest->forest, 0, pforest_refine_flag, NULL));
9420a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)dm;
943792fecdfSBarry Smith         PetscCallP4est(p4est_balance, (pforest->forest, P4EST_CONNECT_FULL, NULL));
94448a46eb9SPierre Jolivet         if (computeAdaptSF) PetscCall(DMPforestComputeLocalCellTransferSF(PetscObjectComm((PetscObject)dm), apforest->forest, apforest->cLocalStart, pforest->forest, 0, &preCoarseToFine, &coarseToPreFine));
9450a96aa3bSJed Brown       }
9460a96aa3bSJed Brown       for (t = flt, count = cLocalStart; t <= llt; t++) {
9470a96aa3bSJed Brown         p4est_tree_t     *atree     = &(((p4est_tree_t *)apforest->forest->trees->array)[t]);
9480a96aa3bSJed Brown         p4est_tree_t     *tree      = &(((p4est_tree_t *)pforest->forest->trees->array)[t]);
9490a96aa3bSJed Brown         PetscInt          anumQuads = (PetscInt)atree->quadrants.elem_count, i;
9500a96aa3bSJed Brown         PetscInt          numQuads  = (PetscInt)tree->quadrants.elem_count;
9510a96aa3bSJed Brown         p4est_quadrant_t *aquads    = (p4est_quadrant_t *)atree->quadrants.array;
9520a96aa3bSJed Brown         p4est_quadrant_t *quads     = (p4est_quadrant_t *)tree->quadrants.array;
9530a96aa3bSJed Brown 
9540a96aa3bSJed Brown         if (anumQuads != numQuads) {
9550a96aa3bSJed Brown           ctx.anyChange = PETSC_TRUE;
9560a96aa3bSJed Brown         } else {
9570a96aa3bSJed Brown           for (i = 0; i < numQuads; i++) {
9580a96aa3bSJed Brown             p4est_quadrant_t *aq = &aquads[i];
9590a96aa3bSJed Brown             p4est_quadrant_t *q  = &quads[i];
9600a96aa3bSJed Brown 
9610a96aa3bSJed Brown             if (aq->level != q->level) {
9620a96aa3bSJed Brown               ctx.anyChange = PETSC_TRUE;
9630a96aa3bSJed Brown               break;
9640a96aa3bSJed Brown             }
9650a96aa3bSJed Brown           }
9660a96aa3bSJed Brown         }
967ad540459SPierre Jolivet         if (ctx.anyChange) break;
9680a96aa3bSJed Brown       }
9690a96aa3bSJed Brown     }
9700a96aa3bSJed Brown     {
9710a96aa3bSJed Brown       PetscInt numLabels, l;
9720a96aa3bSJed Brown 
9739566063dSJacob Faibussowitsch       PetscCall(DMGetNumLabels(adaptFrom, &numLabels));
9740a96aa3bSJed Brown       for (l = 0; l < numLabels; l++) {
9750a96aa3bSJed Brown         PetscBool   isDepth, isCellType, isGhost, isVTK;
9760a96aa3bSJed Brown         DMLabel     label, labelNew;
9770a96aa3bSJed Brown         PetscInt    defVal;
9780a96aa3bSJed Brown         const char *name;
9790a96aa3bSJed Brown 
9809566063dSJacob Faibussowitsch         PetscCall(DMGetLabelName(adaptFrom, l, &name));
9819566063dSJacob Faibussowitsch         PetscCall(DMGetLabelByNum(adaptFrom, l, &label));
9829566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "depth", &isDepth));
9830a96aa3bSJed Brown         if (isDepth) continue;
9849566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "celltype", &isCellType));
9850a96aa3bSJed Brown         if (isCellType) continue;
9869566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "ghost", &isGhost));
9870a96aa3bSJed Brown         if (isGhost) continue;
9889566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(name, "vtk", &isVTK));
9890a96aa3bSJed Brown         if (isVTK) continue;
9909566063dSJacob Faibussowitsch         PetscCall(DMCreateLabel(dm, name));
9919566063dSJacob Faibussowitsch         PetscCall(DMGetLabel(dm, name, &labelNew));
9929566063dSJacob Faibussowitsch         PetscCall(DMLabelGetDefaultValue(label, &defVal));
9939566063dSJacob Faibussowitsch         PetscCall(DMLabelSetDefaultValue(labelNew, defVal));
9940a96aa3bSJed Brown       }
9950a96aa3bSJed Brown     }
9960a96aa3bSJed Brown   } else { /* initial */
9970a96aa3bSJed Brown     PetscInt initLevel, minLevel;
99866c0a4b5SToby Isaac   #if defined(PETSC_HAVE_MPIUNI)
99966c0a4b5SToby Isaac     sc_MPI_Comm comm = sc_MPI_COMM_WORLD;
100066c0a4b5SToby Isaac   #else
100166c0a4b5SToby Isaac     MPI_Comm comm = PetscObjectComm((PetscObject)dm);
100266c0a4b5SToby Isaac   #endif
10030a96aa3bSJed Brown 
10049566063dSJacob Faibussowitsch     PetscCall(DMForestGetInitialRefinement(dm, &initLevel));
10059566063dSJacob Faibussowitsch     PetscCall(DMForestGetMinimumRefinement(dm, &minLevel));
10069371c9d4SSatish Balay     PetscCallP4estReturn(pforest->forest, p4est_new_ext,
10079371c9d4SSatish Balay                          (comm, pforest->topo->conn, 0, /* minimum number of quadrants per processor */
10080a96aa3bSJed Brown                           initLevel,                    /* level of refinement */
10090a96aa3bSJed Brown                           1,                            /* uniform refinement */
10100a96aa3bSJed Brown                           0,                            /* we don't allocate any per quadrant data */
10110a96aa3bSJed Brown                           NULL,                         /* there is no special quadrant initialization */
10120a96aa3bSJed Brown                           (void *)dm));                 /* this dm is the user context */
10130a96aa3bSJed Brown 
10140a96aa3bSJed Brown     if (initLevel > minLevel) pforest->coarsen_hierarchy = PETSC_TRUE;
10150a96aa3bSJed Brown     if (dm->setfromoptionscalled) {
10160a96aa3bSJed Brown       PetscBool   flgPattern, flgFractal;
10170a96aa3bSJed Brown       PetscInt    corner = 0;
10180a96aa3bSJed Brown       PetscInt    corners[P4EST_CHILDREN], ncorner = P4EST_CHILDREN;
10190a96aa3bSJed Brown       PetscReal   likelihood = 1. / P4EST_DIM;
10200a96aa3bSJed Brown       PetscInt    pattern;
10210a96aa3bSJed Brown       const char *prefix;
10220a96aa3bSJed Brown 
10239566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
10249566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetEList(((PetscObject)dm)->options, prefix, "-dm_p4est_refine_pattern", DMRefinePatternName, PATTERN_COUNT, &pattern, &flgPattern));
10259566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetInt(((PetscObject)dm)->options, prefix, "-dm_p4est_refine_corner", &corner, NULL));
10269566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetIntArray(((PetscObject)dm)->options, prefix, "-dm_p4est_refine_fractal_corners", corners, &ncorner, &flgFractal));
10279566063dSJacob Faibussowitsch       PetscCall(PetscOptionsGetReal(((PetscObject)dm)->options, prefix, "-dm_p4est_refine_hash_likelihood", &likelihood, NULL));
10280a96aa3bSJed Brown 
10290a96aa3bSJed Brown       if (flgPattern) {
10300a96aa3bSJed Brown         DMRefinePatternCtx *ctx;
10310a96aa3bSJed Brown         PetscInt            maxLevel;
10320a96aa3bSJed Brown 
10339566063dSJacob Faibussowitsch         PetscCall(DMForestGetMaximumRefinement(dm, &maxLevel));
10344dfa11a4SJacob Faibussowitsch         PetscCall(PetscNew(&ctx));
10350a96aa3bSJed Brown         ctx->maxLevel = PetscMin(maxLevel, P4EST_QMAXLEVEL);
10360a96aa3bSJed Brown         if (initLevel + ctx->maxLevel > minLevel) pforest->coarsen_hierarchy = PETSC_TRUE;
10370a96aa3bSJed Brown         switch (pattern) {
10380a96aa3bSJed Brown         case PATTERN_HASH:
10390a96aa3bSJed Brown           ctx->refine_fn      = DMRefinePattern_Hash;
10400a96aa3bSJed Brown           ctx->hashLikelihood = likelihood;
10410a96aa3bSJed Brown           break;
10420a96aa3bSJed Brown         case PATTERN_CORNER:
10430a96aa3bSJed Brown           ctx->corner    = corner;
10440a96aa3bSJed Brown           ctx->refine_fn = DMRefinePattern_Corner;
10450a96aa3bSJed Brown           break;
1046d71ae5a4SJacob Faibussowitsch         case PATTERN_CENTER:
1047d71ae5a4SJacob Faibussowitsch           ctx->refine_fn = DMRefinePattern_Center;
1048d71ae5a4SJacob Faibussowitsch           break;
10490a96aa3bSJed Brown         case PATTERN_FRACTAL:
10500a96aa3bSJed Brown           if (flgFractal) {
10510a96aa3bSJed Brown             PetscInt i;
10520a96aa3bSJed Brown 
10530a96aa3bSJed Brown             for (i = 0; i < ncorner; i++) ctx->fractal[corners[i]] = PETSC_TRUE;
10540a96aa3bSJed Brown           } else {
10550a96aa3bSJed Brown   #if !defined(P4_TO_P8)
10560a96aa3bSJed Brown             ctx->fractal[0] = ctx->fractal[1] = ctx->fractal[2] = PETSC_TRUE;
10570a96aa3bSJed Brown   #else
10580a96aa3bSJed Brown             ctx->fractal[0] = ctx->fractal[3] = ctx->fractal[5] = ctx->fractal[6] = PETSC_TRUE;
10590a96aa3bSJed Brown   #endif
10600a96aa3bSJed Brown           }
10610a96aa3bSJed Brown           ctx->refine_fn = DMRefinePattern_Fractal;
10620a96aa3bSJed Brown           break;
1063d71ae5a4SJacob Faibussowitsch         default:
1064d71ae5a4SJacob Faibussowitsch           SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Not a valid refinement pattern");
10650a96aa3bSJed Brown         }
10660a96aa3bSJed Brown 
10670a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)ctx;
1068792fecdfSBarry Smith         PetscCallP4est(p4est_refine, (pforest->forest, 1, ctx->refine_fn, NULL));
1069792fecdfSBarry Smith         PetscCallP4est(p4est_balance, (pforest->forest, P4EST_CONNECT_FULL, NULL));
10709566063dSJacob Faibussowitsch         PetscCall(PetscFree(ctx));
10710a96aa3bSJed Brown         pforest->forest->user_pointer = (void *)dm;
10720a96aa3bSJed Brown       }
10730a96aa3bSJed Brown     }
10740a96aa3bSJed Brown   }
10750a96aa3bSJed Brown   if (pforest->coarsen_hierarchy) {
10760a96aa3bSJed Brown     PetscInt initLevel, currLevel, minLevel;
10770a96aa3bSJed Brown 
10789566063dSJacob Faibussowitsch     PetscCall(DMPforestGetRefinementLevel(dm, &currLevel));
10799566063dSJacob Faibussowitsch     PetscCall(DMForestGetInitialRefinement(dm, &initLevel));
10809566063dSJacob Faibussowitsch     PetscCall(DMForestGetMinimumRefinement(dm, &minLevel));
10810a96aa3bSJed Brown     if (currLevel > minLevel) {
10820a96aa3bSJed Brown       DM_Forest_pforest *coarse_pforest;
10830a96aa3bSJed Brown       DMLabel            coarsen;
10840a96aa3bSJed Brown       DM                 coarseDM;
10850a96aa3bSJed Brown 
10869566063dSJacob Faibussowitsch       PetscCall(DMForestTemplate(dm, MPI_COMM_NULL, &coarseDM));
10879566063dSJacob Faibussowitsch       PetscCall(DMForestSetAdaptivityPurpose(coarseDM, DM_ADAPT_COARSEN));
10889566063dSJacob Faibussowitsch       PetscCall(DMLabelCreate(PETSC_COMM_SELF, "coarsen", &coarsen));
10899566063dSJacob Faibussowitsch       PetscCall(DMLabelSetDefaultValue(coarsen, DM_ADAPT_COARSEN));
10909566063dSJacob Faibussowitsch       PetscCall(DMForestSetAdaptivityLabel(coarseDM, coarsen));
10919566063dSJacob Faibussowitsch       PetscCall(DMLabelDestroy(&coarsen));
10929566063dSJacob Faibussowitsch       PetscCall(DMSetCoarseDM(dm, coarseDM));
10939566063dSJacob Faibussowitsch       PetscCall(PetscObjectDereference((PetscObject)coarseDM));
10940a96aa3bSJed Brown       initLevel = currLevel == initLevel ? initLevel - 1 : initLevel;
10959566063dSJacob Faibussowitsch       PetscCall(DMForestSetInitialRefinement(coarseDM, initLevel));
10969566063dSJacob Faibussowitsch       PetscCall(DMForestSetMinimumRefinement(coarseDM, minLevel));
10970a96aa3bSJed Brown       coarse_pforest                    = (DM_Forest_pforest *)((DM_Forest *)coarseDM->data)->data;
10980a96aa3bSJed Brown       coarse_pforest->coarsen_hierarchy = PETSC_TRUE;
10990a96aa3bSJed Brown     }
11000a96aa3bSJed Brown   }
11010a96aa3bSJed Brown 
11020a96aa3bSJed Brown   { /* repartitioning and overlap */
11030a96aa3bSJed Brown     PetscMPIInt size, rank;
11040a96aa3bSJed Brown 
11059566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm), &size));
11069566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
11070a96aa3bSJed Brown     if ((size > 1) && (pforest->partition_for_coarsening || forest->cellWeights || forest->weightCapacity != 1. || forest->weightsFactor != 1.)) {
11080a96aa3bSJed Brown       PetscBool      copyForest  = PETSC_FALSE;
11090a96aa3bSJed Brown       p4est_t       *forest_copy = NULL;
11100a96aa3bSJed Brown       p4est_gloidx_t shipped     = 0;
11110a96aa3bSJed Brown 
11120a96aa3bSJed Brown       if (preCoarseToFine || coarseToPreFine) copyForest = PETSC_TRUE;
1113792fecdfSBarry Smith       if (copyForest) PetscCallP4estReturn(forest_copy, p4est_copy, (pforest->forest, 0));
11140a96aa3bSJed Brown 
11150a96aa3bSJed Brown       if (!forest->cellWeights && forest->weightCapacity == 1. && forest->weightsFactor == 1.) {
1116792fecdfSBarry Smith         PetscCallP4estReturn(shipped, p4est_partition_ext, (pforest->forest, (int)pforest->partition_for_coarsening, NULL));
11170a96aa3bSJed Brown       } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Non-uniform partition cases not implemented yet");
11180a96aa3bSJed Brown       if (shipped) ctx.anyChange = PETSC_TRUE;
11190a96aa3bSJed Brown       if (forest_copy) {
11200a96aa3bSJed Brown         if (preCoarseToFine || coarseToPreFine) {
11210a96aa3bSJed Brown           PetscSF        repartSF; /* repartSF has roots in the old partition */
11220a96aa3bSJed Brown           PetscInt       pStart = -1, pEnd = -1, p;
11230a96aa3bSJed Brown           PetscInt       numRoots, numLeaves;
11240a96aa3bSJed Brown           PetscSFNode   *repartRoots;
11250a96aa3bSJed Brown           p4est_gloidx_t postStart  = pforest->forest->global_first_quadrant[rank];
11260a96aa3bSJed Brown           p4est_gloidx_t postEnd    = pforest->forest->global_first_quadrant[rank + 1];
11270a96aa3bSJed Brown           p4est_gloidx_t partOffset = postStart;
11280a96aa3bSJed Brown 
11290a96aa3bSJed Brown           numRoots  = (PetscInt)(forest_copy->global_first_quadrant[rank + 1] - forest_copy->global_first_quadrant[rank]);
11300a96aa3bSJed Brown           numLeaves = (PetscInt)(postEnd - postStart);
11319566063dSJacob Faibussowitsch           PetscCall(DMPforestComputeOverlappingRanks(size, rank, pforest->forest, forest_copy, &pStart, &pEnd));
11329566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1((PetscInt)pforest->forest->local_num_quadrants, &repartRoots));
11330a96aa3bSJed Brown           for (p = pStart; p < pEnd; p++) {
11340a96aa3bSJed Brown             p4est_gloidx_t preStart = forest_copy->global_first_quadrant[p];
11350a96aa3bSJed Brown             p4est_gloidx_t preEnd   = forest_copy->global_first_quadrant[p + 1];
11360a96aa3bSJed Brown             PetscInt       q;
11370a96aa3bSJed Brown 
11380a96aa3bSJed Brown             if (preEnd == preStart) continue;
113908401ef6SPierre Jolivet             PetscCheck(preStart <= postStart, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Bad partition overlap computation");
11400a96aa3bSJed Brown             preEnd = preEnd > postEnd ? postEnd : preEnd;
11410a96aa3bSJed Brown             for (q = partOffset; q < preEnd; q++) {
11420a96aa3bSJed Brown               repartRoots[q - postStart].rank  = p;
11430a96aa3bSJed Brown               repartRoots[q - postStart].index = partOffset - preStart;
11440a96aa3bSJed Brown             }
11450a96aa3bSJed Brown             partOffset = preEnd;
11460a96aa3bSJed Brown           }
11479566063dSJacob Faibussowitsch           PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &repartSF));
11489566063dSJacob Faibussowitsch           PetscCall(PetscSFSetGraph(repartSF, numRoots, numLeaves, NULL, PETSC_OWN_POINTER, repartRoots, PETSC_OWN_POINTER));
11499566063dSJacob Faibussowitsch           PetscCall(PetscSFSetUp(repartSF));
11500a96aa3bSJed Brown           if (preCoarseToFine) {
11510a96aa3bSJed Brown             PetscSF         repartSFembed, preCoarseToFineNew;
11520a96aa3bSJed Brown             PetscInt        nleaves;
11530a96aa3bSJed Brown             const PetscInt *leaves;
11540a96aa3bSJed Brown 
11559566063dSJacob Faibussowitsch             PetscCall(PetscSFSetUp(preCoarseToFine));
11569566063dSJacob Faibussowitsch             PetscCall(PetscSFGetGraph(preCoarseToFine, NULL, &nleaves, &leaves, NULL));
11570a96aa3bSJed Brown             if (leaves) {
11589566063dSJacob Faibussowitsch               PetscCall(PetscSFCreateEmbeddedRootSF(repartSF, nleaves, leaves, &repartSFembed));
11590a96aa3bSJed Brown             } else {
11600a96aa3bSJed Brown               repartSFembed = repartSF;
11619566063dSJacob Faibussowitsch               PetscCall(PetscObjectReference((PetscObject)repartSFembed));
11620a96aa3bSJed Brown             }
11639566063dSJacob Faibussowitsch             PetscCall(PetscSFCompose(preCoarseToFine, repartSFembed, &preCoarseToFineNew));
11649566063dSJacob Faibussowitsch             PetscCall(PetscSFDestroy(&preCoarseToFine));
11659566063dSJacob Faibussowitsch             PetscCall(PetscSFDestroy(&repartSFembed));
11660a96aa3bSJed Brown             preCoarseToFine = preCoarseToFineNew;
11670a96aa3bSJed Brown           }
11680a96aa3bSJed Brown           if (coarseToPreFine) {
11690a96aa3bSJed Brown             PetscSF repartSFinv, coarseToPreFineNew;
11700a96aa3bSJed Brown 
11719566063dSJacob Faibussowitsch             PetscCall(PetscSFCreateInverseSF(repartSF, &repartSFinv));
11729566063dSJacob Faibussowitsch             PetscCall(PetscSFCompose(repartSFinv, coarseToPreFine, &coarseToPreFineNew));
11739566063dSJacob Faibussowitsch             PetscCall(PetscSFDestroy(&coarseToPreFine));
11749566063dSJacob Faibussowitsch             PetscCall(PetscSFDestroy(&repartSFinv));
11750a96aa3bSJed Brown             coarseToPreFine = coarseToPreFineNew;
11760a96aa3bSJed Brown           }
11779566063dSJacob Faibussowitsch           PetscCall(PetscSFDestroy(&repartSF));
11780a96aa3bSJed Brown         }
1179792fecdfSBarry Smith         PetscCallP4est(p4est_destroy, (forest_copy));
11800a96aa3bSJed Brown       }
11810a96aa3bSJed Brown     }
11820a96aa3bSJed Brown     if (size > 1) {
11830a96aa3bSJed Brown       PetscInt overlap;
11840a96aa3bSJed Brown 
11859566063dSJacob Faibussowitsch       PetscCall(DMForestGetPartitionOverlap(dm, &overlap));
11860a96aa3bSJed Brown 
11870a96aa3bSJed Brown       if (adaptFrom) {
11880a96aa3bSJed Brown         PetscInt aoverlap;
11890a96aa3bSJed Brown 
11909566063dSJacob Faibussowitsch         PetscCall(DMForestGetPartitionOverlap(adaptFrom, &aoverlap));
1191ad540459SPierre Jolivet         if (aoverlap != overlap) ctx.anyChange = PETSC_TRUE;
11920a96aa3bSJed Brown       }
11930a96aa3bSJed Brown 
11940a96aa3bSJed Brown       if (overlap > 0) {
11950a96aa3bSJed Brown         PetscInt i, cLocalStart;
11960a96aa3bSJed Brown         PetscInt cEnd;
11970a96aa3bSJed Brown         PetscSF  preCellSF = NULL, cellSF = NULL;
11980a96aa3bSJed Brown 
1199792fecdfSBarry Smith         PetscCallP4estReturn(pforest->ghost, p4est_ghost_new, (pforest->forest, P4EST_CONNECT_FULL));
1200792fecdfSBarry Smith         PetscCallP4estReturn(pforest->lnodes, p4est_lnodes_new, (pforest->forest, pforest->ghost, -P4EST_DIM));
1201792fecdfSBarry Smith         PetscCallP4est(p4est_ghost_support_lnodes, (pforest->forest, pforest->lnodes, pforest->ghost));
1202792fecdfSBarry Smith         for (i = 1; i < overlap; i++) PetscCallP4est(p4est_ghost_expand_by_lnodes, (pforest->forest, pforest->lnodes, pforest->ghost));
12030a96aa3bSJed Brown 
12040a96aa3bSJed Brown         cLocalStart = pforest->cLocalStart = pforest->ghost->proc_offsets[rank];
12050a96aa3bSJed Brown         cEnd                               = pforest->forest->local_num_quadrants + pforest->ghost->proc_offsets[size];
12060a96aa3bSJed Brown 
12070a96aa3bSJed Brown         /* shift sfs by cLocalStart, expand by cell SFs */
12080a96aa3bSJed Brown         if (preCoarseToFine || coarseToPreFine) {
12099566063dSJacob Faibussowitsch           if (adaptFrom) PetscCall(DMForestGetCellSF(adaptFrom, &preCellSF));
12100a96aa3bSJed Brown           dm->setupcalled = PETSC_TRUE;
12119566063dSJacob Faibussowitsch           PetscCall(DMForestGetCellSF(dm, &cellSF));
12120a96aa3bSJed Brown         }
12130a96aa3bSJed Brown         if (preCoarseToFine) {
12140a96aa3bSJed Brown           PetscSF            preCoarseToFineNew;
12150a96aa3bSJed Brown           PetscInt           nleaves, nroots, *leavesNew, i, nleavesNew;
12160a96aa3bSJed Brown           const PetscInt    *leaves;
12170a96aa3bSJed Brown           const PetscSFNode *remotes;
12180a96aa3bSJed Brown           PetscSFNode       *remotesAll;
12190a96aa3bSJed Brown 
12209566063dSJacob Faibussowitsch           PetscCall(PetscSFSetUp(preCoarseToFine));
12219566063dSJacob Faibussowitsch           PetscCall(PetscSFGetGraph(preCoarseToFine, &nroots, &nleaves, &leaves, &remotes));
12229566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(cEnd, &remotesAll));
12230a96aa3bSJed Brown           for (i = 0; i < cEnd; i++) {
12240a96aa3bSJed Brown             remotesAll[i].rank  = -1;
12250a96aa3bSJed Brown             remotesAll[i].index = -1;
12260a96aa3bSJed Brown           }
12270a96aa3bSJed Brown           for (i = 0; i < nleaves; i++) remotesAll[(leaves ? leaves[i] : i) + cLocalStart] = remotes[i];
12289566063dSJacob Faibussowitsch           PetscCall(PetscSFSetUp(cellSF));
12299566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastBegin(cellSF, MPIU_2INT, remotesAll, remotesAll, MPI_REPLACE));
12309566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastEnd(cellSF, MPIU_2INT, remotesAll, remotesAll, MPI_REPLACE));
12310a96aa3bSJed Brown           nleavesNew = 0;
12320a96aa3bSJed Brown           for (i = 0; i < nleaves; i++) {
12330a96aa3bSJed Brown             if (remotesAll[i].rank >= 0) nleavesNew++;
12340a96aa3bSJed Brown           }
12359566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(nleavesNew, &leavesNew));
12360a96aa3bSJed Brown           nleavesNew = 0;
12370a96aa3bSJed Brown           for (i = 0; i < nleaves; i++) {
12380a96aa3bSJed Brown             if (remotesAll[i].rank >= 0) {
12390a96aa3bSJed Brown               leavesNew[nleavesNew] = i;
12400a96aa3bSJed Brown               if (i > nleavesNew) remotesAll[nleavesNew] = remotesAll[i];
12410a96aa3bSJed Brown               nleavesNew++;
12420a96aa3bSJed Brown             }
12430a96aa3bSJed Brown           }
12449566063dSJacob Faibussowitsch           PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &preCoarseToFineNew));
12450a96aa3bSJed Brown           if (nleavesNew < cEnd) {
12469566063dSJacob Faibussowitsch             PetscCall(PetscSFSetGraph(preCoarseToFineNew, nroots, nleavesNew, leavesNew, PETSC_OWN_POINTER, remotesAll, PETSC_COPY_VALUES));
12470a96aa3bSJed Brown           } else { /* all cells are leaves */
12489566063dSJacob Faibussowitsch             PetscCall(PetscFree(leavesNew));
12499566063dSJacob Faibussowitsch             PetscCall(PetscSFSetGraph(preCoarseToFineNew, nroots, nleavesNew, NULL, PETSC_OWN_POINTER, remotesAll, PETSC_COPY_VALUES));
12500a96aa3bSJed Brown           }
12519566063dSJacob Faibussowitsch           PetscCall(PetscFree(remotesAll));
12529566063dSJacob Faibussowitsch           PetscCall(PetscSFDestroy(&preCoarseToFine));
12530a96aa3bSJed Brown           preCoarseToFine = preCoarseToFineNew;
12540a96aa3bSJed Brown           preCoarseToFine = preCoarseToFineNew;
12550a96aa3bSJed Brown         }
12560a96aa3bSJed Brown         if (coarseToPreFine) {
12570a96aa3bSJed Brown           PetscSF            coarseToPreFineNew;
12580a96aa3bSJed Brown           PetscInt           nleaves, nroots, i, nleavesCellSF, nleavesExpanded, *leavesNew;
12590a96aa3bSJed Brown           const PetscInt    *leaves;
12600a96aa3bSJed Brown           const PetscSFNode *remotes;
12610a96aa3bSJed Brown           PetscSFNode       *remotesNew, *remotesNewRoot, *remotesExpanded;
12620a96aa3bSJed Brown 
12639566063dSJacob Faibussowitsch           PetscCall(PetscSFSetUp(coarseToPreFine));
12649566063dSJacob Faibussowitsch           PetscCall(PetscSFGetGraph(coarseToPreFine, &nroots, &nleaves, &leaves, &remotes));
12659566063dSJacob Faibussowitsch           PetscCall(PetscSFGetGraph(preCellSF, NULL, &nleavesCellSF, NULL, NULL));
12669566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(nroots, &remotesNewRoot));
12679566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(nleaves, &remotesNew));
12680a96aa3bSJed Brown           for (i = 0; i < nroots; i++) {
12690a96aa3bSJed Brown             remotesNewRoot[i].rank  = rank;
12700a96aa3bSJed Brown             remotesNewRoot[i].index = i + cLocalStart;
12710a96aa3bSJed Brown           }
12729566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastBegin(coarseToPreFine, MPIU_2INT, remotesNewRoot, remotesNew, MPI_REPLACE));
12739566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastEnd(coarseToPreFine, MPIU_2INT, remotesNewRoot, remotesNew, MPI_REPLACE));
12749566063dSJacob Faibussowitsch           PetscCall(PetscFree(remotesNewRoot));
12759566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(nleavesCellSF, &remotesExpanded));
12760a96aa3bSJed Brown           for (i = 0; i < nleavesCellSF; i++) {
12770a96aa3bSJed Brown             remotesExpanded[i].rank  = -1;
12780a96aa3bSJed Brown             remotesExpanded[i].index = -1;
12790a96aa3bSJed Brown           }
12800a96aa3bSJed Brown           for (i = 0; i < nleaves; i++) remotesExpanded[leaves ? leaves[i] : i] = remotesNew[i];
12819566063dSJacob Faibussowitsch           PetscCall(PetscFree(remotesNew));
12829566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastBegin(preCellSF, MPIU_2INT, remotesExpanded, remotesExpanded, MPI_REPLACE));
12839566063dSJacob Faibussowitsch           PetscCall(PetscSFBcastEnd(preCellSF, MPIU_2INT, remotesExpanded, remotesExpanded, MPI_REPLACE));
12840a96aa3bSJed Brown 
12850a96aa3bSJed Brown           nleavesExpanded = 0;
12860a96aa3bSJed Brown           for (i = 0; i < nleavesCellSF; i++) {
12870a96aa3bSJed Brown             if (remotesExpanded[i].rank >= 0) nleavesExpanded++;
12880a96aa3bSJed Brown           }
12899566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(nleavesExpanded, &leavesNew));
12900a96aa3bSJed Brown           nleavesExpanded = 0;
12910a96aa3bSJed Brown           for (i = 0; i < nleavesCellSF; i++) {
12920a96aa3bSJed Brown             if (remotesExpanded[i].rank >= 0) {
12930a96aa3bSJed Brown               leavesNew[nleavesExpanded] = i;
12940a96aa3bSJed Brown               if (i > nleavesExpanded) remotesExpanded[nleavesExpanded] = remotes[i];
12950a96aa3bSJed Brown               nleavesExpanded++;
12960a96aa3bSJed Brown             }
12970a96aa3bSJed Brown           }
12989566063dSJacob Faibussowitsch           PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &coarseToPreFineNew));
12990a96aa3bSJed Brown           if (nleavesExpanded < nleavesCellSF) {
13009566063dSJacob Faibussowitsch             PetscCall(PetscSFSetGraph(coarseToPreFineNew, cEnd, nleavesExpanded, leavesNew, PETSC_OWN_POINTER, remotesExpanded, PETSC_COPY_VALUES));
13010a96aa3bSJed Brown           } else {
13029566063dSJacob Faibussowitsch             PetscCall(PetscFree(leavesNew));
13039566063dSJacob Faibussowitsch             PetscCall(PetscSFSetGraph(coarseToPreFineNew, cEnd, nleavesExpanded, NULL, PETSC_OWN_POINTER, remotesExpanded, PETSC_COPY_VALUES));
13040a96aa3bSJed Brown           }
13059566063dSJacob Faibussowitsch           PetscCall(PetscFree(remotesExpanded));
13069566063dSJacob Faibussowitsch           PetscCall(PetscSFDestroy(&coarseToPreFine));
13070a96aa3bSJed Brown           coarseToPreFine = coarseToPreFineNew;
13080a96aa3bSJed Brown         }
13090a96aa3bSJed Brown       }
13100a96aa3bSJed Brown     }
13110a96aa3bSJed Brown   }
13120a96aa3bSJed Brown   forest->preCoarseToFine = preCoarseToFine;
13130a96aa3bSJed Brown   forest->coarseToPreFine = coarseToPreFine;
13140a96aa3bSJed Brown   dm->setupcalled         = PETSC_TRUE;
1315712fec58SPierre Jolivet   PetscCall(MPIU_Allreduce(&ctx.anyChange, &(pforest->adaptivitySuccess), 1, MPIU_BOOL, MPI_LOR, PetscObjectComm((PetscObject)dm)));
13169566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, NULL));
13173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
13180a96aa3bSJed Brown }
13190a96aa3bSJed Brown 
13200a96aa3bSJed Brown   #define DMForestGetAdaptivitySuccess_pforest _append_pforest(DMForestGetAdaptivitySuccess)
1321d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestGetAdaptivitySuccess_pforest(DM dm, PetscBool *success)
1322d71ae5a4SJacob Faibussowitsch {
13230a96aa3bSJed Brown   DM_Forest         *forest;
13240a96aa3bSJed Brown   DM_Forest_pforest *pforest;
13250a96aa3bSJed Brown 
13260a96aa3bSJed Brown   PetscFunctionBegin;
13270a96aa3bSJed Brown   forest   = (DM_Forest *)dm->data;
13280a96aa3bSJed Brown   pforest  = (DM_Forest_pforest *)forest->data;
13290a96aa3bSJed Brown   *success = pforest->adaptivitySuccess;
13303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
13310a96aa3bSJed Brown }
13320a96aa3bSJed Brown 
13330a96aa3bSJed Brown   #define DMView_ASCII_pforest _append_pforest(DMView_ASCII)
1334d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMView_ASCII_pforest(PetscObject odm, PetscViewer viewer)
1335d71ae5a4SJacob Faibussowitsch {
13360a96aa3bSJed Brown   DM dm = (DM)odm;
13370a96aa3bSJed Brown 
13380a96aa3bSJed Brown   PetscFunctionBegin;
13390a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
13400a96aa3bSJed Brown   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
13419566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
13420a96aa3bSJed Brown   switch (viewer->format) {
13430a96aa3bSJed Brown   case PETSC_VIEWER_DEFAULT:
13449371c9d4SSatish Balay   case PETSC_VIEWER_ASCII_INFO: {
13450a96aa3bSJed Brown     PetscInt    dim;
13460a96aa3bSJed Brown     const char *name;
13470a96aa3bSJed Brown 
13489566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)dm, &name));
13499566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(dm, &dim));
135063a3b9bcSJacob Faibussowitsch     if (name) PetscCall(PetscViewerASCIIPrintf(viewer, "Forest %s in %" PetscInt_FMT " dimensions:\n", name, dim));
135163a3b9bcSJacob Faibussowitsch     else PetscCall(PetscViewerASCIIPrintf(viewer, "Forest in %" PetscInt_FMT " dimensions:\n", dim));
1352f4d061e9SPierre Jolivet   } /* fall through */
13530a96aa3bSJed Brown   case PETSC_VIEWER_ASCII_INFO_DETAIL:
13549371c9d4SSatish Balay   case PETSC_VIEWER_LOAD_BALANCE: {
13550a96aa3bSJed Brown     DM plex;
13560a96aa3bSJed Brown 
13579566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(dm, &plex));
13589566063dSJacob Faibussowitsch     PetscCall(DMView(plex, viewer));
13599371c9d4SSatish Balay   } break;
1360d71ae5a4SJacob Faibussowitsch   default:
1361d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "No support for format '%s'", PetscViewerFormats[viewer->format]);
13620a96aa3bSJed Brown   }
13633ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
13640a96aa3bSJed Brown }
13650a96aa3bSJed Brown 
13660a96aa3bSJed Brown   #define DMView_VTK_pforest _append_pforest(DMView_VTK)
1367d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMView_VTK_pforest(PetscObject odm, PetscViewer viewer)
1368d71ae5a4SJacob Faibussowitsch {
13690a96aa3bSJed Brown   DM                 dm      = (DM)odm;
13700a96aa3bSJed Brown   DM_Forest         *forest  = (DM_Forest *)dm->data;
13710a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)forest->data;
13720a96aa3bSJed Brown   PetscBool          isvtk;
13730a96aa3bSJed Brown   PetscReal          vtkScale = 1. - PETSC_MACHINE_EPSILON;
13740a96aa3bSJed Brown   PetscViewer_VTK   *vtk      = (PetscViewer_VTK *)viewer->data;
13750a96aa3bSJed Brown   const char        *name;
13760a96aa3bSJed Brown   char              *filenameStrip = NULL;
13770a96aa3bSJed Brown   PetscBool          hasExt;
13780a96aa3bSJed Brown   size_t             len;
13790a96aa3bSJed Brown   p4est_geometry_t  *geom;
13800a96aa3bSJed Brown 
13810a96aa3bSJed Brown   PetscFunctionBegin;
13820a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
13830a96aa3bSJed Brown   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
13849566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
13850a96aa3bSJed Brown   geom = pforest->topo->geom;
13869566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
138728b400f6SJacob Faibussowitsch   PetscCheck(isvtk, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_INCOMP, "Cannot use viewer type %s", ((PetscObject)viewer)->type_name);
13880a96aa3bSJed Brown   switch (viewer->format) {
13890a96aa3bSJed Brown   case PETSC_VIEWER_VTK_VTU:
139028b400f6SJacob Faibussowitsch     PetscCheck(pforest->forest, PetscObjectComm(odm), PETSC_ERR_ARG_WRONG, "DM has not been setup with a valid forest");
13910a96aa3bSJed Brown     name = vtk->filename;
13929566063dSJacob Faibussowitsch     PetscCall(PetscStrlen(name, &len));
13939566063dSJacob Faibussowitsch     PetscCall(PetscStrcasecmp(name + len - 4, ".vtu", &hasExt));
13940a96aa3bSJed Brown     if (hasExt) {
13959566063dSJacob Faibussowitsch       PetscCall(PetscStrallocpy(name, &filenameStrip));
13960a96aa3bSJed Brown       filenameStrip[len - 4] = '\0';
13970a96aa3bSJed Brown       name                   = filenameStrip;
13980a96aa3bSJed Brown     }
1399792fecdfSBarry Smith     if (!pforest->topo->geom) PetscCallP4estReturn(geom, p4est_geometry_new_connectivity, (pforest->topo->conn));
14000a96aa3bSJed Brown     {
14010a96aa3bSJed Brown       p4est_vtk_context_t *pvtk;
14020a96aa3bSJed Brown       int                  footerr;
14030a96aa3bSJed Brown 
1404792fecdfSBarry Smith       PetscCallP4estReturn(pvtk, p4est_vtk_context_new, (pforest->forest, name));
1405792fecdfSBarry Smith       PetscCallP4est(p4est_vtk_context_set_geom, (pvtk, geom));
1406792fecdfSBarry Smith       PetscCallP4est(p4est_vtk_context_set_scale, (pvtk, (double)vtkScale));
1407792fecdfSBarry Smith       PetscCallP4estReturn(pvtk, p4est_vtk_write_header, (pvtk));
140828b400f6SJacob Faibussowitsch       PetscCheck(pvtk, PetscObjectComm((PetscObject)odm), PETSC_ERR_LIB, P4EST_STRING "_vtk_write_header() failed");
14099371c9d4SSatish Balay       PetscCallP4estReturn(pvtk, p4est_vtk_write_cell_dataf,
14109371c9d4SSatish Balay                            (pvtk, 1, /* write tree */
14110a96aa3bSJed Brown                             1,       /* write level */
14120a96aa3bSJed Brown                             1,       /* write rank */
14130a96aa3bSJed Brown                             0,       /* do not wrap rank */
14140a96aa3bSJed Brown                             0,       /* no scalar fields */
14150a96aa3bSJed Brown                             0,       /* no vector fields */
14160a96aa3bSJed Brown                             pvtk));
141728b400f6SJacob Faibussowitsch       PetscCheck(pvtk, PetscObjectComm((PetscObject)odm), PETSC_ERR_LIB, P4EST_STRING "_vtk_write_cell_dataf() failed");
1418792fecdfSBarry Smith       PetscCallP4estReturn(footerr, p4est_vtk_write_footer, (pvtk));
141928b400f6SJacob Faibussowitsch       PetscCheck(!footerr, PetscObjectComm((PetscObject)odm), PETSC_ERR_LIB, P4EST_STRING "_vtk_write_footer() failed");
14200a96aa3bSJed Brown     }
1421792fecdfSBarry Smith     if (!pforest->topo->geom) PetscCallP4est(p4est_geometry_destroy, (geom));
14229566063dSJacob Faibussowitsch     PetscCall(PetscFree(filenameStrip));
14230a96aa3bSJed Brown     break;
1424d71ae5a4SJacob Faibussowitsch   default:
1425d71ae5a4SJacob Faibussowitsch     SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "No support for format '%s'", PetscViewerFormats[viewer->format]);
14260a96aa3bSJed Brown   }
14273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14280a96aa3bSJed Brown }
14290a96aa3bSJed Brown 
14300a96aa3bSJed Brown   #define DMView_HDF5_pforest _append_pforest(DMView_HDF5)
1431d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMView_HDF5_pforest(DM dm, PetscViewer viewer)
1432d71ae5a4SJacob Faibussowitsch {
14330a96aa3bSJed Brown   DM plex;
14340a96aa3bSJed Brown 
14350a96aa3bSJed Brown   PetscFunctionBegin;
14369566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
14379566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
14389566063dSJacob Faibussowitsch   PetscCall(DMView(plex, viewer));
14393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14400a96aa3bSJed Brown }
14410a96aa3bSJed Brown 
14420a96aa3bSJed Brown   #define DMView_GLVis_pforest _append_pforest(DMView_GLVis)
1443d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMView_GLVis_pforest(DM dm, PetscViewer viewer)
1444d71ae5a4SJacob Faibussowitsch {
14450a96aa3bSJed Brown   DM plex;
14460a96aa3bSJed Brown 
14470a96aa3bSJed Brown   PetscFunctionBegin;
14489566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
14499566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
14509566063dSJacob Faibussowitsch   PetscCall(DMView(plex, viewer));
14513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14520a96aa3bSJed Brown }
14530a96aa3bSJed Brown 
14540a96aa3bSJed Brown   #define DMView_pforest _append_pforest(DMView)
1455d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMView_pforest(DM dm, PetscViewer viewer)
1456d71ae5a4SJacob Faibussowitsch {
14570a96aa3bSJed Brown   PetscBool isascii, isvtk, ishdf5, isglvis;
14580a96aa3bSJed Brown 
14590a96aa3bSJed Brown   PetscFunctionBegin;
14600a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
14610a96aa3bSJed Brown   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
14629566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
14639566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERVTK, &isvtk));
14649566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5));
14659566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERGLVIS, &isglvis));
14660a96aa3bSJed Brown   if (isascii) {
14679566063dSJacob Faibussowitsch     PetscCall(DMView_ASCII_pforest((PetscObject)dm, viewer));
14680a96aa3bSJed Brown   } else if (isvtk) {
14699566063dSJacob Faibussowitsch     PetscCall(DMView_VTK_pforest((PetscObject)dm, viewer));
14700a96aa3bSJed Brown   } else if (ishdf5) {
14719566063dSJacob Faibussowitsch     PetscCall(DMView_HDF5_pforest(dm, viewer));
14720a96aa3bSJed Brown   } else if (isglvis) {
14739566063dSJacob Faibussowitsch     PetscCall(DMView_GLVis_pforest(dm, viewer));
14740a96aa3bSJed Brown   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Viewer not supported (not VTK, HDF5, or GLVis)");
14753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14760a96aa3bSJed Brown }
14770a96aa3bSJed Brown 
1478d71ae5a4SJacob Faibussowitsch static PetscErrorCode PforestConnectivityEnumerateFacets(p4est_connectivity_t *conn, PetscInt **tree_face_to_uniq)
1479d71ae5a4SJacob Faibussowitsch {
14800a96aa3bSJed Brown   PetscInt *ttf, f, t, g, count;
14810a96aa3bSJed Brown   PetscInt  numFacets;
14820a96aa3bSJed Brown 
14830a96aa3bSJed Brown   PetscFunctionBegin;
14840a96aa3bSJed Brown   numFacets = conn->num_trees * P4EST_FACES;
14859566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(numFacets, &ttf));
14860a96aa3bSJed Brown   for (f = 0; f < numFacets; f++) ttf[f] = -1;
14870a96aa3bSJed Brown   for (g = 0, count = 0, t = 0; t < conn->num_trees; t++) {
14880a96aa3bSJed Brown     for (f = 0; f < P4EST_FACES; f++, g++) {
14890a96aa3bSJed Brown       if (ttf[g] == -1) {
14900a96aa3bSJed Brown         PetscInt ng;
14910a96aa3bSJed Brown 
14920a96aa3bSJed Brown         ttf[g]  = count++;
14930a96aa3bSJed Brown         ng      = conn->tree_to_tree[g] * P4EST_FACES + (conn->tree_to_face[g] % P4EST_FACES);
14940a96aa3bSJed Brown         ttf[ng] = ttf[g];
14950a96aa3bSJed Brown       }
14960a96aa3bSJed Brown     }
14970a96aa3bSJed Brown   }
14980a96aa3bSJed Brown   *tree_face_to_uniq = ttf;
14993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15000a96aa3bSJed Brown }
15010a96aa3bSJed Brown 
1502d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPlexCreateConnectivity_pforest(DM dm, p4est_connectivity_t **connOut, PetscInt **tree_face_to_uniq)
1503d71ae5a4SJacob Faibussowitsch {
15040a96aa3bSJed Brown   p4est_topidx_t numTrees, numVerts, numCorns, numCtt;
15050a96aa3bSJed Brown   PetscSection   ctt;
15060a96aa3bSJed Brown   #if defined(P4_TO_P8)
15070a96aa3bSJed Brown   p4est_topidx_t numEdges, numEtt;
15080a96aa3bSJed Brown   PetscSection   ett;
15090a96aa3bSJed Brown   PetscInt       eStart, eEnd, e, ettSize;
15100a96aa3bSJed Brown   PetscInt       vertOff = 1 + P4EST_FACES + P8EST_EDGES;
15110a96aa3bSJed Brown   PetscInt       edgeOff = 1 + P4EST_FACES;
15120a96aa3bSJed Brown   #else
15130a96aa3bSJed Brown   PetscInt vertOff = 1 + P4EST_FACES;
15140a96aa3bSJed Brown   #endif
15150a96aa3bSJed Brown   p4est_connectivity_t *conn;
15160a96aa3bSJed Brown   PetscInt              cStart, cEnd, c, vStart, vEnd, v, fStart, fEnd, f;
15170a96aa3bSJed Brown   PetscInt             *star = NULL, *closure = NULL, closureSize, starSize, cttSize;
15180a96aa3bSJed Brown   PetscInt             *ttf;
15190a96aa3bSJed Brown 
15200a96aa3bSJed Brown   PetscFunctionBegin;
15210a96aa3bSJed Brown   /* 1: count objects, allocate */
15229566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd));
15239566063dSJacob Faibussowitsch   PetscCall(P4estTopidxCast(cEnd - cStart, &numTrees));
15240a96aa3bSJed Brown   numVerts = P4EST_CHILDREN * numTrees;
15259566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
15269566063dSJacob Faibussowitsch   PetscCall(P4estTopidxCast(vEnd - vStart, &numCorns));
15279566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &ctt));
15289566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(ctt, vStart, vEnd));
15290a96aa3bSJed Brown   for (v = vStart; v < vEnd; v++) {
15300a96aa3bSJed Brown     PetscInt s;
15310a96aa3bSJed Brown 
15329566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star));
15330a96aa3bSJed Brown     for (s = 0; s < starSize; s++) {
15340a96aa3bSJed Brown       PetscInt p = star[2 * s];
15350a96aa3bSJed Brown 
15360a96aa3bSJed Brown       if (p >= cStart && p < cEnd) {
15370a96aa3bSJed Brown         /* we want to count every time cell p references v, so we see how many times it comes up in the closure.  This
15380a96aa3bSJed Brown          * only protects against periodicity problems */
15399566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
154063a3b9bcSJacob Faibussowitsch         PetscCheck(closureSize == P4EST_INSUL, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Cell %" PetscInt_FMT " with wrong closure size %" PetscInt_FMT " != %d", p, closureSize, P4EST_INSUL);
15410a96aa3bSJed Brown         for (c = 0; c < P4EST_CHILDREN; c++) {
15420a96aa3bSJed Brown           PetscInt cellVert = closure[2 * (c + vertOff)];
15430a96aa3bSJed Brown 
15441dca8a05SBarry Smith           PetscCheck(cellVert >= vStart && cellVert < vEnd, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Non-standard closure: vertices");
154548a46eb9SPierre Jolivet           if (cellVert == v) PetscCall(PetscSectionAddDof(ctt, v, 1));
15460a96aa3bSJed Brown         }
15479566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
15480a96aa3bSJed Brown       }
15490a96aa3bSJed Brown     }
15509566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star));
15510a96aa3bSJed Brown   }
15529566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(ctt));
15539566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(ctt, &cttSize));
15549566063dSJacob Faibussowitsch   PetscCall(P4estTopidxCast(cttSize, &numCtt));
15550a96aa3bSJed Brown   #if defined(P4_TO_P8)
15569566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(dm, P4EST_DIM - 1, &eStart, &eEnd));
15579566063dSJacob Faibussowitsch   PetscCall(P4estTopidxCast(eEnd - eStart, &numEdges));
15589566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &ett));
15599566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(ett, eStart, eEnd));
15600a96aa3bSJed Brown   for (e = eStart; e < eEnd; e++) {
15610a96aa3bSJed Brown     PetscInt s;
15620a96aa3bSJed Brown 
15639566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, e, PETSC_FALSE, &starSize, &star));
15640a96aa3bSJed Brown     for (s = 0; s < starSize; s++) {
15650a96aa3bSJed Brown       PetscInt p = star[2 * s];
15660a96aa3bSJed Brown 
15670a96aa3bSJed Brown       if (p >= cStart && p < cEnd) {
15680a96aa3bSJed Brown         /* we want to count every time cell p references e, so we see how many times it comes up in the closure.  This
15690a96aa3bSJed Brown          * only protects against periodicity problems */
15709566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
157108401ef6SPierre Jolivet         PetscCheck(closureSize == P4EST_INSUL, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Cell with wrong closure size");
15720a96aa3bSJed Brown         for (c = 0; c < P8EST_EDGES; c++) {
15730a96aa3bSJed Brown           PetscInt cellEdge = closure[2 * (c + edgeOff)];
15740a96aa3bSJed Brown 
15751dca8a05SBarry Smith           PetscCheck(cellEdge >= eStart && cellEdge < eEnd, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Non-standard closure: edges");
157648a46eb9SPierre Jolivet           if (cellEdge == e) PetscCall(PetscSectionAddDof(ett, e, 1));
15770a96aa3bSJed Brown         }
15789566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
15790a96aa3bSJed Brown       }
15800a96aa3bSJed Brown     }
15819566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, e, PETSC_FALSE, &starSize, &star));
15820a96aa3bSJed Brown   }
15839566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(ett));
15849566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(ett, &ettSize));
15859566063dSJacob Faibussowitsch   PetscCall(P4estTopidxCast(ettSize, &numEtt));
15860a96aa3bSJed Brown 
15870a96aa3bSJed Brown   /* This routine allocates space for the arrays, which we fill below */
1588792fecdfSBarry Smith   PetscCallP4estReturn(conn, p8est_connectivity_new, (numVerts, numTrees, numEdges, numEtt, numCorns, numCtt));
15890a96aa3bSJed Brown   #else
1590792fecdfSBarry Smith   PetscCallP4estReturn(conn, p4est_connectivity_new, (numVerts, numTrees, numCorns, numCtt));
15910a96aa3bSJed Brown   #endif
15920a96aa3bSJed Brown 
15930a96aa3bSJed Brown   /* 2: visit every face, determine neighboring cells(trees) */
15949566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(dm, 1, &fStart, &fEnd));
15959566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1((cEnd - cStart) * P4EST_FACES, &ttf));
15960a96aa3bSJed Brown   for (f = fStart; f < fEnd; f++) {
15970a96aa3bSJed Brown     PetscInt        numSupp, s;
15980a96aa3bSJed Brown     PetscInt        myFace[2] = {-1, -1};
15990a96aa3bSJed Brown     PetscInt        myOrnt[2] = {PETSC_MIN_INT, PETSC_MIN_INT};
16000a96aa3bSJed Brown     const PetscInt *supp;
16010a96aa3bSJed Brown 
16029566063dSJacob Faibussowitsch     PetscCall(DMPlexGetSupportSize(dm, f, &numSupp));
16031dca8a05SBarry Smith     PetscCheck(numSupp == 1 || numSupp == 2, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "point %" PetscInt_FMT " has facet with %" PetscInt_FMT " sides: must be 1 or 2 (boundary or conformal)", f, numSupp);
16049566063dSJacob Faibussowitsch     PetscCall(DMPlexGetSupport(dm, f, &supp));
16050a96aa3bSJed Brown 
16060a96aa3bSJed Brown     for (s = 0; s < numSupp; s++) {
16070a96aa3bSJed Brown       PetscInt p = supp[s];
16080a96aa3bSJed Brown 
16090a96aa3bSJed Brown       if (p >= cEnd) {
16100a96aa3bSJed Brown         numSupp--;
16110a96aa3bSJed Brown         if (s) supp = &supp[1 - s];
16120a96aa3bSJed Brown         break;
16130a96aa3bSJed Brown       }
16140a96aa3bSJed Brown     }
16150a96aa3bSJed Brown     for (s = 0; s < numSupp; s++) {
16160a96aa3bSJed Brown       PetscInt        p = supp[s], i;
16170a96aa3bSJed Brown       PetscInt        numCone;
16180a96aa3bSJed Brown       DMPolytopeType  ct;
16190a96aa3bSJed Brown       const PetscInt *cone;
16200a96aa3bSJed Brown       const PetscInt *ornt;
16210a96aa3bSJed Brown       PetscInt        orient = PETSC_MIN_INT;
16220a96aa3bSJed Brown 
16239566063dSJacob Faibussowitsch       PetscCall(DMPlexGetConeSize(dm, p, &numCone));
162463a3b9bcSJacob Faibussowitsch       PetscCheck(numCone == P4EST_FACES, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "cell %" PetscInt_FMT " has %" PetscInt_FMT " facets, expect %d", p, numCone, P4EST_FACES);
16259566063dSJacob Faibussowitsch       PetscCall(DMPlexGetCone(dm, p, &cone));
16269566063dSJacob Faibussowitsch       PetscCall(DMPlexGetCellType(dm, cone[0], &ct));
16279566063dSJacob Faibussowitsch       PetscCall(DMPlexGetConeOrientation(dm, p, &ornt));
16280a96aa3bSJed Brown       for (i = 0; i < P4EST_FACES; i++) {
16290a96aa3bSJed Brown         if (cone[i] == f) {
16300a96aa3bSJed Brown           orient = DMPolytopeConvertNewOrientation_Internal(ct, ornt[i]);
16310a96aa3bSJed Brown           break;
16320a96aa3bSJed Brown         }
16330a96aa3bSJed Brown       }
163463a3b9bcSJacob Faibussowitsch       PetscCheck(i < P4EST_FACES, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "cell %" PetscInt_FMT " faced %" PetscInt_FMT " mismatch", p, f);
16350a96aa3bSJed Brown       if (p < cStart || p >= cEnd) {
16360a96aa3bSJed Brown         DMPolytopeType ct;
16379566063dSJacob Faibussowitsch         PetscCall(DMPlexGetCellType(dm, p, &ct));
163863a3b9bcSJacob Faibussowitsch         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "cell %" PetscInt_FMT " (%s) should be in [%" PetscInt_FMT ", %" PetscInt_FMT ")", p, DMPolytopeTypes[ct], cStart, cEnd);
16390a96aa3bSJed Brown       }
16400a96aa3bSJed Brown       ttf[P4EST_FACES * (p - cStart) + PetscFaceToP4estFace[i]] = f - fStart;
16410a96aa3bSJed Brown       if (numSupp == 1) {
16420a96aa3bSJed Brown         /* boundary faces indicated by self reference */
16430a96aa3bSJed Brown         conn->tree_to_tree[P4EST_FACES * (p - cStart) + PetscFaceToP4estFace[i]] = p - cStart;
16440a96aa3bSJed Brown         conn->tree_to_face[P4EST_FACES * (p - cStart) + PetscFaceToP4estFace[i]] = (int8_t)PetscFaceToP4estFace[i];
16450a96aa3bSJed Brown       } else {
16460a96aa3bSJed Brown         const PetscInt N = P4EST_CHILDREN / 2;
16470a96aa3bSJed Brown 
16480a96aa3bSJed Brown         conn->tree_to_tree[P4EST_FACES * (p - cStart) + PetscFaceToP4estFace[i]] = supp[1 - s] - cStart;
16490a96aa3bSJed Brown         myFace[s]                                                                = PetscFaceToP4estFace[i];
16500a96aa3bSJed Brown         /* get the orientation of cell p in p4est-type closure to facet f, by composing the p4est-closure to
16510a96aa3bSJed Brown          * petsc-closure permutation and the petsc-closure to facet orientation */
16520a96aa3bSJed Brown         myOrnt[s] = DihedralCompose(N, orient, DMPolytopeConvertNewOrientation_Internal(ct, P4estFaceToPetscOrnt[myFace[s]]));
16530a96aa3bSJed Brown       }
16540a96aa3bSJed Brown     }
16550a96aa3bSJed Brown     if (numSupp == 2) {
16560a96aa3bSJed Brown       for (s = 0; s < numSupp; s++) {
16570a96aa3bSJed Brown         PetscInt       p = supp[s];
16580a96aa3bSJed Brown         PetscInt       orntAtoB;
16590a96aa3bSJed Brown         PetscInt       p4estOrient;
16600a96aa3bSJed Brown         const PetscInt N = P4EST_CHILDREN / 2;
16610a96aa3bSJed Brown 
16620a96aa3bSJed Brown         /* composing the forward permutation with the other cell's inverse permutation gives the self-to-neighbor
16630a96aa3bSJed Brown          * permutation of this cell-facet's cone */
16640a96aa3bSJed Brown         orntAtoB = DihedralCompose(N, DihedralInvert(N, myOrnt[1 - s]), myOrnt[s]);
16650a96aa3bSJed Brown 
16660a96aa3bSJed Brown         /* convert cone-description permutation (i.e., edges around facet) to cap-description permutation (i.e.,
16670a96aa3bSJed Brown          * vertices around facet) */
16680a96aa3bSJed Brown   #if !defined(P4_TO_P8)
16690a96aa3bSJed Brown         p4estOrient = orntAtoB < 0 ? -(orntAtoB + 1) : orntAtoB;
16700a96aa3bSJed Brown   #else
16710a96aa3bSJed Brown         {
16720a96aa3bSJed Brown           PetscInt firstVert      = orntAtoB < 0 ? ((-orntAtoB) % N) : orntAtoB;
16730a96aa3bSJed Brown           PetscInt p4estFirstVert = firstVert < 2 ? firstVert : (firstVert ^ 1);
16740a96aa3bSJed Brown 
16750a96aa3bSJed Brown           /* swap bits */
16760a96aa3bSJed Brown           p4estOrient = ((myFace[s] <= myFace[1 - s]) || (orntAtoB < 0)) ? p4estFirstVert : ((p4estFirstVert >> 1) | ((p4estFirstVert & 1) << 1));
16770a96aa3bSJed Brown         }
16780a96aa3bSJed Brown   #endif
16790a96aa3bSJed Brown         /* encode neighbor face and orientation in tree_to_face per p4est_connectivity standard (see
16800a96aa3bSJed Brown          * p4est_connectivity.h, p8est_connectivity.h) */
16810a96aa3bSJed Brown         conn->tree_to_face[P4EST_FACES * (p - cStart) + myFace[s]] = (int8_t)myFace[1 - s] + p4estOrient * P4EST_FACES;
16820a96aa3bSJed Brown       }
16830a96aa3bSJed Brown     }
16840a96aa3bSJed Brown   }
16850a96aa3bSJed Brown 
16860a96aa3bSJed Brown   #if defined(P4_TO_P8)
16870a96aa3bSJed Brown   /* 3: visit every edge */
16880a96aa3bSJed Brown   conn->ett_offset[0] = 0;
16890a96aa3bSJed Brown   for (e = eStart; e < eEnd; e++) {
16900a96aa3bSJed Brown     PetscInt off, s;
16910a96aa3bSJed Brown 
16929566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(ett, e, &off));
16930a96aa3bSJed Brown     conn->ett_offset[e - eStart] = (p4est_topidx_t)off;
16949566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, e, PETSC_FALSE, &starSize, &star));
16950a96aa3bSJed Brown     for (s = 0; s < starSize; s++) {
16960a96aa3bSJed Brown       PetscInt p = star[2 * s];
16970a96aa3bSJed Brown 
16980a96aa3bSJed Brown       if (p >= cStart && p < cEnd) {
16999566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
170008401ef6SPierre Jolivet         PetscCheck(closureSize == P4EST_INSUL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Non-standard closure");
17010a96aa3bSJed Brown         for (c = 0; c < P8EST_EDGES; c++) {
17020a96aa3bSJed Brown           PetscInt       cellEdge = closure[2 * (c + edgeOff)];
17030a96aa3bSJed Brown           PetscInt       cellOrnt = closure[2 * (c + edgeOff) + 1];
17040a96aa3bSJed Brown           DMPolytopeType ct;
17050a96aa3bSJed Brown 
17069566063dSJacob Faibussowitsch           PetscCall(DMPlexGetCellType(dm, cellEdge, &ct));
17070a96aa3bSJed Brown           cellOrnt = DMPolytopeConvertNewOrientation_Internal(ct, cellOrnt);
17080a96aa3bSJed Brown           if (cellEdge == e) {
17090a96aa3bSJed Brown             PetscInt p4estEdge = PetscEdgeToP4estEdge[c];
17100a96aa3bSJed Brown             PetscInt totalOrient;
17110a96aa3bSJed Brown 
17120a96aa3bSJed Brown             /* compose p4est-closure to petsc-closure permutation and petsc-closure to edge orientation */
17130a96aa3bSJed Brown             totalOrient = DihedralCompose(2, cellOrnt, DMPolytopeConvertNewOrientation_Internal(DM_POLYTOPE_SEGMENT, P4estEdgeToPetscOrnt[p4estEdge]));
17140a96aa3bSJed Brown             /* p4est orientations are positive: -2 => 1, -1 => 0 */
17150a96aa3bSJed Brown             totalOrient             = (totalOrient < 0) ? -(totalOrient + 1) : totalOrient;
17160a96aa3bSJed Brown             conn->edge_to_tree[off] = (p4est_locidx_t)(p - cStart);
1717d5b43468SJose E. Roman             /* encode cell-edge and orientation in edge_to_edge per p8est_connectivity standard (see
17180a96aa3bSJed Brown              * p8est_connectivity.h) */
17190a96aa3bSJed Brown             conn->edge_to_edge[off++]                                  = (int8_t)p4estEdge + P8EST_EDGES * totalOrient;
17200a96aa3bSJed Brown             conn->tree_to_edge[P8EST_EDGES * (p - cStart) + p4estEdge] = e - eStart;
17210a96aa3bSJed Brown           }
17220a96aa3bSJed Brown         }
17239566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
17240a96aa3bSJed Brown       }
17250a96aa3bSJed Brown     }
17269566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, e, PETSC_FALSE, &starSize, &star));
17270a96aa3bSJed Brown   }
17289566063dSJacob Faibussowitsch   PetscCall(PetscSectionDestroy(&ett));
17290a96aa3bSJed Brown   #endif
17300a96aa3bSJed Brown 
17310a96aa3bSJed Brown   /* 4: visit every vertex */
17320a96aa3bSJed Brown   conn->ctt_offset[0] = 0;
17330a96aa3bSJed Brown   for (v = vStart; v < vEnd; v++) {
17340a96aa3bSJed Brown     PetscInt off, s;
17350a96aa3bSJed Brown 
17369566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(ctt, v, &off));
17370a96aa3bSJed Brown     conn->ctt_offset[v - vStart] = (p4est_topidx_t)off;
17389566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star));
17390a96aa3bSJed Brown     for (s = 0; s < starSize; s++) {
17400a96aa3bSJed Brown       PetscInt p = star[2 * s];
17410a96aa3bSJed Brown 
17420a96aa3bSJed Brown       if (p >= cStart && p < cEnd) {
17439566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
174408401ef6SPierre Jolivet         PetscCheck(closureSize == P4EST_INSUL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Non-standard closure");
17450a96aa3bSJed Brown         for (c = 0; c < P4EST_CHILDREN; c++) {
17460a96aa3bSJed Brown           PetscInt cellVert = closure[2 * (c + vertOff)];
17470a96aa3bSJed Brown 
17480a96aa3bSJed Brown           if (cellVert == v) {
17490a96aa3bSJed Brown             PetscInt p4estVert = PetscVertToP4estVert[c];
17500a96aa3bSJed Brown 
17510a96aa3bSJed Brown             conn->corner_to_tree[off]                                       = (p4est_locidx_t)(p - cStart);
17520a96aa3bSJed Brown             conn->corner_to_corner[off++]                                   = (int8_t)p4estVert;
17530a96aa3bSJed Brown             conn->tree_to_corner[P4EST_CHILDREN * (p - cStart) + p4estVert] = v - vStart;
17540a96aa3bSJed Brown           }
17550a96aa3bSJed Brown         }
17569566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure));
17570a96aa3bSJed Brown       }
17580a96aa3bSJed Brown     }
17599566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star));
17600a96aa3bSJed Brown   }
17619566063dSJacob Faibussowitsch   PetscCall(PetscSectionDestroy(&ctt));
17620a96aa3bSJed Brown 
17630a96aa3bSJed Brown   /* 5: Compute the coordinates */
17640a96aa3bSJed Brown   {
17650a96aa3bSJed Brown     PetscInt coordDim;
17660a96aa3bSJed Brown 
17679566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDim(dm, &coordDim));
17686858538eSMatthew G. Knepley     PetscCall(DMGetCoordinatesLocalSetUp(dm));
17690a96aa3bSJed Brown     for (c = cStart; c < cEnd; c++) {
17700a96aa3bSJed Brown       PetscInt           dof;
17716858538eSMatthew G. Knepley       PetscBool          isDG;
17720a96aa3bSJed Brown       PetscScalar       *cellCoords = NULL;
17736858538eSMatthew G. Knepley       const PetscScalar *array;
17740a96aa3bSJed Brown 
17756858538eSMatthew G. Knepley       PetscCall(DMPlexGetCellCoordinates(dm, c, &isDG, &dof, &array, &cellCoords));
17766858538eSMatthew G. Knepley       PetscCheck(dof == P4EST_CHILDREN * coordDim, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Need coordinates at the corners: (dof) %" PetscInt_FMT " != %d * %" PetscInt_FMT " (sdim)", dof, P4EST_CHILDREN, coordDim);
17770a96aa3bSJed Brown       for (v = 0; v < P4EST_CHILDREN; v++) {
17780a96aa3bSJed Brown         PetscInt i, lim = PetscMin(3, coordDim);
17790a96aa3bSJed Brown         PetscInt p4estVert = PetscVertToP4estVert[v];
17800a96aa3bSJed Brown 
17810a96aa3bSJed Brown         conn->tree_to_vertex[P4EST_CHILDREN * (c - cStart) + v] = P4EST_CHILDREN * (c - cStart) + v;
17820a96aa3bSJed Brown         /* p4est vertices are always embedded in R^3 */
17830a96aa3bSJed Brown         for (i = 0; i < 3; i++) conn->vertices[3 * (P4EST_CHILDREN * (c - cStart) + p4estVert) + i] = 0.;
17840a96aa3bSJed Brown         for (i = 0; i < lim; i++) conn->vertices[3 * (P4EST_CHILDREN * (c - cStart) + p4estVert) + i] = PetscRealPart(cellCoords[v * coordDim + i]);
17850a96aa3bSJed Brown       }
17866858538eSMatthew G. Knepley       PetscCall(DMPlexRestoreCellCoordinates(dm, c, &isDG, &dof, &array, &cellCoords));
17870a96aa3bSJed Brown     }
17880a96aa3bSJed Brown   }
17890a96aa3bSJed Brown 
17900a96aa3bSJed Brown   #if defined(P4EST_ENABLE_DEBUG)
179108401ef6SPierre Jolivet   PetscCheck(p4est_connectivity_is_valid(conn), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Plex to p4est conversion failed");
17920a96aa3bSJed Brown   #endif
17930a96aa3bSJed Brown 
17940a96aa3bSJed Brown   *connOut = conn;
17950a96aa3bSJed Brown 
17960a96aa3bSJed Brown   *tree_face_to_uniq = ttf;
17970a96aa3bSJed Brown 
17983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
17990a96aa3bSJed Brown }
18000a96aa3bSJed Brown 
1801d71ae5a4SJacob Faibussowitsch static PetscErrorCode locidx_to_PetscInt(sc_array_t *array)
1802d71ae5a4SJacob Faibussowitsch {
18030a96aa3bSJed Brown   sc_array_t *newarray;
18040a96aa3bSJed Brown   size_t      zz, count = array->elem_count;
18050a96aa3bSJed Brown 
18060a96aa3bSJed Brown   PetscFunctionBegin;
180708401ef6SPierre Jolivet   PetscCheck(array->elem_size == sizeof(p4est_locidx_t), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Wrong locidx size");
18080a96aa3bSJed Brown 
18093ba16761SJacob Faibussowitsch   if (sizeof(p4est_locidx_t) == sizeof(PetscInt)) PetscFunctionReturn(PETSC_SUCCESS);
18100a96aa3bSJed Brown 
18110a96aa3bSJed Brown   newarray = sc_array_new_size(sizeof(PetscInt), array->elem_count);
18120a96aa3bSJed Brown   for (zz = 0; zz < count; zz++) {
18130a96aa3bSJed Brown     p4est_locidx_t il = *((p4est_locidx_t *)sc_array_index(array, zz));
18140a96aa3bSJed Brown     PetscInt      *ip = (PetscInt *)sc_array_index(newarray, zz);
18150a96aa3bSJed Brown 
18160a96aa3bSJed Brown     *ip = (PetscInt)il;
18170a96aa3bSJed Brown   }
18180a96aa3bSJed Brown 
18190a96aa3bSJed Brown   sc_array_reset(array);
18200a96aa3bSJed Brown   sc_array_init_size(array, sizeof(PetscInt), count);
18210a96aa3bSJed Brown   sc_array_copy(array, newarray);
18220a96aa3bSJed Brown   sc_array_destroy(newarray);
18233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
18240a96aa3bSJed Brown }
18250a96aa3bSJed Brown 
1826d71ae5a4SJacob Faibussowitsch static PetscErrorCode coords_double_to_PetscScalar(sc_array_t *array, PetscInt dim)
1827d71ae5a4SJacob Faibussowitsch {
18280a96aa3bSJed Brown   sc_array_t *newarray;
18290a96aa3bSJed Brown   size_t      zz, count = array->elem_count;
18300a96aa3bSJed Brown 
18310a96aa3bSJed Brown   PetscFunctionBegin;
18321dca8a05SBarry Smith   PetscCheck(array->elem_size == 3 * sizeof(double), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Wrong coordinate size");
18330a96aa3bSJed Brown   #if !defined(PETSC_USE_COMPLEX)
18343ba16761SJacob Faibussowitsch   if (sizeof(double) == sizeof(PetscScalar) && dim == 3) PetscFunctionReturn(PETSC_SUCCESS);
18350a96aa3bSJed Brown   #endif
18360a96aa3bSJed Brown 
18370a96aa3bSJed Brown   newarray = sc_array_new_size(dim * sizeof(PetscScalar), array->elem_count);
18380a96aa3bSJed Brown   for (zz = 0; zz < count; zz++) {
18390a96aa3bSJed Brown     int          i;
18400a96aa3bSJed Brown     double      *id = (double *)sc_array_index(array, zz);
18410a96aa3bSJed Brown     PetscScalar *ip = (PetscScalar *)sc_array_index(newarray, zz);
18420a96aa3bSJed Brown 
18430a96aa3bSJed Brown     for (i = 0; i < dim; i++) ip[i] = 0.;
18440a96aa3bSJed Brown     for (i = 0; i < PetscMin(dim, 3); i++) ip[i] = (PetscScalar)id[i];
18450a96aa3bSJed Brown   }
18460a96aa3bSJed Brown 
18470a96aa3bSJed Brown   sc_array_reset(array);
18480a96aa3bSJed Brown   sc_array_init_size(array, dim * sizeof(PetscScalar), count);
18490a96aa3bSJed Brown   sc_array_copy(array, newarray);
18500a96aa3bSJed Brown   sc_array_destroy(newarray);
18513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
18520a96aa3bSJed Brown }
18530a96aa3bSJed Brown 
1854d71ae5a4SJacob Faibussowitsch static PetscErrorCode locidx_pair_to_PetscSFNode(sc_array_t *array)
1855d71ae5a4SJacob Faibussowitsch {
18560a96aa3bSJed Brown   sc_array_t *newarray;
18570a96aa3bSJed Brown   size_t      zz, count = array->elem_count;
18580a96aa3bSJed Brown 
18590a96aa3bSJed Brown   PetscFunctionBegin;
18601dca8a05SBarry Smith   PetscCheck(array->elem_size == 2 * sizeof(p4est_locidx_t), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Wrong locidx size");
18610a96aa3bSJed Brown 
18620a96aa3bSJed Brown   newarray = sc_array_new_size(sizeof(PetscSFNode), array->elem_count);
18630a96aa3bSJed Brown   for (zz = 0; zz < count; zz++) {
18640a96aa3bSJed Brown     p4est_locidx_t *il = (p4est_locidx_t *)sc_array_index(array, zz);
18650a96aa3bSJed Brown     PetscSFNode    *ip = (PetscSFNode *)sc_array_index(newarray, zz);
18660a96aa3bSJed Brown 
18670a96aa3bSJed Brown     ip->rank  = (PetscInt)il[0];
18680a96aa3bSJed Brown     ip->index = (PetscInt)il[1];
18690a96aa3bSJed Brown   }
18700a96aa3bSJed Brown 
18710a96aa3bSJed Brown   sc_array_reset(array);
18720a96aa3bSJed Brown   sc_array_init_size(array, sizeof(PetscSFNode), count);
18730a96aa3bSJed Brown   sc_array_copy(array, newarray);
18740a96aa3bSJed Brown   sc_array_destroy(newarray);
18753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
18760a96aa3bSJed Brown }
18770a96aa3bSJed Brown 
1878d71ae5a4SJacob Faibussowitsch static PetscErrorCode P4estToPlex_Local(p4est_t *p4est, DM *plex)
1879d71ae5a4SJacob Faibussowitsch {
18800a96aa3bSJed Brown   PetscFunctionBegin;
18810a96aa3bSJed Brown   {
18820a96aa3bSJed Brown     sc_array_t    *points_per_dim    = sc_array_new(sizeof(p4est_locidx_t));
18830a96aa3bSJed Brown     sc_array_t    *cone_sizes        = sc_array_new(sizeof(p4est_locidx_t));
18840a96aa3bSJed Brown     sc_array_t    *cones             = sc_array_new(sizeof(p4est_locidx_t));
18850a96aa3bSJed Brown     sc_array_t    *cone_orientations = sc_array_new(sizeof(p4est_locidx_t));
18860a96aa3bSJed Brown     sc_array_t    *coords            = sc_array_new(3 * sizeof(double));
18870a96aa3bSJed Brown     sc_array_t    *children          = sc_array_new(sizeof(p4est_locidx_t));
18880a96aa3bSJed Brown     sc_array_t    *parents           = sc_array_new(sizeof(p4est_locidx_t));
18890a96aa3bSJed Brown     sc_array_t    *childids          = sc_array_new(sizeof(p4est_locidx_t));
18900a96aa3bSJed Brown     sc_array_t    *leaves            = sc_array_new(sizeof(p4est_locidx_t));
18910a96aa3bSJed Brown     sc_array_t    *remotes           = sc_array_new(2 * sizeof(p4est_locidx_t));
18920a96aa3bSJed Brown     p4est_locidx_t first_local_quad;
18930a96aa3bSJed Brown 
1894792fecdfSBarry Smith     PetscCallP4est(p4est_get_plex_data, (p4est, P4EST_CONNECT_FULL, 0, &first_local_quad, points_per_dim, cone_sizes, cones, cone_orientations, coords, children, parents, childids, leaves, remotes));
18950a96aa3bSJed Brown 
18969566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(points_per_dim));
18979566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cone_sizes));
18989566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cones));
18999566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cone_orientations));
19009566063dSJacob Faibussowitsch     PetscCall(coords_double_to_PetscScalar(coords, P4EST_DIM));
19010a96aa3bSJed Brown 
19029566063dSJacob Faibussowitsch     PetscCall(DMPlexCreate(PETSC_COMM_SELF, plex));
19039566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(*plex, P4EST_DIM));
19049566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(*plex, P4EST_DIM, (PetscInt *)points_per_dim->array, (PetscInt *)cone_sizes->array, (PetscInt *)cones->array, (PetscInt *)cone_orientations->array, (PetscScalar *)coords->array));
19059566063dSJacob Faibussowitsch     PetscCall(DMPlexConvertOldOrientations_Internal(*plex));
19060a96aa3bSJed Brown     sc_array_destroy(points_per_dim);
19070a96aa3bSJed Brown     sc_array_destroy(cone_sizes);
19080a96aa3bSJed Brown     sc_array_destroy(cones);
19090a96aa3bSJed Brown     sc_array_destroy(cone_orientations);
19100a96aa3bSJed Brown     sc_array_destroy(coords);
19110a96aa3bSJed Brown     sc_array_destroy(children);
19120a96aa3bSJed Brown     sc_array_destroy(parents);
19130a96aa3bSJed Brown     sc_array_destroy(childids);
19140a96aa3bSJed Brown     sc_array_destroy(leaves);
19150a96aa3bSJed Brown     sc_array_destroy(remotes);
19160a96aa3bSJed Brown   }
19173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
19180a96aa3bSJed Brown }
19190a96aa3bSJed Brown 
19200a96aa3bSJed Brown   #define DMReferenceTreeGetChildSymmetry_pforest _append_pforest(DMReferenceTreeGetChildSymmetry)
1921d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMReferenceTreeGetChildSymmetry_pforest(DM dm, PetscInt parent, PetscInt parentOrientA, PetscInt childOrientA, PetscInt childA, PetscInt parentOrientB, PetscInt *childOrientB, PetscInt *childB)
1922d71ae5a4SJacob Faibussowitsch {
19230a96aa3bSJed Brown   PetscInt coneSize, dStart, dEnd, vStart, vEnd, dim, ABswap, oAvert, oBvert, ABswapVert;
19240a96aa3bSJed Brown 
19250a96aa3bSJed Brown   PetscFunctionBegin;
19260a96aa3bSJed Brown   if (parentOrientA == parentOrientB) {
19270a96aa3bSJed Brown     if (childOrientB) *childOrientB = childOrientA;
19280a96aa3bSJed Brown     if (childB) *childB = childA;
19293ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
19300a96aa3bSJed Brown   }
19319566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
19326aad120cSJose E. Roman   if (childA >= vStart && childA < vEnd) { /* vertices (always in the middle) are invariant under rotation */
19330a96aa3bSJed Brown     if (childOrientB) *childOrientB = 0;
19340a96aa3bSJed Brown     if (childB) *childB = childA;
19353ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
19360a96aa3bSJed Brown   }
19370a96aa3bSJed Brown   for (dim = 0; dim < 3; dim++) {
19389566063dSJacob Faibussowitsch     PetscCall(DMPlexGetDepthStratum(dm, dim, &dStart, &dEnd));
19390a96aa3bSJed Brown     if (parent >= dStart && parent <= dEnd) break;
19400a96aa3bSJed Brown   }
194163a3b9bcSJacob Faibussowitsch   PetscCheck(dim <= 2, PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot perform child symmetry for %" PetscInt_FMT "-cells", dim);
194228b400f6SJacob Faibussowitsch   PetscCheck(dim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "A vertex has no children");
19430a96aa3bSJed Brown   if (childA < dStart || childA >= dEnd) { /* a 1-cell in a 2-cell */
19440a96aa3bSJed Brown     /* this is a lower-dimensional child: bootstrap */
19450a96aa3bSJed Brown     PetscInt        size, i, sA = -1, sB, sOrientB, sConeSize;
19460a96aa3bSJed Brown     const PetscInt *supp, *coneA, *coneB, *oA, *oB;
19470a96aa3bSJed Brown 
19489566063dSJacob Faibussowitsch     PetscCall(DMPlexGetSupportSize(dm, childA, &size));
19499566063dSJacob Faibussowitsch     PetscCall(DMPlexGetSupport(dm, childA, &supp));
19500a96aa3bSJed Brown 
19510a96aa3bSJed Brown     /* find a point sA in supp(childA) that has the same parent */
19520a96aa3bSJed Brown     for (i = 0; i < size; i++) {
19530a96aa3bSJed Brown       PetscInt sParent;
19540a96aa3bSJed Brown 
19550a96aa3bSJed Brown       sA = supp[i];
19560a96aa3bSJed Brown       if (sA == parent) continue;
19579566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTreeParent(dm, sA, &sParent, NULL));
19580a96aa3bSJed Brown       if (sParent == parent) break;
19590a96aa3bSJed Brown     }
196008401ef6SPierre Jolivet     PetscCheck(i != size, PETSC_COMM_SELF, PETSC_ERR_PLIB, "could not find support in children");
19610a96aa3bSJed Brown     /* find out which point sB is in an equivalent position to sA under
19620a96aa3bSJed Brown      * parentOrientB */
19639566063dSJacob Faibussowitsch     PetscCall(DMReferenceTreeGetChildSymmetry_pforest(dm, parent, parentOrientA, 0, sA, parentOrientB, &sOrientB, &sB));
19649566063dSJacob Faibussowitsch     PetscCall(DMPlexGetConeSize(dm, sA, &sConeSize));
19659566063dSJacob Faibussowitsch     PetscCall(DMPlexGetCone(dm, sA, &coneA));
19669566063dSJacob Faibussowitsch     PetscCall(DMPlexGetCone(dm, sB, &coneB));
19679566063dSJacob Faibussowitsch     PetscCall(DMPlexGetConeOrientation(dm, sA, &oA));
19689566063dSJacob Faibussowitsch     PetscCall(DMPlexGetConeOrientation(dm, sB, &oB));
19690a96aa3bSJed Brown     /* step through the cone of sA in natural order */
19700a96aa3bSJed Brown     for (i = 0; i < sConeSize; i++) {
19710a96aa3bSJed Brown       if (coneA[i] == childA) {
19720a96aa3bSJed Brown         /* if childA is at position i in coneA,
19730a96aa3bSJed Brown          * then we want the point that is at sOrientB*i in coneB */
19740a96aa3bSJed Brown         PetscInt j = (sOrientB >= 0) ? ((sOrientB + i) % sConeSize) : ((sConeSize - (sOrientB + 1) - i) % sConeSize);
19750a96aa3bSJed Brown         if (childB) *childB = coneB[j];
19760a96aa3bSJed Brown         if (childOrientB) {
19770a96aa3bSJed Brown           DMPolytopeType ct;
19780a96aa3bSJed Brown           PetscInt       oBtrue;
19790a96aa3bSJed Brown 
19809566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm, childA, &coneSize));
19810a96aa3bSJed Brown           /* compose sOrientB and oB[j] */
19821dca8a05SBarry Smith           PetscCheck(coneSize == 0 || coneSize == 2, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected a vertex or an edge");
19830a96aa3bSJed Brown           ct = coneSize ? DM_POLYTOPE_SEGMENT : DM_POLYTOPE_POINT;
19840a96aa3bSJed Brown           /* we may have to flip an edge */
19850a96aa3bSJed Brown           oBtrue        = (sOrientB >= 0) ? oB[j] : DMPolytopeTypeComposeOrientationInv(ct, -1, oB[j]);
19860a96aa3bSJed Brown           oBtrue        = DMPolytopeConvertNewOrientation_Internal(ct, oBtrue);
19870a96aa3bSJed Brown           ABswap        = DihedralSwap(coneSize, DMPolytopeConvertNewOrientation_Internal(ct, oA[i]), oBtrue);
19880a96aa3bSJed Brown           *childOrientB = DihedralCompose(coneSize, childOrientA, ABswap);
19890a96aa3bSJed Brown         }
19900a96aa3bSJed Brown         break;
19910a96aa3bSJed Brown       }
19920a96aa3bSJed Brown     }
199308401ef6SPierre Jolivet     PetscCheck(i != sConeSize, PETSC_COMM_SELF, PETSC_ERR_PLIB, "support cone mismatch");
19943ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
19950a96aa3bSJed Brown   }
19960a96aa3bSJed Brown   /* get the cone size and symmetry swap */
19979566063dSJacob Faibussowitsch   PetscCall(DMPlexGetConeSize(dm, parent, &coneSize));
19980a96aa3bSJed Brown   ABswap = DihedralSwap(coneSize, parentOrientA, parentOrientB);
19990a96aa3bSJed Brown   if (dim == 2) {
20000a96aa3bSJed Brown     /* orientations refer to cones: we want them to refer to vertices:
20010a96aa3bSJed Brown      * if it's a rotation, they are the same, but if the order is reversed, a
20020a96aa3bSJed Brown      * permutation that puts side i first does *not* put vertex i first */
20030a96aa3bSJed Brown     oAvert     = (parentOrientA >= 0) ? parentOrientA : -((-parentOrientA % coneSize) + 1);
20040a96aa3bSJed Brown     oBvert     = (parentOrientB >= 0) ? parentOrientB : -((-parentOrientB % coneSize) + 1);
20050a96aa3bSJed Brown     ABswapVert = DihedralSwap(coneSize, oAvert, oBvert);
20060a96aa3bSJed Brown   } else {
20070a96aa3bSJed Brown     oAvert     = parentOrientA;
20080a96aa3bSJed Brown     oBvert     = parentOrientB;
20090a96aa3bSJed Brown     ABswapVert = ABswap;
20100a96aa3bSJed Brown   }
20110a96aa3bSJed Brown   if (childB) {
20120a96aa3bSJed Brown     /* assume that each child corresponds to a vertex, in the same order */
20130a96aa3bSJed Brown     PetscInt        p, posA = -1, numChildren, i;
20140a96aa3bSJed Brown     const PetscInt *children;
20150a96aa3bSJed Brown 
20160a96aa3bSJed Brown     /* count which position the child is in */
20179566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTreeChildren(dm, parent, &numChildren, &children));
20180a96aa3bSJed Brown     for (i = 0; i < numChildren; i++) {
20190a96aa3bSJed Brown       p = children[i];
20200a96aa3bSJed Brown       if (p == childA) {
20210a96aa3bSJed Brown         if (dim == 1) {
20220a96aa3bSJed Brown           posA = i;
20230a96aa3bSJed Brown         } else { /* 2D Morton to rotation */
20240a96aa3bSJed Brown           posA = (i & 2) ? (i ^ 1) : i;
20250a96aa3bSJed Brown         }
20260a96aa3bSJed Brown         break;
20270a96aa3bSJed Brown       }
20280a96aa3bSJed Brown     }
20290a96aa3bSJed Brown     if (posA >= coneSize) {
20300a96aa3bSJed Brown       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not find childA in children of parent");
20310a96aa3bSJed Brown     } else {
20320a96aa3bSJed Brown       /* figure out position B by applying ABswapVert */
20330a96aa3bSJed Brown       PetscInt posB, childIdB;
20340a96aa3bSJed Brown 
20350a96aa3bSJed Brown       posB = (ABswapVert >= 0) ? ((ABswapVert + posA) % coneSize) : ((coneSize - (ABswapVert + 1) - posA) % coneSize);
20360a96aa3bSJed Brown       if (dim == 1) {
20370a96aa3bSJed Brown         childIdB = posB;
20380a96aa3bSJed Brown       } else { /* 2D rotation to Morton */
20390a96aa3bSJed Brown         childIdB = (posB & 2) ? (posB ^ 1) : posB;
20400a96aa3bSJed Brown       }
20410a96aa3bSJed Brown       if (childB) *childB = children[childIdB];
20420a96aa3bSJed Brown     }
20430a96aa3bSJed Brown   }
20440a96aa3bSJed Brown   if (childOrientB) *childOrientB = DihedralCompose(coneSize, childOrientA, ABswap);
20453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
20460a96aa3bSJed Brown }
20470a96aa3bSJed Brown 
20480a96aa3bSJed Brown   #define DMCreateReferenceTree_pforest _append_pforest(DMCreateReferenceTree)
2049d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateReferenceTree_pforest(MPI_Comm comm, DM *dm)
2050d71ae5a4SJacob Faibussowitsch {
20510a96aa3bSJed Brown   p4est_connectivity_t *refcube;
20520a96aa3bSJed Brown   p4est_t              *root, *refined;
20530a96aa3bSJed Brown   DM                    dmRoot, dmRefined;
20540a96aa3bSJed Brown   DM_Plex              *mesh;
20550a96aa3bSJed Brown   PetscMPIInt           rank;
205666c0a4b5SToby Isaac   #if defined(PETSC_HAVE_MPIUNI)
205766c0a4b5SToby Isaac   sc_MPI_Comm comm_self = sc_MPI_COMM_SELF;
205866c0a4b5SToby Isaac   #else
205966c0a4b5SToby Isaac   MPI_Comm comm_self = PETSC_COMM_SELF;
206066c0a4b5SToby Isaac   #endif
20610a96aa3bSJed Brown 
20620a96aa3bSJed Brown   PetscFunctionBegin;
2063792fecdfSBarry Smith   PetscCallP4estReturn(refcube, p4est_connectivity_new_byname, ("unit"));
20640a96aa3bSJed Brown   { /* [-1,1]^d geometry */
20650a96aa3bSJed Brown     PetscInt i, j;
20660a96aa3bSJed Brown 
20670a96aa3bSJed Brown     for (i = 0; i < P4EST_CHILDREN; i++) {
20680a96aa3bSJed Brown       for (j = 0; j < 3; j++) {
20690a96aa3bSJed Brown         refcube->vertices[3 * i + j] *= 2.;
20700a96aa3bSJed Brown         refcube->vertices[3 * i + j] -= 1.;
20710a96aa3bSJed Brown       }
20720a96aa3bSJed Brown     }
20730a96aa3bSJed Brown   }
2074792fecdfSBarry Smith   PetscCallP4estReturn(root, p4est_new, (comm_self, refcube, 0, NULL, NULL));
2075792fecdfSBarry Smith   PetscCallP4estReturn(refined, p4est_new_ext, (comm_self, refcube, 0, 1, 1, 0, NULL, NULL));
20769566063dSJacob Faibussowitsch   PetscCall(P4estToPlex_Local(root, &dmRoot));
20779566063dSJacob Faibussowitsch   PetscCall(P4estToPlex_Local(refined, &dmRefined));
20780a96aa3bSJed Brown   {
20790a96aa3bSJed Brown   #if !defined(P4_TO_P8)
20800a96aa3bSJed Brown     PetscInt nPoints   = 25;
20819371c9d4SSatish Balay     PetscInt perm[25]  = {0, 1, 2, 3, 4, 12, 8, 14, 6, 9, 15, 5, 13, 10, 7, 11, 16, 22, 20, 24, 17, 21, 18, 23, 19};
20829371c9d4SSatish Balay     PetscInt ident[25] = {0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 0, 0, 0, 0, 5, 6, 7, 8, 1, 2, 3, 4, 0};
20830a96aa3bSJed Brown   #else
20840a96aa3bSJed Brown     PetscInt nPoints    = 125;
20859371c9d4SSatish Balay     PetscInt perm[125]  = {0,  1,  2,  3,  4,  5,  6,  7,  8,  32, 16, 36, 24, 40, 12, 17,  37,  25,  41,  9,   33,  20,  26, 42,  13,  21,  27,  43,  10,  34,  18,  38,  28,  14,  19,  39,  29,  11,  35,  22,  30, 15,
20869371c9d4SSatish Balay                            23, 31, 44, 84, 76, 92, 52, 86, 68, 94, 60, 78, 70, 96, 45, 85,  77,  93,  54,  72,  62,  74,  46, 80,  53,  87,  69,  95,  64,  82,  47,  81,  55,  73,  66,  48,  88,  56,  90,  61,  79, 71,
20879371c9d4SSatish Balay                            97, 49, 89, 58, 63, 75, 50, 57, 91, 65, 83, 51, 59, 67, 98, 106, 110, 122, 114, 120, 118, 124, 99, 111, 115, 119, 100, 107, 116, 121, 101, 117, 102, 108, 112, 123, 103, 113, 104, 109, 105};
20889371c9d4SSatish Balay     PetscInt ident[125] = {0,  0,  0,  0,  0,  0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 6, 6, 0, 0, 0, 0,  0,  0,  0,  0,  0,  0,  0,  0, 7, 7, 8,  8,  9,  9,  10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16,
20899371c9d4SSatish Balay                            16, 17, 17, 18, 18, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 6, 6, 0, 0, 0, 0, 0, 0, 19, 20, 21, 22, 23, 24, 25, 26, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 1,  2,  3,  4,  5,  6,  0};
20900a96aa3bSJed Brown 
20910a96aa3bSJed Brown   #endif
20920a96aa3bSJed Brown     IS permIS;
20930a96aa3bSJed Brown     DM dmPerm;
20940a96aa3bSJed Brown 
20959566063dSJacob Faibussowitsch     PetscCall(ISCreateGeneral(PETSC_COMM_SELF, nPoints, perm, PETSC_USE_POINTER, &permIS));
20969566063dSJacob Faibussowitsch     PetscCall(DMPlexPermute(dmRefined, permIS, &dmPerm));
20970a96aa3bSJed Brown     if (dmPerm) {
20989566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&dmRefined));
20990a96aa3bSJed Brown       dmRefined = dmPerm;
21000a96aa3bSJed Brown     }
21019566063dSJacob Faibussowitsch     PetscCall(ISDestroy(&permIS));
21020a96aa3bSJed Brown     {
21030a96aa3bSJed Brown       PetscInt p;
21049566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(dmRoot, "identity"));
21059566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(dmRefined, "identity"));
210648a46eb9SPierre Jolivet       for (p = 0; p < P4EST_INSUL; p++) PetscCall(DMSetLabelValue(dmRoot, "identity", p, p));
210748a46eb9SPierre Jolivet       for (p = 0; p < nPoints; p++) PetscCall(DMSetLabelValue(dmRefined, "identity", p, ident[p]));
21080a96aa3bSJed Brown     }
21090a96aa3bSJed Brown   }
21109566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceTree_Union(dmRoot, dmRefined, "identity", dm));
21110a96aa3bSJed Brown   mesh                   = (DM_Plex *)(*dm)->data;
21120a96aa3bSJed Brown   mesh->getchildsymmetry = DMReferenceTreeGetChildSymmetry_pforest;
21139566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
21140a96aa3bSJed Brown   if (rank == 0) {
21159566063dSJacob Faibussowitsch     PetscCall(DMViewFromOptions(dmRoot, NULL, "-dm_p4est_ref_root_view"));
21169566063dSJacob Faibussowitsch     PetscCall(DMViewFromOptions(dmRefined, NULL, "-dm_p4est_ref_refined_view"));
21179566063dSJacob Faibussowitsch     PetscCall(DMViewFromOptions(dmRefined, NULL, "-dm_p4est_ref_tree_view"));
21180a96aa3bSJed Brown   }
21199566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&dmRefined));
21209566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&dmRoot));
2121792fecdfSBarry Smith   PetscCallP4est(p4est_destroy, (refined));
2122792fecdfSBarry Smith   PetscCallP4est(p4est_destroy, (root));
2123792fecdfSBarry Smith   PetscCallP4est(p4est_connectivity_destroy, (refcube));
21243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
21250a96aa3bSJed Brown }
21260a96aa3bSJed Brown 
2127d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMShareDiscretization(DM dmA, DM dmB)
2128d71ae5a4SJacob Faibussowitsch {
21290a96aa3bSJed Brown   void     *ctx;
21300a96aa3bSJed Brown   PetscInt  num;
21310a96aa3bSJed Brown   PetscReal val;
21320a96aa3bSJed Brown 
21330a96aa3bSJed Brown   PetscFunctionBegin;
21349566063dSJacob Faibussowitsch   PetscCall(DMGetApplicationContext(dmA, &ctx));
21359566063dSJacob Faibussowitsch   PetscCall(DMSetApplicationContext(dmB, ctx));
21369566063dSJacob Faibussowitsch   PetscCall(DMCopyDisc(dmA, dmB));
21379566063dSJacob Faibussowitsch   PetscCall(DMGetOutputSequenceNumber(dmA, &num, &val));
21389566063dSJacob Faibussowitsch   PetscCall(DMSetOutputSequenceNumber(dmB, num, val));
21390a96aa3bSJed Brown   if (dmB->localSection != dmA->localSection || dmB->globalSection != dmA->globalSection) {
21409566063dSJacob Faibussowitsch     PetscCall(DMClearLocalVectors(dmB));
21419566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)dmA->localSection));
21429566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&(dmB->localSection)));
21430a96aa3bSJed Brown     dmB->localSection = dmA->localSection;
21449566063dSJacob Faibussowitsch     PetscCall(DMClearGlobalVectors(dmB));
21459566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)dmA->globalSection));
21469566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&(dmB->globalSection)));
21470a96aa3bSJed Brown     dmB->globalSection = dmA->globalSection;
21489566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)dmA->defaultConstraint.section));
21499566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&(dmB->defaultConstraint.section)));
21503b8ba7d1SJed Brown     dmB->defaultConstraint.section = dmA->defaultConstraint.section;
21519566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)dmA->defaultConstraint.mat));
21529566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&(dmB->defaultConstraint.mat)));
21533b8ba7d1SJed Brown     dmB->defaultConstraint.mat = dmA->defaultConstraint.mat;
21549566063dSJacob Faibussowitsch     if (dmA->map) PetscCall(PetscLayoutReference(dmA->map, &dmB->map));
21550a96aa3bSJed Brown   }
21560a96aa3bSJed Brown   if (dmB->sectionSF != dmA->sectionSF) {
21579566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)dmA->sectionSF));
21589566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&dmB->sectionSF));
21590a96aa3bSJed Brown     dmB->sectionSF = dmA->sectionSF;
21600a96aa3bSJed Brown   }
21613ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
21620a96aa3bSJed Brown }
21630a96aa3bSJed Brown 
21640a96aa3bSJed Brown /* Get an SF that broadcasts a coarse-cell covering of the local fine cells */
2165d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetCellCoveringSF(MPI_Comm comm, p4est_t *p4estC, p4est_t *p4estF, PetscInt cStart, PetscInt cEnd, PetscSF *coveringSF)
2166d71ae5a4SJacob Faibussowitsch {
21670a96aa3bSJed Brown   PetscInt     startF, endF, startC, endC, p, nLeaves;
21680a96aa3bSJed Brown   PetscSFNode *leaves;
21690a96aa3bSJed Brown   PetscSF      sf;
21700a96aa3bSJed Brown   PetscInt    *recv, *send;
21710a96aa3bSJed Brown   PetscMPIInt  tag;
21720a96aa3bSJed Brown   MPI_Request *recvReqs, *sendReqs;
21730a96aa3bSJed Brown   PetscSection section;
21740a96aa3bSJed Brown 
21750a96aa3bSJed Brown   PetscFunctionBegin;
21769566063dSJacob Faibussowitsch   PetscCall(DMPforestComputeOverlappingRanks(p4estC->mpisize, p4estC->mpirank, p4estF, p4estC, &startC, &endC));
21779566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(2 * (endC - startC), &recv, endC - startC, &recvReqs));
21789566063dSJacob Faibussowitsch   PetscCall(PetscCommGetNewTag(comm, &tag));
21790a96aa3bSJed Brown   for (p = startC; p < endC; p++) {
21800a96aa3bSJed Brown     recvReqs[p - startC] = MPI_REQUEST_NULL;                                        /* just in case we don't initiate a receive */
21810a96aa3bSJed Brown     if (p4estC->global_first_quadrant[p] == p4estC->global_first_quadrant[p + 1]) { /* empty coarse partition */
21820a96aa3bSJed Brown       recv[2 * (p - startC)]     = 0;
21830a96aa3bSJed Brown       recv[2 * (p - startC) + 1] = 0;
21840a96aa3bSJed Brown       continue;
21850a96aa3bSJed Brown     }
21860a96aa3bSJed Brown 
21879566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Irecv(&recv[2 * (p - startC)], 2, MPIU_INT, p, tag, comm, &recvReqs[p - startC]));
21880a96aa3bSJed Brown   }
21899566063dSJacob Faibussowitsch   PetscCall(DMPforestComputeOverlappingRanks(p4estC->mpisize, p4estC->mpirank, p4estC, p4estF, &startF, &endF));
21909566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(2 * (endF - startF), &send, endF - startF, &sendReqs));
21910a96aa3bSJed Brown   /* count the quadrants rank will send to each of [startF,endF) */
21920a96aa3bSJed Brown   for (p = startF; p < endF; p++) {
21930a96aa3bSJed Brown     p4est_quadrant_t *myFineStart = &p4estF->global_first_position[p];
21940a96aa3bSJed Brown     p4est_quadrant_t *myFineEnd   = &p4estF->global_first_position[p + 1];
21950a96aa3bSJed Brown     PetscInt          tStart      = (PetscInt)myFineStart->p.which_tree;
21960a96aa3bSJed Brown     PetscInt          tEnd        = (PetscInt)myFineEnd->p.which_tree;
21970a96aa3bSJed Brown     PetscInt          firstCell = -1, lastCell = -1;
21980a96aa3bSJed Brown     p4est_tree_t     *treeStart = &(((p4est_tree_t *)p4estC->trees->array)[tStart]);
21990a96aa3bSJed Brown     p4est_tree_t     *treeEnd   = (size_t)tEnd < p4estC->trees->elem_count ? &(((p4est_tree_t *)p4estC->trees->array)[tEnd]) : NULL;
22000a96aa3bSJed Brown     ssize_t           overlapIndex;
22010a96aa3bSJed Brown 
22020a96aa3bSJed Brown     sendReqs[p - startF] = MPI_REQUEST_NULL; /* just in case we don't initiate a send */
22030a96aa3bSJed Brown     if (p4estF->global_first_quadrant[p] == p4estF->global_first_quadrant[p + 1]) continue;
22040a96aa3bSJed Brown 
22050a96aa3bSJed Brown     /* locate myFineStart in (or before) a cell */
22060a96aa3bSJed Brown     if (treeStart->quadrants.elem_count) {
2207792fecdfSBarry Smith       PetscCallP4estReturn(overlapIndex, sc_array_bsearch, (&(treeStart->quadrants), myFineStart, p4est_quadrant_disjoint));
22080a96aa3bSJed Brown       if (overlapIndex < 0) {
22090a96aa3bSJed Brown         firstCell = 0;
22100a96aa3bSJed Brown       } else {
22110a96aa3bSJed Brown         firstCell = treeStart->quadrants_offset + overlapIndex;
22120a96aa3bSJed Brown       }
22130a96aa3bSJed Brown     } else {
22140a96aa3bSJed Brown       firstCell = 0;
22150a96aa3bSJed Brown     }
22160a96aa3bSJed Brown     if (treeEnd && treeEnd->quadrants.elem_count) {
2217792fecdfSBarry Smith       PetscCallP4estReturn(overlapIndex, sc_array_bsearch, (&(treeEnd->quadrants), myFineEnd, p4est_quadrant_disjoint));
22180a96aa3bSJed Brown       if (overlapIndex < 0) { /* all of this local section is overlapped */
22190a96aa3bSJed Brown         lastCell = p4estC->local_num_quadrants;
22200a96aa3bSJed Brown       } else {
22210a96aa3bSJed Brown         p4est_quadrant_t *container = &(((p4est_quadrant_t *)treeEnd->quadrants.array)[overlapIndex]);
22220a96aa3bSJed Brown         p4est_quadrant_t  first_desc;
22230a96aa3bSJed Brown         int               equal;
22240a96aa3bSJed Brown 
2225792fecdfSBarry Smith         PetscCallP4est(p4est_quadrant_first_descendant, (container, &first_desc, P4EST_QMAXLEVEL));
2226792fecdfSBarry Smith         PetscCallP4estReturn(equal, p4est_quadrant_is_equal, (myFineEnd, &first_desc));
22270a96aa3bSJed Brown         if (equal) {
22280a96aa3bSJed Brown           lastCell = treeEnd->quadrants_offset + overlapIndex;
22290a96aa3bSJed Brown         } else {
22300a96aa3bSJed Brown           lastCell = treeEnd->quadrants_offset + overlapIndex + 1;
22310a96aa3bSJed Brown         }
22320a96aa3bSJed Brown       }
22330a96aa3bSJed Brown     } else {
22340a96aa3bSJed Brown       lastCell = p4estC->local_num_quadrants;
22350a96aa3bSJed Brown     }
22360a96aa3bSJed Brown     send[2 * (p - startF)]     = firstCell;
22370a96aa3bSJed Brown     send[2 * (p - startF) + 1] = lastCell - firstCell;
22389566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Isend(&send[2 * (p - startF)], 2, MPIU_INT, p, tag, comm, &sendReqs[p - startF]));
22390a96aa3bSJed Brown   }
22409566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Waitall((PetscMPIInt)(endC - startC), recvReqs, MPI_STATUSES_IGNORE));
22419566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &section));
22429566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(section, startC, endC));
22430a96aa3bSJed Brown   for (p = startC; p < endC; p++) {
22440a96aa3bSJed Brown     PetscInt numCells = recv[2 * (p - startC) + 1];
22459566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetDof(section, p, numCells));
22460a96aa3bSJed Brown   }
22479566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(section));
22489566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(section, &nLeaves));
22499566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(nLeaves, &leaves));
22500a96aa3bSJed Brown   for (p = startC; p < endC; p++) {
22510a96aa3bSJed Brown     PetscInt firstCell = recv[2 * (p - startC)];
22520a96aa3bSJed Brown     PetscInt numCells  = recv[2 * (p - startC) + 1];
22530a96aa3bSJed Brown     PetscInt off, i;
22540a96aa3bSJed Brown 
22559566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(section, p, &off));
22560a96aa3bSJed Brown     for (i = 0; i < numCells; i++) {
22570a96aa3bSJed Brown       leaves[off + i].rank  = p;
22580a96aa3bSJed Brown       leaves[off + i].index = firstCell + i;
22590a96aa3bSJed Brown     }
22600a96aa3bSJed Brown   }
22619566063dSJacob Faibussowitsch   PetscCall(PetscSFCreate(comm, &sf));
22629566063dSJacob Faibussowitsch   PetscCall(PetscSFSetGraph(sf, cEnd - cStart, nLeaves, NULL, PETSC_OWN_POINTER, leaves, PETSC_OWN_POINTER));
22639566063dSJacob Faibussowitsch   PetscCall(PetscSectionDestroy(&section));
22649566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Waitall((PetscMPIInt)(endF - startF), sendReqs, MPI_STATUSES_IGNORE));
22659566063dSJacob Faibussowitsch   PetscCall(PetscFree2(send, sendReqs));
22669566063dSJacob Faibussowitsch   PetscCall(PetscFree2(recv, recvReqs));
22670a96aa3bSJed Brown   *coveringSF = sf;
22683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22690a96aa3bSJed Brown }
22700a96aa3bSJed Brown 
22710a96aa3bSJed Brown /* closure points for locally-owned cells */
2272d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetCellSFNodes(DM dm, PetscInt numClosureIndices, PetscInt *numClosurePoints, PetscSFNode **closurePoints, PetscBool redirect)
2273d71ae5a4SJacob Faibussowitsch {
22740a96aa3bSJed Brown   PetscInt           cStart, cEnd;
22750a96aa3bSJed Brown   PetscInt           count, c;
22760a96aa3bSJed Brown   PetscMPIInt        rank;
22770a96aa3bSJed Brown   PetscInt           closureSize = -1;
22780a96aa3bSJed Brown   PetscInt          *closure     = NULL;
22790a96aa3bSJed Brown   PetscSF            pointSF;
22800a96aa3bSJed Brown   PetscInt           nleaves, nroots;
22810a96aa3bSJed Brown   const PetscInt    *ilocal;
22820a96aa3bSJed Brown   const PetscSFNode *iremote;
22830a96aa3bSJed Brown   DM                 plex;
22840a96aa3bSJed Brown   DM_Forest         *forest;
22850a96aa3bSJed Brown   DM_Forest_pforest *pforest;
22860a96aa3bSJed Brown 
22870a96aa3bSJed Brown   PetscFunctionBegin;
22880a96aa3bSJed Brown   forest  = (DM_Forest *)dm->data;
22890a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)forest->data;
22900a96aa3bSJed Brown   cStart  = pforest->cLocalStart;
22910a96aa3bSJed Brown   cEnd    = pforest->cLocalEnd;
22929566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
22939566063dSJacob Faibussowitsch   PetscCall(DMGetPointSF(dm, &pointSF));
22949566063dSJacob Faibussowitsch   PetscCall(PetscSFGetGraph(pointSF, &nroots, &nleaves, &ilocal, &iremote));
22950a96aa3bSJed Brown   nleaves           = PetscMax(0, nleaves);
22960a96aa3bSJed Brown   nroots            = PetscMax(0, nroots);
22970a96aa3bSJed Brown   *numClosurePoints = numClosureIndices * (cEnd - cStart);
22989566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(*numClosurePoints, closurePoints));
22999566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
23000a96aa3bSJed Brown   for (c = cStart, count = 0; c < cEnd; c++) {
23010a96aa3bSJed Brown     PetscInt i;
23029566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(plex, c, PETSC_TRUE, &closureSize, &closure));
23030a96aa3bSJed Brown 
23040a96aa3bSJed Brown     for (i = 0; i < numClosureIndices; i++, count++) {
23050a96aa3bSJed Brown       PetscInt p   = closure[2 * i];
23060a96aa3bSJed Brown       PetscInt loc = -1;
23070a96aa3bSJed Brown 
23089566063dSJacob Faibussowitsch       PetscCall(PetscFindInt(p, nleaves, ilocal, &loc));
23090a96aa3bSJed Brown       if (redirect && loc >= 0) {
23100a96aa3bSJed Brown         (*closurePoints)[count].rank  = iremote[loc].rank;
23110a96aa3bSJed Brown         (*closurePoints)[count].index = iremote[loc].index;
23120a96aa3bSJed Brown       } else {
23130a96aa3bSJed Brown         (*closurePoints)[count].rank  = rank;
23140a96aa3bSJed Brown         (*closurePoints)[count].index = p;
23150a96aa3bSJed Brown       }
23160a96aa3bSJed Brown     }
23179566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(plex, c, PETSC_TRUE, &closureSize, &closure));
23180a96aa3bSJed Brown   }
23193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23200a96aa3bSJed Brown }
23210a96aa3bSJed Brown 
2322d71ae5a4SJacob Faibussowitsch static void MPIAPI DMPforestMaxSFNode(void *a, void *b, PetscMPIInt *len, MPI_Datatype *type)
2323d71ae5a4SJacob Faibussowitsch {
23240a96aa3bSJed Brown   PetscMPIInt i;
23250a96aa3bSJed Brown 
23260a96aa3bSJed Brown   for (i = 0; i < *len; i++) {
23270a96aa3bSJed Brown     PetscSFNode *A = (PetscSFNode *)a;
23280a96aa3bSJed Brown     PetscSFNode *B = (PetscSFNode *)b;
23290a96aa3bSJed Brown 
23300a96aa3bSJed Brown     if (B->rank < 0) *B = *A;
23310a96aa3bSJed Brown   }
23320a96aa3bSJed Brown }
23330a96aa3bSJed Brown 
2334d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetTransferSF_Point(DM coarse, DM fine, PetscSF *sf, PetscBool transferIdent, PetscInt *childIds[])
2335d71ae5a4SJacob Faibussowitsch {
23360a96aa3bSJed Brown   MPI_Comm           comm;
23370a96aa3bSJed Brown   PetscMPIInt        rank, size;
23380a96aa3bSJed Brown   DM_Forest_pforest *pforestC, *pforestF;
23390a96aa3bSJed Brown   p4est_t           *p4estC, *p4estF;
23400a96aa3bSJed Brown   PetscInt           numClosureIndices;
23410a96aa3bSJed Brown   PetscInt           numClosurePointsC, numClosurePointsF;
23420a96aa3bSJed Brown   PetscSFNode       *closurePointsC, *closurePointsF;
23430a96aa3bSJed Brown   p4est_quadrant_t  *coverQuads = NULL;
23440a96aa3bSJed Brown   p4est_quadrant_t **treeQuads;
23450a96aa3bSJed Brown   PetscInt          *treeQuadCounts;
23460a96aa3bSJed Brown   MPI_Datatype       nodeType;
23470a96aa3bSJed Brown   MPI_Datatype       nodeClosureType;
23480a96aa3bSJed Brown   MPI_Op             sfNodeReduce;
23490a96aa3bSJed Brown   p4est_topidx_t     fltF, lltF, t;
23500a96aa3bSJed Brown   DM                 plexC, plexF;
23510a96aa3bSJed Brown   PetscInt           pStartF, pEndF, pStartC, pEndC;
23520a96aa3bSJed Brown   PetscBool          saveInCoarse = PETSC_FALSE;
23530a96aa3bSJed Brown   PetscBool          saveInFine   = PETSC_FALSE;
23540a96aa3bSJed Brown   PetscBool          formCids     = (childIds != NULL) ? PETSC_TRUE : PETSC_FALSE;
23550a96aa3bSJed Brown   PetscInt          *cids         = NULL;
23560a96aa3bSJed Brown 
23570a96aa3bSJed Brown   PetscFunctionBegin;
23580a96aa3bSJed Brown   pforestC = (DM_Forest_pforest *)((DM_Forest *)coarse->data)->data;
23590a96aa3bSJed Brown   pforestF = (DM_Forest_pforest *)((DM_Forest *)fine->data)->data;
23600a96aa3bSJed Brown   p4estC   = pforestC->forest;
23610a96aa3bSJed Brown   p4estF   = pforestF->forest;
236208401ef6SPierre Jolivet   PetscCheck(pforestC->topo == pforestF->topo, PetscObjectComm((PetscObject)coarse), PETSC_ERR_ARG_INCOMP, "DM's must have the same base DM");
23630a96aa3bSJed Brown   comm = PetscObjectComm((PetscObject)coarse);
23649566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
23659566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
23669566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(fine, &plexF));
23679566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plexF, &pStartF, &pEndF));
23689566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(coarse, &plexC));
23699566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plexC, &pStartC, &pEndC));
23700a96aa3bSJed Brown   { /* check if the results have been cached */
23710a96aa3bSJed Brown     DM adaptCoarse, adaptFine;
23720a96aa3bSJed Brown 
23739566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityForest(coarse, &adaptCoarse));
23749566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityForest(fine, &adaptFine));
23750a96aa3bSJed Brown     if (adaptCoarse && adaptCoarse->data == fine->data) { /* coarse is adapted from fine */
23760a96aa3bSJed Brown       if (pforestC->pointSelfToAdaptSF) {
23779566063dSJacob Faibussowitsch         PetscCall(PetscObjectReference((PetscObject)(pforestC->pointSelfToAdaptSF)));
23780a96aa3bSJed Brown         *sf = pforestC->pointSelfToAdaptSF;
23790a96aa3bSJed Brown         if (childIds) {
23809566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(pEndF - pStartF, &cids));
23819566063dSJacob Faibussowitsch           PetscCall(PetscArraycpy(cids, pforestC->pointSelfToAdaptCids, pEndF - pStartF));
23820a96aa3bSJed Brown           *childIds = cids;
23830a96aa3bSJed Brown         }
23843ba16761SJacob Faibussowitsch         PetscFunctionReturn(PETSC_SUCCESS);
23850a96aa3bSJed Brown       } else {
23860a96aa3bSJed Brown         saveInCoarse = PETSC_TRUE;
23870a96aa3bSJed Brown         formCids     = PETSC_TRUE;
23880a96aa3bSJed Brown       }
23890a96aa3bSJed Brown     } else if (adaptFine && adaptFine->data == coarse->data) { /* fine is adapted from coarse */
23900a96aa3bSJed Brown       if (pforestF->pointAdaptToSelfSF) {
23919566063dSJacob Faibussowitsch         PetscCall(PetscObjectReference((PetscObject)(pforestF->pointAdaptToSelfSF)));
23920a96aa3bSJed Brown         *sf = pforestF->pointAdaptToSelfSF;
23930a96aa3bSJed Brown         if (childIds) {
23949566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(pEndF - pStartF, &cids));
23959566063dSJacob Faibussowitsch           PetscCall(PetscArraycpy(cids, pforestF->pointAdaptToSelfCids, pEndF - pStartF));
23960a96aa3bSJed Brown           *childIds = cids;
23970a96aa3bSJed Brown         }
23983ba16761SJacob Faibussowitsch         PetscFunctionReturn(PETSC_SUCCESS);
23990a96aa3bSJed Brown       } else {
24000a96aa3bSJed Brown         saveInFine = PETSC_TRUE;
24010a96aa3bSJed Brown         formCids   = PETSC_TRUE;
24020a96aa3bSJed Brown       }
24030a96aa3bSJed Brown     }
24040a96aa3bSJed Brown   }
24050a96aa3bSJed Brown 
24060a96aa3bSJed Brown   /* count the number of closure points that have dofs and create a list */
24070a96aa3bSJed Brown   numClosureIndices = P4EST_INSUL;
24080a96aa3bSJed Brown   /* create the datatype */
24099566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_contiguous(2, MPIU_INT, &nodeType));
24109566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_commit(&nodeType));
24119566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Op_create(DMPforestMaxSFNode, PETSC_FALSE, &sfNodeReduce));
24129566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_contiguous(numClosureIndices * 2, MPIU_INT, &nodeClosureType));
24139566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_commit(&nodeClosureType));
24140a96aa3bSJed Brown   /* everything has to go through cells: for each cell, create a list of the sfnodes in its closure */
24150a96aa3bSJed Brown   /* get lists of closure point SF nodes for every cell */
24169566063dSJacob Faibussowitsch   PetscCall(DMPforestGetCellSFNodes(coarse, numClosureIndices, &numClosurePointsC, &closurePointsC, PETSC_TRUE));
24179566063dSJacob Faibussowitsch   PetscCall(DMPforestGetCellSFNodes(fine, numClosureIndices, &numClosurePointsF, &closurePointsF, PETSC_FALSE));
24180a96aa3bSJed Brown   /* create pointers for tree lists */
24190a96aa3bSJed Brown   fltF = p4estF->first_local_tree;
24200a96aa3bSJed Brown   lltF = p4estF->last_local_tree;
24219566063dSJacob Faibussowitsch   PetscCall(PetscCalloc2(lltF + 1 - fltF, &treeQuads, lltF + 1 - fltF, &treeQuadCounts));
24220a96aa3bSJed Brown   /* if the partitions don't match, ship the coarse to cover the fine */
24230a96aa3bSJed Brown   if (size > 1) {
24240a96aa3bSJed Brown     PetscInt p;
24250a96aa3bSJed Brown 
24260a96aa3bSJed Brown     for (p = 0; p < size; p++) {
24270a96aa3bSJed Brown       int equal;
24280a96aa3bSJed Brown 
2429792fecdfSBarry Smith       PetscCallP4estReturn(equal, p4est_quadrant_is_equal_piggy, (&p4estC->global_first_position[p], &p4estF->global_first_position[p]));
24300a96aa3bSJed Brown       if (!equal) break;
24310a96aa3bSJed Brown     }
24320a96aa3bSJed Brown     if (p < size) { /* non-matching distribution: send the coarse to cover the fine */
24330a96aa3bSJed Brown       PetscInt          cStartC, cEndC;
24340a96aa3bSJed Brown       PetscSF           coveringSF;
24350a96aa3bSJed Brown       PetscInt          nleaves;
24360a96aa3bSJed Brown       PetscInt          count;
24370a96aa3bSJed Brown       PetscSFNode      *newClosurePointsC;
24380a96aa3bSJed Brown       p4est_quadrant_t *coverQuadsSend;
24390a96aa3bSJed Brown       p4est_topidx_t    fltC = p4estC->first_local_tree;
24400a96aa3bSJed Brown       p4est_topidx_t    lltC = p4estC->last_local_tree;
24410a96aa3bSJed Brown       p4est_topidx_t    t;
24420a96aa3bSJed Brown       PetscMPIInt       blockSizes[4]   = {P4EST_DIM, 2, 1, 1};
24439371c9d4SSatish Balay       MPI_Aint          blockOffsets[4] = {offsetof(p4est_quadrant_t, x), offsetof(p4est_quadrant_t, level), offsetof(p4est_quadrant_t, pad16), offsetof(p4est_quadrant_t, p)};
24440a96aa3bSJed Brown       MPI_Datatype      blockTypes[4]   = {MPI_INT32_T, MPI_INT8_T, MPI_INT16_T, MPI_INT32_T /* p.which_tree */};
24450a96aa3bSJed Brown       MPI_Datatype      quadStruct, quadType;
24460a96aa3bSJed Brown 
24479566063dSJacob Faibussowitsch       PetscCall(DMPlexGetSimplexOrBoxCells(plexC, 0, &cStartC, &cEndC));
24489566063dSJacob Faibussowitsch       PetscCall(DMPforestGetCellCoveringSF(comm, p4estC, p4estF, pforestC->cLocalStart, pforestC->cLocalEnd, &coveringSF));
24499566063dSJacob Faibussowitsch       PetscCall(PetscSFGetGraph(coveringSF, NULL, &nleaves, NULL, NULL));
24509566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numClosureIndices * nleaves, &newClosurePointsC));
24519566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nleaves, &coverQuads));
24529566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(cEndC - cStartC, &coverQuadsSend));
24530a96aa3bSJed Brown       count = 0;
24540a96aa3bSJed Brown       for (t = fltC; t <= lltC; t++) { /* unfortunately, we need to pack a send array, since quads are not stored packed in p4est */
24550a96aa3bSJed Brown         p4est_tree_t *tree = &(((p4est_tree_t *)p4estC->trees->array)[t]);
24560a96aa3bSJed Brown         PetscInt      q;
24570a96aa3bSJed Brown 
24589566063dSJacob Faibussowitsch         PetscCall(PetscMemcpy(&coverQuadsSend[count], tree->quadrants.array, tree->quadrants.elem_count * sizeof(p4est_quadrant_t)));
24590a96aa3bSJed Brown         for (q = 0; (size_t)q < tree->quadrants.elem_count; q++) coverQuadsSend[count + q].p.which_tree = t;
24600a96aa3bSJed Brown         count += tree->quadrants.elem_count;
24610a96aa3bSJed Brown       }
24620a96aa3bSJed Brown       /* p is of a union type p4est_quadrant_data, but only the p.which_tree field is active at this time. So, we
24630a96aa3bSJed Brown          have a simple blockTypes[] to use. Note that quadStruct does not count potential padding in array of
24640a96aa3bSJed Brown          p4est_quadrant_t. We have to call MPI_Type_create_resized() to change upper-bound of quadStruct.
24650a96aa3bSJed Brown        */
24669566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Type_create_struct(4, blockSizes, blockOffsets, blockTypes, &quadStruct));
24679566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Type_create_resized(quadStruct, 0, sizeof(p4est_quadrant_t), &quadType));
24689566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Type_commit(&quadType));
24699566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(coveringSF, nodeClosureType, closurePointsC, newClosurePointsC, MPI_REPLACE));
24709566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(coveringSF, quadType, coverQuadsSend, coverQuads, MPI_REPLACE));
24719566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(coveringSF, nodeClosureType, closurePointsC, newClosurePointsC, MPI_REPLACE));
24729566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(coveringSF, quadType, coverQuadsSend, coverQuads, MPI_REPLACE));
24739566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Type_free(&quadStruct));
24749566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Type_free(&quadType));
24759566063dSJacob Faibussowitsch       PetscCall(PetscFree(coverQuadsSend));
24769566063dSJacob Faibussowitsch       PetscCall(PetscFree(closurePointsC));
24779566063dSJacob Faibussowitsch       PetscCall(PetscSFDestroy(&coveringSF));
24780a96aa3bSJed Brown       closurePointsC = newClosurePointsC;
24790a96aa3bSJed Brown 
24800a96aa3bSJed Brown       /* assign tree quads based on locations in coverQuads */
24810a96aa3bSJed Brown       {
24820a96aa3bSJed Brown         PetscInt q;
24830a96aa3bSJed Brown         for (q = 0; q < nleaves; q++) {
24840a96aa3bSJed Brown           p4est_locidx_t t = coverQuads[q].p.which_tree;
24850a96aa3bSJed Brown           if (!treeQuadCounts[t - fltF]++) treeQuads[t - fltF] = &coverQuads[q];
24860a96aa3bSJed Brown         }
24870a96aa3bSJed Brown       }
24880a96aa3bSJed Brown     }
24890a96aa3bSJed Brown   }
24900a96aa3bSJed Brown   if (!coverQuads) { /* matching partitions: assign tree quads based on locations in p4est native arrays */
24910a96aa3bSJed Brown     for (t = fltF; t <= lltF; t++) {
24920a96aa3bSJed Brown       p4est_tree_t *tree = &(((p4est_tree_t *)p4estC->trees->array)[t]);
24930a96aa3bSJed Brown 
24940a96aa3bSJed Brown       treeQuadCounts[t - fltF] = tree->quadrants.elem_count;
24950a96aa3bSJed Brown       treeQuads[t - fltF]      = (p4est_quadrant_t *)tree->quadrants.array;
24960a96aa3bSJed Brown     }
24970a96aa3bSJed Brown   }
24980a96aa3bSJed Brown 
24990a96aa3bSJed Brown   {
25000a96aa3bSJed Brown     PetscInt     p;
25010a96aa3bSJed Brown     PetscInt     cLocalStartF;
25020a96aa3bSJed Brown     PetscSF      pointSF;
25030a96aa3bSJed Brown     PetscSFNode *roots;
25040a96aa3bSJed Brown     PetscInt    *rootType;
25050a96aa3bSJed Brown     DM           refTree = NULL;
25060a96aa3bSJed Brown     DMLabel      canonical;
25070a96aa3bSJed Brown     PetscInt    *childClosures[P4EST_CHILDREN] = {NULL};
25080a96aa3bSJed Brown     PetscInt    *rootClosure                   = NULL;
25090a96aa3bSJed Brown     PetscInt     coarseOffset;
25100a96aa3bSJed Brown     PetscInt     numCoarseQuads;
25110a96aa3bSJed Brown 
25129566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(pEndF - pStartF, &roots));
25139566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(pEndF - pStartF, &rootType));
25149566063dSJacob Faibussowitsch     PetscCall(DMGetPointSF(fine, &pointSF));
25150a96aa3bSJed Brown     for (p = pStartF; p < pEndF; p++) {
25160a96aa3bSJed Brown       roots[p - pStartF].rank  = -1;
25170a96aa3bSJed Brown       roots[p - pStartF].index = -1;
25180a96aa3bSJed Brown       rootType[p - pStartF]    = -1;
25190a96aa3bSJed Brown     }
25200a96aa3bSJed Brown     if (formCids) {
25210a96aa3bSJed Brown       PetscInt child;
25220a96aa3bSJed Brown 
25239566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndF - pStartF, &cids));
25240a96aa3bSJed Brown       for (p = pStartF; p < pEndF; p++) cids[p - pStartF] = -2;
25259566063dSJacob Faibussowitsch       PetscCall(DMPlexGetReferenceTree(plexF, &refTree));
25269566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(refTree, 0, PETSC_TRUE, NULL, &rootClosure));
25270a96aa3bSJed Brown       for (child = 0; child < P4EST_CHILDREN; child++) { /* get the closures of the child cells in the reference tree */
25289566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTransitiveClosure(refTree, child + 1, PETSC_TRUE, NULL, &childClosures[child]));
25290a96aa3bSJed Brown       }
25309566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(refTree, "canonical", &canonical));
25310a96aa3bSJed Brown     }
25320a96aa3bSJed Brown     cLocalStartF = pforestF->cLocalStart;
25330a96aa3bSJed Brown     for (t = fltF, coarseOffset = 0, numCoarseQuads = 0; t <= lltF; t++, coarseOffset += numCoarseQuads) {
25340a96aa3bSJed Brown       p4est_tree_t     *tree         = &(((p4est_tree_t *)p4estF->trees->array)[t]);
25350a96aa3bSJed Brown       PetscInt          numFineQuads = tree->quadrants.elem_count;
25360a96aa3bSJed Brown       p4est_quadrant_t *coarseQuads  = treeQuads[t - fltF];
25370a96aa3bSJed Brown       p4est_quadrant_t *fineQuads    = (p4est_quadrant_t *)tree->quadrants.array;
25380a96aa3bSJed Brown       PetscInt          i, coarseCount = 0;
25390a96aa3bSJed Brown       PetscInt          offset = tree->quadrants_offset;
25400a96aa3bSJed Brown       sc_array_t        coarseQuadsArray;
25410a96aa3bSJed Brown 
25420a96aa3bSJed Brown       numCoarseQuads = treeQuadCounts[t - fltF];
2543792fecdfSBarry Smith       PetscCallP4est(sc_array_init_data, (&coarseQuadsArray, coarseQuads, sizeof(p4est_quadrant_t), (size_t)numCoarseQuads));
25440a96aa3bSJed Brown       for (i = 0; i < numFineQuads; i++) {
25450a96aa3bSJed Brown         PetscInt          c          = i + offset;
25460a96aa3bSJed Brown         p4est_quadrant_t *quad       = &fineQuads[i];
25470a96aa3bSJed Brown         p4est_quadrant_t *quadCoarse = NULL;
25480a96aa3bSJed Brown         ssize_t           disjoint   = -1;
25490a96aa3bSJed Brown 
25500a96aa3bSJed Brown         while (disjoint < 0 && coarseCount < numCoarseQuads) {
25510a96aa3bSJed Brown           quadCoarse = &coarseQuads[coarseCount];
2552792fecdfSBarry Smith           PetscCallP4estReturn(disjoint, p4est_quadrant_disjoint, (quadCoarse, quad));
25530a96aa3bSJed Brown           if (disjoint < 0) coarseCount++;
25540a96aa3bSJed Brown         }
255508401ef6SPierre Jolivet         PetscCheck(disjoint == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "did not find overlapping coarse quad");
25560a96aa3bSJed Brown         if (quadCoarse->level > quad->level || (quadCoarse->level == quad->level && !transferIdent)) { /* the "coarse" mesh is finer than the fine mesh at the point: continue */
25570a96aa3bSJed Brown           if (transferIdent) {                                                                         /* find corners */
25580a96aa3bSJed Brown             PetscInt j = 0;
25590a96aa3bSJed Brown 
25600a96aa3bSJed Brown             do {
25610a96aa3bSJed Brown               if (j < P4EST_CHILDREN) {
25620a96aa3bSJed Brown                 p4est_quadrant_t cornerQuad;
25630a96aa3bSJed Brown                 int              equal;
25640a96aa3bSJed Brown 
2565792fecdfSBarry Smith                 PetscCallP4est(p4est_quadrant_corner_descendant, (quad, &cornerQuad, j, quadCoarse->level));
2566792fecdfSBarry Smith                 PetscCallP4estReturn(equal, p4est_quadrant_is_equal, (&cornerQuad, quadCoarse));
25670a96aa3bSJed Brown                 if (equal) {
25680a96aa3bSJed Brown                   PetscInt    petscJ = P4estVertToPetscVert[j];
25690a96aa3bSJed Brown                   PetscInt    p      = closurePointsF[numClosureIndices * c + (P4EST_INSUL - P4EST_CHILDREN) + petscJ].index;
25700a96aa3bSJed Brown                   PetscSFNode q      = closurePointsC[numClosureIndices * (coarseCount + coarseOffset) + (P4EST_INSUL - P4EST_CHILDREN) + petscJ];
25710a96aa3bSJed Brown 
25720a96aa3bSJed Brown                   roots[p - pStartF]    = q;
25730a96aa3bSJed Brown                   rootType[p - pStartF] = PETSC_MAX_INT;
25740a96aa3bSJed Brown                   cids[p - pStartF]     = -1;
25750a96aa3bSJed Brown                   j++;
25760a96aa3bSJed Brown                 }
25770a96aa3bSJed Brown               }
25780a96aa3bSJed Brown               coarseCount++;
25790a96aa3bSJed Brown               disjoint = 1;
25800a96aa3bSJed Brown               if (coarseCount < numCoarseQuads) {
25810a96aa3bSJed Brown                 quadCoarse = &coarseQuads[coarseCount];
2582792fecdfSBarry Smith                 PetscCallP4estReturn(disjoint, p4est_quadrant_disjoint, (quadCoarse, quad));
25830a96aa3bSJed Brown               }
25840a96aa3bSJed Brown             } while (!disjoint);
25850a96aa3bSJed Brown           }
25860a96aa3bSJed Brown           continue;
25870a96aa3bSJed Brown         }
25880a96aa3bSJed Brown         if (quadCoarse->level == quad->level) { /* same quad present in coarse and fine mesh */
25890a96aa3bSJed Brown           PetscInt j;
25900a96aa3bSJed Brown           for (j = 0; j < numClosureIndices; j++) {
25910a96aa3bSJed Brown             PetscInt p = closurePointsF[numClosureIndices * c + j].index;
25920a96aa3bSJed Brown 
25930a96aa3bSJed Brown             roots[p - pStartF]    = closurePointsC[numClosureIndices * (coarseCount + coarseOffset) + j];
25940a96aa3bSJed Brown             rootType[p - pStartF] = PETSC_MAX_INT; /* unconditionally accept */
25950a96aa3bSJed Brown             cids[p - pStartF]     = -1;
25960a96aa3bSJed Brown           }
25970a96aa3bSJed Brown         } else {
25980a96aa3bSJed Brown           PetscInt levelDiff                 = quad->level - quadCoarse->level;
25990a96aa3bSJed Brown           PetscInt proposedCids[P4EST_INSUL] = {0};
26000a96aa3bSJed Brown 
26010a96aa3bSJed Brown           if (formCids) {
26020a96aa3bSJed Brown             PetscInt  cl;
26030a96aa3bSJed Brown             PetscInt *pointClosure = NULL;
26040a96aa3bSJed Brown             int       cid;
26050a96aa3bSJed Brown 
260608401ef6SPierre Jolivet             PetscCheck(levelDiff <= 1, PETSC_COMM_SELF, PETSC_ERR_USER, "Recursive child ids not implemented");
2607792fecdfSBarry Smith             PetscCallP4estReturn(cid, p4est_quadrant_child_id, (quad));
26089566063dSJacob Faibussowitsch             PetscCall(DMPlexGetTransitiveClosure(plexF, c + cLocalStartF, PETSC_TRUE, NULL, &pointClosure));
26090a96aa3bSJed Brown             for (cl = 0; cl < P4EST_INSUL; cl++) {
26100a96aa3bSJed Brown               PetscInt       p      = pointClosure[2 * cl];
26110a96aa3bSJed Brown               PetscInt       point  = childClosures[cid][2 * cl];
26120a96aa3bSJed Brown               PetscInt       ornt   = childClosures[cid][2 * cl + 1];
26130a96aa3bSJed Brown               PetscInt       newcid = -1;
26140a96aa3bSJed Brown               DMPolytopeType ct;
26150a96aa3bSJed Brown 
26160a96aa3bSJed Brown               if (rootType[p - pStartF] == PETSC_MAX_INT) continue;
26179566063dSJacob Faibussowitsch               PetscCall(DMPlexGetCellType(refTree, point, &ct));
26180a96aa3bSJed Brown               ornt = DMPolytopeConvertNewOrientation_Internal(ct, ornt);
26190a96aa3bSJed Brown               if (!cl) {
26200a96aa3bSJed Brown                 newcid = cid + 1;
26210a96aa3bSJed Brown               } else {
26220a96aa3bSJed Brown                 PetscInt rcl, parent, parentOrnt = 0;
26230a96aa3bSJed Brown 
26249566063dSJacob Faibussowitsch                 PetscCall(DMPlexGetTreeParent(refTree, point, &parent, NULL));
26250a96aa3bSJed Brown                 if (parent == point) {
26260a96aa3bSJed Brown                   newcid = -1;
26270a96aa3bSJed Brown                 } else if (!parent) { /* in the root */
26280a96aa3bSJed Brown                   newcid = point;
26290a96aa3bSJed Brown                 } else {
26300a96aa3bSJed Brown                   DMPolytopeType rct = DM_POLYTOPE_UNKNOWN;
26310a96aa3bSJed Brown 
26320a96aa3bSJed Brown                   for (rcl = 1; rcl < P4EST_INSUL; rcl++) {
26330a96aa3bSJed Brown                     if (rootClosure[2 * rcl] == parent) {
26349566063dSJacob Faibussowitsch                       PetscCall(DMPlexGetCellType(refTree, parent, &rct));
26350a96aa3bSJed Brown                       parentOrnt = DMPolytopeConvertNewOrientation_Internal(rct, rootClosure[2 * rcl + 1]);
26360a96aa3bSJed Brown                       break;
26370a96aa3bSJed Brown                     }
26380a96aa3bSJed Brown                   }
263908401ef6SPierre Jolivet                   PetscCheck(rcl < P4EST_INSUL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Couldn't find parent in root closure");
26409566063dSJacob Faibussowitsch                   PetscCall(DMPlexReferenceTreeGetChildSymmetry(refTree, parent, parentOrnt, ornt, point, DMPolytopeConvertNewOrientation_Internal(rct, pointClosure[2 * rcl + 1]), NULL, &newcid));
26410a96aa3bSJed Brown                 }
26420a96aa3bSJed Brown               }
26430a96aa3bSJed Brown               if (newcid >= 0) {
264448a46eb9SPierre Jolivet                 if (canonical) PetscCall(DMLabelGetValue(canonical, newcid, &newcid));
26450a96aa3bSJed Brown                 proposedCids[cl] = newcid;
26460a96aa3bSJed Brown               }
26470a96aa3bSJed Brown             }
26489566063dSJacob Faibussowitsch             PetscCall(DMPlexRestoreTransitiveClosure(plexF, c + cLocalStartF, PETSC_TRUE, NULL, &pointClosure));
26490a96aa3bSJed Brown           }
26509371c9d4SSatish Balay           p4est_qcoord_t coarseBound[2][P4EST_DIM] = {
26519371c9d4SSatish Balay             {quadCoarse->x, quadCoarse->y,
26520a96aa3bSJed Brown   #if defined(P4_TO_P8)
26530a96aa3bSJed Brown              quadCoarse->z
26540a96aa3bSJed Brown   #endif
26559371c9d4SSatish Balay             },
26569371c9d4SSatish Balay             {0}
26579371c9d4SSatish Balay           };
26589371c9d4SSatish Balay           p4est_qcoord_t fineBound[2][P4EST_DIM] = {
26599371c9d4SSatish Balay             {quad->x, quad->y,
26600a96aa3bSJed Brown   #if defined(P4_TO_P8)
26610a96aa3bSJed Brown              quad->z
26620a96aa3bSJed Brown   #endif
26639371c9d4SSatish Balay             },
26649371c9d4SSatish Balay             {0}
26659371c9d4SSatish Balay           };
26660a96aa3bSJed Brown           PetscInt j;
26670a96aa3bSJed Brown           for (j = 0; j < P4EST_DIM; j++) { /* get the coordinates of cell boundaries in each direction */
26680a96aa3bSJed Brown             coarseBound[1][j] = coarseBound[0][j] + P4EST_QUADRANT_LEN(quadCoarse->level);
26690a96aa3bSJed Brown             fineBound[1][j]   = fineBound[0][j] + P4EST_QUADRANT_LEN(quad->level);
26700a96aa3bSJed Brown           }
26710a96aa3bSJed Brown           for (j = 0; j < numClosureIndices; j++) {
26720a96aa3bSJed Brown             PetscInt    l, p;
26730a96aa3bSJed Brown             PetscSFNode q;
26740a96aa3bSJed Brown 
26750a96aa3bSJed Brown             p = closurePointsF[numClosureIndices * c + j].index;
26760a96aa3bSJed Brown             if (rootType[p - pStartF] == PETSC_MAX_INT) continue;
26770a96aa3bSJed Brown             if (j == 0) { /* volume: ancestor is volume */
26780a96aa3bSJed Brown               l = 0;
26790a96aa3bSJed Brown             } else if (j < 1 + P4EST_FACES) { /* facet */
26800a96aa3bSJed Brown               PetscInt face       = PetscFaceToP4estFace[j - 1];
26810a96aa3bSJed Brown               PetscInt direction  = face / 2;
26820a96aa3bSJed Brown               PetscInt coarseFace = -1;
26830a96aa3bSJed Brown 
26840a96aa3bSJed Brown               if (coarseBound[face % 2][direction] == fineBound[face % 2][direction]) {
26850a96aa3bSJed Brown                 coarseFace = face;
26860a96aa3bSJed Brown                 l          = 1 + P4estFaceToPetscFace[coarseFace];
26870a96aa3bSJed Brown               } else {
26880a96aa3bSJed Brown                 l = 0;
26890a96aa3bSJed Brown               }
26900a96aa3bSJed Brown   #if defined(P4_TO_P8)
26910a96aa3bSJed Brown             } else if (j < 1 + P4EST_FACES + P8EST_EDGES) {
26920a96aa3bSJed Brown               PetscInt  edge       = PetscEdgeToP4estEdge[j - (1 + P4EST_FACES)];
26930a96aa3bSJed Brown               PetscInt  direction  = edge / 4;
26940a96aa3bSJed Brown               PetscInt  mod        = edge % 4;
26950a96aa3bSJed Brown               PetscInt  coarseEdge = -1, coarseFace = -1;
26960a96aa3bSJed Brown               PetscInt  minDir = PetscMin((direction + 1) % 3, (direction + 2) % 3);
26970a96aa3bSJed Brown               PetscInt  maxDir = PetscMax((direction + 1) % 3, (direction + 2) % 3);
26980a96aa3bSJed Brown               PetscBool dirTest[2];
26990a96aa3bSJed Brown 
27000a96aa3bSJed Brown               dirTest[0] = (PetscBool)(coarseBound[mod % 2][minDir] == fineBound[mod % 2][minDir]);
27010a96aa3bSJed Brown               dirTest[1] = (PetscBool)(coarseBound[mod / 2][maxDir] == fineBound[mod / 2][maxDir]);
27020a96aa3bSJed Brown 
27030a96aa3bSJed Brown               if (dirTest[0] && dirTest[1]) { /* fine edge falls on coarse edge */
27040a96aa3bSJed Brown                 coarseEdge = edge;
27050a96aa3bSJed Brown                 l          = 1 + P4EST_FACES + P4estEdgeToPetscEdge[coarseEdge];
27060a96aa3bSJed Brown               } else if (dirTest[0]) { /* fine edge falls on a coarse face in the minDir direction */
27070a96aa3bSJed Brown                 coarseFace = 2 * minDir + (mod % 2);
27080a96aa3bSJed Brown                 l          = 1 + P4estFaceToPetscFace[coarseFace];
27090a96aa3bSJed Brown               } else if (dirTest[1]) { /* fine edge falls on a coarse face in the maxDir direction */
27100a96aa3bSJed Brown                 coarseFace = 2 * maxDir + (mod / 2);
27110a96aa3bSJed Brown                 l          = 1 + P4estFaceToPetscFace[coarseFace];
27120a96aa3bSJed Brown               } else {
27130a96aa3bSJed Brown                 l = 0;
27140a96aa3bSJed Brown               }
27150a96aa3bSJed Brown   #endif
27160a96aa3bSJed Brown             } else {
27170a96aa3bSJed Brown               PetscInt  vertex = PetscVertToP4estVert[P4EST_CHILDREN - (P4EST_INSUL - j)];
27180a96aa3bSJed Brown               PetscBool dirTest[P4EST_DIM];
27190a96aa3bSJed Brown               PetscInt  m;
27200a96aa3bSJed Brown               PetscInt  numMatch     = 0;
27210a96aa3bSJed Brown               PetscInt  coarseVertex = -1, coarseFace = -1;
27220a96aa3bSJed Brown   #if defined(P4_TO_P8)
27230a96aa3bSJed Brown               PetscInt coarseEdge = -1;
27240a96aa3bSJed Brown   #endif
27250a96aa3bSJed Brown 
27260a96aa3bSJed Brown               for (m = 0; m < P4EST_DIM; m++) {
27270a96aa3bSJed Brown                 dirTest[m] = (PetscBool)(coarseBound[(vertex >> m) & 1][m] == fineBound[(vertex >> m) & 1][m]);
27280a96aa3bSJed Brown                 if (dirTest[m]) numMatch++;
27290a96aa3bSJed Brown               }
27300a96aa3bSJed Brown               if (numMatch == P4EST_DIM) { /* vertex on vertex */
27310a96aa3bSJed Brown                 coarseVertex = vertex;
27320a96aa3bSJed Brown                 l            = P4EST_INSUL - (P4EST_CHILDREN - P4estVertToPetscVert[coarseVertex]);
27330a96aa3bSJed Brown               } else if (numMatch == 1) { /* vertex on face */
27340a96aa3bSJed Brown                 for (m = 0; m < P4EST_DIM; m++) {
27350a96aa3bSJed Brown                   if (dirTest[m]) {
27360a96aa3bSJed Brown                     coarseFace = 2 * m + ((vertex >> m) & 1);
27370a96aa3bSJed Brown                     break;
27380a96aa3bSJed Brown                   }
27390a96aa3bSJed Brown                 }
27400a96aa3bSJed Brown                 l = 1 + P4estFaceToPetscFace[coarseFace];
27410a96aa3bSJed Brown   #if defined(P4_TO_P8)
27420a96aa3bSJed Brown               } else if (numMatch == 2) { /* vertex on edge */
27430a96aa3bSJed Brown                 for (m = 0; m < P4EST_DIM; m++) {
27440a96aa3bSJed Brown                   if (!dirTest[m]) {
27450a96aa3bSJed Brown                     PetscInt otherDir1 = (m + 1) % 3;
27460a96aa3bSJed Brown                     PetscInt otherDir2 = (m + 2) % 3;
27470a96aa3bSJed Brown                     PetscInt minDir    = PetscMin(otherDir1, otherDir2);
27480a96aa3bSJed Brown                     PetscInt maxDir    = PetscMax(otherDir1, otherDir2);
27490a96aa3bSJed Brown 
27500a96aa3bSJed Brown                     coarseEdge = m * 4 + 2 * ((vertex >> maxDir) & 1) + ((vertex >> minDir) & 1);
27510a96aa3bSJed Brown                     break;
27520a96aa3bSJed Brown                   }
27530a96aa3bSJed Brown                 }
27540a96aa3bSJed Brown                 l = 1 + P4EST_FACES + P4estEdgeToPetscEdge[coarseEdge];
27550a96aa3bSJed Brown   #endif
27560a96aa3bSJed Brown               } else { /* volume */
27570a96aa3bSJed Brown                 l = 0;
27580a96aa3bSJed Brown               }
27590a96aa3bSJed Brown             }
27600a96aa3bSJed Brown             q = closurePointsC[numClosureIndices * (coarseCount + coarseOffset) + l];
27610a96aa3bSJed Brown             if (l > rootType[p - pStartF]) {
27620a96aa3bSJed Brown               if (l >= P4EST_INSUL - P4EST_CHILDREN) { /* vertex on vertex: unconditional acceptance */
27630a96aa3bSJed Brown                 if (transferIdent) {
27640a96aa3bSJed Brown                   roots[p - pStartF]    = q;
27650a96aa3bSJed Brown                   rootType[p - pStartF] = PETSC_MAX_INT;
27660a96aa3bSJed Brown                   if (formCids) cids[p - pStartF] = -1;
27670a96aa3bSJed Brown                 }
27680a96aa3bSJed Brown               } else {
27690a96aa3bSJed Brown                 PetscInt k, thisp = p, limit;
27700a96aa3bSJed Brown 
27710a96aa3bSJed Brown                 roots[p - pStartF]    = q;
27720a96aa3bSJed Brown                 rootType[p - pStartF] = l;
27730a96aa3bSJed Brown                 if (formCids) cids[p - pStartF] = proposedCids[j];
27740a96aa3bSJed Brown                 limit = transferIdent ? levelDiff : (levelDiff - 1);
27750a96aa3bSJed Brown                 for (k = 0; k < limit; k++) {
27760a96aa3bSJed Brown                   PetscInt parent;
27770a96aa3bSJed Brown 
27789566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetTreeParent(plexF, thisp, &parent, NULL));
27790a96aa3bSJed Brown                   if (parent == thisp) break;
27800a96aa3bSJed Brown 
27810a96aa3bSJed Brown                   roots[parent - pStartF]    = q;
27820a96aa3bSJed Brown                   rootType[parent - pStartF] = PETSC_MAX_INT;
27830a96aa3bSJed Brown                   if (formCids) cids[parent - pStartF] = -1;
27840a96aa3bSJed Brown                   thisp = parent;
27850a96aa3bSJed Brown                 }
27860a96aa3bSJed Brown               }
27870a96aa3bSJed Brown             }
27880a96aa3bSJed Brown           }
27890a96aa3bSJed Brown         }
27900a96aa3bSJed Brown       }
27910a96aa3bSJed Brown     }
27920a96aa3bSJed Brown 
27930a96aa3bSJed Brown     /* now every cell has labeled the points in its closure, so we first make sure everyone agrees by reducing to roots, and the broadcast the agreements */
27940a96aa3bSJed Brown     if (size > 1) {
27950a96aa3bSJed Brown       PetscInt *rootTypeCopy, p;
27960a96aa3bSJed Brown 
27979566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndF - pStartF, &rootTypeCopy));
27989566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(rootTypeCopy, rootType, pEndF - pStartF));
279957168dbeSPierre Jolivet       PetscCall(PetscSFReduceBegin(pointSF, MPIU_INT, rootTypeCopy, rootTypeCopy, MPI_MAX));
280057168dbeSPierre Jolivet       PetscCall(PetscSFReduceEnd(pointSF, MPIU_INT, rootTypeCopy, rootTypeCopy, MPI_MAX));
28019566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(pointSF, MPIU_INT, rootTypeCopy, rootTypeCopy, MPI_REPLACE));
28029566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(pointSF, MPIU_INT, rootTypeCopy, rootTypeCopy, MPI_REPLACE));
28030a96aa3bSJed Brown       for (p = pStartF; p < pEndF; p++) {
28040a96aa3bSJed Brown         if (rootTypeCopy[p - pStartF] > rootType[p - pStartF]) { /* another process found a root of higher type (e.g. vertex instead of edge), which we want to accept, so nullify this */
28050a96aa3bSJed Brown           roots[p - pStartF].rank  = -1;
28060a96aa3bSJed Brown           roots[p - pStartF].index = -1;
28070a96aa3bSJed Brown         }
28089371c9d4SSatish Balay         if (formCids && rootTypeCopy[p - pStartF] == PETSC_MAX_INT) { cids[p - pStartF] = -1; /* we have found an antecedent that is the same: no child id */ }
28090a96aa3bSJed Brown       }
28109566063dSJacob Faibussowitsch       PetscCall(PetscFree(rootTypeCopy));
28119566063dSJacob Faibussowitsch       PetscCall(PetscSFReduceBegin(pointSF, nodeType, roots, roots, sfNodeReduce));
28129566063dSJacob Faibussowitsch       PetscCall(PetscSFReduceEnd(pointSF, nodeType, roots, roots, sfNodeReduce));
28139566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(pointSF, nodeType, roots, roots, MPI_REPLACE));
28149566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(pointSF, nodeType, roots, roots, MPI_REPLACE));
28150a96aa3bSJed Brown     }
28169566063dSJacob Faibussowitsch     PetscCall(PetscFree(rootType));
28170a96aa3bSJed Brown 
28180a96aa3bSJed Brown     {
28190a96aa3bSJed Brown       PetscInt     numRoots;
28200a96aa3bSJed Brown       PetscInt     numLeaves;
28210a96aa3bSJed Brown       PetscInt    *leaves;
28220a96aa3bSJed Brown       PetscSFNode *iremote;
28230a96aa3bSJed Brown       /* count leaves */
28240a96aa3bSJed Brown 
28250a96aa3bSJed Brown       numRoots = pEndC - pStartC;
28260a96aa3bSJed Brown 
28270a96aa3bSJed Brown       numLeaves = 0;
28280a96aa3bSJed Brown       for (p = pStartF; p < pEndF; p++) {
28290a96aa3bSJed Brown         if (roots[p - pStartF].index >= 0) numLeaves++;
28300a96aa3bSJed Brown       }
28319566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numLeaves, &leaves));
28329566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(numLeaves, &iremote));
28330a96aa3bSJed Brown       numLeaves = 0;
28340a96aa3bSJed Brown       for (p = pStartF; p < pEndF; p++) {
28350a96aa3bSJed Brown         if (roots[p - pStartF].index >= 0) {
28360a96aa3bSJed Brown           leaves[numLeaves]  = p - pStartF;
28370a96aa3bSJed Brown           iremote[numLeaves] = roots[p - pStartF];
28380a96aa3bSJed Brown           numLeaves++;
28390a96aa3bSJed Brown         }
28400a96aa3bSJed Brown       }
28419566063dSJacob Faibussowitsch       PetscCall(PetscFree(roots));
28429566063dSJacob Faibussowitsch       PetscCall(PetscSFCreate(comm, sf));
28430a96aa3bSJed Brown       if (numLeaves == (pEndF - pStartF)) {
28449566063dSJacob Faibussowitsch         PetscCall(PetscFree(leaves));
28459566063dSJacob Faibussowitsch         PetscCall(PetscSFSetGraph(*sf, numRoots, numLeaves, NULL, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER));
28460a96aa3bSJed Brown       } else {
28479566063dSJacob Faibussowitsch         PetscCall(PetscSFSetGraph(*sf, numRoots, numLeaves, leaves, PETSC_OWN_POINTER, iremote, PETSC_OWN_POINTER));
28480a96aa3bSJed Brown       }
28490a96aa3bSJed Brown     }
28500a96aa3bSJed Brown     if (formCids) {
28510a96aa3bSJed Brown       PetscSF  pointSF;
28520a96aa3bSJed Brown       PetscInt child;
28530a96aa3bSJed Brown 
28549566063dSJacob Faibussowitsch       PetscCall(DMPlexGetReferenceTree(plexF, &refTree));
28559566063dSJacob Faibussowitsch       PetscCall(DMGetPointSF(plexF, &pointSF));
285657168dbeSPierre Jolivet       PetscCall(PetscSFReduceBegin(pointSF, MPIU_INT, cids, cids, MPI_MAX));
285757168dbeSPierre Jolivet       PetscCall(PetscSFReduceEnd(pointSF, MPIU_INT, cids, cids, MPI_MAX));
28580a96aa3bSJed Brown       if (childIds) *childIds = cids;
285948a46eb9SPierre Jolivet       for (child = 0; child < P4EST_CHILDREN; child++) PetscCall(DMPlexRestoreTransitiveClosure(refTree, child + 1, PETSC_TRUE, NULL, &childClosures[child]));
28609566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(refTree, 0, PETSC_TRUE, NULL, &rootClosure));
28610a96aa3bSJed Brown     }
28620a96aa3bSJed Brown   }
28630a96aa3bSJed Brown   if (saveInCoarse) { /* cache results */
28649566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)*sf));
28650a96aa3bSJed Brown     pforestC->pointSelfToAdaptSF = *sf;
28660a96aa3bSJed Brown     if (!childIds) {
28670a96aa3bSJed Brown       pforestC->pointSelfToAdaptCids = cids;
28680a96aa3bSJed Brown     } else {
28699566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndF - pStartF, &pforestC->pointSelfToAdaptCids));
28709566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(pforestC->pointSelfToAdaptCids, cids, pEndF - pStartF));
28710a96aa3bSJed Brown     }
28720a96aa3bSJed Brown   } else if (saveInFine) {
28739566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)*sf));
28740a96aa3bSJed Brown     pforestF->pointAdaptToSelfSF = *sf;
28750a96aa3bSJed Brown     if (!childIds) {
28760a96aa3bSJed Brown       pforestF->pointAdaptToSelfCids = cids;
28770a96aa3bSJed Brown     } else {
28789566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndF - pStartF, &pforestF->pointAdaptToSelfCids));
28799566063dSJacob Faibussowitsch       PetscCall(PetscArraycpy(pforestF->pointAdaptToSelfCids, cids, pEndF - pStartF));
28800a96aa3bSJed Brown     }
28810a96aa3bSJed Brown   }
28829566063dSJacob Faibussowitsch   PetscCall(PetscFree2(treeQuads, treeQuadCounts));
28839566063dSJacob Faibussowitsch   PetscCall(PetscFree(coverQuads));
28849566063dSJacob Faibussowitsch   PetscCall(PetscFree(closurePointsC));
28859566063dSJacob Faibussowitsch   PetscCall(PetscFree(closurePointsF));
28869566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_free(&nodeClosureType));
28879566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Op_free(&sfNodeReduce));
28889566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Type_free(&nodeType));
28893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
28900a96aa3bSJed Brown }
28910a96aa3bSJed Brown 
28920a96aa3bSJed Brown /* children are sf leaves of parents */
2893d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetTransferSF_Internal(DM coarse, DM fine, const PetscInt dofPerDim[], PetscSF *sf, PetscBool transferIdent, PetscInt *childIds[])
2894d71ae5a4SJacob Faibussowitsch {
28950a96aa3bSJed Brown   MPI_Comm           comm;
2896d70f29a3SPierre Jolivet   PetscMPIInt        rank;
28970a96aa3bSJed Brown   DM_Forest_pforest *pforestC, *pforestF;
28980a96aa3bSJed Brown   DM                 plexC, plexF;
28990a96aa3bSJed Brown   PetscInt           pStartC, pEndC, pStartF, pEndF;
29000a96aa3bSJed Brown   PetscSF            pointTransferSF;
29010a96aa3bSJed Brown   PetscBool          allOnes = PETSC_TRUE;
29020a96aa3bSJed Brown 
29030a96aa3bSJed Brown   PetscFunctionBegin;
29040a96aa3bSJed Brown   pforestC = (DM_Forest_pforest *)((DM_Forest *)coarse->data)->data;
29050a96aa3bSJed Brown   pforestF = (DM_Forest_pforest *)((DM_Forest *)fine->data)->data;
290608401ef6SPierre Jolivet   PetscCheck(pforestC->topo == pforestF->topo, PetscObjectComm((PetscObject)coarse), PETSC_ERR_ARG_INCOMP, "DM's must have the same base DM");
29070a96aa3bSJed Brown   comm = PetscObjectComm((PetscObject)coarse);
29089566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
29090a96aa3bSJed Brown 
29100a96aa3bSJed Brown   {
29110a96aa3bSJed Brown     PetscInt i;
29120a96aa3bSJed Brown     for (i = 0; i <= P4EST_DIM; i++) {
29130a96aa3bSJed Brown       if (dofPerDim[i] != 1) {
29140a96aa3bSJed Brown         allOnes = PETSC_FALSE;
29150a96aa3bSJed Brown         break;
29160a96aa3bSJed Brown       }
29170a96aa3bSJed Brown     }
29180a96aa3bSJed Brown   }
29199566063dSJacob Faibussowitsch   PetscCall(DMPforestGetTransferSF_Point(coarse, fine, &pointTransferSF, transferIdent, childIds));
29200a96aa3bSJed Brown   if (allOnes) {
29210a96aa3bSJed Brown     *sf = pointTransferSF;
29223ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
29230a96aa3bSJed Brown   }
29240a96aa3bSJed Brown 
29259566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(fine, &plexF));
29269566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plexF, &pStartF, &pEndF));
29279566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(coarse, &plexC));
29289566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plexC, &pStartC, &pEndC));
29290a96aa3bSJed Brown   {
29300a96aa3bSJed Brown     PetscInt           numRoots;
29310a96aa3bSJed Brown     PetscInt           numLeaves;
29320a96aa3bSJed Brown     const PetscInt    *leaves;
29330a96aa3bSJed Brown     const PetscSFNode *iremote;
29340a96aa3bSJed Brown     PetscInt           d;
29350a96aa3bSJed Brown     PetscSection       leafSection, rootSection;
29360a96aa3bSJed Brown     /* count leaves */
29370a96aa3bSJed Brown 
29389566063dSJacob Faibussowitsch     PetscCall(PetscSFGetGraph(pointTransferSF, &numRoots, &numLeaves, &leaves, &iremote));
29399566063dSJacob Faibussowitsch     PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &rootSection));
29409566063dSJacob Faibussowitsch     PetscCall(PetscSectionCreate(PETSC_COMM_SELF, &leafSection));
29419566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(rootSection, pStartC, pEndC));
29429566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(leafSection, pStartF, pEndF));
29430a96aa3bSJed Brown 
29440a96aa3bSJed Brown     for (d = 0; d <= P4EST_DIM; d++) {
29450a96aa3bSJed Brown       PetscInt startC, endC, e;
29460a96aa3bSJed Brown 
29479566063dSJacob Faibussowitsch       PetscCall(DMPlexGetSimplexOrBoxCells(plexC, P4EST_DIM - d, &startC, &endC));
294848a46eb9SPierre Jolivet       for (e = startC; e < endC; e++) PetscCall(PetscSectionSetDof(rootSection, e, dofPerDim[d]));
29490a96aa3bSJed Brown     }
29500a96aa3bSJed Brown 
29510a96aa3bSJed Brown     for (d = 0; d <= P4EST_DIM; d++) {
29520a96aa3bSJed Brown       PetscInt startF, endF, e;
29530a96aa3bSJed Brown 
29549566063dSJacob Faibussowitsch       PetscCall(DMPlexGetSimplexOrBoxCells(plexF, P4EST_DIM - d, &startF, &endF));
295548a46eb9SPierre Jolivet       for (e = startF; e < endF; e++) PetscCall(PetscSectionSetDof(leafSection, e, dofPerDim[d]));
29560a96aa3bSJed Brown     }
29570a96aa3bSJed Brown 
29589566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(rootSection));
29599566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(leafSection));
29600a96aa3bSJed Brown     {
29610a96aa3bSJed Brown       PetscInt     nroots, nleaves;
29620a96aa3bSJed Brown       PetscInt    *mine, i, p;
29630a96aa3bSJed Brown       PetscInt    *offsets, *offsetsRoot;
29640a96aa3bSJed Brown       PetscSFNode *remote;
29650a96aa3bSJed Brown 
29669566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndF - pStartF, &offsets));
29679566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(pEndC - pStartC, &offsetsRoot));
296848a46eb9SPierre Jolivet       for (p = pStartC; p < pEndC; p++) PetscCall(PetscSectionGetOffset(rootSection, p, &offsetsRoot[p - pStartC]));
29699566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(pointTransferSF, MPIU_INT, offsetsRoot, offsets, MPI_REPLACE));
29709566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(pointTransferSF, MPIU_INT, offsetsRoot, offsets, MPI_REPLACE));
29719566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetStorageSize(rootSection, &nroots));
29720a96aa3bSJed Brown       nleaves = 0;
29730a96aa3bSJed Brown       for (i = 0; i < numLeaves; i++) {
29740a96aa3bSJed Brown         PetscInt leaf = leaves ? leaves[i] : i;
29750a96aa3bSJed Brown         PetscInt dof;
29760a96aa3bSJed Brown 
29779566063dSJacob Faibussowitsch         PetscCall(PetscSectionGetDof(leafSection, leaf, &dof));
29780a96aa3bSJed Brown         nleaves += dof;
29790a96aa3bSJed Brown       }
29809566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nleaves, &mine));
29819566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(nleaves, &remote));
29820a96aa3bSJed Brown       nleaves = 0;
29830a96aa3bSJed Brown       for (i = 0; i < numLeaves; i++) {
29840a96aa3bSJed Brown         PetscInt leaf = leaves ? leaves[i] : i;
29850a96aa3bSJed Brown         PetscInt dof;
29860a96aa3bSJed Brown         PetscInt off, j;
29870a96aa3bSJed Brown 
29889566063dSJacob Faibussowitsch         PetscCall(PetscSectionGetDof(leafSection, leaf, &dof));
29899566063dSJacob Faibussowitsch         PetscCall(PetscSectionGetOffset(leafSection, leaf, &off));
29900a96aa3bSJed Brown         for (j = 0; j < dof; j++) {
29910a96aa3bSJed Brown           remote[nleaves].rank  = iremote[i].rank;
29920a96aa3bSJed Brown           remote[nleaves].index = offsets[leaf] + j;
29930a96aa3bSJed Brown           mine[nleaves++]       = off + j;
29940a96aa3bSJed Brown         }
29950a96aa3bSJed Brown       }
29969566063dSJacob Faibussowitsch       PetscCall(PetscFree(offsetsRoot));
29979566063dSJacob Faibussowitsch       PetscCall(PetscFree(offsets));
29989566063dSJacob Faibussowitsch       PetscCall(PetscSFCreate(comm, sf));
29999566063dSJacob Faibussowitsch       PetscCall(PetscSFSetGraph(*sf, nroots, nleaves, mine, PETSC_OWN_POINTER, remote, PETSC_OWN_POINTER));
30000a96aa3bSJed Brown     }
30019566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&leafSection));
30029566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&rootSection));
30039566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&pointTransferSF));
30040a96aa3bSJed Brown   }
30053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30060a96aa3bSJed Brown }
30070a96aa3bSJed Brown 
3008d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetTransferSF(DM dmA, DM dmB, const PetscInt dofPerDim[], PetscSF *sfAtoB, PetscSF *sfBtoA)
3009d71ae5a4SJacob Faibussowitsch {
30100a96aa3bSJed Brown   DM          adaptA, adaptB;
30110a96aa3bSJed Brown   DMAdaptFlag purpose;
30120a96aa3bSJed Brown 
30130a96aa3bSJed Brown   PetscFunctionBegin;
30149566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dmA, &adaptA));
30159566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dmB, &adaptB));
30160a96aa3bSJed Brown   /* it is more efficient when the coarser mesh is the first argument: reorder if we know one is coarser than the other */
30170a96aa3bSJed Brown   if (adaptA && adaptA->data == dmB->data) { /* dmA was adapted from dmB */
30189566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityPurpose(dmA, &purpose));
30190a96aa3bSJed Brown     if (purpose == DM_ADAPT_REFINE) {
30209566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF(dmB, dmA, dofPerDim, sfBtoA, sfAtoB));
30213ba16761SJacob Faibussowitsch       PetscFunctionReturn(PETSC_SUCCESS);
30220a96aa3bSJed Brown     }
30230a96aa3bSJed Brown   } else if (adaptB && adaptB->data == dmA->data) { /* dmB was adapted from dmA */
30249566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityPurpose(dmB, &purpose));
30250a96aa3bSJed Brown     if (purpose == DM_ADAPT_COARSEN) {
30269566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF(dmB, dmA, dofPerDim, sfBtoA, sfAtoB));
30273ba16761SJacob Faibussowitsch       PetscFunctionReturn(PETSC_SUCCESS);
30280a96aa3bSJed Brown     }
30290a96aa3bSJed Brown   }
30301baa6e33SBarry Smith   if (sfAtoB) PetscCall(DMPforestGetTransferSF_Internal(dmA, dmB, dofPerDim, sfAtoB, PETSC_TRUE, NULL));
30311baa6e33SBarry Smith   if (sfBtoA) PetscCall(DMPforestGetTransferSF_Internal(dmB, dmA, dofPerDim, sfBtoA, (PetscBool)(sfAtoB == NULL), NULL));
30323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30330a96aa3bSJed Brown }
30340a96aa3bSJed Brown 
3035d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestLabelsInitialize(DM dm, DM plex)
3036d71ae5a4SJacob Faibussowitsch {
30370a96aa3bSJed Brown   DM_Forest         *forest  = (DM_Forest *)dm->data;
30380a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)forest->data;
30390a96aa3bSJed Brown   PetscInt           cLocalStart, cLocalEnd, cStart, cEnd, fStart, fEnd, eStart, eEnd, vStart, vEnd;
30400a96aa3bSJed Brown   PetscInt           cStartBase, cEndBase, fStartBase, fEndBase, vStartBase, vEndBase, eStartBase, eEndBase;
30410a96aa3bSJed Brown   PetscInt           pStart, pEnd, pStartBase, pEndBase, p;
30420a96aa3bSJed Brown   DM                 base;
30430a96aa3bSJed Brown   PetscInt          *star      = NULL, starSize;
30440a96aa3bSJed Brown   DMLabelLink        next      = dm->labels;
30450a96aa3bSJed Brown   PetscInt           guess     = 0;
30460a96aa3bSJed Brown   p4est_topidx_t     num_trees = pforest->topo->conn->num_trees;
30470a96aa3bSJed Brown 
30480a96aa3bSJed Brown   PetscFunctionBegin;
30490a96aa3bSJed Brown   pforest->labelsFinalized = PETSC_TRUE;
30500a96aa3bSJed Brown   cLocalStart              = pforest->cLocalStart;
30510a96aa3bSJed Brown   cLocalEnd                = pforest->cLocalEnd;
30529566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseDM(dm, &base));
30530a96aa3bSJed Brown   if (!base) {
30540a96aa3bSJed Brown     if (pforest->ghostName) { /* insert a label to make the boundaries, with stratum values denoting which face of the element touches the boundary */
30550a96aa3bSJed Brown       p4est_connectivity_t *conn  = pforest->topo->conn;
30560a96aa3bSJed Brown       p4est_t              *p4est = pforest->forest;
30570a96aa3bSJed Brown       p4est_tree_t         *trees = (p4est_tree_t *)p4est->trees->array;
30580a96aa3bSJed Brown       p4est_topidx_t        t, flt = p4est->first_local_tree;
30590a96aa3bSJed Brown       p4est_topidx_t        llt = pforest->forest->last_local_tree;
30600a96aa3bSJed Brown       DMLabel               ghostLabel;
30610a96aa3bSJed Brown       PetscInt              c;
30620a96aa3bSJed Brown 
30639566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(plex, pforest->ghostName));
30649566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(plex, pforest->ghostName, &ghostLabel));
30650a96aa3bSJed Brown       for (c = cLocalStart, t = flt; t <= llt; t++) {
30660a96aa3bSJed Brown         p4est_tree_t     *tree     = &trees[t];
30670a96aa3bSJed Brown         p4est_quadrant_t *quads    = (p4est_quadrant_t *)tree->quadrants.array;
30680a96aa3bSJed Brown         PetscInt          numQuads = (PetscInt)tree->quadrants.elem_count;
30690a96aa3bSJed Brown         PetscInt          q;
30700a96aa3bSJed Brown 
30710a96aa3bSJed Brown         for (q = 0; q < numQuads; q++, c++) {
30720a96aa3bSJed Brown           p4est_quadrant_t *quad = &quads[q];
30730a96aa3bSJed Brown           PetscInt          f;
30740a96aa3bSJed Brown 
30750a96aa3bSJed Brown           for (f = 0; f < P4EST_FACES; f++) {
30760a96aa3bSJed Brown             p4est_quadrant_t neigh;
30770a96aa3bSJed Brown             int              isOutside;
30780a96aa3bSJed Brown 
3079792fecdfSBarry Smith             PetscCallP4est(p4est_quadrant_face_neighbor, (quad, f, &neigh));
3080792fecdfSBarry Smith             PetscCallP4estReturn(isOutside, p4est_quadrant_is_outside_face, (&neigh));
30810a96aa3bSJed Brown             if (isOutside) {
30820a96aa3bSJed Brown               p4est_topidx_t nt;
30830a96aa3bSJed Brown               PetscInt       nf;
30840a96aa3bSJed Brown 
30850a96aa3bSJed Brown               nt = conn->tree_to_tree[t * P4EST_FACES + f];
30860a96aa3bSJed Brown               nf = (PetscInt)conn->tree_to_face[t * P4EST_FACES + f];
30870a96aa3bSJed Brown               nf = nf % P4EST_FACES;
30880a96aa3bSJed Brown               if (nt == t && nf == f) {
30890a96aa3bSJed Brown                 PetscInt        plexF = P4estFaceToPetscFace[f];
30900a96aa3bSJed Brown                 const PetscInt *cone;
30910a96aa3bSJed Brown 
30929566063dSJacob Faibussowitsch                 PetscCall(DMPlexGetCone(plex, c, &cone));
30939566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(ghostLabel, cone[plexF], plexF + 1));
30940a96aa3bSJed Brown               }
30950a96aa3bSJed Brown             }
30960a96aa3bSJed Brown           }
30970a96aa3bSJed Brown         }
30980a96aa3bSJed Brown       }
30990a96aa3bSJed Brown     }
31003ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
31010a96aa3bSJed Brown   }
31029566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(base, 0, &cStartBase, &cEndBase));
31039566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(base, 1, &fStartBase, &fEndBase));
31049566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(base, P4EST_DIM - 1, &eStartBase, &eEndBase));
31059566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(base, 0, &vStartBase, &vEndBase));
31060a96aa3bSJed Brown 
31079566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(plex, 0, &cStart, &cEnd));
31089566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(plex, 1, &fStart, &fEnd));
31099566063dSJacob Faibussowitsch   PetscCall(DMPlexGetSimplexOrBoxCells(plex, P4EST_DIM - 1, &eStart, &eEnd));
31109566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(plex, 0, &vStart, &vEnd));
31110a96aa3bSJed Brown 
31129566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plex, &pStart, &pEnd));
31139566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(base, &pStartBase, &pEndBase));
31140a96aa3bSJed Brown   /* go through the mesh: use star to find a quadrant that borders a point.  Use the closure to determine the
31150a96aa3bSJed Brown    * orientation of the quadrant relative to that point.  Use that to relate the point to the numbering in the base
31160a96aa3bSJed Brown    * mesh, and extract a label value (since the base mesh is redundantly distributed, can be found locally). */
31170a96aa3bSJed Brown   while (next) {
31180a96aa3bSJed Brown     DMLabel     baseLabel;
31190a96aa3bSJed Brown     DMLabel     label = next->label;
31200a96aa3bSJed Brown     PetscBool   isDepth, isCellType, isGhost, isVTK, isSpmap;
31210a96aa3bSJed Brown     const char *name;
31220a96aa3bSJed Brown 
31239566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)label, &name));
31249566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "depth", &isDepth));
31250a96aa3bSJed Brown     if (isDepth) {
31260a96aa3bSJed Brown       next = next->next;
31270a96aa3bSJed Brown       continue;
31280a96aa3bSJed Brown     }
31299566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "celltype", &isCellType));
31300a96aa3bSJed Brown     if (isCellType) {
31310a96aa3bSJed Brown       next = next->next;
31320a96aa3bSJed Brown       continue;
31330a96aa3bSJed Brown     }
31349566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "ghost", &isGhost));
31350a96aa3bSJed Brown     if (isGhost) {
31360a96aa3bSJed Brown       next = next->next;
31370a96aa3bSJed Brown       continue;
31380a96aa3bSJed Brown     }
31399566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "vtk", &isVTK));
31400a96aa3bSJed Brown     if (isVTK) {
31410a96aa3bSJed Brown       next = next->next;
31420a96aa3bSJed Brown       continue;
31430a96aa3bSJed Brown     }
31449566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "_forest_base_subpoint_map", &isSpmap));
31450a96aa3bSJed Brown     if (!isSpmap) {
31469566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(base, name, &baseLabel));
31470a96aa3bSJed Brown       if (!baseLabel) {
31480a96aa3bSJed Brown         next = next->next;
31490a96aa3bSJed Brown         continue;
31500a96aa3bSJed Brown       }
31519566063dSJacob Faibussowitsch       PetscCall(DMLabelCreateIndex(baseLabel, pStartBase, pEndBase));
31520a96aa3bSJed Brown     } else baseLabel = NULL;
31530a96aa3bSJed Brown 
31540a96aa3bSJed Brown     for (p = pStart; p < pEnd; p++) {
31550a96aa3bSJed Brown       PetscInt          s, c = -1, l;
31560a96aa3bSJed Brown       PetscInt         *closure = NULL, closureSize;
31570a96aa3bSJed Brown       p4est_quadrant_t *ghosts  = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
31580a96aa3bSJed Brown       p4est_tree_t     *trees   = (p4est_tree_t *)pforest->forest->trees->array;
31590a96aa3bSJed Brown       p4est_quadrant_t *q;
31600a96aa3bSJed Brown       PetscInt          t, val;
31610a96aa3bSJed Brown       PetscBool         zerosupportpoint = PETSC_FALSE;
31620a96aa3bSJed Brown 
31639566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(plex, p, PETSC_FALSE, &starSize, &star));
31640a96aa3bSJed Brown       for (s = 0; s < starSize; s++) {
31650a96aa3bSJed Brown         PetscInt point = star[2 * s];
31660a96aa3bSJed Brown 
31670a96aa3bSJed Brown         if (cStart <= point && point < cEnd) {
31689566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTransitiveClosure(plex, point, PETSC_TRUE, &closureSize, &closure));
31690a96aa3bSJed Brown           for (l = 0; l < closureSize; l++) {
31700a96aa3bSJed Brown             PetscInt qParent = closure[2 * l], q, pp = p, pParent = p;
31710a96aa3bSJed Brown             do { /* check parents of q */
31720a96aa3bSJed Brown               q = qParent;
31730a96aa3bSJed Brown               if (q == p) {
31740a96aa3bSJed Brown                 c = point;
31750a96aa3bSJed Brown                 break;
31760a96aa3bSJed Brown               }
31779566063dSJacob Faibussowitsch               PetscCall(DMPlexGetTreeParent(plex, q, &qParent, NULL));
31780a96aa3bSJed Brown             } while (qParent != q);
31790a96aa3bSJed Brown             if (c != -1) break;
31809566063dSJacob Faibussowitsch             PetscCall(DMPlexGetTreeParent(plex, pp, &pParent, NULL));
31810a96aa3bSJed Brown             q = closure[2 * l];
31820a96aa3bSJed Brown             while (pParent != pp) { /* check parents of p */
31830a96aa3bSJed Brown               pp = pParent;
31840a96aa3bSJed Brown               if (pp == q) {
31850a96aa3bSJed Brown                 c = point;
31860a96aa3bSJed Brown                 break;
31870a96aa3bSJed Brown               }
31889566063dSJacob Faibussowitsch               PetscCall(DMPlexGetTreeParent(plex, pp, &pParent, NULL));
31890a96aa3bSJed Brown             }
31900a96aa3bSJed Brown             if (c != -1) break;
31910a96aa3bSJed Brown           }
31929566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreTransitiveClosure(plex, point, PETSC_TRUE, NULL, &closure));
31930a96aa3bSJed Brown           if (l < closureSize) break;
31940a96aa3bSJed Brown         } else {
31950a96aa3bSJed Brown           PetscInt supportSize;
31960a96aa3bSJed Brown 
31979566063dSJacob Faibussowitsch           PetscCall(DMPlexGetSupportSize(plex, point, &supportSize));
31980a96aa3bSJed Brown           zerosupportpoint = (PetscBool)(zerosupportpoint || !supportSize);
31990a96aa3bSJed Brown         }
32000a96aa3bSJed Brown       }
32010a96aa3bSJed Brown       if (c < 0) {
32020a96aa3bSJed Brown         const char *prefix;
32030a96aa3bSJed Brown         PetscBool   print = PETSC_FALSE;
32040a96aa3bSJed Brown 
32059566063dSJacob Faibussowitsch         PetscCall(PetscObjectGetOptionsPrefix((PetscObject)dm, &prefix));
32069566063dSJacob Faibussowitsch         PetscCall(PetscOptionsGetBool(((PetscObject)dm)->options, prefix, "-dm_forest_print_label_error", &print, NULL));
32070a96aa3bSJed Brown         if (print) {
32080a96aa3bSJed Brown           PetscInt i;
32090a96aa3bSJed Brown 
321063a3b9bcSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, "[%d] Failed to find cell with point %" PetscInt_FMT " in its closure for label %s (starSize %" PetscInt_FMT ")\n", PetscGlobalRank, p, baseLabel ? ((PetscObject)baseLabel)->name : "_forest_base_subpoint_map", starSize));
321163a3b9bcSJacob Faibussowitsch           for (i = 0; i < starSize; i++) PetscCall(PetscPrintf(PETSC_COMM_SELF, "  star[%" PetscInt_FMT "] = %" PetscInt_FMT ",%" PetscInt_FMT "\n", i, star[2 * i], star[2 * i + 1]));
32120a96aa3bSJed Brown         }
32139566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreTransitiveClosure(plex, p, PETSC_FALSE, NULL, &star));
32140a96aa3bSJed Brown         if (zerosupportpoint) continue;
32159371c9d4SSatish Balay         else
32169371c9d4SSatish Balay           SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Failed to find cell with point %" PetscInt_FMT " in its closure for label %s. Rerun with -dm_forest_print_label_error for more information", p, baseLabel ? ((PetscObject)baseLabel)->name : "_forest_base_subpoint_map");
32170a96aa3bSJed Brown       }
32189566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(plex, p, PETSC_FALSE, NULL, &star));
32190a96aa3bSJed Brown 
32200a96aa3bSJed Brown       if (c < cLocalStart) {
32210a96aa3bSJed Brown         /* get from the beginning of the ghost layer */
32220a96aa3bSJed Brown         q = &(ghosts[c]);
32230a96aa3bSJed Brown         t = (PetscInt)q->p.which_tree;
32240a96aa3bSJed Brown       } else if (c < cLocalEnd) {
32250a96aa3bSJed Brown         PetscInt lo = 0, hi = num_trees;
32260a96aa3bSJed Brown         /* get from local quadrants: have to find the right tree */
32270a96aa3bSJed Brown 
32280a96aa3bSJed Brown         c -= cLocalStart;
32290a96aa3bSJed Brown 
32300a96aa3bSJed Brown         do {
32310a96aa3bSJed Brown           p4est_tree_t *tree;
32320a96aa3bSJed Brown 
32331dca8a05SBarry Smith           PetscCheck(guess >= lo && guess < num_trees && lo < hi, PETSC_COMM_SELF, PETSC_ERR_PLIB, "failed binary search");
32340a96aa3bSJed Brown           tree = &trees[guess];
32350a96aa3bSJed Brown           if (c < tree->quadrants_offset) {
32360a96aa3bSJed Brown             hi = guess;
32370a96aa3bSJed Brown           } else if (c < tree->quadrants_offset + (PetscInt)tree->quadrants.elem_count) {
32380a96aa3bSJed Brown             q = &((p4est_quadrant_t *)tree->quadrants.array)[c - (PetscInt)tree->quadrants_offset];
32390a96aa3bSJed Brown             t = guess;
32400a96aa3bSJed Brown             break;
32410a96aa3bSJed Brown           } else {
32420a96aa3bSJed Brown             lo = guess + 1;
32430a96aa3bSJed Brown           }
32440a96aa3bSJed Brown           guess = lo + (hi - lo) / 2;
32450a96aa3bSJed Brown         } while (1);
32460a96aa3bSJed Brown       } else {
32470a96aa3bSJed Brown         /* get from the end of the ghost layer */
32480a96aa3bSJed Brown         c -= (cLocalEnd - cLocalStart);
32490a96aa3bSJed Brown 
32500a96aa3bSJed Brown         q = &(ghosts[c]);
32510a96aa3bSJed Brown         t = (PetscInt)q->p.which_tree;
32520a96aa3bSJed Brown       }
32530a96aa3bSJed Brown 
32540a96aa3bSJed Brown       if (l == 0) { /* cell */
32550a96aa3bSJed Brown         if (baseLabel) {
32569566063dSJacob Faibussowitsch           PetscCall(DMLabelGetValue(baseLabel, t + cStartBase, &val));
32570a96aa3bSJed Brown         } else {
32580a96aa3bSJed Brown           val = t + cStartBase;
32590a96aa3bSJed Brown         }
32609566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(label, p, val));
32610a96aa3bSJed Brown       } else if (l >= 1 && l < 1 + P4EST_FACES) { /* facet */
32620a96aa3bSJed Brown         p4est_quadrant_t nq;
32630a96aa3bSJed Brown         int              isInside;
32640a96aa3bSJed Brown 
32650a96aa3bSJed Brown         l = PetscFaceToP4estFace[l - 1];
3266792fecdfSBarry Smith         PetscCallP4est(p4est_quadrant_face_neighbor, (q, l, &nq));
3267792fecdfSBarry Smith         PetscCallP4estReturn(isInside, p4est_quadrant_is_inside_root, (&nq));
32680a96aa3bSJed Brown         if (isInside) {
32690a96aa3bSJed Brown           /* this facet is in the interior of a tree, so it inherits the label of the tree */
32700a96aa3bSJed Brown           if (baseLabel) {
32719566063dSJacob Faibussowitsch             PetscCall(DMLabelGetValue(baseLabel, t + cStartBase, &val));
32720a96aa3bSJed Brown           } else {
32730a96aa3bSJed Brown             val = t + cStartBase;
32740a96aa3bSJed Brown           }
32759566063dSJacob Faibussowitsch           PetscCall(DMLabelSetValue(label, p, val));
32760a96aa3bSJed Brown         } else {
32770a96aa3bSJed Brown           PetscInt f = pforest->topo->tree_face_to_uniq[P4EST_FACES * t + l];
32780a96aa3bSJed Brown 
32790a96aa3bSJed Brown           if (baseLabel) {
32809566063dSJacob Faibussowitsch             PetscCall(DMLabelGetValue(baseLabel, f + fStartBase, &val));
32810a96aa3bSJed Brown           } else {
32820a96aa3bSJed Brown             val = f + fStartBase;
32830a96aa3bSJed Brown           }
32849566063dSJacob Faibussowitsch           PetscCall(DMLabelSetValue(label, p, val));
32850a96aa3bSJed Brown         }
32860a96aa3bSJed Brown   #if defined(P4_TO_P8)
32870a96aa3bSJed Brown       } else if (l >= 1 + P4EST_FACES && l < 1 + P4EST_FACES + P8EST_EDGES) { /* edge */
32880a96aa3bSJed Brown         p4est_quadrant_t nq;
32890a96aa3bSJed Brown         int              isInside;
32900a96aa3bSJed Brown 
32910a96aa3bSJed Brown         l = PetscEdgeToP4estEdge[l - (1 + P4EST_FACES)];
3292792fecdfSBarry Smith         PetscCallP4est(p8est_quadrant_edge_neighbor, (q, l, &nq));
3293792fecdfSBarry Smith         PetscCallP4estReturn(isInside, p4est_quadrant_is_inside_root, (&nq));
32940a96aa3bSJed Brown         if (isInside) {
32950a96aa3bSJed Brown           /* this edge is in the interior of a tree, so it inherits the label of the tree */
32960a96aa3bSJed Brown           if (baseLabel) {
32979566063dSJacob Faibussowitsch             PetscCall(DMLabelGetValue(baseLabel, t + cStartBase, &val));
32980a96aa3bSJed Brown           } else {
32990a96aa3bSJed Brown             val = t + cStartBase;
33000a96aa3bSJed Brown           }
33019566063dSJacob Faibussowitsch           PetscCall(DMLabelSetValue(label, p, val));
33020a96aa3bSJed Brown         } else {
33030a96aa3bSJed Brown           int isOutsideFace;
33040a96aa3bSJed Brown 
3305792fecdfSBarry Smith           PetscCallP4estReturn(isOutsideFace, p4est_quadrant_is_outside_face, (&nq));
33060a96aa3bSJed Brown           if (isOutsideFace) {
33070a96aa3bSJed Brown             PetscInt f;
33080a96aa3bSJed Brown 
33090a96aa3bSJed Brown             if (nq.x < 0) {
33100a96aa3bSJed Brown               f = 0;
33110a96aa3bSJed Brown             } else if (nq.x >= P4EST_ROOT_LEN) {
33120a96aa3bSJed Brown               f = 1;
33130a96aa3bSJed Brown             } else if (nq.y < 0) {
33140a96aa3bSJed Brown               f = 2;
33150a96aa3bSJed Brown             } else if (nq.y >= P4EST_ROOT_LEN) {
33160a96aa3bSJed Brown               f = 3;
33170a96aa3bSJed Brown             } else if (nq.z < 0) {
33180a96aa3bSJed Brown               f = 4;
33190a96aa3bSJed Brown             } else {
33200a96aa3bSJed Brown               f = 5;
33210a96aa3bSJed Brown             }
33220a96aa3bSJed Brown             f = pforest->topo->tree_face_to_uniq[P4EST_FACES * t + f];
33230a96aa3bSJed Brown             if (baseLabel) {
33249566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(baseLabel, f + fStartBase, &val));
33250a96aa3bSJed Brown             } else {
33260a96aa3bSJed Brown               val = f + fStartBase;
33270a96aa3bSJed Brown             }
33289566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(label, p, val));
33290a96aa3bSJed Brown           } else { /* the quadrant edge corresponds to the tree edge */
33300a96aa3bSJed Brown             PetscInt e = pforest->topo->conn->tree_to_edge[P8EST_EDGES * t + l];
33310a96aa3bSJed Brown 
33320a96aa3bSJed Brown             if (baseLabel) {
33339566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(baseLabel, e + eStartBase, &val));
33340a96aa3bSJed Brown             } else {
33350a96aa3bSJed Brown               val = e + eStartBase;
33360a96aa3bSJed Brown             }
33379566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(label, p, val));
33380a96aa3bSJed Brown           }
33390a96aa3bSJed Brown         }
33400a96aa3bSJed Brown   #endif
33410a96aa3bSJed Brown       } else { /* vertex */
33420a96aa3bSJed Brown         p4est_quadrant_t nq;
33430a96aa3bSJed Brown         int              isInside;
33440a96aa3bSJed Brown 
33450a96aa3bSJed Brown   #if defined(P4_TO_P8)
33460a96aa3bSJed Brown         l = PetscVertToP4estVert[l - (1 + P4EST_FACES + P8EST_EDGES)];
33470a96aa3bSJed Brown   #else
33480a96aa3bSJed Brown         l = PetscVertToP4estVert[l - (1 + P4EST_FACES)];
33490a96aa3bSJed Brown   #endif
3350792fecdfSBarry Smith         PetscCallP4est(p4est_quadrant_corner_neighbor, (q, l, &nq));
3351792fecdfSBarry Smith         PetscCallP4estReturn(isInside, p4est_quadrant_is_inside_root, (&nq));
33520a96aa3bSJed Brown         if (isInside) {
33530a96aa3bSJed Brown           if (baseLabel) {
33549566063dSJacob Faibussowitsch             PetscCall(DMLabelGetValue(baseLabel, t + cStartBase, &val));
33550a96aa3bSJed Brown           } else {
33560a96aa3bSJed Brown             val = t + cStartBase;
33570a96aa3bSJed Brown           }
33589566063dSJacob Faibussowitsch           PetscCall(DMLabelSetValue(label, p, val));
33590a96aa3bSJed Brown         } else {
33600a96aa3bSJed Brown           int isOutside;
33610a96aa3bSJed Brown 
3362792fecdfSBarry Smith           PetscCallP4estReturn(isOutside, p4est_quadrant_is_outside_face, (&nq));
33630a96aa3bSJed Brown           if (isOutside) {
33640a96aa3bSJed Brown             PetscInt f = -1;
33650a96aa3bSJed Brown 
33660a96aa3bSJed Brown             if (nq.x < 0) {
33670a96aa3bSJed Brown               f = 0;
33680a96aa3bSJed Brown             } else if (nq.x >= P4EST_ROOT_LEN) {
33690a96aa3bSJed Brown               f = 1;
33700a96aa3bSJed Brown             } else if (nq.y < 0) {
33710a96aa3bSJed Brown               f = 2;
33720a96aa3bSJed Brown             } else if (nq.y >= P4EST_ROOT_LEN) {
33730a96aa3bSJed Brown               f = 3;
33740a96aa3bSJed Brown   #if defined(P4_TO_P8)
33750a96aa3bSJed Brown             } else if (nq.z < 0) {
33760a96aa3bSJed Brown               f = 4;
33770a96aa3bSJed Brown             } else {
33780a96aa3bSJed Brown               f = 5;
33790a96aa3bSJed Brown   #endif
33800a96aa3bSJed Brown             }
33810a96aa3bSJed Brown             f = pforest->topo->tree_face_to_uniq[P4EST_FACES * t + f];
33820a96aa3bSJed Brown             if (baseLabel) {
33839566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(baseLabel, f + fStartBase, &val));
33840a96aa3bSJed Brown             } else {
33850a96aa3bSJed Brown               val = f + fStartBase;
33860a96aa3bSJed Brown             }
33879566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(label, p, val));
33880a96aa3bSJed Brown             continue;
33890a96aa3bSJed Brown           }
33900a96aa3bSJed Brown   #if defined(P4_TO_P8)
3391792fecdfSBarry Smith           PetscCallP4estReturn(isOutside, p8est_quadrant_is_outside_edge, (&nq));
33920a96aa3bSJed Brown           if (isOutside) {
33930a96aa3bSJed Brown             /* outside edge */
33940a96aa3bSJed Brown             PetscInt e = -1;
33950a96aa3bSJed Brown 
33960a96aa3bSJed Brown             if (nq.x >= 0 && nq.x < P4EST_ROOT_LEN) {
33970a96aa3bSJed Brown               if (nq.z < 0) {
33980a96aa3bSJed Brown                 if (nq.y < 0) {
33990a96aa3bSJed Brown                   e = 0;
34000a96aa3bSJed Brown                 } else {
34010a96aa3bSJed Brown                   e = 1;
34020a96aa3bSJed Brown                 }
34030a96aa3bSJed Brown               } else {
34040a96aa3bSJed Brown                 if (nq.y < 0) {
34050a96aa3bSJed Brown                   e = 2;
34060a96aa3bSJed Brown                 } else {
34070a96aa3bSJed Brown                   e = 3;
34080a96aa3bSJed Brown                 }
34090a96aa3bSJed Brown               }
34100a96aa3bSJed Brown             } else if (nq.y >= 0 && nq.y < P4EST_ROOT_LEN) {
34110a96aa3bSJed Brown               if (nq.z < 0) {
34120a96aa3bSJed Brown                 if (nq.x < 0) {
34130a96aa3bSJed Brown                   e = 4;
34140a96aa3bSJed Brown                 } else {
34150a96aa3bSJed Brown                   e = 5;
34160a96aa3bSJed Brown                 }
34170a96aa3bSJed Brown               } else {
34180a96aa3bSJed Brown                 if (nq.x < 0) {
34190a96aa3bSJed Brown                   e = 6;
34200a96aa3bSJed Brown                 } else {
34210a96aa3bSJed Brown                   e = 7;
34220a96aa3bSJed Brown                 }
34230a96aa3bSJed Brown               }
34240a96aa3bSJed Brown             } else {
34250a96aa3bSJed Brown               if (nq.y < 0) {
34260a96aa3bSJed Brown                 if (nq.x < 0) {
34270a96aa3bSJed Brown                   e = 8;
34280a96aa3bSJed Brown                 } else {
34290a96aa3bSJed Brown                   e = 9;
34300a96aa3bSJed Brown                 }
34310a96aa3bSJed Brown               } else {
34320a96aa3bSJed Brown                 if (nq.x < 0) {
34330a96aa3bSJed Brown                   e = 10;
34340a96aa3bSJed Brown                 } else {
34350a96aa3bSJed Brown                   e = 11;
34360a96aa3bSJed Brown                 }
34370a96aa3bSJed Brown               }
34380a96aa3bSJed Brown             }
34390a96aa3bSJed Brown 
34400a96aa3bSJed Brown             e = pforest->topo->conn->tree_to_edge[P8EST_EDGES * t + e];
34410a96aa3bSJed Brown             if (baseLabel) {
34429566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(baseLabel, e + eStartBase, &val));
34430a96aa3bSJed Brown             } else {
34440a96aa3bSJed Brown               val = e + eStartBase;
34450a96aa3bSJed Brown             }
34469566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(label, p, val));
34470a96aa3bSJed Brown             continue;
34480a96aa3bSJed Brown           }
34490a96aa3bSJed Brown   #endif
34500a96aa3bSJed Brown           {
34510a96aa3bSJed Brown             /* outside vertex: same corner as quadrant corner */
34520a96aa3bSJed Brown             PetscInt v = pforest->topo->conn->tree_to_corner[P4EST_CHILDREN * t + l];
34530a96aa3bSJed Brown 
34540a96aa3bSJed Brown             if (baseLabel) {
34559566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(baseLabel, v + vStartBase, &val));
34560a96aa3bSJed Brown             } else {
34570a96aa3bSJed Brown               val = v + vStartBase;
34580a96aa3bSJed Brown             }
34599566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(label, p, val));
34600a96aa3bSJed Brown           }
34610a96aa3bSJed Brown         }
34620a96aa3bSJed Brown       }
34630a96aa3bSJed Brown     }
34640a96aa3bSJed Brown     next = next->next;
34650a96aa3bSJed Brown   }
34663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
34670a96aa3bSJed Brown }
34680a96aa3bSJed Brown 
3469d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestLabelsFinalize(DM dm, DM plex)
3470d71ae5a4SJacob Faibussowitsch {
34710a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
34720a96aa3bSJed Brown   DM                 adapt;
34730a96aa3bSJed Brown 
34740a96aa3bSJed Brown   PetscFunctionBegin;
34753ba16761SJacob Faibussowitsch   if (pforest->labelsFinalized) PetscFunctionReturn(PETSC_SUCCESS);
34760a96aa3bSJed Brown   pforest->labelsFinalized = PETSC_TRUE;
34779566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dm, &adapt));
34780a96aa3bSJed Brown   if (!adapt) {
34790a96aa3bSJed Brown     /* Initialize labels from the base dm */
34809566063dSJacob Faibussowitsch     PetscCall(DMPforestLabelsInitialize(dm, plex));
34810a96aa3bSJed Brown   } else {
34820a96aa3bSJed Brown     PetscInt    dofPerDim[4] = {1, 1, 1, 1};
34830a96aa3bSJed Brown     PetscSF     transferForward, transferBackward, pointSF;
34840a96aa3bSJed Brown     PetscInt    pStart, pEnd, pStartA, pEndA;
34850a96aa3bSJed Brown     PetscInt   *values, *adaptValues;
34860a96aa3bSJed Brown     DMLabelLink next = adapt->labels;
34870a96aa3bSJed Brown     DMLabel     adaptLabel;
34880a96aa3bSJed Brown     DM          adaptPlex;
34890a96aa3bSJed Brown 
34909566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdaptivityLabel(dm, &adaptLabel));
34919566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(adapt, &adaptPlex));
34929566063dSJacob Faibussowitsch     PetscCall(DMPforestGetTransferSF(adapt, dm, dofPerDim, &transferForward, &transferBackward));
34939566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(plex, &pStart, &pEnd));
34949566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(adaptPlex, &pStartA, &pEndA));
34959566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(pEnd - pStart, &values, pEndA - pStartA, &adaptValues));
34969566063dSJacob Faibussowitsch     PetscCall(DMGetPointSF(plex, &pointSF));
34970a96aa3bSJed Brown     if (PetscDefined(USE_DEBUG)) {
34980a96aa3bSJed Brown       PetscInt p;
34990a96aa3bSJed Brown       for (p = pStartA; p < pEndA; p++) adaptValues[p - pStartA] = -1;
35000a96aa3bSJed Brown       for (p = pStart; p < pEnd; p++) values[p - pStart] = -2;
35010a96aa3bSJed Brown       if (transferForward) {
35029566063dSJacob Faibussowitsch         PetscCall(PetscSFBcastBegin(transferForward, MPIU_INT, adaptValues, values, MPI_REPLACE));
35039566063dSJacob Faibussowitsch         PetscCall(PetscSFBcastEnd(transferForward, MPIU_INT, adaptValues, values, MPI_REPLACE));
35040a96aa3bSJed Brown       }
35050a96aa3bSJed Brown       if (transferBackward) {
350657168dbeSPierre Jolivet         PetscCall(PetscSFReduceBegin(transferBackward, MPIU_INT, adaptValues, values, MPI_MAX));
350757168dbeSPierre Jolivet         PetscCall(PetscSFReduceEnd(transferBackward, MPIU_INT, adaptValues, values, MPI_MAX));
35080a96aa3bSJed Brown       }
35090a96aa3bSJed Brown       for (p = pStart; p < pEnd; p++) {
35100a96aa3bSJed Brown         PetscInt q = p, parent;
35110a96aa3bSJed Brown 
35129566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTreeParent(plex, q, &parent, NULL));
35130a96aa3bSJed Brown         while (parent != q) {
35140a96aa3bSJed Brown           if (values[parent] == -2) values[parent] = values[q];
35150a96aa3bSJed Brown           q = parent;
35169566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTreeParent(plex, q, &parent, NULL));
35170a96aa3bSJed Brown         }
35180a96aa3bSJed Brown       }
351957168dbeSPierre Jolivet       PetscCall(PetscSFReduceBegin(pointSF, MPIU_INT, values, values, MPI_MAX));
352057168dbeSPierre Jolivet       PetscCall(PetscSFReduceEnd(pointSF, MPIU_INT, values, values, MPI_MAX));
35219566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(pointSF, MPIU_INT, values, values, MPI_REPLACE));
35229566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(pointSF, MPIU_INT, values, values, MPI_REPLACE));
3523ad540459SPierre Jolivet       for (p = pStart; p < pEnd; p++) PetscCheck(values[p - pStart] != -2, PETSC_COMM_SELF, PETSC_ERR_PLIB, "uncovered point %" PetscInt_FMT, p);
35240a96aa3bSJed Brown     }
35250a96aa3bSJed Brown     while (next) {
35260a96aa3bSJed Brown       DMLabel     nextLabel = next->label;
35270a96aa3bSJed Brown       const char *name;
35280a96aa3bSJed Brown       PetscBool   isDepth, isCellType, isGhost, isVTK;
35290a96aa3bSJed Brown       DMLabel     label;
35300a96aa3bSJed Brown       PetscInt    p;
35310a96aa3bSJed Brown 
35329566063dSJacob Faibussowitsch       PetscCall(PetscObjectGetName((PetscObject)nextLabel, &name));
35339566063dSJacob Faibussowitsch       PetscCall(PetscStrcmp(name, "depth", &isDepth));
35340a96aa3bSJed Brown       if (isDepth) {
35350a96aa3bSJed Brown         next = next->next;
35360a96aa3bSJed Brown         continue;
35370a96aa3bSJed Brown       }
35389566063dSJacob Faibussowitsch       PetscCall(PetscStrcmp(name, "celltype", &isCellType));
35390a96aa3bSJed Brown       if (isCellType) {
35400a96aa3bSJed Brown         next = next->next;
35410a96aa3bSJed Brown         continue;
35420a96aa3bSJed Brown       }
35439566063dSJacob Faibussowitsch       PetscCall(PetscStrcmp(name, "ghost", &isGhost));
35440a96aa3bSJed Brown       if (isGhost) {
35450a96aa3bSJed Brown         next = next->next;
35460a96aa3bSJed Brown         continue;
35470a96aa3bSJed Brown       }
35489566063dSJacob Faibussowitsch       PetscCall(PetscStrcmp(name, "vtk", &isVTK));
35490a96aa3bSJed Brown       if (isVTK) {
35500a96aa3bSJed Brown         next = next->next;
35510a96aa3bSJed Brown         continue;
35520a96aa3bSJed Brown       }
35530a96aa3bSJed Brown       if (nextLabel == adaptLabel) {
35540a96aa3bSJed Brown         next = next->next;
35550a96aa3bSJed Brown         continue;
35560a96aa3bSJed Brown       }
35570a96aa3bSJed Brown       /* label was created earlier */
35589566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm, name, &label));
355948a46eb9SPierre Jolivet       for (p = pStartA; p < pEndA; p++) PetscCall(DMLabelGetValue(nextLabel, p, &adaptValues[p]));
35600a96aa3bSJed Brown       for (p = pStart; p < pEnd; p++) values[p] = PETSC_MIN_INT;
35610a96aa3bSJed Brown 
35621baa6e33SBarry Smith       if (transferForward) PetscCall(PetscSFBcastBegin(transferForward, MPIU_INT, adaptValues, values, MPI_REPLACE));
356357168dbeSPierre Jolivet       if (transferBackward) PetscCall(PetscSFReduceBegin(transferBackward, MPIU_INT, adaptValues, values, MPI_MAX));
35641baa6e33SBarry Smith       if (transferForward) PetscCall(PetscSFBcastEnd(transferForward, MPIU_INT, adaptValues, values, MPI_REPLACE));
356557168dbeSPierre Jolivet       if (transferBackward) PetscCall(PetscSFReduceEnd(transferBackward, MPIU_INT, adaptValues, values, MPI_MAX));
35660a96aa3bSJed Brown       for (p = pStart; p < pEnd; p++) {
35670a96aa3bSJed Brown         PetscInt q = p, parent;
35680a96aa3bSJed Brown 
35699566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTreeParent(plex, q, &parent, NULL));
35700a96aa3bSJed Brown         while (parent != q) {
35710a96aa3bSJed Brown           if (values[parent] == PETSC_MIN_INT) values[parent] = values[q];
35720a96aa3bSJed Brown           q = parent;
35739566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTreeParent(plex, q, &parent, NULL));
35740a96aa3bSJed Brown         }
35750a96aa3bSJed Brown       }
357657168dbeSPierre Jolivet       PetscCall(PetscSFReduceBegin(pointSF, MPIU_INT, values, values, MPI_MAX));
357757168dbeSPierre Jolivet       PetscCall(PetscSFReduceEnd(pointSF, MPIU_INT, values, values, MPI_MAX));
35789566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(pointSF, MPIU_INT, values, values, MPI_REPLACE));
35799566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(pointSF, MPIU_INT, values, values, MPI_REPLACE));
35800a96aa3bSJed Brown 
358148a46eb9SPierre Jolivet       for (p = pStart; p < pEnd; p++) PetscCall(DMLabelSetValue(label, p, values[p]));
35820a96aa3bSJed Brown       next = next->next;
35830a96aa3bSJed Brown     }
35849566063dSJacob Faibussowitsch     PetscCall(PetscFree2(values, adaptValues));
35859566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&transferForward));
35869566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&transferBackward));
35870a96aa3bSJed Brown     pforest->labelsFinalized = PETSC_TRUE;
35880a96aa3bSJed Brown   }
35893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
35900a96aa3bSJed Brown }
35910a96aa3bSJed Brown 
3592d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestMapCoordinates_Cell(DM plex, p4est_geometry_t *geom, PetscInt cell, p4est_quadrant_t *q, p4est_topidx_t t, p4est_connectivity_t *conn, PetscScalar *coords)
3593d71ae5a4SJacob Faibussowitsch {
35940a96aa3bSJed Brown   PetscInt     closureSize, c, coordStart, coordEnd, coordDim;
35950a96aa3bSJed Brown   PetscInt    *closure = NULL;
35960a96aa3bSJed Brown   PetscSection coordSec;
35970a96aa3bSJed Brown 
35980a96aa3bSJed Brown   PetscFunctionBegin;
35999566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(plex, &coordSec));
36009566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetChart(coordSec, &coordStart, &coordEnd));
36019566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(plex, &coordDim));
36029566063dSJacob Faibussowitsch   PetscCall(DMPlexGetTransitiveClosure(plex, cell, PETSC_TRUE, &closureSize, &closure));
36030a96aa3bSJed Brown   for (c = 0; c < closureSize; c++) {
36040a96aa3bSJed Brown     PetscInt point = closure[2 * c];
36050a96aa3bSJed Brown 
36060a96aa3bSJed Brown     if (point >= coordStart && point < coordEnd) {
36070a96aa3bSJed Brown       PetscInt dof, off;
36080a96aa3bSJed Brown       PetscInt nCoords, i;
36099566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetDof(coordSec, point, &dof));
361008401ef6SPierre Jolivet       PetscCheck(dof % coordDim == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Did not understand coordinate layout");
36110a96aa3bSJed Brown       nCoords = dof / coordDim;
36129566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSec, point, &off));
36130a96aa3bSJed Brown       for (i = 0; i < nCoords; i++) {
36140a96aa3bSJed Brown         PetscScalar *coord               = &coords[off + i * coordDim];
36150a96aa3bSJed Brown         double       coordP4est[3]       = {0.};
36160a96aa3bSJed Brown         double       coordP4estMapped[3] = {0.};
36170a96aa3bSJed Brown         PetscInt     j;
36180a96aa3bSJed Brown         PetscReal    treeCoords[P4EST_CHILDREN][3] = {{0.}};
36190a96aa3bSJed Brown         PetscReal    eta[3]                        = {0.};
36200a96aa3bSJed Brown         PetscInt     numRounds                     = 10;
36210a96aa3bSJed Brown         PetscReal    coordGuess[3]                 = {0.};
36220a96aa3bSJed Brown 
36230a96aa3bSJed Brown         eta[0] = (PetscReal)q->x / (PetscReal)P4EST_ROOT_LEN;
36240a96aa3bSJed Brown         eta[1] = (PetscReal)q->y / (PetscReal)P4EST_ROOT_LEN;
36250a96aa3bSJed Brown   #if defined(P4_TO_P8)
36260a96aa3bSJed Brown         eta[2] = (PetscReal)q->z / (PetscReal)P4EST_ROOT_LEN;
36270a96aa3bSJed Brown   #endif
36280a96aa3bSJed Brown 
36290a96aa3bSJed Brown         for (j = 0; j < P4EST_CHILDREN; j++) {
36300a96aa3bSJed Brown           PetscInt k;
36310a96aa3bSJed Brown 
36320a96aa3bSJed Brown           for (k = 0; k < 3; k++) treeCoords[j][k] = conn->vertices[3 * conn->tree_to_vertex[P4EST_CHILDREN * t + j] + k];
36330a96aa3bSJed Brown         }
36340a96aa3bSJed Brown 
36350a96aa3bSJed Brown         for (j = 0; j < P4EST_CHILDREN; j++) {
36360a96aa3bSJed Brown           PetscInt  k;
36370a96aa3bSJed Brown           PetscReal prod = 1.;
36380a96aa3bSJed Brown 
36390a96aa3bSJed Brown           for (k = 0; k < P4EST_DIM; k++) prod *= (j & (1 << k)) ? eta[k] : (1. - eta[k]);
36400a96aa3bSJed Brown           for (k = 0; k < 3; k++) coordGuess[k] += prod * treeCoords[j][k];
36410a96aa3bSJed Brown         }
36420a96aa3bSJed Brown 
36430a96aa3bSJed Brown         for (j = 0; j < numRounds; j++) {
36440a96aa3bSJed Brown           PetscInt dir;
36450a96aa3bSJed Brown 
36460a96aa3bSJed Brown           for (dir = 0; dir < P4EST_DIM; dir++) {
36470a96aa3bSJed Brown             PetscInt  k;
36480a96aa3bSJed Brown             PetscReal diff[3];
36490a96aa3bSJed Brown             PetscReal dXdeta[3] = {0.};
36500a96aa3bSJed Brown             PetscReal rhs, scale, update;
36510a96aa3bSJed Brown 
36520a96aa3bSJed Brown             for (k = 0; k < 3; k++) diff[k] = coordP4est[k] - coordGuess[k];
36530a96aa3bSJed Brown             for (k = 0; k < P4EST_CHILDREN; k++) {
36540a96aa3bSJed Brown               PetscInt  l;
36550a96aa3bSJed Brown               PetscReal prod = 1.;
36560a96aa3bSJed Brown 
36570a96aa3bSJed Brown               for (l = 0; l < P4EST_DIM; l++) {
36580a96aa3bSJed Brown                 if (l == dir) {
36590a96aa3bSJed Brown                   prod *= (k & (1 << l)) ? 1. : -1.;
36600a96aa3bSJed Brown                 } else {
36610a96aa3bSJed Brown                   prod *= (k & (1 << l)) ? eta[l] : (1. - eta[l]);
36620a96aa3bSJed Brown                 }
36630a96aa3bSJed Brown               }
36640a96aa3bSJed Brown               for (l = 0; l < 3; l++) dXdeta[l] += prod * treeCoords[k][l];
36650a96aa3bSJed Brown             }
36660a96aa3bSJed Brown             rhs   = 0.;
36670a96aa3bSJed Brown             scale = 0;
36680a96aa3bSJed Brown             for (k = 0; k < 3; k++) {
36690a96aa3bSJed Brown               rhs += diff[k] * dXdeta[k];
36700a96aa3bSJed Brown               scale += dXdeta[k] * dXdeta[k];
36710a96aa3bSJed Brown             }
36720a96aa3bSJed Brown             update = rhs / scale;
36730a96aa3bSJed Brown             eta[dir] += update;
36740a96aa3bSJed Brown             eta[dir] = PetscMin(eta[dir], 1.);
36750a96aa3bSJed Brown             eta[dir] = PetscMax(eta[dir], 0.);
36760a96aa3bSJed Brown 
36770a96aa3bSJed Brown             coordGuess[0] = coordGuess[1] = coordGuess[2] = 0.;
36780a96aa3bSJed Brown             for (k = 0; k < P4EST_CHILDREN; k++) {
36790a96aa3bSJed Brown               PetscInt  l;
36800a96aa3bSJed Brown               PetscReal prod = 1.;
36810a96aa3bSJed Brown 
36820a96aa3bSJed Brown               for (l = 0; l < P4EST_DIM; l++) prod *= (k & (1 << l)) ? eta[l] : (1. - eta[l]);
36830a96aa3bSJed Brown               for (l = 0; l < 3; l++) coordGuess[l] += prod * treeCoords[k][l];
36840a96aa3bSJed Brown             }
36850a96aa3bSJed Brown           }
36860a96aa3bSJed Brown         }
36870a96aa3bSJed Brown         for (j = 0; j < 3; j++) coordP4est[j] = (double)eta[j];
36880a96aa3bSJed Brown 
36890a96aa3bSJed Brown         if (geom) {
36900a96aa3bSJed Brown           (geom->X)(geom, t, coordP4est, coordP4estMapped);
36910a96aa3bSJed Brown           for (j = 0; j < coordDim; j++) coord[j] = (PetscScalar)coordP4estMapped[j];
36920a96aa3bSJed Brown         } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not coded");
36930a96aa3bSJed Brown       }
36940a96aa3bSJed Brown     }
36950a96aa3bSJed Brown   }
36969566063dSJacob Faibussowitsch   PetscCall(DMPlexRestoreTransitiveClosure(plex, cell, PETSC_TRUE, &closureSize, &closure));
36973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
36980a96aa3bSJed Brown }
36990a96aa3bSJed Brown 
3700d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestMapCoordinates(DM dm, DM plex)
3701d71ae5a4SJacob Faibussowitsch {
37020a96aa3bSJed Brown   DM_Forest         *forest;
37030a96aa3bSJed Brown   DM_Forest_pforest *pforest;
37040a96aa3bSJed Brown   p4est_geometry_t  *geom;
37050a96aa3bSJed Brown   PetscInt           cLocalStart, cLocalEnd;
37060a96aa3bSJed Brown   Vec                coordLocalVec;
37070a96aa3bSJed Brown   PetscScalar       *coords;
37080a96aa3bSJed Brown   p4est_topidx_t     flt, llt, t;
37090a96aa3bSJed Brown   p4est_tree_t      *trees;
37100a96aa3bSJed Brown   PetscErrorCode (*map)(DM, PetscInt, PetscInt, const PetscReal[], PetscReal[], void *);
37110a96aa3bSJed Brown   void *mapCtx;
37120a96aa3bSJed Brown 
37130a96aa3bSJed Brown   PetscFunctionBegin;
37140a96aa3bSJed Brown   forest  = (DM_Forest *)dm->data;
37150a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)forest->data;
37160a96aa3bSJed Brown   geom    = pforest->topo->geom;
37179566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseCoordinateMapping(dm, &map, &mapCtx));
37183ba16761SJacob Faibussowitsch   if (!geom && !map) PetscFunctionReturn(PETSC_SUCCESS);
37199566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(plex, &coordLocalVec));
37209566063dSJacob Faibussowitsch   PetscCall(VecGetArray(coordLocalVec, &coords));
37210a96aa3bSJed Brown   cLocalStart = pforest->cLocalStart;
37220a96aa3bSJed Brown   cLocalEnd   = pforest->cLocalEnd;
37230a96aa3bSJed Brown   flt         = pforest->forest->first_local_tree;
37240a96aa3bSJed Brown   llt         = pforest->forest->last_local_tree;
37250a96aa3bSJed Brown   trees       = (p4est_tree_t *)pforest->forest->trees->array;
37260a96aa3bSJed Brown   if (map) { /* apply the map directly to the existing coordinates */
37270a96aa3bSJed Brown     PetscSection coordSec;
37280a96aa3bSJed Brown     PetscInt     coordStart, coordEnd, p, coordDim, p4estCoordDim, cStart, cEnd, cEndInterior;
37290a96aa3bSJed Brown     DM           base;
37300a96aa3bSJed Brown 
37319566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(plex, 0, &cStart, &cEnd));
3732*2827ebadSStefano Zampini     PetscCall(DMPlexGetCellTypeStratum(plex, DM_POLYTOPE_FV_GHOST, &cEndInterior, NULL));
37330a96aa3bSJed Brown     cEnd = cEndInterior < 0 ? cEnd : cEndInterior;
37349566063dSJacob Faibussowitsch     PetscCall(DMForestGetBaseDM(dm, &base));
37359566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateSection(plex, &coordSec));
37369566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetChart(coordSec, &coordStart, &coordEnd));
37379566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDim(plex, &coordDim));
37380a96aa3bSJed Brown     p4estCoordDim = PetscMin(coordDim, 3);
37390a96aa3bSJed Brown     for (p = coordStart; p < coordEnd; p++) {
37400a96aa3bSJed Brown       PetscInt *star = NULL, starSize;
37410a96aa3bSJed Brown       PetscInt  dof, off, cell = -1, coarsePoint = -1;
37420a96aa3bSJed Brown       PetscInt  nCoords, i;
37439566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetDof(coordSec, p, &dof));
374408401ef6SPierre Jolivet       PetscCheck(dof % coordDim == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Did not understand coordinate layout");
37450a96aa3bSJed Brown       nCoords = dof / coordDim;
37469566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetOffset(coordSec, p, &off));
37479566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(plex, p, PETSC_FALSE, &starSize, &star));
37480a96aa3bSJed Brown       for (i = 0; i < starSize; i++) {
37490a96aa3bSJed Brown         PetscInt point = star[2 * i];
37500a96aa3bSJed Brown 
37510a96aa3bSJed Brown         if (cStart <= point && point < cEnd) {
37520a96aa3bSJed Brown           cell = point;
37530a96aa3bSJed Brown           break;
37540a96aa3bSJed Brown         }
37550a96aa3bSJed Brown       }
37569566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(plex, p, PETSC_FALSE, &starSize, &star));
37570a96aa3bSJed Brown       if (cell >= 0) {
37580a96aa3bSJed Brown         if (cell < cLocalStart) {
37590a96aa3bSJed Brown           p4est_quadrant_t *ghosts = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
37600a96aa3bSJed Brown 
37610a96aa3bSJed Brown           coarsePoint = ghosts[cell].p.which_tree;
37620a96aa3bSJed Brown         } else if (cell < cLocalEnd) {
37630a96aa3bSJed Brown           cell -= cLocalStart;
37640a96aa3bSJed Brown           for (t = flt; t <= llt; t++) {
37650a96aa3bSJed Brown             p4est_tree_t *tree = &(trees[t]);
37660a96aa3bSJed Brown 
37670a96aa3bSJed Brown             if (cell >= tree->quadrants_offset && (size_t)cell < tree->quadrants_offset + tree->quadrants.elem_count) {
37680a96aa3bSJed Brown               coarsePoint = t;
37690a96aa3bSJed Brown               break;
37700a96aa3bSJed Brown             }
37710a96aa3bSJed Brown           }
37720a96aa3bSJed Brown         } else {
37730a96aa3bSJed Brown           p4est_quadrant_t *ghosts = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
37740a96aa3bSJed Brown 
37750a96aa3bSJed Brown           coarsePoint = ghosts[cell - cLocalEnd].p.which_tree;
37760a96aa3bSJed Brown         }
37770a96aa3bSJed Brown       }
37780a96aa3bSJed Brown       for (i = 0; i < nCoords; i++) {
37790a96aa3bSJed Brown         PetscScalar *coord               = &coords[off + i * coordDim];
37800a96aa3bSJed Brown         PetscReal    coordP4est[3]       = {0.};
37810a96aa3bSJed Brown         PetscReal    coordP4estMapped[3] = {0.};
37820a96aa3bSJed Brown         PetscInt     j;
37830a96aa3bSJed Brown 
37840a96aa3bSJed Brown         for (j = 0; j < p4estCoordDim; j++) coordP4est[j] = PetscRealPart(coord[j]);
37859566063dSJacob Faibussowitsch         PetscCall((map)(base, coarsePoint, p4estCoordDim, coordP4est, coordP4estMapped, mapCtx));
37860a96aa3bSJed Brown         for (j = 0; j < p4estCoordDim; j++) coord[j] = (PetscScalar)coordP4estMapped[j];
37870a96aa3bSJed Brown       }
37880a96aa3bSJed Brown     }
37890a96aa3bSJed Brown   } else { /* we have to transform coordinates back to the unit cube (where geom is defined), and then apply geom */
37900a96aa3bSJed Brown     PetscInt cStart, cEnd, cEndInterior;
37910a96aa3bSJed Brown 
37929566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(plex, 0, &cStart, &cEnd));
3793*2827ebadSStefano Zampini     PetscCall(DMPlexGetCellTypeStratum(plex, DM_POLYTOPE_FV_GHOST, &cEndInterior, NULL));
37940a96aa3bSJed Brown     cEnd = cEndInterior < 0 ? cEnd : cEndInterior;
37950a96aa3bSJed Brown     if (cLocalStart > 0) {
37960a96aa3bSJed Brown       p4est_quadrant_t *ghosts = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
37970a96aa3bSJed Brown       PetscInt          count;
37980a96aa3bSJed Brown 
37990a96aa3bSJed Brown       for (count = 0; count < cLocalStart; count++) {
38000a96aa3bSJed Brown         p4est_quadrant_t *quad = &ghosts[count];
38010a96aa3bSJed Brown         p4est_topidx_t    t    = quad->p.which_tree;
38020a96aa3bSJed Brown 
38039566063dSJacob Faibussowitsch         PetscCall(DMPforestMapCoordinates_Cell(plex, geom, count, quad, t, pforest->topo->conn, coords));
38040a96aa3bSJed Brown       }
38050a96aa3bSJed Brown     }
38060a96aa3bSJed Brown     for (t = flt; t <= llt; t++) {
38070a96aa3bSJed Brown       p4est_tree_t     *tree     = &(trees[t]);
38080a96aa3bSJed Brown       PetscInt          offset   = cLocalStart + tree->quadrants_offset, i;
38090a96aa3bSJed Brown       PetscInt          numQuads = (PetscInt)tree->quadrants.elem_count;
38100a96aa3bSJed Brown       p4est_quadrant_t *quads    = (p4est_quadrant_t *)tree->quadrants.array;
38110a96aa3bSJed Brown 
38120a96aa3bSJed Brown       for (i = 0; i < numQuads; i++) {
38130a96aa3bSJed Brown         PetscInt count = i + offset;
38140a96aa3bSJed Brown 
38159566063dSJacob Faibussowitsch         PetscCall(DMPforestMapCoordinates_Cell(plex, geom, count, &quads[i], t, pforest->topo->conn, coords));
38160a96aa3bSJed Brown       }
38170a96aa3bSJed Brown     }
38180a96aa3bSJed Brown     if (cLocalEnd - cLocalStart < cEnd - cStart) {
38190a96aa3bSJed Brown       p4est_quadrant_t *ghosts    = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
38200a96aa3bSJed Brown       PetscInt          numGhosts = (PetscInt)pforest->ghost->ghosts.elem_count;
38210a96aa3bSJed Brown       PetscInt          count;
38220a96aa3bSJed Brown 
38230a96aa3bSJed Brown       for (count = 0; count < numGhosts - cLocalStart; count++) {
38240a96aa3bSJed Brown         p4est_quadrant_t *quad = &ghosts[count + cLocalStart];
38250a96aa3bSJed Brown         p4est_topidx_t    t    = quad->p.which_tree;
38260a96aa3bSJed Brown 
38279566063dSJacob Faibussowitsch         PetscCall(DMPforestMapCoordinates_Cell(plex, geom, count + cLocalEnd, quad, t, pforest->topo->conn, coords));
38280a96aa3bSJed Brown       }
38290a96aa3bSJed Brown     }
38300a96aa3bSJed Brown   }
38319566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(coordLocalVec, &coords));
38323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
38330a96aa3bSJed Brown }
38340a96aa3bSJed Brown 
3835d71ae5a4SJacob Faibussowitsch static PetscErrorCode PforestQuadrantIsInterior(p4est_quadrant_t *quad, PetscBool *is_interior)
3836d71ae5a4SJacob Faibussowitsch {
3837852b71a7SToby Isaac   PetscFunctionBegin;
3838852b71a7SToby Isaac   p4est_qcoord_t h = P4EST_QUADRANT_LEN(quad->level);
3839852b71a7SToby Isaac   if ((quad->x > 0) && (quad->x + h < P4EST_ROOT_LEN)
3840852b71a7SToby Isaac   #ifdef P4_TO_P8
3841852b71a7SToby Isaac       && (quad->z > 0) && (quad->z + h < P4EST_ROOT_LEN)
3842852b71a7SToby Isaac   #endif
3843852b71a7SToby Isaac       && (quad->y > 0) && (quad->y + h < P4EST_ROOT_LEN)) {
3844852b71a7SToby Isaac     *is_interior = PETSC_TRUE;
3845852b71a7SToby Isaac   } else {
3846852b71a7SToby Isaac     *is_interior = PETSC_FALSE;
3847852b71a7SToby Isaac   }
38483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3849852b71a7SToby Isaac }
3850852b71a7SToby Isaac 
3851852b71a7SToby Isaac /* We always use DG coordinates with p4est: if they do not match the vertex
3852852b71a7SToby Isaac    coordinates, add space for them in the section */
3853d71ae5a4SJacob Faibussowitsch static PetscErrorCode PforestCheckLocalizeCell(DM plex, PetscInt cDim, Vec cVecOld, DM_Forest_pforest *pforest, PetscSection oldSection, PetscSection newSection, PetscInt cell, PetscInt coarsePoint, p4est_quadrant_t *quad)
3854d71ae5a4SJacob Faibussowitsch {
3855852b71a7SToby Isaac   PetscBool is_interior;
3856852b71a7SToby Isaac 
3857852b71a7SToby Isaac   PetscFunctionBegin;
3858852b71a7SToby Isaac   PetscCall(PforestQuadrantIsInterior(quad, &is_interior));
3859852b71a7SToby Isaac   if (is_interior) { // quads in the interior of a coarse cell can't touch periodic interfaces
3860852b71a7SToby Isaac     PetscCall(PetscSectionSetDof(newSection, cell, 0));
3861852b71a7SToby Isaac     PetscCall(PetscSectionSetFieldDof(newSection, cell, 0, 0));
3862852b71a7SToby Isaac   } else {
3863852b71a7SToby Isaac     PetscInt     cSize;
3864852b71a7SToby Isaac     PetscScalar *values      = NULL;
3865852b71a7SToby Isaac     PetscBool    same_coords = PETSC_TRUE;
3866852b71a7SToby Isaac 
3867852b71a7SToby Isaac     PetscCall(DMPlexVecGetClosure(plex, oldSection, cVecOld, cell, &cSize, &values));
3868852b71a7SToby Isaac     PetscAssert(cSize == cDim * P4EST_CHILDREN, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected closure size");
3869852b71a7SToby Isaac     for (int c = 0; c < P4EST_CHILDREN; c++) {
3870852b71a7SToby Isaac       p4est_qcoord_t quad_coords[3];
3871852b71a7SToby Isaac       p4est_qcoord_t h = P4EST_QUADRANT_LEN(quad->level);
3872852b71a7SToby Isaac       double         corner_coords[3];
3873852b71a7SToby Isaac       double         vert_coords[3];
3874852b71a7SToby Isaac       PetscInt       corner = PetscVertToP4estVert[c];
3875852b71a7SToby Isaac 
3876ad540459SPierre Jolivet       for (PetscInt d = 0; d < PetscMin(cDim, 3); d++) vert_coords[d] = PetscRealPart(values[c * cDim + d]);
3877852b71a7SToby Isaac 
3878852b71a7SToby Isaac       quad_coords[0] = quad->x;
3879852b71a7SToby Isaac       quad_coords[1] = quad->y;
3880852b71a7SToby Isaac   #ifdef P4_TO_P8
3881852b71a7SToby Isaac       quad_coords[2] = quad->z;
3882852b71a7SToby Isaac   #endif
3883ad540459SPierre Jolivet       for (int d = 0; d < 3; d++) quad_coords[d] += (corner & (1 << d)) ? h : 0;
3884852b71a7SToby Isaac   #ifndef P4_TO_P8
3885648bc8c4SBarry Smith       PetscCallP4est(p4est_qcoord_to_vertex, (pforest->forest->connectivity, coarsePoint, quad_coords[0], quad_coords[1], corner_coords));
3886852b71a7SToby Isaac   #else
3887648bc8c4SBarry Smith       PetscCallP4est(p4est_qcoord_to_vertex, (pforest->forest->connectivity, coarsePoint, quad_coords[0], quad_coords[1], quad_coords[2], corner_coords));
3888852b71a7SToby Isaac   #endif
3889852b71a7SToby Isaac       for (PetscInt d = 0; d < PetscMin(cDim, 3); d++) {
3890852b71a7SToby Isaac         if (fabs(vert_coords[d] - corner_coords[d]) > PETSC_SMALL) {
3891852b71a7SToby Isaac           same_coords = PETSC_FALSE;
3892852b71a7SToby Isaac           break;
3893852b71a7SToby Isaac         }
3894852b71a7SToby Isaac       }
3895852b71a7SToby Isaac     }
3896852b71a7SToby Isaac     if (same_coords) {
3897852b71a7SToby Isaac       PetscCall(PetscSectionSetDof(newSection, cell, 0));
3898852b71a7SToby Isaac       PetscCall(PetscSectionSetFieldDof(newSection, cell, 0, 0));
3899852b71a7SToby Isaac     } else {
3900852b71a7SToby Isaac       PetscCall(PetscSectionSetDof(newSection, cell, cSize));
3901852b71a7SToby Isaac       PetscCall(PetscSectionSetFieldDof(newSection, cell, 0, cSize));
3902852b71a7SToby Isaac     }
3903852b71a7SToby Isaac     PetscCall(DMPlexVecRestoreClosure(plex, oldSection, cVecOld, cell, &cSize, &values));
3904852b71a7SToby Isaac   }
39053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3906852b71a7SToby Isaac }
3907852b71a7SToby Isaac 
3908d71ae5a4SJacob Faibussowitsch static PetscErrorCode PforestLocalizeCell(DM plex, PetscInt cDim, DM_Forest_pforest *pforest, PetscSection newSection, PetscInt cell, PetscInt coarsePoint, p4est_quadrant_t *quad, PetscScalar coords[])
3909d71ae5a4SJacob Faibussowitsch {
3910852b71a7SToby Isaac   PetscInt cdof, off;
3911852b71a7SToby Isaac 
3912852b71a7SToby Isaac   PetscFunctionBegin;
3913852b71a7SToby Isaac   PetscCall(PetscSectionGetDof(newSection, cell, &cdof));
39143ba16761SJacob Faibussowitsch   if (!cdof) PetscFunctionReturn(PETSC_SUCCESS);
3915852b71a7SToby Isaac 
3916852b71a7SToby Isaac   PetscCall(PetscSectionGetOffset(newSection, cell, &off));
3917852b71a7SToby Isaac   for (PetscInt c = 0, pos = off; c < P4EST_CHILDREN; c++) {
3918852b71a7SToby Isaac     p4est_qcoord_t quad_coords[3];
3919852b71a7SToby Isaac     p4est_qcoord_t h = P4EST_QUADRANT_LEN(quad->level);
3920852b71a7SToby Isaac     double         corner_coords[3];
3921852b71a7SToby Isaac     PetscInt       corner = PetscVertToP4estVert[c];
3922852b71a7SToby Isaac 
3923852b71a7SToby Isaac     quad_coords[0] = quad->x;
3924852b71a7SToby Isaac     quad_coords[1] = quad->y;
3925852b71a7SToby Isaac   #ifdef P4_TO_P8
3926852b71a7SToby Isaac     quad_coords[2] = quad->z;
3927852b71a7SToby Isaac   #endif
3928ad540459SPierre Jolivet     for (int d = 0; d < 3; d++) quad_coords[d] += (corner & (1 << d)) ? h : 0;
3929852b71a7SToby Isaac   #ifndef P4_TO_P8
3930648bc8c4SBarry Smith     PetscCallP4est(p4est_qcoord_to_vertex, (pforest->forest->connectivity, coarsePoint, quad_coords[0], quad_coords[1], corner_coords));
3931852b71a7SToby Isaac   #else
3932648bc8c4SBarry Smith     PetscCallP4est(p4est_qcoord_to_vertex, (pforest->forest->connectivity, coarsePoint, quad_coords[0], quad_coords[1], quad_coords[2], corner_coords));
3933852b71a7SToby Isaac   #endif
3934ad540459SPierre Jolivet     for (PetscInt d = 0; d < PetscMin(cDim, 3); d++) coords[pos++] = corner_coords[d];
3935ad540459SPierre Jolivet     for (PetscInt d = PetscMin(cDim, 3); d < cDim; d++) coords[pos++] = 0.;
3936852b71a7SToby Isaac   }
39373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3938852b71a7SToby Isaac }
3939852b71a7SToby Isaac 
3940d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestLocalizeCoordinates(DM dm, DM plex)
3941d71ae5a4SJacob Faibussowitsch {
39420a96aa3bSJed Brown   DM_Forest         *forest;
39430a96aa3bSJed Brown   DM_Forest_pforest *pforest;
39446858538eSMatthew G. Knepley   DM                 base, cdm, cdmCell;
3945852b71a7SToby Isaac   Vec                cVec, cVecOld;
3946852b71a7SToby Isaac   PetscSection       oldSection, newSection;
39470a96aa3bSJed Brown   PetscScalar       *coords2;
39486858538eSMatthew G. Knepley   const PetscReal   *L;
39490a96aa3bSJed Brown   PetscInt           cLocalStart, cLocalEnd, coarsePoint;
3950852b71a7SToby Isaac   PetscInt           cDim, newStart, newEnd;
3951852b71a7SToby Isaac   PetscInt           v, vStart, vEnd, cp, cStart, cEnd, cEndInterior;
39520a96aa3bSJed Brown   p4est_topidx_t     flt, llt, t;
39530a96aa3bSJed Brown   p4est_tree_t      *trees;
39546858538eSMatthew G. Knepley   PetscBool          baseLocalized = PETSC_FALSE;
39550a96aa3bSJed Brown 
39560a96aa3bSJed Brown   PetscFunctionBegin;
39574fb89dddSMatthew G. Knepley   PetscCall(DMGetPeriodicity(dm, NULL, NULL, &L));
39580a96aa3bSJed Brown   /* we localize on all cells if we don't have a base DM or the base DM coordinates have not been localized */
39599566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &cDim));
39609566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseDM(dm, &base));
39616858538eSMatthew G. Knepley   if (base) PetscCall(DMGetCoordinatesLocalized(base, &baseLocalized));
39620a96aa3bSJed Brown   if (!baseLocalized) base = NULL;
39633ba16761SJacob Faibussowitsch   if (!baseLocalized && !L) PetscFunctionReturn(PETSC_SUCCESS);
39649566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(plex, &newStart, &newEnd));
39650a96aa3bSJed Brown 
39669566063dSJacob Faibussowitsch   PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), &newSection));
39679566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetNumFields(newSection, 1));
39689566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetFieldComponents(newSection, 0, cDim));
39699566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetChart(newSection, newStart, newEnd));
39700a96aa3bSJed Brown 
39719566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(plex, &oldSection));
39729566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(plex, 0, &vStart, &vEnd));
3973852b71a7SToby Isaac   PetscCall(DMGetCoordinatesLocal(plex, &cVecOld));
39740a96aa3bSJed Brown 
39750a96aa3bSJed Brown   forest      = (DM_Forest *)dm->data;
39760a96aa3bSJed Brown   pforest     = (DM_Forest_pforest *)forest->data;
39770a96aa3bSJed Brown   cLocalStart = pforest->cLocalStart;
39780a96aa3bSJed Brown   cLocalEnd   = pforest->cLocalEnd;
39790a96aa3bSJed Brown   flt         = pforest->forest->first_local_tree;
39800a96aa3bSJed Brown   llt         = pforest->forest->last_local_tree;
39810a96aa3bSJed Brown   trees       = (p4est_tree_t *)pforest->forest->trees->array;
39820a96aa3bSJed Brown 
39839566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(plex, 0, &cStart, &cEnd));
3984*2827ebadSStefano Zampini   PetscCall(DMPlexGetCellTypeStratum(plex, DM_POLYTOPE_FV_GHOST, &cEndInterior, NULL));
39850a96aa3bSJed Brown   cEnd = cEndInterior < 0 ? cEnd : cEndInterior;
3986852b71a7SToby Isaac   cp   = 0;
39870a96aa3bSJed Brown   if (cLocalStart > 0) {
39880a96aa3bSJed Brown     p4est_quadrant_t *ghosts = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
3989852b71a7SToby Isaac     PetscInt          cell;
39900a96aa3bSJed Brown 
3991852b71a7SToby Isaac     for (cell = 0; cell < cLocalStart; ++cell, cp++) {
3992852b71a7SToby Isaac       p4est_quadrant_t *quad = &ghosts[cell];
3993852b71a7SToby Isaac 
39940a96aa3bSJed Brown       coarsePoint = quad->p.which_tree;
3995852b71a7SToby Isaac       PetscCall(PforestCheckLocalizeCell(plex, cDim, cVecOld, pforest, oldSection, newSection, cell, coarsePoint, quad));
39960a96aa3bSJed Brown     }
39970a96aa3bSJed Brown   }
39980a96aa3bSJed Brown   for (t = flt; t <= llt; t++) {
39990a96aa3bSJed Brown     p4est_tree_t     *tree     = &(trees[t]);
40000a96aa3bSJed Brown     PetscInt          offset   = cLocalStart + tree->quadrants_offset;
40010a96aa3bSJed Brown     PetscInt          numQuads = (PetscInt)tree->quadrants.elem_count;
4002852b71a7SToby Isaac     p4est_quadrant_t *quads    = (p4est_quadrant_t *)tree->quadrants.array;
40030a96aa3bSJed Brown     PetscInt          i;
40040a96aa3bSJed Brown 
40050a96aa3bSJed Brown     if (!numQuads) continue;
40060a96aa3bSJed Brown     coarsePoint = t;
4007852b71a7SToby Isaac     for (i = 0; i < numQuads; i++, cp++) {
4008852b71a7SToby Isaac       PetscInt          cell = i + offset;
4009852b71a7SToby Isaac       p4est_quadrant_t *quad = &quads[i];
40100a96aa3bSJed Brown 
4011852b71a7SToby Isaac       PetscCall(PforestCheckLocalizeCell(plex, cDim, cVecOld, pforest, oldSection, newSection, cell, coarsePoint, quad));
40120a96aa3bSJed Brown     }
40130a96aa3bSJed Brown   }
40140a96aa3bSJed Brown   if (cLocalEnd - cLocalStart < cEnd - cStart) {
40150a96aa3bSJed Brown     p4est_quadrant_t *ghosts    = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
40160a96aa3bSJed Brown     PetscInt          numGhosts = (PetscInt)pforest->ghost->ghosts.elem_count;
40170a96aa3bSJed Brown     PetscInt          count;
40180a96aa3bSJed Brown 
4019852b71a7SToby Isaac     for (count = 0; count < numGhosts - cLocalStart; count++, cp++) {
40200a96aa3bSJed Brown       p4est_quadrant_t *quad = &ghosts[count + cLocalStart];
40210a96aa3bSJed Brown       coarsePoint            = quad->p.which_tree;
4022852b71a7SToby Isaac       PetscInt cell          = count + cLocalEnd;
40230a96aa3bSJed Brown 
4024852b71a7SToby Isaac       PetscCall(PforestCheckLocalizeCell(plex, cDim, cVecOld, pforest, oldSection, newSection, cell, coarsePoint, quad));
40250a96aa3bSJed Brown     }
40260a96aa3bSJed Brown   }
4027852b71a7SToby Isaac   PetscAssert(cp == cEnd - cStart, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected number of fine cells %" PetscInt_FMT " != %" PetscInt_FMT, cp, cEnd - cStart);
40280a96aa3bSJed Brown 
40299566063dSJacob Faibussowitsch   PetscCall(PetscSectionSetUp(newSection));
40306858538eSMatthew G. Knepley   PetscCall(DMGetCoordinateDM(plex, &cdm));
40316858538eSMatthew G. Knepley   PetscCall(DMClone(cdm, &cdmCell));
40326858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateDM(plex, cdmCell));
40336858538eSMatthew G. Knepley   PetscCall(DMDestroy(&cdmCell));
40346858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinateSection(plex, cDim, newSection));
40359566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetStorageSize(newSection, &v));
40369566063dSJacob Faibussowitsch   PetscCall(VecCreate(PETSC_COMM_SELF, &cVec));
40379566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)cVec, "coordinates"));
40389566063dSJacob Faibussowitsch   PetscCall(VecSetBlockSize(cVec, cDim));
40399566063dSJacob Faibussowitsch   PetscCall(VecSetSizes(cVec, v, PETSC_DETERMINE));
40409566063dSJacob Faibussowitsch   PetscCall(VecSetType(cVec, VECSTANDARD));
40419566063dSJacob Faibussowitsch   PetscCall(VecSet(cVec, PETSC_MIN_REAL));
40420a96aa3bSJed Brown 
40430a96aa3bSJed Brown   /* Localize coordinates on cells if needed */
40446858538eSMatthew G. Knepley   PetscCall(VecGetArray(cVec, &coords2));
4045852b71a7SToby Isaac   cp = 0;
4046852b71a7SToby Isaac   if (cLocalStart > 0) {
4047852b71a7SToby Isaac     p4est_quadrant_t *ghosts = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
4048852b71a7SToby Isaac     PetscInt          cell;
4049852b71a7SToby Isaac 
4050852b71a7SToby Isaac     for (cell = 0; cell < cLocalStart; ++cell, cp++) {
4051852b71a7SToby Isaac       p4est_quadrant_t *quad = &ghosts[cell];
4052852b71a7SToby Isaac 
4053852b71a7SToby Isaac       coarsePoint = quad->p.which_tree;
4054852b71a7SToby Isaac       PetscCall(PforestLocalizeCell(plex, cDim, pforest, newSection, cell, coarsePoint, quad, coords2));
4055852b71a7SToby Isaac     }
4056852b71a7SToby Isaac   }
40570a96aa3bSJed Brown   for (t = flt; t <= llt; t++) {
40580a96aa3bSJed Brown     p4est_tree_t     *tree     = &(trees[t]);
40590a96aa3bSJed Brown     PetscInt          offset   = cLocalStart + tree->quadrants_offset;
40600a96aa3bSJed Brown     PetscInt          numQuads = (PetscInt)tree->quadrants.elem_count;
4061852b71a7SToby Isaac     p4est_quadrant_t *quads    = (p4est_quadrant_t *)tree->quadrants.array;
4062852b71a7SToby Isaac     PetscInt          i;
40630a96aa3bSJed Brown 
40640a96aa3bSJed Brown     if (!numQuads) continue;
4065852b71a7SToby Isaac     coarsePoint = t;
4066852b71a7SToby Isaac     for (i = 0; i < numQuads; i++, cp++) {
4067852b71a7SToby Isaac       PetscInt          cell = i + offset;
40680a96aa3bSJed Brown       p4est_quadrant_t *quad = &quads[i];
40690a96aa3bSJed Brown 
4070852b71a7SToby Isaac       PetscCall(PforestLocalizeCell(plex, cDim, pforest, newSection, cell, coarsePoint, quad, coords2));
4071852b71a7SToby Isaac     }
4072852b71a7SToby Isaac   }
4073852b71a7SToby Isaac   if (cLocalEnd - cLocalStart < cEnd - cStart) {
4074852b71a7SToby Isaac     p4est_quadrant_t *ghosts    = (p4est_quadrant_t *)pforest->ghost->ghosts.array;
4075852b71a7SToby Isaac     PetscInt          numGhosts = (PetscInt)pforest->ghost->ghosts.elem_count;
4076852b71a7SToby Isaac     PetscInt          count;
40770a96aa3bSJed Brown 
4078852b71a7SToby Isaac     for (count = 0; count < numGhosts - cLocalStart; count++, cp++) {
4079852b71a7SToby Isaac       p4est_quadrant_t *quad = &ghosts[count + cLocalStart];
4080852b71a7SToby Isaac       coarsePoint            = quad->p.which_tree;
4081852b71a7SToby Isaac       PetscInt cell          = count + cLocalEnd;
40820a96aa3bSJed Brown 
4083852b71a7SToby Isaac       PetscCall(PforestLocalizeCell(plex, cDim, pforest, newSection, cell, coarsePoint, quad, coords2));
40840a96aa3bSJed Brown     }
40850a96aa3bSJed Brown   }
40869566063dSJacob Faibussowitsch   PetscCall(VecRestoreArray(cVec, &coords2));
40876858538eSMatthew G. Knepley   PetscCall(DMSetCellCoordinatesLocal(plex, cVec));
40889566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&cVec));
40899566063dSJacob Faibussowitsch   PetscCall(PetscSectionDestroy(&newSection));
40903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
40910a96aa3bSJed Brown }
40920a96aa3bSJed Brown 
40930a96aa3bSJed Brown   #define DMForestClearAdaptivityForest_pforest _append_pforest(DMForestClearAdaptivityForest)
4094d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestClearAdaptivityForest_pforest(DM dm)
4095d71ae5a4SJacob Faibussowitsch {
40960a96aa3bSJed Brown   DM_Forest         *forest;
40970a96aa3bSJed Brown   DM_Forest_pforest *pforest;
40980a96aa3bSJed Brown 
40990a96aa3bSJed Brown   PetscFunctionBegin;
41000a96aa3bSJed Brown   forest  = (DM_Forest *)dm->data;
41010a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)forest->data;
41029566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&(pforest->pointAdaptToSelfSF)));
41039566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&(pforest->pointSelfToAdaptSF)));
41049566063dSJacob Faibussowitsch   PetscCall(PetscFree(pforest->pointAdaptToSelfCids));
41059566063dSJacob Faibussowitsch   PetscCall(PetscFree(pforest->pointSelfToAdaptCids));
41063ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
41070a96aa3bSJed Brown }
41080a96aa3bSJed Brown 
4109d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMConvert_pforest_plex(DM dm, DMType newtype, DM *plex)
4110d71ae5a4SJacob Faibussowitsch {
41110a96aa3bSJed Brown   DM_Forest           *forest;
41120a96aa3bSJed Brown   DM_Forest_pforest   *pforest;
41130a96aa3bSJed Brown   DM                   refTree, newPlex, base;
41140a96aa3bSJed Brown   PetscInt             adjDim, adjCodim, coordDim;
41150a96aa3bSJed Brown   MPI_Comm             comm;
41160a96aa3bSJed Brown   PetscBool            isPforest;
41170a96aa3bSJed Brown   PetscInt             dim;
41180a96aa3bSJed Brown   PetscInt             overlap;
41190a96aa3bSJed Brown   p4est_connect_type_t ctype;
41200a96aa3bSJed Brown   p4est_locidx_t       first_local_quad = -1;
41210a96aa3bSJed Brown   sc_array_t          *points_per_dim, *cone_sizes, *cones, *cone_orientations, *coords, *children, *parents, *childids, *leaves, *remotes;
41220a96aa3bSJed Brown   PetscSection         parentSection;
41230a96aa3bSJed Brown   PetscSF              pointSF;
41240a96aa3bSJed Brown   size_t               zz, count;
41250a96aa3bSJed Brown   PetscInt             pStart, pEnd;
41260a96aa3bSJed Brown   DMLabel              ghostLabelBase = NULL;
41270a96aa3bSJed Brown 
41280a96aa3bSJed Brown   PetscFunctionBegin;
41290a96aa3bSJed Brown 
41300a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
41310a96aa3bSJed Brown   comm = PetscObjectComm((PetscObject)dm);
41329566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)dm, DMPFOREST, &isPforest));
413328b400f6SJacob Faibussowitsch   PetscCheck(isPforest, comm, PETSC_ERR_ARG_WRONG, "Expected DM type %s, got %s", DMPFOREST, ((PetscObject)dm)->type_name);
41349566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
413563a3b9bcSJacob Faibussowitsch   PetscCheck(dim == P4EST_DIM, comm, PETSC_ERR_ARG_WRONG, "Expected DM dimension %d, got %" PetscInt_FMT, P4EST_DIM, dim);
41360a96aa3bSJed Brown   forest  = (DM_Forest *)dm->data;
41370a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)forest->data;
41389566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseDM(dm, &base));
413948a46eb9SPierre Jolivet   if (base) PetscCall(DMGetLabel(base, "ghost", &ghostLabelBase));
41400a96aa3bSJed Brown   if (!pforest->plex) {
41410a96aa3bSJed Brown     PetscMPIInt size;
41420a96aa3bSJed Brown 
41439566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_size(comm, &size));
41449566063dSJacob Faibussowitsch     PetscCall(DMCreate(comm, &newPlex));
41459566063dSJacob Faibussowitsch     PetscCall(DMSetType(newPlex, DMPLEX));
41469566063dSJacob Faibussowitsch     PetscCall(DMSetMatType(newPlex, dm->mattype));
41470a96aa3bSJed Brown     /* share labels */
41489566063dSJacob Faibussowitsch     PetscCall(DMCopyLabels(dm, newPlex, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
41499566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdjacencyDimension(dm, &adjDim));
41509566063dSJacob Faibussowitsch     PetscCall(DMForestGetAdjacencyCodimension(dm, &adjCodim));
41519566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDim(dm, &coordDim));
41520a96aa3bSJed Brown     if (adjDim == 0) {
41530a96aa3bSJed Brown       ctype = P4EST_CONNECT_FULL;
41540a96aa3bSJed Brown     } else if (adjCodim == 1) {
41550a96aa3bSJed Brown       ctype = P4EST_CONNECT_FACE;
41560a96aa3bSJed Brown   #if defined(P4_TO_P8)
41570a96aa3bSJed Brown     } else if (adjDim == 1) {
41580a96aa3bSJed Brown       ctype = P8EST_CONNECT_EDGE;
41590a96aa3bSJed Brown   #endif
41600a96aa3bSJed Brown     } else {
416163a3b9bcSJacob Faibussowitsch       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Invalid adjacency dimension %" PetscInt_FMT, adjDim);
41620a96aa3bSJed Brown     }
416363a3b9bcSJacob Faibussowitsch     PetscCheck(ctype == P4EST_CONNECT_FULL, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Adjacency dimension %" PetscInt_FMT " / codimension %" PetscInt_FMT " not supported yet", adjDim, adjCodim);
41649566063dSJacob Faibussowitsch     PetscCall(DMForestGetPartitionOverlap(dm, &overlap));
416560667520SVaclav Hapla     PetscCall(DMPlexSetOverlap_Plex(newPlex, NULL, overlap));
41660a96aa3bSJed Brown 
41670a96aa3bSJed Brown     points_per_dim    = sc_array_new(sizeof(p4est_locidx_t));
41680a96aa3bSJed Brown     cone_sizes        = sc_array_new(sizeof(p4est_locidx_t));
41690a96aa3bSJed Brown     cones             = sc_array_new(sizeof(p4est_locidx_t));
41700a96aa3bSJed Brown     cone_orientations = sc_array_new(sizeof(p4est_locidx_t));
41710a96aa3bSJed Brown     coords            = sc_array_new(3 * sizeof(double));
41720a96aa3bSJed Brown     children          = sc_array_new(sizeof(p4est_locidx_t));
41730a96aa3bSJed Brown     parents           = sc_array_new(sizeof(p4est_locidx_t));
41740a96aa3bSJed Brown     childids          = sc_array_new(sizeof(p4est_locidx_t));
41750a96aa3bSJed Brown     leaves            = sc_array_new(sizeof(p4est_locidx_t));
41760a96aa3bSJed Brown     remotes           = sc_array_new(2 * sizeof(p4est_locidx_t));
41770a96aa3bSJed Brown 
4178792fecdfSBarry Smith     PetscCallP4est(p4est_get_plex_data_ext, (pforest->forest, &pforest->ghost, &pforest->lnodes, ctype, (int)((size > 1) ? overlap : 0), &first_local_quad, points_per_dim, cone_sizes, cones, cone_orientations, coords, children, parents, childids, leaves, remotes, 1));
41790a96aa3bSJed Brown 
41800a96aa3bSJed Brown     pforest->cLocalStart = (PetscInt)first_local_quad;
41810a96aa3bSJed Brown     pforest->cLocalEnd   = pforest->cLocalStart + (PetscInt)pforest->forest->local_num_quadrants;
41829566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(points_per_dim));
41839566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cone_sizes));
41849566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cones));
41859566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(cone_orientations));
41869566063dSJacob Faibussowitsch     PetscCall(coords_double_to_PetscScalar(coords, coordDim));
41879566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(children));
41889566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(parents));
41899566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(childids));
41909566063dSJacob Faibussowitsch     PetscCall(locidx_to_PetscInt(leaves));
41919566063dSJacob Faibussowitsch     PetscCall(locidx_pair_to_PetscSFNode(remotes));
41920a96aa3bSJed Brown 
41939566063dSJacob Faibussowitsch     PetscCall(DMSetDimension(newPlex, P4EST_DIM));
41949566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinateDim(newPlex, coordDim));
41959566063dSJacob Faibussowitsch     PetscCall(DMPlexSetMaxProjectionHeight(newPlex, P4EST_DIM - 1));
41969566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateFromDAG(newPlex, P4EST_DIM, (PetscInt *)points_per_dim->array, (PetscInt *)cone_sizes->array, (PetscInt *)cones->array, (PetscInt *)cone_orientations->array, (PetscScalar *)coords->array));
41979566063dSJacob Faibussowitsch     PetscCall(DMPlexConvertOldOrientations_Internal(newPlex));
41989566063dSJacob Faibussowitsch     PetscCall(DMCreateReferenceTree_pforest(comm, &refTree));
41999566063dSJacob Faibussowitsch     PetscCall(DMPlexSetReferenceTree(newPlex, refTree));
42009566063dSJacob Faibussowitsch     PetscCall(PetscSectionCreate(comm, &parentSection));
42019566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(newPlex, &pStart, &pEnd));
42029566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetChart(parentSection, pStart, pEnd));
42030a96aa3bSJed Brown     count = children->elem_count;
42040a96aa3bSJed Brown     for (zz = 0; zz < count; zz++) {
42050a96aa3bSJed Brown       PetscInt child = *((PetscInt *)sc_array_index(children, zz));
42060a96aa3bSJed Brown 
42079566063dSJacob Faibussowitsch       PetscCall(PetscSectionSetDof(parentSection, child, 1));
42080a96aa3bSJed Brown     }
42099566063dSJacob Faibussowitsch     PetscCall(PetscSectionSetUp(parentSection));
42109566063dSJacob Faibussowitsch     PetscCall(DMPlexSetTree(newPlex, parentSection, (PetscInt *)parents->array, (PetscInt *)childids->array));
42119566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&parentSection));
42129566063dSJacob Faibussowitsch     PetscCall(PetscSFCreate(comm, &pointSF));
42130a96aa3bSJed Brown     /*
42140a96aa3bSJed Brown        These arrays defining the sf are from the p4est library, but the code there shows the leaves being populated in increasing order.
42150a96aa3bSJed Brown        https://gitlab.com/petsc/petsc/merge_requests/2248#note_240186391
42160a96aa3bSJed Brown     */
42179566063dSJacob Faibussowitsch     PetscCall(PetscSFSetGraph(pointSF, pEnd - pStart, (PetscInt)leaves->elem_count, (PetscInt *)leaves->array, PETSC_COPY_VALUES, (PetscSFNode *)remotes->array, PETSC_COPY_VALUES));
42189566063dSJacob Faibussowitsch     PetscCall(DMSetPointSF(newPlex, pointSF));
42199566063dSJacob Faibussowitsch     PetscCall(DMSetPointSF(dm, pointSF));
42200a96aa3bSJed Brown     {
42210a96aa3bSJed Brown       DM coordDM;
42220a96aa3bSJed Brown 
42239566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinateDM(newPlex, &coordDM));
42249566063dSJacob Faibussowitsch       PetscCall(DMSetPointSF(coordDM, pointSF));
42250a96aa3bSJed Brown     }
42269566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&pointSF));
42270a96aa3bSJed Brown     sc_array_destroy(points_per_dim);
42280a96aa3bSJed Brown     sc_array_destroy(cone_sizes);
42290a96aa3bSJed Brown     sc_array_destroy(cones);
42300a96aa3bSJed Brown     sc_array_destroy(cone_orientations);
42310a96aa3bSJed Brown     sc_array_destroy(coords);
42320a96aa3bSJed Brown     sc_array_destroy(children);
42330a96aa3bSJed Brown     sc_array_destroy(parents);
42340a96aa3bSJed Brown     sc_array_destroy(childids);
42350a96aa3bSJed Brown     sc_array_destroy(leaves);
42360a96aa3bSJed Brown     sc_array_destroy(remotes);
42370a96aa3bSJed Brown 
42380a96aa3bSJed Brown     {
42394fb89dddSMatthew G. Knepley       const PetscReal *maxCell, *Lstart, *L;
42400a96aa3bSJed Brown 
42414fb89dddSMatthew G. Knepley       PetscCall(DMGetPeriodicity(dm, &maxCell, &Lstart, &L));
42424fb89dddSMatthew G. Knepley       PetscCall(DMSetPeriodicity(newPlex, maxCell, Lstart, L));
42439566063dSJacob Faibussowitsch       PetscCall(DMPforestLocalizeCoordinates(dm, newPlex));
42440a96aa3bSJed Brown     }
42450a96aa3bSJed Brown 
42460a96aa3bSJed Brown     if (overlap > 0) { /* the p4est routine can't set all of the coordinates in its routine if there is overlap */
42470a96aa3bSJed Brown       Vec                coordsGlobal, coordsLocal;
42480a96aa3bSJed Brown       const PetscScalar *globalArray;
42490a96aa3bSJed Brown       PetscScalar       *localArray;
42500a96aa3bSJed Brown       PetscSF            coordSF;
42510a96aa3bSJed Brown       DM                 coordDM;
42520a96aa3bSJed Brown 
42539566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinateDM(newPlex, &coordDM));
42549566063dSJacob Faibussowitsch       PetscCall(DMGetSectionSF(coordDM, &coordSF));
42559566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(newPlex, &coordsGlobal));
42569566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(newPlex, &coordsLocal));
42579566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(coordsGlobal, &globalArray));
42589566063dSJacob Faibussowitsch       PetscCall(VecGetArray(coordsLocal, &localArray));
42599566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastBegin(coordSF, MPIU_SCALAR, globalArray, localArray, MPI_REPLACE));
42609566063dSJacob Faibussowitsch       PetscCall(PetscSFBcastEnd(coordSF, MPIU_SCALAR, globalArray, localArray, MPI_REPLACE));
42619566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(coordsLocal, &localArray));
42629566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(coordsGlobal, &globalArray));
42639566063dSJacob Faibussowitsch       PetscCall(DMSetCoordinatesLocal(newPlex, coordsLocal));
42640a96aa3bSJed Brown     }
42659566063dSJacob Faibussowitsch     PetscCall(DMPforestMapCoordinates(dm, newPlex));
42660a96aa3bSJed Brown 
42670a96aa3bSJed Brown     pforest->plex = newPlex;
42680a96aa3bSJed Brown 
42690a96aa3bSJed Brown     /* copy labels */
42709566063dSJacob Faibussowitsch     PetscCall(DMPforestLabelsFinalize(dm, newPlex));
42710a96aa3bSJed Brown 
42720a96aa3bSJed Brown     if (ghostLabelBase || pforest->ghostName) { /* we have to do this after copying labels because the labels drive the construction of ghost cells */
42730a96aa3bSJed Brown       PetscInt numAdded;
42740a96aa3bSJed Brown       DM       newPlexGhosted;
42750a96aa3bSJed Brown       void    *ctx;
42760a96aa3bSJed Brown 
42779566063dSJacob Faibussowitsch       PetscCall(DMPlexConstructGhostCells(newPlex, pforest->ghostName, &numAdded, &newPlexGhosted));
42789566063dSJacob Faibussowitsch       PetscCall(DMGetApplicationContext(newPlex, &ctx));
42799566063dSJacob Faibussowitsch       PetscCall(DMSetApplicationContext(newPlexGhosted, ctx));
42800a96aa3bSJed Brown       /* we want the sf for the ghost dm to be the one for the p4est dm as well */
42819566063dSJacob Faibussowitsch       PetscCall(DMGetPointSF(newPlexGhosted, &pointSF));
42829566063dSJacob Faibussowitsch       PetscCall(DMSetPointSF(dm, pointSF));
42839566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&newPlex));
42849566063dSJacob Faibussowitsch       PetscCall(DMPlexSetReferenceTree(newPlexGhosted, refTree));
42859566063dSJacob Faibussowitsch       PetscCall(DMForestClearAdaptivityForest_pforest(dm));
42860a96aa3bSJed Brown       newPlex = newPlexGhosted;
42870a96aa3bSJed Brown 
42880a96aa3bSJed Brown       /* share the labels back */
42899566063dSJacob Faibussowitsch       PetscCall(DMDestroyLabelLinkList_Internal(dm));
42909566063dSJacob Faibussowitsch       PetscCall(DMCopyLabels(newPlex, dm, PETSC_OWN_POINTER, PETSC_TRUE, DM_COPY_LABELS_FAIL));
42910a96aa3bSJed Brown       pforest->plex = newPlex;
42920a96aa3bSJed Brown     }
42939566063dSJacob Faibussowitsch     PetscCall(DMDestroy(&refTree));
42940a96aa3bSJed Brown     if (dm->setfromoptionscalled) {
4295d0609cedSBarry Smith       PetscObjectOptionsBegin((PetscObject)newPlex);
4296dbbe0bcdSBarry Smith       PetscCall(DMSetFromOptions_NonRefinement_Plex(newPlex, PetscOptionsObject));
4297dbbe0bcdSBarry Smith       PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)newPlex, PetscOptionsObject));
4298d0609cedSBarry Smith       PetscOptionsEnd();
42990a96aa3bSJed Brown     }
43009566063dSJacob Faibussowitsch     PetscCall(DMViewFromOptions(newPlex, NULL, "-dm_p4est_plex_view"));
43010a96aa3bSJed Brown     {
43026858538eSMatthew G. Knepley       DM           cdm;
43030a96aa3bSJed Brown       PetscSection coordsSec;
43040a96aa3bSJed Brown       Vec          coords;
43050a96aa3bSJed Brown       PetscInt     cDim;
43060a96aa3bSJed Brown 
43079566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinateDim(newPlex, &cDim));
43089566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinateSection(newPlex, &coordsSec));
43099566063dSJacob Faibussowitsch       PetscCall(DMSetCoordinateSection(dm, cDim, coordsSec));
43109566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinatesLocal(newPlex, &coords));
43119566063dSJacob Faibussowitsch       PetscCall(DMSetCoordinatesLocal(dm, coords));
43126858538eSMatthew G. Knepley       PetscCall(DMGetCellCoordinateDM(newPlex, &cdm));
43136858538eSMatthew G. Knepley       if (cdm) PetscCall(DMSetCellCoordinateDM(dm, cdm));
43146858538eSMatthew G. Knepley       PetscCall(DMGetCellCoordinateSection(newPlex, &coordsSec));
43156858538eSMatthew G. Knepley       if (coordsSec) PetscCall(DMSetCellCoordinateSection(dm, cDim, coordsSec));
43166858538eSMatthew G. Knepley       PetscCall(DMGetCellCoordinatesLocal(newPlex, &coords));
43176858538eSMatthew G. Knepley       if (coords) PetscCall(DMSetCellCoordinatesLocal(dm, coords));
43180a96aa3bSJed Brown     }
43190a96aa3bSJed Brown   }
43200a96aa3bSJed Brown   newPlex = pforest->plex;
43210a96aa3bSJed Brown   if (plex) {
43229566063dSJacob Faibussowitsch     PetscCall(DMClone(newPlex, plex));
43236858538eSMatthew G. Knepley   #if 0
43249566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(newPlex,&coordDM));
43259566063dSJacob Faibussowitsch     PetscCall(DMSetCoordinateDM(*plex,coordDM));
43266858538eSMatthew G. Knepley     PetscCall(DMGetCellCoordinateDM(newPlex,&coordDM));
43276858538eSMatthew G. Knepley     PetscCall(DMSetCellCoordinateDM(*plex,coordDM));
43286858538eSMatthew G. Knepley   #endif
43299566063dSJacob Faibussowitsch     PetscCall(DMShareDiscretization(dm, *plex));
43300a96aa3bSJed Brown   }
43313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43320a96aa3bSJed Brown }
43330a96aa3bSJed Brown 
4334d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMSetFromOptions_pforest(DM dm, PetscOptionItems *PetscOptionsObject)
4335d71ae5a4SJacob Faibussowitsch {
43360a96aa3bSJed Brown   DM_Forest_pforest *pforest = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
43370a96aa3bSJed Brown   char               stringBuffer[256];
43380a96aa3bSJed Brown   PetscBool          flg;
43390a96aa3bSJed Brown 
43400a96aa3bSJed Brown   PetscFunctionBegin;
4341dbbe0bcdSBarry Smith   PetscCall(DMSetFromOptions_Forest(dm, PetscOptionsObject));
4342d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "DM" P4EST_STRING " options");
43439566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-dm_p4est_partition_for_coarsening", "partition forest to allow for coarsening", "DMP4estSetPartitionForCoarsening", pforest->partition_for_coarsening, &(pforest->partition_for_coarsening), NULL));
43449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-dm_p4est_ghost_label_name", "the name of the ghost label when converting from a DMPlex", NULL, NULL, stringBuffer, sizeof(stringBuffer), &flg));
4345d0609cedSBarry Smith   PetscOptionsHeadEnd();
43460a96aa3bSJed Brown   if (flg) {
43479566063dSJacob Faibussowitsch     PetscCall(PetscFree(pforest->ghostName));
43489566063dSJacob Faibussowitsch     PetscCall(PetscStrallocpy(stringBuffer, &pforest->ghostName));
43490a96aa3bSJed Brown   }
43503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43510a96aa3bSJed Brown }
43520a96aa3bSJed Brown 
43530a96aa3bSJed Brown   #if !defined(P4_TO_P8)
43540a96aa3bSJed Brown     #define DMPforestGetPartitionForCoarsening DMP4estGetPartitionForCoarsening
43550a96aa3bSJed Brown     #define DMPforestSetPartitionForCoarsening DMP4estSetPartitionForCoarsening
43560a96aa3bSJed Brown   #else
43570a96aa3bSJed Brown     #define DMPforestGetPartitionForCoarsening DMP8estGetPartitionForCoarsening
43580a96aa3bSJed Brown     #define DMPforestSetPartitionForCoarsening DMP8estSetPartitionForCoarsening
43590a96aa3bSJed Brown   #endif
43600a96aa3bSJed Brown 
4361d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMPforestGetPartitionForCoarsening(DM dm, PetscBool *flg)
4362d71ae5a4SJacob Faibussowitsch {
43630a96aa3bSJed Brown   DM_Forest_pforest *pforest;
43640a96aa3bSJed Brown 
43650a96aa3bSJed Brown   PetscFunctionBegin;
43660a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
43670a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
43680a96aa3bSJed Brown   *flg    = pforest->partition_for_coarsening;
43693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43700a96aa3bSJed Brown }
43710a96aa3bSJed Brown 
4372d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMPforestSetPartitionForCoarsening(DM dm, PetscBool flg)
4373d71ae5a4SJacob Faibussowitsch {
43740a96aa3bSJed Brown   DM_Forest_pforest *pforest;
43750a96aa3bSJed Brown 
43760a96aa3bSJed Brown   PetscFunctionBegin;
43770a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
43780a96aa3bSJed Brown   pforest                           = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
43790a96aa3bSJed Brown   pforest->partition_for_coarsening = flg;
43803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43810a96aa3bSJed Brown }
43820a96aa3bSJed Brown 
4383d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMPforestGetPlex(DM dm, DM *plex)
4384d71ae5a4SJacob Faibussowitsch {
43850a96aa3bSJed Brown   DM_Forest_pforest *pforest;
43860a96aa3bSJed Brown 
43870a96aa3bSJed Brown   PetscFunctionBegin;
43880a96aa3bSJed Brown   if (plex) *plex = NULL;
43899566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
43900a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)((DM_Forest *)dm->data)->data;
439148a46eb9SPierre Jolivet   if (!pforest->plex) PetscCall(DMConvert_pforest_plex(dm, DMPLEX, NULL));
43929566063dSJacob Faibussowitsch   PetscCall(DMShareDiscretization(dm, pforest->plex));
43930a96aa3bSJed Brown   if (plex) *plex = pforest->plex;
43943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
43950a96aa3bSJed Brown }
43960a96aa3bSJed Brown 
43970a96aa3bSJed Brown   #define DMCreateInterpolation_pforest _append_pforest(DMCreateInterpolation)
4398d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateInterpolation_pforest(DM dmCoarse, DM dmFine, Mat *interpolation, Vec *scaling)
4399d71ae5a4SJacob Faibussowitsch {
44000a96aa3bSJed Brown   PetscSection gsc, gsf;
44010a96aa3bSJed Brown   PetscInt     m, n;
44020a96aa3bSJed Brown   DM           cdm;
44030a96aa3bSJed Brown 
44040a96aa3bSJed Brown   PetscFunctionBegin;
44059566063dSJacob Faibussowitsch   PetscCall(DMGetGlobalSection(dmFine, &gsf));
44069566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetConstrainedStorageSize(gsf, &m));
44079566063dSJacob Faibussowitsch   PetscCall(DMGetGlobalSection(dmCoarse, &gsc));
44089566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetConstrainedStorageSize(gsc, &n));
44090a96aa3bSJed Brown 
44109566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)dmFine), interpolation));
44119566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(*interpolation, m, n, PETSC_DETERMINE, PETSC_DETERMINE));
44129566063dSJacob Faibussowitsch   PetscCall(MatSetType(*interpolation, MATAIJ));
44130a96aa3bSJed Brown 
44149566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmFine, &cdm));
441508401ef6SPierre Jolivet   PetscCheck(cdm == dmCoarse, PetscObjectComm((PetscObject)dmFine), PETSC_ERR_SUP, "Only interpolation from coarse DM for now");
44160a96aa3bSJed Brown 
44170a96aa3bSJed Brown   {
44180a96aa3bSJed Brown     DM        plexF, plexC;
44190a96aa3bSJed Brown     PetscSF   sf;
44200a96aa3bSJed Brown     PetscInt *cids;
44210a96aa3bSJed Brown     PetscInt  dofPerDim[4] = {1, 1, 1, 1};
44220a96aa3bSJed Brown 
44239566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(dmCoarse, &plexC));
44249566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(dmFine, &plexF));
44259566063dSJacob Faibussowitsch     PetscCall(DMPforestGetTransferSF_Internal(dmCoarse, dmFine, dofPerDim, &sf, PETSC_TRUE, &cids));
44269566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
44279566063dSJacob Faibussowitsch     PetscCall(DMPlexComputeInterpolatorTree(plexC, plexF, sf, cids, *interpolation));
44289566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&sf));
44299566063dSJacob Faibussowitsch     PetscCall(PetscFree(cids));
44300a96aa3bSJed Brown   }
44319566063dSJacob Faibussowitsch   PetscCall(MatViewFromOptions(*interpolation, NULL, "-interp_mat_view"));
44320a96aa3bSJed Brown   /* Use naive scaling */
44339566063dSJacob Faibussowitsch   PetscCall(DMCreateInterpolationScale(dmCoarse, dmFine, *interpolation, scaling));
44343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44350a96aa3bSJed Brown }
44360a96aa3bSJed Brown 
44370a96aa3bSJed Brown   #define DMCreateInjection_pforest _append_pforest(DMCreateInjection)
4438d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateInjection_pforest(DM dmCoarse, DM dmFine, Mat *injection)
4439d71ae5a4SJacob Faibussowitsch {
44400a96aa3bSJed Brown   PetscSection gsc, gsf;
44410a96aa3bSJed Brown   PetscInt     m, n;
44420a96aa3bSJed Brown   DM           cdm;
44430a96aa3bSJed Brown 
44440a96aa3bSJed Brown   PetscFunctionBegin;
44459566063dSJacob Faibussowitsch   PetscCall(DMGetGlobalSection(dmFine, &gsf));
44469566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetConstrainedStorageSize(gsf, &n));
44479566063dSJacob Faibussowitsch   PetscCall(DMGetGlobalSection(dmCoarse, &gsc));
44489566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetConstrainedStorageSize(gsc, &m));
44490a96aa3bSJed Brown 
44509566063dSJacob Faibussowitsch   PetscCall(MatCreate(PetscObjectComm((PetscObject)dmFine), injection));
44519566063dSJacob Faibussowitsch   PetscCall(MatSetSizes(*injection, m, n, PETSC_DETERMINE, PETSC_DETERMINE));
44529566063dSJacob Faibussowitsch   PetscCall(MatSetType(*injection, MATAIJ));
44530a96aa3bSJed Brown 
44549566063dSJacob Faibussowitsch   PetscCall(DMGetCoarseDM(dmFine, &cdm));
445508401ef6SPierre Jolivet   PetscCheck(cdm == dmCoarse, PetscObjectComm((PetscObject)dmFine), PETSC_ERR_SUP, "Only injection to coarse DM for now");
44560a96aa3bSJed Brown 
44570a96aa3bSJed Brown   {
44580a96aa3bSJed Brown     DM        plexF, plexC;
44590a96aa3bSJed Brown     PetscSF   sf;
44600a96aa3bSJed Brown     PetscInt *cids;
44610a96aa3bSJed Brown     PetscInt  dofPerDim[4] = {1, 1, 1, 1};
44620a96aa3bSJed Brown 
44639566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(dmCoarse, &plexC));
44649566063dSJacob Faibussowitsch     PetscCall(DMPforestGetPlex(dmFine, &plexF));
44659566063dSJacob Faibussowitsch     PetscCall(DMPforestGetTransferSF_Internal(dmCoarse, dmFine, dofPerDim, &sf, PETSC_TRUE, &cids));
44669566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(sf));
44679566063dSJacob Faibussowitsch     PetscCall(DMPlexComputeInjectorTree(plexC, plexF, sf, cids, *injection));
44689566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&sf));
44699566063dSJacob Faibussowitsch     PetscCall(PetscFree(cids));
44700a96aa3bSJed Brown   }
44719566063dSJacob Faibussowitsch   PetscCall(MatViewFromOptions(*injection, NULL, "-inject_mat_view"));
44720a96aa3bSJed Brown   /* Use naive scaling */
44733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
44740a96aa3bSJed Brown }
44750a96aa3bSJed Brown 
44760a96aa3bSJed Brown   #define DMForestTransferVecFromBase_pforest _append_pforest(DMForestTransferVecFromBase)
4477d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestTransferVecFromBase_pforest(DM dm, Vec vecIn, Vec vecOut)
4478d71ae5a4SJacob Faibussowitsch {
44790a96aa3bSJed Brown   DM        dmIn, dmVecIn, base, basec, plex, coarseDM;
44800a96aa3bSJed Brown   DM       *hierarchy;
44810a96aa3bSJed Brown   PetscSF   sfRed = NULL;
44820a96aa3bSJed Brown   PetscDS   ds;
44830a96aa3bSJed Brown   Vec       vecInLocal, vecOutLocal;
44840a96aa3bSJed Brown   DMLabel   subpointMap;
44850a96aa3bSJed Brown   PetscInt  minLevel, mh, n_hi, i;
44860a96aa3bSJed Brown   PetscBool hiforest, *hierarchy_forest;
44870a96aa3bSJed Brown 
44880a96aa3bSJed Brown   PetscFunctionBegin;
44899566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vecIn, &dmVecIn));
44909566063dSJacob Faibussowitsch   PetscCall(DMGetDS(dmVecIn, &ds));
449128b400f6SJacob Faibussowitsch   PetscCheck(ds, PetscObjectComm((PetscObject)dmVecIn), PETSC_ERR_SUP, "Cannot transfer without a PetscDS object");
44920a96aa3bSJed Brown   { /* we cannot stick user contexts into function callbacks for DMProjectFieldLocal! */
44930a96aa3bSJed Brown     PetscSection section;
44940a96aa3bSJed Brown     PetscInt     Nf;
44950a96aa3bSJed Brown 
44969566063dSJacob Faibussowitsch     PetscCall(DMGetLocalSection(dmVecIn, &section));
44979566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetNumFields(section, &Nf));
449863a3b9bcSJacob Faibussowitsch     PetscCheck(Nf <= 3, PetscObjectComm((PetscObject)dmVecIn), PETSC_ERR_SUP, "Number of fields %" PetscInt_FMT " are currently not supported! Send an email at petsc-dev@mcs.anl.gov", Nf);
44990a96aa3bSJed Brown   }
45009566063dSJacob Faibussowitsch   PetscCall(DMForestGetMinimumRefinement(dm, &minLevel));
450163a3b9bcSJacob Faibussowitsch   PetscCheck(!minLevel, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Cannot transfer with minimum refinement set to %" PetscInt_FMT ". Rerun with DMForestSetMinimumRefinement(dm,0)", minLevel);
45029566063dSJacob Faibussowitsch   PetscCall(DMForestGetBaseDM(dm, &base));
450328b400f6SJacob Faibussowitsch   PetscCheck(base, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Missing base DM");
45040a96aa3bSJed Brown 
45059566063dSJacob Faibussowitsch   PetscCall(VecSet(vecOut, 0.0));
45060a96aa3bSJed Brown   if (dmVecIn == base) { /* sequential runs */
45079566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)vecIn));
45080a96aa3bSJed Brown   } else {
45090a96aa3bSJed Brown     PetscSection secIn, secInRed;
45100a96aa3bSJed Brown     Vec          vecInRed, vecInLocal;
45110a96aa3bSJed Brown 
45129566063dSJacob Faibussowitsch     PetscCall(PetscObjectQuery((PetscObject)base, "_base_migration_sf", (PetscObject *)&sfRed));
451328b400f6SJacob Faibussowitsch     PetscCheck(sfRed, PETSC_COMM_SELF, PETSC_ERR_SUP, "Not the DM set with DMForestSetBaseDM()");
45149566063dSJacob Faibussowitsch     PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dmVecIn), &secInRed));
45159566063dSJacob Faibussowitsch     PetscCall(VecCreate(PETSC_COMM_SELF, &vecInRed));
45169566063dSJacob Faibussowitsch     PetscCall(DMGetLocalSection(dmVecIn, &secIn));
45179566063dSJacob Faibussowitsch     PetscCall(DMGetLocalVector(dmVecIn, &vecInLocal));
45189566063dSJacob Faibussowitsch     PetscCall(DMGlobalToLocalBegin(dmVecIn, vecIn, INSERT_VALUES, vecInLocal));
45199566063dSJacob Faibussowitsch     PetscCall(DMGlobalToLocalEnd(dmVecIn, vecIn, INSERT_VALUES, vecInLocal));
45209566063dSJacob Faibussowitsch     PetscCall(DMPlexDistributeField(dmVecIn, sfRed, secIn, vecInLocal, secInRed, vecInRed));
45219566063dSJacob Faibussowitsch     PetscCall(DMRestoreLocalVector(dmVecIn, &vecInLocal));
45229566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&secInRed));
45230a96aa3bSJed Brown     vecIn = vecInRed;
45240a96aa3bSJed Brown   }
45250a96aa3bSJed Brown 
45260a96aa3bSJed Brown   /* we first search through the AdaptivityForest hierarchy
45270a96aa3bSJed Brown      once we found the first disconnected forest, we upsweep the DM hierarchy */
45280a96aa3bSJed Brown   hiforest = PETSC_TRUE;
45290a96aa3bSJed Brown 
45300a96aa3bSJed Brown   /* upsweep to the coarsest DM */
45310a96aa3bSJed Brown   n_hi     = 0;
45320a96aa3bSJed Brown   coarseDM = dm;
45330a96aa3bSJed Brown   do {
45340a96aa3bSJed Brown     PetscBool isforest;
45350a96aa3bSJed Brown 
45360a96aa3bSJed Brown     dmIn = coarseDM;
45370a96aa3bSJed Brown     /* need to call DMSetUp to have the hierarchy recursively setup */
45389566063dSJacob Faibussowitsch     PetscCall(DMSetUp(dmIn));
45399566063dSJacob Faibussowitsch     PetscCall(DMIsForest(dmIn, &isforest));
454028b400f6SJacob Faibussowitsch     PetscCheck(isforest, PetscObjectComm((PetscObject)dmIn), PETSC_ERR_SUP, "Cannot currently transfer through a mixed hierarchy! Found DM type %s", ((PetscObject)dmIn)->type_name);
45410a96aa3bSJed Brown     coarseDM = NULL;
454248a46eb9SPierre Jolivet     if (hiforest) PetscCall(DMForestGetAdaptivityForest(dmIn, &coarseDM));
45430a96aa3bSJed Brown     if (!coarseDM) { /* DMForest hierarchy ended, we keep upsweeping through the DM hierarchy */
45440a96aa3bSJed Brown       hiforest = PETSC_FALSE;
45459566063dSJacob Faibussowitsch       PetscCall(DMGetCoarseDM(dmIn, &coarseDM));
45460a96aa3bSJed Brown     }
45470a96aa3bSJed Brown     n_hi++;
45480a96aa3bSJed Brown   } while (coarseDM);
45490a96aa3bSJed Brown 
45509566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(n_hi, &hierarchy, n_hi, &hierarchy_forest));
45510a96aa3bSJed Brown 
45520a96aa3bSJed Brown   i        = 0;
45530a96aa3bSJed Brown   hiforest = PETSC_TRUE;
45540a96aa3bSJed Brown   coarseDM = dm;
45550a96aa3bSJed Brown   do {
45560a96aa3bSJed Brown     dmIn     = coarseDM;
45570a96aa3bSJed Brown     coarseDM = NULL;
455848a46eb9SPierre Jolivet     if (hiforest) PetscCall(DMForestGetAdaptivityForest(dmIn, &coarseDM));
45590a96aa3bSJed Brown     if (!coarseDM) { /* DMForest hierarchy ended, we keep upsweeping through the DM hierarchy */
45600a96aa3bSJed Brown       hiforest = PETSC_FALSE;
45619566063dSJacob Faibussowitsch       PetscCall(DMGetCoarseDM(dmIn, &coarseDM));
45620a96aa3bSJed Brown     }
45630a96aa3bSJed Brown     i++;
45640a96aa3bSJed Brown     hierarchy[n_hi - i] = dmIn;
45650a96aa3bSJed Brown   } while (coarseDM);
45660a96aa3bSJed Brown 
45670a96aa3bSJed Brown   /* project base vector on the coarsest forest (minimum refinement = 0) */
45689566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dmIn, &plex));
45690a96aa3bSJed Brown 
45700a96aa3bSJed Brown   /* Check this plex is compatible with the base */
45710a96aa3bSJed Brown   {
45720a96aa3bSJed Brown     IS       gnum[2];
45730a96aa3bSJed Brown     PetscInt ncells[2], gncells[2];
45740a96aa3bSJed Brown 
45759566063dSJacob Faibussowitsch     PetscCall(DMPlexGetCellNumbering(base, &gnum[0]));
45769566063dSJacob Faibussowitsch     PetscCall(DMPlexGetCellNumbering(plex, &gnum[1]));
45779566063dSJacob Faibussowitsch     PetscCall(ISGetMinMax(gnum[0], NULL, &ncells[0]));
45789566063dSJacob Faibussowitsch     PetscCall(ISGetMinMax(gnum[1], NULL, &ncells[1]));
45791c2dc1cbSBarry Smith     PetscCall(MPIU_Allreduce(ncells, gncells, 2, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm)));
458063a3b9bcSJacob Faibussowitsch     PetscCheck(gncells[0] == gncells[1], PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Invalid number of base cells! Expected %" PetscInt_FMT ", found %" PetscInt_FMT, gncells[0] + 1, gncells[1] + 1);
45810a96aa3bSJed Brown   }
45820a96aa3bSJed Brown 
45839566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dmIn, "_forest_base_subpoint_map", &subpointMap));
458428b400f6SJacob Faibussowitsch   PetscCheck(subpointMap, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing _forest_base_subpoint_map label");
45850a96aa3bSJed Brown 
45869566063dSJacob Faibussowitsch   PetscCall(DMPlexGetMaxProjectionHeight(base, &mh));
45879566063dSJacob Faibussowitsch   PetscCall(DMPlexSetMaxProjectionHeight(plex, mh));
45880a96aa3bSJed Brown 
45899566063dSJacob Faibussowitsch   PetscCall(DMClone(base, &basec));
45909566063dSJacob Faibussowitsch   PetscCall(DMCopyDisc(dmVecIn, basec));
45910a96aa3bSJed Brown   if (sfRed) {
45929566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)vecIn));
45930a96aa3bSJed Brown     vecInLocal = vecIn;
45940a96aa3bSJed Brown   } else {
45959566063dSJacob Faibussowitsch     PetscCall(DMCreateLocalVector(basec, &vecInLocal));
45969566063dSJacob Faibussowitsch     PetscCall(DMGlobalToLocalBegin(basec, vecIn, INSERT_VALUES, vecInLocal));
45979566063dSJacob Faibussowitsch     PetscCall(DMGlobalToLocalEnd(basec, vecIn, INSERT_VALUES, vecInLocal));
45980a96aa3bSJed Brown   }
45990a96aa3bSJed Brown 
46009566063dSJacob Faibussowitsch   PetscCall(DMGetLocalVector(dmIn, &vecOutLocal));
46010a96aa3bSJed Brown   { /* get degrees of freedom ordered onto dmIn */
46020a96aa3bSJed Brown     PetscSF            basetocoarse;
46030a96aa3bSJed Brown     PetscInt           bStart, bEnd, nroots;
46040a96aa3bSJed Brown     PetscInt           iStart, iEnd, nleaves, leaf;
46050a96aa3bSJed Brown     PetscMPIInt        rank;
46060a96aa3bSJed Brown     PetscSFNode       *remotes;
46070a96aa3bSJed Brown     PetscSection       secIn, secOut;
46080a96aa3bSJed Brown     PetscInt          *remoteOffsets;
46090a96aa3bSJed Brown     PetscSF            transferSF;
46100a96aa3bSJed Brown     const PetscScalar *inArray;
46110a96aa3bSJed Brown     PetscScalar       *outArray;
46120a96aa3bSJed Brown 
46139566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)basec), &rank));
46149566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(basec, &bStart, &bEnd));
46150a96aa3bSJed Brown     nroots = PetscMax(bEnd - bStart, 0);
46169566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(plex, &iStart, &iEnd));
46170a96aa3bSJed Brown     nleaves = PetscMax(iEnd - iStart, 0);
46180a96aa3bSJed Brown 
46199566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nleaves, &remotes));
46200a96aa3bSJed Brown     for (leaf = iStart; leaf < iEnd; leaf++) {
46210a96aa3bSJed Brown       PetscInt index;
46220a96aa3bSJed Brown 
46230a96aa3bSJed Brown       remotes[leaf - iStart].rank = rank;
46249566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValue(subpointMap, leaf, &index));
46250a96aa3bSJed Brown       remotes[leaf - iStart].index = index;
46260a96aa3bSJed Brown     }
46270a96aa3bSJed Brown 
46289566063dSJacob Faibussowitsch     PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)basec), &basetocoarse));
46299566063dSJacob Faibussowitsch     PetscCall(PetscSFSetGraph(basetocoarse, nroots, nleaves, NULL, PETSC_OWN_POINTER, remotes, PETSC_OWN_POINTER));
46309566063dSJacob Faibussowitsch     PetscCall(PetscSFSetUp(basetocoarse));
46319566063dSJacob Faibussowitsch     PetscCall(DMGetLocalSection(basec, &secIn));
46329566063dSJacob Faibussowitsch     PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dmIn), &secOut));
46339566063dSJacob Faibussowitsch     PetscCall(PetscSFDistributeSection(basetocoarse, secIn, &remoteOffsets, secOut));
46349566063dSJacob Faibussowitsch     PetscCall(PetscSFCreateSectionSF(basetocoarse, secIn, remoteOffsets, secOut, &transferSF));
46359566063dSJacob Faibussowitsch     PetscCall(PetscFree(remoteOffsets));
46369566063dSJacob Faibussowitsch     PetscCall(VecGetArrayWrite(vecOutLocal, &outArray));
46379566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(vecInLocal, &inArray));
46389566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastBegin(transferSF, MPIU_SCALAR, inArray, outArray, MPI_REPLACE));
46399566063dSJacob Faibussowitsch     PetscCall(PetscSFBcastEnd(transferSF, MPIU_SCALAR, inArray, outArray, MPI_REPLACE));
46409566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(vecInLocal, &inArray));
46419566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayWrite(vecOutLocal, &outArray));
46429566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&transferSF));
46439566063dSJacob Faibussowitsch     PetscCall(PetscSectionDestroy(&secOut));
46449566063dSJacob Faibussowitsch     PetscCall(PetscSFDestroy(&basetocoarse));
46450a96aa3bSJed Brown   }
46469566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&vecInLocal));
46479566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&basec));
46489566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&vecIn));
46490a96aa3bSJed Brown 
46500a96aa3bSJed Brown   /* output */
46510a96aa3bSJed Brown   if (n_hi > 1) { /* downsweep the stored hierarchy */
46520a96aa3bSJed Brown     Vec vecOut1, vecOut2;
46530a96aa3bSJed Brown     DM  fineDM;
46540a96aa3bSJed Brown 
46559566063dSJacob Faibussowitsch     PetscCall(DMGetGlobalVector(dmIn, &vecOut1));
46569566063dSJacob Faibussowitsch     PetscCall(DMLocalToGlobal(dmIn, vecOutLocal, INSERT_VALUES, vecOut1));
46579566063dSJacob Faibussowitsch     PetscCall(DMRestoreLocalVector(dmIn, &vecOutLocal));
46580a96aa3bSJed Brown     for (i = 1; i < n_hi - 1; i++) {
46590a96aa3bSJed Brown       fineDM = hierarchy[i];
46609566063dSJacob Faibussowitsch       PetscCall(DMGetGlobalVector(fineDM, &vecOut2));
46619566063dSJacob Faibussowitsch       PetscCall(DMForestTransferVec(dmIn, vecOut1, fineDM, vecOut2, PETSC_TRUE, 0.0));
46629566063dSJacob Faibussowitsch       PetscCall(DMRestoreGlobalVector(dmIn, &vecOut1));
46630a96aa3bSJed Brown       vecOut1 = vecOut2;
46640a96aa3bSJed Brown       dmIn    = fineDM;
46650a96aa3bSJed Brown     }
46669566063dSJacob Faibussowitsch     PetscCall(DMForestTransferVec(dmIn, vecOut1, dm, vecOut, PETSC_TRUE, 0.0));
46679566063dSJacob Faibussowitsch     PetscCall(DMRestoreGlobalVector(dmIn, &vecOut1));
46680a96aa3bSJed Brown   } else {
46699566063dSJacob Faibussowitsch     PetscCall(DMLocalToGlobal(dmIn, vecOutLocal, INSERT_VALUES, vecOut));
46709566063dSJacob Faibussowitsch     PetscCall(DMRestoreLocalVector(dmIn, &vecOutLocal));
46710a96aa3bSJed Brown   }
46729566063dSJacob Faibussowitsch   PetscCall(PetscFree2(hierarchy, hierarchy_forest));
46733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
46740a96aa3bSJed Brown }
46750a96aa3bSJed Brown 
46760a96aa3bSJed Brown   #define DMForestTransferVec_pforest _append_pforest(DMForestTransferVec)
4677d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestTransferVec_pforest(DM dmIn, Vec vecIn, DM dmOut, Vec vecOut, PetscBool useBCs, PetscReal time)
4678d71ae5a4SJacob Faibussowitsch {
46790a96aa3bSJed Brown   DM          adaptIn, adaptOut, plexIn, plexOut;
46800a96aa3bSJed Brown   DM_Forest  *forestIn, *forestOut, *forestAdaptIn, *forestAdaptOut;
46810a96aa3bSJed Brown   PetscInt    dofPerDim[] = {1, 1, 1, 1};
46820a96aa3bSJed Brown   PetscSF     inSF = NULL, outSF = NULL;
46830a96aa3bSJed Brown   PetscInt   *inCids = NULL, *outCids = NULL;
46840a96aa3bSJed Brown   DMAdaptFlag purposeIn, purposeOut;
46850a96aa3bSJed Brown 
46860a96aa3bSJed Brown   PetscFunctionBegin;
46870a96aa3bSJed Brown   forestOut = (DM_Forest *)dmOut->data;
46880a96aa3bSJed Brown   forestIn  = (DM_Forest *)dmIn->data;
46890a96aa3bSJed Brown 
46909566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dmOut, &adaptOut));
46919566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityPurpose(dmOut, &purposeOut));
46920a96aa3bSJed Brown   forestAdaptOut = adaptOut ? (DM_Forest *)adaptOut->data : NULL;
46930a96aa3bSJed Brown 
46949566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityForest(dmIn, &adaptIn));
46959566063dSJacob Faibussowitsch   PetscCall(DMForestGetAdaptivityPurpose(dmIn, &purposeIn));
46960a96aa3bSJed Brown   forestAdaptIn = adaptIn ? (DM_Forest *)adaptIn->data : NULL;
46970a96aa3bSJed Brown 
46980a96aa3bSJed Brown   if (forestAdaptOut == forestIn) {
46990a96aa3bSJed Brown     switch (purposeOut) {
47000a96aa3bSJed Brown     case DM_ADAPT_REFINE:
47019566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmIn, dmOut, dofPerDim, &inSF, PETSC_TRUE, &inCids));
47029566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(inSF));
47030a96aa3bSJed Brown       break;
47040a96aa3bSJed Brown     case DM_ADAPT_COARSEN:
47050a96aa3bSJed Brown     case DM_ADAPT_COARSEN_LAST:
47069566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmOut, dmIn, dofPerDim, &outSF, PETSC_TRUE, &outCids));
47079566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(outSF));
47080a96aa3bSJed Brown       break;
47090a96aa3bSJed Brown     default:
47109566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmIn, dmOut, dofPerDim, &inSF, PETSC_TRUE, &inCids));
47119566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmOut, dmIn, dofPerDim, &outSF, PETSC_FALSE, &outCids));
47129566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(inSF));
47139566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(outSF));
47140a96aa3bSJed Brown     }
47150a96aa3bSJed Brown   } else if (forestAdaptIn == forestOut) {
47160a96aa3bSJed Brown     switch (purposeIn) {
47170a96aa3bSJed Brown     case DM_ADAPT_REFINE:
47189566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmOut, dmIn, dofPerDim, &outSF, PETSC_TRUE, &inCids));
47199566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(outSF));
47200a96aa3bSJed Brown       break;
47210a96aa3bSJed Brown     case DM_ADAPT_COARSEN:
47220a96aa3bSJed Brown     case DM_ADAPT_COARSEN_LAST:
47239566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmIn, dmOut, dofPerDim, &inSF, PETSC_TRUE, &inCids));
47249566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(inSF));
47250a96aa3bSJed Brown       break;
47260a96aa3bSJed Brown     default:
47279566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmIn, dmOut, dofPerDim, &inSF, PETSC_TRUE, &inCids));
47289566063dSJacob Faibussowitsch       PetscCall(DMPforestGetTransferSF_Internal(dmOut, dmIn, dofPerDim, &outSF, PETSC_FALSE, &outCids));
47299566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(inSF));
47309566063dSJacob Faibussowitsch       PetscCall(PetscSFSetUp(outSF));
47310a96aa3bSJed Brown     }
47320a96aa3bSJed Brown   } else SETERRQ(PetscObjectComm((PetscObject)dmIn), PETSC_ERR_SUP, "Only support transfer from pre-adaptivity to post-adaptivity right now");
47339566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dmIn, &plexIn));
47349566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dmOut, &plexOut));
47350a96aa3bSJed Brown 
47369566063dSJacob Faibussowitsch   PetscCall(DMPlexTransferVecTree(plexIn, vecIn, plexOut, vecOut, inSF, outSF, inCids, outCids, useBCs, time));
47379566063dSJacob Faibussowitsch   PetscCall(PetscFree(inCids));
47389566063dSJacob Faibussowitsch   PetscCall(PetscFree(outCids));
47399566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&inSF));
47409566063dSJacob Faibussowitsch   PetscCall(PetscSFDestroy(&outSF));
47419566063dSJacob Faibussowitsch   PetscCall(PetscFree(inCids));
47429566063dSJacob Faibussowitsch   PetscCall(PetscFree(outCids));
47433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47440a96aa3bSJed Brown }
47450a96aa3bSJed Brown 
47460a96aa3bSJed Brown   #define DMCreateCoordinateDM_pforest _append_pforest(DMCreateCoordinateDM)
4747d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateCoordinateDM_pforest(DM dm, DM *cdm)
4748d71ae5a4SJacob Faibussowitsch {
47490a96aa3bSJed Brown   DM plex;
47500a96aa3bSJed Brown 
47510a96aa3bSJed Brown   PetscFunctionBegin;
47520a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
47539566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
47549566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(plex, cdm));
47559566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)*cdm));
47563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47570a96aa3bSJed Brown }
47580a96aa3bSJed Brown 
47590a96aa3bSJed Brown   #define VecViewLocal_pforest _append_pforest(VecViewLocal)
4760d71ae5a4SJacob Faibussowitsch static PetscErrorCode VecViewLocal_pforest(Vec vec, PetscViewer viewer)
4761d71ae5a4SJacob Faibussowitsch {
47620a96aa3bSJed Brown   DM dm, plex;
47630a96aa3bSJed Brown 
47640a96aa3bSJed Brown   PetscFunctionBegin;
47659566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vec, &dm));
47669566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
47679566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, plex));
47689566063dSJacob Faibussowitsch   PetscCall(VecView_Plex_Local(vec, viewer));
47699566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, dm));
47703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47710a96aa3bSJed Brown }
47720a96aa3bSJed Brown 
47730a96aa3bSJed Brown   #define VecView_pforest _append_pforest(VecView)
4774d71ae5a4SJacob Faibussowitsch static PetscErrorCode VecView_pforest(Vec vec, PetscViewer viewer)
4775d71ae5a4SJacob Faibussowitsch {
47760a96aa3bSJed Brown   DM dm, plex;
47770a96aa3bSJed Brown 
47780a96aa3bSJed Brown   PetscFunctionBegin;
47799566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vec, &dm));
47809566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
47819566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, plex));
47829566063dSJacob Faibussowitsch   PetscCall(VecView_Plex(vec, viewer));
47839566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, dm));
47843ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47850a96aa3bSJed Brown }
47860a96aa3bSJed Brown 
47870a96aa3bSJed Brown   #define VecView_pforest_Native _infix_pforest(VecView, _Native)
4788d71ae5a4SJacob Faibussowitsch static PetscErrorCode VecView_pforest_Native(Vec vec, PetscViewer viewer)
4789d71ae5a4SJacob Faibussowitsch {
47900a96aa3bSJed Brown   DM dm, plex;
47910a96aa3bSJed Brown 
47920a96aa3bSJed Brown   PetscFunctionBegin;
47939566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vec, &dm));
47949566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
47959566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, plex));
47969566063dSJacob Faibussowitsch   PetscCall(VecView_Plex_Native(vec, viewer));
47979566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, dm));
47983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
47990a96aa3bSJed Brown }
48000a96aa3bSJed Brown 
48010a96aa3bSJed Brown   #define VecLoad_pforest _append_pforest(VecLoad)
4802d71ae5a4SJacob Faibussowitsch static PetscErrorCode VecLoad_pforest(Vec vec, PetscViewer viewer)
4803d71ae5a4SJacob Faibussowitsch {
48040a96aa3bSJed Brown   DM dm, plex;
48050a96aa3bSJed Brown 
48060a96aa3bSJed Brown   PetscFunctionBegin;
48079566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vec, &dm));
48089566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48099566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, plex));
48109566063dSJacob Faibussowitsch   PetscCall(VecLoad_Plex(vec, viewer));
48119566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, dm));
48123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48130a96aa3bSJed Brown }
48140a96aa3bSJed Brown 
48150a96aa3bSJed Brown   #define VecLoad_pforest_Native _infix_pforest(VecLoad, _Native)
4816d71ae5a4SJacob Faibussowitsch static PetscErrorCode VecLoad_pforest_Native(Vec vec, PetscViewer viewer)
4817d71ae5a4SJacob Faibussowitsch {
48180a96aa3bSJed Brown   DM dm, plex;
48190a96aa3bSJed Brown 
48200a96aa3bSJed Brown   PetscFunctionBegin;
48219566063dSJacob Faibussowitsch   PetscCall(VecGetDM(vec, &dm));
48229566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48239566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, plex));
48249566063dSJacob Faibussowitsch   PetscCall(VecLoad_Plex_Native(vec, viewer));
48259566063dSJacob Faibussowitsch   PetscCall(VecSetDM(vec, dm));
48263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48270a96aa3bSJed Brown }
48280a96aa3bSJed Brown 
48290a96aa3bSJed Brown   #define DMCreateGlobalVector_pforest _append_pforest(DMCreateGlobalVector)
4830d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateGlobalVector_pforest(DM dm, Vec *vec)
4831d71ae5a4SJacob Faibussowitsch {
48320a96aa3bSJed Brown   PetscFunctionBegin;
48339566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector_Section_Private(dm, vec));
48349566063dSJacob Faibussowitsch   /* PetscCall(VecSetOperation(*vec, VECOP_DUPLICATE, (void(*)(void)) VecDuplicate_MPI_DM)); */
48359566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecView_pforest));
48369566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEWNATIVE, (void (*)(void))VecView_pforest_Native));
48379566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOAD, (void (*)(void))VecLoad_pforest));
48389566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_LOADNATIVE, (void (*)(void))VecLoad_pforest_Native));
48393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48400a96aa3bSJed Brown }
48410a96aa3bSJed Brown 
48420a96aa3bSJed Brown   #define DMCreateLocalVector_pforest _append_pforest(DMCreateLocalVector)
4843d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateLocalVector_pforest(DM dm, Vec *vec)
4844d71ae5a4SJacob Faibussowitsch {
48450a96aa3bSJed Brown   PetscFunctionBegin;
48469566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector_Section_Private(dm, vec));
48479566063dSJacob Faibussowitsch   PetscCall(VecSetOperation(*vec, VECOP_VIEW, (void (*)(void))VecViewLocal_pforest));
48483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48490a96aa3bSJed Brown }
48500a96aa3bSJed Brown 
48510a96aa3bSJed Brown   #define DMCreateMatrix_pforest _append_pforest(DMCreateMatrix)
4852d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateMatrix_pforest(DM dm, Mat *mat)
4853d71ae5a4SJacob Faibussowitsch {
48540a96aa3bSJed Brown   DM plex;
48550a96aa3bSJed Brown 
48560a96aa3bSJed Brown   PetscFunctionBegin;
48570a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
48589566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48590a96aa3bSJed Brown   if (plex->prealloc_only != dm->prealloc_only) plex->prealloc_only = dm->prealloc_only; /* maybe this should go into forest->plex */
48609566063dSJacob Faibussowitsch   PetscCall(DMCreateMatrix(plex, mat));
48619566063dSJacob Faibussowitsch   PetscCall(MatSetDM(*mat, dm));
48623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48630a96aa3bSJed Brown }
48640a96aa3bSJed Brown 
48650a96aa3bSJed Brown   #define DMProjectFunctionLocal_pforest _append_pforest(DMProjectFunctionLocal)
4866d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMProjectFunctionLocal_pforest(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
4867d71ae5a4SJacob Faibussowitsch {
48680a96aa3bSJed Brown   DM plex;
48690a96aa3bSJed Brown 
48700a96aa3bSJed Brown   PetscFunctionBegin;
48710a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
48729566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48739566063dSJacob Faibussowitsch   PetscCall(DMProjectFunctionLocal(plex, time, funcs, ctxs, mode, localX));
48743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48750a96aa3bSJed Brown }
48760a96aa3bSJed Brown 
48770a96aa3bSJed Brown   #define DMProjectFunctionLabelLocal_pforest _append_pforest(DMProjectFunctionLabelLocal)
4878d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMProjectFunctionLabelLocal_pforest(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscInt Ncc, const PetscInt comps[], PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
4879d71ae5a4SJacob Faibussowitsch {
48800a96aa3bSJed Brown   DM plex;
48810a96aa3bSJed Brown 
48820a96aa3bSJed Brown   PetscFunctionBegin;
48830a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
48849566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48859566063dSJacob Faibussowitsch   PetscCall(DMProjectFunctionLabelLocal(plex, time, label, numIds, ids, Ncc, comps, funcs, ctxs, mode, localX));
48863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48870a96aa3bSJed Brown }
48880a96aa3bSJed Brown 
48890a96aa3bSJed Brown   #define DMProjectFieldLocal_pforest _append_pforest(DMProjectFieldLocal)
4890d71ae5a4SJacob Faibussowitsch PetscErrorCode DMProjectFieldLocal_pforest(DM dm, PetscReal time, Vec localU, void (**funcs)(PetscInt, PetscInt, PetscInt, const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), InsertMode mode, Vec localX)
4891d71ae5a4SJacob Faibussowitsch {
48920a96aa3bSJed Brown   DM plex;
48930a96aa3bSJed Brown 
48940a96aa3bSJed Brown   PetscFunctionBegin;
48950a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
48969566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
48979566063dSJacob Faibussowitsch   PetscCall(DMProjectFieldLocal(plex, time, localU, funcs, mode, localX));
48983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48990a96aa3bSJed Brown }
49000a96aa3bSJed Brown 
49010a96aa3bSJed Brown   #define DMComputeL2Diff_pforest _append_pforest(DMComputeL2Diff)
4902d71ae5a4SJacob Faibussowitsch PetscErrorCode DMComputeL2Diff_pforest(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff)
4903d71ae5a4SJacob Faibussowitsch {
49040a96aa3bSJed Brown   DM plex;
49050a96aa3bSJed Brown 
49060a96aa3bSJed Brown   PetscFunctionBegin;
49070a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
49089566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
49099566063dSJacob Faibussowitsch   PetscCall(DMComputeL2Diff(plex, time, funcs, ctxs, X, diff));
49103ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49110a96aa3bSJed Brown }
49120a96aa3bSJed Brown 
49130a96aa3bSJed Brown   #define DMComputeL2FieldDiff_pforest _append_pforest(DMComputeL2FieldDiff)
4914d71ae5a4SJacob Faibussowitsch PetscErrorCode DMComputeL2FieldDiff_pforest(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal diff[])
4915d71ae5a4SJacob Faibussowitsch {
49160a96aa3bSJed Brown   DM plex;
49170a96aa3bSJed Brown 
49180a96aa3bSJed Brown   PetscFunctionBegin;
49190a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
49209566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
49219566063dSJacob Faibussowitsch   PetscCall(DMComputeL2FieldDiff(plex, time, funcs, ctxs, X, diff));
49223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49230a96aa3bSJed Brown }
49240a96aa3bSJed Brown 
49250a96aa3bSJed Brown   #define DMCreatelocalsection_pforest _append_pforest(DMCreatelocalsection)
4926d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreatelocalsection_pforest(DM dm)
4927d71ae5a4SJacob Faibussowitsch {
49280a96aa3bSJed Brown   DM           plex;
49290a96aa3bSJed Brown   PetscSection section;
49300a96aa3bSJed Brown 
49310a96aa3bSJed Brown   PetscFunctionBegin;
49320a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
49339566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
49349566063dSJacob Faibussowitsch   PetscCall(DMGetLocalSection(plex, &section));
49359566063dSJacob Faibussowitsch   PetscCall(DMSetLocalSection(dm, section));
49363ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49370a96aa3bSJed Brown }
49380a96aa3bSJed Brown 
49390a96aa3bSJed Brown   #define DMCreateDefaultConstraints_pforest _append_pforest(DMCreateDefaultConstraints)
4940d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateDefaultConstraints_pforest(DM dm)
4941d71ae5a4SJacob Faibussowitsch {
49420a96aa3bSJed Brown   DM           plex;
49430a96aa3bSJed Brown   Mat          mat;
494479769bd5SJed Brown   Vec          bias;
49450a96aa3bSJed Brown   PetscSection section;
49460a96aa3bSJed Brown 
49470a96aa3bSJed Brown   PetscFunctionBegin;
49480a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
49499566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
49509566063dSJacob Faibussowitsch   PetscCall(DMGetDefaultConstraints(plex, &section, &mat, &bias));
49519566063dSJacob Faibussowitsch   PetscCall(DMSetDefaultConstraints(dm, section, mat, bias));
49523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49530a96aa3bSJed Brown }
49540a96aa3bSJed Brown 
49550a96aa3bSJed Brown   #define DMGetDimPoints_pforest _append_pforest(DMGetDimPoints)
4956d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMGetDimPoints_pforest(DM dm, PetscInt dim, PetscInt *cStart, PetscInt *cEnd)
4957d71ae5a4SJacob Faibussowitsch {
49580a96aa3bSJed Brown   DM plex;
49590a96aa3bSJed Brown 
49600a96aa3bSJed Brown   PetscFunctionBegin;
49610a96aa3bSJed Brown   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
49629566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
49639566063dSJacob Faibussowitsch   PetscCall(DMGetDimPoints(plex, dim, cStart, cEnd));
49643ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49650a96aa3bSJed Brown }
49660a96aa3bSJed Brown 
49670a96aa3bSJed Brown   /* Need to forward declare */
49680a96aa3bSJed Brown   #define DMInitialize_pforest _append_pforest(DMInitialize)
49690a96aa3bSJed Brown static PetscErrorCode DMInitialize_pforest(DM dm);
49700a96aa3bSJed Brown 
49710a96aa3bSJed Brown   #define DMClone_pforest _append_pforest(DMClone)
4972d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMClone_pforest(DM dm, DM *newdm)
4973d71ae5a4SJacob Faibussowitsch {
49740a96aa3bSJed Brown   PetscFunctionBegin;
49759566063dSJacob Faibussowitsch   PetscCall(DMClone_Forest(dm, newdm));
49769566063dSJacob Faibussowitsch   PetscCall(DMInitialize_pforest(*newdm));
49773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49780a96aa3bSJed Brown }
49790a96aa3bSJed Brown 
49800a96aa3bSJed Brown   #define DMForestCreateCellChart_pforest _append_pforest(DMForestCreateCellChart)
4981d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestCreateCellChart_pforest(DM dm, PetscInt *cStart, PetscInt *cEnd)
4982d71ae5a4SJacob Faibussowitsch {
49830a96aa3bSJed Brown   DM_Forest         *forest;
49840a96aa3bSJed Brown   DM_Forest_pforest *pforest;
49850a96aa3bSJed Brown   PetscInt           overlap;
49860a96aa3bSJed Brown 
49870a96aa3bSJed Brown   PetscFunctionBegin;
49889566063dSJacob Faibussowitsch   PetscCall(DMSetUp(dm));
49890a96aa3bSJed Brown   forest  = (DM_Forest *)dm->data;
49900a96aa3bSJed Brown   pforest = (DM_Forest_pforest *)forest->data;
49910a96aa3bSJed Brown   *cStart = 0;
49929566063dSJacob Faibussowitsch   PetscCall(DMForestGetPartitionOverlap(dm, &overlap));
49930a96aa3bSJed Brown   if (overlap && pforest->ghost) {
49940a96aa3bSJed Brown     *cEnd = pforest->forest->local_num_quadrants + pforest->ghost->proc_offsets[pforest->forest->mpisize];
49950a96aa3bSJed Brown   } else {
49960a96aa3bSJed Brown     *cEnd = pforest->forest->local_num_quadrants;
49970a96aa3bSJed Brown   }
49983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
49990a96aa3bSJed Brown }
50000a96aa3bSJed Brown 
50010a96aa3bSJed Brown   #define DMForestCreateCellSF_pforest _append_pforest(DMForestCreateCellSF)
5002d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMForestCreateCellSF_pforest(DM dm, PetscSF *cellSF)
5003d71ae5a4SJacob Faibussowitsch {
50040a96aa3bSJed Brown   DM_Forest         *forest;
50050a96aa3bSJed Brown   DM_Forest_pforest *pforest;
50060a96aa3bSJed Brown   PetscMPIInt        rank;
50070a96aa3bSJed Brown   PetscInt           overlap;
50080a96aa3bSJed Brown   PetscInt           cStart, cEnd, cLocalStart, cLocalEnd;
50090a96aa3bSJed Brown   PetscInt           nRoots, nLeaves, *mine = NULL;
50100a96aa3bSJed Brown   PetscSFNode       *remote = NULL;
50110a96aa3bSJed Brown   PetscSF            sf;
50120a96aa3bSJed Brown 
50130a96aa3bSJed Brown   PetscFunctionBegin;
50149566063dSJacob Faibussowitsch   PetscCall(DMForestGetCellChart(dm, &cStart, &cEnd));
50150a96aa3bSJed Brown   forest      = (DM_Forest *)dm->data;
50160a96aa3bSJed Brown   pforest     = (DM_Forest_pforest *)forest->data;
50170a96aa3bSJed Brown   nRoots      = cEnd - cStart;
50180a96aa3bSJed Brown   cLocalStart = pforest->cLocalStart;
50190a96aa3bSJed Brown   cLocalEnd   = pforest->cLocalEnd;
50200a96aa3bSJed Brown   nLeaves     = 0;
50219566063dSJacob Faibussowitsch   PetscCall(DMForestGetPartitionOverlap(dm, &overlap));
50229566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
50230a96aa3bSJed Brown   if (overlap && pforest->ghost) {
50240a96aa3bSJed Brown     PetscSFNode      *mirror;
50250a96aa3bSJed Brown     p4est_quadrant_t *mirror_array;
50260a96aa3bSJed Brown     PetscInt          nMirror, nGhostPre, nSelf, q;
50270a96aa3bSJed Brown     void            **mirrorPtrs;
50280a96aa3bSJed Brown 
50290a96aa3bSJed Brown     nMirror   = (PetscInt)pforest->ghost->mirrors.elem_count;
50300a96aa3bSJed Brown     nSelf     = cLocalEnd - cLocalStart;
50310a96aa3bSJed Brown     nLeaves   = nRoots - nSelf;
50320a96aa3bSJed Brown     nGhostPre = (PetscInt)pforest->ghost->proc_offsets[rank];
50339566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nLeaves, &mine));
50349566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(nLeaves, &remote));
50359566063dSJacob Faibussowitsch     PetscCall(PetscMalloc2(nMirror, &mirror, nMirror, &mirrorPtrs));
50360a96aa3bSJed Brown     mirror_array = (p4est_quadrant_t *)pforest->ghost->mirrors.array;
50370a96aa3bSJed Brown     for (q = 0; q < nMirror; q++) {
50380a96aa3bSJed Brown       p4est_quadrant_t *mir = &(mirror_array[q]);
50390a96aa3bSJed Brown 
50400a96aa3bSJed Brown       mirror[q].rank  = rank;
50410a96aa3bSJed Brown       mirror[q].index = (PetscInt)mir->p.piggy3.local_num + cLocalStart;
50420a96aa3bSJed Brown       mirrorPtrs[q]   = (void *)&(mirror[q]);
50430a96aa3bSJed Brown     }
5044792fecdfSBarry Smith     PetscCallP4est(p4est_ghost_exchange_custom, (pforest->forest, pforest->ghost, sizeof(PetscSFNode), mirrorPtrs, remote));
50459566063dSJacob Faibussowitsch     PetscCall(PetscFree2(mirror, mirrorPtrs));
50460a96aa3bSJed Brown     for (q = 0; q < nGhostPre; q++) mine[q] = q;
50470a96aa3bSJed Brown     for (; q < nLeaves; q++) mine[q] = (q - nGhostPre) + cLocalEnd;
50480a96aa3bSJed Brown   }
50499566063dSJacob Faibussowitsch   PetscCall(PetscSFCreate(PetscObjectComm((PetscObject)dm), &sf));
50509566063dSJacob Faibussowitsch   PetscCall(PetscSFSetGraph(sf, nRoots, nLeaves, mine, PETSC_OWN_POINTER, remote, PETSC_OWN_POINTER));
50510a96aa3bSJed Brown   *cellSF = sf;
50523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
50530a96aa3bSJed Brown }
50540a96aa3bSJed Brown 
5055d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMCreateNeumannOverlap_pforest(DM dm, IS *ovl, Mat *J, PetscErrorCode (**setup)(Mat, PetscReal, Vec, Vec, PetscReal, IS, void *), void **setup_ctx)
5056d71ae5a4SJacob Faibussowitsch {
50570a96aa3bSJed Brown   DM plex;
50580a96aa3bSJed Brown 
50590a96aa3bSJed Brown   PetscFunctionBegin;
50609566063dSJacob Faibussowitsch   PetscCall(DMPforestGetPlex(dm, &plex));
50619566063dSJacob Faibussowitsch   PetscCall(DMCreateNeumannOverlap_Plex(plex, ovl, J, setup, setup_ctx));
50620a96aa3bSJed Brown   if (!*setup) {
50639566063dSJacob Faibussowitsch     PetscCall(PetscObjectQueryFunction((PetscObject)dm, "MatComputeNeumannOverlap_C", setup));
506448a46eb9SPierre Jolivet     if (*setup) PetscCall(PetscObjectCompose((PetscObject)*ovl, "_DM_Original_HPDDM", (PetscObject)dm));
50650a96aa3bSJed Brown   }
50663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
50670a96aa3bSJed Brown }
50680a96aa3bSJed Brown 
5069d71ae5a4SJacob Faibussowitsch static PetscErrorCode DMInitialize_pforest(DM dm)
5070d71ae5a4SJacob Faibussowitsch {
50710a96aa3bSJed Brown   PetscFunctionBegin;
50720a96aa3bSJed Brown   dm->ops->setup                     = DMSetUp_pforest;
50730a96aa3bSJed Brown   dm->ops->view                      = DMView_pforest;
50740a96aa3bSJed Brown   dm->ops->clone                     = DMClone_pforest;
50750a96aa3bSJed Brown   dm->ops->createinterpolation       = DMCreateInterpolation_pforest;
50760a96aa3bSJed Brown   dm->ops->createinjection           = DMCreateInjection_pforest;
50770a96aa3bSJed Brown   dm->ops->setfromoptions            = DMSetFromOptions_pforest;
50780a96aa3bSJed Brown   dm->ops->createcoordinatedm        = DMCreateCoordinateDM_pforest;
50790a96aa3bSJed Brown   dm->ops->createglobalvector        = DMCreateGlobalVector_pforest;
50800a96aa3bSJed Brown   dm->ops->createlocalvector         = DMCreateLocalVector_pforest;
50810a96aa3bSJed Brown   dm->ops->creatematrix              = DMCreateMatrix_pforest;
50820a96aa3bSJed Brown   dm->ops->projectfunctionlocal      = DMProjectFunctionLocal_pforest;
50830a96aa3bSJed Brown   dm->ops->projectfunctionlabellocal = DMProjectFunctionLabelLocal_pforest;
50840a96aa3bSJed Brown   dm->ops->projectfieldlocal         = DMProjectFieldLocal_pforest;
50850a96aa3bSJed Brown   dm->ops->createlocalsection        = DMCreatelocalsection_pforest;
50860a96aa3bSJed Brown   dm->ops->createdefaultconstraints  = DMCreateDefaultConstraints_pforest;
50870a96aa3bSJed Brown   dm->ops->computel2diff             = DMComputeL2Diff_pforest;
50880a96aa3bSJed Brown   dm->ops->computel2fielddiff        = DMComputeL2FieldDiff_pforest;
50890a96aa3bSJed Brown   dm->ops->getdimpoints              = DMGetDimPoints_pforest;
50900a96aa3bSJed Brown 
50919566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, PetscStringize(DMConvert_plex_pforest) "_C", DMConvert_plex_pforest));
50929566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, PetscStringize(DMConvert_pforest_plex) "_C", DMConvert_pforest_plex));
50939566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMCreateNeumannOverlap_C", DMCreateNeumannOverlap_pforest));
50949566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexGetOverlap_C", DMForestGetPartitionOverlap));
50953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
50960a96aa3bSJed Brown }
50970a96aa3bSJed Brown 
50980a96aa3bSJed Brown   #define DMCreate_pforest _append_pforest(DMCreate)
5099d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode DMCreate_pforest(DM dm)
5100d71ae5a4SJacob Faibussowitsch {
51010a96aa3bSJed Brown   DM_Forest         *forest;
51020a96aa3bSJed Brown   DM_Forest_pforest *pforest;
51030a96aa3bSJed Brown 
51040a96aa3bSJed Brown   PetscFunctionBegin;
51059566063dSJacob Faibussowitsch   PetscCall(PetscP4estInitialize());
51069566063dSJacob Faibussowitsch   PetscCall(DMCreate_Forest(dm));
51079566063dSJacob Faibussowitsch   PetscCall(DMInitialize_pforest(dm));
51089566063dSJacob Faibussowitsch   PetscCall(DMSetDimension(dm, P4EST_DIM));
51090a96aa3bSJed Brown 
51100a96aa3bSJed Brown   /* set forest defaults */
51119566063dSJacob Faibussowitsch   PetscCall(DMForestSetTopology(dm, "unit"));
51129566063dSJacob Faibussowitsch   PetscCall(DMForestSetMinimumRefinement(dm, 0));
51139566063dSJacob Faibussowitsch   PetscCall(DMForestSetInitialRefinement(dm, 0));
51149566063dSJacob Faibussowitsch   PetscCall(DMForestSetMaximumRefinement(dm, P4EST_QMAXLEVEL));
51159566063dSJacob Faibussowitsch   PetscCall(DMForestSetGradeFactor(dm, 2));
51169566063dSJacob Faibussowitsch   PetscCall(DMForestSetAdjacencyDimension(dm, 0));
51179566063dSJacob Faibussowitsch   PetscCall(DMForestSetPartitionOverlap(dm, 0));
51180a96aa3bSJed Brown 
51190a96aa3bSJed Brown   /* create p4est data */
51204dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&pforest));
51210a96aa3bSJed Brown 
51220a96aa3bSJed Brown   forest                            = (DM_Forest *)dm->data;
51230a96aa3bSJed Brown   forest->data                      = pforest;
51240a96aa3bSJed Brown   forest->destroy                   = DMForestDestroy_pforest;
51250a96aa3bSJed Brown   forest->ftemplate                 = DMForestTemplate_pforest;
51260a96aa3bSJed Brown   forest->transfervec               = DMForestTransferVec_pforest;
51270a96aa3bSJed Brown   forest->transfervecfrombase       = DMForestTransferVecFromBase_pforest;
51280a96aa3bSJed Brown   forest->createcellchart           = DMForestCreateCellChart_pforest;
51290a96aa3bSJed Brown   forest->createcellsf              = DMForestCreateCellSF_pforest;
51300a96aa3bSJed Brown   forest->clearadaptivityforest     = DMForestClearAdaptivityForest_pforest;
51310a96aa3bSJed Brown   forest->getadaptivitysuccess      = DMForestGetAdaptivitySuccess_pforest;
51320a96aa3bSJed Brown   pforest->topo                     = NULL;
51330a96aa3bSJed Brown   pforest->forest                   = NULL;
51340a96aa3bSJed Brown   pforest->ghost                    = NULL;
51350a96aa3bSJed Brown   pforest->lnodes                   = NULL;
51360a96aa3bSJed Brown   pforest->partition_for_coarsening = PETSC_TRUE;
51370a96aa3bSJed Brown   pforest->coarsen_hierarchy        = PETSC_FALSE;
51380a96aa3bSJed Brown   pforest->cLocalStart              = -1;
51390a96aa3bSJed Brown   pforest->cLocalEnd                = -1;
51400a96aa3bSJed Brown   pforest->labelsFinalized          = PETSC_FALSE;
51410a96aa3bSJed Brown   pforest->ghostName                = NULL;
51423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
51430a96aa3bSJed Brown }
51440a96aa3bSJed Brown 
51450a96aa3bSJed Brown #endif /* defined(PETSC_HAVE_P4EST) */
5146