xref: /petsc/src/dm/impls/plex/tests/ex3.c (revision 012bc3648b56b7befe35b618b2ec944650f76ae7)
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);
253*012bc364SMatthew G. Knepley       ierr = DMViewFromOptions(refTree,NULL,"-reftree_dm_view");CHKERRQ(ierr);
254c4762a1bSJed Brown       ierr = DMPlexSetReferenceTree(*dm,refTree);CHKERRQ(ierr);
255c4762a1bSJed Brown       ierr = DMDestroy(&refTree);CHKERRQ(ierr);
256c4762a1bSJed Brown       ierr = DMPlexTreeRefineCell(*dm,user->treeCell,&ncdm);CHKERRQ(ierr);
257c4762a1bSJed Brown       if (ncdm) {
258c4762a1bSJed Brown         ierr = DMDestroy(dm);CHKERRQ(ierr);
259c4762a1bSJed Brown         *dm = ncdm;
260c4762a1bSJed Brown         ierr = DMPlexSetRefinementUniform(*dm, PETSC_FALSE);CHKERRQ(ierr);
261c4762a1bSJed Brown       }
262*012bc364SMatthew G. Knepley       ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "tree_");CHKERRQ(ierr);
263*012bc364SMatthew G. Knepley       ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
264*012bc364SMatthew G. Knepley       ierr = DMViewFromOptions(*dm,NULL,"-dm_view");CHKERRQ(ierr);
265c4762a1bSJed Brown     } else {
266c4762a1bSJed Brown       ierr = DMPlexSetRefinementUniform(*dm, PETSC_TRUE);CHKERRQ(ierr);
267c4762a1bSJed Brown     }
26830602db0SMatthew G. Knepley     ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "dist_");CHKERRQ(ierr);
26930602db0SMatthew G. Knepley     ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
27030602db0SMatthew G. Knepley     ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, NULL);CHKERRQ(ierr);
27130602db0SMatthew G. Knepley     if (simplex) {ierr = PetscObjectSetName((PetscObject) *dm, "Simplicial Mesh");CHKERRQ(ierr);}
27230602db0SMatthew G. Knepley     else         {ierr = PetscObjectSetName((PetscObject) *dm, "Hexahedral Mesh");CHKERRQ(ierr);}
273c4762a1bSJed Brown   }
274c4762a1bSJed Brown   ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
275c4762a1bSJed Brown   ierr = TransformCoordinates(*dm, user);CHKERRQ(ierr);
276c4762a1bSJed Brown   ierr = DMViewFromOptions(*dm,NULL,"-dm_view");CHKERRQ(ierr);
277c4762a1bSJed Brown   PetscFunctionReturn(0);
278c4762a1bSJed Brown }
279c4762a1bSJed Brown 
280c4762a1bSJed Brown static void simple_mass(PetscInt dim, PetscInt Nf, PetscInt NfAux,
281c4762a1bSJed Brown                         const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
282c4762a1bSJed Brown                         const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
283c4762a1bSJed Brown                         PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[])
284c4762a1bSJed Brown {
285c4762a1bSJed Brown   PetscInt d, e;
286c4762a1bSJed Brown   for (d = 0, e = 0; d < dim; d++, e+=dim+1) {
287c4762a1bSJed Brown     g0[e] = 1.;
288c4762a1bSJed Brown   }
289c4762a1bSJed Brown }
290c4762a1bSJed Brown 
291c4762a1bSJed Brown /* < \nabla v, 1/2(\nabla u + {\nabla u}^T) > */
292c4762a1bSJed Brown static void symmetric_gradient_inner_product(PetscInt dim, PetscInt Nf, PetscInt NfAux,
293c4762a1bSJed Brown                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
294c4762a1bSJed Brown                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
295c4762a1bSJed Brown                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar C[])
296c4762a1bSJed Brown {
297c4762a1bSJed Brown   PetscInt compI, compJ, d, e;
298c4762a1bSJed Brown 
299c4762a1bSJed Brown   for (compI = 0; compI < dim; ++compI) {
300c4762a1bSJed Brown     for (compJ = 0; compJ < dim; ++compJ) {
301c4762a1bSJed Brown       for (d = 0; d < dim; ++d) {
302c4762a1bSJed Brown         for (e = 0; e < dim; e++) {
303c4762a1bSJed Brown           if (d == e && d == compI && d == compJ) {
304c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 1.0;
305c4762a1bSJed Brown           } else if ((d == compJ && e == compI) || (d == e && compI == compJ)) {
306c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 0.5;
307c4762a1bSJed Brown           } else {
308c4762a1bSJed Brown             C[((compI*dim+compJ)*dim+d)*dim+e] = 0.0;
309c4762a1bSJed Brown           }
310c4762a1bSJed Brown         }
311c4762a1bSJed Brown       }
312c4762a1bSJed Brown     }
313c4762a1bSJed Brown   }
314c4762a1bSJed Brown }
315c4762a1bSJed Brown 
316c4762a1bSJed Brown static PetscErrorCode SetupSection(DM dm, AppCtx *user)
317c4762a1bSJed Brown {
318c4762a1bSJed Brown   PetscErrorCode ierr;
319c4762a1bSJed Brown 
320c4762a1bSJed Brown   PetscFunctionBeginUser;
32130602db0SMatthew G. Knepley   if (user->constraints) {
322c4762a1bSJed Brown     /* test local constraints */
323c4762a1bSJed Brown     DM            coordDM;
324c4762a1bSJed Brown     PetscInt      fStart, fEnd, f, vStart, vEnd, v;
325c4762a1bSJed Brown     PetscInt      edgesx = 2, vertsx;
326c4762a1bSJed Brown     PetscInt      edgesy = 2, vertsy;
327c4762a1bSJed Brown     PetscMPIInt   size;
328c4762a1bSJed Brown     PetscInt      numConst;
329c4762a1bSJed Brown     PetscSection  aSec;
330c4762a1bSJed Brown     PetscInt     *anchors;
331c4762a1bSJed Brown     PetscInt      offset;
332c4762a1bSJed Brown     IS            aIS;
333c4762a1bSJed Brown     MPI_Comm      comm = PetscObjectComm((PetscObject)dm);
334c4762a1bSJed Brown 
335ffc4695bSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
336c4762a1bSJed Brown     if (size > 1) SETERRQ(comm,PETSC_ERR_SUP,"Local constraint test can only be performed in serial");
337c4762a1bSJed Brown 
338c4762a1bSJed Brown     /* we are going to test constraints by using them to enforce periodicity
339c4762a1bSJed Brown      * in one direction, and comparing to the existing method of enforcing
340c4762a1bSJed Brown      * periodicity */
341c4762a1bSJed Brown 
342c4762a1bSJed Brown     /* first create the coordinate section so that it does not clone the
343c4762a1bSJed Brown      * constraints */
344c4762a1bSJed Brown     ierr = DMGetCoordinateDM(dm,&coordDM);CHKERRQ(ierr);
345c4762a1bSJed Brown 
346c4762a1bSJed Brown     /* create the constrained-to-anchor section */
347c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
348c4762a1bSJed Brown     ierr = DMPlexGetDepthStratum(dm,1,&fStart,&fEnd);CHKERRQ(ierr);
349c4762a1bSJed Brown     ierr = PetscSectionCreate(PETSC_COMM_SELF,&aSec);CHKERRQ(ierr);
350c4762a1bSJed Brown     ierr = PetscSectionSetChart(aSec,PetscMin(fStart,vStart),PetscMax(fEnd,vEnd));CHKERRQ(ierr);
351c4762a1bSJed Brown 
352c4762a1bSJed Brown     /* define the constraints */
353c4762a1bSJed Brown     ierr = PetscOptionsGetInt(NULL,NULL,"-da_grid_x",&edgesx,NULL);CHKERRQ(ierr);
354c4762a1bSJed Brown     ierr = PetscOptionsGetInt(NULL,NULL,"-da_grid_y",&edgesy,NULL);CHKERRQ(ierr);
355c4762a1bSJed Brown     vertsx = edgesx + 1;
356c4762a1bSJed Brown     vertsy = edgesy + 1;
357c4762a1bSJed Brown     numConst = vertsy + edgesy;
358c4762a1bSJed Brown     ierr = PetscMalloc1(numConst,&anchors);CHKERRQ(ierr);
359c4762a1bSJed Brown     offset = 0;
360c4762a1bSJed Brown     for (v = vStart + edgesx; v < vEnd; v+= vertsx) {
361c4762a1bSJed Brown       ierr = PetscSectionSetDof(aSec,v,1);CHKERRQ(ierr);
362c4762a1bSJed Brown       anchors[offset++] = v - edgesx;
363c4762a1bSJed Brown     }
364c4762a1bSJed Brown     for (f = fStart + edgesx * vertsy + edgesx * edgesy; f < fEnd; f++) {
365c4762a1bSJed Brown       ierr = PetscSectionSetDof(aSec,f,1);CHKERRQ(ierr);
366c4762a1bSJed Brown       anchors[offset++] = f - edgesx * edgesy;
367c4762a1bSJed Brown     }
368c4762a1bSJed Brown     ierr = PetscSectionSetUp(aSec);CHKERRQ(ierr);
369c4762a1bSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,numConst,anchors,PETSC_OWN_POINTER,&aIS);CHKERRQ(ierr);
370c4762a1bSJed Brown 
371c4762a1bSJed Brown     ierr = DMPlexSetAnchors(dm,aSec,aIS);CHKERRQ(ierr);
372c4762a1bSJed Brown     ierr = PetscSectionDestroy(&aSec);CHKERRQ(ierr);
373c4762a1bSJed Brown     ierr = ISDestroy(&aIS);CHKERRQ(ierr);
374c4762a1bSJed Brown   }
375c4762a1bSJed Brown   ierr = DMSetNumFields(dm, 1);CHKERRQ(ierr);
376c4762a1bSJed Brown   ierr = DMSetField(dm, 0, NULL, (PetscObject) user->fe);CHKERRQ(ierr);
377c4762a1bSJed Brown   ierr = DMCreateDS(dm);CHKERRQ(ierr);
37830602db0SMatthew G. Knepley   if (user->constraints) {
379c4762a1bSJed Brown     /* test getting local constraint matrix that matches section */
380c4762a1bSJed Brown     PetscSection aSec;
381c4762a1bSJed Brown     IS           aIS;
382c4762a1bSJed Brown 
383c4762a1bSJed Brown     ierr = DMPlexGetAnchors(dm,&aSec,&aIS);CHKERRQ(ierr);
384c4762a1bSJed Brown     if (aSec) {
385c4762a1bSJed Brown       PetscDS         ds;
386c4762a1bSJed Brown       PetscSection    cSec, section;
387c4762a1bSJed Brown       PetscInt        cStart, cEnd, c, numComp;
388c4762a1bSJed Brown       Mat             cMat, mass;
389c4762a1bSJed Brown       Vec             local;
390c4762a1bSJed Brown       const PetscInt *anchors;
391c4762a1bSJed Brown 
392c4762a1bSJed Brown       ierr = DMGetLocalSection(dm,&section);CHKERRQ(ierr);
393c4762a1bSJed Brown       /* this creates the matrix and preallocates the matrix structure: we
394c4762a1bSJed Brown        * just have to fill in the values */
395c4762a1bSJed Brown       ierr = DMGetDefaultConstraints(dm,&cSec,&cMat);CHKERRQ(ierr);
396c4762a1bSJed Brown       ierr = PetscSectionGetChart(cSec,&cStart,&cEnd);CHKERRQ(ierr);
397c4762a1bSJed Brown       ierr = ISGetIndices(aIS,&anchors);CHKERRQ(ierr);
398c4762a1bSJed Brown       ierr = PetscFEGetNumComponents(user->fe, &numComp);CHKERRQ(ierr);
399c4762a1bSJed Brown       for (c = cStart; c < cEnd; c++) {
400c4762a1bSJed Brown         PetscInt cDof;
401c4762a1bSJed Brown 
402c4762a1bSJed Brown         /* is this point constrained? (does it have an anchor?) */
403c4762a1bSJed Brown         ierr = PetscSectionGetDof(aSec,c,&cDof);CHKERRQ(ierr);
404c4762a1bSJed Brown         if (cDof) {
405c4762a1bSJed Brown           PetscInt cOff, a, aDof, aOff, j;
406c4762a1bSJed Brown           if (cDof != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Found %d anchor points: should be just one",cDof);
407c4762a1bSJed Brown 
408c4762a1bSJed Brown           /* find the anchor point */
409c4762a1bSJed Brown           ierr = PetscSectionGetOffset(aSec,c,&cOff);CHKERRQ(ierr);
410c4762a1bSJed Brown           a    = anchors[cOff];
411c4762a1bSJed Brown 
412c4762a1bSJed Brown           /* find the constrained dofs (row in constraint matrix) */
413c4762a1bSJed Brown           ierr = PetscSectionGetDof(cSec,c,&cDof);CHKERRQ(ierr);
414c4762a1bSJed Brown           ierr = PetscSectionGetOffset(cSec,c,&cOff);CHKERRQ(ierr);
415c4762a1bSJed Brown 
416c4762a1bSJed Brown           /* find the anchor dofs (column in constraint matrix) */
417c4762a1bSJed Brown           ierr = PetscSectionGetDof(section,a,&aDof);CHKERRQ(ierr);
418c4762a1bSJed Brown           ierr = PetscSectionGetOffset(section,a,&aOff);CHKERRQ(ierr);
419c4762a1bSJed Brown 
420c4762a1bSJed Brown           if (cDof != aDof) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point and anchor have different number of dofs: %d, %d\n",cDof,aDof);
421c4762a1bSJed Brown           if (cDof % numComp) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Point dofs not divisible by field components: %d, %d\n",cDof,numComp);
422c4762a1bSJed Brown 
423c4762a1bSJed Brown           /* put in a simple equality constraint */
424c4762a1bSJed Brown           for (j = 0; j < cDof; j++) {
425c4762a1bSJed Brown             ierr = MatSetValue(cMat,cOff+j,aOff+j,1.,INSERT_VALUES);CHKERRQ(ierr);
426c4762a1bSJed Brown           }
427c4762a1bSJed Brown         }
428c4762a1bSJed Brown       }
429c4762a1bSJed Brown       ierr = MatAssemblyBegin(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
430c4762a1bSJed Brown       ierr = MatAssemblyEnd(cMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
431c4762a1bSJed Brown       ierr = ISRestoreIndices(aIS,&anchors);CHKERRQ(ierr);
432c4762a1bSJed Brown 
433c4762a1bSJed Brown       /* Now that we have constructed the constraint matrix, any FE matrix
434c4762a1bSJed Brown        * that we construct will apply the constraints during construction */
435c4762a1bSJed Brown 
436c4762a1bSJed Brown       ierr = DMCreateMatrix(dm,&mass);CHKERRQ(ierr);
437c4762a1bSJed Brown       /* get a dummy local variable to serve as the solution */
438c4762a1bSJed Brown       ierr = DMGetLocalVector(dm,&local);CHKERRQ(ierr);
439c4762a1bSJed Brown       ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
440c4762a1bSJed Brown       /* set the jacobian to be the mass matrix */
441c4762a1bSJed Brown       ierr = PetscDSSetJacobian(ds, 0, 0, simple_mass, NULL,  NULL, NULL);CHKERRQ(ierr);
442c4762a1bSJed Brown       /* build the mass matrix */
443c4762a1bSJed Brown       ierr = DMPlexSNESComputeJacobianFEM(dm,local,mass,mass,NULL);CHKERRQ(ierr);
444c4762a1bSJed Brown       ierr = MatView(mass,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
445c4762a1bSJed Brown       ierr = MatDestroy(&mass);CHKERRQ(ierr);
446c4762a1bSJed Brown       ierr = DMRestoreLocalVector(dm,&local);CHKERRQ(ierr);
447c4762a1bSJed Brown     }
448c4762a1bSJed Brown   }
449c4762a1bSJed Brown   PetscFunctionReturn(0);
450c4762a1bSJed Brown }
451c4762a1bSJed Brown 
452c4762a1bSJed Brown static PetscErrorCode TestFEJacobian(DM dm, AppCtx *user)
453c4762a1bSJed Brown {
454c4762a1bSJed Brown   PetscErrorCode ierr;
455c4762a1bSJed Brown 
456c4762a1bSJed Brown   PetscFunctionBeginUser;
45730602db0SMatthew G. Knepley   if (!user->useDA) {
458c4762a1bSJed Brown     Vec          local;
459c4762a1bSJed Brown     const Vec    *vecs;
460c4762a1bSJed Brown     Mat          E;
461c4762a1bSJed Brown     MatNullSpace sp;
462c4762a1bSJed Brown     PetscBool    isNullSpace, hasConst;
46330602db0SMatthew G. Knepley     PetscInt     dim, n, i;
464c4762a1bSJed Brown     Vec          res = NULL, localX, localRes;
465c4762a1bSJed Brown     PetscDS      ds;
466c4762a1bSJed Brown 
46730602db0SMatthew G. Knepley     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
46830602db0SMatthew 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);
469c4762a1bSJed Brown     ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
470c4762a1bSJed Brown     ierr = PetscDSSetJacobian(ds,0,0,NULL,NULL,NULL,symmetric_gradient_inner_product);CHKERRQ(ierr);
471c4762a1bSJed Brown     ierr = DMCreateMatrix(dm,&E);CHKERRQ(ierr);
472c4762a1bSJed Brown     ierr = DMGetLocalVector(dm,&local);CHKERRQ(ierr);
473c4762a1bSJed Brown     ierr = DMPlexSNESComputeJacobianFEM(dm,local,E,E,NULL);CHKERRQ(ierr);
47456cf3b9cSMatthew G. Knepley     ierr = DMPlexCreateRigidBody(dm,0,&sp);CHKERRQ(ierr);
475c4762a1bSJed Brown     ierr = MatNullSpaceGetVecs(sp,&hasConst,&n,&vecs);CHKERRQ(ierr);
476c4762a1bSJed Brown     if (n) {ierr = VecDuplicate(vecs[0],&res);CHKERRQ(ierr);}
477c4762a1bSJed Brown     ierr = DMCreateLocalVector(dm,&localX);CHKERRQ(ierr);
478c4762a1bSJed Brown     ierr = DMCreateLocalVector(dm,&localRes);CHKERRQ(ierr);
479c4762a1bSJed Brown     for (i = 0; i < n; i++) { /* also test via matrix-free Jacobian application */
480c4762a1bSJed Brown       PetscReal resNorm;
481c4762a1bSJed Brown 
482c4762a1bSJed Brown       ierr = VecSet(localRes,0.);CHKERRQ(ierr);
483c4762a1bSJed Brown       ierr = VecSet(localX,0.);CHKERRQ(ierr);
484c4762a1bSJed Brown       ierr = VecSet(local,0.);CHKERRQ(ierr);
485c4762a1bSJed Brown       ierr = VecSet(res,0.);CHKERRQ(ierr);
486c4762a1bSJed Brown       ierr = DMGlobalToLocalBegin(dm,vecs[i],INSERT_VALUES,localX);CHKERRQ(ierr);
487c4762a1bSJed Brown       ierr = DMGlobalToLocalEnd(dm,vecs[i],INSERT_VALUES,localX);CHKERRQ(ierr);
4888e3a2eefSMatthew G. Knepley       ierr = DMSNESComputeJacobianAction(dm,local,localX,localRes,NULL);CHKERRQ(ierr);
489c4762a1bSJed Brown       ierr = DMLocalToGlobalBegin(dm,localRes,ADD_VALUES,res);CHKERRQ(ierr);
490c4762a1bSJed Brown       ierr = DMLocalToGlobalEnd(dm,localRes,ADD_VALUES,res);CHKERRQ(ierr);
491c4762a1bSJed Brown       ierr = VecNorm(res,NORM_2,&resNorm);CHKERRQ(ierr);
492c4762a1bSJed Brown       if (resNorm > PETSC_SMALL) {
493c4762a1bSJed Brown         ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient action null space vector %D residual: %E\n",i,resNorm);CHKERRQ(ierr);
494c4762a1bSJed Brown       }
495c4762a1bSJed Brown     }
496c4762a1bSJed Brown     ierr = VecDestroy(&localRes);CHKERRQ(ierr);
497c4762a1bSJed Brown     ierr = VecDestroy(&localX);CHKERRQ(ierr);
498c4762a1bSJed Brown     ierr = VecDestroy(&res);CHKERRQ(ierr);
499c4762a1bSJed Brown     ierr = MatNullSpaceTest(sp,E,&isNullSpace);CHKERRQ(ierr);
500c4762a1bSJed Brown     if (isNullSpace) {
501c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient null space: PASS\n");CHKERRQ(ierr);
502c4762a1bSJed Brown     } else {
503c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Symmetric gradient null space: FAIL\n");CHKERRQ(ierr);
504c4762a1bSJed Brown     }
505c4762a1bSJed Brown     ierr = MatNullSpaceDestroy(&sp);CHKERRQ(ierr);
506c4762a1bSJed Brown     ierr = MatDestroy(&E);CHKERRQ(ierr);
507c4762a1bSJed Brown     ierr = DMRestoreLocalVector(dm,&local);CHKERRQ(ierr);
508c4762a1bSJed Brown   }
509c4762a1bSJed Brown   PetscFunctionReturn(0);
510c4762a1bSJed Brown }
511c4762a1bSJed Brown 
512c4762a1bSJed Brown static PetscErrorCode TestInjector(DM dm, AppCtx *user)
513c4762a1bSJed Brown {
514c4762a1bSJed Brown   DM             refTree;
515c4762a1bSJed Brown   PetscMPIInt    rank;
516c4762a1bSJed Brown   PetscErrorCode ierr;
517c4762a1bSJed Brown 
518c4762a1bSJed Brown   PetscFunctionBegin;
519c4762a1bSJed Brown   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
520c4762a1bSJed Brown   if (refTree) {
521c4762a1bSJed Brown     Mat inj;
522c4762a1bSJed Brown 
523c4762a1bSJed Brown     ierr = DMPlexComputeInjectorReferenceTree(refTree,&inj);CHKERRQ(ierr);
524c4762a1bSJed Brown     ierr = PetscObjectSetName((PetscObject)inj,"Reference Tree Injector");CHKERRQ(ierr);
525ffc4695bSBarry Smith     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);
526c4762a1bSJed Brown     if (!rank) {
527c4762a1bSJed Brown       ierr = MatView(inj,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
528c4762a1bSJed Brown     }
529c4762a1bSJed Brown     ierr = MatDestroy(&inj);CHKERRQ(ierr);
530c4762a1bSJed Brown   }
531c4762a1bSJed Brown   PetscFunctionReturn(0);
532c4762a1bSJed Brown }
533c4762a1bSJed Brown 
534c4762a1bSJed Brown static PetscErrorCode TestFVGrad(DM dm, AppCtx *user)
535c4762a1bSJed Brown {
536c4762a1bSJed Brown   MPI_Comm          comm;
537c4762a1bSJed Brown   DM                dmRedist, dmfv, dmgrad, dmCell, refTree;
538c4762a1bSJed Brown   PetscFV           fv;
53930602db0SMatthew G. Knepley   PetscInt          dim, nvecs, v, cStart, cEnd, cEndInterior;
540c4762a1bSJed Brown   PetscMPIInt       size;
541c4762a1bSJed Brown   Vec               cellgeom, grad, locGrad;
542c4762a1bSJed Brown   const PetscScalar *cgeom;
543c4762a1bSJed Brown   PetscReal         allVecMaxDiff = 0., fvTol = 100. * PETSC_MACHINE_EPSILON;
544c4762a1bSJed Brown   PetscErrorCode    ierr;
545c4762a1bSJed Brown 
546c4762a1bSJed Brown   PetscFunctionBeginUser;
547c4762a1bSJed Brown   comm = PetscObjectComm((PetscObject)dm);
54830602db0SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
549c4762a1bSJed Brown   /* duplicate DM, give dup. a FV discretization */
550c4762a1bSJed Brown   ierr = DMSetBasicAdjacency(dm,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr);
551ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
552c4762a1bSJed Brown   dmRedist = NULL;
553c4762a1bSJed Brown   if (size > 1) {
554c4762a1bSJed Brown     ierr = DMPlexDistributeOverlap(dm,1,NULL,&dmRedist);CHKERRQ(ierr);
555c4762a1bSJed Brown   }
556c4762a1bSJed Brown   if (!dmRedist) {
557c4762a1bSJed Brown     dmRedist = dm;
558c4762a1bSJed Brown     ierr = PetscObjectReference((PetscObject)dmRedist);CHKERRQ(ierr);
559c4762a1bSJed Brown   }
560c4762a1bSJed Brown   ierr = PetscFVCreate(comm,&fv);CHKERRQ(ierr);
561c4762a1bSJed Brown   ierr = PetscFVSetType(fv,PETSCFVLEASTSQUARES);CHKERRQ(ierr);
562c4762a1bSJed Brown   ierr = PetscFVSetNumComponents(fv,user->numComponents);CHKERRQ(ierr);
56330602db0SMatthew G. Knepley   ierr = PetscFVSetSpatialDimension(fv,dim);CHKERRQ(ierr);
564c4762a1bSJed Brown   ierr = PetscFVSetFromOptions(fv);CHKERRQ(ierr);
565c4762a1bSJed Brown   ierr = PetscFVSetUp(fv);CHKERRQ(ierr);
566c4762a1bSJed Brown   ierr = DMPlexConstructGhostCells(dmRedist,NULL,NULL,&dmfv);CHKERRQ(ierr);
567c4762a1bSJed Brown   ierr = DMDestroy(&dmRedist);CHKERRQ(ierr);
568c4762a1bSJed Brown   ierr = DMSetNumFields(dmfv,1);CHKERRQ(ierr);
569c4762a1bSJed Brown   ierr = DMSetField(dmfv, 0, NULL, (PetscObject) fv);CHKERRQ(ierr);
570c4762a1bSJed Brown   ierr = DMCreateDS(dmfv);CHKERRQ(ierr);
571c4762a1bSJed Brown   ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
572c4762a1bSJed Brown   if (refTree) {ierr = DMCopyDisc(dmfv,refTree);CHKERRQ(ierr);}
5733e9753d6SMatthew G. Knepley   ierr = DMPlexGetGradientDM(dmfv, fv, &dmgrad);CHKERRQ(ierr);
574c4762a1bSJed Brown   ierr = DMPlexGetHeightStratum(dmfv,0,&cStart,&cEnd);CHKERRQ(ierr);
57530602db0SMatthew G. Knepley   nvecs = dim * (dim+1) / 2;
5763e9753d6SMatthew G. Knepley   ierr = DMPlexGetGeometryFVM(dmfv,NULL,&cellgeom,NULL);CHKERRQ(ierr);
577c4762a1bSJed Brown   ierr = VecGetDM(cellgeom,&dmCell);CHKERRQ(ierr);
578c4762a1bSJed Brown   ierr = VecGetArrayRead(cellgeom,&cgeom);CHKERRQ(ierr);
579c4762a1bSJed Brown   ierr = DMGetGlobalVector(dmgrad,&grad);CHKERRQ(ierr);
580c4762a1bSJed Brown   ierr = DMGetLocalVector(dmgrad,&locGrad);CHKERRQ(ierr);
581c4762a1bSJed Brown   ierr = DMPlexGetGhostCellStratum(dmgrad,&cEndInterior,NULL);CHKERRQ(ierr);
582c4762a1bSJed Brown   cEndInterior = (cEndInterior < 0) ? cEnd: cEndInterior;
583c4762a1bSJed Brown   for (v = 0; v < nvecs; v++) {
584c4762a1bSJed Brown     Vec               locX;
585c4762a1bSJed Brown     PetscInt          c;
586c4762a1bSJed Brown     PetscScalar       trueGrad[3][3] = {{0.}};
587c4762a1bSJed Brown     const PetscScalar *gradArray;
588c4762a1bSJed Brown     PetscReal         maxDiff, maxDiffGlob;
589c4762a1bSJed Brown 
590c4762a1bSJed Brown     ierr = DMGetLocalVector(dmfv,&locX);CHKERRQ(ierr);
591c4762a1bSJed Brown     /* get the local projection of the rigid body mode */
592c4762a1bSJed Brown     for (c = cStart; c < cEnd; c++) {
593c4762a1bSJed Brown       PetscFVCellGeom *cg;
594c4762a1bSJed Brown       PetscScalar     cx[3] = {0.,0.,0.};
595c4762a1bSJed Brown 
596c4762a1bSJed Brown       ierr = DMPlexPointLocalRead(dmCell, c, cgeom, &cg);CHKERRQ(ierr);
59730602db0SMatthew G. Knepley       if (v < dim) {
598c4762a1bSJed Brown         cx[v] = 1.;
599c4762a1bSJed Brown       } else {
60030602db0SMatthew G. Knepley         PetscInt w = v - dim;
601c4762a1bSJed Brown 
60230602db0SMatthew G. Knepley         cx[(w + 1) % dim] =  cg->centroid[(w + 2) % dim];
60330602db0SMatthew G. Knepley         cx[(w + 2) % dim] = -cg->centroid[(w + 1) % dim];
604c4762a1bSJed Brown       }
605c4762a1bSJed Brown       ierr = DMPlexVecSetClosure(dmfv,NULL,locX,c,cx,INSERT_ALL_VALUES);CHKERRQ(ierr);
606c4762a1bSJed Brown     }
607c4762a1bSJed Brown     /* TODO: this isn't in any header */
608c4762a1bSJed Brown     ierr = DMPlexReconstructGradientsFVM(dmfv,locX,grad);CHKERRQ(ierr);
609c4762a1bSJed Brown     ierr = DMGlobalToLocalBegin(dmgrad,grad,INSERT_VALUES,locGrad);CHKERRQ(ierr);
610c4762a1bSJed Brown     ierr = DMGlobalToLocalEnd(dmgrad,grad,INSERT_VALUES,locGrad);CHKERRQ(ierr);
611c4762a1bSJed Brown     ierr = VecGetArrayRead(locGrad,&gradArray);CHKERRQ(ierr);
612c4762a1bSJed Brown     /* compare computed gradient to exact gradient */
61330602db0SMatthew G. Knepley     if (v >= dim) {
61430602db0SMatthew G. Knepley       PetscInt w = v - dim;
615c4762a1bSJed Brown 
61630602db0SMatthew G. Knepley       trueGrad[(w + 1) % dim][(w + 2) % dim] =  1.;
61730602db0SMatthew G. Knepley       trueGrad[(w + 2) % dim][(w + 1) % dim] = -1.;
618c4762a1bSJed Brown     }
619c4762a1bSJed Brown     maxDiff = 0.;
620c4762a1bSJed Brown     for (c = cStart; c < cEndInterior; c++) {
621c4762a1bSJed Brown       PetscScalar *compGrad;
622c4762a1bSJed Brown       PetscInt    i, j, k;
623c4762a1bSJed Brown       PetscReal   FrobDiff = 0.;
624c4762a1bSJed Brown 
625c4762a1bSJed Brown       ierr = DMPlexPointLocalRead(dmgrad, c, gradArray, &compGrad);CHKERRQ(ierr);
626c4762a1bSJed Brown 
62730602db0SMatthew G. Knepley       for (i = 0, k = 0; i < dim; i++) {
62830602db0SMatthew G. Knepley         for (j = 0; j < dim; j++, k++) {
629c4762a1bSJed Brown           PetscScalar diff = compGrad[k] - trueGrad[i][j];
630c4762a1bSJed Brown           FrobDiff += PetscRealPart(diff * PetscConj(diff));
631c4762a1bSJed Brown         }
632c4762a1bSJed Brown       }
633c4762a1bSJed Brown       FrobDiff = PetscSqrtReal(FrobDiff);
634c4762a1bSJed Brown       maxDiff  = PetscMax(maxDiff,FrobDiff);
635c4762a1bSJed Brown     }
636ffc4695bSBarry Smith     ierr = MPI_Allreduce(&maxDiff,&maxDiffGlob,1,MPIU_REAL,MPIU_MAX,comm);CHKERRMPI(ierr);
637c4762a1bSJed Brown     allVecMaxDiff = PetscMax(allVecMaxDiff,maxDiffGlob);
638c4762a1bSJed Brown     ierr = VecRestoreArrayRead(locGrad,&gradArray);CHKERRQ(ierr);
639c4762a1bSJed Brown     ierr = DMRestoreLocalVector(dmfv,&locX);CHKERRQ(ierr);
640c4762a1bSJed Brown   }
641c4762a1bSJed Brown   if (allVecMaxDiff < fvTol) {
642c4762a1bSJed Brown     ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Finite volume gradient reconstruction: PASS\n");CHKERRQ(ierr);
643c4762a1bSJed Brown   } else {
644c4762a1bSJed Brown     ierr = PetscPrintf(PetscObjectComm((PetscObject)dm),"Finite volume gradient reconstruction: FAIL at tolerance %g with max difference %g\n",fvTol,allVecMaxDiff);CHKERRQ(ierr);
645c4762a1bSJed Brown   }
646c4762a1bSJed Brown   ierr = DMRestoreLocalVector(dmgrad,&locGrad);CHKERRQ(ierr);
647c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(dmgrad,&grad);CHKERRQ(ierr);
648c4762a1bSJed Brown   ierr = VecRestoreArrayRead(cellgeom,&cgeom);CHKERRQ(ierr);
649c4762a1bSJed Brown   ierr = DMDestroy(&dmfv);CHKERRQ(ierr);
650c4762a1bSJed Brown   ierr = PetscFVDestroy(&fv);CHKERRQ(ierr);
651c4762a1bSJed Brown   PetscFunctionReturn(0);
652c4762a1bSJed Brown }
653c4762a1bSJed Brown 
654c4762a1bSJed Brown static PetscErrorCode ComputeError(DM dm, PetscErrorCode (**exactFuncs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *),
655c4762a1bSJed Brown                                    PetscErrorCode (**exactFuncDers)(PetscInt, PetscReal, const PetscReal[], const PetscReal[], PetscInt, PetscScalar *, void *),
656c4762a1bSJed Brown                                    void **exactCtxs, PetscReal *error, PetscReal *errorDer, AppCtx *user)
657c4762a1bSJed Brown {
658c4762a1bSJed Brown   Vec            u;
659c4762a1bSJed Brown   PetscReal      n[3] = {1.0, 1.0, 1.0};
660c4762a1bSJed Brown   PetscErrorCode ierr;
661c4762a1bSJed Brown 
662c4762a1bSJed Brown   PetscFunctionBeginUser;
663c4762a1bSJed Brown   ierr = DMGetGlobalVector(dm, &u);CHKERRQ(ierr);
664c4762a1bSJed Brown   /* Project function into FE function space */
665c4762a1bSJed Brown   ierr = DMProjectFunction(dm, 0.0, exactFuncs, exactCtxs, INSERT_ALL_VALUES, u);CHKERRQ(ierr);
666c4762a1bSJed Brown   ierr = VecViewFromOptions(u, NULL, "-projection_view");CHKERRQ(ierr);
667c4762a1bSJed Brown   /* Compare approximation to exact in L_2 */
668c4762a1bSJed Brown   ierr = DMComputeL2Diff(dm, 0.0, exactFuncs, exactCtxs, u, error);CHKERRQ(ierr);
669c4762a1bSJed Brown   ierr = DMComputeL2GradientDiff(dm, 0.0, exactFuncDers, exactCtxs, u, n, errorDer);CHKERRQ(ierr);
670c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(dm, &u);CHKERRQ(ierr);
671c4762a1bSJed Brown   PetscFunctionReturn(0);
672c4762a1bSJed Brown }
673c4762a1bSJed Brown 
674c4762a1bSJed Brown static PetscErrorCode CheckFunctions(DM dm, PetscInt order, AppCtx *user)
675c4762a1bSJed Brown {
676c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx);
677c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt dim, PetscReal time, const PetscReal x[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx);
678c4762a1bSJed Brown   void            *exactCtxs[3];
679c4762a1bSJed Brown   MPI_Comm         comm;
680c4762a1bSJed Brown   PetscReal        error, errorDer, tol = PETSC_SMALL;
681c4762a1bSJed Brown   PetscErrorCode   ierr;
682c4762a1bSJed Brown 
683c4762a1bSJed Brown   PetscFunctionBeginUser;
684c4762a1bSJed Brown   exactCtxs[0]       = user;
685c4762a1bSJed Brown   exactCtxs[1]       = user;
686c4762a1bSJed Brown   exactCtxs[2]       = user;
687c4762a1bSJed Brown   ierr = PetscObjectGetComm((PetscObject)dm, &comm);CHKERRQ(ierr);
688c4762a1bSJed Brown   /* Setup functions to approximate */
689c4762a1bSJed Brown   switch (order) {
690c4762a1bSJed Brown   case 0:
691c4762a1bSJed Brown     exactFuncs[0]    = constant;
692c4762a1bSJed Brown     exactFuncDers[0] = constantDer;
693c4762a1bSJed Brown     break;
694c4762a1bSJed Brown   case 1:
695c4762a1bSJed Brown     exactFuncs[0]    = linear;
696c4762a1bSJed Brown     exactFuncDers[0] = linearDer;
697c4762a1bSJed Brown     break;
698c4762a1bSJed Brown   case 2:
699c4762a1bSJed Brown     exactFuncs[0]    = quadratic;
700c4762a1bSJed Brown     exactFuncDers[0] = quadraticDer;
701c4762a1bSJed Brown     break;
702c4762a1bSJed Brown   case 3:
703c4762a1bSJed Brown     exactFuncs[0]    = cubic;
704c4762a1bSJed Brown     exactFuncDers[0] = cubicDer;
705c4762a1bSJed Brown     break;
706c4762a1bSJed Brown   default:
70730602db0SMatthew G. Knepley     SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "Could not determine functions to test for order %d", order);
708c4762a1bSJed Brown   }
709c4762a1bSJed Brown   ierr = ComputeError(dm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
710c4762a1bSJed Brown   /* Report result */
711c4762a1bSJed 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);}
712c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Function tests pass for order %D at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
713c4762a1bSJed 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);}
714c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Function tests pass for order %D derivatives at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
715c4762a1bSJed Brown   PetscFunctionReturn(0);
716c4762a1bSJed Brown }
717c4762a1bSJed Brown 
718c4762a1bSJed Brown static PetscErrorCode CheckInterpolation(DM dm, PetscBool checkRestrict, PetscInt order, AppCtx *user)
719c4762a1bSJed Brown {
720c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt, PetscReal, const PetscReal x[], PetscInt, PetscScalar *u, void *ctx);
721c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt, PetscReal, const PetscReal x[], const PetscReal n[], PetscInt, PetscScalar *u, void *ctx);
722c4762a1bSJed Brown   PetscReal       n[3]         = {1.0, 1.0, 1.0};
723c4762a1bSJed Brown   void           *exactCtxs[3];
724c4762a1bSJed Brown   DM              rdm, idm, fdm;
725c4762a1bSJed Brown   Mat             Interp;
726c4762a1bSJed Brown   Vec             iu, fu, scaling;
727c4762a1bSJed Brown   MPI_Comm        comm;
72830602db0SMatthew G. Knepley   PetscInt        dim;
729c4762a1bSJed Brown   PetscReal       error, errorDer, tol = PETSC_SMALL;
730c4762a1bSJed Brown   PetscErrorCode  ierr;
731c4762a1bSJed Brown 
732c4762a1bSJed Brown   PetscFunctionBeginUser;
733c4762a1bSJed Brown   exactCtxs[0]       = user;
734c4762a1bSJed Brown   exactCtxs[1]       = user;
735c4762a1bSJed Brown   exactCtxs[2]       = user;
736c4762a1bSJed Brown   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
73730602db0SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
738c4762a1bSJed Brown   ierr = DMRefine(dm, comm, &rdm);CHKERRQ(ierr);
739c4762a1bSJed Brown   ierr = DMSetCoarseDM(rdm, dm);CHKERRQ(ierr);
740c4762a1bSJed Brown   ierr = DMPlexSetRegularRefinement(rdm, user->convRefine);CHKERRQ(ierr);
74130602db0SMatthew G. Knepley   if (user->tree) {
742c4762a1bSJed Brown     DM refTree;
743c4762a1bSJed Brown     ierr = DMPlexGetReferenceTree(dm,&refTree);CHKERRQ(ierr);
744c4762a1bSJed Brown     ierr = DMPlexSetReferenceTree(rdm,refTree);CHKERRQ(ierr);
745c4762a1bSJed Brown   }
74630602db0SMatthew G. Knepley   if (user->useDA) {ierr = DMDASetVertexCoordinates(rdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
747c4762a1bSJed Brown   ierr = SetupSection(rdm, user);CHKERRQ(ierr);
748c4762a1bSJed Brown   /* Setup functions to approximate */
749c4762a1bSJed Brown   switch (order) {
750c4762a1bSJed Brown   case 0:
751c4762a1bSJed Brown     exactFuncs[0]    = constant;
752c4762a1bSJed Brown     exactFuncDers[0] = constantDer;
753c4762a1bSJed Brown     break;
754c4762a1bSJed Brown   case 1:
755c4762a1bSJed Brown     exactFuncs[0]    = linear;
756c4762a1bSJed Brown     exactFuncDers[0] = linearDer;
757c4762a1bSJed Brown     break;
758c4762a1bSJed Brown   case 2:
759c4762a1bSJed Brown     exactFuncs[0]    = quadratic;
760c4762a1bSJed Brown     exactFuncDers[0] = quadraticDer;
761c4762a1bSJed Brown     break;
762c4762a1bSJed Brown   case 3:
763c4762a1bSJed Brown     exactFuncs[0]    = cubic;
764c4762a1bSJed Brown     exactFuncDers[0] = cubicDer;
765c4762a1bSJed Brown     break;
766c4762a1bSJed Brown   default:
767c4762a1bSJed Brown     SETERRQ2(comm, PETSC_ERR_ARG_OUTOFRANGE, "Could not determine functions to test for dimension %D order %D", dim, order);
768c4762a1bSJed Brown   }
769c4762a1bSJed Brown   idm  = checkRestrict ? rdm :  dm;
770c4762a1bSJed Brown   fdm  = checkRestrict ?  dm : rdm;
771c4762a1bSJed Brown   ierr = DMGetGlobalVector(idm, &iu);CHKERRQ(ierr);
772c4762a1bSJed Brown   ierr = DMGetGlobalVector(fdm, &fu);CHKERRQ(ierr);
773c4762a1bSJed Brown   ierr = DMSetApplicationContext(dm, user);CHKERRQ(ierr);
774c4762a1bSJed Brown   ierr = DMSetApplicationContext(rdm, user);CHKERRQ(ierr);
775c4762a1bSJed Brown   ierr = DMCreateInterpolation(dm, rdm, &Interp, &scaling);CHKERRQ(ierr);
776c4762a1bSJed Brown   /* Project function into initial FE function space */
777c4762a1bSJed Brown   ierr = DMProjectFunction(idm, 0.0, exactFuncs, exactCtxs, INSERT_ALL_VALUES, iu);CHKERRQ(ierr);
778c4762a1bSJed Brown   /* Interpolate function into final FE function space */
779c4762a1bSJed Brown   if (checkRestrict) {ierr = MatRestrict(Interp, iu, fu);CHKERRQ(ierr);ierr = VecPointwiseMult(fu, scaling, fu);CHKERRQ(ierr);}
780c4762a1bSJed Brown   else               {ierr = MatInterpolate(Interp, iu, fu);CHKERRQ(ierr);}
781c4762a1bSJed Brown   /* Compare approximation to exact in L_2 */
782c4762a1bSJed Brown   ierr = DMComputeL2Diff(fdm, 0.0, exactFuncs, exactCtxs, fu, &error);CHKERRQ(ierr);
783c4762a1bSJed Brown   ierr = DMComputeL2GradientDiff(fdm, 0.0, exactFuncDers, exactCtxs, fu, n, &errorDer);CHKERRQ(ierr);
784c4762a1bSJed Brown   /* Report result */
785c4762a1bSJed 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);}
786c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Interpolation tests pass for order %D at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
787c4762a1bSJed 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);}
788c4762a1bSJed Brown   else                {ierr = PetscPrintf(comm, "Interpolation tests pass for order %D derivatives at tolerance %g\n", order, (double)tol);CHKERRQ(ierr);}
789c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(idm, &iu);CHKERRQ(ierr);
790c4762a1bSJed Brown   ierr = DMRestoreGlobalVector(fdm, &fu);CHKERRQ(ierr);
791c4762a1bSJed Brown   ierr = MatDestroy(&Interp);CHKERRQ(ierr);
792c4762a1bSJed Brown   ierr = VecDestroy(&scaling);CHKERRQ(ierr);
793c4762a1bSJed Brown   ierr = DMDestroy(&rdm);CHKERRQ(ierr);
794c4762a1bSJed Brown   PetscFunctionReturn(0);
795c4762a1bSJed Brown }
796c4762a1bSJed Brown 
797c4762a1bSJed Brown static PetscErrorCode CheckConvergence(DM dm, PetscInt Nr, AppCtx *user)
798c4762a1bSJed Brown {
799c4762a1bSJed Brown   DM               odm = dm, rdm = NULL, cdm = NULL;
800c4762a1bSJed Brown   PetscErrorCode (*exactFuncs[1]) (PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx) = {trig};
801c4762a1bSJed Brown   PetscErrorCode (*exactFuncDers[1]) (PetscInt dim, PetscReal time, const PetscReal x[], const PetscReal n[], PetscInt Nf, PetscScalar *u, void *ctx) = {trigDer};
802c4762a1bSJed Brown   void            *exactCtxs[3];
803c4762a1bSJed Brown   PetscInt         r, c, cStart, cEnd;
804c4762a1bSJed Brown   PetscReal        errorOld, errorDerOld, error, errorDer, rel, len, lenOld;
805c4762a1bSJed Brown   double           p;
806c4762a1bSJed Brown   PetscErrorCode   ierr;
807c4762a1bSJed Brown 
808c4762a1bSJed Brown   PetscFunctionBeginUser;
809c4762a1bSJed Brown   if (!user->convergence) PetscFunctionReturn(0);
810c4762a1bSJed Brown   exactCtxs[0] = user;
811c4762a1bSJed Brown   exactCtxs[1] = user;
812c4762a1bSJed Brown   exactCtxs[2] = user;
813c4762a1bSJed Brown   ierr = PetscObjectReference((PetscObject) odm);CHKERRQ(ierr);
814c4762a1bSJed Brown   if (!user->convRefine) {
815c4762a1bSJed Brown     for (r = 0; r < Nr; ++r) {
816c4762a1bSJed Brown       ierr = DMRefine(odm, PetscObjectComm((PetscObject) dm), &rdm);CHKERRQ(ierr);
817c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
818c4762a1bSJed Brown       odm  = rdm;
819c4762a1bSJed Brown     }
820c4762a1bSJed Brown     ierr = SetupSection(odm, user);CHKERRQ(ierr);
821c4762a1bSJed Brown   }
822c4762a1bSJed Brown   ierr = ComputeError(odm, exactFuncs, exactFuncDers, exactCtxs, &errorOld, &errorDerOld, user);CHKERRQ(ierr);
823c4762a1bSJed Brown   if (user->convRefine) {
824c4762a1bSJed Brown     for (r = 0; r < Nr; ++r) {
825c4762a1bSJed Brown       ierr = DMRefine(odm, PetscObjectComm((PetscObject) dm), &rdm);CHKERRQ(ierr);
82630602db0SMatthew G. Knepley       if (user->useDA) {ierr = DMDASetVertexCoordinates(rdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
827c4762a1bSJed Brown       ierr = SetupSection(rdm, user);CHKERRQ(ierr);
828c4762a1bSJed Brown       ierr = ComputeError(rdm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
829c4762a1bSJed Brown       p    = PetscLog2Real(errorOld/error);
830c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Function   convergence rate at refinement %D: %.2f\n", r, (double)p);CHKERRQ(ierr);
831c4762a1bSJed Brown       p    = PetscLog2Real(errorDerOld/errorDer);
832c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Derivative convergence rate at refinement %D: %.2f\n", r, (double)p);CHKERRQ(ierr);
833c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
834c4762a1bSJed Brown       odm         = rdm;
835c4762a1bSJed Brown       errorOld    = error;
836c4762a1bSJed Brown       errorDerOld = errorDer;
837c4762a1bSJed Brown     }
838c4762a1bSJed Brown   } else {
839c4762a1bSJed Brown     /* ierr = ComputeLongestEdge(dm, &lenOld);CHKERRQ(ierr); */
840c4762a1bSJed Brown     ierr = DMPlexGetHeightStratum(odm, 0, &cStart, &cEnd);CHKERRQ(ierr);
841c4762a1bSJed Brown     lenOld = cEnd - cStart;
842c4762a1bSJed Brown     for (c = 0; c < Nr; ++c) {
843c4762a1bSJed Brown       ierr = DMCoarsen(odm, PetscObjectComm((PetscObject) dm), &cdm);CHKERRQ(ierr);
84430602db0SMatthew G. Knepley       if (user->useDA) {ierr = DMDASetVertexCoordinates(cdm, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0);CHKERRQ(ierr);}
845c4762a1bSJed Brown       ierr = SetupSection(cdm, user);CHKERRQ(ierr);
846c4762a1bSJed Brown       ierr = ComputeError(cdm, exactFuncs, exactFuncDers, exactCtxs, &error, &errorDer, user);CHKERRQ(ierr);
847c4762a1bSJed Brown       /* ierr = ComputeLongestEdge(cdm, &len);CHKERRQ(ierr); */
848c4762a1bSJed Brown       ierr = DMPlexGetHeightStratum(cdm, 0, &cStart, &cEnd);CHKERRQ(ierr);
849c4762a1bSJed Brown       len  = cEnd - cStart;
850c4762a1bSJed Brown       rel  = error/errorOld;
851c4762a1bSJed Brown       p    = PetscLogReal(rel) / PetscLogReal(lenOld / len);
852c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Function   convergence rate at coarsening %D: %.2f\n", c, (double)p);CHKERRQ(ierr);
853c4762a1bSJed Brown       rel  = errorDer/errorDerOld;
854c4762a1bSJed Brown       p    = PetscLogReal(rel) / PetscLogReal(lenOld / len);
855c4762a1bSJed Brown       ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Derivative convergence rate at coarsening %D: %.2f\n", c, (double)p);CHKERRQ(ierr);
856c4762a1bSJed Brown       ierr = DMDestroy(&odm);CHKERRQ(ierr);
857c4762a1bSJed Brown       odm         = cdm;
858c4762a1bSJed Brown       errorOld    = error;
859c4762a1bSJed Brown       errorDerOld = errorDer;
860c4762a1bSJed Brown       lenOld      = len;
861c4762a1bSJed Brown     }
862c4762a1bSJed Brown   }
863c4762a1bSJed Brown   ierr = DMDestroy(&odm);CHKERRQ(ierr);
864c4762a1bSJed Brown   PetscFunctionReturn(0);
865c4762a1bSJed Brown }
866c4762a1bSJed Brown 
867c4762a1bSJed Brown int main(int argc, char **argv)
868c4762a1bSJed Brown {
869c4762a1bSJed Brown   DM             dm;
870c4762a1bSJed Brown   AppCtx         user;                 /* user-defined work context */
87130602db0SMatthew G. Knepley   PetscInt       dim = 2;
87230602db0SMatthew G. Knepley   PetscBool      simplex = PETSC_FALSE;
873c4762a1bSJed Brown   PetscErrorCode ierr;
874c4762a1bSJed Brown 
875c4762a1bSJed Brown   ierr = PetscInitialize(&argc, &argv, NULL, help);if (ierr) return ierr;
876c4762a1bSJed Brown   ierr = ProcessOptions(PETSC_COMM_WORLD, &user);CHKERRQ(ierr);
877c4762a1bSJed Brown   ierr = CreateMesh(PETSC_COMM_WORLD, &user, &dm);CHKERRQ(ierr);
87830602db0SMatthew G. Knepley   if (!user.useDA) {
87930602db0SMatthew G. Knepley     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
88030602db0SMatthew G. Knepley     ierr = DMPlexIsSimplex(dm, &simplex);CHKERRQ(ierr);
88130602db0SMatthew G. Knepley   }
88230602db0SMatthew G. Knepley   user.numComponents = user.numComponents < 0 ? dim : user.numComponents;
88330602db0SMatthew G. Knepley   ierr = PetscFECreateDefault(PETSC_COMM_WORLD, dim, user.numComponents, simplex, NULL, user.qorder, &user.fe);CHKERRQ(ierr);
884c4762a1bSJed Brown   ierr = SetupSection(dm, &user);CHKERRQ(ierr);
885c4762a1bSJed Brown   if (user.testFEjacobian) {ierr = TestFEJacobian(dm, &user);CHKERRQ(ierr);}
886c4762a1bSJed Brown   if (user.testFVgrad) {ierr = TestFVGrad(dm, &user);CHKERRQ(ierr);}
887c4762a1bSJed Brown   if (user.testInjector) {ierr = TestInjector(dm, &user);CHKERRQ(ierr);}
888c4762a1bSJed Brown   ierr = CheckFunctions(dm, user.porder, &user);CHKERRQ(ierr);
889c4762a1bSJed Brown   {
890c4762a1bSJed Brown     PetscDualSpace dsp;
891c4762a1bSJed Brown     PetscInt       k;
892c4762a1bSJed Brown 
893c4762a1bSJed Brown     ierr = PetscFEGetDualSpace(user.fe, &dsp);CHKERRQ(ierr);
894c4762a1bSJed Brown     ierr = PetscDualSpaceGetDeRahm(dsp, &k);CHKERRQ(ierr);
89530602db0SMatthew G. Knepley     if (dim == 2 && simplex == PETSC_TRUE && user.tree == PETSC_FALSE && k == 0) {
896c4762a1bSJed Brown       ierr = CheckInterpolation(dm, PETSC_FALSE, user.porder, &user);CHKERRQ(ierr);
897c4762a1bSJed Brown       ierr = CheckInterpolation(dm, PETSC_TRUE,  user.porder, &user);CHKERRQ(ierr);
898c4762a1bSJed Brown     }
899c4762a1bSJed Brown   }
900c4762a1bSJed Brown   ierr = CheckConvergence(dm, 3, &user);CHKERRQ(ierr);
901c4762a1bSJed Brown   ierr = PetscFEDestroy(&user.fe);CHKERRQ(ierr);
902c4762a1bSJed Brown   ierr = DMDestroy(&dm);CHKERRQ(ierr);
903c4762a1bSJed Brown   ierr = PetscFinalize();
904c4762a1bSJed Brown   return ierr;
905c4762a1bSJed Brown }
906c4762a1bSJed Brown 
907c4762a1bSJed Brown /*TEST
908c4762a1bSJed Brown 
909c4762a1bSJed Brown   test:
910c4762a1bSJed Brown     suffix: 1
911c4762a1bSJed Brown     requires: triangle
912c4762a1bSJed Brown 
913c4762a1bSJed Brown   # 2D P_1 on a triangle
914c4762a1bSJed Brown   test:
915c4762a1bSJed Brown     suffix: p1_2d_0
916c4762a1bSJed Brown     requires: triangle
917c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -convergence
918c4762a1bSJed Brown   test:
919c4762a1bSJed Brown     suffix: p1_2d_1
920c4762a1bSJed Brown     requires: triangle
921c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -porder 1
922c4762a1bSJed Brown   test:
923c4762a1bSJed Brown     suffix: p1_2d_2
924c4762a1bSJed Brown     requires: triangle
925c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -porder 2
926c4762a1bSJed Brown   test:
927c4762a1bSJed Brown     suffix: p1_2d_3
928c4762a1bSJed Brown     requires: triangle pragmatic
929c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
930c4762a1bSJed Brown     filter: grep -v DEBUG
931c4762a1bSJed Brown   test:
932c4762a1bSJed Brown     suffix: p1_2d_4
933c4762a1bSJed Brown     requires: triangle pragmatic
934c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
935c4762a1bSJed Brown   test:
936c4762a1bSJed Brown     suffix: p1_2d_5
937c4762a1bSJed Brown     requires: triangle pragmatic
938c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
939c4762a1bSJed Brown 
940c4762a1bSJed Brown   # 3D P_1 on a tetrahedron
941c4762a1bSJed Brown   test:
942c4762a1bSJed Brown     suffix: p1_3d_0
943c4762a1bSJed Brown     requires: ctetgen
94430602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -convergence
945c4762a1bSJed Brown   test:
946c4762a1bSJed Brown     suffix: p1_3d_1
947c4762a1bSJed Brown     requires: ctetgen
94830602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -porder 1
949c4762a1bSJed Brown   test:
950c4762a1bSJed Brown     suffix: p1_3d_2
951c4762a1bSJed Brown     requires: ctetgen
95230602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -porder 2
953c4762a1bSJed Brown   test:
954c4762a1bSJed Brown     suffix: p1_3d_3
955c4762a1bSJed Brown     requires: ctetgen pragmatic
95630602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
957c4762a1bSJed Brown     filter: grep -v DEBUG
958c4762a1bSJed Brown   test:
959c4762a1bSJed Brown     suffix: p1_3d_4
960c4762a1bSJed Brown     requires: ctetgen pragmatic
96130602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
962c4762a1bSJed Brown   test:
963c4762a1bSJed Brown     suffix: p1_3d_5
964c4762a1bSJed Brown     requires: ctetgen pragmatic
96530602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
966c4762a1bSJed Brown 
967c4762a1bSJed Brown   # 2D P_2 on a triangle
968c4762a1bSJed Brown   test:
969c4762a1bSJed Brown     suffix: p2_2d_0
970c4762a1bSJed Brown     requires: triangle
971c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -convergence
972c4762a1bSJed Brown   test:
973c4762a1bSJed Brown     suffix: p2_2d_1
974c4762a1bSJed Brown     requires: triangle
975c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -porder 1
976c4762a1bSJed Brown   test:
977c4762a1bSJed Brown     suffix: p2_2d_2
978c4762a1bSJed Brown     requires: triangle
979c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -porder 2
980c4762a1bSJed Brown   test:
981c4762a1bSJed Brown     suffix: p2_2d_3
982c4762a1bSJed Brown     requires: triangle pragmatic
983c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -convergence -conv_refine 0
984c4762a1bSJed Brown     filter: grep -v DEBUG
985c4762a1bSJed Brown   test:
986c4762a1bSJed Brown     suffix: p2_2d_4
987c4762a1bSJed Brown     requires: triangle pragmatic
988c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 1 -conv_refine 0
989c4762a1bSJed Brown   test:
990c4762a1bSJed Brown     suffix: p2_2d_5
991c4762a1bSJed Brown     requires: triangle pragmatic
992c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 2 -conv_refine 0
993c4762a1bSJed Brown 
994c4762a1bSJed Brown   # 3D P_2 on a tetrahedron
995c4762a1bSJed Brown   test:
996c4762a1bSJed Brown     suffix: p2_3d_0
997c4762a1bSJed Brown     requires: ctetgen
99830602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -convergence
999c4762a1bSJed Brown   test:
1000c4762a1bSJed Brown     suffix: p2_3d_1
1001c4762a1bSJed Brown     requires: ctetgen
100230602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -porder 1
1003c4762a1bSJed Brown   test:
1004c4762a1bSJed Brown     suffix: p2_3d_2
1005c4762a1bSJed Brown     requires: ctetgen
100630602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -porder 2
1007c4762a1bSJed Brown   test:
1008c4762a1bSJed Brown     suffix: p2_3d_3
1009c4762a1bSJed Brown     requires: ctetgen pragmatic
101030602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -convergence -conv_refine 0
1011c4762a1bSJed Brown     filter: grep -v DEBUG
1012c4762a1bSJed Brown   test:
1013c4762a1bSJed Brown     suffix: p2_3d_4
1014c4762a1bSJed Brown     requires: ctetgen pragmatic
101530602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 1 -conv_refine 0
1016c4762a1bSJed Brown   test:
1017c4762a1bSJed Brown     suffix: p2_3d_5
1018c4762a1bSJed Brown     requires: ctetgen pragmatic
101930602db0SMatthew G. Knepley     args: -dm_plex_dim 3 -petscspace_degree 2 -qorder 2 -dm_plex_hash_location -porder 2 -conv_refine 0
1020c4762a1bSJed Brown 
1021c4762a1bSJed Brown   # 2D Q_1 on a quadrilaterial DA
1022c4762a1bSJed Brown   test:
1023c4762a1bSJed Brown     suffix: q1_2d_da_0
1024c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
102530602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -convergence
1026c4762a1bSJed Brown   test:
1027c4762a1bSJed Brown     suffix: q1_2d_da_1
1028c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
102930602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -porder 1
1030c4762a1bSJed Brown   test:
1031c4762a1bSJed Brown     suffix: q1_2d_da_2
1032c4762a1bSJed Brown     requires: mpi_type_get_envelope broken
103330602db0SMatthew G. Knepley     args: -use_da 1 -petscspace_degree 1 -qorder 1 -porder 2
1034c4762a1bSJed Brown 
1035c4762a1bSJed Brown   # 2D Q_1 on a quadrilaterial Plex
1036c4762a1bSJed Brown   test:
1037c4762a1bSJed Brown     suffix: q1_2d_plex_0
103830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -convergence
1039c4762a1bSJed Brown   test:
1040c4762a1bSJed Brown     suffix: q1_2d_plex_1
104130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 1
1042c4762a1bSJed Brown   test:
1043c4762a1bSJed Brown     suffix: q1_2d_plex_2
104430602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 2
1045c4762a1bSJed Brown   test:
1046c4762a1bSJed Brown     suffix: q1_2d_plex_3
104730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 1 -shear_coords
1048c4762a1bSJed Brown   test:
1049c4762a1bSJed Brown     suffix: q1_2d_plex_4
105030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 1 -porder 2 -shear_coords
1051c4762a1bSJed Brown   test:
1052c4762a1bSJed Brown     suffix: q1_2d_plex_5
105330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 0 -non_affine_coords -convergence
1054c4762a1bSJed Brown   test:
1055c4762a1bSJed Brown     suffix: q1_2d_plex_6
105630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 1 -non_affine_coords -convergence
1057c4762a1bSJed Brown   test:
1058c4762a1bSJed Brown     suffix: q1_2d_plex_7
105930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscspace_type tensor -qorder 1 -porder 2 -non_affine_coords -convergence
1060c4762a1bSJed Brown 
1061c4762a1bSJed Brown   # 2D Q_2 on a quadrilaterial
1062c4762a1bSJed Brown   test:
1063c4762a1bSJed Brown     suffix: q2_2d_plex_0
1064c4762a1bSJed Brown     requires: mpi_type_get_envelope
106530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -convergence
1066c4762a1bSJed Brown   test:
1067c4762a1bSJed Brown     suffix: q2_2d_plex_1
1068c4762a1bSJed Brown     requires: mpi_type_get_envelope
106930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1
1070c4762a1bSJed Brown   test:
1071c4762a1bSJed Brown     suffix: q2_2d_plex_2
1072c4762a1bSJed Brown     requires: mpi_type_get_envelope
107330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2
1074c4762a1bSJed Brown   test:
1075c4762a1bSJed Brown     suffix: q2_2d_plex_3
107630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1 -shear_coords
1077c4762a1bSJed Brown   test:
1078c4762a1bSJed Brown     suffix: q2_2d_plex_4
1079c4762a1bSJed Brown     requires: mpi_type_get_envelope
108030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2 -shear_coords
1081c4762a1bSJed Brown   test:
1082c4762a1bSJed Brown     suffix: q2_2d_plex_5
1083c4762a1bSJed Brown     requires: mpi_type_get_envelope
108430602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 0 -non_affine_coords -convergence
1085c4762a1bSJed Brown   test:
1086c4762a1bSJed Brown     suffix: q2_2d_plex_6
1087c4762a1bSJed Brown     requires: mpi_type_get_envelope
108830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 1 -non_affine_coords -convergence
1089c4762a1bSJed Brown   test:
1090c4762a1bSJed Brown     suffix: q2_2d_plex_7
1091c4762a1bSJed Brown     requires: mpi_type_get_envelope
109230602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_type tensor -qorder 2 -porder 2 -non_affine_coords -convergence
1093c4762a1bSJed Brown 
1094c4762a1bSJed Brown   # 2D P_3 on a triangle
1095c4762a1bSJed Brown   test:
1096c4762a1bSJed Brown     suffix: p3_2d_0
1097c4762a1bSJed Brown     requires: triangle !single
1098c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -convergence
1099c4762a1bSJed Brown   test:
1100c4762a1bSJed Brown     suffix: p3_2d_1
1101c4762a1bSJed Brown     requires: triangle !single
1102c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 1
1103c4762a1bSJed Brown   test:
1104c4762a1bSJed Brown     suffix: p3_2d_2
1105c4762a1bSJed Brown     requires: triangle !single
1106c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 2
1107c4762a1bSJed Brown   test:
1108c4762a1bSJed Brown     suffix: p3_2d_3
1109c4762a1bSJed Brown     requires: triangle !single
1110c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -porder 3
1111c4762a1bSJed Brown   test:
1112c4762a1bSJed Brown     suffix: p3_2d_4
1113c4762a1bSJed Brown     requires: triangle pragmatic
1114c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -convergence -conv_refine 0
1115c4762a1bSJed Brown     filter: grep -v DEBUG
1116c4762a1bSJed Brown   test:
1117c4762a1bSJed Brown     suffix: p3_2d_5
1118c4762a1bSJed Brown     requires: triangle pragmatic
1119c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -porder 1 -conv_refine 0
1120c4762a1bSJed Brown   test:
1121c4762a1bSJed Brown     suffix: p3_2d_6
1122c4762a1bSJed Brown     requires: triangle pragmatic
1123c4762a1bSJed Brown     args: -petscspace_degree 3 -qorder 3 -dm_plex_hash_location -porder 3 -conv_refine 0
1124c4762a1bSJed Brown 
1125c4762a1bSJed Brown   # 2D Q_3 on a quadrilaterial
1126c4762a1bSJed Brown   test:
1127c4762a1bSJed Brown     suffix: q3_2d_0
1128c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
112930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -convergence
1130c4762a1bSJed Brown   test:
1131c4762a1bSJed Brown     suffix: q3_2d_1
1132c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
113330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 1
1134c4762a1bSJed Brown   test:
1135c4762a1bSJed Brown     suffix: q3_2d_2
1136c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
113730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 2
1138c4762a1bSJed Brown   test:
1139c4762a1bSJed Brown     suffix: q3_2d_3
1140c4762a1bSJed Brown     requires: mpi_type_get_envelope !single
114130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 3 -qorder 3 -porder 3
1142c4762a1bSJed Brown 
1143c4762a1bSJed Brown   # 2D P_1disc on a triangle/quadrilateral
1144c4762a1bSJed Brown   test:
1145c4762a1bSJed Brown     suffix: p1d_2d_0
1146c4762a1bSJed Brown     requires: triangle
1147c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -convergence
1148c4762a1bSJed Brown   test:
1149c4762a1bSJed Brown     suffix: p1d_2d_1
1150c4762a1bSJed Brown     requires: triangle
1151c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 1
1152c4762a1bSJed Brown   test:
1153c4762a1bSJed Brown     suffix: p1d_2d_2
1154c4762a1bSJed Brown     requires: triangle
1155c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 2
1156c4762a1bSJed Brown   test:
1157c4762a1bSJed Brown     suffix: p1d_2d_3
1158c4762a1bSJed Brown     requires: triangle
115930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -convergence
1160c4762a1bSJed Brown     filter: sed  -e "s/convergence rate at refinement 0: 2/convergence rate at refinement 0: 1.9/g"
1161c4762a1bSJed Brown   test:
1162c4762a1bSJed Brown     suffix: p1d_2d_4
1163c4762a1bSJed Brown     requires: triangle
116430602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 1
1165c4762a1bSJed Brown   test:
1166c4762a1bSJed Brown     suffix: p1d_2d_5
1167c4762a1bSJed Brown     requires: triangle
116830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -petscdualspace_lagrange_continuity 0 -qorder 1 -porder 2
1169c4762a1bSJed Brown 
1170c4762a1bSJed Brown   # 2D BDM_1 on a triangle
1171c4762a1bSJed Brown   test:
1172c4762a1bSJed Brown     suffix: bdm1_2d_0
1173c4762a1bSJed Brown     requires: triangle
1174c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1175c4762a1bSJed Brown           -num_comp 2 -qorder 1 -convergence
1176c4762a1bSJed Brown   test:
1177c4762a1bSJed Brown     suffix: bdm1_2d_1
1178c4762a1bSJed Brown     requires: triangle
1179c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1180c4762a1bSJed Brown           -num_comp 2 -qorder 1 -porder 1
1181c4762a1bSJed Brown   test:
1182c4762a1bSJed Brown     suffix: bdm1_2d_2
1183c4762a1bSJed Brown     requires: triangle
1184c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
1185c4762a1bSJed Brown           -num_comp 2 -qorder 1 -porder 2
1186c4762a1bSJed Brown 
1187c4762a1bSJed Brown   # 2D BDM_1 on a quadrilateral
1188c4762a1bSJed Brown   test:
1189c4762a1bSJed Brown     suffix: bdm1q_2d_0
1190c4762a1bSJed Brown     requires: triangle
1191c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
11923f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
119330602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -convergence
1194c4762a1bSJed Brown   test:
1195c4762a1bSJed Brown     suffix: bdm1q_2d_1
1196c4762a1bSJed Brown     requires: triangle
1197c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
11983f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
119930602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -porder 1
1200c4762a1bSJed Brown   test:
1201c4762a1bSJed Brown     suffix: bdm1q_2d_2
1202c4762a1bSJed Brown     requires: triangle
1203c4762a1bSJed Brown     args: -petscspace_degree 1 -petscdualspace_type bdm \
12043f27d899SToby Isaac           -petscdualspace_lagrange_tensor 1 \
120530602db0SMatthew G. Knepley           -dm_plex_simplex 0 -num_comp 2 -qorder 1 -porder 2
1206c4762a1bSJed Brown 
1207c4762a1bSJed Brown   # Test high order quadrature
1208c4762a1bSJed Brown   test:
1209c4762a1bSJed Brown     suffix: p1_quad_2
1210c4762a1bSJed Brown     requires: triangle
1211c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 2 -porder 1
1212c4762a1bSJed Brown   test:
1213c4762a1bSJed Brown     suffix: p1_quad_5
1214c4762a1bSJed Brown     requires: triangle
1215c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 5 -porder 1
1216c4762a1bSJed Brown   test:
1217c4762a1bSJed Brown     suffix: p2_quad_3
1218c4762a1bSJed Brown     requires: triangle
1219c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 3 -porder 2
1220c4762a1bSJed Brown   test:
1221c4762a1bSJed Brown     suffix: p2_quad_5
1222c4762a1bSJed Brown     requires: triangle
1223c4762a1bSJed Brown     args: -petscspace_degree 2 -qorder 5 -porder 2
1224c4762a1bSJed Brown   test:
1225c4762a1bSJed Brown     suffix: q1_quad_2
1226c4762a1bSJed Brown     requires: mpi_type_get_envelope
122730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 2 -porder 1
1228c4762a1bSJed Brown   test:
1229c4762a1bSJed Brown     suffix: q1_quad_5
1230c4762a1bSJed Brown     requires: mpi_type_get_envelope
123130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 1 -qorder 5 -porder 1
1232c4762a1bSJed Brown   test:
1233c4762a1bSJed Brown     suffix: q2_quad_3
1234c4762a1bSJed Brown     requires: mpi_type_get_envelope
123530602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 3 -porder 1
1236c4762a1bSJed Brown   test:
1237c4762a1bSJed Brown     suffix: q2_quad_5
1238c4762a1bSJed Brown     requires: mpi_type_get_envelope
123930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 5 -porder 1
1240c4762a1bSJed Brown 
1241c4762a1bSJed Brown   # Nonconforming tests
1242c4762a1bSJed Brown   test:
1243c4762a1bSJed Brown     suffix: constraints
124430602db0SMatthew G. Knepley     args: -dm_coord_space 0 -dm_plex_simplex 0 -petscspace_type tensor -petscspace_degree 1 -qorder 0 -constraints
1245c4762a1bSJed Brown   test:
1246c4762a1bSJed Brown     suffix: nonconforming_tensor_2
1247c4762a1bSJed Brown     nsize: 4
124830602db0SMatthew 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
1249c4762a1bSJed Brown   test:
1250c4762a1bSJed Brown     suffix: nonconforming_tensor_3
1251c4762a1bSJed Brown     nsize: 4
125230602db0SMatthew 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
1253c4762a1bSJed Brown   test:
1254c4762a1bSJed Brown     suffix: nonconforming_tensor_2_fv
1255c4762a1bSJed Brown     nsize: 4
125630602db0SMatthew 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
1257c4762a1bSJed Brown   test:
1258c4762a1bSJed Brown     suffix: nonconforming_tensor_3_fv
1259c4762a1bSJed Brown     nsize: 4
126030602db0SMatthew 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
1261c4762a1bSJed Brown   test:
1262c4762a1bSJed Brown     suffix: nonconforming_tensor_2_hi
1263c4762a1bSJed Brown     requires: !single
1264c4762a1bSJed Brown     nsize: 4
126530602db0SMatthew 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
1266c4762a1bSJed Brown   test:
1267c4762a1bSJed Brown     suffix: nonconforming_tensor_3_hi
1268c4762a1bSJed Brown     requires: !single skip
1269c4762a1bSJed Brown     nsize: 4
127030602db0SMatthew 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
1271c4762a1bSJed Brown   test:
1272c4762a1bSJed Brown     suffix: nonconforming_simplex_2
1273c4762a1bSJed Brown     requires: triangle
1274c4762a1bSJed Brown     nsize: 4
127530602db0SMatthew 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
1276c4762a1bSJed Brown   test:
1277c4762a1bSJed Brown     suffix: nonconforming_simplex_2_hi
1278c4762a1bSJed Brown     requires: triangle !single
1279c4762a1bSJed Brown     nsize: 4
128030602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fe_jacobian -petscpartitioner_type simple -tree -dm_plex_max_projection_height 1 -petscspace_degree 4 -qorder 4
1281c4762a1bSJed Brown   test:
1282c4762a1bSJed Brown     suffix: nonconforming_simplex_2_fv
1283c4762a1bSJed Brown     requires: triangle
1284c4762a1bSJed Brown     nsize: 4
128530602db0SMatthew G. Knepley     args: -dist_dm_distribute -test_fv_grad -test_injector -petsclimiter_type none -petscpartitioner_type simple -tree -num_comp 2
1286c4762a1bSJed Brown   test:
1287c4762a1bSJed Brown     suffix: nonconforming_simplex_3
1288c4762a1bSJed Brown     requires: ctetgen
1289c4762a1bSJed Brown     nsize: 4
129030602db0SMatthew 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
1291c4762a1bSJed Brown   test:
1292c4762a1bSJed Brown     suffix: nonconforming_simplex_3_hi
1293c4762a1bSJed Brown     requires: ctetgen skip
1294c4762a1bSJed Brown     nsize: 4
129530602db0SMatthew 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
1296c4762a1bSJed Brown   test:
1297c4762a1bSJed Brown     suffix: nonconforming_simplex_3_fv
1298c4762a1bSJed Brown     requires: ctetgen
1299c4762a1bSJed Brown     nsize: 4
130030602db0SMatthew 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
1301c4762a1bSJed Brown 
1302c4762a1bSJed Brown TEST*/
1303c4762a1bSJed Brown 
1304c4762a1bSJed Brown /*
1305c4762a1bSJed Brown    # 2D Q_2 on a quadrilaterial Plex
1306c4762a1bSJed Brown   test:
1307c4762a1bSJed Brown     suffix: q2_2d_plex_0
130830602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -convergence
1309c4762a1bSJed Brown   test:
1310c4762a1bSJed Brown     suffix: q2_2d_plex_1
131130602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1
1312c4762a1bSJed Brown   test:
1313c4762a1bSJed Brown     suffix: q2_2d_plex_2
131430602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2
1315c4762a1bSJed Brown   test:
1316c4762a1bSJed Brown     suffix: q2_2d_plex_3
131730602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 1 -shear_coords
1318c4762a1bSJed Brown   test:
1319c4762a1bSJed Brown     suffix: q2_2d_plex_4
132030602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -qorder 2 -porder 2 -shear_coords
1321c4762a1bSJed Brown   test:
1322c4762a1bSJed Brown     suffix: q2_2d_plex_5
132330602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 0 -non_affine_coords
1324c4762a1bSJed Brown   test:
1325c4762a1bSJed Brown     suffix: q2_2d_plex_6
132630602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 1 -non_affine_coords
1327c4762a1bSJed Brown   test:
1328c4762a1bSJed Brown     suffix: q2_2d_plex_7
132930602db0SMatthew G. Knepley     args: -dm_plex_simplex 0 -petscspace_degree 2 -petscspace_poly_tensor 1 -qorder 2 -porder 2 -non_affine_coords
1330c4762a1bSJed Brown 
1331c4762a1bSJed Brown   test:
1332c4762a1bSJed Brown     suffix: p1d_2d_6
1333c4762a1bSJed Brown     requires: pragmatic
1334c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -convergence -conv_refine 0
1335c4762a1bSJed Brown   test:
1336c4762a1bSJed Brown     suffix: p1d_2d_7
1337c4762a1bSJed Brown     requires: pragmatic
1338c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 1 -conv_refine 0
1339c4762a1bSJed Brown   test:
1340c4762a1bSJed Brown     suffix: p1d_2d_8
1341c4762a1bSJed Brown     requires: pragmatic
1342c4762a1bSJed Brown     args: -petscspace_degree 1 -qorder 1 -dm_plex_hash_location -porder 2 -conv_refine 0
1343c4762a1bSJed Brown */
1344