120cf1dd8SToby Isaac #include <petsc/private/petscfeimpl.h> /*I "petscfe.h" I*/ 220cf1dd8SToby Isaac #include <petscdmplex.h> 320cf1dd8SToby Isaac 420cf1dd8SToby Isaac PetscClassId PETSCDUALSPACE_CLASSID = 0; 520cf1dd8SToby Isaac 6ead873ccSMatthew G. Knepley PetscLogEvent PETSCDUALSPACE_SetUp; 7ead873ccSMatthew G. Knepley 820cf1dd8SToby Isaac PetscFunctionList PetscDualSpaceList = NULL; 920cf1dd8SToby Isaac PetscBool PetscDualSpaceRegisterAllCalled = PETSC_FALSE; 1020cf1dd8SToby Isaac 116f905325SMatthew G. Knepley /* 126f905325SMatthew G. Knepley PetscDualSpaceLatticePointLexicographic_Internal - Returns all tuples of size 'len' with nonnegative integers that sum up to at most 'max'. 136f905325SMatthew G. Knepley Ordering is lexicographic with lowest index as least significant in ordering. 14b4457527SToby Isaac e.g. for len == 2 and max == 2, this will return, in order, {0,0}, {1,0}, {2,0}, {0,1}, {1,1}, {0,2}. 156f905325SMatthew G. Knepley 166f905325SMatthew G. Knepley Input Parameters: 176f905325SMatthew G. Knepley + len - The length of the tuple 186f905325SMatthew G. Knepley . max - The maximum sum 196f905325SMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition 206f905325SMatthew G. Knepley 216f905325SMatthew G. Knepley Output Parameter: 226f905325SMatthew G. Knepley . tup - A tuple of len integers whos sum is at most 'max' 236f905325SMatthew G. Knepley 246f905325SMatthew G. Knepley Level: developer 256f905325SMatthew G. Knepley 26db781477SPatrick Sanan .seealso: `PetscDualSpaceTensorPointLexicographic_Internal()` 276f905325SMatthew G. Knepley */ 286f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceLatticePointLexicographic_Internal(PetscInt len, PetscInt max, PetscInt tup[]) 296f905325SMatthew G. Knepley { 306f905325SMatthew G. Knepley PetscFunctionBegin; 316f905325SMatthew G. Knepley while (len--) { 326f905325SMatthew G. Knepley max -= tup[len]; 336f905325SMatthew G. Knepley if (!max) { 346f905325SMatthew G. Knepley tup[len] = 0; 356f905325SMatthew G. Knepley break; 366f905325SMatthew G. Knepley } 376f905325SMatthew G. Knepley } 386f905325SMatthew G. Knepley tup[++len]++; 396f905325SMatthew G. Knepley PetscFunctionReturn(0); 406f905325SMatthew G. Knepley } 416f905325SMatthew G. Knepley 426f905325SMatthew G. Knepley /* 436f905325SMatthew G. Knepley PetscDualSpaceTensorPointLexicographic_Internal - Returns all tuples of size 'len' with nonnegative integers that are all less than or equal to 'max'. 446f905325SMatthew G. Knepley Ordering is lexicographic with lowest index as least significant in ordering. 456f905325SMatthew G. Knepley e.g. for len == 2 and max == 2, this will return, in order, {0,0}, {1,0}, {2,0}, {0,1}, {1,1}, {2,1}, {0,2}, {1,2}, {2,2}. 466f905325SMatthew G. Knepley 476f905325SMatthew G. Knepley Input Parameters: 486f905325SMatthew G. Knepley + len - The length of the tuple 496f905325SMatthew G. Knepley . max - The maximum value 506f905325SMatthew G. Knepley - tup - A tuple of length len+1: tup[len] > 0 indicates a stopping condition 516f905325SMatthew G. Knepley 526f905325SMatthew G. Knepley Output Parameter: 536f905325SMatthew G. Knepley . tup - A tuple of len integers whos sum is at most 'max' 546f905325SMatthew G. Knepley 556f905325SMatthew G. Knepley Level: developer 566f905325SMatthew G. Knepley 57db781477SPatrick Sanan .seealso: `PetscDualSpaceLatticePointLexicographic_Internal()` 586f905325SMatthew G. Knepley */ 596f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceTensorPointLexicographic_Internal(PetscInt len, PetscInt max, PetscInt tup[]) 606f905325SMatthew G. Knepley { 616f905325SMatthew G. Knepley PetscInt i; 626f905325SMatthew G. Knepley 636f905325SMatthew G. Knepley PetscFunctionBegin; 646f905325SMatthew G. Knepley for (i = 0; i < len; i++) { 656f905325SMatthew G. Knepley if (tup[i] < max) { 666f905325SMatthew G. Knepley break; 676f905325SMatthew G. Knepley } else { 686f905325SMatthew G. Knepley tup[i] = 0; 696f905325SMatthew G. Knepley } 706f905325SMatthew G. Knepley } 716f905325SMatthew G. Knepley tup[i]++; 726f905325SMatthew G. Knepley PetscFunctionReturn(0); 736f905325SMatthew G. Knepley } 746f905325SMatthew G. Knepley 7520cf1dd8SToby Isaac /*@C 7620cf1dd8SToby Isaac PetscDualSpaceRegister - Adds a new PetscDualSpace implementation 7720cf1dd8SToby Isaac 7820cf1dd8SToby Isaac Not Collective 7920cf1dd8SToby Isaac 8020cf1dd8SToby Isaac Input Parameters: 8120cf1dd8SToby Isaac + name - The name of a new user-defined creation routine 8220cf1dd8SToby Isaac - create_func - The creation routine itself 8320cf1dd8SToby Isaac 8420cf1dd8SToby Isaac Notes: 8520cf1dd8SToby Isaac PetscDualSpaceRegister() may be called multiple times to add several user-defined PetscDualSpaces 8620cf1dd8SToby Isaac 8720cf1dd8SToby Isaac Sample usage: 8820cf1dd8SToby Isaac .vb 8920cf1dd8SToby Isaac PetscDualSpaceRegister("my_space", MyPetscDualSpaceCreate); 9020cf1dd8SToby Isaac .ve 9120cf1dd8SToby Isaac 9220cf1dd8SToby Isaac Then, your PetscDualSpace type can be chosen with the procedural interface via 9320cf1dd8SToby Isaac .vb 9420cf1dd8SToby Isaac PetscDualSpaceCreate(MPI_Comm, PetscDualSpace *); 9520cf1dd8SToby Isaac PetscDualSpaceSetType(PetscDualSpace, "my_dual_space"); 9620cf1dd8SToby Isaac .ve 9720cf1dd8SToby Isaac or at runtime via the option 9820cf1dd8SToby Isaac .vb 9920cf1dd8SToby Isaac -petscdualspace_type my_dual_space 10020cf1dd8SToby Isaac .ve 10120cf1dd8SToby Isaac 10220cf1dd8SToby Isaac Level: advanced 10320cf1dd8SToby Isaac 104db781477SPatrick Sanan .seealso: `PetscDualSpaceRegisterAll()`, `PetscDualSpaceRegisterDestroy()` 10520cf1dd8SToby Isaac 10620cf1dd8SToby Isaac @*/ 10720cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceRegister(const char sname[], PetscErrorCode (*function)(PetscDualSpace)) 10820cf1dd8SToby Isaac { 10920cf1dd8SToby Isaac PetscFunctionBegin; 1109566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&PetscDualSpaceList, sname, function)); 11120cf1dd8SToby Isaac PetscFunctionReturn(0); 11220cf1dd8SToby Isaac } 11320cf1dd8SToby Isaac 11420cf1dd8SToby Isaac /*@C 11520cf1dd8SToby Isaac PetscDualSpaceSetType - Builds a particular PetscDualSpace 11620cf1dd8SToby Isaac 117d083f849SBarry Smith Collective on sp 11820cf1dd8SToby Isaac 11920cf1dd8SToby Isaac Input Parameters: 12020cf1dd8SToby Isaac + sp - The PetscDualSpace object 12120cf1dd8SToby Isaac - name - The kind of space 12220cf1dd8SToby Isaac 12320cf1dd8SToby Isaac Options Database Key: 12420cf1dd8SToby Isaac . -petscdualspace_type <type> - Sets the PetscDualSpace type; use -help for a list of available types 12520cf1dd8SToby Isaac 12620cf1dd8SToby Isaac Level: intermediate 12720cf1dd8SToby Isaac 128db781477SPatrick Sanan .seealso: `PetscDualSpaceGetType()`, `PetscDualSpaceCreate()` 12920cf1dd8SToby Isaac @*/ 13020cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetType(PetscDualSpace sp, PetscDualSpaceType name) 13120cf1dd8SToby Isaac { 13220cf1dd8SToby Isaac PetscErrorCode (*r)(PetscDualSpace); 13320cf1dd8SToby Isaac PetscBool match; 13420cf1dd8SToby Isaac 13520cf1dd8SToby Isaac PetscFunctionBegin; 13620cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1379566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject) sp, name, &match)); 13820cf1dd8SToby Isaac if (match) PetscFunctionReturn(0); 13920cf1dd8SToby Isaac 1409566063dSJacob Faibussowitsch if (!PetscDualSpaceRegisterAllCalled) PetscCall(PetscDualSpaceRegisterAll()); 1419566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(PetscDualSpaceList, name, &r)); 14228b400f6SJacob Faibussowitsch PetscCheck(r,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown PetscDualSpace type: %s", name); 14320cf1dd8SToby Isaac 14420cf1dd8SToby Isaac if (sp->ops->destroy) { 1459566063dSJacob Faibussowitsch PetscCall((*sp->ops->destroy)(sp)); 14620cf1dd8SToby Isaac sp->ops->destroy = NULL; 14720cf1dd8SToby Isaac } 1489566063dSJacob Faibussowitsch PetscCall((*r)(sp)); 1499566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject) sp, name)); 15020cf1dd8SToby Isaac PetscFunctionReturn(0); 15120cf1dd8SToby Isaac } 15220cf1dd8SToby Isaac 15320cf1dd8SToby Isaac /*@C 15420cf1dd8SToby Isaac PetscDualSpaceGetType - Gets the PetscDualSpace type name (as a string) from the object. 15520cf1dd8SToby Isaac 15620cf1dd8SToby Isaac Not Collective 15720cf1dd8SToby Isaac 15820cf1dd8SToby Isaac Input Parameter: 15920cf1dd8SToby Isaac . sp - The PetscDualSpace 16020cf1dd8SToby Isaac 16120cf1dd8SToby Isaac Output Parameter: 16220cf1dd8SToby Isaac . name - The PetscDualSpace type name 16320cf1dd8SToby Isaac 16420cf1dd8SToby Isaac Level: intermediate 16520cf1dd8SToby Isaac 166db781477SPatrick Sanan .seealso: `PetscDualSpaceSetType()`, `PetscDualSpaceCreate()` 16720cf1dd8SToby Isaac @*/ 16820cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetType(PetscDualSpace sp, PetscDualSpaceType *name) 16920cf1dd8SToby Isaac { 17020cf1dd8SToby Isaac PetscFunctionBegin; 17120cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 17220cf1dd8SToby Isaac PetscValidPointer(name, 2); 17320cf1dd8SToby Isaac if (!PetscDualSpaceRegisterAllCalled) { 1749566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceRegisterAll()); 17520cf1dd8SToby Isaac } 17620cf1dd8SToby Isaac *name = ((PetscObject) sp)->type_name; 17720cf1dd8SToby Isaac PetscFunctionReturn(0); 17820cf1dd8SToby Isaac } 17920cf1dd8SToby Isaac 180221d6281SMatthew G. Knepley static PetscErrorCode PetscDualSpaceView_ASCII(PetscDualSpace sp, PetscViewer v) 181221d6281SMatthew G. Knepley { 182221d6281SMatthew G. Knepley PetscViewerFormat format; 183221d6281SMatthew G. Knepley PetscInt pdim, f; 184221d6281SMatthew G. Knepley 185221d6281SMatthew G. Knepley PetscFunctionBegin; 1869566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(sp, &pdim)); 1879566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject) sp, v)); 1889566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(v)); 189b4457527SToby Isaac if (sp->k) { 19063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "Dual space for %" PetscInt_FMT "-forms %swith %" PetscInt_FMT " components, size %" PetscInt_FMT "\n", PetscAbsInt(sp->k), sp->k < 0 ? "(stored in dual form) ": "", sp->Nc, pdim)); 191b4457527SToby Isaac } else { 19263a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "Dual space with %" PetscInt_FMT " components, size %" PetscInt_FMT "\n", sp->Nc, pdim)); 193b4457527SToby Isaac } 1949566063dSJacob Faibussowitsch if (sp->ops->view) PetscCall((*sp->ops->view)(sp, v)); 1959566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(v, &format)); 196221d6281SMatthew G. Knepley if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 1979566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(v)); 198221d6281SMatthew G. Knepley for (f = 0; f < pdim; ++f) { 19963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(v, "Dual basis vector %" PetscInt_FMT "\n", f)); 2009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(v)); 2019566063dSJacob Faibussowitsch PetscCall(PetscQuadratureView(sp->functional[f], v)); 2029566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(v)); 203221d6281SMatthew G. Knepley } 2049566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(v)); 205221d6281SMatthew G. Knepley } 2069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(v)); 207221d6281SMatthew G. Knepley PetscFunctionReturn(0); 208221d6281SMatthew G. Knepley } 209221d6281SMatthew G. Knepley 210fe2efc57SMark /*@C 211fe2efc57SMark PetscDualSpaceViewFromOptions - View from Options 212fe2efc57SMark 213fe2efc57SMark Collective on PetscDualSpace 214fe2efc57SMark 215fe2efc57SMark Input Parameters: 216fe2efc57SMark + A - the PetscDualSpace object 217736c3998SJose E. Roman . obj - Optional object, proivides prefix 218736c3998SJose E. Roman - name - command line option 219fe2efc57SMark 220fe2efc57SMark Level: intermediate 221db781477SPatrick Sanan .seealso: `PetscDualSpace`, `PetscDualSpaceView()`, `PetscObjectViewFromOptions()`, `PetscDualSpaceCreate()` 222fe2efc57SMark @*/ 223fe2efc57SMark PetscErrorCode PetscDualSpaceViewFromOptions(PetscDualSpace A,PetscObject obj,const char name[]) 224fe2efc57SMark { 225fe2efc57SMark PetscFunctionBegin; 226fe2efc57SMark PetscValidHeaderSpecific(A,PETSCDUALSPACE_CLASSID,1); 2279566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name)); 228fe2efc57SMark PetscFunctionReturn(0); 229fe2efc57SMark } 230fe2efc57SMark 23120cf1dd8SToby Isaac /*@ 23220cf1dd8SToby Isaac PetscDualSpaceView - Views a PetscDualSpace 23320cf1dd8SToby Isaac 234d083f849SBarry Smith Collective on sp 23520cf1dd8SToby Isaac 236d8d19677SJose E. Roman Input Parameters: 23720cf1dd8SToby Isaac + sp - the PetscDualSpace object to view 23820cf1dd8SToby Isaac - v - the viewer 23920cf1dd8SToby Isaac 240a4ce7ad1SMatthew G. Knepley Level: beginner 24120cf1dd8SToby Isaac 242db781477SPatrick Sanan .seealso `PetscDualSpaceDestroy()`, `PetscDualSpace` 24320cf1dd8SToby Isaac @*/ 24420cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceView(PetscDualSpace sp, PetscViewer v) 24520cf1dd8SToby Isaac { 246d9bac1caSLisandro Dalcin PetscBool iascii; 24720cf1dd8SToby Isaac 24820cf1dd8SToby Isaac PetscFunctionBegin; 24920cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 250d9bac1caSLisandro Dalcin if (v) PetscValidHeaderSpecific(v, PETSC_VIEWER_CLASSID, 2); 2519566063dSJacob Faibussowitsch if (!v) PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject) sp), &v)); 2529566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject) v, PETSCVIEWERASCII, &iascii)); 2539566063dSJacob Faibussowitsch if (iascii) PetscCall(PetscDualSpaceView_ASCII(sp, v)); 25420cf1dd8SToby Isaac PetscFunctionReturn(0); 25520cf1dd8SToby Isaac } 25620cf1dd8SToby Isaac 25720cf1dd8SToby Isaac /*@ 25820cf1dd8SToby Isaac PetscDualSpaceSetFromOptions - sets parameters in a PetscDualSpace from the options database 25920cf1dd8SToby Isaac 260d083f849SBarry Smith Collective on sp 26120cf1dd8SToby Isaac 26220cf1dd8SToby Isaac Input Parameter: 26320cf1dd8SToby Isaac . sp - the PetscDualSpace object to set options for 26420cf1dd8SToby Isaac 26520cf1dd8SToby Isaac Options Database: 2668f2aacc6SMatthew G. Knepley + -petscdualspace_order <order> - the approximation order of the space 267fe36a153SMatthew G. Knepley . -petscdualspace_form_degree <deg> - the form degree, say 0 for point evaluations, or 2 for area integrals 2688f2aacc6SMatthew G. Knepley . -petscdualspace_components <c> - the number of components, say d for a vector field 2698f2aacc6SMatthew G. Knepley - -petscdualspace_refcell <celltype> - Reference cell type name 27020cf1dd8SToby Isaac 271a4ce7ad1SMatthew G. Knepley Level: intermediate 27220cf1dd8SToby Isaac 273db781477SPatrick Sanan .seealso `PetscDualSpaceView()`, `PetscDualSpace`, `PetscObjectSetFromOptions()` 27420cf1dd8SToby Isaac @*/ 27520cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetFromOptions(PetscDualSpace sp) 27620cf1dd8SToby Isaac { 2772df84da0SMatthew G. Knepley DMPolytopeType refCell = DM_POLYTOPE_TRIANGLE; 27820cf1dd8SToby Isaac const char *defaultType; 27920cf1dd8SToby Isaac char name[256]; 280f783ec47SMatthew G. Knepley PetscBool flg; 28120cf1dd8SToby Isaac 28220cf1dd8SToby Isaac PetscFunctionBegin; 28320cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 28420cf1dd8SToby Isaac if (!((PetscObject) sp)->type_name) { 28520cf1dd8SToby Isaac defaultType = PETSCDUALSPACELAGRANGE; 28620cf1dd8SToby Isaac } else { 28720cf1dd8SToby Isaac defaultType = ((PetscObject) sp)->type_name; 28820cf1dd8SToby Isaac } 2899566063dSJacob Faibussowitsch if (!PetscSpaceRegisterAllCalled) PetscCall(PetscSpaceRegisterAll()); 29020cf1dd8SToby Isaac 291d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject) sp); 2929566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-petscdualspace_type", "Dual space", "PetscDualSpaceSetType", PetscDualSpaceList, defaultType, name, 256, &flg)); 29320cf1dd8SToby Isaac if (flg) { 2949566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSetType(sp, name)); 29520cf1dd8SToby Isaac } else if (!((PetscObject) sp)->type_name) { 2969566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSetType(sp, defaultType)); 29720cf1dd8SToby Isaac } 2989566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-petscdualspace_order", "The approximation order", "PetscDualSpaceSetOrder", sp->order, &sp->order, NULL,0)); 2999566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-petscdualspace_form_degree", "The form degree of the dofs", "PetscDualSpaceSetFormDegree", sp->k, &sp->k, NULL)); 3009566063dSJacob Faibussowitsch PetscCall(PetscOptionsBoundedInt("-petscdualspace_components", "The number of components", "PetscDualSpaceSetNumComponents", sp->Nc, &sp->Nc, NULL,1)); 30120cf1dd8SToby Isaac if (sp->ops->setfromoptions) { 3029566063dSJacob Faibussowitsch PetscCall((*sp->ops->setfromoptions)(PetscOptionsObject,sp)); 30320cf1dd8SToby Isaac } 3049566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-petscdualspace_refcell", "Reference cell shape", "PetscDualSpaceSetReferenceCell", DMPolytopeTypes, (PetscEnum) refCell, (PetscEnum *) &refCell, &flg)); 305063ee4adSMatthew G. Knepley if (flg) { 306063ee4adSMatthew G. Knepley DM K; 307063ee4adSMatthew G. Knepley 3089566063dSJacob Faibussowitsch PetscCall(DMPlexCreateReferenceCell(PETSC_COMM_SELF, refCell, &K)); 3099566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSetDM(sp, K)); 3109566063dSJacob Faibussowitsch PetscCall(DMDestroy(&K)); 311063ee4adSMatthew G. Knepley } 312063ee4adSMatthew G. Knepley 31320cf1dd8SToby Isaac /* process any options handlers added with PetscObjectAddOptionsHandler() */ 3149566063dSJacob Faibussowitsch PetscCall(PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject) sp)); 315d0609cedSBarry Smith PetscOptionsEnd(); 316063ee4adSMatthew G. Knepley sp->setfromoptionscalled = PETSC_TRUE; 31720cf1dd8SToby Isaac PetscFunctionReturn(0); 31820cf1dd8SToby Isaac } 31920cf1dd8SToby Isaac 32020cf1dd8SToby Isaac /*@ 32120cf1dd8SToby Isaac PetscDualSpaceSetUp - Construct a basis for the PetscDualSpace 32220cf1dd8SToby Isaac 323d083f849SBarry Smith Collective on sp 32420cf1dd8SToby Isaac 32520cf1dd8SToby Isaac Input Parameter: 32620cf1dd8SToby Isaac . sp - the PetscDualSpace object to setup 32720cf1dd8SToby Isaac 328a4ce7ad1SMatthew G. Knepley Level: intermediate 32920cf1dd8SToby Isaac 330db781477SPatrick Sanan .seealso `PetscDualSpaceView()`, `PetscDualSpaceDestroy()`, `PetscDualSpace` 33120cf1dd8SToby Isaac @*/ 33220cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetUp(PetscDualSpace sp) 33320cf1dd8SToby Isaac { 33420cf1dd8SToby Isaac PetscFunctionBegin; 33520cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 33620cf1dd8SToby Isaac if (sp->setupcalled) PetscFunctionReturn(0); 3379566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(PETSCDUALSPACE_SetUp, sp, 0, 0, 0)); 33820cf1dd8SToby Isaac sp->setupcalled = PETSC_TRUE; 3399566063dSJacob Faibussowitsch if (sp->ops->setup) PetscCall((*sp->ops->setup)(sp)); 3409566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(PETSCDUALSPACE_SetUp, sp, 0, 0, 0)); 3419566063dSJacob Faibussowitsch if (sp->setfromoptionscalled) PetscCall(PetscDualSpaceViewFromOptions(sp, NULL, "-petscdualspace_view")); 34220cf1dd8SToby Isaac PetscFunctionReturn(0); 34320cf1dd8SToby Isaac } 34420cf1dd8SToby Isaac 345b4457527SToby Isaac static PetscErrorCode PetscDualSpaceClearDMData_Internal(PetscDualSpace sp, DM dm) 346b4457527SToby Isaac { 347b4457527SToby Isaac PetscInt pStart = -1, pEnd = -1, depth = -1; 348b4457527SToby Isaac 349b4457527SToby Isaac PetscFunctionBegin; 350b4457527SToby Isaac if (!dm) PetscFunctionReturn(0); 3519566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 3529566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm, &depth)); 353b4457527SToby Isaac 354b4457527SToby Isaac if (sp->pointSpaces) { 355b4457527SToby Isaac PetscInt i; 356b4457527SToby Isaac 357b4457527SToby Isaac for (i = 0; i < pEnd - pStart; i++) { 3589566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceDestroy(&(sp->pointSpaces[i]))); 359b4457527SToby Isaac } 360b4457527SToby Isaac } 3619566063dSJacob Faibussowitsch PetscCall(PetscFree(sp->pointSpaces)); 362b4457527SToby Isaac 363b4457527SToby Isaac if (sp->heightSpaces) { 364b4457527SToby Isaac PetscInt i; 365b4457527SToby Isaac 366b4457527SToby Isaac for (i = 0; i <= depth; i++) { 3679566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceDestroy(&(sp->heightSpaces[i]))); 368b4457527SToby Isaac } 369b4457527SToby Isaac } 3709566063dSJacob Faibussowitsch PetscCall(PetscFree(sp->heightSpaces)); 371b4457527SToby Isaac 3729566063dSJacob Faibussowitsch PetscCall(PetscSectionDestroy(&(sp->pointSection))); 3739566063dSJacob Faibussowitsch PetscCall(PetscQuadratureDestroy(&(sp->intNodes))); 3749566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(sp->intDofValues))); 3759566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(sp->intNodeValues))); 3769566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(sp->intMat))); 3779566063dSJacob Faibussowitsch PetscCall(PetscQuadratureDestroy(&(sp->allNodes))); 3789566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(sp->allDofValues))); 3799566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(sp->allNodeValues))); 3809566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(sp->allMat))); 3819566063dSJacob Faibussowitsch PetscCall(PetscFree(sp->numDof)); 382b4457527SToby Isaac PetscFunctionReturn(0); 383b4457527SToby Isaac } 384b4457527SToby Isaac 38520cf1dd8SToby Isaac /*@ 38620cf1dd8SToby Isaac PetscDualSpaceDestroy - Destroys a PetscDualSpace object 38720cf1dd8SToby Isaac 388d083f849SBarry Smith Collective on sp 38920cf1dd8SToby Isaac 39020cf1dd8SToby Isaac Input Parameter: 39120cf1dd8SToby Isaac . sp - the PetscDualSpace object to destroy 39220cf1dd8SToby Isaac 393a4ce7ad1SMatthew G. Knepley Level: beginner 39420cf1dd8SToby Isaac 395db781477SPatrick Sanan .seealso `PetscDualSpaceView()`, `PetscDualSpace()`, `PetscDualSpaceCreate()` 39620cf1dd8SToby Isaac @*/ 39720cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceDestroy(PetscDualSpace *sp) 39820cf1dd8SToby Isaac { 39920cf1dd8SToby Isaac PetscInt dim, f; 400b4457527SToby Isaac DM dm; 40120cf1dd8SToby Isaac 40220cf1dd8SToby Isaac PetscFunctionBegin; 40320cf1dd8SToby Isaac if (!*sp) PetscFunctionReturn(0); 40420cf1dd8SToby Isaac PetscValidHeaderSpecific((*sp), PETSCDUALSPACE_CLASSID, 1); 40520cf1dd8SToby Isaac 406ea78f98cSLisandro Dalcin if (--((PetscObject)(*sp))->refct > 0) {*sp = NULL; PetscFunctionReturn(0);} 40720cf1dd8SToby Isaac ((PetscObject) (*sp))->refct = 0; 40820cf1dd8SToby Isaac 4099566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(*sp, &dim)); 410b4457527SToby Isaac dm = (*sp)->dm; 411b4457527SToby Isaac 4129566063dSJacob Faibussowitsch if ((*sp)->ops->destroy) PetscCall((*(*sp)->ops->destroy)(*sp)); 4139566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceClearDMData_Internal(*sp, dm)); 414b4457527SToby Isaac 41520cf1dd8SToby Isaac for (f = 0; f < dim; ++f) { 4169566063dSJacob Faibussowitsch PetscCall(PetscQuadratureDestroy(&(*sp)->functional[f])); 41720cf1dd8SToby Isaac } 4189566063dSJacob Faibussowitsch PetscCall(PetscFree((*sp)->functional)); 4199566063dSJacob Faibussowitsch PetscCall(DMDestroy(&(*sp)->dm)); 4209566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(sp)); 42120cf1dd8SToby Isaac PetscFunctionReturn(0); 42220cf1dd8SToby Isaac } 42320cf1dd8SToby Isaac 42420cf1dd8SToby Isaac /*@ 42520cf1dd8SToby Isaac PetscDualSpaceCreate - Creates an empty PetscDualSpace object. The type can then be set with PetscDualSpaceSetType(). 42620cf1dd8SToby Isaac 427d083f849SBarry Smith Collective 42820cf1dd8SToby Isaac 42920cf1dd8SToby Isaac Input Parameter: 43020cf1dd8SToby Isaac . comm - The communicator for the PetscDualSpace object 43120cf1dd8SToby Isaac 43220cf1dd8SToby Isaac Output Parameter: 43320cf1dd8SToby Isaac . sp - The PetscDualSpace object 43420cf1dd8SToby Isaac 43520cf1dd8SToby Isaac Level: beginner 43620cf1dd8SToby Isaac 437db781477SPatrick Sanan .seealso: `PetscDualSpaceSetType()`, `PETSCDUALSPACELAGRANGE` 43820cf1dd8SToby Isaac @*/ 43920cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceCreate(MPI_Comm comm, PetscDualSpace *sp) 44020cf1dd8SToby Isaac { 44120cf1dd8SToby Isaac PetscDualSpace s; 44220cf1dd8SToby Isaac 44320cf1dd8SToby Isaac PetscFunctionBegin; 44420cf1dd8SToby Isaac PetscValidPointer(sp, 2); 4459566063dSJacob Faibussowitsch PetscCall(PetscCitationsRegister(FECitation,&FEcite)); 44620cf1dd8SToby Isaac *sp = NULL; 4479566063dSJacob Faibussowitsch PetscCall(PetscFEInitializePackage()); 44820cf1dd8SToby Isaac 4499566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(s, PETSCDUALSPACE_CLASSID, "PetscDualSpace", "Dual Space", "PetscDualSpace", comm, PetscDualSpaceDestroy, PetscDualSpaceView)); 45020cf1dd8SToby Isaac 45120cf1dd8SToby Isaac s->order = 0; 45220cf1dd8SToby Isaac s->Nc = 1; 4534bee2e38SMatthew G. Knepley s->k = 0; 454b4457527SToby Isaac s->spdim = -1; 455b4457527SToby Isaac s->spintdim = -1; 456b4457527SToby Isaac s->uniform = PETSC_TRUE; 45720cf1dd8SToby Isaac s->setupcalled = PETSC_FALSE; 45820cf1dd8SToby Isaac 45920cf1dd8SToby Isaac *sp = s; 46020cf1dd8SToby Isaac PetscFunctionReturn(0); 46120cf1dd8SToby Isaac } 46220cf1dd8SToby Isaac 46320cf1dd8SToby Isaac /*@ 46420cf1dd8SToby Isaac PetscDualSpaceDuplicate - Creates a duplicate PetscDualSpace object, however it is not setup. 46520cf1dd8SToby Isaac 466d083f849SBarry Smith Collective on sp 46720cf1dd8SToby Isaac 46820cf1dd8SToby Isaac Input Parameter: 46920cf1dd8SToby Isaac . sp - The original PetscDualSpace 47020cf1dd8SToby Isaac 47120cf1dd8SToby Isaac Output Parameter: 47220cf1dd8SToby Isaac . spNew - The duplicate PetscDualSpace 47320cf1dd8SToby Isaac 47420cf1dd8SToby Isaac Level: beginner 47520cf1dd8SToby Isaac 476db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()`, `PetscDualSpaceSetType()` 47720cf1dd8SToby Isaac @*/ 47820cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceDuplicate(PetscDualSpace sp, PetscDualSpace *spNew) 47920cf1dd8SToby Isaac { 480b4457527SToby Isaac DM dm; 481b4457527SToby Isaac PetscDualSpaceType type; 482b4457527SToby Isaac const char *name; 48320cf1dd8SToby Isaac 48420cf1dd8SToby Isaac PetscFunctionBegin; 48520cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 48620cf1dd8SToby Isaac PetscValidPointer(spNew, 2); 4879566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceCreate(PetscObjectComm((PetscObject)sp), spNew)); 4889566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject) sp, &name)); 4899566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject) *spNew, name)); 4909566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetType(sp, &type)); 4919566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSetType(*spNew, type)); 4929566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 4939566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSetDM(*spNew, dm)); 494b4457527SToby Isaac 495b4457527SToby Isaac (*spNew)->order = sp->order; 496b4457527SToby Isaac (*spNew)->k = sp->k; 497b4457527SToby Isaac (*spNew)->Nc = sp->Nc; 498b4457527SToby Isaac (*spNew)->uniform = sp->uniform; 4999566063dSJacob Faibussowitsch if (sp->ops->duplicate) PetscCall((*sp->ops->duplicate)(sp, *spNew)); 50020cf1dd8SToby Isaac PetscFunctionReturn(0); 50120cf1dd8SToby Isaac } 50220cf1dd8SToby Isaac 50320cf1dd8SToby Isaac /*@ 50420cf1dd8SToby Isaac PetscDualSpaceGetDM - Get the DM representing the reference cell 50520cf1dd8SToby Isaac 50620cf1dd8SToby Isaac Not collective 50720cf1dd8SToby Isaac 50820cf1dd8SToby Isaac Input Parameter: 50920cf1dd8SToby Isaac . sp - The PetscDualSpace 51020cf1dd8SToby Isaac 51120cf1dd8SToby Isaac Output Parameter: 51220cf1dd8SToby Isaac . dm - The reference cell 51320cf1dd8SToby Isaac 51420cf1dd8SToby Isaac Level: intermediate 51520cf1dd8SToby Isaac 516db781477SPatrick Sanan .seealso: `PetscDualSpaceSetDM()`, `PetscDualSpaceCreate()` 51720cf1dd8SToby Isaac @*/ 51820cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetDM(PetscDualSpace sp, DM *dm) 51920cf1dd8SToby Isaac { 52020cf1dd8SToby Isaac PetscFunctionBegin; 52120cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 52220cf1dd8SToby Isaac PetscValidPointer(dm, 2); 52320cf1dd8SToby Isaac *dm = sp->dm; 52420cf1dd8SToby Isaac PetscFunctionReturn(0); 52520cf1dd8SToby Isaac } 52620cf1dd8SToby Isaac 52720cf1dd8SToby Isaac /*@ 52820cf1dd8SToby Isaac PetscDualSpaceSetDM - Get the DM representing the reference cell 52920cf1dd8SToby Isaac 53020cf1dd8SToby Isaac Not collective 53120cf1dd8SToby Isaac 53220cf1dd8SToby Isaac Input Parameters: 53320cf1dd8SToby Isaac + sp - The PetscDualSpace 53420cf1dd8SToby Isaac - dm - The reference cell 53520cf1dd8SToby Isaac 53620cf1dd8SToby Isaac Level: intermediate 53720cf1dd8SToby Isaac 538db781477SPatrick Sanan .seealso: `PetscDualSpaceGetDM()`, `PetscDualSpaceCreate()` 53920cf1dd8SToby Isaac @*/ 54020cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetDM(PetscDualSpace sp, DM dm) 54120cf1dd8SToby Isaac { 54220cf1dd8SToby Isaac PetscFunctionBegin; 54320cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 54420cf1dd8SToby Isaac PetscValidHeaderSpecific(dm, DM_CLASSID, 2); 54528b400f6SJacob Faibussowitsch PetscCheck(!sp->setupcalled,PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONGSTATE, "Cannot change DM after dualspace is set up"); 5469566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) dm)); 547b4457527SToby Isaac if (sp->dm && sp->dm != dm) { 5489566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceClearDMData_Internal(sp, sp->dm)); 549b4457527SToby Isaac } 5509566063dSJacob Faibussowitsch PetscCall(DMDestroy(&sp->dm)); 55120cf1dd8SToby Isaac sp->dm = dm; 55220cf1dd8SToby Isaac PetscFunctionReturn(0); 55320cf1dd8SToby Isaac } 55420cf1dd8SToby Isaac 55520cf1dd8SToby Isaac /*@ 55620cf1dd8SToby Isaac PetscDualSpaceGetOrder - Get the order of the dual space 55720cf1dd8SToby Isaac 55820cf1dd8SToby Isaac Not collective 55920cf1dd8SToby Isaac 56020cf1dd8SToby Isaac Input Parameter: 56120cf1dd8SToby Isaac . sp - The PetscDualSpace 56220cf1dd8SToby Isaac 56320cf1dd8SToby Isaac Output Parameter: 56420cf1dd8SToby Isaac . order - The order 56520cf1dd8SToby Isaac 56620cf1dd8SToby Isaac Level: intermediate 56720cf1dd8SToby Isaac 568db781477SPatrick Sanan .seealso: `PetscDualSpaceSetOrder()`, `PetscDualSpaceCreate()` 56920cf1dd8SToby Isaac @*/ 57020cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetOrder(PetscDualSpace sp, PetscInt *order) 57120cf1dd8SToby Isaac { 57220cf1dd8SToby Isaac PetscFunctionBegin; 57320cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 574dadcf809SJacob Faibussowitsch PetscValidIntPointer(order, 2); 57520cf1dd8SToby Isaac *order = sp->order; 57620cf1dd8SToby Isaac PetscFunctionReturn(0); 57720cf1dd8SToby Isaac } 57820cf1dd8SToby Isaac 57920cf1dd8SToby Isaac /*@ 58020cf1dd8SToby Isaac PetscDualSpaceSetOrder - Set the order of the dual space 58120cf1dd8SToby Isaac 58220cf1dd8SToby Isaac Not collective 58320cf1dd8SToby Isaac 58420cf1dd8SToby Isaac Input Parameters: 58520cf1dd8SToby Isaac + sp - The PetscDualSpace 58620cf1dd8SToby Isaac - order - The order 58720cf1dd8SToby Isaac 58820cf1dd8SToby Isaac Level: intermediate 58920cf1dd8SToby Isaac 590db781477SPatrick Sanan .seealso: `PetscDualSpaceGetOrder()`, `PetscDualSpaceCreate()` 59120cf1dd8SToby Isaac @*/ 59220cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetOrder(PetscDualSpace sp, PetscInt order) 59320cf1dd8SToby Isaac { 59420cf1dd8SToby Isaac PetscFunctionBegin; 59520cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 59628b400f6SJacob Faibussowitsch PetscCheck(!sp->setupcalled,PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONGSTATE, "Cannot change order after dualspace is set up"); 59720cf1dd8SToby Isaac sp->order = order; 59820cf1dd8SToby Isaac PetscFunctionReturn(0); 59920cf1dd8SToby Isaac } 60020cf1dd8SToby Isaac 60120cf1dd8SToby Isaac /*@ 60220cf1dd8SToby Isaac PetscDualSpaceGetNumComponents - Return the number of components for this space 60320cf1dd8SToby Isaac 60420cf1dd8SToby Isaac Input Parameter: 60520cf1dd8SToby Isaac . sp - The PetscDualSpace 60620cf1dd8SToby Isaac 60720cf1dd8SToby Isaac Output Parameter: 60820cf1dd8SToby Isaac . Nc - The number of components 60920cf1dd8SToby Isaac 61020cf1dd8SToby Isaac Note: A vector space, for example, will have d components, where d is the spatial dimension 61120cf1dd8SToby Isaac 61220cf1dd8SToby Isaac Level: intermediate 61320cf1dd8SToby Isaac 614db781477SPatrick Sanan .seealso: `PetscDualSpaceSetNumComponents()`, `PetscDualSpaceGetDimension()`, `PetscDualSpaceCreate()`, `PetscDualSpace` 61520cf1dd8SToby Isaac @*/ 61620cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetNumComponents(PetscDualSpace sp, PetscInt *Nc) 61720cf1dd8SToby Isaac { 61820cf1dd8SToby Isaac PetscFunctionBegin; 61920cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 620dadcf809SJacob Faibussowitsch PetscValidIntPointer(Nc, 2); 62120cf1dd8SToby Isaac *Nc = sp->Nc; 62220cf1dd8SToby Isaac PetscFunctionReturn(0); 62320cf1dd8SToby Isaac } 62420cf1dd8SToby Isaac 62520cf1dd8SToby Isaac /*@ 62620cf1dd8SToby Isaac PetscDualSpaceSetNumComponents - Set the number of components for this space 62720cf1dd8SToby Isaac 62820cf1dd8SToby Isaac Input Parameters: 62920cf1dd8SToby Isaac + sp - The PetscDualSpace 63020cf1dd8SToby Isaac - order - The number of components 63120cf1dd8SToby Isaac 63220cf1dd8SToby Isaac Level: intermediate 63320cf1dd8SToby Isaac 634db781477SPatrick Sanan .seealso: `PetscDualSpaceGetNumComponents()`, `PetscDualSpaceCreate()`, `PetscDualSpace` 63520cf1dd8SToby Isaac @*/ 63620cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceSetNumComponents(PetscDualSpace sp, PetscInt Nc) 63720cf1dd8SToby Isaac { 63820cf1dd8SToby Isaac PetscFunctionBegin; 63920cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 64028b400f6SJacob Faibussowitsch PetscCheck(!sp->setupcalled,PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONGSTATE, "Cannot change number of components after dualspace is set up"); 64120cf1dd8SToby Isaac sp->Nc = Nc; 64220cf1dd8SToby Isaac PetscFunctionReturn(0); 64320cf1dd8SToby Isaac } 64420cf1dd8SToby Isaac 64520cf1dd8SToby Isaac /*@ 64620cf1dd8SToby Isaac PetscDualSpaceGetFunctional - Get the i-th basis functional in the dual space 64720cf1dd8SToby Isaac 64820cf1dd8SToby Isaac Not collective 64920cf1dd8SToby Isaac 65020cf1dd8SToby Isaac Input Parameters: 65120cf1dd8SToby Isaac + sp - The PetscDualSpace 65220cf1dd8SToby Isaac - i - The basis number 65320cf1dd8SToby Isaac 65420cf1dd8SToby Isaac Output Parameter: 65520cf1dd8SToby Isaac . functional - The basis functional 65620cf1dd8SToby Isaac 65720cf1dd8SToby Isaac Level: intermediate 65820cf1dd8SToby Isaac 659db781477SPatrick Sanan .seealso: `PetscDualSpaceGetDimension()`, `PetscDualSpaceCreate()` 66020cf1dd8SToby Isaac @*/ 66120cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetFunctional(PetscDualSpace sp, PetscInt i, PetscQuadrature *functional) 66220cf1dd8SToby Isaac { 66320cf1dd8SToby Isaac PetscInt dim; 66420cf1dd8SToby Isaac 66520cf1dd8SToby Isaac PetscFunctionBegin; 66620cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 66720cf1dd8SToby Isaac PetscValidPointer(functional, 3); 6689566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(sp, &dim)); 66963a3b9bcSJacob Faibussowitsch PetscCheck(!(i < 0) && !(i >= dim),PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Functional index %" PetscInt_FMT " must be in [0, %" PetscInt_FMT ")", i, dim); 67020cf1dd8SToby Isaac *functional = sp->functional[i]; 67120cf1dd8SToby Isaac PetscFunctionReturn(0); 67220cf1dd8SToby Isaac } 67320cf1dd8SToby Isaac 67420cf1dd8SToby Isaac /*@ 67520cf1dd8SToby Isaac PetscDualSpaceGetDimension - Get the dimension of the dual space, i.e. the number of basis functionals 67620cf1dd8SToby Isaac 67720cf1dd8SToby Isaac Not collective 67820cf1dd8SToby Isaac 67920cf1dd8SToby Isaac Input Parameter: 68020cf1dd8SToby Isaac . sp - The PetscDualSpace 68120cf1dd8SToby Isaac 68220cf1dd8SToby Isaac Output Parameter: 68320cf1dd8SToby Isaac . dim - The dimension 68420cf1dd8SToby Isaac 68520cf1dd8SToby Isaac Level: intermediate 68620cf1dd8SToby Isaac 687db781477SPatrick Sanan .seealso: `PetscDualSpaceGetFunctional()`, `PetscDualSpaceCreate()` 68820cf1dd8SToby Isaac @*/ 68920cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetDimension(PetscDualSpace sp, PetscInt *dim) 69020cf1dd8SToby Isaac { 69120cf1dd8SToby Isaac PetscFunctionBegin; 69220cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 693dadcf809SJacob Faibussowitsch PetscValidIntPointer(dim, 2); 694b4457527SToby Isaac if (sp->spdim < 0) { 695b4457527SToby Isaac PetscSection section; 696b4457527SToby Isaac 6979566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetSection(sp, §ion)); 698b4457527SToby Isaac if (section) { 6999566063dSJacob Faibussowitsch PetscCall(PetscSectionGetStorageSize(section, &(sp->spdim))); 700b4457527SToby Isaac } else sp->spdim = 0; 701b4457527SToby Isaac } 702b4457527SToby Isaac *dim = sp->spdim; 70320cf1dd8SToby Isaac PetscFunctionReturn(0); 70420cf1dd8SToby Isaac } 70520cf1dd8SToby Isaac 706b4457527SToby Isaac /*@ 707b4457527SToby Isaac PetscDualSpaceGetInteriorDimension - Get the interior dimension of the dual space, i.e. the number of basis functionals assigned to the interior of the reference domain 708b4457527SToby Isaac 709b4457527SToby Isaac Not collective 710b4457527SToby Isaac 711b4457527SToby Isaac Input Parameter: 712b4457527SToby Isaac . sp - The PetscDualSpace 713b4457527SToby Isaac 714b4457527SToby Isaac Output Parameter: 715b4457527SToby Isaac . dim - The dimension 716b4457527SToby Isaac 717b4457527SToby Isaac Level: intermediate 718b4457527SToby Isaac 719db781477SPatrick Sanan .seealso: `PetscDualSpaceGetFunctional()`, `PetscDualSpaceCreate()` 720b4457527SToby Isaac @*/ 721b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetInteriorDimension(PetscDualSpace sp, PetscInt *intdim) 722b4457527SToby Isaac { 723b4457527SToby Isaac PetscFunctionBegin; 724b4457527SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 725dadcf809SJacob Faibussowitsch PetscValidIntPointer(intdim, 2); 726b4457527SToby Isaac if (sp->spintdim < 0) { 727b4457527SToby Isaac PetscSection section; 728b4457527SToby Isaac 7299566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetSection(sp, §ion)); 730b4457527SToby Isaac if (section) { 7319566063dSJacob Faibussowitsch PetscCall(PetscSectionGetConstrainedStorageSize(section, &(sp->spintdim))); 732b4457527SToby Isaac } else sp->spintdim = 0; 733b4457527SToby Isaac } 734b4457527SToby Isaac *intdim = sp->spintdim; 735b4457527SToby Isaac PetscFunctionReturn(0); 736b4457527SToby Isaac } 737b4457527SToby Isaac 738b4457527SToby Isaac /*@ 739b4457527SToby Isaac PetscDualSpaceGetUniform - Whether this dual space is uniform 740b4457527SToby Isaac 741b4457527SToby Isaac Not collective 742b4457527SToby Isaac 743b4457527SToby Isaac Input Parameters: 744b4457527SToby Isaac . sp - A dual space 745b4457527SToby Isaac 746b4457527SToby Isaac Output Parameters: 747b4457527SToby Isaac . uniform - PETSC_TRUE if (a) the dual space is the same for each point in a stratum of the reference DMPlex, and 748b4457527SToby Isaac (b) every symmetry of each point in the reference DMPlex is also a symmetry of the point's dual space. 749b4457527SToby Isaac 750b4457527SToby Isaac Level: advanced 751b4457527SToby Isaac 752b4457527SToby Isaac Note: all of the usual spaces on simplex or tensor-product elements will be uniform, only reference cells 753b4457527SToby Isaac with non-uniform strata (like trianguar-prisms) or anisotropic hp dual spaces will not be uniform. 754b4457527SToby Isaac 755db781477SPatrick Sanan .seealso: `PetscDualSpaceGetPointSubspace()`, `PetscDualSpaceGetSymmetries()` 756b4457527SToby Isaac @*/ 757b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetUniform(PetscDualSpace sp, PetscBool *uniform) 758b4457527SToby Isaac { 759b4457527SToby Isaac PetscFunctionBegin; 760b4457527SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 761dadcf809SJacob Faibussowitsch PetscValidBoolPointer(uniform, 2); 762b4457527SToby Isaac *uniform = sp->uniform; 763b4457527SToby Isaac PetscFunctionReturn(0); 764b4457527SToby Isaac } 765b4457527SToby Isaac 76620cf1dd8SToby Isaac /*@C 76720cf1dd8SToby Isaac PetscDualSpaceGetNumDof - Get the number of degrees of freedom for each spatial (topological) dimension 76820cf1dd8SToby Isaac 76920cf1dd8SToby Isaac Not collective 77020cf1dd8SToby Isaac 77120cf1dd8SToby Isaac Input Parameter: 77220cf1dd8SToby Isaac . sp - The PetscDualSpace 77320cf1dd8SToby Isaac 77420cf1dd8SToby Isaac Output Parameter: 77520cf1dd8SToby Isaac . numDof - An array of length dim+1 which holds the number of dofs for each dimension 77620cf1dd8SToby Isaac 77720cf1dd8SToby Isaac Level: intermediate 77820cf1dd8SToby Isaac 779db781477SPatrick Sanan .seealso: `PetscDualSpaceGetFunctional()`, `PetscDualSpaceCreate()` 78020cf1dd8SToby Isaac @*/ 78120cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetNumDof(PetscDualSpace sp, const PetscInt **numDof) 78220cf1dd8SToby Isaac { 78320cf1dd8SToby Isaac PetscFunctionBegin; 78420cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 78520cf1dd8SToby Isaac PetscValidPointer(numDof, 2); 78628b400f6SJacob Faibussowitsch PetscCheck(sp->uniform,PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "A non-uniform space does not have a fixed number of dofs for each height"); 787b4457527SToby Isaac if (!sp->numDof) { 788b4457527SToby Isaac DM dm; 789b4457527SToby Isaac PetscInt depth, d; 790b4457527SToby Isaac PetscSection section; 791b4457527SToby Isaac 7929566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 7939566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm, &depth)); 7949566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(depth+1,&(sp->numDof))); 7959566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetSection(sp, §ion)); 796b4457527SToby Isaac for (d = 0; d <= depth; d++) { 797b4457527SToby Isaac PetscInt dStart, dEnd; 798b4457527SToby Isaac 7999566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, d, &dStart, &dEnd)); 800b4457527SToby Isaac if (dEnd <= dStart) continue; 8019566063dSJacob Faibussowitsch PetscCall(PetscSectionGetDof(section, dStart, &(sp->numDof[d]))); 802b4457527SToby Isaac 803b4457527SToby Isaac } 804b4457527SToby Isaac } 805b4457527SToby Isaac *numDof = sp->numDof; 80608401ef6SPierre Jolivet PetscCheck(*numDof,PetscObjectComm((PetscObject) sp), PETSC_ERR_LIB, "Empty numDof[] returned from dual space implementation"); 80720cf1dd8SToby Isaac PetscFunctionReturn(0); 80820cf1dd8SToby Isaac } 80920cf1dd8SToby Isaac 810b4457527SToby Isaac /* create the section of the right size and set a permutation for topological ordering */ 811b4457527SToby Isaac PetscErrorCode PetscDualSpaceSectionCreate_Internal(PetscDualSpace sp, PetscSection *topSection) 812b4457527SToby Isaac { 813b4457527SToby Isaac DM dm; 814b4457527SToby Isaac PetscInt pStart, pEnd, cStart, cEnd, c, depth, count, i; 815b4457527SToby Isaac PetscInt *seen, *perm; 816b4457527SToby Isaac PetscSection section; 817b4457527SToby Isaac 818b4457527SToby Isaac PetscFunctionBegin; 819b4457527SToby Isaac dm = sp->dm; 8209566063dSJacob Faibussowitsch PetscCall(PetscSectionCreate(PETSC_COMM_SELF, §ion)); 8219566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 8229566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(section, pStart, pEnd)); 8239566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(pEnd - pStart, &seen)); 8249566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(pEnd - pStart, &perm)); 8259566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm, &depth)); 8269566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd)); 827b4457527SToby Isaac for (c = cStart, count = 0; c < cEnd; c++) { 828b4457527SToby Isaac PetscInt closureSize = -1, e; 829b4457527SToby Isaac PetscInt *closure = NULL; 830b4457527SToby Isaac 831b4457527SToby Isaac perm[count++] = c; 832b4457527SToby Isaac seen[c-pStart] = 1; 8339566063dSJacob Faibussowitsch PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 834b4457527SToby Isaac for (e = 0; e < closureSize; e++) { 835b4457527SToby Isaac PetscInt point = closure[2*e]; 836b4457527SToby Isaac 837b4457527SToby Isaac if (seen[point-pStart]) continue; 838b4457527SToby Isaac perm[count++] = point; 839b4457527SToby Isaac seen[point-pStart] = 1; 840b4457527SToby Isaac } 8419566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure)); 842b4457527SToby Isaac } 8431dca8a05SBarry Smith PetscCheck(count == pEnd - pStart,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Bad topological ordering"); 844b4457527SToby Isaac for (i = 0; i < pEnd - pStart; i++) if (perm[i] != i) break; 845b4457527SToby Isaac if (i < pEnd - pStart) { 846b4457527SToby Isaac IS permIS; 847b4457527SToby Isaac 8489566063dSJacob Faibussowitsch PetscCall(ISCreateGeneral(PETSC_COMM_SELF, pEnd - pStart, perm, PETSC_OWN_POINTER, &permIS)); 8499566063dSJacob Faibussowitsch PetscCall(ISSetPermutation(permIS)); 8509566063dSJacob Faibussowitsch PetscCall(PetscSectionSetPermutation(section, permIS)); 8519566063dSJacob Faibussowitsch PetscCall(ISDestroy(&permIS)); 852b4457527SToby Isaac } else { 8539566063dSJacob Faibussowitsch PetscCall(PetscFree(perm)); 854b4457527SToby Isaac } 8559566063dSJacob Faibussowitsch PetscCall(PetscFree(seen)); 856b4457527SToby Isaac *topSection = section; 857b4457527SToby Isaac PetscFunctionReturn(0); 858b4457527SToby Isaac } 859b4457527SToby Isaac 860b4457527SToby Isaac /* mark boundary points and set up */ 861b4457527SToby Isaac PetscErrorCode PetscDualSpaceSectionSetUp_Internal(PetscDualSpace sp, PetscSection section) 862b4457527SToby Isaac { 863b4457527SToby Isaac DM dm; 864b4457527SToby Isaac DMLabel boundary; 865b4457527SToby Isaac PetscInt pStart, pEnd, p; 866b4457527SToby Isaac 867b4457527SToby Isaac PetscFunctionBegin; 868b4457527SToby Isaac dm = sp->dm; 8699566063dSJacob Faibussowitsch PetscCall(DMLabelCreate(PETSC_COMM_SELF,"boundary",&boundary)); 8709566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp,&dm)); 8719566063dSJacob Faibussowitsch PetscCall(DMPlexMarkBoundaryFaces(dm,1,boundary)); 8729566063dSJacob Faibussowitsch PetscCall(DMPlexLabelComplete(dm,boundary)); 8739566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 874b4457527SToby Isaac for (p = pStart; p < pEnd; p++) { 875b4457527SToby Isaac PetscInt bval; 876b4457527SToby Isaac 8779566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(boundary, p, &bval)); 878b4457527SToby Isaac if (bval == 1) { 879b4457527SToby Isaac PetscInt dof; 880b4457527SToby Isaac 8819566063dSJacob Faibussowitsch PetscCall(PetscSectionGetDof(section, p, &dof)); 8829566063dSJacob Faibussowitsch PetscCall(PetscSectionSetConstraintDof(section, p, dof)); 883b4457527SToby Isaac } 884b4457527SToby Isaac } 8859566063dSJacob Faibussowitsch PetscCall(DMLabelDestroy(&boundary)); 8869566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(section)); 887b4457527SToby Isaac PetscFunctionReturn(0); 888b4457527SToby Isaac } 889b4457527SToby Isaac 890a4ce7ad1SMatthew G. Knepley /*@ 891b4457527SToby Isaac PetscDualSpaceGetSection - Create a PetscSection over the reference cell with the layout from this space 892a4ce7ad1SMatthew G. Knepley 893a4ce7ad1SMatthew G. Knepley Collective on sp 894a4ce7ad1SMatthew G. Knepley 895a4ce7ad1SMatthew G. Knepley Input Parameters: 896f0fc11ceSJed Brown . sp - The PetscDualSpace 897a4ce7ad1SMatthew G. Knepley 898a4ce7ad1SMatthew G. Knepley Output Parameter: 899a4ce7ad1SMatthew G. Knepley . section - The section 900a4ce7ad1SMatthew G. Knepley 901a4ce7ad1SMatthew G. Knepley Level: advanced 902a4ce7ad1SMatthew G. Knepley 903db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()`, `DMPLEX` 904a4ce7ad1SMatthew G. Knepley @*/ 905b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetSection(PetscDualSpace sp, PetscSection *section) 90620cf1dd8SToby Isaac { 907b4457527SToby Isaac PetscInt pStart, pEnd, p; 908b4457527SToby Isaac 909b4457527SToby Isaac PetscFunctionBegin; 910*78f1d139SRomain Beucher if (!sp->dm) { 911*78f1d139SRomain Beucher *section = NULL; 912*78f1d139SRomain Beucher PetscFunctionReturn(0); 913*78f1d139SRomain Beucher } 914b4457527SToby Isaac if (!sp->pointSection) { 915b4457527SToby Isaac /* mark the boundary */ 9169566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSectionCreate_Internal(sp, &(sp->pointSection))); 9179566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(sp->dm,&pStart,&pEnd)); 918b4457527SToby Isaac for (p = pStart; p < pEnd; p++) { 919b4457527SToby Isaac PetscDualSpace psp; 920b4457527SToby Isaac 9219566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetPointSubspace(sp, p, &psp)); 922b4457527SToby Isaac if (psp) { 923b4457527SToby Isaac PetscInt dof; 924b4457527SToby Isaac 9259566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetInteriorDimension(psp, &dof)); 9269566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(sp->pointSection,p,dof)); 927b4457527SToby Isaac } 928b4457527SToby Isaac } 9299566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceSectionSetUp_Internal(sp,sp->pointSection)); 930b4457527SToby Isaac } 931b4457527SToby Isaac *section = sp->pointSection; 932b4457527SToby Isaac PetscFunctionReturn(0); 933b4457527SToby Isaac } 934b4457527SToby Isaac 935b4457527SToby Isaac /* this assumes that all of the point dual spaces store their interior dofs first, which is true when the point DMs 936b4457527SToby Isaac * have one cell */ 937b4457527SToby Isaac PetscErrorCode PetscDualSpacePushForwardSubspaces_Internal(PetscDualSpace sp, PetscInt sStart, PetscInt sEnd) 938b4457527SToby Isaac { 939b4457527SToby Isaac PetscReal *sv0, *v0, *J; 940b4457527SToby Isaac PetscSection section; 941b4457527SToby Isaac PetscInt dim, s, k; 94220cf1dd8SToby Isaac DM dm; 94320cf1dd8SToby Isaac 94420cf1dd8SToby Isaac PetscFunctionBegin; 9459566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 9469566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 9479566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetSection(sp, §ion)); 9489566063dSJacob Faibussowitsch PetscCall(PetscMalloc3(dim, &v0, dim, &sv0, dim*dim, &J)); 9499566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFormDegree(sp, &k)); 950b4457527SToby Isaac for (s = sStart; s < sEnd; s++) { 951b4457527SToby Isaac PetscReal detJ, hdetJ; 952b4457527SToby Isaac PetscDualSpace ssp; 953b4457527SToby Isaac PetscInt dof, off, f, sdim; 954b4457527SToby Isaac PetscInt i, j; 955b4457527SToby Isaac DM sdm; 95620cf1dd8SToby Isaac 9579566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetPointSubspace(sp, s, &ssp)); 958b4457527SToby Isaac if (!ssp) continue; 9599566063dSJacob Faibussowitsch PetscCall(PetscSectionGetDof(section, s, &dof)); 9609566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(section, s, &off)); 961b4457527SToby Isaac /* get the first vertex of the reference cell */ 9629566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(ssp, &sdm)); 9639566063dSJacob Faibussowitsch PetscCall(DMGetDimension(sdm, &sdim)); 9649566063dSJacob Faibussowitsch PetscCall(DMPlexComputeCellGeometryAffineFEM(sdm, 0, sv0, NULL, NULL, &hdetJ)); 9659566063dSJacob Faibussowitsch PetscCall(DMPlexComputeCellGeometryAffineFEM(dm, s, v0, J, NULL, &detJ)); 966b4457527SToby Isaac /* compactify Jacobian */ 967b4457527SToby Isaac for (i = 0; i < dim; i++) for (j = 0; j < sdim; j++) J[i* sdim + j] = J[i * dim + j]; 968b4457527SToby Isaac for (f = 0; f < dof; f++) { 969b4457527SToby Isaac PetscQuadrature fn; 97020cf1dd8SToby Isaac 9719566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(ssp, f, &fn)); 9729566063dSJacob Faibussowitsch PetscCall(PetscQuadraturePushForward(fn, dim, sv0, v0, J, k, &(sp->functional[off+f]))); 97320cf1dd8SToby Isaac } 97420cf1dd8SToby Isaac } 9759566063dSJacob Faibussowitsch PetscCall(PetscFree3(v0, sv0, J)); 97620cf1dd8SToby Isaac PetscFunctionReturn(0); 97720cf1dd8SToby Isaac } 97820cf1dd8SToby Isaac 97920cf1dd8SToby Isaac /*@C 98020cf1dd8SToby Isaac PetscDualSpaceApply - Apply a functional from the dual space basis to an input function 98120cf1dd8SToby Isaac 98220cf1dd8SToby Isaac Input Parameters: 98320cf1dd8SToby Isaac + sp - The PetscDualSpace object 98420cf1dd8SToby Isaac . f - The basis functional index 98520cf1dd8SToby Isaac . time - The time 98620cf1dd8SToby Isaac . cgeom - A context with geometric information for this cell, we use v0 (the initial vertex) and J (the Jacobian) (or evaluated at the coordinates of the functional) 98720cf1dd8SToby Isaac . numComp - The number of components for the function 98820cf1dd8SToby Isaac . func - The input function 98920cf1dd8SToby Isaac - ctx - A context for the function 99020cf1dd8SToby Isaac 99120cf1dd8SToby Isaac Output Parameter: 99220cf1dd8SToby Isaac . value - numComp output values 99320cf1dd8SToby Isaac 99420cf1dd8SToby Isaac Note: The calling sequence for the callback func is given by: 99520cf1dd8SToby Isaac 99620cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[], 99720cf1dd8SToby Isaac $ PetscInt numComponents, PetscScalar values[], void *ctx) 99820cf1dd8SToby Isaac 999a4ce7ad1SMatthew G. Knepley Level: beginner 100020cf1dd8SToby Isaac 1001db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 100220cf1dd8SToby Isaac @*/ 100320cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApply(PetscDualSpace sp, PetscInt f, PetscReal time, PetscFEGeom *cgeom, PetscInt numComp, PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void *ctx, PetscScalar *value) 100420cf1dd8SToby Isaac { 100520cf1dd8SToby Isaac PetscFunctionBegin; 100620cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 100720cf1dd8SToby Isaac PetscValidPointer(cgeom, 4); 1008dadcf809SJacob Faibussowitsch PetscValidScalarPointer(value, 8); 10099566063dSJacob Faibussowitsch PetscCall((*sp->ops->apply)(sp, f, time, cgeom, numComp, func, ctx, value)); 101020cf1dd8SToby Isaac PetscFunctionReturn(0); 101120cf1dd8SToby Isaac } 101220cf1dd8SToby Isaac 101320cf1dd8SToby Isaac /*@C 1014b4457527SToby Isaac PetscDualSpaceApplyAll - Apply all functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetAllData() 101520cf1dd8SToby Isaac 101620cf1dd8SToby Isaac Input Parameters: 101720cf1dd8SToby Isaac + sp - The PetscDualSpace object 1018b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetAllData() 101920cf1dd8SToby Isaac 102020cf1dd8SToby Isaac Output Parameter: 102120cf1dd8SToby Isaac . spValue - The values of all dual space functionals 102220cf1dd8SToby Isaac 1023a4ce7ad1SMatthew G. Knepley Level: beginner 102420cf1dd8SToby Isaac 1025db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 102620cf1dd8SToby Isaac @*/ 102720cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyAll(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue) 102820cf1dd8SToby Isaac { 102920cf1dd8SToby Isaac PetscFunctionBegin; 103020cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 10319566063dSJacob Faibussowitsch PetscCall((*sp->ops->applyall)(sp, pointEval, spValue)); 103220cf1dd8SToby Isaac PetscFunctionReturn(0); 103320cf1dd8SToby Isaac } 103420cf1dd8SToby Isaac 103520cf1dd8SToby Isaac /*@C 1036b4457527SToby Isaac PetscDualSpaceApplyInterior - Apply interior functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetInteriorData() 1037b4457527SToby Isaac 1038b4457527SToby Isaac Input Parameters: 1039b4457527SToby Isaac + sp - The PetscDualSpace object 1040b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetInteriorData() 1041b4457527SToby Isaac 1042b4457527SToby Isaac Output Parameter: 1043b4457527SToby Isaac . spValue - The values of interior dual space functionals 1044b4457527SToby Isaac 1045b4457527SToby Isaac Level: beginner 1046b4457527SToby Isaac 1047db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 1048b4457527SToby Isaac @*/ 1049b4457527SToby Isaac PetscErrorCode PetscDualSpaceApplyInterior(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue) 1050b4457527SToby Isaac { 1051b4457527SToby Isaac PetscFunctionBegin; 1052b4457527SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 10539566063dSJacob Faibussowitsch PetscCall((*sp->ops->applyint)(sp, pointEval, spValue)); 1054b4457527SToby Isaac PetscFunctionReturn(0); 1055b4457527SToby Isaac } 1056b4457527SToby Isaac 1057b4457527SToby Isaac /*@C 105820cf1dd8SToby Isaac PetscDualSpaceApplyDefault - Apply a functional from the dual space basis to an input function by assuming a point evaluation functional. 105920cf1dd8SToby Isaac 106020cf1dd8SToby Isaac Input Parameters: 106120cf1dd8SToby Isaac + sp - The PetscDualSpace object 106220cf1dd8SToby Isaac . f - The basis functional index 106320cf1dd8SToby Isaac . time - The time 106420cf1dd8SToby Isaac . cgeom - A context with geometric information for this cell, we use v0 (the initial vertex) and J (the Jacobian) 106520cf1dd8SToby Isaac . Nc - The number of components for the function 106620cf1dd8SToby Isaac . func - The input function 106720cf1dd8SToby Isaac - ctx - A context for the function 106820cf1dd8SToby Isaac 106920cf1dd8SToby Isaac Output Parameter: 107020cf1dd8SToby Isaac . value - The output value 107120cf1dd8SToby Isaac 107220cf1dd8SToby Isaac Note: The calling sequence for the callback func is given by: 107320cf1dd8SToby Isaac 107420cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[], 107520cf1dd8SToby Isaac $ PetscInt numComponents, PetscScalar values[], void *ctx) 107620cf1dd8SToby Isaac 107720cf1dd8SToby Isaac and the idea is to evaluate the functional as an integral 107820cf1dd8SToby Isaac 107920cf1dd8SToby Isaac $ n(f) = int dx n(x) . f(x) 108020cf1dd8SToby Isaac 108120cf1dd8SToby Isaac where both n and f have Nc components. 108220cf1dd8SToby Isaac 1083a4ce7ad1SMatthew G. Knepley Level: beginner 108420cf1dd8SToby Isaac 1085db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 108620cf1dd8SToby Isaac @*/ 108720cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyDefault(PetscDualSpace sp, PetscInt f, PetscReal time, PetscFEGeom *cgeom, PetscInt Nc, PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void *ctx, PetscScalar *value) 108820cf1dd8SToby Isaac { 108920cf1dd8SToby Isaac DM dm; 109020cf1dd8SToby Isaac PetscQuadrature n; 109120cf1dd8SToby Isaac const PetscReal *points, *weights; 109220cf1dd8SToby Isaac PetscReal x[3]; 109320cf1dd8SToby Isaac PetscScalar *val; 109420cf1dd8SToby Isaac PetscInt dim, dE, qNc, c, Nq, q; 109520cf1dd8SToby Isaac PetscBool isAffine; 109620cf1dd8SToby Isaac 109720cf1dd8SToby Isaac PetscFunctionBegin; 109820cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1099dadcf809SJacob Faibussowitsch PetscValidScalarPointer(value, 8); 11009566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 11019566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp, f, &n)); 11029566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(n, &dim, &qNc, &Nq, &points, &weights)); 110363a3b9bcSJacob Faibussowitsch PetscCheck(dim == cgeom->dim,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_SIZ, "The quadrature spatial dimension %" PetscInt_FMT " != cell geometry dimension %" PetscInt_FMT, dim, cgeom->dim); 110463a3b9bcSJacob Faibussowitsch PetscCheck(qNc == Nc,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_SIZ, "The quadrature components %" PetscInt_FMT " != function components %" PetscInt_FMT, qNc, Nc); 11059566063dSJacob Faibussowitsch PetscCall(DMGetWorkArray(dm, Nc, MPIU_SCALAR, &val)); 110620cf1dd8SToby Isaac *value = 0.0; 110720cf1dd8SToby Isaac isAffine = cgeom->isAffine; 110820cf1dd8SToby Isaac dE = cgeom->dimEmbed; 110920cf1dd8SToby Isaac for (q = 0; q < Nq; ++q) { 111020cf1dd8SToby Isaac if (isAffine) { 111120cf1dd8SToby Isaac CoordinatesRefToReal(dE, cgeom->dim, cgeom->xi, cgeom->v, cgeom->J, &points[q*dim], x); 11129566063dSJacob Faibussowitsch PetscCall((*func)(dE, time, x, Nc, val, ctx)); 111320cf1dd8SToby Isaac } else { 11149566063dSJacob Faibussowitsch PetscCall((*func)(dE, time, &cgeom->v[dE*q], Nc, val, ctx)); 111520cf1dd8SToby Isaac } 111620cf1dd8SToby Isaac for (c = 0; c < Nc; ++c) { 111720cf1dd8SToby Isaac *value += val[c]*weights[q*Nc+c]; 111820cf1dd8SToby Isaac } 111920cf1dd8SToby Isaac } 11209566063dSJacob Faibussowitsch PetscCall(DMRestoreWorkArray(dm, Nc, MPIU_SCALAR, &val)); 112120cf1dd8SToby Isaac PetscFunctionReturn(0); 112220cf1dd8SToby Isaac } 112320cf1dd8SToby Isaac 112420cf1dd8SToby Isaac /*@C 1125b4457527SToby Isaac PetscDualSpaceApplyAllDefault - Apply all functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetAllData() 112620cf1dd8SToby Isaac 112720cf1dd8SToby Isaac Input Parameters: 112820cf1dd8SToby Isaac + sp - The PetscDualSpace object 1129b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetAllData() 113020cf1dd8SToby Isaac 113120cf1dd8SToby Isaac Output Parameter: 113220cf1dd8SToby Isaac . spValue - The values of all dual space functionals 113320cf1dd8SToby Isaac 1134a4ce7ad1SMatthew G. Knepley Level: beginner 113520cf1dd8SToby Isaac 1136db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 113720cf1dd8SToby Isaac @*/ 113820cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyAllDefault(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue) 113920cf1dd8SToby Isaac { 1140b4457527SToby Isaac Vec pointValues, dofValues; 1141b4457527SToby Isaac Mat allMat; 114220cf1dd8SToby Isaac 114320cf1dd8SToby Isaac PetscFunctionBegin; 114420cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 114520cf1dd8SToby Isaac PetscValidScalarPointer(pointEval, 2); 1146064a246eSJacob Faibussowitsch PetscValidScalarPointer(spValue, 3); 11479566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetAllData(sp, NULL, &allMat)); 1148b4457527SToby Isaac if (!(sp->allNodeValues)) { 11499566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(allMat, &(sp->allNodeValues), NULL)); 115020cf1dd8SToby Isaac } 1151b4457527SToby Isaac pointValues = sp->allNodeValues; 1152b4457527SToby Isaac if (!(sp->allDofValues)) { 11539566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(allMat, NULL, &(sp->allDofValues))); 115420cf1dd8SToby Isaac } 1155b4457527SToby Isaac dofValues = sp->allDofValues; 11569566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(pointValues, pointEval)); 11579566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(dofValues, spValue)); 11589566063dSJacob Faibussowitsch PetscCall(MatMult(allMat, pointValues, dofValues)); 11599566063dSJacob Faibussowitsch PetscCall(VecResetArray(dofValues)); 11609566063dSJacob Faibussowitsch PetscCall(VecResetArray(pointValues)); 1161b4457527SToby Isaac PetscFunctionReturn(0); 116220cf1dd8SToby Isaac } 1163b4457527SToby Isaac 1164b4457527SToby Isaac /*@C 1165b4457527SToby Isaac PetscDualSpaceApplyInteriorDefault - Apply interior functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetInteriorData() 1166b4457527SToby Isaac 1167b4457527SToby Isaac Input Parameters: 1168b4457527SToby Isaac + sp - The PetscDualSpace object 1169b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetInteriorData() 1170b4457527SToby Isaac 1171b4457527SToby Isaac Output Parameter: 1172b4457527SToby Isaac . spValue - The values of interior dual space functionals 1173b4457527SToby Isaac 1174b4457527SToby Isaac Level: beginner 1175b4457527SToby Isaac 1176db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 1177b4457527SToby Isaac @*/ 1178b4457527SToby Isaac PetscErrorCode PetscDualSpaceApplyInteriorDefault(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue) 1179b4457527SToby Isaac { 1180b4457527SToby Isaac Vec pointValues, dofValues; 1181b4457527SToby Isaac Mat intMat; 1182b4457527SToby Isaac 1183b4457527SToby Isaac PetscFunctionBegin; 1184b4457527SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1185b4457527SToby Isaac PetscValidScalarPointer(pointEval, 2); 1186064a246eSJacob Faibussowitsch PetscValidScalarPointer(spValue, 3); 11879566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetInteriorData(sp, NULL, &intMat)); 1188b4457527SToby Isaac if (!(sp->intNodeValues)) { 11899566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(intMat, &(sp->intNodeValues), NULL)); 1190b4457527SToby Isaac } 1191b4457527SToby Isaac pointValues = sp->intNodeValues; 1192b4457527SToby Isaac if (!(sp->intDofValues)) { 11939566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(intMat, NULL, &(sp->intDofValues))); 1194b4457527SToby Isaac } 1195b4457527SToby Isaac dofValues = sp->intDofValues; 11969566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(pointValues, pointEval)); 11979566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(dofValues, spValue)); 11989566063dSJacob Faibussowitsch PetscCall(MatMult(intMat, pointValues, dofValues)); 11999566063dSJacob Faibussowitsch PetscCall(VecResetArray(dofValues)); 12009566063dSJacob Faibussowitsch PetscCall(VecResetArray(pointValues)); 120120cf1dd8SToby Isaac PetscFunctionReturn(0); 120220cf1dd8SToby Isaac } 120320cf1dd8SToby Isaac 1204a4ce7ad1SMatthew G. Knepley /*@ 1205b4457527SToby Isaac PetscDualSpaceGetAllData - Get all quadrature nodes from this space, and the matrix that sends quadrature node values to degree-of-freedom values 1206a4ce7ad1SMatthew G. Knepley 1207a4ce7ad1SMatthew G. Knepley Input Parameter: 1208a4ce7ad1SMatthew G. Knepley . sp - The dualspace 1209a4ce7ad1SMatthew G. Knepley 1210d8d19677SJose E. Roman Output Parameters: 1211b4457527SToby Isaac + allNodes - A PetscQuadrature object containing all evaluation nodes 1212b4457527SToby Isaac - allMat - A Mat for the node-to-dof transformation 1213a4ce7ad1SMatthew G. Knepley 1214a4ce7ad1SMatthew G. Knepley Level: advanced 1215a4ce7ad1SMatthew G. Knepley 1216db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 1217a4ce7ad1SMatthew G. Knepley @*/ 1218b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetAllData(PetscDualSpace sp, PetscQuadrature *allNodes, Mat *allMat) 121920cf1dd8SToby Isaac { 122020cf1dd8SToby Isaac PetscFunctionBegin; 122120cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1222b4457527SToby Isaac if (allNodes) PetscValidPointer(allNodes,2); 1223b4457527SToby Isaac if (allMat) PetscValidPointer(allMat,3); 1224b4457527SToby Isaac if ((!sp->allNodes || !sp->allMat) && sp->ops->createalldata) { 1225b4457527SToby Isaac PetscQuadrature qpoints; 1226b4457527SToby Isaac Mat amat; 1227b4457527SToby Isaac 12289566063dSJacob Faibussowitsch PetscCall((*sp->ops->createalldata)(sp,&qpoints,&amat)); 12299566063dSJacob Faibussowitsch PetscCall(PetscQuadratureDestroy(&(sp->allNodes))); 12309566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(sp->allMat))); 1231b4457527SToby Isaac sp->allNodes = qpoints; 1232b4457527SToby Isaac sp->allMat = amat; 123320cf1dd8SToby Isaac } 1234b4457527SToby Isaac if (allNodes) *allNodes = sp->allNodes; 1235b4457527SToby Isaac if (allMat) *allMat = sp->allMat; 123620cf1dd8SToby Isaac PetscFunctionReturn(0); 123720cf1dd8SToby Isaac } 123820cf1dd8SToby Isaac 1239a4ce7ad1SMatthew G. Knepley /*@ 1240b4457527SToby Isaac PetscDualSpaceCreateAllDataDefault - Create all evaluation nodes and the node-to-dof matrix by examining functionals 1241a4ce7ad1SMatthew G. Knepley 1242a4ce7ad1SMatthew G. Knepley Input Parameter: 1243a4ce7ad1SMatthew G. Knepley . sp - The dualspace 1244a4ce7ad1SMatthew G. Knepley 1245d8d19677SJose E. Roman Output Parameters: 1246b4457527SToby Isaac + allNodes - A PetscQuadrature object containing all evaluation nodes 1247b4457527SToby Isaac - allMat - A Mat for the node-to-dof transformation 1248a4ce7ad1SMatthew G. Knepley 1249a4ce7ad1SMatthew G. Knepley Level: advanced 1250a4ce7ad1SMatthew G. Knepley 1251db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 1252a4ce7ad1SMatthew G. Knepley @*/ 1253b4457527SToby Isaac PetscErrorCode PetscDualSpaceCreateAllDataDefault(PetscDualSpace sp, PetscQuadrature *allNodes, Mat *allMat) 125420cf1dd8SToby Isaac { 125520cf1dd8SToby Isaac PetscInt spdim; 125620cf1dd8SToby Isaac PetscInt numPoints, offset; 125720cf1dd8SToby Isaac PetscReal *points; 125820cf1dd8SToby Isaac PetscInt f, dim; 1259b4457527SToby Isaac PetscInt Nc, nrows, ncols; 1260b4457527SToby Isaac PetscInt maxNumPoints; 126120cf1dd8SToby Isaac PetscQuadrature q; 1262b4457527SToby Isaac Mat A; 126320cf1dd8SToby Isaac 126420cf1dd8SToby Isaac PetscFunctionBegin; 12659566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetNumComponents(sp, &Nc)); 12669566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(sp,&spdim)); 126720cf1dd8SToby Isaac if (!spdim) { 12689566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,allNodes)); 12699566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*allNodes,0,0,0,NULL,NULL)); 127020cf1dd8SToby Isaac } 1271b4457527SToby Isaac nrows = spdim; 12729566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp,0,&q)); 12739566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q,&dim,NULL,&numPoints,NULL,NULL)); 1274b4457527SToby Isaac maxNumPoints = numPoints; 127520cf1dd8SToby Isaac for (f = 1; f < spdim; f++) { 127620cf1dd8SToby Isaac PetscInt Np; 127720cf1dd8SToby Isaac 12789566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp,f,&q)); 12799566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,NULL,NULL)); 128020cf1dd8SToby Isaac numPoints += Np; 1281b4457527SToby Isaac maxNumPoints = PetscMax(maxNumPoints,Np); 128220cf1dd8SToby Isaac } 1283b4457527SToby Isaac ncols = numPoints * Nc; 12849566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim*numPoints,&points)); 12859566063dSJacob Faibussowitsch PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF, nrows, ncols, maxNumPoints * Nc, NULL, &A)); 128620cf1dd8SToby Isaac for (f = 0, offset = 0; f < spdim; f++) { 1287b4457527SToby Isaac const PetscReal *p, *w; 128820cf1dd8SToby Isaac PetscInt Np, i; 1289b4457527SToby Isaac PetscInt fnc; 129020cf1dd8SToby Isaac 12919566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp,f,&q)); 12929566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q,NULL,&fnc,&Np,&p,&w)); 129308401ef6SPierre Jolivet PetscCheck(fnc == Nc,PETSC_COMM_SELF, PETSC_ERR_PLIB, "functional component mismatch"); 1294b4457527SToby Isaac for (i = 0; i < Np * dim; i++) { 1295b4457527SToby Isaac points[offset* dim + i] = p[i]; 1296b4457527SToby Isaac } 1297b4457527SToby Isaac for (i = 0; i < Np * Nc; i++) { 12989566063dSJacob Faibussowitsch PetscCall(MatSetValue(A, f, offset * Nc, w[i], INSERT_VALUES)); 1299b4457527SToby Isaac } 1300b4457527SToby Isaac offset += Np; 1301b4457527SToby Isaac } 13029566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 13039566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 13049566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,allNodes)); 13059566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*allNodes,dim,0,numPoints,points,NULL)); 1306b4457527SToby Isaac *allMat = A; 1307b4457527SToby Isaac PetscFunctionReturn(0); 1308b4457527SToby Isaac } 1309b4457527SToby Isaac 1310b4457527SToby Isaac /*@ 1311b4457527SToby Isaac PetscDualSpaceGetInteriorData - Get all quadrature points necessary to compute the interior degrees of freedom from 1312b4457527SToby Isaac this space, as well as the matrix that computes the degrees of freedom from the quadrature values. Degrees of 1313b4457527SToby Isaac freedom are interior degrees of freedom if they belong (by PetscDualSpaceGetSection()) to interior points in the 1314b4457527SToby Isaac reference DMPlex: complementary boundary degrees of freedom are marked as constrained in the section returned by 1315b4457527SToby Isaac PetscDualSpaceGetSection()). 1316b4457527SToby Isaac 1317b4457527SToby Isaac Input Parameter: 1318b4457527SToby Isaac . sp - The dualspace 1319b4457527SToby Isaac 1320d8d19677SJose E. Roman Output Parameters: 1321b4457527SToby Isaac + intNodes - A PetscQuadrature object containing all evaluation points needed to evaluate interior degrees of freedom 1322b4457527SToby Isaac - intMat - A matrix that computes dual space values from point values: size [spdim0 x (npoints * nc)], where spdim0 is 1323b4457527SToby Isaac the size of the constrained layout (PetscSectionGetConstrainStorageSize()) of the dual space section, 1324b4457527SToby Isaac npoints is the number of points in intNodes and nc is PetscDualSpaceGetNumComponents(). 1325b4457527SToby Isaac 1326b4457527SToby Isaac Level: advanced 1327b4457527SToby Isaac 1328db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()`, `PetscDualSpaceGetDimension()`, `PetscDualSpaceGetNumComponents()`, `PetscQuadratureGetData()` 1329b4457527SToby Isaac @*/ 1330b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetInteriorData(PetscDualSpace sp, PetscQuadrature *intNodes, Mat *intMat) 1331b4457527SToby Isaac { 1332b4457527SToby Isaac PetscFunctionBegin; 1333b4457527SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1334b4457527SToby Isaac if (intNodes) PetscValidPointer(intNodes,2); 1335b4457527SToby Isaac if (intMat) PetscValidPointer(intMat,3); 1336b4457527SToby Isaac if ((!sp->intNodes || !sp->intMat) && sp->ops->createintdata) { 1337b4457527SToby Isaac PetscQuadrature qpoints; 1338b4457527SToby Isaac Mat imat; 1339b4457527SToby Isaac 13409566063dSJacob Faibussowitsch PetscCall((*sp->ops->createintdata)(sp,&qpoints,&imat)); 13419566063dSJacob Faibussowitsch PetscCall(PetscQuadratureDestroy(&(sp->intNodes))); 13429566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(sp->intMat))); 1343b4457527SToby Isaac sp->intNodes = qpoints; 1344b4457527SToby Isaac sp->intMat = imat; 1345b4457527SToby Isaac } 1346b4457527SToby Isaac if (intNodes) *intNodes = sp->intNodes; 1347b4457527SToby Isaac if (intMat) *intMat = sp->intMat; 1348b4457527SToby Isaac PetscFunctionReturn(0); 1349b4457527SToby Isaac } 1350b4457527SToby Isaac 1351b4457527SToby Isaac /*@ 1352b4457527SToby Isaac PetscDualSpaceCreateInteriorDataDefault - Create quadrature points by examining interior functionals and create the matrix mapping quadrature point values to interior dual space values 1353b4457527SToby Isaac 1354b4457527SToby Isaac Input Parameter: 1355b4457527SToby Isaac . sp - The dualspace 1356b4457527SToby Isaac 1357d8d19677SJose E. Roman Output Parameters: 1358b4457527SToby Isaac + intNodes - A PetscQuadrature object containing all evaluation points needed to evaluate interior degrees of freedom 1359b4457527SToby Isaac - intMat - A matrix that computes dual space values from point values: size [spdim0 x (npoints * nc)], where spdim0 is 1360b4457527SToby Isaac the size of the constrained layout (PetscSectionGetConstrainStorageSize()) of the dual space section, 1361b4457527SToby Isaac npoints is the number of points in allNodes and nc is PetscDualSpaceGetNumComponents(). 1362b4457527SToby Isaac 1363b4457527SToby Isaac Level: advanced 1364b4457527SToby Isaac 1365db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()`, `PetscDualSpaceGetInteriorData()` 1366b4457527SToby Isaac @*/ 1367b4457527SToby Isaac PetscErrorCode PetscDualSpaceCreateInteriorDataDefault(PetscDualSpace sp, PetscQuadrature *intNodes, Mat *intMat) 1368b4457527SToby Isaac { 1369b4457527SToby Isaac DM dm; 1370b4457527SToby Isaac PetscInt spdim0; 1371b4457527SToby Isaac PetscInt Nc; 1372b4457527SToby Isaac PetscInt pStart, pEnd, p, f; 1373b4457527SToby Isaac PetscSection section; 1374b4457527SToby Isaac PetscInt numPoints, offset, matoffset; 1375b4457527SToby Isaac PetscReal *points; 1376b4457527SToby Isaac PetscInt dim; 1377b4457527SToby Isaac PetscInt *nnz; 1378b4457527SToby Isaac PetscQuadrature q; 1379b4457527SToby Isaac Mat imat; 1380b4457527SToby Isaac 1381b4457527SToby Isaac PetscFunctionBegin; 1382b4457527SToby Isaac PetscValidHeaderSpecific(sp,PETSCDUALSPACE_CLASSID,1); 13839566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetSection(sp, §ion)); 13849566063dSJacob Faibussowitsch PetscCall(PetscSectionGetConstrainedStorageSize(section, &spdim0)); 1385b4457527SToby Isaac if (!spdim0) { 1386b4457527SToby Isaac *intNodes = NULL; 1387b4457527SToby Isaac *intMat = NULL; 1388b4457527SToby Isaac PetscFunctionReturn(0); 1389b4457527SToby Isaac } 13909566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetNumComponents(sp, &Nc)); 13919566063dSJacob Faibussowitsch PetscCall(PetscSectionGetChart(section, &pStart, &pEnd)); 13929566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 13939566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 13949566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(spdim0, &nnz)); 1395b4457527SToby Isaac for (p = pStart, f = 0, numPoints = 0; p < pEnd; p++) { 1396b4457527SToby Isaac PetscInt dof, cdof, off, d; 1397b4457527SToby Isaac 13989566063dSJacob Faibussowitsch PetscCall(PetscSectionGetDof(section, p, &dof)); 13999566063dSJacob Faibussowitsch PetscCall(PetscSectionGetConstraintDof(section, p, &cdof)); 1400b4457527SToby Isaac if (!(dof - cdof)) continue; 14019566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(section, p, &off)); 1402b4457527SToby Isaac for (d = 0; d < dof; d++, off++, f++) { 1403b4457527SToby Isaac PetscInt Np; 1404b4457527SToby Isaac 14059566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp,off,&q)); 14069566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,NULL,NULL)); 1407b4457527SToby Isaac nnz[f] = Np * Nc; 1408b4457527SToby Isaac numPoints += Np; 1409b4457527SToby Isaac } 1410b4457527SToby Isaac } 14119566063dSJacob Faibussowitsch PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF, spdim0, numPoints * Nc, 0, nnz, &imat)); 14129566063dSJacob Faibussowitsch PetscCall(PetscFree(nnz)); 14139566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(dim*numPoints,&points)); 1414b4457527SToby Isaac for (p = pStart, f = 0, offset = 0, matoffset = 0; p < pEnd; p++) { 1415b4457527SToby Isaac PetscInt dof, cdof, off, d; 1416b4457527SToby Isaac 14179566063dSJacob Faibussowitsch PetscCall(PetscSectionGetDof(section, p, &dof)); 14189566063dSJacob Faibussowitsch PetscCall(PetscSectionGetConstraintDof(section, p, &cdof)); 1419b4457527SToby Isaac if (!(dof - cdof)) continue; 14209566063dSJacob Faibussowitsch PetscCall(PetscSectionGetOffset(section, p, &off)); 1421b4457527SToby Isaac for (d = 0; d < dof; d++, off++, f++) { 1422b4457527SToby Isaac const PetscReal *p; 1423b4457527SToby Isaac const PetscReal *w; 1424b4457527SToby Isaac PetscInt Np, i; 1425b4457527SToby Isaac 14269566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp,off,&q)); 14279566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,&p,&w)); 142820cf1dd8SToby Isaac for (i = 0; i < Np * dim; i++) { 142920cf1dd8SToby Isaac points[offset + i] = p[i]; 143020cf1dd8SToby Isaac } 1431b4457527SToby Isaac for (i = 0; i < Np * Nc; i++) { 14329566063dSJacob Faibussowitsch PetscCall(MatSetValue(imat, f, matoffset + i, w[i],INSERT_VALUES)); 143320cf1dd8SToby Isaac } 1434b4457527SToby Isaac offset += Np * dim; 1435b4457527SToby Isaac matoffset += Np * Nc; 1436b4457527SToby Isaac } 1437b4457527SToby Isaac } 14389566063dSJacob Faibussowitsch PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,intNodes)); 14399566063dSJacob Faibussowitsch PetscCall(PetscQuadratureSetData(*intNodes,dim,0,numPoints,points,NULL)); 14409566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(imat, MAT_FINAL_ASSEMBLY)); 14419566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(imat, MAT_FINAL_ASSEMBLY)); 1442b4457527SToby Isaac *intMat = imat; 144320cf1dd8SToby Isaac PetscFunctionReturn(0); 144420cf1dd8SToby Isaac } 144520cf1dd8SToby Isaac 14464f9ab2b4SJed Brown /*@ 14474f9ab2b4SJed Brown PetscDualSpaceEqual - Determine if a dual space is equivalent 14484f9ab2b4SJed Brown 14494f9ab2b4SJed Brown Input Parameters: 14504f9ab2b4SJed Brown + A - A PetscDualSpace object 14514f9ab2b4SJed Brown - B - Another PetscDualSpace object 14524f9ab2b4SJed Brown 14534f9ab2b4SJed Brown Output Parameter: 14544f9ab2b4SJed Brown . equal - PETSC_TRUE if the dual spaces are equivalent 14554f9ab2b4SJed Brown 14564f9ab2b4SJed Brown Level: advanced 14574f9ab2b4SJed Brown 1458db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 14594f9ab2b4SJed Brown @*/ 14604f9ab2b4SJed Brown PetscErrorCode PetscDualSpaceEqual(PetscDualSpace A, PetscDualSpace B, PetscBool *equal) 14614f9ab2b4SJed Brown { 14624f9ab2b4SJed Brown PetscInt sizeA, sizeB, dimA, dimB; 14634f9ab2b4SJed Brown const PetscInt *dofA, *dofB; 14644f9ab2b4SJed Brown PetscQuadrature quadA, quadB; 14654f9ab2b4SJed Brown Mat matA, matB; 14664f9ab2b4SJed Brown 14674f9ab2b4SJed Brown PetscFunctionBegin; 14684f9ab2b4SJed Brown PetscValidHeaderSpecific(A,PETSCDUALSPACE_CLASSID,1); 14694f9ab2b4SJed Brown PetscValidHeaderSpecific(B,PETSCDUALSPACE_CLASSID,2); 14704f9ab2b4SJed Brown PetscValidBoolPointer(equal,3); 14714f9ab2b4SJed Brown *equal = PETSC_FALSE; 14729566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(A, &sizeA)); 14739566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDimension(B, &sizeB)); 14744f9ab2b4SJed Brown if (sizeB != sizeA) { 14754f9ab2b4SJed Brown PetscFunctionReturn(0); 14764f9ab2b4SJed Brown } 14779566063dSJacob Faibussowitsch PetscCall(DMGetDimension(A->dm, &dimA)); 14789566063dSJacob Faibussowitsch PetscCall(DMGetDimension(B->dm, &dimB)); 14794f9ab2b4SJed Brown if (dimA != dimB) { 14804f9ab2b4SJed Brown PetscFunctionReturn(0); 14814f9ab2b4SJed Brown } 14824f9ab2b4SJed Brown 14839566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetNumDof(A, &dofA)); 14849566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetNumDof(B, &dofB)); 14854f9ab2b4SJed Brown for (PetscInt d=0; d<dimA; d++) { 14864f9ab2b4SJed Brown if (dofA[d] != dofB[d]) { 14874f9ab2b4SJed Brown PetscFunctionReturn(0); 14884f9ab2b4SJed Brown } 14894f9ab2b4SJed Brown } 14904f9ab2b4SJed Brown 14919566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetInteriorData(A, &quadA, &matA)); 14929566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetInteriorData(B, &quadB, &matB)); 14934f9ab2b4SJed Brown if (!quadA && !quadB) { 14944f9ab2b4SJed Brown *equal = PETSC_TRUE; 14954f9ab2b4SJed Brown } else if (quadA && quadB) { 14969566063dSJacob Faibussowitsch PetscCall(PetscQuadratureEqual(quadA, quadB, equal)); 14974f9ab2b4SJed Brown if (*equal == PETSC_FALSE) PetscFunctionReturn(0); 14984f9ab2b4SJed Brown if (!matA && !matB) PetscFunctionReturn(0); 14999566063dSJacob Faibussowitsch if (matA && matB) PetscCall(MatEqual(matA, matB, equal)); 15004f9ab2b4SJed Brown else *equal = PETSC_FALSE; 15014f9ab2b4SJed Brown } 15024f9ab2b4SJed Brown PetscFunctionReturn(0); 15034f9ab2b4SJed Brown } 15044f9ab2b4SJed Brown 150520cf1dd8SToby Isaac /*@C 150620cf1dd8SToby Isaac PetscDualSpaceApplyFVM - Apply a functional from the dual space basis to an input function by assuming a point evaluation functional at the cell centroid. 150720cf1dd8SToby Isaac 150820cf1dd8SToby Isaac Input Parameters: 150920cf1dd8SToby Isaac + sp - The PetscDualSpace object 151020cf1dd8SToby Isaac . f - The basis functional index 151120cf1dd8SToby Isaac . time - The time 151220cf1dd8SToby Isaac . cgeom - A context with geometric information for this cell, we currently just use the centroid 151320cf1dd8SToby Isaac . Nc - The number of components for the function 151420cf1dd8SToby Isaac . func - The input function 151520cf1dd8SToby Isaac - ctx - A context for the function 151620cf1dd8SToby Isaac 151720cf1dd8SToby Isaac Output Parameter: 151820cf1dd8SToby Isaac . value - The output value (scalar) 151920cf1dd8SToby Isaac 152020cf1dd8SToby Isaac Note: The calling sequence for the callback func is given by: 152120cf1dd8SToby Isaac 152220cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[], 152320cf1dd8SToby Isaac $ PetscInt numComponents, PetscScalar values[], void *ctx) 152420cf1dd8SToby Isaac 152520cf1dd8SToby Isaac and the idea is to evaluate the functional as an integral 152620cf1dd8SToby Isaac 152720cf1dd8SToby Isaac $ n(f) = int dx n(x) . f(x) 152820cf1dd8SToby Isaac 152920cf1dd8SToby Isaac where both n and f have Nc components. 153020cf1dd8SToby Isaac 1531a4ce7ad1SMatthew G. Knepley Level: beginner 153220cf1dd8SToby Isaac 1533db781477SPatrick Sanan .seealso: `PetscDualSpaceCreate()` 153420cf1dd8SToby Isaac @*/ 153520cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyFVM(PetscDualSpace sp, PetscInt f, PetscReal time, PetscFVCellGeom *cgeom, PetscInt Nc, PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void *ctx, PetscScalar *value) 153620cf1dd8SToby Isaac { 153720cf1dd8SToby Isaac DM dm; 153820cf1dd8SToby Isaac PetscQuadrature n; 153920cf1dd8SToby Isaac const PetscReal *points, *weights; 154020cf1dd8SToby Isaac PetscScalar *val; 154120cf1dd8SToby Isaac PetscInt dimEmbed, qNc, c, Nq, q; 154220cf1dd8SToby Isaac 154320cf1dd8SToby Isaac PetscFunctionBegin; 154420cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1545dadcf809SJacob Faibussowitsch PetscValidScalarPointer(value, 8); 15469566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDM(sp, &dm)); 15479566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDim(dm, &dimEmbed)); 15489566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetFunctional(sp, f, &n)); 15499566063dSJacob Faibussowitsch PetscCall(PetscQuadratureGetData(n, NULL, &qNc, &Nq, &points, &weights)); 155063a3b9bcSJacob Faibussowitsch PetscCheck(qNc == Nc,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_SIZ, "The quadrature components %" PetscInt_FMT " != function components %" PetscInt_FMT, qNc, Nc); 15519566063dSJacob Faibussowitsch PetscCall(DMGetWorkArray(dm, Nc, MPIU_SCALAR, &val)); 155220cf1dd8SToby Isaac *value = 0.; 155320cf1dd8SToby Isaac for (q = 0; q < Nq; ++q) { 15549566063dSJacob Faibussowitsch PetscCall((*func)(dimEmbed, time, cgeom->centroid, Nc, val, ctx)); 155520cf1dd8SToby Isaac for (c = 0; c < Nc; ++c) { 155620cf1dd8SToby Isaac *value += val[c]*weights[q*Nc+c]; 155720cf1dd8SToby Isaac } 155820cf1dd8SToby Isaac } 15599566063dSJacob Faibussowitsch PetscCall(DMRestoreWorkArray(dm, Nc, MPIU_SCALAR, &val)); 156020cf1dd8SToby Isaac PetscFunctionReturn(0); 156120cf1dd8SToby Isaac } 156220cf1dd8SToby Isaac 156320cf1dd8SToby Isaac /*@ 156420cf1dd8SToby Isaac PetscDualSpaceGetHeightSubspace - Get the subset of the dual space basis that is supported on a mesh point of a 156520cf1dd8SToby Isaac given height. This assumes that the reference cell is symmetric over points of this height. 156620cf1dd8SToby Isaac 156720cf1dd8SToby Isaac If the dual space is not defined on mesh points of the given height (e.g. if the space is discontinuous and 156820cf1dd8SToby Isaac pointwise values are not defined on the element boundaries), or if the implementation of PetscDualSpace does not 156920cf1dd8SToby Isaac support extracting subspaces, then NULL is returned. 157020cf1dd8SToby Isaac 157120cf1dd8SToby Isaac This does not increment the reference count on the returned dual space, and the user should not destroy it. 157220cf1dd8SToby Isaac 157320cf1dd8SToby Isaac Not collective 157420cf1dd8SToby Isaac 157520cf1dd8SToby Isaac Input Parameters: 157620cf1dd8SToby Isaac + sp - the PetscDualSpace object 157720cf1dd8SToby Isaac - height - the height of the mesh point for which the subspace is desired 157820cf1dd8SToby Isaac 157920cf1dd8SToby Isaac Output Parameter: 158020cf1dd8SToby Isaac . subsp - the subspace. Note that the functionals in the subspace are with respect to the intrinsic geometry of the 158120cf1dd8SToby Isaac point, which will be of lesser dimension if height > 0. 158220cf1dd8SToby Isaac 158320cf1dd8SToby Isaac Level: advanced 158420cf1dd8SToby Isaac 1585db781477SPatrick Sanan .seealso: `PetscSpaceGetHeightSubspace()`, `PetscDualSpace` 158620cf1dd8SToby Isaac @*/ 158720cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetHeightSubspace(PetscDualSpace sp, PetscInt height, PetscDualSpace *subsp) 158820cf1dd8SToby Isaac { 1589b4457527SToby Isaac PetscInt depth = -1, cStart, cEnd; 1590b4457527SToby Isaac DM dm; 159120cf1dd8SToby Isaac 159220cf1dd8SToby Isaac PetscFunctionBegin; 159320cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1594064a246eSJacob Faibussowitsch PetscValidPointer(subsp,3); 159508401ef6SPierre Jolivet PetscCheck((sp->uniform),PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "A non-uniform dual space does not have a single dual space at each height"); 159620cf1dd8SToby Isaac *subsp = NULL; 1597b4457527SToby Isaac dm = sp->dm; 15989566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm, &depth)); 15991dca8a05SBarry Smith PetscCheck(height >= 0 && height <= depth,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid height"); 16009566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd)); 1601b4457527SToby Isaac if (height == 0 && cEnd == cStart + 1) { 1602b4457527SToby Isaac *subsp = sp; 1603b4457527SToby Isaac PetscFunctionReturn(0); 1604b4457527SToby Isaac } 1605b4457527SToby Isaac if (!sp->heightSpaces) { 1606b4457527SToby Isaac PetscInt h; 16079566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(depth+1, &(sp->heightSpaces))); 1608b4457527SToby Isaac 1609b4457527SToby Isaac for (h = 0; h <= depth; h++) { 1610b4457527SToby Isaac if (h == 0 && cEnd == cStart + 1) continue; 16119566063dSJacob Faibussowitsch if (sp->ops->createheightsubspace) PetscCall((*sp->ops->createheightsubspace)(sp,height,&(sp->heightSpaces[h]))); 1612b4457527SToby Isaac else if (sp->pointSpaces) { 1613b4457527SToby Isaac PetscInt hStart, hEnd; 1614b4457527SToby Isaac 16159566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm,h,&hStart,&hEnd)); 1616b4457527SToby Isaac if (hEnd > hStart) { 1617665f567fSMatthew G. Knepley const char *name; 1618665f567fSMatthew G. Knepley 16199566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)(sp->pointSpaces[hStart]))); 1620665f567fSMatthew G. Knepley if (sp->pointSpaces[hStart]) { 16219566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject) sp, &name)); 16229566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject) sp->pointSpaces[hStart], name)); 1623665f567fSMatthew G. Knepley } 1624b4457527SToby Isaac sp->heightSpaces[h] = sp->pointSpaces[hStart]; 1625b4457527SToby Isaac } 1626b4457527SToby Isaac } 1627b4457527SToby Isaac } 1628b4457527SToby Isaac } 1629b4457527SToby Isaac *subsp = sp->heightSpaces[height]; 163020cf1dd8SToby Isaac PetscFunctionReturn(0); 163120cf1dd8SToby Isaac } 163220cf1dd8SToby Isaac 163320cf1dd8SToby Isaac /*@ 163420cf1dd8SToby Isaac PetscDualSpaceGetPointSubspace - Get the subset of the dual space basis that is supported on a particular mesh point. 163520cf1dd8SToby Isaac 163620cf1dd8SToby Isaac If the dual space is not defined on the mesh point (e.g. if the space is discontinuous and pointwise values are not 163720cf1dd8SToby Isaac defined on the element boundaries), or if the implementation of PetscDualSpace does not support extracting 163820cf1dd8SToby Isaac subspaces, then NULL is returned. 163920cf1dd8SToby Isaac 164020cf1dd8SToby Isaac This does not increment the reference count on the returned dual space, and the user should not destroy it. 164120cf1dd8SToby Isaac 164220cf1dd8SToby Isaac Not collective 164320cf1dd8SToby Isaac 164420cf1dd8SToby Isaac Input Parameters: 164520cf1dd8SToby Isaac + sp - the PetscDualSpace object 164620cf1dd8SToby Isaac - point - the point (in the dual space's DM) for which the subspace is desired 164720cf1dd8SToby Isaac 164820cf1dd8SToby Isaac Output Parameters: 164920cf1dd8SToby Isaac bdsp - the subspace. Note that the functionals in the subspace are with respect to the intrinsic geometry of the 165020cf1dd8SToby Isaac point, which will be of lesser dimension if height > 0. 165120cf1dd8SToby Isaac 165220cf1dd8SToby Isaac Level: advanced 165320cf1dd8SToby Isaac 1654db781477SPatrick Sanan .seealso: `PetscDualSpace` 165520cf1dd8SToby Isaac @*/ 165620cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetPointSubspace(PetscDualSpace sp, PetscInt point, PetscDualSpace *bdsp) 165720cf1dd8SToby Isaac { 1658b4457527SToby Isaac PetscInt pStart = 0, pEnd = 0, cStart, cEnd; 1659b4457527SToby Isaac DM dm; 166020cf1dd8SToby Isaac 166120cf1dd8SToby Isaac PetscFunctionBegin; 166220cf1dd8SToby Isaac PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1); 1663064a246eSJacob Faibussowitsch PetscValidPointer(bdsp,3); 166420cf1dd8SToby Isaac *bdsp = NULL; 1665b4457527SToby Isaac dm = sp->dm; 16669566063dSJacob Faibussowitsch PetscCall(DMPlexGetChart(dm, &pStart, &pEnd)); 16671dca8a05SBarry Smith PetscCheck(point >= pStart && point <= pEnd,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid point"); 16689566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd)); 1669b4457527SToby Isaac if (point == cStart && cEnd == cStart + 1) { /* the dual space is only equivalent to the dual space on a cell if the reference mesh has just one cell */ 1670b4457527SToby Isaac *bdsp = sp; 1671b4457527SToby Isaac PetscFunctionReturn(0); 1672b4457527SToby Isaac } 1673b4457527SToby Isaac if (!sp->pointSpaces) { 1674b4457527SToby Isaac PetscInt p; 16759566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(pEnd - pStart, &(sp->pointSpaces))); 167620cf1dd8SToby Isaac 1677b4457527SToby Isaac for (p = 0; p < pEnd - pStart; p++) { 1678b4457527SToby Isaac if (p + pStart == cStart && cEnd == cStart + 1) continue; 16799566063dSJacob Faibussowitsch if (sp->ops->createpointsubspace) PetscCall((*sp->ops->createpointsubspace)(sp,p+pStart,&(sp->pointSpaces[p]))); 1680b4457527SToby Isaac else if (sp->heightSpaces || sp->ops->createheightsubspace) { 1681b4457527SToby Isaac PetscInt dim, depth, height; 1682b4457527SToby Isaac DMLabel label; 1683b4457527SToby Isaac 16849566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepth(dm,&dim)); 16859566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthLabel(dm,&label)); 16869566063dSJacob Faibussowitsch PetscCall(DMLabelGetValue(label,p+pStart,&depth)); 168720cf1dd8SToby Isaac height = dim - depth; 16889566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetHeightSubspace(sp, height, &(sp->pointSpaces[p]))); 16899566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)sp->pointSpaces[p])); 169020cf1dd8SToby Isaac } 1691b4457527SToby Isaac } 1692b4457527SToby Isaac } 1693b4457527SToby Isaac *bdsp = sp->pointSpaces[point - pStart]; 169420cf1dd8SToby Isaac PetscFunctionReturn(0); 169520cf1dd8SToby Isaac } 169620cf1dd8SToby Isaac 16976f905325SMatthew G. Knepley /*@C 16986f905325SMatthew G. Knepley PetscDualSpaceGetSymmetries - Returns a description of the symmetries of this basis 16996f905325SMatthew G. Knepley 17006f905325SMatthew G. Knepley Not collective 17016f905325SMatthew G. Knepley 17026f905325SMatthew G. Knepley Input Parameter: 17036f905325SMatthew G. Knepley . sp - the PetscDualSpace object 17046f905325SMatthew G. Knepley 17056f905325SMatthew G. Knepley Output Parameters: 1706b4457527SToby Isaac + perms - Permutations of the interior degrees of freedom, parameterized by the point orientation 1707b4457527SToby Isaac - flips - Sign reversal of the interior degrees of freedom, parameterized by the point orientation 17086f905325SMatthew G. Knepley 17096f905325SMatthew G. Knepley Note: The permutation and flip arrays are organized in the following way 17106f905325SMatthew G. Knepley $ perms[p][ornt][dof # on point] = new local dof # 17116f905325SMatthew G. Knepley $ flips[p][ornt][dof # on point] = reversal or not 17126f905325SMatthew G. Knepley 17136f905325SMatthew G. Knepley Level: developer 17146f905325SMatthew G. Knepley 17156f905325SMatthew G. Knepley @*/ 17166f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceGetSymmetries(PetscDualSpace sp, const PetscInt ****perms, const PetscScalar ****flips) 17176f905325SMatthew G. Knepley { 17186f905325SMatthew G. Knepley PetscFunctionBegin; 17196f905325SMatthew G. Knepley PetscValidHeaderSpecific(sp,PETSCDUALSPACE_CLASSID,1); 17206f905325SMatthew G. Knepley if (perms) {PetscValidPointer(perms,2); *perms = NULL;} 17216f905325SMatthew G. Knepley if (flips) {PetscValidPointer(flips,3); *flips = NULL;} 17229566063dSJacob Faibussowitsch if (sp->ops->getsymmetries) PetscCall((sp->ops->getsymmetries)(sp,perms,flips)); 17236f905325SMatthew G. Knepley PetscFunctionReturn(0); 17246f905325SMatthew G. Knepley } 17254bee2e38SMatthew G. Knepley 17264bee2e38SMatthew G. Knepley /*@ 1727b4457527SToby Isaac PetscDualSpaceGetFormDegree - Get the form degree k for the k-form the describes the pushforwards/pullbacks of this 1728b4457527SToby Isaac dual space's functionals. 1729b4457527SToby Isaac 1730b4457527SToby Isaac Input Parameter: 1731b4457527SToby Isaac . dsp - The PetscDualSpace 1732b4457527SToby Isaac 1733b4457527SToby Isaac Output Parameter: 1734b4457527SToby Isaac . k - The *signed* degree k of the k. If k >= 0, this means that the degrees of freedom are k-forms, and are stored 1735b4457527SToby Isaac in lexicographic order according to the basis of k-forms constructed from the wedge product of 1-forms. So for example, 1736b4457527SToby Isaac the 1-form basis in 3-D is (dx, dy, dz), and the 2-form basis in 3-D is (dx wedge dy, dx wedge dz, dy wedge dz). 1737b4457527SToby Isaac If k < 0, this means that the degrees transform as k-forms, but are stored as (N-k) forms according to the 1738b4457527SToby Isaac Hodge star map. So for example if k = -2 and N = 3, this means that the degrees of freedom transform as 2-forms 1739b4457527SToby Isaac but are stored as 1-forms. 1740b4457527SToby Isaac 1741b4457527SToby Isaac Level: developer 1742b4457527SToby Isaac 1743db781477SPatrick Sanan .seealso: `PetscDTAltV`, `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransform()`, `PetscDualSpaceTransformType` 1744b4457527SToby Isaac @*/ 1745b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetFormDegree(PetscDualSpace dsp, PetscInt *k) 1746b4457527SToby Isaac { 1747b4457527SToby Isaac PetscFunctionBeginHot; 1748b4457527SToby Isaac PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 1749dadcf809SJacob Faibussowitsch PetscValidIntPointer(k, 2); 1750b4457527SToby Isaac *k = dsp->k; 1751b4457527SToby Isaac PetscFunctionReturn(0); 1752b4457527SToby Isaac } 1753b4457527SToby Isaac 1754b4457527SToby Isaac /*@ 1755b4457527SToby Isaac PetscDualSpaceSetFormDegree - Set the form degree k for the k-form the describes the pushforwards/pullbacks of this 1756b4457527SToby Isaac dual space's functionals. 1757b4457527SToby Isaac 1758d8d19677SJose E. Roman Input Parameters: 1759b4457527SToby Isaac + dsp - The PetscDualSpace 1760b4457527SToby Isaac - k - The *signed* degree k of the k. If k >= 0, this means that the degrees of freedom are k-forms, and are stored 1761b4457527SToby Isaac in lexicographic order according to the basis of k-forms constructed from the wedge product of 1-forms. So for example, 1762b4457527SToby Isaac the 1-form basis in 3-D is (dx, dy, dz), and the 2-form basis in 3-D is (dx wedge dy, dx wedge dz, dy wedge dz). 1763b4457527SToby Isaac If k < 0, this means that the degrees transform as k-forms, but are stored as (N-k) forms according to the 1764b4457527SToby Isaac Hodge star map. So for example if k = -2 and N = 3, this means that the degrees of freedom transform as 2-forms 1765b4457527SToby Isaac but are stored as 1-forms. 1766b4457527SToby Isaac 1767b4457527SToby Isaac Level: developer 1768b4457527SToby Isaac 1769db781477SPatrick Sanan .seealso: `PetscDTAltV`, `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransform()`, `PetscDualSpaceTransformType` 1770b4457527SToby Isaac @*/ 1771b4457527SToby Isaac PetscErrorCode PetscDualSpaceSetFormDegree(PetscDualSpace dsp, PetscInt k) 1772b4457527SToby Isaac { 1773b4457527SToby Isaac PetscInt dim; 1774b4457527SToby Isaac 1775b4457527SToby Isaac PetscFunctionBeginHot; 1776b4457527SToby Isaac PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 177728b400f6SJacob Faibussowitsch PetscCheck(!dsp->setupcalled,PetscObjectComm((PetscObject)dsp), PETSC_ERR_ARG_WRONGSTATE, "Cannot change number of components after dualspace is set up"); 1778b4457527SToby Isaac dim = dsp->dm->dim; 17791dca8a05SBarry Smith PetscCheck(k >= -dim && k <= dim,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unsupported %" PetscInt_FMT "-form on %" PetscInt_FMT "-dimensional reference cell", PetscAbsInt(k), dim); 1780b4457527SToby Isaac dsp->k = k; 1781b4457527SToby Isaac PetscFunctionReturn(0); 1782b4457527SToby Isaac } 1783b4457527SToby Isaac 1784b4457527SToby Isaac /*@ 17854bee2e38SMatthew G. Knepley PetscDualSpaceGetDeRahm - Get the k-simplex associated with the functionals in this dual space 17864bee2e38SMatthew G. Knepley 17874bee2e38SMatthew G. Knepley Input Parameter: 17884bee2e38SMatthew G. Knepley . dsp - The PetscDualSpace 17894bee2e38SMatthew G. Knepley 17904bee2e38SMatthew G. Knepley Output Parameter: 17914bee2e38SMatthew G. Knepley . k - The simplex dimension 17924bee2e38SMatthew G. Knepley 1793a4ce7ad1SMatthew G. Knepley Level: developer 17944bee2e38SMatthew G. Knepley 17954bee2e38SMatthew G. Knepley Note: Currently supported values are 17964bee2e38SMatthew G. Knepley $ 0: These are H_1 methods that only transform coordinates 17974bee2e38SMatthew G. Knepley $ 1: These are Hcurl methods that transform functions using the covariant Piola transform (COVARIANT_PIOLA_TRANSFORM) 17984bee2e38SMatthew G. Knepley $ 2: These are the same as 1 17994bee2e38SMatthew G. Knepley $ 3: These are Hdiv methods that transform functions using the contravariant Piola transform (CONTRAVARIANT_PIOLA_TRANSFORM) 18004bee2e38SMatthew G. Knepley 1801db781477SPatrick Sanan .seealso: `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransform()`, `PetscDualSpaceTransformType` 18024bee2e38SMatthew G. Knepley @*/ 18034bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceGetDeRahm(PetscDualSpace dsp, PetscInt *k) 18044bee2e38SMatthew G. Knepley { 1805b4457527SToby Isaac PetscInt dim; 1806b4457527SToby Isaac 18074bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 18084bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 1809dadcf809SJacob Faibussowitsch PetscValidIntPointer(k, 2); 1810b4457527SToby Isaac dim = dsp->dm->dim; 1811b4457527SToby Isaac if (!dsp->k) *k = IDENTITY_TRANSFORM; 1812b4457527SToby Isaac else if (dsp->k == 1) *k = COVARIANT_PIOLA_TRANSFORM; 1813b4457527SToby Isaac else if (dsp->k == -(dim - 1)) *k = CONTRAVARIANT_PIOLA_TRANSFORM; 1814b4457527SToby Isaac else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unsupported transformation"); 18154bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 18164bee2e38SMatthew G. Knepley } 18174bee2e38SMatthew G. Knepley 18184bee2e38SMatthew G. Knepley /*@C 18194bee2e38SMatthew G. Knepley PetscDualSpaceTransform - Transform the function values 18204bee2e38SMatthew G. Knepley 18214bee2e38SMatthew G. Knepley Input Parameters: 18224bee2e38SMatthew G. Knepley + dsp - The PetscDualSpace 18234bee2e38SMatthew G. Knepley . trans - The type of transform 18244bee2e38SMatthew G. Knepley . isInverse - Flag to invert the transform 18254bee2e38SMatthew G. Knepley . fegeom - The cell geometry 18264bee2e38SMatthew G. Knepley . Nv - The number of function samples 18274bee2e38SMatthew G. Knepley . Nc - The number of function components 18284bee2e38SMatthew G. Knepley - vals - The function values 18294bee2e38SMatthew G. Knepley 18304bee2e38SMatthew G. Knepley Output Parameter: 18314bee2e38SMatthew G. Knepley . vals - The transformed function values 18324bee2e38SMatthew G. Knepley 1833a4ce7ad1SMatthew G. Knepley Level: intermediate 18344bee2e38SMatthew G. Knepley 1835f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 18362edcad52SToby Isaac 1837db781477SPatrick Sanan .seealso: `PetscDualSpaceTransformGradient()`, `PetscDualSpaceTransformHessian()`, `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransformType` 18384bee2e38SMatthew G. Knepley @*/ 18394bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransform(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[]) 18404bee2e38SMatthew G. Knepley { 1841b4457527SToby Isaac PetscReal Jstar[9] = {0}; 1842b4457527SToby Isaac PetscInt dim, v, c, Nk; 18434bee2e38SMatthew G. Knepley 18444bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 18454bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 18464bee2e38SMatthew G. Knepley PetscValidPointer(fegeom, 4); 1847dadcf809SJacob Faibussowitsch PetscValidScalarPointer(vals, 7); 1848b4457527SToby Isaac /* TODO: not handling dimEmbed != dim right now */ 18492ae266adSMatthew G. Knepley dim = dsp->dm->dim; 1850b4457527SToby Isaac /* No change needed for 0-forms */ 1851b4457527SToby Isaac if (!dsp->k) PetscFunctionReturn(0); 18529566063dSJacob Faibussowitsch PetscCall(PetscDTBinomialInt(dim, PetscAbsInt(dsp->k), &Nk)); 1853b4457527SToby Isaac /* TODO: use fegeom->isAffine */ 18549566063dSJacob Faibussowitsch PetscCall(PetscDTAltVPullbackMatrix(dim, dim, isInverse ? fegeom->J : fegeom->invJ, dsp->k, Jstar)); 18554bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 1856b4457527SToby Isaac switch (Nk) { 1857b4457527SToby Isaac case 1: 1858b4457527SToby Isaac for (c = 0; c < Nc; c++) vals[v*Nc + c] *= Jstar[0]; 18594bee2e38SMatthew G. Knepley break; 1860b4457527SToby Isaac case 2: 1861b4457527SToby Isaac for (c = 0; c < Nc; c += 2) DMPlex_Mult2DReal_Internal(Jstar, 1, &vals[v*Nc + c], &vals[v*Nc + c]); 18624bee2e38SMatthew G. Knepley break; 1863b4457527SToby Isaac case 3: 1864b4457527SToby Isaac for (c = 0; c < Nc; c += 3) DMPlex_Mult3DReal_Internal(Jstar, 1, &vals[v*Nc + c], &vals[v*Nc + c]); 1865b4457527SToby Isaac break; 186663a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported form size %" PetscInt_FMT " for transformation", Nk); 1867b4457527SToby Isaac } 18684bee2e38SMatthew G. Knepley } 18694bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 18704bee2e38SMatthew G. Knepley } 1871b4457527SToby Isaac 18724bee2e38SMatthew G. Knepley /*@C 18734bee2e38SMatthew G. Knepley PetscDualSpaceTransformGradient - Transform the function gradient values 18744bee2e38SMatthew G. Knepley 18754bee2e38SMatthew G. Knepley Input Parameters: 18764bee2e38SMatthew G. Knepley + dsp - The PetscDualSpace 18774bee2e38SMatthew G. Knepley . trans - The type of transform 18784bee2e38SMatthew G. Knepley . isInverse - Flag to invert the transform 18794bee2e38SMatthew G. Knepley . fegeom - The cell geometry 18804bee2e38SMatthew G. Knepley . Nv - The number of function gradient samples 18814bee2e38SMatthew G. Knepley . Nc - The number of function components 18824bee2e38SMatthew G. Knepley - vals - The function gradient values 18834bee2e38SMatthew G. Knepley 18844bee2e38SMatthew G. Knepley Output Parameter: 1885f9244615SMatthew G. Knepley . vals - The transformed function gradient values 18864bee2e38SMatthew G. Knepley 1887a4ce7ad1SMatthew G. Knepley Level: intermediate 18884bee2e38SMatthew G. Knepley 1889f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 18902edcad52SToby Isaac 1891db781477SPatrick Sanan .seealso: `PetscDualSpaceTransform()`, `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransformType` 18924bee2e38SMatthew G. Knepley @*/ 18934bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransformGradient(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[]) 18944bee2e38SMatthew G. Knepley { 189527f02ce8SMatthew G. Knepley const PetscInt dim = dsp->dm->dim, dE = fegeom->dimEmbed; 189627f02ce8SMatthew G. Knepley PetscInt v, c, d; 18974bee2e38SMatthew G. Knepley 18984bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 18994bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 19004bee2e38SMatthew G. Knepley PetscValidPointer(fegeom, 4); 1901dadcf809SJacob Faibussowitsch PetscValidScalarPointer(vals, 7); 190227f02ce8SMatthew G. Knepley #ifdef PETSC_USE_DEBUG 190363a3b9bcSJacob Faibussowitsch PetscCheck(dE > 0,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid embedding dimension %" PetscInt_FMT, dE); 190427f02ce8SMatthew G. Knepley #endif 19054bee2e38SMatthew G. Knepley /* Transform gradient */ 190627f02ce8SMatthew G. Knepley if (dim == dE) { 19074bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 19084bee2e38SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 19094bee2e38SMatthew G. Knepley switch (dim) 19104bee2e38SMatthew G. Knepley { 1911100a78e1SStefano Zampini case 1: vals[(v*Nc+c)*dim] *= fegeom->invJ[0];break; 19126142fa51SMatthew G. Knepley case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->invJ, 1, &vals[(v*Nc+c)*dim], &vals[(v*Nc+c)*dim]);break; 19136142fa51SMatthew G. Knepley case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->invJ, 1, &vals[(v*Nc+c)*dim], &vals[(v*Nc+c)*dim]);break; 191463a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 19154bee2e38SMatthew G. Knepley } 19164bee2e38SMatthew G. Knepley } 19174bee2e38SMatthew G. Knepley } 191827f02ce8SMatthew G. Knepley } else { 191927f02ce8SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 192027f02ce8SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 192127f02ce8SMatthew G. Knepley DMPlex_MultTransposeReal_Internal(fegeom->invJ, dim, dE, 1, &vals[(v*Nc+c)*dE], &vals[(v*Nc+c)*dE]); 192227f02ce8SMatthew G. Knepley } 192327f02ce8SMatthew G. Knepley } 192427f02ce8SMatthew G. Knepley } 19254bee2e38SMatthew G. Knepley /* Assume its a vector, otherwise assume its a bunch of scalars */ 19264bee2e38SMatthew G. Knepley if (Nc == 1 || Nc != dim) PetscFunctionReturn(0); 19274bee2e38SMatthew G. Knepley switch (trans) { 19284bee2e38SMatthew G. Knepley case IDENTITY_TRANSFORM: break; 19294bee2e38SMatthew G. Knepley case COVARIANT_PIOLA_TRANSFORM: /* Covariant Piola mapping $\sigma^*(F) = J^{-T} F \circ \phi^{-1)$ */ 19304bee2e38SMatthew G. Knepley if (isInverse) { 19314bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 19324bee2e38SMatthew G. Knepley for (d = 0; d < dim; ++d) { 19334bee2e38SMatthew G. Knepley switch (dim) 19344bee2e38SMatthew G. Knepley { 19356142fa51SMatthew G. Knepley case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 19366142fa51SMatthew G. Knepley case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 193763a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 19384bee2e38SMatthew G. Knepley } 19394bee2e38SMatthew G. Knepley } 19404bee2e38SMatthew G. Knepley } 19414bee2e38SMatthew G. Knepley } else { 19424bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 19434bee2e38SMatthew G. Knepley for (d = 0; d < dim; ++d) { 19444bee2e38SMatthew G. Knepley switch (dim) 19454bee2e38SMatthew G. Knepley { 19466142fa51SMatthew G. Knepley case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 19476142fa51SMatthew G. Knepley case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 194863a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 19494bee2e38SMatthew G. Knepley } 19504bee2e38SMatthew G. Knepley } 19514bee2e38SMatthew G. Knepley } 19524bee2e38SMatthew G. Knepley } 19534bee2e38SMatthew G. Knepley break; 19544bee2e38SMatthew G. Knepley case CONTRAVARIANT_PIOLA_TRANSFORM: /* Contravariant Piola mapping $\sigma^*(F) = \frac{1}{|\det J|} J F \circ \phi^{-1}$ */ 19554bee2e38SMatthew G. Knepley if (isInverse) { 19564bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 19574bee2e38SMatthew G. Knepley for (d = 0; d < dim; ++d) { 19584bee2e38SMatthew G. Knepley switch (dim) 19594bee2e38SMatthew G. Knepley { 19606142fa51SMatthew G. Knepley case 2: DMPlex_Mult2DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 19616142fa51SMatthew G. Knepley case 3: DMPlex_Mult3DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 196263a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 19634bee2e38SMatthew G. Knepley } 19644bee2e38SMatthew G. Knepley for (c = 0; c < Nc; ++c) vals[(v*Nc+c)*dim+d] *= fegeom->detJ[0]; 19654bee2e38SMatthew G. Knepley } 19664bee2e38SMatthew G. Knepley } 19674bee2e38SMatthew G. Knepley } else { 19684bee2e38SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 19694bee2e38SMatthew G. Knepley for (d = 0; d < dim; ++d) { 19704bee2e38SMatthew G. Knepley switch (dim) 19714bee2e38SMatthew G. Knepley { 19726142fa51SMatthew G. Knepley case 2: DMPlex_Mult2DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 19736142fa51SMatthew G. Knepley case 3: DMPlex_Mult3DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break; 197463a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 19754bee2e38SMatthew G. Knepley } 19764bee2e38SMatthew G. Knepley for (c = 0; c < Nc; ++c) vals[(v*Nc+c)*dim+d] /= fegeom->detJ[0]; 19774bee2e38SMatthew G. Knepley } 19784bee2e38SMatthew G. Knepley } 19794bee2e38SMatthew G. Knepley } 19804bee2e38SMatthew G. Knepley break; 19814bee2e38SMatthew G. Knepley } 19824bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 19834bee2e38SMatthew G. Knepley } 19844bee2e38SMatthew G. Knepley 19854bee2e38SMatthew G. Knepley /*@C 1986f9244615SMatthew G. Knepley PetscDualSpaceTransformHessian - Transform the function Hessian values 1987f9244615SMatthew G. Knepley 1988f9244615SMatthew G. Knepley Input Parameters: 1989f9244615SMatthew G. Knepley + dsp - The PetscDualSpace 1990f9244615SMatthew G. Knepley . trans - The type of transform 1991f9244615SMatthew G. Knepley . isInverse - Flag to invert the transform 1992f9244615SMatthew G. Knepley . fegeom - The cell geometry 1993f9244615SMatthew G. Knepley . Nv - The number of function Hessian samples 1994f9244615SMatthew G. Knepley . Nc - The number of function components 1995f9244615SMatthew G. Knepley - vals - The function gradient values 1996f9244615SMatthew G. Knepley 1997f9244615SMatthew G. Knepley Output Parameter: 1998f9244615SMatthew G. Knepley . vals - The transformed function Hessian values 1999f9244615SMatthew G. Knepley 2000f9244615SMatthew G. Knepley Level: intermediate 2001f9244615SMatthew G. Knepley 2002f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 2003f9244615SMatthew G. Knepley 2004db781477SPatrick Sanan .seealso: `PetscDualSpaceTransform()`, `PetscDualSpacePullback()`, `PetscDualSpacePushforward()`, `PetscDualSpaceTransformType` 2005f9244615SMatthew G. Knepley @*/ 2006f9244615SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransformHessian(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[]) 2007f9244615SMatthew G. Knepley { 2008f9244615SMatthew G. Knepley const PetscInt dim = dsp->dm->dim, dE = fegeom->dimEmbed; 2009f9244615SMatthew G. Knepley PetscInt v, c; 2010f9244615SMatthew G. Knepley 2011f9244615SMatthew G. Knepley PetscFunctionBeginHot; 2012f9244615SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 2013f9244615SMatthew G. Knepley PetscValidPointer(fegeom, 4); 2014dadcf809SJacob Faibussowitsch PetscValidScalarPointer(vals, 7); 2015f9244615SMatthew G. Knepley #ifdef PETSC_USE_DEBUG 201663a3b9bcSJacob Faibussowitsch PetscCheck(dE > 0,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid embedding dimension %" PetscInt_FMT, dE); 2017f9244615SMatthew G. Knepley #endif 2018f9244615SMatthew G. Knepley /* Transform Hessian: J^{-T}_{ik} J^{-T}_{jl} H(f)_{kl} = J^{-T}_{ik} H(f)_{kl} J^{-1}_{lj} */ 2019f9244615SMatthew G. Knepley if (dim == dE) { 2020f9244615SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 2021f9244615SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 2022f9244615SMatthew G. Knepley switch (dim) 2023f9244615SMatthew G. Knepley { 2024f9244615SMatthew G. Knepley case 1: vals[(v*Nc+c)*dim*dim] *= PetscSqr(fegeom->invJ[0]);break; 2025f9244615SMatthew G. Knepley case 2: DMPlex_PTAP2DReal_Internal(fegeom->invJ, &vals[(v*Nc+c)*dim*dim], &vals[(v*Nc+c)*dim*dim]);break; 2026f9244615SMatthew G. Knepley case 3: DMPlex_PTAP3DReal_Internal(fegeom->invJ, &vals[(v*Nc+c)*dim*dim], &vals[(v*Nc+c)*dim*dim]);break; 202763a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim); 2028f9244615SMatthew G. Knepley } 2029f9244615SMatthew G. Knepley } 2030f9244615SMatthew G. Knepley } 2031f9244615SMatthew G. Knepley } else { 2032f9244615SMatthew G. Knepley for (v = 0; v < Nv; ++v) { 2033f9244615SMatthew G. Knepley for (c = 0; c < Nc; ++c) { 2034f9244615SMatthew G. Knepley DMPlex_PTAPReal_Internal(fegeom->invJ, dim, dE, &vals[(v*Nc+c)*dE*dE], &vals[(v*Nc+c)*dE*dE]); 2035f9244615SMatthew G. Knepley } 2036f9244615SMatthew G. Knepley } 2037f9244615SMatthew G. Knepley } 2038f9244615SMatthew G. Knepley /* Assume its a vector, otherwise assume its a bunch of scalars */ 2039f9244615SMatthew G. Knepley if (Nc == 1 || Nc != dim) PetscFunctionReturn(0); 2040f9244615SMatthew G. Knepley switch (trans) { 2041f9244615SMatthew G. Knepley case IDENTITY_TRANSFORM: break; 2042f9244615SMatthew G. Knepley case COVARIANT_PIOLA_TRANSFORM: /* Covariant Piola mapping $\sigma^*(F) = J^{-T} F \circ \phi^{-1)$ */ 2043f9244615SMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Piola mapping for Hessians not yet supported"); 2044f9244615SMatthew G. Knepley case CONTRAVARIANT_PIOLA_TRANSFORM: /* Contravariant Piola mapping $\sigma^*(F) = \frac{1}{|\det J|} J F \circ \phi^{-1}$ */ 2045f9244615SMatthew G. Knepley SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Piola mapping for Hessians not yet supported"); 2046f9244615SMatthew G. Knepley } 2047f9244615SMatthew G. Knepley PetscFunctionReturn(0); 2048f9244615SMatthew G. Knepley } 2049f9244615SMatthew G. Knepley 2050f9244615SMatthew G. Knepley /*@C 20514bee2e38SMatthew G. Knepley PetscDualSpacePullback - Transform the given functional so that it operates on real space, rather than the reference element. Operationally, this means that we map the function evaluations depending on continuity requirements of our finite element method. 20524bee2e38SMatthew G. Knepley 20534bee2e38SMatthew G. Knepley Input Parameters: 20544bee2e38SMatthew G. Knepley + dsp - The PetscDualSpace 20554bee2e38SMatthew G. Knepley . fegeom - The geometry for this cell 20564bee2e38SMatthew G. Knepley . Nq - The number of function samples 20574bee2e38SMatthew G. Knepley . Nc - The number of function components 20584bee2e38SMatthew G. Knepley - pointEval - The function values 20594bee2e38SMatthew G. Knepley 20604bee2e38SMatthew G. Knepley Output Parameter: 20614bee2e38SMatthew G. Knepley . pointEval - The transformed function values 20624bee2e38SMatthew G. Knepley 20634bee2e38SMatthew G. Knepley Level: advanced 20644bee2e38SMatthew G. Knepley 20654bee2e38SMatthew G. Knepley Note: Functions transform in a complementary way (pushforward) to functionals, so that the scalar product is invariant. The type of transform is dependent on the associated k-simplex from the DeRahm complex. 20664bee2e38SMatthew G. Knepley 20672edcad52SToby Isaac Note: This only handles tranformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 20682edcad52SToby Isaac 2069db781477SPatrick Sanan .seealso: `PetscDualSpacePushforward()`, `PetscDualSpaceTransform()`, `PetscDualSpaceGetDeRahm()` 20704bee2e38SMatthew G. Knepley @*/ 20712edcad52SToby Isaac PetscErrorCode PetscDualSpacePullback(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[]) 20724bee2e38SMatthew G. Knepley { 20734bee2e38SMatthew G. Knepley PetscDualSpaceTransformType trans; 2074b4457527SToby Isaac PetscInt k; 20754bee2e38SMatthew G. Knepley 20764bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 20774bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 20784bee2e38SMatthew G. Knepley PetscValidPointer(fegeom, 2); 2079dadcf809SJacob Faibussowitsch PetscValidScalarPointer(pointEval, 5); 20804bee2e38SMatthew G. Knepley /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k. 20814bee2e38SMatthew G. Knepley This determines their transformation properties. */ 20829566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDeRahm(dsp, &k)); 2083b4457527SToby Isaac switch (k) 20844bee2e38SMatthew G. Knepley { 20854bee2e38SMatthew G. Knepley case 0: /* H^1 point evaluations */ 20864bee2e38SMatthew G. Knepley trans = IDENTITY_TRANSFORM;break; 20874bee2e38SMatthew G. Knepley case 1: /* Hcurl preserves tangential edge traces */ 20884bee2e38SMatthew G. Knepley trans = COVARIANT_PIOLA_TRANSFORM;break; 2089b4457527SToby Isaac case 2: 20904bee2e38SMatthew G. Knepley case 3: /* Hdiv preserve normal traces */ 20914bee2e38SMatthew G. Knepley trans = CONTRAVARIANT_PIOLA_TRANSFORM;break; 209263a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k); 20934bee2e38SMatthew G. Knepley } 20949566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceTransform(dsp, trans, PETSC_TRUE, fegeom, Nq, Nc, pointEval)); 20954bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 20964bee2e38SMatthew G. Knepley } 20974bee2e38SMatthew G. Knepley 20984bee2e38SMatthew G. Knepley /*@C 20994bee2e38SMatthew G. Knepley PetscDualSpacePushforward - Transform the given function so that it operates on real space, rather than the reference element. Operationally, this means that we map the function evaluations depending on continuity requirements of our finite element method. 21004bee2e38SMatthew G. Knepley 21014bee2e38SMatthew G. Knepley Input Parameters: 21024bee2e38SMatthew G. Knepley + dsp - The PetscDualSpace 21034bee2e38SMatthew G. Knepley . fegeom - The geometry for this cell 21044bee2e38SMatthew G. Knepley . Nq - The number of function samples 21054bee2e38SMatthew G. Knepley . Nc - The number of function components 21064bee2e38SMatthew G. Knepley - pointEval - The function values 21074bee2e38SMatthew G. Knepley 21084bee2e38SMatthew G. Knepley Output Parameter: 21094bee2e38SMatthew G. Knepley . pointEval - The transformed function values 21104bee2e38SMatthew G. Knepley 21114bee2e38SMatthew G. Knepley Level: advanced 21124bee2e38SMatthew G. Knepley 21134bee2e38SMatthew G. Knepley Note: Functionals transform in a complementary way (pullback) to functions, so that the scalar product is invariant. The type of transform is dependent on the associated k-simplex from the DeRahm complex. 21144bee2e38SMatthew G. Knepley 2115f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 21162edcad52SToby Isaac 2117db781477SPatrick Sanan .seealso: `PetscDualSpacePullback()`, `PetscDualSpaceTransform()`, `PetscDualSpaceGetDeRahm()` 21184bee2e38SMatthew G. Knepley @*/ 21192edcad52SToby Isaac PetscErrorCode PetscDualSpacePushforward(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[]) 21204bee2e38SMatthew G. Knepley { 21214bee2e38SMatthew G. Knepley PetscDualSpaceTransformType trans; 2122b4457527SToby Isaac PetscInt k; 21234bee2e38SMatthew G. Knepley 21244bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 21254bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 21264bee2e38SMatthew G. Knepley PetscValidPointer(fegeom, 2); 2127dadcf809SJacob Faibussowitsch PetscValidScalarPointer(pointEval, 5); 21284bee2e38SMatthew G. Knepley /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k. 21294bee2e38SMatthew G. Knepley This determines their transformation properties. */ 21309566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDeRahm(dsp, &k)); 2131b4457527SToby Isaac switch (k) 21324bee2e38SMatthew G. Knepley { 21334bee2e38SMatthew G. Knepley case 0: /* H^1 point evaluations */ 21344bee2e38SMatthew G. Knepley trans = IDENTITY_TRANSFORM;break; 21354bee2e38SMatthew G. Knepley case 1: /* Hcurl preserves tangential edge traces */ 21364bee2e38SMatthew G. Knepley trans = COVARIANT_PIOLA_TRANSFORM;break; 2137b4457527SToby Isaac case 2: 21384bee2e38SMatthew G. Knepley case 3: /* Hdiv preserve normal traces */ 21394bee2e38SMatthew G. Knepley trans = CONTRAVARIANT_PIOLA_TRANSFORM;break; 214063a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k); 21414bee2e38SMatthew G. Knepley } 21429566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceTransform(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval)); 21434bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 21444bee2e38SMatthew G. Knepley } 21454bee2e38SMatthew G. Knepley 21464bee2e38SMatthew G. Knepley /*@C 21474bee2e38SMatthew G. Knepley PetscDualSpacePushforwardGradient - Transform the given function gradient so that it operates on real space, rather than the reference element. Operationally, this means that we map the function evaluations depending on continuity requirements of our finite element method. 21484bee2e38SMatthew G. Knepley 21494bee2e38SMatthew G. Knepley Input Parameters: 21504bee2e38SMatthew G. Knepley + dsp - The PetscDualSpace 21514bee2e38SMatthew G. Knepley . fegeom - The geometry for this cell 21524bee2e38SMatthew G. Knepley . Nq - The number of function gradient samples 21534bee2e38SMatthew G. Knepley . Nc - The number of function components 21544bee2e38SMatthew G. Knepley - pointEval - The function gradient values 21554bee2e38SMatthew G. Knepley 21564bee2e38SMatthew G. Knepley Output Parameter: 21574bee2e38SMatthew G. Knepley . pointEval - The transformed function gradient values 21584bee2e38SMatthew G. Knepley 21594bee2e38SMatthew G. Knepley Level: advanced 21604bee2e38SMatthew G. Knepley 21614bee2e38SMatthew G. Knepley Note: Functionals transform in a complementary way (pullback) to functions, so that the scalar product is invariant. The type of transform is dependent on the associated k-simplex from the DeRahm complex. 21624bee2e38SMatthew G. Knepley 2163f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 21642edcad52SToby Isaac 2165db781477SPatrick Sanan .seealso: `PetscDualSpacePushforward()`, `PPetscDualSpacePullback()`, `PetscDualSpaceTransform()`, `PetscDualSpaceGetDeRahm()` 2166dc0529c6SBarry Smith @*/ 21672edcad52SToby Isaac PetscErrorCode PetscDualSpacePushforwardGradient(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[]) 21684bee2e38SMatthew G. Knepley { 21694bee2e38SMatthew G. Knepley PetscDualSpaceTransformType trans; 2170b4457527SToby Isaac PetscInt k; 21714bee2e38SMatthew G. Knepley 21724bee2e38SMatthew G. Knepley PetscFunctionBeginHot; 21734bee2e38SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 21744bee2e38SMatthew G. Knepley PetscValidPointer(fegeom, 2); 2175dadcf809SJacob Faibussowitsch PetscValidScalarPointer(pointEval, 5); 21764bee2e38SMatthew G. Knepley /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k. 21774bee2e38SMatthew G. Knepley This determines their transformation properties. */ 21789566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDeRahm(dsp, &k)); 2179b4457527SToby Isaac switch (k) 21804bee2e38SMatthew G. Knepley { 21814bee2e38SMatthew G. Knepley case 0: /* H^1 point evaluations */ 21824bee2e38SMatthew G. Knepley trans = IDENTITY_TRANSFORM;break; 21834bee2e38SMatthew G. Knepley case 1: /* Hcurl preserves tangential edge traces */ 21844bee2e38SMatthew G. Knepley trans = COVARIANT_PIOLA_TRANSFORM;break; 2185b4457527SToby Isaac case 2: 21864bee2e38SMatthew G. Knepley case 3: /* Hdiv preserve normal traces */ 21874bee2e38SMatthew G. Knepley trans = CONTRAVARIANT_PIOLA_TRANSFORM;break; 218863a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k); 21894bee2e38SMatthew G. Knepley } 21909566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceTransformGradient(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval)); 21914bee2e38SMatthew G. Knepley PetscFunctionReturn(0); 21924bee2e38SMatthew G. Knepley } 2193f9244615SMatthew G. Knepley 2194f9244615SMatthew G. Knepley /*@C 2195f9244615SMatthew G. Knepley PetscDualSpacePushforwardHessian - Transform the given function Hessian so that it operates on real space, rather than the reference element. Operationally, this means that we map the function evaluations depending on continuity requirements of our finite element method. 2196f9244615SMatthew G. Knepley 2197f9244615SMatthew G. Knepley Input Parameters: 2198f9244615SMatthew G. Knepley + dsp - The PetscDualSpace 2199f9244615SMatthew G. Knepley . fegeom - The geometry for this cell 2200f9244615SMatthew G. Knepley . Nq - The number of function Hessian samples 2201f9244615SMatthew G. Knepley . Nc - The number of function components 2202f9244615SMatthew G. Knepley - pointEval - The function gradient values 2203f9244615SMatthew G. Knepley 2204f9244615SMatthew G. Knepley Output Parameter: 2205f9244615SMatthew G. Knepley . pointEval - The transformed function Hessian values 2206f9244615SMatthew G. Knepley 2207f9244615SMatthew G. Knepley Level: advanced 2208f9244615SMatthew G. Knepley 2209f9244615SMatthew G. Knepley Note: Functionals transform in a complementary way (pullback) to functions, so that the scalar product is invariant. The type of transform is dependent on the associated k-simplex from the DeRahm complex. 2210f9244615SMatthew G. Knepley 2211f9244615SMatthew G. Knepley Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension. 2212f9244615SMatthew G. Knepley 2213db781477SPatrick Sanan .seealso: `PetscDualSpacePushforward()`, `PPetscDualSpacePullback()`, `PetscDualSpaceTransform()`, `PetscDualSpaceGetDeRahm()` 2214f9244615SMatthew G. Knepley @*/ 2215f9244615SMatthew G. Knepley PetscErrorCode PetscDualSpacePushforwardHessian(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[]) 2216f9244615SMatthew G. Knepley { 2217f9244615SMatthew G. Knepley PetscDualSpaceTransformType trans; 2218f9244615SMatthew G. Knepley PetscInt k; 2219f9244615SMatthew G. Knepley 2220f9244615SMatthew G. Knepley PetscFunctionBeginHot; 2221f9244615SMatthew G. Knepley PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1); 2222f9244615SMatthew G. Knepley PetscValidPointer(fegeom, 2); 2223dadcf809SJacob Faibussowitsch PetscValidScalarPointer(pointEval, 5); 2224f9244615SMatthew G. Knepley /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k. 2225f9244615SMatthew G. Knepley This determines their transformation properties. */ 22269566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceGetDeRahm(dsp, &k)); 2227f9244615SMatthew G. Knepley switch (k) 2228f9244615SMatthew G. Knepley { 2229f9244615SMatthew G. Knepley case 0: /* H^1 point evaluations */ 2230f9244615SMatthew G. Knepley trans = IDENTITY_TRANSFORM;break; 2231f9244615SMatthew G. Knepley case 1: /* Hcurl preserves tangential edge traces */ 2232f9244615SMatthew G. Knepley trans = COVARIANT_PIOLA_TRANSFORM;break; 2233f9244615SMatthew G. Knepley case 2: 2234f9244615SMatthew G. Knepley case 3: /* Hdiv preserve normal traces */ 2235f9244615SMatthew G. Knepley trans = CONTRAVARIANT_PIOLA_TRANSFORM;break; 223663a3b9bcSJacob Faibussowitsch default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k); 2237f9244615SMatthew G. Knepley } 22389566063dSJacob Faibussowitsch PetscCall(PetscDualSpaceTransformHessian(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval)); 2239f9244615SMatthew G. Knepley PetscFunctionReturn(0); 2240f9244615SMatthew G. Knepley } 2241