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 167fe2efc57SMark PetscFEViewFromOptions - View from Options 168fe2efc57SMark 169fe2efc57SMark Collective on PetscFE 170fe2efc57SMark 171fe2efc57SMark Input Parameters: 172fe2efc57SMark + A - the PetscFE object 173fe2efc57SMark . obj - Optional object 174fe2efc57SMark - name - command line option 175fe2efc57SMark 176fe2efc57SMark Level: intermediate 177fe2efc57SMark .seealso: PetscFE(), PetscFEView(), PetscObjectViewFromOptions(), PetscFECreate() 178fe2efc57SMark @*/ 179fe2efc57SMark PetscErrorCode PetscFEViewFromOptions(PetscFE A,PetscObject obj,const char name[]) 180fe2efc57SMark { 181fe2efc57SMark PetscErrorCode ierr; 182fe2efc57SMark 183fe2efc57SMark PetscFunctionBegin; 184fe2efc57SMark PetscValidHeaderSpecific(A,PETSCFE_CLASSID,1); 185fe2efc57SMark ierr = PetscObjectViewFromOptions((PetscObject)A,obj,name);CHKERRQ(ierr); 186fe2efc57SMark PetscFunctionReturn(0); 187fe2efc57SMark } 188fe2efc57SMark 189fe2efc57SMark /*@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); 323ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&(*fem)->T);CHKERRQ(ierr); 324ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&(*fem)->Tf);CHKERRQ(ierr); 325ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&(*fem)->Tc);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; 369ef0bb6c7SMatthew G. Knepley f->T = NULL; 370ef0bb6c7SMatthew G. Knepley f->Tf = NULL; 371ef0bb6c7SMatthew G. Knepley f->Tc = NULL; 372580bdb30SBarry Smith ierr = PetscArrayzero(&f->quadrature, 1);CHKERRQ(ierr); 373580bdb30SBarry Smith ierr = PetscArrayzero(&f->faceQuadrature, 1);CHKERRQ(ierr); 37420cf1dd8SToby Isaac f->blockSize = 0; 37520cf1dd8SToby Isaac f->numBlocks = 1; 37620cf1dd8SToby Isaac f->batchSize = 0; 37720cf1dd8SToby Isaac f->numBatches = 1; 37820cf1dd8SToby Isaac 37920cf1dd8SToby Isaac *fem = f; 38020cf1dd8SToby Isaac PetscFunctionReturn(0); 38120cf1dd8SToby Isaac } 38220cf1dd8SToby Isaac 38320cf1dd8SToby Isaac /*@ 38420cf1dd8SToby Isaac PetscFEGetSpatialDimension - Returns the spatial dimension of the element 38520cf1dd8SToby Isaac 38620cf1dd8SToby Isaac Not collective 38720cf1dd8SToby Isaac 38820cf1dd8SToby Isaac Input Parameter: 38920cf1dd8SToby Isaac . fem - The PetscFE object 39020cf1dd8SToby Isaac 39120cf1dd8SToby Isaac Output Parameter: 39220cf1dd8SToby Isaac . dim - The spatial dimension 39320cf1dd8SToby Isaac 39420cf1dd8SToby Isaac Level: intermediate 39520cf1dd8SToby Isaac 39620cf1dd8SToby Isaac .seealso: PetscFECreate() 39720cf1dd8SToby Isaac @*/ 39820cf1dd8SToby Isaac PetscErrorCode PetscFEGetSpatialDimension(PetscFE fem, PetscInt *dim) 39920cf1dd8SToby Isaac { 40020cf1dd8SToby Isaac DM dm; 40120cf1dd8SToby Isaac PetscErrorCode ierr; 40220cf1dd8SToby Isaac 40320cf1dd8SToby Isaac PetscFunctionBegin; 40420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 40520cf1dd8SToby Isaac PetscValidPointer(dim, 2); 40620cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(fem->dualSpace, &dm);CHKERRQ(ierr); 40720cf1dd8SToby Isaac ierr = DMGetDimension(dm, dim);CHKERRQ(ierr); 40820cf1dd8SToby Isaac PetscFunctionReturn(0); 40920cf1dd8SToby Isaac } 41020cf1dd8SToby Isaac 41120cf1dd8SToby Isaac /*@ 41220cf1dd8SToby Isaac PetscFESetNumComponents - Sets the number of components in the element 41320cf1dd8SToby Isaac 41420cf1dd8SToby Isaac Not collective 41520cf1dd8SToby Isaac 41620cf1dd8SToby Isaac Input Parameters: 41720cf1dd8SToby Isaac + fem - The PetscFE object 41820cf1dd8SToby Isaac - comp - The number of field components 41920cf1dd8SToby Isaac 42020cf1dd8SToby Isaac Level: intermediate 42120cf1dd8SToby Isaac 42220cf1dd8SToby Isaac .seealso: PetscFECreate() 42320cf1dd8SToby Isaac @*/ 42420cf1dd8SToby Isaac PetscErrorCode PetscFESetNumComponents(PetscFE fem, PetscInt comp) 42520cf1dd8SToby Isaac { 42620cf1dd8SToby Isaac PetscFunctionBegin; 42720cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 42820cf1dd8SToby Isaac fem->numComponents = comp; 42920cf1dd8SToby Isaac PetscFunctionReturn(0); 43020cf1dd8SToby Isaac } 43120cf1dd8SToby Isaac 43220cf1dd8SToby Isaac /*@ 43320cf1dd8SToby Isaac PetscFEGetNumComponents - Returns the number of components in the element 43420cf1dd8SToby Isaac 43520cf1dd8SToby Isaac Not collective 43620cf1dd8SToby Isaac 43720cf1dd8SToby Isaac Input Parameter: 43820cf1dd8SToby Isaac . fem - The PetscFE object 43920cf1dd8SToby Isaac 44020cf1dd8SToby Isaac Output Parameter: 44120cf1dd8SToby Isaac . comp - The number of field components 44220cf1dd8SToby Isaac 44320cf1dd8SToby Isaac Level: intermediate 44420cf1dd8SToby Isaac 44520cf1dd8SToby Isaac .seealso: PetscFECreate() 44620cf1dd8SToby Isaac @*/ 44720cf1dd8SToby Isaac PetscErrorCode PetscFEGetNumComponents(PetscFE fem, PetscInt *comp) 44820cf1dd8SToby Isaac { 44920cf1dd8SToby Isaac PetscFunctionBegin; 45020cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 45120cf1dd8SToby Isaac PetscValidPointer(comp, 2); 45220cf1dd8SToby Isaac *comp = fem->numComponents; 45320cf1dd8SToby Isaac PetscFunctionReturn(0); 45420cf1dd8SToby Isaac } 45520cf1dd8SToby Isaac 45620cf1dd8SToby Isaac /*@ 45720cf1dd8SToby Isaac PetscFESetTileSizes - Sets the tile sizes for evaluation 45820cf1dd8SToby Isaac 45920cf1dd8SToby Isaac Not collective 46020cf1dd8SToby Isaac 46120cf1dd8SToby Isaac Input Parameters: 46220cf1dd8SToby Isaac + fem - The PetscFE object 46320cf1dd8SToby Isaac . blockSize - The number of elements in a block 46420cf1dd8SToby Isaac . numBlocks - The number of blocks in a batch 46520cf1dd8SToby Isaac . batchSize - The number of elements in a batch 46620cf1dd8SToby Isaac - numBatches - The number of batches in a chunk 46720cf1dd8SToby Isaac 46820cf1dd8SToby Isaac Level: intermediate 46920cf1dd8SToby Isaac 47020cf1dd8SToby Isaac .seealso: PetscFECreate() 47120cf1dd8SToby Isaac @*/ 47220cf1dd8SToby Isaac PetscErrorCode PetscFESetTileSizes(PetscFE fem, PetscInt blockSize, PetscInt numBlocks, PetscInt batchSize, PetscInt numBatches) 47320cf1dd8SToby Isaac { 47420cf1dd8SToby Isaac PetscFunctionBegin; 47520cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 47620cf1dd8SToby Isaac fem->blockSize = blockSize; 47720cf1dd8SToby Isaac fem->numBlocks = numBlocks; 47820cf1dd8SToby Isaac fem->batchSize = batchSize; 47920cf1dd8SToby Isaac fem->numBatches = numBatches; 48020cf1dd8SToby Isaac PetscFunctionReturn(0); 48120cf1dd8SToby Isaac } 48220cf1dd8SToby Isaac 48320cf1dd8SToby Isaac /*@ 48420cf1dd8SToby Isaac PetscFEGetTileSizes - Returns the tile sizes for evaluation 48520cf1dd8SToby Isaac 48620cf1dd8SToby Isaac Not collective 48720cf1dd8SToby Isaac 48820cf1dd8SToby Isaac Input Parameter: 48920cf1dd8SToby Isaac . fem - The PetscFE object 49020cf1dd8SToby Isaac 49120cf1dd8SToby Isaac Output Parameters: 49220cf1dd8SToby Isaac + blockSize - The number of elements in a block 49320cf1dd8SToby Isaac . numBlocks - The number of blocks in a batch 49420cf1dd8SToby Isaac . batchSize - The number of elements in a batch 49520cf1dd8SToby Isaac - numBatches - The number of batches in a chunk 49620cf1dd8SToby Isaac 49720cf1dd8SToby Isaac Level: intermediate 49820cf1dd8SToby Isaac 49920cf1dd8SToby Isaac .seealso: PetscFECreate() 50020cf1dd8SToby Isaac @*/ 50120cf1dd8SToby Isaac PetscErrorCode PetscFEGetTileSizes(PetscFE fem, PetscInt *blockSize, PetscInt *numBlocks, PetscInt *batchSize, PetscInt *numBatches) 50220cf1dd8SToby Isaac { 50320cf1dd8SToby Isaac PetscFunctionBegin; 50420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 50520cf1dd8SToby Isaac if (blockSize) PetscValidPointer(blockSize, 2); 50620cf1dd8SToby Isaac if (numBlocks) PetscValidPointer(numBlocks, 3); 50720cf1dd8SToby Isaac if (batchSize) PetscValidPointer(batchSize, 4); 50820cf1dd8SToby Isaac if (numBatches) PetscValidPointer(numBatches, 5); 50920cf1dd8SToby Isaac if (blockSize) *blockSize = fem->blockSize; 51020cf1dd8SToby Isaac if (numBlocks) *numBlocks = fem->numBlocks; 51120cf1dd8SToby Isaac if (batchSize) *batchSize = fem->batchSize; 51220cf1dd8SToby Isaac if (numBatches) *numBatches = fem->numBatches; 51320cf1dd8SToby Isaac PetscFunctionReturn(0); 51420cf1dd8SToby Isaac } 51520cf1dd8SToby Isaac 51620cf1dd8SToby Isaac /*@ 51720cf1dd8SToby Isaac PetscFEGetBasisSpace - Returns the PetscSpace used for approximation of the solution 51820cf1dd8SToby Isaac 51920cf1dd8SToby Isaac Not collective 52020cf1dd8SToby Isaac 52120cf1dd8SToby Isaac Input Parameter: 52220cf1dd8SToby Isaac . fem - The PetscFE object 52320cf1dd8SToby Isaac 52420cf1dd8SToby Isaac Output Parameter: 52520cf1dd8SToby Isaac . sp - The PetscSpace object 52620cf1dd8SToby Isaac 52720cf1dd8SToby Isaac Level: intermediate 52820cf1dd8SToby Isaac 52920cf1dd8SToby Isaac .seealso: PetscFECreate() 53020cf1dd8SToby Isaac @*/ 53120cf1dd8SToby Isaac PetscErrorCode PetscFEGetBasisSpace(PetscFE fem, PetscSpace *sp) 53220cf1dd8SToby Isaac { 53320cf1dd8SToby Isaac PetscFunctionBegin; 53420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 53520cf1dd8SToby Isaac PetscValidPointer(sp, 2); 53620cf1dd8SToby Isaac *sp = fem->basisSpace; 53720cf1dd8SToby Isaac PetscFunctionReturn(0); 53820cf1dd8SToby Isaac } 53920cf1dd8SToby Isaac 54020cf1dd8SToby Isaac /*@ 54120cf1dd8SToby Isaac PetscFESetBasisSpace - Sets the PetscSpace used for approximation of the solution 54220cf1dd8SToby Isaac 54320cf1dd8SToby Isaac Not collective 54420cf1dd8SToby Isaac 54520cf1dd8SToby Isaac Input Parameters: 54620cf1dd8SToby Isaac + fem - The PetscFE object 54720cf1dd8SToby Isaac - sp - The PetscSpace object 54820cf1dd8SToby Isaac 54920cf1dd8SToby Isaac Level: intermediate 55020cf1dd8SToby Isaac 55120cf1dd8SToby Isaac .seealso: PetscFECreate() 55220cf1dd8SToby Isaac @*/ 55320cf1dd8SToby Isaac PetscErrorCode PetscFESetBasisSpace(PetscFE fem, PetscSpace sp) 55420cf1dd8SToby Isaac { 55520cf1dd8SToby Isaac PetscErrorCode ierr; 55620cf1dd8SToby Isaac 55720cf1dd8SToby Isaac PetscFunctionBegin; 55820cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 55920cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCSPACE_CLASSID, 2); 56020cf1dd8SToby Isaac ierr = PetscSpaceDestroy(&fem->basisSpace);CHKERRQ(ierr); 56120cf1dd8SToby Isaac fem->basisSpace = sp; 56220cf1dd8SToby Isaac ierr = PetscObjectReference((PetscObject) fem->basisSpace);CHKERRQ(ierr); 56320cf1dd8SToby Isaac PetscFunctionReturn(0); 56420cf1dd8SToby Isaac } 56520cf1dd8SToby Isaac 56620cf1dd8SToby Isaac /*@ 56720cf1dd8SToby Isaac PetscFEGetDualSpace - Returns the PetscDualSpace used to define the inner product 56820cf1dd8SToby Isaac 56920cf1dd8SToby Isaac Not collective 57020cf1dd8SToby Isaac 57120cf1dd8SToby Isaac Input Parameter: 57220cf1dd8SToby Isaac . fem - The PetscFE object 57320cf1dd8SToby Isaac 57420cf1dd8SToby Isaac Output Parameter: 57520cf1dd8SToby Isaac . sp - The PetscDualSpace object 57620cf1dd8SToby Isaac 57720cf1dd8SToby Isaac Level: intermediate 57820cf1dd8SToby Isaac 57920cf1dd8SToby Isaac .seealso: PetscFECreate() 58020cf1dd8SToby Isaac @*/ 58120cf1dd8SToby Isaac PetscErrorCode PetscFEGetDualSpace(PetscFE fem, PetscDualSpace *sp) 58220cf1dd8SToby Isaac { 58320cf1dd8SToby Isaac PetscFunctionBegin; 58420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 58520cf1dd8SToby Isaac PetscValidPointer(sp, 2); 58620cf1dd8SToby Isaac *sp = fem->dualSpace; 58720cf1dd8SToby Isaac PetscFunctionReturn(0); 58820cf1dd8SToby Isaac } 58920cf1dd8SToby Isaac 59020cf1dd8SToby Isaac /*@ 59120cf1dd8SToby Isaac PetscFESetDualSpace - Sets the PetscDualSpace used to define the inner product 59220cf1dd8SToby Isaac 59320cf1dd8SToby Isaac Not collective 59420cf1dd8SToby Isaac 59520cf1dd8SToby Isaac Input Parameters: 59620cf1dd8SToby Isaac + fem - The PetscFE object 59720cf1dd8SToby Isaac - sp - The PetscDualSpace object 59820cf1dd8SToby Isaac 59920cf1dd8SToby Isaac Level: intermediate 60020cf1dd8SToby Isaac 60120cf1dd8SToby Isaac .seealso: PetscFECreate() 60220cf1dd8SToby Isaac @*/ 60320cf1dd8SToby Isaac PetscErrorCode PetscFESetDualSpace(PetscFE fem, PetscDualSpace sp) 60420cf1dd8SToby Isaac { 60520cf1dd8SToby Isaac PetscErrorCode ierr; 60620cf1dd8SToby Isaac 60720cf1dd8SToby Isaac PetscFunctionBegin; 60820cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 60920cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 2); 61020cf1dd8SToby Isaac ierr = PetscDualSpaceDestroy(&fem->dualSpace);CHKERRQ(ierr); 61120cf1dd8SToby Isaac fem->dualSpace = sp; 61220cf1dd8SToby Isaac ierr = PetscObjectReference((PetscObject) fem->dualSpace);CHKERRQ(ierr); 61320cf1dd8SToby Isaac PetscFunctionReturn(0); 61420cf1dd8SToby Isaac } 61520cf1dd8SToby Isaac 61620cf1dd8SToby Isaac /*@ 61720cf1dd8SToby Isaac PetscFEGetQuadrature - Returns the PetscQuadrature used to calculate inner products 61820cf1dd8SToby Isaac 61920cf1dd8SToby Isaac Not collective 62020cf1dd8SToby Isaac 62120cf1dd8SToby Isaac Input Parameter: 62220cf1dd8SToby Isaac . fem - The PetscFE object 62320cf1dd8SToby Isaac 62420cf1dd8SToby Isaac Output Parameter: 62520cf1dd8SToby Isaac . q - The PetscQuadrature object 62620cf1dd8SToby Isaac 62720cf1dd8SToby Isaac Level: intermediate 62820cf1dd8SToby Isaac 62920cf1dd8SToby Isaac .seealso: PetscFECreate() 63020cf1dd8SToby Isaac @*/ 63120cf1dd8SToby Isaac PetscErrorCode PetscFEGetQuadrature(PetscFE fem, PetscQuadrature *q) 63220cf1dd8SToby Isaac { 63320cf1dd8SToby Isaac PetscFunctionBegin; 63420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 63520cf1dd8SToby Isaac PetscValidPointer(q, 2); 63620cf1dd8SToby Isaac *q = fem->quadrature; 63720cf1dd8SToby Isaac PetscFunctionReturn(0); 63820cf1dd8SToby Isaac } 63920cf1dd8SToby Isaac 64020cf1dd8SToby Isaac /*@ 64120cf1dd8SToby Isaac PetscFESetQuadrature - Sets the PetscQuadrature used to calculate inner products 64220cf1dd8SToby Isaac 64320cf1dd8SToby Isaac Not collective 64420cf1dd8SToby Isaac 64520cf1dd8SToby Isaac Input Parameters: 64620cf1dd8SToby Isaac + fem - The PetscFE object 64720cf1dd8SToby Isaac - q - The PetscQuadrature object 64820cf1dd8SToby Isaac 64920cf1dd8SToby Isaac Level: intermediate 65020cf1dd8SToby Isaac 65120cf1dd8SToby Isaac .seealso: PetscFECreate() 65220cf1dd8SToby Isaac @*/ 65320cf1dd8SToby Isaac PetscErrorCode PetscFESetQuadrature(PetscFE fem, PetscQuadrature q) 65420cf1dd8SToby Isaac { 65520cf1dd8SToby Isaac PetscInt Nc, qNc; 65620cf1dd8SToby Isaac PetscErrorCode ierr; 65720cf1dd8SToby Isaac 65820cf1dd8SToby Isaac PetscFunctionBegin; 65920cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 660*fd2fdbddSMatthew G. Knepley if (q == fem->quadrature) PetscFunctionReturn(0); 66120cf1dd8SToby Isaac ierr = PetscFEGetNumComponents(fem, &Nc);CHKERRQ(ierr); 66220cf1dd8SToby Isaac ierr = PetscQuadratureGetNumComponents(q, &qNc);CHKERRQ(ierr); 66320cf1dd8SToby 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); 664ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&fem->T);CHKERRQ(ierr); 665ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&fem->Tc);CHKERRQ(ierr); 666*fd2fdbddSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) q);CHKERRQ(ierr); 66720cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&fem->quadrature);CHKERRQ(ierr); 66820cf1dd8SToby Isaac fem->quadrature = q; 66920cf1dd8SToby Isaac PetscFunctionReturn(0); 67020cf1dd8SToby Isaac } 67120cf1dd8SToby Isaac 67220cf1dd8SToby Isaac /*@ 67320cf1dd8SToby Isaac PetscFEGetFaceQuadrature - Returns the PetscQuadrature used to calculate inner products on faces 67420cf1dd8SToby Isaac 67520cf1dd8SToby Isaac Not collective 67620cf1dd8SToby Isaac 67720cf1dd8SToby Isaac Input Parameter: 67820cf1dd8SToby Isaac . fem - The PetscFE object 67920cf1dd8SToby Isaac 68020cf1dd8SToby Isaac Output Parameter: 68120cf1dd8SToby Isaac . q - The PetscQuadrature object 68220cf1dd8SToby Isaac 68320cf1dd8SToby Isaac Level: intermediate 68420cf1dd8SToby Isaac 68520cf1dd8SToby Isaac .seealso: PetscFECreate() 68620cf1dd8SToby Isaac @*/ 68720cf1dd8SToby Isaac PetscErrorCode PetscFEGetFaceQuadrature(PetscFE fem, PetscQuadrature *q) 68820cf1dd8SToby Isaac { 68920cf1dd8SToby Isaac PetscFunctionBegin; 69020cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 69120cf1dd8SToby Isaac PetscValidPointer(q, 2); 69220cf1dd8SToby Isaac *q = fem->faceQuadrature; 69320cf1dd8SToby Isaac PetscFunctionReturn(0); 69420cf1dd8SToby Isaac } 69520cf1dd8SToby Isaac 69620cf1dd8SToby Isaac /*@ 69720cf1dd8SToby Isaac PetscFESetFaceQuadrature - Sets the PetscQuadrature used to calculate inner products on faces 69820cf1dd8SToby Isaac 69920cf1dd8SToby Isaac Not collective 70020cf1dd8SToby Isaac 70120cf1dd8SToby Isaac Input Parameters: 70220cf1dd8SToby Isaac + fem - The PetscFE object 70320cf1dd8SToby Isaac - q - The PetscQuadrature object 70420cf1dd8SToby Isaac 70520cf1dd8SToby Isaac Level: intermediate 70620cf1dd8SToby Isaac 70720cf1dd8SToby Isaac .seealso: PetscFECreate() 70820cf1dd8SToby Isaac @*/ 70920cf1dd8SToby Isaac PetscErrorCode PetscFESetFaceQuadrature(PetscFE fem, PetscQuadrature q) 71020cf1dd8SToby Isaac { 711ef0bb6c7SMatthew G. Knepley PetscInt Nc, qNc; 71220cf1dd8SToby Isaac PetscErrorCode ierr; 71320cf1dd8SToby Isaac 71420cf1dd8SToby Isaac PetscFunctionBegin; 71520cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 716ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetNumComponents(fem, &Nc);CHKERRQ(ierr); 717ef0bb6c7SMatthew G. Knepley ierr = PetscQuadratureGetNumComponents(q, &qNc);CHKERRQ(ierr); 718ef0bb6c7SMatthew G. Knepley 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); 719ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&fem->Tf);CHKERRQ(ierr); 72020cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&fem->faceQuadrature);CHKERRQ(ierr); 72120cf1dd8SToby Isaac fem->faceQuadrature = q; 72220cf1dd8SToby Isaac ierr = PetscObjectReference((PetscObject) q);CHKERRQ(ierr); 72320cf1dd8SToby Isaac PetscFunctionReturn(0); 72420cf1dd8SToby Isaac } 72520cf1dd8SToby Isaac 7265dc5c000SMatthew G. Knepley /*@ 7275dc5c000SMatthew G. Knepley PetscFECopyQuadrature - Copy both volumetric and surface quadrature 7285dc5c000SMatthew G. Knepley 7295dc5c000SMatthew G. Knepley Not collective 7305dc5c000SMatthew G. Knepley 7315dc5c000SMatthew G. Knepley Input Parameters: 7325dc5c000SMatthew G. Knepley + sfe - The PetscFE source for the quadratures 7335dc5c000SMatthew G. Knepley - tfe - The PetscFE target for the quadratures 7345dc5c000SMatthew G. Knepley 7355dc5c000SMatthew G. Knepley Level: intermediate 7365dc5c000SMatthew G. Knepley 7375dc5c000SMatthew G. Knepley .seealso: PetscFECreate(), PetscFESetQuadrature(), PetscFESetFaceQuadrature() 7385dc5c000SMatthew G. Knepley @*/ 7395dc5c000SMatthew G. Knepley PetscErrorCode PetscFECopyQuadrature(PetscFE sfe, PetscFE tfe) 7405dc5c000SMatthew G. Knepley { 7415dc5c000SMatthew G. Knepley PetscQuadrature q; 7425dc5c000SMatthew G. Knepley PetscErrorCode ierr; 7435dc5c000SMatthew G. Knepley 7445dc5c000SMatthew G. Knepley PetscFunctionBegin; 7455dc5c000SMatthew G. Knepley PetscValidHeaderSpecific(sfe, PETSCFE_CLASSID, 1); 7465dc5c000SMatthew G. Knepley PetscValidHeaderSpecific(tfe, PETSCFE_CLASSID, 2); 7475dc5c000SMatthew G. Knepley ierr = PetscFEGetQuadrature(sfe, &q);CHKERRQ(ierr); 7485dc5c000SMatthew G. Knepley ierr = PetscFESetQuadrature(tfe, q);CHKERRQ(ierr); 7495dc5c000SMatthew G. Knepley ierr = PetscFEGetFaceQuadrature(sfe, &q);CHKERRQ(ierr); 7505dc5c000SMatthew G. Knepley ierr = PetscFESetFaceQuadrature(tfe, q);CHKERRQ(ierr); 7515dc5c000SMatthew G. Knepley PetscFunctionReturn(0); 7525dc5c000SMatthew G. Knepley } 7535dc5c000SMatthew G. Knepley 75420cf1dd8SToby Isaac /*@C 75520cf1dd8SToby Isaac PetscFEGetNumDof - Returns the number of dofs (dual basis vectors) associated to mesh points on the reference cell of a given dimension 75620cf1dd8SToby Isaac 75720cf1dd8SToby Isaac Not collective 75820cf1dd8SToby Isaac 75920cf1dd8SToby Isaac Input Parameter: 76020cf1dd8SToby Isaac . fem - The PetscFE object 76120cf1dd8SToby Isaac 76220cf1dd8SToby Isaac Output Parameter: 76320cf1dd8SToby Isaac . numDof - Array with the number of dofs per dimension 76420cf1dd8SToby Isaac 76520cf1dd8SToby Isaac Level: intermediate 76620cf1dd8SToby Isaac 76720cf1dd8SToby Isaac .seealso: PetscFECreate() 76820cf1dd8SToby Isaac @*/ 76920cf1dd8SToby Isaac PetscErrorCode PetscFEGetNumDof(PetscFE fem, const PetscInt **numDof) 77020cf1dd8SToby Isaac { 77120cf1dd8SToby Isaac PetscErrorCode ierr; 77220cf1dd8SToby Isaac 77320cf1dd8SToby Isaac PetscFunctionBegin; 77420cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 77520cf1dd8SToby Isaac PetscValidPointer(numDof, 2); 77620cf1dd8SToby Isaac ierr = PetscDualSpaceGetNumDof(fem->dualSpace, numDof);CHKERRQ(ierr); 77720cf1dd8SToby Isaac PetscFunctionReturn(0); 77820cf1dd8SToby Isaac } 77920cf1dd8SToby Isaac 78020cf1dd8SToby Isaac /*@C 781ef0bb6c7SMatthew G. Knepley PetscFEGetCellTabulation - Returns the tabulation of the basis functions at the quadrature points on the reference cell 78220cf1dd8SToby Isaac 78320cf1dd8SToby Isaac Not collective 78420cf1dd8SToby Isaac 78520cf1dd8SToby Isaac Input Parameter: 78620cf1dd8SToby Isaac . fem - The PetscFE object 78720cf1dd8SToby Isaac 788ef0bb6c7SMatthew G. Knepley Output Parameter: 789ef0bb6c7SMatthew G. Knepley . T - The basis function values and derivatives at quadrature points 79020cf1dd8SToby Isaac 79120cf1dd8SToby Isaac Note: 792ef0bb6c7SMatthew G. Knepley $ T->T[0] = B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c 793ef0bb6c7SMatthew G. Knepley $ T->T[1] = D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d 794ef0bb6c7SMatthew G. Knepley $ T->T[2] = 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 79520cf1dd8SToby Isaac 79620cf1dd8SToby Isaac Level: intermediate 79720cf1dd8SToby Isaac 798ef0bb6c7SMatthew G. Knepley .seealso: PetscFECreateTabulation(), PetscTabulationDestroy() 79920cf1dd8SToby Isaac @*/ 800ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEGetCellTabulation(PetscFE fem, PetscTabulation *T) 80120cf1dd8SToby Isaac { 80220cf1dd8SToby Isaac PetscInt npoints; 80320cf1dd8SToby Isaac const PetscReal *points; 80420cf1dd8SToby Isaac PetscErrorCode ierr; 80520cf1dd8SToby Isaac 80620cf1dd8SToby Isaac PetscFunctionBegin; 80720cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 808ef0bb6c7SMatthew G. Knepley PetscValidPointer(T, 2); 80920cf1dd8SToby Isaac ierr = PetscQuadratureGetData(fem->quadrature, NULL, NULL, &npoints, &points, NULL);CHKERRQ(ierr); 810ef0bb6c7SMatthew G. Knepley if (!fem->T) {ierr = PetscFECreateTabulation(fem, 1, npoints, points, 1, &fem->T);CHKERRQ(ierr);} 811ef0bb6c7SMatthew G. Knepley *T = fem->T; 81220cf1dd8SToby Isaac PetscFunctionReturn(0); 81320cf1dd8SToby Isaac } 81420cf1dd8SToby Isaac 8152b99622eSMatthew G. Knepley /*@C 816ef0bb6c7SMatthew G. Knepley PetscFEGetFaceTabulation - Returns the tabulation of the basis functions at the face quadrature points for each face of the reference cell 8172b99622eSMatthew G. Knepley 8182b99622eSMatthew G. Knepley Not collective 8192b99622eSMatthew G. Knepley 8202b99622eSMatthew G. Knepley Input Parameter: 8212b99622eSMatthew G. Knepley . fem - The PetscFE object 8222b99622eSMatthew G. Knepley 8232b99622eSMatthew G. Knepley Output Parameters: 824ef0bb6c7SMatthew G. Knepley . Tf - The basis function values and derviatives at face quadrature points 8252b99622eSMatthew G. Knepley 8262b99622eSMatthew G. Knepley Note: 827ef0bb6c7SMatthew G. Knepley $ T->T[0] = Bf[((f*Nq + q)*pdim + i)*Nc + c] is the value at point f,q for basis function i and component c 828ef0bb6c7SMatthew G. Knepley $ T->T[1] = 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 829ef0bb6c7SMatthew G. Knepley $ T->T[2] = 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 8302b99622eSMatthew G. Knepley 8312b99622eSMatthew G. Knepley Level: intermediate 8322b99622eSMatthew G. Knepley 833ef0bb6c7SMatthew G. Knepley .seealso: PetscFEGetCellTabulation(), PetscFECreateTabulation(), PetscTabulationDestroy() 8342b99622eSMatthew G. Knepley @*/ 835ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEGetFaceTabulation(PetscFE fem, PetscTabulation *Tf) 83620cf1dd8SToby Isaac { 83720cf1dd8SToby Isaac PetscErrorCode ierr; 83820cf1dd8SToby Isaac 83920cf1dd8SToby Isaac PetscFunctionBegin; 84020cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 841ef0bb6c7SMatthew G. Knepley PetscValidPointer(Tf, 2); 842ef0bb6c7SMatthew G. Knepley if (!fem->Tf) { 84320cf1dd8SToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 84420cf1dd8SToby Isaac PetscReal v0[3], J[9], detJ; 84520cf1dd8SToby Isaac PetscQuadrature fq; 84620cf1dd8SToby Isaac PetscDualSpace sp; 84720cf1dd8SToby Isaac DM dm; 84820cf1dd8SToby Isaac const PetscInt *faces; 84920cf1dd8SToby Isaac PetscInt dim, numFaces, f, npoints, q; 85020cf1dd8SToby Isaac const PetscReal *points; 85120cf1dd8SToby Isaac PetscReal *facePoints; 85220cf1dd8SToby Isaac 85320cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fem, &sp);CHKERRQ(ierr); 85420cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr); 85520cf1dd8SToby Isaac ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 85620cf1dd8SToby Isaac ierr = DMPlexGetConeSize(dm, 0, &numFaces);CHKERRQ(ierr); 85720cf1dd8SToby Isaac ierr = DMPlexGetCone(dm, 0, &faces);CHKERRQ(ierr); 85820cf1dd8SToby Isaac ierr = PetscFEGetFaceQuadrature(fem, &fq);CHKERRQ(ierr); 85920cf1dd8SToby Isaac if (fq) { 86020cf1dd8SToby Isaac ierr = PetscQuadratureGetData(fq, NULL, NULL, &npoints, &points, NULL);CHKERRQ(ierr); 86120cf1dd8SToby Isaac ierr = PetscMalloc1(numFaces*npoints*dim, &facePoints);CHKERRQ(ierr); 86220cf1dd8SToby Isaac for (f = 0; f < numFaces; ++f) { 86320cf1dd8SToby Isaac ierr = DMPlexComputeCellGeometryFEM(dm, faces[f], NULL, v0, J, NULL, &detJ);CHKERRQ(ierr); 86420cf1dd8SToby Isaac for (q = 0; q < npoints; ++q) CoordinatesRefToReal(dim, dim-1, xi0, v0, J, &points[q*(dim-1)], &facePoints[(f*npoints+q)*dim]); 86520cf1dd8SToby Isaac } 866ef0bb6c7SMatthew G. Knepley ierr = PetscFECreateTabulation(fem, numFaces, npoints, facePoints, 1, &fem->Tf);CHKERRQ(ierr); 86720cf1dd8SToby Isaac ierr = PetscFree(facePoints);CHKERRQ(ierr); 86820cf1dd8SToby Isaac } 86920cf1dd8SToby Isaac } 870ef0bb6c7SMatthew G. Knepley *Tf = fem->Tf; 87120cf1dd8SToby Isaac PetscFunctionReturn(0); 87220cf1dd8SToby Isaac } 87320cf1dd8SToby Isaac 8742b99622eSMatthew G. Knepley /*@C 875ef0bb6c7SMatthew G. Knepley PetscFEGetFaceCentroidTabulation - Returns the tabulation of the basis functions at the face centroid points 8762b99622eSMatthew G. Knepley 8772b99622eSMatthew G. Knepley Not collective 8782b99622eSMatthew G. Knepley 8792b99622eSMatthew G. Knepley Input Parameter: 8802b99622eSMatthew G. Knepley . fem - The PetscFE object 8812b99622eSMatthew G. Knepley 8822b99622eSMatthew G. Knepley Output Parameters: 883ef0bb6c7SMatthew G. Knepley . Tc - The basis function values at face centroid points 8842b99622eSMatthew G. Knepley 8852b99622eSMatthew G. Knepley Note: 886ef0bb6c7SMatthew G. Knepley $ T->T[0] = Bf[(f*pdim + i)*Nc + c] is the value at point f for basis function i and component c 8872b99622eSMatthew G. Knepley 8882b99622eSMatthew G. Knepley Level: intermediate 8892b99622eSMatthew G. Knepley 890ef0bb6c7SMatthew G. Knepley .seealso: PetscFEGetFaceTabulation(), PetscFEGetCellTabulation(), PetscFECreateTabulation(), PetscTabulationDestroy() 8912b99622eSMatthew G. Knepley @*/ 892ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEGetFaceCentroidTabulation(PetscFE fem, PetscTabulation *Tc) 89320cf1dd8SToby Isaac { 89420cf1dd8SToby Isaac PetscErrorCode ierr; 89520cf1dd8SToby Isaac 89620cf1dd8SToby Isaac PetscFunctionBegin; 89720cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 898ef0bb6c7SMatthew G. Knepley PetscValidPointer(Tc, 2); 899ef0bb6c7SMatthew G. Knepley if (!fem->Tc) { 90020cf1dd8SToby Isaac PetscDualSpace sp; 90120cf1dd8SToby Isaac DM dm; 90220cf1dd8SToby Isaac const PetscInt *cone; 90320cf1dd8SToby Isaac PetscReal *centroids; 90420cf1dd8SToby Isaac PetscInt dim, numFaces, f; 90520cf1dd8SToby Isaac 90620cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fem, &sp);CHKERRQ(ierr); 90720cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr); 90820cf1dd8SToby Isaac ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 90920cf1dd8SToby Isaac ierr = DMPlexGetConeSize(dm, 0, &numFaces);CHKERRQ(ierr); 91020cf1dd8SToby Isaac ierr = DMPlexGetCone(dm, 0, &cone);CHKERRQ(ierr); 91120cf1dd8SToby Isaac ierr = PetscMalloc1(numFaces*dim, ¢roids);CHKERRQ(ierr); 91220cf1dd8SToby Isaac for (f = 0; f < numFaces; ++f) {ierr = DMPlexComputeCellGeometryFVM(dm, cone[f], NULL, ¢roids[f*dim], NULL);CHKERRQ(ierr);} 913ef0bb6c7SMatthew G. Knepley ierr = PetscFECreateTabulation(fem, 1, numFaces, centroids, 0, &fem->Tc);CHKERRQ(ierr); 91420cf1dd8SToby Isaac ierr = PetscFree(centroids);CHKERRQ(ierr); 91520cf1dd8SToby Isaac } 916ef0bb6c7SMatthew G. Knepley *Tc = fem->Tc; 91720cf1dd8SToby Isaac PetscFunctionReturn(0); 91820cf1dd8SToby Isaac } 91920cf1dd8SToby Isaac 92020cf1dd8SToby Isaac /*@C 921ef0bb6c7SMatthew G. Knepley PetscFECreateTabulation - Tabulates the basis functions, and perhaps derivatives, at the points provided. 92220cf1dd8SToby Isaac 92320cf1dd8SToby Isaac Not collective 92420cf1dd8SToby Isaac 92520cf1dd8SToby Isaac Input Parameters: 92620cf1dd8SToby Isaac + fem - The PetscFE object 927ef0bb6c7SMatthew G. Knepley . nrepl - The number of replicas 928ef0bb6c7SMatthew G. Knepley . npoints - The number of tabulation points in a replica 929ef0bb6c7SMatthew G. Knepley . points - The tabulation point coordinates 930ef0bb6c7SMatthew G. Knepley - K - The number of derivatives calculated 93120cf1dd8SToby Isaac 932ef0bb6c7SMatthew G. Knepley Output Parameter: 933ef0bb6c7SMatthew G. Knepley . T - The basis function values and derivatives at tabulation points 93420cf1dd8SToby Isaac 93520cf1dd8SToby Isaac Note: 936ef0bb6c7SMatthew G. Knepley $ T->T[0] = B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c 937ef0bb6c7SMatthew G. Knepley $ T->T[1] = D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d 938ef0bb6c7SMatthew G. Knepley $ T->T[2] = 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 93920cf1dd8SToby Isaac 94020cf1dd8SToby Isaac Level: intermediate 94120cf1dd8SToby Isaac 942ef0bb6c7SMatthew G. Knepley .seealso: PetscFEGetCellTabulation(), PetscTabulationDestroy() 94320cf1dd8SToby Isaac @*/ 944ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFECreateTabulation(PetscFE fem, PetscInt nrepl, PetscInt npoints, const PetscReal points[], PetscInt K, PetscTabulation *T) 94520cf1dd8SToby Isaac { 94620cf1dd8SToby Isaac DM dm; 947ef0bb6c7SMatthew G. Knepley PetscDualSpace Q; 948ef0bb6c7SMatthew G. Knepley PetscInt Nb; /* Dimension of FE space P */ 949ef0bb6c7SMatthew G. Knepley PetscInt Nc; /* Field components */ 950ef0bb6c7SMatthew G. Knepley PetscInt cdim; /* Reference coordinate dimension */ 951ef0bb6c7SMatthew G. Knepley PetscInt k; 95220cf1dd8SToby Isaac PetscErrorCode ierr; 95320cf1dd8SToby Isaac 95420cf1dd8SToby Isaac PetscFunctionBegin; 955ef0bb6c7SMatthew G. Knepley if (!npoints || !fem->dualSpace || K < 0) { 956ef0bb6c7SMatthew G. Knepley *T = NULL; 95720cf1dd8SToby Isaac PetscFunctionReturn(0); 95820cf1dd8SToby Isaac } 95920cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 96040a2aa30SMatthew G. Knepley PetscValidPointer(points, 4); 96140a2aa30SMatthew G. Knepley PetscValidPointer(T, 6); 962ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetDualSpace(fem, &Q);CHKERRQ(ierr); 963ef0bb6c7SMatthew G. Knepley ierr = PetscDualSpaceGetDM(Q, &dm);CHKERRQ(ierr); 964ef0bb6c7SMatthew G. Knepley ierr = DMGetDimension(dm, &cdim);CHKERRQ(ierr); 965ef0bb6c7SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &Nb);CHKERRQ(ierr); 966ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetNumComponents(fem, &Nc);CHKERRQ(ierr); 967ef0bb6c7SMatthew G. Knepley ierr = PetscMalloc1(1, T);CHKERRQ(ierr); 968ef0bb6c7SMatthew G. Knepley (*T)->K = !cdim ? 0 : K; 969ef0bb6c7SMatthew G. Knepley (*T)->Nr = nrepl; 970ef0bb6c7SMatthew G. Knepley (*T)->Np = npoints; 971ef0bb6c7SMatthew G. Knepley (*T)->Nb = Nb; 972ef0bb6c7SMatthew G. Knepley (*T)->Nc = Nc; 973ef0bb6c7SMatthew G. Knepley (*T)->cdim = cdim; 974ef0bb6c7SMatthew G. Knepley ierr = PetscMalloc1((*T)->K+1, &(*T)->T);CHKERRQ(ierr); 975ef0bb6c7SMatthew G. Knepley for (k = 0; k <= (*T)->K; ++k) { 976ef0bb6c7SMatthew G. Knepley ierr = PetscMalloc1(nrepl*npoints*Nb*Nc*PetscPowInt(cdim, k), &(*T)->T[k]);CHKERRQ(ierr); 97720cf1dd8SToby Isaac } 978ef0bb6c7SMatthew G. Knepley ierr = (*fem->ops->createtabulation)(fem, nrepl*npoints, points, K, *T);CHKERRQ(ierr); 97920cf1dd8SToby Isaac PetscFunctionReturn(0); 98020cf1dd8SToby Isaac } 98120cf1dd8SToby Isaac 9822b99622eSMatthew G. Knepley /*@C 983ef0bb6c7SMatthew G. Knepley PetscFEComputeTabulation - Tabulates the basis functions, and perhaps derivatives, at the points provided. 9842b99622eSMatthew G. Knepley 9852b99622eSMatthew G. Knepley Not collective 9862b99622eSMatthew G. Knepley 9872b99622eSMatthew G. Knepley Input Parameters: 9882b99622eSMatthew G. Knepley + fem - The PetscFE object 9892b99622eSMatthew G. Knepley . npoints - The number of tabulation points 9902b99622eSMatthew G. Knepley . points - The tabulation point coordinates 991ef0bb6c7SMatthew G. Knepley . K - The number of derivatives calculated 992ef0bb6c7SMatthew G. Knepley - T - An existing tabulation object with enough allocated space 993ef0bb6c7SMatthew G. Knepley 994ef0bb6c7SMatthew G. Knepley Output Parameter: 995ef0bb6c7SMatthew G. Knepley . T - The basis function values and derivatives at tabulation points 9962b99622eSMatthew G. Knepley 9972b99622eSMatthew G. Knepley Note: 998ef0bb6c7SMatthew G. Knepley $ T->T[0] = B[(p*pdim + i)*Nc + c] is the value at point p for basis function i and component c 999ef0bb6c7SMatthew G. Knepley $ T->T[1] = D[((p*pdim + i)*Nc + c)*dim + d] is the derivative value at point p for basis function i, component c, in direction d 1000ef0bb6c7SMatthew G. Knepley $ T->T[2] = 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 10012b99622eSMatthew G. Knepley 10022b99622eSMatthew G. Knepley Level: intermediate 10032b99622eSMatthew G. Knepley 1004ef0bb6c7SMatthew G. Knepley .seealso: PetscFEGetCellTabulation(), PetscTabulationDestroy() 10052b99622eSMatthew G. Knepley @*/ 1006ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEComputeTabulation(PetscFE fem, PetscInt npoints, const PetscReal points[], PetscInt K, PetscTabulation T) 1007ef0bb6c7SMatthew G. Knepley { 1008ef0bb6c7SMatthew G. Knepley PetscErrorCode ierr; 1009ef0bb6c7SMatthew G. Knepley 1010ef0bb6c7SMatthew G. Knepley PetscFunctionBeginHot; 1011ef0bb6c7SMatthew G. Knepley if (!npoints || !fem->dualSpace || K < 0) PetscFunctionReturn(0); 1012ef0bb6c7SMatthew G. Knepley PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 1013ef0bb6c7SMatthew G. Knepley PetscValidPointer(points, 3); 1014ef0bb6c7SMatthew G. Knepley PetscValidPointer(T, 5); 1015ef0bb6c7SMatthew G. Knepley #ifdef PETSC_USE_DEBUG 101620cf1dd8SToby Isaac { 101720cf1dd8SToby Isaac DM dm; 1018ef0bb6c7SMatthew G. Knepley PetscDualSpace Q; 1019ef0bb6c7SMatthew G. Knepley PetscInt Nb; /* Dimension of FE space P */ 1020ef0bb6c7SMatthew G. Knepley PetscInt Nc; /* Field components */ 1021ef0bb6c7SMatthew G. Knepley PetscInt cdim; /* Reference coordinate dimension */ 1022ef0bb6c7SMatthew G. Knepley 1023ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetDualSpace(fem, &Q);CHKERRQ(ierr); 1024ef0bb6c7SMatthew G. Knepley ierr = PetscDualSpaceGetDM(Q, &dm);CHKERRQ(ierr); 1025ef0bb6c7SMatthew G. Knepley ierr = DMGetDimension(dm, &cdim);CHKERRQ(ierr); 1026ef0bb6c7SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &Nb);CHKERRQ(ierr); 1027ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetNumComponents(fem, &Nc);CHKERRQ(ierr); 1028ef0bb6c7SMatthew G. Knepley if (T->K != (!cdim ? 0 : K)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Tabulation K %D must match requested K %D", T->K, !cdim ? 0 : K); 1029ef0bb6c7SMatthew G. Knepley if (T->Nb != Nb) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Tabulation Nb %D must match requested Nb %D", T->Nb, Nb); 1030ef0bb6c7SMatthew G. Knepley if (T->Nc != Nc) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Tabulation Nc %D must match requested Nc %D", T->Nc, Nc); 1031ef0bb6c7SMatthew G. Knepley if (T->cdim != cdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Tabulation cdim %D must match requested cdim %D", T->cdim, cdim); 1032ef0bb6c7SMatthew G. Knepley } 1033ef0bb6c7SMatthew G. Knepley #endif 1034ef0bb6c7SMatthew G. Knepley T->Nr = 1; 1035ef0bb6c7SMatthew G. Knepley T->Np = npoints; 1036ef0bb6c7SMatthew G. Knepley ierr = (*fem->ops->createtabulation)(fem, npoints, points, K, T);CHKERRQ(ierr); 1037ef0bb6c7SMatthew G. Knepley PetscFunctionReturn(0); 1038ef0bb6c7SMatthew G. Knepley } 1039ef0bb6c7SMatthew G. Knepley 1040ef0bb6c7SMatthew G. Knepley /*@C 1041ef0bb6c7SMatthew G. Knepley PetscTabulationDestroy - Frees memory from the associated tabulation. 1042ef0bb6c7SMatthew G. Knepley 1043ef0bb6c7SMatthew G. Knepley Not collective 1044ef0bb6c7SMatthew G. Knepley 1045ef0bb6c7SMatthew G. Knepley Input Parameter: 1046ef0bb6c7SMatthew G. Knepley . T - The tabulation 1047ef0bb6c7SMatthew G. Knepley 1048ef0bb6c7SMatthew G. Knepley Level: intermediate 1049ef0bb6c7SMatthew G. Knepley 1050ef0bb6c7SMatthew G. Knepley .seealso: PetscFECreateTabulation(), PetscFEGetCellTabulation() 1051ef0bb6c7SMatthew G. Knepley @*/ 1052ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscTabulationDestroy(PetscTabulation *T) 1053ef0bb6c7SMatthew G. Knepley { 1054ef0bb6c7SMatthew G. Knepley PetscInt k; 105520cf1dd8SToby Isaac PetscErrorCode ierr; 105620cf1dd8SToby Isaac 105720cf1dd8SToby Isaac PetscFunctionBegin; 1058ef0bb6c7SMatthew G. Knepley PetscValidPointer(T, 1); 1059ef0bb6c7SMatthew G. Knepley if (!T || !(*T)) PetscFunctionReturn(0); 1060ef0bb6c7SMatthew G. Knepley for (k = 0; k <= (*T)->K; ++k) {ierr = PetscFree((*T)->T[k]);CHKERRQ(ierr);} 1061ef0bb6c7SMatthew G. Knepley ierr = PetscFree((*T)->T);CHKERRQ(ierr); 1062ef0bb6c7SMatthew G. Knepley ierr = PetscFree(*T);CHKERRQ(ierr); 1063ef0bb6c7SMatthew G. Knepley *T = NULL; 106420cf1dd8SToby Isaac PetscFunctionReturn(0); 106520cf1dd8SToby Isaac } 106620cf1dd8SToby Isaac 106720cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode PetscFECreatePointTrace(PetscFE fe, PetscInt refPoint, PetscFE *trFE) 106820cf1dd8SToby Isaac { 106920cf1dd8SToby Isaac PetscSpace bsp, bsubsp; 107020cf1dd8SToby Isaac PetscDualSpace dsp, dsubsp; 107120cf1dd8SToby Isaac PetscInt dim, depth, numComp, i, j, coneSize, order; 107220cf1dd8SToby Isaac PetscFEType type; 107320cf1dd8SToby Isaac DM dm; 107420cf1dd8SToby Isaac DMLabel label; 107520cf1dd8SToby Isaac PetscReal *xi, *v, *J, detJ; 1076db11e2ebSMatthew G. Knepley const char *name; 107720cf1dd8SToby Isaac PetscQuadrature origin, fullQuad, subQuad; 107820cf1dd8SToby Isaac PetscErrorCode ierr; 107920cf1dd8SToby Isaac 108020cf1dd8SToby Isaac PetscFunctionBegin; 108120cf1dd8SToby Isaac PetscValidHeaderSpecific(fe,PETSCFE_CLASSID,1); 108220cf1dd8SToby Isaac PetscValidPointer(trFE,3); 108320cf1dd8SToby Isaac ierr = PetscFEGetBasisSpace(fe,&bsp);CHKERRQ(ierr); 108420cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fe,&dsp);CHKERRQ(ierr); 108520cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(dsp,&dm);CHKERRQ(ierr); 108620cf1dd8SToby Isaac ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr); 108720cf1dd8SToby Isaac ierr = DMPlexGetDepthLabel(dm,&label);CHKERRQ(ierr); 108820cf1dd8SToby Isaac ierr = DMLabelGetValue(label,refPoint,&depth);CHKERRQ(ierr); 108920cf1dd8SToby Isaac ierr = PetscCalloc1(depth,&xi);CHKERRQ(ierr); 109020cf1dd8SToby Isaac ierr = PetscMalloc1(dim,&v);CHKERRQ(ierr); 109120cf1dd8SToby Isaac ierr = PetscMalloc1(dim*dim,&J);CHKERRQ(ierr); 109220cf1dd8SToby Isaac for (i = 0; i < depth; i++) xi[i] = 0.; 109320cf1dd8SToby Isaac ierr = PetscQuadratureCreate(PETSC_COMM_SELF,&origin);CHKERRQ(ierr); 109420cf1dd8SToby Isaac ierr = PetscQuadratureSetData(origin,depth,0,1,xi,NULL);CHKERRQ(ierr); 109520cf1dd8SToby Isaac ierr = DMPlexComputeCellGeometryFEM(dm,refPoint,origin,v,J,NULL,&detJ);CHKERRQ(ierr); 109620cf1dd8SToby Isaac /* CellGeometryFEM computes the expanded Jacobian, we want the true jacobian */ 109720cf1dd8SToby Isaac for (i = 1; i < dim; i++) { 109820cf1dd8SToby Isaac for (j = 0; j < depth; j++) { 109920cf1dd8SToby Isaac J[i * depth + j] = J[i * dim + j]; 110020cf1dd8SToby Isaac } 110120cf1dd8SToby Isaac } 110220cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&origin);CHKERRQ(ierr); 110320cf1dd8SToby Isaac ierr = PetscDualSpaceGetPointSubspace(dsp,refPoint,&dsubsp);CHKERRQ(ierr); 110420cf1dd8SToby Isaac ierr = PetscSpaceCreateSubspace(bsp,dsubsp,v,J,NULL,NULL,PETSC_OWN_POINTER,&bsubsp);CHKERRQ(ierr); 110520cf1dd8SToby Isaac ierr = PetscSpaceSetUp(bsubsp);CHKERRQ(ierr); 110620cf1dd8SToby Isaac ierr = PetscFECreate(PetscObjectComm((PetscObject)fe),trFE);CHKERRQ(ierr); 110720cf1dd8SToby Isaac ierr = PetscFEGetType(fe,&type);CHKERRQ(ierr); 110820cf1dd8SToby Isaac ierr = PetscFESetType(*trFE,type);CHKERRQ(ierr); 110920cf1dd8SToby Isaac ierr = PetscFEGetNumComponents(fe,&numComp);CHKERRQ(ierr); 111020cf1dd8SToby Isaac ierr = PetscFESetNumComponents(*trFE,numComp);CHKERRQ(ierr); 111120cf1dd8SToby Isaac ierr = PetscFESetBasisSpace(*trFE,bsubsp);CHKERRQ(ierr); 111220cf1dd8SToby Isaac ierr = PetscFESetDualSpace(*trFE,dsubsp);CHKERRQ(ierr); 1113db11e2ebSMatthew G. Knepley ierr = PetscObjectGetName((PetscObject) fe, &name);CHKERRQ(ierr); 1114db11e2ebSMatthew G. Knepley if (name) {ierr = PetscFESetName(*trFE, name);CHKERRQ(ierr);} 111520cf1dd8SToby Isaac ierr = PetscFEGetQuadrature(fe,&fullQuad);CHKERRQ(ierr); 111620cf1dd8SToby Isaac ierr = PetscQuadratureGetOrder(fullQuad,&order);CHKERRQ(ierr); 111720cf1dd8SToby Isaac ierr = DMPlexGetConeSize(dm,refPoint,&coneSize);CHKERRQ(ierr); 111820cf1dd8SToby Isaac if (coneSize == 2 * depth) { 111920cf1dd8SToby Isaac ierr = PetscDTGaussTensorQuadrature(depth,1,(order + 1)/2,-1.,1.,&subQuad);CHKERRQ(ierr); 112020cf1dd8SToby Isaac } else { 1121e6a796c3SToby Isaac ierr = PetscDTStroudConicalQuadrature(depth,1,(order + 1)/2,-1.,1.,&subQuad);CHKERRQ(ierr); 112220cf1dd8SToby Isaac } 112320cf1dd8SToby Isaac ierr = PetscFESetQuadrature(*trFE,subQuad);CHKERRQ(ierr); 112420cf1dd8SToby Isaac ierr = PetscFESetUp(*trFE);CHKERRQ(ierr); 112520cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&subQuad);CHKERRQ(ierr); 112620cf1dd8SToby Isaac ierr = PetscSpaceDestroy(&bsubsp);CHKERRQ(ierr); 112720cf1dd8SToby Isaac PetscFunctionReturn(0); 112820cf1dd8SToby Isaac } 112920cf1dd8SToby Isaac 113020cf1dd8SToby Isaac PetscErrorCode PetscFECreateHeightTrace(PetscFE fe, PetscInt height, PetscFE *trFE) 113120cf1dd8SToby Isaac { 113220cf1dd8SToby Isaac PetscInt hStart, hEnd; 113320cf1dd8SToby Isaac PetscDualSpace dsp; 113420cf1dd8SToby Isaac DM dm; 113520cf1dd8SToby Isaac PetscErrorCode ierr; 113620cf1dd8SToby Isaac 113720cf1dd8SToby Isaac PetscFunctionBegin; 113820cf1dd8SToby Isaac PetscValidHeaderSpecific(fe,PETSCFE_CLASSID,1); 113920cf1dd8SToby Isaac PetscValidPointer(trFE,3); 114020cf1dd8SToby Isaac *trFE = NULL; 114120cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fe,&dsp);CHKERRQ(ierr); 114220cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(dsp,&dm);CHKERRQ(ierr); 114320cf1dd8SToby Isaac ierr = DMPlexGetHeightStratum(dm,height,&hStart,&hEnd);CHKERRQ(ierr); 114420cf1dd8SToby Isaac if (hEnd <= hStart) PetscFunctionReturn(0); 114520cf1dd8SToby Isaac ierr = PetscFECreatePointTrace(fe,hStart,trFE);CHKERRQ(ierr); 114620cf1dd8SToby Isaac PetscFunctionReturn(0); 114720cf1dd8SToby Isaac } 114820cf1dd8SToby Isaac 114920cf1dd8SToby Isaac 115020cf1dd8SToby Isaac /*@ 115120cf1dd8SToby Isaac PetscFEGetDimension - Get the dimension of the finite element space on a cell 115220cf1dd8SToby Isaac 115320cf1dd8SToby Isaac Not collective 115420cf1dd8SToby Isaac 115520cf1dd8SToby Isaac Input Parameter: 115620cf1dd8SToby Isaac . fe - The PetscFE 115720cf1dd8SToby Isaac 115820cf1dd8SToby Isaac Output Parameter: 115920cf1dd8SToby Isaac . dim - The dimension 116020cf1dd8SToby Isaac 116120cf1dd8SToby Isaac Level: intermediate 116220cf1dd8SToby Isaac 116320cf1dd8SToby Isaac .seealso: PetscFECreate(), PetscSpaceGetDimension(), PetscDualSpaceGetDimension() 116420cf1dd8SToby Isaac @*/ 116520cf1dd8SToby Isaac PetscErrorCode PetscFEGetDimension(PetscFE fem, PetscInt *dim) 116620cf1dd8SToby Isaac { 116720cf1dd8SToby Isaac PetscErrorCode ierr; 116820cf1dd8SToby Isaac 116920cf1dd8SToby Isaac PetscFunctionBegin; 117020cf1dd8SToby Isaac PetscValidHeaderSpecific(fem, PETSCFE_CLASSID, 1); 117120cf1dd8SToby Isaac PetscValidPointer(dim, 2); 117220cf1dd8SToby Isaac if (fem->ops->getdimension) {ierr = (*fem->ops->getdimension)(fem, dim);CHKERRQ(ierr);} 117320cf1dd8SToby Isaac PetscFunctionReturn(0); 117420cf1dd8SToby Isaac } 117520cf1dd8SToby Isaac 11764bee2e38SMatthew G. Knepley /*@C 11774bee2e38SMatthew G. Knepley PetscFEPushforward - Map the reference element function to real space 11784bee2e38SMatthew G. Knepley 11794bee2e38SMatthew G. Knepley Input Parameters: 11804bee2e38SMatthew G. Knepley + fe - The PetscFE 11814bee2e38SMatthew G. Knepley . fegeom - The cell geometry 11824bee2e38SMatthew G. Knepley . Nv - The number of function values 11834bee2e38SMatthew G. Knepley - vals - The function values 11844bee2e38SMatthew G. Knepley 11854bee2e38SMatthew G. Knepley Output Parameter: 11864bee2e38SMatthew G. Knepley . vals - The transformed function values 11874bee2e38SMatthew G. Knepley 11884bee2e38SMatthew G. Knepley Level: advanced 11894bee2e38SMatthew G. Knepley 11904bee2e38SMatthew G. Knepley Note: This just forwards the call onto PetscDualSpacePushforward(). 11914bee2e38SMatthew G. Knepley 11922edcad52SToby Isaac Note: This only handles tranformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 11932edcad52SToby Isaac 11944bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePushforward() 11954bee2e38SMatthew G. Knepley @*/ 11962edcad52SToby Isaac PetscErrorCode PetscFEPushforward(PetscFE fe, PetscFEGeom *fegeom, PetscInt Nv, PetscScalar vals[]) 11974bee2e38SMatthew G. Knepley { 11984bee2e38SMatthew G. Knepley PetscErrorCode ierr; 11994bee2e38SMatthew G. Knepley 12002ae266adSMatthew G. Knepley PetscFunctionBeginHot; 12012edcad52SToby Isaac ierr = PetscDualSpacePushforward(fe->dualSpace, fegeom, Nv, fe->numComponents, vals);CHKERRQ(ierr); 12024bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 12034bee2e38SMatthew G. Knepley } 12044bee2e38SMatthew G. Knepley 12054bee2e38SMatthew G. Knepley /*@C 12064bee2e38SMatthew G. Knepley PetscFEPushforwardGradient - Map the reference element function gradient to real space 12074bee2e38SMatthew G. Knepley 12084bee2e38SMatthew G. Knepley Input Parameters: 12094bee2e38SMatthew G. Knepley + fe - The PetscFE 12104bee2e38SMatthew G. Knepley . fegeom - The cell geometry 12114bee2e38SMatthew G. Knepley . Nv - The number of function gradient values 12124bee2e38SMatthew G. Knepley - vals - The function gradient values 12134bee2e38SMatthew G. Knepley 12144bee2e38SMatthew G. Knepley Output Parameter: 12154bee2e38SMatthew G. Knepley . vals - The transformed function gradient values 12164bee2e38SMatthew G. Knepley 12174bee2e38SMatthew G. Knepley Level: advanced 12184bee2e38SMatthew G. Knepley 12194bee2e38SMatthew G. Knepley Note: This just forwards the call onto PetscDualSpacePushforwardGradient(). 12204bee2e38SMatthew G. Knepley 12212edcad52SToby Isaac Note: This only handles tranformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 12222edcad52SToby Isaac 12234bee2e38SMatthew G. Knepley .seealso: PetscFEPushforward(), PetscDualSpacePushforwardGradient(), PetscDualSpacePushforward() 12244bee2e38SMatthew G. Knepley @*/ 12252edcad52SToby Isaac PetscErrorCode PetscFEPushforwardGradient(PetscFE fe, PetscFEGeom *fegeom, PetscInt Nv, PetscScalar vals[]) 12264bee2e38SMatthew G. Knepley { 12274bee2e38SMatthew G. Knepley PetscErrorCode ierr; 12284bee2e38SMatthew G. Knepley 12292ae266adSMatthew G. Knepley PetscFunctionBeginHot; 12302edcad52SToby Isaac ierr = PetscDualSpacePushforwardGradient(fe->dualSpace, fegeom, Nv, fe->numComponents, vals);CHKERRQ(ierr); 12314bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 12324bee2e38SMatthew G. Knepley } 12334bee2e38SMatthew G. Knepley 123420cf1dd8SToby Isaac /* 123520cf1dd8SToby Isaac Purpose: Compute element vector for chunk of elements 123620cf1dd8SToby Isaac 123720cf1dd8SToby Isaac Input: 123820cf1dd8SToby Isaac Sizes: 123920cf1dd8SToby Isaac Ne: number of elements 124020cf1dd8SToby Isaac Nf: number of fields 124120cf1dd8SToby Isaac PetscFE 124220cf1dd8SToby Isaac dim: spatial dimension 124320cf1dd8SToby Isaac Nb: number of basis functions 124420cf1dd8SToby Isaac Nc: number of field components 124520cf1dd8SToby Isaac PetscQuadrature 124620cf1dd8SToby Isaac Nq: number of quadrature points 124720cf1dd8SToby Isaac 124820cf1dd8SToby Isaac Geometry: 124920cf1dd8SToby Isaac PetscFEGeom[Ne] possibly *Nq 125020cf1dd8SToby Isaac PetscReal v0s[dim] 125120cf1dd8SToby Isaac PetscReal n[dim] 125220cf1dd8SToby Isaac PetscReal jacobians[dim*dim] 125320cf1dd8SToby Isaac PetscReal jacobianInverses[dim*dim] 125420cf1dd8SToby Isaac PetscReal jacobianDeterminants 125520cf1dd8SToby Isaac FEM: 125620cf1dd8SToby Isaac PetscFE 125720cf1dd8SToby Isaac PetscQuadrature 125820cf1dd8SToby Isaac PetscReal quadPoints[Nq*dim] 125920cf1dd8SToby Isaac PetscReal quadWeights[Nq] 126020cf1dd8SToby Isaac PetscReal basis[Nq*Nb*Nc] 126120cf1dd8SToby Isaac PetscReal basisDer[Nq*Nb*Nc*dim] 126220cf1dd8SToby Isaac PetscScalar coefficients[Ne*Nb*Nc] 126320cf1dd8SToby Isaac PetscScalar elemVec[Ne*Nb*Nc] 126420cf1dd8SToby Isaac 126520cf1dd8SToby Isaac Problem: 126620cf1dd8SToby Isaac PetscInt f: the active field 126720cf1dd8SToby Isaac f0, f1 126820cf1dd8SToby Isaac 126920cf1dd8SToby Isaac Work Space: 127020cf1dd8SToby Isaac PetscFE 127120cf1dd8SToby Isaac PetscScalar f0[Nq*dim]; 127220cf1dd8SToby Isaac PetscScalar f1[Nq*dim*dim]; 127320cf1dd8SToby Isaac PetscScalar u[Nc]; 127420cf1dd8SToby Isaac PetscScalar gradU[Nc*dim]; 127520cf1dd8SToby Isaac PetscReal x[dim]; 127620cf1dd8SToby Isaac PetscScalar realSpaceDer[dim]; 127720cf1dd8SToby Isaac 127820cf1dd8SToby Isaac Purpose: Compute element vector for N_cb batches of elements 127920cf1dd8SToby Isaac 128020cf1dd8SToby Isaac Input: 128120cf1dd8SToby Isaac Sizes: 128220cf1dd8SToby Isaac N_cb: Number of serial cell batches 128320cf1dd8SToby Isaac 128420cf1dd8SToby Isaac Geometry: 128520cf1dd8SToby Isaac PetscReal v0s[Ne*dim] 128620cf1dd8SToby Isaac PetscReal jacobians[Ne*dim*dim] possibly *Nq 128720cf1dd8SToby Isaac PetscReal jacobianInverses[Ne*dim*dim] possibly *Nq 128820cf1dd8SToby Isaac PetscReal jacobianDeterminants[Ne] possibly *Nq 128920cf1dd8SToby Isaac FEM: 129020cf1dd8SToby Isaac static PetscReal quadPoints[Nq*dim] 129120cf1dd8SToby Isaac static PetscReal quadWeights[Nq] 129220cf1dd8SToby Isaac static PetscReal basis[Nq*Nb*Nc] 129320cf1dd8SToby Isaac static PetscReal basisDer[Nq*Nb*Nc*dim] 129420cf1dd8SToby Isaac PetscScalar coefficients[Ne*Nb*Nc] 129520cf1dd8SToby Isaac PetscScalar elemVec[Ne*Nb*Nc] 129620cf1dd8SToby Isaac 129720cf1dd8SToby Isaac ex62.c: 129820cf1dd8SToby Isaac PetscErrorCode PetscFEIntegrateResidualBatch(PetscInt Ne, PetscInt numFields, PetscInt field, PetscQuadrature quad[], const PetscScalar coefficients[], 129920cf1dd8SToby Isaac const PetscReal v0s[], const PetscReal jacobians[], const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], 130020cf1dd8SToby Isaac void (*f0_func)(const PetscScalar u[], const PetscScalar gradU[], const PetscReal x[], PetscScalar f0[]), 130120cf1dd8SToby Isaac void (*f1_func)(const PetscScalar u[], const PetscScalar gradU[], const PetscReal x[], PetscScalar f1[]), PetscScalar elemVec[]) 130220cf1dd8SToby Isaac 130320cf1dd8SToby Isaac ex52.c: 130420cf1dd8SToby 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) 130520cf1dd8SToby 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) 130620cf1dd8SToby Isaac 130720cf1dd8SToby Isaac ex52_integrateElement.cu 130820cf1dd8SToby Isaac __global__ void integrateElementQuadrature(int N_cb, realType *coefficients, realType *jacobianInverses, realType *jacobianDeterminants, realType *elemVec) 130920cf1dd8SToby Isaac 131020cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode IntegrateElementBatchGPU(PetscInt spatial_dim, PetscInt Ne, PetscInt Ncb, PetscInt Nbc, PetscInt Nbl, const PetscScalar coefficients[], 131120cf1dd8SToby Isaac const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscScalar elemVec[], 131220cf1dd8SToby Isaac PetscLogEvent event, PetscInt debug, PetscInt pde_op) 131320cf1dd8SToby Isaac 131420cf1dd8SToby Isaac ex52_integrateElementOpenCL.c: 131520cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode IntegrateElementBatchGPU(PetscInt spatial_dim, PetscInt Ne, PetscInt Ncb, PetscInt Nbc, PetscInt N_bl, const PetscScalar coefficients[], 131620cf1dd8SToby Isaac const PetscReal jacobianInverses[], const PetscReal jacobianDeterminants[], PetscScalar elemVec[], 131720cf1dd8SToby Isaac PetscLogEvent event, PetscInt debug, PetscInt pde_op) 131820cf1dd8SToby Isaac 131920cf1dd8SToby Isaac __kernel void integrateElementQuadrature(int N_cb, __global float *coefficients, __global float *jacobianInverses, __global float *jacobianDeterminants, __global float *elemVec) 132020cf1dd8SToby Isaac */ 132120cf1dd8SToby Isaac 132220cf1dd8SToby Isaac /*@C 132320cf1dd8SToby Isaac PetscFEIntegrate - Produce the integral for the given field for a chunk of elements by quadrature integration 132420cf1dd8SToby Isaac 132520cf1dd8SToby Isaac Not collective 132620cf1dd8SToby Isaac 132720cf1dd8SToby Isaac Input Parameters: 132820cf1dd8SToby Isaac + fem - The PetscFE object for the field being integrated 132920cf1dd8SToby Isaac . prob - The PetscDS specifying the discretizations and continuum functions 133020cf1dd8SToby Isaac . field - The field being integrated 133120cf1dd8SToby Isaac . Ne - The number of elements in the chunk 133220cf1dd8SToby Isaac . cgeom - The cell geometry for each cell in the chunk 133320cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements 133420cf1dd8SToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 133520cf1dd8SToby Isaac - coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 133620cf1dd8SToby Isaac 13377a7aea1fSJed Brown Output Parameter: 133820cf1dd8SToby Isaac . integral - the integral for this field 133920cf1dd8SToby Isaac 13402b99622eSMatthew G. Knepley Level: intermediate 134120cf1dd8SToby Isaac 134220cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual() 134320cf1dd8SToby Isaac @*/ 13444bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrate(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *cgeom, 134520cf1dd8SToby Isaac const PetscScalar coefficients[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscScalar integral[]) 134620cf1dd8SToby Isaac { 13474bee2e38SMatthew G. Knepley PetscFE fe; 134820cf1dd8SToby Isaac PetscErrorCode ierr; 134920cf1dd8SToby Isaac 135020cf1dd8SToby Isaac PetscFunctionBegin; 13514bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 13524bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 13534bee2e38SMatthew G. Knepley if (fe->ops->integrate) {ierr = (*fe->ops->integrate)(prob, field, Ne, cgeom, coefficients, probAux, coefficientsAux, integral);CHKERRQ(ierr);} 135420cf1dd8SToby Isaac PetscFunctionReturn(0); 135520cf1dd8SToby Isaac } 135620cf1dd8SToby Isaac 135720cf1dd8SToby Isaac /*@C 1358afe6d6adSToby Isaac PetscFEIntegrateBd - Produce the integral for the given field for a chunk of elements by quadrature integration 1359afe6d6adSToby Isaac 1360afe6d6adSToby Isaac Not collective 1361afe6d6adSToby Isaac 1362afe6d6adSToby Isaac Input Parameters: 1363afe6d6adSToby Isaac + fem - The PetscFE object for the field being integrated 1364afe6d6adSToby Isaac . prob - The PetscDS specifying the discretizations and continuum functions 1365afe6d6adSToby Isaac . field - The field being integrated 1366afe6d6adSToby Isaac . obj_func - The function to be integrated 1367afe6d6adSToby Isaac . Ne - The number of elements in the chunk 1368afe6d6adSToby Isaac . fgeom - The face geometry for each face in the chunk 1369afe6d6adSToby Isaac . coefficients - The array of FEM basis coefficients for the elements 1370afe6d6adSToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 1371afe6d6adSToby Isaac - coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 1372afe6d6adSToby Isaac 13737a7aea1fSJed Brown Output Parameter: 1374afe6d6adSToby Isaac . integral - the integral for this field 1375afe6d6adSToby Isaac 13762b99622eSMatthew G. Knepley Level: intermediate 1377afe6d6adSToby Isaac 1378afe6d6adSToby Isaac .seealso: PetscFEIntegrateResidual() 1379afe6d6adSToby Isaac @*/ 13804bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBd(PetscDS prob, PetscInt field, 1381afe6d6adSToby Isaac void (*obj_func)(PetscInt, PetscInt, PetscInt, 1382afe6d6adSToby Isaac const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 1383afe6d6adSToby Isaac const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 1384afe6d6adSToby Isaac PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), 1385afe6d6adSToby Isaac PetscInt Ne, PetscFEGeom *geom, const PetscScalar coefficients[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscScalar integral[]) 1386afe6d6adSToby Isaac { 13874bee2e38SMatthew G. Knepley PetscFE fe; 1388afe6d6adSToby Isaac PetscErrorCode ierr; 1389afe6d6adSToby Isaac 1390afe6d6adSToby Isaac PetscFunctionBegin; 13914bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 13924bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 13934bee2e38SMatthew G. Knepley if (fe->ops->integratebd) {ierr = (*fe->ops->integratebd)(prob, field, obj_func, Ne, geom, coefficients, probAux, coefficientsAux, integral);CHKERRQ(ierr);} 1394afe6d6adSToby Isaac PetscFunctionReturn(0); 1395afe6d6adSToby Isaac } 1396afe6d6adSToby Isaac 1397afe6d6adSToby Isaac /*@C 139820cf1dd8SToby Isaac PetscFEIntegrateResidual - Produce the element residual vector for a chunk of elements by quadrature integration 139920cf1dd8SToby Isaac 140020cf1dd8SToby Isaac Not collective 140120cf1dd8SToby Isaac 140220cf1dd8SToby Isaac Input Parameters: 140320cf1dd8SToby Isaac + fem - The PetscFE object for the field being integrated 140420cf1dd8SToby Isaac . prob - The PetscDS specifying the discretizations and continuum functions 140520cf1dd8SToby Isaac . field - The field being integrated 140620cf1dd8SToby Isaac . Ne - The number of elements in the chunk 140720cf1dd8SToby Isaac . cgeom - The cell geometry for each cell in the chunk 140820cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements 140920cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements 141020cf1dd8SToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 141120cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 141220cf1dd8SToby Isaac - t - The time 141320cf1dd8SToby Isaac 14147a7aea1fSJed Brown Output Parameter: 141520cf1dd8SToby Isaac . elemVec - the element residual vectors from each element 141620cf1dd8SToby Isaac 141720cf1dd8SToby Isaac Note: 141820cf1dd8SToby Isaac $ Loop over batch of elements (e): 141920cf1dd8SToby Isaac $ Loop over quadrature points (q): 142020cf1dd8SToby Isaac $ Make u_q and gradU_q (loops over fields,Nb,Ncomp) and x_q 142120cf1dd8SToby Isaac $ Call f_0 and f_1 142220cf1dd8SToby Isaac $ Loop over element vector entries (f,fc --> i): 142320cf1dd8SToby Isaac $ elemVec[i] += \psi^{fc}_f(q) f0_{fc}(u, \nabla u) + \nabla\psi^{fc}_f(q) \cdot f1_{fc,df}(u, \nabla u) 142420cf1dd8SToby Isaac 14252b99622eSMatthew G. Knepley Level: intermediate 142620cf1dd8SToby Isaac 142720cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual() 142820cf1dd8SToby Isaac @*/ 14294bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateResidual(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *cgeom, 143020cf1dd8SToby Isaac const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscScalar elemVec[]) 143120cf1dd8SToby Isaac { 14324bee2e38SMatthew G. Knepley PetscFE fe; 143320cf1dd8SToby Isaac PetscErrorCode ierr; 143420cf1dd8SToby Isaac 143520cf1dd8SToby Isaac PetscFunctionBegin; 14364bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 14374bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 14384bee2e38SMatthew G. Knepley if (fe->ops->integrateresidual) {ierr = (*fe->ops->integrateresidual)(prob, field, Ne, cgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, elemVec);CHKERRQ(ierr);} 143920cf1dd8SToby Isaac PetscFunctionReturn(0); 144020cf1dd8SToby Isaac } 144120cf1dd8SToby Isaac 144220cf1dd8SToby Isaac /*@C 144320cf1dd8SToby Isaac PetscFEIntegrateBdResidual - Produce the element residual vector for a chunk of elements by quadrature integration over a boundary 144420cf1dd8SToby Isaac 144520cf1dd8SToby Isaac Not collective 144620cf1dd8SToby Isaac 144720cf1dd8SToby Isaac Input Parameters: 144820cf1dd8SToby Isaac + fem - The PetscFE object for the field being integrated 144920cf1dd8SToby Isaac . prob - The PetscDS specifying the discretizations and continuum functions 145020cf1dd8SToby Isaac . field - The field being integrated 145120cf1dd8SToby Isaac . Ne - The number of elements in the chunk 145220cf1dd8SToby Isaac . fgeom - The face geometry for each cell in the chunk 145320cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements 145420cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements 145520cf1dd8SToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 145620cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 145720cf1dd8SToby Isaac - t - The time 145820cf1dd8SToby Isaac 14597a7aea1fSJed Brown Output Parameter: 146020cf1dd8SToby Isaac . elemVec - the element residual vectors from each element 146120cf1dd8SToby Isaac 14622b99622eSMatthew G. Knepley Level: intermediate 146320cf1dd8SToby Isaac 146420cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual() 146520cf1dd8SToby Isaac @*/ 14664bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBdResidual(PetscDS prob, PetscInt field, PetscInt Ne, PetscFEGeom *fgeom, 146720cf1dd8SToby Isaac const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscScalar elemVec[]) 146820cf1dd8SToby Isaac { 14694bee2e38SMatthew G. Knepley PetscFE fe; 147020cf1dd8SToby Isaac PetscErrorCode ierr; 147120cf1dd8SToby Isaac 147220cf1dd8SToby Isaac PetscFunctionBegin; 14734bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 14744bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 14754bee2e38SMatthew G. Knepley if (fe->ops->integratebdresidual) {ierr = (*fe->ops->integratebdresidual)(prob, field, Ne, fgeom, coefficients, coefficients_t, probAux, coefficientsAux, t, elemVec);CHKERRQ(ierr);} 147620cf1dd8SToby Isaac PetscFunctionReturn(0); 147720cf1dd8SToby Isaac } 147820cf1dd8SToby Isaac 147920cf1dd8SToby Isaac /*@C 148020cf1dd8SToby Isaac PetscFEIntegrateJacobian - Produce the element Jacobian for a chunk of elements by quadrature integration 148120cf1dd8SToby Isaac 148220cf1dd8SToby Isaac Not collective 148320cf1dd8SToby Isaac 148420cf1dd8SToby Isaac Input Parameters: 148520cf1dd8SToby Isaac + fem - The PetscFE object for the field being integrated 148620cf1dd8SToby Isaac . prob - The PetscDS specifying the discretizations and continuum functions 148720cf1dd8SToby Isaac . jtype - The type of matrix pointwise functions that should be used 148820cf1dd8SToby Isaac . fieldI - The test field being integrated 148920cf1dd8SToby Isaac . fieldJ - The basis field being integrated 149020cf1dd8SToby Isaac . Ne - The number of elements in the chunk 149120cf1dd8SToby Isaac . cgeom - The cell geometry for each cell in the chunk 149220cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements for the Jacobian evaluation point 149320cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements 149420cf1dd8SToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 149520cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 149620cf1dd8SToby Isaac . t - The time 149720cf1dd8SToby Isaac - u_tShift - A multiplier for the dF/du_t term (as opposed to the dF/du term) 149820cf1dd8SToby Isaac 14997a7aea1fSJed Brown Output Parameter: 150020cf1dd8SToby Isaac . elemMat - the element matrices for the Jacobian from each element 150120cf1dd8SToby Isaac 150220cf1dd8SToby Isaac Note: 150320cf1dd8SToby Isaac $ Loop over batch of elements (e): 150420cf1dd8SToby Isaac $ Loop over element matrix entries (f,fc,g,gc --> i,j): 150520cf1dd8SToby Isaac $ Loop over quadrature points (q): 150620cf1dd8SToby Isaac $ Make u_q and gradU_q (loops over fields,Nb,Ncomp) 150720cf1dd8SToby Isaac $ elemMat[i,j] += \psi^{fc}_f(q) g0_{fc,gc}(u, \nabla u) \phi^{gc}_g(q) 150820cf1dd8SToby Isaac $ + \psi^{fc}_f(q) \cdot g1_{fc,gc,dg}(u, \nabla u) \nabla\phi^{gc}_g(q) 150920cf1dd8SToby Isaac $ + \nabla\psi^{fc}_f(q) \cdot g2_{fc,gc,df}(u, \nabla u) \phi^{gc}_g(q) 151020cf1dd8SToby Isaac $ + \nabla\psi^{fc}_f(q) \cdot g3_{fc,gc,df,dg}(u, \nabla u) \nabla\phi^{gc}_g(q) 15112b99622eSMatthew G. Knepley Level: intermediate 151220cf1dd8SToby Isaac 151320cf1dd8SToby Isaac .seealso: PetscFEIntegrateResidual() 151420cf1dd8SToby Isaac @*/ 15154bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateJacobian(PetscDS prob, PetscFEJacobianType jtype, PetscInt fieldI, PetscInt fieldJ, PetscInt Ne, PetscFEGeom *cgeom, 151620cf1dd8SToby Isaac const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscReal u_tshift, PetscScalar elemMat[]) 151720cf1dd8SToby Isaac { 15184bee2e38SMatthew G. Knepley PetscFE fe; 151920cf1dd8SToby Isaac PetscErrorCode ierr; 152020cf1dd8SToby Isaac 152120cf1dd8SToby Isaac PetscFunctionBegin; 15224bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 15234bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fe);CHKERRQ(ierr); 15244bee2e38SMatthew 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);} 152520cf1dd8SToby Isaac PetscFunctionReturn(0); 152620cf1dd8SToby Isaac } 152720cf1dd8SToby Isaac 152820cf1dd8SToby Isaac /*@C 152920cf1dd8SToby Isaac PetscFEIntegrateBdJacobian - Produce the boundary element Jacobian for a chunk of elements by quadrature integration 153020cf1dd8SToby Isaac 153120cf1dd8SToby Isaac Not collective 153220cf1dd8SToby Isaac 153320cf1dd8SToby Isaac Input Parameters: 1534f0fc11ceSJed Brown + prob - The PetscDS specifying the discretizations and continuum functions 153520cf1dd8SToby Isaac . fieldI - The test field being integrated 153620cf1dd8SToby Isaac . fieldJ - The basis field being integrated 153720cf1dd8SToby Isaac . Ne - The number of elements in the chunk 153820cf1dd8SToby Isaac . fgeom - The face geometry for each cell in the chunk 153920cf1dd8SToby Isaac . coefficients - The array of FEM basis coefficients for the elements for the Jacobian evaluation point 154020cf1dd8SToby Isaac . coefficients_t - The array of FEM basis time derivative coefficients for the elements 154120cf1dd8SToby Isaac . probAux - The PetscDS specifying the auxiliary discretizations 154220cf1dd8SToby Isaac . coefficientsAux - The array of FEM auxiliary basis coefficients for the elements 154320cf1dd8SToby Isaac . t - The time 154420cf1dd8SToby Isaac - u_tShift - A multiplier for the dF/du_t term (as opposed to the dF/du term) 154520cf1dd8SToby Isaac 15467a7aea1fSJed Brown Output Parameter: 154720cf1dd8SToby Isaac . elemMat - the element matrices for the Jacobian from each element 154820cf1dd8SToby Isaac 154920cf1dd8SToby Isaac Note: 155020cf1dd8SToby Isaac $ Loop over batch of elements (e): 155120cf1dd8SToby Isaac $ Loop over element matrix entries (f,fc,g,gc --> i,j): 155220cf1dd8SToby Isaac $ Loop over quadrature points (q): 155320cf1dd8SToby Isaac $ Make u_q and gradU_q (loops over fields,Nb,Ncomp) 155420cf1dd8SToby Isaac $ elemMat[i,j] += \psi^{fc}_f(q) g0_{fc,gc}(u, \nabla u) \phi^{gc}_g(q) 155520cf1dd8SToby Isaac $ + \psi^{fc}_f(q) \cdot g1_{fc,gc,dg}(u, \nabla u) \nabla\phi^{gc}_g(q) 155620cf1dd8SToby Isaac $ + \nabla\psi^{fc}_f(q) \cdot g2_{fc,gc,df}(u, \nabla u) \phi^{gc}_g(q) 155720cf1dd8SToby Isaac $ + \nabla\psi^{fc}_f(q) \cdot g3_{fc,gc,df,dg}(u, \nabla u) \nabla\phi^{gc}_g(q) 15582b99622eSMatthew G. Knepley Level: intermediate 155920cf1dd8SToby Isaac 156020cf1dd8SToby Isaac .seealso: PetscFEIntegrateJacobian(), PetscFEIntegrateResidual() 156120cf1dd8SToby Isaac @*/ 15624bee2e38SMatthew G. Knepley PetscErrorCode PetscFEIntegrateBdJacobian(PetscDS prob, PetscInt fieldI, PetscInt fieldJ, PetscInt Ne, PetscFEGeom *fgeom, 156320cf1dd8SToby Isaac const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscDS probAux, const PetscScalar coefficientsAux[], PetscReal t, PetscReal u_tshift, PetscScalar elemMat[]) 156420cf1dd8SToby Isaac { 15654bee2e38SMatthew G. Knepley PetscFE fe; 156620cf1dd8SToby Isaac PetscErrorCode ierr; 156720cf1dd8SToby Isaac 156820cf1dd8SToby Isaac PetscFunctionBegin; 15694bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1); 15704bee2e38SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fe);CHKERRQ(ierr); 15714bee2e38SMatthew 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);} 157220cf1dd8SToby Isaac PetscFunctionReturn(0); 157320cf1dd8SToby Isaac } 157420cf1dd8SToby Isaac 15752b99622eSMatthew G. Knepley /*@ 15762b99622eSMatthew G. Knepley PetscFEGetHeightSubspace - Get the subspace of this space for a mesh point of a given height 15772b99622eSMatthew G. Knepley 15782b99622eSMatthew G. Knepley Input Parameters: 15792b99622eSMatthew G. Knepley + fe - The finite element space 15802b99622eSMatthew G. Knepley - height - The height of the Plex point 15812b99622eSMatthew G. Knepley 15822b99622eSMatthew G. Knepley Output Parameter: 15832b99622eSMatthew G. Knepley . subfe - The subspace of this FE space 15842b99622eSMatthew G. Knepley 15852b99622eSMatthew G. Knepley Note: For example, if we want the subspace of this space for a face, we would choose height = 1. 15862b99622eSMatthew G. Knepley 15872b99622eSMatthew G. Knepley Level: advanced 15882b99622eSMatthew G. Knepley 15892b99622eSMatthew G. Knepley .seealso: PetscFECreateDefault() 15902b99622eSMatthew G. Knepley @*/ 159120cf1dd8SToby Isaac PetscErrorCode PetscFEGetHeightSubspace(PetscFE fe, PetscInt height, PetscFE *subfe) 159220cf1dd8SToby Isaac { 159320cf1dd8SToby Isaac PetscSpace P, subP; 159420cf1dd8SToby Isaac PetscDualSpace Q, subQ; 159520cf1dd8SToby Isaac PetscQuadrature subq; 159620cf1dd8SToby Isaac PetscFEType fetype; 159720cf1dd8SToby Isaac PetscInt dim, Nc; 159820cf1dd8SToby Isaac PetscErrorCode ierr; 159920cf1dd8SToby Isaac 160020cf1dd8SToby Isaac PetscFunctionBegin; 160120cf1dd8SToby Isaac PetscValidHeaderSpecific(fe, PETSCFE_CLASSID, 1); 160220cf1dd8SToby Isaac PetscValidPointer(subfe, 3); 160320cf1dd8SToby Isaac if (height == 0) { 160420cf1dd8SToby Isaac *subfe = fe; 160520cf1dd8SToby Isaac PetscFunctionReturn(0); 160620cf1dd8SToby Isaac } 160720cf1dd8SToby Isaac ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr); 160820cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr); 160920cf1dd8SToby Isaac ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 161020cf1dd8SToby Isaac ierr = PetscFEGetFaceQuadrature(fe, &subq);CHKERRQ(ierr); 161120cf1dd8SToby Isaac ierr = PetscDualSpaceGetDimension(Q, &dim);CHKERRQ(ierr); 161220cf1dd8SToby 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);} 161320cf1dd8SToby Isaac if (!fe->subspaces) {ierr = PetscCalloc1(dim, &fe->subspaces);CHKERRQ(ierr);} 161420cf1dd8SToby Isaac if (height <= dim) { 161520cf1dd8SToby Isaac if (!fe->subspaces[height-1]) { 161620cf1dd8SToby Isaac PetscFE sub; 16173f6b16c7SMatthew G. Knepley const char *name; 161820cf1dd8SToby Isaac 161920cf1dd8SToby Isaac ierr = PetscSpaceGetHeightSubspace(P, height, &subP);CHKERRQ(ierr); 162020cf1dd8SToby Isaac ierr = PetscDualSpaceGetHeightSubspace(Q, height, &subQ);CHKERRQ(ierr); 162120cf1dd8SToby Isaac ierr = PetscFECreate(PetscObjectComm((PetscObject) fe), &sub);CHKERRQ(ierr); 16223f6b16c7SMatthew G. Knepley ierr = PetscObjectGetName((PetscObject) fe, &name);CHKERRQ(ierr); 16233f6b16c7SMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) sub, name);CHKERRQ(ierr); 162420cf1dd8SToby Isaac ierr = PetscFEGetType(fe, &fetype);CHKERRQ(ierr); 162520cf1dd8SToby Isaac ierr = PetscFESetType(sub, fetype);CHKERRQ(ierr); 162620cf1dd8SToby Isaac ierr = PetscFESetBasisSpace(sub, subP);CHKERRQ(ierr); 162720cf1dd8SToby Isaac ierr = PetscFESetDualSpace(sub, subQ);CHKERRQ(ierr); 162820cf1dd8SToby Isaac ierr = PetscFESetNumComponents(sub, Nc);CHKERRQ(ierr); 162920cf1dd8SToby Isaac ierr = PetscFESetUp(sub);CHKERRQ(ierr); 163020cf1dd8SToby Isaac ierr = PetscFESetQuadrature(sub, subq);CHKERRQ(ierr); 163120cf1dd8SToby Isaac fe->subspaces[height-1] = sub; 163220cf1dd8SToby Isaac } 163320cf1dd8SToby Isaac *subfe = fe->subspaces[height-1]; 163420cf1dd8SToby Isaac } else { 163520cf1dd8SToby Isaac *subfe = NULL; 163620cf1dd8SToby Isaac } 163720cf1dd8SToby Isaac PetscFunctionReturn(0); 163820cf1dd8SToby Isaac } 163920cf1dd8SToby Isaac 164020cf1dd8SToby Isaac /*@ 164120cf1dd8SToby Isaac PetscFERefine - Create a "refined" PetscFE object that refines the reference cell into smaller copies. This is typically used 164220cf1dd8SToby Isaac to precondition a higher order method with a lower order method on a refined mesh having the same number of dofs (but more 164320cf1dd8SToby Isaac sparsity). It is also used to create an interpolation between regularly refined meshes. 164420cf1dd8SToby Isaac 1645d083f849SBarry Smith Collective on fem 164620cf1dd8SToby Isaac 164720cf1dd8SToby Isaac Input Parameter: 164820cf1dd8SToby Isaac . fe - The initial PetscFE 164920cf1dd8SToby Isaac 165020cf1dd8SToby Isaac Output Parameter: 165120cf1dd8SToby Isaac . feRef - The refined PetscFE 165220cf1dd8SToby Isaac 16532b99622eSMatthew G. Knepley Level: advanced 165420cf1dd8SToby Isaac 165520cf1dd8SToby Isaac .seealso: PetscFEType, PetscFECreate(), PetscFESetType() 165620cf1dd8SToby Isaac @*/ 165720cf1dd8SToby Isaac PetscErrorCode PetscFERefine(PetscFE fe, PetscFE *feRef) 165820cf1dd8SToby Isaac { 165920cf1dd8SToby Isaac PetscSpace P, Pref; 166020cf1dd8SToby Isaac PetscDualSpace Q, Qref; 166120cf1dd8SToby Isaac DM K, Kref; 166220cf1dd8SToby Isaac PetscQuadrature q, qref; 166320cf1dd8SToby Isaac const PetscReal *v0, *jac; 166420cf1dd8SToby Isaac PetscInt numComp, numSubelements; 16651ac17e89SToby Isaac PetscInt cStart, cEnd, c; 16661ac17e89SToby Isaac PetscDualSpace *cellSpaces; 166720cf1dd8SToby Isaac PetscErrorCode ierr; 166820cf1dd8SToby Isaac 166920cf1dd8SToby Isaac PetscFunctionBegin; 167020cf1dd8SToby Isaac ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr); 167120cf1dd8SToby Isaac ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr); 167220cf1dd8SToby Isaac ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr); 167320cf1dd8SToby Isaac ierr = PetscDualSpaceGetDM(Q, &K);CHKERRQ(ierr); 167420cf1dd8SToby Isaac /* Create space */ 167520cf1dd8SToby Isaac ierr = PetscObjectReference((PetscObject) P);CHKERRQ(ierr); 167620cf1dd8SToby Isaac Pref = P; 167720cf1dd8SToby Isaac /* Create dual space */ 167820cf1dd8SToby Isaac ierr = PetscDualSpaceDuplicate(Q, &Qref);CHKERRQ(ierr); 16791ac17e89SToby Isaac ierr = PetscDualSpaceSetType(Qref, PETSCDUALSPACEREFINED);CHKERRQ(ierr); 168020cf1dd8SToby Isaac ierr = DMRefine(K, PetscObjectComm((PetscObject) fe), &Kref);CHKERRQ(ierr); 168120cf1dd8SToby Isaac ierr = PetscDualSpaceSetDM(Qref, Kref);CHKERRQ(ierr); 16821ac17e89SToby Isaac ierr = DMPlexGetHeightStratum(Kref, 0, &cStart, &cEnd);CHKERRQ(ierr); 16831ac17e89SToby Isaac ierr = PetscMalloc1(cEnd - cStart, &cellSpaces);CHKERRQ(ierr); 16841ac17e89SToby Isaac /* TODO: fix for non-uniform refinement */ 16851ac17e89SToby Isaac for (c = 0; c < cEnd - cStart; c++) cellSpaces[c] = Q; 16861ac17e89SToby Isaac ierr = PetscDualSpaceRefinedSetCellSpaces(Qref, cellSpaces);CHKERRQ(ierr); 16871ac17e89SToby Isaac ierr = PetscFree(cellSpaces);CHKERRQ(ierr); 168820cf1dd8SToby Isaac ierr = DMDestroy(&Kref);CHKERRQ(ierr); 168920cf1dd8SToby Isaac ierr = PetscDualSpaceSetUp(Qref);CHKERRQ(ierr); 169020cf1dd8SToby Isaac /* Create element */ 169120cf1dd8SToby Isaac ierr = PetscFECreate(PetscObjectComm((PetscObject) fe), feRef);CHKERRQ(ierr); 169220cf1dd8SToby Isaac ierr = PetscFESetType(*feRef, PETSCFECOMPOSITE);CHKERRQ(ierr); 169320cf1dd8SToby Isaac ierr = PetscFESetBasisSpace(*feRef, Pref);CHKERRQ(ierr); 169420cf1dd8SToby Isaac ierr = PetscFESetDualSpace(*feRef, Qref);CHKERRQ(ierr); 169520cf1dd8SToby Isaac ierr = PetscFEGetNumComponents(fe, &numComp);CHKERRQ(ierr); 169620cf1dd8SToby Isaac ierr = PetscFESetNumComponents(*feRef, numComp);CHKERRQ(ierr); 169720cf1dd8SToby Isaac ierr = PetscFESetUp(*feRef);CHKERRQ(ierr); 169820cf1dd8SToby Isaac ierr = PetscSpaceDestroy(&Pref);CHKERRQ(ierr); 169920cf1dd8SToby Isaac ierr = PetscDualSpaceDestroy(&Qref);CHKERRQ(ierr); 170020cf1dd8SToby Isaac /* Create quadrature */ 170120cf1dd8SToby Isaac ierr = PetscFECompositeGetMapping(*feRef, &numSubelements, &v0, &jac, NULL);CHKERRQ(ierr); 170220cf1dd8SToby Isaac ierr = PetscQuadratureExpandComposite(q, numSubelements, v0, jac, &qref);CHKERRQ(ierr); 170320cf1dd8SToby Isaac ierr = PetscFESetQuadrature(*feRef, qref);CHKERRQ(ierr); 170420cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&qref);CHKERRQ(ierr); 170520cf1dd8SToby Isaac PetscFunctionReturn(0); 170620cf1dd8SToby Isaac } 170720cf1dd8SToby Isaac 170820cf1dd8SToby Isaac /*@C 170920cf1dd8SToby Isaac PetscFECreateDefault - Create a PetscFE for basic FEM computation 171020cf1dd8SToby Isaac 1711d083f849SBarry Smith Collective 171220cf1dd8SToby Isaac 171320cf1dd8SToby Isaac Input Parameters: 17147be5e748SToby Isaac + comm - The MPI comm 171520cf1dd8SToby Isaac . dim - The spatial dimension 171620cf1dd8SToby Isaac . Nc - The number of components 171720cf1dd8SToby Isaac . isSimplex - Flag for simplex reference cell, otherwise its a tensor product 171820cf1dd8SToby Isaac . prefix - The options prefix, or NULL 1719727cddd5SJacob Faibussowitsch - qorder - The quadrature order or PETSC_DETERMINE to use PetscSpace polynomial degree 172020cf1dd8SToby Isaac 172120cf1dd8SToby Isaac Output Parameter: 172220cf1dd8SToby Isaac . fem - The PetscFE object 172320cf1dd8SToby Isaac 1724e703855dSMatthew G. Knepley Note: 1725e703855dSMatthew G. Knepley Each object is SetFromOption() during creation, so that the object may be customized from the command line. 1726e703855dSMatthew G. Knepley 172720cf1dd8SToby Isaac Level: beginner 172820cf1dd8SToby Isaac 172920cf1dd8SToby Isaac .seealso: PetscFECreate(), PetscSpaceCreate(), PetscDualSpaceCreate() 173020cf1dd8SToby Isaac @*/ 17317be5e748SToby Isaac PetscErrorCode PetscFECreateDefault(MPI_Comm comm, PetscInt dim, PetscInt Nc, PetscBool isSimplex, const char prefix[], PetscInt qorder, PetscFE *fem) 173220cf1dd8SToby Isaac { 173320cf1dd8SToby Isaac PetscQuadrature q, fq; 173420cf1dd8SToby Isaac DM K; 173520cf1dd8SToby Isaac PetscSpace P; 173620cf1dd8SToby Isaac PetscDualSpace Q; 173720cf1dd8SToby Isaac PetscInt order, quadPointsPerEdge; 173820cf1dd8SToby Isaac PetscBool tensor = isSimplex ? PETSC_FALSE : PETSC_TRUE; 173920cf1dd8SToby Isaac PetscErrorCode ierr; 174020cf1dd8SToby Isaac 174120cf1dd8SToby Isaac PetscFunctionBegin; 174220cf1dd8SToby Isaac /* Create space */ 17437be5e748SToby Isaac ierr = PetscSpaceCreate(comm, &P);CHKERRQ(ierr); 174420cf1dd8SToby Isaac ierr = PetscObjectSetOptionsPrefix((PetscObject) P, prefix);CHKERRQ(ierr); 174520cf1dd8SToby Isaac ierr = PetscSpacePolynomialSetTensor(P, tensor);CHKERRQ(ierr); 174620cf1dd8SToby Isaac ierr = PetscSpaceSetNumComponents(P, Nc);CHKERRQ(ierr); 174720cf1dd8SToby Isaac ierr = PetscSpaceSetNumVariables(P, dim);CHKERRQ(ierr); 1748028afddaSToby Isaac ierr = PetscSpaceSetFromOptions(P);CHKERRQ(ierr); 174920cf1dd8SToby Isaac ierr = PetscSpaceSetUp(P);CHKERRQ(ierr); 175020cf1dd8SToby Isaac ierr = PetscSpaceGetDegree(P, &order, NULL);CHKERRQ(ierr); 175120cf1dd8SToby Isaac ierr = PetscSpacePolynomialGetTensor(P, &tensor);CHKERRQ(ierr); 175220cf1dd8SToby Isaac /* Create dual space */ 17537be5e748SToby Isaac ierr = PetscDualSpaceCreate(comm, &Q);CHKERRQ(ierr); 175420cf1dd8SToby Isaac ierr = PetscDualSpaceSetType(Q,PETSCDUALSPACELAGRANGE);CHKERRQ(ierr); 175520cf1dd8SToby Isaac ierr = PetscObjectSetOptionsPrefix((PetscObject) Q, prefix);CHKERRQ(ierr); 175620cf1dd8SToby Isaac ierr = PetscDualSpaceCreateReferenceCell(Q, dim, isSimplex, &K);CHKERRQ(ierr); 175720cf1dd8SToby Isaac ierr = PetscDualSpaceSetDM(Q, K);CHKERRQ(ierr); 175820cf1dd8SToby Isaac ierr = DMDestroy(&K);CHKERRQ(ierr); 175920cf1dd8SToby Isaac ierr = PetscDualSpaceSetNumComponents(Q, Nc);CHKERRQ(ierr); 176020cf1dd8SToby Isaac ierr = PetscDualSpaceSetOrder(Q, order);CHKERRQ(ierr); 176120cf1dd8SToby Isaac ierr = PetscDualSpaceLagrangeSetTensor(Q, tensor);CHKERRQ(ierr); 176220cf1dd8SToby Isaac ierr = PetscDualSpaceSetFromOptions(Q);CHKERRQ(ierr); 176320cf1dd8SToby Isaac ierr = PetscDualSpaceSetUp(Q);CHKERRQ(ierr); 176420cf1dd8SToby Isaac /* Create element */ 17657be5e748SToby Isaac ierr = PetscFECreate(comm, fem);CHKERRQ(ierr); 176620cf1dd8SToby Isaac ierr = PetscObjectSetOptionsPrefix((PetscObject) *fem, prefix);CHKERRQ(ierr); 176720cf1dd8SToby Isaac ierr = PetscFESetBasisSpace(*fem, P);CHKERRQ(ierr); 176820cf1dd8SToby Isaac ierr = PetscFESetDualSpace(*fem, Q);CHKERRQ(ierr); 176920cf1dd8SToby Isaac ierr = PetscFESetNumComponents(*fem, Nc);CHKERRQ(ierr); 177091e89cf0SMatthew G. Knepley ierr = PetscFESetFromOptions(*fem);CHKERRQ(ierr); 177120cf1dd8SToby Isaac ierr = PetscFESetUp(*fem);CHKERRQ(ierr); 177220cf1dd8SToby Isaac ierr = PetscSpaceDestroy(&P);CHKERRQ(ierr); 177320cf1dd8SToby Isaac ierr = PetscDualSpaceDestroy(&Q);CHKERRQ(ierr); 177420cf1dd8SToby Isaac /* Create quadrature (with specified order if given) */ 177520cf1dd8SToby Isaac qorder = qorder >= 0 ? qorder : order; 177620cf1dd8SToby Isaac ierr = PetscObjectOptionsBegin((PetscObject)*fem);CHKERRQ(ierr); 17775a856986SBarry Smith ierr = PetscOptionsBoundedInt("-petscfe_default_quadrature_order","Quadrature order is one less than quadrature points per edge","PetscFECreateDefault",qorder,&qorder,NULL,0);CHKERRQ(ierr); 177820cf1dd8SToby Isaac ierr = PetscOptionsEnd();CHKERRQ(ierr); 177920cf1dd8SToby Isaac quadPointsPerEdge = PetscMax(qorder + 1,1); 178020cf1dd8SToby Isaac if (isSimplex) { 1781e6a796c3SToby Isaac ierr = PetscDTStroudConicalQuadrature(dim, 1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr); 1782e6a796c3SToby Isaac ierr = PetscDTStroudConicalQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr); 17834ccfa306SStefano Zampini } else { 178420cf1dd8SToby Isaac ierr = PetscDTGaussTensorQuadrature(dim, 1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr); 178520cf1dd8SToby Isaac ierr = PetscDTGaussTensorQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr); 178620cf1dd8SToby Isaac } 178720cf1dd8SToby Isaac ierr = PetscFESetQuadrature(*fem, q);CHKERRQ(ierr); 178820cf1dd8SToby Isaac ierr = PetscFESetFaceQuadrature(*fem, fq);CHKERRQ(ierr); 178920cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&q);CHKERRQ(ierr); 179020cf1dd8SToby Isaac ierr = PetscQuadratureDestroy(&fq);CHKERRQ(ierr); 179120cf1dd8SToby Isaac PetscFunctionReturn(0); 179220cf1dd8SToby Isaac } 17933f6b16c7SMatthew G. Knepley 1794e703855dSMatthew G. Knepley /*@ 1795e703855dSMatthew G. Knepley PetscFECreateLagrange - Create a PetscFE for the basic Lagrange space of degree k 1796e703855dSMatthew G. Knepley 1797e703855dSMatthew G. Knepley Collective 1798e703855dSMatthew G. Knepley 1799e703855dSMatthew G. Knepley Input Parameters: 1800e703855dSMatthew G. Knepley + comm - The MPI comm 1801e703855dSMatthew G. Knepley . dim - The spatial dimension 1802e703855dSMatthew G. Knepley . Nc - The number of components 1803e703855dSMatthew G. Knepley . isSimplex - Flag for simplex reference cell, otherwise its a tensor product 1804e703855dSMatthew G. Knepley . k - The degree k of the space 1805e703855dSMatthew G. Knepley - qorder - The quadrature order or PETSC_DETERMINE to use PetscSpace polynomial degree 1806e703855dSMatthew G. Knepley 1807e703855dSMatthew G. Knepley Output Parameter: 1808e703855dSMatthew G. Knepley . fem - The PetscFE object 1809e703855dSMatthew G. Knepley 1810e703855dSMatthew G. Knepley Level: beginner 1811e703855dSMatthew G. Knepley 1812e703855dSMatthew G. Knepley Notes: 1813e703855dSMatthew G. Knepley For simplices, this element is the space of maximum polynomial degree k, otherwise it is a tensor product of 1D polynomials, each with maximal degree k. 1814e703855dSMatthew G. Knepley 1815e703855dSMatthew G. Knepley .seealso: PetscFECreate(), PetscSpaceCreate(), PetscDualSpaceCreate() 1816e703855dSMatthew G. Knepley @*/ 1817e703855dSMatthew G. Knepley PetscErrorCode PetscFECreateLagrange(MPI_Comm comm, PetscInt dim, PetscInt Nc, PetscBool isSimplex, PetscInt k, PetscInt qorder, PetscFE *fem) 1818e703855dSMatthew G. Knepley { 1819e703855dSMatthew G. Knepley PetscQuadrature q, fq; 1820e703855dSMatthew G. Knepley DM K; 1821e703855dSMatthew G. Knepley PetscSpace P; 1822e703855dSMatthew G. Knepley PetscDualSpace Q; 1823e703855dSMatthew G. Knepley PetscInt quadPointsPerEdge; 1824e703855dSMatthew G. Knepley PetscBool tensor = isSimplex ? PETSC_FALSE : PETSC_TRUE; 1825e703855dSMatthew G. Knepley char name[64]; 1826e703855dSMatthew G. Knepley PetscErrorCode ierr; 1827e703855dSMatthew G. Knepley 1828e703855dSMatthew G. Knepley PetscFunctionBegin; 1829e703855dSMatthew G. Knepley /* Create space */ 1830e703855dSMatthew G. Knepley ierr = PetscSpaceCreate(comm, &P);CHKERRQ(ierr); 1831e703855dSMatthew G. Knepley ierr = PetscSpaceSetType(P, PETSCSPACEPOLYNOMIAL);CHKERRQ(ierr); 1832e703855dSMatthew G. Knepley ierr = PetscSpacePolynomialSetTensor(P, tensor);CHKERRQ(ierr); 1833e703855dSMatthew G. Knepley ierr = PetscSpaceSetNumComponents(P, Nc);CHKERRQ(ierr); 1834e703855dSMatthew G. Knepley ierr = PetscSpaceSetNumVariables(P, dim);CHKERRQ(ierr); 1835e703855dSMatthew G. Knepley ierr = PetscSpaceSetDegree(P, k, PETSC_DETERMINE);CHKERRQ(ierr); 1836e703855dSMatthew G. Knepley ierr = PetscSpaceSetUp(P);CHKERRQ(ierr); 1837e703855dSMatthew G. Knepley /* Create dual space */ 1838e703855dSMatthew G. Knepley ierr = PetscDualSpaceCreate(comm, &Q);CHKERRQ(ierr); 1839e703855dSMatthew G. Knepley ierr = PetscDualSpaceSetType(Q, PETSCDUALSPACELAGRANGE);CHKERRQ(ierr); 1840e703855dSMatthew G. Knepley ierr = PetscDualSpaceCreateReferenceCell(Q, dim, isSimplex, &K);CHKERRQ(ierr); 1841e703855dSMatthew G. Knepley ierr = PetscDualSpaceSetDM(Q, K);CHKERRQ(ierr); 1842e703855dSMatthew G. Knepley ierr = DMDestroy(&K);CHKERRQ(ierr); 1843e703855dSMatthew G. Knepley ierr = PetscDualSpaceSetNumComponents(Q, Nc);CHKERRQ(ierr); 1844e703855dSMatthew G. Knepley ierr = PetscDualSpaceSetOrder(Q, k);CHKERRQ(ierr); 1845e703855dSMatthew G. Knepley ierr = PetscDualSpaceLagrangeSetTensor(Q, tensor);CHKERRQ(ierr); 1846e703855dSMatthew G. Knepley ierr = PetscDualSpaceSetUp(Q);CHKERRQ(ierr); 1847e703855dSMatthew G. Knepley /* Create element */ 1848e703855dSMatthew G. Knepley ierr = PetscFECreate(comm, fem);CHKERRQ(ierr); 1849e703855dSMatthew G. Knepley ierr = PetscSNPrintf(name, 64, "P%d", (int) k);CHKERRQ(ierr); 1850e703855dSMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) *fem, name);CHKERRQ(ierr); 1851e703855dSMatthew G. Knepley ierr = PetscFESetType(*fem, PETSCFEBASIC);CHKERRQ(ierr); 1852e703855dSMatthew G. Knepley ierr = PetscFESetBasisSpace(*fem, P);CHKERRQ(ierr); 1853e703855dSMatthew G. Knepley ierr = PetscFESetDualSpace(*fem, Q);CHKERRQ(ierr); 1854e703855dSMatthew G. Knepley ierr = PetscFESetNumComponents(*fem, Nc);CHKERRQ(ierr); 1855e703855dSMatthew G. Knepley ierr = PetscFESetUp(*fem);CHKERRQ(ierr); 1856e703855dSMatthew G. Knepley ierr = PetscSpaceDestroy(&P);CHKERRQ(ierr); 1857e703855dSMatthew G. Knepley ierr = PetscDualSpaceDestroy(&Q);CHKERRQ(ierr); 1858e703855dSMatthew G. Knepley /* Create quadrature (with specified order if given) */ 1859e703855dSMatthew G. Knepley qorder = qorder >= 0 ? qorder : k; 1860e703855dSMatthew G. Knepley quadPointsPerEdge = PetscMax(qorder + 1,1); 1861e703855dSMatthew G. Knepley if (isSimplex) { 1862e6a796c3SToby Isaac ierr = PetscDTStroudConicalQuadrature(dim, 1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr); 1863e6a796c3SToby Isaac ierr = PetscDTStroudConicalQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr); 1864e703855dSMatthew G. Knepley } else { 1865e703855dSMatthew G. Knepley ierr = PetscDTGaussTensorQuadrature(dim, 1, quadPointsPerEdge, -1.0, 1.0, &q);CHKERRQ(ierr); 1866e703855dSMatthew G. Knepley ierr = PetscDTGaussTensorQuadrature(dim-1, 1, quadPointsPerEdge, -1.0, 1.0, &fq);CHKERRQ(ierr); 1867e703855dSMatthew G. Knepley } 1868e703855dSMatthew G. Knepley ierr = PetscFESetQuadrature(*fem, q);CHKERRQ(ierr); 1869e703855dSMatthew G. Knepley ierr = PetscFESetFaceQuadrature(*fem, fq);CHKERRQ(ierr); 1870e703855dSMatthew G. Knepley ierr = PetscQuadratureDestroy(&q);CHKERRQ(ierr); 1871e703855dSMatthew G. Knepley ierr = PetscQuadratureDestroy(&fq);CHKERRQ(ierr); 1872e703855dSMatthew G. Knepley PetscFunctionReturn(0); 1873e703855dSMatthew G. Knepley } 1874e703855dSMatthew G. Knepley 18753f6b16c7SMatthew G. Knepley /*@C 18763f6b16c7SMatthew G. Knepley PetscFESetName - Names the FE and its subobjects 18773f6b16c7SMatthew G. Knepley 18783f6b16c7SMatthew G. Knepley Not collective 18793f6b16c7SMatthew G. Knepley 18803f6b16c7SMatthew G. Knepley Input Parameters: 18813f6b16c7SMatthew G. Knepley + fe - The PetscFE 18823f6b16c7SMatthew G. Knepley - name - The name 18833f6b16c7SMatthew G. Knepley 18842b99622eSMatthew G. Knepley Level: intermediate 18853f6b16c7SMatthew G. Knepley 18863f6b16c7SMatthew G. Knepley .seealso: PetscFECreate(), PetscSpaceCreate(), PetscDualSpaceCreate() 18873f6b16c7SMatthew G. Knepley @*/ 18883f6b16c7SMatthew G. Knepley PetscErrorCode PetscFESetName(PetscFE fe, const char name[]) 18893f6b16c7SMatthew G. Knepley { 18903f6b16c7SMatthew G. Knepley PetscSpace P; 18913f6b16c7SMatthew G. Knepley PetscDualSpace Q; 18923f6b16c7SMatthew G. Knepley PetscErrorCode ierr; 18933f6b16c7SMatthew G. Knepley 18943f6b16c7SMatthew G. Knepley PetscFunctionBegin; 18953f6b16c7SMatthew G. Knepley ierr = PetscFEGetBasisSpace(fe, &P);CHKERRQ(ierr); 18963f6b16c7SMatthew G. Knepley ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr); 18973f6b16c7SMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) fe, name);CHKERRQ(ierr); 18983f6b16c7SMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) P, name);CHKERRQ(ierr); 18993f6b16c7SMatthew G. Knepley ierr = PetscObjectSetName((PetscObject) Q, name);CHKERRQ(ierr); 19003f6b16c7SMatthew G. Knepley PetscFunctionReturn(0); 19013f6b16c7SMatthew G. Knepley } 1902a8f1f9e5SMatthew G. Knepley 1903ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEEvaluateFieldJets_Internal(PetscDS ds, PetscInt Nf, PetscInt r, PetscInt q, PetscTabulation T[], PetscFEGeom *fegeom, const PetscScalar coefficients[], const PetscScalar coefficients_t[], PetscScalar u[], PetscScalar u_x[], PetscScalar u_t[]) 1904a8f1f9e5SMatthew G. Knepley { 1905a8f1f9e5SMatthew G. Knepley PetscInt dOffset = 0, fOffset = 0, f; 1906a8f1f9e5SMatthew G. Knepley PetscErrorCode ierr; 1907a8f1f9e5SMatthew G. Knepley 1908a8f1f9e5SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 1909a8f1f9e5SMatthew G. Knepley PetscFE fe; 1910ef0bb6c7SMatthew G. Knepley const PetscInt cdim = T[f]->cdim; 1911ef0bb6c7SMatthew G. Knepley const PetscInt Nq = T[f]->Np; 1912ef0bb6c7SMatthew G. Knepley const PetscInt Nbf = T[f]->Nb; 1913ef0bb6c7SMatthew G. Knepley const PetscInt Ncf = T[f]->Nc; 1914ef0bb6c7SMatthew G. Knepley const PetscReal *Bq = &T[f]->T[0][(r*Nq+q)*Nbf*Ncf]; 1915ef0bb6c7SMatthew G. Knepley const PetscReal *Dq = &T[f]->T[1][(r*Nq+q)*Nbf*Ncf*cdim]; 1916a8f1f9e5SMatthew G. Knepley PetscInt b, c, d; 1917a8f1f9e5SMatthew G. Knepley 1918a8f1f9e5SMatthew G. Knepley ierr = PetscDSGetDiscretization(ds, f, (PetscObject *) &fe);CHKERRQ(ierr); 1919a8f1f9e5SMatthew G. Knepley for (c = 0; c < Ncf; ++c) u[fOffset+c] = 0.0; 1920ef0bb6c7SMatthew G. Knepley for (d = 0; d < cdim*Ncf; ++d) u_x[fOffset*cdim+d] = 0.0; 1921a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nbf; ++b) { 1922a8f1f9e5SMatthew G. Knepley for (c = 0; c < Ncf; ++c) { 1923a8f1f9e5SMatthew G. Knepley const PetscInt cidx = b*Ncf+c; 1924a8f1f9e5SMatthew G. Knepley 1925a8f1f9e5SMatthew G. Knepley u[fOffset+c] += Bq[cidx]*coefficients[dOffset+b]; 1926ef0bb6c7SMatthew G. Knepley for (d = 0; d < cdim; ++d) u_x[(fOffset+c)*cdim+d] += Dq[cidx*cdim+d]*coefficients[dOffset+b]; 1927a8f1f9e5SMatthew G. Knepley } 1928a8f1f9e5SMatthew G. Knepley } 19292edcad52SToby Isaac ierr = PetscFEPushforward(fe, fegeom, 1, &u[fOffset]);CHKERRQ(ierr); 19302edcad52SToby Isaac ierr = PetscFEPushforwardGradient(fe, fegeom, 1, &u_x[fOffset*cdim]);CHKERRQ(ierr); 1931a8f1f9e5SMatthew G. Knepley if (u_t) { 1932a8f1f9e5SMatthew G. Knepley for (c = 0; c < Ncf; ++c) u_t[fOffset+c] = 0.0; 1933a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nbf; ++b) { 1934a8f1f9e5SMatthew G. Knepley for (c = 0; c < Ncf; ++c) { 1935a8f1f9e5SMatthew G. Knepley const PetscInt cidx = b*Ncf+c; 1936a8f1f9e5SMatthew G. Knepley 1937a8f1f9e5SMatthew G. Knepley u_t[fOffset+c] += Bq[cidx]*coefficients_t[dOffset+b]; 1938a8f1f9e5SMatthew G. Knepley } 1939a8f1f9e5SMatthew G. Knepley } 19402edcad52SToby Isaac ierr = PetscFEPushforward(fe, fegeom, 1, &u_t[fOffset]);CHKERRQ(ierr); 1941a8f1f9e5SMatthew G. Knepley } 1942a8f1f9e5SMatthew G. Knepley fOffset += Ncf; 1943a8f1f9e5SMatthew G. Knepley dOffset += Nbf; 1944a8f1f9e5SMatthew G. Knepley } 1945a8f1f9e5SMatthew G. Knepley return 0; 1946a8f1f9e5SMatthew G. Knepley } 1947a8f1f9e5SMatthew G. Knepley 1948a8f1f9e5SMatthew G. Knepley PetscErrorCode PetscFEEvaluateFaceFields_Internal(PetscDS prob, PetscInt field, PetscInt faceLoc, const PetscScalar coefficients[], PetscScalar u[]) 1949a8f1f9e5SMatthew G. Knepley { 1950a8f1f9e5SMatthew G. Knepley PetscFE fe; 1951ef0bb6c7SMatthew G. Knepley PetscTabulation Tc; 1952ef0bb6c7SMatthew G. Knepley PetscInt b, c; 1953a8f1f9e5SMatthew G. Knepley PetscErrorCode ierr; 1954a8f1f9e5SMatthew G. Knepley 1955a8f1f9e5SMatthew G. Knepley if (!prob) return 0; 1956a8f1f9e5SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 1957ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetFaceCentroidTabulation(fe, &Tc);CHKERRQ(ierr); 1958ef0bb6c7SMatthew G. Knepley { 1959ef0bb6c7SMatthew G. Knepley const PetscReal *faceBasis = Tc->T[0]; 1960ef0bb6c7SMatthew G. Knepley const PetscInt Nb = Tc->Nb; 1961ef0bb6c7SMatthew G. Knepley const PetscInt Nc = Tc->Nc; 1962ef0bb6c7SMatthew G. Knepley 1963a8f1f9e5SMatthew G. Knepley for (c = 0; c < Nc; ++c) {u[c] = 0.0;} 1964a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nb; ++b) { 1965a8f1f9e5SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 1966a8f1f9e5SMatthew G. Knepley const PetscInt cidx = b*Nc+c; 1967a8f1f9e5SMatthew G. Knepley 1968a8f1f9e5SMatthew G. Knepley u[c] += coefficients[cidx]*faceBasis[faceLoc*Nb*Nc+cidx]; 1969a8f1f9e5SMatthew G. Knepley } 1970a8f1f9e5SMatthew G. Knepley } 1971ef0bb6c7SMatthew G. Knepley } 1972a8f1f9e5SMatthew G. Knepley return 0; 1973a8f1f9e5SMatthew G. Knepley } 1974a8f1f9e5SMatthew G. Knepley 1975ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEUpdateElementVec_Internal(PetscFE fe, PetscTabulation T, PetscInt r, PetscScalar tmpBasis[], PetscScalar tmpBasisDer[], PetscFEGeom *fegeom, PetscScalar f0[], PetscScalar f1[], PetscScalar elemVec[]) 1976a8f1f9e5SMatthew G. Knepley { 1977ef0bb6c7SMatthew G. Knepley const PetscInt dim = T->cdim; 1978ef0bb6c7SMatthew G. Knepley const PetscInt Nq = T->Np; 1979ef0bb6c7SMatthew G. Knepley const PetscInt Nb = T->Nb; 1980ef0bb6c7SMatthew G. Knepley const PetscInt Nc = T->Nc; 1981ef0bb6c7SMatthew G. Knepley const PetscReal *basis = &T->T[0][r*Nq*Nb*Nc]; 1982ef0bb6c7SMatthew G. Knepley const PetscReal *basisDer = &T->T[1][r*Nq*Nb*Nc*dim]; 1983a8f1f9e5SMatthew G. Knepley PetscInt q, b, c, d; 1984a8f1f9e5SMatthew G. Knepley PetscErrorCode ierr; 1985a8f1f9e5SMatthew G. Knepley 1986a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nb; ++b) elemVec[b] = 0.0; 1987a8f1f9e5SMatthew G. Knepley for (q = 0; q < Nq; ++q) { 1988a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nb; ++b) { 1989a8f1f9e5SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 1990a8f1f9e5SMatthew G. Knepley const PetscInt bcidx = b*Nc+c; 1991a8f1f9e5SMatthew G. Knepley 1992a8f1f9e5SMatthew G. Knepley tmpBasis[bcidx] = basis[q*Nb*Nc+bcidx]; 1993a8f1f9e5SMatthew G. Knepley for (d = 0; d < dim; ++d) tmpBasisDer[bcidx*dim+d] = basisDer[q*Nb*Nc*dim+bcidx*dim+d]; 1994a8f1f9e5SMatthew G. Knepley } 1995a8f1f9e5SMatthew G. Knepley } 19962edcad52SToby Isaac ierr = PetscFEPushforward(fe, fegeom, Nb, tmpBasis);CHKERRQ(ierr); 19972edcad52SToby Isaac ierr = PetscFEPushforwardGradient(fe, fegeom, Nb, tmpBasisDer);CHKERRQ(ierr); 1998a8f1f9e5SMatthew G. Knepley for (b = 0; b < Nb; ++b) { 1999a8f1f9e5SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 2000a8f1f9e5SMatthew G. Knepley const PetscInt bcidx = b*Nc+c; 2001a8f1f9e5SMatthew G. Knepley const PetscInt qcidx = q*Nc+c; 2002a8f1f9e5SMatthew G. Knepley 2003a8f1f9e5SMatthew G. Knepley elemVec[b] += tmpBasis[bcidx]*f0[qcidx]; 2004a8f1f9e5SMatthew G. Knepley for (d = 0; d < dim; ++d) elemVec[b] += tmpBasisDer[bcidx*dim+d]*f1[qcidx*dim+d]; 2005a8f1f9e5SMatthew G. Knepley } 2006a8f1f9e5SMatthew G. Knepley } 2007a8f1f9e5SMatthew G. Knepley } 2008a8f1f9e5SMatthew G. Knepley return(0); 2009a8f1f9e5SMatthew G. Knepley } 2010a8f1f9e5SMatthew G. Knepley 2011ef0bb6c7SMatthew G. Knepley PetscErrorCode PetscFEUpdateElementMat_Internal(PetscFE feI, PetscFE feJ, PetscInt r, PetscInt q, PetscTabulation TI, PetscScalar tmpBasisI[], PetscScalar tmpBasisDerI[], PetscTabulation TJ, 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[]) 2012a8f1f9e5SMatthew G. Knepley { 2013ef0bb6c7SMatthew G. Knepley const PetscInt dim = TI->cdim; 2014ef0bb6c7SMatthew G. Knepley const PetscInt NqI = TI->Np; 2015ef0bb6c7SMatthew G. Knepley const PetscInt NbI = TI->Nb; 2016ef0bb6c7SMatthew G. Knepley const PetscInt NcI = TI->Nc; 2017ef0bb6c7SMatthew G. Knepley const PetscReal *basisI = &TI->T[0][(r*NqI+q)*NbI*NcI]; 2018ef0bb6c7SMatthew G. Knepley const PetscReal *basisDerI = &TI->T[1][(r*NqI+q)*NbI*NcI*dim]; 2019ef0bb6c7SMatthew G. Knepley const PetscInt NqJ = TJ->Np; 2020ef0bb6c7SMatthew G. Knepley const PetscInt NbJ = TJ->Nb; 2021ef0bb6c7SMatthew G. Knepley const PetscInt NcJ = TJ->Nc; 2022ef0bb6c7SMatthew G. Knepley const PetscReal *basisJ = &TJ->T[0][(r*NqJ+q)*NbJ*NcJ]; 2023ef0bb6c7SMatthew G. Knepley const PetscReal *basisDerJ = &TJ->T[1][(r*NqJ+q)*NbJ*NcJ*dim]; 2024a8f1f9e5SMatthew G. Knepley PetscInt f, fc, g, gc, df, dg; 2025a8f1f9e5SMatthew G. Knepley PetscErrorCode ierr; 2026a8f1f9e5SMatthew G. Knepley 2027a8f1f9e5SMatthew G. Knepley for (f = 0; f < NbI; ++f) { 2028a8f1f9e5SMatthew G. Knepley for (fc = 0; fc < NcI; ++fc) { 2029a8f1f9e5SMatthew G. Knepley const PetscInt fidx = f*NcI+fc; /* Test function basis index */ 2030a8f1f9e5SMatthew G. Knepley 2031a8f1f9e5SMatthew G. Knepley tmpBasisI[fidx] = basisI[fidx]; 2032a8f1f9e5SMatthew G. Knepley for (df = 0; df < dim; ++df) tmpBasisDerI[fidx*dim+df] = basisDerI[fidx*dim+df]; 2033a8f1f9e5SMatthew G. Knepley } 2034a8f1f9e5SMatthew G. Knepley } 20352edcad52SToby Isaac ierr = PetscFEPushforward(feI, fegeom, NbI, tmpBasisI);CHKERRQ(ierr); 20362edcad52SToby Isaac ierr = PetscFEPushforwardGradient(feI, fegeom, NbI, tmpBasisDerI);CHKERRQ(ierr); 2037a8f1f9e5SMatthew G. Knepley for (g = 0; g < NbJ; ++g) { 2038a8f1f9e5SMatthew G. Knepley for (gc = 0; gc < NcJ; ++gc) { 2039a8f1f9e5SMatthew G. Knepley const PetscInt gidx = g*NcJ+gc; /* Trial function basis index */ 2040a8f1f9e5SMatthew G. Knepley 2041a8f1f9e5SMatthew G. Knepley tmpBasisJ[gidx] = basisJ[gidx]; 2042a8f1f9e5SMatthew G. Knepley for (dg = 0; dg < dim; ++dg) tmpBasisDerJ[gidx*dim+dg] = basisDerJ[gidx*dim+dg]; 2043a8f1f9e5SMatthew G. Knepley } 2044a8f1f9e5SMatthew G. Knepley } 20452edcad52SToby Isaac ierr = PetscFEPushforward(feJ, fegeom, NbJ, tmpBasisJ);CHKERRQ(ierr); 20462edcad52SToby Isaac ierr = PetscFEPushforwardGradient(feJ, fegeom, NbJ, tmpBasisDerJ);CHKERRQ(ierr); 2047a8f1f9e5SMatthew G. Knepley for (f = 0; f < NbI; ++f) { 2048a8f1f9e5SMatthew G. Knepley for (fc = 0; fc < NcI; ++fc) { 2049a8f1f9e5SMatthew G. Knepley const PetscInt fidx = f*NcI+fc; /* Test function basis index */ 2050a8f1f9e5SMatthew G. Knepley const PetscInt i = offsetI+f; /* Element matrix row */ 2051a8f1f9e5SMatthew G. Knepley for (g = 0; g < NbJ; ++g) { 2052a8f1f9e5SMatthew G. Knepley for (gc = 0; gc < NcJ; ++gc) { 2053a8f1f9e5SMatthew G. Knepley const PetscInt gidx = g*NcJ+gc; /* Trial function basis index */ 2054a8f1f9e5SMatthew G. Knepley const PetscInt j = offsetJ+g; /* Element matrix column */ 2055a8f1f9e5SMatthew G. Knepley const PetscInt fOff = eOffset+i*totDim+j; 2056a8f1f9e5SMatthew G. Knepley 2057a8f1f9e5SMatthew G. Knepley elemMat[fOff] += tmpBasisI[fidx]*g0[fc*NcJ+gc]*tmpBasisJ[gidx]; 2058a8f1f9e5SMatthew G. Knepley for (df = 0; df < dim; ++df) { 2059a8f1f9e5SMatthew G. Knepley elemMat[fOff] += tmpBasisI[fidx]*g1[(fc*NcJ+gc)*dim+df]*tmpBasisDerJ[gidx*dim+df]; 2060a8f1f9e5SMatthew G. Knepley elemMat[fOff] += tmpBasisDerI[fidx*dim+df]*g2[(fc*NcJ+gc)*dim+df]*tmpBasisJ[gidx]; 2061a8f1f9e5SMatthew G. Knepley for (dg = 0; dg < dim; ++dg) { 2062a8f1f9e5SMatthew G. Knepley elemMat[fOff] += tmpBasisDerI[fidx*dim+df]*g3[((fc*NcJ+gc)*dim+df)*dim+dg]*tmpBasisDerJ[gidx*dim+dg]; 2063a8f1f9e5SMatthew G. Knepley } 2064a8f1f9e5SMatthew G. Knepley } 2065a8f1f9e5SMatthew G. Knepley } 2066a8f1f9e5SMatthew G. Knepley } 2067a8f1f9e5SMatthew G. Knepley } 2068a8f1f9e5SMatthew G. Knepley } 2069a8f1f9e5SMatthew G. Knepley return(0); 2070a8f1f9e5SMatthew G. Knepley } 2071c9ba7969SMatthew G. Knepley 2072c9ba7969SMatthew G. Knepley PetscErrorCode PetscFECreateCellGeometry(PetscFE fe, PetscQuadrature quad, PetscFEGeom *cgeom) 2073c9ba7969SMatthew G. Knepley { 2074c9ba7969SMatthew G. Knepley PetscDualSpace dsp; 2075c9ba7969SMatthew G. Knepley DM dm; 2076c9ba7969SMatthew G. Knepley PetscQuadrature quadDef; 2077c9ba7969SMatthew G. Knepley PetscInt dim, cdim, Nq; 2078c9ba7969SMatthew G. Knepley PetscErrorCode ierr; 2079c9ba7969SMatthew G. Knepley 2080c9ba7969SMatthew G. Knepley PetscFunctionBegin; 2081c9ba7969SMatthew G. Knepley ierr = PetscFEGetDualSpace(fe, &dsp);CHKERRQ(ierr); 2082c9ba7969SMatthew G. Knepley ierr = PetscDualSpaceGetDM(dsp, &dm);CHKERRQ(ierr); 2083c9ba7969SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 2084c9ba7969SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr); 2085c9ba7969SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quadDef);CHKERRQ(ierr); 2086c9ba7969SMatthew G. Knepley quad = quad ? quad : quadDef; 2087c9ba7969SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 2088c9ba7969SMatthew G. Knepley ierr = PetscMalloc1(Nq*cdim, &cgeom->v);CHKERRQ(ierr); 2089c9ba7969SMatthew G. Knepley ierr = PetscMalloc1(Nq*cdim*cdim, &cgeom->J);CHKERRQ(ierr); 2090c9ba7969SMatthew G. Knepley ierr = PetscMalloc1(Nq*cdim*cdim, &cgeom->invJ);CHKERRQ(ierr); 2091c9ba7969SMatthew G. Knepley ierr = PetscMalloc1(Nq, &cgeom->detJ);CHKERRQ(ierr); 2092c9ba7969SMatthew G. Knepley cgeom->dim = dim; 2093c9ba7969SMatthew G. Knepley cgeom->dimEmbed = cdim; 2094c9ba7969SMatthew G. Knepley cgeom->numCells = 1; 2095c9ba7969SMatthew G. Knepley cgeom->numPoints = Nq; 2096c9ba7969SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, 0, quad, cgeom->v, cgeom->J, cgeom->invJ, cgeom->detJ);CHKERRQ(ierr); 2097c9ba7969SMatthew G. Knepley PetscFunctionReturn(0); 2098c9ba7969SMatthew G. Knepley } 2099c9ba7969SMatthew G. Knepley 2100c9ba7969SMatthew G. Knepley PetscErrorCode PetscFEDestroyCellGeometry(PetscFE fe, PetscFEGeom *cgeom) 2101c9ba7969SMatthew G. Knepley { 2102c9ba7969SMatthew G. Knepley PetscErrorCode ierr; 2103c9ba7969SMatthew G. Knepley 2104c9ba7969SMatthew G. Knepley PetscFunctionBegin; 2105c9ba7969SMatthew G. Knepley ierr = PetscFree(cgeom->v);CHKERRQ(ierr); 2106c9ba7969SMatthew G. Knepley ierr = PetscFree(cgeom->J);CHKERRQ(ierr); 2107c9ba7969SMatthew G. Knepley ierr = PetscFree(cgeom->invJ);CHKERRQ(ierr); 2108c9ba7969SMatthew G. Knepley ierr = PetscFree(cgeom->detJ);CHKERRQ(ierr); 2109c9ba7969SMatthew G. Knepley PetscFunctionReturn(0); 2110c9ba7969SMatthew G. Knepley } 2111