xref: /petsc/src/dm/impls/plex/plexfem.c (revision e1d0b1ad164e07e647faa7c83936d7bfc07dfed7)
1 #include <petsc-private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
2 
3 #include <petsc-private/petscfeimpl.h>
4 #include <petsc-private/petscfvimpl.h>
5 
6 #undef __FUNCT__
7 #define __FUNCT__ "DMPlexGetScale"
8 PetscErrorCode DMPlexGetScale(DM dm, PetscUnit unit, PetscReal *scale)
9 {
10   DM_Plex *mesh = (DM_Plex*) dm->data;
11 
12   PetscFunctionBegin;
13   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
14   PetscValidPointer(scale, 3);
15   *scale = mesh->scale[unit];
16   PetscFunctionReturn(0);
17 }
18 
19 #undef __FUNCT__
20 #define __FUNCT__ "DMPlexSetScale"
21 PetscErrorCode DMPlexSetScale(DM dm, PetscUnit unit, PetscReal scale)
22 {
23   DM_Plex *mesh = (DM_Plex*) dm->data;
24 
25   PetscFunctionBegin;
26   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
27   mesh->scale[unit] = scale;
28   PetscFunctionReturn(0);
29 }
30 
31 PETSC_STATIC_INLINE PetscInt epsilon(PetscInt i, PetscInt j, PetscInt k)
32 {
33   switch (i) {
34   case 0:
35     switch (j) {
36     case 0: return 0;
37     case 1:
38       switch (k) {
39       case 0: return 0;
40       case 1: return 0;
41       case 2: return 1;
42       }
43     case 2:
44       switch (k) {
45       case 0: return 0;
46       case 1: return -1;
47       case 2: return 0;
48       }
49     }
50   case 1:
51     switch (j) {
52     case 0:
53       switch (k) {
54       case 0: return 0;
55       case 1: return 0;
56       case 2: return -1;
57       }
58     case 1: return 0;
59     case 2:
60       switch (k) {
61       case 0: return 1;
62       case 1: return 0;
63       case 2: return 0;
64       }
65     }
66   case 2:
67     switch (j) {
68     case 0:
69       switch (k) {
70       case 0: return 0;
71       case 1: return 1;
72       case 2: return 0;
73       }
74     case 1:
75       switch (k) {
76       case 0: return -1;
77       case 1: return 0;
78       case 2: return 0;
79       }
80     case 2: return 0;
81     }
82   }
83   return 0;
84 }
85 
86 #undef __FUNCT__
87 #define __FUNCT__ "DMPlexCreateRigidBody"
88 /*@C
89   DMPlexCreateRigidBody - create rigid body modes from coordinates
90 
91   Collective on DM
92 
93   Input Arguments:
94 + dm - the DM
95 . section - the local section associated with the rigid field, or NULL for the default section
96 - globalSection - the global section associated with the rigid field, or NULL for the default section
97 
98   Output Argument:
99 . sp - the null space
100 
101   Note: This is necessary to take account of Dirichlet conditions on the displacements
102 
103   Level: advanced
104 
105 .seealso: MatNullSpaceCreate()
106 @*/
107 PetscErrorCode DMPlexCreateRigidBody(DM dm, PetscSection section, PetscSection globalSection, MatNullSpace *sp)
108 {
109   MPI_Comm       comm;
110   Vec            coordinates, localMode, mode[6];
111   PetscSection   coordSection;
112   PetscScalar   *coords;
113   PetscInt       dim, vStart, vEnd, v, n, m, d, i, j;
114   PetscErrorCode ierr;
115 
116   PetscFunctionBegin;
117   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
118   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
119   if (dim == 1) {
120     ierr = MatNullSpaceCreate(comm, PETSC_TRUE, 0, NULL, sp);CHKERRQ(ierr);
121     PetscFunctionReturn(0);
122   }
123   if (!section)       {ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);}
124   if (!globalSection) {ierr = DMGetDefaultGlobalSection(dm, &globalSection);CHKERRQ(ierr);}
125   ierr = PetscSectionGetConstrainedStorageSize(globalSection, &n);CHKERRQ(ierr);
126   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
127   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
128   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
129   m    = (dim*(dim+1))/2;
130   ierr = VecCreate(comm, &mode[0]);CHKERRQ(ierr);
131   ierr = VecSetSizes(mode[0], n, PETSC_DETERMINE);CHKERRQ(ierr);
132   ierr = VecSetUp(mode[0]);CHKERRQ(ierr);
133   for (i = 1; i < m; ++i) {ierr = VecDuplicate(mode[0], &mode[i]);CHKERRQ(ierr);}
134   /* Assume P1 */
135   ierr = DMGetLocalVector(dm, &localMode);CHKERRQ(ierr);
136   for (d = 0; d < dim; ++d) {
137     PetscScalar values[3] = {0.0, 0.0, 0.0};
138 
139     values[d] = 1.0;
140     ierr      = VecSet(localMode, 0.0);CHKERRQ(ierr);
141     for (v = vStart; v < vEnd; ++v) {
142       ierr = DMPlexVecSetClosure(dm, section, localMode, v, values, INSERT_VALUES);CHKERRQ(ierr);
143     }
144     ierr = DMLocalToGlobalBegin(dm, localMode, INSERT_VALUES, mode[d]);CHKERRQ(ierr);
145     ierr = DMLocalToGlobalEnd(dm, localMode, INSERT_VALUES, mode[d]);CHKERRQ(ierr);
146   }
147   ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
148   for (d = dim; d < dim*(dim+1)/2; ++d) {
149     PetscInt i, j, k = dim > 2 ? d - dim : d;
150 
151     ierr = VecSet(localMode, 0.0);CHKERRQ(ierr);
152     for (v = vStart; v < vEnd; ++v) {
153       PetscScalar values[3] = {0.0, 0.0, 0.0};
154       PetscInt    off;
155 
156       ierr = PetscSectionGetOffset(coordSection, v, &off);CHKERRQ(ierr);
157       for (i = 0; i < dim; ++i) {
158         for (j = 0; j < dim; ++j) {
159           values[j] += epsilon(i, j, k)*PetscRealPart(coords[off+i]);
160         }
161       }
162       ierr = DMPlexVecSetClosure(dm, section, localMode, v, values, INSERT_VALUES);CHKERRQ(ierr);
163     }
164     ierr = DMLocalToGlobalBegin(dm, localMode, INSERT_VALUES, mode[d]);CHKERRQ(ierr);
165     ierr = DMLocalToGlobalEnd(dm, localMode, INSERT_VALUES, mode[d]);CHKERRQ(ierr);
166   }
167   ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
168   ierr = DMRestoreLocalVector(dm, &localMode);CHKERRQ(ierr);
169   for (i = 0; i < dim; ++i) {ierr = VecNormalize(mode[i], NULL);CHKERRQ(ierr);}
170   /* Orthonormalize system */
171   for (i = dim; i < m; ++i) {
172     PetscScalar dots[6];
173 
174     ierr = VecMDot(mode[i], i, mode, dots);CHKERRQ(ierr);
175     for (j = 0; j < i; ++j) dots[j] *= -1.0;
176     ierr = VecMAXPY(mode[i], i, dots, mode);CHKERRQ(ierr);
177     ierr = VecNormalize(mode[i], NULL);CHKERRQ(ierr);
178   }
179   ierr = MatNullSpaceCreate(comm, PETSC_FALSE, m, mode, sp);CHKERRQ(ierr);
180   for (i = 0; i< m; ++i) {ierr = VecDestroy(&mode[i]);CHKERRQ(ierr);}
181   PetscFunctionReturn(0);
182 }
183 
184 #undef __FUNCT__
185 #define __FUNCT__ "DMPlexProjectFunctionLabelLocal"
186 PetscErrorCode DMPlexProjectFunctionLabelLocal(DM dm, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscFE fe[], void (**funcs)(const PetscReal [], PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
187 {
188   PetscDualSpace *sp;
189   PetscSection    section;
190   PetscScalar    *values;
191   PetscBool      *fieldActive;
192   PetscInt        numFields, numComp, dim, spDim, totDim = 0, numValues, cStart, cEnd, f, d, v, i, comp;
193   PetscErrorCode  ierr;
194 
195   PetscFunctionBegin;
196   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
197   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
198   ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);
199   ierr = PetscMalloc1(numFields,&sp);CHKERRQ(ierr);
200   for (f = 0; f < numFields; ++f) {
201     ierr = PetscFEGetDualSpace(fe[f], &sp[f]);CHKERRQ(ierr);
202     ierr = PetscFEGetNumComponents(fe[f], &numComp);CHKERRQ(ierr);
203     ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr);
204     totDim += spDim*numComp;
205   }
206   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
207   ierr = DMPlexVecGetClosure(dm, section, localX, cStart, &numValues, NULL);CHKERRQ(ierr);
208   if (numValues != totDim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The section cell closure size %d != dual space dimension %d", numValues, totDim);
209   ierr = DMGetWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
210   ierr = DMGetWorkArray(dm, numFields, PETSC_BOOL, &fieldActive);CHKERRQ(ierr);
211   for (f = 0; f < numFields; ++f) fieldActive[f] = funcs[f] ? PETSC_TRUE : PETSC_FALSE;
212   for (i = 0; i < numIds; ++i) {
213     IS              pointIS;
214     const PetscInt *points;
215     PetscInt        n, p;
216 
217     ierr = DMLabelGetStratumIS(label, ids[i], &pointIS);CHKERRQ(ierr);
218     ierr = ISGetLocalSize(pointIS, &n);CHKERRQ(ierr);
219     ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr);
220     for (p = 0; p < n; ++p) {
221       const PetscInt  point = points[p];
222       PetscFECellGeom geom;
223 
224       if ((point < cStart) || (point >= cEnd)) continue;
225       ierr = DMPlexComputeCellGeometryFEM(dm, point, NULL, geom.v0, geom.J, NULL, &geom.detJ);CHKERRQ(ierr);
226       for (f = 0, v = 0; f < numFields; ++f) {
227         void * const ctx = ctxs ? ctxs[f] : NULL;
228         ierr = PetscFEGetNumComponents(fe[f], &numComp);CHKERRQ(ierr);
229         ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr);
230         for (d = 0; d < spDim; ++d) {
231           if (funcs[f]) {
232             ierr = PetscDualSpaceApply(sp[f], d, &geom, numComp, funcs[f], ctx, &values[v]);CHKERRQ(ierr);
233           } else {
234             for (comp = 0; comp < numComp; ++comp) values[v+comp] = 0.0;
235           }
236           v += numComp;
237         }
238       }
239       ierr = DMPlexVecSetFieldClosure_Internal(dm, section, localX, fieldActive, point, values, mode);CHKERRQ(ierr);
240     }
241     ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr);
242     ierr = ISDestroy(&pointIS);CHKERRQ(ierr);
243   }
244   ierr = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
245   ierr = DMRestoreWorkArray(dm, numFields, PETSC_BOOL, &fieldActive);CHKERRQ(ierr);
246   ierr = PetscFree(sp);CHKERRQ(ierr);
247   PetscFunctionReturn(0);
248 }
249 
250 #undef __FUNCT__
251 #define __FUNCT__ "DMPlexProjectFunctionLocal"
252 PetscErrorCode DMPlexProjectFunctionLocal(DM dm, void (**funcs)(const PetscReal [], PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
253 {
254   PetscDualSpace *sp;
255   PetscInt       *numComp;
256   PetscSection    section;
257   PetscScalar    *values;
258   PetscInt        numFields, dim, spDim, totDim = 0, numValues, cStart, cEnd, c, f, d, v, comp;
259   PetscErrorCode  ierr;
260 
261   PetscFunctionBegin;
262   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
263   ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);
264   ierr = PetscMalloc2(numFields, &sp, numFields, &numComp);CHKERRQ(ierr);
265   for (f = 0; f < numFields; ++f) {
266     PetscObject  obj;
267     PetscClassId id;
268 
269     ierr = DMGetField(dm, f, &obj);CHKERRQ(ierr);
270     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
271     if (id == PETSCFE_CLASSID) {
272       PetscFE fe = (PetscFE) obj;
273 
274       ierr = PetscFEGetNumComponents(fe, &numComp[f]);CHKERRQ(ierr);
275       ierr = PetscFEGetDualSpace(fe, &sp[f]);CHKERRQ(ierr);
276       ierr = PetscObjectReference((PetscObject) sp[f]);CHKERRQ(ierr);
277     } else if (id == PETSCFV_CLASSID) {
278       PetscFV         fv = (PetscFV) obj;
279       PetscQuadrature q;
280 
281       ierr = PetscFVGetNumComponents(fv, &numComp[f]);CHKERRQ(ierr);
282       ierr = PetscFVGetQuadrature(fv, &q);CHKERRQ(ierr);
283       ierr = PetscDualSpaceCreate(PetscObjectComm((PetscObject) fv), &sp[f]);CHKERRQ(ierr);
284       ierr = PetscDualSpaceSetDM(sp[f], dm);CHKERRQ(ierr);
285       ierr = PetscDualSpaceSetType(sp[f], PETSCDUALSPACESIMPLE);CHKERRQ(ierr);
286       ierr = PetscDualSpaceSimpleSetDimension(sp[f], 1);CHKERRQ(ierr);
287       ierr = PetscDualSpaceSimpleSetFunctional(sp[f], 0, q);CHKERRQ(ierr);
288     } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", f);
289     ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr);
290     totDim += spDim*numComp[f];
291   }
292   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
293   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
294   ierr = DMPlexVecGetClosure(dm, section, localX, cStart, &numValues, NULL);CHKERRQ(ierr);
295   if (numValues != totDim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The section cell closure size %d != dual space dimension %d", numValues, totDim);
296   ierr = DMGetWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
297   for (c = cStart; c < cEnd; ++c) {
298     PetscFECellGeom geom;
299 
300     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, geom.v0, geom.J, NULL, &geom.detJ);CHKERRQ(ierr);
301     for (f = 0, v = 0; f < numFields; ++f) {
302       void *const ctx = ctxs ? ctxs[f] : NULL;
303 
304       ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr);
305       for (d = 0; d < spDim; ++d) {
306         if (funcs[f]) {
307           ierr = PetscDualSpaceApply(sp[f], d, &geom, numComp[f], funcs[f], ctx, &values[v]);CHKERRQ(ierr);
308         } else {
309           for (comp = 0; comp < numComp[f]; ++comp) values[v+comp] = 0.0;
310         }
311         v += numComp[f];
312       }
313     }
314     ierr = DMPlexVecSetClosure(dm, section, localX, c, values, mode);CHKERRQ(ierr);
315   }
316   ierr = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
317   for (f = 0; f < numFields; ++f) {ierr = PetscDualSpaceDestroy(&sp[f]);CHKERRQ(ierr);}
318   ierr = PetscFree2(sp, numComp);CHKERRQ(ierr);
319   PetscFunctionReturn(0);
320 }
321 
322 #undef __FUNCT__
323 #define __FUNCT__ "DMPlexProjectFunction"
324 /*@C
325   DMPlexProjectFunction - This projects the given function into the function space provided.
326 
327   Input Parameters:
328 + dm      - The DM
329 . funcs   - The coordinate functions to evaluate, one per field
330 . ctxs    - Optional array of contexts to pass to each coordinate function.  ctxs itself may be null.
331 - mode    - The insertion mode for values
332 
333   Output Parameter:
334 . X - vector
335 
336   Level: developer
337 
338 .seealso: DMPlexComputeL2Diff()
339 @*/
340 PetscErrorCode DMPlexProjectFunction(DM dm, void (**funcs)(const PetscReal [], PetscScalar *, void *), void **ctxs, InsertMode mode, Vec X)
341 {
342   Vec            localX;
343   PetscErrorCode ierr;
344 
345   PetscFunctionBegin;
346   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
347   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
348   ierr = DMPlexProjectFunctionLocal(dm, funcs, ctxs, mode, localX);CHKERRQ(ierr);
349   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
350   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
351   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
352   PetscFunctionReturn(0);
353 }
354 
355 #undef __FUNCT__
356 #define __FUNCT__ "DMPlexProjectFieldLocal"
357 PetscErrorCode DMPlexProjectFieldLocal(DM dm, Vec localU, void (**funcs)(const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscReal [], PetscScalar []), InsertMode mode, Vec localX)
358 {
359   DM                dmAux;
360   PetscDS           prob, probAux;
361   Vec               A;
362   PetscSection      section, sectionAux;
363   PetscScalar      *values, *u, *u_x, *a, *a_x;
364   PetscReal        *x, *v0, *J, *invJ, detJ, **basisField, **basisFieldDer, **basisFieldAux, **basisFieldDerAux;
365   PetscInt          Nf, dim, spDim, totDim, numValues, cStart, cEnd, c, f, d, v, comp;
366   PetscErrorCode    ierr;
367 
368   PetscFunctionBegin;
369   ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
370   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
371   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
372   ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr);
373   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
374   ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr);
375   ierr = PetscDSGetTabulation(prob, &basisField, &basisFieldDer);CHKERRQ(ierr);
376   ierr = PetscDSGetEvaluationArrays(prob, &u, NULL, &u_x);CHKERRQ(ierr);
377   ierr = PetscDSGetRefCoordArrays(prob, &x, NULL);CHKERRQ(ierr);
378   ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr);
379   ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &A);CHKERRQ(ierr);
380   if (dmAux) {
381     ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr);
382     ierr = DMGetDefaultSection(dmAux, &sectionAux);CHKERRQ(ierr);
383     ierr = PetscDSGetTabulation(prob, &basisFieldAux, &basisFieldDerAux);CHKERRQ(ierr);
384     ierr = PetscDSGetEvaluationArrays(probAux, &a, NULL, &a_x);CHKERRQ(ierr);
385   }
386   ierr = DMPlexInsertBoundaryValuesFEM(dm, localU);CHKERRQ(ierr);
387   ierr = DMPlexVecGetClosure(dm, section, localX, cStart, &numValues, NULL);CHKERRQ(ierr);
388   if (numValues != totDim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The section cell closure size %d != dual space dimension %d", numValues, totDim);
389   ierr = DMGetWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
390   ierr = PetscMalloc3(dim,&v0,dim*dim,&J,dim*dim,&invJ);CHKERRQ(ierr);
391   for (c = cStart; c < cEnd; ++c) {
392     PetscScalar *coefficients = NULL, *coefficientsAux = NULL;
393 
394     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
395     ierr = DMPlexVecGetClosure(dm, section, localU, c, NULL, &coefficients);CHKERRQ(ierr);
396     if (dmAux) {ierr = DMPlexVecGetClosure(dmAux, sectionAux, A, c, NULL, &coefficientsAux);CHKERRQ(ierr);}
397     for (f = 0, v = 0; f < Nf; ++f) {
398       PetscFE        fe;
399       PetscDualSpace sp;
400       PetscInt       Ncf;
401 
402       ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fe);CHKERRQ(ierr);
403       ierr = PetscFEGetDualSpace(fe, &sp);CHKERRQ(ierr);
404       ierr = PetscFEGetNumComponents(fe, &Ncf);CHKERRQ(ierr);
405       ierr = PetscDualSpaceGetDimension(sp, &spDim);CHKERRQ(ierr);
406       for (d = 0; d < spDim; ++d) {
407         PetscQuadrature  quad;
408         const PetscReal *points, *weights;
409         PetscInt         numPoints, q;
410 
411         if (funcs[f]) {
412           ierr = PetscDualSpaceGetFunctional(sp, d, &quad);CHKERRQ(ierr);
413           ierr = PetscQuadratureGetData(quad, NULL, &numPoints, &points, &weights);CHKERRQ(ierr);
414           ierr = PetscFEGetTabulation(fe, numPoints, points, &basisField[f], &basisFieldDer[f], NULL);CHKERRQ(ierr);
415           for (q = 0; q < numPoints; ++q) {
416             CoordinatesRefToReal(dim, dim, v0, J, &points[q*dim], x);
417             ierr = EvaluateFieldJets(prob,    PETSC_FALSE, q, invJ, coefficients,    NULL, u, u_x, NULL);CHKERRQ(ierr);
418             ierr = EvaluateFieldJets(probAux, PETSC_FALSE, q, invJ, coefficientsAux, NULL, a, a_x, NULL);CHKERRQ(ierr);
419             (*funcs[f])(u, NULL, u_x, a, NULL, a_x, x, &values[v]);
420           }
421           ierr = PetscFERestoreTabulation(fe, numPoints, points, &basisField[f], &basisFieldDer[f], NULL);CHKERRQ(ierr);
422         } else {
423           for (comp = 0; comp < Ncf; ++comp) values[v+comp] = 0.0;
424         }
425         v += Ncf;
426       }
427     }
428     ierr = DMPlexVecRestoreClosure(dm, section, localU, c, NULL, &coefficients);CHKERRQ(ierr);
429     if (dmAux) {ierr = DMPlexVecRestoreClosure(dmAux, sectionAux, A, c, NULL, &coefficientsAux);CHKERRQ(ierr);}
430     ierr = DMPlexVecSetClosure(dm, section, localX, c, values, mode);CHKERRQ(ierr);
431   }
432   ierr = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr);
433   ierr = PetscFree3(v0,J,invJ);CHKERRQ(ierr);
434   PetscFunctionReturn(0);
435 }
436 
437 #undef __FUNCT__
438 #define __FUNCT__ "DMPlexProjectField"
439 /*@C
440   DMPlexProjectField - This projects the given function of the fields into the function space provided.
441 
442   Input Parameters:
443 + dm      - The DM
444 . U       - The input field vector
445 . funcs   - The functions to evaluate, one per field
446 - mode    - The insertion mode for values
447 
448   Output Parameter:
449 . X       - The output vector
450 
451   Level: developer
452 
453 .seealso: DMPlexProjectFunction(), DMPlexComputeL2Diff()
454 @*/
455 PetscErrorCode DMPlexProjectField(DM dm, Vec U, void (**funcs)(const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscScalar[], const PetscReal [], PetscScalar []), InsertMode mode, Vec X)
456 {
457   Vec            localX, localU;
458   PetscErrorCode ierr;
459 
460   PetscFunctionBegin;
461   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
462   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
463   ierr = DMGetLocalVector(dm, &localU);CHKERRQ(ierr);
464   ierr = DMGlobalToLocalBegin(dm, U, INSERT_VALUES, localU);CHKERRQ(ierr);
465   ierr = DMGlobalToLocalEnd(dm, U, INSERT_VALUES, localU);CHKERRQ(ierr);
466   ierr = DMPlexProjectFieldLocal(dm, localU, funcs, mode, localX);CHKERRQ(ierr);
467   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
468   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
469   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
470   ierr = DMRestoreLocalVector(dm, &localU);CHKERRQ(ierr);
471   PetscFunctionReturn(0);
472 }
473 
474 #undef __FUNCT__
475 #define __FUNCT__ "DMPlexInsertBoundaryValuesFEM"
476 PetscErrorCode DMPlexInsertBoundaryValuesFEM(DM dm, Vec localX)
477 {
478   void        (**funcs)(const PetscReal x[], PetscScalar *u, void *ctx);
479   void         **ctxs;
480   PetscFE       *fe;
481   PetscInt       numFields, f, numBd, b;
482   PetscErrorCode ierr;
483 
484   PetscFunctionBegin;
485   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
486   PetscValidHeaderSpecific(localX, VEC_CLASSID, 2);
487   ierr = DMGetNumFields(dm, &numFields);CHKERRQ(ierr);
488   ierr = PetscMalloc3(numFields,&fe,numFields,&funcs,numFields,&ctxs);CHKERRQ(ierr);
489   for (f = 0; f < numFields; ++f) {ierr = DMGetField(dm, f, (PetscObject *) &fe[f]);CHKERRQ(ierr);}
490   /* OPT: Could attempt to do multiple BCs at once */
491   ierr = DMPlexGetNumBoundary(dm, &numBd);CHKERRQ(ierr);
492   for (b = 0; b < numBd; ++b) {
493     DMLabel         label;
494     const PetscInt *ids;
495     const char     *labelname;
496     PetscInt        numids, field;
497     PetscBool       isEssential;
498     void          (*func)();
499     void           *ctx;
500 
501     ierr = DMPlexGetBoundary(dm, b, &isEssential, NULL, &labelname, &field, &func, &numids, &ids, &ctx);CHKERRQ(ierr);
502     ierr = DMPlexGetLabel(dm, labelname, &label);CHKERRQ(ierr);
503     for (f = 0; f < numFields; ++f) {
504       funcs[f] = field == f ? (void (*)(const PetscReal[], PetscScalar *, void *)) func : NULL;
505       ctxs[f]  = field == f ? ctx : NULL;
506     }
507     ierr = DMPlexProjectFunctionLabelLocal(dm, label, numids, ids, fe, funcs, ctxs, INSERT_BC_VALUES, localX);CHKERRQ(ierr);
508   }
509   ierr = PetscFree3(fe,funcs,ctxs);CHKERRQ(ierr);
510   PetscFunctionReturn(0);
511 }
512 
513 #undef __FUNCT__
514 #define __FUNCT__ "DMPlexComputeL2Diff"
515 /*@C
516   DMPlexComputeL2Diff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h.
517 
518   Input Parameters:
519 + dm    - The DM
520 . funcs - The functions to evaluate for each field component
521 . ctxs  - Optional array of contexts to pass to each function, or NULL.
522 - X     - The coefficient vector u_h
523 
524   Output Parameter:
525 . diff - The diff ||u - u_h||_2
526 
527   Level: developer
528 
529 .seealso: DMPlexProjectFunction(), DMPlexComputeL2FieldDiff(), DMPlexComputeL2GradientDiff()
530 @*/
531 PetscErrorCode DMPlexComputeL2Diff(DM dm, void (**funcs)(const PetscReal [], PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff)
532 {
533   const PetscInt   debug = 0;
534   PetscSection     section;
535   PetscQuadrature  quad;
536   Vec              localX;
537   PetscScalar     *funcVal, *interpolant;
538   PetscReal       *coords, *v0, *J, *invJ, detJ;
539   PetscReal        localDiff = 0.0;
540   const PetscReal *quadPoints, *quadWeights;
541   PetscInt         dim, numFields, numComponents = 0, numQuadPoints, cStart, cEnd, c, field, fieldOffset;
542   PetscErrorCode   ierr;
543 
544   PetscFunctionBegin;
545   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
546   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
547   ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);
548   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
549   ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
550   ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
551   for (field = 0; field < numFields; ++field) {
552     PetscObject  obj;
553     PetscClassId id;
554     PetscInt     Nc;
555 
556     ierr = DMGetField(dm, field, &obj);CHKERRQ(ierr);
557     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
558     if (id == PETSCFE_CLASSID) {
559       PetscFE fe = (PetscFE) obj;
560 
561       ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr);
562       ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
563     } else if (id == PETSCFV_CLASSID) {
564       PetscFV fv = (PetscFV) obj;
565 
566       ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr);
567       ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr);
568     } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
569     numComponents += Nc;
570   }
571   ierr = PetscQuadratureGetData(quad, NULL, &numQuadPoints, &quadPoints, &quadWeights);CHKERRQ(ierr);
572   ierr = DMPlexProjectFunctionLocal(dm, funcs, ctxs, INSERT_BC_VALUES, localX);CHKERRQ(ierr);
573   ierr = PetscMalloc6(numComponents,&funcVal,numComponents,&interpolant,dim,&coords,dim,&v0,dim*dim,&J,dim*dim,&invJ);CHKERRQ(ierr);
574   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
575   for (c = cStart; c < cEnd; ++c) {
576     PetscScalar *x = NULL;
577     PetscReal    elemDiff = 0.0;
578 
579     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
580     if (detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", detJ, c);
581     ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
582 
583     for (field = 0, fieldOffset = 0; field < numFields; ++field) {
584       PetscObject  obj;
585       PetscClassId id;
586       void * const ctx = ctxs ? ctxs[field] : NULL;
587       PetscInt     Nb, Nc, q, fc;
588 
589       ierr = DMGetField(dm, field, &obj);CHKERRQ(ierr);
590       ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
591       if (id == PETSCFE_CLASSID)      {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);}
592       else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;}
593       else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
594       if (debug) {
595         char title[1024];
596         ierr = PetscSNPrintf(title, 1023, "Solution for Field %d", field);CHKERRQ(ierr);
597         ierr = DMPrintCellVector(c, title, Nb*Nc, &x[fieldOffset]);CHKERRQ(ierr);
598       }
599       for (q = 0; q < numQuadPoints; ++q) {
600         CoordinatesRefToReal(dim, dim, v0, J, &quadPoints[q*dim], coords);
601         (*funcs[field])(coords, funcVal, ctx);
602         if (id == PETSCFE_CLASSID)      {ierr = PetscFEInterpolate_Static((PetscFE) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);}
603         else if (id == PETSCFV_CLASSID) {ierr = PetscFVInterpolate_Static((PetscFV) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);}
604         else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
605         for (fc = 0; fc < Nc; ++fc) {
606           if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, "    elem %d field %d diff %g\n", c, field, PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*quadWeights[q]*detJ);CHKERRQ(ierr);}
607           elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*quadWeights[q]*detJ;
608         }
609       }
610       fieldOffset += Nb*Nc;
611     }
612     ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
613     if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, "  elem %d diff %g\n", c, elemDiff);CHKERRQ(ierr);}
614     localDiff += elemDiff;
615   }
616   ierr  = PetscFree6(funcVal,interpolant,coords,v0,J,invJ);CHKERRQ(ierr);
617   ierr  = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
618   ierr  = MPI_Allreduce(&localDiff, diff, 1, MPIU_REAL, MPI_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
619   *diff = PetscSqrtReal(*diff);
620   PetscFunctionReturn(0);
621 }
622 
623 #undef __FUNCT__
624 #define __FUNCT__ "DMPlexComputeL2GradientDiff"
625 /*@C
626   DMPlexComputeL2GradientDiff - This function computes the L_2 difference between the gradient of a function u and an FEM interpolant solution grad u_h.
627 
628   Input Parameters:
629 + dm    - The DM
630 . funcs - The gradient functions to evaluate for each field component
631 . ctxs  - Optional array of contexts to pass to each function, or NULL.
632 . X     - The coefficient vector u_h
633 - n     - The vector to project along
634 
635   Output Parameter:
636 . diff - The diff ||(grad u - grad u_h) . n||_2
637 
638   Level: developer
639 
640 .seealso: DMPlexProjectFunction(), DMPlexComputeL2Diff()
641 @*/
642 PetscErrorCode DMPlexComputeL2GradientDiff(DM dm, void (**funcs)(const PetscReal [], const PetscReal [], PetscScalar *, void *), void **ctxs, Vec X, const PetscReal n[], PetscReal *diff)
643 {
644   const PetscInt  debug = 0;
645   PetscSection    section;
646   PetscQuadrature quad;
647   Vec             localX;
648   PetscScalar    *funcVal, *interpolantVec;
649   PetscReal      *coords, *realSpaceDer, *v0, *J, *invJ, detJ;
650   PetscReal       localDiff = 0.0;
651   PetscInt        dim, numFields, numComponents = 0, cStart, cEnd, c, field, fieldOffset, comp;
652   PetscErrorCode  ierr;
653 
654   PetscFunctionBegin;
655   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
656   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
657   ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);
658   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
659   ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
660   ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
661   for (field = 0; field < numFields; ++field) {
662     PetscFE  fe;
663     PetscInt Nc;
664 
665     ierr = DMGetField(dm, field, (PetscObject *) &fe);CHKERRQ(ierr);
666     ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr);
667     ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
668     numComponents += Nc;
669   }
670   /* ierr = DMPlexProjectFunctionLocal(dm, fe, funcs, INSERT_BC_VALUES, localX);CHKERRQ(ierr); */
671   ierr = PetscMalloc7(numComponents,&funcVal,dim,&coords,dim,&realSpaceDer,dim,&v0,dim*dim,&J,dim*dim,&invJ,dim,&interpolantVec);CHKERRQ(ierr);
672   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
673   for (c = cStart; c < cEnd; ++c) {
674     PetscScalar *x = NULL;
675     PetscReal    elemDiff = 0.0;
676 
677     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
678     if (detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", detJ, c);
679     ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
680 
681     for (field = 0, comp = 0, fieldOffset = 0; field < numFields; ++field) {
682       PetscFE          fe;
683       void * const     ctx = ctxs ? ctxs[field] : NULL;
684       const PetscReal *quadPoints, *quadWeights;
685       PetscReal       *basisDer;
686       PetscInt         numQuadPoints, Nb, Ncomp, q, d, e, fc, f, g;
687 
688       ierr = DMGetField(dm, field, (PetscObject *) &fe);CHKERRQ(ierr);
689       ierr = PetscQuadratureGetData(quad, NULL, &numQuadPoints, &quadPoints, &quadWeights);CHKERRQ(ierr);
690       ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr);
691       ierr = PetscFEGetNumComponents(fe, &Ncomp);CHKERRQ(ierr);
692       ierr = PetscFEGetDefaultTabulation(fe, NULL, &basisDer, NULL);CHKERRQ(ierr);
693       if (debug) {
694         char title[1024];
695         ierr = PetscSNPrintf(title, 1023, "Solution for Field %d", field);CHKERRQ(ierr);
696         ierr = DMPrintCellVector(c, title, Nb*Ncomp, &x[fieldOffset]);CHKERRQ(ierr);
697       }
698       for (q = 0; q < numQuadPoints; ++q) {
699         for (d = 0; d < dim; d++) {
700           coords[d] = v0[d];
701           for (e = 0; e < dim; e++) {
702             coords[d] += J[d*dim+e]*(quadPoints[q*dim+e] + 1.0);
703           }
704         }
705         (*funcs[field])(coords, n, funcVal, ctx);
706         for (fc = 0; fc < Ncomp; ++fc) {
707           PetscScalar interpolant = 0.0;
708 
709           for (d = 0; d < dim; ++d) interpolantVec[d] = 0.0;
710           for (f = 0; f < Nb; ++f) {
711             const PetscInt fidx = f*Ncomp+fc;
712 
713             for (d = 0; d < dim; ++d) {
714               realSpaceDer[d] = 0.0;
715               for (g = 0; g < dim; ++g) {
716                 realSpaceDer[d] += invJ[g*dim+d]*basisDer[(q*Nb*Ncomp+fidx)*dim+g];
717               }
718               interpolantVec[d] += x[fieldOffset+fidx]*realSpaceDer[d];
719             }
720           }
721           for (d = 0; d < dim; ++d) interpolant += interpolantVec[d]*n[d];
722           if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, "    elem %d fieldDer %d diff %g\n", c, field, PetscSqr(PetscRealPart(interpolant - funcVal[fc]))*quadWeights[q]*detJ);CHKERRQ(ierr);}
723           elemDiff += PetscSqr(PetscRealPart(interpolant - funcVal[fc]))*quadWeights[q]*detJ;
724         }
725       }
726       comp        += Ncomp;
727       fieldOffset += Nb*Ncomp;
728     }
729     ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
730     if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, "  elem %d diff %g\n", c, elemDiff);CHKERRQ(ierr);}
731     localDiff += elemDiff;
732   }
733   ierr  = PetscFree7(funcVal,coords,realSpaceDer,v0,J,invJ,interpolantVec);CHKERRQ(ierr);
734   ierr  = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
735   ierr  = MPI_Allreduce(&localDiff, diff, 1, MPIU_REAL, MPI_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
736   *diff = PetscSqrtReal(*diff);
737   PetscFunctionReturn(0);
738 }
739 
740 #undef __FUNCT__
741 #define __FUNCT__ "DMPlexComputeL2FieldDiff"
742 /*@C
743   DMPlexComputeL2FieldDiff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h, separated into field components.
744 
745   Input Parameters:
746 + dm    - The DM
747 . funcs - The functions to evaluate for each field component
748 . ctxs  - Optional array of contexts to pass to each function, or NULL.
749 - X     - The coefficient vector u_h
750 
751   Output Parameter:
752 . diff - The array of differences, ||u^f - u^f_h||_2
753 
754   Level: developer
755 
756 .seealso: DMPlexProjectFunction(), DMPlexComputeL2Diff(), DMPlexComputeL2GradientDiff()
757 @*/
758 PetscErrorCode DMPlexComputeL2FieldDiff(DM dm, void (**funcs)(const PetscReal [], PetscScalar *, void *), void **ctxs, Vec X, PetscReal diff[])
759 {
760   const PetscInt   debug = 0;
761   PetscSection     section;
762   PetscQuadrature  quad;
763   Vec              localX;
764   PetscScalar     *funcVal, *interpolant;
765   PetscReal       *coords, *v0, *J, *invJ, detJ;
766   PetscReal       *localDiff;
767   const PetscReal *quadPoints, *quadWeights;
768   PetscInt         dim, numFields, numComponents = 0, numQuadPoints, cStart, cEnd, c, field, fieldOffset;
769   PetscErrorCode   ierr;
770 
771   PetscFunctionBegin;
772   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
773   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
774   ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);
775   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
776   ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
777   ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
778   for (field = 0; field < numFields; ++field) {
779     PetscObject  obj;
780     PetscClassId id;
781     PetscInt     Nc;
782 
783     ierr = DMGetField(dm, field, &obj);CHKERRQ(ierr);
784     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
785     if (id == PETSCFE_CLASSID) {
786       PetscFE fe = (PetscFE) obj;
787 
788       ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr);
789       ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
790     } else if (id == PETSCFV_CLASSID) {
791       PetscFV fv = (PetscFV) obj;
792 
793       ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr);
794       ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr);
795     } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
796     numComponents += Nc;
797   }
798   ierr = PetscQuadratureGetData(quad, NULL, &numQuadPoints, &quadPoints, &quadWeights);CHKERRQ(ierr);
799   ierr = DMPlexProjectFunctionLocal(dm, funcs, ctxs, INSERT_BC_VALUES, localX);CHKERRQ(ierr);
800   ierr = PetscCalloc7(numFields,&localDiff,numComponents,&funcVal,numComponents,&interpolant,dim,&coords,dim,&v0,dim*dim,&J,dim*dim,&invJ);CHKERRQ(ierr);
801   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
802   for (c = cStart; c < cEnd; ++c) {
803     PetscScalar *x = NULL;
804 
805     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
806     if (detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", detJ, c);
807     ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
808 
809     for (field = 0, fieldOffset = 0; field < numFields; ++field) {
810       PetscObject  obj;
811       PetscClassId id;
812       void * const ctx = ctxs ? ctxs[field] : NULL;
813       PetscInt     Nb, Nc, q, fc;
814 
815       PetscReal       elemDiff = 0.0;
816 
817       ierr = DMGetField(dm, field, &obj);CHKERRQ(ierr);
818       ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
819       if (id == PETSCFE_CLASSID)      {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);}
820       else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;}
821       else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
822       if (debug) {
823         char title[1024];
824         ierr = PetscSNPrintf(title, 1023, "Solution for Field %d", field);CHKERRQ(ierr);
825         ierr = DMPrintCellVector(c, title, Nb*Nc, &x[fieldOffset]);CHKERRQ(ierr);
826       }
827       for (q = 0; q < numQuadPoints; ++q) {
828         CoordinatesRefToReal(dim, dim, v0, J, &quadPoints[q*dim], coords);
829         (*funcs[field])(coords, funcVal, ctx);
830         if (id == PETSCFE_CLASSID)      {ierr = PetscFEInterpolate_Static((PetscFE) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);}
831         else if (id == PETSCFV_CLASSID) {ierr = PetscFVInterpolate_Static((PetscFV) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);}
832         else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", field);
833         for (fc = 0; fc < Nc; ++fc) {
834           if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, "    elem %d field %d diff %g\n", c, field, PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*quadWeights[q]*detJ);CHKERRQ(ierr);}
835           elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*quadWeights[q]*detJ;
836         }
837       }
838       fieldOffset += Nb*Nc;
839       localDiff[field] += elemDiff;
840     }
841     ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr);
842   }
843   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
844   ierr = MPI_Allreduce(localDiff, diff, numFields, MPIU_REAL, MPI_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
845   for (field = 0; field < numFields; ++field) diff[field] = PetscSqrtReal(diff[field]);
846   ierr = PetscFree7(localDiff,funcVal,interpolant,coords,v0,J,invJ);CHKERRQ(ierr);
847   PetscFunctionReturn(0);
848 }
849 
850 #undef __FUNCT__
851 #define __FUNCT__ "DMPlexComputeIntegralFEM"
852 /*@
853   DMPlexComputeIntegralFEM - Form the local integral F from the local input X using pointwise functions specified by the user
854 
855   Input Parameters:
856 + dm - The mesh
857 . X  - Local input vector
858 - user - The user context
859 
860   Output Parameter:
861 . integral - Local integral for each field
862 
863   Level: developer
864 
865 .seealso: DMPlexComputeResidualFEM()
866 @*/
867 PetscErrorCode DMPlexComputeIntegralFEM(DM dm, Vec X, PetscReal *integral, void *user)
868 {
869   DM_Plex          *mesh  = (DM_Plex *) dm->data;
870   DM                dmAux;
871   Vec               localX, A;
872   PetscDS           prob, probAux = NULL;
873   PetscQuadrature   q;
874   PetscCellGeometry geom;
875   PetscSection      section, sectionAux;
876   PetscReal        *v0, *J, *invJ, *detJ;
877   PetscScalar      *u, *a = NULL;
878   PetscInt          dim, Nf, f, numCells, cStart, cEnd, c;
879   PetscInt          totDim, totDimAux;
880   PetscErrorCode    ierr;
881 
882   PetscFunctionBegin;
883   /*ierr = PetscLogEventBegin(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr);*/
884   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
885   ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
886   ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr);
887   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
888   ierr = DMGetDefaultSection(dm, &section);CHKERRQ(ierr);
889   ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
890   ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr);
891   ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr);
892   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
893   numCells = cEnd - cStart;
894   for (f = 0; f < Nf; ++f) {integral[f]    = 0.0;}
895   ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr);
896   ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &A);CHKERRQ(ierr);
897   if (dmAux) {
898     ierr = DMGetDefaultSection(dmAux, &sectionAux);CHKERRQ(ierr);
899     ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr);
900     ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr);
901   }
902   ierr = DMPlexInsertBoundaryValuesFEM(dm, localX);CHKERRQ(ierr);
903   ierr = PetscMalloc5(numCells*totDim,&u,numCells*dim,&v0,numCells*dim*dim,&J,numCells*dim*dim,&invJ,numCells,&detJ);CHKERRQ(ierr);
904   if (dmAux) {ierr = PetscMalloc1(numCells*totDimAux, &a);CHKERRQ(ierr);}
905   for (c = cStart; c < cEnd; ++c) {
906     PetscScalar *x = NULL;
907     PetscInt     i;
908 
909     ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, &v0[c*dim], &J[c*dim*dim], &invJ[c*dim*dim], &detJ[c]);CHKERRQ(ierr);
910     if (detJ[c] <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", detJ[c], c);
911     ierr = DMPlexVecGetClosure(dm, section, localX, c, NULL, &x);CHKERRQ(ierr);
912     for (i = 0; i < totDim; ++i) u[c*totDim+i] = x[i];
913     ierr = DMPlexVecRestoreClosure(dm, section, localX, c, NULL, &x);CHKERRQ(ierr);
914     if (dmAux) {
915       ierr = DMPlexVecGetClosure(dmAux, sectionAux, A, c, NULL, &x);CHKERRQ(ierr);
916       for (i = 0; i < totDimAux; ++i) a[c*totDimAux+i] = x[i];
917       ierr = DMPlexVecRestoreClosure(dmAux, sectionAux, A, c, NULL, &x);CHKERRQ(ierr);
918     }
919   }
920   for (f = 0; f < Nf; ++f) {
921     PetscFE  fe;
922     PetscInt numQuadPoints, Nb;
923     /* Conforming batches */
924     PetscInt numChunks, numBatches, numBlocks, Ne, blockSize, batchSize;
925     /* Remainder */
926     PetscInt Nr, offset;
927 
928     ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fe);CHKERRQ(ierr);
929     ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr);
930     ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr);
931     ierr = PetscFEGetTileSizes(fe, NULL, &numBlocks, NULL, &numBatches);CHKERRQ(ierr);
932     ierr = PetscQuadratureGetData(q, NULL, &numQuadPoints, NULL, NULL);CHKERRQ(ierr);
933     blockSize = Nb*numQuadPoints;
934     batchSize = numBlocks * blockSize;
935     ierr =  PetscFESetTileSizes(fe, blockSize, numBlocks, batchSize, numBatches);CHKERRQ(ierr);
936     numChunks = numCells / (numBatches*batchSize);
937     Ne        = numChunks*numBatches*batchSize;
938     Nr        = numCells % (numBatches*batchSize);
939     offset    = numCells - Nr;
940     geom.v0   = v0;
941     geom.J    = J;
942     geom.invJ = invJ;
943     geom.detJ = detJ;
944     ierr = PetscFEIntegrate(fe, prob, f, Ne, geom, u, probAux, a, integral);CHKERRQ(ierr);
945     geom.v0   = &v0[offset*dim];
946     geom.J    = &J[offset*dim*dim];
947     geom.invJ = &invJ[offset*dim*dim];
948     geom.detJ = &detJ[offset];
949     ierr = PetscFEIntegrate(fe, prob, f, Nr, geom, &u[offset*totDim], probAux, &a[offset*totDimAux], integral);CHKERRQ(ierr);
950   }
951   ierr = PetscFree5(u,v0,J,invJ,detJ);CHKERRQ(ierr);
952   if (dmAux) {ierr = PetscFree(a);CHKERRQ(ierr);}
953   if (mesh->printFEM) {
954     ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Local integral:");CHKERRQ(ierr);
955     for (f = 0; f < Nf; ++f) {ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), " %g", integral[f]);CHKERRQ(ierr);}
956     ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "\n");CHKERRQ(ierr);
957   }
958   ierr  = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
959   /* TODO: Allreduce for integral */
960   /*ierr = PetscLogEventEnd(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr);*/
961   PetscFunctionReturn(0);
962 }
963 
964 #undef __FUNCT__
965 #define __FUNCT__ "DMPlexComputeInterpolatorFEM"
966 /*@
967   DMPlexComputeInterpolatorFEM - Form the local portion of the interpolation matrix I from the coarse DM to the uniformly refined DM.
968 
969   Input Parameters:
970 + dmf  - The fine mesh
971 . dmc  - The coarse mesh
972 - user - The user context
973 
974   Output Parameter:
975 . In  - The interpolation matrix
976 
977   Note:
978   The first member of the user context must be an FEMContext.
979 
980   We form the residual one batch of elements at a time. This allows us to offload work onto an accelerator,
981   like a GPU, or vectorize on a multicore machine.
982 
983   Level: developer
984 
985 .seealso: DMPlexComputeJacobianFEM()
986 @*/
987 PetscErrorCode DMPlexComputeInterpolatorFEM(DM dmc, DM dmf, Mat In, void *user)
988 {
989   DM_Plex          *mesh  = (DM_Plex *) dmc->data;
990   const char       *name  = "Interpolator";
991   PetscDS           prob;
992   PetscFE          *feRef;
993   PetscSection      fsection, fglobalSection;
994   PetscSection      csection, cglobalSection;
995   PetscScalar      *elemMat;
996   PetscInt          dim, Nf, f, fieldI, fieldJ, offsetI, offsetJ, cStart, cEnd, c;
997   PetscInt          cTotDim, rTotDim = 0;
998   PetscErrorCode    ierr;
999 
1000   PetscFunctionBegin;
1001   ierr = PetscLogEventBegin(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr);
1002   ierr = DMGetDimension(dmf, &dim);CHKERRQ(ierr);
1003   ierr = DMGetDefaultSection(dmf, &fsection);CHKERRQ(ierr);
1004   ierr = DMGetDefaultGlobalSection(dmf, &fglobalSection);CHKERRQ(ierr);
1005   ierr = DMGetDefaultSection(dmc, &csection);CHKERRQ(ierr);
1006   ierr = DMGetDefaultGlobalSection(dmc, &cglobalSection);CHKERRQ(ierr);
1007   ierr = PetscSectionGetNumFields(fsection, &Nf);CHKERRQ(ierr);
1008   ierr = DMPlexGetHeightStratum(dmc, 0, &cStart, &cEnd);CHKERRQ(ierr);
1009   ierr = DMGetDS(dmf, &prob);CHKERRQ(ierr);
1010   ierr = PetscMalloc1(Nf,&feRef);CHKERRQ(ierr);
1011   for (f = 0; f < Nf; ++f) {
1012     PetscFE  fe;
1013     PetscInt rNb, Nc;
1014 
1015     ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fe);CHKERRQ(ierr);
1016     ierr = PetscFERefine(fe, &feRef[f]);CHKERRQ(ierr);
1017     ierr = PetscFEGetDimension(feRef[f], &rNb);CHKERRQ(ierr);
1018     ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
1019     rTotDim += rNb*Nc;
1020   }
1021   ierr = PetscDSGetTotalDimension(prob, &cTotDim);CHKERRQ(ierr);
1022   ierr = PetscMalloc1(rTotDim*cTotDim,&elemMat);CHKERRQ(ierr);
1023   ierr = PetscMemzero(elemMat, rTotDim*cTotDim * sizeof(PetscScalar));CHKERRQ(ierr);
1024   for (fieldI = 0, offsetI = 0; fieldI < Nf; ++fieldI) {
1025     PetscDualSpace   Qref;
1026     PetscQuadrature  f;
1027     const PetscReal *qpoints, *qweights;
1028     PetscReal       *points;
1029     PetscInt         npoints = 0, Nc, Np, fpdim, i, k, p, d;
1030 
1031     /* Compose points from all dual basis functionals */
1032     ierr = PetscFEGetDualSpace(feRef[fieldI], &Qref);CHKERRQ(ierr);
1033     ierr = PetscFEGetNumComponents(feRef[fieldI], &Nc);CHKERRQ(ierr);
1034     ierr = PetscDualSpaceGetDimension(Qref, &fpdim);CHKERRQ(ierr);
1035     for (i = 0; i < fpdim; ++i) {
1036       ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr);
1037       ierr = PetscQuadratureGetData(f, NULL, &Np, NULL, NULL);CHKERRQ(ierr);
1038       npoints += Np;
1039     }
1040     ierr = PetscMalloc1(npoints*dim,&points);CHKERRQ(ierr);
1041     for (i = 0, k = 0; i < fpdim; ++i) {
1042       ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr);
1043       ierr = PetscQuadratureGetData(f, NULL, &Np, &qpoints, NULL);CHKERRQ(ierr);
1044       for (p = 0; p < Np; ++p, ++k) for (d = 0; d < dim; ++d) points[k*dim+d] = qpoints[p*dim+d];
1045     }
1046 
1047     for (fieldJ = 0, offsetJ = 0; fieldJ < Nf; ++fieldJ) {
1048       PetscFE    fe;
1049       PetscReal *B;
1050       PetscInt   NcJ, cpdim, j;
1051 
1052       /* Evaluate basis at points */
1053       ierr = PetscDSGetDiscretization(prob, fieldJ, (PetscObject *) &fe);CHKERRQ(ierr);
1054       ierr = PetscFEGetNumComponents(fe, &NcJ);CHKERRQ(ierr);
1055       ierr = PetscFEGetDimension(fe, &cpdim);CHKERRQ(ierr);
1056       /* For now, fields only interpolate themselves */
1057       if (fieldI == fieldJ) {
1058         if (Nc != NcJ) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in fine space field %d does not match coarse field %d", Nc, NcJ);
1059         ierr = PetscFEGetTabulation(fe, npoints, points, &B, NULL, NULL);CHKERRQ(ierr);
1060         for (i = 0, k = 0; i < fpdim; ++i) {
1061           ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr);
1062           ierr = PetscQuadratureGetData(f, NULL, &Np, NULL, &qweights);CHKERRQ(ierr);
1063           for (p = 0; p < Np; ++p, ++k) {
1064             for (j = 0; j < cpdim; ++j) {
1065               for (c = 0; c < Nc; ++c) elemMat[(offsetI + i*Nc + c)*cTotDim + offsetJ + j*NcJ + c] += B[k*cpdim*NcJ+j*Nc+c]*qweights[p];
1066             }
1067           }
1068         }
1069         ierr = PetscFERestoreTabulation(fe, npoints, points, &B, NULL, NULL);CHKERRQ(ierr);CHKERRQ(ierr);
1070       }
1071       offsetJ += cpdim*NcJ;
1072     }
1073     offsetI += fpdim*Nc;
1074     ierr = PetscFree(points);CHKERRQ(ierr);
1075   }
1076   if (mesh->printFEM > 1) {ierr = DMPrintCellMatrix(0, name, rTotDim, cTotDim, elemMat);CHKERRQ(ierr);}
1077   /* Preallocate matrix */
1078   {
1079     PetscHashJK ht;
1080     PetscLayout rLayout;
1081     PetscInt   *dnz, *onz, *cellCIndices, *cellFIndices;
1082     PetscInt    locRows, rStart, rEnd, cell, r;
1083 
1084     ierr = MatGetLocalSize(In, &locRows, NULL);CHKERRQ(ierr);
1085     ierr = PetscLayoutCreate(PetscObjectComm((PetscObject) In), &rLayout);CHKERRQ(ierr);
1086     ierr = PetscLayoutSetLocalSize(rLayout, locRows);CHKERRQ(ierr);
1087     ierr = PetscLayoutSetBlockSize(rLayout, 1);CHKERRQ(ierr);
1088     ierr = PetscLayoutSetUp(rLayout);CHKERRQ(ierr);
1089     ierr = PetscLayoutGetRange(rLayout, &rStart, &rEnd);CHKERRQ(ierr);
1090     ierr = PetscLayoutDestroy(&rLayout);CHKERRQ(ierr);
1091     ierr = PetscCalloc4(locRows,&dnz,locRows,&onz,cTotDim,&cellCIndices,rTotDim,&cellFIndices);CHKERRQ(ierr);
1092     ierr = PetscHashJKCreate(&ht);CHKERRQ(ierr);
1093     for (cell = cStart; cell < cEnd; ++cell) {
1094       ierr = DMPlexMatGetClosureIndicesRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, cell, cellCIndices, cellFIndices);CHKERRQ(ierr);
1095       for (r = 0; r < rTotDim; ++r) {
1096         PetscHashJKKey  key;
1097         PetscHashJKIter missing, iter;
1098 
1099         key.j = cellFIndices[r];
1100         if (key.j < 0) continue;
1101         for (c = 0; c < cTotDim; ++c) {
1102           key.k = cellCIndices[c];
1103           if (key.k < 0) continue;
1104           ierr = PetscHashJKPut(ht, key, &missing, &iter);CHKERRQ(ierr);
1105           if (missing) {
1106             ierr = PetscHashJKSet(ht, iter, 1);CHKERRQ(ierr);
1107             if ((key.k >= rStart) && (key.k < rEnd)) ++dnz[key.j-rStart];
1108             else                                     ++onz[key.j-rStart];
1109           }
1110         }
1111       }
1112     }
1113     ierr = PetscHashJKDestroy(&ht);CHKERRQ(ierr);
1114     ierr = MatXAIJSetPreallocation(In, 1, dnz, onz, NULL, NULL);CHKERRQ(ierr);
1115     ierr = MatSetOption(In, MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
1116     ierr = PetscFree4(dnz,onz,cellCIndices,cellFIndices);CHKERRQ(ierr);
1117   }
1118   /* Fill matrix */
1119   ierr = MatZeroEntries(In);CHKERRQ(ierr);
1120   for (c = cStart; c < cEnd; ++c) {
1121     ierr = DMPlexMatSetClosureRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, In, c, elemMat, INSERT_VALUES);CHKERRQ(ierr);
1122   }
1123   for (f = 0; f < Nf; ++f) {ierr = PetscFEDestroy(&feRef[f]);CHKERRQ(ierr);}
1124   ierr = PetscFree(feRef);CHKERRQ(ierr);
1125   ierr = PetscFree(elemMat);CHKERRQ(ierr);
1126   ierr = MatAssemblyBegin(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1127   ierr = MatAssemblyEnd(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1128   if (mesh->printFEM) {
1129     ierr = PetscPrintf(PETSC_COMM_WORLD, "%s:\n", name);CHKERRQ(ierr);
1130     ierr = MatChop(In, 1.0e-10);CHKERRQ(ierr);
1131     ierr = MatView(In, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
1132   }
1133   ierr = PetscLogEventEnd(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr);
1134   PetscFunctionReturn(0);
1135 }
1136 
1137 #undef __FUNCT__
1138 #define __FUNCT__ "DMPlexComputeInjectorFEM"
1139 PetscErrorCode DMPlexComputeInjectorFEM(DM dmc, DM dmf, VecScatter *sc, void *user)
1140 {
1141   PetscDS        prob;
1142   PetscFE       *feRef;
1143   Vec            fv, cv;
1144   IS             fis, cis;
1145   PetscSection   fsection, fglobalSection, csection, cglobalSection;
1146   PetscInt      *cmap, *cellCIndices, *cellFIndices, *cindices, *findices;
1147   PetscInt       cTotDim, fTotDim = 0, Nf, f, field, cStart, cEnd, c, dim, d, startC, offsetC, offsetF, m;
1148   PetscErrorCode ierr;
1149 
1150   PetscFunctionBegin;
1151   ierr = PetscLogEventBegin(DMPLEX_InjectorFEM,dmc,dmf,0,0);CHKERRQ(ierr);
1152   ierr = DMGetDimension(dmf, &dim);CHKERRQ(ierr);
1153   ierr = DMGetDefaultSection(dmf, &fsection);CHKERRQ(ierr);
1154   ierr = DMGetDefaultGlobalSection(dmf, &fglobalSection);CHKERRQ(ierr);
1155   ierr = DMGetDefaultSection(dmc, &csection);CHKERRQ(ierr);
1156   ierr = DMGetDefaultGlobalSection(dmc, &cglobalSection);CHKERRQ(ierr);
1157   ierr = PetscSectionGetNumFields(fsection, &Nf);CHKERRQ(ierr);
1158   ierr = DMPlexGetHeightStratum(dmc, 0, &cStart, &cEnd);CHKERRQ(ierr);
1159   ierr = DMGetDS(dmc, &prob);CHKERRQ(ierr);
1160   ierr = PetscMalloc1(Nf,&feRef);CHKERRQ(ierr);
1161   for (f = 0; f < Nf; ++f) {
1162     PetscFE  fe;
1163     PetscInt fNb, Nc;
1164 
1165     ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fe);CHKERRQ(ierr);
1166     ierr = PetscFERefine(fe, &feRef[f]);CHKERRQ(ierr);
1167     ierr = PetscFEGetDimension(feRef[f], &fNb);CHKERRQ(ierr);
1168     ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
1169     fTotDim += fNb*Nc;
1170   }
1171   ierr = PetscDSGetTotalDimension(prob, &cTotDim);CHKERRQ(ierr);
1172   ierr = PetscMalloc1(cTotDim,&cmap);CHKERRQ(ierr);
1173   for (field = 0, offsetC = 0, offsetF = 0; field < Nf; ++field) {
1174     PetscFE        feC;
1175     PetscDualSpace QF, QC;
1176     PetscInt       NcF, NcC, fpdim, cpdim;
1177 
1178     ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &feC);CHKERRQ(ierr);
1179     ierr = PetscFEGetNumComponents(feC, &NcC);CHKERRQ(ierr);
1180     ierr = PetscFEGetNumComponents(feRef[field], &NcF);CHKERRQ(ierr);
1181     if (NcF != NcC) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in fine space field %d does not match coarse field %d", NcF, NcC);
1182     ierr = PetscFEGetDualSpace(feRef[field], &QF);CHKERRQ(ierr);
1183     ierr = PetscDualSpaceGetDimension(QF, &fpdim);CHKERRQ(ierr);
1184     ierr = PetscFEGetDualSpace(feC, &QC);CHKERRQ(ierr);
1185     ierr = PetscDualSpaceGetDimension(QC, &cpdim);CHKERRQ(ierr);
1186     for (c = 0; c < cpdim; ++c) {
1187       PetscQuadrature  cfunc;
1188       const PetscReal *cqpoints;
1189       PetscInt         NpC;
1190 
1191       ierr = PetscDualSpaceGetFunctional(QC, c, &cfunc);CHKERRQ(ierr);
1192       ierr = PetscQuadratureGetData(cfunc, NULL, &NpC, &cqpoints, NULL);CHKERRQ(ierr);
1193       if (NpC != 1) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not know how to do injection for moments");
1194       for (f = 0; f < fpdim; ++f) {
1195         PetscQuadrature  ffunc;
1196         const PetscReal *fqpoints;
1197         PetscReal        sum = 0.0;
1198         PetscInt         NpF, comp;
1199 
1200         ierr = PetscDualSpaceGetFunctional(QF, f, &ffunc);CHKERRQ(ierr);
1201         ierr = PetscQuadratureGetData(ffunc, NULL, &NpF, &fqpoints, NULL);CHKERRQ(ierr);
1202         if (NpC != NpF) continue;
1203         for (d = 0; d < dim; ++d) sum += PetscAbsReal(cqpoints[d] - fqpoints[d]);
1204         if (sum > 1.0e-9) continue;
1205         for (comp = 0; comp < NcC; ++comp) {
1206           cmap[(offsetC+c)*NcC+comp] = (offsetF+f)*NcF+comp;
1207         }
1208         break;
1209       }
1210     }
1211     offsetC += cpdim*NcC;
1212     offsetF += fpdim*NcF;
1213   }
1214   for (f = 0; f < Nf; ++f) {ierr = PetscFEDestroy(&feRef[f]);CHKERRQ(ierr);}
1215   ierr = PetscFree(feRef);CHKERRQ(ierr);
1216 
1217   ierr = DMGetGlobalVector(dmf, &fv);CHKERRQ(ierr);
1218   ierr = DMGetGlobalVector(dmc, &cv);CHKERRQ(ierr);
1219   ierr = VecGetOwnershipRange(cv, &startC, NULL);CHKERRQ(ierr);
1220   ierr = PetscSectionGetConstrainedStorageSize(cglobalSection, &m);CHKERRQ(ierr);
1221   ierr = PetscMalloc2(cTotDim,&cellCIndices,fTotDim,&cellFIndices);CHKERRQ(ierr);
1222   ierr = PetscMalloc1(m,&cindices);CHKERRQ(ierr);
1223   ierr = PetscMalloc1(m,&findices);CHKERRQ(ierr);
1224   for (d = 0; d < m; ++d) cindices[d] = findices[d] = -1;
1225   for (c = cStart; c < cEnd; ++c) {
1226     ierr = DMPlexMatGetClosureIndicesRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, c, cellCIndices, cellFIndices);CHKERRQ(ierr);
1227     for (d = 0; d < cTotDim; ++d) {
1228       if (cellCIndices[d] < 0) continue;
1229       if ((findices[cellCIndices[d]-startC] >= 0) && (findices[cellCIndices[d]-startC] != cellFIndices[cmap[d]])) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Coarse dof %d maps to both %d and %d", cindices[cellCIndices[d]-startC], findices[cellCIndices[d]-startC], cellFIndices[cmap[d]]);
1230       cindices[cellCIndices[d]-startC] = cellCIndices[d];
1231       findices[cellCIndices[d]-startC] = cellFIndices[cmap[d]];
1232     }
1233   }
1234   ierr = PetscFree(cmap);CHKERRQ(ierr);
1235   ierr = PetscFree2(cellCIndices,cellFIndices);CHKERRQ(ierr);
1236 
1237   ierr = ISCreateGeneral(PETSC_COMM_SELF, m, cindices, PETSC_OWN_POINTER, &cis);CHKERRQ(ierr);
1238   ierr = ISCreateGeneral(PETSC_COMM_SELF, m, findices, PETSC_OWN_POINTER, &fis);CHKERRQ(ierr);
1239   ierr = VecScatterCreate(cv, cis, fv, fis, sc);CHKERRQ(ierr);
1240   ierr = ISDestroy(&cis);CHKERRQ(ierr);
1241   ierr = ISDestroy(&fis);CHKERRQ(ierr);
1242   ierr = DMRestoreGlobalVector(dmf, &fv);CHKERRQ(ierr);
1243   ierr = DMRestoreGlobalVector(dmc, &cv);CHKERRQ(ierr);
1244   ierr = PetscLogEventEnd(DMPLEX_InjectorFEM,dmc,dmf,0,0);CHKERRQ(ierr);
1245   PetscFunctionReturn(0);
1246 }
1247