xref: /petsc/src/dm/dt/fe/interface/fe.c (revision fe2efc57b7777594bce7568e90822861480cbdc8)
120cf1dd8SToby Isaac /* Basis Jet Tabulation
220cf1dd8SToby Isaac 
320cf1dd8SToby Isaac We would like to tabulate the nodal basis functions and derivatives at a set of points, usually quadrature points. We
420cf1dd8SToby Isaac follow here the derviation in http://www.math.ttu.edu/~kirby/papers/fiat-toms-2004.pdf. The nodal basis $\psi_i$ can
520cf1dd8SToby Isaac be expressed in terms of a prime basis $\phi_i$ which can be stably evaluated. In PETSc, we will use the Legendre basis
620cf1dd8SToby Isaac as a prime basis.
720cf1dd8SToby Isaac 
820cf1dd8SToby Isaac   \psi_i = \sum_k \alpha_{ki} \phi_k
920cf1dd8SToby Isaac 
1020cf1dd8SToby Isaac Our nodal basis is defined in terms of the dual basis $n_j$
1120cf1dd8SToby Isaac 
1220cf1dd8SToby Isaac   n_j \cdot \psi_i = \delta_{ji}
1320cf1dd8SToby Isaac 
1420cf1dd8SToby Isaac and we may act on the first equation to obtain
1520cf1dd8SToby Isaac 
1620cf1dd8SToby Isaac   n_j \cdot \psi_i = \sum_k \alpha_{ki} n_j \cdot \phi_k
1720cf1dd8SToby Isaac        \delta_{ji} = \sum_k \alpha_{ki} V_{jk}
1820cf1dd8SToby Isaac                  I = V \alpha
1920cf1dd8SToby Isaac 
2020cf1dd8SToby Isaac so the coefficients of the nodal basis in the prime basis are
2120cf1dd8SToby Isaac 
2220cf1dd8SToby Isaac    \alpha = V^{-1}
2320cf1dd8SToby Isaac 
2420cf1dd8SToby Isaac We will define the dual basis vectors $n_j$ using a quadrature rule.
2520cf1dd8SToby Isaac 
2620cf1dd8SToby Isaac Right now, we will just use the polynomial spaces P^k. I know some elements use the space of symmetric polynomials
2720cf1dd8SToby Isaac (I think Nedelec), but we will neglect this for now. Constraints in the space, e.g. Arnold-Winther elements, can
2820cf1dd8SToby Isaac be implemented exactly as in FIAT using functionals $L_j$.
2920cf1dd8SToby Isaac 
3020cf1dd8SToby Isaac I will have to count the degrees correctly for the Legendre product when we are on simplices.
3120cf1dd8SToby Isaac 
3220cf1dd8SToby Isaac We will have three objects:
3320cf1dd8SToby Isaac  - Space, P: this just need point evaluation I think
3420cf1dd8SToby Isaac  - Dual Space, P'+K: This looks like a set of functionals that can act on members of P, each n is defined by a Q
3520cf1dd8SToby Isaac  - FEM: This keeps {P, P', Q}
3620cf1dd8SToby Isaac */
3720cf1dd8SToby Isaac #include <petsc/private/petscfeimpl.h> /*I "petscfe.h" I*/
3820cf1dd8SToby Isaac #include <petscdmplex.h>
3920cf1dd8SToby Isaac 
4020cf1dd8SToby Isaac PetscBool FEcite = PETSC_FALSE;
4120cf1dd8SToby Isaac const char FECitation[] = "@article{kirby2004,\n"
4220cf1dd8SToby Isaac                           "  title   = {Algorithm 839: FIAT, a New Paradigm for Computing Finite Element Basis Functions},\n"
4320cf1dd8SToby Isaac                           "  journal = {ACM Transactions on Mathematical Software},\n"
4420cf1dd8SToby Isaac                           "  author  = {Robert C. Kirby},\n"
4520cf1dd8SToby Isaac                           "  volume  = {30},\n"
4620cf1dd8SToby Isaac                           "  number  = {4},\n"
4720cf1dd8SToby Isaac                           "  pages   = {502--516},\n"
4820cf1dd8SToby Isaac                           "  doi     = {10.1145/1039813.1039820},\n"
4920cf1dd8SToby Isaac                           "  year    = {2004}\n}\n";
5020cf1dd8SToby Isaac 
5120cf1dd8SToby Isaac PetscClassId PETSCFE_CLASSID = 0;
5220cf1dd8SToby Isaac 
5320cf1dd8SToby Isaac PetscFunctionList PetscFEList              = NULL;
5420cf1dd8SToby Isaac PetscBool         PetscFERegisterAllCalled = PETSC_FALSE;
5520cf1dd8SToby Isaac 
5620cf1dd8SToby Isaac /*@C
5720cf1dd8SToby Isaac   PetscFERegister - Adds a new PetscFE implementation
5820cf1dd8SToby Isaac 
5920cf1dd8SToby Isaac   Not Collective
6020cf1dd8SToby Isaac 
6120cf1dd8SToby Isaac   Input Parameters:
6220cf1dd8SToby Isaac + name        - The name of a new user-defined creation routine
6320cf1dd8SToby Isaac - create_func - The creation routine itself
6420cf1dd8SToby Isaac 
6520cf1dd8SToby Isaac   Notes:
6620cf1dd8SToby Isaac   PetscFERegister() may be called multiple times to add several user-defined PetscFEs
6720cf1dd8SToby Isaac 
6820cf1dd8SToby Isaac   Sample usage:
6920cf1dd8SToby Isaac .vb
7020cf1dd8SToby Isaac     PetscFERegister("my_fe", MyPetscFECreate);
7120cf1dd8SToby Isaac .ve
7220cf1dd8SToby Isaac 
7320cf1dd8SToby Isaac   Then, your PetscFE type can be chosen with the procedural interface via
7420cf1dd8SToby Isaac .vb
7520cf1dd8SToby Isaac     PetscFECreate(MPI_Comm, PetscFE *);
7620cf1dd8SToby Isaac     PetscFESetType(PetscFE, "my_fe");
7720cf1dd8SToby Isaac .ve
7820cf1dd8SToby Isaac    or at runtime via the option
7920cf1dd8SToby Isaac .vb
8020cf1dd8SToby Isaac     -petscfe_type my_fe
8120cf1dd8SToby Isaac .ve
8220cf1dd8SToby Isaac 
8320cf1dd8SToby Isaac   Level: advanced
8420cf1dd8SToby Isaac 
8520cf1dd8SToby Isaac .seealso: PetscFERegisterAll(), PetscFERegisterDestroy()
8620cf1dd8SToby Isaac 
8720cf1dd8SToby Isaac @*/
8820cf1dd8SToby Isaac PetscErrorCode PetscFERegister(const char sname[], PetscErrorCode (*function)(PetscFE))
8920cf1dd8SToby Isaac {
9020cf1dd8SToby Isaac   PetscErrorCode ierr;
9120cf1dd8SToby Isaac 
9220cf1dd8SToby Isaac   PetscFunctionBegin;
9320cf1dd8SToby Isaac   ierr = PetscFunctionListAdd(&PetscFEList, sname, function);CHKERRQ(ierr);
9420cf1dd8SToby Isaac   PetscFunctionReturn(0);
9520cf1dd8SToby Isaac }
9620cf1dd8SToby Isaac 
9720cf1dd8SToby Isaac /*@C
9820cf1dd8SToby Isaac   PetscFESetType - Builds a particular PetscFE
9920cf1dd8SToby Isaac 
100d083f849SBarry Smith   Collective on fem
10120cf1dd8SToby Isaac 
10220cf1dd8SToby Isaac   Input Parameters:
10320cf1dd8SToby Isaac + fem  - The PetscFE object
10420cf1dd8SToby Isaac - name - The kind of FEM space
10520cf1dd8SToby Isaac 
10620cf1dd8SToby Isaac   Options Database Key:
10720cf1dd8SToby Isaac . -petscfe_type <type> - Sets the PetscFE type; use -help for a list of available types
10820cf1dd8SToby Isaac 
10920cf1dd8SToby Isaac   Level: intermediate
11020cf1dd8SToby Isaac 
11120cf1dd8SToby Isaac .seealso: PetscFEGetType(), PetscFECreate()
11220cf1dd8SToby Isaac @*/
11320cf1dd8SToby Isaac PetscErrorCode PetscFESetType(PetscFE fem, PetscFEType name)
11420cf1dd8SToby Isaac {
11520cf1dd8SToby Isaac   PetscErrorCode (*r)(PetscFE);
11620cf1dd8SToby Isaac   PetscBool      match;
11720cf1dd8SToby Isaac   PetscErrorCode ierr;
11820cf1dd8SToby Isaac 
11920cf1dd8SToby Isaac   PetscFunctionBegin;
12020cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
12120cf1dd8SToby Isaac   ierr = PetscObjectTypeCompare((PetscObject) fem, name, &match);CHKERRQ(ierr);
12220cf1dd8SToby Isaac   if (match) PetscFunctionReturn(0);
12320cf1dd8SToby Isaac 
12420cf1dd8SToby Isaac   if (!PetscFERegisterAllCalled) {ierr = PetscFERegisterAll();CHKERRQ(ierr);}
12520cf1dd8SToby Isaac   ierr = PetscFunctionListFind(PetscFEList, name, &r);CHKERRQ(ierr);
12620cf1dd8SToby Isaac   if (!r) SETERRQ1(PetscObjectComm((PetscObject) fem), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown PetscFE type: %s", name);
12720cf1dd8SToby Isaac 
12820cf1dd8SToby Isaac   if (fem->ops->destroy) {
12920cf1dd8SToby Isaac     ierr              = (*fem->ops->destroy)(fem);CHKERRQ(ierr);
13020cf1dd8SToby Isaac     fem->ops->destroy = NULL;
13120cf1dd8SToby Isaac   }
13220cf1dd8SToby Isaac   ierr = (*r)(fem);CHKERRQ(ierr);
13320cf1dd8SToby Isaac   ierr = PetscObjectChangeTypeName((PetscObject) fem, name);CHKERRQ(ierr);
13420cf1dd8SToby Isaac   PetscFunctionReturn(0);
13520cf1dd8SToby Isaac }
13620cf1dd8SToby Isaac 
13720cf1dd8SToby Isaac /*@C
13820cf1dd8SToby Isaac   PetscFEGetType - Gets the PetscFE type name (as a string) from the object.
13920cf1dd8SToby Isaac 
14020cf1dd8SToby Isaac   Not Collective
14120cf1dd8SToby Isaac 
14220cf1dd8SToby Isaac   Input Parameter:
14320cf1dd8SToby Isaac . fem  - The PetscFE
14420cf1dd8SToby Isaac 
14520cf1dd8SToby Isaac   Output Parameter:
14620cf1dd8SToby Isaac . name - The PetscFE type name
14720cf1dd8SToby Isaac 
14820cf1dd8SToby Isaac   Level: intermediate
14920cf1dd8SToby Isaac 
15020cf1dd8SToby Isaac .seealso: PetscFESetType(), PetscFECreate()
15120cf1dd8SToby Isaac @*/
15220cf1dd8SToby Isaac PetscErrorCode PetscFEGetType(PetscFE fem, PetscFEType *name)
15320cf1dd8SToby Isaac {
15420cf1dd8SToby Isaac   PetscErrorCode ierr;
15520cf1dd8SToby Isaac 
15620cf1dd8SToby Isaac   PetscFunctionBegin;
15720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
15820cf1dd8SToby Isaac   PetscValidPointer(name, 2);
15920cf1dd8SToby Isaac   if (!PetscFERegisterAllCalled) {
16020cf1dd8SToby Isaac     ierr = PetscFERegisterAll();CHKERRQ(ierr);
16120cf1dd8SToby Isaac   }
16220cf1dd8SToby Isaac   *name = ((PetscObject) fem)->type_name;
16320cf1dd8SToby Isaac   PetscFunctionReturn(0);
16420cf1dd8SToby Isaac }
16520cf1dd8SToby Isaac 
16620cf1dd8SToby Isaac /*@C
167*fe2efc57SMark    PetscFEViewFromOptions - View from Options
168*fe2efc57SMark 
169*fe2efc57SMark    Collective on PetscFE
170*fe2efc57SMark 
171*fe2efc57SMark    Input Parameters:
172*fe2efc57SMark +  A - the PetscFE object
173*fe2efc57SMark .  obj - Optional object
174*fe2efc57SMark -  name - command line option
175*fe2efc57SMark 
176*fe2efc57SMark    Level: intermediate
177*fe2efc57SMark .seealso:  PetscFE(), PetscFEView(), PetscObjectViewFromOptions(), PetscFECreate()
178*fe2efc57SMark @*/
179*fe2efc57SMark PetscErrorCode  PetscFEViewFromOptions(PetscFE A,PetscObject obj,const char name[])
180*fe2efc57SMark {
181*fe2efc57SMark   PetscErrorCode ierr;
182*fe2efc57SMark 
183*fe2efc57SMark   PetscFunctionBegin;
184*fe2efc57SMark   PetscValidHeaderSpecific(A,PETSCFE_CLASSID,1);
185*fe2efc57SMark   ierr = PetscObjectViewFromOptions((PetscObject)A,obj,name);CHKERRQ(ierr);
186*fe2efc57SMark   PetscFunctionReturn(0);
187*fe2efc57SMark }
188*fe2efc57SMark 
189*fe2efc57SMark /*@C
19020cf1dd8SToby Isaac   PetscFEView - Views a PetscFE
19120cf1dd8SToby Isaac 
192d083f849SBarry Smith   Collective on fem
19320cf1dd8SToby Isaac 
19420cf1dd8SToby Isaac   Input Parameter:
19520cf1dd8SToby Isaac + fem - the PetscFE object to view
196d9bac1caSLisandro Dalcin - viewer   - the viewer
19720cf1dd8SToby Isaac 
1982b99622eSMatthew G. Knepley   Level: beginner
19920cf1dd8SToby Isaac 
20020cf1dd8SToby Isaac .seealso PetscFEDestroy()
20120cf1dd8SToby Isaac @*/
202d9bac1caSLisandro Dalcin PetscErrorCode PetscFEView(PetscFE fem, PetscViewer viewer)
20320cf1dd8SToby Isaac {
204d9bac1caSLisandro Dalcin   PetscBool      iascii;
20520cf1dd8SToby Isaac   PetscErrorCode ierr;
20620cf1dd8SToby Isaac 
20720cf1dd8SToby Isaac   PetscFunctionBegin;
20820cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
209d9bac1caSLisandro Dalcin   if (viewer) PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
210d9bac1caSLisandro Dalcin   if (!viewer) {ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject) fem), &viewer);CHKERRQ(ierr);}
211d9bac1caSLisandro Dalcin   ierr = PetscObjectPrintClassNamePrefixType((PetscObject)fem, viewer);CHKERRQ(ierr);
212d9bac1caSLisandro Dalcin   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
213d9bac1caSLisandro Dalcin   if (fem->ops->view) {ierr = (*fem->ops->view)(fem, viewer);CHKERRQ(ierr);}
21420cf1dd8SToby Isaac   PetscFunctionReturn(0);
21520cf1dd8SToby Isaac }
21620cf1dd8SToby Isaac 
21720cf1dd8SToby Isaac /*@
21820cf1dd8SToby Isaac   PetscFESetFromOptions - sets parameters in a PetscFE from the options database
21920cf1dd8SToby Isaac 
220d083f849SBarry Smith   Collective on fem
22120cf1dd8SToby Isaac 
22220cf1dd8SToby Isaac   Input Parameter:
22320cf1dd8SToby Isaac . fem - the PetscFE object to set options for
22420cf1dd8SToby Isaac 
22520cf1dd8SToby Isaac   Options Database:
226a2b725a8SWilliam Gropp + -petscfe_num_blocks  - the number of cell blocks to integrate concurrently
227a2b725a8SWilliam Gropp - -petscfe_num_batches - the number of cell batches to integrate serially
22820cf1dd8SToby Isaac 
2292b99622eSMatthew G. Knepley   Level: intermediate
23020cf1dd8SToby Isaac 
23120cf1dd8SToby Isaac .seealso PetscFEView()
23220cf1dd8SToby Isaac @*/
23320cf1dd8SToby Isaac PetscErrorCode PetscFESetFromOptions(PetscFE fem)
23420cf1dd8SToby Isaac {
23520cf1dd8SToby Isaac   const char    *defaultType;
23620cf1dd8SToby Isaac   char           name[256];
23720cf1dd8SToby Isaac   PetscBool      flg;
23820cf1dd8SToby Isaac   PetscErrorCode ierr;
23920cf1dd8SToby Isaac 
24020cf1dd8SToby Isaac   PetscFunctionBegin;
24120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
24220cf1dd8SToby Isaac   if (!((PetscObject) fem)->type_name) {
24320cf1dd8SToby Isaac     defaultType = PETSCFEBASIC;
24420cf1dd8SToby Isaac   } else {
24520cf1dd8SToby Isaac     defaultType = ((PetscObject) fem)->type_name;
24620cf1dd8SToby Isaac   }
24720cf1dd8SToby Isaac   if (!PetscFERegisterAllCalled) {ierr = PetscFERegisterAll();CHKERRQ(ierr);}
24820cf1dd8SToby Isaac 
24920cf1dd8SToby Isaac   ierr = PetscObjectOptionsBegin((PetscObject) fem);CHKERRQ(ierr);
25020cf1dd8SToby Isaac   ierr = PetscOptionsFList("-petscfe_type", "Finite element space", "PetscFESetType", PetscFEList, defaultType, name, 256, &flg);CHKERRQ(ierr);
25120cf1dd8SToby Isaac   if (flg) {
25220cf1dd8SToby Isaac     ierr = PetscFESetType(fem, name);CHKERRQ(ierr);
25320cf1dd8SToby Isaac   } else if (!((PetscObject) fem)->type_name) {
25420cf1dd8SToby Isaac     ierr = PetscFESetType(fem, defaultType);CHKERRQ(ierr);
25520cf1dd8SToby Isaac   }
2565a856986SBarry Smith   ierr = PetscOptionsBoundedInt("-petscfe_num_blocks", "The number of cell blocks to integrate concurrently", "PetscSpaceSetTileSizes", fem->numBlocks, &fem->numBlocks, NULL,1);CHKERRQ(ierr);
2575a856986SBarry Smith   ierr = PetscOptionsBoundedInt("-petscfe_num_batches", "The number of cell batches to integrate serially", "PetscSpaceSetTileSizes", fem->numBatches, &fem->numBatches, NULL,1);CHKERRQ(ierr);
25820cf1dd8SToby Isaac   if (fem->ops->setfromoptions) {
25920cf1dd8SToby Isaac     ierr = (*fem->ops->setfromoptions)(PetscOptionsObject,fem);CHKERRQ(ierr);
26020cf1dd8SToby Isaac   }
26120cf1dd8SToby Isaac   /* process any options handlers added with PetscObjectAddOptionsHandler() */
26220cf1dd8SToby Isaac   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject) fem);CHKERRQ(ierr);
26320cf1dd8SToby Isaac   ierr = PetscOptionsEnd();CHKERRQ(ierr);
26420cf1dd8SToby Isaac   ierr = PetscFEViewFromOptions(fem, NULL, "-petscfe_view");CHKERRQ(ierr);
26520cf1dd8SToby Isaac   PetscFunctionReturn(0);
26620cf1dd8SToby Isaac }
26720cf1dd8SToby Isaac 
26820cf1dd8SToby Isaac /*@C
26920cf1dd8SToby Isaac   PetscFESetUp - Construct data structures for the PetscFE
27020cf1dd8SToby Isaac 
271d083f849SBarry Smith   Collective on fem
27220cf1dd8SToby Isaac 
27320cf1dd8SToby Isaac   Input Parameter:
27420cf1dd8SToby Isaac . fem - the PetscFE object to setup
27520cf1dd8SToby Isaac 
2762b99622eSMatthew G. Knepley   Level: intermediate
27720cf1dd8SToby Isaac 
27820cf1dd8SToby Isaac .seealso PetscFEView(), PetscFEDestroy()
27920cf1dd8SToby Isaac @*/
28020cf1dd8SToby Isaac PetscErrorCode PetscFESetUp(PetscFE fem)
28120cf1dd8SToby Isaac {
28220cf1dd8SToby Isaac   PetscErrorCode ierr;
28320cf1dd8SToby Isaac 
28420cf1dd8SToby Isaac   PetscFunctionBegin;
28520cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
28620cf1dd8SToby Isaac   if (fem->setupcalled) PetscFunctionReturn(0);
28720cf1dd8SToby Isaac   fem->setupcalled = PETSC_TRUE;
28820cf1dd8SToby Isaac   if (fem->ops->setup) {ierr = (*fem->ops->setup)(fem);CHKERRQ(ierr);}
28920cf1dd8SToby Isaac   PetscFunctionReturn(0);
29020cf1dd8SToby Isaac }
29120cf1dd8SToby Isaac 
29220cf1dd8SToby Isaac /*@
29320cf1dd8SToby Isaac   PetscFEDestroy - Destroys a PetscFE object
29420cf1dd8SToby Isaac 
295d083f849SBarry Smith   Collective on fem
29620cf1dd8SToby Isaac 
29720cf1dd8SToby Isaac   Input Parameter:
29820cf1dd8SToby Isaac . fem - the PetscFE object to destroy
29920cf1dd8SToby Isaac 
3002b99622eSMatthew G. Knepley   Level: beginner
30120cf1dd8SToby Isaac 
30220cf1dd8SToby Isaac .seealso PetscFEView()
30320cf1dd8SToby Isaac @*/
30420cf1dd8SToby Isaac PetscErrorCode PetscFEDestroy(PetscFE *fem)
30520cf1dd8SToby Isaac {
30620cf1dd8SToby Isaac   PetscErrorCode ierr;
30720cf1dd8SToby Isaac 
30820cf1dd8SToby Isaac   PetscFunctionBegin;
30920cf1dd8SToby Isaac   if (!*fem) PetscFunctionReturn(0);
31020cf1dd8SToby Isaac   PetscValidHeaderSpecific((*fem), PETSCFE_CLASSID, 1);
31120cf1dd8SToby Isaac 
31220cf1dd8SToby Isaac   if (--((PetscObject)(*fem))->refct > 0) {*fem = 0; PetscFunctionReturn(0);}
31320cf1dd8SToby Isaac   ((PetscObject) (*fem))->refct = 0;
31420cf1dd8SToby Isaac 
31520cf1dd8SToby Isaac   if ((*fem)->subspaces) {
31620cf1dd8SToby Isaac     PetscInt dim, d;
31720cf1dd8SToby Isaac 
31820cf1dd8SToby Isaac     ierr = PetscDualSpaceGetDimension((*fem)->dualSpace, &dim);CHKERRQ(ierr);
31920cf1dd8SToby Isaac     for (d = 0; d < dim; ++d) {ierr = PetscFEDestroy(&(*fem)->subspaces[d]);CHKERRQ(ierr);}
32020cf1dd8SToby Isaac   }
32120cf1dd8SToby Isaac   ierr = PetscFree((*fem)->subspaces);CHKERRQ(ierr);
32220cf1dd8SToby Isaac   ierr = PetscFree((*fem)->invV);CHKERRQ(ierr);
32320cf1dd8SToby Isaac   ierr = PetscFERestoreTabulation((*fem), 0, NULL, &(*fem)->B, &(*fem)->D, NULL /*&(*fem)->H*/);CHKERRQ(ierr);
32420cf1dd8SToby Isaac   ierr = PetscFERestoreTabulation((*fem), 0, NULL, &(*fem)->Bf, &(*fem)->Df, NULL /*&(*fem)->Hf*/);CHKERRQ(ierr);
32520cf1dd8SToby Isaac   ierr = PetscFERestoreTabulation((*fem), 0, NULL, &(*fem)->F, NULL, NULL);CHKERRQ(ierr);
32620cf1dd8SToby Isaac   ierr = PetscSpaceDestroy(&(*fem)->basisSpace);CHKERRQ(ierr);
32720cf1dd8SToby Isaac   ierr = PetscDualSpaceDestroy(&(*fem)->dualSpace);CHKERRQ(ierr);
32820cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&(*fem)->quadrature);CHKERRQ(ierr);
32920cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&(*fem)->faceQuadrature);CHKERRQ(ierr);
33020cf1dd8SToby Isaac 
33120cf1dd8SToby Isaac   if ((*fem)->ops->destroy) {ierr = (*(*fem)->ops->destroy)(*fem);CHKERRQ(ierr);}
33220cf1dd8SToby Isaac   ierr = PetscHeaderDestroy(fem);CHKERRQ(ierr);
33320cf1dd8SToby Isaac   PetscFunctionReturn(0);
33420cf1dd8SToby Isaac }
33520cf1dd8SToby Isaac 
33620cf1dd8SToby Isaac /*@
33720cf1dd8SToby Isaac   PetscFECreate - Creates an empty PetscFE object. The type can then be set with PetscFESetType().
33820cf1dd8SToby Isaac 
339d083f849SBarry Smith   Collective
34020cf1dd8SToby Isaac 
34120cf1dd8SToby Isaac   Input Parameter:
34220cf1dd8SToby Isaac . comm - The communicator for the PetscFE object
34320cf1dd8SToby Isaac 
34420cf1dd8SToby Isaac   Output Parameter:
34520cf1dd8SToby Isaac . fem - The PetscFE object
34620cf1dd8SToby Isaac 
34720cf1dd8SToby Isaac   Level: beginner
34820cf1dd8SToby Isaac 
34920cf1dd8SToby Isaac .seealso: PetscFESetType(), PETSCFEGALERKIN
35020cf1dd8SToby Isaac @*/
35120cf1dd8SToby Isaac PetscErrorCode PetscFECreate(MPI_Comm comm, PetscFE *fem)
35220cf1dd8SToby Isaac {
35320cf1dd8SToby Isaac   PetscFE        f;
35420cf1dd8SToby Isaac   PetscErrorCode ierr;
35520cf1dd8SToby Isaac 
35620cf1dd8SToby Isaac   PetscFunctionBegin;
35720cf1dd8SToby Isaac   PetscValidPointer(fem, 2);
35820cf1dd8SToby Isaac   ierr = PetscCitationsRegister(FECitation,&FEcite);CHKERRQ(ierr);
35920cf1dd8SToby Isaac   *fem = NULL;
36020cf1dd8SToby Isaac   ierr = PetscFEInitializePackage();CHKERRQ(ierr);
36120cf1dd8SToby Isaac 
36220cf1dd8SToby Isaac   ierr = PetscHeaderCreate(f, PETSCFE_CLASSID, "PetscFE", "Finite Element", "PetscFE", comm, PetscFEDestroy, PetscFEView);CHKERRQ(ierr);
36320cf1dd8SToby Isaac 
36420cf1dd8SToby Isaac   f->basisSpace    = NULL;
36520cf1dd8SToby Isaac   f->dualSpace     = NULL;
36620cf1dd8SToby Isaac   f->numComponents = 1;
36720cf1dd8SToby Isaac   f->subspaces     = NULL;
36820cf1dd8SToby Isaac   f->invV          = NULL;
36920cf1dd8SToby Isaac   f->B             = NULL;
37020cf1dd8SToby Isaac   f->D             = NULL;
37120cf1dd8SToby Isaac   f->H             = NULL;
37220cf1dd8SToby Isaac   f->Bf            = NULL;
37320cf1dd8SToby Isaac   f->Df            = NULL;
37420cf1dd8SToby Isaac   f->Hf            = NULL;
375580bdb30SBarry Smith   ierr = PetscArrayzero(&f->quadrature, 1);CHKERRQ(ierr);
376580bdb30SBarry Smith   ierr = PetscArrayzero(&f->faceQuadrature, 1);CHKERRQ(ierr);
37720cf1dd8SToby Isaac   f->blockSize     = 0;
37820cf1dd8SToby Isaac   f->numBlocks     = 1;
37920cf1dd8SToby Isaac   f->batchSize     = 0;
38020cf1dd8SToby Isaac   f->numBatches    = 1;
38120cf1dd8SToby Isaac 
38220cf1dd8SToby Isaac   *fem = f;
38320cf1dd8SToby Isaac   PetscFunctionReturn(0);
38420cf1dd8SToby Isaac }
38520cf1dd8SToby Isaac 
38620cf1dd8SToby Isaac /*@
38720cf1dd8SToby Isaac   PetscFEGetSpatialDimension - Returns the spatial dimension of the element
38820cf1dd8SToby Isaac 
38920cf1dd8SToby Isaac   Not collective
39020cf1dd8SToby Isaac 
39120cf1dd8SToby Isaac   Input Parameter:
39220cf1dd8SToby Isaac . fem - The PetscFE object
39320cf1dd8SToby Isaac 
39420cf1dd8SToby Isaac   Output Parameter:
39520cf1dd8SToby Isaac . dim - The spatial dimension
39620cf1dd8SToby Isaac 
39720cf1dd8SToby Isaac   Level: intermediate
39820cf1dd8SToby Isaac 
39920cf1dd8SToby Isaac .seealso: PetscFECreate()
40020cf1dd8SToby Isaac @*/
40120cf1dd8SToby Isaac PetscErrorCode PetscFEGetSpatialDimension(PetscFE fem, PetscInt *dim)
40220cf1dd8SToby Isaac {
40320cf1dd8SToby Isaac   DM             dm;
40420cf1dd8SToby Isaac   PetscErrorCode ierr;
40520cf1dd8SToby Isaac 
40620cf1dd8SToby Isaac   PetscFunctionBegin;
40720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
40820cf1dd8SToby Isaac   PetscValidPointer(dim, 2);
40920cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(fem->dualSpace, &dm);CHKERRQ(ierr);
41020cf1dd8SToby Isaac   ierr = DMGetDimension(dm, dim);CHKERRQ(ierr);
41120cf1dd8SToby Isaac   PetscFunctionReturn(0);
41220cf1dd8SToby Isaac }
41320cf1dd8SToby Isaac 
41420cf1dd8SToby Isaac /*@
41520cf1dd8SToby Isaac   PetscFESetNumComponents - Sets the number of components in the element
41620cf1dd8SToby Isaac 
41720cf1dd8SToby Isaac   Not collective
41820cf1dd8SToby Isaac 
41920cf1dd8SToby Isaac   Input Parameters:
42020cf1dd8SToby Isaac + fem - The PetscFE object
42120cf1dd8SToby Isaac - comp - The number of field components
42220cf1dd8SToby Isaac 
42320cf1dd8SToby Isaac   Level: intermediate
42420cf1dd8SToby Isaac 
42520cf1dd8SToby Isaac .seealso: PetscFECreate()
42620cf1dd8SToby Isaac @*/
42720cf1dd8SToby Isaac PetscErrorCode PetscFESetNumComponents(PetscFE fem, PetscInt comp)
42820cf1dd8SToby Isaac {
42920cf1dd8SToby Isaac   PetscFunctionBegin;
43020cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
43120cf1dd8SToby Isaac   fem->numComponents = comp;
43220cf1dd8SToby Isaac   PetscFunctionReturn(0);
43320cf1dd8SToby Isaac }
43420cf1dd8SToby Isaac 
43520cf1dd8SToby Isaac /*@
43620cf1dd8SToby Isaac   PetscFEGetNumComponents - Returns the number of components in the element
43720cf1dd8SToby Isaac 
43820cf1dd8SToby Isaac   Not collective
43920cf1dd8SToby Isaac 
44020cf1dd8SToby Isaac   Input Parameter:
44120cf1dd8SToby Isaac . fem - The PetscFE object
44220cf1dd8SToby Isaac 
44320cf1dd8SToby Isaac   Output Parameter:
44420cf1dd8SToby Isaac . comp - The number of field components
44520cf1dd8SToby Isaac 
44620cf1dd8SToby Isaac   Level: intermediate
44720cf1dd8SToby Isaac 
44820cf1dd8SToby Isaac .seealso: PetscFECreate()
44920cf1dd8SToby Isaac @*/
45020cf1dd8SToby Isaac PetscErrorCode PetscFEGetNumComponents(PetscFE fem, PetscInt *comp)
45120cf1dd8SToby Isaac {
45220cf1dd8SToby Isaac   PetscFunctionBegin;
45320cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
45420cf1dd8SToby Isaac   PetscValidPointer(comp, 2);
45520cf1dd8SToby Isaac   *comp = fem->numComponents;
45620cf1dd8SToby Isaac   PetscFunctionReturn(0);
45720cf1dd8SToby Isaac }
45820cf1dd8SToby Isaac 
45920cf1dd8SToby Isaac /*@
46020cf1dd8SToby Isaac   PetscFESetTileSizes - Sets the tile sizes for evaluation
46120cf1dd8SToby Isaac 
46220cf1dd8SToby Isaac   Not collective
46320cf1dd8SToby Isaac 
46420cf1dd8SToby Isaac   Input Parameters:
46520cf1dd8SToby Isaac + fem - The PetscFE object
46620cf1dd8SToby Isaac . blockSize - The number of elements in a block
46720cf1dd8SToby Isaac . numBlocks - The number of blocks in a batch
46820cf1dd8SToby Isaac . batchSize - The number of elements in a batch
46920cf1dd8SToby Isaac - numBatches - The number of batches in a chunk
47020cf1dd8SToby Isaac 
47120cf1dd8SToby Isaac   Level: intermediate
47220cf1dd8SToby Isaac 
47320cf1dd8SToby Isaac .seealso: PetscFECreate()
47420cf1dd8SToby Isaac @*/
47520cf1dd8SToby Isaac PetscErrorCode PetscFESetTileSizes(PetscFE fem, PetscInt blockSize, PetscInt numBlocks, PetscInt batchSize, PetscInt numBatches)
47620cf1dd8SToby Isaac {
47720cf1dd8SToby Isaac   PetscFunctionBegin;
47820cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
47920cf1dd8SToby Isaac   fem->blockSize  = blockSize;
48020cf1dd8SToby Isaac   fem->numBlocks  = numBlocks;
48120cf1dd8SToby Isaac   fem->batchSize  = batchSize;
48220cf1dd8SToby Isaac   fem->numBatches = numBatches;
48320cf1dd8SToby Isaac   PetscFunctionReturn(0);
48420cf1dd8SToby Isaac }
48520cf1dd8SToby Isaac 
48620cf1dd8SToby Isaac /*@
48720cf1dd8SToby Isaac   PetscFEGetTileSizes - Returns the tile sizes for evaluation
48820cf1dd8SToby Isaac 
48920cf1dd8SToby Isaac   Not collective
49020cf1dd8SToby Isaac 
49120cf1dd8SToby Isaac   Input Parameter:
49220cf1dd8SToby Isaac . fem - The PetscFE object
49320cf1dd8SToby Isaac 
49420cf1dd8SToby Isaac   Output Parameters:
49520cf1dd8SToby Isaac + blockSize - The number of elements in a block
49620cf1dd8SToby Isaac . numBlocks - The number of blocks in a batch
49720cf1dd8SToby Isaac . batchSize - The number of elements in a batch
49820cf1dd8SToby Isaac - numBatches - The number of batches in a chunk
49920cf1dd8SToby Isaac 
50020cf1dd8SToby Isaac   Level: intermediate
50120cf1dd8SToby Isaac 
50220cf1dd8SToby Isaac .seealso: PetscFECreate()
50320cf1dd8SToby Isaac @*/
50420cf1dd8SToby Isaac PetscErrorCode PetscFEGetTileSizes(PetscFE fem, PetscInt *blockSize, PetscInt *numBlocks, PetscInt *batchSize, PetscInt *numBatches)
50520cf1dd8SToby Isaac {
50620cf1dd8SToby Isaac   PetscFunctionBegin;
50720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
50820cf1dd8SToby Isaac   if (blockSize)  PetscValidPointer(blockSize,  2);
50920cf1dd8SToby Isaac   if (numBlocks)  PetscValidPointer(numBlocks,  3);
51020cf1dd8SToby Isaac   if (batchSize)  PetscValidPointer(batchSize,  4);
51120cf1dd8SToby Isaac   if (numBatches) PetscValidPointer(numBatches, 5);
51220cf1dd8SToby Isaac   if (blockSize)  *blockSize  = fem->blockSize;
51320cf1dd8SToby Isaac   if (numBlocks)  *numBlocks  = fem->numBlocks;
51420cf1dd8SToby Isaac   if (batchSize)  *batchSize  = fem->batchSize;
51520cf1dd8SToby Isaac   if (numBatches) *numBatches = fem->numBatches;
51620cf1dd8SToby Isaac   PetscFunctionReturn(0);
51720cf1dd8SToby Isaac }
51820cf1dd8SToby Isaac 
51920cf1dd8SToby Isaac /*@
52020cf1dd8SToby Isaac   PetscFEGetBasisSpace - Returns the PetscSpace used for approximation of the solution
52120cf1dd8SToby Isaac 
52220cf1dd8SToby Isaac   Not collective
52320cf1dd8SToby Isaac 
52420cf1dd8SToby Isaac   Input Parameter:
52520cf1dd8SToby Isaac . fem - The PetscFE object
52620cf1dd8SToby Isaac 
52720cf1dd8SToby Isaac   Output Parameter:
52820cf1dd8SToby Isaac . sp - The PetscSpace object
52920cf1dd8SToby Isaac 
53020cf1dd8SToby Isaac   Level: intermediate
53120cf1dd8SToby Isaac 
53220cf1dd8SToby Isaac .seealso: PetscFECreate()
53320cf1dd8SToby Isaac @*/
53420cf1dd8SToby Isaac PetscErrorCode PetscFEGetBasisSpace(PetscFE fem, PetscSpace *sp)
53520cf1dd8SToby Isaac {
53620cf1dd8SToby Isaac   PetscFunctionBegin;
53720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
53820cf1dd8SToby Isaac   PetscValidPointer(sp, 2);
53920cf1dd8SToby Isaac   *sp = fem->basisSpace;
54020cf1dd8SToby Isaac   PetscFunctionReturn(0);
54120cf1dd8SToby Isaac }
54220cf1dd8SToby Isaac 
54320cf1dd8SToby Isaac /*@
54420cf1dd8SToby Isaac   PetscFESetBasisSpace - Sets the PetscSpace used for approximation of the solution
54520cf1dd8SToby Isaac 
54620cf1dd8SToby Isaac   Not collective
54720cf1dd8SToby Isaac 
54820cf1dd8SToby Isaac   Input Parameters:
54920cf1dd8SToby Isaac + fem - The PetscFE object
55020cf1dd8SToby Isaac - sp - The PetscSpace object
55120cf1dd8SToby Isaac 
55220cf1dd8SToby Isaac   Level: intermediate
55320cf1dd8SToby Isaac 
55420cf1dd8SToby Isaac .seealso: PetscFECreate()
55520cf1dd8SToby Isaac @*/
55620cf1dd8SToby Isaac PetscErrorCode PetscFESetBasisSpace(PetscFE fem, PetscSpace sp)
55720cf1dd8SToby Isaac {
55820cf1dd8SToby Isaac   PetscErrorCode ierr;
55920cf1dd8SToby Isaac 
56020cf1dd8SToby Isaac   PetscFunctionBegin;
56120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
56220cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCSPACE_CLASSID, 2);
56320cf1dd8SToby Isaac   ierr = PetscSpaceDestroy(&fem->basisSpace);CHKERRQ(ierr);
56420cf1dd8SToby Isaac   fem->basisSpace = sp;
56520cf1dd8SToby Isaac   ierr = PetscObjectReference((PetscObject) fem->basisSpace);CHKERRQ(ierr);
56620cf1dd8SToby Isaac   PetscFunctionReturn(0);
56720cf1dd8SToby Isaac }
56820cf1dd8SToby Isaac 
56920cf1dd8SToby Isaac /*@
57020cf1dd8SToby Isaac   PetscFEGetDualSpace - Returns the PetscDualSpace used to define the inner product
57120cf1dd8SToby Isaac 
57220cf1dd8SToby Isaac   Not collective
57320cf1dd8SToby Isaac 
57420cf1dd8SToby Isaac   Input Parameter:
57520cf1dd8SToby Isaac . fem - The PetscFE object
57620cf1dd8SToby Isaac 
57720cf1dd8SToby Isaac   Output Parameter:
57820cf1dd8SToby Isaac . sp - The PetscDualSpace object
57920cf1dd8SToby Isaac 
58020cf1dd8SToby Isaac   Level: intermediate
58120cf1dd8SToby Isaac 
58220cf1dd8SToby Isaac .seealso: PetscFECreate()
58320cf1dd8SToby Isaac @*/
58420cf1dd8SToby Isaac PetscErrorCode PetscFEGetDualSpace(PetscFE fem, PetscDualSpace *sp)
58520cf1dd8SToby Isaac {
58620cf1dd8SToby Isaac   PetscFunctionBegin;
58720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
58820cf1dd8SToby Isaac   PetscValidPointer(sp, 2);
58920cf1dd8SToby Isaac   *sp = fem->dualSpace;
59020cf1dd8SToby Isaac   PetscFunctionReturn(0);
59120cf1dd8SToby Isaac }
59220cf1dd8SToby Isaac 
59320cf1dd8SToby Isaac /*@
59420cf1dd8SToby Isaac   PetscFESetDualSpace - Sets the PetscDualSpace used to define the inner product
59520cf1dd8SToby Isaac 
59620cf1dd8SToby Isaac   Not collective
59720cf1dd8SToby Isaac 
59820cf1dd8SToby Isaac   Input Parameters:
59920cf1dd8SToby Isaac + fem - The PetscFE object
60020cf1dd8SToby Isaac - sp - The PetscDualSpace object
60120cf1dd8SToby Isaac 
60220cf1dd8SToby Isaac   Level: intermediate
60320cf1dd8SToby Isaac 
60420cf1dd8SToby Isaac .seealso: PetscFECreate()
60520cf1dd8SToby Isaac @*/
60620cf1dd8SToby Isaac PetscErrorCode PetscFESetDualSpace(PetscFE fem, PetscDualSpace sp)
60720cf1dd8SToby Isaac {
60820cf1dd8SToby Isaac   PetscErrorCode ierr;
60920cf1dd8SToby Isaac 
61020cf1dd8SToby Isaac   PetscFunctionBegin;
61120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
61220cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 2);
61320cf1dd8SToby Isaac   ierr = PetscDualSpaceDestroy(&fem->dualSpace);CHKERRQ(ierr);
61420cf1dd8SToby Isaac   fem->dualSpace = sp;
61520cf1dd8SToby Isaac   ierr = PetscObjectReference((PetscObject) fem->dualSpace);CHKERRQ(ierr);
61620cf1dd8SToby Isaac   PetscFunctionReturn(0);
61720cf1dd8SToby Isaac }
61820cf1dd8SToby Isaac 
61920cf1dd8SToby Isaac /*@
62020cf1dd8SToby Isaac   PetscFEGetQuadrature - Returns the PetscQuadrature used to calculate inner products
62120cf1dd8SToby Isaac 
62220cf1dd8SToby Isaac   Not collective
62320cf1dd8SToby Isaac 
62420cf1dd8SToby Isaac   Input Parameter:
62520cf1dd8SToby Isaac . fem - The PetscFE object
62620cf1dd8SToby Isaac 
62720cf1dd8SToby Isaac   Output Parameter:
62820cf1dd8SToby Isaac . q - The PetscQuadrature object
62920cf1dd8SToby Isaac 
63020cf1dd8SToby Isaac   Level: intermediate
63120cf1dd8SToby Isaac 
63220cf1dd8SToby Isaac .seealso: PetscFECreate()
63320cf1dd8SToby Isaac @*/
63420cf1dd8SToby Isaac PetscErrorCode PetscFEGetQuadrature(PetscFE fem, PetscQuadrature *q)
63520cf1dd8SToby Isaac {
63620cf1dd8SToby Isaac   PetscFunctionBegin;
63720cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
63820cf1dd8SToby Isaac   PetscValidPointer(q, 2);
63920cf1dd8SToby Isaac   *q = fem->quadrature;
64020cf1dd8SToby Isaac   PetscFunctionReturn(0);
64120cf1dd8SToby Isaac }
64220cf1dd8SToby Isaac 
64320cf1dd8SToby Isaac /*@
64420cf1dd8SToby Isaac   PetscFESetQuadrature - Sets the PetscQuadrature used to calculate inner products
64520cf1dd8SToby Isaac 
64620cf1dd8SToby Isaac   Not collective
64720cf1dd8SToby Isaac 
64820cf1dd8SToby Isaac   Input Parameters:
64920cf1dd8SToby Isaac + fem - The PetscFE object
65020cf1dd8SToby Isaac - q - The PetscQuadrature object
65120cf1dd8SToby Isaac 
65220cf1dd8SToby Isaac   Level: intermediate
65320cf1dd8SToby Isaac 
65420cf1dd8SToby Isaac .seealso: PetscFECreate()
65520cf1dd8SToby Isaac @*/
65620cf1dd8SToby Isaac PetscErrorCode PetscFESetQuadrature(PetscFE fem, PetscQuadrature q)
65720cf1dd8SToby Isaac {
65820cf1dd8SToby Isaac   PetscInt       Nc, qNc;
65920cf1dd8SToby Isaac   PetscErrorCode ierr;
66020cf1dd8SToby Isaac 
66120cf1dd8SToby Isaac   PetscFunctionBegin;
66220cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
66320cf1dd8SToby Isaac   ierr = PetscFEGetNumComponents(fem, &Nc);CHKERRQ(ierr);
66420cf1dd8SToby Isaac   ierr = PetscQuadratureGetNumComponents(q, &qNc);CHKERRQ(ierr);
66520cf1dd8SToby Isaac   if ((qNc != 1) && (Nc != qNc)) SETERRQ2(PetscObjectComm((PetscObject) fem), PETSC_ERR_ARG_SIZ, "FE components %D != Quadrature components %D and non-scalar quadrature", Nc, qNc);
66620cf1dd8SToby Isaac   ierr = PetscFERestoreTabulation(fem, 0, NULL, &fem->B, &fem->D, NULL /*&(*fem)->H*/);CHKERRQ(ierr);
66720cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&fem->quadrature);CHKERRQ(ierr);
66820cf1dd8SToby Isaac   fem->quadrature = q;
66920cf1dd8SToby Isaac   ierr = PetscObjectReference((PetscObject) q);CHKERRQ(ierr);
67020cf1dd8SToby Isaac   PetscFunctionReturn(0);
67120cf1dd8SToby Isaac }
67220cf1dd8SToby Isaac 
67320cf1dd8SToby Isaac /*@
67420cf1dd8SToby Isaac   PetscFEGetFaceQuadrature - Returns the PetscQuadrature used to calculate inner products on faces
67520cf1dd8SToby Isaac 
67620cf1dd8SToby Isaac   Not collective
67720cf1dd8SToby Isaac 
67820cf1dd8SToby Isaac   Input Parameter:
67920cf1dd8SToby Isaac . fem - The PetscFE object
68020cf1dd8SToby Isaac 
68120cf1dd8SToby Isaac   Output Parameter:
68220cf1dd8SToby Isaac . q - The PetscQuadrature object
68320cf1dd8SToby Isaac 
68420cf1dd8SToby Isaac   Level: intermediate
68520cf1dd8SToby Isaac 
68620cf1dd8SToby Isaac .seealso: PetscFECreate()
68720cf1dd8SToby Isaac @*/
68820cf1dd8SToby Isaac PetscErrorCode PetscFEGetFaceQuadrature(PetscFE fem, PetscQuadrature *q)
68920cf1dd8SToby Isaac {
69020cf1dd8SToby Isaac   PetscFunctionBegin;
69120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
69220cf1dd8SToby Isaac   PetscValidPointer(q, 2);
69320cf1dd8SToby Isaac   *q = fem->faceQuadrature;
69420cf1dd8SToby Isaac   PetscFunctionReturn(0);
69520cf1dd8SToby Isaac }
69620cf1dd8SToby Isaac 
69720cf1dd8SToby Isaac /*@
69820cf1dd8SToby Isaac   PetscFESetFaceQuadrature - Sets the PetscQuadrature used to calculate inner products on faces
69920cf1dd8SToby Isaac 
70020cf1dd8SToby Isaac   Not collective
70120cf1dd8SToby Isaac 
70220cf1dd8SToby Isaac   Input Parameters:
70320cf1dd8SToby Isaac + fem - The PetscFE object
70420cf1dd8SToby Isaac - q - The PetscQuadrature object
70520cf1dd8SToby Isaac 
70620cf1dd8SToby Isaac   Level: intermediate
70720cf1dd8SToby Isaac 
70820cf1dd8SToby Isaac .seealso: PetscFECreate()
70920cf1dd8SToby Isaac @*/
71020cf1dd8SToby Isaac PetscErrorCode PetscFESetFaceQuadrature(PetscFE fem, PetscQuadrature q)
71120cf1dd8SToby Isaac {
71220cf1dd8SToby Isaac   PetscErrorCode ierr;
71320cf1dd8SToby Isaac 
71420cf1dd8SToby Isaac   PetscFunctionBegin;
71520cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
71620cf1dd8SToby Isaac   ierr = PetscFERestoreTabulation(fem, 0, NULL, &fem->Bf, &fem->Df, NULL /*&(*fem)->Hf*/);CHKERRQ(ierr);
71720cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&fem->faceQuadrature);CHKERRQ(ierr);
71820cf1dd8SToby Isaac   fem->faceQuadrature = q;
71920cf1dd8SToby Isaac   ierr = PetscObjectReference((PetscObject) q);CHKERRQ(ierr);
72020cf1dd8SToby Isaac   PetscFunctionReturn(0);
72120cf1dd8SToby Isaac }
72220cf1dd8SToby Isaac 
7235dc5c000SMatthew G. Knepley /*@
7245dc5c000SMatthew G. Knepley   PetscFECopyQuadrature - Copy both volumetric and surface quadrature
7255dc5c000SMatthew G. Knepley 
7265dc5c000SMatthew G. Knepley   Not collective
7275dc5c000SMatthew G. Knepley 
7285dc5c000SMatthew G. Knepley   Input Parameters:
7295dc5c000SMatthew G. Knepley + sfe - The PetscFE source for the quadratures
7305dc5c000SMatthew G. Knepley - tfe - The PetscFE target for the quadratures
7315dc5c000SMatthew G. Knepley 
7325dc5c000SMatthew G. Knepley   Level: intermediate
7335dc5c000SMatthew G. Knepley 
7345dc5c000SMatthew G. Knepley .seealso: PetscFECreate(), PetscFESetQuadrature(), PetscFESetFaceQuadrature()
7355dc5c000SMatthew G. Knepley @*/
7365dc5c000SMatthew G. Knepley PetscErrorCode PetscFECopyQuadrature(PetscFE sfe, PetscFE tfe)
7375dc5c000SMatthew G. Knepley {
7385dc5c000SMatthew G. Knepley   PetscQuadrature q;
7395dc5c000SMatthew G. Knepley   PetscErrorCode  ierr;
7405dc5c000SMatthew G. Knepley 
7415dc5c000SMatthew G. Knepley   PetscFunctionBegin;
7425dc5c000SMatthew G. Knepley   PetscValidHeaderSpecific(sfe, PETSCFE_CLASSID, 1);
7435dc5c000SMatthew G. Knepley   PetscValidHeaderSpecific(tfe, PETSCFE_CLASSID, 2);
7445dc5c000SMatthew G. Knepley   ierr = PetscFEGetQuadrature(sfe, &q);CHKERRQ(ierr);
7455dc5c000SMatthew G. Knepley   ierr = PetscFESetQuadrature(tfe,  q);CHKERRQ(ierr);
7465dc5c000SMatthew G. Knepley   ierr = PetscFEGetFaceQuadrature(sfe, &q);CHKERRQ(ierr);
7475dc5c000SMatthew G. Knepley   ierr = PetscFESetFaceQuadrature(tfe,  q);CHKERRQ(ierr);
7485dc5c000SMatthew G. Knepley   PetscFunctionReturn(0);
7495dc5c000SMatthew G. Knepley }
7505dc5c000SMatthew G. Knepley 
75120cf1dd8SToby Isaac /*@C
75220cf1dd8SToby Isaac   PetscFEGetNumDof - Returns the number of dofs (dual basis vectors) associated to mesh points on the reference cell of a given dimension
75320cf1dd8SToby Isaac 
75420cf1dd8SToby Isaac   Not collective
75520cf1dd8SToby Isaac 
75620cf1dd8SToby Isaac   Input Parameter:
75720cf1dd8SToby Isaac . fem - The PetscFE object
75820cf1dd8SToby Isaac 
75920cf1dd8SToby Isaac   Output Parameter:
76020cf1dd8SToby Isaac . numDof - Array with the number of dofs per dimension
76120cf1dd8SToby Isaac 
76220cf1dd8SToby Isaac   Level: intermediate
76320cf1dd8SToby Isaac 
76420cf1dd8SToby Isaac .seealso: PetscFECreate()
76520cf1dd8SToby Isaac @*/
76620cf1dd8SToby Isaac PetscErrorCode PetscFEGetNumDof(PetscFE fem, const PetscInt **numDof)
76720cf1dd8SToby Isaac {
76820cf1dd8SToby Isaac   PetscErrorCode ierr;
76920cf1dd8SToby Isaac 
77020cf1dd8SToby Isaac   PetscFunctionBegin;
77120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
77220cf1dd8SToby Isaac   PetscValidPointer(numDof, 2);
77320cf1dd8SToby Isaac   ierr = PetscDualSpaceGetNumDof(fem->dualSpace, numDof);CHKERRQ(ierr);
77420cf1dd8SToby Isaac   PetscFunctionReturn(0);
77520cf1dd8SToby Isaac }
77620cf1dd8SToby Isaac 
77720cf1dd8SToby Isaac /*@C
77820cf1dd8SToby Isaac   PetscFEGetDefaultTabulation - Returns the tabulation of the basis functions at the quadrature points
77920cf1dd8SToby Isaac 
78020cf1dd8SToby Isaac   Not collective
78120cf1dd8SToby Isaac 
78220cf1dd8SToby Isaac   Input Parameter:
78320cf1dd8SToby Isaac . fem - The PetscFE object
78420cf1dd8SToby Isaac 
78520cf1dd8SToby Isaac   Output Parameters:
78620cf1dd8SToby Isaac + B - The basis function values at quadrature points
78720cf1dd8SToby Isaac . D - The basis function derivatives at quadrature points
78820cf1dd8SToby Isaac - H - The basis function second derivatives at quadrature points
78920cf1dd8SToby Isaac 
79020cf1dd8SToby Isaac   Note:
79120cf1dd8SToby Isaac $ B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c
79220cf1dd8SToby Isaac $ D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d
79320cf1dd8SToby Isaac $ H[(((p*pdim + i)*Nc + c)*dim + d)*dim + e] is the value at point p for basis function i, component c, in directions d and e
79420cf1dd8SToby Isaac 
79520cf1dd8SToby Isaac   Level: intermediate
79620cf1dd8SToby Isaac 
79720cf1dd8SToby Isaac .seealso: PetscFEGetTabulation(), PetscFERestoreTabulation()
79820cf1dd8SToby Isaac @*/
79920cf1dd8SToby Isaac PetscErrorCode PetscFEGetDefaultTabulation(PetscFE fem, PetscReal **B, PetscReal **D, PetscReal **H)
80020cf1dd8SToby Isaac {
80120cf1dd8SToby Isaac   PetscInt         npoints;
80220cf1dd8SToby Isaac   const PetscReal *points;
80320cf1dd8SToby Isaac   PetscErrorCode   ierr;
80420cf1dd8SToby Isaac 
80520cf1dd8SToby Isaac   PetscFunctionBegin;
80620cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
80720cf1dd8SToby Isaac   if (B) PetscValidPointer(B, 2);
80820cf1dd8SToby Isaac   if (D) PetscValidPointer(D, 3);
80920cf1dd8SToby Isaac   if (H) PetscValidPointer(H, 4);
81020cf1dd8SToby Isaac   ierr = PetscQuadratureGetData(fem->quadrature, NULL, NULL, &npoints, &points, NULL);CHKERRQ(ierr);
81120cf1dd8SToby Isaac   if (!fem->B) {ierr = PetscFEGetTabulation(fem, npoints, points, &fem->B, &fem->D, NULL/*&fem->H*/);CHKERRQ(ierr);}
81220cf1dd8SToby Isaac   if (B) *B = fem->B;
81320cf1dd8SToby Isaac   if (D) *D = fem->D;
81420cf1dd8SToby Isaac   if (H) *H = fem->H;
81520cf1dd8SToby Isaac   PetscFunctionReturn(0);
81620cf1dd8SToby Isaac }
81720cf1dd8SToby Isaac 
8182b99622eSMatthew G. Knepley /*@C
8192b99622eSMatthew G. Knepley   PetscFEGetFaceTabulation - Returns the tabulation of the basis functions at the face quadrature points
8202b99622eSMatthew G. Knepley 
8212b99622eSMatthew G. Knepley   Not collective
8222b99622eSMatthew G. Knepley 
8232b99622eSMatthew G. Knepley   Input Parameter:
8242b99622eSMatthew G. Knepley . fem - The PetscFE object
8252b99622eSMatthew G. Knepley 
8262b99622eSMatthew G. Knepley   Output Parameters:
8272b99622eSMatthew G. Knepley + B - The basis function values at face quadrature points
8282b99622eSMatthew G. Knepley . D - The basis function derivatives at face quadrature points
8292b99622eSMatthew G. Knepley - H - The basis function second derivatives at face quadrature points
8302b99622eSMatthew G. Knepley 
8312b99622eSMatthew G. Knepley   Note:
8322b99622eSMatthew G. Knepley $ Bf[((f*Nq + q)*pdim + i)*Nc + c] is the value at point f,q for basis function i and component c
8332b99622eSMatthew G. Knepley $ Df[(((f*Nq + q)*pdim + i)*Nc + c)*dim + d] is the derivative value at point f,q for basis function i, component c, in direction d
8342b99622eSMatthew G. Knepley $ Hf[((((f*Nq + q)*pdim + i)*Nc + c)*dim + d)*dim + e] is the value at point f,q for basis function i, component c, in directions d and e
8352b99622eSMatthew G. Knepley 
8362b99622eSMatthew G. Knepley   Level: intermediate
8372b99622eSMatthew G. Knepley 
8382b99622eSMatthew G. Knepley .seealso: PetscFEGetDefaultTabulation(), PetscFEGetTabulation(), PetscFERestoreTabulation()
8392b99622eSMatthew G. Knepley @*/
84020cf1dd8SToby Isaac PetscErrorCode PetscFEGetFaceTabulation(PetscFE fem, PetscReal **Bf, PetscReal **Df, PetscReal **Hf)
84120cf1dd8SToby Isaac {
84220cf1dd8SToby Isaac   PetscErrorCode   ierr;
84320cf1dd8SToby Isaac 
84420cf1dd8SToby Isaac   PetscFunctionBegin;
84520cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
84620cf1dd8SToby Isaac   if (Bf) PetscValidPointer(Bf, 2);
84720cf1dd8SToby Isaac   if (Df) PetscValidPointer(Df, 3);
84820cf1dd8SToby Isaac   if (Hf) PetscValidPointer(Hf, 4);
84920cf1dd8SToby Isaac   if (!fem->Bf) {
85020cf1dd8SToby Isaac     const PetscReal  xi0[3] = {-1., -1., -1.};
85120cf1dd8SToby Isaac     PetscReal        v0[3], J[9], detJ;
85220cf1dd8SToby Isaac     PetscQuadrature  fq;
85320cf1dd8SToby Isaac     PetscDualSpace   sp;
85420cf1dd8SToby Isaac     DM               dm;
85520cf1dd8SToby Isaac     const PetscInt  *faces;
85620cf1dd8SToby Isaac     PetscInt         dim, numFaces, f, npoints, q;
85720cf1dd8SToby Isaac     const PetscReal *points;
85820cf1dd8SToby Isaac     PetscReal       *facePoints;
85920cf1dd8SToby Isaac 
86020cf1dd8SToby Isaac     ierr = PetscFEGetDualSpace(fem, &sp);CHKERRQ(ierr);
86120cf1dd8SToby Isaac     ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
86220cf1dd8SToby Isaac     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
86320cf1dd8SToby Isaac     ierr = DMPlexGetConeSize(dm, 0, &numFaces);CHKERRQ(ierr);
86420cf1dd8SToby Isaac     ierr = DMPlexGetCone(dm, 0, &faces);CHKERRQ(ierr);
86520cf1dd8SToby Isaac     ierr = PetscFEGetFaceQuadrature(fem, &fq);CHKERRQ(ierr);
86620cf1dd8SToby Isaac     if (fq) {
86720cf1dd8SToby Isaac       ierr = PetscQuadratureGetData(fq, NULL, NULL, &npoints, &points, NULL);CHKERRQ(ierr);
86820cf1dd8SToby Isaac       ierr = PetscMalloc1(numFaces*npoints*dim, &facePoints);CHKERRQ(ierr);
86920cf1dd8SToby Isaac       for (f = 0; f < numFaces; ++f) {
87020cf1dd8SToby Isaac         ierr = DMPlexComputeCellGeometryFEM(dm, faces[f], NULL, v0, J, NULL, &detJ);CHKERRQ(ierr);
87120cf1dd8SToby Isaac         for (q = 0; q < npoints; ++q) CoordinatesRefToReal(dim, dim-1, xi0, v0, J, &points[q*(dim-1)], &facePoints[(f*npoints+q)*dim]);
87220cf1dd8SToby Isaac       }
87320cf1dd8SToby Isaac       ierr = PetscFEGetTabulation(fem, numFaces*npoints, facePoints, &fem->Bf, &fem->Df, NULL/*&fem->Hf*/);CHKERRQ(ierr);
87420cf1dd8SToby Isaac       ierr = PetscFree(facePoints);CHKERRQ(ierr);
87520cf1dd8SToby Isaac     }
87620cf1dd8SToby Isaac   }
87720cf1dd8SToby Isaac   if (Bf) *Bf = fem->Bf;
87820cf1dd8SToby Isaac   if (Df) *Df = fem->Df;
87920cf1dd8SToby Isaac   if (Hf) *Hf = fem->Hf;
88020cf1dd8SToby Isaac   PetscFunctionReturn(0);
88120cf1dd8SToby Isaac }
88220cf1dd8SToby Isaac 
8832b99622eSMatthew G. Knepley /*@C
8842b99622eSMatthew G. Knepley   PetscFEGetFaceTabulation - Returns the tabulation of the basis functions at the face centroid points
8852b99622eSMatthew G. Knepley 
8862b99622eSMatthew G. Knepley   Not collective
8872b99622eSMatthew G. Knepley 
8882b99622eSMatthew G. Knepley   Input Parameter:
8892b99622eSMatthew G. Knepley . fem - The PetscFE object
8902b99622eSMatthew G. Knepley 
8912b99622eSMatthew G. Knepley   Output Parameters:
8922b99622eSMatthew G. Knepley + B - The basis function values at face centroid points
8932b99622eSMatthew G. Knepley . D - The basis function derivatives at face centroid points
8942b99622eSMatthew G. Knepley - H - The basis function second derivatives at face centroid points
8952b99622eSMatthew G. Knepley 
8962b99622eSMatthew G. Knepley   Note:
8972b99622eSMatthew G. Knepley $ Bf[(f*pdim + i)*Nc + c] is the value at point f for basis function i and component c
8982b99622eSMatthew G. Knepley $ Df[((f*pdim + i)*Nc + c)*dim + d] is the derivative value at point f for basis function i, component c, in direction d
8992b99622eSMatthew G. Knepley $ Hf[(((f*pdim + i)*Nc + c)*dim + d)*dim + e] is the value at point f for basis function i, component c, in directions d and e
9002b99622eSMatthew G. Knepley 
9012b99622eSMatthew G. Knepley   Level: intermediate
9022b99622eSMatthew G. Knepley 
9032b99622eSMatthew G. Knepley .seealso: PetscFEGetFaceTabulation(), PetscFEGetDefaultTabulation(), PetscFEGetTabulation(), PetscFERestoreTabulation()
9042b99622eSMatthew G. Knepley @*/
90520cf1dd8SToby Isaac PetscErrorCode PetscFEGetFaceCentroidTabulation(PetscFE fem, PetscReal **F)
90620cf1dd8SToby Isaac {
90720cf1dd8SToby Isaac   PetscErrorCode   ierr;
90820cf1dd8SToby Isaac 
90920cf1dd8SToby Isaac   PetscFunctionBegin;
91020cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
91120cf1dd8SToby Isaac   PetscValidPointer(F, 2);
91220cf1dd8SToby Isaac   if (!fem->F) {
91320cf1dd8SToby Isaac     PetscDualSpace  sp;
91420cf1dd8SToby Isaac     DM              dm;
91520cf1dd8SToby Isaac     const PetscInt *cone;
91620cf1dd8SToby Isaac     PetscReal      *centroids;
91720cf1dd8SToby Isaac     PetscInt        dim, numFaces, f;
91820cf1dd8SToby Isaac 
91920cf1dd8SToby Isaac     ierr = PetscFEGetDualSpace(fem, &sp);CHKERRQ(ierr);
92020cf1dd8SToby Isaac     ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
92120cf1dd8SToby Isaac     ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
92220cf1dd8SToby Isaac     ierr = DMPlexGetConeSize(dm, 0, &numFaces);CHKERRQ(ierr);
92320cf1dd8SToby Isaac     ierr = DMPlexGetCone(dm, 0, &cone);CHKERRQ(ierr);
92420cf1dd8SToby Isaac     ierr = PetscMalloc1(numFaces*dim, &centroids);CHKERRQ(ierr);
92520cf1dd8SToby Isaac     for (f = 0; f < numFaces; ++f) {ierr = DMPlexComputeCellGeometryFVM(dm, cone[f], NULL, &centroids[f*dim], NULL);CHKERRQ(ierr);}
92620cf1dd8SToby Isaac     ierr = PetscFEGetTabulation(fem, numFaces, centroids, &fem->F, NULL, NULL);CHKERRQ(ierr);
92720cf1dd8SToby Isaac     ierr = PetscFree(centroids);CHKERRQ(ierr);
92820cf1dd8SToby Isaac   }
92920cf1dd8SToby Isaac   *F = fem->F;
93020cf1dd8SToby Isaac   PetscFunctionReturn(0);
93120cf1dd8SToby Isaac }
93220cf1dd8SToby Isaac 
93320cf1dd8SToby Isaac /*@C
93420cf1dd8SToby Isaac   PetscFEGetTabulation - Tabulates the basis functions, and perhaps derivatives, at the points provided.
93520cf1dd8SToby Isaac 
93620cf1dd8SToby Isaac   Not collective
93720cf1dd8SToby Isaac 
93820cf1dd8SToby Isaac   Input Parameters:
93920cf1dd8SToby Isaac + fem     - The PetscFE object
94020cf1dd8SToby Isaac . npoints - The number of tabulation points
94120cf1dd8SToby Isaac - points  - The tabulation point coordinates
94220cf1dd8SToby Isaac 
94320cf1dd8SToby Isaac   Output Parameters:
94420cf1dd8SToby Isaac + B - The basis function values at tabulation points
94520cf1dd8SToby Isaac . D - The basis function derivatives at tabulation points
94620cf1dd8SToby Isaac - H - The basis function second derivatives at tabulation points
94720cf1dd8SToby Isaac 
94820cf1dd8SToby Isaac   Note:
94920cf1dd8SToby Isaac $ B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c
95020cf1dd8SToby Isaac $ D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d
95120cf1dd8SToby Isaac $ H[(((p*pdim + i)*Nc + c)*dim + d)*dim + e] is the value at point p for basis function i, component c, in directions d and e
95220cf1dd8SToby Isaac 
95320cf1dd8SToby Isaac   Level: intermediate
95420cf1dd8SToby Isaac 
95520cf1dd8SToby Isaac .seealso: PetscFERestoreTabulation(), PetscFEGetDefaultTabulation()
95620cf1dd8SToby Isaac @*/
95720cf1dd8SToby Isaac PetscErrorCode PetscFEGetTabulation(PetscFE fem, PetscInt npoints, const PetscReal points[], PetscReal **B, PetscReal **D, PetscReal **H)
95820cf1dd8SToby Isaac {
95920cf1dd8SToby Isaac   DM               dm;
96020cf1dd8SToby Isaac   PetscInt         pdim; /* Dimension of FE space P */
96120cf1dd8SToby Isaac   PetscInt         dim;  /* Spatial dimension */
96220cf1dd8SToby Isaac   PetscInt         comp; /* Field components */
96320cf1dd8SToby Isaac   PetscErrorCode   ierr;
96420cf1dd8SToby Isaac 
96520cf1dd8SToby Isaac   PetscFunctionBegin;
96620cf1dd8SToby Isaac   if (!npoints) {
96720cf1dd8SToby Isaac     if (B) *B = NULL;
96820cf1dd8SToby Isaac     if (D) *D = NULL;
96920cf1dd8SToby Isaac     if (H) *H = NULL;
97020cf1dd8SToby Isaac     PetscFunctionReturn(0);
97120cf1dd8SToby Isaac   }
97220cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
97320cf1dd8SToby Isaac   PetscValidPointer(points, 3);
97420cf1dd8SToby Isaac   if (B) PetscValidPointer(B, 4);
97520cf1dd8SToby Isaac   if (D) PetscValidPointer(D, 5);
97620cf1dd8SToby Isaac   if (H) PetscValidPointer(H, 6);
97720cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(fem->dualSpace, &dm);CHKERRQ(ierr);
97820cf1dd8SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
97920cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDimension(fem->dualSpace, &pdim);CHKERRQ(ierr);
98020cf1dd8SToby Isaac   ierr = PetscFEGetNumComponents(fem, &comp);CHKERRQ(ierr);
98120cf1dd8SToby Isaac   if (B) {ierr = DMGetWorkArray(dm, npoints*pdim*comp, MPIU_REAL, B);CHKERRQ(ierr);}
98220cf1dd8SToby Isaac   if (!dim) {
98320cf1dd8SToby Isaac     if (D) *D = NULL;
98420cf1dd8SToby Isaac     if (H) *H = NULL;
98520cf1dd8SToby Isaac   } else {
98620cf1dd8SToby Isaac     if (D) {ierr = DMGetWorkArray(dm, npoints*pdim*comp*dim, MPIU_REAL, D);CHKERRQ(ierr);}
98720cf1dd8SToby Isaac     if (H) {ierr = DMGetWorkArray(dm, npoints*pdim*comp*dim*dim, MPIU_REAL, H);CHKERRQ(ierr);}
98820cf1dd8SToby Isaac   }
98920cf1dd8SToby Isaac   ierr = (*fem->ops->gettabulation)(fem, npoints, points, B ? *B : NULL, D ? *D : NULL, H ? *H : NULL);CHKERRQ(ierr);
99020cf1dd8SToby Isaac   PetscFunctionReturn(0);
99120cf1dd8SToby Isaac }
99220cf1dd8SToby Isaac 
9932b99622eSMatthew G. Knepley /*@C
9942b99622eSMatthew G. Knepley   PetscFERestoreTabulation - Frees memory from the associated tabulation.
9952b99622eSMatthew G. Knepley 
9962b99622eSMatthew G. Knepley   Not collective
9972b99622eSMatthew G. Knepley 
9982b99622eSMatthew G. Knepley   Input Parameters:
9992b99622eSMatthew G. Knepley + fem     - The PetscFE object
10002b99622eSMatthew G. Knepley . npoints - The number of tabulation points
10012b99622eSMatthew G. Knepley . points  - The tabulation point coordinates
10022b99622eSMatthew G. Knepley . B - The basis function values at tabulation points
10032b99622eSMatthew G. Knepley . D - The basis function derivatives at tabulation points
10042b99622eSMatthew G. Knepley - H - The basis function second derivatives at tabulation points
10052b99622eSMatthew G. Knepley 
10062b99622eSMatthew G. Knepley   Note:
10072b99622eSMatthew G. Knepley $ B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c
10082b99622eSMatthew G. Knepley $ D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d
10092b99622eSMatthew G. Knepley $ H[(((p*pdim + i)*Nc + c)*dim + d)*dim + e] is the value at point p for basis function i, component c, in directions d and e
10102b99622eSMatthew G. Knepley 
10112b99622eSMatthew G. Knepley   Level: intermediate
10122b99622eSMatthew G. Knepley 
10132b99622eSMatthew G. Knepley .seealso: PetscFEGetTabulation(), PetscFEGetDefaultTabulation()
10142b99622eSMatthew G. Knepley @*/
101520cf1dd8SToby Isaac PetscErrorCode PetscFERestoreTabulation(PetscFE fem, PetscInt npoints, const PetscReal points[], PetscReal **B, PetscReal **D, PetscReal **H)
101620cf1dd8SToby Isaac {
101720cf1dd8SToby Isaac   DM             dm;
101820cf1dd8SToby Isaac   PetscErrorCode ierr;
101920cf1dd8SToby Isaac 
102020cf1dd8SToby Isaac   PetscFunctionBegin;
102120cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
102220cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(fem->dualSpace, &dm);CHKERRQ(ierr);
102320cf1dd8SToby Isaac   if (B && *B) {ierr = DMRestoreWorkArray(dm, 0, MPIU_REAL, B);CHKERRQ(ierr);}
102420cf1dd8SToby Isaac   if (D && *D) {ierr = DMRestoreWorkArray(dm, 0, MPIU_REAL, D);CHKERRQ(ierr);}
102520cf1dd8SToby Isaac   if (H && *H) {ierr = DMRestoreWorkArray(dm, 0, MPIU_REAL, H);CHKERRQ(ierr);}
102620cf1dd8SToby Isaac   PetscFunctionReturn(0);
102720cf1dd8SToby Isaac }
102820cf1dd8SToby Isaac 
102920cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode PetscFECreatePointTrace(PetscFE fe, PetscInt refPoint, PetscFE *trFE)
103020cf1dd8SToby Isaac {
103120cf1dd8SToby Isaac   PetscSpace     bsp, bsubsp;
103220cf1dd8SToby Isaac   PetscDualSpace dsp, dsubsp;
103320cf1dd8SToby Isaac   PetscInt       dim, depth, numComp, i, j, coneSize, order;
103420cf1dd8SToby Isaac   PetscFEType    type;
103520cf1dd8SToby Isaac   DM             dm;
103620cf1dd8SToby Isaac   DMLabel        label;
103720cf1dd8SToby Isaac   PetscReal      *xi, *v, *J, detJ;
1038db11e2ebSMatthew G. Knepley   const char     *name;
103920cf1dd8SToby Isaac   PetscQuadrature origin, fullQuad, subQuad;
104020cf1dd8SToby Isaac   PetscErrorCode ierr;
104120cf1dd8SToby Isaac 
104220cf1dd8SToby Isaac   PetscFunctionBegin;
104320cf1dd8SToby Isaac   PetscValidHeaderSpecific(fe,PETSCFE_CLASSID,1);
104420cf1dd8SToby Isaac   PetscValidPointer(trFE,3);
104520cf1dd8SToby Isaac   ierr = PetscFEGetBasisSpace(fe,&bsp);CHKERRQ(ierr);
104620cf1dd8SToby Isaac   ierr = PetscFEGetDualSpace(fe,&dsp);CHKERRQ(ierr);
104720cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(dsp,&dm);CHKERRQ(ierr);
104820cf1dd8SToby Isaac   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
104920cf1dd8SToby Isaac   ierr = DMPlexGetDepthLabel(dm,&label);CHKERRQ(ierr);
105020cf1dd8SToby Isaac   ierr = DMLabelGetValue(label,refPoint,&depth);CHKERRQ(ierr);
105120cf1dd8SToby Isaac   ierr = PetscCalloc1(depth,&xi);CHKERRQ(ierr);
105220cf1dd8SToby Isaac   ierr = PetscMalloc1(dim,&v);CHKERRQ(ierr);
105320cf1dd8SToby Isaac   ierr = PetscMalloc1(dim*dim,&J);CHKERRQ(ierr);
105420cf1dd8SToby Isaac   for (i = 0; i < depth; i++) xi[i] = 0.;
105520cf1dd8SToby Isaac   ierr = PetscQuadratureCreate(PETSC_COMM_SELF,&origin);CHKERRQ(ierr);
105620cf1dd8SToby Isaac   ierr = PetscQuadratureSetData(origin,depth,0,1,xi,NULL);CHKERRQ(ierr);
105720cf1dd8SToby Isaac   ierr = DMPlexComputeCellGeometryFEM(dm,refPoint,origin,v,J,NULL,&detJ);CHKERRQ(ierr);
105820cf1dd8SToby Isaac   /* CellGeometryFEM computes the expanded Jacobian, we want the true jacobian */
105920cf1dd8SToby Isaac   for (i = 1; i < dim; i++) {
106020cf1dd8SToby Isaac     for (j = 0; j < depth; j++) {
106120cf1dd8SToby Isaac       J[i * depth + j] = J[i * dim + j];
106220cf1dd8SToby Isaac     }
106320cf1dd8SToby Isaac   }
106420cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&origin);CHKERRQ(ierr);
106520cf1dd8SToby Isaac   ierr = PetscDualSpaceGetPointSubspace(dsp,refPoint,&dsubsp);CHKERRQ(ierr);
106620cf1dd8SToby Isaac   ierr = PetscSpaceCreateSubspace(bsp,dsubsp,v,J,NULL,NULL,PETSC_OWN_POINTER,&bsubsp);CHKERRQ(ierr);
106720cf1dd8SToby Isaac   ierr = PetscSpaceSetUp(bsubsp);CHKERRQ(ierr);
106820cf1dd8SToby Isaac   ierr = PetscFECreate(PetscObjectComm((PetscObject)fe),trFE);CHKERRQ(ierr);
106920cf1dd8SToby Isaac   ierr = PetscFEGetType(fe,&type);CHKERRQ(ierr);
107020cf1dd8SToby Isaac   ierr = PetscFESetType(*trFE,type);CHKERRQ(ierr);
107120cf1dd8SToby Isaac   ierr = PetscFEGetNumComponents(fe,&numComp);CHKERRQ(ierr);
107220cf1dd8SToby Isaac   ierr = PetscFESetNumComponents(*trFE,numComp);CHKERRQ(ierr);
107320cf1dd8SToby Isaac   ierr = PetscFESetBasisSpace(*trFE,bsubsp);CHKERRQ(ierr);
107420cf1dd8SToby Isaac   ierr = PetscFESetDualSpace(*trFE,dsubsp);CHKERRQ(ierr);
1075db11e2ebSMatthew G. Knepley   ierr = PetscObjectGetName((PetscObject) fe, &name);CHKERRQ(ierr);
1076db11e2ebSMatthew G. Knepley   if (name) {ierr = PetscFESetName(*trFE, name);CHKERRQ(ierr);}
107720cf1dd8SToby Isaac   ierr = PetscFEGetQuadrature(fe,&fullQuad);CHKERRQ(ierr);
107820cf1dd8SToby Isaac   ierr = PetscQuadratureGetOrder(fullQuad,&order);CHKERRQ(ierr);
107920cf1dd8SToby Isaac   ierr = DMPlexGetConeSize(dm,refPoint,&coneSize);CHKERRQ(ierr);
108020cf1dd8SToby Isaac   if (coneSize == 2 * depth) {
108120cf1dd8SToby Isaac     ierr = PetscDTGaussTensorQuadrature(depth,1,(order + 1)/2,-1.,1.,&subQuad);CHKERRQ(ierr);
108220cf1dd8SToby Isaac   } else {
108320cf1dd8SToby Isaac     ierr = PetscDTGaussJacobiQuadrature(depth,1,(order + 1)/2,-1.,1.,&subQuad);CHKERRQ(ierr);
108420cf1dd8SToby Isaac   }
108520cf1dd8SToby Isaac   ierr = PetscFESetQuadrature(*trFE,subQuad);CHKERRQ(ierr);
108620cf1dd8SToby Isaac   ierr = PetscFESetUp(*trFE);CHKERRQ(ierr);
108720cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&subQuad);CHKERRQ(ierr);
108820cf1dd8SToby Isaac   ierr = PetscSpaceDestroy(&bsubsp);CHKERRQ(ierr);
108920cf1dd8SToby Isaac   PetscFunctionReturn(0);
109020cf1dd8SToby Isaac }
109120cf1dd8SToby Isaac 
109220cf1dd8SToby Isaac PetscErrorCode PetscFECreateHeightTrace(PetscFE fe, PetscInt height, PetscFE *trFE)
109320cf1dd8SToby Isaac {
109420cf1dd8SToby Isaac   PetscInt       hStart, hEnd;
109520cf1dd8SToby Isaac   PetscDualSpace dsp;
109620cf1dd8SToby Isaac   DM             dm;
109720cf1dd8SToby Isaac   PetscErrorCode ierr;
109820cf1dd8SToby Isaac 
109920cf1dd8SToby Isaac   PetscFunctionBegin;
110020cf1dd8SToby Isaac   PetscValidHeaderSpecific(fe,PETSCFE_CLASSID,1);
110120cf1dd8SToby Isaac   PetscValidPointer(trFE,3);
110220cf1dd8SToby Isaac   *trFE = NULL;
110320cf1dd8SToby Isaac   ierr = PetscFEGetDualSpace(fe,&dsp);CHKERRQ(ierr);
110420cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(dsp,&dm);CHKERRQ(ierr);
110520cf1dd8SToby Isaac   ierr = DMPlexGetHeightStratum(dm,height,&hStart,&hEnd);CHKERRQ(ierr);
110620cf1dd8SToby Isaac   if (hEnd <= hStart) PetscFunctionReturn(0);
110720cf1dd8SToby Isaac   ierr = PetscFECreatePointTrace(fe,hStart,trFE);CHKERRQ(ierr);
110820cf1dd8SToby Isaac   PetscFunctionReturn(0);
110920cf1dd8SToby Isaac }
111020cf1dd8SToby Isaac 
111120cf1dd8SToby Isaac 
111220cf1dd8SToby Isaac /*@
111320cf1dd8SToby Isaac   PetscFEGetDimension - Get the dimension of the finite element space on a cell
111420cf1dd8SToby Isaac 
111520cf1dd8SToby Isaac   Not collective
111620cf1dd8SToby Isaac 
111720cf1dd8SToby Isaac   Input Parameter:
111820cf1dd8SToby Isaac . fe - The PetscFE
111920cf1dd8SToby Isaac 
112020cf1dd8SToby Isaac   Output Parameter:
112120cf1dd8SToby Isaac . dim - The dimension
112220cf1dd8SToby Isaac 
112320cf1dd8SToby Isaac   Level: intermediate
112420cf1dd8SToby Isaac 
112520cf1dd8SToby Isaac .seealso: PetscFECreate(), PetscSpaceGetDimension(), PetscDualSpaceGetDimension()
112620cf1dd8SToby Isaac @*/
112720cf1dd8SToby Isaac PetscErrorCode PetscFEGetDimension(PetscFE fem, PetscInt *dim)
112820cf1dd8SToby Isaac {
112920cf1dd8SToby Isaac   PetscErrorCode ierr;
113020cf1dd8SToby Isaac 
113120cf1dd8SToby Isaac   PetscFunctionBegin;
113220cf1dd8SToby Isaac   PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1);
113320cf1dd8SToby Isaac   PetscValidPointer(dim, 2);
113420cf1dd8SToby Isaac   if (fem->ops->getdimension) {ierr = (*fem->ops->getdimension)(fem, dim);CHKERRQ(ierr);}
113520cf1dd8SToby Isaac   PetscFunctionReturn(0);
113620cf1dd8SToby Isaac }
113720cf1dd8SToby Isaac 
11384bee2e38SMatthew G. Knepley /*@C
11394bee2e38SMatthew G. Knepley   PetscFEPushforward - Map the reference element function to real space
11404bee2e38SMatthew G. Knepley 
11414bee2e38SMatthew G. Knepley   Input Parameters:
11424bee2e38SMatthew G. Knepley + fe     - The PetscFE
11434bee2e38SMatthew G. Knepley . fegeom - The cell geometry
11444bee2e38SMatthew G. Knepley . Nv     - The number of function values
11454bee2e38SMatthew G. Knepley - vals   - The function values
11464bee2e38SMatthew G. Knepley 
11474bee2e38SMatthew G. Knepley   Output Parameter:
11484bee2e38SMatthew G. Knepley . vals   - The transformed function values
11494bee2e38SMatthew G. Knepley 
11504bee2e38SMatthew G. Knepley   Level: advanced
11514bee2e38SMatthew G. Knepley 
11524bee2e38SMatthew G. Knepley   Note: This just forwards the call onto PetscDualSpacePushforward().
11534bee2e38SMatthew G. Knepley 
11544bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePushforward()
11554bee2e38SMatthew G. Knepley @*/
11564bee2e38SMatthew G. Knepley PetscErrorCode PetscFEPushforward(PetscFE fe, PetscFEGeom *fegeom, PetscInt Nv, PetscScalar vals[])
11574bee2e38SMatthew G. Knepley {
11584bee2e38SMatthew G. Knepley   PetscErrorCode ierr;
11594bee2e38SMatthew G. Knepley 
11602ae266adSMatthew G. Knepley   PetscFunctionBeginHot;
11612ae266adSMatthew G. Knepley   ierr = PetscDualSpacePushforward(fe->dualSpace, fegeom, Nv, fe->numComponents, vals);CHKERRQ(ierr);
11624bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
11634bee2e38SMatthew G. Knepley }
11644bee2e38SMatthew G. Knepley 
11654bee2e38SMatthew G. Knepley /*@C
11664bee2e38SMatthew G. Knepley   PetscFEPushforwardGradient - Map the reference element function gradient to real space
11674bee2e38SMatthew G. Knepley 
11684bee2e38SMatthew G. Knepley   Input Parameters:
11694bee2e38SMatthew G. Knepley + fe     - The PetscFE
11704bee2e38SMatthew G. Knepley . fegeom - The cell geometry
11714bee2e38SMatthew G. Knepley . Nv     - The number of function gradient values
11724bee2e38SMatthew G. Knepley - vals   - The function gradient values
11734bee2e38SMatthew G. Knepley 
11744bee2e38SMatthew G. Knepley   Output Parameter:
11754bee2e38SMatthew G. Knepley . vals   - The transformed function gradient values
11764bee2e38SMatthew G. Knepley 
11774bee2e38SMatthew G. Knepley   Level: advanced
11784bee2e38SMatthew G. Knepley 
11794bee2e38SMatthew G. Knepley   Note: This just forwards the call onto PetscDualSpacePushforwardGradient().
11804bee2e38SMatthew G. Knepley 
11814bee2e38SMatthew G. Knepley .seealso: PetscFEPushforward(), PetscDualSpacePushforwardGradient(), PetscDualSpacePushforward()
11824bee2e38SMatthew G. Knepley @*/
11834bee2e38SMatthew G. Knepley PetscErrorCode PetscFEPushforwardGradient(PetscFE fe, PetscFEGeom *fegeom, PetscInt Nv, PetscScalar vals[])
11844bee2e38SMatthew G. Knepley {
11854bee2e38SMatthew G. Knepley   PetscErrorCode ierr;
11864bee2e38SMatthew G. Knepley 
11872ae266adSMatthew G. Knepley   PetscFunctionBeginHot;
11882ae266adSMatthew G. Knepley   ierr = PetscDualSpacePushforwardGradient(fe->dualSpace, fegeom, Nv, fe->numComponents, vals);CHKERRQ(ierr);
11894bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
11904bee2e38SMatthew G. Knepley }
11914bee2e38SMatthew G. Knepley 
119220cf1dd8SToby Isaac /*
119320cf1dd8SToby Isaac Purpose: Compute element vector for chunk of elements
119420cf1dd8SToby Isaac 
119520cf1dd8SToby Isaac Input:
119620cf1dd8SToby Isaac   Sizes:
119720cf1dd8SToby Isaac      Ne:  number of elements
119820cf1dd8SToby Isaac      Nf:  number of fields
119920cf1dd8SToby Isaac      PetscFE
120020cf1dd8SToby Isaac        dim: spatial dimension
120120cf1dd8SToby Isaac        Nb:  number of basis functions
120220cf1dd8SToby Isaac        Nc:  number of field components
120320cf1dd8SToby Isaac        PetscQuadrature
120420cf1dd8SToby Isaac          Nq:  number of quadrature points
120520cf1dd8SToby Isaac 
120620cf1dd8SToby Isaac   Geometry:
120720cf1dd8SToby Isaac      PetscFEGeom[Ne] possibly *Nq
120820cf1dd8SToby Isaac        PetscReal v0s[dim]
120920cf1dd8SToby Isaac        PetscReal n[dim]
121020cf1dd8SToby Isaac        PetscReal jacobians[dim*dim]
121120cf1dd8SToby Isaac        PetscReal jacobianInverses[dim*dim]
121220cf1dd8SToby Isaac        PetscReal jacobianDeterminants
121320cf1dd8SToby Isaac   FEM:
121420cf1dd8SToby Isaac      PetscFE
121520cf1dd8SToby Isaac        PetscQuadrature
121620cf1dd8SToby Isaac          PetscReal   quadPoints[Nq*dim]
121720cf1dd8SToby Isaac          PetscReal   quadWeights[Nq]
121820cf1dd8SToby Isaac        PetscReal   basis[Nq*Nb*Nc]
121920cf1dd8SToby Isaac        PetscReal   basisDer[Nq*Nb*Nc*dim]
122020cf1dd8SToby Isaac      PetscScalar coefficients[Ne*Nb*Nc]
122120cf1dd8SToby Isaac      PetscScalar elemVec[Ne*Nb*Nc]
122220cf1dd8SToby Isaac 
122320cf1dd8SToby Isaac   Problem:
122420cf1dd8SToby Isaac      PetscInt f: the active field
122520cf1dd8SToby Isaac      f0, f1
122620cf1dd8SToby Isaac 
122720cf1dd8SToby Isaac   Work Space:
122820cf1dd8SToby Isaac      PetscFE
122920cf1dd8SToby Isaac        PetscScalar f0[Nq*dim];
123020cf1dd8SToby Isaac        PetscScalar f1[Nq*dim*dim];
123120cf1dd8SToby Isaac        PetscScalar u[Nc];
123220cf1dd8SToby Isaac        PetscScalar gradU[Nc*dim];
123320cf1dd8SToby Isaac        PetscReal   x[dim];
123420cf1dd8SToby Isaac        PetscScalar realSpaceDer[dim];
123520cf1dd8SToby Isaac 
123620cf1dd8SToby Isaac Purpose: Compute element vector for N_cb batches of elements
123720cf1dd8SToby Isaac 
123820cf1dd8SToby Isaac Input:
123920cf1dd8SToby Isaac   Sizes:
124020cf1dd8SToby Isaac      N_cb: Number of serial cell batches
124120cf1dd8SToby Isaac 
124220cf1dd8SToby Isaac   Geometry:
124320cf1dd8SToby Isaac      PetscReal v0s[Ne*dim]
124420cf1dd8SToby Isaac      PetscReal jacobians[Ne*dim*dim]        possibly *Nq
124520cf1dd8SToby Isaac      PetscReal jacobianInverses[Ne*dim*dim] possibly *Nq
124620cf1dd8SToby Isaac      PetscReal jacobianDeterminants[Ne]     possibly *Nq
124720cf1dd8SToby Isaac   FEM:
124820cf1dd8SToby Isaac      static PetscReal   quadPoints[Nq*dim]
124920cf1dd8SToby Isaac      static PetscReal   quadWeights[Nq]
125020cf1dd8SToby Isaac      static PetscReal   basis[Nq*Nb*Nc]
125120cf1dd8SToby Isaac      static PetscReal   basisDer[Nq*Nb*Nc*dim]
125220cf1dd8SToby Isaac      PetscScalar coefficients[Ne*Nb*Nc]
125320cf1dd8SToby Isaac      PetscScalar elemVec[Ne*Nb*Nc]
125420cf1dd8SToby Isaac 
125520cf1dd8SToby Isaac ex62.c:
125620cf1dd8SToby Isaac   PetscErrorCode PetscFEIntegrateResidualBatch(PetscInt Ne, PetscInt numFields, PetscInt field, PetscQuadrature quad[], const PetscScalar coefficients[],
125720cf1dd8SToby Isaac                                                const PetscReal v0s[], const PetscReal jacobians[], const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[],
125820cf1dd8SToby Isaac                                                void (*f0_func)(const PetscScalar u[], const PetscScalar gradU[], const PetscReal x[], PetscScalar f0[]),
125920cf1dd8SToby Isaac                                                void (*f1_func)(const PetscScalar u[], const PetscScalar gradU[], const PetscReal x[], PetscScalar f1[]), PetscScalar elemVec[])
126020cf1dd8SToby Isaac 
126120cf1dd8SToby Isaac ex52.c:
126220cf1dd8SToby Isaac   PetscErrorCode IntegrateLaplacianBatchCPU(PetscInt Ne, PetscInt Nb, const PetscScalar coefficients[], const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscInt Nq, const PetscReal quadPoints[], const PetscReal quadWeights[], const PetscReal basisTabulation[], const PetscReal basisDerTabulation[], PetscScalar elemVec[], AppCtx *user)
126320cf1dd8SToby Isaac   PetscErrorCode IntegrateElasticityBatchCPU(PetscInt Ne, PetscInt Nb, PetscInt Ncomp, const PetscScalar coefficients[], const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscInt Nq, const PetscReal quadPoints[], const PetscReal quadWeights[], const PetscReal basisTabulation[], const PetscReal basisDerTabulation[], PetscScalar elemVec[], AppCtx *user)
126420cf1dd8SToby Isaac 
126520cf1dd8SToby Isaac ex52_integrateElement.cu
126620cf1dd8SToby Isaac __global__ void integrateElementQuadrature(int N_cb, realType *coefficients, realType *jacobianInverses, realType *jacobianDeterminants, realType *elemVec)
126720cf1dd8SToby Isaac 
126820cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode IntegrateElementBatchGPU(PetscInt spatial_dim, PetscInt Ne, PetscInt Ncb, PetscInt Nbc, PetscInt Nbl, const PetscScalar coefficients[],
126920cf1dd8SToby Isaac                                                      const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscScalar elemVec[],
127020cf1dd8SToby Isaac                                                      PetscLogEvent event, PetscInt debug, PetscInt pde_op)
127120cf1dd8SToby Isaac 
127220cf1dd8SToby Isaac ex52_integrateElementOpenCL.c:
127320cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode IntegrateElementBatchGPU(PetscInt spatial_dim, PetscInt Ne, PetscInt Ncb, PetscInt Nbc, PetscInt N_bl, const PetscScalar coefficients[],
127420cf1dd8SToby Isaac                                                      const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscScalar elemVec[],
127520cf1dd8SToby Isaac                                                      PetscLogEvent event, PetscInt debug, PetscInt pde_op)
127620cf1dd8SToby Isaac 
127720cf1dd8SToby Isaac __kernel void integrateElementQuadrature(int N_cb, __global float *coefficients, __global float *jacobianInverses, __global float *jacobianDeterminants, __global float *elemVec)
127820cf1dd8SToby Isaac */
127920cf1dd8SToby Isaac 
128020cf1dd8SToby Isaac /*@C
128120cf1dd8SToby Isaac   PetscFEIntegrate - Produce the integral for the given field for a chunk of elements by quadrature integration
128220cf1dd8SToby Isaac 
128320cf1dd8SToby Isaac   Not collective
128420cf1dd8SToby Isaac 
128520cf1dd8SToby Isaac   Input Parameters:
128620cf1dd8SToby Isaac + fem          - The PetscFE object for the field being integrated
128720cf1dd8SToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
128820cf1dd8SToby Isaac . field        - The field being integrated
128920cf1dd8SToby Isaac . Ne           - The number of elements in the chunk
129020cf1dd8SToby Isaac . cgeom        - The cell geometry for each cell in the chunk
129120cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements
129220cf1dd8SToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
129320cf1dd8SToby Isaac - coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
129420cf1dd8SToby Isaac 
129520cf1dd8SToby Isaac   Output Parameter
129620cf1dd8SToby Isaac . integral     - the integral for this field
129720cf1dd8SToby Isaac 
12982b99622eSMatthew G. Knepley   Level: intermediate
129920cf1dd8SToby Isaac 
130020cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual()
130120cf1dd8SToby Isaac @*/
13024bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrate(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *cgeom,
130320cf1dd8SToby Isaac                                 const PetscScalar coefficients[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscScalar integral[])
130420cf1dd8SToby Isaac {
13054bee2e38SMatthew G. Knepley   PetscFE        fe;
130620cf1dd8SToby Isaac   PetscErrorCode ierr;
130720cf1dd8SToby Isaac 
130820cf1dd8SToby Isaac   PetscFunctionBegin;
13094bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
13104bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr);
13114bee2e38SMatthew G. Knepley   if (fe->ops->integrate) {ierr = (*fe->ops->integrate)(prob, field, Ne, cgeom, coefficients, probAux, coefficientsAux, integral);CHKERRQ(ierr);}
131220cf1dd8SToby Isaac   PetscFunctionReturn(0);
131320cf1dd8SToby Isaac }
131420cf1dd8SToby Isaac 
131520cf1dd8SToby Isaac /*@C
1316afe6d6adSToby Isaac   PetscFEIntegrateBd - Produce the integral for the given field for a chunk of elements by quadrature integration
1317afe6d6adSToby Isaac 
1318afe6d6adSToby Isaac   Not collective
1319afe6d6adSToby Isaac 
1320afe6d6adSToby Isaac   Input Parameters:
1321afe6d6adSToby Isaac + fem          - The PetscFE object for the field being integrated
1322afe6d6adSToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
1323afe6d6adSToby Isaac . field        - The field being integrated
1324afe6d6adSToby Isaac . obj_func     - The function to be integrated
1325afe6d6adSToby Isaac . Ne           - The number of elements in the chunk
1326afe6d6adSToby Isaac . fgeom        - The face geometry for each face in the chunk
1327afe6d6adSToby Isaac . coefficients - The array of FEM basis coefficients for the elements
1328afe6d6adSToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
1329afe6d6adSToby Isaac - coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
1330afe6d6adSToby Isaac 
1331afe6d6adSToby Isaac   Output Parameter
1332afe6d6adSToby Isaac . integral     - the integral for this field
1333afe6d6adSToby Isaac 
13342b99622eSMatthew G. Knepley   Level: intermediate
1335afe6d6adSToby Isaac 
1336afe6d6adSToby Isaac .seealso: PetscFEIntegrateResidual()
1337afe6d6adSToby Isaac @*/
13384bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBd(PetscDS prob, PetscInt field,
1339afe6d6adSToby Isaac                                   void (*obj_func)(PetscInt, PetscInt, PetscInt,
1340afe6d6adSToby Isaac                                                    const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
1341afe6d6adSToby Isaac                                                    const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
1342afe6d6adSToby Isaac                                                    PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]),
1343afe6d6adSToby Isaac                                   PetscInt Ne, PetscFEGeom *geom, const PetscScalar coefficients[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscScalar integral[])
1344afe6d6adSToby Isaac {
13454bee2e38SMatthew G. Knepley   PetscFE        fe;
1346afe6d6adSToby Isaac   PetscErrorCode ierr;
1347afe6d6adSToby Isaac 
1348afe6d6adSToby Isaac   PetscFunctionBegin;
13494bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
13504bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr);
13514bee2e38SMatthew G. Knepley   if (fe->ops->integratebd) {ierr = (*fe->ops->integratebd)(prob, field, obj_func, Ne, geom, coefficients, probAux, coefficientsAux, integral);CHKERRQ(ierr);}
1352afe6d6adSToby Isaac   PetscFunctionReturn(0);
1353afe6d6adSToby Isaac }
1354afe6d6adSToby Isaac 
1355afe6d6adSToby Isaac /*@C
135620cf1dd8SToby Isaac   PetscFEIntegrateResidual - Produce the element residual vector for a chunk of elements by quadrature integration
135720cf1dd8SToby Isaac 
135820cf1dd8SToby Isaac   Not collective
135920cf1dd8SToby Isaac 
136020cf1dd8SToby Isaac   Input Parameters:
136120cf1dd8SToby Isaac + fem          - The PetscFE object for the field being integrated
136220cf1dd8SToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
136320cf1dd8SToby Isaac . field        - The field being integrated
136420cf1dd8SToby Isaac . Ne           - The number of elements in the chunk
136520cf1dd8SToby Isaac . cgeom        - The cell geometry for each cell in the chunk
136620cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements
136720cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements
136820cf1dd8SToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
136920cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
137020cf1dd8SToby Isaac - t            - The time
137120cf1dd8SToby Isaac 
137220cf1dd8SToby Isaac   Output Parameter
137320cf1dd8SToby Isaac . elemVec      - the element residual vectors from each element
137420cf1dd8SToby Isaac 
137520cf1dd8SToby Isaac   Note:
137620cf1dd8SToby Isaac $ Loop over batch of elements (e):
137720cf1dd8SToby Isaac $   Loop over quadrature points (q):
137820cf1dd8SToby Isaac $     Make u_q and gradU_q (loops over fields,Nb,Ncomp) and x_q
137920cf1dd8SToby Isaac $     Call f_0 and f_1
138020cf1dd8SToby Isaac $   Loop over element vector entries (f,fc --> i):
138120cf1dd8SToby Isaac $     elemVec[i] += \psi^{fc}_f(q) f0_{fc}(u, \nabla u) + \nabla\psi^{fc}_f(q) \cdot f1_{fc,df}(u, \nabla u)
138220cf1dd8SToby Isaac 
13832b99622eSMatthew G. Knepley   Level: intermediate
138420cf1dd8SToby Isaac 
138520cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual()
138620cf1dd8SToby Isaac @*/
13874bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateResidual(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *cgeom,
138820cf1dd8SToby Isaac                                         const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscScalar elemVec[])
138920cf1dd8SToby Isaac {
13904bee2e38SMatthew G. Knepley   PetscFE        fe;
139120cf1dd8SToby Isaac   PetscErrorCode ierr;
139220cf1dd8SToby Isaac 
139320cf1dd8SToby Isaac   PetscFunctionBegin;
13944bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
13954bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr);
13964bee2e38SMatthew G. Knepley   if (fe->ops->integrateresidual) {ierr = (*fe->ops->integrateresidual)(prob, field, Ne, cgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, elemVec);CHKERRQ(ierr);}
139720cf1dd8SToby Isaac   PetscFunctionReturn(0);
139820cf1dd8SToby Isaac }
139920cf1dd8SToby Isaac 
140020cf1dd8SToby Isaac /*@C
140120cf1dd8SToby Isaac   PetscFEIntegrateBdResidual - Produce the element residual vector for a chunk of elements by quadrature integration over a boundary
140220cf1dd8SToby Isaac 
140320cf1dd8SToby Isaac   Not collective
140420cf1dd8SToby Isaac 
140520cf1dd8SToby Isaac   Input Parameters:
140620cf1dd8SToby Isaac + fem          - The PetscFE object for the field being integrated
140720cf1dd8SToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
140820cf1dd8SToby Isaac . field        - The field being integrated
140920cf1dd8SToby Isaac . Ne           - The number of elements in the chunk
141020cf1dd8SToby Isaac . fgeom        - The face geometry for each cell in the chunk
141120cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements
141220cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements
141320cf1dd8SToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
141420cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
141520cf1dd8SToby Isaac - t            - The time
141620cf1dd8SToby Isaac 
141720cf1dd8SToby Isaac   Output Parameter
141820cf1dd8SToby Isaac . elemVec      - the element residual vectors from each element
141920cf1dd8SToby Isaac 
14202b99622eSMatthew G. Knepley   Level: intermediate
142120cf1dd8SToby Isaac 
142220cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual()
142320cf1dd8SToby Isaac @*/
14244bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBdResidual(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *fgeom,
142520cf1dd8SToby Isaac                                           const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscScalar elemVec[])
142620cf1dd8SToby Isaac {
14274bee2e38SMatthew G. Knepley   PetscFE        fe;
142820cf1dd8SToby Isaac   PetscErrorCode ierr;
142920cf1dd8SToby Isaac 
143020cf1dd8SToby Isaac   PetscFunctionBegin;
14314bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
14324bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr);
14334bee2e38SMatthew G. Knepley   if (fe->ops->integratebdresidual) {ierr = (*fe->ops->integratebdresidual)(prob, field, Ne, fgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, elemVec);CHKERRQ(ierr);}
143420cf1dd8SToby Isaac   PetscFunctionReturn(0);
143520cf1dd8SToby Isaac }
143620cf1dd8SToby Isaac 
143720cf1dd8SToby Isaac /*@C
143820cf1dd8SToby Isaac   PetscFEIntegrateJacobian - Produce the element Jacobian for a chunk of elements by quadrature integration
143920cf1dd8SToby Isaac 
144020cf1dd8SToby Isaac   Not collective
144120cf1dd8SToby Isaac 
144220cf1dd8SToby Isaac   Input Parameters:
144320cf1dd8SToby Isaac + fem          - The PetscFE object for the field being integrated
144420cf1dd8SToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
144520cf1dd8SToby Isaac . jtype        - The type of matrix pointwise functions that should be used
144620cf1dd8SToby Isaac . fieldI       - The test field being integrated
144720cf1dd8SToby Isaac . fieldJ       - The basis field being integrated
144820cf1dd8SToby Isaac . Ne           - The number of elements in the chunk
144920cf1dd8SToby Isaac . cgeom        - The cell geometry for each cell in the chunk
145020cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements for the Jacobian evaluation point
145120cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements
145220cf1dd8SToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
145320cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
145420cf1dd8SToby Isaac . t            - The time
145520cf1dd8SToby Isaac - u_tShift     - A multiplier for the dF/du_t term (as opposed to the dF/du term)
145620cf1dd8SToby Isaac 
145720cf1dd8SToby Isaac   Output Parameter
145820cf1dd8SToby Isaac . elemMat      - the element matrices for the Jacobian from each element
145920cf1dd8SToby Isaac 
146020cf1dd8SToby Isaac   Note:
146120cf1dd8SToby Isaac $ Loop over batch of elements (e):
146220cf1dd8SToby Isaac $   Loop over element matrix entries (f,fc,g,gc --> i,j):
146320cf1dd8SToby Isaac $     Loop over quadrature points (q):
146420cf1dd8SToby Isaac $       Make u_q and gradU_q (loops over fields,Nb,Ncomp)
146520cf1dd8SToby Isaac $         elemMat[i,j] += \psi^{fc}_f(q) g0_{fc,gc}(u, \nabla u) \phi^{gc}_g(q)
146620cf1dd8SToby Isaac $                      + \psi^{fc}_f(q) \cdot g1_{fc,gc,dg}(u, \nabla u) \nabla\phi^{gc}_g(q)
146720cf1dd8SToby Isaac $                      + \nabla\psi^{fc}_f(q) \cdot g2_{fc,gc,df}(u, \nabla u) \phi^{gc}_g(q)
146820cf1dd8SToby Isaac $                      + \nabla\psi^{fc}_f(q) \cdot g3_{fc,gc,df,dg}(u, \nabla u) \nabla\phi^{gc}_g(q)
14692b99622eSMatthew G. Knepley   Level: intermediate
147020cf1dd8SToby Isaac 
147120cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual()
147220cf1dd8SToby Isaac @*/
14734bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateJacobian(PetscDS prob, PetscFEJacobianType jtype, PetscInt fieldI, PetscInt fieldJ, PetscInt Ne, PetscFEGeom *cgeom,
147420cf1dd8SToby Isaac                                         const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscReal u_tshift, PetscScalar elemMat[])
147520cf1dd8SToby Isaac {
14764bee2e38SMatthew G. Knepley   PetscFE        fe;
147720cf1dd8SToby Isaac   PetscErrorCode ierr;
147820cf1dd8SToby Isaac 
147920cf1dd8SToby Isaac   PetscFunctionBegin;
14804bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
14814bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fe);CHKERRQ(ierr);
14824bee2e38SMatthew G. Knepley   if (fe->ops->integratejacobian) {ierr = (*fe->ops->integratejacobian)(prob, jtype, fieldI, fieldJ, Ne, cgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, u_tshift, elemMat);CHKERRQ(ierr);}
148320cf1dd8SToby Isaac   PetscFunctionReturn(0);
148420cf1dd8SToby Isaac }
148520cf1dd8SToby Isaac 
148620cf1dd8SToby Isaac /*@C
148720cf1dd8SToby Isaac   PetscFEIntegrateBdJacobian - Produce the boundary element Jacobian for a chunk of elements by quadrature integration
148820cf1dd8SToby Isaac 
148920cf1dd8SToby Isaac   Not collective
149020cf1dd8SToby Isaac 
149120cf1dd8SToby Isaac   Input Parameters:
149220cf1dd8SToby Isaac . prob         - The PetscDS specifying the discretizations and continuum functions
149320cf1dd8SToby Isaac . fieldI       - The test field being integrated
149420cf1dd8SToby Isaac . fieldJ       - The basis field being integrated
149520cf1dd8SToby Isaac . Ne           - The number of elements in the chunk
149620cf1dd8SToby Isaac . fgeom        - The face geometry for each cell in the chunk
149720cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements for the Jacobian evaluation point
149820cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements
149920cf1dd8SToby Isaac . probAux      - The PetscDS specifying the auxiliary discretizations
150020cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements
150120cf1dd8SToby Isaac . t            - The time
150220cf1dd8SToby Isaac - u_tShift     - A multiplier for the dF/du_t term (as opposed to the dF/du term)
150320cf1dd8SToby Isaac 
150420cf1dd8SToby Isaac   Output Parameter
150520cf1dd8SToby Isaac . elemMat              - the element matrices for the Jacobian from each element
150620cf1dd8SToby Isaac 
150720cf1dd8SToby Isaac   Note:
150820cf1dd8SToby Isaac $ Loop over batch of elements (e):
150920cf1dd8SToby Isaac $   Loop over element matrix entries (f,fc,g,gc --> i,j):
151020cf1dd8SToby Isaac $     Loop over quadrature points (q):
151120cf1dd8SToby Isaac $       Make u_q and gradU_q (loops over fields,Nb,Ncomp)
151220cf1dd8SToby Isaac $         elemMat[i,j] += \psi^{fc}_f(q) g0_{fc,gc}(u, \nabla u) \phi^{gc}_g(q)
151320cf1dd8SToby Isaac $                      + \psi^{fc}_f(q) \cdot g1_{fc,gc,dg}(u, \nabla u) \nabla\phi^{gc}_g(q)
151420cf1dd8SToby Isaac $                      + \nabla\psi^{fc}_f(q) \cdot g2_{fc,gc,df}(u, \nabla u) \phi^{gc}_g(q)
151520cf1dd8SToby Isaac $                      + \nabla\psi^{fc}_f(q) \cdot g3_{fc,gc,df,dg}(u, \nabla u) \nabla\phi^{gc}_g(q)
15162b99622eSMatthew G. Knepley   Level: intermediate
151720cf1dd8SToby Isaac 
151820cf1dd8SToby Isaac .seealso: PetscFEIntegrateJacobian(), PetscFEIntegrateResidual()
151920cf1dd8SToby Isaac @*/
15204bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBdJacobian(PetscDS prob, PetscInt fieldI, PetscInt fieldJ, PetscInt Ne, PetscFEGeom *fgeom,
152120cf1dd8SToby Isaac                                           const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscReal u_tshift, PetscScalar elemMat[])
152220cf1dd8SToby Isaac {
15234bee2e38SMatthew G. Knepley   PetscFE        fe;
152420cf1dd8SToby Isaac   PetscErrorCode ierr;
152520cf1dd8SToby Isaac 
152620cf1dd8SToby Isaac   PetscFunctionBegin;
15274bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
15284bee2e38SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fe);CHKERRQ(ierr);
15294bee2e38SMatthew G. Knepley   if (fe->ops->integratebdjacobian) {ierr = (*fe->ops->integratebdjacobian)(prob, fieldI, fieldJ, Ne, fgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, u_tshift, elemMat);CHKERRQ(ierr);}
153020cf1dd8SToby Isaac   PetscFunctionReturn(0);
153120cf1dd8SToby Isaac }
153220cf1dd8SToby Isaac 
15332b99622eSMatthew G. Knepley /*@
15342b99622eSMatthew G. Knepley   PetscFEGetHeightSubspace - Get the subspace of this space for a mesh point of a given height
15352b99622eSMatthew G. Knepley 
15362b99622eSMatthew G. Knepley   Input Parameters:
15372b99622eSMatthew G. Knepley + fe     - The finite element space
15382b99622eSMatthew G. Knepley - height - The height of the Plex point
15392b99622eSMatthew G. Knepley 
15402b99622eSMatthew G. Knepley   Output Parameter:
15412b99622eSMatthew G. Knepley . subfe  - The subspace of this FE space
15422b99622eSMatthew G. Knepley 
15432b99622eSMatthew G. Knepley   Note: For example, if we want the subspace of this space for a face, we would choose height = 1.
15442b99622eSMatthew G. Knepley 
15452b99622eSMatthew G. Knepley   Level: advanced
15462b99622eSMatthew G. Knepley 
15472b99622eSMatthew G. Knepley .seealso: PetscFECreateDefault()
15482b99622eSMatthew G. Knepley @*/
154920cf1dd8SToby Isaac PetscErrorCode PetscFEGetHeightSubspace(PetscFE fe, PetscInt height, PetscFE *subfe)
155020cf1dd8SToby Isaac {
155120cf1dd8SToby Isaac   PetscSpace      P, subP;
155220cf1dd8SToby Isaac   PetscDualSpace  Q, subQ;
155320cf1dd8SToby Isaac   PetscQuadrature subq;
155420cf1dd8SToby Isaac   PetscFEType     fetype;
155520cf1dd8SToby Isaac   PetscInt        dim, Nc;
155620cf1dd8SToby Isaac   PetscErrorCode  ierr;
155720cf1dd8SToby Isaac 
155820cf1dd8SToby Isaac   PetscFunctionBegin;
155920cf1dd8SToby Isaac   PetscValidHeaderSpecific(fe, PETSCFE_CLASSID, 1);
156020cf1dd8SToby Isaac   PetscValidPointer(subfe, 3);
156120cf1dd8SToby Isaac   if (height == 0) {
156220cf1dd8SToby Isaac     *subfe = fe;
156320cf1dd8SToby Isaac     PetscFunctionReturn(0);
156420cf1dd8SToby Isaac   }
156520cf1dd8SToby Isaac   ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr);
156620cf1dd8SToby Isaac   ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr);
156720cf1dd8SToby Isaac   ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
156820cf1dd8SToby Isaac   ierr = PetscFEGetFaceQuadrature(fe, &subq);CHKERRQ(ierr);
156920cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDimension(Q, &dim);CHKERRQ(ierr);
157020cf1dd8SToby Isaac   if (height > dim || height < 0) {SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Asked for space at height %D for dimension %D space", height, dim);}
157120cf1dd8SToby Isaac   if (!fe->subspaces) {ierr = PetscCalloc1(dim, &fe->subspaces);CHKERRQ(ierr);}
157220cf1dd8SToby Isaac   if (height <= dim) {
157320cf1dd8SToby Isaac     if (!fe->subspaces[height-1]) {
157420cf1dd8SToby Isaac       PetscFE     sub;
15753f6b16c7SMatthew G. Knepley       const char *name;
157620cf1dd8SToby Isaac 
157720cf1dd8SToby Isaac       ierr = PetscSpaceGetHeightSubspace(P, height, &subP);CHKERRQ(ierr);
157820cf1dd8SToby Isaac       ierr = PetscDualSpaceGetHeightSubspace(Q, height, &subQ);CHKERRQ(ierr);
157920cf1dd8SToby Isaac       ierr = PetscFECreate(PetscObjectComm((PetscObject) fe), &sub);CHKERRQ(ierr);
15803f6b16c7SMatthew G. Knepley       ierr = PetscObjectGetName((PetscObject) fe,  &name);CHKERRQ(ierr);
15813f6b16c7SMatthew G. Knepley       ierr = PetscObjectSetName((PetscObject) sub,  name);CHKERRQ(ierr);
158220cf1dd8SToby Isaac       ierr = PetscFEGetType(fe, &fetype);CHKERRQ(ierr);
158320cf1dd8SToby Isaac       ierr = PetscFESetType(sub, fetype);CHKERRQ(ierr);
158420cf1dd8SToby Isaac       ierr = PetscFESetBasisSpace(sub, subP);CHKERRQ(ierr);
158520cf1dd8SToby Isaac       ierr = PetscFESetDualSpace(sub, subQ);CHKERRQ(ierr);
158620cf1dd8SToby Isaac       ierr = PetscFESetNumComponents(sub, Nc);CHKERRQ(ierr);
158720cf1dd8SToby Isaac       ierr = PetscFESetUp(sub);CHKERRQ(ierr);
158820cf1dd8SToby Isaac       ierr = PetscFESetQuadrature(sub, subq);CHKERRQ(ierr);
158920cf1dd8SToby Isaac       fe->subspaces[height-1] = sub;
159020cf1dd8SToby Isaac     }
159120cf1dd8SToby Isaac     *subfe = fe->subspaces[height-1];
159220cf1dd8SToby Isaac   } else {
159320cf1dd8SToby Isaac     *subfe = NULL;
159420cf1dd8SToby Isaac   }
159520cf1dd8SToby Isaac   PetscFunctionReturn(0);
159620cf1dd8SToby Isaac }
159720cf1dd8SToby Isaac 
159820cf1dd8SToby Isaac /*@
159920cf1dd8SToby Isaac   PetscFERefine - Create a "refined" PetscFE object that refines the reference cell into smaller copies. This is typically used
160020cf1dd8SToby Isaac   to precondition a higher order method with a lower order method on a refined mesh having the same number of dofs (but more
160120cf1dd8SToby Isaac   sparsity). It is also used to create an interpolation between regularly refined meshes.
160220cf1dd8SToby Isaac 
1603d083f849SBarry Smith   Collective on fem
160420cf1dd8SToby Isaac 
160520cf1dd8SToby Isaac   Input Parameter:
160620cf1dd8SToby Isaac . fe - The initial PetscFE
160720cf1dd8SToby Isaac 
160820cf1dd8SToby Isaac   Output Parameter:
160920cf1dd8SToby Isaac . feRef - The refined PetscFE
161020cf1dd8SToby Isaac 
16112b99622eSMatthew G. Knepley   Level: advanced
161220cf1dd8SToby Isaac 
161320cf1dd8SToby Isaac .seealso: PetscFEType, PetscFECreate(), PetscFESetType()
161420cf1dd8SToby Isaac @*/
161520cf1dd8SToby Isaac PetscErrorCode PetscFERefine(PetscFE fe, PetscFE *feRef)
161620cf1dd8SToby Isaac {
161720cf1dd8SToby Isaac   PetscSpace       P, Pref;
161820cf1dd8SToby Isaac   PetscDualSpace   Q, Qref;
161920cf1dd8SToby Isaac   DM               K, Kref;
162020cf1dd8SToby Isaac   PetscQuadrature  q, qref;
162120cf1dd8SToby Isaac   const PetscReal *v0, *jac;
162220cf1dd8SToby Isaac   PetscInt         numComp, numSubelements;
162320cf1dd8SToby Isaac   PetscErrorCode   ierr;
162420cf1dd8SToby Isaac 
162520cf1dd8SToby Isaac   PetscFunctionBegin;
162620cf1dd8SToby Isaac   ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr);
162720cf1dd8SToby Isaac   ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr);
162820cf1dd8SToby Isaac   ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr);
162920cf1dd8SToby Isaac   ierr = PetscDualSpaceGetDM(Q, &K);CHKERRQ(ierr);
163020cf1dd8SToby Isaac   /* Create space */
163120cf1dd8SToby Isaac   ierr = PetscObjectReference((PetscObject) P);CHKERRQ(ierr);
163220cf1dd8SToby Isaac   Pref = P;
163320cf1dd8SToby Isaac   /* Create dual space */
163420cf1dd8SToby Isaac   ierr = PetscDualSpaceDuplicate(Q, &Qref);CHKERRQ(ierr);
163520cf1dd8SToby Isaac   ierr = DMRefine(K, PetscObjectComm((PetscObject) fe), &Kref);CHKERRQ(ierr);
163620cf1dd8SToby Isaac   ierr = PetscDualSpaceSetDM(Qref, Kref);CHKERRQ(ierr);
163720cf1dd8SToby Isaac   ierr = DMDestroy(&Kref);CHKERRQ(ierr);
163820cf1dd8SToby Isaac   ierr = PetscDualSpaceSetUp(Qref);CHKERRQ(ierr);
163920cf1dd8SToby Isaac   /* Create element */
164020cf1dd8SToby Isaac   ierr = PetscFECreate(PetscObjectComm((PetscObject) fe), feRef);CHKERRQ(ierr);
164120cf1dd8SToby Isaac   ierr = PetscFESetType(*feRef, PETSCFECOMPOSITE);CHKERRQ(ierr);
164220cf1dd8SToby Isaac   ierr = PetscFESetBasisSpace(*feRef, Pref);CHKERRQ(ierr);
164320cf1dd8SToby Isaac   ierr = PetscFESetDualSpace(*feRef, Qref);CHKERRQ(ierr);
164420cf1dd8SToby Isaac   ierr = PetscFEGetNumComponents(fe,    &numComp);CHKERRQ(ierr);
164520cf1dd8SToby Isaac   ierr = PetscFESetNumComponents(*feRef, numComp);CHKERRQ(ierr);
164620cf1dd8SToby Isaac   ierr = PetscFESetUp(*feRef);CHKERRQ(ierr);
164720cf1dd8SToby Isaac   ierr = PetscSpaceDestroy(&Pref);CHKERRQ(ierr);
164820cf1dd8SToby Isaac   ierr = PetscDualSpaceDestroy(&Qref);CHKERRQ(ierr);
164920cf1dd8SToby Isaac   /* Create quadrature */
165020cf1dd8SToby Isaac   ierr = PetscFECompositeGetMapping(*feRef, &numSubelements, &v0, &jac, NULL);CHKERRQ(ierr);
165120cf1dd8SToby Isaac   ierr = PetscQuadratureExpandComposite(q, numSubelements, v0, jac, &qref);CHKERRQ(ierr);
165220cf1dd8SToby Isaac   ierr = PetscFESetQuadrature(*feRef, qref);CHKERRQ(ierr);
165320cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&qref);CHKERRQ(ierr);
165420cf1dd8SToby Isaac   PetscFunctionReturn(0);
165520cf1dd8SToby Isaac }
165620cf1dd8SToby Isaac 
165720cf1dd8SToby Isaac /*@C
165820cf1dd8SToby Isaac   PetscFECreateDefault - Create a PetscFE for basic FEM computation
165920cf1dd8SToby Isaac 
1660d083f849SBarry Smith   Collective
166120cf1dd8SToby Isaac 
166220cf1dd8SToby Isaac   Input Parameters:
16637be5e748SToby Isaac + comm      - The MPI comm
166420cf1dd8SToby Isaac . dim       - The spatial dimension
166520cf1dd8SToby Isaac . Nc        - The number of components
166620cf1dd8SToby Isaac . isSimplex - Flag for simplex reference cell, otherwise its a tensor product
166720cf1dd8SToby Isaac . prefix    - The options prefix, or NULL
1668727cddd5SJacob Faibussowitsch - qorder    - The quadrature order or PETSC_DETERMINE to use PetscSpace polynomial degree
166920cf1dd8SToby Isaac 
167020cf1dd8SToby Isaac   Output Parameter:
167120cf1dd8SToby Isaac . fem - The PetscFE object
167220cf1dd8SToby Isaac 
167320cf1dd8SToby Isaac   Level: beginner
167420cf1dd8SToby Isaac 
167520cf1dd8SToby Isaac .seealso: PetscFECreate(), PetscSpaceCreate(), PetscDualSpaceCreate()
167620cf1dd8SToby Isaac @*/
16777be5e748SToby Isaac PetscErrorCode PetscFECreateDefault(MPI_Comm comm, PetscInt dim, PetscInt Nc, PetscBool isSimplex, const char prefix[], PetscInt qorder, PetscFE *fem)
167820cf1dd8SToby Isaac {
167920cf1dd8SToby Isaac   PetscQuadrature q, fq;
168020cf1dd8SToby Isaac   DM              K;
168120cf1dd8SToby Isaac   PetscSpace      P;
168220cf1dd8SToby Isaac   PetscDualSpace  Q;
168320cf1dd8SToby Isaac   PetscInt        order, quadPointsPerEdge;
168420cf1dd8SToby Isaac   PetscBool       tensor = isSimplex ? PETSC_FALSE : PETSC_TRUE;
168520cf1dd8SToby Isaac   PetscErrorCode  ierr;
168620cf1dd8SToby Isaac 
168720cf1dd8SToby Isaac   PetscFunctionBegin;
168820cf1dd8SToby Isaac   /* Create space */
16897be5e748SToby Isaac   ierr = PetscSpaceCreate(comm, &P);CHKERRQ(ierr);
169020cf1dd8SToby Isaac   ierr = PetscObjectSetOptionsPrefix((PetscObject) P, prefix);CHKERRQ(ierr);
169120cf1dd8SToby Isaac   ierr = PetscSpacePolynomialSetTensor(P, tensor);CHKERRQ(ierr);
169220cf1dd8SToby Isaac   ierr = PetscSpaceSetNumComponents(P, Nc);CHKERRQ(ierr);
169320cf1dd8SToby Isaac   ierr = PetscSpaceSetNumVariables(P, dim);CHKERRQ(ierr);
1694028afddaSToby Isaac   ierr = PetscSpaceSetFromOptions(P);CHKERRQ(ierr);
169520cf1dd8SToby Isaac   ierr = PetscSpaceSetUp(P);CHKERRQ(ierr);
169620cf1dd8SToby Isaac   ierr = PetscSpaceGetDegree(P, &order, NULL);CHKERRQ(ierr);
169720cf1dd8SToby Isaac   ierr = PetscSpacePolynomialGetTensor(P, &tensor);CHKERRQ(ierr);
169820cf1dd8SToby Isaac   /* Create dual space */
16997be5e748SToby Isaac   ierr = PetscDualSpaceCreate(comm, &Q);CHKERRQ(ierr);
170020cf1dd8SToby Isaac   ierr = PetscDualSpaceSetType(Q,PETSCDUALSPACELAGRANGE);CHKERRQ(ierr);
170120cf1dd8SToby Isaac   ierr = PetscObjectSetOptionsPrefix((PetscObject) Q, prefix);CHKERRQ(ierr);
170220cf1dd8SToby Isaac   ierr = PetscDualSpaceCreateReferenceCell(Q, dim, isSimplex, &K);CHKERRQ(ierr);
170320cf1dd8SToby Isaac   ierr = PetscDualSpaceSetDM(Q, K);CHKERRQ(ierr);
170420cf1dd8SToby Isaac   ierr = DMDestroy(&K);CHKERRQ(ierr);
170520cf1dd8SToby Isaac   ierr = PetscDualSpaceSetNumComponents(Q, Nc);CHKERRQ(ierr);
170620cf1dd8SToby Isaac   ierr = PetscDualSpaceSetOrder(Q, order);CHKERRQ(ierr);
170720cf1dd8SToby Isaac   ierr = PetscDualSpaceLagrangeSetTensor(Q, tensor);CHKERRQ(ierr);
170820cf1dd8SToby Isaac   ierr = PetscDualSpaceSetFromOptions(Q);CHKERRQ(ierr);
170920cf1dd8SToby Isaac   ierr = PetscDualSpaceSetUp(Q);CHKERRQ(ierr);
171020cf1dd8SToby Isaac   /* Create element */
17117be5e748SToby Isaac   ierr = PetscFECreate(comm, fem);CHKERRQ(ierr);
171220cf1dd8SToby Isaac   ierr = PetscObjectSetOptionsPrefix((PetscObject) *fem, prefix);CHKERRQ(ierr);
171320cf1dd8SToby Isaac   ierr = PetscFESetBasisSpace(*fem, P);CHKERRQ(ierr);
171420cf1dd8SToby Isaac   ierr = PetscFESetDualSpace(*fem, Q);CHKERRQ(ierr);
171520cf1dd8SToby Isaac   ierr = PetscFESetNumComponents(*fem, Nc);CHKERRQ(ierr);
171691e89cf0SMatthew G. Knepley   ierr = PetscFESetFromOptions(*fem);CHKERRQ(ierr);
171720cf1dd8SToby Isaac   ierr = PetscFESetUp(*fem);CHKERRQ(ierr);
171820cf1dd8SToby Isaac   ierr = PetscSpaceDestroy(&P);CHKERRQ(ierr);
171920cf1dd8SToby Isaac   ierr = PetscDualSpaceDestroy(&Q);CHKERRQ(ierr);
172020cf1dd8SToby Isaac   /* Create quadrature (with specified order if given) */
172120cf1dd8SToby Isaac   qorder = qorder >= 0 ? qorder : order;
172220cf1dd8SToby Isaac   ierr = PetscObjectOptionsBegin((PetscObject)*fem);CHKERRQ(ierr);
17235a856986SBarry Smith   ierr = PetscOptionsBoundedInt("-petscfe_default_quadrature_order","Quadrature order is one less than quadrature points per edge","PetscFECreateDefault",qorder,&qorder,NULL,0);CHKERRQ(ierr);
172420cf1dd8SToby Isaac   ierr = PetscOptionsEnd();CHKERRQ(ierr);
172520cf1dd8SToby Isaac   quadPointsPerEdge = PetscMax(qorder + 1,1);
172620cf1dd8SToby Isaac   if (isSimplex) {
172720cf1dd8SToby Isaac     ierr = PetscDTGaussJacobiQuadrature(dim,   1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr);
172820cf1dd8SToby Isaac     ierr = PetscDTGaussJacobiQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr);
17294ccfa306SStefano Zampini   } else {
173020cf1dd8SToby Isaac     ierr = PetscDTGaussTensorQuadrature(dim,   1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr);
173120cf1dd8SToby Isaac     ierr = PetscDTGaussTensorQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr);
173220cf1dd8SToby Isaac   }
173320cf1dd8SToby Isaac   ierr = PetscFESetQuadrature(*fem, q);CHKERRQ(ierr);
173420cf1dd8SToby Isaac   ierr = PetscFESetFaceQuadrature(*fem, fq);CHKERRQ(ierr);
173520cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&q);CHKERRQ(ierr);
173620cf1dd8SToby Isaac   ierr = PetscQuadratureDestroy(&fq);CHKERRQ(ierr);
173720cf1dd8SToby Isaac   PetscFunctionReturn(0);
173820cf1dd8SToby Isaac }
17393f6b16c7SMatthew G. Knepley 
17403f6b16c7SMatthew G. Knepley /*@C
17413f6b16c7SMatthew G. Knepley   PetscFESetName - Names the FE and its subobjects
17423f6b16c7SMatthew G. Knepley 
17433f6b16c7SMatthew G. Knepley   Not collective
17443f6b16c7SMatthew G. Knepley 
17453f6b16c7SMatthew G. Knepley   Input Parameters:
17463f6b16c7SMatthew G. Knepley + fe   - The PetscFE
17473f6b16c7SMatthew G. Knepley - name - The name
17483f6b16c7SMatthew G. Knepley 
17492b99622eSMatthew G. Knepley   Level: intermediate
17503f6b16c7SMatthew G. Knepley 
17513f6b16c7SMatthew G. Knepley .seealso: PetscFECreate(), PetscSpaceCreate(), PetscDualSpaceCreate()
17523f6b16c7SMatthew G. Knepley @*/
17533f6b16c7SMatthew G. Knepley PetscErrorCode PetscFESetName(PetscFE fe, const char name[])
17543f6b16c7SMatthew G. Knepley {
17553f6b16c7SMatthew G. Knepley   PetscSpace     P;
17563f6b16c7SMatthew G. Knepley   PetscDualSpace Q;
17573f6b16c7SMatthew G. Knepley   PetscErrorCode ierr;
17583f6b16c7SMatthew G. Knepley 
17593f6b16c7SMatthew G. Knepley   PetscFunctionBegin;
17603f6b16c7SMatthew G. Knepley   ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr);
17613f6b16c7SMatthew G. Knepley   ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr);
17623f6b16c7SMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) fe, name);CHKERRQ(ierr);
17633f6b16c7SMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) P,  name);CHKERRQ(ierr);
17643f6b16c7SMatthew G. Knepley   ierr = PetscObjectSetName((PetscObject) Q,  name);CHKERRQ(ierr);
17653f6b16c7SMatthew G. Knepley   PetscFunctionReturn(0);
17663f6b16c7SMatthew G. Knepley }
1767a8f1f9e5SMatthew G. Knepley 
1768a8f1f9e5SMatthew G. Knepley PetscErrorCode PetscFEEvaluateFieldJets_Internal(PetscDS ds, PetscInt dim, PetscInt Nf, const PetscInt Nb[], const PetscInt Nc[], PetscInt q, PetscReal *basisField[], PetscReal *basisFieldDer[], PetscFEGeom *fegeom, const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscScalar u[], PetscScalar u_x[], PetscScalar u_t[])
1769a8f1f9e5SMatthew G. Knepley {
1770a8f1f9e5SMatthew G. Knepley   PetscInt       dOffset = 0, fOffset = 0, f;
1771a8f1f9e5SMatthew G. Knepley   PetscErrorCode ierr;
1772a8f1f9e5SMatthew G. Knepley 
1773a8f1f9e5SMatthew G. Knepley   for (f = 0; f < Nf; ++f) {
1774a8f1f9e5SMatthew G. Knepley     PetscFE          fe;
1775a8f1f9e5SMatthew G. Knepley     const PetscInt   Nbf = Nb[f], Ncf = Nc[f];
1776a8f1f9e5SMatthew G. Knepley     const PetscReal *Bq = &basisField[f][q*Nbf*Ncf];
1777a8f1f9e5SMatthew G. Knepley     const PetscReal *Dq = &basisFieldDer[f][q*Nbf*Ncf*dim];
1778a8f1f9e5SMatthew G. Knepley     PetscInt         b, c, d;
1779a8f1f9e5SMatthew G. Knepley 
1780a8f1f9e5SMatthew G. Knepley     ierr = PetscDSGetDiscretization(ds, f, (PetscObject *) &fe);CHKERRQ(ierr);
1781a8f1f9e5SMatthew G. Knepley     for (c = 0; c < Ncf; ++c)     u[fOffset+c] = 0.0;
1782a8f1f9e5SMatthew G. Knepley     for (d = 0; d < dim*Ncf; ++d) u_x[fOffset*dim+d] = 0.0;
1783a8f1f9e5SMatthew G. Knepley     for (b = 0; b < Nbf; ++b) {
1784a8f1f9e5SMatthew G. Knepley       for (c = 0; c < Ncf; ++c) {
1785a8f1f9e5SMatthew G. Knepley         const PetscInt cidx = b*Ncf+c;
1786a8f1f9e5SMatthew G. Knepley 
1787a8f1f9e5SMatthew G. Knepley         u[fOffset+c] += Bq[cidx]*coefficients[dOffset+b];
1788a8f1f9e5SMatthew G. Knepley         for (d = 0; d < dim; ++d) u_x[(fOffset+c)*dim+d] += Dq[cidx*dim+d]*coefficients[dOffset+b];
1789a8f1f9e5SMatthew G. Knepley       }
1790a8f1f9e5SMatthew G. Knepley     }
1791a8f1f9e5SMatthew G. Knepley     ierr = PetscFEPushforward(fe, fegeom, 1, &u[fOffset]);CHKERRQ(ierr);
1792a8f1f9e5SMatthew G. Knepley     ierr = PetscFEPushforwardGradient(fe, fegeom, 1, &u_x[fOffset*dim]);CHKERRQ(ierr);
1793a8f1f9e5SMatthew G. Knepley     if (u_t) {
1794a8f1f9e5SMatthew G. Knepley       for (c = 0; c < Ncf; ++c) u_t[fOffset+c] = 0.0;
1795a8f1f9e5SMatthew G. Knepley       for (b = 0; b < Nbf; ++b) {
1796a8f1f9e5SMatthew G. Knepley         for (c = 0; c < Ncf; ++c) {
1797a8f1f9e5SMatthew G. Knepley           const PetscInt cidx = b*Ncf+c;
1798a8f1f9e5SMatthew G. Knepley 
1799a8f1f9e5SMatthew G. Knepley           u_t[fOffset+c] += Bq[cidx]*coefficients_t[dOffset+b];
1800a8f1f9e5SMatthew G. Knepley         }
1801a8f1f9e5SMatthew G. Knepley       }
1802a8f1f9e5SMatthew G. Knepley       ierr = PetscFEPushforward(fe, fegeom, 1, &u_t[fOffset]);CHKERRQ(ierr);
1803a8f1f9e5SMatthew G. Knepley     }
1804a8f1f9e5SMatthew G. Knepley     fOffset += Ncf;
1805a8f1f9e5SMatthew G. Knepley     dOffset += Nbf;
1806a8f1f9e5SMatthew G. Knepley   }
1807a8f1f9e5SMatthew G. Knepley   return 0;
1808a8f1f9e5SMatthew G. Knepley }
1809a8f1f9e5SMatthew G. Knepley 
1810a8f1f9e5SMatthew G. Knepley PetscErrorCode PetscFEEvaluateFaceFields_Internal(PetscDS prob, PetscInt field, PetscInt faceLoc, const PetscScalar coefficients[], PetscScalar u[])
1811a8f1f9e5SMatthew G. Knepley {
1812a8f1f9e5SMatthew G. Knepley   PetscFE        fe;
1813a8f1f9e5SMatthew G. Knepley   PetscReal     *faceBasis;
1814a8f1f9e5SMatthew G. Knepley   PetscInt       Nb, Nc, b, c;
1815a8f1f9e5SMatthew G. Knepley   PetscErrorCode ierr;
1816a8f1f9e5SMatthew G. Knepley 
1817a8f1f9e5SMatthew G. Knepley   if (!prob) return 0;
1818a8f1f9e5SMatthew G. Knepley   ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr);
1819a8f1f9e5SMatthew G. Knepley   ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr);
1820a8f1f9e5SMatthew G. Knepley   ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
1821a8f1f9e5SMatthew G. Knepley   ierr = PetscFEGetFaceCentroidTabulation(fe, &faceBasis);CHKERRQ(ierr);
1822a8f1f9e5SMatthew G. Knepley   for (c = 0; c < Nc; ++c) {u[c] = 0.0;}
1823a8f1f9e5SMatthew G. Knepley   for (b = 0; b < Nb; ++b) {
1824a8f1f9e5SMatthew G. Knepley     for (c = 0; c < Nc; ++c) {
1825a8f1f9e5SMatthew G. Knepley       const PetscInt cidx = b*Nc+c;
1826a8f1f9e5SMatthew G. Knepley 
1827a8f1f9e5SMatthew G. Knepley       u[c] += coefficients[cidx]*faceBasis[faceLoc*Nb*Nc+cidx];
1828a8f1f9e5SMatthew G. Knepley     }
1829a8f1f9e5SMatthew G. Knepley   }
1830a8f1f9e5SMatthew G. Knepley   return 0;
1831a8f1f9e5SMatthew G. Knepley }
1832a8f1f9e5SMatthew G. Knepley 
18336142fa51SMatthew G. Knepley PetscErrorCode PetscFEUpdateElementVec_Internal(PetscFE fe, PetscInt dim, PetscInt Nq, PetscInt Nb, PetscInt Nc, PetscReal basis[], PetscReal basisDer[], PetscScalar tmpBasis[], PetscScalar tmpBasisDer[], PetscFEGeom *fegeom, PetscScalar f0[], PetscScalar f1[], PetscScalar elemVec[])
1834a8f1f9e5SMatthew G. Knepley {
1835a8f1f9e5SMatthew G. Knepley   PetscInt       q, b, c, d;
1836a8f1f9e5SMatthew G. Knepley   PetscErrorCode ierr;
1837a8f1f9e5SMatthew G. Knepley 
1838a8f1f9e5SMatthew G. Knepley   for (b = 0; b < Nb; ++b) elemVec[b] = 0.0;
1839a8f1f9e5SMatthew G. Knepley   for (q = 0; q < Nq; ++q) {
1840a8f1f9e5SMatthew G. Knepley     for (b = 0; b < Nb; ++b) {
1841a8f1f9e5SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
1842a8f1f9e5SMatthew G. Knepley         const PetscInt bcidx = b*Nc+c;
1843a8f1f9e5SMatthew G. Knepley 
1844a8f1f9e5SMatthew G. Knepley         tmpBasis[bcidx] = basis[q*Nb*Nc+bcidx];
1845a8f1f9e5SMatthew G. Knepley         for (d = 0; d < dim; ++d) tmpBasisDer[bcidx*dim+d] = basisDer[q*Nb*Nc*dim+bcidx*dim+d];
1846a8f1f9e5SMatthew G. Knepley       }
1847a8f1f9e5SMatthew G. Knepley     }
1848a8f1f9e5SMatthew G. Knepley     ierr = PetscFEPushforward(fe, fegeom, Nb, tmpBasis);CHKERRQ(ierr);
1849a8f1f9e5SMatthew G. Knepley     ierr = PetscFEPushforwardGradient(fe, fegeom, Nb, tmpBasisDer);CHKERRQ(ierr);
1850a8f1f9e5SMatthew G. Knepley     for (b = 0; b < Nb; ++b) {
1851a8f1f9e5SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
1852a8f1f9e5SMatthew G. Knepley         const PetscInt bcidx = b*Nc+c;
1853a8f1f9e5SMatthew G. Knepley         const PetscInt qcidx = q*Nc+c;
1854a8f1f9e5SMatthew G. Knepley 
1855a8f1f9e5SMatthew G. Knepley         elemVec[b] += tmpBasis[bcidx]*f0[qcidx];
1856a8f1f9e5SMatthew G. Knepley         for (d = 0; d < dim; ++d) elemVec[b] += tmpBasisDer[bcidx*dim+d]*f1[qcidx*dim+d];
1857a8f1f9e5SMatthew G. Knepley       }
1858a8f1f9e5SMatthew G. Knepley     }
1859a8f1f9e5SMatthew G. Knepley   }
1860a8f1f9e5SMatthew G. Knepley   return(0);
1861a8f1f9e5SMatthew G. Knepley }
1862a8f1f9e5SMatthew G. Knepley 
18636142fa51SMatthew G. Knepley PetscErrorCode PetscFEUpdateElementMat_Internal(PetscFE feI, PetscFE feJ, PetscInt dim, PetscInt NbI, PetscInt NcI, const PetscReal basisI[], const PetscReal basisDerI[], PetscScalar tmpBasisI[], PetscScalar tmpBasisDerI[], PetscInt NbJ, PetscInt NcJ, const PetscReal basisJ[], const PetscReal basisDerJ[], PetscScalar tmpBasisJ[], PetscScalar tmpBasisDerJ[], PetscFEGeom *fegeom, const PetscScalar g0[], const PetscScalar g1[], const PetscScalar g2[], const PetscScalar g3[], PetscInt eOffset, PetscInt totDim, PetscInt offsetI, PetscInt offsetJ, PetscScalar elemMat[])
1864a8f1f9e5SMatthew G. Knepley {
1865a8f1f9e5SMatthew G. Knepley   PetscInt       f, fc, g, gc, df, dg;
1866a8f1f9e5SMatthew G. Knepley   PetscErrorCode ierr;
1867a8f1f9e5SMatthew G. Knepley 
1868a8f1f9e5SMatthew G. Knepley   for (f = 0; f < NbI; ++f) {
1869a8f1f9e5SMatthew G. Knepley     for (fc = 0; fc < NcI; ++fc) {
1870a8f1f9e5SMatthew G. Knepley       const PetscInt fidx = f*NcI+fc; /* Test function basis index */
1871a8f1f9e5SMatthew G. Knepley 
1872a8f1f9e5SMatthew G. Knepley       tmpBasisI[fidx] = basisI[fidx];
1873a8f1f9e5SMatthew G. Knepley       for (df = 0; df < dim; ++df) tmpBasisDerI[fidx*dim+df] = basisDerI[fidx*dim+df];
1874a8f1f9e5SMatthew G. Knepley     }
1875a8f1f9e5SMatthew G. Knepley   }
1876a8f1f9e5SMatthew G. Knepley   ierr = PetscFEPushforward(feI, fegeom, NbI, tmpBasisI);CHKERRQ(ierr);
1877a8f1f9e5SMatthew G. Knepley   ierr = PetscFEPushforwardGradient(feI, fegeom, NbI, tmpBasisDerI);CHKERRQ(ierr);
1878a8f1f9e5SMatthew G. Knepley   for (g = 0; g < NbJ; ++g) {
1879a8f1f9e5SMatthew G. Knepley     for (gc = 0; gc < NcJ; ++gc) {
1880a8f1f9e5SMatthew G. Knepley       const PetscInt gidx = g*NcJ+gc; /* Trial function basis index */
1881a8f1f9e5SMatthew G. Knepley 
1882a8f1f9e5SMatthew G. Knepley       tmpBasisJ[gidx] = basisJ[gidx];
1883a8f1f9e5SMatthew G. Knepley       for (dg = 0; dg < dim; ++dg) tmpBasisDerJ[gidx*dim+dg] = basisDerJ[gidx*dim+dg];
1884a8f1f9e5SMatthew G. Knepley     }
1885a8f1f9e5SMatthew G. Knepley   }
1886a8f1f9e5SMatthew G. Knepley   ierr = PetscFEPushforward(feJ, fegeom, NbJ, tmpBasisJ);CHKERRQ(ierr);
1887a8f1f9e5SMatthew G. Knepley   ierr = PetscFEPushforwardGradient(feJ, fegeom, NbJ, tmpBasisDerJ);CHKERRQ(ierr);
1888a8f1f9e5SMatthew G. Knepley   for (f = 0; f < NbI; ++f) {
1889a8f1f9e5SMatthew G. Knepley     for (fc = 0; fc < NcI; ++fc) {
1890a8f1f9e5SMatthew G. Knepley       const PetscInt fidx = f*NcI+fc; /* Test function basis index */
1891a8f1f9e5SMatthew G. Knepley       const PetscInt i    = offsetI+f; /* Element matrix row */
1892a8f1f9e5SMatthew G. Knepley       for (g = 0; g < NbJ; ++g) {
1893a8f1f9e5SMatthew G. Knepley         for (gc = 0; gc < NcJ; ++gc) {
1894a8f1f9e5SMatthew G. Knepley           const PetscInt gidx = g*NcJ+gc; /* Trial function basis index */
1895a8f1f9e5SMatthew G. Knepley           const PetscInt j    = offsetJ+g; /* Element matrix column */
1896a8f1f9e5SMatthew G. Knepley           const PetscInt fOff = eOffset+i*totDim+j;
1897a8f1f9e5SMatthew G. Knepley 
1898a8f1f9e5SMatthew G. Knepley           elemMat[fOff] += tmpBasisI[fidx]*g0[fc*NcJ+gc]*tmpBasisJ[gidx];
1899a8f1f9e5SMatthew G. Knepley           for (df = 0; df < dim; ++df) {
1900a8f1f9e5SMatthew G. Knepley             elemMat[fOff] += tmpBasisI[fidx]*g1[(fc*NcJ+gc)*dim+df]*tmpBasisDerJ[gidx*dim+df];
1901a8f1f9e5SMatthew G. Knepley             elemMat[fOff] += tmpBasisDerI[fidx*dim+df]*g2[(fc*NcJ+gc)*dim+df]*tmpBasisJ[gidx];
1902a8f1f9e5SMatthew G. Knepley             for (dg = 0; dg < dim; ++dg) {
1903a8f1f9e5SMatthew G. Knepley               elemMat[fOff] += tmpBasisDerI[fidx*dim+df]*g3[((fc*NcJ+gc)*dim+df)*dim+dg]*tmpBasisDerJ[gidx*dim+dg];
1904a8f1f9e5SMatthew G. Knepley             }
1905a8f1f9e5SMatthew G. Knepley           }
1906a8f1f9e5SMatthew G. Knepley         }
1907a8f1f9e5SMatthew G. Knepley       }
1908a8f1f9e5SMatthew G. Knepley     }
1909a8f1f9e5SMatthew G. Knepley   }
1910a8f1f9e5SMatthew G. Knepley   return(0);
1911a8f1f9e5SMatthew G. Knepley }
1912c9ba7969SMatthew G. Knepley 
1913c9ba7969SMatthew G. Knepley PetscErrorCode PetscFECreateCellGeometry(PetscFE fe, PetscQuadrature quad, PetscFEGeom *cgeom)
1914c9ba7969SMatthew G. Knepley {
1915c9ba7969SMatthew G. Knepley   PetscDualSpace  dsp;
1916c9ba7969SMatthew G. Knepley   DM              dm;
1917c9ba7969SMatthew G. Knepley   PetscQuadrature quadDef;
1918c9ba7969SMatthew G. Knepley   PetscInt        dim, cdim, Nq;
1919c9ba7969SMatthew G. Knepley   PetscErrorCode  ierr;
1920c9ba7969SMatthew G. Knepley 
1921c9ba7969SMatthew G. Knepley   PetscFunctionBegin;
1922c9ba7969SMatthew G. Knepley   ierr = PetscFEGetDualSpace(fe, &dsp);CHKERRQ(ierr);
1923c9ba7969SMatthew G. Knepley   ierr = PetscDualSpaceGetDM(dsp, &dm);CHKERRQ(ierr);
1924c9ba7969SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1925c9ba7969SMatthew G. Knepley   ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr);
1926c9ba7969SMatthew G. Knepley   ierr = PetscFEGetQuadrature(fe, &quadDef);CHKERRQ(ierr);
1927c9ba7969SMatthew G. Knepley   quad = quad ? quad : quadDef;
1928c9ba7969SMatthew G. Knepley   ierr = PetscQuadratureGetData(quad, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr);
1929c9ba7969SMatthew G. Knepley   ierr = PetscMalloc1(Nq*cdim,      &cgeom->v);CHKERRQ(ierr);
1930c9ba7969SMatthew G. Knepley   ierr = PetscMalloc1(Nq*cdim*cdim, &cgeom->J);CHKERRQ(ierr);
1931c9ba7969SMatthew G. Knepley   ierr = PetscMalloc1(Nq*cdim*cdim, &cgeom->invJ);CHKERRQ(ierr);
1932c9ba7969SMatthew G. Knepley   ierr = PetscMalloc1(Nq,           &cgeom->detJ);CHKERRQ(ierr);
1933c9ba7969SMatthew G. Knepley   cgeom->dim       = dim;
1934c9ba7969SMatthew G. Knepley   cgeom->dimEmbed  = cdim;
1935c9ba7969SMatthew G. Knepley   cgeom->numCells  = 1;
1936c9ba7969SMatthew G. Knepley   cgeom->numPoints = Nq;
1937c9ba7969SMatthew G. Knepley   ierr = DMPlexComputeCellGeometryFEM(dm, 0, quad, cgeom->v, cgeom->J, cgeom->invJ, cgeom->detJ);CHKERRQ(ierr);
1938c9ba7969SMatthew G. Knepley   PetscFunctionReturn(0);
1939c9ba7969SMatthew G. Knepley }
1940c9ba7969SMatthew G. Knepley 
1941c9ba7969SMatthew G. Knepley PetscErrorCode PetscFEDestroyCellGeometry(PetscFE fe, PetscFEGeom *cgeom)
1942c9ba7969SMatthew G. Knepley {
1943c9ba7969SMatthew G. Knepley   PetscErrorCode ierr;
1944c9ba7969SMatthew G. Knepley 
1945c9ba7969SMatthew G. Knepley   PetscFunctionBegin;
1946c9ba7969SMatthew G. Knepley   ierr = PetscFree(cgeom->v);CHKERRQ(ierr);
1947c9ba7969SMatthew G. Knepley   ierr = PetscFree(cgeom->J);CHKERRQ(ierr);
1948c9ba7969SMatthew G. Knepley   ierr = PetscFree(cgeom->invJ);CHKERRQ(ierr);
1949c9ba7969SMatthew G. Knepley   ierr = PetscFree(cgeom->detJ);CHKERRQ(ierr);
1950c9ba7969SMatthew G. Knepley   PetscFunctionReturn(0);
1951c9ba7969SMatthew G. Knepley }
1952