xref: /petsc/src/dm/dt/dualspace/interface/dualspace.c (revision 1dca8a0504492127e77eac64bc165d7372dd6d63)
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 
266f905325SMatthew G. Knepley .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 
576f905325SMatthew G. Knepley .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 
10420cf1dd8SToby Isaac .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 
12820cf1dd8SToby Isaac .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 
16620cf1dd8SToby Isaac .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
221fe2efc57SMark .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 
242fe2efc57SMark .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 
273fe2efc57SMark .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 
330fe2efc57SMark .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 
395fe2efc57SMark .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 
43720cf1dd8SToby Isaac .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 
47620cf1dd8SToby Isaac .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 
51620cf1dd8SToby Isaac .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 
53820cf1dd8SToby Isaac .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 
56820cf1dd8SToby Isaac .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 
59020cf1dd8SToby Isaac .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 
61420cf1dd8SToby Isaac .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 
63420cf1dd8SToby Isaac .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 
65920cf1dd8SToby Isaac .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 
68720cf1dd8SToby Isaac .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, &section));
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 
719b4457527SToby Isaac .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, &section));
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 
755b4457527SToby Isaac .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 
77920cf1dd8SToby Isaac .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, &section));
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, &section));
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   }
843*1dca8a05SBarry 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 
903a4ce7ad1SMatthew G. Knepley .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;
910b4457527SToby Isaac   if (!sp->pointSection) {
911b4457527SToby Isaac     /* mark the boundary */
9129566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceSectionCreate_Internal(sp, &(sp->pointSection)));
9139566063dSJacob Faibussowitsch     PetscCall(DMPlexGetChart(sp->dm,&pStart,&pEnd));
914b4457527SToby Isaac     for (p = pStart; p < pEnd; p++) {
915b4457527SToby Isaac       PetscDualSpace psp;
916b4457527SToby Isaac 
9179566063dSJacob Faibussowitsch       PetscCall(PetscDualSpaceGetPointSubspace(sp, p, &psp));
918b4457527SToby Isaac       if (psp) {
919b4457527SToby Isaac         PetscInt dof;
920b4457527SToby Isaac 
9219566063dSJacob Faibussowitsch         PetscCall(PetscDualSpaceGetInteriorDimension(psp, &dof));
9229566063dSJacob Faibussowitsch         PetscCall(PetscSectionSetDof(sp->pointSection,p,dof));
923b4457527SToby Isaac       }
924b4457527SToby Isaac     }
9259566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceSectionSetUp_Internal(sp,sp->pointSection));
926b4457527SToby Isaac   }
927b4457527SToby Isaac   *section = sp->pointSection;
928b4457527SToby Isaac   PetscFunctionReturn(0);
929b4457527SToby Isaac }
930b4457527SToby Isaac 
931b4457527SToby Isaac /* this assumes that all of the point dual spaces store their interior dofs first, which is true when the point DMs
932b4457527SToby Isaac  * have one cell */
933b4457527SToby Isaac PetscErrorCode PetscDualSpacePushForwardSubspaces_Internal(PetscDualSpace sp, PetscInt sStart, PetscInt sEnd)
934b4457527SToby Isaac {
935b4457527SToby Isaac   PetscReal *sv0, *v0, *J;
936b4457527SToby Isaac   PetscSection section;
937b4457527SToby Isaac   PetscInt     dim, s, k;
93820cf1dd8SToby Isaac   DM             dm;
93920cf1dd8SToby Isaac 
94020cf1dd8SToby Isaac   PetscFunctionBegin;
9419566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDM(sp, &dm));
9429566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
9439566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetSection(sp, &section));
9449566063dSJacob Faibussowitsch   PetscCall(PetscMalloc3(dim, &v0, dim, &sv0, dim*dim, &J));
9459566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetFormDegree(sp, &k));
946b4457527SToby Isaac   for (s = sStart; s < sEnd; s++) {
947b4457527SToby Isaac     PetscReal detJ, hdetJ;
948b4457527SToby Isaac     PetscDualSpace ssp;
949b4457527SToby Isaac     PetscInt dof, off, f, sdim;
950b4457527SToby Isaac     PetscInt i, j;
951b4457527SToby Isaac     DM sdm;
95220cf1dd8SToby Isaac 
9539566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceGetPointSubspace(sp, s, &ssp));
954b4457527SToby Isaac     if (!ssp) continue;
9559566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetDof(section, s, &dof));
9569566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(section, s, &off));
957b4457527SToby Isaac     /* get the first vertex of the reference cell */
9589566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceGetDM(ssp, &sdm));
9599566063dSJacob Faibussowitsch     PetscCall(DMGetDimension(sdm, &sdim));
9609566063dSJacob Faibussowitsch     PetscCall(DMPlexComputeCellGeometryAffineFEM(sdm, 0, sv0, NULL, NULL, &hdetJ));
9619566063dSJacob Faibussowitsch     PetscCall(DMPlexComputeCellGeometryAffineFEM(dm, s, v0, J, NULL, &detJ));
962b4457527SToby Isaac     /* compactify Jacobian */
963b4457527SToby Isaac     for (i = 0; i < dim; i++) for (j = 0; j < sdim; j++) J[i* sdim + j] = J[i * dim + j];
964b4457527SToby Isaac     for (f = 0; f < dof; f++) {
965b4457527SToby Isaac       PetscQuadrature fn;
96620cf1dd8SToby Isaac 
9679566063dSJacob Faibussowitsch       PetscCall(PetscDualSpaceGetFunctional(ssp, f, &fn));
9689566063dSJacob Faibussowitsch       PetscCall(PetscQuadraturePushForward(fn, dim, sv0, v0, J, k, &(sp->functional[off+f])));
96920cf1dd8SToby Isaac     }
97020cf1dd8SToby Isaac   }
9719566063dSJacob Faibussowitsch   PetscCall(PetscFree3(v0, sv0, J));
97220cf1dd8SToby Isaac   PetscFunctionReturn(0);
97320cf1dd8SToby Isaac }
97420cf1dd8SToby Isaac 
97520cf1dd8SToby Isaac /*@C
97620cf1dd8SToby Isaac   PetscDualSpaceApply - Apply a functional from the dual space basis to an input function
97720cf1dd8SToby Isaac 
97820cf1dd8SToby Isaac   Input Parameters:
97920cf1dd8SToby Isaac + sp      - The PetscDualSpace object
98020cf1dd8SToby Isaac . f       - The basis functional index
98120cf1dd8SToby Isaac . time    - The time
98220cf1dd8SToby 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)
98320cf1dd8SToby Isaac . numComp - The number of components for the function
98420cf1dd8SToby Isaac . func    - The input function
98520cf1dd8SToby Isaac - ctx     - A context for the function
98620cf1dd8SToby Isaac 
98720cf1dd8SToby Isaac   Output Parameter:
98820cf1dd8SToby Isaac . value   - numComp output values
98920cf1dd8SToby Isaac 
99020cf1dd8SToby Isaac   Note: The calling sequence for the callback func is given by:
99120cf1dd8SToby Isaac 
99220cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[],
99320cf1dd8SToby Isaac $      PetscInt numComponents, PetscScalar values[], void *ctx)
99420cf1dd8SToby Isaac 
995a4ce7ad1SMatthew G. Knepley   Level: beginner
99620cf1dd8SToby Isaac 
99720cf1dd8SToby Isaac .seealso: PetscDualSpaceCreate()
99820cf1dd8SToby Isaac @*/
99920cf1dd8SToby 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)
100020cf1dd8SToby Isaac {
100120cf1dd8SToby Isaac   PetscFunctionBegin;
100220cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
100320cf1dd8SToby Isaac   PetscValidPointer(cgeom, 4);
1004dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(value, 8);
10059566063dSJacob Faibussowitsch   PetscCall((*sp->ops->apply)(sp, f, time, cgeom, numComp, func, ctx, value));
100620cf1dd8SToby Isaac   PetscFunctionReturn(0);
100720cf1dd8SToby Isaac }
100820cf1dd8SToby Isaac 
100920cf1dd8SToby Isaac /*@C
1010b4457527SToby Isaac   PetscDualSpaceApplyAll - Apply all functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetAllData()
101120cf1dd8SToby Isaac 
101220cf1dd8SToby Isaac   Input Parameters:
101320cf1dd8SToby Isaac + sp        - The PetscDualSpace object
1014b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetAllData()
101520cf1dd8SToby Isaac 
101620cf1dd8SToby Isaac   Output Parameter:
101720cf1dd8SToby Isaac . spValue   - The values of all dual space functionals
101820cf1dd8SToby Isaac 
1019a4ce7ad1SMatthew G. Knepley   Level: beginner
102020cf1dd8SToby Isaac 
102120cf1dd8SToby Isaac .seealso: PetscDualSpaceCreate()
102220cf1dd8SToby Isaac @*/
102320cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyAll(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue)
102420cf1dd8SToby Isaac {
102520cf1dd8SToby Isaac   PetscFunctionBegin;
102620cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
10279566063dSJacob Faibussowitsch   PetscCall((*sp->ops->applyall)(sp, pointEval, spValue));
102820cf1dd8SToby Isaac   PetscFunctionReturn(0);
102920cf1dd8SToby Isaac }
103020cf1dd8SToby Isaac 
103120cf1dd8SToby Isaac /*@C
1032b4457527SToby Isaac   PetscDualSpaceApplyInterior - Apply interior functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetInteriorData()
1033b4457527SToby Isaac 
1034b4457527SToby Isaac   Input Parameters:
1035b4457527SToby Isaac + sp        - The PetscDualSpace object
1036b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetInteriorData()
1037b4457527SToby Isaac 
1038b4457527SToby Isaac   Output Parameter:
1039b4457527SToby Isaac . spValue   - The values of interior dual space functionals
1040b4457527SToby Isaac 
1041b4457527SToby Isaac   Level: beginner
1042b4457527SToby Isaac 
1043b4457527SToby Isaac .seealso: PetscDualSpaceCreate()
1044b4457527SToby Isaac @*/
1045b4457527SToby Isaac PetscErrorCode PetscDualSpaceApplyInterior(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue)
1046b4457527SToby Isaac {
1047b4457527SToby Isaac   PetscFunctionBegin;
1048b4457527SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
10499566063dSJacob Faibussowitsch   PetscCall((*sp->ops->applyint)(sp, pointEval, spValue));
1050b4457527SToby Isaac   PetscFunctionReturn(0);
1051b4457527SToby Isaac }
1052b4457527SToby Isaac 
1053b4457527SToby Isaac /*@C
105420cf1dd8SToby Isaac   PetscDualSpaceApplyDefault - Apply a functional from the dual space basis to an input function by assuming a point evaluation functional.
105520cf1dd8SToby Isaac 
105620cf1dd8SToby Isaac   Input Parameters:
105720cf1dd8SToby Isaac + sp    - The PetscDualSpace object
105820cf1dd8SToby Isaac . f     - The basis functional index
105920cf1dd8SToby Isaac . time  - The time
106020cf1dd8SToby Isaac . cgeom - A context with geometric information for this cell, we use v0 (the initial vertex) and J (the Jacobian)
106120cf1dd8SToby Isaac . Nc    - The number of components for the function
106220cf1dd8SToby Isaac . func  - The input function
106320cf1dd8SToby Isaac - ctx   - A context for the function
106420cf1dd8SToby Isaac 
106520cf1dd8SToby Isaac   Output Parameter:
106620cf1dd8SToby Isaac . value   - The output value
106720cf1dd8SToby Isaac 
106820cf1dd8SToby Isaac   Note: The calling sequence for the callback func is given by:
106920cf1dd8SToby Isaac 
107020cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[],
107120cf1dd8SToby Isaac $      PetscInt numComponents, PetscScalar values[], void *ctx)
107220cf1dd8SToby Isaac 
107320cf1dd8SToby Isaac and the idea is to evaluate the functional as an integral
107420cf1dd8SToby Isaac 
107520cf1dd8SToby Isaac $ n(f) = int dx n(x) . f(x)
107620cf1dd8SToby Isaac 
107720cf1dd8SToby Isaac where both n and f have Nc components.
107820cf1dd8SToby Isaac 
1079a4ce7ad1SMatthew G. Knepley   Level: beginner
108020cf1dd8SToby Isaac 
108120cf1dd8SToby Isaac .seealso: PetscDualSpaceCreate()
108220cf1dd8SToby Isaac @*/
108320cf1dd8SToby 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)
108420cf1dd8SToby Isaac {
108520cf1dd8SToby Isaac   DM               dm;
108620cf1dd8SToby Isaac   PetscQuadrature  n;
108720cf1dd8SToby Isaac   const PetscReal *points, *weights;
108820cf1dd8SToby Isaac   PetscReal        x[3];
108920cf1dd8SToby Isaac   PetscScalar     *val;
109020cf1dd8SToby Isaac   PetscInt         dim, dE, qNc, c, Nq, q;
109120cf1dd8SToby Isaac   PetscBool        isAffine;
109220cf1dd8SToby Isaac 
109320cf1dd8SToby Isaac   PetscFunctionBegin;
109420cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1095dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(value, 8);
10969566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDM(sp, &dm));
10979566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetFunctional(sp, f, &n));
10989566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureGetData(n, &dim, &qNc, &Nq, &points, &weights));
109963a3b9bcSJacob 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);
110063a3b9bcSJacob Faibussowitsch   PetscCheck(qNc == Nc,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_SIZ, "The quadrature components %" PetscInt_FMT " != function components %" PetscInt_FMT, qNc, Nc);
11019566063dSJacob Faibussowitsch   PetscCall(DMGetWorkArray(dm, Nc, MPIU_SCALAR, &val));
110220cf1dd8SToby Isaac   *value = 0.0;
110320cf1dd8SToby Isaac   isAffine = cgeom->isAffine;
110420cf1dd8SToby Isaac   dE = cgeom->dimEmbed;
110520cf1dd8SToby Isaac   for (q = 0; q < Nq; ++q) {
110620cf1dd8SToby Isaac     if (isAffine) {
110720cf1dd8SToby Isaac       CoordinatesRefToReal(dE, cgeom->dim, cgeom->xi, cgeom->v, cgeom->J, &points[q*dim], x);
11089566063dSJacob Faibussowitsch       PetscCall((*func)(dE, time, x, Nc, val, ctx));
110920cf1dd8SToby Isaac     } else {
11109566063dSJacob Faibussowitsch       PetscCall((*func)(dE, time, &cgeom->v[dE*q], Nc, val, ctx));
111120cf1dd8SToby Isaac     }
111220cf1dd8SToby Isaac     for (c = 0; c < Nc; ++c) {
111320cf1dd8SToby Isaac       *value += val[c]*weights[q*Nc+c];
111420cf1dd8SToby Isaac     }
111520cf1dd8SToby Isaac   }
11169566063dSJacob Faibussowitsch   PetscCall(DMRestoreWorkArray(dm, Nc, MPIU_SCALAR, &val));
111720cf1dd8SToby Isaac   PetscFunctionReturn(0);
111820cf1dd8SToby Isaac }
111920cf1dd8SToby Isaac 
112020cf1dd8SToby Isaac /*@C
1121b4457527SToby Isaac   PetscDualSpaceApplyAllDefault - Apply all functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetAllData()
112220cf1dd8SToby Isaac 
112320cf1dd8SToby Isaac   Input Parameters:
112420cf1dd8SToby Isaac + sp        - The PetscDualSpace object
1125b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetAllData()
112620cf1dd8SToby Isaac 
112720cf1dd8SToby Isaac   Output Parameter:
112820cf1dd8SToby Isaac . spValue   - The values of all dual space functionals
112920cf1dd8SToby Isaac 
1130a4ce7ad1SMatthew G. Knepley   Level: beginner
113120cf1dd8SToby Isaac 
113220cf1dd8SToby Isaac .seealso: PetscDualSpaceCreate()
113320cf1dd8SToby Isaac @*/
113420cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceApplyAllDefault(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue)
113520cf1dd8SToby Isaac {
1136b4457527SToby Isaac   Vec              pointValues, dofValues;
1137b4457527SToby Isaac   Mat              allMat;
113820cf1dd8SToby Isaac 
113920cf1dd8SToby Isaac   PetscFunctionBegin;
114020cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
114120cf1dd8SToby Isaac   PetscValidScalarPointer(pointEval, 2);
1142064a246eSJacob Faibussowitsch   PetscValidScalarPointer(spValue, 3);
11439566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetAllData(sp, NULL, &allMat));
1144b4457527SToby Isaac   if (!(sp->allNodeValues)) {
11459566063dSJacob Faibussowitsch     PetscCall(MatCreateVecs(allMat, &(sp->allNodeValues), NULL));
114620cf1dd8SToby Isaac   }
1147b4457527SToby Isaac   pointValues = sp->allNodeValues;
1148b4457527SToby Isaac   if (!(sp->allDofValues)) {
11499566063dSJacob Faibussowitsch     PetscCall(MatCreateVecs(allMat, NULL, &(sp->allDofValues)));
115020cf1dd8SToby Isaac   }
1151b4457527SToby Isaac   dofValues = sp->allDofValues;
11529566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(pointValues, pointEval));
11539566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(dofValues, spValue));
11549566063dSJacob Faibussowitsch   PetscCall(MatMult(allMat, pointValues, dofValues));
11559566063dSJacob Faibussowitsch   PetscCall(VecResetArray(dofValues));
11569566063dSJacob Faibussowitsch   PetscCall(VecResetArray(pointValues));
1157b4457527SToby Isaac   PetscFunctionReturn(0);
115820cf1dd8SToby Isaac }
1159b4457527SToby Isaac 
1160b4457527SToby Isaac /*@C
1161b4457527SToby Isaac   PetscDualSpaceApplyInteriorDefault - Apply interior functionals from the dual space basis to the result of an evaluation at the points returned by PetscDualSpaceGetInteriorData()
1162b4457527SToby Isaac 
1163b4457527SToby Isaac   Input Parameters:
1164b4457527SToby Isaac + sp        - The PetscDualSpace object
1165b4457527SToby Isaac - pointEval - Evaluation at the points returned by PetscDualSpaceGetInteriorData()
1166b4457527SToby Isaac 
1167b4457527SToby Isaac   Output Parameter:
1168b4457527SToby Isaac . spValue   - The values of interior dual space functionals
1169b4457527SToby Isaac 
1170b4457527SToby Isaac   Level: beginner
1171b4457527SToby Isaac 
1172b4457527SToby Isaac .seealso: PetscDualSpaceCreate()
1173b4457527SToby Isaac @*/
1174b4457527SToby Isaac PetscErrorCode PetscDualSpaceApplyInteriorDefault(PetscDualSpace sp, const PetscScalar *pointEval, PetscScalar *spValue)
1175b4457527SToby Isaac {
1176b4457527SToby Isaac   Vec              pointValues, dofValues;
1177b4457527SToby Isaac   Mat              intMat;
1178b4457527SToby Isaac 
1179b4457527SToby Isaac   PetscFunctionBegin;
1180b4457527SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1181b4457527SToby Isaac   PetscValidScalarPointer(pointEval, 2);
1182064a246eSJacob Faibussowitsch   PetscValidScalarPointer(spValue, 3);
11839566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetInteriorData(sp, NULL, &intMat));
1184b4457527SToby Isaac   if (!(sp->intNodeValues)) {
11859566063dSJacob Faibussowitsch     PetscCall(MatCreateVecs(intMat, &(sp->intNodeValues), NULL));
1186b4457527SToby Isaac   }
1187b4457527SToby Isaac   pointValues = sp->intNodeValues;
1188b4457527SToby Isaac   if (!(sp->intDofValues)) {
11899566063dSJacob Faibussowitsch     PetscCall(MatCreateVecs(intMat, NULL, &(sp->intDofValues)));
1190b4457527SToby Isaac   }
1191b4457527SToby Isaac   dofValues = sp->intDofValues;
11929566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(pointValues, pointEval));
11939566063dSJacob Faibussowitsch   PetscCall(VecPlaceArray(dofValues, spValue));
11949566063dSJacob Faibussowitsch   PetscCall(MatMult(intMat, pointValues, dofValues));
11959566063dSJacob Faibussowitsch   PetscCall(VecResetArray(dofValues));
11969566063dSJacob Faibussowitsch   PetscCall(VecResetArray(pointValues));
119720cf1dd8SToby Isaac   PetscFunctionReturn(0);
119820cf1dd8SToby Isaac }
119920cf1dd8SToby Isaac 
1200a4ce7ad1SMatthew G. Knepley /*@
1201b4457527SToby Isaac   PetscDualSpaceGetAllData - Get all quadrature nodes from this space, and the matrix that sends quadrature node values to degree-of-freedom values
1202a4ce7ad1SMatthew G. Knepley 
1203a4ce7ad1SMatthew G. Knepley   Input Parameter:
1204a4ce7ad1SMatthew G. Knepley . sp - The dualspace
1205a4ce7ad1SMatthew G. Knepley 
1206d8d19677SJose E. Roman   Output Parameters:
1207b4457527SToby Isaac + allNodes - A PetscQuadrature object containing all evaluation nodes
1208b4457527SToby Isaac - allMat - A Mat for the node-to-dof transformation
1209a4ce7ad1SMatthew G. Knepley 
1210a4ce7ad1SMatthew G. Knepley   Level: advanced
1211a4ce7ad1SMatthew G. Knepley 
1212a4ce7ad1SMatthew G. Knepley .seealso: PetscDualSpaceCreate()
1213a4ce7ad1SMatthew G. Knepley @*/
1214b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetAllData(PetscDualSpace sp, PetscQuadrature *allNodes, Mat *allMat)
121520cf1dd8SToby Isaac {
121620cf1dd8SToby Isaac   PetscFunctionBegin;
121720cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1218b4457527SToby Isaac   if (allNodes) PetscValidPointer(allNodes,2);
1219b4457527SToby Isaac   if (allMat) PetscValidPointer(allMat,3);
1220b4457527SToby Isaac   if ((!sp->allNodes || !sp->allMat) && sp->ops->createalldata) {
1221b4457527SToby Isaac     PetscQuadrature qpoints;
1222b4457527SToby Isaac     Mat amat;
1223b4457527SToby Isaac 
12249566063dSJacob Faibussowitsch     PetscCall((*sp->ops->createalldata)(sp,&qpoints,&amat));
12259566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureDestroy(&(sp->allNodes)));
12269566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&(sp->allMat)));
1227b4457527SToby Isaac     sp->allNodes = qpoints;
1228b4457527SToby Isaac     sp->allMat = amat;
122920cf1dd8SToby Isaac   }
1230b4457527SToby Isaac   if (allNodes) *allNodes = sp->allNodes;
1231b4457527SToby Isaac   if (allMat) *allMat = sp->allMat;
123220cf1dd8SToby Isaac   PetscFunctionReturn(0);
123320cf1dd8SToby Isaac }
123420cf1dd8SToby Isaac 
1235a4ce7ad1SMatthew G. Knepley /*@
1236b4457527SToby Isaac   PetscDualSpaceCreateAllDataDefault - Create all evaluation nodes and the node-to-dof matrix by examining functionals
1237a4ce7ad1SMatthew G. Knepley 
1238a4ce7ad1SMatthew G. Knepley   Input Parameter:
1239a4ce7ad1SMatthew G. Knepley . sp - The dualspace
1240a4ce7ad1SMatthew G. Knepley 
1241d8d19677SJose E. Roman   Output Parameters:
1242b4457527SToby Isaac + allNodes - A PetscQuadrature object containing all evaluation nodes
1243b4457527SToby Isaac - allMat - A Mat for the node-to-dof transformation
1244a4ce7ad1SMatthew G. Knepley 
1245a4ce7ad1SMatthew G. Knepley   Level: advanced
1246a4ce7ad1SMatthew G. Knepley 
1247a4ce7ad1SMatthew G. Knepley .seealso: PetscDualSpaceCreate()
1248a4ce7ad1SMatthew G. Knepley @*/
1249b4457527SToby Isaac PetscErrorCode PetscDualSpaceCreateAllDataDefault(PetscDualSpace sp, PetscQuadrature *allNodes, Mat *allMat)
125020cf1dd8SToby Isaac {
125120cf1dd8SToby Isaac   PetscInt        spdim;
125220cf1dd8SToby Isaac   PetscInt        numPoints, offset;
125320cf1dd8SToby Isaac   PetscReal       *points;
125420cf1dd8SToby Isaac   PetscInt        f, dim;
1255b4457527SToby Isaac   PetscInt        Nc, nrows, ncols;
1256b4457527SToby Isaac   PetscInt        maxNumPoints;
125720cf1dd8SToby Isaac   PetscQuadrature q;
1258b4457527SToby Isaac   Mat             A;
125920cf1dd8SToby Isaac 
126020cf1dd8SToby Isaac   PetscFunctionBegin;
12619566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetNumComponents(sp, &Nc));
12629566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDimension(sp,&spdim));
126320cf1dd8SToby Isaac   if (!spdim) {
12649566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,allNodes));
12659566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureSetData(*allNodes,0,0,0,NULL,NULL));
126620cf1dd8SToby Isaac   }
1267b4457527SToby Isaac   nrows = spdim;
12689566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetFunctional(sp,0,&q));
12699566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureGetData(q,&dim,NULL,&numPoints,NULL,NULL));
1270b4457527SToby Isaac   maxNumPoints = numPoints;
127120cf1dd8SToby Isaac   for (f = 1; f < spdim; f++) {
127220cf1dd8SToby Isaac     PetscInt Np;
127320cf1dd8SToby Isaac 
12749566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceGetFunctional(sp,f,&q));
12759566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,NULL,NULL));
127620cf1dd8SToby Isaac     numPoints += Np;
1277b4457527SToby Isaac     maxNumPoints = PetscMax(maxNumPoints,Np);
127820cf1dd8SToby Isaac   }
1279b4457527SToby Isaac   ncols = numPoints * Nc;
12809566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(dim*numPoints,&points));
12819566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF, nrows, ncols, maxNumPoints * Nc, NULL, &A));
128220cf1dd8SToby Isaac   for (f = 0, offset = 0; f < spdim; f++) {
1283b4457527SToby Isaac     const PetscReal *p, *w;
128420cf1dd8SToby Isaac     PetscInt        Np, i;
1285b4457527SToby Isaac     PetscInt        fnc;
128620cf1dd8SToby Isaac 
12879566063dSJacob Faibussowitsch     PetscCall(PetscDualSpaceGetFunctional(sp,f,&q));
12889566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureGetData(q,NULL,&fnc,&Np,&p,&w));
128908401ef6SPierre Jolivet     PetscCheck(fnc == Nc,PETSC_COMM_SELF, PETSC_ERR_PLIB, "functional component mismatch");
1290b4457527SToby Isaac     for (i = 0; i < Np * dim; i++) {
1291b4457527SToby Isaac       points[offset* dim + i] = p[i];
1292b4457527SToby Isaac     }
1293b4457527SToby Isaac     for (i = 0; i < Np * Nc; i++) {
12949566063dSJacob Faibussowitsch       PetscCall(MatSetValue(A, f, offset * Nc, w[i], INSERT_VALUES));
1295b4457527SToby Isaac     }
1296b4457527SToby Isaac     offset += Np;
1297b4457527SToby Isaac   }
12989566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
12999566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
13009566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,allNodes));
13019566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureSetData(*allNodes,dim,0,numPoints,points,NULL));
1302b4457527SToby Isaac   *allMat = A;
1303b4457527SToby Isaac   PetscFunctionReturn(0);
1304b4457527SToby Isaac }
1305b4457527SToby Isaac 
1306b4457527SToby Isaac /*@
1307b4457527SToby Isaac   PetscDualSpaceGetInteriorData - Get all quadrature points necessary to compute the interior degrees of freedom from
1308b4457527SToby Isaac   this space, as well as the matrix that computes the degrees of freedom from the quadrature values.  Degrees of
1309b4457527SToby Isaac   freedom are interior degrees of freedom if they belong (by PetscDualSpaceGetSection()) to interior points in the
1310b4457527SToby Isaac   reference DMPlex: complementary boundary degrees of freedom are marked as constrained in the section returned by
1311b4457527SToby Isaac   PetscDualSpaceGetSection()).
1312b4457527SToby Isaac 
1313b4457527SToby Isaac   Input Parameter:
1314b4457527SToby Isaac . sp - The dualspace
1315b4457527SToby Isaac 
1316d8d19677SJose E. Roman   Output Parameters:
1317b4457527SToby Isaac + intNodes - A PetscQuadrature object containing all evaluation points needed to evaluate interior degrees of freedom
1318b4457527SToby Isaac - intMat   - A matrix that computes dual space values from point values: size [spdim0 x (npoints * nc)], where spdim0 is
1319b4457527SToby Isaac              the size of the constrained layout (PetscSectionGetConstrainStorageSize()) of the dual space section,
1320b4457527SToby Isaac              npoints is the number of points in intNodes and nc is PetscDualSpaceGetNumComponents().
1321b4457527SToby Isaac 
1322b4457527SToby Isaac   Level: advanced
1323b4457527SToby Isaac 
1324b4457527SToby Isaac .seealso: PetscDualSpaceCreate(), PetscDualSpaceGetDimension(), PetscDualSpaceGetNumComponents(), PetscQuadratureGetData()
1325b4457527SToby Isaac @*/
1326b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetInteriorData(PetscDualSpace sp, PetscQuadrature *intNodes, Mat *intMat)
1327b4457527SToby Isaac {
1328b4457527SToby Isaac   PetscFunctionBegin;
1329b4457527SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1330b4457527SToby Isaac   if (intNodes) PetscValidPointer(intNodes,2);
1331b4457527SToby Isaac   if (intMat) PetscValidPointer(intMat,3);
1332b4457527SToby Isaac   if ((!sp->intNodes || !sp->intMat) && sp->ops->createintdata) {
1333b4457527SToby Isaac     PetscQuadrature qpoints;
1334b4457527SToby Isaac     Mat imat;
1335b4457527SToby Isaac 
13369566063dSJacob Faibussowitsch     PetscCall((*sp->ops->createintdata)(sp,&qpoints,&imat));
13379566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureDestroy(&(sp->intNodes)));
13389566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&(sp->intMat)));
1339b4457527SToby Isaac     sp->intNodes = qpoints;
1340b4457527SToby Isaac     sp->intMat = imat;
1341b4457527SToby Isaac   }
1342b4457527SToby Isaac   if (intNodes) *intNodes = sp->intNodes;
1343b4457527SToby Isaac   if (intMat) *intMat = sp->intMat;
1344b4457527SToby Isaac   PetscFunctionReturn(0);
1345b4457527SToby Isaac }
1346b4457527SToby Isaac 
1347b4457527SToby Isaac /*@
1348b4457527SToby Isaac   PetscDualSpaceCreateInteriorDataDefault - Create quadrature points by examining interior functionals and create the matrix mapping quadrature point values to interior dual space values
1349b4457527SToby Isaac 
1350b4457527SToby Isaac   Input Parameter:
1351b4457527SToby Isaac . sp - The dualspace
1352b4457527SToby Isaac 
1353d8d19677SJose E. Roman   Output Parameters:
1354b4457527SToby Isaac + intNodes - A PetscQuadrature object containing all evaluation points needed to evaluate interior degrees of freedom
1355b4457527SToby Isaac - intMat    - A matrix that computes dual space values from point values: size [spdim0 x (npoints * nc)], where spdim0 is
1356b4457527SToby Isaac               the size of the constrained layout (PetscSectionGetConstrainStorageSize()) of the dual space section,
1357b4457527SToby Isaac               npoints is the number of points in allNodes and nc is PetscDualSpaceGetNumComponents().
1358b4457527SToby Isaac 
1359b4457527SToby Isaac   Level: advanced
1360b4457527SToby Isaac 
1361b4457527SToby Isaac .seealso: PetscDualSpaceCreate(), PetscDualSpaceGetInteriorData()
1362b4457527SToby Isaac @*/
1363b4457527SToby Isaac PetscErrorCode PetscDualSpaceCreateInteriorDataDefault(PetscDualSpace sp, PetscQuadrature *intNodes, Mat *intMat)
1364b4457527SToby Isaac {
1365b4457527SToby Isaac   DM              dm;
1366b4457527SToby Isaac   PetscInt        spdim0;
1367b4457527SToby Isaac   PetscInt        Nc;
1368b4457527SToby Isaac   PetscInt        pStart, pEnd, p, f;
1369b4457527SToby Isaac   PetscSection    section;
1370b4457527SToby Isaac   PetscInt        numPoints, offset, matoffset;
1371b4457527SToby Isaac   PetscReal       *points;
1372b4457527SToby Isaac   PetscInt        dim;
1373b4457527SToby Isaac   PetscInt        *nnz;
1374b4457527SToby Isaac   PetscQuadrature q;
1375b4457527SToby Isaac   Mat             imat;
1376b4457527SToby Isaac 
1377b4457527SToby Isaac   PetscFunctionBegin;
1378b4457527SToby Isaac   PetscValidHeaderSpecific(sp,PETSCDUALSPACE_CLASSID,1);
13799566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetSection(sp, &section));
13809566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetConstrainedStorageSize(section, &spdim0));
1381b4457527SToby Isaac   if (!spdim0) {
1382b4457527SToby Isaac     *intNodes = NULL;
1383b4457527SToby Isaac     *intMat = NULL;
1384b4457527SToby Isaac     PetscFunctionReturn(0);
1385b4457527SToby Isaac   }
13869566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetNumComponents(sp, &Nc));
13879566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetChart(section, &pStart, &pEnd));
13889566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDM(sp, &dm));
13899566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm, &dim));
13909566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(spdim0, &nnz));
1391b4457527SToby Isaac   for (p = pStart, f = 0, numPoints = 0; p < pEnd; p++) {
1392b4457527SToby Isaac     PetscInt dof, cdof, off, d;
1393b4457527SToby Isaac 
13949566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetDof(section, p, &dof));
13959566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetConstraintDof(section, p, &cdof));
1396b4457527SToby Isaac     if (!(dof - cdof)) continue;
13979566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(section, p, &off));
1398b4457527SToby Isaac     for (d = 0; d < dof; d++, off++, f++) {
1399b4457527SToby Isaac       PetscInt Np;
1400b4457527SToby Isaac 
14019566063dSJacob Faibussowitsch       PetscCall(PetscDualSpaceGetFunctional(sp,off,&q));
14029566063dSJacob Faibussowitsch       PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,NULL,NULL));
1403b4457527SToby Isaac       nnz[f] = Np * Nc;
1404b4457527SToby Isaac       numPoints += Np;
1405b4457527SToby Isaac     }
1406b4457527SToby Isaac   }
14079566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF, spdim0, numPoints * Nc, 0, nnz, &imat));
14089566063dSJacob Faibussowitsch   PetscCall(PetscFree(nnz));
14099566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(dim*numPoints,&points));
1410b4457527SToby Isaac   for (p = pStart, f = 0, offset = 0, matoffset = 0; p < pEnd; p++) {
1411b4457527SToby Isaac     PetscInt dof, cdof, off, d;
1412b4457527SToby Isaac 
14139566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetDof(section, p, &dof));
14149566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetConstraintDof(section, p, &cdof));
1415b4457527SToby Isaac     if (!(dof - cdof)) continue;
14169566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(section, p, &off));
1417b4457527SToby Isaac     for (d = 0; d < dof; d++, off++, f++) {
1418b4457527SToby Isaac       const PetscReal *p;
1419b4457527SToby Isaac       const PetscReal *w;
1420b4457527SToby Isaac       PetscInt        Np, i;
1421b4457527SToby Isaac 
14229566063dSJacob Faibussowitsch       PetscCall(PetscDualSpaceGetFunctional(sp,off,&q));
14239566063dSJacob Faibussowitsch       PetscCall(PetscQuadratureGetData(q,NULL,NULL,&Np,&p,&w));
142420cf1dd8SToby Isaac       for (i = 0; i < Np * dim; i++) {
142520cf1dd8SToby Isaac         points[offset + i] = p[i];
142620cf1dd8SToby Isaac       }
1427b4457527SToby Isaac       for (i = 0; i < Np * Nc; i++) {
14289566063dSJacob Faibussowitsch         PetscCall(MatSetValue(imat, f, matoffset + i, w[i],INSERT_VALUES));
142920cf1dd8SToby Isaac       }
1430b4457527SToby Isaac       offset += Np * dim;
1431b4457527SToby Isaac       matoffset += Np * Nc;
1432b4457527SToby Isaac     }
1433b4457527SToby Isaac   }
14349566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureCreate(PETSC_COMM_SELF,intNodes));
14359566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureSetData(*intNodes,dim,0,numPoints,points,NULL));
14369566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(imat, MAT_FINAL_ASSEMBLY));
14379566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(imat, MAT_FINAL_ASSEMBLY));
1438b4457527SToby Isaac   *intMat = imat;
143920cf1dd8SToby Isaac   PetscFunctionReturn(0);
144020cf1dd8SToby Isaac }
144120cf1dd8SToby Isaac 
14424f9ab2b4SJed Brown /*@
14434f9ab2b4SJed Brown   PetscDualSpaceEqual - Determine if a dual space is equivalent
14444f9ab2b4SJed Brown 
14454f9ab2b4SJed Brown   Input Parameters:
14464f9ab2b4SJed Brown + A    - A PetscDualSpace object
14474f9ab2b4SJed Brown - B    - Another PetscDualSpace object
14484f9ab2b4SJed Brown 
14494f9ab2b4SJed Brown   Output Parameter:
14504f9ab2b4SJed Brown . equal - PETSC_TRUE if the dual spaces are equivalent
14514f9ab2b4SJed Brown 
14524f9ab2b4SJed Brown   Level: advanced
14534f9ab2b4SJed Brown 
14544f9ab2b4SJed Brown .seealso: PetscDualSpaceCreate()
14554f9ab2b4SJed Brown @*/
14564f9ab2b4SJed Brown PetscErrorCode PetscDualSpaceEqual(PetscDualSpace A, PetscDualSpace B, PetscBool *equal)
14574f9ab2b4SJed Brown {
14584f9ab2b4SJed Brown   PetscInt sizeA, sizeB, dimA, dimB;
14594f9ab2b4SJed Brown   const PetscInt *dofA, *dofB;
14604f9ab2b4SJed Brown   PetscQuadrature quadA, quadB;
14614f9ab2b4SJed Brown   Mat matA, matB;
14624f9ab2b4SJed Brown 
14634f9ab2b4SJed Brown   PetscFunctionBegin;
14644f9ab2b4SJed Brown   PetscValidHeaderSpecific(A,PETSCDUALSPACE_CLASSID,1);
14654f9ab2b4SJed Brown   PetscValidHeaderSpecific(B,PETSCDUALSPACE_CLASSID,2);
14664f9ab2b4SJed Brown   PetscValidBoolPointer(equal,3);
14674f9ab2b4SJed Brown   *equal = PETSC_FALSE;
14689566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDimension(A, &sizeA));
14699566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDimension(B, &sizeB));
14704f9ab2b4SJed Brown   if (sizeB != sizeA) {
14714f9ab2b4SJed Brown     PetscFunctionReturn(0);
14724f9ab2b4SJed Brown   }
14739566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(A->dm, &dimA));
14749566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(B->dm, &dimB));
14754f9ab2b4SJed Brown   if (dimA != dimB) {
14764f9ab2b4SJed Brown     PetscFunctionReturn(0);
14774f9ab2b4SJed Brown   }
14784f9ab2b4SJed Brown 
14799566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetNumDof(A, &dofA));
14809566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetNumDof(B, &dofB));
14814f9ab2b4SJed Brown   for (PetscInt d=0; d<dimA; d++) {
14824f9ab2b4SJed Brown     if (dofA[d] != dofB[d]) {
14834f9ab2b4SJed Brown       PetscFunctionReturn(0);
14844f9ab2b4SJed Brown     }
14854f9ab2b4SJed Brown   }
14864f9ab2b4SJed Brown 
14879566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetInteriorData(A, &quadA, &matA));
14889566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetInteriorData(B, &quadB, &matB));
14894f9ab2b4SJed Brown   if (!quadA && !quadB) {
14904f9ab2b4SJed Brown     *equal = PETSC_TRUE;
14914f9ab2b4SJed Brown   } else if (quadA && quadB) {
14929566063dSJacob Faibussowitsch     PetscCall(PetscQuadratureEqual(quadA, quadB, equal));
14934f9ab2b4SJed Brown     if (*equal == PETSC_FALSE) PetscFunctionReturn(0);
14944f9ab2b4SJed Brown     if (!matA && !matB) PetscFunctionReturn(0);
14959566063dSJacob Faibussowitsch     if (matA && matB) PetscCall(MatEqual(matA, matB, equal));
14964f9ab2b4SJed Brown     else *equal = PETSC_FALSE;
14974f9ab2b4SJed Brown   }
14984f9ab2b4SJed Brown   PetscFunctionReturn(0);
14994f9ab2b4SJed Brown }
15004f9ab2b4SJed Brown 
150120cf1dd8SToby Isaac /*@C
150220cf1dd8SToby Isaac   PetscDualSpaceApplyFVM - Apply a functional from the dual space basis to an input function by assuming a point evaluation functional at the cell centroid.
150320cf1dd8SToby Isaac 
150420cf1dd8SToby Isaac   Input Parameters:
150520cf1dd8SToby Isaac + sp    - The PetscDualSpace object
150620cf1dd8SToby Isaac . f     - The basis functional index
150720cf1dd8SToby Isaac . time  - The time
150820cf1dd8SToby Isaac . cgeom - A context with geometric information for this cell, we currently just use the centroid
150920cf1dd8SToby Isaac . Nc    - The number of components for the function
151020cf1dd8SToby Isaac . func  - The input function
151120cf1dd8SToby Isaac - ctx   - A context for the function
151220cf1dd8SToby Isaac 
151320cf1dd8SToby Isaac   Output Parameter:
151420cf1dd8SToby Isaac . value - The output value (scalar)
151520cf1dd8SToby Isaac 
151620cf1dd8SToby Isaac   Note: The calling sequence for the callback func is given by:
151720cf1dd8SToby Isaac 
151820cf1dd8SToby Isaac $ func(PetscInt dim, PetscReal time, const PetscReal x[],
151920cf1dd8SToby Isaac $      PetscInt numComponents, PetscScalar values[], void *ctx)
152020cf1dd8SToby Isaac 
152120cf1dd8SToby Isaac and the idea is to evaluate the functional as an integral
152220cf1dd8SToby Isaac 
152320cf1dd8SToby Isaac $ n(f) = int dx n(x) . f(x)
152420cf1dd8SToby Isaac 
152520cf1dd8SToby Isaac where both n and f have Nc components.
152620cf1dd8SToby Isaac 
1527a4ce7ad1SMatthew G. Knepley   Level: beginner
152820cf1dd8SToby Isaac 
152920cf1dd8SToby Isaac .seealso: PetscDualSpaceCreate()
153020cf1dd8SToby Isaac @*/
153120cf1dd8SToby 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)
153220cf1dd8SToby Isaac {
153320cf1dd8SToby Isaac   DM               dm;
153420cf1dd8SToby Isaac   PetscQuadrature  n;
153520cf1dd8SToby Isaac   const PetscReal *points, *weights;
153620cf1dd8SToby Isaac   PetscScalar     *val;
153720cf1dd8SToby Isaac   PetscInt         dimEmbed, qNc, c, Nq, q;
153820cf1dd8SToby Isaac 
153920cf1dd8SToby Isaac   PetscFunctionBegin;
154020cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1541dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(value, 8);
15429566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDM(sp, &dm));
15439566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dimEmbed));
15449566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetFunctional(sp, f, &n));
15459566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureGetData(n, NULL, &qNc, &Nq, &points, &weights));
154663a3b9bcSJacob Faibussowitsch   PetscCheck(qNc == Nc,PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_SIZ, "The quadrature components %" PetscInt_FMT " != function components %" PetscInt_FMT, qNc, Nc);
15479566063dSJacob Faibussowitsch   PetscCall(DMGetWorkArray(dm, Nc, MPIU_SCALAR, &val));
154820cf1dd8SToby Isaac   *value = 0.;
154920cf1dd8SToby Isaac   for (q = 0; q < Nq; ++q) {
15509566063dSJacob Faibussowitsch     PetscCall((*func)(dimEmbed, time, cgeom->centroid, Nc, val, ctx));
155120cf1dd8SToby Isaac     for (c = 0; c < Nc; ++c) {
155220cf1dd8SToby Isaac       *value += val[c]*weights[q*Nc+c];
155320cf1dd8SToby Isaac     }
155420cf1dd8SToby Isaac   }
15559566063dSJacob Faibussowitsch   PetscCall(DMRestoreWorkArray(dm, Nc, MPIU_SCALAR, &val));
155620cf1dd8SToby Isaac   PetscFunctionReturn(0);
155720cf1dd8SToby Isaac }
155820cf1dd8SToby Isaac 
155920cf1dd8SToby Isaac /*@
156020cf1dd8SToby Isaac   PetscDualSpaceGetHeightSubspace - Get the subset of the dual space basis that is supported on a mesh point of a
156120cf1dd8SToby Isaac   given height.  This assumes that the reference cell is symmetric over points of this height.
156220cf1dd8SToby Isaac 
156320cf1dd8SToby Isaac   If the dual space is not defined on mesh points of the given height (e.g. if the space is discontinuous and
156420cf1dd8SToby Isaac   pointwise values are not defined on the element boundaries), or if the implementation of PetscDualSpace does not
156520cf1dd8SToby Isaac   support extracting subspaces, then NULL is returned.
156620cf1dd8SToby Isaac 
156720cf1dd8SToby Isaac   This does not increment the reference count on the returned dual space, and the user should not destroy it.
156820cf1dd8SToby Isaac 
156920cf1dd8SToby Isaac   Not collective
157020cf1dd8SToby Isaac 
157120cf1dd8SToby Isaac   Input Parameters:
157220cf1dd8SToby Isaac + sp - the PetscDualSpace object
157320cf1dd8SToby Isaac - height - the height of the mesh point for which the subspace is desired
157420cf1dd8SToby Isaac 
157520cf1dd8SToby Isaac   Output Parameter:
157620cf1dd8SToby Isaac . subsp - the subspace.  Note that the functionals in the subspace are with respect to the intrinsic geometry of the
157720cf1dd8SToby Isaac   point, which will be of lesser dimension if height > 0.
157820cf1dd8SToby Isaac 
157920cf1dd8SToby Isaac   Level: advanced
158020cf1dd8SToby Isaac 
158120cf1dd8SToby Isaac .seealso: PetscSpaceGetHeightSubspace(), PetscDualSpace
158220cf1dd8SToby Isaac @*/
158320cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetHeightSubspace(PetscDualSpace sp, PetscInt height, PetscDualSpace *subsp)
158420cf1dd8SToby Isaac {
1585b4457527SToby Isaac   PetscInt       depth = -1, cStart, cEnd;
1586b4457527SToby Isaac   DM             dm;
158720cf1dd8SToby Isaac 
158820cf1dd8SToby Isaac   PetscFunctionBegin;
158920cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1590064a246eSJacob Faibussowitsch   PetscValidPointer(subsp,3);
159108401ef6SPierre 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");
159220cf1dd8SToby Isaac   *subsp = NULL;
1593b4457527SToby Isaac   dm = sp->dm;
15949566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm, &depth));
1595*1dca8a05SBarry Smith   PetscCheck(height >= 0 && height <= depth,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid height");
15969566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd));
1597b4457527SToby Isaac   if (height == 0 && cEnd == cStart + 1) {
1598b4457527SToby Isaac     *subsp = sp;
1599b4457527SToby Isaac     PetscFunctionReturn(0);
1600b4457527SToby Isaac   }
1601b4457527SToby Isaac   if (!sp->heightSpaces) {
1602b4457527SToby Isaac     PetscInt h;
16039566063dSJacob Faibussowitsch     PetscCall(PetscCalloc1(depth+1, &(sp->heightSpaces)));
1604b4457527SToby Isaac 
1605b4457527SToby Isaac     for (h = 0; h <= depth; h++) {
1606b4457527SToby Isaac       if (h == 0 && cEnd == cStart + 1) continue;
16079566063dSJacob Faibussowitsch       if (sp->ops->createheightsubspace) PetscCall((*sp->ops->createheightsubspace)(sp,height,&(sp->heightSpaces[h])));
1608b4457527SToby Isaac       else if (sp->pointSpaces) {
1609b4457527SToby Isaac         PetscInt hStart, hEnd;
1610b4457527SToby Isaac 
16119566063dSJacob Faibussowitsch         PetscCall(DMPlexGetHeightStratum(dm,h,&hStart,&hEnd));
1612b4457527SToby Isaac         if (hEnd > hStart) {
1613665f567fSMatthew G. Knepley           const char *name;
1614665f567fSMatthew G. Knepley 
16159566063dSJacob Faibussowitsch           PetscCall(PetscObjectReference((PetscObject)(sp->pointSpaces[hStart])));
1616665f567fSMatthew G. Knepley           if (sp->pointSpaces[hStart]) {
16179566063dSJacob Faibussowitsch             PetscCall(PetscObjectGetName((PetscObject) sp,                     &name));
16189566063dSJacob Faibussowitsch             PetscCall(PetscObjectSetName((PetscObject) sp->pointSpaces[hStart], name));
1619665f567fSMatthew G. Knepley           }
1620b4457527SToby Isaac           sp->heightSpaces[h] = sp->pointSpaces[hStart];
1621b4457527SToby Isaac         }
1622b4457527SToby Isaac       }
1623b4457527SToby Isaac     }
1624b4457527SToby Isaac   }
1625b4457527SToby Isaac   *subsp = sp->heightSpaces[height];
162620cf1dd8SToby Isaac   PetscFunctionReturn(0);
162720cf1dd8SToby Isaac }
162820cf1dd8SToby Isaac 
162920cf1dd8SToby Isaac /*@
163020cf1dd8SToby Isaac   PetscDualSpaceGetPointSubspace - Get the subset of the dual space basis that is supported on a particular mesh point.
163120cf1dd8SToby Isaac 
163220cf1dd8SToby Isaac   If the dual space is not defined on the mesh point (e.g. if the space is discontinuous and pointwise values are not
163320cf1dd8SToby Isaac   defined on the element boundaries), or if the implementation of PetscDualSpace does not support extracting
163420cf1dd8SToby Isaac   subspaces, then NULL is returned.
163520cf1dd8SToby Isaac 
163620cf1dd8SToby Isaac   This does not increment the reference count on the returned dual space, and the user should not destroy it.
163720cf1dd8SToby Isaac 
163820cf1dd8SToby Isaac   Not collective
163920cf1dd8SToby Isaac 
164020cf1dd8SToby Isaac   Input Parameters:
164120cf1dd8SToby Isaac + sp - the PetscDualSpace object
164220cf1dd8SToby Isaac - point - the point (in the dual space's DM) for which the subspace is desired
164320cf1dd8SToby Isaac 
164420cf1dd8SToby Isaac   Output Parameters:
164520cf1dd8SToby Isaac   bdsp - the subspace.  Note that the functionals in the subspace are with respect to the intrinsic geometry of the
164620cf1dd8SToby Isaac   point, which will be of lesser dimension if height > 0.
164720cf1dd8SToby Isaac 
164820cf1dd8SToby Isaac   Level: advanced
164920cf1dd8SToby Isaac 
165020cf1dd8SToby Isaac .seealso: PetscDualSpace
165120cf1dd8SToby Isaac @*/
165220cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceGetPointSubspace(PetscDualSpace sp, PetscInt point, PetscDualSpace *bdsp)
165320cf1dd8SToby Isaac {
1654b4457527SToby Isaac   PetscInt       pStart = 0, pEnd = 0, cStart, cEnd;
1655b4457527SToby Isaac   DM             dm;
165620cf1dd8SToby Isaac 
165720cf1dd8SToby Isaac   PetscFunctionBegin;
165820cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
1659064a246eSJacob Faibussowitsch   PetscValidPointer(bdsp,3);
166020cf1dd8SToby Isaac   *bdsp = NULL;
1661b4457527SToby Isaac   dm = sp->dm;
16629566063dSJacob Faibussowitsch   PetscCall(DMPlexGetChart(dm, &pStart, &pEnd));
1663*1dca8a05SBarry Smith   PetscCheck(point >= pStart && point <= pEnd,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid point");
16649566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd));
1665b4457527SToby 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 */
1666b4457527SToby Isaac     *bdsp = sp;
1667b4457527SToby Isaac     PetscFunctionReturn(0);
1668b4457527SToby Isaac   }
1669b4457527SToby Isaac   if (!sp->pointSpaces) {
1670b4457527SToby Isaac     PetscInt p;
16719566063dSJacob Faibussowitsch     PetscCall(PetscCalloc1(pEnd - pStart, &(sp->pointSpaces)));
167220cf1dd8SToby Isaac 
1673b4457527SToby Isaac     for (p = 0; p < pEnd - pStart; p++) {
1674b4457527SToby Isaac       if (p + pStart == cStart && cEnd == cStart + 1) continue;
16759566063dSJacob Faibussowitsch       if (sp->ops->createpointsubspace) PetscCall((*sp->ops->createpointsubspace)(sp,p+pStart,&(sp->pointSpaces[p])));
1676b4457527SToby Isaac       else if (sp->heightSpaces || sp->ops->createheightsubspace) {
1677b4457527SToby Isaac         PetscInt dim, depth, height;
1678b4457527SToby Isaac         DMLabel  label;
1679b4457527SToby Isaac 
16809566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepth(dm,&dim));
16819566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthLabel(dm,&label));
16829566063dSJacob Faibussowitsch         PetscCall(DMLabelGetValue(label,p+pStart,&depth));
168320cf1dd8SToby Isaac         height = dim - depth;
16849566063dSJacob Faibussowitsch         PetscCall(PetscDualSpaceGetHeightSubspace(sp, height, &(sp->pointSpaces[p])));
16859566063dSJacob Faibussowitsch         PetscCall(PetscObjectReference((PetscObject)sp->pointSpaces[p]));
168620cf1dd8SToby Isaac       }
1687b4457527SToby Isaac     }
1688b4457527SToby Isaac   }
1689b4457527SToby Isaac   *bdsp = sp->pointSpaces[point - pStart];
169020cf1dd8SToby Isaac   PetscFunctionReturn(0);
169120cf1dd8SToby Isaac }
169220cf1dd8SToby Isaac 
16936f905325SMatthew G. Knepley /*@C
16946f905325SMatthew G. Knepley   PetscDualSpaceGetSymmetries - Returns a description of the symmetries of this basis
16956f905325SMatthew G. Knepley 
16966f905325SMatthew G. Knepley   Not collective
16976f905325SMatthew G. Knepley 
16986f905325SMatthew G. Knepley   Input Parameter:
16996f905325SMatthew G. Knepley . sp - the PetscDualSpace object
17006f905325SMatthew G. Knepley 
17016f905325SMatthew G. Knepley   Output Parameters:
1702b4457527SToby Isaac + perms - Permutations of the interior degrees of freedom, parameterized by the point orientation
1703b4457527SToby Isaac - flips - Sign reversal of the interior degrees of freedom, parameterized by the point orientation
17046f905325SMatthew G. Knepley 
17056f905325SMatthew G. Knepley   Note: The permutation and flip arrays are organized in the following way
17066f905325SMatthew G. Knepley $ perms[p][ornt][dof # on point] = new local dof #
17076f905325SMatthew G. Knepley $ flips[p][ornt][dof # on point] = reversal or not
17086f905325SMatthew G. Knepley 
17096f905325SMatthew G. Knepley   Level: developer
17106f905325SMatthew G. Knepley 
17116f905325SMatthew G. Knepley @*/
17126f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceGetSymmetries(PetscDualSpace sp, const PetscInt ****perms, const PetscScalar ****flips)
17136f905325SMatthew G. Knepley {
17146f905325SMatthew G. Knepley   PetscFunctionBegin;
17156f905325SMatthew G. Knepley   PetscValidHeaderSpecific(sp,PETSCDUALSPACE_CLASSID,1);
17166f905325SMatthew G. Knepley   if (perms) {PetscValidPointer(perms,2); *perms = NULL;}
17176f905325SMatthew G. Knepley   if (flips) {PetscValidPointer(flips,3); *flips = NULL;}
17189566063dSJacob Faibussowitsch   if (sp->ops->getsymmetries) PetscCall((sp->ops->getsymmetries)(sp,perms,flips));
17196f905325SMatthew G. Knepley   PetscFunctionReturn(0);
17206f905325SMatthew G. Knepley }
17214bee2e38SMatthew G. Knepley 
17224bee2e38SMatthew G. Knepley /*@
1723b4457527SToby Isaac   PetscDualSpaceGetFormDegree - Get the form degree k for the k-form the describes the pushforwards/pullbacks of this
1724b4457527SToby Isaac   dual space's functionals.
1725b4457527SToby Isaac 
1726b4457527SToby Isaac   Input Parameter:
1727b4457527SToby Isaac . dsp - The PetscDualSpace
1728b4457527SToby Isaac 
1729b4457527SToby Isaac   Output Parameter:
1730b4457527SToby 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
1731b4457527SToby Isaac         in lexicographic order according to the basis of k-forms constructed from the wedge product of 1-forms.  So for example,
1732b4457527SToby 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).
1733b4457527SToby Isaac         If k < 0, this means that the degrees transform as k-forms, but are stored as (N-k) forms according to the
1734b4457527SToby Isaac         Hodge star map.  So for example if k = -2 and N = 3, this means that the degrees of freedom transform as 2-forms
1735b4457527SToby Isaac         but are stored as 1-forms.
1736b4457527SToby Isaac 
1737b4457527SToby Isaac   Level: developer
1738b4457527SToby Isaac 
1739b4457527SToby Isaac .seealso: PetscDTAltV, PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransform(), PetscDualSpaceTransformType
1740b4457527SToby Isaac @*/
1741b4457527SToby Isaac PetscErrorCode PetscDualSpaceGetFormDegree(PetscDualSpace dsp, PetscInt *k)
1742b4457527SToby Isaac {
1743b4457527SToby Isaac   PetscFunctionBeginHot;
1744b4457527SToby Isaac   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
1745dadcf809SJacob Faibussowitsch   PetscValidIntPointer(k, 2);
1746b4457527SToby Isaac   *k = dsp->k;
1747b4457527SToby Isaac   PetscFunctionReturn(0);
1748b4457527SToby Isaac }
1749b4457527SToby Isaac 
1750b4457527SToby Isaac /*@
1751b4457527SToby Isaac   PetscDualSpaceSetFormDegree - Set the form degree k for the k-form the describes the pushforwards/pullbacks of this
1752b4457527SToby Isaac   dual space's functionals.
1753b4457527SToby Isaac 
1754d8d19677SJose E. Roman   Input Parameters:
1755b4457527SToby Isaac + dsp - The PetscDualSpace
1756b4457527SToby 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
1757b4457527SToby Isaac         in lexicographic order according to the basis of k-forms constructed from the wedge product of 1-forms.  So for example,
1758b4457527SToby 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).
1759b4457527SToby Isaac         If k < 0, this means that the degrees transform as k-forms, but are stored as (N-k) forms according to the
1760b4457527SToby Isaac         Hodge star map.  So for example if k = -2 and N = 3, this means that the degrees of freedom transform as 2-forms
1761b4457527SToby Isaac         but are stored as 1-forms.
1762b4457527SToby Isaac 
1763b4457527SToby Isaac   Level: developer
1764b4457527SToby Isaac 
1765b4457527SToby Isaac .seealso: PetscDTAltV, PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransform(), PetscDualSpaceTransformType
1766b4457527SToby Isaac @*/
1767b4457527SToby Isaac PetscErrorCode PetscDualSpaceSetFormDegree(PetscDualSpace dsp, PetscInt k)
1768b4457527SToby Isaac {
1769b4457527SToby Isaac   PetscInt dim;
1770b4457527SToby Isaac 
1771b4457527SToby Isaac   PetscFunctionBeginHot;
1772b4457527SToby Isaac   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
177328b400f6SJacob Faibussowitsch   PetscCheck(!dsp->setupcalled,PetscObjectComm((PetscObject)dsp), PETSC_ERR_ARG_WRONGSTATE, "Cannot change number of components after dualspace is set up");
1774b4457527SToby Isaac   dim = dsp->dm->dim;
1775*1dca8a05SBarry 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);
1776b4457527SToby Isaac   dsp->k = k;
1777b4457527SToby Isaac   PetscFunctionReturn(0);
1778b4457527SToby Isaac }
1779b4457527SToby Isaac 
1780b4457527SToby Isaac /*@
17814bee2e38SMatthew G. Knepley   PetscDualSpaceGetDeRahm - Get the k-simplex associated with the functionals in this dual space
17824bee2e38SMatthew G. Knepley 
17834bee2e38SMatthew G. Knepley   Input Parameter:
17844bee2e38SMatthew G. Knepley . dsp - The PetscDualSpace
17854bee2e38SMatthew G. Knepley 
17864bee2e38SMatthew G. Knepley   Output Parameter:
17874bee2e38SMatthew G. Knepley . k   - The simplex dimension
17884bee2e38SMatthew G. Knepley 
1789a4ce7ad1SMatthew G. Knepley   Level: developer
17904bee2e38SMatthew G. Knepley 
17914bee2e38SMatthew G. Knepley   Note: Currently supported values are
17924bee2e38SMatthew G. Knepley $ 0: These are H_1 methods that only transform coordinates
17934bee2e38SMatthew G. Knepley $ 1: These are Hcurl methods that transform functions using the covariant Piola transform (COVARIANT_PIOLA_TRANSFORM)
17944bee2e38SMatthew G. Knepley $ 2: These are the same as 1
17954bee2e38SMatthew G. Knepley $ 3: These are Hdiv methods that transform functions using the contravariant Piola transform (CONTRAVARIANT_PIOLA_TRANSFORM)
17964bee2e38SMatthew G. Knepley 
17974bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransform(), PetscDualSpaceTransformType
17984bee2e38SMatthew G. Knepley @*/
17994bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceGetDeRahm(PetscDualSpace dsp, PetscInt *k)
18004bee2e38SMatthew G. Knepley {
1801b4457527SToby Isaac   PetscInt dim;
1802b4457527SToby Isaac 
18034bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
18044bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
1805dadcf809SJacob Faibussowitsch   PetscValidIntPointer(k, 2);
1806b4457527SToby Isaac   dim = dsp->dm->dim;
1807b4457527SToby Isaac   if (!dsp->k) *k = IDENTITY_TRANSFORM;
1808b4457527SToby Isaac   else if (dsp->k == 1) *k = COVARIANT_PIOLA_TRANSFORM;
1809b4457527SToby Isaac   else if (dsp->k == -(dim - 1)) *k = CONTRAVARIANT_PIOLA_TRANSFORM;
1810b4457527SToby Isaac   else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unsupported transformation");
18114bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
18124bee2e38SMatthew G. Knepley }
18134bee2e38SMatthew G. Knepley 
18144bee2e38SMatthew G. Knepley /*@C
18154bee2e38SMatthew G. Knepley   PetscDualSpaceTransform - Transform the function values
18164bee2e38SMatthew G. Knepley 
18174bee2e38SMatthew G. Knepley   Input Parameters:
18184bee2e38SMatthew G. Knepley + dsp       - The PetscDualSpace
18194bee2e38SMatthew G. Knepley . trans     - The type of transform
18204bee2e38SMatthew G. Knepley . isInverse - Flag to invert the transform
18214bee2e38SMatthew G. Knepley . fegeom    - The cell geometry
18224bee2e38SMatthew G. Knepley . Nv        - The number of function samples
18234bee2e38SMatthew G. Knepley . Nc        - The number of function components
18244bee2e38SMatthew G. Knepley - vals      - The function values
18254bee2e38SMatthew G. Knepley 
18264bee2e38SMatthew G. Knepley   Output Parameter:
18274bee2e38SMatthew G. Knepley . vals      - The transformed function values
18284bee2e38SMatthew G. Knepley 
1829a4ce7ad1SMatthew G. Knepley   Level: intermediate
18304bee2e38SMatthew G. Knepley 
1831f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
18322edcad52SToby Isaac 
1833f9244615SMatthew G. Knepley .seealso: PetscDualSpaceTransformGradient(), PetscDualSpaceTransformHessian(), PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransformType
18344bee2e38SMatthew G. Knepley @*/
18354bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransform(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[])
18364bee2e38SMatthew G. Knepley {
1837b4457527SToby Isaac   PetscReal Jstar[9] = {0};
1838b4457527SToby Isaac   PetscInt dim, v, c, Nk;
18394bee2e38SMatthew G. Knepley 
18404bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
18414bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
18424bee2e38SMatthew G. Knepley   PetscValidPointer(fegeom, 4);
1843dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(vals, 7);
1844b4457527SToby Isaac   /* TODO: not handling dimEmbed != dim right now */
18452ae266adSMatthew G. Knepley   dim = dsp->dm->dim;
1846b4457527SToby Isaac   /* No change needed for 0-forms */
1847b4457527SToby Isaac   if (!dsp->k) PetscFunctionReturn(0);
18489566063dSJacob Faibussowitsch   PetscCall(PetscDTBinomialInt(dim, PetscAbsInt(dsp->k), &Nk));
1849b4457527SToby Isaac   /* TODO: use fegeom->isAffine */
18509566063dSJacob Faibussowitsch   PetscCall(PetscDTAltVPullbackMatrix(dim, dim, isInverse ? fegeom->J : fegeom->invJ, dsp->k, Jstar));
18514bee2e38SMatthew G. Knepley   for (v = 0; v < Nv; ++v) {
1852b4457527SToby Isaac     switch (Nk) {
1853b4457527SToby Isaac     case 1:
1854b4457527SToby Isaac       for (c = 0; c < Nc; c++) vals[v*Nc + c] *= Jstar[0];
18554bee2e38SMatthew G. Knepley       break;
1856b4457527SToby Isaac     case 2:
1857b4457527SToby Isaac       for (c = 0; c < Nc; c += 2) DMPlex_Mult2DReal_Internal(Jstar, 1, &vals[v*Nc + c], &vals[v*Nc + c]);
18584bee2e38SMatthew G. Knepley       break;
1859b4457527SToby Isaac     case 3:
1860b4457527SToby Isaac       for (c = 0; c < Nc; c += 3) DMPlex_Mult3DReal_Internal(Jstar, 1, &vals[v*Nc + c], &vals[v*Nc + c]);
1861b4457527SToby Isaac       break;
186263a3b9bcSJacob Faibussowitsch     default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported form size %" PetscInt_FMT " for transformation", Nk);
1863b4457527SToby Isaac     }
18644bee2e38SMatthew G. Knepley   }
18654bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
18664bee2e38SMatthew G. Knepley }
1867b4457527SToby Isaac 
18684bee2e38SMatthew G. Knepley /*@C
18694bee2e38SMatthew G. Knepley   PetscDualSpaceTransformGradient - Transform the function gradient values
18704bee2e38SMatthew G. Knepley 
18714bee2e38SMatthew G. Knepley   Input Parameters:
18724bee2e38SMatthew G. Knepley + dsp       - The PetscDualSpace
18734bee2e38SMatthew G. Knepley . trans     - The type of transform
18744bee2e38SMatthew G. Knepley . isInverse - Flag to invert the transform
18754bee2e38SMatthew G. Knepley . fegeom    - The cell geometry
18764bee2e38SMatthew G. Knepley . Nv        - The number of function gradient samples
18774bee2e38SMatthew G. Knepley . Nc        - The number of function components
18784bee2e38SMatthew G. Knepley - vals      - The function gradient values
18794bee2e38SMatthew G. Knepley 
18804bee2e38SMatthew G. Knepley   Output Parameter:
1881f9244615SMatthew G. Knepley . vals      - The transformed function gradient values
18824bee2e38SMatthew G. Knepley 
1883a4ce7ad1SMatthew G. Knepley   Level: intermediate
18844bee2e38SMatthew G. Knepley 
1885f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
18862edcad52SToby Isaac 
1887625e0966SMatthew G. Knepley .seealso: PetscDualSpaceTransform(), PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransformType
18884bee2e38SMatthew G. Knepley @*/
18894bee2e38SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransformGradient(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[])
18904bee2e38SMatthew G. Knepley {
189127f02ce8SMatthew G. Knepley   const PetscInt dim = dsp->dm->dim, dE = fegeom->dimEmbed;
189227f02ce8SMatthew G. Knepley   PetscInt       v, c, d;
18934bee2e38SMatthew G. Knepley 
18944bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
18954bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
18964bee2e38SMatthew G. Knepley   PetscValidPointer(fegeom, 4);
1897dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(vals, 7);
189827f02ce8SMatthew G. Knepley #ifdef PETSC_USE_DEBUG
189963a3b9bcSJacob Faibussowitsch   PetscCheck(dE > 0,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid embedding dimension %" PetscInt_FMT, dE);
190027f02ce8SMatthew G. Knepley #endif
19014bee2e38SMatthew G. Knepley   /* Transform gradient */
190227f02ce8SMatthew G. Knepley   if (dim == dE) {
19034bee2e38SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
19044bee2e38SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
19054bee2e38SMatthew G. Knepley         switch (dim)
19064bee2e38SMatthew G. Knepley         {
1907100a78e1SStefano Zampini           case 1: vals[(v*Nc+c)*dim] *= fegeom->invJ[0];break;
19086142fa51SMatthew G. Knepley           case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->invJ, 1, &vals[(v*Nc+c)*dim], &vals[(v*Nc+c)*dim]);break;
19096142fa51SMatthew G. Knepley           case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->invJ, 1, &vals[(v*Nc+c)*dim], &vals[(v*Nc+c)*dim]);break;
191063a3b9bcSJacob Faibussowitsch           default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
19114bee2e38SMatthew G. Knepley         }
19124bee2e38SMatthew G. Knepley       }
19134bee2e38SMatthew G. Knepley     }
191427f02ce8SMatthew G. Knepley   } else {
191527f02ce8SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
191627f02ce8SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
191727f02ce8SMatthew G. Knepley         DMPlex_MultTransposeReal_Internal(fegeom->invJ, dim, dE, 1, &vals[(v*Nc+c)*dE], &vals[(v*Nc+c)*dE]);
191827f02ce8SMatthew G. Knepley       }
191927f02ce8SMatthew G. Knepley     }
192027f02ce8SMatthew G. Knepley   }
19214bee2e38SMatthew G. Knepley   /* Assume its a vector, otherwise assume its a bunch of scalars */
19224bee2e38SMatthew G. Knepley   if (Nc == 1 || Nc != dim) PetscFunctionReturn(0);
19234bee2e38SMatthew G. Knepley   switch (trans) {
19244bee2e38SMatthew G. Knepley     case IDENTITY_TRANSFORM: break;
19254bee2e38SMatthew G. Knepley     case COVARIANT_PIOLA_TRANSFORM: /* Covariant Piola mapping $\sigma^*(F) = J^{-T} F \circ \phi^{-1)$ */
19264bee2e38SMatthew G. Knepley     if (isInverse) {
19274bee2e38SMatthew G. Knepley       for (v = 0; v < Nv; ++v) {
19284bee2e38SMatthew G. Knepley         for (d = 0; d < dim; ++d) {
19294bee2e38SMatthew G. Knepley           switch (dim)
19304bee2e38SMatthew G. Knepley           {
19316142fa51SMatthew G. Knepley             case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
19326142fa51SMatthew G. Knepley             case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
193363a3b9bcSJacob Faibussowitsch             default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
19344bee2e38SMatthew G. Knepley           }
19354bee2e38SMatthew G. Knepley         }
19364bee2e38SMatthew G. Knepley       }
19374bee2e38SMatthew G. Knepley     } else {
19384bee2e38SMatthew G. Knepley       for (v = 0; v < Nv; ++v) {
19394bee2e38SMatthew G. Knepley         for (d = 0; d < dim; ++d) {
19404bee2e38SMatthew G. Knepley           switch (dim)
19414bee2e38SMatthew G. Knepley           {
19426142fa51SMatthew G. Knepley             case 2: DMPlex_MultTranspose2DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
19436142fa51SMatthew G. Knepley             case 3: DMPlex_MultTranspose3DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
194463a3b9bcSJacob Faibussowitsch             default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
19454bee2e38SMatthew G. Knepley           }
19464bee2e38SMatthew G. Knepley         }
19474bee2e38SMatthew G. Knepley       }
19484bee2e38SMatthew G. Knepley     }
19494bee2e38SMatthew G. Knepley     break;
19504bee2e38SMatthew G. Knepley     case CONTRAVARIANT_PIOLA_TRANSFORM: /* Contravariant Piola mapping $\sigma^*(F) = \frac{1}{|\det J|} J F \circ \phi^{-1}$ */
19514bee2e38SMatthew G. Knepley     if (isInverse) {
19524bee2e38SMatthew G. Knepley       for (v = 0; v < Nv; ++v) {
19534bee2e38SMatthew G. Knepley         for (d = 0; d < dim; ++d) {
19544bee2e38SMatthew G. Knepley           switch (dim)
19554bee2e38SMatthew G. Knepley           {
19566142fa51SMatthew G. Knepley             case 2: DMPlex_Mult2DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
19576142fa51SMatthew G. Knepley             case 3: DMPlex_Mult3DReal_Internal(fegeom->invJ, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
195863a3b9bcSJacob Faibussowitsch             default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
19594bee2e38SMatthew G. Knepley           }
19604bee2e38SMatthew G. Knepley           for (c = 0; c < Nc; ++c) vals[(v*Nc+c)*dim+d] *= fegeom->detJ[0];
19614bee2e38SMatthew G. Knepley         }
19624bee2e38SMatthew G. Knepley       }
19634bee2e38SMatthew G. Knepley     } else {
19644bee2e38SMatthew G. Knepley       for (v = 0; v < Nv; ++v) {
19654bee2e38SMatthew G. Knepley         for (d = 0; d < dim; ++d) {
19664bee2e38SMatthew G. Knepley           switch (dim)
19674bee2e38SMatthew G. Knepley           {
19686142fa51SMatthew G. Knepley             case 2: DMPlex_Mult2DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
19696142fa51SMatthew G. Knepley             case 3: DMPlex_Mult3DReal_Internal(fegeom->J, dim, &vals[v*Nc*dim+d], &vals[v*Nc*dim+d]);break;
197063a3b9bcSJacob Faibussowitsch             default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
19714bee2e38SMatthew G. Knepley           }
19724bee2e38SMatthew G. Knepley           for (c = 0; c < Nc; ++c) vals[(v*Nc+c)*dim+d] /= fegeom->detJ[0];
19734bee2e38SMatthew G. Knepley         }
19744bee2e38SMatthew G. Knepley       }
19754bee2e38SMatthew G. Knepley     }
19764bee2e38SMatthew G. Knepley     break;
19774bee2e38SMatthew G. Knepley   }
19784bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
19794bee2e38SMatthew G. Knepley }
19804bee2e38SMatthew G. Knepley 
19814bee2e38SMatthew G. Knepley /*@C
1982f9244615SMatthew G. Knepley   PetscDualSpaceTransformHessian - Transform the function Hessian values
1983f9244615SMatthew G. Knepley 
1984f9244615SMatthew G. Knepley   Input Parameters:
1985f9244615SMatthew G. Knepley + dsp       - The PetscDualSpace
1986f9244615SMatthew G. Knepley . trans     - The type of transform
1987f9244615SMatthew G. Knepley . isInverse - Flag to invert the transform
1988f9244615SMatthew G. Knepley . fegeom    - The cell geometry
1989f9244615SMatthew G. Knepley . Nv        - The number of function Hessian samples
1990f9244615SMatthew G. Knepley . Nc        - The number of function components
1991f9244615SMatthew G. Knepley - vals      - The function gradient values
1992f9244615SMatthew G. Knepley 
1993f9244615SMatthew G. Knepley   Output Parameter:
1994f9244615SMatthew G. Knepley . vals      - The transformed function Hessian values
1995f9244615SMatthew G. Knepley 
1996f9244615SMatthew G. Knepley   Level: intermediate
1997f9244615SMatthew G. Knepley 
1998f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
1999f9244615SMatthew G. Knepley 
2000f9244615SMatthew G. Knepley .seealso: PetscDualSpaceTransform(), PetscDualSpacePullback(), PetscDualSpacePushforward(), PetscDualSpaceTransformType
2001f9244615SMatthew G. Knepley @*/
2002f9244615SMatthew G. Knepley PetscErrorCode PetscDualSpaceTransformHessian(PetscDualSpace dsp, PetscDualSpaceTransformType trans, PetscBool isInverse, PetscFEGeom *fegeom, PetscInt Nv, PetscInt Nc, PetscScalar vals[])
2003f9244615SMatthew G. Knepley {
2004f9244615SMatthew G. Knepley   const PetscInt dim = dsp->dm->dim, dE = fegeom->dimEmbed;
2005f9244615SMatthew G. Knepley   PetscInt       v, c;
2006f9244615SMatthew G. Knepley 
2007f9244615SMatthew G. Knepley   PetscFunctionBeginHot;
2008f9244615SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
2009f9244615SMatthew G. Knepley   PetscValidPointer(fegeom, 4);
2010dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(vals, 7);
2011f9244615SMatthew G. Knepley #ifdef PETSC_USE_DEBUG
201263a3b9bcSJacob Faibussowitsch   PetscCheck(dE > 0,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid embedding dimension %" PetscInt_FMT, dE);
2013f9244615SMatthew G. Knepley #endif
2014f9244615SMatthew G. Knepley   /* Transform Hessian: J^{-T}_{ik} J^{-T}_{jl} H(f)_{kl} = J^{-T}_{ik} H(f)_{kl} J^{-1}_{lj} */
2015f9244615SMatthew G. Knepley   if (dim == dE) {
2016f9244615SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
2017f9244615SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
2018f9244615SMatthew G. Knepley         switch (dim)
2019f9244615SMatthew G. Knepley         {
2020f9244615SMatthew G. Knepley           case 1: vals[(v*Nc+c)*dim*dim] *= PetscSqr(fegeom->invJ[0]);break;
2021f9244615SMatthew G. Knepley           case 2: DMPlex_PTAP2DReal_Internal(fegeom->invJ, &vals[(v*Nc+c)*dim*dim], &vals[(v*Nc+c)*dim*dim]);break;
2022f9244615SMatthew G. Knepley           case 3: DMPlex_PTAP3DReal_Internal(fegeom->invJ, &vals[(v*Nc+c)*dim*dim], &vals[(v*Nc+c)*dim*dim]);break;
202363a3b9bcSJacob Faibussowitsch           default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported dim %" PetscInt_FMT " for transformation", dim);
2024f9244615SMatthew G. Knepley         }
2025f9244615SMatthew G. Knepley       }
2026f9244615SMatthew G. Knepley     }
2027f9244615SMatthew G. Knepley   } else {
2028f9244615SMatthew G. Knepley     for (v = 0; v < Nv; ++v) {
2029f9244615SMatthew G. Knepley       for (c = 0; c < Nc; ++c) {
2030f9244615SMatthew G. Knepley         DMPlex_PTAPReal_Internal(fegeom->invJ, dim, dE, &vals[(v*Nc+c)*dE*dE], &vals[(v*Nc+c)*dE*dE]);
2031f9244615SMatthew G. Knepley       }
2032f9244615SMatthew G. Knepley     }
2033f9244615SMatthew G. Knepley   }
2034f9244615SMatthew G. Knepley   /* Assume its a vector, otherwise assume its a bunch of scalars */
2035f9244615SMatthew G. Knepley   if (Nc == 1 || Nc != dim) PetscFunctionReturn(0);
2036f9244615SMatthew G. Knepley   switch (trans) {
2037f9244615SMatthew G. Knepley     case IDENTITY_TRANSFORM: break;
2038f9244615SMatthew G. Knepley     case COVARIANT_PIOLA_TRANSFORM: /* Covariant Piola mapping $\sigma^*(F) = J^{-T} F \circ \phi^{-1)$ */
2039f9244615SMatthew G. Knepley     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Piola mapping for Hessians not yet supported");
2040f9244615SMatthew G. Knepley     case CONTRAVARIANT_PIOLA_TRANSFORM: /* Contravariant Piola mapping $\sigma^*(F) = \frac{1}{|\det J|} J F \circ \phi^{-1}$ */
2041f9244615SMatthew G. Knepley     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Piola mapping for Hessians not yet supported");
2042f9244615SMatthew G. Knepley   }
2043f9244615SMatthew G. Knepley   PetscFunctionReturn(0);
2044f9244615SMatthew G. Knepley }
2045f9244615SMatthew G. Knepley 
2046f9244615SMatthew G. Knepley /*@C
20474bee2e38SMatthew 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.
20484bee2e38SMatthew G. Knepley 
20494bee2e38SMatthew G. Knepley   Input Parameters:
20504bee2e38SMatthew G. Knepley + dsp        - The PetscDualSpace
20514bee2e38SMatthew G. Knepley . fegeom     - The geometry for this cell
20524bee2e38SMatthew G. Knepley . Nq         - The number of function samples
20534bee2e38SMatthew G. Knepley . Nc         - The number of function components
20544bee2e38SMatthew G. Knepley - pointEval  - The function values
20554bee2e38SMatthew G. Knepley 
20564bee2e38SMatthew G. Knepley   Output Parameter:
20574bee2e38SMatthew G. Knepley . pointEval  - The transformed function values
20584bee2e38SMatthew G. Knepley 
20594bee2e38SMatthew G. Knepley   Level: advanced
20604bee2e38SMatthew G. Knepley 
20614bee2e38SMatthew 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.
20624bee2e38SMatthew G. Knepley 
20632edcad52SToby Isaac   Note: This only handles tranformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
20642edcad52SToby Isaac 
20654bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePushforward(), PetscDualSpaceTransform(), PetscDualSpaceGetDeRahm()
20664bee2e38SMatthew G. Knepley @*/
20672edcad52SToby Isaac PetscErrorCode PetscDualSpacePullback(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[])
20684bee2e38SMatthew G. Knepley {
20694bee2e38SMatthew G. Knepley   PetscDualSpaceTransformType trans;
2070b4457527SToby Isaac   PetscInt                    k;
20714bee2e38SMatthew G. Knepley 
20724bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
20734bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
20744bee2e38SMatthew G. Knepley   PetscValidPointer(fegeom, 2);
2075dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(pointEval, 5);
20764bee2e38SMatthew G. Knepley   /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k.
20774bee2e38SMatthew G. Knepley      This determines their transformation properties. */
20789566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDeRahm(dsp, &k));
2079b4457527SToby Isaac   switch (k)
20804bee2e38SMatthew G. Knepley   {
20814bee2e38SMatthew G. Knepley     case 0: /* H^1 point evaluations */
20824bee2e38SMatthew G. Knepley     trans = IDENTITY_TRANSFORM;break;
20834bee2e38SMatthew G. Knepley     case 1: /* Hcurl preserves tangential edge traces  */
20844bee2e38SMatthew G. Knepley     trans = COVARIANT_PIOLA_TRANSFORM;break;
2085b4457527SToby Isaac     case 2:
20864bee2e38SMatthew G. Knepley     case 3: /* Hdiv preserve normal traces */
20874bee2e38SMatthew G. Knepley     trans = CONTRAVARIANT_PIOLA_TRANSFORM;break;
208863a3b9bcSJacob Faibussowitsch     default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k);
20894bee2e38SMatthew G. Knepley   }
20909566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceTransform(dsp, trans, PETSC_TRUE, fegeom, Nq, Nc, pointEval));
20914bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
20924bee2e38SMatthew G. Knepley }
20934bee2e38SMatthew G. Knepley 
20944bee2e38SMatthew G. Knepley /*@C
20954bee2e38SMatthew 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.
20964bee2e38SMatthew G. Knepley 
20974bee2e38SMatthew G. Knepley   Input Parameters:
20984bee2e38SMatthew G. Knepley + dsp        - The PetscDualSpace
20994bee2e38SMatthew G. Knepley . fegeom     - The geometry for this cell
21004bee2e38SMatthew G. Knepley . Nq         - The number of function samples
21014bee2e38SMatthew G. Knepley . Nc         - The number of function components
21024bee2e38SMatthew G. Knepley - pointEval  - The function values
21034bee2e38SMatthew G. Knepley 
21044bee2e38SMatthew G. Knepley   Output Parameter:
21054bee2e38SMatthew G. Knepley . pointEval  - The transformed function values
21064bee2e38SMatthew G. Knepley 
21074bee2e38SMatthew G. Knepley   Level: advanced
21084bee2e38SMatthew G. Knepley 
21094bee2e38SMatthew 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.
21104bee2e38SMatthew G. Knepley 
2111f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
21122edcad52SToby Isaac 
21134bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePullback(), PetscDualSpaceTransform(), PetscDualSpaceGetDeRahm()
21144bee2e38SMatthew G. Knepley @*/
21152edcad52SToby Isaac PetscErrorCode PetscDualSpacePushforward(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[])
21164bee2e38SMatthew G. Knepley {
21174bee2e38SMatthew G. Knepley   PetscDualSpaceTransformType trans;
2118b4457527SToby Isaac   PetscInt                    k;
21194bee2e38SMatthew G. Knepley 
21204bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
21214bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
21224bee2e38SMatthew G. Knepley   PetscValidPointer(fegeom, 2);
2123dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(pointEval, 5);
21244bee2e38SMatthew G. Knepley   /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k.
21254bee2e38SMatthew G. Knepley      This determines their transformation properties. */
21269566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDeRahm(dsp, &k));
2127b4457527SToby Isaac   switch (k)
21284bee2e38SMatthew G. Knepley   {
21294bee2e38SMatthew G. Knepley     case 0: /* H^1 point evaluations */
21304bee2e38SMatthew G. Knepley     trans = IDENTITY_TRANSFORM;break;
21314bee2e38SMatthew G. Knepley     case 1: /* Hcurl preserves tangential edge traces  */
21324bee2e38SMatthew G. Knepley     trans = COVARIANT_PIOLA_TRANSFORM;break;
2133b4457527SToby Isaac     case 2:
21344bee2e38SMatthew G. Knepley     case 3: /* Hdiv preserve normal traces */
21354bee2e38SMatthew G. Knepley     trans = CONTRAVARIANT_PIOLA_TRANSFORM;break;
213663a3b9bcSJacob Faibussowitsch     default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k);
21374bee2e38SMatthew G. Knepley   }
21389566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceTransform(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval));
21394bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
21404bee2e38SMatthew G. Knepley }
21414bee2e38SMatthew G. Knepley 
21424bee2e38SMatthew G. Knepley /*@C
21434bee2e38SMatthew 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.
21444bee2e38SMatthew G. Knepley 
21454bee2e38SMatthew G. Knepley   Input Parameters:
21464bee2e38SMatthew G. Knepley + dsp        - The PetscDualSpace
21474bee2e38SMatthew G. Knepley . fegeom     - The geometry for this cell
21484bee2e38SMatthew G. Knepley . Nq         - The number of function gradient samples
21494bee2e38SMatthew G. Knepley . Nc         - The number of function components
21504bee2e38SMatthew G. Knepley - pointEval  - The function gradient values
21514bee2e38SMatthew G. Knepley 
21524bee2e38SMatthew G. Knepley   Output Parameter:
21534bee2e38SMatthew G. Knepley . pointEval  - The transformed function gradient values
21544bee2e38SMatthew G. Knepley 
21554bee2e38SMatthew G. Knepley   Level: advanced
21564bee2e38SMatthew G. Knepley 
21574bee2e38SMatthew 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.
21584bee2e38SMatthew G. Knepley 
2159f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
21602edcad52SToby Isaac 
21614bee2e38SMatthew G. Knepley .seealso: PetscDualSpacePushforward(), PPetscDualSpacePullback(), PetscDualSpaceTransform(), PetscDualSpaceGetDeRahm()
2162dc0529c6SBarry Smith @*/
21632edcad52SToby Isaac PetscErrorCode PetscDualSpacePushforwardGradient(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[])
21644bee2e38SMatthew G. Knepley {
21654bee2e38SMatthew G. Knepley   PetscDualSpaceTransformType trans;
2166b4457527SToby Isaac   PetscInt                    k;
21674bee2e38SMatthew G. Knepley 
21684bee2e38SMatthew G. Knepley   PetscFunctionBeginHot;
21694bee2e38SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
21704bee2e38SMatthew G. Knepley   PetscValidPointer(fegeom, 2);
2171dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(pointEval, 5);
21724bee2e38SMatthew G. Knepley   /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k.
21734bee2e38SMatthew G. Knepley      This determines their transformation properties. */
21749566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDeRahm(dsp, &k));
2175b4457527SToby Isaac   switch (k)
21764bee2e38SMatthew G. Knepley   {
21774bee2e38SMatthew G. Knepley     case 0: /* H^1 point evaluations */
21784bee2e38SMatthew G. Knepley     trans = IDENTITY_TRANSFORM;break;
21794bee2e38SMatthew G. Knepley     case 1: /* Hcurl preserves tangential edge traces  */
21804bee2e38SMatthew G. Knepley     trans = COVARIANT_PIOLA_TRANSFORM;break;
2181b4457527SToby Isaac     case 2:
21824bee2e38SMatthew G. Knepley     case 3: /* Hdiv preserve normal traces */
21834bee2e38SMatthew G. Knepley     trans = CONTRAVARIANT_PIOLA_TRANSFORM;break;
218463a3b9bcSJacob Faibussowitsch     default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k);
21854bee2e38SMatthew G. Knepley   }
21869566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceTransformGradient(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval));
21874bee2e38SMatthew G. Knepley   PetscFunctionReturn(0);
21884bee2e38SMatthew G. Knepley }
2189f9244615SMatthew G. Knepley 
2190f9244615SMatthew G. Knepley /*@C
2191f9244615SMatthew 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.
2192f9244615SMatthew G. Knepley 
2193f9244615SMatthew G. Knepley   Input Parameters:
2194f9244615SMatthew G. Knepley + dsp        - The PetscDualSpace
2195f9244615SMatthew G. Knepley . fegeom     - The geometry for this cell
2196f9244615SMatthew G. Knepley . Nq         - The number of function Hessian samples
2197f9244615SMatthew G. Knepley . Nc         - The number of function components
2198f9244615SMatthew G. Knepley - pointEval  - The function gradient values
2199f9244615SMatthew G. Knepley 
2200f9244615SMatthew G. Knepley   Output Parameter:
2201f9244615SMatthew G. Knepley . pointEval  - The transformed function Hessian values
2202f9244615SMatthew G. Knepley 
2203f9244615SMatthew G. Knepley   Level: advanced
2204f9244615SMatthew G. Knepley 
2205f9244615SMatthew 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.
2206f9244615SMatthew G. Knepley 
2207f9244615SMatthew G. Knepley   Note: This only handles transformations when the embedding dimension of the geometry in fegeom is the same as the reference dimension.
2208f9244615SMatthew G. Knepley 
2209f9244615SMatthew G. Knepley .seealso: PetscDualSpacePushforward(), PPetscDualSpacePullback(), PetscDualSpaceTransform(), PetscDualSpaceGetDeRahm()
2210f9244615SMatthew G. Knepley @*/
2211f9244615SMatthew G. Knepley PetscErrorCode PetscDualSpacePushforwardHessian(PetscDualSpace dsp, PetscFEGeom *fegeom, PetscInt Nq, PetscInt Nc, PetscScalar pointEval[])
2212f9244615SMatthew G. Knepley {
2213f9244615SMatthew G. Knepley   PetscDualSpaceTransformType trans;
2214f9244615SMatthew G. Knepley   PetscInt                    k;
2215f9244615SMatthew G. Knepley 
2216f9244615SMatthew G. Knepley   PetscFunctionBeginHot;
2217f9244615SMatthew G. Knepley   PetscValidHeaderSpecific(dsp, PETSCDUALSPACE_CLASSID, 1);
2218f9244615SMatthew G. Knepley   PetscValidPointer(fegeom, 2);
2219dadcf809SJacob Faibussowitsch   PetscValidScalarPointer(pointEval, 5);
2220f9244615SMatthew G. Knepley   /* The dualspace dofs correspond to some simplex in the DeRahm complex, which we label by k.
2221f9244615SMatthew G. Knepley      This determines their transformation properties. */
22229566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDeRahm(dsp, &k));
2223f9244615SMatthew G. Knepley   switch (k)
2224f9244615SMatthew G. Knepley   {
2225f9244615SMatthew G. Knepley     case 0: /* H^1 point evaluations */
2226f9244615SMatthew G. Knepley     trans = IDENTITY_TRANSFORM;break;
2227f9244615SMatthew G. Knepley     case 1: /* Hcurl preserves tangential edge traces  */
2228f9244615SMatthew G. Knepley     trans = COVARIANT_PIOLA_TRANSFORM;break;
2229f9244615SMatthew G. Knepley     case 2:
2230f9244615SMatthew G. Knepley     case 3: /* Hdiv preserve normal traces */
2231f9244615SMatthew G. Knepley     trans = CONTRAVARIANT_PIOLA_TRANSFORM;break;
223263a3b9bcSJacob Faibussowitsch     default: SETERRQ(PetscObjectComm((PetscObject) dsp), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported simplex dim %" PetscInt_FMT " for transformation", k);
2233f9244615SMatthew G. Knepley   }
22349566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceTransformHessian(dsp, trans, PETSC_FALSE, fegeom, Nq, Nc, pointEval));
2235f9244615SMatthew G. Knepley   PetscFunctionReturn(0);
2236f9244615SMatthew G. Knepley }
2237