xref: /petsc/src/dm/impls/plex/plextree.c (revision 947b95d86705d3df95f59e9a72133be5fe596209)
1af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
2d6a7ad0dSToby Isaac #include <../src/sys/utils/hash.h>
3af0996ceSBarry Smith #include <petsc/private/isimpl.h>
4af0996ceSBarry Smith #include <petsc/private/petscfeimpl.h>
5d6a7ad0dSToby Isaac #include <petscsf.h>
60c37af3bSToby Isaac #include <petscds.h>
7d6a7ad0dSToby Isaac 
8d6a7ad0dSToby Isaac /** hierarchy routines */
9d6a7ad0dSToby Isaac 
10d6a7ad0dSToby Isaac #undef __FUNCT__
11d6a7ad0dSToby Isaac #define __FUNCT__ "DMPlexSetReferenceTree"
12d6a7ad0dSToby Isaac /*@
13d6a7ad0dSToby Isaac   DMPlexSetReferenceTree - set the reference tree for hierarchically non-conforming meshes.
14d6a7ad0dSToby Isaac 
15d6a7ad0dSToby Isaac   Not collective
16d6a7ad0dSToby Isaac 
17d6a7ad0dSToby Isaac   Input Parameters:
18d6a7ad0dSToby Isaac + dm - The DMPlex object
19d6a7ad0dSToby Isaac - ref - The reference tree DMPlex object
20d6a7ad0dSToby Isaac 
210b7167a0SToby Isaac   Level: intermediate
22d6a7ad0dSToby Isaac 
23da43764aSToby Isaac .seealso: DMPlexGetReferenceTree(), DMPlexCreateDefaultReferenceTree()
24d6a7ad0dSToby Isaac @*/
25d6a7ad0dSToby Isaac PetscErrorCode DMPlexSetReferenceTree(DM dm, DM ref)
26d6a7ad0dSToby Isaac {
27d6a7ad0dSToby Isaac   DM_Plex        *mesh = (DM_Plex *)dm->data;
28d6a7ad0dSToby Isaac   PetscErrorCode  ierr;
29d6a7ad0dSToby Isaac 
30d6a7ad0dSToby Isaac   PetscFunctionBegin;
31d6a7ad0dSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3247a1df27SMatthew G. Knepley   if (ref) {PetscValidHeaderSpecific(ref, DM_CLASSID, 2);}
33d6a7ad0dSToby Isaac   ierr = PetscObjectReference((PetscObject)ref);CHKERRQ(ierr);
34d6a7ad0dSToby Isaac   ierr = DMDestroy(&mesh->referenceTree);CHKERRQ(ierr);
35d6a7ad0dSToby Isaac   mesh->referenceTree = ref;
36d6a7ad0dSToby Isaac   PetscFunctionReturn(0);
37d6a7ad0dSToby Isaac }
38d6a7ad0dSToby Isaac 
39d6a7ad0dSToby Isaac #undef __FUNCT__
40d6a7ad0dSToby Isaac #define __FUNCT__ "DMPlexGetReferenceTree"
41d6a7ad0dSToby Isaac /*@
42d6a7ad0dSToby Isaac   DMPlexGetReferenceTree - get the reference tree for hierarchically non-conforming meshes.
43d6a7ad0dSToby Isaac 
44d6a7ad0dSToby Isaac   Not collective
45d6a7ad0dSToby Isaac 
46d6a7ad0dSToby Isaac   Input Parameters:
47d6a7ad0dSToby Isaac . dm - The DMPlex object
48d6a7ad0dSToby Isaac 
49d6a7ad0dSToby Isaac   Output Parameters
50d6a7ad0dSToby Isaac . ref - The reference tree DMPlex object
51d6a7ad0dSToby Isaac 
520b7167a0SToby Isaac   Level: intermediate
53d6a7ad0dSToby Isaac 
54da43764aSToby Isaac .seealso: DMPlexSetReferenceTree(), DMPlexCreateDefaultReferenceTree()
55d6a7ad0dSToby Isaac @*/
56d6a7ad0dSToby Isaac PetscErrorCode DMPlexGetReferenceTree(DM dm, DM *ref)
57d6a7ad0dSToby Isaac {
58d6a7ad0dSToby Isaac   DM_Plex        *mesh = (DM_Plex *)dm->data;
59d6a7ad0dSToby Isaac 
60d6a7ad0dSToby Isaac   PetscFunctionBegin;
61d6a7ad0dSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
62d6a7ad0dSToby Isaac   PetscValidPointer(ref,2);
63d6a7ad0dSToby Isaac   *ref = mesh->referenceTree;
64d6a7ad0dSToby Isaac   PetscFunctionReturn(0);
65d6a7ad0dSToby Isaac }
66d6a7ad0dSToby Isaac 
67dcbd3bf7SToby Isaac #undef __FUNCT__
68dcbd3bf7SToby Isaac #define __FUNCT__ "DMPlexReferenceTreeGetChildSymmetry_Default"
69dcbd3bf7SToby Isaac static PetscErrorCode DMPlexReferenceTreeGetChildSymmetry_Default(DM dm, PetscInt parent, PetscInt parentOrientA, PetscInt childOrientA, PetscInt childA, PetscInt parentOrientB, PetscInt *childOrientB, PetscInt *childB)
70dcbd3bf7SToby Isaac {
71dcbd3bf7SToby Isaac   PetscInt       coneSize, dStart, dEnd, dim, ABswap, oAvert, oBvert, ABswapVert;
72dcbd3bf7SToby Isaac   PetscErrorCode ierr;
73dcbd3bf7SToby Isaac 
74dcbd3bf7SToby Isaac   PetscFunctionBegin;
75dcbd3bf7SToby Isaac   if (parentOrientA == parentOrientB) {
76dcbd3bf7SToby Isaac     if (childOrientB) *childOrientB = childOrientA;
77dcbd3bf7SToby Isaac     if (childB) *childB = childA;
78dcbd3bf7SToby Isaac     PetscFunctionReturn(0);
79dcbd3bf7SToby Isaac   }
80dcbd3bf7SToby Isaac   for (dim = 0; dim < 3; dim++) {
81dcbd3bf7SToby Isaac     ierr = DMPlexGetDepthStratum(dm,dim,&dStart,&dEnd);CHKERRQ(ierr);
82dcbd3bf7SToby Isaac     if (parent >= dStart && parent <= dEnd) {
83dcbd3bf7SToby Isaac       break;
84dcbd3bf7SToby Isaac     }
85dcbd3bf7SToby Isaac   }
86dcbd3bf7SToby Isaac   if (dim > 2) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot perform child symmetry for %d-cells",dim);
87dcbd3bf7SToby Isaac   if (!dim) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"A vertex has no children");
88dcbd3bf7SToby Isaac   if (childA < dStart || childA >= dEnd) {
89dcbd3bf7SToby Isaac     /* this is a lower-dimensional child: bootstrap */
90dcbd3bf7SToby Isaac     PetscInt size, i, sA = -1, sB, sOrientB, sConeSize;
91dcbd3bf7SToby Isaac     const PetscInt *supp, *coneA, *coneB, *oA, *oB;
92dcbd3bf7SToby Isaac 
93dcbd3bf7SToby Isaac     ierr = DMPlexGetSupportSize(dm,childA,&size);CHKERRQ(ierr);
94dcbd3bf7SToby Isaac     ierr = DMPlexGetSupport(dm,childA,&supp);CHKERRQ(ierr);
95dcbd3bf7SToby Isaac 
96dcbd3bf7SToby Isaac     /* find a point sA in supp(childA) that has the same parent */
97dcbd3bf7SToby Isaac     for (i = 0; i < size; i++) {
98dcbd3bf7SToby Isaac       PetscInt sParent;
99dcbd3bf7SToby Isaac 
100dcbd3bf7SToby Isaac       sA   = supp[i];
101dcbd3bf7SToby Isaac       if (sA == parent) continue;
102dcbd3bf7SToby Isaac       ierr = DMPlexGetTreeParent(dm,sA,&sParent,NULL);CHKERRQ(ierr);
103dcbd3bf7SToby Isaac       if (sParent == parent) {
104dcbd3bf7SToby Isaac         break;
105dcbd3bf7SToby Isaac       }
106dcbd3bf7SToby Isaac     }
107dcbd3bf7SToby Isaac     if (i == size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"could not find support in children");
108dcbd3bf7SToby Isaac     /* find out which point sB is in an equivalent position to sA under
109dcbd3bf7SToby Isaac      * parentOrientB */
110dcbd3bf7SToby Isaac     ierr = DMPlexReferenceTreeGetChildSymmetry_Default(dm,parent,parentOrientA,0,sA,parentOrientB,&sOrientB,&sB);CHKERRQ(ierr);
111dcbd3bf7SToby Isaac     ierr = DMPlexGetConeSize(dm,sA,&sConeSize);CHKERRQ(ierr);
112dcbd3bf7SToby Isaac     ierr = DMPlexGetCone(dm,sA,&coneA);CHKERRQ(ierr);
113dcbd3bf7SToby Isaac     ierr = DMPlexGetCone(dm,sB,&coneB);CHKERRQ(ierr);
114dcbd3bf7SToby Isaac     ierr = DMPlexGetConeOrientation(dm,sA,&oA);CHKERRQ(ierr);
115dcbd3bf7SToby Isaac     ierr = DMPlexGetConeOrientation(dm,sB,&oB);CHKERRQ(ierr);
116dcbd3bf7SToby Isaac     /* step through the cone of sA in natural order */
117dcbd3bf7SToby Isaac     for (i = 0; i < sConeSize; i++) {
118dcbd3bf7SToby Isaac       if (coneA[i] == childA) {
119dcbd3bf7SToby Isaac         /* if childA is at position i in coneA,
120dcbd3bf7SToby Isaac          * then we want the point that is at sOrientB*i in coneB */
121dcbd3bf7SToby Isaac         PetscInt j = (sOrientB >= 0) ? ((sOrientB + i) % sConeSize) : ((sConeSize -(sOrientB+1) - i) % sConeSize);
122dcbd3bf7SToby Isaac         if (childB) *childB = coneB[j];
123dcbd3bf7SToby Isaac         if (childOrientB) {
124dcbd3bf7SToby Isaac           PetscInt oBtrue;
125dcbd3bf7SToby Isaac 
126dcbd3bf7SToby Isaac           ierr          = DMPlexGetConeSize(dm,childA,&coneSize);CHKERRQ(ierr);
127dcbd3bf7SToby Isaac           /* compose sOrientB and oB[j] */
128dcbd3bf7SToby Isaac           if (coneSize != 0 && coneSize != 2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expected a vertex or an edge");
129dcbd3bf7SToby Isaac           /* we may have to flip an edge */
130dcbd3bf7SToby Isaac           oBtrue        = coneSize ? ((sOrientB >= 0) ? oB[j] : -(oB[j] + 2)) : 0;
131dcbd3bf7SToby Isaac           ABswap        = DihedralSwap(coneSize,oA[i],oBtrue);CHKERRQ(ierr);
132dcbd3bf7SToby Isaac           *childOrientB = DihedralCompose(coneSize,childOrientA,ABswap);
133dcbd3bf7SToby Isaac         }
134dcbd3bf7SToby Isaac         break;
135dcbd3bf7SToby Isaac       }
136dcbd3bf7SToby Isaac     }
137dcbd3bf7SToby Isaac     if (i == sConeSize) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"support cone mismatch");
138dcbd3bf7SToby Isaac     PetscFunctionReturn(0);
139dcbd3bf7SToby Isaac   }
140dcbd3bf7SToby Isaac   /* get the cone size and symmetry swap */
141dcbd3bf7SToby Isaac   ierr   = DMPlexGetConeSize(dm,parent,&coneSize);CHKERRQ(ierr);
142dcbd3bf7SToby Isaac   ABswap = DihedralSwap(coneSize, parentOrientA, parentOrientB);
143dcbd3bf7SToby Isaac   if (dim == 2) {
144dcbd3bf7SToby Isaac     /* orientations refer to cones: we want them to refer to vertices:
145dcbd3bf7SToby Isaac      * if it's a rotation, they are the same, but if the order is reversed, a
146dcbd3bf7SToby Isaac      * permutation that puts side i first does *not* put vertex i first */
147dcbd3bf7SToby Isaac     oAvert     = (parentOrientA >= 0) ? parentOrientA : -((-parentOrientA % coneSize) + 1);
148dcbd3bf7SToby Isaac     oBvert     = (parentOrientB >= 0) ? parentOrientB : -((-parentOrientB % coneSize) + 1);
149dcbd3bf7SToby Isaac     ABswapVert = DihedralSwap(coneSize, oAvert, oBvert);
150*947b95d8SBarry Smith   } else {
151dcbd3bf7SToby Isaac     ABswapVert = ABswap;
152dcbd3bf7SToby Isaac   }
153dcbd3bf7SToby Isaac   if (childB) {
154dcbd3bf7SToby Isaac     /* assume that each child corresponds to a vertex, in the same order */
155dcbd3bf7SToby Isaac     PetscInt p, posA = -1, numChildren, i;
156dcbd3bf7SToby Isaac     const PetscInt *children;
157dcbd3bf7SToby Isaac 
158dcbd3bf7SToby Isaac     /* count which position the child is in */
159dcbd3bf7SToby Isaac     ierr = DMPlexGetTreeChildren(dm,parent,&numChildren,&children);CHKERRQ(ierr);
160dcbd3bf7SToby Isaac     for (i = 0; i < numChildren; i++) {
161dcbd3bf7SToby Isaac       p = children[i];
162dcbd3bf7SToby Isaac       if (p == childA) {
163dcbd3bf7SToby Isaac         posA = i;
164dcbd3bf7SToby Isaac         break;
165dcbd3bf7SToby Isaac       }
166dcbd3bf7SToby Isaac     }
167dcbd3bf7SToby Isaac     if (posA >= coneSize) {
168dcbd3bf7SToby Isaac       /* this is the triangle in the middle of a uniformly refined triangle: it
169dcbd3bf7SToby Isaac        * is invariant */
170dcbd3bf7SToby Isaac       if (dim != 2 || posA != 3) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Expected a middle triangle, got something else");
171dcbd3bf7SToby Isaac       *childB = childA;
172dcbd3bf7SToby Isaac     }
173dcbd3bf7SToby Isaac     else {
174dcbd3bf7SToby Isaac       /* figure out position B by applying ABswapVert */
175dcbd3bf7SToby Isaac       PetscInt posB;
176dcbd3bf7SToby Isaac 
177dcbd3bf7SToby Isaac       posB = (ABswapVert >= 0) ? ((ABswapVert + posA) % coneSize) : ((coneSize -(ABswapVert + 1) - posA) % coneSize);
178dcbd3bf7SToby Isaac       if (childB) *childB = children[posB];
179dcbd3bf7SToby Isaac     }
180dcbd3bf7SToby Isaac   }
181dcbd3bf7SToby Isaac   if (childOrientB) *childOrientB = DihedralCompose(coneSize,childOrientA,ABswap);
182dcbd3bf7SToby Isaac   PetscFunctionReturn(0);
183dcbd3bf7SToby Isaac }
184dcbd3bf7SToby Isaac 
185dcbd3bf7SToby Isaac #undef __FUNCT__
186dcbd3bf7SToby Isaac #define __FUNCT__ "DMPlexReferenceTreeGetChildSymmetry"
187dcbd3bf7SToby Isaac /*@
188dcbd3bf7SToby Isaac   DMPlexReferenceTreeGetChildSymmetry - Given a reference tree, transform a childid and orientation from one parent frame to another
189dcbd3bf7SToby Isaac 
190dcbd3bf7SToby Isaac   Input Parameters:
191dcbd3bf7SToby Isaac + dm - the reference tree DMPlex object
192dcbd3bf7SToby Isaac . parent - the parent point
193dcbd3bf7SToby Isaac . parentOrientA - the reference orientation for describing the parent
194dcbd3bf7SToby Isaac . childOrientA - the reference orientation for describing the child
195dcbd3bf7SToby Isaac . childA - the reference childID for describing the child
196dcbd3bf7SToby Isaac - parentOrientB - the new orientation for describing the parent
197dcbd3bf7SToby Isaac 
198dcbd3bf7SToby Isaac   Output Parameters:
199dcbd3bf7SToby Isaac + childOrientB - if not NULL, set to the new oreintation for describing the child
200dcbd3bf7SToby Isaac . childB - if not NULL, the new childID for describing the child
201dcbd3bf7SToby Isaac 
202dcbd3bf7SToby Isaac   Level: developer
203dcbd3bf7SToby Isaac 
204dcbd3bf7SToby Isaac .seealso: DMPlexGetReferenceTree(), DMPlexSetReferenceTree(), DMPlexSetTree()
205dcbd3bf7SToby Isaac @*/
206dcbd3bf7SToby Isaac PetscErrorCode DMPlexReferenceTreeGetChildSymmetry(DM dm, PetscInt parent, PetscInt parentOrientA, PetscInt childOrientA, PetscInt childA, PetscInt parentOrientB, PetscInt *childOrientB, PetscInt *childB)
207dcbd3bf7SToby Isaac {
208dcbd3bf7SToby Isaac   DM_Plex        *mesh = (DM_Plex *)dm->data;
209dcbd3bf7SToby Isaac   PetscErrorCode ierr;
210dcbd3bf7SToby Isaac 
211dcbd3bf7SToby Isaac   PetscFunctionBegin;
212dcbd3bf7SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
213dcbd3bf7SToby Isaac   if (!mesh->getchildsymmetry) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"DMPlexReferenceTreeGetChildSymmetry not implemented");
214dcbd3bf7SToby Isaac   ierr = mesh->getchildsymmetry(dm,parent,parentOrientA,childOrientA,childA,parentOrientB,childOrientB,childB);CHKERRQ(ierr);
215dcbd3bf7SToby Isaac   PetscFunctionReturn(0);
216dcbd3bf7SToby Isaac }
217dcbd3bf7SToby Isaac 
218776742edSToby Isaac static PetscErrorCode DMPlexSetTree_Internal(DM,PetscSection,PetscInt*,PetscInt*,PetscBool,PetscBool);
219f9f063d4SToby Isaac 
220da43764aSToby Isaac #undef __FUNCT__
2210e2cc29aSToby Isaac #define __FUNCT__ "DMPlexCreateReferenceTree_Union"
2220e2cc29aSToby Isaac PetscErrorCode DMPlexCreateReferenceTree_Union(DM K, DM Kref, const char *labelName, DM *ref)
223da43764aSToby Isaac {
2240e2cc29aSToby Isaac   MPI_Comm       comm;
2250e2cc29aSToby Isaac   PetscInt       dim, p, pStart, pEnd, pRefStart, pRefEnd, d, offset, parentSize, *parents, *childIDs;
226da43764aSToby Isaac   PetscInt      *permvals, *unionCones, *coneSizes, *unionOrientations, numUnionPoints, *numDimPoints, numCones, numVerts;
227da43764aSToby Isaac   DMLabel        identity, identityRef;
22810f7e118SToby Isaac   PetscSection   unionSection, unionConeSection, parentSection;
229da43764aSToby Isaac   PetscScalar   *unionCoords;
230da43764aSToby Isaac   IS             perm;
231da43764aSToby Isaac   PetscErrorCode ierr;
232da43764aSToby Isaac 
233da43764aSToby Isaac   PetscFunctionBegin;
2340e2cc29aSToby Isaac   comm = PetscObjectComm((PetscObject)K);
2350e2cc29aSToby Isaac   ierr = DMGetDimension(K, &dim);CHKERRQ(ierr);
236da43764aSToby Isaac   ierr = DMPlexGetChart(K, &pStart, &pEnd);CHKERRQ(ierr);
2370e2cc29aSToby Isaac   ierr = DMGetLabel(K, labelName, &identity);CHKERRQ(ierr);
2380e2cc29aSToby Isaac   ierr = DMGetLabel(Kref, labelName, &identityRef);CHKERRQ(ierr);
239da43764aSToby Isaac   ierr = DMPlexGetChart(Kref, &pRefStart, &pRefEnd);CHKERRQ(ierr);
240da43764aSToby Isaac   ierr = PetscSectionCreate(comm, &unionSection);CHKERRQ(ierr);
241da43764aSToby Isaac   ierr = PetscSectionSetChart(unionSection, 0, (pEnd - pStart) + (pRefEnd - pRefStart));CHKERRQ(ierr);
242da43764aSToby Isaac   /* count points that will go in the union */
243da43764aSToby Isaac   for (p = pStart; p < pEnd; p++) {
244da43764aSToby Isaac     ierr = PetscSectionSetDof(unionSection, p - pStart, 1);CHKERRQ(ierr);
245da43764aSToby Isaac   }
246da43764aSToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
247da43764aSToby Isaac     PetscInt q, qSize;
248da43764aSToby Isaac     ierr = DMLabelGetValue(identityRef, p, &q);CHKERRQ(ierr);
249da43764aSToby Isaac     ierr = DMLabelGetStratumSize(identityRef, q, &qSize);CHKERRQ(ierr);
250da43764aSToby Isaac     if (qSize > 1) {
251da43764aSToby Isaac       ierr = PetscSectionSetDof(unionSection, p - pRefStart + (pEnd - pStart), 1);CHKERRQ(ierr);
252da43764aSToby Isaac     }
253da43764aSToby Isaac   }
254854ce69bSBarry Smith   ierr = PetscMalloc1(pEnd - pStart + pRefEnd - pRefStart,&permvals);CHKERRQ(ierr);
255da43764aSToby Isaac   offset = 0;
256da43764aSToby Isaac   /* stratify points in the union by topological dimension */
257da43764aSToby Isaac   for (d = 0; d <= dim; d++) {
258da43764aSToby Isaac     PetscInt cStart, cEnd, c;
259da43764aSToby Isaac 
260da43764aSToby Isaac     ierr = DMPlexGetHeightStratum(K, d, &cStart, &cEnd);CHKERRQ(ierr);
261da43764aSToby Isaac     for (c = cStart; c < cEnd; c++) {
262da43764aSToby Isaac       permvals[offset++] = c;
263da43764aSToby Isaac     }
264da43764aSToby Isaac 
265da43764aSToby Isaac     ierr = DMPlexGetHeightStratum(Kref, d, &cStart, &cEnd);CHKERRQ(ierr);
266da43764aSToby Isaac     for (c = cStart; c < cEnd; c++) {
267da43764aSToby Isaac       permvals[offset++] = c + (pEnd - pStart);
268da43764aSToby Isaac     }
269da43764aSToby Isaac   }
270da43764aSToby Isaac   ierr = ISCreateGeneral(comm, (pEnd - pStart) + (pRefEnd - pRefStart), permvals, PETSC_OWN_POINTER, &perm);CHKERRQ(ierr);
271da43764aSToby Isaac   ierr = PetscSectionSetPermutation(unionSection,perm);CHKERRQ(ierr);
272da43764aSToby Isaac   ierr = PetscSectionSetUp(unionSection);CHKERRQ(ierr);
273da43764aSToby Isaac   ierr = PetscSectionGetStorageSize(unionSection,&numUnionPoints);CHKERRQ(ierr);
274da43764aSToby Isaac   ierr = PetscMalloc2(numUnionPoints,&coneSizes,dim+1,&numDimPoints);CHKERRQ(ierr);
275da43764aSToby Isaac   /* count dimension points */
276da43764aSToby Isaac   for (d = 0; d <= dim; d++) {
277da43764aSToby Isaac     PetscInt cStart, cOff, cOff2;
278da43764aSToby Isaac     ierr = DMPlexGetHeightStratum(K,d,&cStart,NULL);CHKERRQ(ierr);
279da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionSection,cStart-pStart,&cOff);CHKERRQ(ierr);
280da43764aSToby Isaac     if (d < dim) {
281da43764aSToby Isaac       ierr = DMPlexGetHeightStratum(K,d+1,&cStart,NULL);CHKERRQ(ierr);
282da43764aSToby Isaac       ierr = PetscSectionGetOffset(unionSection,cStart-pStart,&cOff2);CHKERRQ(ierr);
283da43764aSToby Isaac     }
284da43764aSToby Isaac     else {
285da43764aSToby Isaac       cOff2 = numUnionPoints;
286da43764aSToby Isaac     }
287da43764aSToby Isaac     numDimPoints[dim - d] = cOff2 - cOff;
288da43764aSToby Isaac   }
289da43764aSToby Isaac   ierr = PetscSectionCreate(comm, &unionConeSection);CHKERRQ(ierr);
290da43764aSToby Isaac   ierr = PetscSectionSetChart(unionConeSection, 0, numUnionPoints);CHKERRQ(ierr);
291da43764aSToby Isaac   /* count the cones in the union */
292da43764aSToby Isaac   for (p = pStart; p < pEnd; p++) {
293da43764aSToby Isaac     PetscInt dof, uOff;
294da43764aSToby Isaac 
295da43764aSToby Isaac     ierr = DMPlexGetConeSize(K, p, &dof);CHKERRQ(ierr);
296da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pStart,&uOff);CHKERRQ(ierr);
297da43764aSToby Isaac     ierr = PetscSectionSetDof(unionConeSection, uOff, dof);CHKERRQ(ierr);
298da43764aSToby Isaac     coneSizes[uOff] = dof;
299da43764aSToby Isaac   }
300da43764aSToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
301da43764aSToby Isaac     PetscInt dof, uDof, uOff;
302da43764aSToby Isaac 
303da43764aSToby Isaac     ierr = DMPlexGetConeSize(Kref, p, &dof);CHKERRQ(ierr);
304da43764aSToby Isaac     ierr = PetscSectionGetDof(unionSection, p - pRefStart + (pEnd - pStart),&uDof);CHKERRQ(ierr);
305da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pRefStart + (pEnd - pStart),&uOff);CHKERRQ(ierr);
306da43764aSToby Isaac     if (uDof) {
307da43764aSToby Isaac       ierr = PetscSectionSetDof(unionConeSection, uOff, dof);CHKERRQ(ierr);
308da43764aSToby Isaac       coneSizes[uOff] = dof;
309da43764aSToby Isaac     }
310da43764aSToby Isaac   }
311da43764aSToby Isaac   ierr = PetscSectionSetUp(unionConeSection);CHKERRQ(ierr);
312da43764aSToby Isaac   ierr = PetscSectionGetStorageSize(unionConeSection,&numCones);CHKERRQ(ierr);
313da43764aSToby Isaac   ierr = PetscMalloc2(numCones,&unionCones,numCones,&unionOrientations);CHKERRQ(ierr);
314da43764aSToby Isaac   /* write the cones in the union */
315da43764aSToby Isaac   for (p = pStart; p < pEnd; p++) {
316da43764aSToby Isaac     PetscInt dof, uOff, c, cOff;
317da43764aSToby Isaac     const PetscInt *cone, *orientation;
318da43764aSToby Isaac 
319da43764aSToby Isaac     ierr = DMPlexGetConeSize(K, p, &dof);CHKERRQ(ierr);
320da43764aSToby Isaac     ierr = DMPlexGetCone(K, p, &cone);CHKERRQ(ierr);
321da43764aSToby Isaac     ierr = DMPlexGetConeOrientation(K, p, &orientation);CHKERRQ(ierr);
322da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pStart,&uOff);CHKERRQ(ierr);
323da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionConeSection,uOff,&cOff);CHKERRQ(ierr);
324da43764aSToby Isaac     for (c = 0; c < dof; c++) {
325da43764aSToby Isaac       PetscInt e, eOff;
326da43764aSToby Isaac       e                           = cone[c];
327da43764aSToby Isaac       ierr                        = PetscSectionGetOffset(unionSection, e - pStart, &eOff);CHKERRQ(ierr);
328da43764aSToby Isaac       unionCones[cOff + c]        = eOff;
329da43764aSToby Isaac       unionOrientations[cOff + c] = orientation[c];
330da43764aSToby Isaac     }
331da43764aSToby Isaac   }
332da43764aSToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
333da43764aSToby Isaac     PetscInt dof, uDof, uOff, c, cOff;
334da43764aSToby Isaac     const PetscInt *cone, *orientation;
335da43764aSToby Isaac 
336da43764aSToby Isaac     ierr = DMPlexGetConeSize(Kref, p, &dof);CHKERRQ(ierr);
337da43764aSToby Isaac     ierr = DMPlexGetCone(Kref, p, &cone);CHKERRQ(ierr);
338da43764aSToby Isaac     ierr = DMPlexGetConeOrientation(Kref, p, &orientation);CHKERRQ(ierr);
339da43764aSToby Isaac     ierr = PetscSectionGetDof(unionSection, p - pRefStart + (pEnd - pStart),&uDof);CHKERRQ(ierr);
340da43764aSToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pRefStart + (pEnd - pStart),&uOff);CHKERRQ(ierr);
341da43764aSToby Isaac     if (uDof) {
342da43764aSToby Isaac       ierr = PetscSectionGetOffset(unionConeSection,uOff,&cOff);CHKERRQ(ierr);
343da43764aSToby Isaac       for (c = 0; c < dof; c++) {
344da43764aSToby Isaac         PetscInt e, eOff, eDof;
345da43764aSToby Isaac 
346da43764aSToby Isaac         e    = cone[c];
347da43764aSToby Isaac         ierr = PetscSectionGetDof(unionSection, e - pRefStart + (pEnd - pStart),&eDof);CHKERRQ(ierr);
348da43764aSToby Isaac         if (eDof) {
349da43764aSToby Isaac           ierr = PetscSectionGetOffset(unionSection, e - pRefStart + (pEnd - pStart), &eOff);CHKERRQ(ierr);
350da43764aSToby Isaac         }
351da43764aSToby Isaac         else {
352da43764aSToby Isaac           ierr = DMLabelGetValue(identityRef, e, &e);CHKERRQ(ierr);
353da43764aSToby Isaac           ierr = PetscSectionGetOffset(unionSection, e - pStart, &eOff);CHKERRQ(ierr);
354da43764aSToby Isaac         }
355da43764aSToby Isaac         unionCones[cOff + c]        = eOff;
356da43764aSToby Isaac         unionOrientations[cOff + c] = orientation[c];
357da43764aSToby Isaac       }
358da43764aSToby Isaac     }
359da43764aSToby Isaac   }
360da43764aSToby Isaac   /* get the coordinates */
361da43764aSToby Isaac   {
362da43764aSToby Isaac     PetscInt vStart, vEnd, vRefStart, vRefEnd, v, vDof, vOff;
363da43764aSToby Isaac     PetscSection KcoordsSec, KrefCoordsSec;
364da43764aSToby Isaac     Vec      KcoordsVec, KrefCoordsVec;
365da43764aSToby Isaac     PetscScalar *Kcoords;
366da43764aSToby Isaac 
367da43764aSToby Isaac     DMGetCoordinateSection(K, &KcoordsSec);CHKERRQ(ierr);
368da43764aSToby Isaac     DMGetCoordinatesLocal(K, &KcoordsVec);CHKERRQ(ierr);
369da43764aSToby Isaac     DMGetCoordinateSection(Kref, &KrefCoordsSec);CHKERRQ(ierr);
370da43764aSToby Isaac     DMGetCoordinatesLocal(Kref, &KrefCoordsVec);CHKERRQ(ierr);
371da43764aSToby Isaac 
372da43764aSToby Isaac     numVerts = numDimPoints[0];
373da43764aSToby Isaac     ierr     = PetscMalloc1(numVerts * dim,&unionCoords);CHKERRQ(ierr);
374da43764aSToby Isaac     ierr     = DMPlexGetDepthStratum(K,0,&vStart,&vEnd);CHKERRQ(ierr);
375da43764aSToby Isaac 
376da43764aSToby Isaac     offset = 0;
377da43764aSToby Isaac     for (v = vStart; v < vEnd; v++) {
378da43764aSToby Isaac       ierr = PetscSectionGetOffset(unionSection,v - pStart,&vOff);CHKERRQ(ierr);
379da43764aSToby Isaac       ierr = VecGetValuesSection(KcoordsVec, KcoordsSec, v, &Kcoords);CHKERRQ(ierr);
380da43764aSToby Isaac       for (d = 0; d < dim; d++) {
381da43764aSToby Isaac         unionCoords[offset * dim + d] = Kcoords[d];
382da43764aSToby Isaac       }
383da43764aSToby Isaac       offset++;
384da43764aSToby Isaac     }
385da43764aSToby Isaac     ierr = DMPlexGetDepthStratum(Kref,0,&vRefStart,&vRefEnd);CHKERRQ(ierr);
386da43764aSToby Isaac     for (v = vRefStart; v < vRefEnd; v++) {
387da43764aSToby Isaac       ierr = PetscSectionGetDof(unionSection,v - pRefStart + (pEnd - pStart),&vDof);CHKERRQ(ierr);
388da43764aSToby Isaac       ierr = PetscSectionGetOffset(unionSection,v - pRefStart + (pEnd - pStart),&vOff);CHKERRQ(ierr);
389da43764aSToby Isaac       ierr = VecGetValuesSection(KrefCoordsVec, KrefCoordsSec, v, &Kcoords);CHKERRQ(ierr);
390da43764aSToby Isaac       if (vDof) {
391da43764aSToby Isaac         for (d = 0; d < dim; d++) {
392da43764aSToby Isaac           unionCoords[offset * dim + d] = Kcoords[d];
393da43764aSToby Isaac         }
394da43764aSToby Isaac         offset++;
395da43764aSToby Isaac       }
396da43764aSToby Isaac     }
397da43764aSToby Isaac   }
398da43764aSToby Isaac   ierr = DMCreate(comm,ref);CHKERRQ(ierr);
399da43764aSToby Isaac   ierr = DMSetType(*ref,DMPLEX);CHKERRQ(ierr);
40028f4b327SMatthew G. Knepley   ierr = DMSetDimension(*ref,dim);CHKERRQ(ierr);
401da43764aSToby Isaac   ierr = DMPlexCreateFromDAG(*ref,dim,numDimPoints,coneSizes,unionCones,unionOrientations,unionCoords);CHKERRQ(ierr);
40210f7e118SToby Isaac   /* set the tree */
40310f7e118SToby Isaac   ierr = PetscSectionCreate(comm,&parentSection);CHKERRQ(ierr);
40410f7e118SToby Isaac   ierr = PetscSectionSetChart(parentSection,0,numUnionPoints);CHKERRQ(ierr);
40510f7e118SToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
40610f7e118SToby Isaac     PetscInt uDof, uOff;
40710f7e118SToby Isaac 
40810f7e118SToby Isaac     ierr = PetscSectionGetDof(unionSection, p - pRefStart + (pEnd - pStart),&uDof);CHKERRQ(ierr);
40910f7e118SToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pRefStart + (pEnd - pStart),&uOff);CHKERRQ(ierr);
41010f7e118SToby Isaac     if (uDof) {
41110f7e118SToby Isaac       PetscSectionSetDof(parentSection,uOff,1);CHKERRQ(ierr);
41210f7e118SToby Isaac     }
41310f7e118SToby Isaac   }
41410f7e118SToby Isaac   ierr = PetscSectionSetUp(parentSection);CHKERRQ(ierr);
41510f7e118SToby Isaac   ierr = PetscSectionGetStorageSize(parentSection,&parentSize);CHKERRQ(ierr);
41610f7e118SToby Isaac   ierr = PetscMalloc2(parentSize,&parents,parentSize,&childIDs);CHKERRQ(ierr);
41710f7e118SToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
41810f7e118SToby Isaac     PetscInt uDof, uOff;
41910f7e118SToby Isaac 
42010f7e118SToby Isaac     ierr = PetscSectionGetDof(unionSection, p - pRefStart + (pEnd - pStart),&uDof);CHKERRQ(ierr);
42110f7e118SToby Isaac     ierr = PetscSectionGetOffset(unionSection, p - pRefStart + (pEnd - pStart),&uOff);CHKERRQ(ierr);
42210f7e118SToby Isaac     if (uDof) {
42310f7e118SToby Isaac       PetscInt pOff, parent, parentU;
42410f7e118SToby Isaac       PetscSectionGetOffset(parentSection,uOff,&pOff);CHKERRQ(ierr);
42510f7e118SToby Isaac       DMLabelGetValue(identityRef,p,&parent);CHKERRQ(ierr);
42610f7e118SToby Isaac       ierr = PetscSectionGetOffset(unionSection, parent - pStart,&parentU);CHKERRQ(ierr);
42710f7e118SToby Isaac       parents[pOff] = parentU;
42810f7e118SToby Isaac       childIDs[pOff] = uOff;
42910f7e118SToby Isaac     }
43010f7e118SToby Isaac   }
431776742edSToby Isaac   ierr = DMPlexSetTree_Internal(*ref,parentSection,parents,childIDs,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
43210f7e118SToby Isaac   ierr = PetscSectionDestroy(&parentSection);CHKERRQ(ierr);
43310f7e118SToby Isaac   ierr = PetscFree2(parents,childIDs);CHKERRQ(ierr);
43410f7e118SToby Isaac 
435da43764aSToby Isaac   /* clean up */
436da43764aSToby Isaac   ierr = PetscSectionDestroy(&unionSection);CHKERRQ(ierr);
437da43764aSToby Isaac   ierr = PetscSectionDestroy(&unionConeSection);CHKERRQ(ierr);
438da43764aSToby Isaac   ierr = ISDestroy(&perm);CHKERRQ(ierr);
439da43764aSToby Isaac   ierr = PetscFree(unionCoords);CHKERRQ(ierr);
440da43764aSToby Isaac   ierr = PetscFree2(unionCones,unionOrientations);CHKERRQ(ierr);
441da43764aSToby Isaac   ierr = PetscFree2(coneSizes,numDimPoints);CHKERRQ(ierr);
4420e2cc29aSToby Isaac   PetscFunctionReturn(0);
4430e2cc29aSToby Isaac }
4440e2cc29aSToby Isaac 
4450e2cc29aSToby Isaac #undef __FUNCT__
4460e2cc29aSToby Isaac #define __FUNCT__ "DMPlexCreateDefaultReferenceTree"
4470e2cc29aSToby Isaac /*@
4480e2cc29aSToby Isaac   DMPlexCreateDefaultReferenceTree - create a reference tree for isotropic hierarchical mesh refinement.
4490e2cc29aSToby Isaac 
4500e2cc29aSToby Isaac   Collective on comm
4510e2cc29aSToby Isaac 
4520e2cc29aSToby Isaac   Input Parameters:
4530e2cc29aSToby Isaac + comm    - the MPI communicator
4540e2cc29aSToby Isaac . dim     - the spatial dimension
4550e2cc29aSToby Isaac - simplex - Flag for simplex, otherwise use a tensor-product cell
4560e2cc29aSToby Isaac 
4570e2cc29aSToby Isaac   Output Parameters:
4580e2cc29aSToby Isaac . ref     - the reference tree DMPlex object
4590e2cc29aSToby Isaac 
4600e2cc29aSToby Isaac   Level: intermediate
4610e2cc29aSToby Isaac 
4620e2cc29aSToby Isaac .keywords: reference cell
4630e2cc29aSToby Isaac .seealso: DMPlexSetReferenceTree(), DMPlexGetReferenceTree()
4640e2cc29aSToby Isaac @*/
4650e2cc29aSToby Isaac PetscErrorCode DMPlexCreateDefaultReferenceTree(MPI_Comm comm, PetscInt dim, PetscBool simplex, DM *ref)
4660e2cc29aSToby Isaac {
4670e2cc29aSToby Isaac   DM_Plex       *mesh;
4680e2cc29aSToby Isaac   DM             K, Kref;
4690e2cc29aSToby Isaac   PetscInt       p, pStart, pEnd;
4700e2cc29aSToby Isaac   DMLabel        identity;
4710e2cc29aSToby Isaac   PetscErrorCode ierr;
4720e2cc29aSToby Isaac 
4730e2cc29aSToby Isaac   PetscFunctionBegin;
4740e2cc29aSToby Isaac #if 1
4750e2cc29aSToby Isaac   comm = PETSC_COMM_SELF;
4760e2cc29aSToby Isaac #endif
4770e2cc29aSToby Isaac   /* create a reference element */
4780e2cc29aSToby Isaac   ierr = DMPlexCreateReferenceCell(comm, dim, simplex, &K);CHKERRQ(ierr);
4790e2cc29aSToby Isaac   ierr = DMCreateLabel(K, "identity");CHKERRQ(ierr);
4800e2cc29aSToby Isaac   ierr = DMGetLabel(K, "identity", &identity);CHKERRQ(ierr);
4810e2cc29aSToby Isaac   ierr = DMPlexGetChart(K, &pStart, &pEnd);CHKERRQ(ierr);
4820e2cc29aSToby Isaac   for (p = pStart; p < pEnd; p++) {
4830e2cc29aSToby Isaac     ierr = DMLabelSetValue(identity, p, p);CHKERRQ(ierr);
4840e2cc29aSToby Isaac   }
4850e2cc29aSToby Isaac   /* refine it */
4860e2cc29aSToby Isaac   ierr = DMRefine(K,comm,&Kref);CHKERRQ(ierr);
4870e2cc29aSToby Isaac 
4880e2cc29aSToby Isaac   /* the reference tree is the union of these two, without duplicating
4890e2cc29aSToby Isaac    * points that appear in both */
4900e2cc29aSToby Isaac   ierr = DMPlexCreateReferenceTree_Union(K, Kref, "identity", ref);CHKERRQ(ierr);
4910e2cc29aSToby Isaac   mesh = (DM_Plex *) (*ref)->data;
4920e2cc29aSToby Isaac   mesh->getchildsymmetry = DMPlexReferenceTreeGetChildSymmetry_Default;
493da43764aSToby Isaac   ierr = DMDestroy(&K);CHKERRQ(ierr);
494da43764aSToby Isaac   ierr = DMDestroy(&Kref);CHKERRQ(ierr);
495da43764aSToby Isaac   PetscFunctionReturn(0);
496da43764aSToby Isaac }
497da43764aSToby Isaac 
498d961a43aSToby Isaac #undef __FUNCT__
499878b19aaSToby Isaac #define __FUNCT__ "DMPlexTreeSymmetrize"
500878b19aaSToby Isaac static PetscErrorCode DMPlexTreeSymmetrize(DM dm)
501878b19aaSToby Isaac {
502878b19aaSToby Isaac   DM_Plex        *mesh = (DM_Plex *)dm->data;
503878b19aaSToby Isaac   PetscSection   childSec, pSec;
504878b19aaSToby Isaac   PetscInt       p, pSize, cSize, parMax = PETSC_MIN_INT, parMin = PETSC_MAX_INT;
505878b19aaSToby Isaac   PetscInt       *offsets, *children, pStart, pEnd;
506878b19aaSToby Isaac   PetscErrorCode ierr;
507878b19aaSToby Isaac 
508878b19aaSToby Isaac   PetscFunctionBegin;
509878b19aaSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
510878b19aaSToby Isaac   ierr = PetscSectionDestroy(&mesh->childSection);CHKERRQ(ierr);
511878b19aaSToby Isaac   ierr = PetscFree(mesh->children);CHKERRQ(ierr);
512878b19aaSToby Isaac   pSec = mesh->parentSection;
513878b19aaSToby Isaac   if (!pSec) PetscFunctionReturn(0);
514878b19aaSToby Isaac   ierr = PetscSectionGetStorageSize(pSec,&pSize);CHKERRQ(ierr);
515878b19aaSToby Isaac   for (p = 0; p < pSize; p++) {
516878b19aaSToby Isaac     PetscInt par = mesh->parents[p];
517878b19aaSToby Isaac 
518878b19aaSToby Isaac     parMax = PetscMax(parMax,par+1);
519878b19aaSToby Isaac     parMin = PetscMin(parMin,par);
520878b19aaSToby Isaac   }
521878b19aaSToby Isaac   if (parMin > parMax) {
522878b19aaSToby Isaac     parMin = -1;
523878b19aaSToby Isaac     parMax = -1;
524878b19aaSToby Isaac   }
525878b19aaSToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)pSec),&childSec);CHKERRQ(ierr);
526878b19aaSToby Isaac   ierr = PetscSectionSetChart(childSec,parMin,parMax);CHKERRQ(ierr);
527878b19aaSToby Isaac   for (p = 0; p < pSize; p++) {
528878b19aaSToby Isaac     PetscInt par = mesh->parents[p];
529878b19aaSToby Isaac 
530878b19aaSToby Isaac     ierr = PetscSectionAddDof(childSec,par,1);CHKERRQ(ierr);
531878b19aaSToby Isaac   }
532878b19aaSToby Isaac   ierr = PetscSectionSetUp(childSec);CHKERRQ(ierr);
533878b19aaSToby Isaac   ierr = PetscSectionGetStorageSize(childSec,&cSize);CHKERRQ(ierr);
534878b19aaSToby Isaac   ierr = PetscMalloc1(cSize,&children);CHKERRQ(ierr);
535878b19aaSToby Isaac   ierr = PetscCalloc1(parMax-parMin,&offsets);CHKERRQ(ierr);
536878b19aaSToby Isaac   ierr = PetscSectionGetChart(pSec,&pStart,&pEnd);CHKERRQ(ierr);
537878b19aaSToby Isaac   for (p = pStart; p < pEnd; p++) {
538878b19aaSToby Isaac     PetscInt dof, off, i;
539878b19aaSToby Isaac 
540878b19aaSToby Isaac     ierr = PetscSectionGetDof(pSec,p,&dof);CHKERRQ(ierr);
541878b19aaSToby Isaac     ierr = PetscSectionGetOffset(pSec,p,&off);CHKERRQ(ierr);
542878b19aaSToby Isaac     for (i = 0; i < dof; i++) {
543878b19aaSToby Isaac       PetscInt par = mesh->parents[off + i], cOff;
544878b19aaSToby Isaac 
545878b19aaSToby Isaac       ierr = PetscSectionGetOffset(childSec,par,&cOff);CHKERRQ(ierr);
546878b19aaSToby Isaac       children[cOff + offsets[par-parMin]++] = p;
547878b19aaSToby Isaac     }
548878b19aaSToby Isaac   }
549878b19aaSToby Isaac   mesh->childSection = childSec;
550878b19aaSToby Isaac   mesh->children = children;
551878b19aaSToby Isaac   ierr = PetscFree(offsets);CHKERRQ(ierr);
552878b19aaSToby Isaac   PetscFunctionReturn(0);
553878b19aaSToby Isaac }
554878b19aaSToby Isaac 
555878b19aaSToby Isaac #undef __FUNCT__
5566dd5a8c8SToby Isaac #define __FUNCT__ "AnchorsFlatten"
5576dd5a8c8SToby Isaac static PetscErrorCode AnchorsFlatten (PetscSection section, IS is, PetscSection *sectionNew, IS *isNew)
5586dd5a8c8SToby Isaac {
5596dd5a8c8SToby Isaac   PetscInt       pStart, pEnd, size, sizeNew, i, p, *valsNew = NULL;
5606dd5a8c8SToby Isaac   const PetscInt *vals;
5616dd5a8c8SToby Isaac   PetscSection   secNew;
5626dd5a8c8SToby Isaac   PetscBool      anyNew, globalAnyNew;
5636dd5a8c8SToby Isaac   PetscBool      compress;
5646dd5a8c8SToby Isaac   PetscErrorCode ierr;
5656dd5a8c8SToby Isaac 
5666dd5a8c8SToby Isaac   PetscFunctionBegin;
5676dd5a8c8SToby Isaac   ierr = PetscSectionGetChart(section,&pStart,&pEnd);CHKERRQ(ierr);
5686dd5a8c8SToby Isaac   ierr = ISGetLocalSize(is,&size);CHKERRQ(ierr);
5696dd5a8c8SToby Isaac   ierr = ISGetIndices(is,&vals);CHKERRQ(ierr);
5706dd5a8c8SToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)section),&secNew);CHKERRQ(ierr);
5716dd5a8c8SToby Isaac   ierr = PetscSectionSetChart(secNew,pStart,pEnd);CHKERRQ(ierr);
5726dd5a8c8SToby Isaac   for (i = 0; i < size; i++) {
5736dd5a8c8SToby Isaac     PetscInt dof;
5746dd5a8c8SToby Isaac 
5756dd5a8c8SToby Isaac     p = vals[i];
5766dd5a8c8SToby Isaac     if (p < pStart || p >= pEnd) continue;
5776dd5a8c8SToby Isaac     ierr = PetscSectionGetDof(section, p, &dof);CHKERRQ(ierr);
5786dd5a8c8SToby Isaac     if (dof) break;
5796dd5a8c8SToby Isaac   }
5806dd5a8c8SToby Isaac   if (i == size) {
5816dd5a8c8SToby Isaac     ierr     = PetscSectionSetUp(secNew);CHKERRQ(ierr);
5826dd5a8c8SToby Isaac     anyNew   = PETSC_FALSE;
5836dd5a8c8SToby Isaac     compress = PETSC_FALSE;
5846dd5a8c8SToby Isaac     sizeNew  = 0;
5856dd5a8c8SToby Isaac   }
5866dd5a8c8SToby Isaac   else {
5876dd5a8c8SToby Isaac     anyNew = PETSC_TRUE;
5886dd5a8c8SToby Isaac     for (p = pStart; p < pEnd; p++) {
5896dd5a8c8SToby Isaac       PetscInt dof, off;
5906dd5a8c8SToby Isaac 
5916dd5a8c8SToby Isaac       ierr = PetscSectionGetDof(section, p, &dof);CHKERRQ(ierr);
5926dd5a8c8SToby Isaac       ierr = PetscSectionGetOffset(section, p, &off);CHKERRQ(ierr);
5936dd5a8c8SToby Isaac       for (i = 0; i < dof; i++) {
5946dd5a8c8SToby Isaac         PetscInt q = vals[off + i], qDof = 0;
5956dd5a8c8SToby Isaac 
5966dd5a8c8SToby Isaac         if (q >= pStart && q < pEnd) {
5976dd5a8c8SToby Isaac           ierr = PetscSectionGetDof(section, q, &qDof);CHKERRQ(ierr);
5986dd5a8c8SToby Isaac         }
5996dd5a8c8SToby Isaac         if (qDof) {
6006dd5a8c8SToby Isaac           ierr = PetscSectionAddDof(secNew, p, qDof);CHKERRQ(ierr);
6016dd5a8c8SToby Isaac         }
6026dd5a8c8SToby Isaac         else {
6036dd5a8c8SToby Isaac           ierr = PetscSectionAddDof(secNew, p, 1);CHKERRQ(ierr);
6046dd5a8c8SToby Isaac         }
6056dd5a8c8SToby Isaac       }
6066dd5a8c8SToby Isaac     }
6076dd5a8c8SToby Isaac     ierr = PetscSectionSetUp(secNew);CHKERRQ(ierr);
6086dd5a8c8SToby Isaac     ierr = PetscSectionGetStorageSize(secNew,&sizeNew);CHKERRQ(ierr);
6096dd5a8c8SToby Isaac     ierr = PetscMalloc1(sizeNew,&valsNew);CHKERRQ(ierr);
6106dd5a8c8SToby Isaac     compress = PETSC_FALSE;
6116dd5a8c8SToby Isaac     for (p = pStart; p < pEnd; p++) {
6126dd5a8c8SToby Isaac       PetscInt dof, off, count, offNew, dofNew;
6136dd5a8c8SToby Isaac 
6146dd5a8c8SToby Isaac       ierr  = PetscSectionGetDof(section, p, &dof);CHKERRQ(ierr);
6156dd5a8c8SToby Isaac       ierr  = PetscSectionGetOffset(section, p, &off);CHKERRQ(ierr);
6166dd5a8c8SToby Isaac       ierr  = PetscSectionGetDof(secNew, p, &dofNew);CHKERRQ(ierr);
6176dd5a8c8SToby Isaac       ierr  = PetscSectionGetOffset(secNew, p, &offNew);CHKERRQ(ierr);
6186dd5a8c8SToby Isaac       count = 0;
6196dd5a8c8SToby Isaac       for (i = 0; i < dof; i++) {
6206dd5a8c8SToby Isaac         PetscInt q = vals[off + i], qDof = 0, qOff = 0, j;
6216dd5a8c8SToby Isaac 
6226dd5a8c8SToby Isaac         if (q >= pStart && q < pEnd) {
6236dd5a8c8SToby Isaac           ierr = PetscSectionGetDof(section, q, &qDof);CHKERRQ(ierr);
6246dd5a8c8SToby Isaac           ierr = PetscSectionGetOffset(section, q, &qOff);CHKERRQ(ierr);
6256dd5a8c8SToby Isaac         }
6266dd5a8c8SToby Isaac         if (qDof) {
6276dd5a8c8SToby Isaac           PetscInt oldCount = count;
6286dd5a8c8SToby Isaac 
6296dd5a8c8SToby Isaac           for (j = 0; j < qDof; j++) {
6306dd5a8c8SToby Isaac             PetscInt k, r = vals[qOff + j];
6316dd5a8c8SToby Isaac 
6326dd5a8c8SToby Isaac             for (k = 0; k < oldCount; k++) {
6336dd5a8c8SToby Isaac               if (valsNew[offNew + k] == r) {
6346dd5a8c8SToby Isaac                 break;
6356dd5a8c8SToby Isaac               }
6366dd5a8c8SToby Isaac             }
6376dd5a8c8SToby Isaac             if (k == oldCount) {
6386dd5a8c8SToby Isaac               valsNew[offNew + count++] = r;
6396dd5a8c8SToby Isaac             }
6406dd5a8c8SToby Isaac           }
6416dd5a8c8SToby Isaac         }
6426dd5a8c8SToby Isaac         else {
6436dd5a8c8SToby Isaac           PetscInt k, oldCount = count;
6446dd5a8c8SToby Isaac 
6456dd5a8c8SToby Isaac           for (k = 0; k < oldCount; k++) {
6466dd5a8c8SToby Isaac             if (valsNew[offNew + k] == q) {
6476dd5a8c8SToby Isaac               break;
6486dd5a8c8SToby Isaac             }
6496dd5a8c8SToby Isaac           }
6506dd5a8c8SToby Isaac           if (k == oldCount) {
6516dd5a8c8SToby Isaac             valsNew[offNew + count++] = q;
6526dd5a8c8SToby Isaac           }
6536dd5a8c8SToby Isaac         }
6546dd5a8c8SToby Isaac       }
6556dd5a8c8SToby Isaac       if (count < dofNew) {
6566dd5a8c8SToby Isaac         ierr = PetscSectionSetDof(secNew, p, count);CHKERRQ(ierr);
6576dd5a8c8SToby Isaac         compress = PETSC_TRUE;
6586dd5a8c8SToby Isaac       }
6596dd5a8c8SToby Isaac     }
6606dd5a8c8SToby Isaac   }
6616dd5a8c8SToby Isaac   ierr = ISRestoreIndices(is,&vals);CHKERRQ(ierr);
662b2566f29SBarry Smith   ierr = MPIU_Allreduce(&anyNew,&globalAnyNew,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)secNew));CHKERRQ(ierr);
6636dd5a8c8SToby Isaac   if (!globalAnyNew) {
6646dd5a8c8SToby Isaac     ierr = PetscSectionDestroy(&secNew);CHKERRQ(ierr);
6656dd5a8c8SToby Isaac     *sectionNew = NULL;
6666dd5a8c8SToby Isaac     *isNew = NULL;
6676dd5a8c8SToby Isaac   }
6686dd5a8c8SToby Isaac   else {
6696dd5a8c8SToby Isaac     PetscBool globalCompress;
6706dd5a8c8SToby Isaac 
671b2566f29SBarry Smith     ierr = MPIU_Allreduce(&compress,&globalCompress,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)secNew));CHKERRQ(ierr);
6726dd5a8c8SToby Isaac     if (compress) {
6736dd5a8c8SToby Isaac       PetscSection secComp;
6746dd5a8c8SToby Isaac       PetscInt *valsComp = NULL;
6756dd5a8c8SToby Isaac 
6766dd5a8c8SToby Isaac       ierr = PetscSectionCreate(PetscObjectComm((PetscObject)section),&secComp);CHKERRQ(ierr);
6776dd5a8c8SToby Isaac       ierr = PetscSectionSetChart(secComp,pStart,pEnd);CHKERRQ(ierr);
6786dd5a8c8SToby Isaac       for (p = pStart; p < pEnd; p++) {
6796dd5a8c8SToby Isaac         PetscInt dof;
6806dd5a8c8SToby Isaac 
6816dd5a8c8SToby Isaac         ierr = PetscSectionGetDof(secNew, p, &dof);CHKERRQ(ierr);
6826dd5a8c8SToby Isaac         ierr = PetscSectionSetDof(secComp, p, dof);CHKERRQ(ierr);
6836dd5a8c8SToby Isaac       }
6846dd5a8c8SToby Isaac       ierr = PetscSectionSetUp(secComp);CHKERRQ(ierr);
6856dd5a8c8SToby Isaac       ierr = PetscSectionGetStorageSize(secComp,&sizeNew);CHKERRQ(ierr);
6866dd5a8c8SToby Isaac       ierr = PetscMalloc1(sizeNew,&valsComp);CHKERRQ(ierr);
6876dd5a8c8SToby Isaac       for (p = pStart; p < pEnd; p++) {
6886dd5a8c8SToby Isaac         PetscInt dof, off, offNew, j;
6896dd5a8c8SToby Isaac 
6906dd5a8c8SToby Isaac         ierr = PetscSectionGetDof(secNew, p, &dof);CHKERRQ(ierr);
6916dd5a8c8SToby Isaac         ierr = PetscSectionGetOffset(secNew, p, &off);CHKERRQ(ierr);
6926dd5a8c8SToby Isaac         ierr = PetscSectionGetOffset(secComp, p, &offNew);CHKERRQ(ierr);
6936dd5a8c8SToby Isaac         for (j = 0; j < dof; j++) {
6946dd5a8c8SToby Isaac           valsComp[offNew + j] = valsNew[off + j];
6956dd5a8c8SToby Isaac         }
6966dd5a8c8SToby Isaac       }
6976dd5a8c8SToby Isaac       ierr    = PetscSectionDestroy(&secNew);CHKERRQ(ierr);
6986dd5a8c8SToby Isaac       secNew  = secComp;
6996dd5a8c8SToby Isaac       ierr    = PetscFree(valsNew);CHKERRQ(ierr);
7006dd5a8c8SToby Isaac       valsNew = valsComp;
7016dd5a8c8SToby Isaac     }
7026dd5a8c8SToby Isaac     ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is),sizeNew,valsNew,PETSC_OWN_POINTER,isNew);CHKERRQ(ierr);
7036dd5a8c8SToby Isaac   }
7046dd5a8c8SToby Isaac   PetscFunctionReturn(0);
7056dd5a8c8SToby Isaac }
7066dd5a8c8SToby Isaac 
7076dd5a8c8SToby Isaac #undef __FUNCT__
708f7c74593SToby Isaac #define __FUNCT__ "DMPlexCreateAnchors_Tree"
709f7c74593SToby Isaac static PetscErrorCode DMPlexCreateAnchors_Tree(DM dm)
71066af876cSToby Isaac {
71166af876cSToby Isaac   PetscInt       p, pStart, pEnd, *anchors, size;
71266af876cSToby Isaac   PetscInt       aMin = PETSC_MAX_INT, aMax = PETSC_MIN_INT;
71366af876cSToby Isaac   PetscSection   aSec;
714f9f063d4SToby Isaac   DMLabel        canonLabel;
71566af876cSToby Isaac   IS             aIS;
71666af876cSToby Isaac   PetscErrorCode ierr;
71766af876cSToby Isaac 
71866af876cSToby Isaac   PetscFunctionBegin;
71966af876cSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
72066af876cSToby Isaac   ierr = DMPlexGetChart(dm,&pStart,&pEnd);CHKERRQ(ierr);
721c58f1c22SToby Isaac   ierr = DMGetLabel(dm,"canonical",&canonLabel);CHKERRQ(ierr);
72266af876cSToby Isaac   for (p = pStart; p < pEnd; p++) {
72366af876cSToby Isaac     PetscInt parent;
72466af876cSToby Isaac 
725f9f063d4SToby Isaac     if (canonLabel) {
726f9f063d4SToby Isaac       PetscInt canon;
727f9f063d4SToby Isaac 
728f9f063d4SToby Isaac       ierr = DMLabelGetValue(canonLabel,p,&canon);CHKERRQ(ierr);
729f9f063d4SToby Isaac       if (p != canon) continue;
730f9f063d4SToby Isaac     }
73166af876cSToby Isaac     ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
73266af876cSToby Isaac     if (parent != p) {
73366af876cSToby Isaac       aMin = PetscMin(aMin,p);
73466af876cSToby Isaac       aMax = PetscMax(aMax,p+1);
73566af876cSToby Isaac     }
73666af876cSToby Isaac   }
73766af876cSToby Isaac   if (aMin > aMax) {
73866af876cSToby Isaac     aMin = -1;
73966af876cSToby Isaac     aMax = -1;
74066af876cSToby Isaac   }
741e228b242SToby Isaac   ierr = PetscSectionCreate(PETSC_COMM_SELF,&aSec);CHKERRQ(ierr);
74266af876cSToby Isaac   ierr = PetscSectionSetChart(aSec,aMin,aMax);CHKERRQ(ierr);
74366af876cSToby Isaac   for (p = aMin; p < aMax; p++) {
74466af876cSToby Isaac     PetscInt parent, ancestor = p;
74566af876cSToby Isaac 
746f9f063d4SToby Isaac     if (canonLabel) {
747f9f063d4SToby Isaac       PetscInt canon;
748f9f063d4SToby Isaac 
749f9f063d4SToby Isaac       ierr = DMLabelGetValue(canonLabel,p,&canon);CHKERRQ(ierr);
750f9f063d4SToby Isaac       if (p != canon) continue;
751f9f063d4SToby Isaac     }
75266af876cSToby Isaac     ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
75366af876cSToby Isaac     while (parent != ancestor) {
75466af876cSToby Isaac       ancestor = parent;
75566af876cSToby Isaac       ierr     = DMPlexGetTreeParent(dm,ancestor,&parent,NULL);CHKERRQ(ierr);
75666af876cSToby Isaac     }
75766af876cSToby Isaac     if (ancestor != p) {
75866af876cSToby Isaac       PetscInt closureSize, *closure = NULL;
75966af876cSToby Isaac 
76066af876cSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,ancestor,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
76166af876cSToby Isaac       ierr = PetscSectionSetDof(aSec,p,closureSize);CHKERRQ(ierr);
76266af876cSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,ancestor,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
76366af876cSToby Isaac     }
76466af876cSToby Isaac   }
76566af876cSToby Isaac   ierr = PetscSectionSetUp(aSec);CHKERRQ(ierr);
76666af876cSToby Isaac   ierr = PetscSectionGetStorageSize(aSec,&size);CHKERRQ(ierr);
76766af876cSToby Isaac   ierr = PetscMalloc1(size,&anchors);CHKERRQ(ierr);
76866af876cSToby Isaac   for (p = aMin; p < aMax; p++) {
76966af876cSToby Isaac     PetscInt parent, ancestor = p;
77066af876cSToby Isaac 
771f9f063d4SToby Isaac     if (canonLabel) {
772f9f063d4SToby Isaac       PetscInt canon;
773f9f063d4SToby Isaac 
774f9f063d4SToby Isaac       ierr = DMLabelGetValue(canonLabel,p,&canon);CHKERRQ(ierr);
775f9f063d4SToby Isaac       if (p != canon) continue;
776f9f063d4SToby Isaac     }
77766af876cSToby Isaac     ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
77866af876cSToby Isaac     while (parent != ancestor) {
77966af876cSToby Isaac       ancestor = parent;
78066af876cSToby Isaac       ierr     = DMPlexGetTreeParent(dm,ancestor,&parent,NULL);CHKERRQ(ierr);
78166af876cSToby Isaac     }
78266af876cSToby Isaac     if (ancestor != p) {
78366af876cSToby Isaac       PetscInt j, closureSize, *closure = NULL, aOff;
78466af876cSToby Isaac 
78566af876cSToby Isaac       ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
78666af876cSToby Isaac 
78766af876cSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,ancestor,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
78866af876cSToby Isaac       for (j = 0; j < closureSize; j++) {
78966af876cSToby Isaac         anchors[aOff + j] = closure[2*j];
79066af876cSToby Isaac       }
79166af876cSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,ancestor,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
79266af876cSToby Isaac     }
79366af876cSToby Isaac   }
794e228b242SToby Isaac   ierr = ISCreateGeneral(PETSC_COMM_SELF,size,anchors,PETSC_OWN_POINTER,&aIS);CHKERRQ(ierr);
7956dd5a8c8SToby Isaac   {
7966dd5a8c8SToby Isaac     PetscSection aSecNew = aSec;
7976dd5a8c8SToby Isaac     IS           aISNew  = aIS;
7986dd5a8c8SToby Isaac 
7996dd5a8c8SToby Isaac     ierr = PetscObjectReference((PetscObject)aSec);CHKERRQ(ierr);
8006dd5a8c8SToby Isaac     ierr = PetscObjectReference((PetscObject)aIS);CHKERRQ(ierr);
8016dd5a8c8SToby Isaac     while (aSecNew) {
8026dd5a8c8SToby Isaac       ierr    = PetscSectionDestroy(&aSec);CHKERRQ(ierr);
8036dd5a8c8SToby Isaac       ierr    = ISDestroy(&aIS);CHKERRQ(ierr);
8046dd5a8c8SToby Isaac       aSec    = aSecNew;
8056dd5a8c8SToby Isaac       aIS     = aISNew;
8066dd5a8c8SToby Isaac       aSecNew = NULL;
8076dd5a8c8SToby Isaac       aISNew  = NULL;
8086dd5a8c8SToby Isaac       ierr    = AnchorsFlatten(aSec,aIS,&aSecNew,&aISNew);CHKERRQ(ierr);
8096dd5a8c8SToby Isaac     }
8106dd5a8c8SToby Isaac   }
811a17985deSToby Isaac   ierr = DMPlexSetAnchors(dm,aSec,aIS);CHKERRQ(ierr);
81266af876cSToby Isaac   ierr = PetscSectionDestroy(&aSec);CHKERRQ(ierr);
81366af876cSToby Isaac   ierr = ISDestroy(&aIS);CHKERRQ(ierr);
81466af876cSToby Isaac   PetscFunctionReturn(0);
81566af876cSToby Isaac }
81666af876cSToby Isaac 
81766af876cSToby Isaac #undef __FUNCT__
8186461c1adSToby Isaac #define __FUNCT__ "DMPlexGetTrueSupportSize"
8196461c1adSToby Isaac static PetscErrorCode DMPlexGetTrueSupportSize(DM dm,PetscInt p,PetscInt *dof,PetscInt *numTrueSupp)
8206461c1adSToby Isaac {
8216461c1adSToby Isaac   PetscErrorCode ierr;
8226461c1adSToby Isaac 
8236461c1adSToby Isaac   PetscFunctionBegin;
8246461c1adSToby Isaac   if (numTrueSupp[p] == -1) {
8256461c1adSToby Isaac     PetscInt i, alldof;
8266461c1adSToby Isaac     const PetscInt *supp;
8276461c1adSToby Isaac     PetscInt count = 0;
8286461c1adSToby Isaac 
8296461c1adSToby Isaac     ierr = DMPlexGetSupportSize(dm,p,&alldof);CHKERRQ(ierr);
8306461c1adSToby Isaac     ierr = DMPlexGetSupport(dm,p,&supp);CHKERRQ(ierr);
8316461c1adSToby Isaac     for (i = 0; i < alldof; i++) {
8326461c1adSToby Isaac       PetscInt q = supp[i], numCones, j;
8336461c1adSToby Isaac       const PetscInt *cone;
8346461c1adSToby Isaac 
8356461c1adSToby Isaac       ierr = DMPlexGetConeSize(dm,q,&numCones);CHKERRQ(ierr);
8366461c1adSToby Isaac       ierr = DMPlexGetCone(dm,q,&cone);CHKERRQ(ierr);
8376461c1adSToby Isaac       for (j = 0; j < numCones; j++) {
8386461c1adSToby Isaac         if (cone[j] == p) break;
8396461c1adSToby Isaac       }
8406461c1adSToby Isaac       if (j < numCones) count++;
8416461c1adSToby Isaac     }
8426461c1adSToby Isaac     numTrueSupp[p] = count;
8436461c1adSToby Isaac   }
8446461c1adSToby Isaac   *dof = numTrueSupp[p];
8456461c1adSToby Isaac   PetscFunctionReturn(0);
8466461c1adSToby Isaac }
8476461c1adSToby Isaac 
8486461c1adSToby Isaac #undef __FUNCT__
849776742edSToby Isaac #define __FUNCT__ "DMPlexTreeExchangeSupports"
850776742edSToby Isaac static PetscErrorCode DMPlexTreeExchangeSupports(DM dm)
851776742edSToby Isaac {
852776742edSToby Isaac   DM_Plex *mesh = (DM_Plex *)dm->data;
853776742edSToby Isaac   PetscSection newSupportSection;
854776742edSToby Isaac   PetscInt newSize, *newSupports, pStart, pEnd, p, d, depth;
8556461c1adSToby Isaac   PetscInt *numTrueSupp;
856776742edSToby Isaac   PetscInt *offsets;
857776742edSToby Isaac   PetscErrorCode ierr;
858776742edSToby Isaac 
859776742edSToby Isaac   PetscFunctionBegin;
860776742edSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
861776742edSToby Isaac   /* symmetrize the hierarchy */
862776742edSToby Isaac   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
863e228b242SToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)(mesh->supportSection)),&newSupportSection);CHKERRQ(ierr);
864776742edSToby Isaac   ierr = DMPlexGetChart(dm,&pStart,&pEnd);CHKERRQ(ierr);
865776742edSToby Isaac   ierr = PetscSectionSetChart(newSupportSection,pStart,pEnd);CHKERRQ(ierr);
866776742edSToby Isaac   ierr = PetscCalloc1(pEnd,&offsets);CHKERRQ(ierr);
8676461c1adSToby Isaac   ierr = PetscMalloc1(pEnd,&numTrueSupp);CHKERRQ(ierr);
8686461c1adSToby Isaac   for (p = 0; p < pEnd; p++) numTrueSupp[p] = -1;
8696461c1adSToby Isaac   /* if a point is in the (true) support of q, it should be in the support of
870776742edSToby Isaac    * parent(q) */
871776742edSToby Isaac   for (d = 0; d <= depth; d++) {
872776742edSToby Isaac     ierr = DMPlexGetHeightStratum(dm,d,&pStart,&pEnd);CHKERRQ(ierr);
873776742edSToby Isaac     for (p = pStart; p < pEnd; ++p) {
874776742edSToby Isaac       PetscInt dof, q, qdof, parent;
875776742edSToby Isaac 
8766461c1adSToby Isaac       ierr = DMPlexGetTrueSupportSize(dm,p,&dof,numTrueSupp);CHKERRQ(ierr);
877776742edSToby Isaac       ierr = PetscSectionAddDof(newSupportSection, p, dof);CHKERRQ(ierr);
878776742edSToby Isaac       q    = p;
879776742edSToby Isaac       ierr = DMPlexGetTreeParent(dm,q,&parent,NULL);CHKERRQ(ierr);
880776742edSToby Isaac       while (parent != q && parent >= pStart && parent < pEnd) {
881776742edSToby Isaac         q = parent;
882776742edSToby Isaac 
8836461c1adSToby Isaac         ierr = DMPlexGetTrueSupportSize(dm,q,&qdof,numTrueSupp);CHKERRQ(ierr);
884776742edSToby Isaac         ierr = PetscSectionAddDof(newSupportSection,p,qdof);CHKERRQ(ierr);
885776742edSToby Isaac         ierr = PetscSectionAddDof(newSupportSection,q,dof);CHKERRQ(ierr);
886776742edSToby Isaac         ierr = DMPlexGetTreeParent(dm,q,&parent,NULL);CHKERRQ(ierr);
887776742edSToby Isaac       }
888776742edSToby Isaac     }
889776742edSToby Isaac   }
890776742edSToby Isaac   ierr = PetscSectionSetUp(newSupportSection);CHKERRQ(ierr);
891776742edSToby Isaac   ierr = PetscSectionGetStorageSize(newSupportSection,&newSize);CHKERRQ(ierr);
892776742edSToby Isaac   ierr = PetscMalloc1(newSize,&newSupports);CHKERRQ(ierr);
893776742edSToby Isaac   for (d = 0; d <= depth; d++) {
894776742edSToby Isaac     ierr = DMPlexGetHeightStratum(dm,d,&pStart,&pEnd);CHKERRQ(ierr);
895776742edSToby Isaac     for (p = pStart; p < pEnd; p++) {
896776742edSToby Isaac       PetscInt dof, off, q, qdof, qoff, newDof, newOff, newqOff, i, parent;
897776742edSToby Isaac 
898776742edSToby Isaac       ierr = PetscSectionGetDof(mesh->supportSection, p, &dof);CHKERRQ(ierr);
899776742edSToby Isaac       ierr = PetscSectionGetOffset(mesh->supportSection, p, &off);CHKERRQ(ierr);
900776742edSToby Isaac       ierr = PetscSectionGetDof(newSupportSection, p, &newDof);CHKERRQ(ierr);
901776742edSToby Isaac       ierr = PetscSectionGetOffset(newSupportSection, p, &newOff);CHKERRQ(ierr);
902776742edSToby Isaac       for (i = 0; i < dof; i++) {
9036461c1adSToby Isaac         PetscInt numCones, j;
9046461c1adSToby Isaac         const PetscInt *cone;
9056461c1adSToby Isaac         PetscInt q = mesh->supports[off + i];
9066461c1adSToby Isaac 
9076461c1adSToby Isaac         ierr = DMPlexGetConeSize(dm,q,&numCones);CHKERRQ(ierr);
9086461c1adSToby Isaac         ierr = DMPlexGetCone(dm,q,&cone);CHKERRQ(ierr);
9096461c1adSToby Isaac         for (j = 0; j < numCones; j++) {
9106461c1adSToby Isaac           if (cone[j] == p) break;
9116461c1adSToby Isaac         }
9126461c1adSToby Isaac         if (j < numCones) newSupports[newOff+offsets[p]++] = q;
913776742edSToby Isaac       }
914776742edSToby Isaac       mesh->maxSupportSize = PetscMax(mesh->maxSupportSize,newDof);
915776742edSToby Isaac 
916776742edSToby Isaac       q    = p;
917776742edSToby Isaac       ierr = DMPlexGetTreeParent(dm,q,&parent,NULL);CHKERRQ(ierr);
918776742edSToby Isaac       while (parent != q && parent >= pStart && parent < pEnd) {
919776742edSToby Isaac         q = parent;
920776742edSToby Isaac         ierr = PetscSectionGetDof(mesh->supportSection, q, &qdof);CHKERRQ(ierr);
921776742edSToby Isaac         ierr = PetscSectionGetOffset(mesh->supportSection, q, &qoff);CHKERRQ(ierr);
922776742edSToby Isaac         ierr = PetscSectionGetOffset(newSupportSection, q, &newqOff);CHKERRQ(ierr);
923776742edSToby Isaac         for (i = 0; i < qdof; i++) {
9246461c1adSToby Isaac           PetscInt numCones, j;
9256461c1adSToby Isaac           const PetscInt *cone;
9266461c1adSToby Isaac           PetscInt r = mesh->supports[qoff + i];
9276461c1adSToby Isaac 
9286461c1adSToby Isaac           ierr = DMPlexGetConeSize(dm,r,&numCones);CHKERRQ(ierr);
9296461c1adSToby Isaac           ierr = DMPlexGetCone(dm,r,&cone);CHKERRQ(ierr);
9306461c1adSToby Isaac           for (j = 0; j < numCones; j++) {
9316461c1adSToby Isaac             if (cone[j] == q) break;
9326461c1adSToby Isaac           }
9336461c1adSToby Isaac           if (j < numCones) newSupports[newOff+offsets[p]++] = r;
934776742edSToby Isaac         }
935776742edSToby Isaac         for (i = 0; i < dof; i++) {
9366461c1adSToby Isaac           PetscInt numCones, j;
9376461c1adSToby Isaac           const PetscInt *cone;
9386461c1adSToby Isaac           PetscInt r = mesh->supports[off + i];
9396461c1adSToby Isaac 
9406461c1adSToby Isaac           ierr = DMPlexGetConeSize(dm,r,&numCones);CHKERRQ(ierr);
9416461c1adSToby Isaac           ierr = DMPlexGetCone(dm,r,&cone);CHKERRQ(ierr);
9426461c1adSToby Isaac           for (j = 0; j < numCones; j++) {
9436461c1adSToby Isaac             if (cone[j] == p) break;
9446461c1adSToby Isaac           }
9456461c1adSToby Isaac           if (j < numCones) newSupports[newqOff+offsets[q]++] = r;
946776742edSToby Isaac         }
947776742edSToby Isaac         ierr = DMPlexGetTreeParent(dm,q,&parent,NULL);CHKERRQ(ierr);
948776742edSToby Isaac       }
949776742edSToby Isaac     }
950776742edSToby Isaac   }
951776742edSToby Isaac   ierr = PetscSectionDestroy(&mesh->supportSection);CHKERRQ(ierr);
952776742edSToby Isaac   mesh->supportSection = newSupportSection;
953776742edSToby Isaac   ierr = PetscFree(mesh->supports);CHKERRQ(ierr);
954776742edSToby Isaac   mesh->supports = newSupports;
955776742edSToby Isaac   ierr = PetscFree(offsets);CHKERRQ(ierr);
9566461c1adSToby Isaac   ierr = PetscFree(numTrueSupp);CHKERRQ(ierr);
957776742edSToby Isaac 
958776742edSToby Isaac   PetscFunctionReturn(0);
959776742edSToby Isaac }
960776742edSToby Isaac 
961f7c74593SToby Isaac static PetscErrorCode DMPlexComputeAnchorMatrix_Tree_Direct(DM,PetscSection,PetscSection,Mat);
962f7c74593SToby Isaac static PetscErrorCode DMPlexComputeAnchorMatrix_Tree_FromReference(DM,PetscSection,PetscSection,Mat);
963f7c74593SToby Isaac 
964776742edSToby Isaac #undef __FUNCT__
965f9f063d4SToby Isaac #define __FUNCT__ "DMPlexSetTree_Internal"
966776742edSToby Isaac static PetscErrorCode DMPlexSetTree_Internal(DM dm, PetscSection parentSection, PetscInt *parents, PetscInt *childIDs, PetscBool computeCanonical, PetscBool exchangeSupports)
967f9f063d4SToby Isaac {
968f9f063d4SToby Isaac   DM_Plex       *mesh = (DM_Plex *)dm->data;
969f9f063d4SToby Isaac   DM             refTree;
970f9f063d4SToby Isaac   PetscInt       size;
971f9f063d4SToby Isaac   PetscErrorCode ierr;
972f9f063d4SToby Isaac 
973f9f063d4SToby Isaac   PetscFunctionBegin;
974f9f063d4SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
975f9f063d4SToby Isaac   PetscValidHeaderSpecific(parentSection, PETSC_SECTION_CLASSID, 2);
976f9f063d4SToby Isaac   ierr = PetscObjectReference((PetscObject)parentSection);CHKERRQ(ierr);
977f9f063d4SToby Isaac   ierr = PetscSectionDestroy(&mesh->parentSection);CHKERRQ(ierr);
978f9f063d4SToby Isaac   mesh->parentSection = parentSection;
979f9f063d4SToby Isaac   ierr = PetscSectionGetStorageSize(parentSection,&size);CHKERRQ(ierr);
980f9f063d4SToby Isaac   if (parents != mesh->parents) {
981f9f063d4SToby Isaac     ierr = PetscFree(mesh->parents);CHKERRQ(ierr);
982f9f063d4SToby Isaac     ierr = PetscMalloc1(size,&mesh->parents);CHKERRQ(ierr);
983f9f063d4SToby Isaac     ierr = PetscMemcpy(mesh->parents, parents, size * sizeof(*parents));CHKERRQ(ierr);
984f9f063d4SToby Isaac   }
985f9f063d4SToby Isaac   if (childIDs != mesh->childIDs) {
986f9f063d4SToby Isaac     ierr = PetscFree(mesh->childIDs);CHKERRQ(ierr);
987f9f063d4SToby Isaac     ierr = PetscMalloc1(size,&mesh->childIDs);CHKERRQ(ierr);
988f9f063d4SToby Isaac     ierr = PetscMemcpy(mesh->childIDs, childIDs, size * sizeof(*childIDs));CHKERRQ(ierr);
989f9f063d4SToby Isaac   }
990f9f063d4SToby Isaac   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
991f9f063d4SToby Isaac   if (refTree) {
992f9f063d4SToby Isaac     DMLabel canonLabel;
993f9f063d4SToby Isaac 
994c58f1c22SToby Isaac     ierr = DMGetLabel(refTree,"canonical",&canonLabel);CHKERRQ(ierr);
995f9f063d4SToby Isaac     if (canonLabel) {
996f9f063d4SToby Isaac       PetscInt i;
997f9f063d4SToby Isaac 
998f9f063d4SToby Isaac       for (i = 0; i < size; i++) {
999f9f063d4SToby Isaac         PetscInt canon;
1000f9f063d4SToby Isaac         ierr = DMLabelGetValue(canonLabel, mesh->childIDs[i], &canon);CHKERRQ(ierr);
1001f9f063d4SToby Isaac         if (canon >= 0) {
1002f9f063d4SToby Isaac           mesh->childIDs[i] = canon;
1003f9f063d4SToby Isaac         }
1004f9f063d4SToby Isaac       }
1005f9f063d4SToby Isaac     }
1006f7c74593SToby Isaac     mesh->computeanchormatrix = DMPlexComputeAnchorMatrix_Tree_FromReference;
1007f7c74593SToby Isaac   }
1008f7c74593SToby Isaac   else {
1009f7c74593SToby Isaac     mesh->computeanchormatrix = DMPlexComputeAnchorMatrix_Tree_Direct;
1010f9f063d4SToby Isaac   }
1011f9f063d4SToby Isaac   ierr = DMPlexTreeSymmetrize(dm);CHKERRQ(ierr);
1012f9f063d4SToby Isaac   if (computeCanonical) {
1013f9f063d4SToby Isaac     PetscInt d, dim;
1014f9f063d4SToby Isaac 
1015f9f063d4SToby Isaac     /* add the canonical label */
101628f4b327SMatthew G. Knepley     ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
1017c58f1c22SToby Isaac     ierr = DMCreateLabel(dm,"canonical");CHKERRQ(ierr);
1018f9f063d4SToby Isaac     for (d = 0; d <= dim; d++) {
1019f9f063d4SToby Isaac       PetscInt p, dStart, dEnd, canon = -1, cNumChildren;
1020f9f063d4SToby Isaac       const PetscInt *cChildren;
1021f9f063d4SToby Isaac 
1022f9f063d4SToby Isaac       ierr = DMPlexGetDepthStratum(dm,d,&dStart,&dEnd);CHKERRQ(ierr);
1023f9f063d4SToby Isaac       for (p = dStart; p < dEnd; p++) {
1024f9f063d4SToby Isaac         ierr = DMPlexGetTreeChildren(dm,p,&cNumChildren,&cChildren);CHKERRQ(ierr);
1025f9f063d4SToby Isaac         if (cNumChildren) {
1026f9f063d4SToby Isaac           canon = p;
1027f9f063d4SToby Isaac           break;
1028f9f063d4SToby Isaac         }
1029f9f063d4SToby Isaac       }
1030f9f063d4SToby Isaac       if (canon == -1) continue;
1031f9f063d4SToby Isaac       for (p = dStart; p < dEnd; p++) {
1032f9f063d4SToby Isaac         PetscInt numChildren, i;
1033f9f063d4SToby Isaac         const PetscInt *children;
1034f9f063d4SToby Isaac 
1035f9f063d4SToby Isaac         ierr = DMPlexGetTreeChildren(dm,p,&numChildren,&children);CHKERRQ(ierr);
1036f9f063d4SToby Isaac         if (numChildren) {
1037f9f063d4SToby Isaac           if (numChildren != cNumChildren) SETERRQ2(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"All parent points in a stratum should have the same number of children: %d != %d", numChildren, cNumChildren);
1038c58f1c22SToby Isaac           ierr = DMSetLabelValue(dm,"canonical",p,canon);CHKERRQ(ierr);
1039f9f063d4SToby Isaac           for (i = 0; i < numChildren; i++) {
1040c58f1c22SToby Isaac             ierr = DMSetLabelValue(dm,"canonical",children[i],cChildren[i]);CHKERRQ(ierr);
1041f9f063d4SToby Isaac           }
1042f9f063d4SToby Isaac         }
1043f9f063d4SToby Isaac       }
1044f9f063d4SToby Isaac     }
1045f9f063d4SToby Isaac   }
1046776742edSToby Isaac   if (exchangeSupports) {
1047776742edSToby Isaac     ierr = DMPlexTreeExchangeSupports(dm);CHKERRQ(ierr);
1048776742edSToby Isaac   }
1049f7c74593SToby Isaac   mesh->createanchors = DMPlexCreateAnchors_Tree;
1050f7c74593SToby Isaac   /* reset anchors */
1051f7c74593SToby Isaac   ierr = DMPlexSetAnchors(dm,NULL,NULL);CHKERRQ(ierr);
1052f9f063d4SToby Isaac   PetscFunctionReturn(0);
1053f9f063d4SToby Isaac }
1054f9f063d4SToby Isaac 
1055f9f063d4SToby Isaac #undef __FUNCT__
10560b7167a0SToby Isaac #define __FUNCT__ "DMPlexSetTree"
10570b7167a0SToby Isaac /*@
10580b7167a0SToby Isaac   DMPlexSetTree - set the tree that describes the hierarchy of non-conforming mesh points.  This routine also creates
10590b7167a0SToby Isaac   the point-to-point constraints determined by the tree: a point is constained to the points in the closure of its
10600b7167a0SToby Isaac   tree root.
10610b7167a0SToby Isaac 
10620b7167a0SToby Isaac   Collective on dm
10630b7167a0SToby Isaac 
10640b7167a0SToby Isaac   Input Parameters:
10650b7167a0SToby Isaac + dm - the DMPlex object
10660b7167a0SToby Isaac . parentSection - a section describing the tree: a point has a parent if it has 1 dof in the section; the section
10670b7167a0SToby Isaac                   offset indexes the parent and childID list; the reference count of parentSection is incremented
10680b7167a0SToby Isaac . parents - a list of the point parents; copied, can be destroyed
10690b7167a0SToby Isaac - childIDs - identifies the relationship of the child point to the parent point; if there is a reference tree, then
10700b7167a0SToby Isaac              the child corresponds to the point in the reference tree with index childIDs; copied, can be destroyed
10710b7167a0SToby Isaac 
10720b7167a0SToby Isaac   Level: intermediate
10730b7167a0SToby Isaac 
1074a17985deSToby Isaac .seealso: DMPlexGetTree(), DMPlexSetReferenceTree(), DMPlexSetAnchors(), DMPlexGetTreeParent(), DMPlexGetTreeChildren()
10750b7167a0SToby Isaac @*/
1076b2f41788SToby Isaac PetscErrorCode DMPlexSetTree(DM dm, PetscSection parentSection, PetscInt parents[], PetscInt childIDs[])
10770b7167a0SToby Isaac {
10780b7167a0SToby Isaac   PetscErrorCode ierr;
10790b7167a0SToby Isaac 
10800b7167a0SToby Isaac   PetscFunctionBegin;
1081776742edSToby Isaac   ierr = DMPlexSetTree_Internal(dm,parentSection,parents,childIDs,PETSC_FALSE,PETSC_TRUE);CHKERRQ(ierr);
10820b7167a0SToby Isaac   PetscFunctionReturn(0);
10830b7167a0SToby Isaac }
10840b7167a0SToby Isaac 
10850b7167a0SToby Isaac #undef __FUNCT__
1086b2f41788SToby Isaac #define __FUNCT__ "DMPlexGetTree"
1087b2f41788SToby Isaac /*@
1088b2f41788SToby Isaac   DMPlexGetTree - get the tree that describes the hierarchy of non-conforming mesh points.
1089b2f41788SToby Isaac   Collective on dm
1090b2f41788SToby Isaac 
1091b2f41788SToby Isaac   Input Parameters:
1092b2f41788SToby Isaac . dm - the DMPlex object
1093b2f41788SToby Isaac 
1094b2f41788SToby Isaac   Output Parameters:
1095b2f41788SToby Isaac + parentSection - a section describing the tree: a point has a parent if it has 1 dof in the section; the section
1096b2f41788SToby Isaac                   offset indexes the parent and childID list
1097b2f41788SToby Isaac . parents - a list of the point parents
1098b2f41788SToby Isaac . childIDs - identifies the relationship of the child point to the parent point; if there is a reference tree, then
1099b2f41788SToby Isaac              the child corresponds to the point in the reference tree with index childID
1100b2f41788SToby Isaac . childSection - the inverse of the parent section
1101b2f41788SToby Isaac - children - a list of the point children
1102b2f41788SToby Isaac 
1103b2f41788SToby Isaac   Level: intermediate
1104b2f41788SToby Isaac 
1105a17985deSToby Isaac .seealso: DMPlexSetTree(), DMPlexSetReferenceTree(), DMPlexSetAnchors(), DMPlexGetTreeParent(), DMPlexGetTreeChildren()
1106b2f41788SToby Isaac @*/
1107b2f41788SToby Isaac PetscErrorCode DMPlexGetTree(DM dm, PetscSection *parentSection, PetscInt *parents[], PetscInt *childIDs[], PetscSection *childSection, PetscInt *children[])
1108b2f41788SToby Isaac {
1109b2f41788SToby Isaac   DM_Plex        *mesh = (DM_Plex *)dm->data;
1110b2f41788SToby Isaac 
1111b2f41788SToby Isaac   PetscFunctionBegin;
1112b2f41788SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1113b2f41788SToby Isaac   if (parentSection) *parentSection = mesh->parentSection;
1114b2f41788SToby Isaac   if (parents)       *parents       = mesh->parents;
1115b2f41788SToby Isaac   if (childIDs)      *childIDs      = mesh->childIDs;
1116b2f41788SToby Isaac   if (childSection)  *childSection  = mesh->childSection;
1117b2f41788SToby Isaac   if (children)      *children      = mesh->children;
1118b2f41788SToby Isaac   PetscFunctionReturn(0);
1119b2f41788SToby Isaac }
1120b2f41788SToby Isaac 
1121b2f41788SToby Isaac #undef __FUNCT__
1122d961a43aSToby Isaac #define __FUNCT__ "DMPlexGetTreeParent"
1123d961a43aSToby Isaac /*@
1124d961a43aSToby Isaac   DMPlexGetTreeParent - get the parent of a point in the tree describing the point hierarchy (not the Sieve DAG)
1125d961a43aSToby Isaac 
1126d961a43aSToby Isaac   Input Parameters:
1127d961a43aSToby Isaac + dm - the DMPlex object
1128d961a43aSToby Isaac - point - the query point
1129d961a43aSToby Isaac 
1130d961a43aSToby Isaac   Output Parameters:
1131d961a43aSToby Isaac + parent - if not NULL, set to the parent of the point, or the point itself if the point does not have a parent
1132d961a43aSToby Isaac - childID - if not NULL, set to the child ID of the point with respect to its parent, or 0 if the point
1133d961a43aSToby Isaac             does not have a parent
1134d961a43aSToby Isaac 
1135d961a43aSToby Isaac   Level: intermediate
1136d961a43aSToby Isaac 
1137d961a43aSToby Isaac .seealso: DMPlexSetTree(), DMPlexGetTree(), DMPlexGetTreeChildren()
1138d961a43aSToby Isaac @*/
1139d961a43aSToby Isaac PetscErrorCode DMPlexGetTreeParent(DM dm, PetscInt point, PetscInt *parent, PetscInt *childID)
1140d961a43aSToby Isaac {
1141d961a43aSToby Isaac   DM_Plex       *mesh = (DM_Plex *)dm->data;
1142d961a43aSToby Isaac   PetscSection   pSec;
1143d961a43aSToby Isaac   PetscErrorCode ierr;
1144d961a43aSToby Isaac 
1145d961a43aSToby Isaac   PetscFunctionBegin;
1146d961a43aSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1147d961a43aSToby Isaac   pSec = mesh->parentSection;
1148d961a43aSToby Isaac   if (pSec && point >= pSec->pStart && point < pSec->pEnd) {
1149d961a43aSToby Isaac     PetscInt dof;
1150d961a43aSToby Isaac 
1151d961a43aSToby Isaac     ierr = PetscSectionGetDof (pSec, point, &dof);CHKERRQ(ierr);
1152d961a43aSToby Isaac     if (dof) {
1153d961a43aSToby Isaac       PetscInt off;
1154d961a43aSToby Isaac 
1155d961a43aSToby Isaac       ierr = PetscSectionGetOffset (pSec, point, &off);CHKERRQ(ierr);
1156d961a43aSToby Isaac       if (parent)  *parent = mesh->parents[off];
1157d961a43aSToby Isaac       if (childID) *childID = mesh->childIDs[off];
1158d961a43aSToby Isaac       PetscFunctionReturn(0);
1159d961a43aSToby Isaac     }
1160d961a43aSToby Isaac   }
1161d961a43aSToby Isaac   if (parent) {
1162d961a43aSToby Isaac     *parent = point;
1163d961a43aSToby Isaac   }
1164d961a43aSToby Isaac   if (childID) {
1165d961a43aSToby Isaac     *childID = 0;
1166d961a43aSToby Isaac   }
1167d961a43aSToby Isaac   PetscFunctionReturn(0);
1168d961a43aSToby Isaac }
1169d961a43aSToby Isaac 
1170d961a43aSToby Isaac #undef __FUNCT__
1171d961a43aSToby Isaac #define __FUNCT__ "DMPlexGetTreeChildren"
1172d961a43aSToby Isaac /*@C
1173d961a43aSToby Isaac   DMPlexGetTreeChildren - get the children of a point in the tree describing the point hierarchy (not the Sieve DAG)
1174d961a43aSToby Isaac 
1175d961a43aSToby Isaac   Input Parameters:
1176d961a43aSToby Isaac + dm - the DMPlex object
1177d961a43aSToby Isaac - point - the query point
1178d961a43aSToby Isaac 
1179d961a43aSToby Isaac   Output Parameters:
1180d961a43aSToby Isaac + numChildren - if not NULL, set to the number of children
1181d961a43aSToby Isaac - children - if not NULL, set to a list children, or set to NULL if the point has no children
1182d961a43aSToby Isaac 
1183d961a43aSToby Isaac   Level: intermediate
1184d961a43aSToby Isaac 
1185d961a43aSToby Isaac   Fortran Notes:
1186d961a43aSToby Isaac   Since it returns an array, this routine is only available in Fortran 90, and you must
1187d961a43aSToby Isaac   include petsc.h90 in your code.
1188d961a43aSToby Isaac 
1189d961a43aSToby Isaac .seealso: DMPlexSetTree(), DMPlexGetTree(), DMPlexGetTreeParent()
1190d961a43aSToby Isaac @*/
1191d961a43aSToby Isaac PetscErrorCode DMPlexGetTreeChildren(DM dm, PetscInt point, PetscInt *numChildren, const PetscInt *children[])
1192d961a43aSToby Isaac {
1193d961a43aSToby Isaac   DM_Plex       *mesh = (DM_Plex *)dm->data;
1194d961a43aSToby Isaac   PetscSection   childSec;
1195d961a43aSToby Isaac   PetscInt       dof = 0;
1196d961a43aSToby Isaac   PetscErrorCode ierr;
1197d961a43aSToby Isaac 
1198d961a43aSToby Isaac   PetscFunctionBegin;
1199d961a43aSToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1200d961a43aSToby Isaac   childSec = mesh->childSection;
1201d961a43aSToby Isaac   if (childSec && point >= childSec->pStart && point < childSec->pEnd) {
1202d961a43aSToby Isaac     ierr = PetscSectionGetDof (childSec, point, &dof);CHKERRQ(ierr);
1203d961a43aSToby Isaac   }
1204d961a43aSToby Isaac   if (numChildren) *numChildren = dof;
1205d961a43aSToby Isaac   if (children) {
1206d961a43aSToby Isaac     if (dof) {
1207d961a43aSToby Isaac       PetscInt off;
1208d961a43aSToby Isaac 
1209d961a43aSToby Isaac       ierr = PetscSectionGetOffset (childSec, point, &off);CHKERRQ(ierr);
1210d961a43aSToby Isaac       *children = &mesh->children[off];
1211d961a43aSToby Isaac     }
1212d961a43aSToby Isaac     else {
1213d961a43aSToby Isaac       *children = NULL;
1214d961a43aSToby Isaac     }
1215d961a43aSToby Isaac   }
1216d961a43aSToby Isaac   PetscFunctionReturn(0);
1217d961a43aSToby Isaac }
12180c37af3bSToby Isaac 
12190c37af3bSToby Isaac #undef __FUNCT__
1220f7c74593SToby Isaac #define __FUNCT__ "DMPlexComputeAnchorMatrix_Tree_Direct"
1221f7c74593SToby Isaac static PetscErrorCode DMPlexComputeAnchorMatrix_Tree_Direct(DM dm, PetscSection section, PetscSection cSec, Mat cMat)
12220c37af3bSToby Isaac {
12230c37af3bSToby Isaac   PetscDS        ds;
12240c37af3bSToby Isaac   PetscInt       spdim;
12250c37af3bSToby Isaac   PetscInt       numFields, f, c, cStart, cEnd, pStart, pEnd, conStart, conEnd;
12260c37af3bSToby Isaac   const PetscInt *anchors;
1227f7c74593SToby Isaac   PetscSection   aSec;
12280c37af3bSToby Isaac   PetscReal      *v0, *v0parent, *vtmp, *J, *Jparent, *invJparent, detJ, detJparent;
12290c37af3bSToby Isaac   IS             aIS;
12300c37af3bSToby Isaac   PetscErrorCode ierr;
12310c37af3bSToby Isaac 
12320c37af3bSToby Isaac   PetscFunctionBegin;
12330c37af3bSToby Isaac   ierr = DMPlexGetChart(dm,&pStart,&pEnd);CHKERRQ(ierr);
12340c37af3bSToby Isaac   ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
12350c37af3bSToby Isaac   ierr = PetscDSGetNumFields(ds,&numFields);CHKERRQ(ierr);
12360c37af3bSToby Isaac   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
1237a17985deSToby Isaac   ierr = DMPlexGetAnchors(dm,&aSec,&aIS);CHKERRQ(ierr);
12380c37af3bSToby Isaac   ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
12390c37af3bSToby Isaac   ierr = PetscSectionGetChart(cSec,&conStart,&conEnd);CHKERRQ(ierr);
124028f4b327SMatthew G. Knepley   ierr = DMGetDimension(dm,&spdim);CHKERRQ(ierr);
12410c37af3bSToby Isaac   ierr = PetscMalloc6(spdim,&v0,spdim,&v0parent,spdim,&vtmp,spdim*spdim,&J,spdim*spdim,&Jparent,spdim*spdim,&invJparent);CHKERRQ(ierr);
12420c37af3bSToby Isaac 
12430c37af3bSToby Isaac   for (f = 0; f < numFields; f++) {
12440dd1b1feSToby Isaac     PetscObject disc;
12450dd1b1feSToby Isaac     PetscFE fe = NULL;
12460dd1b1feSToby Isaac     PetscFV fv = NULL;
12470dd1b1feSToby Isaac     PetscClassId id;
12480c37af3bSToby Isaac     PetscDualSpace space;
12490c37af3bSToby Isaac     PetscInt i, j, k, nPoints, offset;
12500c37af3bSToby Isaac     PetscInt fSize, fComp;
1251c111c6b7SMatthew G. Knepley     PetscReal *B = NULL;
12520c37af3bSToby Isaac     PetscReal *weights, *pointsRef, *pointsReal;
12530c37af3bSToby Isaac     Mat Amat, Bmat, Xmat;
12540c37af3bSToby Isaac 
12550dd1b1feSToby Isaac     ierr = PetscDSGetDiscretization(ds,f,&disc);CHKERRQ(ierr);
12560dd1b1feSToby Isaac     ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
12570dd1b1feSToby Isaac     if (id == PETSCFE_CLASSID) {
12580dd1b1feSToby Isaac       fe = (PetscFE) disc;
12590c37af3bSToby Isaac       ierr = PetscFEGetDualSpace(fe,&space);CHKERRQ(ierr);
12600c37af3bSToby Isaac       ierr = PetscDualSpaceGetDimension(space,&fSize);CHKERRQ(ierr);
12610c37af3bSToby Isaac       ierr = PetscFEGetNumComponents(fe,&fComp);CHKERRQ(ierr);
12620dd1b1feSToby Isaac     }
12630dd1b1feSToby Isaac     else if (id == PETSCFV_CLASSID) {
12640dd1b1feSToby Isaac       fv = (PetscFV) disc;
12650dd1b1feSToby Isaac       ierr = PetscFVGetDualSpace(fv,&space);CHKERRQ(ierr);
12660dd1b1feSToby Isaac       ierr = PetscDualSpaceGetDimension(space,&fSize);CHKERRQ(ierr);
12670dd1b1feSToby Isaac       ierr = PetscFVGetNumComponents(fv,&fComp);CHKERRQ(ierr);
12680dd1b1feSToby Isaac     }
12690dd1b1feSToby Isaac     else SETERRQ1(PetscObjectComm(disc),PETSC_ERR_ARG_UNKNOWN_TYPE, "PetscDS discretization id %d not recognized.", id);
12700dd1b1feSToby Isaac 
12710c37af3bSToby Isaac     ierr = MatCreate(PETSC_COMM_SELF,&Amat);CHKERRQ(ierr);
12720c37af3bSToby Isaac     ierr = MatSetSizes(Amat,fSize,fSize,fSize,fSize);CHKERRQ(ierr);
12730c37af3bSToby Isaac     ierr = MatSetType(Amat,MATSEQDENSE);CHKERRQ(ierr);
12740c37af3bSToby Isaac     ierr = MatSetUp(Amat);CHKERRQ(ierr);
12750c37af3bSToby Isaac     ierr = MatDuplicate(Amat,MAT_DO_NOT_COPY_VALUES,&Bmat);CHKERRQ(ierr);
12760c37af3bSToby Isaac     ierr = MatDuplicate(Amat,MAT_DO_NOT_COPY_VALUES,&Xmat);CHKERRQ(ierr);
12770c37af3bSToby Isaac     nPoints = 0;
12780c37af3bSToby Isaac     for (i = 0; i < fSize; i++) {
12790c37af3bSToby Isaac       PetscInt        qPoints;
12800c37af3bSToby Isaac       PetscQuadrature quad;
12810c37af3bSToby Isaac 
12820c37af3bSToby Isaac       ierr = PetscDualSpaceGetFunctional(space,i,&quad);CHKERRQ(ierr);
12830c37af3bSToby Isaac       ierr = PetscQuadratureGetData(quad,NULL,&qPoints,NULL,NULL);CHKERRQ(ierr);
12840c37af3bSToby Isaac       nPoints += qPoints;
12850c37af3bSToby Isaac     }
12860c37af3bSToby Isaac     ierr = PetscMalloc3(nPoints,&weights,spdim*nPoints,&pointsRef,spdim*nPoints,&pointsReal);CHKERRQ(ierr);
12870c37af3bSToby Isaac     offset = 0;
12880c37af3bSToby Isaac     for (i = 0; i < fSize; i++) {
12890c37af3bSToby Isaac       PetscInt        qPoints;
12900c37af3bSToby Isaac       const PetscReal    *p, *w;
12910c37af3bSToby Isaac       PetscQuadrature quad;
12920c37af3bSToby Isaac 
12930c37af3bSToby Isaac       ierr = PetscDualSpaceGetFunctional(space,i,&quad);CHKERRQ(ierr);
12940c37af3bSToby Isaac       ierr = PetscQuadratureGetData(quad,NULL,&qPoints,&p,&w);CHKERRQ(ierr);
12950c37af3bSToby Isaac       ierr = PetscMemcpy(weights+offset,w,qPoints*sizeof(*w));CHKERRQ(ierr);
12960c37af3bSToby Isaac       ierr = PetscMemcpy(pointsRef+spdim*offset,p,spdim*qPoints*sizeof(*p));CHKERRQ(ierr);
12970c37af3bSToby Isaac       offset += qPoints;
12980c37af3bSToby Isaac     }
12990dd1b1feSToby Isaac     if (id == PETSCFE_CLASSID) {
13000c37af3bSToby Isaac       ierr = PetscFEGetTabulation(fe,nPoints,pointsRef,&B,NULL,NULL);CHKERRQ(ierr);
13010dd1b1feSToby Isaac     }
13020dd1b1feSToby Isaac     else {
13030dd1b1feSToby Isaac       ierr = PetscFVGetTabulation(fv,nPoints,pointsRef,&B,NULL,NULL);CHKERRQ(ierr);
13040dd1b1feSToby Isaac     }
13050c37af3bSToby Isaac     offset = 0;
13060c37af3bSToby Isaac     for (i = 0; i < fSize; i++) {
13070c37af3bSToby Isaac       PetscInt        qPoints;
13080c37af3bSToby Isaac       PetscQuadrature quad;
13090c37af3bSToby Isaac 
13100c37af3bSToby Isaac       ierr = PetscDualSpaceGetFunctional(space,i,&quad);CHKERRQ(ierr);
13110c37af3bSToby Isaac       ierr = PetscQuadratureGetData(quad,NULL,&qPoints,NULL,NULL);CHKERRQ(ierr);
13120c37af3bSToby Isaac       for (j = 0; j < fSize; j++) {
13130c37af3bSToby Isaac         PetscScalar val = 0.;
13140c37af3bSToby Isaac 
13150c37af3bSToby Isaac         for (k = 0; k < qPoints; k++) {
13160c37af3bSToby Isaac           val += B[((offset + k) * fSize + j) * fComp] * weights[k];
13170c37af3bSToby Isaac         }
13180c37af3bSToby Isaac         ierr = MatSetValue(Amat,i,j,val,INSERT_VALUES);CHKERRQ(ierr);
13190c37af3bSToby Isaac       }
13200c37af3bSToby Isaac       offset += qPoints;
13210c37af3bSToby Isaac     }
13220dd1b1feSToby Isaac     if (id == PETSCFE_CLASSID) {
13230c37af3bSToby Isaac       ierr = PetscFERestoreTabulation(fe,nPoints,pointsRef,&B,NULL,NULL);CHKERRQ(ierr);
13240dd1b1feSToby Isaac     }
13250dd1b1feSToby Isaac     else {
13260dd1b1feSToby Isaac       ierr = PetscFVRestoreTabulation(fv,nPoints,pointsRef,&B,NULL,NULL);CHKERRQ(ierr);
13270dd1b1feSToby Isaac     }
13280c37af3bSToby Isaac     ierr = MatAssemblyBegin(Amat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13290c37af3bSToby Isaac     ierr = MatAssemblyEnd(Amat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13300c37af3bSToby Isaac     ierr = MatLUFactor(Amat,NULL,NULL,NULL);CHKERRQ(ierr);
13310c37af3bSToby Isaac     for (c = cStart; c < cEnd; c++) {
13320c37af3bSToby Isaac       PetscInt parent;
13330c37af3bSToby Isaac       PetscInt closureSize, closureSizeP, *closure = NULL, *closureP = NULL;
13340c37af3bSToby Isaac       PetscInt *childOffsets, *parentOffsets;
13350c37af3bSToby Isaac 
13360c37af3bSToby Isaac       ierr = DMPlexGetTreeParent(dm,c,&parent,NULL);CHKERRQ(ierr);
13370c37af3bSToby Isaac       if (parent == c) continue;
13380c37af3bSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,c,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
13390c37af3bSToby Isaac       for (i = 0; i < closureSize; i++) {
13400c37af3bSToby Isaac         PetscInt p = closure[2*i];
13410c37af3bSToby Isaac         PetscInt conDof;
13420c37af3bSToby Isaac 
13430c37af3bSToby Isaac         if (p < conStart || p >= conEnd) continue;
13440c37af3bSToby Isaac         if (numFields > 1) {
13450c37af3bSToby Isaac           ierr = PetscSectionGetFieldDof(cSec,p,f,&conDof);CHKERRQ(ierr);
13460c37af3bSToby Isaac         }
13470c37af3bSToby Isaac         else {
13480c37af3bSToby Isaac           ierr = PetscSectionGetDof(cSec,p,&conDof);CHKERRQ(ierr);
13490c37af3bSToby Isaac         }
13500c37af3bSToby Isaac         if (conDof) break;
13510c37af3bSToby Isaac       }
13520c37af3bSToby Isaac       if (i == closureSize) {
13530c37af3bSToby Isaac         ierr = DMPlexRestoreTransitiveClosure(dm,c,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
13540c37af3bSToby Isaac         continue;
13550c37af3bSToby Isaac       }
13560c37af3bSToby Isaac 
135773a7f2aaSMatthew G. Knepley       ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, NULL, &detJ);CHKERRQ(ierr);
135873a7f2aaSMatthew G. Knepley       ierr = DMPlexComputeCellGeometryFEM(dm, parent, NULL, v0parent, Jparent, invJparent, &detJparent);CHKERRQ(ierr);
13590c37af3bSToby Isaac       for (i = 0; i < nPoints; i++) {
13600c37af3bSToby Isaac         CoordinatesRefToReal(spdim, spdim, v0, J, &pointsRef[i*spdim],vtmp);CHKERRQ(ierr);
13610c37af3bSToby Isaac         CoordinatesRealToRef(spdim, spdim, v0parent, invJparent, vtmp, &pointsReal[i*spdim]);CHKERRQ(ierr);
13620c37af3bSToby Isaac       }
13630dd1b1feSToby Isaac       if (id == PETSCFE_CLASSID) {
13640c37af3bSToby Isaac         ierr = PetscFEGetTabulation(fe,nPoints,pointsReal,&B,NULL,NULL);CHKERRQ(ierr);
13650dd1b1feSToby Isaac       }
13660dd1b1feSToby Isaac       else {
13670dd1b1feSToby Isaac         ierr = PetscFVGetTabulation(fv,nPoints,pointsReal,&B,NULL,NULL);CHKERRQ(ierr);
13680dd1b1feSToby Isaac       }
13690c37af3bSToby Isaac       offset = 0;
13700c37af3bSToby Isaac       for (i = 0; i < fSize; i++) {
13710c37af3bSToby Isaac         PetscInt        qPoints;
13720c37af3bSToby Isaac         PetscQuadrature quad;
13730c37af3bSToby Isaac 
13740c37af3bSToby Isaac         ierr = PetscDualSpaceGetFunctional(space,i,&quad);CHKERRQ(ierr);
13750c37af3bSToby Isaac         ierr = PetscQuadratureGetData(quad,NULL,&qPoints,NULL,NULL);CHKERRQ(ierr);
13760c37af3bSToby Isaac         for (j = 0; j < fSize; j++) {
13770c37af3bSToby Isaac           PetscScalar val = 0.;
13780c37af3bSToby Isaac 
13790c37af3bSToby Isaac           for (k = 0; k < qPoints; k++) {
13800c37af3bSToby Isaac             val += B[((offset + k) * fSize + j) * fComp] * weights[k];
13810c37af3bSToby Isaac           }
13820c37af3bSToby Isaac           MatSetValue(Bmat,i,j,val,INSERT_VALUES);CHKERRQ(ierr);
13830c37af3bSToby Isaac         }
13840c37af3bSToby Isaac         offset += qPoints;
13850c37af3bSToby Isaac       }
13860dd1b1feSToby Isaac       if (id == PETSCFE_CLASSID) {
13870c37af3bSToby Isaac         ierr = PetscFERestoreTabulation(fe,nPoints,pointsReal,&B,NULL,NULL);CHKERRQ(ierr);
13880dd1b1feSToby Isaac       }
13890dd1b1feSToby Isaac       else {
13900dd1b1feSToby Isaac         ierr = PetscFVRestoreTabulation(fv,nPoints,pointsReal,&B,NULL,NULL);CHKERRQ(ierr);
13910dd1b1feSToby Isaac       }
13920c37af3bSToby Isaac       ierr = MatAssemblyBegin(Bmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13930c37af3bSToby Isaac       ierr = MatAssemblyEnd(Bmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13940c37af3bSToby Isaac       ierr = MatMatSolve(Amat,Bmat,Xmat);CHKERRQ(ierr);
13950c37af3bSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,parent,PETSC_TRUE,&closureSizeP,&closureP);CHKERRQ(ierr);
13960c37af3bSToby Isaac       ierr = PetscMalloc2(closureSize+1,&childOffsets,closureSizeP+1,&parentOffsets);CHKERRQ(ierr);
13970c37af3bSToby Isaac       childOffsets[0] = 0;
13980c37af3bSToby Isaac       for (i = 0; i < closureSize; i++) {
13990c37af3bSToby Isaac         PetscInt p = closure[2*i];
14000c37af3bSToby Isaac         PetscInt dof;
14010c37af3bSToby Isaac 
14020c37af3bSToby Isaac         if (numFields > 1) {
14030c37af3bSToby Isaac           ierr = PetscSectionGetFieldDof(section,p,f,&dof);CHKERRQ(ierr);
14040c37af3bSToby Isaac         }
14050c37af3bSToby Isaac         else {
14060c37af3bSToby Isaac           ierr = PetscSectionGetDof(section,p,&dof);CHKERRQ(ierr);
14070c37af3bSToby Isaac         }
14080c37af3bSToby Isaac         childOffsets[i+1]=childOffsets[i]+dof / fComp;
14090c37af3bSToby Isaac       }
14100c37af3bSToby Isaac       parentOffsets[0] = 0;
14110c37af3bSToby Isaac       for (i = 0; i < closureSizeP; i++) {
14120c37af3bSToby Isaac         PetscInt p = closureP[2*i];
14130c37af3bSToby Isaac         PetscInt dof;
14140c37af3bSToby Isaac 
14150c37af3bSToby Isaac         if (numFields > 1) {
14160c37af3bSToby Isaac           ierr = PetscSectionGetFieldDof(section,p,f,&dof);CHKERRQ(ierr);
14170c37af3bSToby Isaac         }
14180c37af3bSToby Isaac         else {
14190c37af3bSToby Isaac           ierr = PetscSectionGetDof(section,p,&dof);CHKERRQ(ierr);
14200c37af3bSToby Isaac         }
14210c37af3bSToby Isaac         parentOffsets[i+1]=parentOffsets[i]+dof / fComp;
14220c37af3bSToby Isaac       }
14230c37af3bSToby Isaac       for (i = 0; i < closureSize; i++) {
14240c37af3bSToby Isaac         PetscInt conDof, conOff, aDof, aOff;
14250c37af3bSToby Isaac         PetscInt p = closure[2*i];
14260c37af3bSToby Isaac         PetscInt o = closure[2*i+1];
14270c37af3bSToby Isaac 
14280c37af3bSToby Isaac         if (p < conStart || p >= conEnd) continue;
14290c37af3bSToby Isaac         if (numFields > 1) {
14300c37af3bSToby Isaac           ierr = PetscSectionGetFieldDof(cSec,p,f,&conDof);CHKERRQ(ierr);
14310c37af3bSToby Isaac           ierr = PetscSectionGetFieldOffset(cSec,p,f,&conOff);CHKERRQ(ierr);
14320c37af3bSToby Isaac         }
14330c37af3bSToby Isaac         else {
14340c37af3bSToby Isaac           ierr = PetscSectionGetDof(cSec,p,&conDof);CHKERRQ(ierr);
14350c37af3bSToby Isaac           ierr = PetscSectionGetOffset(cSec,p,&conOff);CHKERRQ(ierr);
14360c37af3bSToby Isaac         }
14370c37af3bSToby Isaac         if (!conDof) continue;
14380c37af3bSToby Isaac         ierr = PetscSectionGetDof(aSec,p,&aDof);CHKERRQ(ierr);
14390c37af3bSToby Isaac         ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
14400c37af3bSToby Isaac         for (k = 0; k < aDof; k++) {
14410c37af3bSToby Isaac           PetscInt a = anchors[aOff + k];
14420c37af3bSToby Isaac           PetscInt aSecDof, aSecOff;
14430c37af3bSToby Isaac 
14440c37af3bSToby Isaac           if (numFields > 1) {
14450c37af3bSToby Isaac             ierr = PetscSectionGetFieldDof(section,a,f,&aSecDof);CHKERRQ(ierr);
14460c37af3bSToby Isaac             ierr = PetscSectionGetFieldOffset(section,a,f,&aSecOff);CHKERRQ(ierr);
14470c37af3bSToby Isaac           }
14480c37af3bSToby Isaac           else {
14490c37af3bSToby Isaac             ierr = PetscSectionGetDof(section,a,&aSecDof);CHKERRQ(ierr);
14500c37af3bSToby Isaac             ierr = PetscSectionGetOffset(section,a,&aSecOff);CHKERRQ(ierr);
14510c37af3bSToby Isaac           }
14520c37af3bSToby Isaac           if (!aSecDof) continue;
14530c37af3bSToby Isaac 
14540c37af3bSToby Isaac           for (j = 0; j < closureSizeP; j++) {
14550c37af3bSToby Isaac             PetscInt q = closureP[2*j];
14560c37af3bSToby Isaac             PetscInt oq = closureP[2*j+1];
14570c37af3bSToby Isaac 
14580c37af3bSToby Isaac             if (q == a) {
14590c37af3bSToby Isaac               PetscInt r, s, t;
14600c37af3bSToby Isaac 
14610c37af3bSToby Isaac               for (r = childOffsets[i]; r < childOffsets[i+1]; r++) {
14620c37af3bSToby Isaac                 for (s = parentOffsets[j]; s < parentOffsets[j+1]; s++) {
14630c37af3bSToby Isaac                   PetscScalar val;
14640c37af3bSToby Isaac                   PetscInt insertCol, insertRow;
14650c37af3bSToby Isaac 
14660c37af3bSToby Isaac                   ierr = MatGetValue(Xmat,r,s,&val);CHKERRQ(ierr);
14670c37af3bSToby Isaac                   if (o >= 0) {
14680c37af3bSToby Isaac                     insertRow = conOff + fComp * (r - childOffsets[i]);
14690c37af3bSToby Isaac                   }
14700c37af3bSToby Isaac                   else {
14710c37af3bSToby Isaac                     insertRow = conOff + fComp * (childOffsets[i + 1] - 1 - r);
14720c37af3bSToby Isaac                   }
14730c37af3bSToby Isaac                   if (oq >= 0) {
14740c37af3bSToby Isaac                     insertCol = aSecOff + fComp * (s - parentOffsets[j]);
14750c37af3bSToby Isaac                   }
14760c37af3bSToby Isaac                   else {
14770c37af3bSToby Isaac                     insertCol = aSecOff + fComp * (parentOffsets[j + 1] - 1 - s);
14780c37af3bSToby Isaac                   }
14790c37af3bSToby Isaac                   for (t = 0; t < fComp; t++) {
14800c37af3bSToby Isaac                     ierr = MatSetValue(cMat,insertRow + t,insertCol + t,val,INSERT_VALUES);CHKERRQ(ierr);
14810c37af3bSToby Isaac                   }
14820c37af3bSToby Isaac                 }
14830c37af3bSToby Isaac               }
14840c37af3bSToby Isaac             }
14850c37af3bSToby Isaac           }
14860c37af3bSToby Isaac         }
14870c37af3bSToby Isaac       }
14880c37af3bSToby Isaac       ierr = PetscFree2(childOffsets,parentOffsets);CHKERRQ(ierr);
14890c37af3bSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,c,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
14900c37af3bSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,parent,PETSC_TRUE,&closureSizeP,&closureP);CHKERRQ(ierr);
14910c37af3bSToby Isaac     }
14920c37af3bSToby Isaac     ierr = MatDestroy(&Amat);CHKERRQ(ierr);
14930c37af3bSToby Isaac     ierr = MatDestroy(&Bmat);CHKERRQ(ierr);
14940c37af3bSToby Isaac     ierr = MatDestroy(&Xmat);CHKERRQ(ierr);
14950c37af3bSToby Isaac     ierr = PetscFree3(weights,pointsRef,pointsReal);CHKERRQ(ierr);
14960c37af3bSToby Isaac   }
14970c37af3bSToby Isaac   ierr = MatAssemblyBegin(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14980c37af3bSToby Isaac   ierr = MatAssemblyEnd(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14990c37af3bSToby Isaac   ierr = PetscFree6(v0,v0parent,vtmp,J,Jparent,invJparent);CHKERRQ(ierr);
15000c37af3bSToby Isaac   ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
15010c37af3bSToby Isaac 
15020c37af3bSToby Isaac   PetscFunctionReturn(0);
15030c37af3bSToby Isaac }
150495a0b26dSToby Isaac 
150595a0b26dSToby Isaac #undef __FUNCT__
150621968bf8SToby Isaac #define __FUNCT__ "DMPlexReferenceTreeGetChildrenMatrices"
150721968bf8SToby Isaac static PetscErrorCode DMPlexReferenceTreeGetChildrenMatrices(DM refTree, PetscScalar ****childrenMats, PetscInt ***childrenN)
150895a0b26dSToby Isaac {
1509f7c74593SToby Isaac   Mat            refCmat;
151021968bf8SToby Isaac   PetscDS        ds;
151121968bf8SToby Isaac   PetscInt       numFields, f, pRefStart, pRefEnd, p, *rows, *cols, maxDof, maxAnDof, **refPointFieldN;
151221968bf8SToby Isaac   PetscScalar    ***refPointFieldMats;
151321968bf8SToby Isaac   PetscSection   refConSec, refAnSec, refSection;
151421968bf8SToby Isaac   IS             refAnIS;
151521968bf8SToby Isaac   const PetscInt *refAnchors;
151695a0b26dSToby Isaac   PetscErrorCode ierr;
151795a0b26dSToby Isaac 
151895a0b26dSToby Isaac   PetscFunctionBegin;
151921968bf8SToby Isaac   ierr = DMGetDS(refTree,&ds);CHKERRQ(ierr);
152095a0b26dSToby Isaac   ierr = PetscDSGetNumFields(ds,&numFields);CHKERRQ(ierr);
1521f7c74593SToby Isaac   ierr = DMGetDefaultConstraints(refTree,&refConSec,&refCmat);CHKERRQ(ierr);
1522a17985deSToby Isaac   ierr = DMPlexGetAnchors(refTree,&refAnSec,&refAnIS);CHKERRQ(ierr);
152395a0b26dSToby Isaac   ierr = ISGetIndices(refAnIS,&refAnchors);CHKERRQ(ierr);
152495a0b26dSToby Isaac   ierr = DMGetDefaultSection(refTree,&refSection);CHKERRQ(ierr);
152595a0b26dSToby Isaac   ierr = PetscSectionGetChart(refConSec,&pRefStart,&pRefEnd);CHKERRQ(ierr);
152695a0b26dSToby Isaac   ierr = PetscMalloc1(pRefEnd-pRefStart,&refPointFieldMats);CHKERRQ(ierr);
152795a0b26dSToby Isaac   ierr = PetscMalloc1(pRefEnd-pRefStart,&refPointFieldN);CHKERRQ(ierr);
152895a0b26dSToby Isaac   ierr = PetscSectionGetMaxDof(refConSec,&maxDof);CHKERRQ(ierr);
152995a0b26dSToby Isaac   ierr = PetscSectionGetMaxDof(refAnSec,&maxAnDof);CHKERRQ(ierr);
153095a0b26dSToby Isaac   ierr = PetscMalloc1(maxDof,&rows);CHKERRQ(ierr);
153195a0b26dSToby Isaac   ierr = PetscMalloc1(maxDof*maxAnDof,&cols);CHKERRQ(ierr);
153295a0b26dSToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
153395a0b26dSToby Isaac     PetscInt parent, closureSize, *closure = NULL, pDof;
153495a0b26dSToby Isaac 
153595a0b26dSToby Isaac     ierr = DMPlexGetTreeParent(refTree,p,&parent,NULL);CHKERRQ(ierr);
153695a0b26dSToby Isaac     ierr = PetscSectionGetDof(refConSec,p,&pDof);CHKERRQ(ierr);
153795a0b26dSToby Isaac     if (!pDof || parent == p) continue;
153895a0b26dSToby Isaac 
153995a0b26dSToby Isaac     ierr = PetscMalloc1(numFields,&refPointFieldMats[p-pRefStart]);CHKERRQ(ierr);
154095a0b26dSToby Isaac     ierr = PetscCalloc1(numFields,&refPointFieldN[p-pRefStart]);CHKERRQ(ierr);
154195a0b26dSToby Isaac     ierr = DMPlexGetTransitiveClosure(refTree,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
154295a0b26dSToby Isaac     for (f = 0; f < numFields; f++) {
154395a0b26dSToby Isaac       PetscInt cDof, cOff, numCols, r, i, fComp;
15440dd1b1feSToby Isaac       PetscObject disc;
15450dd1b1feSToby Isaac       PetscClassId id;
15460dd1b1feSToby Isaac       PetscFE fe = NULL;
15470dd1b1feSToby Isaac       PetscFV fv = NULL;
154895a0b26dSToby Isaac 
15490dd1b1feSToby Isaac       ierr = PetscDSGetDiscretization(ds,f,&disc);CHKERRQ(ierr);
15500dd1b1feSToby Isaac       ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
15510dd1b1feSToby Isaac       if (id == PETSCFE_CLASSID) {
15520dd1b1feSToby Isaac         fe = (PetscFE) disc;
155395a0b26dSToby Isaac         ierr = PetscFEGetNumComponents(fe,&fComp);CHKERRQ(ierr);
15540dd1b1feSToby Isaac       }
15550dd1b1feSToby Isaac       else if (id == PETSCFV_CLASSID) {
15560dd1b1feSToby Isaac         fv = (PetscFV) disc;
15570dd1b1feSToby Isaac         ierr = PetscFVGetNumComponents(fv,&fComp);CHKERRQ(ierr);
15580dd1b1feSToby Isaac       }
15590dd1b1feSToby Isaac       else SETERRQ1(PetscObjectComm(disc),PETSC_ERR_ARG_UNKNOWN_TYPE, "PetscDS discretization id %d not recognized.", id);
156095a0b26dSToby Isaac 
156195a0b26dSToby Isaac       if (numFields > 1) {
156295a0b26dSToby Isaac         ierr = PetscSectionGetFieldDof(refConSec,p,f,&cDof);CHKERRQ(ierr);
156395a0b26dSToby Isaac         ierr = PetscSectionGetFieldOffset(refConSec,p,f,&cOff);CHKERRQ(ierr);
156495a0b26dSToby Isaac       }
156595a0b26dSToby Isaac       else {
156695a0b26dSToby Isaac         ierr = PetscSectionGetDof(refConSec,p,&cDof);CHKERRQ(ierr);
156795a0b26dSToby Isaac         ierr = PetscSectionGetOffset(refConSec,p,&cOff);CHKERRQ(ierr);
156895a0b26dSToby Isaac       }
156995a0b26dSToby Isaac 
157095a0b26dSToby Isaac       if (!cDof) continue;
157195a0b26dSToby Isaac       for (r = 0; r < cDof; r++) {
157295a0b26dSToby Isaac         rows[r] = cOff + r;
157395a0b26dSToby Isaac       }
157495a0b26dSToby Isaac       numCols = 0;
157595a0b26dSToby Isaac       for (i = 0; i < closureSize; i++) {
157695a0b26dSToby Isaac         PetscInt q = closure[2*i];
157795a0b26dSToby Isaac         PetscInt o = closure[2*i+1];
157895a0b26dSToby Isaac         PetscInt aDof, aOff, j;
157995a0b26dSToby Isaac 
158095a0b26dSToby Isaac         if (numFields > 1) {
158195a0b26dSToby Isaac           ierr = PetscSectionGetFieldDof(refSection,q,f,&aDof);CHKERRQ(ierr);
158295a0b26dSToby Isaac           ierr = PetscSectionGetFieldOffset(refSection,q,f,&aOff);CHKERRQ(ierr);
158395a0b26dSToby Isaac         }
158495a0b26dSToby Isaac         else {
158595a0b26dSToby Isaac           ierr = PetscSectionGetDof(refSection,q,&aDof);CHKERRQ(ierr);
158695a0b26dSToby Isaac           ierr = PetscSectionGetOffset(refSection,q,&aOff);CHKERRQ(ierr);
158795a0b26dSToby Isaac         }
158895a0b26dSToby Isaac 
158995a0b26dSToby Isaac         for (j = 0; j < aDof; j++) {
159095a0b26dSToby Isaac           PetscInt node = (o >= 0) ? (j / fComp) : ((aDof - 1 - j) / fComp);
159195a0b26dSToby Isaac           PetscInt comp = (j % fComp);
159295a0b26dSToby Isaac 
159395a0b26dSToby Isaac           cols[numCols++] = aOff + node * fComp + comp;
159495a0b26dSToby Isaac         }
159595a0b26dSToby Isaac       }
159695a0b26dSToby Isaac       refPointFieldN[p-pRefStart][f] = numCols;
159795a0b26dSToby Isaac       ierr = PetscMalloc1(cDof*numCols,&refPointFieldMats[p-pRefStart][f]);CHKERRQ(ierr);
159895a0b26dSToby Isaac       ierr = MatGetValues(refCmat,cDof,rows,numCols,cols,refPointFieldMats[p-pRefStart][f]);CHKERRQ(ierr);
159995a0b26dSToby Isaac     }
160095a0b26dSToby Isaac     ierr = DMPlexRestoreTransitiveClosure(refTree,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
160195a0b26dSToby Isaac   }
160221968bf8SToby Isaac   *childrenMats = refPointFieldMats;
160321968bf8SToby Isaac   *childrenN = refPointFieldN;
160421968bf8SToby Isaac   ierr = ISRestoreIndices(refAnIS,&refAnchors);CHKERRQ(ierr);
160521968bf8SToby Isaac   ierr = PetscFree(rows);CHKERRQ(ierr);
160621968bf8SToby Isaac   ierr = PetscFree(cols);CHKERRQ(ierr);
160721968bf8SToby Isaac   PetscFunctionReturn(0);
160821968bf8SToby Isaac }
160921968bf8SToby Isaac 
161021968bf8SToby Isaac #undef __FUNCT__
161121968bf8SToby Isaac #define __FUNCT__ "DMPlexReferenceTreeRestoreChildrenMatrices"
161221968bf8SToby Isaac static PetscErrorCode DMPlexReferenceTreeRestoreChildrenMatrices(DM refTree, PetscScalar ****childrenMats, PetscInt ***childrenN)
161321968bf8SToby Isaac {
161421968bf8SToby Isaac   PetscDS        ds;
161521968bf8SToby Isaac   PetscInt       **refPointFieldN;
161621968bf8SToby Isaac   PetscScalar    ***refPointFieldMats;
161721968bf8SToby Isaac   PetscInt       numFields, pRefStart, pRefEnd, p, f;
161821968bf8SToby Isaac   PetscSection   refConSec;
161921968bf8SToby Isaac   PetscErrorCode ierr;
162021968bf8SToby Isaac 
162121968bf8SToby Isaac   PetscFunctionBegin;
162221968bf8SToby Isaac   refPointFieldN = *childrenN;
162321968bf8SToby Isaac   *childrenN = NULL;
162421968bf8SToby Isaac   refPointFieldMats = *childrenMats;
162521968bf8SToby Isaac   *childrenMats = NULL;
162621968bf8SToby Isaac   ierr = DMGetDS(refTree,&ds);CHKERRQ(ierr);
162721968bf8SToby Isaac   ierr = PetscDSGetNumFields(ds,&numFields);CHKERRQ(ierr);
162821968bf8SToby Isaac   ierr = DMGetDefaultConstraints(refTree,&refConSec,NULL);CHKERRQ(ierr);
1629e44e4e7fSToby Isaac   ierr = PetscSectionGetChart(refConSec,&pRefStart,&pRefEnd);CHKERRQ(ierr);
163021968bf8SToby Isaac   for (p = pRefStart; p < pRefEnd; p++) {
163121968bf8SToby Isaac     PetscInt parent, pDof;
163221968bf8SToby Isaac 
163321968bf8SToby Isaac     ierr = DMPlexGetTreeParent(refTree,p,&parent,NULL);CHKERRQ(ierr);
163421968bf8SToby Isaac     ierr = PetscSectionGetDof(refConSec,p,&pDof);CHKERRQ(ierr);
163521968bf8SToby Isaac     if (!pDof || parent == p) continue;
163621968bf8SToby Isaac 
163721968bf8SToby Isaac     for (f = 0; f < numFields; f++) {
163821968bf8SToby Isaac       PetscInt cDof;
163921968bf8SToby Isaac 
164021968bf8SToby Isaac       if (numFields > 1) {
164121968bf8SToby Isaac         ierr = PetscSectionGetFieldDof(refConSec,p,f,&cDof);CHKERRQ(ierr);
164221968bf8SToby Isaac       }
164321968bf8SToby Isaac       else {
164421968bf8SToby Isaac         ierr = PetscSectionGetDof(refConSec,p,&cDof);CHKERRQ(ierr);
164521968bf8SToby Isaac       }
164621968bf8SToby Isaac 
164721968bf8SToby Isaac       if (!cDof) continue;
164821968bf8SToby Isaac       ierr = PetscFree(refPointFieldMats[p - pRefStart][f]);CHKERRQ(ierr);
164921968bf8SToby Isaac     }
165021968bf8SToby Isaac     ierr = PetscFree(refPointFieldMats[p - pRefStart]);CHKERRQ(ierr);
165121968bf8SToby Isaac     ierr = PetscFree(refPointFieldN[p - pRefStart]);CHKERRQ(ierr);
165221968bf8SToby Isaac   }
165321968bf8SToby Isaac   ierr = PetscFree(refPointFieldMats);CHKERRQ(ierr);
165421968bf8SToby Isaac   ierr = PetscFree(refPointFieldN);CHKERRQ(ierr);
165521968bf8SToby Isaac   PetscFunctionReturn(0);
165621968bf8SToby Isaac }
165721968bf8SToby Isaac 
165821968bf8SToby Isaac #undef __FUNCT__
165921968bf8SToby Isaac #define __FUNCT__ "DMPlexComputeAnchorMatrix_Tree_FromReference"
166021968bf8SToby Isaac static PetscErrorCode DMPlexComputeAnchorMatrix_Tree_FromReference(DM dm, PetscSection section, PetscSection conSec, Mat cMat)
166121968bf8SToby Isaac {
166221968bf8SToby Isaac   DM             refTree;
166321968bf8SToby Isaac   PetscDS        ds;
166421968bf8SToby Isaac   Mat            refCmat;
166521968bf8SToby Isaac   PetscInt       numFields, f, pRefStart, pRefEnd, p, maxDof, maxAnDof, *perm, *iperm, pStart, pEnd, conStart, conEnd, **refPointFieldN;
166621968bf8SToby Isaac   PetscScalar ***refPointFieldMats, *pointWork;
166721968bf8SToby Isaac   PetscSection   refConSec, refAnSec, anSec;
166821968bf8SToby Isaac   IS             refAnIS, anIS;
166921968bf8SToby Isaac   const PetscInt *anchors;
167021968bf8SToby Isaac   PetscErrorCode ierr;
167121968bf8SToby Isaac 
167221968bf8SToby Isaac   PetscFunctionBegin;
167321968bf8SToby Isaac   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
167421968bf8SToby Isaac   ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
167521968bf8SToby Isaac   ierr = PetscDSGetNumFields(ds,&numFields);CHKERRQ(ierr);
167621968bf8SToby Isaac   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
167721968bf8SToby Isaac   ierr = DMSetDS(refTree,ds);CHKERRQ(ierr);
167821968bf8SToby Isaac   ierr = DMGetDefaultConstraints(refTree,&refConSec,&refCmat);CHKERRQ(ierr);
167921968bf8SToby Isaac   ierr = DMPlexGetAnchors(refTree,&refAnSec,&refAnIS);CHKERRQ(ierr);
168021968bf8SToby Isaac   ierr = DMPlexGetAnchors(dm,&anSec,&anIS);CHKERRQ(ierr);
168121968bf8SToby Isaac   ierr = ISGetIndices(anIS,&anchors);CHKERRQ(ierr);
168221968bf8SToby Isaac   ierr = PetscSectionGetChart(refConSec,&pRefStart,&pRefEnd);CHKERRQ(ierr);
168321968bf8SToby Isaac   ierr = PetscSectionGetChart(conSec,&conStart,&conEnd);CHKERRQ(ierr);
168421968bf8SToby Isaac   ierr = PetscSectionGetMaxDof(refConSec,&maxDof);CHKERRQ(ierr);
168521968bf8SToby Isaac   ierr = PetscSectionGetMaxDof(refAnSec,&maxAnDof);CHKERRQ(ierr);
168621968bf8SToby Isaac   ierr = PetscMalloc1(maxDof*maxDof*maxAnDof,&pointWork);CHKERRQ(ierr);
168721968bf8SToby Isaac 
168821968bf8SToby Isaac   /* step 1: get submats for every constrained point in the reference tree */
168921968bf8SToby Isaac   ierr = DMPlexReferenceTreeGetChildrenMatrices(refTree,&refPointFieldMats,&refPointFieldN);CHKERRQ(ierr);
169095a0b26dSToby Isaac 
169195a0b26dSToby Isaac   /* step 2: compute the preorder */
169295a0b26dSToby Isaac   ierr = DMPlexGetChart(dm,&pStart,&pEnd);CHKERRQ(ierr);
169395a0b26dSToby Isaac   ierr = PetscMalloc2(pEnd-pStart,&perm,pEnd-pStart,&iperm);CHKERRQ(ierr);
169495a0b26dSToby Isaac   for (p = pStart; p < pEnd; p++) {
169595a0b26dSToby Isaac     perm[p - pStart] = p;
169695a0b26dSToby Isaac     iperm[p - pStart] = p-pStart;
169795a0b26dSToby Isaac   }
169895a0b26dSToby Isaac   for (p = 0; p < pEnd - pStart;) {
169995a0b26dSToby Isaac     PetscInt point = perm[p];
170095a0b26dSToby Isaac     PetscInt parent;
170195a0b26dSToby Isaac 
170295a0b26dSToby Isaac     ierr = DMPlexGetTreeParent(dm,point,&parent,NULL);CHKERRQ(ierr);
170395a0b26dSToby Isaac     if (parent == point) {
170495a0b26dSToby Isaac       p++;
170595a0b26dSToby Isaac     }
170695a0b26dSToby Isaac     else {
170795a0b26dSToby Isaac       PetscInt size, closureSize, *closure = NULL, i;
170895a0b26dSToby Isaac 
170995a0b26dSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
171095a0b26dSToby Isaac       for (i = 0; i < closureSize; i++) {
171195a0b26dSToby Isaac         PetscInt q = closure[2*i];
171295a0b26dSToby Isaac         if (iperm[q-pStart] > iperm[point-pStart]) {
171395a0b26dSToby Isaac           /* swap */
171495a0b26dSToby Isaac           perm[p]               = q;
171595a0b26dSToby Isaac           perm[iperm[q-pStart]] = point;
171695a0b26dSToby Isaac           iperm[point-pStart]   = iperm[q-pStart];
171795a0b26dSToby Isaac           iperm[q-pStart]       = p;
171895a0b26dSToby Isaac           break;
171995a0b26dSToby Isaac         }
172095a0b26dSToby Isaac       }
172195a0b26dSToby Isaac       size = closureSize;
172295a0b26dSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
172395a0b26dSToby Isaac       if (i == size) {
172495a0b26dSToby Isaac         p++;
172595a0b26dSToby Isaac       }
172695a0b26dSToby Isaac     }
172795a0b26dSToby Isaac   }
172895a0b26dSToby Isaac 
172995a0b26dSToby Isaac   /* step 3: fill the constraint matrix */
173095a0b26dSToby Isaac   /* we are going to use a preorder progressive fill strategy.  Mat doesn't
173195a0b26dSToby Isaac    * allow progressive fill without assembly, so we are going to set up the
173295a0b26dSToby Isaac    * values outside of the Mat first.
173395a0b26dSToby Isaac    */
173495a0b26dSToby Isaac   {
173595a0b26dSToby Isaac     PetscInt nRows, row, nnz;
173695a0b26dSToby Isaac     PetscBool done;
173795a0b26dSToby Isaac     const PetscInt *ia, *ja;
173895a0b26dSToby Isaac     PetscScalar *vals;
173995a0b26dSToby Isaac 
174095a0b26dSToby Isaac     ierr = MatGetRowIJ(cMat,0,PETSC_FALSE,PETSC_FALSE,&nRows,&ia,&ja,&done);CHKERRQ(ierr);
174195a0b26dSToby Isaac     if (!done) SETERRQ(PetscObjectComm((PetscObject)cMat),PETSC_ERR_PLIB,"Could not get RowIJ of constraint matrix");
174295a0b26dSToby Isaac     nnz  = ia[nRows];
174395a0b26dSToby Isaac     /* malloc and then zero rows right before we fill them: this way valgrind
174495a0b26dSToby Isaac      * can tell if we are doing progressive fill in the wrong order */
174595a0b26dSToby Isaac     ierr = PetscMalloc1(nnz,&vals);CHKERRQ(ierr);
174695a0b26dSToby Isaac     for (p = 0; p < pEnd - pStart; p++) {
174795a0b26dSToby Isaac       PetscInt parent, childid, closureSize, *closure = NULL;
174895a0b26dSToby Isaac       PetscInt point = perm[p], pointDof;
174995a0b26dSToby Isaac 
175095a0b26dSToby Isaac       ierr = DMPlexGetTreeParent(dm,point,&parent,&childid);CHKERRQ(ierr);
175195a0b26dSToby Isaac       if ((point < conStart) || (point >= conEnd) || (parent == point)) continue;
175295a0b26dSToby Isaac       ierr = PetscSectionGetDof(conSec,point,&pointDof);CHKERRQ(ierr);
175395a0b26dSToby Isaac       if (!pointDof) continue;
175495a0b26dSToby Isaac       ierr = DMPlexGetTransitiveClosure(dm,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
175595a0b26dSToby Isaac       for (f = 0; f < numFields; f++) {
175695a0b26dSToby Isaac         PetscInt cDof, cOff, numCols, numFillCols, i, r, fComp, matOffset, offset;
175795a0b26dSToby Isaac         PetscScalar *pointMat;
17580dd1b1feSToby Isaac         PetscObject disc;
17590dd1b1feSToby Isaac         PetscClassId id;
17600dd1b1feSToby Isaac         PetscFE fe = NULL;
17610dd1b1feSToby Isaac         PetscFV fv = NULL;
176295a0b26dSToby Isaac 
17630dd1b1feSToby Isaac         ierr = PetscDSGetDiscretization(ds,f,&disc);CHKERRQ(ierr);
17640dd1b1feSToby Isaac         ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
17650dd1b1feSToby Isaac         if (id == PETSCFE_CLASSID) {
17660dd1b1feSToby Isaac           fe = (PetscFE) disc;
176795a0b26dSToby Isaac           ierr = PetscFEGetNumComponents(fe,&fComp);CHKERRQ(ierr);
17680dd1b1feSToby Isaac         }
17690dd1b1feSToby Isaac         else if (id == PETSCFV_CLASSID) {
17700dd1b1feSToby Isaac           fv = (PetscFV) disc;
17710dd1b1feSToby Isaac           ierr = PetscFVGetNumComponents(fv,&fComp);CHKERRQ(ierr);
17720dd1b1feSToby Isaac         }
177395a0b26dSToby Isaac 
177495a0b26dSToby Isaac         if (numFields > 1) {
177595a0b26dSToby Isaac           ierr = PetscSectionGetFieldDof(conSec,point,f,&cDof);CHKERRQ(ierr);
177695a0b26dSToby Isaac           ierr = PetscSectionGetFieldOffset(conSec,point,f,&cOff);CHKERRQ(ierr);
177795a0b26dSToby Isaac         }
177895a0b26dSToby Isaac         else {
177995a0b26dSToby Isaac           ierr = PetscSectionGetDof(conSec,point,&cDof);CHKERRQ(ierr);
178095a0b26dSToby Isaac           ierr = PetscSectionGetOffset(conSec,point,&cOff);CHKERRQ(ierr);
178195a0b26dSToby Isaac         }
178295a0b26dSToby Isaac         if (!cDof) continue;
178395a0b26dSToby Isaac 
178495a0b26dSToby Isaac         /* make sure that every row for this point is the same size */
178595a0b26dSToby Isaac #if defined(PETSC_USE_DEBUG)
178695a0b26dSToby Isaac         for (r = 0; r < cDof; r++) {
178795a0b26dSToby Isaac           if (cDof > 1 && r) {
178821968bf8SToby Isaac             if ((ia[cOff+r+1]-ia[cOff+r]) != (ia[cOff+r]-ia[cOff+r-1])) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Two point rows have different nnz: %D vs. %D", (ia[cOff+r+1]-ia[cOff+r]), (ia[cOff+r]-ia[cOff+r-1]));
178995a0b26dSToby Isaac           }
179095a0b26dSToby Isaac         }
179195a0b26dSToby Isaac #endif
179295a0b26dSToby Isaac         /* zero rows */
179395a0b26dSToby Isaac         for (i = ia[cOff] ; i< ia[cOff+cDof];i++) {
179495a0b26dSToby Isaac           vals[i] = 0.;
179595a0b26dSToby Isaac         }
179695a0b26dSToby Isaac         matOffset = ia[cOff];
179795a0b26dSToby Isaac         numFillCols = ia[cOff+1] - matOffset;
179895a0b26dSToby Isaac         pointMat = refPointFieldMats[childid-pRefStart][f];
179995a0b26dSToby Isaac         numCols = refPointFieldN[childid-pRefStart][f];
180095a0b26dSToby Isaac         offset = 0;
180195a0b26dSToby Isaac         for (i = 0; i < closureSize; i++) {
180295a0b26dSToby Isaac           PetscInt q = closure[2*i];
180395a0b26dSToby Isaac           PetscInt o = closure[2*i+1];
180495a0b26dSToby Isaac           PetscInt aDof, aOff, j, k, qConDof, qConOff;
180595a0b26dSToby Isaac 
180695a0b26dSToby Isaac           qConDof = qConOff = 0;
180795a0b26dSToby Isaac           if (numFields > 1) {
180895a0b26dSToby Isaac             ierr = PetscSectionGetFieldDof(section,q,f,&aDof);CHKERRQ(ierr);
180995a0b26dSToby Isaac             ierr = PetscSectionGetFieldOffset(section,q,f,&aOff);CHKERRQ(ierr);
181095a0b26dSToby Isaac             if (q >= conStart && q < conEnd) {
181195a0b26dSToby Isaac               ierr = PetscSectionGetFieldDof(conSec,q,f,&qConDof);CHKERRQ(ierr);
181295a0b26dSToby Isaac               ierr = PetscSectionGetFieldOffset(conSec,q,f,&qConOff);CHKERRQ(ierr);
181395a0b26dSToby Isaac             }
181495a0b26dSToby Isaac           }
181595a0b26dSToby Isaac           else {
181695a0b26dSToby Isaac             ierr = PetscSectionGetDof(section,q,&aDof);CHKERRQ(ierr);
181795a0b26dSToby Isaac             ierr = PetscSectionGetOffset(section,q,&aOff);CHKERRQ(ierr);
181895a0b26dSToby Isaac             if (q >= conStart && q < conEnd) {
181995a0b26dSToby Isaac               ierr = PetscSectionGetDof(conSec,q,&qConDof);CHKERRQ(ierr);
182095a0b26dSToby Isaac               ierr = PetscSectionGetOffset(conSec,q,&qConOff);CHKERRQ(ierr);
182195a0b26dSToby Isaac             }
182295a0b26dSToby Isaac           }
182395a0b26dSToby Isaac           if (!aDof) continue;
182495a0b26dSToby Isaac           if (qConDof) {
182595a0b26dSToby Isaac             /* this point has anchors: its rows of the matrix should already
182695a0b26dSToby Isaac              * be filled, thanks to preordering */
182795a0b26dSToby Isaac             /* first multiply into pointWork, then set in matrix */
182895a0b26dSToby Isaac             PetscInt aMatOffset = ia[qConOff];
182995a0b26dSToby Isaac             PetscInt aNumFillCols = ia[qConOff + 1] - aMatOffset;
183095a0b26dSToby Isaac             for (r = 0; r < cDof; r++) {
183195a0b26dSToby Isaac               for (j = 0; j < aNumFillCols; j++) {
183295a0b26dSToby Isaac                 PetscScalar inVal = 0;
183395a0b26dSToby Isaac                 for (k = 0; k < aDof; k++) {
183495a0b26dSToby Isaac                   PetscInt node = (o >= 0) ? (k / fComp) : ((aDof - 1 - k) / fComp);
183595a0b26dSToby Isaac                   PetscInt comp = (k % fComp);
183695a0b26dSToby Isaac                   PetscInt col  = node * fComp + comp;
183795a0b26dSToby Isaac 
183895a0b26dSToby Isaac                   inVal += pointMat[r * numCols + offset + col] * vals[aMatOffset + aNumFillCols * k + j];
183995a0b26dSToby Isaac                 }
184095a0b26dSToby Isaac                 pointWork[r * aNumFillCols + j] = inVal;
184195a0b26dSToby Isaac               }
184295a0b26dSToby Isaac             }
184395a0b26dSToby Isaac             /* assume that the columns are sorted, spend less time searching */
184495a0b26dSToby Isaac             for (j = 0, k = 0; j < aNumFillCols; j++) {
184595a0b26dSToby Isaac               PetscInt col = ja[aMatOffset + j];
184695a0b26dSToby Isaac               for (;k < numFillCols; k++) {
184795a0b26dSToby Isaac                 if (ja[matOffset + k] == col) {
184895a0b26dSToby Isaac                   break;
184995a0b26dSToby Isaac                 }
185095a0b26dSToby Isaac               }
185195a0b26dSToby Isaac               if (k == numFillCols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"No nonzero space for (%d, %d)", cOff, col);
185295a0b26dSToby Isaac               for (r = 0; r < cDof; r++) {
185395a0b26dSToby Isaac                 vals[matOffset + numFillCols * r + k] = pointWork[r * aNumFillCols + j];
185495a0b26dSToby Isaac               }
185595a0b26dSToby Isaac             }
185695a0b26dSToby Isaac           }
185795a0b26dSToby Isaac           else {
185895a0b26dSToby Isaac             /* find where to put this portion of pointMat into the matrix */
185995a0b26dSToby Isaac             for (k = 0; k < numFillCols; k++) {
186095a0b26dSToby Isaac               if (ja[matOffset + k] == aOff) {
186195a0b26dSToby Isaac                 break;
186295a0b26dSToby Isaac               }
186395a0b26dSToby Isaac             }
186495a0b26dSToby Isaac             if (k == numFillCols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"No nonzero space for (%d, %d)", cOff, aOff);
186595a0b26dSToby Isaac             for (j = 0; j < aDof; j++) {
186695a0b26dSToby Isaac               PetscInt node = (o >= 0) ? (j / fComp) : ((aDof - 1 - j) / fComp);
186795a0b26dSToby Isaac               PetscInt comp = (j % fComp);
186895a0b26dSToby Isaac               PetscInt col  = node * fComp + comp;
186995a0b26dSToby Isaac               for (r = 0; r < cDof; r++) {
187095a0b26dSToby Isaac                 vals[matOffset + numFillCols * r + k + col] += pointMat[r * numCols + offset + col];
187195a0b26dSToby Isaac               }
187295a0b26dSToby Isaac             }
187395a0b26dSToby Isaac           }
187495a0b26dSToby Isaac           offset += aDof;
187595a0b26dSToby Isaac         }
187695a0b26dSToby Isaac       }
187795a0b26dSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(dm,parent,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
187895a0b26dSToby Isaac     }
187995a0b26dSToby Isaac     for (row = 0; row < nRows; row++) {
188095a0b26dSToby Isaac       ierr = MatSetValues(cMat,1,&row,ia[row+1]-ia[row],&ja[ia[row]],&vals[ia[row]],INSERT_VALUES);CHKERRQ(ierr);
188195a0b26dSToby Isaac     }
188295a0b26dSToby Isaac     ierr = MatRestoreRowIJ(cMat,0,PETSC_FALSE,PETSC_FALSE,&nRows,&ia,&ja,&done);CHKERRQ(ierr);
188395a0b26dSToby Isaac     if (!done) SETERRQ(PetscObjectComm((PetscObject)cMat),PETSC_ERR_PLIB,"Could not restore RowIJ of constraint matrix");
188495a0b26dSToby Isaac     ierr = MatAssemblyBegin(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
188595a0b26dSToby Isaac     ierr = MatAssemblyEnd(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
188695a0b26dSToby Isaac     ierr = PetscFree(vals);CHKERRQ(ierr);
188795a0b26dSToby Isaac   }
188895a0b26dSToby Isaac 
188995a0b26dSToby Isaac   /* clean up */
189095a0b26dSToby Isaac   ierr = ISRestoreIndices(anIS,&anchors);CHKERRQ(ierr);
189195a0b26dSToby Isaac   ierr = PetscFree2(perm,iperm);CHKERRQ(ierr);
189295a0b26dSToby Isaac   ierr = PetscFree(pointWork);CHKERRQ(ierr);
189321968bf8SToby Isaac   ierr = DMPlexReferenceTreeRestoreChildrenMatrices(refTree,&refPointFieldMats,&refPointFieldN);CHKERRQ(ierr);
189495a0b26dSToby Isaac   PetscFunctionReturn(0);
189595a0b26dSToby Isaac }
189695a0b26dSToby Isaac 
18976f5f1567SToby Isaac #undef __FUNCT__
18986f5f1567SToby Isaac #define __FUNCT__ "DMPlexTreeRefineCell"
18996f5f1567SToby Isaac /* refine a single cell on rank 0: this is not intended to provide good local refinement, only to create an example of
19006f5f1567SToby Isaac  * a non-conforming mesh.  Local refinement comes later */
19016f5f1567SToby Isaac PetscErrorCode DMPlexTreeRefineCell (DM dm, PetscInt cell, DM *ncdm)
19026f5f1567SToby Isaac {
19036f5f1567SToby Isaac   DM K;
1904420f55faSMatthew G. Knepley   PetscMPIInt rank;
19056f5f1567SToby Isaac   PetscInt dim, *pNewStart, *pNewEnd, *pNewCount, *pOldStart, *pOldEnd, offset, d, pStart, pEnd;
19066f5f1567SToby Isaac   PetscInt numNewCones, *newConeSizes, *newCones, *newOrientations;
19076f5f1567SToby Isaac   PetscInt *Kembedding;
19086f5f1567SToby Isaac   PetscInt *cellClosure=NULL, nc;
19096f5f1567SToby Isaac   PetscScalar *newVertexCoords;
19106f5f1567SToby Isaac   PetscInt numPointsWithParents, *parents, *childIDs, *perm, *iperm, *preOrient, pOffset;
19116f5f1567SToby Isaac   PetscSection parentSection;
19126f5f1567SToby Isaac   PetscErrorCode ierr;
19136f5f1567SToby Isaac 
19146f5f1567SToby Isaac   PetscFunctionBegin;
19156f5f1567SToby Isaac   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm),&rank);CHKERRQ(ierr);
191628f4b327SMatthew G. Knepley   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
19176f5f1567SToby Isaac   ierr = DMPlexCreate(PetscObjectComm((PetscObject)dm), ncdm);CHKERRQ(ierr);
191828f4b327SMatthew G. Knepley   ierr = DMSetDimension(*ncdm,dim);CHKERRQ(ierr);
19196f5f1567SToby Isaac 
19206f5f1567SToby Isaac   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
19216f5f1567SToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)dm),&parentSection);CHKERRQ(ierr);
19226f5f1567SToby Isaac   ierr = DMPlexGetReferenceTree(dm,&K);CHKERRQ(ierr);
19236f5f1567SToby Isaac   if (!rank) {
19246f5f1567SToby Isaac     /* compute the new charts */
19256f5f1567SToby Isaac     ierr = PetscMalloc5(dim+1,&pNewCount,dim+1,&pNewStart,dim+1,&pNewEnd,dim+1,&pOldStart,dim+1,&pOldEnd);CHKERRQ(ierr);
19266f5f1567SToby Isaac     offset = 0;
19276f5f1567SToby Isaac     for (d = 0; d <= dim; d++) {
19286f5f1567SToby Isaac       PetscInt pOldCount, kStart, kEnd, k;
19296f5f1567SToby Isaac 
19306f5f1567SToby Isaac       pNewStart[d] = offset;
19316f5f1567SToby Isaac       ierr = DMPlexGetHeightStratum(dm,d,&pOldStart[d],&pOldEnd[d]);CHKERRQ(ierr);
19326f5f1567SToby Isaac       ierr = DMPlexGetHeightStratum(K,d,&kStart,&kEnd);CHKERRQ(ierr);
19336f5f1567SToby Isaac       pOldCount = pOldEnd[d] - pOldStart[d];
19346f5f1567SToby Isaac       /* adding the new points */
19356f5f1567SToby Isaac       pNewCount[d] = pOldCount + kEnd - kStart;
19366f5f1567SToby Isaac       if (!d) {
19376f5f1567SToby Isaac         /* removing the cell */
19386f5f1567SToby Isaac         pNewCount[d]--;
19396f5f1567SToby Isaac       }
19406f5f1567SToby Isaac       for (k = kStart; k < kEnd; k++) {
19416f5f1567SToby Isaac         PetscInt parent;
19426f5f1567SToby Isaac         ierr = DMPlexGetTreeParent(K,k,&parent,NULL);CHKERRQ(ierr);
19436f5f1567SToby Isaac         if (parent == k) {
19446f5f1567SToby Isaac           /* avoid double counting points that won't actually be new */
19456f5f1567SToby Isaac           pNewCount[d]--;
19466f5f1567SToby Isaac         }
19476f5f1567SToby Isaac       }
19486f5f1567SToby Isaac       pNewEnd[d] = pNewStart[d] + pNewCount[d];
19496f5f1567SToby Isaac       offset = pNewEnd[d];
19506f5f1567SToby Isaac 
19516f5f1567SToby Isaac     }
19526f5f1567SToby Isaac     if (cell < pOldStart[0] || cell >= pOldEnd[0]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"%d not in cell range [%d, %d)", cell, pOldStart[0], pOldEnd[0]);
19536f5f1567SToby Isaac     /* get the current closure of the cell that we are removing */
19546f5f1567SToby Isaac     ierr = DMPlexGetTransitiveClosure(dm,cell,PETSC_TRUE,&nc,&cellClosure);CHKERRQ(ierr);
19556f5f1567SToby Isaac 
19566f5f1567SToby Isaac     ierr = PetscMalloc1(pNewEnd[dim],&newConeSizes);CHKERRQ(ierr);
19576f5f1567SToby Isaac     {
19586f5f1567SToby Isaac       PetscInt kStart, kEnd, k, closureSizeK, *closureK = NULL, j;
19596f5f1567SToby Isaac 
19606f5f1567SToby Isaac       ierr = DMPlexGetChart(K,&kStart,&kEnd);CHKERRQ(ierr);
19616f5f1567SToby Isaac       ierr = PetscMalloc4(kEnd-kStart,&Kembedding,kEnd-kStart,&perm,kEnd-kStart,&iperm,kEnd-kStart,&preOrient);CHKERRQ(ierr);
19626f5f1567SToby Isaac 
19636f5f1567SToby Isaac       for (k = kStart; k < kEnd; k++) {
19646f5f1567SToby Isaac         perm[k - kStart] = k;
19656f5f1567SToby Isaac         iperm [k - kStart] = k - kStart;
19666f5f1567SToby Isaac         preOrient[k - kStart] = 0;
19676f5f1567SToby Isaac       }
19686f5f1567SToby Isaac 
19696f5f1567SToby Isaac       ierr = DMPlexGetTransitiveClosure(K,0,PETSC_TRUE,&closureSizeK,&closureK);CHKERRQ(ierr);
19706f5f1567SToby Isaac       for (j = 1; j < closureSizeK; j++) {
19716f5f1567SToby Isaac         PetscInt parentOrientA = closureK[2*j+1];
19726f5f1567SToby Isaac         PetscInt parentOrientB = cellClosure[2*j+1];
19736f5f1567SToby Isaac         PetscInt p, q;
19746f5f1567SToby Isaac 
19756f5f1567SToby Isaac         p = closureK[2*j];
19766f5f1567SToby Isaac         q = cellClosure[2*j];
19776f5f1567SToby Isaac         for (d = 0; d <= dim; d++) {
19786f5f1567SToby Isaac           if (q >= pOldStart[d] && q < pOldEnd[d]) {
19796f5f1567SToby Isaac             Kembedding[p] = (q - pOldStart[d]) + pNewStart[d];
19806f5f1567SToby Isaac           }
19816f5f1567SToby Isaac         }
19826f5f1567SToby Isaac         if (parentOrientA != parentOrientB) {
19836f5f1567SToby Isaac           PetscInt numChildren, i;
19846f5f1567SToby Isaac           const PetscInt *children;
19856f5f1567SToby Isaac 
19866f5f1567SToby Isaac           ierr = DMPlexGetTreeChildren(K,p,&numChildren,&children);CHKERRQ(ierr);
19876f5f1567SToby Isaac           for (i = 0; i < numChildren; i++) {
19886f5f1567SToby Isaac             PetscInt kPerm, oPerm;
19896f5f1567SToby Isaac 
19906f5f1567SToby Isaac             k    = children[i];
19916f5f1567SToby Isaac             ierr = DMPlexReferenceTreeGetChildSymmetry(K,p,parentOrientA,0,k,parentOrientB,&oPerm,&kPerm);CHKERRQ(ierr);
19926f5f1567SToby Isaac             /* perm = what refTree position I'm in */
19936f5f1567SToby Isaac             perm[kPerm-kStart]      = k;
19946f5f1567SToby Isaac             /* iperm = who is at this position */
19956f5f1567SToby Isaac             iperm[k-kStart]         = kPerm-kStart;
19966f5f1567SToby Isaac             preOrient[kPerm-kStart] = oPerm;
19976f5f1567SToby Isaac           }
19986f5f1567SToby Isaac         }
19996f5f1567SToby Isaac       }
20006f5f1567SToby Isaac       ierr = DMPlexRestoreTransitiveClosure(K,0,PETSC_TRUE,&closureSizeK,&closureK);CHKERRQ(ierr);
20016f5f1567SToby Isaac     }
20026f5f1567SToby Isaac     ierr = PetscSectionSetChart(parentSection,0,pNewEnd[dim]);CHKERRQ(ierr);
20036f5f1567SToby Isaac     offset = 0;
20046f5f1567SToby Isaac     numNewCones = 0;
20056f5f1567SToby Isaac     for (d = 0; d <= dim; d++) {
20066f5f1567SToby Isaac       PetscInt kStart, kEnd, k;
20076f5f1567SToby Isaac       PetscInt p;
20086f5f1567SToby Isaac       PetscInt size;
20096f5f1567SToby Isaac 
20106f5f1567SToby Isaac       for (p = pOldStart[d]; p < pOldEnd[d]; p++) {
20116f5f1567SToby Isaac         /* skip cell 0 */
20126f5f1567SToby Isaac         if (p == cell) continue;
20136f5f1567SToby Isaac         /* old cones to new cones */
20146f5f1567SToby Isaac         ierr = DMPlexGetConeSize(dm,p,&size);CHKERRQ(ierr);
20156f5f1567SToby Isaac         newConeSizes[offset++] = size;
20166f5f1567SToby Isaac         numNewCones += size;
20176f5f1567SToby Isaac       }
20186f5f1567SToby Isaac 
20196f5f1567SToby Isaac       ierr = DMPlexGetHeightStratum(K,d,&kStart,&kEnd);CHKERRQ(ierr);
20206f5f1567SToby Isaac       for (k = kStart; k < kEnd; k++) {
20216f5f1567SToby Isaac         PetscInt kParent;
20226f5f1567SToby Isaac 
20236f5f1567SToby Isaac         ierr = DMPlexGetTreeParent(K,k,&kParent,NULL);CHKERRQ(ierr);
20246f5f1567SToby Isaac         if (kParent != k) {
20256f5f1567SToby Isaac           Kembedding[k] = offset;
20266f5f1567SToby Isaac           ierr = DMPlexGetConeSize(K,k,&size);CHKERRQ(ierr);
20276f5f1567SToby Isaac           newConeSizes[offset++] = size;
20286f5f1567SToby Isaac           numNewCones += size;
20296f5f1567SToby Isaac           if (kParent != 0) {
20306f5f1567SToby Isaac             ierr = PetscSectionSetDof(parentSection,Kembedding[k],1);CHKERRQ(ierr);
20316f5f1567SToby Isaac           }
20326f5f1567SToby Isaac         }
20336f5f1567SToby Isaac       }
20346f5f1567SToby Isaac     }
20356f5f1567SToby Isaac 
20366f5f1567SToby Isaac     ierr = PetscSectionSetUp(parentSection);CHKERRQ(ierr);
20376f5f1567SToby Isaac     ierr = PetscSectionGetStorageSize(parentSection,&numPointsWithParents);CHKERRQ(ierr);
20386f5f1567SToby Isaac     ierr = PetscMalloc2(numNewCones,&newCones,numNewCones,&newOrientations);CHKERRQ(ierr);
20396f5f1567SToby Isaac     ierr = PetscMalloc2(numPointsWithParents,&parents,numPointsWithParents,&childIDs);CHKERRQ(ierr);
20406f5f1567SToby Isaac 
20416f5f1567SToby Isaac     /* fill new cones */
20426f5f1567SToby Isaac     offset = 0;
20436f5f1567SToby Isaac     for (d = 0; d <= dim; d++) {
20446f5f1567SToby Isaac       PetscInt kStart, kEnd, k, l;
20456f5f1567SToby Isaac       PetscInt p;
20466f5f1567SToby Isaac       PetscInt size;
20476f5f1567SToby Isaac       const PetscInt *cone, *orientation;
20486f5f1567SToby Isaac 
20496f5f1567SToby Isaac       for (p = pOldStart[d]; p < pOldEnd[d]; p++) {
20506f5f1567SToby Isaac         /* skip cell 0 */
20516f5f1567SToby Isaac         if (p == cell) continue;
20526f5f1567SToby Isaac         /* old cones to new cones */
20536f5f1567SToby Isaac         ierr = DMPlexGetConeSize(dm,p,&size);CHKERRQ(ierr);
20546f5f1567SToby Isaac         ierr = DMPlexGetCone(dm,p,&cone);CHKERRQ(ierr);
20556f5f1567SToby Isaac         ierr = DMPlexGetConeOrientation(dm,p,&orientation);CHKERRQ(ierr);
20566f5f1567SToby Isaac         for (l = 0; l < size; l++) {
20576f5f1567SToby Isaac           newCones[offset]          = (cone[l] - pOldStart[d + 1]) + pNewStart[d + 1];
20586f5f1567SToby Isaac           newOrientations[offset++] = orientation[l];
20596f5f1567SToby Isaac         }
20606f5f1567SToby Isaac       }
20616f5f1567SToby Isaac 
20626f5f1567SToby Isaac       ierr = DMPlexGetHeightStratum(K,d,&kStart,&kEnd);CHKERRQ(ierr);
20636f5f1567SToby Isaac       for (k = kStart; k < kEnd; k++) {
20646f5f1567SToby Isaac         PetscInt kPerm = perm[k], kParent;
20656f5f1567SToby Isaac         PetscInt preO  = preOrient[k];
20666f5f1567SToby Isaac 
20676f5f1567SToby Isaac         ierr = DMPlexGetTreeParent(K,k,&kParent,NULL);CHKERRQ(ierr);
20686f5f1567SToby Isaac         if (kParent != k) {
20696f5f1567SToby Isaac           /* embed new cones */
20706f5f1567SToby Isaac           ierr = DMPlexGetConeSize(K,k,&size);CHKERRQ(ierr);
20716f5f1567SToby Isaac           ierr = DMPlexGetCone(K,kPerm,&cone);CHKERRQ(ierr);
20726f5f1567SToby Isaac           ierr = DMPlexGetConeOrientation(K,kPerm,&orientation);CHKERRQ(ierr);
20736f5f1567SToby Isaac           for (l = 0; l < size; l++) {
20746f5f1567SToby Isaac             PetscInt q, m = (preO >= 0) ? ((preO + l) % size) : ((size -(preO + 1) - l) % size);
20756f5f1567SToby Isaac             PetscInt newO, lSize, oTrue;
20766f5f1567SToby Isaac 
20776f5f1567SToby Isaac             q                         = iperm[cone[m]];
20786f5f1567SToby Isaac             newCones[offset]          = Kembedding[q];
20796f5f1567SToby Isaac             ierr                      = DMPlexGetConeSize(K,q,&lSize);CHKERRQ(ierr);
20806f5f1567SToby Isaac             oTrue                     = orientation[m];
20816f5f1567SToby Isaac             oTrue                     = ((!lSize) || (preOrient[k] >= 0)) ? oTrue : -(oTrue + 2);
20826f5f1567SToby Isaac             newO                      = DihedralCompose(lSize,oTrue,preOrient[q]);
20836f5f1567SToby Isaac             newOrientations[offset++] = newO;
20846f5f1567SToby Isaac           }
20856f5f1567SToby Isaac           if (kParent != 0) {
20866f5f1567SToby Isaac             PetscInt newPoint = Kembedding[kParent];
20876f5f1567SToby Isaac             ierr              = PetscSectionGetOffset(parentSection,Kembedding[k],&pOffset);CHKERRQ(ierr);
20886f5f1567SToby Isaac             parents[pOffset]  = newPoint;
20896f5f1567SToby Isaac             childIDs[pOffset] = k;
20906f5f1567SToby Isaac           }
20916f5f1567SToby Isaac         }
20926f5f1567SToby Isaac       }
20936f5f1567SToby Isaac     }
20946f5f1567SToby Isaac 
20956f5f1567SToby Isaac     ierr = PetscMalloc1(dim*(pNewEnd[dim]-pNewStart[dim]),&newVertexCoords);CHKERRQ(ierr);
20966f5f1567SToby Isaac 
20976f5f1567SToby Isaac     /* fill coordinates */
20986f5f1567SToby Isaac     offset = 0;
20996f5f1567SToby Isaac     {
2100d90620a3SMatthew G. Knepley       PetscInt kStart, kEnd, l;
21016f5f1567SToby Isaac       PetscSection vSection;
21026f5f1567SToby Isaac       PetscInt v;
21036f5f1567SToby Isaac       Vec coords;
21046f5f1567SToby Isaac       PetscScalar *coordvals;
21056f5f1567SToby Isaac       PetscInt dof, off;
2106c111c6b7SMatthew G. Knepley       PetscReal v0[3], J[9], detJ;
21076f5f1567SToby Isaac 
21086f5f1567SToby Isaac #if defined(PETSC_USE_DEBUG)
2109d90620a3SMatthew G. Knepley       {
2110d90620a3SMatthew G. Knepley         PetscInt k;
21116f5f1567SToby Isaac         ierr = DMPlexGetHeightStratum(K,0,&kStart,&kEnd);CHKERRQ(ierr);
21126f5f1567SToby Isaac         for (k = kStart; k < kEnd; k++) {
211373a7f2aaSMatthew G. Knepley           ierr = DMPlexComputeCellGeometryFEM(K, k, NULL, v0, J, NULL, &detJ);CHKERRQ(ierr);
21146f5f1567SToby Isaac           if (detJ <= 0.) SETERRQ1 (PETSC_COMM_SELF,PETSC_ERR_PLIB,"reference tree cell %d has bad determinant",k);
21156f5f1567SToby Isaac         }
2116d90620a3SMatthew G. Knepley       }
21176f5f1567SToby Isaac #endif
211873a7f2aaSMatthew G. Knepley       ierr = DMPlexComputeCellGeometryFEM(dm, cell, NULL, v0, J, NULL, &detJ);CHKERRQ(ierr);
21196f5f1567SToby Isaac       ierr = DMGetCoordinateSection(dm,&vSection);CHKERRQ(ierr);
21206f5f1567SToby Isaac       ierr = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr);
21216f5f1567SToby Isaac       ierr = VecGetArray(coords,&coordvals);CHKERRQ(ierr);
21226f5f1567SToby Isaac       for (v = pOldStart[dim]; v < pOldEnd[dim]; v++) {
21236f5f1567SToby Isaac 
21246f5f1567SToby Isaac         ierr = PetscSectionGetDof(vSection,v,&dof);CHKERRQ(ierr);
21256f5f1567SToby Isaac         ierr = PetscSectionGetOffset(vSection,v,&off);CHKERRQ(ierr);
21266f5f1567SToby Isaac         for (l = 0; l < dof; l++) {
21276f5f1567SToby Isaac           newVertexCoords[offset++] = coordvals[off + l];
21286f5f1567SToby Isaac         }
21296f5f1567SToby Isaac       }
21306f5f1567SToby Isaac       ierr = VecRestoreArray(coords,&coordvals);CHKERRQ(ierr);
21316f5f1567SToby Isaac 
21326f5f1567SToby Isaac       ierr = DMGetCoordinateSection(K,&vSection);CHKERRQ(ierr);
21336f5f1567SToby Isaac       ierr = DMGetCoordinatesLocal(K,&coords);CHKERRQ(ierr);
21346f5f1567SToby Isaac       ierr = VecGetArray(coords,&coordvals);CHKERRQ(ierr);
21356f5f1567SToby Isaac       ierr = DMPlexGetDepthStratum(K,0,&kStart,&kEnd);CHKERRQ(ierr);
21366f5f1567SToby Isaac       for (v = kStart; v < kEnd; v++) {
21379bc368c7SMatthew G. Knepley         PetscReal coord[3], newCoord[3];
21386f5f1567SToby Isaac         PetscInt  vPerm = perm[v];
21396f5f1567SToby Isaac         PetscInt  kParent;
21406f5f1567SToby Isaac 
21416f5f1567SToby Isaac         ierr = DMPlexGetTreeParent(K,v,&kParent,NULL);CHKERRQ(ierr);
21426f5f1567SToby Isaac         if (kParent != v) {
21436f5f1567SToby Isaac           /* this is a new vertex */
21446f5f1567SToby Isaac           ierr = PetscSectionGetOffset(vSection,vPerm,&off);CHKERRQ(ierr);
21459bc368c7SMatthew G. Knepley           for (l = 0; l < dim; ++l) coord[l] = PetscRealPart(coordvals[off+l]);
21469bc368c7SMatthew G. Knepley           CoordinatesRefToReal(dim, dim, v0, J, coord, newCoord);CHKERRQ(ierr);
21479bc368c7SMatthew G. Knepley           for (l = 0; l < dim; ++l) newVertexCoords[offset+l] = newCoord[l];
21486f5f1567SToby Isaac           offset += dim;
21496f5f1567SToby Isaac         }
21506f5f1567SToby Isaac       }
21516f5f1567SToby Isaac       ierr = VecRestoreArray(coords,&coordvals);CHKERRQ(ierr);
21526f5f1567SToby Isaac     }
21536f5f1567SToby Isaac 
21546f5f1567SToby Isaac     /* need to reverse the order of pNewCount: vertices first, cells last */
21556f5f1567SToby Isaac     for (d = 0; d < (dim + 1) / 2; d++) {
21566f5f1567SToby Isaac       PetscInt tmp;
21576f5f1567SToby Isaac 
21586f5f1567SToby Isaac       tmp = pNewCount[d];
21596f5f1567SToby Isaac       pNewCount[d] = pNewCount[dim - d];
21606f5f1567SToby Isaac       pNewCount[dim - d] = tmp;
21616f5f1567SToby Isaac     }
21626f5f1567SToby Isaac 
21636f5f1567SToby Isaac     ierr = DMPlexCreateFromDAG(*ncdm,dim,pNewCount,newConeSizes,newCones,newOrientations,newVertexCoords);CHKERRQ(ierr);
21646f5f1567SToby Isaac     ierr = DMPlexSetReferenceTree(*ncdm,K);CHKERRQ(ierr);
21656f5f1567SToby Isaac     ierr = DMPlexSetTree(*ncdm,parentSection,parents,childIDs);CHKERRQ(ierr);
21666f5f1567SToby Isaac 
21676f5f1567SToby Isaac     /* clean up */
21686f5f1567SToby Isaac     ierr = DMPlexRestoreTransitiveClosure(dm,cell,PETSC_TRUE,&nc,&cellClosure);CHKERRQ(ierr);
21696f5f1567SToby Isaac     ierr = PetscFree5(pNewCount,pNewStart,pNewEnd,pOldStart,pOldEnd);CHKERRQ(ierr);
21706f5f1567SToby Isaac     ierr = PetscFree(newConeSizes);CHKERRQ(ierr);
21716f5f1567SToby Isaac     ierr = PetscFree2(newCones,newOrientations);CHKERRQ(ierr);
21726f5f1567SToby Isaac     ierr = PetscFree(newVertexCoords);CHKERRQ(ierr);
21736f5f1567SToby Isaac     ierr = PetscFree2(parents,childIDs);CHKERRQ(ierr);
21746f5f1567SToby Isaac     ierr = PetscFree4(Kembedding,perm,iperm,preOrient);CHKERRQ(ierr);
21756f5f1567SToby Isaac   }
21766f5f1567SToby Isaac   else {
21776f5f1567SToby Isaac     PetscInt    p, counts[4];
21786f5f1567SToby Isaac     PetscInt    *coneSizes, *cones, *orientations;
21796f5f1567SToby Isaac     Vec         coordVec;
21806f5f1567SToby Isaac     PetscScalar *coords;
21816f5f1567SToby Isaac 
21826f5f1567SToby Isaac     for (d = 0; d <= dim; d++) {
21836f5f1567SToby Isaac       PetscInt dStart, dEnd;
21846f5f1567SToby Isaac 
21856f5f1567SToby Isaac       ierr = DMPlexGetDepthStratum(dm,d,&dStart,&dEnd);CHKERRQ(ierr);
21866f5f1567SToby Isaac       counts[d] = dEnd - dStart;
21876f5f1567SToby Isaac     }
21886f5f1567SToby Isaac     ierr = PetscMalloc1(pEnd-pStart,&coneSizes);CHKERRQ(ierr);
21896f5f1567SToby Isaac     for (p = pStart; p < pEnd; p++) {
21906f5f1567SToby Isaac       ierr = DMPlexGetConeSize(dm,p,&coneSizes[p-pStart]);CHKERRQ(ierr);
21916f5f1567SToby Isaac     }
21926f5f1567SToby Isaac     ierr = DMPlexGetCones(dm, &cones);CHKERRQ(ierr);
21936f5f1567SToby Isaac     ierr = DMPlexGetConeOrientations(dm, &orientations);CHKERRQ(ierr);
21946f5f1567SToby Isaac     ierr = DMGetCoordinatesLocal(dm,&coordVec);CHKERRQ(ierr);
21956f5f1567SToby Isaac     ierr = VecGetArray(coordVec,&coords);CHKERRQ(ierr);
21966f5f1567SToby Isaac 
21976f5f1567SToby Isaac     ierr = PetscSectionSetChart(parentSection,pStart,pEnd);CHKERRQ(ierr);
21986f5f1567SToby Isaac     ierr = PetscSectionSetUp(parentSection);CHKERRQ(ierr);
21996f5f1567SToby Isaac     ierr = DMPlexCreateFromDAG(*ncdm,dim,counts,coneSizes,cones,orientations,NULL);CHKERRQ(ierr);
22006f5f1567SToby Isaac     ierr = DMPlexSetReferenceTree(*ncdm,K);CHKERRQ(ierr);
22016f5f1567SToby Isaac     ierr = DMPlexSetTree(*ncdm,parentSection,NULL,NULL);CHKERRQ(ierr);
22026f5f1567SToby Isaac     ierr = VecRestoreArray(coordVec,&coords);CHKERRQ(ierr);
22036f5f1567SToby Isaac   }
22046f5f1567SToby Isaac   ierr = PetscSectionDestroy(&parentSection);CHKERRQ(ierr);
22056f5f1567SToby Isaac 
22066f5f1567SToby Isaac   PetscFunctionReturn(0);
22076f5f1567SToby Isaac }
22086ecaa68aSToby Isaac 
22096ecaa68aSToby Isaac PETSC_EXTERN PetscErrorCode indicesPointFields_private(PetscSection,PetscInt,PetscInt,PetscInt[],PetscBool,PetscInt,PetscInt[]);
22106ecaa68aSToby Isaac PETSC_EXTERN PetscErrorCode indicesPoint_private(PetscSection,PetscInt,PetscInt,PetscInt *,PetscBool,PetscInt,PetscInt[]);
221121968bf8SToby Isaac 
22126ecaa68aSToby Isaac #undef __FUNCT__
22136ecaa68aSToby Isaac #define __FUNCT__ "DMPlexComputeInterpolatorTree"
22146ecaa68aSToby Isaac PetscErrorCode DMPlexComputeInterpolatorTree(DM coarse, DM fine, PetscSF coarseToFine, PetscInt *childIds, Mat mat)
22156ecaa68aSToby Isaac {
22166ecaa68aSToby Isaac   PetscSF        coarseToFineEmbedded;
22176ecaa68aSToby Isaac   PetscSection   globalCoarse, globalFine;
22186ecaa68aSToby Isaac   PetscSection   localCoarse, localFine;
22196ecaa68aSToby Isaac   PetscSection   aSec, cSec;
22206ecaa68aSToby Isaac   PetscSection   rootIndicesSec, rootMatricesSec;
222146bdb399SToby Isaac   PetscSection   leafIndicesSec, leafMatricesSec;
222246bdb399SToby Isaac   PetscInt       *rootIndices, *leafIndices;
222346bdb399SToby Isaac   PetscScalar    *rootMatrices, *leafMatrices;
22246ecaa68aSToby Isaac   IS             aIS;
22256ecaa68aSToby Isaac   const PetscInt *anchors;
22266ecaa68aSToby Isaac   Mat            cMat;
22276ecaa68aSToby Isaac   PetscInt       numFields;
22286ecaa68aSToby Isaac   PetscInt       pStartC, pEndC, pStartF, pEndF, p;
22296ecaa68aSToby Isaac   PetscInt       aStart, aEnd, cStart, cEnd;
22301c58ffc4SToby Isaac   PetscInt       *maxChildIds;
2231e44e4e7fSToby Isaac   PetscInt       *offsets, *newOffsets, *offsetsCopy, *newOffsetsCopy, *rowOffsets, *numD, *numO;
22326ecaa68aSToby Isaac   PetscErrorCode ierr;
22336ecaa68aSToby Isaac 
22346ecaa68aSToby Isaac   PetscFunctionBegin;
22356ecaa68aSToby Isaac   ierr = DMPlexGetChart(coarse,&pStartC,&pEndC);CHKERRQ(ierr);
22366ecaa68aSToby Isaac   ierr = DMPlexGetChart(fine,&pStartF,&pEndF);CHKERRQ(ierr);
22376ecaa68aSToby Isaac   ierr = DMGetDefaultGlobalSection(fine,&globalFine);CHKERRQ(ierr);
22386ecaa68aSToby Isaac   { /* winnow fine points that don't have global dofs out of the sf */
22396ecaa68aSToby Isaac     PetscInt dof, cdof, numPointsWithDofs, offset, *pointsWithDofs;
22406ecaa68aSToby Isaac 
22416ecaa68aSToby Isaac     for (p = pStartF, numPointsWithDofs = 0; p < pEndF; p++) {
22426ecaa68aSToby Isaac       ierr = PetscSectionGetDof(globalFine,p,&dof);CHKERRQ(ierr);
22436ecaa68aSToby Isaac       ierr = PetscSectionGetConstraintDof(globalFine,p,&cdof);CHKERRQ(ierr);
22446ecaa68aSToby Isaac       if ((dof - cdof) > 0) {
22456ecaa68aSToby Isaac         numPointsWithDofs++;
22466ecaa68aSToby Isaac       }
22476ecaa68aSToby Isaac     }
22486ecaa68aSToby Isaac     ierr = PetscMalloc1(numPointsWithDofs,&pointsWithDofs);CHKERRQ(ierr);
22496ecaa68aSToby Isaac     for (p = pStartF, offset = 0; p < pEndF; p++) {
22506ecaa68aSToby Isaac       ierr = PetscSectionGetDof(globalFine,p,&dof);CHKERRQ(ierr);
22516ecaa68aSToby Isaac       ierr = PetscSectionGetConstraintDof(globalFine,p,&cdof);CHKERRQ(ierr);
22526ecaa68aSToby Isaac       if ((dof - cdof) > 0) {
22536ecaa68aSToby Isaac         pointsWithDofs[offset++] = p - pStartF;
22546ecaa68aSToby Isaac       }
22556ecaa68aSToby Isaac     }
22566ecaa68aSToby Isaac     ierr = PetscSFCreateEmbeddedLeafSF(coarseToFine, numPointsWithDofs, pointsWithDofs, &coarseToFineEmbedded);CHKERRQ(ierr);
22576ecaa68aSToby Isaac   }
22586ecaa68aSToby Isaac   /* communicate back to the coarse mesh which coarse points have children (that may require interpolation) */
22596ecaa68aSToby Isaac   ierr = PetscMalloc1(pEndC-pStartC,&maxChildIds);CHKERRQ(ierr);
22606ecaa68aSToby Isaac   for (p = pStartC; p < pEndC; p++) {
22616ecaa68aSToby Isaac     maxChildIds[p - pStartC] = -1;
22626ecaa68aSToby Isaac   }
22636ecaa68aSToby Isaac   ierr = PetscSFReduceBegin(coarseToFineEmbedded,MPIU_INT,childIds,maxChildIds,MPIU_MAX);CHKERRQ(ierr);
22646ecaa68aSToby Isaac   ierr = PetscSFReduceEnd(coarseToFineEmbedded,MPIU_INT,childIds,maxChildIds,MPIU_MAX);CHKERRQ(ierr);
226546bdb399SToby Isaac 
22666ecaa68aSToby Isaac   ierr = DMGetDefaultSection(coarse,&localCoarse);CHKERRQ(ierr);
22676ecaa68aSToby Isaac   ierr = DMGetDefaultGlobalSection(coarse,&globalCoarse);CHKERRQ(ierr);
226846bdb399SToby Isaac 
22696ecaa68aSToby Isaac   ierr = DMPlexGetAnchors(coarse,&aSec,&aIS);CHKERRQ(ierr);
22706ecaa68aSToby Isaac   ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
22716ecaa68aSToby Isaac   ierr = PetscSectionGetChart(aSec,&aStart,&aEnd);CHKERRQ(ierr);
227246bdb399SToby Isaac 
22736ecaa68aSToby Isaac   ierr = DMGetDefaultConstraints(coarse,&cSec,&cMat);CHKERRQ(ierr);
22746ecaa68aSToby Isaac   ierr = PetscSectionGetChart(cSec,&cStart,&cEnd);CHKERRQ(ierr);
227546bdb399SToby Isaac 
227646bdb399SToby Isaac   /* create sections that will send to children the indices and matrices they will need to construct the interpolator */
22776ecaa68aSToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)coarse),&rootIndicesSec);CHKERRQ(ierr);
22786ecaa68aSToby Isaac   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)coarse),&rootMatricesSec);CHKERRQ(ierr);
22796ecaa68aSToby Isaac   ierr = PetscSectionSetChart(rootIndicesSec,pStartC,pEndC);CHKERRQ(ierr);
22806ecaa68aSToby Isaac   ierr = PetscSectionSetChart(rootMatricesSec,pStartC,pEndC);CHKERRQ(ierr);
22816ecaa68aSToby Isaac   ierr = PetscSectionGetNumFields(globalCoarse,&numFields);CHKERRQ(ierr);
228221968bf8SToby Isaac   {
2283267d4f3fSToby Isaac     PetscInt maxFields = PetscMax(1,numFields) + 1;
2284e44e4e7fSToby Isaac     ierr = PetscMalloc7(maxFields,&offsets,maxFields,&offsetsCopy,maxFields,&newOffsets,maxFields,&newOffsetsCopy,maxFields,&rowOffsets,maxFields,&numD,maxFields,&numO);CHKERRQ(ierr);
228521968bf8SToby Isaac   }
228646bdb399SToby Isaac 
228746bdb399SToby Isaac   for (p = pStartC; p < pEndC; p++) { /* count the sizes of the indices and matrices */
22886ecaa68aSToby Isaac     PetscInt dof, matSize;
22896ecaa68aSToby Isaac     PetscInt aDof           = 0;
22906ecaa68aSToby Isaac     PetscInt cDof           = 0;
22916ecaa68aSToby Isaac     PetscInt maxChildId     = maxChildIds[p - pStartC];
22926ecaa68aSToby Isaac     PetscInt numRowIndices  = 0;
22936ecaa68aSToby Isaac     PetscInt numColIndices  = 0;
22946ecaa68aSToby Isaac 
22956ecaa68aSToby Isaac     ierr = PetscSectionGetDof(globalCoarse,p,&dof);CHKERRQ(ierr);
22961cfc5b76SToby Isaac     if (dof < 0) {
22971cfc5b76SToby Isaac       dof = -(dof + 1);
22981cfc5b76SToby Isaac     }
22996ecaa68aSToby Isaac     if (p >= aStart && p < aEnd) {
23006ecaa68aSToby Isaac       ierr = PetscSectionGetDof(aSec,p,&aDof);CHKERRQ(ierr);
23016ecaa68aSToby Isaac     }
23026ecaa68aSToby Isaac     if (p >= cStart && p < cEnd) {
23036ecaa68aSToby Isaac       ierr = PetscSectionGetDof(cSec,p,&cDof);CHKERRQ(ierr);
23046ecaa68aSToby Isaac     }
230521968bf8SToby Isaac     offsets[0]    = 0;
230621968bf8SToby Isaac     newOffsets[0] = 0;
23076ecaa68aSToby Isaac     if (maxChildId >= 0) { /* this point has children (with dofs) that will need to be interpolated from the closure of p */
23086ecaa68aSToby Isaac       PetscInt *closure = NULL, closureSize, cl, f;
23096ecaa68aSToby Isaac 
23106ecaa68aSToby Isaac       ierr = DMPlexGetTransitiveClosure(coarse,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
231146bdb399SToby Isaac       for (cl = 0; cl < closureSize; cl++) { /* get the closure */
23126ecaa68aSToby Isaac         PetscInt c = closure[2 * cl], clDof;
23136ecaa68aSToby Isaac 
23146ecaa68aSToby Isaac         ierr = PetscSectionGetDof(localCoarse,c,&clDof);CHKERRQ(ierr);
23156ecaa68aSToby Isaac         numRowIndices += clDof;
23166ecaa68aSToby Isaac         for (f = 0; f < numFields; f++) {
23176ecaa68aSToby Isaac           ierr = PetscSectionGetFieldDof(localCoarse,c,f,&clDof);CHKERRQ(ierr);
23186ecaa68aSToby Isaac           offsets[f + 1] += clDof;
23196ecaa68aSToby Isaac         }
23206ecaa68aSToby Isaac       }
23216ecaa68aSToby Isaac       for (f = 0; f < numFields; f++) {
23226ecaa68aSToby Isaac         offsets[f + 1]   += offsets[f];
23236ecaa68aSToby Isaac         newOffsets[f + 1] = offsets[f + 1];
23246ecaa68aSToby Isaac       }
232546bdb399SToby Isaac       /* get the number of indices needed and their field offsets */
23266ecaa68aSToby Isaac       ierr = DMPlexAnchorsModifyMat(coarse,localCoarse,closureSize,numRowIndices,closure,NULL,NULL,&numColIndices,NULL,NULL,newOffsets,PETSC_FALSE);CHKERRQ(ierr);
23276ecaa68aSToby Isaac       ierr = DMPlexRestoreTransitiveClosure(coarse,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
23286ecaa68aSToby Isaac       if (!numColIndices) { /* there are no hanging constraint modifications, so the matrix is just the identity: do not send it */
23296ecaa68aSToby Isaac         numColIndices = numRowIndices;
23306ecaa68aSToby Isaac         matSize = 0;
23316ecaa68aSToby Isaac       }
233246bdb399SToby Isaac       else if (numFields) { /* we send one submat for each field: sum their sizes */
23336ecaa68aSToby Isaac         matSize = 0;
23346ecaa68aSToby Isaac         for (f = 0; f < numFields; f++) {
23356ecaa68aSToby Isaac           PetscInt numRow, numCol;
23366ecaa68aSToby Isaac 
23376ecaa68aSToby Isaac           numRow = offsets[f + 1] - offsets[f];
23386ecaa68aSToby Isaac           numCol = newOffsets[f + 1] - newOffsets[f + 1];
23396ecaa68aSToby Isaac           matSize += numRow * numCol;
23406ecaa68aSToby Isaac         }
23416ecaa68aSToby Isaac       }
23426ecaa68aSToby Isaac       else {
23436ecaa68aSToby Isaac         matSize = numRowIndices * numColIndices;
23446ecaa68aSToby Isaac       }
23456ecaa68aSToby Isaac     }
23466ecaa68aSToby Isaac     else if (cDof > 0) { /* this point's dofs are interpolated via cMat: get the submatrix of cMat */
23476ecaa68aSToby Isaac       PetscInt aOff, a, f;
23486ecaa68aSToby Isaac 
23496ecaa68aSToby Isaac       ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
23506ecaa68aSToby Isaac       for (f = 0; f < numFields; f++) {
23516ecaa68aSToby Isaac         PetscInt fDof;
23526ecaa68aSToby Isaac 
23536ecaa68aSToby Isaac         ierr = PetscSectionGetFieldDof(localCoarse,p,f,&fDof);CHKERRQ(ierr);
235421968bf8SToby Isaac         offsets[f+1] = fDof;
23556ecaa68aSToby Isaac       }
23566ecaa68aSToby Isaac       for (a = 0; a < aDof; a++) {
23576ecaa68aSToby Isaac         PetscInt anchor = anchors[a + aOff], aLocalDof;
23586ecaa68aSToby Isaac 
23596ecaa68aSToby Isaac         ierr = PetscSectionGetDof(localCoarse,anchor,&aLocalDof);CHKERRQ(ierr);
23606ecaa68aSToby Isaac         numColIndices += aLocalDof;
23616ecaa68aSToby Isaac         for (f = 0; f < numFields; f++) {
23626ecaa68aSToby Isaac           PetscInt fDof;
23636ecaa68aSToby Isaac 
23646ecaa68aSToby Isaac           ierr = PetscSectionGetFieldDof(localCoarse,anchor,f,&fDof);CHKERRQ(ierr);
236521968bf8SToby Isaac           newOffsets[f+1] += fDof;
23666ecaa68aSToby Isaac         }
23676ecaa68aSToby Isaac       }
23686ecaa68aSToby Isaac       if (numFields) {
23696ecaa68aSToby Isaac         matSize = 0;
23706ecaa68aSToby Isaac         for (f = 0; f < numFields; f++) {
237121968bf8SToby Isaac           matSize += offsets[f+1] * newOffsets[f+1];
23726ecaa68aSToby Isaac         }
23736ecaa68aSToby Isaac       }
23746ecaa68aSToby Isaac       else {
23756ecaa68aSToby Isaac         matSize = numColIndices * dof;
23766ecaa68aSToby Isaac       }
23776ecaa68aSToby Isaac     }
23786ecaa68aSToby Isaac     else { /* no children, and no constraints on dofs: just get the global indices */
23796ecaa68aSToby Isaac       numColIndices = dof;
23806ecaa68aSToby Isaac       matSize       = 0;
23816ecaa68aSToby Isaac     }
238246bdb399SToby Isaac     /* we will pack the column indices with the field offsets */
23836ecaa68aSToby Isaac     ierr = PetscSectionSetDof(rootIndicesSec,p,numColIndices ? numColIndices+2*numFields : 0);CHKERRQ(ierr);
23846ecaa68aSToby Isaac     ierr = PetscSectionSetDof(rootMatricesSec,p,matSize);CHKERRQ(ierr);
23856ecaa68aSToby Isaac   }
23866ecaa68aSToby Isaac   ierr = PetscSectionSetUp(rootIndicesSec);CHKERRQ(ierr);
23876ecaa68aSToby Isaac   ierr = PetscSectionSetUp(rootMatricesSec);CHKERRQ(ierr);
23886ecaa68aSToby Isaac   {
23896ecaa68aSToby Isaac     PetscInt    numRootIndices, numRootMatrices;
23906ecaa68aSToby Isaac 
23916ecaa68aSToby Isaac     ierr = PetscSectionGetStorageSize(rootIndicesSec,&numRootIndices);CHKERRQ(ierr);
23926ecaa68aSToby Isaac     ierr = PetscSectionGetStorageSize(rootMatricesSec,&numRootMatrices);CHKERRQ(ierr);
23936ecaa68aSToby Isaac     ierr = PetscMalloc2(numRootIndices,&rootIndices,numRootMatrices,&rootMatrices);CHKERRQ(ierr);
23946ecaa68aSToby Isaac     for (p = pStartC; p < pEndC; p++) {
23956ecaa68aSToby Isaac       PetscInt    numRowIndices, numColIndices, matSize, dof;
23966ecaa68aSToby Isaac       PetscInt    pIndOff, pMatOff;
23976ecaa68aSToby Isaac       PetscInt    *pInd;
23986ecaa68aSToby Isaac       PetscInt    maxChildId = maxChildIds[p - pStartC];
23996ecaa68aSToby Isaac       PetscScalar *pMat = NULL;
24006ecaa68aSToby Isaac 
24016ecaa68aSToby Isaac       ierr = PetscSectionGetDof(rootIndicesSec,p,&numColIndices);CHKERRQ(ierr);
24026ecaa68aSToby Isaac       if (!numColIndices) {
24036ecaa68aSToby Isaac         continue;
24046ecaa68aSToby Isaac       }
240521968bf8SToby Isaac       offsets[0]        = 0;
240621968bf8SToby Isaac       newOffsets[0]     = 0;
240721968bf8SToby Isaac       offsetsCopy[0]    = 0;
240821968bf8SToby Isaac       newOffsetsCopy[0] = 0;
24096ecaa68aSToby Isaac       numColIndices -= 2 * numFields;
24106ecaa68aSToby Isaac       ierr = PetscSectionGetOffset(rootIndicesSec,p,&pIndOff);CHKERRQ(ierr);
24116ecaa68aSToby Isaac       pInd = &(rootIndices[pIndOff]);
24126ecaa68aSToby Isaac       ierr = PetscSectionGetDof(rootMatricesSec,p,&matSize);CHKERRQ(ierr);
24136ecaa68aSToby Isaac       if (matSize) {
24146ecaa68aSToby Isaac         ierr = PetscSectionGetOffset(rootMatricesSec,p,&pMatOff);CHKERRQ(ierr);
24156ecaa68aSToby Isaac         pMat = &rootMatrices[pMatOff];
24166ecaa68aSToby Isaac       }
24176ecaa68aSToby Isaac       ierr = PetscSectionGetDof(globalCoarse,p,&dof);CHKERRQ(ierr);
24181cfc5b76SToby Isaac       if (dof < 0) {
24191cfc5b76SToby Isaac         dof = -(dof + 1);
24201cfc5b76SToby Isaac       }
24216ecaa68aSToby Isaac       if (maxChildId >= 0) { /* build an identity matrix, apply matrix constraints on the right */
24226ecaa68aSToby Isaac         PetscInt i, j;
24236ecaa68aSToby Isaac         PetscInt numRowIndices = matSize / numColIndices;
24246ecaa68aSToby Isaac 
24256ecaa68aSToby Isaac         if (!numRowIndices) { /* don't need to calculate the mat, just the indices */
24266ecaa68aSToby Isaac           PetscInt numIndices, *indices;
24276ecaa68aSToby Isaac           ierr = DMPlexGetClosureIndices(coarse,localCoarse,globalCoarse,p,&numIndices,&indices,offsets);CHKERRQ(ierr);
24286ecaa68aSToby Isaac           if (numIndices != numColIndices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"mismatching constraint indices calculations");
24296ecaa68aSToby Isaac           for (i = 0; i < numColIndices; i++) {
24306ecaa68aSToby Isaac             pInd[i] = indices[i];
24316ecaa68aSToby Isaac           }
24326ecaa68aSToby Isaac           for (i = 0; i < numFields; i++) {
243346bdb399SToby Isaac             pInd[numColIndices + i]             = offsets[i+1];
243446bdb399SToby Isaac             pInd[numColIndices + numFields + i] = offsets[i+1];
24356ecaa68aSToby Isaac           }
243646bdb399SToby Isaac           ierr = DMPlexRestoreClosureIndices(coarse,localCoarse,globalCoarse,p,&numIndices,&indices,offsets);CHKERRQ(ierr);
24376ecaa68aSToby Isaac         }
24386ecaa68aSToby Isaac         else {
24396ecaa68aSToby Isaac           PetscInt closureSize, *closure = NULL, cl;
24406ecaa68aSToby Isaac           PetscScalar *pMatIn, *pMatModified;
24416ecaa68aSToby Isaac           PetscInt numPoints,*points;
24426ecaa68aSToby Isaac 
24436ecaa68aSToby Isaac           ierr = DMGetWorkArray(coarse,numRowIndices * numRowIndices,PETSC_SCALAR,&pMatIn);CHKERRQ(ierr);
24446ecaa68aSToby Isaac           for (i = 0; i < numRowIndices; i++) { /* initialize to the identity */
24456ecaa68aSToby Isaac             for (j = 0; j < numRowIndices; j++) {
24466ecaa68aSToby Isaac               pMatIn[i * numRowIndices + j] = (i == j) ? 1. : 0.;
24476ecaa68aSToby Isaac             }
24486ecaa68aSToby Isaac           }
24496ecaa68aSToby Isaac           ierr = DMPlexGetTransitiveClosure(coarse, p, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
24506ecaa68aSToby Isaac           if (numFields) {
24516ecaa68aSToby Isaac             PetscInt f;
24526ecaa68aSToby Isaac 
24536ecaa68aSToby Isaac             for (cl = 0; cl < closureSize; cl++) {
24546ecaa68aSToby Isaac               PetscInt c = closure[2 * cl];
24556ecaa68aSToby Isaac 
24566ecaa68aSToby Isaac               for (f = 0; f < numFields; f++) {
24576ecaa68aSToby Isaac                 PetscInt fDof;
24586ecaa68aSToby Isaac 
24596ecaa68aSToby Isaac                 ierr = PetscSectionGetFieldDof(localCoarse,c,f,&fDof);CHKERRQ(ierr);
24606ecaa68aSToby Isaac                 offsets[f + 1] += fDof;
24616ecaa68aSToby Isaac               }
24626ecaa68aSToby Isaac             }
24636ecaa68aSToby Isaac             for (f = 0; f < numFields; f++) {
24646ecaa68aSToby Isaac               offsets[f + 1]   += offsets[f];
24656ecaa68aSToby Isaac               newOffsets[f + 1] = offsets[f + 1];
24666ecaa68aSToby Isaac             }
24676ecaa68aSToby Isaac           }
24686ecaa68aSToby Isaac           /* apply hanging node constraints on the right, get the new points and the new offsets */
2469267d4f3fSToby Isaac           ierr = DMPlexAnchorsModifyMat(coarse,localCoarse,closureSize,numRowIndices,closure,pMatIn,&numPoints,NULL,&points,&pMatModified,newOffsets,PETSC_FALSE);CHKERRQ(ierr);
24706ecaa68aSToby Isaac           if (!numFields) {
24716ecaa68aSToby Isaac             for (i = 0; i < numRowIndices * numColIndices; i++) {
24726ecaa68aSToby Isaac               pMat[i] = pMatModified[i];
24736ecaa68aSToby Isaac             }
24746ecaa68aSToby Isaac           }
24756ecaa68aSToby Isaac           else {
24766ecaa68aSToby Isaac             PetscInt f, i, j, count;
24776ecaa68aSToby Isaac             for (f = 0, count = 0; f < numFields; f++) {
24786ecaa68aSToby Isaac               for (i = offsets[f]; i < offsets[f+1]; i++) {
24796ecaa68aSToby Isaac                 for (j = newOffsets[f]; j < newOffsets[f+1]; j++, count++) {
24806ecaa68aSToby Isaac                   pMat[count] = pMatModified[i * numColIndices + j];
24816ecaa68aSToby Isaac                 }
24826ecaa68aSToby Isaac               }
24836ecaa68aSToby Isaac             }
24846ecaa68aSToby Isaac           }
24856ecaa68aSToby Isaac           ierr = DMRestoreWorkArray(coarse,numRowIndices * numColIndices,PETSC_SCALAR,&pMatModified);CHKERRQ(ierr);
24866ecaa68aSToby Isaac           ierr = DMPlexRestoreTransitiveClosure(coarse, p, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
24876ecaa68aSToby Isaac           ierr = DMRestoreWorkArray(coarse,numRowIndices * numColIndices,PETSC_SCALAR,&pMatIn);CHKERRQ(ierr);
24886ecaa68aSToby Isaac           if (numFields) {
248946bdb399SToby Isaac             PetscInt f;
249046bdb399SToby Isaac             for (f = 0; f < numFields; f++) {
249146bdb399SToby Isaac               pInd[numColIndices + f]             = offsets[f+1];
249246bdb399SToby Isaac               pInd[numColIndices + numFields + f] = newOffsets[f+1];
24936ecaa68aSToby Isaac             }
24946ecaa68aSToby Isaac             for (cl = 0; cl < numPoints*2; cl += 2) {
24956ecaa68aSToby Isaac               PetscInt o = points[cl+1], globalOff, c = points[cl];
24966ecaa68aSToby Isaac               ierr = PetscSectionGetOffset(globalCoarse, c, &globalOff);CHKERRQ(ierr);
249746bdb399SToby Isaac               indicesPointFields_private(localCoarse, c, globalOff < 0 ? -(globalOff+1) : globalOff, newOffsets, PETSC_FALSE, o, pInd);
24986ecaa68aSToby Isaac             }
24996ecaa68aSToby Isaac           } else {
25006ecaa68aSToby Isaac             for (cl = 0; cl < numPoints*2; cl += 2) {
25016ecaa68aSToby Isaac               PetscInt o = points[cl+1], c = points[cl], globalOff;
25026ecaa68aSToby Isaac               ierr = PetscSectionGetOffset(globalCoarse, c, &globalOff);CHKERRQ(ierr);
25036ecaa68aSToby Isaac               indicesPoint_private(localCoarse, c, globalOff < 0 ? -(globalOff+1) : globalOff, newOffsets, PETSC_FALSE, o, pInd);
25046ecaa68aSToby Isaac             }
25056ecaa68aSToby Isaac           }
25066ecaa68aSToby Isaac         }
25076ecaa68aSToby Isaac       }
25086ecaa68aSToby Isaac       else if (matSize) {
25096ecaa68aSToby Isaac         PetscInt cOff;
25106ecaa68aSToby Isaac         PetscInt *rowIndices, *colIndices, a, aDof, aOff;
25116ecaa68aSToby Isaac 
25126ecaa68aSToby Isaac         numRowIndices = matSize / numColIndices;
2513628cbfb8SToby Isaac         if (numRowIndices != dof) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Miscounted dofs");
25146ecaa68aSToby Isaac         ierr = DMGetWorkArray(coarse,numRowIndices,PETSC_INT,&rowIndices);CHKERRQ(ierr);
25156ecaa68aSToby Isaac         ierr = DMGetWorkArray(coarse,numColIndices,PETSC_INT,&colIndices);CHKERRQ(ierr);
25166ecaa68aSToby Isaac         ierr = PetscSectionGetOffset(cSec,p,&cOff);CHKERRQ(ierr);
25176ecaa68aSToby Isaac         ierr = PetscSectionGetDof(aSec,p,&aDof);CHKERRQ(ierr);
25186ecaa68aSToby Isaac         ierr = PetscSectionGetOffset(aSec,p,&aOff);CHKERRQ(ierr);
25196ecaa68aSToby Isaac         if (numFields) {
25206ecaa68aSToby Isaac           PetscInt f;
25216ecaa68aSToby Isaac 
25226ecaa68aSToby Isaac           for (f = 0; f < numFields; f++) {
25236ecaa68aSToby Isaac             PetscInt fDof;
25246ecaa68aSToby Isaac             ierr = PetscSectionGetFieldDof(cSec,p,f,&fDof);CHKERRQ(ierr);
25256ecaa68aSToby Isaac             offsets[f + 1] = fDof;
25266ecaa68aSToby Isaac             for (a = 0; a < aDof; a++) {
25276ecaa68aSToby Isaac               PetscInt anchor = anchors[a + aOff];
25286ecaa68aSToby Isaac               ierr = PetscSectionGetFieldDof(localCoarse,anchor,f,&fDof);CHKERRQ(ierr);
25296ecaa68aSToby Isaac               newOffsets[f + 1] += fDof;
25306ecaa68aSToby Isaac             }
25316ecaa68aSToby Isaac           }
25326ecaa68aSToby Isaac           for (f = 0; f < numFields; f++) {
25336ecaa68aSToby Isaac             offsets[f + 1]       += offsets[f];
25346ecaa68aSToby Isaac             offsetsCopy[f + 1]    = offsets[f + 1];
25356ecaa68aSToby Isaac             newOffsets[f + 1]    += newOffsets[f];
25366ecaa68aSToby Isaac             newOffsetsCopy[f + 1] = newOffsets[f + 1];
25376ecaa68aSToby Isaac           }
25386ecaa68aSToby Isaac           indicesPointFields_private(cSec,p,cOff,offsetsCopy,PETSC_TRUE,0,rowIndices);CHKERRQ(ierr);
25396ecaa68aSToby Isaac           for (a = 0; a < aDof; a++) {
25406ecaa68aSToby Isaac             PetscInt anchor = anchors[a + aOff], lOff;
25416ecaa68aSToby Isaac             ierr = PetscSectionGetOffset(localCoarse,anchor,&lOff);CHKERRQ(ierr);
254221968bf8SToby Isaac             indicesPointFields_private(localCoarse,anchor,lOff,newOffsetsCopy,PETSC_TRUE,0,colIndices);CHKERRQ(ierr);
25436ecaa68aSToby Isaac           }
25446ecaa68aSToby Isaac         }
25456ecaa68aSToby Isaac         else {
2546267d4f3fSToby Isaac           indicesPoint_private(cSec,p,cOff,offsetsCopy,PETSC_TRUE,0,rowIndices);CHKERRQ(ierr);
25476ecaa68aSToby Isaac           for (a = 0; a < aDof; a++) {
25486ecaa68aSToby Isaac             PetscInt anchor = anchors[a + aOff], lOff;
25496ecaa68aSToby Isaac             ierr = PetscSectionGetOffset(localCoarse,anchor,&lOff);CHKERRQ(ierr);
2550267d4f3fSToby Isaac             indicesPoint_private(localCoarse,anchor,lOff,newOffsetsCopy,PETSC_TRUE,0,colIndices);CHKERRQ(ierr);
25516ecaa68aSToby Isaac           }
25526ecaa68aSToby Isaac         }
25536ecaa68aSToby Isaac         if (numFields) {
25546ecaa68aSToby Isaac           PetscInt count, f, a;
25556ecaa68aSToby Isaac           for (f = 0, count = 0; f < numFields; f++) {
25566ecaa68aSToby Isaac             PetscInt iSize = offsets[f + 1] - offsets[f];
25576ecaa68aSToby Isaac             PetscInt jSize = newOffsets[f + 1] - newOffsets[f];
25586ecaa68aSToby Isaac             ierr = MatGetValues(cMat,iSize,&rowIndices[offsets[f]],jSize,&colIndices[newOffsets[f]],&pMat[count]);CHKERRQ(ierr);
25596ecaa68aSToby Isaac             count += iSize * jSize;
256046bdb399SToby Isaac             pInd[numColIndices + f]             = offsets[f+1];
256146bdb399SToby Isaac             pInd[numColIndices + numFields + f] = newOffsets[f+1];
25626ecaa68aSToby Isaac           }
25636ecaa68aSToby Isaac           for (a = 0; a < aDof; a++) {
25646ecaa68aSToby Isaac             PetscInt anchor = anchors[a + aOff];
25656ecaa68aSToby Isaac             PetscInt gOff;
25666ecaa68aSToby Isaac             ierr = PetscSectionGetOffset(globalCoarse,anchor,&gOff);CHKERRQ(ierr);
25676ecaa68aSToby Isaac             indicesPointFields_private(localCoarse,anchor,gOff < 0 ? -(gOff + 1) : gOff,newOffsets,PETSC_FALSE,0,pInd);CHKERRQ(ierr);
25686ecaa68aSToby Isaac           }
25696ecaa68aSToby Isaac         }
25706ecaa68aSToby Isaac         else {
25716ecaa68aSToby Isaac           PetscInt a;
25726ecaa68aSToby Isaac           ierr = MatGetValues(cMat,numRowIndices,rowIndices,numColIndices,colIndices,pMat);CHKERRQ(ierr);
25736ecaa68aSToby Isaac           for (a = 0; a < aDof; a++) {
25746ecaa68aSToby Isaac             PetscInt anchor = anchors[a + aOff];
25756ecaa68aSToby Isaac             PetscInt gOff;
25766ecaa68aSToby Isaac             ierr = PetscSectionGetOffset(globalCoarse,anchor,&gOff);CHKERRQ(ierr);
25776ecaa68aSToby Isaac             indicesPoint_private(localCoarse,anchor,gOff < 0 ? -(gOff + 1) : gOff,newOffsets,PETSC_FALSE,0,pInd);CHKERRQ(ierr);
25786ecaa68aSToby Isaac           }
25796ecaa68aSToby Isaac         }
25806ecaa68aSToby Isaac         ierr = DMRestoreWorkArray(coarse,numColIndices,PETSC_INT,&colIndices);CHKERRQ(ierr);
25816ecaa68aSToby Isaac         ierr = DMRestoreWorkArray(coarse,numRowIndices,PETSC_INT,&rowIndices);CHKERRQ(ierr);
25826ecaa68aSToby Isaac       }
25836ecaa68aSToby Isaac       else {
25846ecaa68aSToby Isaac         PetscInt gOff;
25856ecaa68aSToby Isaac 
25866ecaa68aSToby Isaac         ierr = PetscSectionGetOffset(globalCoarse,p,&gOff);CHKERRQ(ierr);
25876ecaa68aSToby Isaac         if (numFields) {
25886ecaa68aSToby Isaac           PetscInt f;
25896ecaa68aSToby Isaac 
25906ecaa68aSToby Isaac           for (f = 0; f < numFields; f++) {
25916ecaa68aSToby Isaac             PetscInt fDof;
25926ecaa68aSToby Isaac             ierr = PetscSectionGetFieldDof(localCoarse,p,f,&fDof);CHKERRQ(ierr);
25936ecaa68aSToby Isaac             offsets[f + 1] = fDof + offsets[f];
25946ecaa68aSToby Isaac           }
25956ecaa68aSToby Isaac           for (f = 0; f < numFields; f++) {
259646bdb399SToby Isaac             pInd[numColIndices + f]             = offsets[f+1];
259746bdb399SToby Isaac             pInd[numColIndices + numFields + f] = offsets[f+1];
25986ecaa68aSToby Isaac           }
25996ecaa68aSToby Isaac           indicesPointFields_private(localCoarse,p,gOff < 0 ? -(gOff + 1) : gOff,offsets,PETSC_FALSE,0,pInd);CHKERRQ(ierr);
26006ecaa68aSToby Isaac         }
26016ecaa68aSToby Isaac         else {
26026ecaa68aSToby Isaac           indicesPoint_private(localCoarse,p,gOff < 0 ? -(gOff + 1) : gOff,offsets,PETSC_FALSE,0,pInd);CHKERRQ(ierr);
26036ecaa68aSToby Isaac         }
26046ecaa68aSToby Isaac       }
26056ecaa68aSToby Isaac     }
2606e44e4e7fSToby Isaac     ierr = PetscFree(maxChildIds);CHKERRQ(ierr);
26076ecaa68aSToby Isaac   }
260846bdb399SToby Isaac   {
260946bdb399SToby Isaac     PetscSF  indicesSF, matricesSF;
261046bdb399SToby Isaac     PetscInt *remoteOffsetsIndices, *remoteOffsetsMatrices, numLeafIndices, numLeafMatrices;
261146bdb399SToby Isaac 
261246bdb399SToby Isaac     ierr = PetscSectionCreate(PetscObjectComm((PetscObject)fine),&leafIndicesSec);CHKERRQ(ierr);
261346bdb399SToby Isaac     ierr = PetscSectionCreate(PetscObjectComm((PetscObject)fine),&leafMatricesSec);CHKERRQ(ierr);
261446bdb399SToby Isaac     ierr = PetscSFDistributeSection(coarseToFineEmbedded,rootIndicesSec,&remoteOffsetsIndices,leafIndicesSec);CHKERRQ(ierr);
261546bdb399SToby Isaac     ierr = PetscSFDistributeSection(coarseToFineEmbedded,rootMatricesSec,&remoteOffsetsMatrices,leafMatricesSec);CHKERRQ(ierr);
261646bdb399SToby Isaac     ierr = PetscSFCreateSectionSF(coarseToFineEmbedded,rootIndicesSec,remoteOffsetsIndices,leafIndicesSec,&indicesSF);CHKERRQ(ierr);
261746bdb399SToby Isaac     ierr = PetscSFCreateSectionSF(coarseToFineEmbedded,rootMatricesSec,remoteOffsetsMatrices,leafMatricesSec,&matricesSF);CHKERRQ(ierr);
2618e44e4e7fSToby Isaac     ierr = PetscSFDestroy(&coarseToFineEmbedded);CHKERRQ(ierr);
261946bdb399SToby Isaac     ierr = PetscFree(remoteOffsetsIndices);CHKERRQ(ierr);
262046bdb399SToby Isaac     ierr = PetscFree(remoteOffsetsMatrices);CHKERRQ(ierr);
262146bdb399SToby Isaac     ierr = PetscSectionGetStorageSize(leafIndicesSec,&numLeafIndices);CHKERRQ(ierr);
262246bdb399SToby Isaac     ierr = PetscSectionGetStorageSize(leafMatricesSec,&numLeafMatrices);CHKERRQ(ierr);
262346bdb399SToby Isaac     ierr = PetscMalloc2(numLeafIndices,&leafIndices,numLeafMatrices,&leafMatrices);CHKERRQ(ierr);
262446bdb399SToby Isaac     ierr = PetscSFBcastBegin(indicesSF,MPIU_INT,rootIndices,leafIndices);CHKERRQ(ierr);
2625267d4f3fSToby Isaac     ierr = PetscSFBcastBegin(matricesSF,MPIU_SCALAR,rootMatrices,leafMatrices);CHKERRQ(ierr);
262646bdb399SToby Isaac     ierr = PetscSFBcastEnd(indicesSF,MPIU_INT,rootIndices,leafIndices);CHKERRQ(ierr);
2627267d4f3fSToby Isaac     ierr = PetscSFBcastEnd(matricesSF,MPIU_SCALAR,rootMatrices,leafMatrices);CHKERRQ(ierr);
262846bdb399SToby Isaac     ierr = PetscSFDestroy(&matricesSF);CHKERRQ(ierr);
262946bdb399SToby Isaac     ierr = PetscSFDestroy(&indicesSF);CHKERRQ(ierr);
263046bdb399SToby Isaac     ierr = PetscFree2(rootIndices,rootMatrices);CHKERRQ(ierr);
263146bdb399SToby Isaac     ierr = PetscSectionDestroy(&rootIndicesSec);CHKERRQ(ierr);
263246bdb399SToby Isaac     ierr = PetscSectionDestroy(&rootMatricesSec);CHKERRQ(ierr);
263346bdb399SToby Isaac   }
263446bdb399SToby Isaac   /* count to preallocate */
263546bdb399SToby Isaac   ierr = DMGetDefaultSection(fine,&localFine);CHKERRQ(ierr);
263646bdb399SToby Isaac   {
263746bdb399SToby Isaac     PetscInt    nGlobal;
263846bdb399SToby Isaac     PetscInt    *dnnz, *onnz;
2639b9a5774bSToby Isaac     PetscLayout rowMap, colMap;
2640b9a5774bSToby Isaac     PetscInt    rowStart, rowEnd, colStart, colEnd;
26411c58ffc4SToby Isaac     PetscInt    maxDof;
26421c58ffc4SToby Isaac     PetscInt    *rowIndices;
26431c58ffc4SToby Isaac     DM           refTree;
26441c58ffc4SToby Isaac     PetscInt     **refPointFieldN;
26451c58ffc4SToby Isaac     PetscScalar  ***refPointFieldMats;
26461c58ffc4SToby Isaac     PetscSection refConSec, refAnSec;
26477c0540e0SToby Isaac     PetscInt     pRefStart,pRefEnd,maxConDof,maxColumns;
26481c58ffc4SToby Isaac     PetscScalar  *pointWork;
264946bdb399SToby Isaac 
265046bdb399SToby Isaac     ierr = PetscSectionGetConstrainedStorageSize(globalFine,&nGlobal);CHKERRQ(ierr);
265146bdb399SToby Isaac     ierr = PetscCalloc2(nGlobal,&dnnz,nGlobal,&onnz);CHKERRQ(ierr);
2652b9a5774bSToby Isaac     ierr = MatGetLayouts(mat,&rowMap,&colMap);CHKERRQ(ierr);
2653b9a5774bSToby Isaac     ierr = PetscLayoutSetUp(rowMap);CHKERRQ(ierr);
26541c58ffc4SToby Isaac     ierr = PetscLayoutSetUp(colMap);CHKERRQ(ierr);
2655b9a5774bSToby Isaac     ierr = PetscLayoutGetRange(rowMap,&rowStart,&rowEnd);CHKERRQ(ierr);
265646bdb399SToby Isaac     ierr = PetscLayoutGetRange(colMap,&colStart,&colEnd);CHKERRQ(ierr);
26571c58ffc4SToby Isaac     ierr = PetscSectionGetMaxDof(globalFine,&maxDof);CHKERRQ(ierr);
26581c58ffc4SToby Isaac     ierr = DMGetWorkArray(fine,maxDof,PETSC_INT,&rowIndices);CHKERRQ(ierr);
265946bdb399SToby Isaac     for (p = pStartF; p < pEndF; p++) {
266046bdb399SToby Isaac       PetscInt    gDof, gcDof, gOff;
266146bdb399SToby Isaac       PetscInt    numColIndices, pIndOff, *pInd;
266246bdb399SToby Isaac       PetscInt    matSize;
266321968bf8SToby Isaac       PetscInt    i;
266446bdb399SToby Isaac 
266546bdb399SToby Isaac       ierr = PetscSectionGetDof(globalFine,p,&gDof);CHKERRQ(ierr);
266646bdb399SToby Isaac       ierr = PetscSectionGetConstraintDof(globalFine,p,&gcDof);CHKERRQ(ierr);
266746bdb399SToby Isaac       if ((gDof - gcDof) <= 0) {
266846bdb399SToby Isaac         continue;
266946bdb399SToby Isaac       }
267046bdb399SToby Isaac       ierr = PetscSectionGetOffset(globalFine,p,&gOff);CHKERRQ(ierr);
267146bdb399SToby Isaac       if (gOff < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"I though having global dofs meant a non-negative offset");
2672b9a5774bSToby Isaac       if ((gOff < rowStart) || ((gOff + gDof - gcDof) > rowEnd)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"I thought the row map would constrain the global dofs");
267346bdb399SToby Isaac       ierr = PetscSectionGetDof(leafIndicesSec,p,&numColIndices);CHKERRQ(ierr);
267446bdb399SToby Isaac       ierr = PetscSectionGetOffset(leafIndicesSec,p,&pIndOff);CHKERRQ(ierr);
267546bdb399SToby Isaac       numColIndices -= 2 * numFields;
26761c58ffc4SToby Isaac       if (numColIndices <= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"global fine dof with no dofs to interpolate from");
267746bdb399SToby Isaac       pInd = &leafIndices[pIndOff];
267821968bf8SToby Isaac       offsets[0]        = 0;
267921968bf8SToby Isaac       offsetsCopy[0]    = 0;
268021968bf8SToby Isaac       newOffsets[0]     = 0;
268121968bf8SToby Isaac       newOffsetsCopy[0] = 0;
268246bdb399SToby Isaac       if (numFields) {
268321968bf8SToby Isaac         PetscInt f;
268446bdb399SToby Isaac         for (f = 0; f < numFields; f++) {
268546bdb399SToby Isaac           PetscInt rowDof;
268646bdb399SToby Isaac 
268746bdb399SToby Isaac           ierr = PetscSectionGetFieldDof(localFine,p,f,&rowDof);CHKERRQ(ierr);
268821968bf8SToby Isaac           offsets[f + 1]        = offsets[f] + rowDof;
268921968bf8SToby Isaac           offsetsCopy[f + 1]    = offsets[f + 1];
269021968bf8SToby Isaac           newOffsets[f + 1]     = pInd[numColIndices + numFields + f];
269121968bf8SToby Isaac           numD[f] = 0;
269221968bf8SToby Isaac           numO[f] = 0;
269346bdb399SToby Isaac         }
269421968bf8SToby Isaac         ierr = indicesPointFields_private(localFine,p,gOff,offsetsCopy,PETSC_FALSE,0,rowIndices);CHKERRQ(ierr);
269546bdb399SToby Isaac         for (f = 0; f < numFields; f++) {
269621968bf8SToby Isaac           PetscInt colOffset    = newOffsets[f];
269721968bf8SToby Isaac           PetscInt numFieldCols = newOffsets[f + 1] - newOffsets[f];
269846bdb399SToby Isaac 
269946bdb399SToby Isaac           for (i = 0; i < numFieldCols; i++) {
270046bdb399SToby Isaac             PetscInt gInd = pInd[i + colOffset];
270146bdb399SToby Isaac 
270246bdb399SToby Isaac             if (gInd >= colStart && gInd < colEnd) {
270321968bf8SToby Isaac               numD[f]++;
270446bdb399SToby Isaac             }
270546bdb399SToby Isaac             else if (gInd >= 0) { /* negative means non-entry */
270621968bf8SToby Isaac               numO[f]++;
270746bdb399SToby Isaac             }
270846bdb399SToby Isaac           }
270946bdb399SToby Isaac         }
271046bdb399SToby Isaac       }
271146bdb399SToby Isaac       else {
271221968bf8SToby Isaac         ierr = indicesPoint_private(localFine,p,gOff,offsetsCopy,PETSC_FALSE,0,rowIndices);CHKERRQ(ierr);
271321968bf8SToby Isaac         numD[0] = 0;
271421968bf8SToby Isaac         numO[0] = 0;
271546bdb399SToby Isaac         for (i = 0; i < numColIndices; i++) {
271646bdb399SToby Isaac           PetscInt gInd = pInd[i];
271746bdb399SToby Isaac 
271846bdb399SToby Isaac           if (gInd >= colStart && gInd < colEnd) {
271921968bf8SToby Isaac             numD[0]++;
272046bdb399SToby Isaac           }
272146bdb399SToby Isaac           else if (gInd >= 0) { /* negative means non-entry */
272221968bf8SToby Isaac             numO[0]++;
272346bdb399SToby Isaac           }
272446bdb399SToby Isaac         }
272546bdb399SToby Isaac       }
272646bdb399SToby Isaac       ierr = PetscSectionGetDof(leafMatricesSec,p,&matSize);CHKERRQ(ierr);
272746bdb399SToby Isaac       if (!matSize) { /* incoming matrix is identity */
272846bdb399SToby Isaac         PetscInt childId;
272946bdb399SToby Isaac 
273046bdb399SToby Isaac         childId = childIds[p-pStartF];
273121968bf8SToby Isaac         if (childId < 0) { /* no child interpolation: one nnz per */
273246bdb399SToby Isaac           if (numFields) {
2733b9a5774bSToby Isaac             PetscInt f;
2734b9a5774bSToby Isaac             for (f = 0; f < numFields; f++) {
273521968bf8SToby Isaac               PetscInt numRows = offsets[f+1] - offsets[f], row;
273646bdb399SToby Isaac               for (row = 0; row < numRows; row++) {
273721968bf8SToby Isaac                 PetscInt gIndCoarse = pInd[newOffsets[f] + row];
273821968bf8SToby Isaac                 PetscInt gIndFine   = rowIndices[offsets[f] + row];
273946bdb399SToby Isaac                 if (gIndCoarse >= colStart && gIndCoarse < colEnd) { /* local */
2740b9a5774bSToby Isaac                   if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2741b9a5774bSToby Isaac                   dnnz[gIndFine - rowStart] = 1;
274246bdb399SToby Isaac                 }
274346bdb399SToby Isaac                 else if (gIndCoarse >= 0) { /* remote */
2744b9a5774bSToby Isaac                   if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2745b9a5774bSToby Isaac                   onnz[gIndFine - rowStart] = 1;
274646bdb399SToby Isaac                 }
274746bdb399SToby Isaac                 else { /* constrained */
274846bdb399SToby Isaac                   if (gIndFine >= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
274946bdb399SToby Isaac                 }
275046bdb399SToby Isaac               }
275146bdb399SToby Isaac             }
275246bdb399SToby Isaac           }
275346bdb399SToby Isaac           else {
2754b9a5774bSToby Isaac             PetscInt i;
2755b9a5774bSToby Isaac             for (i = 0; i < gDof; i++) {
275646bdb399SToby Isaac               PetscInt gIndCoarse = pInd[i];
275746bdb399SToby Isaac               PetscInt gIndFine   = rowIndices[i];
275846bdb399SToby Isaac               if (gIndCoarse >= colStart && gIndCoarse < colEnd) { /* local */
2759b9a5774bSToby Isaac                 if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2760b9a5774bSToby Isaac                 dnnz[gIndFine - rowStart] = 1;
276146bdb399SToby Isaac               }
276246bdb399SToby Isaac               else if (gIndCoarse >= 0) { /* remote */
2763b9a5774bSToby Isaac                 if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2764b9a5774bSToby Isaac                 onnz[gIndFine - rowStart] = 1;
276546bdb399SToby Isaac               }
276646bdb399SToby Isaac               else { /* constrained */
276746bdb399SToby Isaac                 if (gIndFine >= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
276846bdb399SToby Isaac               }
276946bdb399SToby Isaac             }
277046bdb399SToby Isaac           }
277146bdb399SToby Isaac         }
277246bdb399SToby Isaac         else { /* interpolate from all */
277346bdb399SToby Isaac           if (numFields) {
2774b9a5774bSToby Isaac             PetscInt f;
2775b9a5774bSToby Isaac             for (f = 0; f < numFields; f++) {
277621968bf8SToby Isaac               PetscInt numRows = offsets[f+1] - offsets[f], row;
277746bdb399SToby Isaac               for (row = 0; row < numRows; row++) {
277821968bf8SToby Isaac                 PetscInt gIndFine = rowIndices[offsets[f] + row];
277946bdb399SToby Isaac                 if (gIndFine >= 0) {
2780b9a5774bSToby Isaac                   if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2781b9a5774bSToby Isaac                   dnnz[gIndFine - rowStart] = numD[f];
2782b9a5774bSToby Isaac                   onnz[gIndFine - rowStart] = numO[f];
278346bdb399SToby Isaac                 }
278446bdb399SToby Isaac               }
278546bdb399SToby Isaac             }
278646bdb399SToby Isaac           }
278746bdb399SToby Isaac           else {
2788b9a5774bSToby Isaac             PetscInt i;
2789b9a5774bSToby Isaac             for (i = 0; i < gDof; i++) {
279046bdb399SToby Isaac               PetscInt gIndFine = rowIndices[i];
279146bdb399SToby Isaac               if (gIndFine >= 0) {
2792b9a5774bSToby Isaac                 if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2793b9a5774bSToby Isaac                 dnnz[gIndFine - rowStart] = numD[0];
2794b9a5774bSToby Isaac                 onnz[gIndFine - rowStart] = numO[0];
279546bdb399SToby Isaac               }
279646bdb399SToby Isaac             }
279746bdb399SToby Isaac           }
279846bdb399SToby Isaac         }
279946bdb399SToby Isaac       }
280046bdb399SToby Isaac       else { /* interpolate from all */
280146bdb399SToby Isaac         if (numFields) {
2802b9a5774bSToby Isaac           PetscInt f;
2803b9a5774bSToby Isaac           for (f = 0; f < numFields; f++) {
280421968bf8SToby Isaac             PetscInt numRows = offsets[f+1] - offsets[f], row;
280546bdb399SToby Isaac             for (row = 0; row < numRows; row++) {
280621968bf8SToby Isaac               PetscInt gIndFine = rowIndices[offsets[f] + row];
280746bdb399SToby Isaac               if (gIndFine >= 0) {
2808b9a5774bSToby Isaac                 if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2809b9a5774bSToby Isaac                 dnnz[gIndFine - rowStart] = numD[f];
2810b9a5774bSToby Isaac                 onnz[gIndFine - rowStart] = numO[f];
281146bdb399SToby Isaac               }
281246bdb399SToby Isaac             }
281346bdb399SToby Isaac           }
281446bdb399SToby Isaac         }
281546bdb399SToby Isaac         else { /* every dof get a full row */
2816b9a5774bSToby Isaac           PetscInt i;
2817b9a5774bSToby Isaac           for (i = 0; i < gDof; i++) {
281846bdb399SToby Isaac             PetscInt gIndFine = rowIndices[i];
281946bdb399SToby Isaac             if (gIndFine >= 0) {
2820b9a5774bSToby Isaac               if (gIndFine < rowStart || gIndFine >= rowEnd) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Mismatched number of constrained dofs");
2821b9a5774bSToby Isaac               dnnz[gIndFine - rowStart] = numD[0];
2822b9a5774bSToby Isaac               onnz[gIndFine - rowStart] = numO[0];
282346bdb399SToby Isaac             }
282446bdb399SToby Isaac           }
282546bdb399SToby Isaac         }
282646bdb399SToby Isaac       }
282746bdb399SToby Isaac     }
282846bdb399SToby Isaac     ierr = MatXAIJSetPreallocation(mat,1,dnnz,onnz,NULL,NULL);CHKERRQ(ierr);
282946bdb399SToby Isaac     ierr = PetscFree2(dnnz,onnz);CHKERRQ(ierr);
283021968bf8SToby Isaac 
283121968bf8SToby Isaac     ierr = DMPlexGetReferenceTree(fine,&refTree);CHKERRQ(ierr);
283221968bf8SToby Isaac     ierr = DMPlexReferenceTreeGetChildrenMatrices(refTree,&refPointFieldMats,&refPointFieldN);CHKERRQ(ierr);
2833e44e4e7fSToby Isaac     ierr = DMGetDefaultConstraints(refTree,&refConSec,NULL);CHKERRQ(ierr);
2834e44e4e7fSToby Isaac     ierr = DMPlexGetAnchors(refTree,&refAnSec,NULL);CHKERRQ(ierr);
2835e44e4e7fSToby Isaac     ierr = PetscSectionGetChart(refConSec,&pRefStart,&pRefEnd);CHKERRQ(ierr);
28361c58ffc4SToby Isaac     ierr = PetscSectionGetMaxDof(refConSec,&maxConDof);CHKERRQ(ierr);
28377c0540e0SToby Isaac     ierr = PetscSectionGetMaxDof(leafIndicesSec,&maxColumns);CHKERRQ(ierr);
28387c0540e0SToby Isaac     ierr = PetscMalloc1(maxConDof*maxColumns,&pointWork);CHKERRQ(ierr);
283921968bf8SToby Isaac     for (p = pStartF; p < pEndF; p++) {
2840e44e4e7fSToby Isaac       PetscInt    gDof, gcDof, gOff;
2841e44e4e7fSToby Isaac       PetscInt    numColIndices, pIndOff, *pInd;
2842e44e4e7fSToby Isaac       PetscInt    matSize;
2843e44e4e7fSToby Isaac       PetscInt    childId;
2844e44e4e7fSToby Isaac 
2845e44e4e7fSToby Isaac 
2846e44e4e7fSToby Isaac       ierr = PetscSectionGetDof(globalFine,p,&gDof);CHKERRQ(ierr);
2847e44e4e7fSToby Isaac       ierr = PetscSectionGetConstraintDof(globalFine,p,&gcDof);CHKERRQ(ierr);
2848e44e4e7fSToby Isaac       if ((gDof - gcDof) <= 0) {
2849e44e4e7fSToby Isaac         continue;
2850e44e4e7fSToby Isaac       }
2851e44e4e7fSToby Isaac       childId = childIds[p-pStartF];
2852e44e4e7fSToby Isaac       ierr = PetscSectionGetOffset(globalFine,p,&gOff);CHKERRQ(ierr);
2853e44e4e7fSToby Isaac       ierr = PetscSectionGetDof(leafIndicesSec,p,&numColIndices);CHKERRQ(ierr);
2854e44e4e7fSToby Isaac       ierr = PetscSectionGetOffset(leafIndicesSec,p,&pIndOff);CHKERRQ(ierr);
2855e44e4e7fSToby Isaac       numColIndices -= 2 * numFields;
2856e44e4e7fSToby Isaac       pInd = &leafIndices[pIndOff];
2857e44e4e7fSToby Isaac       offsets[0]        = 0;
2858e44e4e7fSToby Isaac       offsetsCopy[0]    = 0;
2859e44e4e7fSToby Isaac       newOffsets[0]     = 0;
2860e44e4e7fSToby Isaac       newOffsetsCopy[0] = 0;
2861e44e4e7fSToby Isaac       rowOffsets[0]     = 0;
2862e44e4e7fSToby Isaac       if (numFields) {
2863e44e4e7fSToby Isaac         PetscInt f;
2864e44e4e7fSToby Isaac         for (f = 0; f < numFields; f++) {
2865e44e4e7fSToby Isaac           PetscInt rowDof;
2866e44e4e7fSToby Isaac 
2867e44e4e7fSToby Isaac           ierr = PetscSectionGetFieldDof(localFine,p,f,&rowDof);CHKERRQ(ierr);
2868e44e4e7fSToby Isaac           offsets[f + 1]     = offsets[f] + rowDof;
2869e44e4e7fSToby Isaac           offsetsCopy[f + 1] = offsets[f + 1];
2870e44e4e7fSToby Isaac           rowOffsets[f + 1]  = pInd[numColIndices + f];
2871e44e4e7fSToby Isaac           newOffsets[f + 1]  = pInd[numColIndices + numFields + f];
2872e44e4e7fSToby Isaac         }
2873e44e4e7fSToby Isaac         ierr = indicesPointFields_private(localFine,p,gOff,offsetsCopy,PETSC_FALSE,0,rowIndices);CHKERRQ(ierr);
2874e44e4e7fSToby Isaac       }
28751c58ffc4SToby Isaac       else {
28761c58ffc4SToby Isaac         ierr = indicesPoint_private(localFine,p,gOff,offsetsCopy,PETSC_FALSE,0,rowIndices);CHKERRQ(ierr);
28771c58ffc4SToby Isaac       }
2878e44e4e7fSToby Isaac       ierr = PetscSectionGetDof(leafMatricesSec,p,&matSize);CHKERRQ(ierr);
2879e44e4e7fSToby Isaac       if (!matSize) { /* incoming matrix is identity */
2880e44e4e7fSToby Isaac         if (childId < 0) { /* no child interpolation: scatter */
2881e44e4e7fSToby Isaac           if (numFields) {
2882e44e4e7fSToby Isaac             PetscInt f;
2883e44e4e7fSToby Isaac             for (f = 0; f < numFields; f++) {
2884e44e4e7fSToby Isaac               PetscInt numRows = offsets[f+1] - offsets[f], row;
2885e44e4e7fSToby Isaac               for (row = 0; row < numRows; row++) {
2886e44e4e7fSToby Isaac                 ierr = MatSetValue(mat,rowIndices[offsets[f]+row],pInd[newOffsets[f]+row],1.,INSERT_VALUES);CHKERRQ(ierr);
288721968bf8SToby Isaac               }
288821968bf8SToby Isaac             }
2889e44e4e7fSToby Isaac           }
2890e44e4e7fSToby Isaac           else {
2891e44e4e7fSToby Isaac             PetscInt numRows = gDof, row;
2892e44e4e7fSToby Isaac             for (row = 0; row < numRows; row++) {
2893e44e4e7fSToby Isaac               ierr = MatSetValue(mat,rowIndices[row],pInd[row],1.,INSERT_VALUES);CHKERRQ(ierr);
2894e44e4e7fSToby Isaac             }
2895e44e4e7fSToby Isaac           }
2896e44e4e7fSToby Isaac         }
2897e44e4e7fSToby Isaac         else { /* interpolate from all */
2898e44e4e7fSToby Isaac           if (numFields) {
2899e44e4e7fSToby Isaac             PetscInt f;
2900e44e4e7fSToby Isaac             for (f = 0; f < numFields; f++) {
2901e44e4e7fSToby Isaac               PetscInt numRows = offsets[f+1] - offsets[f];
2902e44e4e7fSToby Isaac               PetscInt numCols = newOffsets[f + 1] - newOffsets[f];
2903e44e4e7fSToby Isaac               ierr = MatSetValues(mat,numRows,&rowIndices[offsets[f]],numCols,&pInd[newOffsets[f]],refPointFieldMats[childId - pRefStart][f],INSERT_VALUES);CHKERRQ(ierr);
2904e44e4e7fSToby Isaac             }
2905e44e4e7fSToby Isaac           }
2906e44e4e7fSToby Isaac           else {
2907e44e4e7fSToby Isaac             ierr = MatSetValues(mat,gDof,rowIndices,numColIndices,pInd,refPointFieldMats[childId - pRefStart][0],INSERT_VALUES);CHKERRQ(ierr);
2908e44e4e7fSToby Isaac           }
2909e44e4e7fSToby Isaac         }
2910e44e4e7fSToby Isaac       }
2911e44e4e7fSToby Isaac       else { /* interpolate from all */
2912e44e4e7fSToby Isaac         PetscInt    pMatOff;
2913e44e4e7fSToby Isaac         PetscScalar *pMat;
2914e44e4e7fSToby Isaac 
2915e44e4e7fSToby Isaac         ierr = PetscSectionGetOffset(leafMatricesSec,p,&pMatOff);CHKERRQ(ierr);
2916e44e4e7fSToby Isaac         pMat = &leafMatrices[pMatOff];
2917e44e4e7fSToby Isaac         if (childId < 0) { /* copy the incoming matrix */
2918e44e4e7fSToby Isaac           if (numFields) {
2919e44e4e7fSToby Isaac             PetscInt f, count;
2920e44e4e7fSToby Isaac             for (f = 0, count = 0; f < numFields; f++) {
2921e44e4e7fSToby Isaac               PetscInt numRows = offsets[f+1]-offsets[f];
2922e44e4e7fSToby Isaac               PetscInt numCols = newOffsets[f+1]-newOffsets[f];
2923e44e4e7fSToby Isaac               PetscInt numInRows = rowOffsets[f+1]-rowOffsets[f];
2924e44e4e7fSToby Isaac               PetscScalar *inMat = &pMat[count];
2925e44e4e7fSToby Isaac 
2926e44e4e7fSToby Isaac               ierr = MatSetValues(mat,numRows,&rowIndices[offsets[f]],numCols,&pInd[newOffsets[f]],inMat,INSERT_VALUES);CHKERRQ(ierr);
2927e44e4e7fSToby Isaac               count += numCols * numInRows;
2928e44e4e7fSToby Isaac             }
2929e44e4e7fSToby Isaac           }
2930e44e4e7fSToby Isaac           else {
2931e44e4e7fSToby Isaac             ierr = MatSetValues(mat,gDof,rowIndices,numColIndices,pInd,pMat,INSERT_VALUES);CHKERRQ(ierr);
2932e44e4e7fSToby Isaac           }
2933e44e4e7fSToby Isaac         }
2934e44e4e7fSToby Isaac         else { /* multiply the incoming matrix by the child interpolation */
2935e44e4e7fSToby Isaac           if (numFields) {
2936e44e4e7fSToby Isaac             PetscInt f, count;
2937e44e4e7fSToby Isaac             for (f = 0, count = 0; f < numFields; f++) {
2938e44e4e7fSToby Isaac               PetscInt numRows = offsets[f+1]-offsets[f];
2939e44e4e7fSToby Isaac               PetscInt numCols = newOffsets[f+1]-newOffsets[f];
2940e44e4e7fSToby Isaac               PetscInt numInRows = rowOffsets[f+1]-rowOffsets[f];
2941e44e4e7fSToby Isaac               PetscScalar *inMat = &pMat[count];
2942e44e4e7fSToby Isaac               PetscInt i, j, k;
2943e44e4e7fSToby Isaac               if (refPointFieldN[childId - pRefStart][f] != numInRows) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point constraint matrix multiply dimension mismatch");
2944e44e4e7fSToby Isaac               for (i = 0; i < numRows; i++) {
2945e44e4e7fSToby Isaac                 for (j = 0; j < numCols; j++) {
2946e44e4e7fSToby Isaac                   PetscScalar val = 0.;
2947e44e4e7fSToby Isaac                   for (k = 0; k < numInRows; k++) {
2948e44e4e7fSToby Isaac                     val += refPointFieldMats[childId - pRefStart][f][i * numInRows + k] * inMat[k * numCols + j];
2949e44e4e7fSToby Isaac                   }
2950e44e4e7fSToby Isaac                   pointWork[i * numCols + j] = val;
2951e44e4e7fSToby Isaac                 }
2952e44e4e7fSToby Isaac               }
2953e44e4e7fSToby Isaac               ierr = MatSetValues(mat,numRows,&rowIndices[offsets[f]],numCols,&pInd[newOffsets[f]],pointWork,INSERT_VALUES);CHKERRQ(ierr);
2954e44e4e7fSToby Isaac               count += numCols * numInRows;
2955e44e4e7fSToby Isaac             }
2956e44e4e7fSToby Isaac           }
2957267d4f3fSToby Isaac           else { /* every dof gets a full row */
2958e44e4e7fSToby Isaac             PetscInt numRows   = gDof;
2959e44e4e7fSToby Isaac             PetscInt numCols   = numColIndices;
2960e44e4e7fSToby Isaac             PetscInt numInRows = matSize / numColIndices;
2961e44e4e7fSToby Isaac             PetscInt i, j, k;
2962e44e4e7fSToby Isaac             if (refPointFieldN[childId - pRefStart][0] != numInRows) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point constraint matrix multiply dimension mismatch");
2963e44e4e7fSToby Isaac             for (i = 0; i < numRows; i++) {
2964e44e4e7fSToby Isaac               for (j = 0; j < numCols; j++) {
2965e44e4e7fSToby Isaac                 PetscScalar val = 0.;
2966e44e4e7fSToby Isaac                 for (k = 0; k < numInRows; k++) {
2967e44e4e7fSToby Isaac                   val += refPointFieldMats[childId - pRefStart][0][i * numInRows + k] * pMat[k * numCols + j];
2968e44e4e7fSToby Isaac                 }
2969e44e4e7fSToby Isaac                 pointWork[i * numCols + j] = val;
2970e44e4e7fSToby Isaac               }
2971e44e4e7fSToby Isaac             }
2972e44e4e7fSToby Isaac             ierr = MatSetValues(mat,numRows,rowIndices,numCols,pInd,pointWork,INSERT_VALUES);CHKERRQ(ierr);
2973e44e4e7fSToby Isaac           }
2974e44e4e7fSToby Isaac         }
2975e44e4e7fSToby Isaac       }
2976e44e4e7fSToby Isaac     }
29771c58ffc4SToby Isaac     ierr = DMPlexReferenceTreeRestoreChildrenMatrices(refTree,&refPointFieldMats,&refPointFieldN);CHKERRQ(ierr);
29781c58ffc4SToby Isaac     ierr = DMRestoreWorkArray(fine,maxDof,PETSC_INT,&rowIndices);CHKERRQ(ierr);
2979e44e4e7fSToby Isaac     ierr = PetscFree(pointWork);CHKERRQ(ierr);
2980e44e4e7fSToby Isaac   }
2981e44e4e7fSToby Isaac   ierr = MatAssemblyBegin(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2982e44e4e7fSToby Isaac   ierr = MatAssemblyEnd(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2983e44e4e7fSToby Isaac   ierr = PetscSectionDestroy(&leafIndicesSec);CHKERRQ(ierr);
2984e44e4e7fSToby Isaac   ierr = PetscSectionDestroy(&leafMatricesSec);CHKERRQ(ierr);
2985e44e4e7fSToby Isaac   ierr = PetscFree2(leafIndices,leafMatrices);CHKERRQ(ierr);
2986e44e4e7fSToby Isaac   ierr = PetscFree7(offsets,offsetsCopy,newOffsets,newOffsetsCopy,rowOffsets,numD,numO);CHKERRQ(ierr);
29876ecaa68aSToby Isaac   ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
29886ecaa68aSToby Isaac   PetscFunctionReturn(0);
29896ecaa68aSToby Isaac }
2990