xref: /petsc/src/dm/impls/plex/plexgeometry.c (revision feff33ee0b5b037fa8f9f294dede656a2f85cc47)
1 #include <petsc/private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
2 #include <petsc/private/petscfeimpl.h>  /*I      "petscfe.h"       I*/
3 #include <petscblaslapack.h>
4 #include <petsctime.h>
5 
6 static PetscErrorCode DMPlexGetLineIntersection_2D_Internal(const PetscReal segmentA[], const PetscReal segmentB[], PetscReal intersection[], PetscBool *hasIntersection)
7 {
8   const PetscReal p0_x  = segmentA[0*2+0];
9   const PetscReal p0_y  = segmentA[0*2+1];
10   const PetscReal p1_x  = segmentA[1*2+0];
11   const PetscReal p1_y  = segmentA[1*2+1];
12   const PetscReal p2_x  = segmentB[0*2+0];
13   const PetscReal p2_y  = segmentB[0*2+1];
14   const PetscReal p3_x  = segmentB[1*2+0];
15   const PetscReal p3_y  = segmentB[1*2+1];
16   const PetscReal s1_x  = p1_x - p0_x;
17   const PetscReal s1_y  = p1_y - p0_y;
18   const PetscReal s2_x  = p3_x - p2_x;
19   const PetscReal s2_y  = p3_y - p2_y;
20   const PetscReal denom = (-s2_x * s1_y + s1_x * s2_y);
21 
22   PetscFunctionBegin;
23   *hasIntersection = PETSC_FALSE;
24   /* Non-parallel lines */
25   if (denom != 0.0) {
26     const PetscReal s = (-s1_y * (p0_x - p2_x) + s1_x * (p0_y - p2_y)) / denom;
27     const PetscReal t = ( s2_x * (p0_y - p2_y) - s2_y * (p0_x - p2_x)) / denom;
28 
29     if (s >= 0 && s <= 1 && t >= 0 && t <= 1) {
30       *hasIntersection = PETSC_TRUE;
31       if (intersection) {
32         intersection[0] = p0_x + (t * s1_x);
33         intersection[1] = p0_y + (t * s1_y);
34       }
35     }
36   }
37   PetscFunctionReturn(0);
38 }
39 
40 static PetscErrorCode DMPlexLocatePoint_Simplex_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell)
41 {
42   const PetscInt  embedDim = 2;
43   const PetscReal eps      = PETSC_SQRT_MACHINE_EPSILON;
44   PetscReal       x        = PetscRealPart(point[0]);
45   PetscReal       y        = PetscRealPart(point[1]);
46   PetscReal       v0[2], J[4], invJ[4], detJ;
47   PetscReal       xi, eta;
48   PetscErrorCode  ierr;
49 
50   PetscFunctionBegin;
51   ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
52   xi  = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]);
53   eta = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]);
54 
55   if ((xi >= -eps) && (eta >= -eps) && (xi + eta <= 2.0+eps)) *cell = c;
56   else *cell = DMLOCATEPOINT_POINT_NOT_FOUND;
57   PetscFunctionReturn(0);
58 }
59 
60 static PetscErrorCode DMPlexClosestPoint_Simplex_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscReal cpoint[])
61 {
62   const PetscInt  embedDim = 2;
63   PetscReal       x        = PetscRealPart(point[0]);
64   PetscReal       y        = PetscRealPart(point[1]);
65   PetscReal       v0[2], J[4], invJ[4], detJ;
66   PetscReal       xi, eta, r;
67   PetscErrorCode  ierr;
68 
69   PetscFunctionBegin;
70   ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
71   xi  = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]);
72   eta = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]);
73 
74   xi  = PetscMax(xi,  0.0);
75   eta = PetscMax(eta, 0.0);
76   if (xi + eta > 2.0) {
77     r    = (xi + eta)/2.0;
78     xi  /= r;
79     eta /= r;
80   }
81   cpoint[0] = J[0*embedDim+0]*xi + J[0*embedDim+1]*eta + v0[0];
82   cpoint[1] = J[1*embedDim+0]*xi + J[1*embedDim+1]*eta + v0[1];
83   PetscFunctionReturn(0);
84 }
85 
86 static PetscErrorCode DMPlexLocatePoint_General_2D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell)
87 {
88   PetscSection       coordSection;
89   Vec             coordsLocal;
90   PetscScalar    *coords = NULL;
91   const PetscInt  faces[8]  = {0, 1, 1, 2, 2, 3, 3, 0};
92   PetscReal       x         = PetscRealPart(point[0]);
93   PetscReal       y         = PetscRealPart(point[1]);
94   PetscInt        crossings = 0, f;
95   PetscErrorCode  ierr;
96 
97   PetscFunctionBegin;
98   ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr);
99   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
100   ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr);
101   for (f = 0; f < 4; ++f) {
102     PetscReal x_i   = PetscRealPart(coords[faces[2*f+0]*2+0]);
103     PetscReal y_i   = PetscRealPart(coords[faces[2*f+0]*2+1]);
104     PetscReal x_j   = PetscRealPart(coords[faces[2*f+1]*2+0]);
105     PetscReal y_j   = PetscRealPart(coords[faces[2*f+1]*2+1]);
106     PetscReal slope = (y_j - y_i) / (x_j - x_i);
107     PetscBool cond1 = (x_i <= x) && (x < x_j) ? PETSC_TRUE : PETSC_FALSE;
108     PetscBool cond2 = (x_j <= x) && (x < x_i) ? PETSC_TRUE : PETSC_FALSE;
109     PetscBool above = (y < slope * (x - x_i) + y_i) ? PETSC_TRUE : PETSC_FALSE;
110     if ((cond1 || cond2)  && above) ++crossings;
111   }
112   if (crossings % 2) *cell = c;
113   else *cell = DMLOCATEPOINT_POINT_NOT_FOUND;
114   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr);
115   PetscFunctionReturn(0);
116 }
117 
118 static PetscErrorCode DMPlexLocatePoint_Simplex_3D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell)
119 {
120   const PetscInt embedDim = 3;
121   PetscReal      v0[3], J[9], invJ[9], detJ;
122   PetscReal      x = PetscRealPart(point[0]);
123   PetscReal      y = PetscRealPart(point[1]);
124   PetscReal      z = PetscRealPart(point[2]);
125   PetscReal      xi, eta, zeta;
126   PetscErrorCode ierr;
127 
128   PetscFunctionBegin;
129   ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr);
130   xi   = invJ[0*embedDim+0]*(x - v0[0]) + invJ[0*embedDim+1]*(y - v0[1]) + invJ[0*embedDim+2]*(z - v0[2]);
131   eta  = invJ[1*embedDim+0]*(x - v0[0]) + invJ[1*embedDim+1]*(y - v0[1]) + invJ[1*embedDim+2]*(z - v0[2]);
132   zeta = invJ[2*embedDim+0]*(x - v0[0]) + invJ[2*embedDim+1]*(y - v0[1]) + invJ[2*embedDim+2]*(z - v0[2]);
133 
134   if ((xi >= 0.0) && (eta >= 0.0) && (zeta >= 0.0) && (xi + eta + zeta <= 2.0)) *cell = c;
135   else *cell = DMLOCATEPOINT_POINT_NOT_FOUND;
136   PetscFunctionReturn(0);
137 }
138 
139 static PetscErrorCode DMPlexLocatePoint_General_3D_Internal(DM dm, const PetscScalar point[], PetscInt c, PetscInt *cell)
140 {
141   PetscSection   coordSection;
142   Vec            coordsLocal;
143   PetscScalar   *coords = NULL;
144   const PetscInt faces[24] = {0, 3, 2, 1,  5, 4, 7, 6,  3, 0, 4, 5,
145                               1, 2, 6, 7,  3, 5, 6, 2,  0, 1, 7, 4};
146   PetscBool      found = PETSC_TRUE;
147   PetscInt       f;
148   PetscErrorCode ierr;
149 
150   PetscFunctionBegin;
151   ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr);
152   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
153   ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr);
154   for (f = 0; f < 6; ++f) {
155     /* Check the point is under plane */
156     /*   Get face normal */
157     PetscReal v_i[3];
158     PetscReal v_j[3];
159     PetscReal normal[3];
160     PetscReal pp[3];
161     PetscReal dot;
162 
163     v_i[0]    = PetscRealPart(coords[faces[f*4+3]*3+0]-coords[faces[f*4+0]*3+0]);
164     v_i[1]    = PetscRealPart(coords[faces[f*4+3]*3+1]-coords[faces[f*4+0]*3+1]);
165     v_i[2]    = PetscRealPart(coords[faces[f*4+3]*3+2]-coords[faces[f*4+0]*3+2]);
166     v_j[0]    = PetscRealPart(coords[faces[f*4+1]*3+0]-coords[faces[f*4+0]*3+0]);
167     v_j[1]    = PetscRealPart(coords[faces[f*4+1]*3+1]-coords[faces[f*4+0]*3+1]);
168     v_j[2]    = PetscRealPart(coords[faces[f*4+1]*3+2]-coords[faces[f*4+0]*3+2]);
169     normal[0] = v_i[1]*v_j[2] - v_i[2]*v_j[1];
170     normal[1] = v_i[2]*v_j[0] - v_i[0]*v_j[2];
171     normal[2] = v_i[0]*v_j[1] - v_i[1]*v_j[0];
172     pp[0]     = PetscRealPart(coords[faces[f*4+0]*3+0] - point[0]);
173     pp[1]     = PetscRealPart(coords[faces[f*4+0]*3+1] - point[1]);
174     pp[2]     = PetscRealPart(coords[faces[f*4+0]*3+2] - point[2]);
175     dot       = normal[0]*pp[0] + normal[1]*pp[1] + normal[2]*pp[2];
176 
177     /* Check that projected point is in face (2D location problem) */
178     if (dot < 0.0) {
179       found = PETSC_FALSE;
180       break;
181     }
182   }
183   if (found) *cell = c;
184   else *cell = DMLOCATEPOINT_POINT_NOT_FOUND;
185   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &coords);CHKERRQ(ierr);
186   PetscFunctionReturn(0);
187 }
188 
189 static PetscErrorCode PetscGridHashInitialize_Internal(PetscGridHash box, PetscInt dim, const PetscScalar point[])
190 {
191   PetscInt d;
192 
193   PetscFunctionBegin;
194   box->dim = dim;
195   for (d = 0; d < dim; ++d) box->lower[d] = box->upper[d] = PetscRealPart(point[d]);
196   PetscFunctionReturn(0);
197 }
198 
199 PetscErrorCode PetscGridHashCreate(MPI_Comm comm, PetscInt dim, const PetscScalar point[], PetscGridHash *box)
200 {
201   PetscErrorCode ierr;
202 
203   PetscFunctionBegin;
204   ierr = PetscMalloc1(1, box);CHKERRQ(ierr);
205   ierr = PetscGridHashInitialize_Internal(*box, dim, point);CHKERRQ(ierr);
206   PetscFunctionReturn(0);
207 }
208 
209 PetscErrorCode PetscGridHashEnlarge(PetscGridHash box, const PetscScalar point[])
210 {
211   PetscInt d;
212 
213   PetscFunctionBegin;
214   for (d = 0; d < box->dim; ++d) {
215     box->lower[d] = PetscMin(box->lower[d], PetscRealPart(point[d]));
216     box->upper[d] = PetscMax(box->upper[d], PetscRealPart(point[d]));
217   }
218   PetscFunctionReturn(0);
219 }
220 
221 /*
222   PetscGridHashSetGrid - Divide the grid into boxes
223 
224   Not collective
225 
226   Input Parameters:
227 + box - The grid hash object
228 . n   - The number of boxes in each dimension, or PETSC_DETERMINE
229 - h   - The box size in each dimension, only used if n[d] == PETSC_DETERMINE
230 
231   Level: developer
232 
233 .seealso: PetscGridHashCreate()
234 */
235 PetscErrorCode PetscGridHashSetGrid(PetscGridHash box, const PetscInt n[], const PetscReal h[])
236 {
237   PetscInt d;
238 
239   PetscFunctionBegin;
240   for (d = 0; d < box->dim; ++d) {
241     box->extent[d] = box->upper[d] - box->lower[d];
242     if (n[d] == PETSC_DETERMINE) {
243       box->h[d] = h[d];
244       box->n[d] = PetscCeilReal(box->extent[d]/h[d]);
245     } else {
246       box->n[d] = n[d];
247       box->h[d] = box->extent[d]/n[d];
248     }
249   }
250   PetscFunctionReturn(0);
251 }
252 
253 /*
254   PetscGridHashGetEnclosingBox - Find the grid boxes containing each input point
255 
256   Not collective
257 
258   Input Parameters:
259 + box       - The grid hash object
260 . numPoints - The number of input points
261 - points    - The input point coordinates
262 
263   Output Parameters:
264 + dboxes    - An array of numPoints*dim integers expressing the enclosing box as (i_0, i_1, ..., i_dim)
265 - boxes     - An array of numPoints integers expressing the enclosing box as single number, or NULL
266 
267   Level: developer
268 
269 .seealso: PetscGridHashCreate()
270 */
271 PetscErrorCode PetscGridHashGetEnclosingBox(PetscGridHash box, PetscInt numPoints, const PetscScalar points[], PetscInt dboxes[], PetscInt boxes[])
272 {
273   const PetscReal *lower = box->lower;
274   const PetscReal *upper = box->upper;
275   const PetscReal *h     = box->h;
276   const PetscInt  *n     = box->n;
277   const PetscInt   dim   = box->dim;
278   PetscInt         d, p;
279 
280   PetscFunctionBegin;
281   for (p = 0; p < numPoints; ++p) {
282     for (d = 0; d < dim; ++d) {
283       PetscInt dbox = PetscFloorReal((PetscRealPart(points[p*dim+d]) - lower[d])/h[d]);
284 
285       if (dbox == n[d] && PetscAbsReal(PetscRealPart(points[p*dim+d]) - upper[d]) < 1.0e-9) dbox = n[d]-1;
286       if (dbox == -1   && PetscAbsReal(PetscRealPart(points[p*dim+d]) - lower[d]) < 1.0e-9) dbox = 0;
287       if (dbox < 0 || dbox >= n[d]) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Input point %d (%g, %g, %g) is outside of our bounding box",
288                                              p, PetscRealPart(points[p*dim+0]), dim > 1 ? PetscRealPart(points[p*dim+1]) : 0.0, dim > 2 ? PetscRealPart(points[p*dim+2]) : 0.0);
289       dboxes[p*dim+d] = dbox;
290     }
291     if (boxes) for (d = 1, boxes[p] = dboxes[p*dim]; d < dim; ++d) boxes[p] += dboxes[p*dim+d]*n[d-1];
292   }
293   PetscFunctionReturn(0);
294 }
295 
296 /*
297  PetscGridHashGetEnclosingBoxQuery - Find the grid boxes containing each input point
298 
299  Not collective
300 
301   Input Parameters:
302 + box       - The grid hash object
303 . numPoints - The number of input points
304 - points    - The input point coordinates
305 
306   Output Parameters:
307 + dboxes    - An array of numPoints*dim integers expressing the enclosing box as (i_0, i_1, ..., i_dim)
308 . boxes     - An array of numPoints integers expressing the enclosing box as single number, or NULL
309 - found     - Flag indicating if point was located within a box
310 
311   Level: developer
312 
313 .seealso: PetscGridHashGetEnclosingBox()
314 */
315 PetscErrorCode PetscGridHashGetEnclosingBoxQuery(PetscGridHash box, PetscInt numPoints, const PetscScalar points[], PetscInt dboxes[], PetscInt boxes[],PetscBool *found)
316 {
317   const PetscReal *lower = box->lower;
318   const PetscReal *upper = box->upper;
319   const PetscReal *h     = box->h;
320   const PetscInt  *n     = box->n;
321   const PetscInt   dim   = box->dim;
322   PetscInt         d, p;
323 
324   PetscFunctionBegin;
325   *found = PETSC_FALSE;
326   for (p = 0; p < numPoints; ++p) {
327     for (d = 0; d < dim; ++d) {
328       PetscInt dbox = PetscFloorReal((PetscRealPart(points[p*dim+d]) - lower[d])/h[d]);
329 
330       if (dbox == n[d] && PetscAbsReal(PetscRealPart(points[p*dim+d]) - upper[d]) < 1.0e-9) dbox = n[d]-1;
331       if (dbox < 0 || dbox >= n[d]) {
332         PetscFunctionReturn(0);
333       }
334       dboxes[p*dim+d] = dbox;
335     }
336     if (boxes) for (d = 1, boxes[p] = dboxes[p*dim]; d < dim; ++d) boxes[p] += dboxes[p*dim+d]*n[d-1];
337   }
338   *found = PETSC_TRUE;
339   PetscFunctionReturn(0);
340 }
341 
342 PetscErrorCode PetscGridHashDestroy(PetscGridHash *box)
343 {
344   PetscErrorCode ierr;
345 
346   PetscFunctionBegin;
347   if (*box) {
348     ierr = PetscSectionDestroy(&(*box)->cellSection);CHKERRQ(ierr);
349     ierr = ISDestroy(&(*box)->cells);CHKERRQ(ierr);
350     ierr = DMLabelDestroy(&(*box)->cellsSparse);CHKERRQ(ierr);
351   }
352   ierr = PetscFree(*box);CHKERRQ(ierr);
353   PetscFunctionReturn(0);
354 }
355 
356 PetscErrorCode DMPlexLocatePoint_Internal(DM dm, PetscInt dim, const PetscScalar point[], PetscInt cellStart, PetscInt *cell)
357 {
358   PetscInt       coneSize;
359   PetscErrorCode ierr;
360 
361   PetscFunctionBegin;
362   switch (dim) {
363   case 2:
364     ierr = DMPlexGetConeSize(dm, cellStart, &coneSize);CHKERRQ(ierr);
365     switch (coneSize) {
366     case 3:
367       ierr = DMPlexLocatePoint_Simplex_2D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr);
368       break;
369     case 4:
370       ierr = DMPlexLocatePoint_General_2D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr);
371       break;
372     default:
373       SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for cell with cone size %D", coneSize);
374     }
375     break;
376   case 3:
377     ierr = DMPlexGetConeSize(dm, cellStart, &coneSize);CHKERRQ(ierr);
378     switch (coneSize) {
379     case 4:
380       ierr = DMPlexLocatePoint_Simplex_3D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr);
381       break;
382     case 6:
383       ierr = DMPlexLocatePoint_General_3D_Internal(dm, point, cellStart, cell);CHKERRQ(ierr);
384       break;
385     default:
386       SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for cell with cone size %D", coneSize);
387     }
388     break;
389   default:
390     SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No point location for mesh dimension %D", dim);
391   }
392   PetscFunctionReturn(0);
393 }
394 
395 /*
396   DMPlexClosestPoint_Internal - Returns the closest point in the cell to the given point
397 */
398 PetscErrorCode DMPlexClosestPoint_Internal(DM dm, PetscInt dim, const PetscScalar point[], PetscInt cell, PetscReal cpoint[])
399 {
400   PetscInt       coneSize;
401   PetscErrorCode ierr;
402 
403   PetscFunctionBegin;
404   switch (dim) {
405   case 2:
406     ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr);
407     switch (coneSize) {
408     case 3:
409       ierr = DMPlexClosestPoint_Simplex_2D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr);
410       break;
411 #if 0
412     case 4:
413       ierr = DMPlexClosestPoint_General_2D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr);
414       break;
415 #endif
416     default:
417       SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for cell with cone size %D", coneSize);
418     }
419     break;
420 #if 0
421   case 3:
422     ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr);
423     switch (coneSize) {
424     case 4:
425       ierr = DMPlexClosestPoint_Simplex_3D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr);
426       break;
427     case 6:
428       ierr = DMPlexClosestPoint_General_3D_Internal(dm, point, cell, cpoint);CHKERRQ(ierr);
429       break;
430     default:
431       SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for cell with cone size %D", coneSize);
432     }
433     break;
434 #endif
435   default:
436     SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "No closest point location for mesh dimension %D", dim);
437   }
438   PetscFunctionReturn(0);
439 }
440 
441 /*
442   DMPlexComputeGridHash_Internal - Create a grid hash structure covering the Plex
443 
444   Collective on DM
445 
446   Input Parameter:
447 . dm - The Plex
448 
449   Output Parameter:
450 . localBox - The grid hash object
451 
452   Level: developer
453 
454 .seealso: PetscGridHashCreate(), PetscGridHashGetEnclosingBox()
455 */
456 PetscErrorCode DMPlexComputeGridHash_Internal(DM dm, PetscGridHash *localBox)
457 {
458   MPI_Comm           comm;
459   PetscGridHash      lbox;
460   Vec                coordinates;
461   PetscSection       coordSection;
462   Vec                coordsLocal;
463   const PetscScalar *coords;
464   PetscInt          *dboxes, *boxes;
465   PetscInt           n[3] = {10, 10, 10};
466   PetscInt           dim, N, cStart, cEnd, cMax, c, i;
467   PetscErrorCode     ierr;
468 
469   PetscFunctionBegin;
470   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
471   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
472   ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr);
473   if (dim != 2) SETERRQ(comm, PETSC_ERR_SUP, "I have only coded this for 2D");
474   ierr = VecGetLocalSize(coordinates, &N);CHKERRQ(ierr);
475   ierr = VecGetArrayRead(coordinates, &coords);CHKERRQ(ierr);
476   ierr = PetscGridHashCreate(comm, dim, coords, &lbox);CHKERRQ(ierr);
477   for (i = 0; i < N; i += dim) {ierr = PetscGridHashEnlarge(lbox, &coords[i]);CHKERRQ(ierr);}
478   ierr = VecRestoreArrayRead(coordinates, &coords);CHKERRQ(ierr);
479   ierr = PetscOptionsGetInt(NULL,NULL,"-dm_plex_hash_box_nijk",&n[0],NULL);CHKERRQ(ierr);
480   n[1] = n[0];
481   n[2] = n[0];
482   ierr = PetscGridHashSetGrid(lbox, n, NULL);CHKERRQ(ierr);
483 #if 0
484   /* Could define a custom reduction to merge these */
485   ierr = MPIU_Allreduce(lbox->lower, gbox->lower, 3, MPIU_REAL, MPI_MIN, comm);CHKERRQ(ierr);
486   ierr = MPIU_Allreduce(lbox->upper, gbox->upper, 3, MPIU_REAL, MPI_MAX, comm);CHKERRQ(ierr);
487 #endif
488   /* Is there a reason to snap the local bounding box to a division of the global box? */
489   /* Should we compute all overlaps of local boxes? We could do this with a rendevouz scheme partitioning the global box */
490   /* Create label */
491   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
492   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
493   if (cMax >= 0) cEnd = PetscMin(cEnd, cMax);
494   ierr = DMLabelCreate("cells", &lbox->cellsSparse);CHKERRQ(ierr);
495   ierr = DMLabelCreateIndex(lbox->cellsSparse, cStart, cEnd);CHKERRQ(ierr);
496   /* Compute boxes which overlap each cell: http://stackoverflow.com/questions/13790208/triangle-square-intersection-test-in-2d */
497   ierr = DMGetCoordinatesLocal(dm, &coordsLocal);CHKERRQ(ierr);
498   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
499   ierr = PetscCalloc2(16 * dim, &dboxes, 16, &boxes);CHKERRQ(ierr);
500   for (c = cStart; c < cEnd; ++c) {
501     const PetscReal *h       = lbox->h;
502     PetscScalar     *ccoords = NULL;
503     PetscInt         csize   = 0;
504     PetscScalar      point[3];
505     PetscInt         dlim[6], d, e, i, j, k;
506 
507     /* Find boxes enclosing each vertex */
508     ierr = DMPlexVecGetClosure(dm, coordSection, coordsLocal, c, &csize, &ccoords);CHKERRQ(ierr);
509     ierr = PetscGridHashGetEnclosingBox(lbox, csize/dim, ccoords, dboxes, boxes);CHKERRQ(ierr);
510     /* Mark cells containing the vertices */
511     for (e = 0; e < csize/dim; ++e) {ierr = DMLabelSetValue(lbox->cellsSparse, c, boxes[e]);CHKERRQ(ierr);}
512     /* Get grid of boxes containing these */
513     for (d = 0;   d < dim; ++d) {dlim[d*2+0] = dlim[d*2+1] = dboxes[d];}
514     for (d = dim; d < 3;   ++d) {dlim[d*2+0] = dlim[d*2+1] = 0;}
515     for (e = 1; e < dim+1; ++e) {
516       for (d = 0; d < dim; ++d) {
517         dlim[d*2+0] = PetscMin(dlim[d*2+0], dboxes[e*dim+d]);
518         dlim[d*2+1] = PetscMax(dlim[d*2+1], dboxes[e*dim+d]);
519       }
520     }
521     /* Check for intersection of box with cell */
522     for (k = dlim[2*2+0], point[2] = lbox->lower[2] + k*h[2]; k <= dlim[2*2+1]; ++k, point[2] += h[2]) {
523       for (j = dlim[1*2+0], point[1] = lbox->lower[1] + j*h[1]; j <= dlim[1*2+1]; ++j, point[1] += h[1]) {
524         for (i = dlim[0*2+0], point[0] = lbox->lower[0] + i*h[0]; i <= dlim[0*2+1]; ++i, point[0] += h[0]) {
525           const PetscInt box = (k*lbox->n[1] + j)*lbox->n[0] + i;
526           PetscScalar    cpoint[3];
527           PetscInt       cell, edge, ii, jj, kk;
528 
529           /* Check whether cell contains any vertex of these subboxes TODO vectorize this */
530           for (kk = 0, cpoint[2] = point[2]; kk < (dim > 2 ? 2 : 1); ++kk, cpoint[2] += h[2]) {
531             for (jj = 0, cpoint[1] = point[1]; jj < (dim > 1 ? 2 : 1); ++jj, cpoint[1] += h[1]) {
532               for (ii = 0, cpoint[0] = point[0]; ii < 2; ++ii, cpoint[0] += h[0]) {
533 
534                 ierr = DMPlexLocatePoint_Internal(dm, dim, cpoint, c, &cell);CHKERRQ(ierr);
535                 if (cell >= 0) {DMLabelSetValue(lbox->cellsSparse, c, box);CHKERRQ(ierr); ii = jj = kk = 2;}
536               }
537             }
538           }
539           /* Check whether cell edge intersects any edge of these subboxes TODO vectorize this */
540           for (edge = 0; edge < dim+1; ++edge) {
541             PetscReal segA[6], segB[6];
542 
543             if (PetscUnlikely(dim > 3)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected dim %d > 3",dim);
544             for (d = 0; d < dim; ++d) {segA[d] = PetscRealPart(ccoords[edge*dim+d]); segA[dim+d] = PetscRealPart(ccoords[((edge+1)%(dim+1))*dim+d]);}
545             for (kk = 0; kk < (dim > 2 ? 2 : 1); ++kk) {
546               if (dim > 2) {segB[2]     = PetscRealPart(point[2]);
547                             segB[dim+2] = PetscRealPart(point[2]) + kk*h[2];}
548               for (jj = 0; jj < (dim > 1 ? 2 : 1); ++jj) {
549                 if (dim > 1) {segB[1]     = PetscRealPart(point[1]);
550                               segB[dim+1] = PetscRealPart(point[1]) + jj*h[1];}
551                 for (ii = 0; ii < 2; ++ii) {
552                   PetscBool intersects;
553 
554                   segB[0]     = PetscRealPart(point[0]);
555                   segB[dim+0] = PetscRealPart(point[0]) + ii*h[0];
556                   ierr = DMPlexGetLineIntersection_2D_Internal(segA, segB, NULL, &intersects);CHKERRQ(ierr);
557                   if (intersects) {DMLabelSetValue(lbox->cellsSparse, c, box);CHKERRQ(ierr); edge = ii = jj = kk = dim+1;}
558                 }
559               }
560             }
561           }
562         }
563       }
564     }
565     ierr = DMPlexVecRestoreClosure(dm, coordSection, coordsLocal, c, NULL, &ccoords);CHKERRQ(ierr);
566   }
567   ierr = PetscFree2(dboxes, boxes);CHKERRQ(ierr);
568   ierr = DMLabelConvertToSection(lbox->cellsSparse, &lbox->cellSection, &lbox->cells);CHKERRQ(ierr);
569   ierr = DMLabelDestroy(&lbox->cellsSparse);CHKERRQ(ierr);
570   *localBox = lbox;
571   PetscFunctionReturn(0);
572 }
573 
574 PetscErrorCode DMLocatePoints_Plex(DM dm, Vec v, DMPointLocationType ltype, PetscSF cellSF)
575 {
576   DM_Plex        *mesh = (DM_Plex *) dm->data;
577   PetscBool       hash = mesh->useHashLocation, reuse = PETSC_FALSE;
578   PetscInt        bs, numPoints, p, numFound, *found = NULL;
579   PetscInt        dim, cStart, cEnd, cMax, numCells, c, d;
580   const PetscInt *boxCells;
581   PetscSFNode    *cells;
582   PetscScalar    *a;
583   PetscMPIInt     result;
584   PetscLogDouble  t0,t1;
585   PetscReal       gmin[3],gmax[3];
586   PetscInt        terminating_query_type[] = { 0, 0, 0 };
587   PetscErrorCode  ierr;
588 
589   PetscFunctionBegin;
590   ierr = PetscTime(&t0);CHKERRQ(ierr);
591   if (ltype == DM_POINTLOCATION_NEAREST && !hash) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_SUP, "Nearest point location only supported with grid hashing. Use -dm_plex_hash_location to enable it.");
592   ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr);
593   ierr = VecGetBlockSize(v, &bs);CHKERRQ(ierr);
594   ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)cellSF),PETSC_COMM_SELF,&result);CHKERRQ(ierr);
595   if (result != MPI_IDENT && result != MPI_CONGRUENT) SETERRQ(PetscObjectComm((PetscObject)cellSF),PETSC_ERR_SUP, "Trying parallel point location: only local point location supported");
596   if (bs != dim) SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Block size for point vector %D must be the mesh coordinate dimension %D", bs, dim);
597   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
598   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
599   if (cMax >= 0) cEnd = PetscMin(cEnd, cMax);
600   ierr = VecGetLocalSize(v, &numPoints);CHKERRQ(ierr);
601   ierr = VecGetArray(v, &a);CHKERRQ(ierr);
602   numPoints /= bs;
603   {
604     const PetscSFNode *sf_cells;
605 
606     ierr = PetscSFGetGraph(cellSF,NULL,NULL,NULL,&sf_cells);CHKERRQ(ierr);
607     if (sf_cells) {
608       ierr = PetscInfo(dm,"[DMLocatePoints_Plex] Re-using existing StarForest node list\n");CHKERRQ(ierr);
609       cells = (PetscSFNode*)sf_cells;
610       reuse = PETSC_TRUE;
611     } else {
612       ierr = PetscInfo(dm,"[DMLocatePoints_Plex] Creating and initializing new StarForest node list\n");CHKERRQ(ierr);
613       ierr = PetscMalloc1(numPoints, &cells);CHKERRQ(ierr);
614       /* initialize cells if created */
615       for (p=0; p<numPoints; p++) {
616         cells[p].rank  = 0;
617         cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND;
618       }
619     }
620   }
621   /* define domain bounding box */
622   {
623     Vec coorglobal;
624 
625     ierr = DMGetCoordinates(dm,&coorglobal);CHKERRQ(ierr);
626     ierr = VecStrideMaxAll(coorglobal,NULL,gmax);CHKERRQ(ierr);
627     ierr = VecStrideMinAll(coorglobal,NULL,gmin);CHKERRQ(ierr);
628   }
629   if (hash) {
630     if (!mesh->lbox) {ierr = PetscInfo(dm, "Initializing grid hashing");CHKERRQ(ierr);ierr = DMPlexComputeGridHash_Internal(dm, &mesh->lbox);CHKERRQ(ierr);}
631     /* Designate the local box for each point */
632     /* Send points to correct process */
633     /* Search cells that lie in each subbox */
634     /*   Should we bin points before doing search? */
635     ierr = ISGetIndices(mesh->lbox->cells, &boxCells);CHKERRQ(ierr);
636   }
637   for (p = 0, numFound = 0; p < numPoints; ++p) {
638     const PetscScalar *point = &a[p*bs];
639     PetscInt           dbin[3] = {-1,-1,-1}, bin, cell = -1, cellOffset;
640     PetscBool          point_outside_domain = PETSC_FALSE;
641 
642     /* check bounding box of domain */
643     for (d=0; d<dim; d++) {
644       if (PetscRealPart(point[d]) < gmin[d]) { point_outside_domain = PETSC_TRUE; break; }
645       if (PetscRealPart(point[d]) > gmax[d]) { point_outside_domain = PETSC_TRUE; break; }
646     }
647     if (point_outside_domain) {
648       cells[p].rank = 0;
649       cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND;
650       terminating_query_type[0]++;
651       continue;
652     }
653 
654     /* check initial values in cells[].index - abort early if found */
655     if (cells[p].index != DMLOCATEPOINT_POINT_NOT_FOUND) {
656       c = cells[p].index;
657       cells[p].index = DMLOCATEPOINT_POINT_NOT_FOUND;
658       ierr = DMPlexLocatePoint_Internal(dm, dim, point, c, &cell);CHKERRQ(ierr);
659       if (cell >= 0) {
660         cells[p].rank = 0;
661         cells[p].index = cell;
662         numFound++;
663       }
664     }
665     if (cells[p].index != DMLOCATEPOINT_POINT_NOT_FOUND) {
666       terminating_query_type[1]++;
667       continue;
668     }
669 
670     if (hash) {
671       PetscBool found_box;
672 
673       /* allow for case that point is outside box - abort early */
674       ierr = PetscGridHashGetEnclosingBoxQuery(mesh->lbox, 1, point, dbin, &bin,&found_box);CHKERRQ(ierr);
675       if (found_box) {
676         /* TODO Lay an interface over this so we can switch between Section (dense) and Label (sparse) */
677         ierr = PetscSectionGetDof(mesh->lbox->cellSection, bin, &numCells);CHKERRQ(ierr);
678         ierr = PetscSectionGetOffset(mesh->lbox->cellSection, bin, &cellOffset);CHKERRQ(ierr);
679         for (c = cellOffset; c < cellOffset + numCells; ++c) {
680           ierr = DMPlexLocatePoint_Internal(dm, dim, point, boxCells[c], &cell);CHKERRQ(ierr);
681           if (cell >= 0) {
682             cells[p].rank = 0;
683             cells[p].index = cell;
684             numFound++;
685             terminating_query_type[2]++;
686             break;
687           }
688         }
689       }
690     } else {
691       for (c = cStart; c < cEnd; ++c) {
692         ierr = DMPlexLocatePoint_Internal(dm, dim, point, c, &cell);CHKERRQ(ierr);
693         if (cell >= 0) {
694           cells[p].rank = 0;
695           cells[p].index = cell;
696           numFound++;
697           terminating_query_type[2]++;
698           break;
699         }
700       }
701     }
702   }
703   if (hash) {ierr = ISRestoreIndices(mesh->lbox->cells, &boxCells);CHKERRQ(ierr);}
704   if (ltype == DM_POINTLOCATION_NEAREST && hash && numFound < numPoints) {
705     for (p = 0; p < numPoints; p++) {
706       const PetscScalar *point = &a[p*bs];
707       PetscReal          cpoint[3], diff[3], dist, distMax = PETSC_MAX_REAL;
708       PetscInt           dbin[3] = {-1,-1,-1}, bin, cellOffset, d;
709 
710       if (cells[p].index < 0) {
711         ++numFound;
712         ierr = PetscGridHashGetEnclosingBox(mesh->lbox, 1, point, dbin, &bin);CHKERRQ(ierr);
713         ierr = PetscSectionGetDof(mesh->lbox->cellSection, bin, &numCells);CHKERRQ(ierr);
714         ierr = PetscSectionGetOffset(mesh->lbox->cellSection, bin, &cellOffset);CHKERRQ(ierr);
715         for (c = cellOffset; c < cellOffset + numCells; ++c) {
716           ierr = DMPlexClosestPoint_Internal(dm, dim, point, boxCells[c], cpoint);CHKERRQ(ierr);
717           for (d = 0; d < dim; ++d) diff[d] = cpoint[d] - PetscRealPart(point[d]);
718           dist = DMPlex_NormD_Internal(dim, diff);
719           if (dist < distMax) {
720             for (d = 0; d < dim; ++d) a[p*bs+d] = cpoint[d];
721             cells[p].rank  = 0;
722             cells[p].index = boxCells[c];
723             distMax = dist;
724             break;
725           }
726         }
727       }
728     }
729   }
730   /* This code is only be relevant when interfaced to parallel point location */
731   /* Check for highest numbered proc that claims a point (do we care?) */
732   if (ltype == DM_POINTLOCATION_REMOVE && numFound < numPoints) {
733     ierr = PetscMalloc1(numFound,&found);CHKERRQ(ierr);
734     for (p = 0, numFound = 0; p < numPoints; p++) {
735       if (cells[p].rank >= 0 && cells[p].index >= 0) {
736         if (numFound < p) {
737           cells[numFound] = cells[p];
738         }
739         found[numFound++] = p;
740       }
741     }
742   }
743   ierr = VecRestoreArray(v, &a);CHKERRQ(ierr);
744   if (!reuse) {
745     ierr = PetscSFSetGraph(cellSF, cEnd - cStart, numFound, found, PETSC_OWN_POINTER, cells, PETSC_OWN_POINTER);CHKERRQ(ierr);
746   }
747   ierr = PetscTime(&t1);CHKERRQ(ierr);
748   if (hash) {
749     ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] terminating_query_type : %D [outside domain] : %D [inside intial cell] : %D [hash]\n",terminating_query_type[0],terminating_query_type[1],terminating_query_type[2]);CHKERRQ(ierr);
750   } else {
751     ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] terminating_query_type : %D [outside domain] : %D [inside intial cell] : %D [brute-force]\n",terminating_query_type[0],terminating_query_type[1],terminating_query_type[2]);CHKERRQ(ierr);
752   }
753   ierr = PetscInfo3(dm,"[DMLocatePoints_Plex] npoints %D : time(rank0) %1.2e (sec): points/sec %1.4e\n",numPoints,t1-t0,(double)((double)numPoints/(t1-t0)));CHKERRQ(ierr);
754   PetscFunctionReturn(0);
755 }
756 
757 /*@C
758   DMPlexComputeProjection2Dto1D - Rewrite coordinates to be the 1D projection of the 2D coordinates
759 
760   Not collective
761 
762   Input Parameter:
763 . coords - The coordinates of a segment
764 
765   Output Parameters:
766 + coords - The new y-coordinate, and 0 for x
767 - R - The rotation which accomplishes the projection
768 
769   Level: developer
770 
771 .seealso: DMPlexComputeProjection3Dto1D(), DMPlexComputeProjection3Dto2D()
772 @*/
773 PetscErrorCode DMPlexComputeProjection2Dto1D(PetscScalar coords[], PetscReal R[])
774 {
775   const PetscReal x = PetscRealPart(coords[2] - coords[0]);
776   const PetscReal y = PetscRealPart(coords[3] - coords[1]);
777   const PetscReal r = PetscSqrtReal(x*x + y*y), c = x/r, s = y/r;
778 
779   PetscFunctionBegin;
780   R[0] = c; R[1] = -s;
781   R[2] = s; R[3] =  c;
782   coords[0] = 0.0;
783   coords[1] = r;
784   PetscFunctionReturn(0);
785 }
786 
787 /*@C
788   DMPlexComputeProjection3Dto1D - Rewrite coordinates to be the 1D projection of the 3D coordinates
789 
790   Not collective
791 
792   Input Parameter:
793 . coords - The coordinates of a segment
794 
795   Output Parameters:
796 + coords - The new y-coordinate, and 0 for x and z
797 - R - The rotation which accomplishes the projection
798 
799   Note: This uses the basis completion described by Frisvad in http://www.imm.dtu.dk/~jerf/papers/abstracts/onb.html, DOI:10.1080/2165347X.2012.689606
800 
801   Level: developer
802 
803 .seealso: DMPlexComputeProjection2Dto1D(), DMPlexComputeProjection3Dto2D()
804 @*/
805 PetscErrorCode DMPlexComputeProjection3Dto1D(PetscScalar coords[], PetscReal R[])
806 {
807   PetscReal      x    = PetscRealPart(coords[3] - coords[0]);
808   PetscReal      y    = PetscRealPart(coords[4] - coords[1]);
809   PetscReal      z    = PetscRealPart(coords[5] - coords[2]);
810   PetscReal      r    = PetscSqrtReal(x*x + y*y + z*z);
811   PetscReal      rinv = 1. / r;
812   PetscFunctionBegin;
813 
814   x *= rinv; y *= rinv; z *= rinv;
815   if (x > 0.) {
816     PetscReal inv1pX   = 1./ (1. + x);
817 
818     R[0] = x; R[1] = -y;              R[2] = -z;
819     R[3] = y; R[4] = 1. - y*y*inv1pX; R[5] =     -y*z*inv1pX;
820     R[6] = z; R[7] =     -y*z*inv1pX; R[8] = 1. - z*z*inv1pX;
821   }
822   else {
823     PetscReal inv1mX   = 1./ (1. - x);
824 
825     R[0] = x; R[1] = z;               R[2] = y;
826     R[3] = y; R[4] =     -y*z*inv1mX; R[5] = 1. - y*y*inv1mX;
827     R[6] = z; R[7] = 1. - z*z*inv1mX; R[8] =     -y*z*inv1mX;
828   }
829   coords[0] = 0.0;
830   coords[1] = r;
831   PetscFunctionReturn(0);
832 }
833 
834 /*@
835   DMPlexComputeProjection3Dto2D - Rewrite coordinates to be the 2D projection of the 3D coordinates
836 
837   Not collective
838 
839   Input Parameter:
840 . coords - The coordinates of a segment
841 
842   Output Parameters:
843 + coords - The new y- and z-coordinates, and 0 for x
844 - R - The rotation which accomplishes the projection
845 
846   Level: developer
847 
848 .seealso: DMPlexComputeProjection2Dto1D(), DMPlexComputeProjection3Dto1D()
849 @*/
850 PetscErrorCode DMPlexComputeProjection3Dto2D(PetscInt coordSize, PetscScalar coords[], PetscReal R[])
851 {
852   PetscReal      x1[3],  x2[3], n[3], norm;
853   PetscReal      x1p[3], x2p[3], xnp[3];
854   PetscReal      sqrtz, alpha;
855   const PetscInt dim = 3;
856   PetscInt       d, e, p;
857 
858   PetscFunctionBegin;
859   /* 0) Calculate normal vector */
860   for (d = 0; d < dim; ++d) {
861     x1[d] = PetscRealPart(coords[1*dim+d] - coords[0*dim+d]);
862     x2[d] = PetscRealPart(coords[2*dim+d] - coords[0*dim+d]);
863   }
864   n[0] = x1[1]*x2[2] - x1[2]*x2[1];
865   n[1] = x1[2]*x2[0] - x1[0]*x2[2];
866   n[2] = x1[0]*x2[1] - x1[1]*x2[0];
867   norm = PetscSqrtReal(n[0]*n[0] + n[1]*n[1] + n[2]*n[2]);
868   n[0] /= norm;
869   n[1] /= norm;
870   n[2] /= norm;
871   /* 1) Take the normal vector and rotate until it is \hat z
872 
873     Let the normal vector be <nx, ny, nz> and alpha = 1/sqrt(1 - nz^2), then
874 
875     R = /  alpha nx nz  alpha ny nz -1/alpha \
876         | -alpha ny     alpha nx        0    |
877         \     nx            ny         nz    /
878 
879     will rotate the normal vector to \hat z
880   */
881   sqrtz = PetscSqrtReal(1.0 - n[2]*n[2]);
882   /* Check for n = z */
883   if (sqrtz < 1.0e-10) {
884     const PetscInt s = PetscSign(n[2]);
885     /* If nz < 0, rotate 180 degrees around x-axis */
886     for (p = 3; p < coordSize/3; ++p) {
887       coords[p*2+0] = PetscRealPart(coords[p*dim+0] - coords[0*dim+0]);
888       coords[p*2+1] = (PetscRealPart(coords[p*dim+1] - coords[0*dim+1])) * s;
889     }
890     coords[0] = 0.0;
891     coords[1] = 0.0;
892     coords[2] = x1[0];
893     coords[3] = x1[1] * s;
894     coords[4] = x2[0];
895     coords[5] = x2[1] * s;
896     R[0] = 1.0;     R[1] = 0.0;     R[2] = 0.0;
897     R[3] = 0.0;     R[4] = 1.0 * s; R[5] = 0.0;
898     R[6] = 0.0;     R[7] = 0.0;     R[8] = 1.0 * s;
899     PetscFunctionReturn(0);
900   }
901   alpha = 1.0/sqrtz;
902   R[0] =  alpha*n[0]*n[2]; R[1] = alpha*n[1]*n[2]; R[2] = -sqrtz;
903   R[3] = -alpha*n[1];      R[4] = alpha*n[0];      R[5] = 0.0;
904   R[6] =  n[0];            R[7] = n[1];            R[8] = n[2];
905   for (d = 0; d < dim; ++d) {
906     x1p[d] = 0.0;
907     x2p[d] = 0.0;
908     for (e = 0; e < dim; ++e) {
909       x1p[d] += R[d*dim+e]*x1[e];
910       x2p[d] += R[d*dim+e]*x2[e];
911     }
912   }
913   if (PetscAbsReal(x1p[2]) > 10. * PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid rotation calculated");
914   if (PetscAbsReal(x2p[2]) > 10. * PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid rotation calculated");
915   /* 2) Project to (x, y) */
916   for (p = 3; p < coordSize/3; ++p) {
917     for (d = 0; d < dim; ++d) {
918       xnp[d] = 0.0;
919       for (e = 0; e < dim; ++e) {
920         xnp[d] += R[d*dim+e]*PetscRealPart(coords[p*dim+e] - coords[0*dim+e]);
921       }
922       if (d < dim-1) coords[p*2+d] = xnp[d];
923     }
924   }
925   coords[0] = 0.0;
926   coords[1] = 0.0;
927   coords[2] = x1p[0];
928   coords[3] = x1p[1];
929   coords[4] = x2p[0];
930   coords[5] = x2p[1];
931   /* Output R^T which rotates \hat z to the input normal */
932   for (d = 0; d < dim; ++d) {
933     for (e = d+1; e < dim; ++e) {
934       PetscReal tmp;
935 
936       tmp        = R[d*dim+e];
937       R[d*dim+e] = R[e*dim+d];
938       R[e*dim+d] = tmp;
939     }
940   }
941   PetscFunctionReturn(0);
942 }
943 
944 PETSC_UNUSED
945 PETSC_STATIC_INLINE void Volume_Triangle_Internal(PetscReal *vol, PetscReal coords[])
946 {
947   /* Signed volume is 1/2 the determinant
948 
949    |  1  1  1 |
950    | x0 x1 x2 |
951    | y0 y1 y2 |
952 
953      but if x0,y0 is the origin, we have
954 
955    | x1 x2 |
956    | y1 y2 |
957   */
958   const PetscReal x1 = coords[2] - coords[0], y1 = coords[3] - coords[1];
959   const PetscReal x2 = coords[4] - coords[0], y2 = coords[5] - coords[1];
960   PetscReal       M[4], detM;
961   M[0] = x1; M[1] = x2;
962   M[2] = y1; M[3] = y2;
963   DMPlex_Det2D_Internal(&detM, M);
964   *vol = 0.5*detM;
965   (void)PetscLogFlops(5.0);
966 }
967 
968 PETSC_STATIC_INLINE void Volume_Triangle_Origin_Internal(PetscReal *vol, PetscReal coords[])
969 {
970   DMPlex_Det2D_Internal(vol, coords);
971   *vol *= 0.5;
972 }
973 
974 PETSC_UNUSED
975 PETSC_STATIC_INLINE void Volume_Tetrahedron_Internal(PetscReal *vol, PetscReal coords[])
976 {
977   /* Signed volume is 1/6th of the determinant
978 
979    |  1  1  1  1 |
980    | x0 x1 x2 x3 |
981    | y0 y1 y2 y3 |
982    | z0 z1 z2 z3 |
983 
984      but if x0,y0,z0 is the origin, we have
985 
986    | x1 x2 x3 |
987    | y1 y2 y3 |
988    | z1 z2 z3 |
989   */
990   const PetscReal x1 = coords[3] - coords[0], y1 = coords[4]  - coords[1], z1 = coords[5]  - coords[2];
991   const PetscReal x2 = coords[6] - coords[0], y2 = coords[7]  - coords[1], z2 = coords[8]  - coords[2];
992   const PetscReal x3 = coords[9] - coords[0], y3 = coords[10] - coords[1], z3 = coords[11] - coords[2];
993   const PetscReal onesixth = ((PetscReal)1./(PetscReal)6.);
994   PetscReal       M[9], detM;
995   M[0] = x1; M[1] = x2; M[2] = x3;
996   M[3] = y1; M[4] = y2; M[5] = y3;
997   M[6] = z1; M[7] = z2; M[8] = z3;
998   DMPlex_Det3D_Internal(&detM, M);
999   *vol = -onesixth*detM;
1000   (void)PetscLogFlops(10.0);
1001 }
1002 
1003 PETSC_STATIC_INLINE void Volume_Tetrahedron_Origin_Internal(PetscReal *vol, PetscReal coords[])
1004 {
1005   const PetscReal onesixth = ((PetscReal)1./(PetscReal)6.);
1006   DMPlex_Det3D_Internal(vol, coords);
1007   *vol *= -onesixth;
1008 }
1009 
1010 static PetscErrorCode DMPlexComputePointGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1011 {
1012   PetscSection   coordSection;
1013   Vec            coordinates;
1014   const PetscScalar *coords;
1015   PetscInt       dim, d, off;
1016   PetscErrorCode ierr;
1017 
1018   PetscFunctionBegin;
1019   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1020   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1021   ierr = PetscSectionGetDof(coordSection,e,&dim);CHKERRQ(ierr);
1022   if (!dim) PetscFunctionReturn(0);
1023   ierr = PetscSectionGetOffset(coordSection,e,&off);CHKERRQ(ierr);
1024   ierr = VecGetArrayRead(coordinates,&coords);CHKERRQ(ierr);
1025   if (v0) {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[off + d]);}
1026   ierr = VecRestoreArrayRead(coordinates,&coords);CHKERRQ(ierr);
1027   *detJ = 1.;
1028   if (J) {
1029     for (d = 0; d < dim * dim; d++) J[d] = 0.;
1030     for (d = 0; d < dim; d++) J[d * dim + d] = 1.;
1031     if (invJ) {
1032       for (d = 0; d < dim * dim; d++) invJ[d] = 0.;
1033       for (d = 0; d < dim; d++) invJ[d * dim + d] = 1.;
1034     }
1035   }
1036   PetscFunctionReturn(0);
1037 }
1038 
1039 static PetscErrorCode DMPlexComputeLineGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1040 {
1041   PetscSection   coordSection;
1042   Vec            coordinates;
1043   PetscScalar   *coords = NULL;
1044   PetscInt       numCoords, d, pStart, pEnd, numSelfCoords = 0;
1045   PetscErrorCode ierr;
1046 
1047   PetscFunctionBegin;
1048   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1049   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1050   ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr);
1051   if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);}
1052   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1053   numCoords = numSelfCoords ? numSelfCoords : numCoords;
1054   if (invJ && !J) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "In order to compute invJ, J must not be NULL");
1055   *detJ = 0.0;
1056   if (numCoords == 6) {
1057     const PetscInt dim = 3;
1058     PetscReal      R[9], J0;
1059 
1060     if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1061     ierr = DMPlexComputeProjection3Dto1D(coords, R);CHKERRQ(ierr);
1062     if (J)    {
1063       J0   = 0.5*PetscRealPart(coords[1]);
1064       J[0] = R[0]*J0; J[1] = R[1]; J[2] = R[2];
1065       J[3] = R[3]*J0; J[4] = R[4]; J[5] = R[5];
1066       J[6] = R[6]*J0; J[7] = R[7]; J[8] = R[8];
1067       DMPlex_Det3D_Internal(detJ, J);
1068       if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);}
1069     }
1070   } else if (numCoords == 4) {
1071     const PetscInt dim = 2;
1072     PetscReal      R[4], J0;
1073 
1074     if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1075     ierr = DMPlexComputeProjection2Dto1D(coords, R);CHKERRQ(ierr);
1076     if (J)    {
1077       J0   = 0.5*PetscRealPart(coords[1]);
1078       J[0] = R[0]*J0; J[1] = R[1];
1079       J[2] = R[2]*J0; J[3] = R[3];
1080       DMPlex_Det2D_Internal(detJ, J);
1081       if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);}
1082     }
1083   } else if (numCoords == 2) {
1084     const PetscInt dim = 1;
1085 
1086     if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1087     if (J)    {
1088       J[0]  = 0.5*(PetscRealPart(coords[1]) - PetscRealPart(coords[0]));
1089       *detJ = J[0];
1090       ierr = PetscLogFlops(2.0);CHKERRQ(ierr);
1091       if (invJ) {invJ[0] = 1.0/J[0]; ierr = PetscLogFlops(1.0);CHKERRQ(ierr);}
1092     }
1093   } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this segment is %D != 2", numCoords);
1094   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1095   PetscFunctionReturn(0);
1096 }
1097 
1098 static PetscErrorCode DMPlexComputeTriangleGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1099 {
1100   PetscSection   coordSection;
1101   Vec            coordinates;
1102   PetscScalar   *coords = NULL;
1103   PetscInt       numCoords, numSelfCoords = 0, d, f, g, pStart, pEnd;
1104   PetscErrorCode ierr;
1105 
1106   PetscFunctionBegin;
1107   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1108   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1109   ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr);
1110   if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);}
1111   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1112   numCoords = numSelfCoords ? numSelfCoords : numCoords;
1113   *detJ = 0.0;
1114   if (numCoords == 9) {
1115     const PetscInt dim = 3;
1116     PetscReal      R[9], J0[9] = {1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0};
1117 
1118     if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1119     ierr = DMPlexComputeProjection3Dto2D(numCoords, coords, R);CHKERRQ(ierr);
1120     if (J)    {
1121       const PetscInt pdim = 2;
1122 
1123       for (d = 0; d < pdim; d++) {
1124         for (f = 0; f < pdim; f++) {
1125           J0[d*dim+f] = 0.5*(PetscRealPart(coords[(f+1)*pdim+d]) - PetscRealPart(coords[0*pdim+d]));
1126         }
1127       }
1128       ierr = PetscLogFlops(8.0);CHKERRQ(ierr);
1129       DMPlex_Det3D_Internal(detJ, J0);
1130       for (d = 0; d < dim; d++) {
1131         for (f = 0; f < dim; f++) {
1132           J[d*dim+f] = 0.0;
1133           for (g = 0; g < dim; g++) {
1134             J[d*dim+f] += R[d*dim+g]*J0[g*dim+f];
1135           }
1136         }
1137       }
1138       ierr = PetscLogFlops(18.0);CHKERRQ(ierr);
1139     }
1140     if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);}
1141   } else if (numCoords == 6) {
1142     const PetscInt dim = 2;
1143 
1144     if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1145     if (J)    {
1146       for (d = 0; d < dim; d++) {
1147         for (f = 0; f < dim; f++) {
1148           J[d*dim+f] = 0.5*(PetscRealPart(coords[(f+1)*dim+d]) - PetscRealPart(coords[0*dim+d]));
1149         }
1150       }
1151       ierr = PetscLogFlops(8.0);CHKERRQ(ierr);
1152       DMPlex_Det2D_Internal(detJ, J);
1153     }
1154     if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);}
1155   } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this triangle is %D != 6 or 9", numCoords);
1156   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1157   PetscFunctionReturn(0);
1158 }
1159 
1160 static PetscErrorCode DMPlexComputeRectangleGeometry_Internal(DM dm, PetscInt e, PetscInt Nq, const PetscReal points[], PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1161 {
1162   PetscSection   coordSection;
1163   Vec            coordinates;
1164   PetscScalar   *coords = NULL;
1165   PetscInt       numCoords, numSelfCoords = 0, d, f, g, pStart, pEnd;
1166   PetscErrorCode ierr;
1167 
1168   PetscFunctionBegin;
1169   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1170   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1171   ierr = PetscSectionGetChart(coordSection,&pStart,&pEnd);CHKERRQ(ierr);
1172   if (e >= pStart && e < pEnd) {ierr = PetscSectionGetDof(coordSection,e,&numSelfCoords);CHKERRQ(ierr);}
1173   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1174   numCoords = numSelfCoords ? numSelfCoords : numCoords;
1175   if (!Nq) {
1176     *detJ = 0.0;
1177     if (numCoords == 12) {
1178       const PetscInt dim = 3;
1179       PetscReal      R[9], J0[9] = {1.0,0.0,0.0,0.0,1.0,0.0,0.0,0.0,1.0};
1180 
1181       if (v)   {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);}
1182       ierr = DMPlexComputeProjection3Dto2D(numCoords, coords, R);CHKERRQ(ierr);
1183       if (J)    {
1184         const PetscInt pdim = 2;
1185 
1186         for (d = 0; d < pdim; d++) {
1187           J0[d*dim+0] = 0.5*(PetscRealPart(coords[1*pdim+d]) - PetscRealPart(coords[0*pdim+d]));
1188           J0[d*dim+1] = 0.5*(PetscRealPart(coords[3*pdim+d]) - PetscRealPart(coords[0*pdim+d]));
1189         }
1190         ierr = PetscLogFlops(8.0);CHKERRQ(ierr);
1191         DMPlex_Det3D_Internal(detJ, J0);
1192         for (d = 0; d < dim; d++) {
1193           for (f = 0; f < dim; f++) {
1194             J[d*dim+f] = 0.0;
1195             for (g = 0; g < dim; g++) {
1196               J[d*dim+f] += R[d*dim+g]*J0[g*dim+f];
1197             }
1198           }
1199         }
1200         ierr = PetscLogFlops(18.0);CHKERRQ(ierr);
1201       }
1202       if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);}
1203     } else if (numCoords == 8) {
1204       const PetscInt dim = 2;
1205 
1206       if (v)   {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);}
1207       if (J)    {
1208         for (d = 0; d < dim; d++) {
1209           J[d*dim+0] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d]));
1210           J[d*dim+1] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d]));
1211         }
1212         ierr = PetscLogFlops(8.0);CHKERRQ(ierr);
1213         DMPlex_Det2D_Internal(detJ, J);
1214       }
1215       if (invJ) {DMPlex_Invert2D_Internal(invJ, J, *detJ);}
1216     } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this quadrilateral is %D != 8 or 12", numCoords);
1217   } else {
1218     const PetscInt Nv = 4;
1219     const PetscInt dimR = 2;
1220     const PetscInt zToPlex[4] = {0, 1, 3, 2};
1221     PetscReal zOrder[12];
1222     PetscReal zCoeff[12];
1223     PetscInt  i, j, k, l, dim;
1224 
1225     if (numCoords == 12) {
1226       dim = 3;
1227     } else if (numCoords == 8) {
1228       dim = 2;
1229     } else SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "The number of coordinates for this quadrilateral is %D != 8 or 12", numCoords);
1230     for (i = 0; i < Nv; i++) {
1231       PetscInt zi = zToPlex[i];
1232 
1233       for (j = 0; j < dim; j++) {
1234         zOrder[dim * i + j] = PetscRealPart(coords[dim * zi + j]);
1235       }
1236     }
1237     for (j = 0; j < dim; j++) {
1238       zCoeff[dim * 0 + j] = 0.25 * (  zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j]);
1239       zCoeff[dim * 1 + j] = 0.25 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j]);
1240       zCoeff[dim * 2 + j] = 0.25 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j]);
1241       zCoeff[dim * 3 + j] = 0.25 * (  zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j]);
1242     }
1243     for (i = 0; i < Nq; i++) {
1244       PetscReal xi = points[dimR * i], eta = points[dimR * i + 1];
1245 
1246       if (v) {
1247         PetscReal extPoint[4];
1248 
1249         extPoint[0] = 1.;
1250         extPoint[1] = xi;
1251         extPoint[2] = eta;
1252         extPoint[3] = xi * eta;
1253         for (j = 0; j < dim; j++) {
1254           PetscReal val = 0.;
1255 
1256           for (k = 0; k < Nv; k++) {
1257             val += extPoint[k] * zCoeff[dim * k + j];
1258           }
1259           v[i * dim + j] = val;
1260         }
1261       }
1262       if (J) {
1263         PetscReal extJ[8];
1264 
1265         extJ[0] = 0.;
1266         extJ[1] = 0.;
1267         extJ[2] = 1.;
1268         extJ[3] = 0.;
1269         extJ[4] = 0.;
1270         extJ[5] = 1.;
1271         extJ[6] = eta;
1272         extJ[7] = xi;
1273         for (j = 0; j < dim; j++) {
1274           for (k = 0; k < dimR; k++) {
1275             PetscReal val = 0.;
1276 
1277             for (l = 0; l < Nv; l++) {
1278               val += zCoeff[dim * l + j] * extJ[dimR * l + k];
1279             }
1280             J[i * dim * dim + dim * j + k] = val;
1281           }
1282         }
1283         if (dim == 3) { /* put the cross product in the third component of the Jacobian */
1284           PetscReal x, y, z;
1285           PetscReal *iJ = &J[i * dim * dim];
1286           PetscReal norm;
1287 
1288           x = iJ[1 * dim + 0] * iJ[2 * dim + 1] - iJ[1 * dim + 1] * iJ[2 * dim + 0];
1289           y = iJ[0 * dim + 1] * iJ[2 * dim + 0] - iJ[0 * dim + 0] * iJ[2 * dim + 1];
1290           z = iJ[0 * dim + 0] * iJ[1 * dim + 1] - iJ[0 * dim + 1] * iJ[1 * dim + 0];
1291           norm = PetscSqrtReal(x * x + y * y + z * z);
1292           iJ[2] = x / norm;
1293           iJ[5] = y / norm;
1294           iJ[8] = z / norm;
1295           DMPlex_Det3D_Internal(&detJ[i], &J[i * dim * dim]);
1296           if (invJ) {DMPlex_Invert3D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);}
1297         } else {
1298           DMPlex_Det2D_Internal(&detJ[i], &J[i * dim * dim]);
1299           if (invJ) {DMPlex_Invert2D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);}
1300         }
1301       }
1302     }
1303   }
1304   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, &numCoords, &coords);CHKERRQ(ierr);
1305   PetscFunctionReturn(0);
1306 }
1307 
1308 static PetscErrorCode DMPlexComputeTetrahedronGeometry_Internal(DM dm, PetscInt e, PetscReal v0[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1309 {
1310   PetscSection   coordSection;
1311   Vec            coordinates;
1312   PetscScalar   *coords = NULL;
1313   const PetscInt dim = 3;
1314   PetscInt       d;
1315   PetscErrorCode ierr;
1316 
1317   PetscFunctionBegin;
1318   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1319   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1320   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr);
1321   *detJ = 0.0;
1322   if (v0)   {for (d = 0; d < dim; d++) v0[d] = PetscRealPart(coords[d]);}
1323   if (J)    {
1324     for (d = 0; d < dim; d++) {
1325       /* I orient with outward face normals */
1326       J[d*dim+0] = 0.5*(PetscRealPart(coords[2*dim+d]) - PetscRealPart(coords[0*dim+d]));
1327       J[d*dim+1] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d]));
1328       J[d*dim+2] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d]));
1329     }
1330     ierr = PetscLogFlops(18.0);CHKERRQ(ierr);
1331     DMPlex_Det3D_Internal(detJ, J);
1332   }
1333   if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);}
1334   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr);
1335   PetscFunctionReturn(0);
1336 }
1337 
1338 static PetscErrorCode DMPlexComputeHexahedronGeometry_Internal(DM dm, PetscInt e, PetscInt Nq, const PetscReal points[], PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1339 {
1340   PetscSection   coordSection;
1341   Vec            coordinates;
1342   PetscScalar   *coords = NULL;
1343   const PetscInt dim = 3;
1344   PetscInt       d;
1345   PetscErrorCode ierr;
1346 
1347   PetscFunctionBegin;
1348   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1349   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1350   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr);
1351   if (!Nq) {
1352     *detJ = 0.0;
1353     if (v)   {for (d = 0; d < dim; d++) v[d] = PetscRealPart(coords[d]);}
1354     if (J)    {
1355       for (d = 0; d < dim; d++) {
1356         J[d*dim+0] = 0.5*(PetscRealPart(coords[3*dim+d]) - PetscRealPart(coords[0*dim+d]));
1357         J[d*dim+1] = 0.5*(PetscRealPart(coords[1*dim+d]) - PetscRealPart(coords[0*dim+d]));
1358         J[d*dim+2] = 0.5*(PetscRealPart(coords[4*dim+d]) - PetscRealPart(coords[0*dim+d]));
1359       }
1360       ierr = PetscLogFlops(18.0);CHKERRQ(ierr);
1361       DMPlex_Det3D_Internal(detJ, J);
1362     }
1363     if (invJ) {DMPlex_Invert3D_Internal(invJ, J, *detJ);}
1364   } else {
1365     const PetscInt Nv = 8;
1366     const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6};
1367     const PetscInt dim = 3;
1368     const PetscInt dimR = 3;
1369     PetscReal zOrder[24];
1370     PetscReal zCoeff[24];
1371     PetscInt  i, j, k, l;
1372 
1373     for (i = 0; i < Nv; i++) {
1374       PetscInt zi = zToPlex[i];
1375 
1376       for (j = 0; j < dim; j++) {
1377         zOrder[dim * i + j] = PetscRealPart(coords[dim * zi + j]);
1378       }
1379     }
1380     for (j = 0; j < dim; j++) {
1381       zCoeff[dim * 0 + j] = 0.125 * (  zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j] + zOrder[dim * 4 + j] + zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1382       zCoeff[dim * 1 + j] = 0.125 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j] - zOrder[dim * 4 + j] + zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1383       zCoeff[dim * 2 + j] = 0.125 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] + zOrder[dim * 3 + j] - zOrder[dim * 4 + j] - zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1384       zCoeff[dim * 3 + j] = 0.125 * (  zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] + zOrder[dim * 3 + j] + zOrder[dim * 4 + j] - zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1385       zCoeff[dim * 4 + j] = 0.125 * (- zOrder[dim * 0 + j] - zOrder[dim * 1 + j] - zOrder[dim * 2 + j] - zOrder[dim * 3 + j] + zOrder[dim * 4 + j] + zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1386       zCoeff[dim * 5 + j] = 0.125 * (+ zOrder[dim * 0 + j] - zOrder[dim * 1 + j] + zOrder[dim * 2 + j] - zOrder[dim * 3 + j] - zOrder[dim * 4 + j] + zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1387       zCoeff[dim * 6 + j] = 0.125 * (+ zOrder[dim * 0 + j] + zOrder[dim * 1 + j] - zOrder[dim * 2 + j] - zOrder[dim * 3 + j] - zOrder[dim * 4 + j] - zOrder[dim * 5 + j] + zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1388       zCoeff[dim * 7 + j] = 0.125 * (- zOrder[dim * 0 + j] + zOrder[dim * 1 + j] + zOrder[dim * 2 + j] - zOrder[dim * 3 + j] + zOrder[dim * 4 + j] - zOrder[dim * 5 + j] - zOrder[dim * 6 + j] + zOrder[dim * 7 + j]);
1389     }
1390     for (i = 0; i < Nq; i++) {
1391       PetscReal xi = points[dimR * i], eta = points[dimR * i + 1], theta = points[dimR * i + 2];
1392 
1393       if (v) {
1394         PetscReal extPoint[8];
1395 
1396         extPoint[0] = 1.;
1397         extPoint[1] = xi;
1398         extPoint[2] = eta;
1399         extPoint[3] = xi * eta;
1400         extPoint[4] = theta;
1401         extPoint[5] = theta * xi;
1402         extPoint[6] = theta * eta;
1403         extPoint[7] = theta * eta * xi;
1404         for (j = 0; j < dim; j++) {
1405           PetscReal val = 0.;
1406 
1407           for (k = 0; k < Nv; k++) {
1408             val += extPoint[k] * zCoeff[dim * k + j];
1409           }
1410           v[i * dim + j] = val;
1411         }
1412       }
1413       if (J) {
1414         PetscReal extJ[24];
1415 
1416         extJ[0]  = 0.         ; extJ[1]  = 0.        ; extJ[2]  = 0.      ;
1417         extJ[3]  = 1.         ; extJ[4]  = 0.        ; extJ[5]  = 0.      ;
1418         extJ[6]  = 0.         ; extJ[7]  = 1.        ; extJ[8]  = 0.      ;
1419         extJ[9]  = eta        ; extJ[10] = xi        ; extJ[11] = 0.      ;
1420         extJ[12] = 0.         ; extJ[13] = 0.        ; extJ[14] = 1.      ;
1421         extJ[15] = theta      ; extJ[16] = 0.        ; extJ[17] = xi      ;
1422         extJ[18] = 0.         ; extJ[19] = theta     ; extJ[20] = eta     ;
1423         extJ[21] = theta * eta; extJ[22] = theta * xi; extJ[23] = eta * xi;
1424 
1425         for (j = 0; j < dim; j++) {
1426           for (k = 0; k < dimR; k++) {
1427             PetscReal val = 0.;
1428 
1429             for (l = 0; l < Nv; l++) {
1430               val += zCoeff[dim * l + j] * extJ[dimR * l + k];
1431             }
1432             J[i * dim * dim + dim * j + k] = val;
1433           }
1434         }
1435         DMPlex_Det3D_Internal(&detJ[i], &J[i * dim * dim]);
1436         if (invJ) {DMPlex_Invert3D_Internal(&invJ[i * dim * dim], &J[i * dim * dim], detJ[i]);}
1437       }
1438     }
1439   }
1440   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, e, NULL, &coords);CHKERRQ(ierr);
1441   PetscFunctionReturn(0);
1442 }
1443 
1444 static PetscErrorCode DMPlexComputeCellGeometryFEM_Implicit(DM dm, PetscInt cell, PetscQuadrature quad, PetscReal *v, PetscReal *J, PetscReal *invJ, PetscReal *detJ)
1445 {
1446   PetscInt        depth, dim, coordDim, coneSize, i;
1447   PetscInt        Nq = 0;
1448   const PetscReal *points = NULL;
1449   DMLabel         depthLabel;
1450   PetscReal       v0[3] = {-1.}, J0[9] = {-1.}, detJ0 = -1.;
1451   PetscBool       isAffine = PETSC_TRUE;
1452   PetscErrorCode  ierr;
1453 
1454   PetscFunctionBegin;
1455   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
1456   ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr);
1457   ierr = DMPlexGetDepthLabel(dm, &depthLabel);CHKERRQ(ierr);
1458   ierr = DMLabelGetValue(depthLabel, cell, &dim);CHKERRQ(ierr);
1459   if (depth == 1 && dim == 1) {
1460     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1461   }
1462   ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr);
1463   if (coordDim > 3) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported coordinate dimension %D > 3", coordDim);
1464   if (quad) {ierr = PetscQuadratureGetData(quad, NULL, NULL, &Nq, &points, NULL);CHKERRQ(ierr);}
1465   switch (dim) {
1466   case 0:
1467     ierr = DMPlexComputePointGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr);
1468     isAffine = PETSC_FALSE;
1469     break;
1470   case 1:
1471     if (Nq) {
1472       ierr = DMPlexComputeLineGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr);
1473     } else {
1474       ierr = DMPlexComputeLineGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr);
1475     }
1476     break;
1477   case 2:
1478     switch (coneSize) {
1479     case 3:
1480       if (Nq) {
1481         ierr = DMPlexComputeTriangleGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr);
1482       } else {
1483         ierr = DMPlexComputeTriangleGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr);
1484       }
1485       break;
1486     case 4:
1487       ierr = DMPlexComputeRectangleGeometry_Internal(dm, cell, Nq, points, v, J, invJ, detJ);CHKERRQ(ierr);
1488       isAffine = PETSC_FALSE;
1489       break;
1490     default:
1491       SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported number of faces %D in cell %D for element geometry computation", coneSize, cell);
1492     }
1493     break;
1494   case 3:
1495     switch (coneSize) {
1496     case 4:
1497       if (Nq) {
1498         ierr = DMPlexComputeTetrahedronGeometry_Internal(dm, cell, v0, J0, NULL, &detJ0);CHKERRQ(ierr);
1499       } else {
1500         ierr = DMPlexComputeTetrahedronGeometry_Internal(dm, cell, v, J, invJ, detJ);CHKERRQ(ierr);
1501       }
1502       break;
1503     case 6: /* Faces */
1504     case 8: /* Vertices */
1505       ierr = DMPlexComputeHexahedronGeometry_Internal(dm, cell, Nq, points, v, J, invJ, detJ);CHKERRQ(ierr);
1506       isAffine = PETSC_FALSE;
1507       break;
1508     default:
1509       SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported number of faces %D in cell %D for element geometry computation", coneSize, cell);
1510     }
1511     break;
1512   default:
1513     SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension %D for element geometry computation", dim);
1514   }
1515   if (isAffine && Nq) {
1516     if (v) {
1517       for (i = 0; i < Nq; i++) {
1518         CoordinatesRefToReal(coordDim,dim,v0, J0, &points[dim * i], &v[coordDim * i]);
1519       }
1520     }
1521     if (detJ) {
1522       for (i = 0; i < Nq; i++) {
1523         detJ[i] = detJ0;
1524       }
1525     }
1526     if (J) {
1527       PetscInt k;
1528 
1529       for (i = 0, k = 0; i < Nq; i++) {
1530         PetscInt j;
1531 
1532         for (j = 0; j < coordDim * coordDim; j++, k++) {
1533           J[k] = J0[j];
1534         }
1535       }
1536     }
1537     if (invJ) {
1538       PetscInt k;
1539       switch (coordDim) {
1540       case 0:
1541         break;
1542       case 1:
1543         invJ[0] = 1./J0[0];
1544         break;
1545       case 2:
1546         DMPlex_Invert2D_Internal(invJ, J0, detJ0);
1547         break;
1548       case 3:
1549         DMPlex_Invert3D_Internal(invJ, J0, detJ0);
1550         break;
1551       }
1552       for (i = 1, k = coordDim * coordDim; i < Nq; i++) {
1553         PetscInt j;
1554 
1555         for (j = 0; j < coordDim * coordDim; j++, k++) {
1556           invJ[k] = invJ[j];
1557         }
1558       }
1559     }
1560   }
1561   PetscFunctionReturn(0);
1562 }
1563 
1564 /*@C
1565   DMPlexComputeCellGeometryAffineFEM - Assuming an affine map, compute the Jacobian, inverse Jacobian, and Jacobian determinant for a given cell
1566 
1567   Collective on DM
1568 
1569   Input Arguments:
1570 + dm   - the DM
1571 - cell - the cell
1572 
1573   Output Arguments:
1574 + v0   - the translation part of this affine transform
1575 . J    - the Jacobian of the transform from the reference element
1576 . invJ - the inverse of the Jacobian
1577 - detJ - the Jacobian determinant
1578 
1579   Level: advanced
1580 
1581   Fortran Notes:
1582   Since it returns arrays, this routine is only available in Fortran 90, and you must
1583   include petsc.h90 in your code.
1584 
1585 .seealso: DMPlexComputeCellGeometryFEM(), DMGetCoordinateSection(), DMGetCoordinates()
1586 @*/
1587 PetscErrorCode DMPlexComputeCellGeometryAffineFEM(DM dm, PetscInt cell, PetscReal *v0, PetscReal *J, PetscReal *invJ, PetscReal *detJ)
1588 {
1589   PetscErrorCode ierr;
1590 
1591   PetscFunctionBegin;
1592   ierr = DMPlexComputeCellGeometryFEM_Implicit(dm,cell,NULL,v0,J,invJ,detJ);CHKERRQ(ierr);
1593   PetscFunctionReturn(0);
1594 }
1595 
1596 static PetscErrorCode DMPlexComputeCellGeometryFEM_FE(DM dm, PetscFE fe, PetscInt point, PetscQuadrature quad, PetscReal v[], PetscReal J[], PetscReal invJ[], PetscReal *detJ)
1597 {
1598   PetscQuadrature  feQuad;
1599   PetscSection     coordSection;
1600   Vec              coordinates;
1601   PetscScalar     *coords = NULL;
1602   const PetscReal *quadPoints;
1603   PetscReal       *basisDer, *basis, detJt;
1604   PetscInt         dim, cdim, pdim, qdim, Nq, numCoords, q;
1605   PetscErrorCode   ierr;
1606 
1607   PetscFunctionBegin;
1608   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1609   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1610   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, point, &numCoords, &coords);CHKERRQ(ierr);
1611   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1612   ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr);
1613   if (!quad) { /* use the first point of the first functional of the dual space */
1614     PetscDualSpace dsp;
1615 
1616     ierr = PetscFEGetDualSpace(fe, &dsp);CHKERRQ(ierr);
1617     ierr = PetscDualSpaceGetFunctional(dsp, 0, &quad);CHKERRQ(ierr);
1618     ierr = PetscQuadratureGetData(quad, &qdim, NULL, &Nq, &quadPoints, NULL);CHKERRQ(ierr);
1619     Nq = 1;
1620   } else {
1621     ierr = PetscQuadratureGetData(quad, &qdim, NULL, &Nq, &quadPoints, NULL);CHKERRQ(ierr);
1622   }
1623   ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr);
1624   ierr = PetscFEGetQuadrature(fe, &feQuad);CHKERRQ(ierr);
1625   if (feQuad == quad) {
1626     ierr = PetscFEGetDefaultTabulation(fe, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr);
1627     if (numCoords != pdim*cdim) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "There are %d coordinates for point %d != %d*%d", numCoords, point, pdim, cdim);
1628   } else {
1629     ierr = PetscFEGetTabulation(fe, Nq, quadPoints, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr);
1630   }
1631   if (qdim != dim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Point dimension %d != quadrature dimension %d", dim, qdim);
1632   if (v) {
1633     ierr = PetscMemzero(v, Nq*cdim*sizeof(PetscReal));CHKERRQ(ierr);
1634     for (q = 0; q < Nq; ++q) {
1635       PetscInt i, k;
1636 
1637       for (k = 0; k < pdim; ++k)
1638         for (i = 0; i < cdim; ++i)
1639           v[q*cdim + i] += basis[q*pdim + k] * PetscRealPart(coords[k*cdim + i]);
1640       ierr = PetscLogFlops(2.0*pdim*cdim);CHKERRQ(ierr);
1641     }
1642   }
1643   if (J) {
1644     ierr = PetscMemzero(J, Nq*cdim*cdim*sizeof(PetscReal));CHKERRQ(ierr);
1645     for (q = 0; q < Nq; ++q) {
1646       PetscInt i, j, k, c, r;
1647 
1648       /* J = dx_i/d\xi_j = sum[k=0,n-1] dN_k/d\xi_j * x_i(k) */
1649       for (k = 0; k < pdim; ++k)
1650         for (j = 0; j < dim; ++j)
1651           for (i = 0; i < cdim; ++i)
1652             J[(q*cdim + i)*cdim + j] += basisDer[(q*pdim + k)*dim + j] * PetscRealPart(coords[k*cdim + i]);
1653       ierr = PetscLogFlops(2.0*pdim*dim*cdim);CHKERRQ(ierr);
1654       if (cdim > dim) {
1655         for (c = dim; c < cdim; ++c)
1656           for (r = 0; r < cdim; ++r)
1657             J[r*cdim+c] = r == c ? 1.0 : 0.0;
1658       }
1659       if (!detJ && !invJ) continue;
1660       detJt = 0.;
1661       switch (cdim) {
1662       case 3:
1663         DMPlex_Det3D_Internal(&detJt, &J[q*cdim*dim]);
1664         if (invJ) {DMPlex_Invert3D_Internal(&invJ[q*cdim*dim], &J[q*cdim*dim], detJt);}
1665         break;
1666       case 2:
1667         DMPlex_Det2D_Internal(&detJt, &J[q*cdim*dim]);
1668         if (invJ) {DMPlex_Invert2D_Internal(&invJ[q*cdim*dim], &J[q*cdim*dim], detJt);}
1669         break;
1670       case 1:
1671         detJt = J[q*cdim*dim];
1672         if (invJ) invJ[q*cdim*dim] = 1.0/detJt;
1673       }
1674       if (detJ) detJ[q] = detJt;
1675     }
1676   }
1677   else if (detJ || invJ) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Need J to compute invJ or detJ");
1678   if (feQuad != quad) {
1679     ierr = PetscFERestoreTabulation(fe, Nq, quadPoints, &basis, J ? &basisDer : NULL, NULL);CHKERRQ(ierr);
1680   }
1681   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, point, &numCoords, &coords);CHKERRQ(ierr);
1682   PetscFunctionReturn(0);
1683 }
1684 
1685 /*@C
1686   DMPlexComputeCellGeometryFEM - Compute the Jacobian, inverse Jacobian, and Jacobian determinant at each quadrature point in the given cell
1687 
1688   Collective on DM
1689 
1690   Input Arguments:
1691 + dm   - the DM
1692 . cell - the cell
1693 - quad - the quadrature containing the points in the reference element where the geometry will be evaluated.  If quad == NULL, geometry will be
1694          evaluated at the first vertex of the reference element
1695 
1696   Output Arguments:
1697 + v    - the image of the transformed quadrature points, otherwise the image of the first vertex in the closure of the reference element
1698 . J    - the Jacobian of the transform from the reference element at each quadrature point
1699 . invJ - the inverse of the Jacobian at each quadrature point
1700 - detJ - the Jacobian determinant at each quadrature point
1701 
1702   Level: advanced
1703 
1704   Fortran Notes:
1705   Since it returns arrays, this routine is only available in Fortran 90, and you must
1706   include petsc.h90 in your code.
1707 
1708 .seealso: DMGetCoordinateSection(), DMGetCoordinates()
1709 @*/
1710 PetscErrorCode DMPlexComputeCellGeometryFEM(DM dm, PetscInt cell, PetscQuadrature quad, PetscReal *v, PetscReal *J, PetscReal *invJ, PetscReal *detJ)
1711 {
1712   PetscFE        fe = NULL;
1713   PetscErrorCode ierr;
1714 
1715   PetscFunctionBegin;
1716   if (dm->coordinateDM) {
1717     PetscClassId id;
1718     PetscInt     numFields;
1719     PetscDS      prob = dm->coordinateDM->prob;
1720     PetscObject  disc;
1721 
1722     ierr = PetscDSGetNumFields(prob, &numFields);CHKERRQ(ierr);
1723     if (numFields) {
1724       ierr = PetscDSGetDiscretization(prob,0,&disc);CHKERRQ(ierr);
1725       ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
1726       if (id == PETSCFE_CLASSID) {
1727         fe = (PetscFE) disc;
1728       }
1729     }
1730   }
1731   if (!fe) {ierr = DMPlexComputeCellGeometryFEM_Implicit(dm, cell, quad, v, J, invJ, detJ);CHKERRQ(ierr);}
1732   else     {ierr = DMPlexComputeCellGeometryFEM_FE(dm, fe, cell, quad, v, J, invJ, detJ);CHKERRQ(ierr);}
1733   PetscFunctionReturn(0);
1734 }
1735 
1736 static PetscErrorCode DMPlexComputeGeometryFVM_1D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[])
1737 {
1738   PetscSection   coordSection;
1739   Vec            coordinates;
1740   PetscScalar   *coords = NULL;
1741   PetscScalar    tmp[2];
1742   PetscInt       coordSize;
1743   PetscErrorCode ierr;
1744 
1745   PetscFunctionBegin;
1746   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1747   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1748   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr);
1749   if (dim != 2) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "We only support 2D edges right now");
1750   ierr = DMLocalizeCoordinate_Internal(dm, dim, coords, &coords[dim], tmp);CHKERRQ(ierr);
1751   if (centroid) {
1752     centroid[0] = 0.5*PetscRealPart(coords[0] + tmp[0]);
1753     centroid[1] = 0.5*PetscRealPart(coords[1] + tmp[1]);
1754   }
1755   if (normal) {
1756     PetscReal norm;
1757 
1758     normal[0]  = -PetscRealPart(coords[1] - tmp[1]);
1759     normal[1]  =  PetscRealPart(coords[0] - tmp[0]);
1760     norm       = PetscSqrtReal(normal[0]*normal[0] + normal[1]*normal[1]);
1761     normal[0] /= norm;
1762     normal[1] /= norm;
1763   }
1764   if (vol) {
1765     *vol = PetscSqrtReal(PetscSqr(PetscRealPart(coords[0] - tmp[0])) + PetscSqr(PetscRealPart(coords[1] - tmp[1])));
1766   }
1767   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr);
1768   PetscFunctionReturn(0);
1769 }
1770 
1771 /* Centroid_i = (\sum_n A_n Cn_i ) / A */
1772 static PetscErrorCode DMPlexComputeGeometryFVM_2D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[])
1773 {
1774   PetscSection   coordSection;
1775   Vec            coordinates;
1776   PetscScalar   *coords = NULL;
1777   PetscReal      vsum = 0.0, csum[3] = {0.0, 0.0, 0.0}, vtmp, ctmp[4], v0[3], R[9];
1778   PetscInt       tdim = 2, coordSize, numCorners, p, d, e;
1779   PetscErrorCode ierr;
1780 
1781   PetscFunctionBegin;
1782   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1783   ierr = DMPlexGetConeSize(dm, cell, &numCorners);CHKERRQ(ierr);
1784   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1785   ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr);
1786   ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr);
1787   if (dim > 2 && centroid) {
1788     v0[0] = PetscRealPart(coords[0]);
1789     v0[1] = PetscRealPart(coords[1]);
1790     v0[2] = PetscRealPart(coords[2]);
1791   }
1792   if (normal) {
1793     if (dim > 2) {
1794       const PetscReal x0 = PetscRealPart(coords[dim+0] - coords[0]), x1 = PetscRealPart(coords[dim*2+0] - coords[0]);
1795       const PetscReal y0 = PetscRealPart(coords[dim+1] - coords[1]), y1 = PetscRealPart(coords[dim*2+1] - coords[1]);
1796       const PetscReal z0 = PetscRealPart(coords[dim+2] - coords[2]), z1 = PetscRealPart(coords[dim*2+2] - coords[2]);
1797       PetscReal       norm;
1798 
1799       normal[0] = y0*z1 - z0*y1;
1800       normal[1] = z0*x1 - x0*z1;
1801       normal[2] = x0*y1 - y0*x1;
1802       norm = PetscSqrtReal(normal[0]*normal[0] + normal[1]*normal[1] + normal[2]*normal[2]);
1803       normal[0] /= norm;
1804       normal[1] /= norm;
1805       normal[2] /= norm;
1806     } else {
1807       for (d = 0; d < dim; ++d) normal[d] = 0.0;
1808     }
1809   }
1810   if (dim == 3) {ierr = DMPlexComputeProjection3Dto2D(coordSize, coords, R);CHKERRQ(ierr);}
1811   for (p = 0; p < numCorners; ++p) {
1812     /* Need to do this copy to get types right */
1813     for (d = 0; d < tdim; ++d) {
1814       ctmp[d]      = PetscRealPart(coords[p*tdim+d]);
1815       ctmp[tdim+d] = PetscRealPart(coords[((p+1)%numCorners)*tdim+d]);
1816     }
1817     Volume_Triangle_Origin_Internal(&vtmp, ctmp);
1818     vsum += vtmp;
1819     for (d = 0; d < tdim; ++d) {
1820       csum[d] += (ctmp[d] + ctmp[tdim+d])*vtmp;
1821     }
1822   }
1823   for (d = 0; d < tdim; ++d) {
1824     csum[d] /= (tdim+1)*vsum;
1825   }
1826   ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, cell, &coordSize, &coords);CHKERRQ(ierr);
1827   if (vol) *vol = PetscAbsReal(vsum);
1828   if (centroid) {
1829     if (dim > 2) {
1830       for (d = 0; d < dim; ++d) {
1831         centroid[d] = v0[d];
1832         for (e = 0; e < dim; ++e) {
1833           centroid[d] += R[d*dim+e]*csum[e];
1834         }
1835       }
1836     } else for (d = 0; d < dim; ++d) centroid[d] = csum[d];
1837   }
1838   PetscFunctionReturn(0);
1839 }
1840 
1841 /* Centroid_i = (\sum_n V_n Cn_i ) / V */
1842 static PetscErrorCode DMPlexComputeGeometryFVM_3D_Internal(DM dm, PetscInt dim, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[])
1843 {
1844   PetscSection    coordSection;
1845   Vec             coordinates;
1846   PetscScalar    *coords = NULL;
1847   PetscReal       vsum = 0.0, vtmp, coordsTmp[3*3];
1848   const PetscInt *faces, *facesO;
1849   PetscInt        numFaces, f, coordSize, numCorners, p, d;
1850   PetscErrorCode  ierr;
1851 
1852   PetscFunctionBegin;
1853   if (PetscUnlikely(dim > 3)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"No support for dim %D > 3",dim);
1854   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
1855   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1856 
1857   if (centroid) for (d = 0; d < dim; ++d) centroid[d] = 0.0;
1858   ierr = DMPlexGetConeSize(dm, cell, &numFaces);CHKERRQ(ierr);
1859   ierr = DMPlexGetCone(dm, cell, &faces);CHKERRQ(ierr);
1860   ierr = DMPlexGetConeOrientation(dm, cell, &facesO);CHKERRQ(ierr);
1861   for (f = 0; f < numFaces; ++f) {
1862     ierr = DMPlexVecGetClosure(dm, coordSection, coordinates, faces[f], &coordSize, &coords);CHKERRQ(ierr);
1863     numCorners = coordSize/dim;
1864     switch (numCorners) {
1865     case 3:
1866       for (d = 0; d < dim; ++d) {
1867         coordsTmp[0*dim+d] = PetscRealPart(coords[0*dim+d]);
1868         coordsTmp[1*dim+d] = PetscRealPart(coords[1*dim+d]);
1869         coordsTmp[2*dim+d] = PetscRealPart(coords[2*dim+d]);
1870       }
1871       Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp);
1872       if (facesO[f] < 0) vtmp = -vtmp;
1873       vsum += vtmp;
1874       if (centroid) {           /* Centroid of OABC = (a+b+c)/4 */
1875         for (d = 0; d < dim; ++d) {
1876           for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp;
1877         }
1878       }
1879       break;
1880     case 4:
1881       /* DO FOR PYRAMID */
1882       /* First tet */
1883       for (d = 0; d < dim; ++d) {
1884         coordsTmp[0*dim+d] = PetscRealPart(coords[0*dim+d]);
1885         coordsTmp[1*dim+d] = PetscRealPart(coords[1*dim+d]);
1886         coordsTmp[2*dim+d] = PetscRealPart(coords[3*dim+d]);
1887       }
1888       Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp);
1889       if (facesO[f] < 0) vtmp = -vtmp;
1890       vsum += vtmp;
1891       if (centroid) {
1892         for (d = 0; d < dim; ++d) {
1893           for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp;
1894         }
1895       }
1896       /* Second tet */
1897       for (d = 0; d < dim; ++d) {
1898         coordsTmp[0*dim+d] = PetscRealPart(coords[1*dim+d]);
1899         coordsTmp[1*dim+d] = PetscRealPart(coords[2*dim+d]);
1900         coordsTmp[2*dim+d] = PetscRealPart(coords[3*dim+d]);
1901       }
1902       Volume_Tetrahedron_Origin_Internal(&vtmp, coordsTmp);
1903       if (facesO[f] < 0) vtmp = -vtmp;
1904       vsum += vtmp;
1905       if (centroid) {
1906         for (d = 0; d < dim; ++d) {
1907           for (p = 0; p < 3; ++p) centroid[d] += coordsTmp[p*dim+d]*vtmp;
1908         }
1909       }
1910       break;
1911     default:
1912       SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle faces with %D vertices", numCorners);
1913     }
1914     ierr = DMPlexVecRestoreClosure(dm, coordSection, coordinates, faces[f], &coordSize, &coords);CHKERRQ(ierr);
1915   }
1916   if (vol)     *vol = PetscAbsReal(vsum);
1917   if (normal)   for (d = 0; d < dim; ++d) normal[d]    = 0.0;
1918   if (centroid) for (d = 0; d < dim; ++d) centroid[d] /= (vsum*4);
1919   PetscFunctionReturn(0);
1920 }
1921 
1922 /*@C
1923   DMPlexComputeCellGeometryFVM - Compute the volume for a given cell
1924 
1925   Collective on DM
1926 
1927   Input Arguments:
1928 + dm   - the DM
1929 - cell - the cell
1930 
1931   Output Arguments:
1932 + volume   - the cell volume
1933 . centroid - the cell centroid
1934 - normal - the cell normal, if appropriate
1935 
1936   Level: advanced
1937 
1938   Fortran Notes:
1939   Since it returns arrays, this routine is only available in Fortran 90, and you must
1940   include petsc.h90 in your code.
1941 
1942 .seealso: DMGetCoordinateSection(), DMGetCoordinates()
1943 @*/
1944 PetscErrorCode DMPlexComputeCellGeometryFVM(DM dm, PetscInt cell, PetscReal *vol, PetscReal centroid[], PetscReal normal[])
1945 {
1946   PetscInt       depth, dim;
1947   PetscErrorCode ierr;
1948 
1949   PetscFunctionBegin;
1950   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
1951   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1952   if (depth != dim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh must be interpolated");
1953   /* We need to keep a pointer to the depth label */
1954   ierr = DMGetLabelValue(dm, "depth", cell, &depth);CHKERRQ(ierr);
1955   /* Cone size is now the number of faces */
1956   switch (depth) {
1957   case 1:
1958     ierr = DMPlexComputeGeometryFVM_1D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr);
1959     break;
1960   case 2:
1961     ierr = DMPlexComputeGeometryFVM_2D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr);
1962     break;
1963   case 3:
1964     ierr = DMPlexComputeGeometryFVM_3D_Internal(dm, dim, cell, vol, centroid, normal);CHKERRQ(ierr);
1965     break;
1966   default:
1967     SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported dimension %D for element geometry computation", dim);
1968   }
1969   PetscFunctionReturn(0);
1970 }
1971 
1972 /*@
1973   DMPlexComputeGeometryFEM - Precompute cell geometry for the entire mesh
1974 
1975   Collective on dm
1976 
1977   Input Parameter:
1978 . dm - The DMPlex
1979 
1980   Output Parameter:
1981 . cellgeom - A vector with the cell geometry data for each cell
1982 
1983   Level: beginner
1984 
1985 .keywords: DMPlexComputeCellGeometryFEM()
1986 @*/
1987 PetscErrorCode DMPlexComputeGeometryFEM(DM dm, Vec *cellgeom)
1988 {
1989   DM             dmCell;
1990   Vec            coordinates;
1991   PetscSection   coordSection, sectionCell;
1992   PetscScalar   *cgeom;
1993   PetscInt       cStart, cEnd, cMax, c;
1994   PetscErrorCode ierr;
1995 
1996   PetscFunctionBegin;
1997   ierr = DMClone(dm, &dmCell);CHKERRQ(ierr);
1998   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
1999   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
2000   ierr = DMSetCoordinateSection(dmCell, PETSC_DETERMINE, coordSection);CHKERRQ(ierr);
2001   ierr = DMSetCoordinatesLocal(dmCell, coordinates);CHKERRQ(ierr);
2002   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &sectionCell);CHKERRQ(ierr);
2003   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
2004   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
2005   cEnd = cMax < 0 ? cEnd : cMax;
2006   ierr = PetscSectionSetChart(sectionCell, cStart, cEnd);CHKERRQ(ierr);
2007   /* TODO This needs to be multiplied by Nq for non-affine */
2008   for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionCell, c, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFECellGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);}
2009   ierr = PetscSectionSetUp(sectionCell);CHKERRQ(ierr);
2010   ierr = DMSetDefaultSection(dmCell, sectionCell);CHKERRQ(ierr);
2011   ierr = PetscSectionDestroy(&sectionCell);CHKERRQ(ierr);
2012   ierr = DMCreateLocalVector(dmCell, cellgeom);CHKERRQ(ierr);
2013   ierr = VecGetArray(*cellgeom, &cgeom);CHKERRQ(ierr);
2014   for (c = cStart; c < cEnd; ++c) {
2015     PetscFECellGeom *cg;
2016 
2017     ierr = DMPlexPointLocalRef(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
2018     ierr = PetscMemzero(cg, sizeof(*cg));CHKERRQ(ierr);
2019     ierr = DMPlexComputeCellGeometryFEM(dmCell, c, NULL, cg->v0, cg->J, cg->invJ, &cg->detJ);CHKERRQ(ierr);
2020     if (cg->detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %d", cg->detJ, c);
2021   }
2022   ierr = VecRestoreArray(*cellgeom, &cgeom);CHKERRQ(ierr);
2023   ierr = DMDestroy(&dmCell);CHKERRQ(ierr);
2024   PetscFunctionReturn(0);
2025 }
2026 
2027 /*@
2028   DMPlexComputeGeometryFVM - Computes the cell and face geometry for a finite volume method
2029 
2030   Input Parameter:
2031 . dm - The DM
2032 
2033   Output Parameters:
2034 + cellgeom - A Vec of PetscFVCellGeom data
2035 . facegeom - A Vec of PetscFVFaceGeom data
2036 
2037   Level: developer
2038 
2039 .seealso: PetscFVFaceGeom, PetscFVCellGeom, DMPlexComputeGeometryFEM()
2040 @*/
2041 PetscErrorCode DMPlexComputeGeometryFVM(DM dm, Vec *cellgeom, Vec *facegeom)
2042 {
2043   DM             dmFace, dmCell;
2044   DMLabel        ghostLabel;
2045   PetscSection   sectionFace, sectionCell;
2046   PetscSection   coordSection;
2047   Vec            coordinates;
2048   PetscScalar   *fgeom, *cgeom;
2049   PetscReal      minradius, gminradius;
2050   PetscInt       dim, cStart, cEnd, cEndInterior, c, fStart, fEnd, f;
2051   PetscErrorCode ierr;
2052 
2053   PetscFunctionBegin;
2054   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
2055   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
2056   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
2057   /* Make cell centroids and volumes */
2058   ierr = DMClone(dm, &dmCell);CHKERRQ(ierr);
2059   ierr = DMSetCoordinateSection(dmCell, PETSC_DETERMINE, coordSection);CHKERRQ(ierr);
2060   ierr = DMSetCoordinatesLocal(dmCell, coordinates);CHKERRQ(ierr);
2061   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &sectionCell);CHKERRQ(ierr);
2062   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
2063   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr);
2064   ierr = PetscSectionSetChart(sectionCell, cStart, cEnd);CHKERRQ(ierr);
2065   for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionCell, c, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFVCellGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);}
2066   ierr = PetscSectionSetUp(sectionCell);CHKERRQ(ierr);
2067   ierr = DMSetDefaultSection(dmCell, sectionCell);CHKERRQ(ierr);
2068   ierr = PetscSectionDestroy(&sectionCell);CHKERRQ(ierr);
2069   ierr = DMCreateLocalVector(dmCell, cellgeom);CHKERRQ(ierr);
2070   if (cEndInterior < 0) {
2071     cEndInterior = cEnd;
2072   }
2073   ierr = VecGetArray(*cellgeom, &cgeom);CHKERRQ(ierr);
2074   for (c = cStart; c < cEndInterior; ++c) {
2075     PetscFVCellGeom *cg;
2076 
2077     ierr = DMPlexPointLocalRef(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
2078     ierr = PetscMemzero(cg, sizeof(*cg));CHKERRQ(ierr);
2079     ierr = DMPlexComputeCellGeometryFVM(dmCell, c, &cg->volume, cg->centroid, NULL);CHKERRQ(ierr);
2080   }
2081   /* Compute face normals and minimum cell radius */
2082   ierr = DMClone(dm, &dmFace);CHKERRQ(ierr);
2083   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &sectionFace);CHKERRQ(ierr);
2084   ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr);
2085   ierr = PetscSectionSetChart(sectionFace, fStart, fEnd);CHKERRQ(ierr);
2086   for (f = fStart; f < fEnd; ++f) {ierr = PetscSectionSetDof(sectionFace, f, (PetscInt) PetscCeilReal(((PetscReal) sizeof(PetscFVFaceGeom))/sizeof(PetscScalar)));CHKERRQ(ierr);}
2087   ierr = PetscSectionSetUp(sectionFace);CHKERRQ(ierr);
2088   ierr = DMSetDefaultSection(dmFace, sectionFace);CHKERRQ(ierr);
2089   ierr = PetscSectionDestroy(&sectionFace);CHKERRQ(ierr);
2090   ierr = DMCreateLocalVector(dmFace, facegeom);CHKERRQ(ierr);
2091   ierr = VecGetArray(*facegeom, &fgeom);CHKERRQ(ierr);
2092   ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr);
2093   minradius = PETSC_MAX_REAL;
2094   for (f = fStart; f < fEnd; ++f) {
2095     PetscFVFaceGeom *fg;
2096     PetscReal        area;
2097     PetscInt         ghost = -1, d, numChildren;
2098 
2099     if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);}
2100     ierr = DMPlexGetTreeChildren(dm,f,&numChildren,NULL);CHKERRQ(ierr);
2101     if (ghost >= 0 || numChildren) continue;
2102     ierr = DMPlexPointLocalRef(dmFace, f, fgeom, &fg);CHKERRQ(ierr);
2103     ierr = DMPlexComputeCellGeometryFVM(dm, f, &area, fg->centroid, fg->normal);CHKERRQ(ierr);
2104     for (d = 0; d < dim; ++d) fg->normal[d] *= area;
2105     /* Flip face orientation if necessary to match ordering in support, and Update minimum radius */
2106     {
2107       PetscFVCellGeom *cL, *cR;
2108       PetscInt         ncells;
2109       const PetscInt  *cells;
2110       PetscReal       *lcentroid, *rcentroid;
2111       PetscReal        l[3], r[3], v[3];
2112 
2113       ierr = DMPlexGetSupport(dm, f, &cells);CHKERRQ(ierr);
2114       ierr = DMPlexGetSupportSize(dm, f, &ncells);CHKERRQ(ierr);
2115       ierr = DMPlexPointLocalRead(dmCell, cells[0], cgeom, &cL);CHKERRQ(ierr);
2116       lcentroid = cells[0] >= cEndInterior ? fg->centroid : cL->centroid;
2117       if (ncells > 1) {
2118         ierr = DMPlexPointLocalRead(dmCell, cells[1], cgeom, &cR);CHKERRQ(ierr);
2119         rcentroid = cells[1] >= cEndInterior ? fg->centroid : cR->centroid;
2120       }
2121       else {
2122         rcentroid = fg->centroid;
2123       }
2124       ierr = DMLocalizeCoordinateReal_Internal(dm, dim, fg->centroid, lcentroid, l);CHKERRQ(ierr);
2125       ierr = DMLocalizeCoordinateReal_Internal(dm, dim, fg->centroid, rcentroid, r);CHKERRQ(ierr);
2126       DMPlex_WaxpyD_Internal(dim, -1, l, r, v);
2127       if (DMPlex_DotRealD_Internal(dim, fg->normal, v) < 0) {
2128         for (d = 0; d < dim; ++d) fg->normal[d] = -fg->normal[d];
2129       }
2130       if (DMPlex_DotRealD_Internal(dim, fg->normal, v) <= 0) {
2131         if (dim == 2) SETERRQ5(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed, normal (%g,%g) v (%g,%g)", f, (double) fg->normal[0], (double) fg->normal[1], (double) v[0], (double) v[1]);
2132         if (dim == 3) SETERRQ7(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed, normal (%g,%g,%g) v (%g,%g,%g)", f, (double) fg->normal[0], (double) fg->normal[1], (double) fg->normal[2], (double) v[0], (double) v[1], (double) v[2]);
2133         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Direction for face %d could not be fixed", f);
2134       }
2135       if (cells[0] < cEndInterior) {
2136         DMPlex_WaxpyD_Internal(dim, -1, fg->centroid, cL->centroid, v);
2137         minradius = PetscMin(minradius, DMPlex_NormD_Internal(dim, v));
2138       }
2139       if (ncells > 1 && cells[1] < cEndInterior) {
2140         DMPlex_WaxpyD_Internal(dim, -1, fg->centroid, cR->centroid, v);
2141         minradius = PetscMin(minradius, DMPlex_NormD_Internal(dim, v));
2142       }
2143     }
2144   }
2145   ierr = MPIU_Allreduce(&minradius, &gminradius, 1, MPIU_REAL, MPIU_MIN, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
2146   ierr = DMPlexSetMinRadius(dm, gminradius);CHKERRQ(ierr);
2147   /* Compute centroids of ghost cells */
2148   for (c = cEndInterior; c < cEnd; ++c) {
2149     PetscFVFaceGeom *fg;
2150     const PetscInt  *cone,    *support;
2151     PetscInt         coneSize, supportSize, s;
2152 
2153     ierr = DMPlexGetConeSize(dmCell, c, &coneSize);CHKERRQ(ierr);
2154     if (coneSize != 1) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Ghost cell %d has cone size %d != 1", c, coneSize);
2155     ierr = DMPlexGetCone(dmCell, c, &cone);CHKERRQ(ierr);
2156     ierr = DMPlexGetSupportSize(dmCell, cone[0], &supportSize);CHKERRQ(ierr);
2157     if (supportSize != 2) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %d has support size %d != 2", cone[0], supportSize);
2158     ierr = DMPlexGetSupport(dmCell, cone[0], &support);CHKERRQ(ierr);
2159     ierr = DMPlexPointLocalRef(dmFace, cone[0], fgeom, &fg);CHKERRQ(ierr);
2160     for (s = 0; s < 2; ++s) {
2161       /* Reflect ghost centroid across plane of face */
2162       if (support[s] == c) {
2163         PetscFVCellGeom       *ci;
2164         PetscFVCellGeom       *cg;
2165         PetscReal              c2f[3], a;
2166 
2167         ierr = DMPlexPointLocalRead(dmCell, support[(s+1)%2], cgeom, &ci);CHKERRQ(ierr);
2168         DMPlex_WaxpyD_Internal(dim, -1, ci->centroid, fg->centroid, c2f); /* cell to face centroid */
2169         a    = DMPlex_DotRealD_Internal(dim, c2f, fg->normal)/DMPlex_DotRealD_Internal(dim, fg->normal, fg->normal);
2170         ierr = DMPlexPointLocalRef(dmCell, support[s], cgeom, &cg);CHKERRQ(ierr);
2171         DMPlex_WaxpyD_Internal(dim, 2*a, fg->normal, ci->centroid, cg->centroid);
2172         cg->volume = ci->volume;
2173       }
2174     }
2175   }
2176   ierr = VecRestoreArray(*facegeom, &fgeom);CHKERRQ(ierr);
2177   ierr = VecRestoreArray(*cellgeom, &cgeom);CHKERRQ(ierr);
2178   ierr = DMDestroy(&dmCell);CHKERRQ(ierr);
2179   ierr = DMDestroy(&dmFace);CHKERRQ(ierr);
2180   PetscFunctionReturn(0);
2181 }
2182 
2183 /*@C
2184   DMPlexGetMinRadius - Returns the minimum distance from any cell centroid to a face
2185 
2186   Not collective
2187 
2188   Input Argument:
2189 . dm - the DM
2190 
2191   Output Argument:
2192 . minradius - the minium cell radius
2193 
2194   Level: developer
2195 
2196 .seealso: DMGetCoordinates()
2197 @*/
2198 PetscErrorCode DMPlexGetMinRadius(DM dm, PetscReal *minradius)
2199 {
2200   PetscFunctionBegin;
2201   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2202   PetscValidPointer(minradius,2);
2203   *minradius = ((DM_Plex*) dm->data)->minradius;
2204   PetscFunctionReturn(0);
2205 }
2206 
2207 /*@C
2208   DMPlexSetMinRadius - Sets the minimum distance from the cell centroid to a face
2209 
2210   Logically collective
2211 
2212   Input Arguments:
2213 + dm - the DM
2214 - minradius - the minium cell radius
2215 
2216   Level: developer
2217 
2218 .seealso: DMSetCoordinates()
2219 @*/
2220 PetscErrorCode DMPlexSetMinRadius(DM dm, PetscReal minradius)
2221 {
2222   PetscFunctionBegin;
2223   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2224   ((DM_Plex*) dm->data)->minradius = minradius;
2225   PetscFunctionReturn(0);
2226 }
2227 
2228 static PetscErrorCode BuildGradientReconstruction_Internal(DM dm, PetscFV fvm, DM dmFace, PetscScalar *fgeom, DM dmCell, PetscScalar *cgeom)
2229 {
2230   DMLabel        ghostLabel;
2231   PetscScalar   *dx, *grad, **gref;
2232   PetscInt       dim, cStart, cEnd, c, cEndInterior, maxNumFaces;
2233   PetscErrorCode ierr;
2234 
2235   PetscFunctionBegin;
2236   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
2237   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
2238   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr);
2239   ierr = DMPlexGetMaxSizes(dm, &maxNumFaces, NULL);CHKERRQ(ierr);
2240   ierr = PetscFVLeastSquaresSetMaxFaces(fvm, maxNumFaces);CHKERRQ(ierr);
2241   ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr);
2242   ierr = PetscMalloc3(maxNumFaces*dim, &dx, maxNumFaces*dim, &grad, maxNumFaces, &gref);CHKERRQ(ierr);
2243   for (c = cStart; c < cEndInterior; c++) {
2244     const PetscInt        *faces;
2245     PetscInt               numFaces, usedFaces, f, d;
2246     PetscFVCellGeom        *cg;
2247     PetscBool              boundary;
2248     PetscInt               ghost;
2249 
2250     ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
2251     ierr = DMPlexGetConeSize(dm, c, &numFaces);CHKERRQ(ierr);
2252     ierr = DMPlexGetCone(dm, c, &faces);CHKERRQ(ierr);
2253     if (numFaces < dim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Cell %D has only %D faces, not enough for gradient reconstruction", c, numFaces);
2254     for (f = 0, usedFaces = 0; f < numFaces; ++f) {
2255       PetscFVCellGeom       *cg1;
2256       PetscFVFaceGeom       *fg;
2257       const PetscInt        *fcells;
2258       PetscInt               ncell, side;
2259 
2260       ierr = DMLabelGetValue(ghostLabel, faces[f], &ghost);CHKERRQ(ierr);
2261       ierr = DMIsBoundaryPoint(dm, faces[f], &boundary);CHKERRQ(ierr);
2262       if ((ghost >= 0) || boundary) continue;
2263       ierr  = DMPlexGetSupport(dm, faces[f], &fcells);CHKERRQ(ierr);
2264       side  = (c != fcells[0]); /* c is on left=0 or right=1 of face */
2265       ncell = fcells[!side];    /* the neighbor */
2266       ierr  = DMPlexPointLocalRef(dmFace, faces[f], fgeom, &fg);CHKERRQ(ierr);
2267       ierr  = DMPlexPointLocalRead(dmCell, ncell, cgeom, &cg1);CHKERRQ(ierr);
2268       for (d = 0; d < dim; ++d) dx[usedFaces*dim+d] = cg1->centroid[d] - cg->centroid[d];
2269       gref[usedFaces++] = fg->grad[side];  /* Gradient reconstruction term will go here */
2270     }
2271     if (!usedFaces) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_USER, "Mesh contains isolated cell (no neighbors). Is it intentional?");
2272     ierr = PetscFVComputeGradient(fvm, usedFaces, dx, grad);CHKERRQ(ierr);
2273     for (f = 0, usedFaces = 0; f < numFaces; ++f) {
2274       ierr = DMLabelGetValue(ghostLabel, faces[f], &ghost);CHKERRQ(ierr);
2275       ierr = DMIsBoundaryPoint(dm, faces[f], &boundary);CHKERRQ(ierr);
2276       if ((ghost >= 0) || boundary) continue;
2277       for (d = 0; d < dim; ++d) gref[usedFaces][d] = grad[usedFaces*dim+d];
2278       ++usedFaces;
2279     }
2280   }
2281   ierr = PetscFree3(dx, grad, gref);CHKERRQ(ierr);
2282   PetscFunctionReturn(0);
2283 }
2284 
2285 static PetscErrorCode BuildGradientReconstruction_Internal_Tree(DM dm, PetscFV fvm, DM dmFace, PetscScalar *fgeom, DM dmCell, PetscScalar *cgeom)
2286 {
2287   DMLabel        ghostLabel;
2288   PetscScalar   *dx, *grad, **gref;
2289   PetscInt       dim, cStart, cEnd, c, cEndInterior, fStart, fEnd, f, nStart, nEnd, maxNumFaces = 0;
2290   PetscSection   neighSec;
2291   PetscInt     (*neighbors)[2];
2292   PetscInt      *counter;
2293   PetscErrorCode ierr;
2294 
2295   PetscFunctionBegin;
2296   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
2297   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
2298   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr);
2299   if (cEndInterior < 0) {
2300     cEndInterior = cEnd;
2301   }
2302   ierr = PetscSectionCreate(PetscObjectComm((PetscObject)dm),&neighSec);CHKERRQ(ierr);
2303   ierr = PetscSectionSetChart(neighSec,cStart,cEndInterior);CHKERRQ(ierr);
2304   ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr);
2305   ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr);
2306   for (f = fStart; f < fEnd; f++) {
2307     const PetscInt        *fcells;
2308     PetscBool              boundary;
2309     PetscInt               ghost = -1;
2310     PetscInt               numChildren, numCells, c;
2311 
2312     if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);}
2313     ierr = DMIsBoundaryPoint(dm, f, &boundary);CHKERRQ(ierr);
2314     ierr = DMPlexGetTreeChildren(dm, f, &numChildren, NULL);CHKERRQ(ierr);
2315     if ((ghost >= 0) || boundary || numChildren) continue;
2316     ierr = DMPlexGetSupportSize(dm, f, &numCells);CHKERRQ(ierr);
2317     if (numCells == 2) {
2318       ierr = DMPlexGetSupport(dm, f, &fcells);CHKERRQ(ierr);
2319       for (c = 0; c < 2; c++) {
2320         PetscInt cell = fcells[c];
2321 
2322         if (cell >= cStart && cell < cEndInterior) {
2323           ierr = PetscSectionAddDof(neighSec,cell,1);CHKERRQ(ierr);
2324         }
2325       }
2326     }
2327   }
2328   ierr = PetscSectionSetUp(neighSec);CHKERRQ(ierr);
2329   ierr = PetscSectionGetMaxDof(neighSec,&maxNumFaces);CHKERRQ(ierr);
2330   ierr = PetscFVLeastSquaresSetMaxFaces(fvm, maxNumFaces);CHKERRQ(ierr);
2331   nStart = 0;
2332   ierr = PetscSectionGetStorageSize(neighSec,&nEnd);CHKERRQ(ierr);
2333   ierr = PetscMalloc1((nEnd-nStart),&neighbors);CHKERRQ(ierr);
2334   ierr = PetscCalloc1((cEndInterior-cStart),&counter);CHKERRQ(ierr);
2335   for (f = fStart; f < fEnd; f++) {
2336     const PetscInt        *fcells;
2337     PetscBool              boundary;
2338     PetscInt               ghost = -1;
2339     PetscInt               numChildren, numCells, c;
2340 
2341     if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, f, &ghost);CHKERRQ(ierr);}
2342     ierr = DMIsBoundaryPoint(dm, f, &boundary);CHKERRQ(ierr);
2343     ierr = DMPlexGetTreeChildren(dm, f, &numChildren, NULL);CHKERRQ(ierr);
2344     if ((ghost >= 0) || boundary || numChildren) continue;
2345     ierr = DMPlexGetSupportSize(dm, f, &numCells);CHKERRQ(ierr);
2346     if (numCells == 2) {
2347       ierr  = DMPlexGetSupport(dm, f, &fcells);CHKERRQ(ierr);
2348       for (c = 0; c < 2; c++) {
2349         PetscInt cell = fcells[c], off;
2350 
2351         if (cell >= cStart && cell < cEndInterior) {
2352           ierr = PetscSectionGetOffset(neighSec,cell,&off);CHKERRQ(ierr);
2353           off += counter[cell - cStart]++;
2354           neighbors[off][0] = f;
2355           neighbors[off][1] = fcells[1 - c];
2356         }
2357       }
2358     }
2359   }
2360   ierr = PetscFree(counter);CHKERRQ(ierr);
2361   ierr = PetscMalloc3(maxNumFaces*dim, &dx, maxNumFaces*dim, &grad, maxNumFaces, &gref);CHKERRQ(ierr);
2362   for (c = cStart; c < cEndInterior; c++) {
2363     PetscInt               numFaces, f, d, off, ghost = -1;
2364     PetscFVCellGeom        *cg;
2365 
2366     ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
2367     ierr = PetscSectionGetDof(neighSec, c, &numFaces);CHKERRQ(ierr);
2368     ierr = PetscSectionGetOffset(neighSec, c, &off);CHKERRQ(ierr);
2369     if (ghostLabel) {ierr = DMLabelGetValue(ghostLabel, c, &ghost);CHKERRQ(ierr);}
2370     if (ghost < 0 && numFaces < dim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Cell %D has only %D faces, not enough for gradient reconstruction", c, numFaces);
2371     for (f = 0; f < numFaces; ++f) {
2372       PetscFVCellGeom       *cg1;
2373       PetscFVFaceGeom       *fg;
2374       const PetscInt        *fcells;
2375       PetscInt               ncell, side, nface;
2376 
2377       nface = neighbors[off + f][0];
2378       ncell = neighbors[off + f][1];
2379       ierr  = DMPlexGetSupport(dm,nface,&fcells);CHKERRQ(ierr);
2380       side  = (c != fcells[0]);
2381       ierr  = DMPlexPointLocalRef(dmFace, nface, fgeom, &fg);CHKERRQ(ierr);
2382       ierr  = DMPlexPointLocalRead(dmCell, ncell, cgeom, &cg1);CHKERRQ(ierr);
2383       for (d = 0; d < dim; ++d) dx[f*dim+d] = cg1->centroid[d] - cg->centroid[d];
2384       gref[f] = fg->grad[side];  /* Gradient reconstruction term will go here */
2385     }
2386     ierr = PetscFVComputeGradient(fvm, numFaces, dx, grad);CHKERRQ(ierr);
2387     for (f = 0; f < numFaces; ++f) {
2388       for (d = 0; d < dim; ++d) gref[f][d] = grad[f*dim+d];
2389     }
2390   }
2391   ierr = PetscFree3(dx, grad, gref);CHKERRQ(ierr);
2392   ierr = PetscSectionDestroy(&neighSec);CHKERRQ(ierr);
2393   ierr = PetscFree(neighbors);CHKERRQ(ierr);
2394   PetscFunctionReturn(0);
2395 }
2396 
2397 /*@
2398   DMPlexComputeGradientFVM - Compute geometric factors for gradient reconstruction, which are stored in the geometry data, and compute layout for gradient data
2399 
2400   Collective on DM
2401 
2402   Input Arguments:
2403 + dm  - The DM
2404 . fvm - The PetscFV
2405 . faceGeometry - The face geometry from DMPlexComputeFaceGeometryFVM()
2406 - cellGeometry - The face geometry from DMPlexComputeCellGeometryFVM()
2407 
2408   Output Parameters:
2409 + faceGeometry - The geometric factors for gradient calculation are inserted
2410 - dmGrad - The DM describing the layout of gradient data
2411 
2412   Level: developer
2413 
2414 .seealso: DMPlexGetFaceGeometryFVM(), DMPlexGetCellGeometryFVM()
2415 @*/
2416 PetscErrorCode DMPlexComputeGradientFVM(DM dm, PetscFV fvm, Vec faceGeometry, Vec cellGeometry, DM *dmGrad)
2417 {
2418   DM             dmFace, dmCell;
2419   PetscScalar   *fgeom, *cgeom;
2420   PetscSection   sectionGrad, parentSection;
2421   PetscInt       dim, pdim, cStart, cEnd, cEndInterior, c;
2422   PetscErrorCode ierr;
2423 
2424   PetscFunctionBegin;
2425   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
2426   ierr = PetscFVGetNumComponents(fvm, &pdim);CHKERRQ(ierr);
2427   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
2428   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr);
2429   /* Construct the interpolant corresponding to each face from the least-square solution over the cell neighborhood */
2430   ierr = VecGetDM(faceGeometry, &dmFace);CHKERRQ(ierr);
2431   ierr = VecGetDM(cellGeometry, &dmCell);CHKERRQ(ierr);
2432   ierr = VecGetArray(faceGeometry, &fgeom);CHKERRQ(ierr);
2433   ierr = VecGetArray(cellGeometry, &cgeom);CHKERRQ(ierr);
2434   ierr = DMPlexGetTree(dm,&parentSection,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
2435   if (!parentSection) {
2436     ierr = BuildGradientReconstruction_Internal(dm, fvm, dmFace, fgeom, dmCell, cgeom);CHKERRQ(ierr);
2437   } else {
2438     ierr = BuildGradientReconstruction_Internal_Tree(dm, fvm, dmFace, fgeom, dmCell, cgeom);CHKERRQ(ierr);
2439   }
2440   ierr = VecRestoreArray(faceGeometry, &fgeom);CHKERRQ(ierr);
2441   ierr = VecRestoreArray(cellGeometry, &cgeom);CHKERRQ(ierr);
2442   /* Create storage for gradients */
2443   ierr = DMClone(dm, dmGrad);CHKERRQ(ierr);
2444   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &sectionGrad);CHKERRQ(ierr);
2445   ierr = PetscSectionSetChart(sectionGrad, cStart, cEnd);CHKERRQ(ierr);
2446   for (c = cStart; c < cEnd; ++c) {ierr = PetscSectionSetDof(sectionGrad, c, pdim*dim);CHKERRQ(ierr);}
2447   ierr = PetscSectionSetUp(sectionGrad);CHKERRQ(ierr);
2448   ierr = DMSetDefaultSection(*dmGrad, sectionGrad);CHKERRQ(ierr);
2449   ierr = PetscSectionDestroy(&sectionGrad);CHKERRQ(ierr);
2450   PetscFunctionReturn(0);
2451 }
2452 
2453 /*@
2454   DMPlexGetDataFVM - Retrieve precomputed cell geometry
2455 
2456   Collective on DM
2457 
2458   Input Arguments:
2459 + dm  - The DM
2460 - fvm - The PetscFV
2461 
2462   Output Parameters:
2463 + cellGeometry - The cell geometry
2464 . faceGeometry - The face geometry
2465 - dmGrad       - The gradient matrices
2466 
2467   Level: developer
2468 
2469 .seealso: DMPlexComputeGeometryFVM()
2470 @*/
2471 PetscErrorCode DMPlexGetDataFVM(DM dm, PetscFV fv, Vec *cellgeom, Vec *facegeom, DM *gradDM)
2472 {
2473   PetscObject    cellgeomobj, facegeomobj;
2474   PetscErrorCode ierr;
2475 
2476   PetscFunctionBegin;
2477   ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_cellgeom_fvm", &cellgeomobj);CHKERRQ(ierr);
2478   if (!cellgeomobj) {
2479     Vec cellgeomInt, facegeomInt;
2480 
2481     ierr = DMPlexComputeGeometryFVM(dm, &cellgeomInt, &facegeomInt);CHKERRQ(ierr);
2482     ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_cellgeom_fvm",(PetscObject)cellgeomInt);CHKERRQ(ierr);
2483     ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_facegeom_fvm",(PetscObject)facegeomInt);CHKERRQ(ierr);
2484     ierr = VecDestroy(&cellgeomInt);CHKERRQ(ierr);
2485     ierr = VecDestroy(&facegeomInt);CHKERRQ(ierr);
2486     ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_cellgeom_fvm", &cellgeomobj);CHKERRQ(ierr);
2487   }
2488   ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_facegeom_fvm", &facegeomobj);CHKERRQ(ierr);
2489   if (cellgeom) *cellgeom = (Vec) cellgeomobj;
2490   if (facegeom) *facegeom = (Vec) facegeomobj;
2491   if (gradDM) {
2492     PetscObject gradobj;
2493     PetscBool   computeGradients;
2494 
2495     ierr = PetscFVGetComputeGradients(fv,&computeGradients);CHKERRQ(ierr);
2496     if (!computeGradients) {
2497       *gradDM = NULL;
2498       PetscFunctionReturn(0);
2499     }
2500     ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_dmgrad_fvm", &gradobj);CHKERRQ(ierr);
2501     if (!gradobj) {
2502       DM dmGradInt;
2503 
2504       ierr = DMPlexComputeGradientFVM(dm,fv,(Vec) facegeomobj,(Vec) cellgeomobj,&dmGradInt);CHKERRQ(ierr);
2505       ierr = PetscObjectCompose((PetscObject) dm, "DMPlex_dmgrad_fvm", (PetscObject)dmGradInt);CHKERRQ(ierr);
2506       ierr = DMDestroy(&dmGradInt);CHKERRQ(ierr);
2507       ierr = PetscObjectQuery((PetscObject) dm, "DMPlex_dmgrad_fvm", &gradobj);CHKERRQ(ierr);
2508     }
2509     *gradDM = (DM) gradobj;
2510   }
2511   PetscFunctionReturn(0);
2512 }
2513 
2514 static PetscErrorCode DMPlexCoordinatesToReference_NewtonUpdate(PetscInt dimC, PetscInt dimR, PetscScalar *J, PetscScalar *invJ, PetscScalar *work,  PetscReal *resNeg, PetscReal *guess)
2515 {
2516   PetscInt l, m;
2517 
2518   PetscFunctionBeginHot;
2519   if (dimC == dimR && dimR <= 3) {
2520     /* invert Jacobian, multiply */
2521     PetscScalar det, idet;
2522 
2523     switch (dimR) {
2524     case 1:
2525       invJ[0] = 1./ J[0];
2526       break;
2527     case 2:
2528       det = J[0] * J[3] - J[1] * J[2];
2529       idet = 1./det;
2530       invJ[0] =  J[3] * idet;
2531       invJ[1] = -J[1] * idet;
2532       invJ[2] = -J[2] * idet;
2533       invJ[3] =  J[0] * idet;
2534       break;
2535     case 3:
2536       {
2537         invJ[0] = J[4] * J[8] - J[5] * J[7];
2538         invJ[1] = J[2] * J[7] - J[1] * J[8];
2539         invJ[2] = J[1] * J[5] - J[2] * J[4];
2540         det = invJ[0] * J[0] + invJ[1] * J[3] + invJ[2] * J[6];
2541         idet = 1./det;
2542         invJ[0] *= idet;
2543         invJ[1] *= idet;
2544         invJ[2] *= idet;
2545         invJ[3]  = idet * (J[5] * J[6] - J[3] * J[8]);
2546         invJ[4]  = idet * (J[0] * J[8] - J[2] * J[6]);
2547         invJ[5]  = idet * (J[2] * J[3] - J[0] * J[5]);
2548         invJ[6]  = idet * (J[3] * J[7] - J[4] * J[6]);
2549         invJ[7]  = idet * (J[1] * J[6] - J[0] * J[7]);
2550         invJ[8]  = idet * (J[0] * J[4] - J[1] * J[3]);
2551       }
2552       break;
2553     }
2554     for (l = 0; l < dimR; l++) {
2555       for (m = 0; m < dimC; m++) {
2556         guess[l] += PetscRealPart(invJ[l * dimC + m]) * resNeg[m];
2557       }
2558     }
2559   } else {
2560 #if defined(PETSC_USE_COMPLEX)
2561     char transpose = 'C';
2562 #else
2563     char transpose = 'T';
2564 #endif
2565     PetscBLASInt m = dimR;
2566     PetscBLASInt n = dimC;
2567     PetscBLASInt one = 1;
2568     PetscBLASInt worksize = dimR * dimC, info;
2569 
2570     for (l = 0; l < dimC; l++) {invJ[l] = resNeg[l];}
2571 
2572     PetscStackCallBLAS("LAPACKgels",LAPACKgels_(&transpose,&m,&n,&one,J,&m,invJ,&n,work,&worksize, &info));
2573     if (info != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad argument to GELS");
2574 
2575     for (l = 0; l < dimR; l++) {guess[l] += PetscRealPart(invJ[l]);}
2576   }
2577   PetscFunctionReturn(0);
2578 }
2579 
2580 static PetscErrorCode DMPlexCoordinatesToReference_Tensor(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[], Vec coords, PetscInt dimC, PetscInt dimR)
2581 {
2582   PetscInt       coordSize, i, j, k, l, m, maxIts = 7, numV = (1 << dimR);
2583   PetscScalar    *coordsScalar = NULL;
2584   PetscReal      *cellData, *cellCoords, *cellCoeffs, *extJ, *resNeg;
2585   PetscScalar    *J, *invJ, *work;
2586   PetscErrorCode ierr;
2587 
2588   PetscFunctionBegin;
2589   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2590   ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr);
2591   if (coordSize < dimC * numV) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expecting at least %D coordinates, got %D",dimC * (1 << dimR), coordSize);
2592   ierr = DMGetWorkArray(dm, 2 * coordSize + dimR + dimC, MPIU_REAL, &cellData);CHKERRQ(ierr);
2593   ierr = DMGetWorkArray(dm, 3 * dimR * dimC, MPIU_SCALAR, &J);CHKERRQ(ierr);
2594   cellCoords = &cellData[0];
2595   cellCoeffs = &cellData[coordSize];
2596   extJ       = &cellData[2 * coordSize];
2597   resNeg     = &cellData[2 * coordSize + dimR];
2598   invJ       = &J[dimR * dimC];
2599   work       = &J[2 * dimR * dimC];
2600   if (dimR == 2) {
2601     const PetscInt zToPlex[4] = {0, 1, 3, 2};
2602 
2603     for (i = 0; i < 4; i++) {
2604       PetscInt plexI = zToPlex[i];
2605 
2606       for (j = 0; j < dimC; j++) {
2607         cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]);
2608       }
2609     }
2610   } else if (dimR == 3) {
2611     const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6};
2612 
2613     for (i = 0; i < 8; i++) {
2614       PetscInt plexI = zToPlex[i];
2615 
2616       for (j = 0; j < dimC; j++) {
2617         cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]);
2618       }
2619     }
2620   } else {
2621     for (i = 0; i < coordSize; i++) {cellCoords[i] = PetscRealPart(coordsScalar[i]);}
2622   }
2623   /* Perform the shuffling transform that converts values at the corners of [-1,1]^d to coefficients */
2624   for (i = 0; i < dimR; i++) {
2625     PetscReal *swap;
2626 
2627     for (j = 0; j < (numV / 2); j++) {
2628       for (k = 0; k < dimC; k++) {
2629         cellCoeffs[dimC * j + k]                = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] + cellCoords[dimC * 2 * j + k]);
2630         cellCoeffs[dimC * (j + (numV / 2)) + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] - cellCoords[dimC * 2 * j + k]);
2631       }
2632     }
2633 
2634     if (i < dimR - 1) {
2635       swap = cellCoeffs;
2636       cellCoeffs = cellCoords;
2637       cellCoords = swap;
2638     }
2639   }
2640   ierr = PetscMemzero(refCoords,numPoints * dimR * sizeof (PetscReal));CHKERRQ(ierr);
2641   for (j = 0; j < numPoints; j++) {
2642     for (i = 0; i < maxIts; i++) {
2643       PetscReal *guess = &refCoords[dimR * j];
2644 
2645       /* compute -residual and Jacobian */
2646       for (k = 0; k < dimC; k++) {resNeg[k] = realCoords[dimC * j + k];}
2647       for (k = 0; k < dimC * dimR; k++) {J[k] = 0.;}
2648       for (k = 0; k < numV; k++) {
2649         PetscReal extCoord = 1.;
2650         for (l = 0; l < dimR; l++) {
2651           PetscReal coord = guess[l];
2652           PetscInt  dep   = (k & (1 << l)) >> l;
2653 
2654           extCoord *= dep * coord + !dep;
2655           extJ[l] = dep;
2656 
2657           for (m = 0; m < dimR; m++) {
2658             PetscReal coord = guess[m];
2659             PetscInt  dep   = ((k & (1 << m)) >> m) && (m != l);
2660             PetscReal mult  = dep * coord + !dep;
2661 
2662             extJ[l] *= mult;
2663           }
2664         }
2665         for (l = 0; l < dimC; l++) {
2666           PetscReal coeff = cellCoeffs[dimC * k + l];
2667 
2668           resNeg[l] -= coeff * extCoord;
2669           for (m = 0; m < dimR; m++) {
2670             J[dimR * l + m] += coeff * extJ[m];
2671           }
2672         }
2673       }
2674 #if 0 && defined(PETSC_USE_DEBUG)
2675       {
2676         PetscReal maxAbs = 0.;
2677 
2678         for (l = 0; l < dimC; l++) {
2679           maxAbs = PetscMax(maxAbs,PetscAbsReal(resNeg[l]));
2680         }
2681         ierr = PetscInfo4(dm,"cell %D, point %D, iter %D: res %g\n",cell,j,i,maxAbs);CHKERRQ(ierr);
2682       }
2683 #endif
2684 
2685       ierr = DMPlexCoordinatesToReference_NewtonUpdate(dimC,dimR,J,invJ,work,resNeg,guess);CHKERRQ(ierr);
2686     }
2687   }
2688   ierr = DMRestoreWorkArray(dm, 3 * dimR * dimC, MPIU_SCALAR, &J);CHKERRQ(ierr);
2689   ierr = DMRestoreWorkArray(dm, 2 * coordSize + dimR + dimC, MPIU_REAL, &cellData);CHKERRQ(ierr);
2690   ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr);
2691   PetscFunctionReturn(0);
2692 }
2693 
2694 static PetscErrorCode DMPlexReferenceToCoordinates_Tensor(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[], Vec coords, PetscInt dimC, PetscInt dimR)
2695 {
2696   PetscInt       coordSize, i, j, k, l, numV = (1 << dimR);
2697   PetscScalar    *coordsScalar = NULL;
2698   PetscReal      *cellData, *cellCoords, *cellCoeffs;
2699   PetscErrorCode ierr;
2700 
2701   PetscFunctionBegin;
2702   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2703   ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr);
2704   if (coordSize < dimC * numV) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Expecting at least %D coordinates, got %D",dimC * (1 << dimR), coordSize);
2705   ierr = DMGetWorkArray(dm, 2 * coordSize, MPIU_REAL, &cellData);CHKERRQ(ierr);
2706   cellCoords = &cellData[0];
2707   cellCoeffs = &cellData[coordSize];
2708   if (dimR == 2) {
2709     const PetscInt zToPlex[4] = {0, 1, 3, 2};
2710 
2711     for (i = 0; i < 4; i++) {
2712       PetscInt plexI = zToPlex[i];
2713 
2714       for (j = 0; j < dimC; j++) {
2715         cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]);
2716       }
2717     }
2718   } else if (dimR == 3) {
2719     const PetscInt zToPlex[8] = {0, 3, 1, 2, 4, 5, 7, 6};
2720 
2721     for (i = 0; i < 8; i++) {
2722       PetscInt plexI = zToPlex[i];
2723 
2724       for (j = 0; j < dimC; j++) {
2725         cellCoords[dimC * i + j] = PetscRealPart(coordsScalar[dimC * plexI + j]);
2726       }
2727     }
2728   } else {
2729     for (i = 0; i < coordSize; i++) {cellCoords[i] = PetscRealPart(coordsScalar[i]);}
2730   }
2731   /* Perform the shuffling transform that converts values at the corners of [-1,1]^d to coefficients */
2732   for (i = 0; i < dimR; i++) {
2733     PetscReal *swap;
2734 
2735     for (j = 0; j < (numV / 2); j++) {
2736       for (k = 0; k < dimC; k++) {
2737         cellCoeffs[dimC * j + k]                = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] + cellCoords[dimC * 2 * j + k]);
2738         cellCoeffs[dimC * (j + (numV / 2)) + k] = 0.5 * (cellCoords[dimC * (2 * j + 1) + k] - cellCoords[dimC * 2 * j + k]);
2739       }
2740     }
2741 
2742     if (i < dimR - 1) {
2743       swap = cellCoeffs;
2744       cellCoeffs = cellCoords;
2745       cellCoords = swap;
2746     }
2747   }
2748   ierr = PetscMemzero(realCoords,numPoints * dimC * sizeof (PetscReal));CHKERRQ(ierr);
2749   for (j = 0; j < numPoints; j++) {
2750     const PetscReal *guess  = &refCoords[dimR * j];
2751     PetscReal       *mapped = &realCoords[dimC * j];
2752 
2753     for (k = 0; k < numV; k++) {
2754       PetscReal extCoord = 1.;
2755       for (l = 0; l < dimR; l++) {
2756         PetscReal coord = guess[l];
2757         PetscInt  dep   = (k & (1 << l)) >> l;
2758 
2759         extCoord *= dep * coord + !dep;
2760       }
2761       for (l = 0; l < dimC; l++) {
2762         PetscReal coeff = cellCoeffs[dimC * k + l];
2763 
2764         mapped[l] += coeff * extCoord;
2765       }
2766     }
2767   }
2768   ierr = DMRestoreWorkArray(dm, 2 * coordSize, MPIU_REAL, &cellData);CHKERRQ(ierr);
2769   ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &coordsScalar);CHKERRQ(ierr);
2770   PetscFunctionReturn(0);
2771 }
2772 
2773 /* TODO: TOBY please fix this for Nc > 1 */
2774 static PetscErrorCode DMPlexCoordinatesToReference_FE(DM dm, PetscFE fe, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[], Vec coords, PetscInt Nc, PetscInt dimR)
2775 {
2776   PetscInt       numComp, pdim, i, j, k, l, m, maxIter = 7, coordSize;
2777   PetscScalar    *nodes = NULL;
2778   PetscReal      *invV, *modes;
2779   PetscReal      *B, *D, *resNeg;
2780   PetscScalar    *J, *invJ, *work;
2781   PetscErrorCode ierr;
2782 
2783   PetscFunctionBegin;
2784   ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr);
2785   ierr = PetscFEGetNumComponents(fe, &numComp);CHKERRQ(ierr);
2786   if (numComp != Nc) SETERRQ2(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"coordinate discretization must have as many components (%D) as embedding dimension (!= %D)",numComp,Nc);
2787   ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr);
2788   /* convert nodes to values in the stable evaluation basis */
2789   ierr = DMGetWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr);
2790   invV = fe->invV;
2791   for (i = 0; i < pdim; ++i) {
2792     modes[i] = 0.;
2793     for (j = 0; j < pdim; ++j) {
2794       modes[i] += invV[i * pdim + j] * PetscRealPart(nodes[j]);
2795     }
2796   }
2797   ierr   = DMGetWorkArray(dm,pdim * Nc + pdim * Nc * dimR + Nc,MPIU_REAL,&B);CHKERRQ(ierr);
2798   D      = &B[pdim*Nc];
2799   resNeg = &D[pdim*Nc * dimR];
2800   ierr = DMGetWorkArray(dm,3 * Nc * dimR,MPIU_SCALAR,&J);CHKERRQ(ierr);
2801   invJ = &J[Nc * dimR];
2802   work = &invJ[Nc * dimR];
2803   for (i = 0; i < numPoints * dimR; i++) {refCoords[i] = 0.;}
2804   for (j = 0; j < numPoints; j++) {
2805       for (i = 0; i < maxIter; i++) { /* we could batch this so that we're not making big B and D arrays all the time */
2806       PetscReal *guess = &refCoords[j * dimR];
2807       ierr = PetscSpaceEvaluate(fe->basisSpace, 1, guess, B, D, NULL);CHKERRQ(ierr);
2808       for (k = 0; k < Nc; k++) {resNeg[k] = realCoords[j * Nc + k];}
2809       for (k = 0; k < Nc * dimR; k++) {J[k] = 0.;}
2810       for (k = 0; k < pdim; k++) {
2811         for (l = 0; l < Nc; l++) {
2812           resNeg[l] -= modes[k] * B[k * Nc + l];
2813           for (m = 0; m < dimR; m++) {
2814             J[l * dimR + m] += modes[k] * D[(k * Nc + l) * dimR + m];
2815           }
2816         }
2817       }
2818 #if 0 && defined(PETSC_USE_DEBUG)
2819       {
2820         PetscReal maxAbs = 0.;
2821 
2822         for (l = 0; l < Nc; l++) {
2823           maxAbs = PetscMax(maxAbs,PetscAbsReal(resNeg[l]));
2824         }
2825         ierr = PetscInfo4(dm,"cell %D, point %D, iter %D: res %g\n",cell,j,i,maxAbs);CHKERRQ(ierr);
2826       }
2827 #endif
2828       ierr = DMPlexCoordinatesToReference_NewtonUpdate(Nc,dimR,J,invJ,work,resNeg,guess);CHKERRQ(ierr);
2829     }
2830   }
2831   ierr = DMRestoreWorkArray(dm,3 * Nc * dimR,MPIU_SCALAR,&J);CHKERRQ(ierr);
2832   ierr = DMRestoreWorkArray(dm,pdim * Nc + pdim * Nc * dimR + Nc,MPIU_REAL,&B);CHKERRQ(ierr);
2833   ierr = DMRestoreWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr);
2834   ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr);
2835   PetscFunctionReturn(0);
2836 }
2837 
2838 /* TODO: TOBY please fix this for Nc > 1 */
2839 static PetscErrorCode DMPlexReferenceToCoordinates_FE(DM dm, PetscFE fe, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[], Vec coords, PetscInt Nc, PetscInt dimR)
2840 {
2841   PetscInt       numComp, pdim, i, j, k, l, coordSize;
2842   PetscScalar    *nodes = NULL;
2843   PetscReal      *invV, *modes;
2844   PetscReal      *B;
2845   PetscErrorCode ierr;
2846 
2847   PetscFunctionBegin;
2848   ierr = PetscFEGetDimension(fe, &pdim);CHKERRQ(ierr);
2849   ierr = PetscFEGetNumComponents(fe, &numComp);CHKERRQ(ierr);
2850   if (numComp != Nc) SETERRQ2(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"coordinate discretization must have as many components (%D) as embedding dimension (!= %D)",numComp,Nc);
2851   ierr = DMPlexVecGetClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr);
2852   /* convert nodes to values in the stable evaluation basis */
2853   ierr = DMGetWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr);
2854   invV = fe->invV;
2855   for (i = 0; i < pdim; ++i) {
2856     modes[i] = 0.;
2857     for (j = 0; j < pdim; ++j) {
2858       modes[i] += invV[i * pdim + j] * PetscRealPart(nodes[j]);
2859     }
2860   }
2861   ierr = DMGetWorkArray(dm,numPoints * pdim * Nc,MPIU_REAL,&B);CHKERRQ(ierr);
2862   ierr = PetscSpaceEvaluate(fe->basisSpace, numPoints, refCoords, B, NULL, NULL);CHKERRQ(ierr);
2863   for (i = 0; i < numPoints * Nc; i++) {realCoords[i] = 0.;}
2864   for (j = 0; j < numPoints; j++) {
2865     PetscReal *mapped = &realCoords[j * Nc];
2866 
2867     for (k = 0; k < pdim; k++) {
2868       for (l = 0; l < Nc; l++) {
2869         mapped[l] += modes[k] * B[(j * pdim + k) * Nc + l];
2870       }
2871     }
2872   }
2873   ierr = DMRestoreWorkArray(dm,numPoints * pdim * Nc,MPIU_REAL,&B);CHKERRQ(ierr);
2874   ierr = DMRestoreWorkArray(dm,pdim,MPIU_REAL,&modes);CHKERRQ(ierr);
2875   ierr = DMPlexVecRestoreClosure(dm, NULL, coords, cell, &coordSize, &nodes);CHKERRQ(ierr);
2876   PetscFunctionReturn(0);
2877 }
2878 
2879 /*@
2880   DMPlexCoordinatesToReference - Pull coordinates back from the mesh to the reference element using a single element
2881   map.  This inversion will be accurate inside the reference element, but may be inaccurate for mappings that do not
2882   extend uniquely outside the reference cell (e.g, most non-affine maps)
2883 
2884   Not collective
2885 
2886   Input Parameters:
2887 + dm         - The mesh, with coordinate maps defined either by a PetscDS for the coordinate DM (see DMGetCoordinateDM()) or
2888                implicitly by the coordinates of the corner vertices of the cell: as an affine map for simplicial elements, or
2889                as a multilinear map for tensor-product elements
2890 . cell       - the cell whose map is used.
2891 . numPoints  - the number of points to locate
2892 - realCoords - (numPoints x coordinate dimension) array of coordinates (see DMGetCoordinateDim())
2893 
2894   Output Parameters:
2895 . refCoords  - (numPoints x dimension) array of reference coordinates (see DMGetDimension())
2896 
2897   Level: intermediate
2898 @*/
2899 PetscErrorCode DMPlexCoordinatesToReference(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal realCoords[], PetscReal refCoords[])
2900 {
2901   PetscInt       dimC, dimR, depth, cStart, cEnd, cEndInterior, i;
2902   DM             coordDM = NULL;
2903   Vec            coords;
2904   PetscFE        fe = NULL;
2905   PetscErrorCode ierr;
2906 
2907   PetscFunctionBegin;
2908   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2909   ierr = DMGetDimension(dm,&dimR);CHKERRQ(ierr);
2910   ierr = DMGetCoordinateDim(dm,&dimC);CHKERRQ(ierr);
2911   if (dimR <= 0 || dimC <= 0 || numPoints <= 0) PetscFunctionReturn(0);
2912   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
2913   ierr = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr);
2914   ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr);
2915   if (coordDM) {
2916     PetscInt coordFields;
2917 
2918     ierr = DMGetNumFields(coordDM,&coordFields);CHKERRQ(ierr);
2919     if (coordFields) {
2920       PetscClassId id;
2921       PetscObject  disc;
2922 
2923       ierr = DMGetField(coordDM,0,&disc);CHKERRQ(ierr);
2924       ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
2925       if (id == PETSCFE_CLASSID) {
2926         fe = (PetscFE) disc;
2927       }
2928     }
2929   }
2930   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
2931   ierr = DMPlexGetHybridBounds(dm,&cEndInterior,NULL,NULL,NULL);CHKERRQ(ierr);
2932   cEnd = cEndInterior > 0 ? cEndInterior : cEnd;
2933   if (cell < cStart || cell >= cEnd) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"point %D not in cell range [%D,%D)",cell,cStart,cEnd);
2934   if (!fe) { /* implicit discretization: affine or multilinear */
2935     PetscInt  coneSize;
2936     PetscBool isSimplex, isTensor;
2937 
2938     ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr);
2939     isSimplex = (coneSize == (dimR + 1)) ? PETSC_TRUE : PETSC_FALSE;
2940     isTensor  = (coneSize == ((depth == 1) ? (1 << dimR) : (2 * dimR))) ? PETSC_TRUE : PETSC_FALSE;
2941     if (isSimplex) {
2942       PetscReal detJ, *v0, *J, *invJ;
2943 
2944       ierr = DMGetWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr);
2945       J    = &v0[dimC];
2946       invJ = &J[dimC * dimC];
2947       ierr = DMPlexComputeCellGeometryAffineFEM(dm, cell, v0, J, invJ, &detJ);CHKERRQ(ierr);
2948       for (i = 0; i < numPoints; i++) { /* Apply the inverse affine transformation for each point */
2949         CoordinatesRealToRef(dimC, dimR, v0, invJ, &realCoords[dimC * i], &refCoords[dimR * i]);
2950       }
2951       ierr = DMRestoreWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr);
2952     } else if (isTensor) {
2953       ierr = DMPlexCoordinatesToReference_Tensor(coordDM, cell, numPoints, realCoords, refCoords, coords, dimC, dimR);CHKERRQ(ierr);
2954     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unrecognized cone size %D",coneSize);
2955   } else {
2956     ierr = DMPlexCoordinatesToReference_FE(coordDM, fe, cell, numPoints, realCoords, refCoords, coords, dimC, dimR);CHKERRQ(ierr);
2957   }
2958   PetscFunctionReturn(0);
2959 }
2960 
2961 /*@
2962   DMPlexReferenceToCoordinates - Map references coordinates to coordinates in the the mesh for a single element map.
2963 
2964   Not collective
2965 
2966   Input Parameters:
2967 + dm         - The mesh, with coordinate maps defined either by a PetscDS for the coordinate DM (see DMGetCoordinateDM()) or
2968                implicitly by the coordinates of the corner vertices of the cell: as an affine map for simplicial elements, or
2969                as a multilinear map for tensor-product elements
2970 . cell       - the cell whose map is used.
2971 . numPoints  - the number of points to locate
2972 + refCoords  - (numPoints x dimension) array of reference coordinates (see DMGetDimension())
2973 
2974   Output Parameters:
2975 . realCoords - (numPoints x coordinate dimension) array of coordinates (see DMGetCoordinateDim())
2976 
2977    Level: intermediate
2978 @*/
2979 PetscErrorCode DMPlexReferenceToCoordinates(DM dm, PetscInt cell, PetscInt numPoints, const PetscReal refCoords[], PetscReal realCoords[])
2980 {
2981   PetscInt       dimC, dimR, depth, cStart, cEnd, cEndInterior, i;
2982   DM             coordDM = NULL;
2983   Vec            coords;
2984   PetscFE        fe = NULL;
2985   PetscErrorCode ierr;
2986 
2987   PetscFunctionBegin;
2988   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2989   ierr = DMGetDimension(dm,&dimR);CHKERRQ(ierr);
2990   ierr = DMGetCoordinateDim(dm,&dimC);CHKERRQ(ierr);
2991   if (dimR <= 0 || dimC <= 0 || numPoints <= 0) PetscFunctionReturn(0);
2992   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
2993   ierr = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr);
2994   ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr);
2995   if (coordDM) {
2996     PetscInt coordFields;
2997 
2998     ierr = DMGetNumFields(coordDM,&coordFields);CHKERRQ(ierr);
2999     if (coordFields) {
3000       PetscClassId id;
3001       PetscObject  disc;
3002 
3003       ierr = DMGetField(coordDM,0,&disc);CHKERRQ(ierr);
3004       ierr = PetscObjectGetClassId(disc,&id);CHKERRQ(ierr);
3005       if (id == PETSCFE_CLASSID) {
3006         fe = (PetscFE) disc;
3007       }
3008     }
3009   }
3010   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
3011   ierr = DMPlexGetHybridBounds(dm,&cEndInterior,NULL,NULL,NULL);CHKERRQ(ierr);
3012   cEnd = cEndInterior > 0 ? cEndInterior : cEnd;
3013   if (cell < cStart || cell >= cEnd) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"point %D not in cell range [%D,%D)",cell,cStart,cEnd);
3014   if (!fe) { /* implicit discretization: affine or multilinear */
3015     PetscInt  coneSize;
3016     PetscBool isSimplex, isTensor;
3017 
3018     ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr);
3019     isSimplex = (coneSize == (dimR + 1)) ? PETSC_TRUE : PETSC_FALSE;
3020     isTensor  = (coneSize == ((depth == 1) ? (1 << dimR) : (2 * dimR))) ? PETSC_TRUE : PETSC_FALSE;
3021     if (isSimplex) {
3022       PetscReal detJ, *v0, *J;
3023 
3024       ierr = DMGetWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr);
3025       J    = &v0[dimC];
3026       ierr = DMPlexComputeCellGeometryAffineFEM(dm, cell, v0, J, NULL, &detJ);CHKERRQ(ierr);
3027       for (i = 0; i < numPoints; i++) { /* Apply the inverse affine transformation for each point */
3028         CoordinatesRefToReal(dimC, dimR, v0, J, &refCoords[dimR * i], &realCoords[dimC * i]);
3029       }
3030       ierr = DMRestoreWorkArray(dm,dimC + 2 * dimC * dimC, MPIU_REAL, &v0);CHKERRQ(ierr);
3031     } else if (isTensor) {
3032       ierr = DMPlexReferenceToCoordinates_Tensor(coordDM, cell, numPoints, refCoords, realCoords, coords, dimC, dimR);CHKERRQ(ierr);
3033     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unrecognized cone size %D",coneSize);
3034   } else {
3035     ierr = DMPlexReferenceToCoordinates_FE(coordDM, fe, cell, numPoints, refCoords, realCoords, coords, dimC, dimR);CHKERRQ(ierr);
3036   }
3037   PetscFunctionReturn(0);
3038 }
3039