xref: /petsc/src/dm/impls/plex/tests/ex3.c (revision 8e3a2eeff93b8092aa720bf50e8b48bf277d7f13)
1c4762a1bSJed Brown static char help[] = "Check that a DM can accurately represent and interpolate functions of a given polynomial order\n\n";
2c4762a1bSJed Brown 
3c4762a1bSJed Brown #include <petscdmplex.h>
4c4762a1bSJed Brown #include <petscdm.h>
5c4762a1bSJed Brown #include <petscdmda.h>
6c4762a1bSJed Brown #include <petscfe.h>
7c4762a1bSJed Brown #include <petscds.h>
8c4762a1bSJed Brown #include <petscksp.h>
9c4762a1bSJed Brown #include <petscsnes.h>
10c4762a1bSJed Brown 
11c4762a1bSJed Brown typedef struct {
12c4762a1bSJed Brown   /* Domain and mesh definition */
13c4762a1bSJed Brown   PetscBool useDA;             /* Flag DMDA tensor product mesh */
14c4762a1bSJed Brown   PetscBool shearCoords;       /* Flag for shear transform */
15c4762a1bSJed Brown   PetscBool nonaffineCoords;   /* Flag for non-affine transform */
16c4762a1bSJed Brown   /* Element definition */
17c4762a1bSJed Brown   PetscInt  qorder;            /* Order of the quadrature */
18c4762a1bSJed Brown   PetscInt  numComponents;     /* Number of field components */
19c4762a1bSJed Brown   PetscFE   fe;                /* The finite element */
20c4762a1bSJed Brown   /* Testing space */
21c4762a1bSJed Brown   PetscInt  porder;            /* Order of polynomials to test */
22c4762a1bSJed Brown   PetscBool convergence;       /* Test for order of convergence */
23c4762a1bSJed Brown   PetscBool convRefine;        /* Test for convergence using refinement, otherwise use coarsening */
24c4762a1bSJed Brown   PetscBool constraints;       /* Test local constraints */
25c4762a1bSJed Brown   PetscBool tree;              /* Test tree routines */
26c4762a1bSJed Brown   PetscBool testFEjacobian;    /* Test finite element Jacobian assembly */
27c4762a1bSJed Brown   PetscBool testFVgrad;        /* Test finite difference gradient routine */
28c4762a1bSJed Brown   PetscBool testInjector;      /* Test finite element injection routines */
29c4762a1bSJed Brown   PetscInt  treeCell;          /* Cell to refine in tree test */
30c4762a1bSJed Brown   PetscReal constants[3];      /* Constant values for each dimension */
31c4762a1bSJed Brown } AppCtx;
32c4762a1bSJed Brown 
33c4762a1bSJed Brown /* u = 1 */
34c4762a1bSJed Brown PetscErrorCode constant(PetscInt dim, PetscReal time, const PetscReal coords[], PetscInt Nf, PetscScalar *u, void *ctx)
35c4762a1bSJed Brown {
36c4762a1bSJed Brown   AppCtx   *user = (AppCtx *) ctx;
37c4762a1bSJed Brown   PetscInt d;
3830602db0SMatthew G. Knepley   for (d = 0; d < dim; ++d) u[d] = user->constants[d];
39c4762a1bSJed Brown   return 0;
40c4762a1bSJed Brown }
41c4762a1bSJed Brown PetscErrorCode constantDer(PetscInt dim, PetscReal time, const PetscReal coords[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx)
42c4762a1bSJed Brown {
43c4762a1bSJed Brown   PetscInt d;
4430602db0SMatthew G. Knepley   for (d = 0; d < dim; ++d) u[d] = 0.0;
45c4762a1bSJed Brown   return 0;
46c4762a1bSJed Brown }
47c4762a1bSJed Brown 
48c4762a1bSJed Brown /* u = x */
49c4762a1bSJed Brown PetscErrorCode linear(PetscInt dim, PetscReal time, const PetscReal coords[], PetscInt Nf, PetscScalar *u, void *ctx)
50c4762a1bSJed Brown {
51c4762a1bSJed Brown   PetscInt d;
52c4762a1bSJed Brown   for (d = 0; d < dim; ++d) u[d] = coords[d];
53c4762a1bSJed Brown   return 0;
54c4762a1bSJed Brown }
55c4762a1bSJed Brown PetscErrorCode linearDer(PetscInt dim, PetscReal time, const PetscReal coords[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx)
56c4762a1bSJed Brown {
57c4762a1bSJed Brown   PetscInt d, e;
58c4762a1bSJed Brown   for (d = 0; d < dim; ++d) {
59c4762a1bSJed Brown     u[d] = 0.0;
60c4762a1bSJed Brown     for (e = 0; e < dim; ++e) u[d] += (d == e ? 1.0 : 0.0) * n[e];
61c4762a1bSJed Brown   }
62c4762a1bSJed Brown   return 0;
63c4762a1bSJed Brown }
64c4762a1bSJed Brown 
65c4762a1bSJed Brown /* u = x^2 or u = (x^2, xy) or u = (xy, yz, zx) */
66c4762a1bSJed Brown PetscErrorCode quadratic(PetscInt dim, PetscReal time, const PetscReal coords[], PetscInt Nf, PetscScalar *u, void *ctx)
67c4762a1bSJed Brown {
6830602db0SMatthew G. Knepley   if (dim > 2)      {u[0] = coords[0]*coords[1]; u[1] = coords[1]*coords[2]; u[2] = coords[2]*coords[0];}
6930602db0SMatthew G. Knepley   else if (dim > 1) {u[0] = coords[0]*coords[0]; u[1] = coords[0]*coords[1];}
7030602db0SMatthew G. Knepley   else if (dim > 0) {u[0] = coords[0]*coords[0];}
71c4762a1bSJed Brown   return 0;
72c4762a1bSJed Brown }
73c4762a1bSJed Brown PetscErrorCode quadraticDer(PetscInt dim, PetscReal time, const PetscReal coords[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx)
74c4762a1bSJed Brown {
7530602db0SMatthew G. Knepley   if (dim > 2)      {u[0] = coords[1]*n[0] + coords[0]*n[1]; u[1] = coords[2]*n[1] + coords[1]*n[2]; u[2] = coords[2]*n[0] + coords[0]*n[2];}
7630602db0SMatthew G. Knepley   else if (dim > 1) {u[0] = 2.0*coords[0]*n[0]; u[1] = coords[1]*n[0] + coords[0]*n[1];}
7730602db0SMatthew G. Knepley   else if (dim > 0) {u[0] = 2.0*coords[0]*n[0];}
78c4762a1bSJed Brown   return 0;
79c4762a1bSJed Brown }
80c4762a1bSJed Brown 
81c4762a1bSJed Brown /* u = x^3 or u = (x^3, x^2y) or u = (x^2y, y^2z, z^2x) */
82c4762a1bSJed Brown PetscErrorCode cubic(PetscInt dim, PetscReal time, const PetscReal coords[], PetscInt Nf, PetscScalar *u, void *ctx)
83c4762a1bSJed Brown {
8430602db0SMatthew G. Knepley   if (dim > 2)      {u[0] = coords[0]*coords[0]*coords[1]; u[1] = coords[1]*coords[1]*coords[2]; u[2] = coords[2]*coords[2]*coords[0];}
8530602db0SMatthew G. Knepley   else if (dim > 1) {u[0] = coords[0]*coords[0]*coords[0]; u[1] = coords[0]*coords[0]*coords[1];}
8630602db0SMatthew G. Knepley   else if (dim > 0) {u[0] = coords[0]*coords[0]*coords[0];}
87c4762a1bSJed Brown   return 0;
88c4762a1bSJed Brown }
89c4762a1bSJed Brown PetscErrorCode cubicDer(PetscInt dim, PetscReal time, const PetscReal coords[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx)
90c4762a1bSJed Brown {
9130602db0SMatthew G. Knepley   if (dim > 2)      {u[0] = 2.0*coords[0]*coords[1]*n[0] + coords[0]*coords[0]*n[1]; u[1] = 2.0*coords[1]*coords[2]*n[1] + coords[1]*coords[1]*n[2]; u[2] = 2.0*coords[2]*coords[0]*n[2] + coords[2]*coords[2]*n[0];}
9230602db0SMatthew G. Knepley   else if (dim > 1) {u[0] = 3.0*coords[0]*coords[0]*n[0]; u[1] = 2.0*coords[0]*coords[1]*n[0] + coords[0]*coords[0]*n[1];}
9330602db0SMatthew G. Knepley   else if (dim > 0) {u[0] = 3.0*coords[0]*coords[0]*n[0];}
94c4762a1bSJed Brown   return 0;
95c4762a1bSJed Brown }
96c4762a1bSJed Brown 
97c4762a1bSJed Brown /* u = tanh(x) */
98c4762a1bSJed Brown PetscErrorCode trig(PetscInt dim, PetscReal time, const PetscReal coords[], PetscInt Nf, PetscScalar *u, void *ctx)
99c4762a1bSJed Brown {
100c4762a1bSJed Brown   PetscInt d;
10130602db0SMatthew G. Knepley   for (d = 0; d < dim; ++d) u[d] = PetscTanhReal(coords[d] - 0.5);
102c4762a1bSJed Brown   return 0;
103c4762a1bSJed Brown }
104c4762a1bSJed Brown PetscErrorCode trigDer(PetscInt dim, PetscReal time, const PetscReal coords[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx)
105c4762a1bSJed Brown {
106c4762a1bSJed Brown   PetscInt d;
10730602db0SMatthew G. Knepley   for (d = 0; d < dim; ++d) u[d] = 1.0/PetscSqr(PetscCoshReal(coords[d] - 0.5)) * n[d];
108c4762a1bSJed Brown   return 0;
109c4762a1bSJed Brown }
110c4762a1bSJed Brown 
111c4762a1bSJed Brown static PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options)
112c4762a1bSJed Brown {
113c4762a1bSJed Brown   PetscInt       n = 3;
114c4762a1bSJed Brown   PetscErrorCode ierr;
115c4762a1bSJed Brown 
116c4762a1bSJed Brown   PetscFunctionBeginUser;
11730602db0SMatthew G. Knepley   options->useDA           = PETSC_FALSE;
118c4762a1bSJed Brown   options->shearCoords     = PETSC_FALSE;
119c4762a1bSJed Brown   options->nonaffineCoords = PETSC_FALSE;
120c4762a1bSJed Brown   options->qorder          = 0;
121c4762a1bSJed Brown   options->numComponents   = PETSC_DEFAULT;
122c4762a1bSJed Brown   options->porder          = 0;
123c4762a1bSJed Brown   options->convergence     = PETSC_FALSE;
124c4762a1bSJed Brown   options->convRefine      = PETSC_TRUE;
125c4762a1bSJed Brown   options->constraints     = PETSC_FALSE;
126c4762a1bSJed Brown   options->tree            = PETSC_FALSE;
127c4762a1bSJed Brown   options->treeCell        = 0;
128c4762a1bSJed Brown   options->testFEjacobian  = PETSC_FALSE;
129c4762a1bSJed Brown   options->testFVgrad      = PETSC_FALSE;
130c4762a1bSJed Brown   options->testInjector    = PETSC_FALSE;
131c4762a1bSJed Brown   options->constants[0]    = 1.0;
132c4762a1bSJed Brown   options->constants[1]    = 2.0;
133c4762a1bSJed Brown   options->constants[2]    = 3.0;
134c4762a1bSJed Brown 
135c4762a1bSJed Brown   ierr = PetscOptionsBegin(comm, "", "Projection Test Options", "DMPlex");CHKERRQ(ierr);
136c4762a1bSJed Brown   ierr = PetscOptionsBool("-use_da", "Flag for DMDA mesh", "ex3.c", options->useDA, &options->useDA, NULL);CHKERRQ(ierr);
137c4762a1bSJed Brown   ierr = PetscOptionsBool("-shear_coords", "Transform coordinates with a shear", "ex3.c", options->shearCoords, &options->shearCoords, NULL);CHKERRQ(ierr);
138c4762a1bSJed Brown   ierr = PetscOptionsBool("-non_affine_coords", "Transform coordinates with a non-affine transform", "ex3.c", options->nonaffineCoords, &options->nonaffineCoords, NULL);CHKERRQ(ierr);
139c4762a1bSJed Brown   ierr = PetscOptionsBoundedInt("-qorder", "The quadrature order", "ex3.c", options->qorder, &options->qorder, NULL,0);CHKERRQ(ierr);
140c4762a1bSJed Brown   ierr = PetscOptionsBoundedInt("-num_comp", "The number of field components", "ex3.c", options->numComponents, &options->numComponents, NULL,PETSC_DEFAULT);CHKERRQ(ierr);
141c4762a1bSJed Brown   ierr = PetscOptionsBoundedInt("-porder", "The order of polynomials to test", "ex3.c", options->porder, &options->porder, NULL,0);CHKERRQ(ierr);
142c4762a1bSJed Brown   ierr = PetscOptionsBool("-convergence", "Check the convergence rate", "ex3.c", options->convergence, &options->convergence, NULL);CHKERRQ(ierr);
143c4762a1bSJed Brown   ierr = PetscOptionsBool("-conv_refine", "Use refinement for the convergence rate", "ex3.c", options->convRefine, &options->convRefine, NULL);CHKERRQ(ierr);
144c4762a1bSJed Brown   ierr = PetscOptionsBool("-constraints", "Test local constraints (serial only)", "ex3.c", options->constraints, &options->constraints, NULL);CHKERRQ(ierr);
145c4762a1bSJed Brown   ierr = PetscOptionsBool("-tree", "Test tree routines", "ex3.c", options->tree, &options->tree, NULL);CHKERRQ(ierr);
146c4762a1bSJed Brown   ierr = PetscOptionsBoundedInt("-tree_cell", "cell to refine in tree test", "ex3.c", options->treeCell, &options->treeCell, NULL,0);CHKERRQ(ierr);
147c4762a1bSJed Brown   ierr = PetscOptionsBool("-test_fe_jacobian", "Test finite element Jacobian assembly", "ex3.c", options->testFEjacobian, &options->testFEjacobian, NULL);CHKERRQ(ierr);
148c4762a1bSJed Brown   ierr = PetscOptionsBool("-test_fv_grad", "Test finite volume gradient reconstruction", "ex3.c", options->testFVgrad, &options->testFVgrad, NULL);CHKERRQ(ierr);
149c4762a1bSJed Brown   ierr = PetscOptionsBool("-test_injector","Test finite element injection", "ex3.c", options->testInjector, &options->testInjector,NULL);CHKERRQ(ierr);
150c4762a1bSJed Brown   ierr = PetscOptionsRealArray("-constants","Set the constant values", "ex3.c", options->constants, &n,NULL);CHKERRQ(ierr);
151c4762a1bSJed Brown   ierr = PetscOptionsEnd();CHKERRQ(ierr);
152c4762a1bSJed Brown 
153c4762a1bSJed Brown   PetscFunctionReturn(0);
154c4762a1bSJed Brown }
155c4762a1bSJed Brown 
156c4762a1bSJed Brown static PetscErrorCode TransformCoordinates(DM dm, AppCtx *user)
157c4762a1bSJed Brown {
158c4762a1bSJed Brown   PetscSection   coordSection;
159c4762a1bSJed Brown   Vec            coordinates;
160c4762a1bSJed Brown   PetscScalar   *coords;
161c4762a1bSJed Brown   PetscInt       vStart, vEnd, v;
162c4762a1bSJed Brown   PetscErrorCode ierr;
163c4762a1bSJed Brown 
164c4762a1bSJed Brown   PetscFunctionBeginUser;
165c4762a1bSJed Brown   if (user->nonaffineCoords) {
166c4762a1bSJed Brown     /* x' = r^(1/p) (x/r), y' = r^(1/p) (y/r), z' = z */
167c4762a1bSJed Brown     ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
168c4762a1bSJed Brown     ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
169c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
170c4762a1bSJed Brown     ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
171c4762a1bSJed Brown     for (v = vStart; v < vEnd; ++v) {
172c4762a1bSJed Brown       PetscInt  dof, off;
173c4762a1bSJed Brown       PetscReal p = 4.0, r;
174c4762a1bSJed Brown 
175c4762a1bSJed Brown       ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
176c4762a1bSJed Brown       ierr = PetscSectionGetOffset(coordSection, v, &off);CHKERRQ(ierr);
177c4762a1bSJed Brown       switch (dof) {
178c4762a1bSJed Brown       case 2:
179c4762a1bSJed Brown         r             = PetscSqr(PetscRealPart(coords[off+0])) + PetscSqr(PetscRealPart(coords[off+1]));
180c4762a1bSJed Brown         coords[off+0] = r == 0.0 ? 0.0 : PetscPowReal(r, (1 - p)/(2*p))*coords[off+0];
181c4762a1bSJed Brown         coords[off+1] = r == 0.0 ? 0.0 : PetscPowReal(r, (1 - p)/(2*p))*coords[off+1];
182c4762a1bSJed Brown         break;
183c4762a1bSJed Brown       case 3:
184c4762a1bSJed Brown         r             = PetscSqr(PetscRealPart(coords[off+0])) + PetscSqr(PetscRealPart(coords[off+1]));
185c4762a1bSJed Brown         coords[off+0] = r == 0.0 ? 0.0 : PetscPowReal(r, (1 - p)/(2*p))*coords[off+0];
186c4762a1bSJed Brown         coords[off+1] = r == 0.0 ? 0.0 : PetscPowReal(r, (1 - p)/(2*p))*coords[off+1];
187c4762a1bSJed Brown         coords[off+2] = coords[off+2];
188c4762a1bSJed Brown         break;
189c4762a1bSJed Brown       }
190c4762a1bSJed Brown     }
191c4762a1bSJed Brown     ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
192c4762a1bSJed Brown   }
193c4762a1bSJed Brown   if (user->shearCoords) {
194c4762a1bSJed Brown     /* x' = x + m y + m z, y' = y + m z,  z' = z */
195c4762a1bSJed Brown     ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
196c4762a1bSJed Brown     ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
197c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
198c4762a1bSJed Brown     ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
199c4762a1bSJed Brown     for (v = vStart; v < vEnd; ++v) {
200c4762a1bSJed Brown       PetscInt  dof, off;
201c4762a1bSJed Brown       PetscReal m = 1.0;
202c4762a1bSJed Brown 
203c4762a1bSJed Brown       ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
204c4762a1bSJed Brown       ierr = PetscSectionGetOffset(coordSection, v, &off);CHKERRQ(ierr);
205c4762a1bSJed Brown       switch (dof) {
206c4762a1bSJed Brown       case 2:
207c4762a1bSJed Brown         coords[off+0] = coords[off+0] + m*coords[off+1];
208c4762a1bSJed Brown         coords[off+1] = coords[off+1];
209c4762a1bSJed Brown         break;
210c4762a1bSJed Brown       case 3:
211c4762a1bSJed Brown         coords[off+0] = coords[off+0] + m*coords[off+1] + m*coords[off+2];
212c4762a1bSJed Brown         coords[off+1] = coords[off+1] + m*coords[off+2];
213c4762a1bSJed Brown         coords[off+2] = coords[off+2];
214c4762a1bSJed Brown         break;
215c4762a1bSJed Brown       }
216c4762a1bSJed Brown     }
217c4762a1bSJed Brown     ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
218c4762a1bSJed Brown   }
219c4762a1bSJed Brown   PetscFunctionReturn(0);
220c4762a1bSJed Brown }
221c4762a1bSJed Brown 
222c4762a1bSJed Brown static PetscErrorCode CreateMesh(MPI_Comm comm, AppCtx *user, DM *dm)
223c4762a1bSJed Brown {
22430602db0SMatthew G. Knepley   PetscInt       dim = 2;
22530602db0SMatthew G. Knepley   PetscBool      simplex;
226c4762a1bSJed Brown   PetscErrorCode ierr;
227c4762a1bSJed Brown 
228c4762a1bSJed Brown   PetscFunctionBeginUser;
22930602db0SMatthew G. Knepley   if (user->useDA) {
23030602db0SMatthew G. Knepley     switch (dim) {
231c4762a1bSJed Brown     case 2:
232c4762a1bSJed Brown       ierr = DMDACreate2d(comm, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, 2, 2, PETSC_DETERMINE, PETSC_DETERMINE, 1, 1, NULL, NULL, dm);CHKERRQ(ierr);
233c4762a1bSJed Brown       ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
234c4762a1bSJed Brown       ierr = DMSetUp(*dm);CHKERRQ(ierr);
235c4762a1bSJed Brown       ierr = DMDASetVertexCoordinates(*dm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);
236c4762a1bSJed Brown       break;
237c4762a1bSJed Brown     default:
238c4762a1bSJed Brown       SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot create structured mesh of dimension %d", dim);
239c4762a1bSJed Brown     }
240c4762a1bSJed Brown     ierr = PetscObjectSetName((PetscObject) *dm, "Hexahedral Mesh");CHKERRQ(ierr);
24130602db0SMatthew G. Knepley   } else {
24230602db0SMatthew G. Knepley     ierr = DMCreate(comm, dm);CHKERRQ(ierr);
24330602db0SMatthew G. Knepley     ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr);
24430602db0SMatthew G. Knepley     ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
245c4762a1bSJed Brown 
24630602db0SMatthew G. Knepley     ierr = DMGetDimension(*dm, &dim);CHKERRQ(ierr);
24730602db0SMatthew G. Knepley     ierr = DMPlexIsSimplex(*dm, &simplex);CHKERRQ(ierr);
24830602db0SMatthew G. Knepley     ierr = MPI_Bcast(&simplex, 1, MPIU_BOOL, 0, comm);CHKERRMPI(ierr);
249c4762a1bSJed Brown     if (user->tree) {
25030602db0SMatthew G. Knepley       DM refTree, ncdm = NULL;
251c4762a1bSJed Brown 
25230602db0SMatthew G. Knepley       ierr = DMPlexCreateDefaultReferenceTree(comm,dim,simplex,&refTree);CHKERRQ(ierr);
253c4762a1bSJed Brown       ierr = DMPlexSetReferenceTree(*dm,refTree);CHKERRQ(ierr);
254c4762a1bSJed Brown       ierr = DMDestroy(&refTree);CHKERRQ(ierr);
255c4762a1bSJed Brown       ierr = DMPlexTreeRefineCell(*dm,user->treeCell,&ncdm);CHKERRQ(ierr);
256c4762a1bSJed Brown       if (ncdm) {
257c4762a1bSJed Brown         ierr = DMDestroy(dm);CHKERRQ(ierr);
258c4762a1bSJed Brown         *dm = ncdm;
259c4762a1bSJed Brown         ierr = DMPlexSetRefinementUniform(*dm, PETSC_FALSE);CHKERRQ(ierr);
260c4762a1bSJed Brown       }
26130602db0SMatthew G. Knepley       ierr = DMViewFromOptions(*dm,NULL,"-tree_dm_view");CHKERRQ(ierr);
262c4762a1bSJed Brown     } else {
263c4762a1bSJed Brown       ierr = DMPlexSetRefinementUniform(*dm, PETSC_TRUE);CHKERRQ(ierr);
264c4762a1bSJed Brown     }
26530602db0SMatthew G. Knepley     ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "dist_");CHKERRQ(ierr);
26630602db0SMatthew G. Knepley     ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
26730602db0SMatthew G. Knepley     ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, NULL);CHKERRQ(ierr);
26830602db0SMatthew G. Knepley     if (simplex) {ierr = PetscObjectSetName((PetscObject) *dm, "Simplicial Mesh");CHKERRQ(ierr);}
26930602db0SMatthew G. Knepley     else         {ierr = PetscObjectSetName((PetscObject) *dm, "Hexahedral Mesh");CHKERRQ(ierr);}
270c4762a1bSJed Brown   }
271c4762a1bSJed Brown   ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
272c4762a1bSJed Brown   ierr = TransformCoordinates(*dm, user);CHKERRQ(ierr);
273c4762a1bSJed Brown   ierr = DMViewFromOptions(*dm,NULL,"-dm_view");CHKERRQ(ierr);
274c4762a1bSJed Brown   PetscFunctionReturn(0);
275c4762a1bSJed Brown }
276c4762a1bSJed Brown 
277c4762a1bSJed Brown static void simple_mass(PetscInt dim, PetscInt Nf, PetscInt NfAux,
278c4762a1bSJed Brown                         const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
279c4762a1bSJed Brown                         const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
280c4762a1bSJed Brown                         PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[])
281c4762a1bSJed Brown {
282c4762a1bSJed Brown   PetscInt d, e;
283c4762a1bSJed Brown   for (d = 0, e = 0; d < dim; d++, e+=dim+1) {
284c4762a1bSJed Brown     g0[e] = 1.;
285c4762a1bSJed Brown   }
286c4762a1bSJed Brown }
287c4762a1bSJed Brown 
288c4762a1bSJed Brown /* < \nabla v, 1/2(\nabla u + {\nabla u}^T) > */
289c4762a1bSJed Brown static void symmetric_gradient_inner_product(PetscInt dim, PetscInt Nf, PetscInt NfAux,
290c4762a1bSJed Brown                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
291c4762a1bSJed Brown                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
292c4762a1bSJed Brown                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar C[])
293c4762a1bSJed Brown {
294c4762a1bSJed Brown   PetscInt compI, compJ, d, e;
295c4762a1bSJed Brown 
296c4762a1bSJed Brown   for (compI = 0; compI < dim; ++compI) {
297c4762a1bSJed Brown     for (compJ = 0; compJ < dim; ++compJ) {
298c4762a1bSJed Brown       for (d = 0; d < dim; ++d) {
299c4762a1bSJed Brown         for (e = 0; e < dim; e++) {
300c4762a1bSJed Brown           if (d == e && d == compI && d == compJ) {
301c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 1.0;
302c4762a1bSJed Brown           } else if ((d == compJ && e == compI) || (d == e && compI == compJ)) {
303c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 0.5;
304c4762a1bSJed Brown           } else {
305c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 0.0;
306c4762a1bSJed Brown           }
307c4762a1bSJed Brown         }
308c4762a1bSJed Brown       }
309c4762a1bSJed Brown     }
310c4762a1bSJed Brown   }
311c4762a1bSJed Brown }
312c4762a1bSJed Brown 
313c4762a1bSJed Brown static PetscErrorCode SetupSection(DM dm, AppCtx *user)
314c4762a1bSJed Brown {
315c4762a1bSJed Brown   PetscErrorCode ierr;
316c4762a1bSJed Brown 
317c4762a1bSJed Brown   PetscFunctionBeginUser;
31830602db0SMatthew G. Knepley   if (user->constraints) {
319c4762a1bSJed Brown     /* test local constraints */
320c4762a1bSJed Brown     DM            coordDM;
321c4762a1bSJed Brown     PetscInt      fStart, fEnd, f, vStart, vEnd, v;
322c4762a1bSJed Brown     PetscInt      edgesx = 2, vertsx;
323c4762a1bSJed Brown     PetscInt      edgesy = 2, vertsy;
324c4762a1bSJed Brown     PetscMPIInt   size;
325c4762a1bSJed Brown     PetscInt      numConst;
326c4762a1bSJed Brown     PetscSection  aSec;
327c4762a1bSJed Brown     PetscInt     *anchors;
328c4762a1bSJed Brown     PetscInt      offset;
329c4762a1bSJed Brown     IS            aIS;
330c4762a1bSJed Brown     MPI_Comm      comm = PetscObjectComm((PetscObject)dm);
331c4762a1bSJed Brown 
332ffc4695bSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
333c4762a1bSJed Brown     if (size > 1) SETERRQ(comm,PETSC_ERR_SUP,"Local constraint test can only be performed in serial");
334c4762a1bSJed Brown 
335c4762a1bSJed Brown     /* we are going to test constraints by using them to enforce periodicity
336c4762a1bSJed Brown      * in one direction, and comparing to the existing method of enforcing
337c4762a1bSJed Brown      * periodicity */
338c4762a1bSJed Brown 
339c4762a1bSJed Brown     /* first create the coordinate section so that it does not clone the
340c4762a1bSJed Brown      * constraints */
341c4762a1bSJed Brown     ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr);
342c4762a1bSJed Brown 
343c4762a1bSJed Brown     /* create the constrained-to-anchor section */
344c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
345c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm,1,&fStart,&fEnd);CHKERRQ(ierr);
346c4762a1bSJed Brown     ierr = PetscSectionCreate(PETSC_COMM_SELF,&aSec);CHKERRQ(ierr);
347c4762a1bSJed Brown     ierr = PetscSectionSetChart(aSec,PetscMin(fStart,vStart),PetscMax(fEnd,vEnd));CHKERRQ(ierr);
348c4762a1bSJed Brown 
349c4762a1bSJed Brown     /* define the constraints */
350c4762a1bSJed Brown     ierr = PetscOptionsGetInt(NULL,NULL,"-da_grid_x",&edgesx,NULL);CHKERRQ(ierr);
351c4762a1bSJed Brown     ierr = PetscOptionsGetInt(NULL,NULL,"-da_grid_y",&edgesy,NULL);CHKERRQ(ierr);
352c4762a1bSJed Brown     vertsx = edgesx + 1;
353c4762a1bSJed Brown     vertsy = edgesy + 1;
354c4762a1bSJed Brown     numConst = vertsy + edgesy;
355c4762a1bSJed Brown     ierr = PetscMalloc1(numConst,&anchors);CHKERRQ(ierr);
356c4762a1bSJed Brown     offset = 0;
357c4762a1bSJed Brown     for (v = vStart + edgesx; v < vEnd; v+= vertsx) {
358c4762a1bSJed Brown       ierr = PetscSectionSetDof(aSec,v,1);CHKERRQ(ierr);
359c4762a1bSJed Brown       anchors[offset++] = v - edgesx;
360c4762a1bSJed Brown     }
361c4762a1bSJed Brown     for (f = fStart + edgesx * vertsy + edgesx * edgesy; f < fEnd; f++) {
362c4762a1bSJed Brown       ierr = PetscSectionSetDof(aSec,f,1);CHKERRQ(ierr);
363c4762a1bSJed Brown       anchors[offset++] = f - edgesx * edgesy;
364c4762a1bSJed Brown     }
365c4762a1bSJed Brown     ierr = PetscSectionSetUp(aSec);CHKERRQ(ierr);
366c4762a1bSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,numConst,anchors,PETSC_OWN_POINTER,&aIS);CHKERRQ(ierr);
367c4762a1bSJed Brown 
368c4762a1bSJed Brown     ierr = DMPlexSetAnchors(dm,aSec,aIS);CHKERRQ(ierr);
369c4762a1bSJed Brown     ierr = PetscSectionDestroy(&aSec);CHKERRQ(ierr);
370c4762a1bSJed Brown     ierr = ISDestroy(&aIS);CHKERRQ(ierr);
371c4762a1bSJed Brown   }
372c4762a1bSJed Brown   ierr = DMSetNumFields(dm, 1);CHKERRQ(ierr);
373c4762a1bSJed Brown   ierr = DMSetField(dm, 0, NULL, (PetscObject) user->fe);CHKERRQ(ierr);
374c4762a1bSJed Brown   ierr = DMCreateDS(dm);CHKERRQ(ierr);
37530602db0SMatthew G. Knepley   if (user->constraints) {
376c4762a1bSJed Brown     /* test getting local constraint matrix that matches section */
377c4762a1bSJed Brown     PetscSection aSec;
378c4762a1bSJed Brown     IS           aIS;
379c4762a1bSJed Brown 
380c4762a1bSJed Brown     ierr = DMPlexGetAnchors(dm,&aSec,&aIS);CHKERRQ(ierr);
381c4762a1bSJed Brown     if (aSec) {
382c4762a1bSJed Brown       PetscDS         ds;
383c4762a1bSJed Brown       PetscSection    cSec, section;
384c4762a1bSJed Brown       PetscInt        cStart, cEnd, c, numComp;
385c4762a1bSJed Brown       Mat             cMat, mass;
386c4762a1bSJed Brown       Vec             local;
387c4762a1bSJed Brown       const PetscInt *anchors;
388c4762a1bSJed Brown 
389c4762a1bSJed Brown       ierr = DMGetLocalSection(dm,&section);CHKERRQ(ierr);
390c4762a1bSJed Brown       /* this creates the matrix and preallocates the matrix structure: we
391c4762a1bSJed Brown        * just have to fill in the values */
392c4762a1bSJed Brown       ierr = DMGetDefaultConstraints(dm,&cSec,&cMat);CHKERRQ(ierr);
393c4762a1bSJed Brown       ierr = PetscSectionGetChart(cSec,&cStart,&cEnd);CHKERRQ(ierr);
394c4762a1bSJed Brown       ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
395c4762a1bSJed Brown       ierr = PetscFEGetNumComponents(user->fe, &numComp);CHKERRQ(ierr);
396c4762a1bSJed Brown       for (c = cStart; c < cEnd; c++) {
397c4762a1bSJed Brown         PetscInt cDof;
398c4762a1bSJed Brown 
399c4762a1bSJed Brown         /* is this point constrained? (does it have an anchor?) */
400c4762a1bSJed Brown         ierr = PetscSectionGetDof(aSec,c,&cDof);CHKERRQ(ierr);
401c4762a1bSJed Brown         if (cDof) {
402c4762a1bSJed Brown           PetscInt cOff, a, aDof, aOff, j;
403c4762a1bSJed Brown           if (cDof != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Found %d anchor points: should be just one",cDof);
404c4762a1bSJed Brown 
405c4762a1bSJed Brown           /* find the anchor point */
406c4762a1bSJed Brown           ierr = PetscSectionGetOffset(aSec,c,&cOff);CHKERRQ(ierr);
407c4762a1bSJed Brown           a    = anchors[cOff];
408c4762a1bSJed Brown 
409c4762a1bSJed Brown           /* find the constrained dofs (row in constraint matrix) */
410c4762a1bSJed Brown           ierr = PetscSectionGetDof(cSec,c,&cDof);CHKERRQ(ierr);
411c4762a1bSJed Brown           ierr = PetscSectionGetOffset(cSec,c,&cOff);CHKERRQ(ierr);
412c4762a1bSJed Brown 
413c4762a1bSJed Brown           /* find the anchor dofs (column in constraint matrix) */
414c4762a1bSJed Brown           ierr = PetscSectionGetDof(section,a,&aDof);CHKERRQ(ierr);
415c4762a1bSJed Brown           ierr = PetscSectionGetOffset(section,a,&aOff);CHKERRQ(ierr);
416c4762a1bSJed Brown 
417c4762a1bSJed Brown           if (cDof != aDof) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point and anchor have different number of dofs: %d, %d\n",cDof,aDof);
418c4762a1bSJed Brown           if (cDof % numComp) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point dofs not divisible by field components: %d, %d\n",cDof,numComp);
419c4762a1bSJed Brown 
420c4762a1bSJed Brown           /* put in a simple equality constraint */
421c4762a1bSJed Brown           for (j = 0; j < cDof; j++) {
422c4762a1bSJed Brown             ierr = MatSetValue(cMat,cOff+j,aOff+j,1.,INSERT_VALUES);CHKERRQ(ierr);
423c4762a1bSJed Brown           }
424c4762a1bSJed Brown         }
425c4762a1bSJed Brown       }
426c4762a1bSJed Brown       ierr = MatAssemblyBegin(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
427c4762a1bSJed Brown       ierr = MatAssemblyEnd(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
428c4762a1bSJed Brown       ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
429c4762a1bSJed Brown 
430c4762a1bSJed Brown       /* Now that we have constructed the constraint matrix, any FE matrix
431c4762a1bSJed Brown        * that we construct will apply the constraints during construction */
432c4762a1bSJed Brown 
433c4762a1bSJed Brown       ierr = DMCreateMatrix(dm,&mass);CHKERRQ(ierr);
434c4762a1bSJed Brown       /* get a dummy local variable to serve as the solution */
435c4762a1bSJed Brown       ierr = DMGetLocalVector(dm,&local);CHKERRQ(ierr);
436c4762a1bSJed Brown       ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
437c4762a1bSJed Brown       /* set the jacobian to be the mass matrix */
438c4762a1bSJed Brown       ierr = PetscDSSetJacobian(ds, 0, 0, simple_mass, NULL,  NULL, NULL);CHKERRQ(ierr);
439c4762a1bSJed Brown       /* build the mass matrix */
440c4762a1bSJed Brown       ierr = DMPlexSNESComputeJacobianFEM(dm,local,mass,mass,NULL);CHKERRQ(ierr);
441c4762a1bSJed Brown       ierr = MatView(mass,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
442c4762a1bSJed Brown       ierr = MatDestroy(&mass);CHKERRQ(ierr);
443c4762a1bSJed Brown       ierr = DMRestoreLocalVector(dm,&local);CHKERRQ(ierr);
444c4762a1bSJed Brown     }
445c4762a1bSJed Brown   }
446c4762a1bSJed Brown   PetscFunctionReturn(0);
447c4762a1bSJed Brown }
448c4762a1bSJed Brown 
449c4762a1bSJed Brown static PetscErrorCode TestFEJacobian(DM dm, AppCtx *user)
450c4762a1bSJed Brown {
451c4762a1bSJed Brown   PetscErrorCode ierr;
452c4762a1bSJed Brown 
453c4762a1bSJed Brown   PetscFunctionBeginUser;
45430602db0SMatthew G. Knepley   if (!user->useDA) {
455c4762a1bSJed Brown     Vec          local;
456c4762a1bSJed Brown     const Vec    *vecs;
457c4762a1bSJed Brown     Mat          E;
458c4762a1bSJed Brown     MatNullSpace sp;
459c4762a1bSJed Brown     PetscBool    isNullSpace, hasConst;
46030602db0SMatthew G. Knepley     PetscInt     dim, n, i;
461c4762a1bSJed Brown     Vec          res = NULL, localX, localRes;
462c4762a1bSJed Brown     PetscDS      ds;
463c4762a1bSJed Brown 
46430602db0SMatthew G. Knepley     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
46530602db0SMatthew G. Knepley     if (user->numComponents != dim) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "The number of components %d must be equal to the dimension %d for this test", user->numComponents, dim);
466c4762a1bSJed Brown     ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
467c4762a1bSJed Brown     ierr = PetscDSSetJacobian(ds,0,0,NULL,NULL,NULL,symmetric_gradient_inner_product);CHKERRQ(ierr);
468c4762a1bSJed Brown     ierr = DMCreateMatrix(dm,&E);CHKERRQ(ierr);
469c4762a1bSJed Brown     ierr = DMGetLocalVector(dm,&local);CHKERRQ(ierr);
470c4762a1bSJed Brown     ierr = DMPlexSNESComputeJacobianFEM(dm,local,E,E,NULL);CHKERRQ(ierr);
47156cf3b9cSMatthew G. Knepley     ierr = DMPlexCreateRigidBody(dm,0,&sp);CHKERRQ(ierr);
472c4762a1bSJed Brown     ierr = MatNullSpaceGetVecs(sp,&hasConst,&n,&vecs);CHKERRQ(ierr);
473c4762a1bSJed Brown     if (n) {ierr = VecDuplicate(vecs[0],&res);CHKERRQ(ierr);}
474c4762a1bSJed Brown     ierr = DMCreateLocalVector(dm,&localX);CHKERRQ(ierr);
475c4762a1bSJed Brown     ierr = DMCreateLocalVector(dm,&localRes);CHKERRQ(ierr);
476c4762a1bSJed Brown     for (i = 0; i < n; i++) { /* also test via matrix-free Jacobian application */
477c4762a1bSJed Brown       PetscReal resNorm;
478c4762a1bSJed Brown 
479c4762a1bSJed Brown       ierr = VecSet(localRes,0.);CHKERRQ(ierr);
480c4762a1bSJed Brown       ierr = VecSet(localX,0.);CHKERRQ(ierr);
481c4762a1bSJed Brown       ierr = VecSet(local,0.);CHKERRQ(ierr);
482c4762a1bSJed Brown       ierr = VecSet(res,0.);CHKERRQ(ierr);
483c4762a1bSJed Brown       ierr = DMGlobalToLocalBegin(dm,vecs[i],INSERT_VALUES,localX);CHKERRQ(ierr);
484c4762a1bSJed Brown       ierr = DMGlobalToLocalEnd(dm,vecs[i],INSERT_VALUES,localX);CHKERRQ(ierr);
485*8e3a2eefSMatthew G. Knepley       ierr = DMSNESComputeJacobianAction(dm,local,localX,localRes,NULL);CHKERRQ(ierr);
486c4762a1bSJed Brown       ierr = DMLocalToGlobalBegin(dm,localRes,ADD_VALUES,res);CHKERRQ(ierr);
487c4762a1bSJed Brown       ierr = DMLocalToGlobalEnd(dm,localRes,ADD_VALUES,res);CHKERRQ(ierr);
488c4762a1bSJed Brown       ierr = VecNorm(res,NORM_2,&resNorm);CHKERRQ(ierr);
489c4762a1bSJed Brown       if (resNorm > PETSC_SMALL) {
490c4762a1bSJed Brown         ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient action null space vector %D residual: %E\n",i,resNorm);CHKERRQ(ierr);
491c4762a1bSJed Brown       }
492c4762a1bSJed Brown     }
493c4762a1bSJed Brown     ierr = VecDestroy(&localRes);CHKERRQ(ierr);
494c4762a1bSJed Brown     ierr = VecDestroy(&localX);CHKERRQ(ierr);
495c4762a1bSJed Brown     ierr = VecDestroy(&res);CHKERRQ(ierr);
496c4762a1bSJed Brown     ierr = MatNullSpaceTest(sp,E,&isNullSpace);CHKERRQ(ierr);
497c4762a1bSJed Brown     if (isNullSpace) {
498c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient null space: PASS\n");CHKERRQ(ierr);
499c4762a1bSJed Brown     } else {
500c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient null space: FAIL\n");CHKERRQ(ierr);
501c4762a1bSJed Brown     }
502c4762a1bSJed Brown     ierr = MatNullSpaceDestroy(&sp);CHKERRQ(ierr);
503c4762a1bSJed Brown     ierr = MatDestroy(&E);CHKERRQ(ierr);
504c4762a1bSJed Brown     ierr = DMRestoreLocalVector(dm,&local);CHKERRQ(ierr);
505c4762a1bSJed Brown   }
506c4762a1bSJed Brown   PetscFunctionReturn(0);
507c4762a1bSJed Brown }
508c4762a1bSJed Brown 
509c4762a1bSJed Brown static PetscErrorCode TestInjector(DM dm, AppCtx *user)
510c4762a1bSJed Brown {
511c4762a1bSJed Brown   DM             refTree;
512c4762a1bSJed Brown   PetscMPIInt    rank;
513c4762a1bSJed Brown   PetscErrorCode ierr;
514c4762a1bSJed Brown 
515c4762a1bSJed Brown   PetscFunctionBegin;
516c4762a1bSJed Brown   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
517c4762a1bSJed Brown   if (refTree) {
518c4762a1bSJed Brown     Mat inj;
519c4762a1bSJed Brown 
520c4762a1bSJed Brown     ierr = DMPlexComputeInjectorReferenceTree(refTree,&inj);CHKERRQ(ierr);
521c4762a1bSJed Brown     ierr = PetscObjectSetName((PetscObject)inj,"Reference Tree Injector");CHKERRQ(ierr);
522ffc4695bSBarry Smith     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);
523c4762a1bSJed Brown     if (!rank) {
524c4762a1bSJed Brown       ierr = MatView(inj,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
525c4762a1bSJed Brown     }
526c4762a1bSJed Brown     ierr = MatDestroy(&inj);CHKERRQ(ierr);
527c4762a1bSJed Brown   }
528c4762a1bSJed Brown   PetscFunctionReturn(0);
529c4762a1bSJed Brown }
530c4762a1bSJed Brown 
531c4762a1bSJed Brown static PetscErrorCode TestFVGrad(DM dm, AppCtx *user)
532c4762a1bSJed Brown {
533c4762a1bSJed Brown   MPI_Comm          comm;
534c4762a1bSJed Brown   DM                dmRedist, dmfv, dmgrad, dmCell, refTree;
535c4762a1bSJed Brown   PetscFV           fv;
53630602db0SMatthew G. Knepley   PetscInt          dim, nvecs, v, cStart, cEnd, cEndInterior;
537c4762a1bSJed Brown   PetscMPIInt       size;
538c4762a1bSJed Brown   Vec               cellgeom, grad, locGrad;
539c4762a1bSJed Brown   const PetscScalar *cgeom;
540c4762a1bSJed Brown   PetscReal         allVecMaxDiff = 0., fvTol = 100. * PETSC_MACHINE_EPSILON;
541c4762a1bSJed Brown   PetscErrorCode    ierr;
542c4762a1bSJed Brown 
543c4762a1bSJed Brown   PetscFunctionBeginUser;
544c4762a1bSJed Brown   comm = PetscObjectComm((PetscObject)dm);
54530602db0SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
546c4762a1bSJed Brown   /* duplicate DM, give dup. a FV discretization */
547c4762a1bSJed Brown   ierr = DMSetBasicAdjacency(dm,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
548ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
549c4762a1bSJed Brown   dmRedist = NULL;
550c4762a1bSJed Brown   if (size > 1) {
551c4762a1bSJed Brown     ierr = DMPlexDistributeOverlap(dm,1,NULL,&dmRedist);CHKERRQ(ierr);
552c4762a1bSJed Brown   }
553c4762a1bSJed Brown   if (!dmRedist) {
554c4762a1bSJed Brown     dmRedist = dm;
555c4762a1bSJed Brown     ierr = PetscObjectReference((PetscObject)dmRedist);CHKERRQ(ierr);
556c4762a1bSJed Brown   }
557c4762a1bSJed Brown   ierr = PetscFVCreate(comm,&fv);CHKERRQ(ierr);
558c4762a1bSJed Brown   ierr = PetscFVSetType(fv,PETSCFVLEASTSQUARES);CHKERRQ(ierr);
559c4762a1bSJed Brown   ierr = PetscFVSetNumComponents(fv,user->numComponents);CHKERRQ(ierr);
56030602db0SMatthew G. Knepley   ierr = PetscFVSetSpatialDimension(fv,dim);CHKERRQ(ierr);
561c4762a1bSJed Brown   ierr = PetscFVSetFromOptions(fv);CHKERRQ(ierr);
562c4762a1bSJed Brown   ierr = PetscFVSetUp(fv);CHKERRQ(ierr);
563c4762a1bSJed Brown   ierr = DMPlexConstructGhostCells(dmRedist,NULL,NULL,&dmfv);CHKERRQ(ierr);
564c4762a1bSJed Brown   ierr = DMDestroy(&dmRedist);CHKERRQ(ierr);
565c4762a1bSJed Brown   ierr = DMSetNumFields(dmfv,1);CHKERRQ(ierr);
566c4762a1bSJed Brown   ierr = DMSetField(dmfv, 0, NULL, (PetscObject) fv);CHKERRQ(ierr);
567c4762a1bSJed Brown   ierr = DMCreateDS(dmfv);CHKERRQ(ierr);
568c4762a1bSJed Brown   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
569c4762a1bSJed Brown   if (refTree) {ierr = DMCopyDisc(dmfv,refTree);CHKERRQ(ierr);}
5703e9753d6SMatthew G. Knepley   ierr = DMPlexGetGradientDM(dmfv, fv, &dmgrad);CHKERRQ(ierr);
571c4762a1bSJed Brown   ierr = DMPlexGetHeightStratum(dmfv,0,&cStart,&cEnd);CHKERRQ(ierr);
57230602db0SMatthew G. Knepley   nvecs = dim * (dim+1) / 2;
5733e9753d6SMatthew G. Knepley   ierr = DMPlexGetGeometryFVM(dmfv,NULL,&cellgeom,NULL);CHKERRQ(ierr);
574c4762a1bSJed Brown   ierr = VecGetDM(cellgeom,&dmCell);CHKERRQ(ierr);
575c4762a1bSJed Brown   ierr = VecGetArrayRead(cellgeom,&cgeom);CHKERRQ(ierr);
576c4762a1bSJed Brown   ierr = DMGetGlobalVector(dmgrad,&grad);CHKERRQ(ierr);
577c4762a1bSJed Brown   ierr = DMGetLocalVector(dmgrad,&locGrad);CHKERRQ(ierr);
578c4762a1bSJed Brown   ierr = DMPlexGetGhostCellStratum(dmgrad,&cEndInterior,NULL);CHKERRQ(ierr);
579c4762a1bSJed Brown   cEndInterior = (cEndInterior < 0) ? cEnd: cEndInterior;
580c4762a1bSJed Brown   for (v = 0; v < nvecs; v++) {
581c4762a1bSJed Brown     Vec               locX;
582c4762a1bSJed Brown     PetscInt          c;
583c4762a1bSJed Brown     PetscScalar       trueGrad[3][3] = {{0.}};
584c4762a1bSJed Brown     const PetscScalar *gradArray;
585c4762a1bSJed Brown     PetscReal         maxDiff, maxDiffGlob;
586c4762a1bSJed Brown 
587c4762a1bSJed Brown     ierr = DMGetLocalVector(dmfv,&locX);CHKERRQ(ierr);
588c4762a1bSJed Brown     /* get the local projection of the rigid body mode */
589c4762a1bSJed Brown     for (c = cStart; c < cEnd; c++) {
590c4762a1bSJed Brown       PetscFVCellGeom *cg;
591c4762a1bSJed Brown       PetscScalar     cx[3] = {0.,0.,0.};
592c4762a1bSJed Brown 
593c4762a1bSJed Brown       ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
59430602db0SMatthew G. Knepley       if (v < dim) {
595c4762a1bSJed Brown         cx[v] = 1.;
596c4762a1bSJed Brown       } else {
59730602db0SMatthew G. Knepley         PetscInt w = v - dim;
598c4762a1bSJed Brown 
59930602db0SMatthew G. Knepley         cx[(w + 1) % dim] =  cg->centroid[(w + 2) % dim];
60030602db0SMatthew G. Knepley         cx[(w + 2) % dim] = -cg->centroid[(w + 1) % dim];
601c4762a1bSJed Brown       }
602c4762a1bSJed Brown       ierr = DMPlexVecSetClosure(dmfv,NULL,locX,c,cx,INSERT_ALL_VALUES);CHKERRQ(ierr);
603c4762a1bSJed Brown     }
604c4762a1bSJed Brown     /* TODO: this isn't in any header */
605c4762a1bSJed Brown     ierr = DMPlexReconstructGradientsFVM(dmfv,locX,grad);CHKERRQ(ierr);
606c4762a1bSJed Brown     ierr = DMGlobalToLocalBegin(dmgrad,grad,INSERT_VALUES,locGrad);CHKERRQ(ierr);
607c4762a1bSJed Brown     ierr = DMGlobalToLocalEnd(dmgrad,grad,INSERT_VALUES,locGrad);CHKERRQ(ierr);
608c4762a1bSJed Brown     ierr = VecGetArrayRead(locGrad,&gradArray);CHKERRQ(ierr);
609c4762a1bSJed Brown     /* compare computed gradient to exact gradient */
61030602db0SMatthew G. Knepley     if (v >= dim) {
61130602db0SMatthew G. Knepley       PetscInt w = v - dim;
612c4762a1bSJed Brown 
61330602db0SMatthew G. Knepley       trueGrad[(w + 1) % dim][(w + 2) % dim] =  1.;
61430602db0SMatthew G. Knepley       trueGrad[(w + 2) % dim][(w + 1) % dim] = -1.;
615c4762a1bSJed Brown     }
616c4762a1bSJed Brown     maxDiff = 0.;
617c4762a1bSJed Brown     for (c = cStart; c < cEndInterior; c++) {
618c4762a1bSJed Brown       PetscScalar *compGrad;
619c4762a1bSJed Brown       PetscInt    i, j, k;
620c4762a1bSJed Brown       PetscReal   FrobDiff = 0.;
621c4762a1bSJed Brown 
622c4762a1bSJed Brown       ierr = DMPlexPointLocalRead(dmgrad, c, gradArray, &compGrad);CHKERRQ(ierr);
623c4762a1bSJed Brown 
62430602db0SMatthew G. Knepley       for (i = 0, k = 0; i < dim; i++) {
62530602db0SMatthew G. Knepley         for (j = 0; j < dim; j++, k++) {
626c4762a1bSJed Brown           PetscScalar diff = compGrad[k] - trueGrad[i][j];
627c4762a1bSJed Brown           FrobDiff += PetscRealPart(diff * PetscConj(diff));
628c4762a1bSJed Brown         }
629c4762a1bSJed Brown       }
630c4762a1bSJed Brown       FrobDiff = PetscSqrtReal(FrobDiff);
631c4762a1bSJed Brown       maxDiff  = PetscMax(maxDiff,FrobDiff);
632c4762a1bSJed Brown     }
633ffc4695bSBarry Smith     ierr = MPI_Allreduce(&maxDiff,&maxDiffGlob,1,MPIU_REAL,MPIU_MAX,comm);CHKERRMPI(ierr);
634c4762a1bSJed Brown     allVecMaxDiff = PetscMax(allVecMaxDiff,maxDiffGlob);
635c4762a1bSJed Brown     ierr = VecRestoreArrayRead(locGrad,&gradArray);CHKERRQ(ierr);
636c4762a1bSJed Brown     ierr = DMRestoreLocalVector(dmfv,&locX);CHKERRQ(ierr);
637c4762a1bSJed Brown   }
638c4762a1bSJed Brown   if (allVecMaxDiff < fvTol) {
639c4762a1bSJed Brown     ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Finite volume gradient reconstruction: PASS\n");CHKERRQ(ierr);
640c4762a1bSJed Brown   } else {
641c4762a1bSJed Brown     ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Finite volume gradient reconstruction: FAIL at tolerance %g with max difference %g\n",fvTol,allVecMaxDiff);CHKERRQ(ierr);
642c4762a1bSJed Brown   }
643c4762a1bSJed Brown   ierr = DMRestoreLocalVector(dmgrad,&locGrad);CHKERRQ(ierr);
644c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(dmgrad,&grad);CHKERRQ(ierr);
645c4762a1bSJed Brown   ierr = VecRestoreArrayRead(cellgeom,&cgeom);CHKERRQ(ierr);
646c4762a1bSJed Brown   ierr = DMDestroy(&dmfv);CHKERRQ(ierr);
647c4762a1bSJed Brown   ierr = PetscFVDestroy(&fv);CHKERRQ(ierr);
648c4762a1bSJed Brown   PetscFunctionReturn(0);
649c4762a1bSJed Brown }
650c4762a1bSJed Brown 
651c4762a1bSJed Brown static PetscErrorCode ComputeError(DM dm, PetscErrorCode (**exactFuncs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *),
652c4762a1bSJed Brown                                    PetscErrorCode (**exactFuncDers)(PetscInt, PetscReal, const PetscReal[], const PetscReal[], PetscInt, PetscScalar *, void *),
653c4762a1bSJed Brown                                    void **exactCtxs, PetscReal *error, PetscReal *errorDer, AppCtx *user)
654c4762a1bSJed Brown {
655c4762a1bSJed Brown   Vec            u;
656c4762a1bSJed Brown   PetscReal      n[3] = {1.0, 1.0, 1.0};
657c4762a1bSJed Brown   PetscErrorCode ierr;
658c4762a1bSJed Brown 
659c4762a1bSJed Brown   PetscFunctionBeginUser;
660c4762a1bSJed Brown   ierr = DMGetGlobalVector(dm, &u);CHKERRQ(ierr);
661c4762a1bSJed Brown   /* Project function into FE function space */
662c4762a1bSJed Brown   ierr = DMProjectFunction(dm, 0.0, exactFuncs, exactCtxs, INSERT_ALL_VALUES, u);CHKERRQ(ierr);
663c4762a1bSJed Brown   ierr = VecViewFromOptions(u, NULL, "-projection_view");CHKERRQ(ierr);
664c4762a1bSJed Brown   /* Compare approximation to exact in L_2 */
665c4762a1bSJed Brown   ierr = DMComputeL2Diff(dm, 0.0, exactFuncs, exactCtxs, u, error);CHKERRQ(ierr);
666c4762a1bSJed Brown   ierr = DMComputeL2GradientDiff(dm, 0.0, exactFuncDers, exactCtxs, u, n, errorDer);CHKERRQ(ierr);
667c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(dm, &u);CHKERRQ(ierr);
668c4762a1bSJed Brown   PetscFunctionReturn(0);
669c4762a1bSJed Brown }
670c4762a1bSJed Brown 
671c4762a1bSJed Brown static PetscErrorCode CheckFunctions(DM dm, PetscInt order, AppCtx *user)
672c4762a1bSJed Brown {
673c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx);
674c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt dim, PetscReal time, const PetscReal x[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx);
675c4762a1bSJed Brown   void            *exactCtxs[3];
676c4762a1bSJed Brown   MPI_Comm         comm;
677c4762a1bSJed Brown   PetscReal        error, errorDer, tol = PETSC_SMALL;
678c4762a1bSJed Brown   PetscErrorCode   ierr;
679c4762a1bSJed Brown 
680c4762a1bSJed Brown   PetscFunctionBeginUser;
681c4762a1bSJed Brown   exactCtxs[0]       = user;
682c4762a1bSJed Brown   exactCtxs[1]       = user;
683c4762a1bSJed Brown   exactCtxs[2]       = user;
684c4762a1bSJed Brown   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
685c4762a1bSJed Brown   /* Setup functions to approximate */
686c4762a1bSJed Brown   switch (order) {
687c4762a1bSJed Brown   case 0:
688c4762a1bSJed Brown     exactFuncs[0]    = constant;
689c4762a1bSJed Brown     exactFuncDers[0] = constantDer;
690c4762a1bSJed Brown     break;
691c4762a1bSJed Brown   case 1:
692c4762a1bSJed Brown     exactFuncs[0]    = linear;
693c4762a1bSJed Brown     exactFuncDers[0] = linearDer;
694c4762a1bSJed Brown     break;
695c4762a1bSJed Brown   case 2:
696c4762a1bSJed Brown     exactFuncs[0]    = quadratic;
697c4762a1bSJed Brown     exactFuncDers[0] = quadraticDer;
698c4762a1bSJed Brown     break;
699c4762a1bSJed Brown   case 3:
700c4762a1bSJed Brown     exactFuncs[0]    = cubic;
701c4762a1bSJed Brown     exactFuncDers[0] = cubicDer;
702c4762a1bSJed Brown     break;
703c4762a1bSJed Brown   default:
70430602db0SMatthew G. Knepley     SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "Could not determine functions to test for order %d", order);
705c4762a1bSJed Brown   }
706c4762a1bSJed Brown   ierr = ComputeError(dm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
707c4762a1bSJed Brown   /* Report result */
708c4762a1bSJed Brown   if (error > tol)    {ierr = PetscPrintf(comm, "Function tests FAIL for order %D at tolerance %g error %g\n", order, (double)tol,(double) error);CHKERRQ(ierr);}
709c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Function tests pass for order %D at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
710c4762a1bSJed Brown   if (errorDer > tol) {ierr = PetscPrintf(comm, "Function tests FAIL for order %D derivatives at tolerance %g error %g\n", order, (double)tol, (double)errorDer);CHKERRQ(ierr);}
711c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Function tests pass for order %D derivatives at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
712c4762a1bSJed Brown   PetscFunctionReturn(0);
713c4762a1bSJed Brown }
714c4762a1bSJed Brown 
715c4762a1bSJed Brown static PetscErrorCode CheckInterpolation(DM dm, PetscBool checkRestrict, PetscInt order, AppCtx *user)
716c4762a1bSJed Brown {
717c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt, PetscReal, const PetscReal x[], PetscInt, PetscScalar *u, void *ctx);
718c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt, PetscReal, const PetscReal x[], const PetscReal n[], PetscInt, PetscScalar *u, void *ctx);
719c4762a1bSJed Brown   PetscReal       n[3]         = {1.0, 1.0, 1.0};
720c4762a1bSJed Brown   void           *exactCtxs[3];
721c4762a1bSJed Brown   DM              rdm, idm, fdm;
722c4762a1bSJed Brown   Mat             Interp;
723c4762a1bSJed Brown   Vec             iu, fu, scaling;
724c4762a1bSJed Brown   MPI_Comm        comm;
72530602db0SMatthew G. Knepley   PetscInt        dim;
726c4762a1bSJed Brown   PetscReal       error, errorDer, tol = PETSC_SMALL;
727c4762a1bSJed Brown   PetscErrorCode  ierr;
728c4762a1bSJed Brown 
729c4762a1bSJed Brown   PetscFunctionBeginUser;
730c4762a1bSJed Brown   exactCtxs[0]       = user;
731c4762a1bSJed Brown   exactCtxs[1]       = user;
732c4762a1bSJed Brown   exactCtxs[2]       = user;
733c4762a1bSJed Brown   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
73430602db0SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
735c4762a1bSJed Brown   ierr = DMRefine(dm, comm, &rdm);CHKERRQ(ierr);
736c4762a1bSJed Brown   ierr = DMSetCoarseDM(rdm, dm);CHKERRQ(ierr);
737c4762a1bSJed Brown   ierr = DMPlexSetRegularRefinement(rdm, user->convRefine);CHKERRQ(ierr);
73830602db0SMatthew G. Knepley   if (user->tree) {
739c4762a1bSJed Brown     DM refTree;
740c4762a1bSJed Brown     ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
741c4762a1bSJed Brown     ierr = DMPlexSetReferenceTree(rdm,refTree);CHKERRQ(ierr);
742c4762a1bSJed Brown   }
74330602db0SMatthew G. Knepley   if (user->useDA) {ierr = DMDASetVertexCoordinates(rdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
744c4762a1bSJed Brown   ierr = SetupSection(rdm, user);CHKERRQ(ierr);
745c4762a1bSJed Brown   /* Setup functions to approximate */
746c4762a1bSJed Brown   switch (order) {
747c4762a1bSJed Brown   case 0:
748c4762a1bSJed Brown     exactFuncs[0]    = constant;
749c4762a1bSJed Brown     exactFuncDers[0] = constantDer;
750c4762a1bSJed Brown     break;
751c4762a1bSJed Brown   case 1:
752c4762a1bSJed Brown     exactFuncs[0]    = linear;
753c4762a1bSJed Brown     exactFuncDers[0] = linearDer;
754c4762a1bSJed Brown     break;
755c4762a1bSJed Brown   case 2:
756c4762a1bSJed Brown     exactFuncs[0]    = quadratic;
757c4762a1bSJed Brown     exactFuncDers[0] = quadraticDer;
758c4762a1bSJed Brown     break;
759c4762a1bSJed Brown   case 3:
760c4762a1bSJed Brown     exactFuncs[0]    = cubic;
761c4762a1bSJed Brown     exactFuncDers[0] = cubicDer;
762c4762a1bSJed Brown     break;
763c4762a1bSJed Brown   default:
764c4762a1bSJed Brown     SETERRQ2(comm, PETSC_ERR_ARG_OUTOFRANGE, "Could not determine functions to test for dimension %D order %D", dim, order);
765c4762a1bSJed Brown   }
766c4762a1bSJed Brown   idm  = checkRestrict ? rdm :  dm;
767c4762a1bSJed Brown   fdm  = checkRestrict ?  dm : rdm;
768c4762a1bSJed Brown   ierr = DMGetGlobalVector(idm, &iu);CHKERRQ(ierr);
769c4762a1bSJed Brown   ierr = DMGetGlobalVector(fdm, &fu);CHKERRQ(ierr);
770c4762a1bSJed Brown   ierr = DMSetApplicationContext(dm, user);CHKERRQ(ierr);
771c4762a1bSJed Brown   ierr = DMSetApplicationContext(rdm, user);CHKERRQ(ierr);
772c4762a1bSJed Brown   ierr = DMCreateInterpolation(dm, rdm, &Interp, &scaling);CHKERRQ(ierr);
773c4762a1bSJed Brown   /* Project function into initial FE function space */
774c4762a1bSJed Brown   ierr = DMProjectFunction(idm, 0.0, exactFuncs, exactCtxs, INSERT_ALL_VALUES, iu);CHKERRQ(ierr);
775c4762a1bSJed Brown   /* Interpolate function into final FE function space */
776c4762a1bSJed Brown   if (checkRestrict) {ierr = MatRestrict(Interp, iu, fu);CHKERRQ(ierr);ierr = VecPointwiseMult(fu, scaling, fu);CHKERRQ(ierr);}
777c4762a1bSJed Brown   else               {ierr = MatInterpolate(Interp, iu, fu);CHKERRQ(ierr);}
778c4762a1bSJed Brown   /* Compare approximation to exact in L_2 */
779c4762a1bSJed Brown   ierr = DMComputeL2Diff(fdm, 0.0, exactFuncs, exactCtxs, fu, &error);CHKERRQ(ierr);
780c4762a1bSJed Brown   ierr = DMComputeL2GradientDiff(fdm, 0.0, exactFuncDers, exactCtxs, fu, n, &errorDer);CHKERRQ(ierr);
781c4762a1bSJed Brown   /* Report result */
782c4762a1bSJed Brown   if (error > tol)    {ierr = PetscPrintf(comm, "Interpolation tests FAIL for order %D at tolerance %g error %g\n", order, (double)tol, (double)error);CHKERRQ(ierr);}
783c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Interpolation tests pass for order %D at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
784c4762a1bSJed Brown   if (errorDer > tol) {ierr = PetscPrintf(comm, "Interpolation tests FAIL for order %D derivatives at tolerance %g error %g\n", order, (double)tol, (double)errorDer);CHKERRQ(ierr);}
785c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Interpolation tests pass for order %D derivatives at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
786c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(idm, &iu);CHKERRQ(ierr);
787c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(fdm, &fu);CHKERRQ(ierr);
788c4762a1bSJed Brown   ierr = MatDestroy(&Interp);CHKERRQ(ierr);
789c4762a1bSJed Brown   ierr = VecDestroy(&scaling);CHKERRQ(ierr);
790c4762a1bSJed Brown   ierr = DMDestroy(&rdm);CHKERRQ(ierr);
791c4762a1bSJed Brown   PetscFunctionReturn(0);
792c4762a1bSJed Brown }
793c4762a1bSJed Brown 
794c4762a1bSJed Brown static PetscErrorCode CheckConvergence(DM dm, PetscInt Nr, AppCtx *user)
795c4762a1bSJed Brown {
796c4762a1bSJed Brown   DM               odm = dm, rdm = NULL, cdm = NULL;
797c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx) = {trig};
798c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt dim, PetscReal time, const PetscReal x[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx) = {trigDer};
799c4762a1bSJed Brown   void            *exactCtxs[3];
800c4762a1bSJed Brown   PetscInt         r, c, cStart, cEnd;
801c4762a1bSJed Brown   PetscReal        errorOld, errorDerOld, error, errorDer, rel, len, lenOld;
802c4762a1bSJed Brown   double           p;
803c4762a1bSJed Brown   PetscErrorCode   ierr;
804c4762a1bSJed Brown 
805c4762a1bSJed Brown   PetscFunctionBeginUser;
806c4762a1bSJed Brown   if (!user->convergence) PetscFunctionReturn(0);
807c4762a1bSJed Brown   exactCtxs[0] = user;
808c4762a1bSJed Brown   exactCtxs[1] = user;
809c4762a1bSJed Brown   exactCtxs[2] = user;
810c4762a1bSJed Brown   ierr = PetscObjectReference((PetscObject) odm);CHKERRQ(ierr);
811c4762a1bSJed Brown   if (!user->convRefine) {
812c4762a1bSJed Brown     for (r = 0; r < Nr; ++r) {
813c4762a1bSJed Brown       ierr = DMRefine(odm, PetscObjectComm((PetscObject) dm), &rdm);CHKERRQ(ierr);
814c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
815c4762a1bSJed Brown       odm  = rdm;
816c4762a1bSJed Brown     }
817c4762a1bSJed Brown     ierr = SetupSection(odm, user);CHKERRQ(ierr);
818c4762a1bSJed Brown   }
819c4762a1bSJed Brown   ierr = ComputeError(odm, exactFuncs, exactFuncDers, exactCtxs, &errorOld, &errorDerOld, user);CHKERRQ(ierr);
820c4762a1bSJed Brown   if (user->convRefine) {
821c4762a1bSJed Brown     for (r = 0; r < Nr; ++r) {
822c4762a1bSJed Brown       ierr = DMRefine(odm, PetscObjectComm((PetscObject) dm), &rdm);CHKERRQ(ierr);
82330602db0SMatthew G. Knepley       if (user->useDA) {ierr = DMDASetVertexCoordinates(rdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
824c4762a1bSJed Brown       ierr = SetupSection(rdm, user);CHKERRQ(ierr);
825c4762a1bSJed Brown       ierr = ComputeError(rdm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
826c4762a1bSJed Brown       p    = PetscLog2Real(errorOld/error);
827c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Function   convergence rate at refinement %D: %.2f\n", r, (double)p);CHKERRQ(ierr);
828c4762a1bSJed Brown       p    = PetscLog2Real(errorDerOld/errorDer);
829c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Derivative convergence rate at refinement %D: %.2f\n", r, (double)p);CHKERRQ(ierr);
830c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
831c4762a1bSJed Brown       odm         = rdm;
832c4762a1bSJed Brown       errorOld    = error;
833c4762a1bSJed Brown       errorDerOld = errorDer;
834c4762a1bSJed Brown     }
835c4762a1bSJed Brown   } else {
836c4762a1bSJed Brown     /* ierr = ComputeLongestEdge(dm, &lenOld);CHKERRQ(ierr); */
837c4762a1bSJed Brown     ierr = DMPlexGetHeightStratum(odm, 0, &cStart, &cEnd);CHKERRQ(ierr);
838c4762a1bSJed Brown     lenOld = cEnd - cStart;
839c4762a1bSJed Brown     for (c = 0; c < Nr; ++c) {
840c4762a1bSJed Brown       ierr = DMCoarsen(odm, PetscObjectComm((PetscObject) dm), &cdm);CHKERRQ(ierr);
84130602db0SMatthew G. Knepley       if (user->useDA) {ierr = DMDASetVertexCoordinates(cdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
842c4762a1bSJed Brown       ierr = SetupSection(cdm, user);CHKERRQ(ierr);
843c4762a1bSJed Brown       ierr = ComputeError(cdm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
844c4762a1bSJed Brown       /* ierr = ComputeLongestEdge(cdm, &len);CHKERRQ(ierr); */
845c4762a1bSJed Brown       ierr = DMPlexGetHeightStratum(cdm, 0, &cStart, &cEnd);CHKERRQ(ierr);
846c4762a1bSJed Brown       len  = cEnd - cStart;
847c4762a1bSJed Brown       rel  = error/errorOld;
848c4762a1bSJed Brown       p    = PetscLogReal(rel) / PetscLogReal(lenOld / len);
849c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Function   convergence rate at coarsening %D: %.2f\n", c, (double)p);CHKERRQ(ierr);
850c4762a1bSJed Brown       rel  = errorDer/errorDerOld;
851c4762a1bSJed Brown       p    = PetscLogReal(rel) / PetscLogReal(lenOld / len);
852c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Derivative convergence rate at coarsening %D: %.2f\n", c, (double)p);CHKERRQ(ierr);
853c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
854c4762a1bSJed Brown       odm         = cdm;
855c4762a1bSJed Brown       errorOld    = error;
856c4762a1bSJed Brown       errorDerOld = errorDer;
857c4762a1bSJed Brown       lenOld      = len;
858c4762a1bSJed Brown     }
859c4762a1bSJed Brown   }
860c4762a1bSJed Brown   ierr = DMDestroy(&odm);CHKERRQ(ierr);
861c4762a1bSJed Brown   PetscFunctionReturn(0);
862c4762a1bSJed Brown }
863c4762a1bSJed Brown 
864c4762a1bSJed Brown int main(int argc, char **argv)
865c4762a1bSJed Brown {
866c4762a1bSJed Brown   DM             dm;
867c4762a1bSJed Brown   AppCtx         user;                 /* user-defined work context */
86830602db0SMatthew G. Knepley   PetscInt       dim = 2;
86930602db0SMatthew G. Knepley   PetscBool      simplex = PETSC_FALSE;
870c4762a1bSJed Brown   PetscErrorCode ierr;
871c4762a1bSJed Brown 
872c4762a1bSJed Brown   ierr = PetscInitialize(&argc, &argv, NULL, help);if (ierr) return ierr;
873c4762a1bSJed Brown   ierr = ProcessOptions(PETSC_COMM_WORLD, &user);CHKERRQ(ierr);
874c4762a1bSJed Brown   ierr = CreateMesh(PETSC_COMM_WORLD, &user, &dm);CHKERRQ(ierr);
87530602db0SMatthew G. Knepley   if (!user.useDA) {
87630602db0SMatthew G. Knepley     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
87730602db0SMatthew G. Knepley     ierr = DMPlexIsSimplex(dm, &simplex);CHKERRQ(ierr);
87830602db0SMatthew G. Knepley   }
87930602db0SMatthew G. Knepley   user.numComponents = user.numComponents < 0 ? dim : user.numComponents;
88030602db0SMatthew G. Knepley   ierr = PetscFECreateDefault(PETSC_COMM_WORLD, dim, user.numComponents, simplex, NULL, user.qorder, &user.fe);CHKERRQ(ierr);
881c4762a1bSJed Brown   ierr = SetupSection(dm, &user);CHKERRQ(ierr);
882c4762a1bSJed Brown   if (user.testFEjacobian) {ierr = TestFEJacobian(dm, &user);CHKERRQ(ierr);}
883c4762a1bSJed Brown   if (user.testFVgrad) {ierr = TestFVGrad(dm, &user);CHKERRQ(ierr);}
884c4762a1bSJed Brown   if (user.testInjector) {ierr = TestInjector(dm, &user);CHKERRQ(ierr);}
885c4762a1bSJed Brown   ierr = CheckFunctions(dm, user.porder, &user);CHKERRQ(ierr);
886c4762a1bSJed Brown   {
887c4762a1bSJed Brown     PetscDualSpace dsp;
888c4762a1bSJed Brown     PetscInt       k;
889c4762a1bSJed Brown 
890c4762a1bSJed Brown     ierr = PetscFEGetDualSpace(user.fe, &dsp);CHKERRQ(ierr);
891c4762a1bSJed Brown     ierr = PetscDualSpaceGetDeRahm(dsp, &k);CHKERRQ(ierr);
89230602db0SMatthew G. Knepley     if (dim == 2 && simplex == PETSC_TRUE && user.tree == PETSC_FALSE && k == 0) {
893c4762a1bSJed Brown       ierr = CheckInterpolation(dm, PETSC_FALSE, user.porder, &user);CHKERRQ(ierr);
894c4762a1bSJed Brown       ierr = CheckInterpolation(dm, PETSC_TRUE,  user.porder, &user);CHKERRQ(ierr);
895c4762a1bSJed Brown     }
896c4762a1bSJed Brown   }
897c4762a1bSJed Brown   ierr = CheckConvergence(dm, 3, &user);CHKERRQ(ierr);
898c4762a1bSJed Brown   ierr = PetscFEDestroy(&user.fe);CHKERRQ(ierr);
899c4762a1bSJed Brown   ierr = DMDestroy(&dm);CHKERRQ(ierr);
900c4762a1bSJed Brown   ierr = PetscFinalize();
901c4762a1bSJed Brown   return ierr;
902c4762a1bSJed Brown }
903c4762a1bSJed Brown 
904c4762a1bSJed Brown /*TEST
905c4762a1bSJed Brown 
906c4762a1bSJed Brown   test:
907c4762a1bSJed Brown     suffix: 1
908c4762a1bSJed Brown     requires: triangle
909c4762a1bSJed Brown 
910c4762a1bSJed Brown   # 2D P_1 on a triangle
911c4762a1bSJed Brown   test:
912c4762a1bSJed Brown     suffix: p1_2d_0
913c4762a1bSJed Brown     requires: triangle
914c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -convergence
915c4762a1bSJed Brown   test:
916c4762a1bSJed Brown     suffix: p1_2d_1
917c4762a1bSJed Brown     requires: triangle
918c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -porder 1
919c4762a1bSJed Brown   test:
920c4762a1bSJed Brown     suffix: p1_2d_2
921c4762a1bSJed Brown     requires: triangle
922c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -porder 2
923c4762a1bSJed Brown   test:
924c4762a1bSJed Brown     suffix: p1_2d_3
925c4762a1bSJed Brown     requires: triangle pragmatic
926c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
927c4762a1bSJed Brown     filter: grep -v DEBUG
928c4762a1bSJed Brown   test:
929c4762a1bSJed Brown     suffix: p1_2d_4
930c4762a1bSJed Brown     requires: triangle pragmatic
931c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
932c4762a1bSJed Brown   test:
933c4762a1bSJed Brown     suffix: p1_2d_5
934c4762a1bSJed Brown     requires: triangle pragmatic
935c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
936c4762a1bSJed Brown 
937c4762a1bSJed Brown   # 3D P_1 on a tetrahedron
938c4762a1bSJed Brown   test:
939c4762a1bSJed Brown     suffix: p1_3d_0
940c4762a1bSJed Brown     requires: ctetgen
94130602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -convergence
942c4762a1bSJed Brown   test:
943c4762a1bSJed Brown     suffix: p1_3d_1
944c4762a1bSJed Brown     requires: ctetgen
94530602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -porder 1
946c4762a1bSJed Brown   test:
947c4762a1bSJed Brown     suffix: p1_3d_2
948c4762a1bSJed Brown     requires: ctetgen
94930602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -porder 2
950c4762a1bSJed Brown   test:
951c4762a1bSJed Brown     suffix: p1_3d_3
952c4762a1bSJed Brown     requires: ctetgen pragmatic
95330602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
954c4762a1bSJed Brown     filter: grep -v DEBUG
955c4762a1bSJed Brown   test:
956c4762a1bSJed Brown     suffix: p1_3d_4
957c4762a1bSJed Brown     requires: ctetgen pragmatic
95830602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
959c4762a1bSJed Brown   test:
960c4762a1bSJed Brown     suffix: p1_3d_5
961c4762a1bSJed Brown     requires: ctetgen pragmatic
96230602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
963c4762a1bSJed Brown 
964c4762a1bSJed Brown   # 2D P_2 on a triangle
965c4762a1bSJed Brown   test:
966c4762a1bSJed Brown     suffix: p2_2d_0
967c4762a1bSJed Brown     requires: triangle
968c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -convergence
969c4762a1bSJed Brown   test:
970c4762a1bSJed Brown     suffix: p2_2d_1
971c4762a1bSJed Brown     requires: triangle
972c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -porder 1
973c4762a1bSJed Brown   test:
974c4762a1bSJed Brown     suffix: p2_2d_2
975c4762a1bSJed Brown     requires: triangle
976c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -porder 2
977c4762a1bSJed Brown   test:
978c4762a1bSJed Brown     suffix: p2_2d_3
979c4762a1bSJed Brown     requires: triangle pragmatic
980c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -convergence -conv_refine 0
981c4762a1bSJed Brown     filter: grep -v DEBUG
982c4762a1bSJed Brown   test:
983c4762a1bSJed Brown     suffix: p2_2d_4
984c4762a1bSJed Brown     requires: triangle pragmatic
985c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 1 -conv_refine 0
986c4762a1bSJed Brown   test:
987c4762a1bSJed Brown     suffix: p2_2d_5
988c4762a1bSJed Brown     requires: triangle pragmatic
989c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 2 -conv_refine 0
990c4762a1bSJed Brown 
991c4762a1bSJed Brown   # 3D P_2 on a tetrahedron
992c4762a1bSJed Brown   test:
993c4762a1bSJed Brown     suffix: p2_3d_0
994c4762a1bSJed Brown     requires: ctetgen
99530602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -convergence
996c4762a1bSJed Brown   test:
997c4762a1bSJed Brown     suffix: p2_3d_1
998c4762a1bSJed Brown     requires: ctetgen
99930602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -porder 1
1000c4762a1bSJed Brown   test:
1001c4762a1bSJed Brown     suffix: p2_3d_2
1002c4762a1bSJed Brown     requires: ctetgen
100330602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -porder 2
1004c4762a1bSJed Brown   test:
1005c4762a1bSJed Brown     suffix: p2_3d_3
1006c4762a1bSJed Brown     requires: ctetgen pragmatic
100730602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -convergence -conv_refine 0
1008c4762a1bSJed Brown     filter: grep -v DEBUG
1009c4762a1bSJed Brown   test:
1010c4762a1bSJed Brown     suffix: p2_3d_4
1011c4762a1bSJed Brown     requires: ctetgen pragmatic
101230602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 1 -conv_refine 0
1013c4762a1bSJed Brown   test:
1014c4762a1bSJed Brown     suffix: p2_3d_5
1015c4762a1bSJed Brown     requires: ctetgen pragmatic
101630602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 2 -conv_refine 0
1017c4762a1bSJed Brown 
1018c4762a1bSJed Brown   # 2D Q_1 on a quadrilaterial DA
1019c4762a1bSJed Brown   test:
1020c4762a1bSJed Brown     suffix: q1_2d_da_0
1021c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
102230602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -convergence
1023c4762a1bSJed Brown   test:
1024c4762a1bSJed Brown     suffix: q1_2d_da_1
1025c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
102630602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -porder 1
1027c4762a1bSJed Brown   test:
1028c4762a1bSJed Brown     suffix: q1_2d_da_2
1029c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
103030602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -porder 2
1031c4762a1bSJed Brown 
1032c4762a1bSJed Brown   # 2D Q_1 on a quadrilaterial Plex
1033c4762a1bSJed Brown   test:
1034c4762a1bSJed Brown     suffix: q1_2d_plex_0
103530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -convergence
1036c4762a1bSJed Brown   test:
1037c4762a1bSJed Brown     suffix: q1_2d_plex_1
103830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 1
1039c4762a1bSJed Brown   test:
1040c4762a1bSJed Brown     suffix: q1_2d_plex_2
104130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 2
1042c4762a1bSJed Brown   test:
1043c4762a1bSJed Brown     suffix: q1_2d_plex_3
104430602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 1 -shear_coords
1045c4762a1bSJed Brown   test:
1046c4762a1bSJed Brown     suffix: q1_2d_plex_4
104730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 2 -shear_coords
1048c4762a1bSJed Brown   test:
1049c4762a1bSJed Brown     suffix: q1_2d_plex_5
105030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 0 -non_affine_coords -convergence
1051c4762a1bSJed Brown   test:
1052c4762a1bSJed Brown     suffix: q1_2d_plex_6
105330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 1 -non_affine_coords -convergence
1054c4762a1bSJed Brown   test:
1055c4762a1bSJed Brown     suffix: q1_2d_plex_7
105630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 2 -non_affine_coords -convergence
1057c4762a1bSJed Brown 
1058c4762a1bSJed Brown   # 2D Q_2 on a quadrilaterial
1059c4762a1bSJed Brown   test:
1060c4762a1bSJed Brown     suffix: q2_2d_plex_0
1061c4762a1bSJed Brown     requires: mpi_type_get_envelope
106230602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -convergence
1063c4762a1bSJed Brown   test:
1064c4762a1bSJed Brown     suffix: q2_2d_plex_1
1065c4762a1bSJed Brown     requires: mpi_type_get_envelope
106630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1
1067c4762a1bSJed Brown   test:
1068c4762a1bSJed Brown     suffix: q2_2d_plex_2
1069c4762a1bSJed Brown     requires: mpi_type_get_envelope
107030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2
1071c4762a1bSJed Brown   test:
1072c4762a1bSJed Brown     suffix: q2_2d_plex_3
107330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1 -shear_coords
1074c4762a1bSJed Brown   test:
1075c4762a1bSJed Brown     suffix: q2_2d_plex_4
1076c4762a1bSJed Brown     requires: mpi_type_get_envelope
107730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2 -shear_coords
1078c4762a1bSJed Brown   test:
1079c4762a1bSJed Brown     suffix: q2_2d_plex_5
1080c4762a1bSJed Brown     requires: mpi_type_get_envelope
108130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 0 -non_affine_coords -convergence
1082c4762a1bSJed Brown   test:
1083c4762a1bSJed Brown     suffix: q2_2d_plex_6
1084c4762a1bSJed Brown     requires: mpi_type_get_envelope
108530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 1 -non_affine_coords -convergence
1086c4762a1bSJed Brown   test:
1087c4762a1bSJed Brown     suffix: q2_2d_plex_7
1088c4762a1bSJed Brown     requires: mpi_type_get_envelope
108930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 2 -non_affine_coords -convergence
1090c4762a1bSJed Brown 
1091c4762a1bSJed Brown 
1092c4762a1bSJed Brown   # 2D P_3 on a triangle
1093c4762a1bSJed Brown   test:
1094c4762a1bSJed Brown     suffix: p3_2d_0
1095c4762a1bSJed Brown     requires: triangle !single
1096c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -convergence
1097c4762a1bSJed Brown   test:
1098c4762a1bSJed Brown     suffix: p3_2d_1
1099c4762a1bSJed Brown     requires: triangle !single
1100c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 1
1101c4762a1bSJed Brown   test:
1102c4762a1bSJed Brown     suffix: p3_2d_2
1103c4762a1bSJed Brown     requires: triangle !single
1104c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 2
1105c4762a1bSJed Brown   test:
1106c4762a1bSJed Brown     suffix: p3_2d_3
1107c4762a1bSJed Brown     requires: triangle !single
1108c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 3
1109c4762a1bSJed Brown   test:
1110c4762a1bSJed Brown     suffix: p3_2d_4
1111c4762a1bSJed Brown     requires: triangle pragmatic
1112c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -convergence -conv_refine 0
1113c4762a1bSJed Brown     filter: grep -v DEBUG
1114c4762a1bSJed Brown   test:
1115c4762a1bSJed Brown     suffix: p3_2d_5
1116c4762a1bSJed Brown     requires: triangle pragmatic
1117c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -porder 1 -conv_refine 0
1118c4762a1bSJed Brown   test:
1119c4762a1bSJed Brown     suffix: p3_2d_6
1120c4762a1bSJed Brown     requires: triangle pragmatic
1121c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -porder 3 -conv_refine 0
1122c4762a1bSJed Brown 
1123c4762a1bSJed Brown   # 2D Q_3 on a quadrilaterial
1124c4762a1bSJed Brown   test:
1125c4762a1bSJed Brown     suffix: q3_2d_0
1126c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
112730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -convergence
1128c4762a1bSJed Brown   test:
1129c4762a1bSJed Brown     suffix: q3_2d_1
1130c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
113130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 1
1132c4762a1bSJed Brown   test:
1133c4762a1bSJed Brown     suffix: q3_2d_2
1134c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
113530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 2
1136c4762a1bSJed Brown   test:
1137c4762a1bSJed Brown     suffix: q3_2d_3
1138c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
113930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 3
1140c4762a1bSJed Brown 
1141c4762a1bSJed Brown   # 2D P_1disc on a triangle/quadrilateral
1142c4762a1bSJed Brown   test:
1143c4762a1bSJed Brown     suffix: p1d_2d_0
1144c4762a1bSJed Brown     requires: triangle
1145c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -convergence
1146c4762a1bSJed Brown   test:
1147c4762a1bSJed Brown     suffix: p1d_2d_1
1148c4762a1bSJed Brown     requires: triangle
1149c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 1
1150c4762a1bSJed Brown   test:
1151c4762a1bSJed Brown     suffix: p1d_2d_2
1152c4762a1bSJed Brown     requires: triangle
1153c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 2
1154c4762a1bSJed Brown   test:
1155c4762a1bSJed Brown     suffix: p1d_2d_3
1156c4762a1bSJed Brown     requires: triangle
115730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -convergence
1158c4762a1bSJed Brown     filter: sed  -e "s/convergence rate at refinement 0: 2/convergence rate at refinement 0: 1.9/g"
1159c4762a1bSJed Brown   test:
1160c4762a1bSJed Brown     suffix: p1d_2d_4
1161c4762a1bSJed Brown     requires: triangle
116230602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 1
1163c4762a1bSJed Brown   test:
1164c4762a1bSJed Brown     suffix: p1d_2d_5
1165c4762a1bSJed Brown     requires: triangle
116630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 2
1167c4762a1bSJed Brown 
1168c4762a1bSJed Brown   # 2D BDM_1 on a triangle
1169c4762a1bSJed Brown   test:
1170c4762a1bSJed Brown     suffix: bdm1_2d_0
1171c4762a1bSJed Brown     requires: triangle
1172c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1173c4762a1bSJed Brown           -num_comp 2 -qorder 1 -convergence
1174c4762a1bSJed Brown   test:
1175c4762a1bSJed Brown     suffix: bdm1_2d_1
1176c4762a1bSJed Brown     requires: triangle
1177c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1178c4762a1bSJed Brown           -num_comp 2 -qorder 1 -porder 1
1179c4762a1bSJed Brown   test:
1180c4762a1bSJed Brown     suffix: bdm1_2d_2
1181c4762a1bSJed Brown     requires: triangle
1182c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1183c4762a1bSJed Brown           -num_comp 2 -qorder 1 -porder 2
1184c4762a1bSJed Brown 
1185c4762a1bSJed Brown   # 2D BDM_1 on a quadrilateral
1186c4762a1bSJed Brown   test:
1187c4762a1bSJed Brown     suffix: bdm1q_2d_0
1188c4762a1bSJed Brown     requires: triangle
1189c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
11903f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
119130602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -convergence
1192c4762a1bSJed Brown   test:
1193c4762a1bSJed Brown     suffix: bdm1q_2d_1
1194c4762a1bSJed Brown     requires: triangle
1195c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
11963f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
119730602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -porder 1
1198c4762a1bSJed Brown   test:
1199c4762a1bSJed Brown     suffix: bdm1q_2d_2
1200c4762a1bSJed Brown     requires: triangle
1201c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
12023f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
120330602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -porder 2
1204c4762a1bSJed Brown 
1205c4762a1bSJed Brown   # Test high order quadrature
1206c4762a1bSJed Brown   test:
1207c4762a1bSJed Brown     suffix: p1_quad_2
1208c4762a1bSJed Brown     requires: triangle
1209c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 2 -porder 1
1210c4762a1bSJed Brown   test:
1211c4762a1bSJed Brown     suffix: p1_quad_5
1212c4762a1bSJed Brown     requires: triangle
1213c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 5 -porder 1
1214c4762a1bSJed Brown   test:
1215c4762a1bSJed Brown     suffix: p2_quad_3
1216c4762a1bSJed Brown     requires: triangle
1217c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 3 -porder 2
1218c4762a1bSJed Brown   test:
1219c4762a1bSJed Brown     suffix: p2_quad_5
1220c4762a1bSJed Brown     requires: triangle
1221c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 5 -porder 2
1222c4762a1bSJed Brown   test:
1223c4762a1bSJed Brown     suffix: q1_quad_2
1224c4762a1bSJed Brown     requires: mpi_type_get_envelope
122530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 2 -porder 1
1226c4762a1bSJed Brown   test:
1227c4762a1bSJed Brown     suffix: q1_quad_5
1228c4762a1bSJed Brown     requires: mpi_type_get_envelope
122930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 5 -porder 1
1230c4762a1bSJed Brown   test:
1231c4762a1bSJed Brown     suffix: q2_quad_3
1232c4762a1bSJed Brown     requires: mpi_type_get_envelope
123330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 3 -porder 1
1234c4762a1bSJed Brown   test:
1235c4762a1bSJed Brown     suffix: q2_quad_5
1236c4762a1bSJed Brown     requires: mpi_type_get_envelope
123730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 5 -porder 1
1238c4762a1bSJed Brown 
1239c4762a1bSJed Brown 
1240c4762a1bSJed Brown   # Nonconforming tests
1241c4762a1bSJed Brown   test:
1242c4762a1bSJed Brown     suffix: constraints
124330602db0SMatthew G. Knepley     args: -dm_coord_space 0 -dm_plex_simplex 0 -petscspace_type tensor -petscspace_degree 1 -qorder 0 -constraints
1244c4762a1bSJed Brown   test:
1245c4762a1bSJed Brown     suffix: nonconforming_tensor_2
1246c4762a1bSJed Brown     nsize: 4
124730602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -test_injector -petscpartitioner_type simple -tree -dm_plex_simplex 0 -dm_plex_max_projection_height 1 -petscspace_type tensor -petscspace_degree 2 -qorder 2 -dm_view ascii::ASCII_INFO_DETAIL
1248c4762a1bSJed Brown   test:
1249c4762a1bSJed Brown     suffix: nonconforming_tensor_3
1250c4762a1bSJed Brown     nsize: 4
125130602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_simplex 0 -dm_plex_dim 3 -dm_plex_box_faces 2,2,2 -dm_plex_max_projection_height 2 -petscspace_type tensor -petscspace_degree 1 -qorder 1 -dm_view ascii::ASCII_INFO_DETAIL
1252c4762a1bSJed Brown   test:
1253c4762a1bSJed Brown     suffix: nonconforming_tensor_2_fv
1254c4762a1bSJed Brown     nsize: 4
125530602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fv_grad -test_injector -petsclimiter_type none -petscpartitioner_type simple -tree -dm_plex_simplex 0 -num_comp 2
1256c4762a1bSJed Brown   test:
1257c4762a1bSJed Brown     suffix: nonconforming_tensor_3_fv
1258c4762a1bSJed Brown     nsize: 4
125930602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fv_grad -test_injector -petsclimiter_type none -petscpartitioner_type simple -tree -dm_plex_simplex 0 -dm_plex_dim 3 -dm_plex_box_faces 2,2,2 -num_comp 3
1260c4762a1bSJed Brown   test:
1261c4762a1bSJed Brown     suffix: nonconforming_tensor_2_hi
1262c4762a1bSJed Brown     requires: !single
1263c4762a1bSJed Brown     nsize: 4
126430602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_simplex 0 -dm_plex_max_projection_height 1 -petscspace_type tensor -petscspace_degree 4 -qorder 4
1265c4762a1bSJed Brown   test:
1266c4762a1bSJed Brown     suffix: nonconforming_tensor_3_hi
1267c4762a1bSJed Brown     requires: !single skip
1268c4762a1bSJed Brown     nsize: 4
126930602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_simplex 0 -dm_plex_dim 3 -dm_plex_box_faces 2,2,2 -dm_plex_max_projection_height 2 -petscspace_type tensor -petscspace_degree 4 -qorder 4
1270c4762a1bSJed Brown   test:
1271c4762a1bSJed Brown     suffix: nonconforming_simplex_2
1272c4762a1bSJed Brown     requires: triangle
1273c4762a1bSJed Brown     nsize: 4
127430602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -test_injector -petscpartitioner_type simple -tree -dm_plex_max_projection_height 1 -petscspace_degree 2 -qorder 2 -dm_view ascii::ASCII_INFO_DETAIL
1275c4762a1bSJed Brown   test:
1276c4762a1bSJed Brown     suffix: nonconforming_simplex_2_hi
1277c4762a1bSJed Brown     requires: triangle !single
1278c4762a1bSJed Brown     nsize: 4
127930602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_max_projection_height 1 -petscspace_degree 4 -qorder 4
1280c4762a1bSJed Brown   test:
1281c4762a1bSJed Brown     suffix: nonconforming_simplex_2_fv
1282c4762a1bSJed Brown     requires: triangle
1283c4762a1bSJed Brown     nsize: 4
128430602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fv_grad -test_injector -petsclimiter_type none -petscpartitioner_type simple -tree -num_comp 2
1285c4762a1bSJed Brown   test:
1286c4762a1bSJed Brown     suffix: nonconforming_simplex_3
1287c4762a1bSJed Brown     requires: ctetgen
1288c4762a1bSJed Brown     nsize: 4
128930602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -test_injector -petscpartitioner_type simple -tree -dm_plex_dim 3 -dm_plex_max_projection_height 2 -petscspace_degree 2 -qorder 2 -dm_view ascii::ASCII_INFO_DETAIL
1290c4762a1bSJed Brown   test:
1291c4762a1bSJed Brown     suffix: nonconforming_simplex_3_hi
1292c4762a1bSJed Brown     requires: ctetgen skip
1293c4762a1bSJed Brown     nsize: 4
129430602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_dim 3 -dm_plex_max_projection_height 2 -petscspace_degree 4 -qorder 4
1295c4762a1bSJed Brown   test:
1296c4762a1bSJed Brown     suffix: nonconforming_simplex_3_fv
1297c4762a1bSJed Brown     requires: ctetgen
1298c4762a1bSJed Brown     nsize: 4
129930602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fv_grad -test_injector -petsclimiter_type none -petscpartitioner_type simple -tree -dm_plex_dim 3 -num_comp 3
1300c4762a1bSJed Brown 
1301c4762a1bSJed Brown TEST*/
1302c4762a1bSJed Brown 
1303c4762a1bSJed Brown /*
1304c4762a1bSJed Brown    # 2D Q_2 on a quadrilaterial Plex
1305c4762a1bSJed Brown   test:
1306c4762a1bSJed Brown     suffix: q2_2d_plex_0
130730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -convergence
1308c4762a1bSJed Brown   test:
1309c4762a1bSJed Brown     suffix: q2_2d_plex_1
131030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1
1311c4762a1bSJed Brown   test:
1312c4762a1bSJed Brown     suffix: q2_2d_plex_2
131330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2
1314c4762a1bSJed Brown   test:
1315c4762a1bSJed Brown     suffix: q2_2d_plex_3
131630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1 -shear_coords
1317c4762a1bSJed Brown   test:
1318c4762a1bSJed Brown     suffix: q2_2d_plex_4
131930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2 -shear_coords
1320c4762a1bSJed Brown   test:
1321c4762a1bSJed Brown     suffix: q2_2d_plex_5
132230602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 0 -non_affine_coords
1323c4762a1bSJed Brown   test:
1324c4762a1bSJed Brown     suffix: q2_2d_plex_6
132530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 1 -non_affine_coords
1326c4762a1bSJed Brown   test:
1327c4762a1bSJed Brown     suffix: q2_2d_plex_7
132830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 2 -non_affine_coords
1329c4762a1bSJed Brown 
1330c4762a1bSJed Brown   test:
1331c4762a1bSJed Brown     suffix: p1d_2d_6
1332c4762a1bSJed Brown     requires: pragmatic
1333c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
1334c4762a1bSJed Brown   test:
1335c4762a1bSJed Brown     suffix: p1d_2d_7
1336c4762a1bSJed Brown     requires: pragmatic
1337c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
1338c4762a1bSJed Brown   test:
1339c4762a1bSJed Brown     suffix: p1d_2d_8
1340c4762a1bSJed Brown     requires: pragmatic
1341c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
1342c4762a1bSJed Brown */
1343