xref: /petsc/src/dm/dt/interface/dtds.c (revision 0d3e9b515e31516aa8181fd0cd44d7f8716b7dc2)
1af0996ceSBarry Smith #include <petsc/private/petscdsimpl.h> /*I "petscds.h" I*/
22764a2aaSMatthew G. Knepley 
32764a2aaSMatthew G. Knepley PetscClassId PETSCDS_CLASSID = 0;
42764a2aaSMatthew G. Knepley 
52764a2aaSMatthew G. Knepley PetscFunctionList PetscDSList              = NULL;
62764a2aaSMatthew G. Knepley PetscBool         PetscDSRegisterAllCalled = PETSC_FALSE;
72764a2aaSMatthew G. Knepley 
82764a2aaSMatthew G. Knepley /*@C
92764a2aaSMatthew G. Knepley   PetscDSRegister - Adds a new PetscDS implementation
102764a2aaSMatthew G. Knepley 
112764a2aaSMatthew G. Knepley   Not Collective
122764a2aaSMatthew G. Knepley 
132764a2aaSMatthew G. Knepley   Input Parameters:
142764a2aaSMatthew G. Knepley + name        - The name of a new user-defined creation routine
152764a2aaSMatthew G. Knepley - create_func - The creation routine itself
162764a2aaSMatthew G. Knepley 
172764a2aaSMatthew G. Knepley   Notes:
182764a2aaSMatthew G. Knepley   PetscDSRegister() may be called multiple times to add several user-defined PetscDSs
192764a2aaSMatthew G. Knepley 
202764a2aaSMatthew G. Knepley   Sample usage:
212764a2aaSMatthew G. Knepley .vb
222764a2aaSMatthew G. Knepley     PetscDSRegister("my_ds", MyPetscDSCreate);
232764a2aaSMatthew G. Knepley .ve
242764a2aaSMatthew G. Knepley 
252764a2aaSMatthew G. Knepley   Then, your PetscDS type can be chosen with the procedural interface via
262764a2aaSMatthew G. Knepley .vb
272764a2aaSMatthew G. Knepley     PetscDSCreate(MPI_Comm, PetscDS *);
282764a2aaSMatthew G. Knepley     PetscDSSetType(PetscDS, "my_ds");
292764a2aaSMatthew G. Knepley .ve
302764a2aaSMatthew G. Knepley    or at runtime via the option
312764a2aaSMatthew G. Knepley .vb
322764a2aaSMatthew G. Knepley     -petscds_type my_ds
332764a2aaSMatthew G. Knepley .ve
342764a2aaSMatthew G. Knepley 
352764a2aaSMatthew G. Knepley   Level: advanced
362764a2aaSMatthew G. Knepley 
372764a2aaSMatthew G. Knepley .keywords: PetscDS, register
382764a2aaSMatthew G. Knepley .seealso: PetscDSRegisterAll(), PetscDSRegisterDestroy()
392764a2aaSMatthew G. Knepley 
402764a2aaSMatthew G. Knepley @*/
412764a2aaSMatthew G. Knepley PetscErrorCode PetscDSRegister(const char sname[], PetscErrorCode (*function)(PetscDS))
422764a2aaSMatthew G. Knepley {
432764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
442764a2aaSMatthew G. Knepley 
452764a2aaSMatthew G. Knepley   PetscFunctionBegin;
462764a2aaSMatthew G. Knepley   ierr = PetscFunctionListAdd(&PetscDSList, sname, function);CHKERRQ(ierr);
472764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
482764a2aaSMatthew G. Knepley }
492764a2aaSMatthew G. Knepley 
502764a2aaSMatthew G. Knepley /*@C
512764a2aaSMatthew G. Knepley   PetscDSSetType - Builds a particular PetscDS
522764a2aaSMatthew G. Knepley 
532764a2aaSMatthew G. Knepley   Collective on PetscDS
542764a2aaSMatthew G. Knepley 
552764a2aaSMatthew G. Knepley   Input Parameters:
562764a2aaSMatthew G. Knepley + prob - The PetscDS object
572764a2aaSMatthew G. Knepley - name - The kind of system
582764a2aaSMatthew G. Knepley 
592764a2aaSMatthew G. Knepley   Options Database Key:
602764a2aaSMatthew G. Knepley . -petscds_type <type> - Sets the PetscDS type; use -help for a list of available types
612764a2aaSMatthew G. Knepley 
622764a2aaSMatthew G. Knepley   Level: intermediate
632764a2aaSMatthew G. Knepley 
642764a2aaSMatthew G. Knepley .keywords: PetscDS, set, type
652764a2aaSMatthew G. Knepley .seealso: PetscDSGetType(), PetscDSCreate()
662764a2aaSMatthew G. Knepley @*/
672764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetType(PetscDS prob, PetscDSType name)
682764a2aaSMatthew G. Knepley {
692764a2aaSMatthew G. Knepley   PetscErrorCode (*r)(PetscDS);
702764a2aaSMatthew G. Knepley   PetscBool      match;
712764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
722764a2aaSMatthew G. Knepley 
732764a2aaSMatthew G. Knepley   PetscFunctionBegin;
742764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
752764a2aaSMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) prob, name, &match);CHKERRQ(ierr);
762764a2aaSMatthew G. Knepley   if (match) PetscFunctionReturn(0);
772764a2aaSMatthew G. Knepley 
780f51fdf8SToby Isaac   ierr = PetscDSRegisterAll();CHKERRQ(ierr);
792764a2aaSMatthew G. Knepley   ierr = PetscFunctionListFind(PetscDSList, name, &r);CHKERRQ(ierr);
802764a2aaSMatthew G. Knepley   if (!r) SETERRQ1(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown PetscDS type: %s", name);
812764a2aaSMatthew G. Knepley 
822764a2aaSMatthew G. Knepley   if (prob->ops->destroy) {
832764a2aaSMatthew G. Knepley     ierr             = (*prob->ops->destroy)(prob);CHKERRQ(ierr);
842764a2aaSMatthew G. Knepley     prob->ops->destroy = NULL;
852764a2aaSMatthew G. Knepley   }
862764a2aaSMatthew G. Knepley   ierr = (*r)(prob);CHKERRQ(ierr);
872764a2aaSMatthew G. Knepley   ierr = PetscObjectChangeTypeName((PetscObject) prob, name);CHKERRQ(ierr);
882764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
892764a2aaSMatthew G. Knepley }
902764a2aaSMatthew G. Knepley 
912764a2aaSMatthew G. Knepley /*@C
922764a2aaSMatthew G. Knepley   PetscDSGetType - Gets the PetscDS type name (as a string) from the object.
932764a2aaSMatthew G. Knepley 
942764a2aaSMatthew G. Knepley   Not Collective
952764a2aaSMatthew G. Knepley 
962764a2aaSMatthew G. Knepley   Input Parameter:
972764a2aaSMatthew G. Knepley . prob  - The PetscDS
982764a2aaSMatthew G. Knepley 
992764a2aaSMatthew G. Knepley   Output Parameter:
1002764a2aaSMatthew G. Knepley . name - The PetscDS type name
1012764a2aaSMatthew G. Knepley 
1022764a2aaSMatthew G. Knepley   Level: intermediate
1032764a2aaSMatthew G. Knepley 
1042764a2aaSMatthew G. Knepley .keywords: PetscDS, get, type, name
1052764a2aaSMatthew G. Knepley .seealso: PetscDSSetType(), PetscDSCreate()
1062764a2aaSMatthew G. Knepley @*/
1072764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetType(PetscDS prob, PetscDSType *name)
1082764a2aaSMatthew G. Knepley {
1092764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
1102764a2aaSMatthew G. Knepley 
1112764a2aaSMatthew G. Knepley   PetscFunctionBegin;
1122764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
113c959eef4SJed Brown   PetscValidPointer(name, 2);
1140f51fdf8SToby Isaac   ierr = PetscDSRegisterAll();CHKERRQ(ierr);
1152764a2aaSMatthew G. Knepley   *name = ((PetscObject) prob)->type_name;
1162764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
1172764a2aaSMatthew G. Knepley }
1182764a2aaSMatthew G. Knepley 
1197d8a60eaSMatthew G. Knepley static PetscErrorCode PetscDSView_Ascii(PetscDS prob, PetscViewer viewer)
1207d8a60eaSMatthew G. Knepley {
1217d8a60eaSMatthew G. Knepley   PetscViewerFormat  format;
12297b6e6e8SMatthew G. Knepley   const PetscScalar *constants;
12397b6e6e8SMatthew G. Knepley   PetscInt           numConstants, f;
1247d8a60eaSMatthew G. Knepley   PetscErrorCode     ierr;
1257d8a60eaSMatthew G. Knepley 
1267d8a60eaSMatthew G. Knepley   PetscFunctionBegin;
1277d8a60eaSMatthew G. Knepley   ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
1287d8a60eaSMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "Discrete System with %d fields\n", prob->Nf);CHKERRQ(ierr);
1297d8a60eaSMatthew G. Knepley   ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
1307d8a60eaSMatthew G. Knepley   for (f = 0; f < prob->Nf; ++f) {
1317d8a60eaSMatthew G. Knepley     PetscObject  obj;
1327d8a60eaSMatthew G. Knepley     PetscClassId id;
1337d8a60eaSMatthew G. Knepley     const char  *name;
1347d8a60eaSMatthew G. Knepley     PetscInt     Nc;
1357d8a60eaSMatthew G. Knepley 
1367d8a60eaSMatthew G. Knepley     ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr);
1377d8a60eaSMatthew G. Knepley     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
1387d8a60eaSMatthew G. Knepley     ierr = PetscObjectGetName(obj, &name);CHKERRQ(ierr);
1397d8a60eaSMatthew G. Knepley     ierr = PetscViewerASCIIPrintf(viewer, "Field %s", name ? name : "<unknown>");CHKERRQ(ierr);
1407d8a60eaSMatthew G. Knepley     if (id == PETSCFE_CLASSID)      {
1417d8a60eaSMatthew G. Knepley       ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);
1427d8a60eaSMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, " FEM");CHKERRQ(ierr);
1437d8a60eaSMatthew G. Knepley     } else if (id == PETSCFV_CLASSID) {
1447d8a60eaSMatthew G. Knepley       ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);
1457d8a60eaSMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, " FVM");CHKERRQ(ierr);
1467d8a60eaSMatthew G. Knepley     }
14797b6e6e8SMatthew G. Knepley     else SETERRQ1(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", f);
14897b6e6e8SMatthew G. Knepley     if (Nc > 1) {ierr = PetscViewerASCIIPrintf(viewer, "%D components", Nc);CHKERRQ(ierr);}
14997b6e6e8SMatthew G. Knepley     else        {ierr = PetscViewerASCIIPrintf(viewer, "%D component ", Nc);CHKERRQ(ierr);}
150249df284SMatthew G. Knepley     if (prob->implicit[f]) {ierr = PetscViewerASCIIPrintf(viewer, " (implicit)");CHKERRQ(ierr);}
151249df284SMatthew G. Knepley     else                   {ierr = PetscViewerASCIIPrintf(viewer, " (explicit)");CHKERRQ(ierr);}
152a6cbbb48SMatthew G. Knepley     if (prob->adjacency[f*2+0]) {
153a6cbbb48SMatthew G. Knepley       if (prob->adjacency[f*2+1]) {ierr = PetscViewerASCIIPrintf(viewer, " (adj FVM++)");CHKERRQ(ierr);}
154a6cbbb48SMatthew G. Knepley       else                        {ierr = PetscViewerASCIIPrintf(viewer, " (adj FVM)");CHKERRQ(ierr);}
155a6cbbb48SMatthew G. Knepley     } else {
156a6cbbb48SMatthew G. Knepley       if (prob->adjacency[f*2+1]) {ierr = PetscViewerASCIIPrintf(viewer, " (adj FEM)");CHKERRQ(ierr);}
157a6cbbb48SMatthew G. Knepley       else                        {ierr = PetscViewerASCIIPrintf(viewer, " (adj FUNKY)");CHKERRQ(ierr);}
158a6cbbb48SMatthew G. Knepley     }
1597d8a60eaSMatthew G. Knepley     ierr = PetscViewerASCIIPrintf(viewer, "\n");CHKERRQ(ierr);
1607d8a60eaSMatthew G. Knepley     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
1617d8a60eaSMatthew G. Knepley       if (id == PETSCFE_CLASSID)      {ierr = PetscFEView((PetscFE) obj, viewer);CHKERRQ(ierr);}
1627d8a60eaSMatthew G. Knepley       else if (id == PETSCFV_CLASSID) {ierr = PetscFVView((PetscFV) obj, viewer);CHKERRQ(ierr);}
1637d8a60eaSMatthew G. Knepley     }
1647d8a60eaSMatthew G. Knepley   }
16597b6e6e8SMatthew G. Knepley   ierr = PetscDSGetConstants(prob, &numConstants, &constants);CHKERRQ(ierr);
16697b6e6e8SMatthew G. Knepley   if (numConstants) {
16797b6e6e8SMatthew G. Knepley     ierr = PetscViewerASCIIPrintf(viewer, "%D constants\n", numConstants);CHKERRQ(ierr);
16897b6e6e8SMatthew G. Knepley     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
16997b6e6e8SMatthew G. Knepley     for (f = 0; f < numConstants; ++f) {ierr = PetscViewerASCIIPrintf(viewer, "%g\n", constants[f]);CHKERRQ(ierr);}
17097b6e6e8SMatthew G. Knepley     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
17197b6e6e8SMatthew G. Knepley   }
1727d8a60eaSMatthew G. Knepley   ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
1737d8a60eaSMatthew G. Knepley   PetscFunctionReturn(0);
1747d8a60eaSMatthew G. Knepley }
1757d8a60eaSMatthew G. Knepley 
1762764a2aaSMatthew G. Knepley /*@C
1772764a2aaSMatthew G. Knepley   PetscDSView - Views a PetscDS
1782764a2aaSMatthew G. Knepley 
1792764a2aaSMatthew G. Knepley   Collective on PetscDS
1802764a2aaSMatthew G. Knepley 
1812764a2aaSMatthew G. Knepley   Input Parameter:
1822764a2aaSMatthew G. Knepley + prob - the PetscDS object to view
1832764a2aaSMatthew G. Knepley - v  - the viewer
1842764a2aaSMatthew G. Knepley 
1852764a2aaSMatthew G. Knepley   Level: developer
1862764a2aaSMatthew G. Knepley 
1872764a2aaSMatthew G. Knepley .seealso PetscDSDestroy()
1882764a2aaSMatthew G. Knepley @*/
1892764a2aaSMatthew G. Knepley PetscErrorCode PetscDSView(PetscDS prob, PetscViewer v)
1902764a2aaSMatthew G. Knepley {
1917d8a60eaSMatthew G. Knepley   PetscBool      iascii;
1922764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
1932764a2aaSMatthew G. Knepley 
1942764a2aaSMatthew G. Knepley   PetscFunctionBegin;
1952764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1962764a2aaSMatthew G. Knepley   if (!v) {ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject) prob), &v);CHKERRQ(ierr);}
1977d8a60eaSMatthew G. Knepley   else    {PetscValidHeaderSpecific(v, PETSC_VIEWER_CLASSID, 2);}
1987d8a60eaSMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) v, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
1997d8a60eaSMatthew G. Knepley   if (iascii) {ierr = PetscDSView_Ascii(prob, v);CHKERRQ(ierr);}
2002764a2aaSMatthew G. Knepley   if (prob->ops->view) {ierr = (*prob->ops->view)(prob, v);CHKERRQ(ierr);}
2012764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
2022764a2aaSMatthew G. Knepley }
2032764a2aaSMatthew G. Knepley 
2042764a2aaSMatthew G. Knepley /*@
2052764a2aaSMatthew G. Knepley   PetscDSSetFromOptions - sets parameters in a PetscDS from the options database
2062764a2aaSMatthew G. Knepley 
2072764a2aaSMatthew G. Knepley   Collective on PetscDS
2082764a2aaSMatthew G. Knepley 
2092764a2aaSMatthew G. Knepley   Input Parameter:
2102764a2aaSMatthew G. Knepley . prob - the PetscDS object to set options for
2112764a2aaSMatthew G. Knepley 
2122764a2aaSMatthew G. Knepley   Options Database:
2132764a2aaSMatthew G. Knepley 
2142764a2aaSMatthew G. Knepley   Level: developer
2152764a2aaSMatthew G. Knepley 
2162764a2aaSMatthew G. Knepley .seealso PetscDSView()
2172764a2aaSMatthew G. Knepley @*/
2182764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetFromOptions(PetscDS prob)
2192764a2aaSMatthew G. Knepley {
220f1fd5e65SToby Isaac   DSBoundary     b;
2212764a2aaSMatthew G. Knepley   const char    *defaultType;
2222764a2aaSMatthew G. Knepley   char           name[256];
2232764a2aaSMatthew G. Knepley   PetscBool      flg;
2242764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
2252764a2aaSMatthew G. Knepley 
2262764a2aaSMatthew G. Knepley   PetscFunctionBegin;
2272764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
2282764a2aaSMatthew G. Knepley   if (!((PetscObject) prob)->type_name) {
2292764a2aaSMatthew G. Knepley     defaultType = PETSCDSBASIC;
2302764a2aaSMatthew G. Knepley   } else {
2312764a2aaSMatthew G. Knepley     defaultType = ((PetscObject) prob)->type_name;
2322764a2aaSMatthew G. Knepley   }
2330f51fdf8SToby Isaac   ierr = PetscDSRegisterAll();CHKERRQ(ierr);
2342764a2aaSMatthew G. Knepley 
2352764a2aaSMatthew G. Knepley   ierr = PetscObjectOptionsBegin((PetscObject) prob);CHKERRQ(ierr);
236f1fd5e65SToby Isaac   for (b = prob->boundary; b; b = b->next) {
237f1fd5e65SToby Isaac     char       optname[1024];
238f1fd5e65SToby Isaac     PetscInt   ids[1024], len = 1024;
239f1fd5e65SToby Isaac     PetscBool  flg;
240f1fd5e65SToby Isaac 
241f1fd5e65SToby Isaac     ierr = PetscSNPrintf(optname, sizeof(optname), "-bc_%s", b->name);CHKERRQ(ierr);
242f1fd5e65SToby Isaac     ierr = PetscMemzero(ids, sizeof(ids));CHKERRQ(ierr);
243f1fd5e65SToby Isaac     ierr = PetscOptionsIntArray(optname, "List of boundary IDs", "", ids, &len, &flg);CHKERRQ(ierr);
244f1fd5e65SToby Isaac     if (flg) {
245f1fd5e65SToby Isaac       b->numids = len;
246f1fd5e65SToby Isaac       ierr = PetscFree(b->ids);CHKERRQ(ierr);
247f1fd5e65SToby Isaac       ierr = PetscMalloc1(len, &b->ids);CHKERRQ(ierr);
248f1fd5e65SToby Isaac       ierr = PetscMemcpy(b->ids, ids, len*sizeof(PetscInt));CHKERRQ(ierr);
249f1fd5e65SToby Isaac     }
250e7b0402cSSander Arens     len = 1024;
251f1fd5e65SToby Isaac     ierr = PetscSNPrintf(optname, sizeof(optname), "-bc_%s_comp", b->name);CHKERRQ(ierr);
252f1fd5e65SToby Isaac     ierr = PetscMemzero(ids, sizeof(ids));CHKERRQ(ierr);
253f1fd5e65SToby Isaac     ierr = PetscOptionsIntArray(optname, "List of boundary field components", "", ids, &len, &flg);CHKERRQ(ierr);
254f1fd5e65SToby Isaac     if (flg) {
255f1fd5e65SToby Isaac       b->numcomps = len;
256f1fd5e65SToby Isaac       ierr = PetscFree(b->comps);CHKERRQ(ierr);
257f1fd5e65SToby Isaac       ierr = PetscMalloc1(len, &b->comps);CHKERRQ(ierr);
258f1fd5e65SToby Isaac       ierr = PetscMemcpy(b->comps, ids, len*sizeof(PetscInt));CHKERRQ(ierr);
259f1fd5e65SToby Isaac     }
260f1fd5e65SToby Isaac   }
2612764a2aaSMatthew G. Knepley   ierr = PetscOptionsFList("-petscds_type", "Discrete System", "PetscDSSetType", PetscDSList, defaultType, name, 256, &flg);CHKERRQ(ierr);
2622764a2aaSMatthew G. Knepley   if (flg) {
2632764a2aaSMatthew G. Knepley     ierr = PetscDSSetType(prob, name);CHKERRQ(ierr);
2642764a2aaSMatthew G. Knepley   } else if (!((PetscObject) prob)->type_name) {
2652764a2aaSMatthew G. Knepley     ierr = PetscDSSetType(prob, defaultType);CHKERRQ(ierr);
2662764a2aaSMatthew G. Knepley   }
2672764a2aaSMatthew G. Knepley   if (prob->ops->setfromoptions) {ierr = (*prob->ops->setfromoptions)(prob);CHKERRQ(ierr);}
2682764a2aaSMatthew G. Knepley   /* process any options handlers added with PetscObjectAddOptionsHandler() */
2690633abcbSJed Brown   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject) prob);CHKERRQ(ierr);
2702764a2aaSMatthew G. Knepley   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2712764a2aaSMatthew G. Knepley   ierr = PetscDSViewFromOptions(prob, NULL, "-petscds_view");CHKERRQ(ierr);
2722764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
2732764a2aaSMatthew G. Knepley }
2742764a2aaSMatthew G. Knepley 
2752764a2aaSMatthew G. Knepley /*@C
2762764a2aaSMatthew G. Knepley   PetscDSSetUp - Construct data structures for the PetscDS
2772764a2aaSMatthew G. Knepley 
2782764a2aaSMatthew G. Knepley   Collective on PetscDS
2792764a2aaSMatthew G. Knepley 
2802764a2aaSMatthew G. Knepley   Input Parameter:
2812764a2aaSMatthew G. Knepley . prob - the PetscDS object to setup
2822764a2aaSMatthew G. Knepley 
2832764a2aaSMatthew G. Knepley   Level: developer
2842764a2aaSMatthew G. Knepley 
2852764a2aaSMatthew G. Knepley .seealso PetscDSView(), PetscDSDestroy()
2862764a2aaSMatthew G. Knepley @*/
2872764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetUp(PetscDS prob)
2882764a2aaSMatthew G. Knepley {
2892764a2aaSMatthew G. Knepley   const PetscInt Nf = prob->Nf;
2902764a2aaSMatthew G. Knepley   PetscInt       dim, work, NcMax = 0, NqMax = 0, f;
2912764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
2922764a2aaSMatthew G. Knepley 
2932764a2aaSMatthew G. Knepley   PetscFunctionBegin;
2942764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
2952764a2aaSMatthew G. Knepley   if (prob->setup) PetscFunctionReturn(0);
2962764a2aaSMatthew G. Knepley   /* Calculate sizes */
2972764a2aaSMatthew G. Knepley   ierr = PetscDSGetSpatialDimension(prob, &dim);CHKERRQ(ierr);
298f744cafaSSander Arens   prob->totDim = prob->totComp = 0;
29947e57110SSander Arens   ierr = PetscMalloc2(Nf,&prob->Nc,Nf,&prob->Nb);CHKERRQ(ierr);
300f744cafaSSander Arens   ierr = PetscCalloc2(Nf+1,&prob->off,Nf+1,&prob->offDer);CHKERRQ(ierr);
301f744cafaSSander Arens   ierr = PetscMalloc4(Nf,&prob->basis,Nf,&prob->basisDer,Nf,&prob->basisFace,Nf,&prob->basisDerFace);CHKERRQ(ierr);
3022764a2aaSMatthew G. Knepley   for (f = 0; f < Nf; ++f) {
3039de99aefSMatthew G. Knepley     PetscObject     obj;
3049de99aefSMatthew G. Knepley     PetscClassId    id;
3052764a2aaSMatthew G. Knepley     PetscQuadrature q;
3069de99aefSMatthew G. Knepley     PetscInt        Nq = 0, Nb, Nc;
3072764a2aaSMatthew G. Knepley 
3089de99aefSMatthew G. Knepley     ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr);
3099de99aefSMatthew G. Knepley     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
3109de99aefSMatthew G. Knepley     if (id == PETSCFE_CLASSID)      {
3119de99aefSMatthew G. Knepley       PetscFE fe = (PetscFE) obj;
3129de99aefSMatthew G. Knepley 
3132764a2aaSMatthew G. Knepley       ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr);
3142764a2aaSMatthew G. Knepley       ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr);
3152764a2aaSMatthew G. Knepley       ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr);
3162764a2aaSMatthew G. Knepley       ierr = PetscFEGetDefaultTabulation(fe, &prob->basis[f], &prob->basisDer[f], NULL);CHKERRQ(ierr);
3174d0b9603SSander Arens       ierr = PetscFEGetFaceTabulation(fe, &prob->basisFace[f], &prob->basisDerFace[f], NULL);CHKERRQ(ierr);
3189de99aefSMatthew G. Knepley     } else if (id == PETSCFV_CLASSID) {
3199de99aefSMatthew G. Knepley       PetscFV fv = (PetscFV) obj;
3209de99aefSMatthew G. Knepley 
3219de99aefSMatthew G. Knepley       ierr = PetscFVGetQuadrature(fv, &q);CHKERRQ(ierr);
3229de99aefSMatthew G. Knepley       ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr);
3239c3cf19fSMatthew G. Knepley       Nb   = Nc;
3246c1a3d01SMatthew G. Knepley       ierr = PetscFVGetDefaultTabulation(fv, &prob->basis[f], &prob->basisDer[f], NULL);CHKERRQ(ierr);
3254d0b9603SSander Arens       /* TODO: should PetscFV also have face tabulation? Otherwise there will be a null pointer in prob->basisFace */
326abac5ca0SMatthew G. Knepley     } else SETERRQ1(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", f);
32747e57110SSander Arens     prob->Nc[f]       = Nc;
32847e57110SSander Arens     prob->Nb[f]       = Nb;
329194d53e6SMatthew G. Knepley     prob->off[f+1]    = Nc     + prob->off[f];
330194d53e6SMatthew G. Knepley     prob->offDer[f+1] = Nc*dim + prob->offDer[f];
331a6b92713SMatthew G. Knepley     if (q) {ierr = PetscQuadratureGetData(q, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr);}
3322764a2aaSMatthew G. Knepley     NqMax          = PetscMax(NqMax, Nq);
3332764a2aaSMatthew G. Knepley     NcMax          = PetscMax(NcMax, Nc);
3349c3cf19fSMatthew G. Knepley     prob->totDim  += Nb;
3352764a2aaSMatthew G. Knepley     prob->totComp += Nc;
3362764a2aaSMatthew G. Knepley   }
3372764a2aaSMatthew G. Knepley   work = PetscMax(prob->totComp*dim, PetscSqr(NcMax*dim));
3382764a2aaSMatthew G. Knepley   /* Allocate works space */
3392764a2aaSMatthew G. Knepley   ierr = PetscMalloc5(prob->totComp,&prob->u,prob->totComp,&prob->u_t,prob->totComp*dim,&prob->u_x,dim,&prob->x,work,&prob->refSpaceDer);CHKERRQ(ierr);
3402764a2aaSMatthew G. Knepley   ierr = PetscMalloc6(NqMax*NcMax,&prob->f0,NqMax*NcMax*dim,&prob->f1,NqMax*NcMax*NcMax,&prob->g0,NqMax*NcMax*NcMax*dim,&prob->g1,NqMax*NcMax*NcMax*dim,&prob->g2,NqMax*NcMax*NcMax*dim*dim,&prob->g3);CHKERRQ(ierr);
3412764a2aaSMatthew G. Knepley   if (prob->ops->setup) {ierr = (*prob->ops->setup)(prob);CHKERRQ(ierr);}
3422764a2aaSMatthew G. Knepley   prob->setup = PETSC_TRUE;
3432764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
3442764a2aaSMatthew G. Knepley }
3452764a2aaSMatthew G. Knepley 
3462764a2aaSMatthew G. Knepley static PetscErrorCode PetscDSDestroyStructs_Static(PetscDS prob)
3472764a2aaSMatthew G. Knepley {
3482764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
3492764a2aaSMatthew G. Knepley 
3502764a2aaSMatthew G. Knepley   PetscFunctionBegin;
35147e57110SSander Arens   ierr = PetscFree2(prob->Nc,prob->Nb);CHKERRQ(ierr);
352f744cafaSSander Arens   ierr = PetscFree2(prob->off,prob->offDer);CHKERRQ(ierr);
353f744cafaSSander Arens   ierr = PetscFree4(prob->basis,prob->basisDer,prob->basisFace,prob->basisDerFace);CHKERRQ(ierr);
3542764a2aaSMatthew G. Knepley   ierr = PetscFree5(prob->u,prob->u_t,prob->u_x,prob->x,prob->refSpaceDer);CHKERRQ(ierr);
3552764a2aaSMatthew G. Knepley   ierr = PetscFree6(prob->f0,prob->f1,prob->g0,prob->g1,prob->g2,prob->g3);CHKERRQ(ierr);
3562764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
3572764a2aaSMatthew G. Knepley }
3582764a2aaSMatthew G. Knepley 
3592764a2aaSMatthew G. Knepley static PetscErrorCode PetscDSEnlarge_Static(PetscDS prob, PetscInt NfNew)
3602764a2aaSMatthew G. Knepley {
361f744cafaSSander Arens   PetscObject      *tmpd;
362a6cbbb48SMatthew G. Knepley   PetscBool        *tmpi, *tmpa;
3632aa1fc23SMatthew G. Knepley   PetscPointFunc   *tmpobj, *tmpf;
364b7e05686SMatthew G. Knepley   PetscPointJac    *tmpg, *tmpgp, *tmpgt;
3652aa1fc23SMatthew G. Knepley   PetscBdPointFunc *tmpfbd;
3662aa1fc23SMatthew G. Knepley   PetscBdPointJac  *tmpgbd;
367194d53e6SMatthew G. Knepley   PetscRiemannFunc *tmpr;
3680c2f2876SMatthew G. Knepley   void            **tmpctx;
369a6cbbb48SMatthew G. Knepley   PetscInt          Nf = prob->Nf, f, i;
3702764a2aaSMatthew G. Knepley   PetscErrorCode    ierr;
3712764a2aaSMatthew G. Knepley 
3722764a2aaSMatthew G. Knepley   PetscFunctionBegin;
3732764a2aaSMatthew G. Knepley   if (Nf >= NfNew) PetscFunctionReturn(0);
3742764a2aaSMatthew G. Knepley   prob->setup = PETSC_FALSE;
3752764a2aaSMatthew G. Knepley   ierr = PetscDSDestroyStructs_Static(prob);CHKERRQ(ierr);
376f744cafaSSander Arens   ierr = PetscMalloc3(NfNew, &tmpd, NfNew, &tmpi, NfNew*2, &tmpa);CHKERRQ(ierr);
377f744cafaSSander Arens   for (f = 0; f < Nf; ++f) {tmpd[f] = prob->disc[f]; tmpi[f] = prob->implicit[f]; for (i = 0; i < 2; ++i) tmpa[f*2+i] = prob->adjacency[f*2+i];}
378f744cafaSSander Arens   for (f = Nf; f < NfNew; ++f) {tmpd[f] = NULL; tmpi[f] = PETSC_TRUE; tmpa[f*2+0] = PETSC_FALSE; tmpa[f*2+1] = PETSC_TRUE;}
379f744cafaSSander Arens   ierr = PetscFree3(prob->disc, prob->implicit, prob->adjacency);CHKERRQ(ierr);
3802764a2aaSMatthew G. Knepley   prob->Nf        = NfNew;
3812764a2aaSMatthew G. Knepley   prob->disc      = tmpd;
382249df284SMatthew G. Knepley   prob->implicit  = tmpi;
383a6cbbb48SMatthew G. Knepley   prob->adjacency = tmpa;
384b7e05686SMatthew G. Knepley   ierr = PetscCalloc7(NfNew, &tmpobj, NfNew*2, &tmpf, NfNew*NfNew*4, &tmpg, NfNew*NfNew*4, &tmpgp, NfNew*NfNew*4, &tmpgt, NfNew, &tmpr, NfNew, &tmpctx);CHKERRQ(ierr);
3852764a2aaSMatthew G. Knepley   for (f = 0; f < Nf; ++f) tmpobj[f] = prob->obj[f];
3862764a2aaSMatthew G. Knepley   for (f = 0; f < Nf*2; ++f) tmpf[f] = prob->f[f];
3872764a2aaSMatthew G. Knepley   for (f = 0; f < Nf*Nf*4; ++f) tmpg[f] = prob->g[f];
388475e0ac9SMatthew G. Knepley   for (f = 0; f < Nf*Nf*4; ++f) tmpgp[f] = prob->gp[f];
3890c2f2876SMatthew G. Knepley   for (f = 0; f < Nf; ++f) tmpr[f] = prob->r[f];
3900c2f2876SMatthew G. Knepley   for (f = 0; f < Nf; ++f) tmpctx[f] = prob->ctx[f];
3912764a2aaSMatthew G. Knepley   for (f = Nf; f < NfNew; ++f) tmpobj[f] = NULL;
3922764a2aaSMatthew G. Knepley   for (f = Nf*2; f < NfNew*2; ++f) tmpf[f] = NULL;
3932764a2aaSMatthew G. Knepley   for (f = Nf*Nf*4; f < NfNew*NfNew*4; ++f) tmpg[f] = NULL;
394475e0ac9SMatthew G. Knepley   for (f = Nf*Nf*4; f < NfNew*NfNew*4; ++f) tmpgp[f] = NULL;
395b7e05686SMatthew G. Knepley   for (f = Nf*Nf*4; f < NfNew*NfNew*4; ++f) tmpgt[f] = NULL;
3960c2f2876SMatthew G. Knepley   for (f = Nf; f < NfNew; ++f) tmpr[f] = NULL;
3970c2f2876SMatthew G. Knepley   for (f = Nf; f < NfNew; ++f) tmpctx[f] = NULL;
398b7e05686SMatthew G. Knepley   ierr = PetscFree7(prob->obj, prob->f, prob->g, prob->gp, prob->gt, prob->r, prob->ctx);CHKERRQ(ierr);
3992764a2aaSMatthew G. Knepley   prob->obj = tmpobj;
4002764a2aaSMatthew G. Knepley   prob->f   = tmpf;
4012764a2aaSMatthew G. Knepley   prob->g   = tmpg;
402475e0ac9SMatthew G. Knepley   prob->gp  = tmpgp;
403b7e05686SMatthew G. Knepley   prob->gt  = tmpgt;
4040c2f2876SMatthew G. Knepley   prob->r   = tmpr;
4050c2f2876SMatthew G. Knepley   prob->ctx = tmpctx;
4062764a2aaSMatthew G. Knepley   ierr = PetscCalloc2(NfNew*2, &tmpfbd, NfNew*NfNew*4, &tmpgbd);CHKERRQ(ierr);
4072764a2aaSMatthew G. Knepley   for (f = 0; f < Nf*2; ++f) tmpfbd[f] = prob->fBd[f];
4082764a2aaSMatthew G. Knepley   for (f = 0; f < Nf*Nf*4; ++f) tmpgbd[f] = prob->gBd[f];
4092764a2aaSMatthew G. Knepley   for (f = Nf*2; f < NfNew*2; ++f) tmpfbd[f] = NULL;
4102764a2aaSMatthew G. Knepley   for (f = Nf*Nf*4; f < NfNew*NfNew*4; ++f) tmpgbd[f] = NULL;
4112764a2aaSMatthew G. Knepley   ierr = PetscFree2(prob->fBd, prob->gBd);CHKERRQ(ierr);
4122764a2aaSMatthew G. Knepley   prob->fBd = tmpfbd;
4132764a2aaSMatthew G. Knepley   prob->gBd = tmpgbd;
4142764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
4152764a2aaSMatthew G. Knepley }
4162764a2aaSMatthew G. Knepley 
4172764a2aaSMatthew G. Knepley /*@
4182764a2aaSMatthew G. Knepley   PetscDSDestroy - Destroys a PetscDS object
4192764a2aaSMatthew G. Knepley 
4202764a2aaSMatthew G. Knepley   Collective on PetscDS
4212764a2aaSMatthew G. Knepley 
4222764a2aaSMatthew G. Knepley   Input Parameter:
4232764a2aaSMatthew G. Knepley . prob - the PetscDS object to destroy
4242764a2aaSMatthew G. Knepley 
4252764a2aaSMatthew G. Knepley   Level: developer
4262764a2aaSMatthew G. Knepley 
4272764a2aaSMatthew G. Knepley .seealso PetscDSView()
4282764a2aaSMatthew G. Knepley @*/
4292764a2aaSMatthew G. Knepley PetscErrorCode PetscDSDestroy(PetscDS *prob)
4302764a2aaSMatthew G. Knepley {
4312764a2aaSMatthew G. Knepley   PetscInt       f;
43258ebd649SToby Isaac   DSBoundary     next;
4332764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
4342764a2aaSMatthew G. Knepley 
4352764a2aaSMatthew G. Knepley   PetscFunctionBegin;
4362764a2aaSMatthew G. Knepley   if (!*prob) PetscFunctionReturn(0);
4372764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific((*prob), PETSCDS_CLASSID, 1);
4382764a2aaSMatthew G. Knepley 
4392764a2aaSMatthew G. Knepley   if (--((PetscObject)(*prob))->refct > 0) {*prob = 0; PetscFunctionReturn(0);}
4402764a2aaSMatthew G. Knepley   ((PetscObject) (*prob))->refct = 0;
4412764a2aaSMatthew G. Knepley   ierr = PetscDSDestroyStructs_Static(*prob);CHKERRQ(ierr);
4422764a2aaSMatthew G. Knepley   for (f = 0; f < (*prob)->Nf; ++f) {
4432764a2aaSMatthew G. Knepley     ierr = PetscObjectDereference((*prob)->disc[f]);CHKERRQ(ierr);
4442764a2aaSMatthew G. Knepley   }
445f744cafaSSander Arens   ierr = PetscFree3((*prob)->disc, (*prob)->implicit, (*prob)->adjacency);CHKERRQ(ierr);
446b7e05686SMatthew G. Knepley   ierr = PetscFree7((*prob)->obj,(*prob)->f,(*prob)->g,(*prob)->gp,(*prob)->gt,(*prob)->r,(*prob)->ctx);CHKERRQ(ierr);
4472764a2aaSMatthew G. Knepley   ierr = PetscFree2((*prob)->fBd,(*prob)->gBd);CHKERRQ(ierr);
4482764a2aaSMatthew G. Knepley   if ((*prob)->ops->destroy) {ierr = (*(*prob)->ops->destroy)(*prob);CHKERRQ(ierr);}
44958ebd649SToby Isaac   next = (*prob)->boundary;
45058ebd649SToby Isaac   while (next) {
45158ebd649SToby Isaac     DSBoundary b = next;
45258ebd649SToby Isaac 
45358ebd649SToby Isaac     next = b->next;
45458ebd649SToby Isaac     ierr = PetscFree(b->comps);CHKERRQ(ierr);
45558ebd649SToby Isaac     ierr = PetscFree(b->ids);CHKERRQ(ierr);
45658ebd649SToby Isaac     ierr = PetscFree(b->name);CHKERRQ(ierr);
45758ebd649SToby Isaac     ierr = PetscFree(b->labelname);CHKERRQ(ierr);
45858ebd649SToby Isaac     ierr = PetscFree(b);CHKERRQ(ierr);
45958ebd649SToby Isaac   }
46097b6e6e8SMatthew G. Knepley   ierr = PetscFree((*prob)->constants);CHKERRQ(ierr);
4612764a2aaSMatthew G. Knepley   ierr = PetscHeaderDestroy(prob);CHKERRQ(ierr);
4622764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
4632764a2aaSMatthew G. Knepley }
4642764a2aaSMatthew G. Knepley 
4652764a2aaSMatthew G. Knepley /*@
4662764a2aaSMatthew G. Knepley   PetscDSCreate - Creates an empty PetscDS object. The type can then be set with PetscDSSetType().
4672764a2aaSMatthew G. Knepley 
4682764a2aaSMatthew G. Knepley   Collective on MPI_Comm
4692764a2aaSMatthew G. Knepley 
4702764a2aaSMatthew G. Knepley   Input Parameter:
4712764a2aaSMatthew G. Knepley . comm - The communicator for the PetscDS object
4722764a2aaSMatthew G. Knepley 
4732764a2aaSMatthew G. Knepley   Output Parameter:
4742764a2aaSMatthew G. Knepley . prob - The PetscDS object
4752764a2aaSMatthew G. Knepley 
4762764a2aaSMatthew G. Knepley   Level: beginner
4772764a2aaSMatthew G. Knepley 
4782764a2aaSMatthew G. Knepley .seealso: PetscDSSetType(), PETSCDSBASIC
4792764a2aaSMatthew G. Knepley @*/
4802764a2aaSMatthew G. Knepley PetscErrorCode PetscDSCreate(MPI_Comm comm, PetscDS *prob)
4812764a2aaSMatthew G. Knepley {
4822764a2aaSMatthew G. Knepley   PetscDS   p;
4832764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
4842764a2aaSMatthew G. Knepley 
4852764a2aaSMatthew G. Knepley   PetscFunctionBegin;
4862764a2aaSMatthew G. Knepley   PetscValidPointer(prob, 2);
4872764a2aaSMatthew G. Knepley   *prob  = NULL;
4882764a2aaSMatthew G. Knepley   ierr = PetscDSInitializePackage();CHKERRQ(ierr);
4892764a2aaSMatthew G. Knepley 
49073107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(p, PETSCDS_CLASSID, "PetscDS", "Discrete System", "PetscDS", comm, PetscDSDestroy, PetscDSView);CHKERRQ(ierr);
4912764a2aaSMatthew G. Knepley 
4922764a2aaSMatthew G. Knepley   p->Nf    = 0;
4932764a2aaSMatthew G. Knepley   p->setup = PETSC_FALSE;
49497b6e6e8SMatthew G. Knepley   p->numConstants = 0;
49597b6e6e8SMatthew G. Knepley   p->constants    = NULL;
4962764a2aaSMatthew G. Knepley 
4972764a2aaSMatthew G. Knepley   *prob = p;
4982764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
4992764a2aaSMatthew G. Knepley }
5002764a2aaSMatthew G. Knepley 
501bc4ae4beSMatthew G. Knepley /*@
502bc4ae4beSMatthew G. Knepley   PetscDSGetNumFields - Returns the number of fields in the DS
503bc4ae4beSMatthew G. Knepley 
504bc4ae4beSMatthew G. Knepley   Not collective
505bc4ae4beSMatthew G. Knepley 
506bc4ae4beSMatthew G. Knepley   Input Parameter:
507bc4ae4beSMatthew G. Knepley . prob - The PetscDS object
508bc4ae4beSMatthew G. Knepley 
509bc4ae4beSMatthew G. Knepley   Output Parameter:
510bc4ae4beSMatthew G. Knepley . Nf - The number of fields
511bc4ae4beSMatthew G. Knepley 
512bc4ae4beSMatthew G. Knepley   Level: beginner
513bc4ae4beSMatthew G. Knepley 
514bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetSpatialDimension(), PetscDSCreate()
515bc4ae4beSMatthew G. Knepley @*/
5162764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetNumFields(PetscDS prob, PetscInt *Nf)
5172764a2aaSMatthew G. Knepley {
5182764a2aaSMatthew G. Knepley   PetscFunctionBegin;
5192764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
5202764a2aaSMatthew G. Knepley   PetscValidPointer(Nf, 2);
5212764a2aaSMatthew G. Knepley   *Nf = prob->Nf;
5222764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
5232764a2aaSMatthew G. Knepley }
5242764a2aaSMatthew G. Knepley 
525bc4ae4beSMatthew G. Knepley /*@
526bc4ae4beSMatthew G. Knepley   PetscDSGetSpatialDimension - Returns the spatial dimension of the DS
527bc4ae4beSMatthew G. Knepley 
528bc4ae4beSMatthew G. Knepley   Not collective
529bc4ae4beSMatthew G. Knepley 
530bc4ae4beSMatthew G. Knepley   Input Parameter:
531bc4ae4beSMatthew G. Knepley . prob - The PetscDS object
532bc4ae4beSMatthew G. Knepley 
533bc4ae4beSMatthew G. Knepley   Output Parameter:
534bc4ae4beSMatthew G. Knepley . dim - The spatial dimension
535bc4ae4beSMatthew G. Knepley 
536bc4ae4beSMatthew G. Knepley   Level: beginner
537bc4ae4beSMatthew G. Knepley 
538bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetNumFields(), PetscDSCreate()
539bc4ae4beSMatthew G. Knepley @*/
5402764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetSpatialDimension(PetscDS prob, PetscInt *dim)
5412764a2aaSMatthew G. Knepley {
5422764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
5432764a2aaSMatthew G. Knepley 
5442764a2aaSMatthew G. Knepley   PetscFunctionBegin;
5452764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
5462764a2aaSMatthew G. Knepley   PetscValidPointer(dim, 2);
5472764a2aaSMatthew G. Knepley   *dim = 0;
5489de99aefSMatthew G. Knepley   if (prob->Nf) {
5499de99aefSMatthew G. Knepley     PetscObject  obj;
5509de99aefSMatthew G. Knepley     PetscClassId id;
5519de99aefSMatthew G. Knepley 
5529de99aefSMatthew G. Knepley     ierr = PetscDSGetDiscretization(prob, 0, &obj);CHKERRQ(ierr);
5539de99aefSMatthew G. Knepley     ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr);
5549de99aefSMatthew G. Knepley     if (id == PETSCFE_CLASSID)      {ierr = PetscFEGetSpatialDimension((PetscFE) obj, dim);CHKERRQ(ierr);}
5559de99aefSMatthew G. Knepley     else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetSpatialDimension((PetscFV) obj, dim);CHKERRQ(ierr);}
5569de99aefSMatthew G. Knepley     else SETERRQ1(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", 0);
5579de99aefSMatthew G. Knepley   }
5582764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
5592764a2aaSMatthew G. Knepley }
5602764a2aaSMatthew G. Knepley 
561bc4ae4beSMatthew G. Knepley /*@
562bc4ae4beSMatthew G. Knepley   PetscDSGetTotalDimension - Returns the total size of the approximation space for this system
563bc4ae4beSMatthew G. Knepley 
564bc4ae4beSMatthew G. Knepley   Not collective
565bc4ae4beSMatthew G. Knepley 
566bc4ae4beSMatthew G. Knepley   Input Parameter:
567bc4ae4beSMatthew G. Knepley . prob - The PetscDS object
568bc4ae4beSMatthew G. Knepley 
569bc4ae4beSMatthew G. Knepley   Output Parameter:
570bc4ae4beSMatthew G. Knepley . dim - The total problem dimension
571bc4ae4beSMatthew G. Knepley 
572bc4ae4beSMatthew G. Knepley   Level: beginner
573bc4ae4beSMatthew G. Knepley 
574bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetNumFields(), PetscDSCreate()
575bc4ae4beSMatthew G. Knepley @*/
5762764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetTotalDimension(PetscDS prob, PetscInt *dim)
5772764a2aaSMatthew G. Knepley {
5782764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
5792764a2aaSMatthew G. Knepley 
5802764a2aaSMatthew G. Knepley   PetscFunctionBegin;
5812764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
5822764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
5832764a2aaSMatthew G. Knepley   PetscValidPointer(dim, 2);
5842764a2aaSMatthew G. Knepley   *dim = prob->totDim;
5852764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
5862764a2aaSMatthew G. Knepley }
5872764a2aaSMatthew G. Knepley 
588bc4ae4beSMatthew G. Knepley /*@
589bc4ae4beSMatthew G. Knepley   PetscDSGetTotalComponents - Returns the total number of components in this system
590bc4ae4beSMatthew G. Knepley 
591bc4ae4beSMatthew G. Knepley   Not collective
592bc4ae4beSMatthew G. Knepley 
593bc4ae4beSMatthew G. Knepley   Input Parameter:
594bc4ae4beSMatthew G. Knepley . prob - The PetscDS object
595bc4ae4beSMatthew G. Knepley 
596bc4ae4beSMatthew G. Knepley   Output Parameter:
597bc4ae4beSMatthew G. Knepley . dim - The total number of components
598bc4ae4beSMatthew G. Knepley 
599bc4ae4beSMatthew G. Knepley   Level: beginner
600bc4ae4beSMatthew G. Knepley 
601bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetNumFields(), PetscDSCreate()
602bc4ae4beSMatthew G. Knepley @*/
6032764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetTotalComponents(PetscDS prob, PetscInt *Nc)
6042764a2aaSMatthew G. Knepley {
6052764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
6062764a2aaSMatthew G. Knepley 
6072764a2aaSMatthew G. Knepley   PetscFunctionBegin;
6082764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
6092764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
6102764a2aaSMatthew G. Knepley   PetscValidPointer(Nc, 2);
6112764a2aaSMatthew G. Knepley   *Nc = prob->totComp;
6122764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
6132764a2aaSMatthew G. Knepley }
6142764a2aaSMatthew G. Knepley 
615bc4ae4beSMatthew G. Knepley /*@
616bc4ae4beSMatthew G. Knepley   PetscDSGetDiscretization - Returns the discretization object for the given field
617bc4ae4beSMatthew G. Knepley 
618bc4ae4beSMatthew G. Knepley   Not collective
619bc4ae4beSMatthew G. Knepley 
620bc4ae4beSMatthew G. Knepley   Input Parameters:
621bc4ae4beSMatthew G. Knepley + prob - The PetscDS object
622bc4ae4beSMatthew G. Knepley - f - The field number
623bc4ae4beSMatthew G. Knepley 
624bc4ae4beSMatthew G. Knepley   Output Parameter:
625bc4ae4beSMatthew G. Knepley . disc - The discretization object
626bc4ae4beSMatthew G. Knepley 
627bc4ae4beSMatthew G. Knepley   Level: beginner
628bc4ae4beSMatthew G. Knepley 
629f744cafaSSander Arens .seealso: PetscDSSetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
630bc4ae4beSMatthew G. Knepley @*/
6312764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetDiscretization(PetscDS prob, PetscInt f, PetscObject *disc)
6322764a2aaSMatthew G. Knepley {
6332764a2aaSMatthew G. Knepley   PetscFunctionBegin;
6342764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
6352764a2aaSMatthew G. Knepley   PetscValidPointer(disc, 3);
6362764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
6372764a2aaSMatthew G. Knepley   *disc = prob->disc[f];
6382764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
6392764a2aaSMatthew G. Knepley }
6402764a2aaSMatthew G. Knepley 
641bc4ae4beSMatthew G. Knepley /*@
642bc4ae4beSMatthew G. Knepley   PetscDSSetDiscretization - Sets the discretization object for the given field
643bc4ae4beSMatthew G. Knepley 
644bc4ae4beSMatthew G. Knepley   Not collective
645bc4ae4beSMatthew G. Knepley 
646bc4ae4beSMatthew G. Knepley   Input Parameters:
647bc4ae4beSMatthew G. Knepley + prob - The PetscDS object
648bc4ae4beSMatthew G. Knepley . f - The field number
649bc4ae4beSMatthew G. Knepley - disc - The discretization object
650bc4ae4beSMatthew G. Knepley 
651bc4ae4beSMatthew G. Knepley   Level: beginner
652bc4ae4beSMatthew G. Knepley 
653bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
654bc4ae4beSMatthew G. Knepley @*/
6552764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetDiscretization(PetscDS prob, PetscInt f, PetscObject disc)
6562764a2aaSMatthew G. Knepley {
6572764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
6582764a2aaSMatthew G. Knepley 
6592764a2aaSMatthew G. Knepley   PetscFunctionBegin;
6602764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
6612764a2aaSMatthew G. Knepley   PetscValidPointer(disc, 3);
6622764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
6632764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
6642764a2aaSMatthew G. Knepley   if (prob->disc[f]) {ierr = PetscObjectDereference(prob->disc[f]);CHKERRQ(ierr);}
6652764a2aaSMatthew G. Knepley   prob->disc[f] = disc;
6662764a2aaSMatthew G. Knepley   ierr = PetscObjectReference(disc);CHKERRQ(ierr);
667249df284SMatthew G. Knepley   {
668249df284SMatthew G. Knepley     PetscClassId id;
669249df284SMatthew G. Knepley 
670249df284SMatthew G. Knepley     ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr);
671a6cbbb48SMatthew G. Knepley     if (id == PETSCFV_CLASSID) {
672a6cbbb48SMatthew G. Knepley       prob->implicit[f]      = PETSC_FALSE;
673a6cbbb48SMatthew G. Knepley       prob->adjacency[f*2+0] = PETSC_TRUE;
674a6cbbb48SMatthew G. Knepley       prob->adjacency[f*2+1] = PETSC_FALSE;
675a6cbbb48SMatthew G. Knepley     }
676249df284SMatthew G. Knepley   }
6772764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
6782764a2aaSMatthew G. Knepley }
6792764a2aaSMatthew G. Knepley 
680bc4ae4beSMatthew G. Knepley /*@
681bc4ae4beSMatthew G. Knepley   PetscDSAddDiscretization - Adds a discretization object
682bc4ae4beSMatthew G. Knepley 
683bc4ae4beSMatthew G. Knepley   Not collective
684bc4ae4beSMatthew G. Knepley 
685bc4ae4beSMatthew G. Knepley   Input Parameters:
686bc4ae4beSMatthew G. Knepley + prob - The PetscDS object
687bc4ae4beSMatthew G. Knepley - disc - The boundary discretization object
688bc4ae4beSMatthew G. Knepley 
689bc4ae4beSMatthew G. Knepley   Level: beginner
690bc4ae4beSMatthew G. Knepley 
691bc4ae4beSMatthew G. Knepley .seealso: PetscDSGetDiscretization(), PetscDSSetDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
692bc4ae4beSMatthew G. Knepley @*/
6932764a2aaSMatthew G. Knepley PetscErrorCode PetscDSAddDiscretization(PetscDS prob, PetscObject disc)
6942764a2aaSMatthew G. Knepley {
6952764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
6962764a2aaSMatthew G. Knepley 
6972764a2aaSMatthew G. Knepley   PetscFunctionBegin;
6982764a2aaSMatthew G. Knepley   ierr = PetscDSSetDiscretization(prob, prob->Nf, disc);CHKERRQ(ierr);
6992764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
7002764a2aaSMatthew G. Knepley }
7012764a2aaSMatthew G. Knepley 
702249df284SMatthew G. Knepley /*@
703249df284SMatthew G. Knepley   PetscDSGetImplicit - Returns the flag for implicit solve for this field. This is just a guide for IMEX
704249df284SMatthew G. Knepley 
705249df284SMatthew G. Knepley   Not collective
706249df284SMatthew G. Knepley 
707249df284SMatthew G. Knepley   Input Parameters:
708249df284SMatthew G. Knepley + prob - The PetscDS object
709249df284SMatthew G. Knepley - f - The field number
710249df284SMatthew G. Knepley 
711249df284SMatthew G. Knepley   Output Parameter:
712249df284SMatthew G. Knepley . implicit - The flag indicating what kind of solve to use for this field
713249df284SMatthew G. Knepley 
714249df284SMatthew G. Knepley   Level: developer
715249df284SMatthew G. Knepley 
716f744cafaSSander Arens .seealso: PetscDSSetImplicit(), PetscDSSetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
717249df284SMatthew G. Knepley @*/
718249df284SMatthew G. Knepley PetscErrorCode PetscDSGetImplicit(PetscDS prob, PetscInt f, PetscBool *implicit)
719249df284SMatthew G. Knepley {
720249df284SMatthew G. Knepley   PetscFunctionBegin;
721249df284SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
722249df284SMatthew G. Knepley   PetscValidPointer(implicit, 3);
723249df284SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
724249df284SMatthew G. Knepley   *implicit = prob->implicit[f];
725249df284SMatthew G. Knepley   PetscFunctionReturn(0);
726249df284SMatthew G. Knepley }
727249df284SMatthew G. Knepley 
728249df284SMatthew G. Knepley /*@
729249df284SMatthew G. Knepley   PetscDSSetImplicit - Set the flag for implicit solve for this field. This is just a guide for IMEX
730249df284SMatthew G. Knepley 
731249df284SMatthew G. Knepley   Not collective
732249df284SMatthew G. Knepley 
733249df284SMatthew G. Knepley   Input Parameters:
734249df284SMatthew G. Knepley + prob - The PetscDS object
735249df284SMatthew G. Knepley . f - The field number
736249df284SMatthew G. Knepley - implicit - The flag indicating what kind of solve to use for this field
737249df284SMatthew G. Knepley 
738249df284SMatthew G. Knepley   Level: developer
739249df284SMatthew G. Knepley 
740f744cafaSSander Arens .seealso: PetscDSGetImplicit(), PetscDSSetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
741249df284SMatthew G. Knepley @*/
742249df284SMatthew G. Knepley PetscErrorCode PetscDSSetImplicit(PetscDS prob, PetscInt f, PetscBool implicit)
743249df284SMatthew G. Knepley {
744249df284SMatthew G. Knepley   PetscFunctionBegin;
745249df284SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
746249df284SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
747249df284SMatthew G. Knepley   prob->implicit[f] = implicit;
748249df284SMatthew G. Knepley   PetscFunctionReturn(0);
749249df284SMatthew G. Knepley }
750249df284SMatthew G. Knepley 
751a6cbbb48SMatthew G. Knepley /*@
752a6cbbb48SMatthew G. Knepley   PetscDSGetAdjacency - Returns the flags for determining variable influence
753a6cbbb48SMatthew G. Knepley 
754a6cbbb48SMatthew G. Knepley   Not collective
755a6cbbb48SMatthew G. Knepley 
756a6cbbb48SMatthew G. Knepley   Input Parameters:
757a6cbbb48SMatthew G. Knepley + prob - The PetscDS object
758a6cbbb48SMatthew G. Knepley - f - The field number
759a6cbbb48SMatthew G. Knepley 
760a6cbbb48SMatthew G. Knepley   Output Parameter:
761a6cbbb48SMatthew G. Knepley + useCone    - Flag for variable influence starting with the cone operation
762a6cbbb48SMatthew G. Knepley - useClosure - Flag for variable influence using transitive closure
763a6cbbb48SMatthew G. Knepley 
764a6cbbb48SMatthew G. Knepley   Note: See the discussion in DMPlexGetAdjacencyUseCone() and DMPlexGetAdjacencyUseClosure()
765a6cbbb48SMatthew G. Knepley 
766a6cbbb48SMatthew G. Knepley   Level: developer
767a6cbbb48SMatthew G. Knepley 
768f744cafaSSander Arens .seealso: PetscDSSetAdjacency(), DMPlexGetAdjacencyUseCone(), DMPlexGetAdjacencyUseClosure(), PetscDSSetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
769a6cbbb48SMatthew G. Knepley @*/
770a6cbbb48SMatthew G. Knepley PetscErrorCode PetscDSGetAdjacency(PetscDS prob, PetscInt f, PetscBool *useCone, PetscBool *useClosure)
771a6cbbb48SMatthew G. Knepley {
772a6cbbb48SMatthew G. Knepley   PetscFunctionBegin;
773a6cbbb48SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
774a6cbbb48SMatthew G. Knepley   PetscValidPointer(useCone, 3);
775a6cbbb48SMatthew G. Knepley   PetscValidPointer(useClosure, 4);
776a6cbbb48SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
777a6cbbb48SMatthew G. Knepley   *useCone    = prob->adjacency[f*2+0];
778a6cbbb48SMatthew G. Knepley   *useClosure = prob->adjacency[f*2+1];
779a6cbbb48SMatthew G. Knepley   PetscFunctionReturn(0);
780a6cbbb48SMatthew G. Knepley }
781a6cbbb48SMatthew G. Knepley 
782a6cbbb48SMatthew G. Knepley /*@
783a6cbbb48SMatthew G. Knepley   PetscDSSetAdjacency - Set the flags for determining variable influence
784a6cbbb48SMatthew G. Knepley 
785a6cbbb48SMatthew G. Knepley   Not collective
786a6cbbb48SMatthew G. Knepley 
787a6cbbb48SMatthew G. Knepley   Input Parameters:
788a6cbbb48SMatthew G. Knepley + prob - The PetscDS object
789a6cbbb48SMatthew G. Knepley . f - The field number
790a6cbbb48SMatthew G. Knepley . useCone    - Flag for variable influence starting with the cone operation
791a6cbbb48SMatthew G. Knepley - useClosure - Flag for variable influence using transitive closure
792a6cbbb48SMatthew G. Knepley 
793a6cbbb48SMatthew G. Knepley   Note: See the discussion in DMPlexGetAdjacencyUseCone() and DMPlexGetAdjacencyUseClosure()
794a6cbbb48SMatthew G. Knepley 
795a6cbbb48SMatthew G. Knepley   Level: developer
796a6cbbb48SMatthew G. Knepley 
797f744cafaSSander Arens .seealso: PetscDSGetAdjacency(), DMPlexGetAdjacencyUseCone(), DMPlexGetAdjacencyUseClosure(), PetscDSSetDiscretization(), PetscDSAddDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
798a6cbbb48SMatthew G. Knepley @*/
799a6cbbb48SMatthew G. Knepley PetscErrorCode PetscDSSetAdjacency(PetscDS prob, PetscInt f, PetscBool useCone, PetscBool useClosure)
800a6cbbb48SMatthew G. Knepley {
801a6cbbb48SMatthew G. Knepley   PetscFunctionBegin;
802a6cbbb48SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
803a6cbbb48SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
804a6cbbb48SMatthew G. Knepley   prob->adjacency[f*2+0] = useCone;
805a6cbbb48SMatthew G. Knepley   prob->adjacency[f*2+1] = useClosure;
806a6cbbb48SMatthew G. Knepley   PetscFunctionReturn(0);
807a6cbbb48SMatthew G. Knepley }
808a6cbbb48SMatthew G. Knepley 
8092764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetObjective(PetscDS prob, PetscInt f,
81030b9ff8bSMatthew G. Knepley                                    void (**obj)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
811194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
812194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
81397b6e6e8SMatthew G. Knepley                                                 PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar obj[]))
8142764a2aaSMatthew G. Knepley {
8152764a2aaSMatthew G. Knepley   PetscFunctionBegin;
8162764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
8172764a2aaSMatthew G. Knepley   PetscValidPointer(obj, 2);
8182764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
8192764a2aaSMatthew G. Knepley   *obj = prob->obj[f];
8202764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
8212764a2aaSMatthew G. Knepley }
8222764a2aaSMatthew G. Knepley 
8232764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetObjective(PetscDS prob, PetscInt f,
82430b9ff8bSMatthew G. Knepley                                    void (*obj)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
825194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
826194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
82797b6e6e8SMatthew G. Knepley                                                PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar obj[]))
8282764a2aaSMatthew G. Knepley {
8292764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
8302764a2aaSMatthew G. Knepley 
8312764a2aaSMatthew G. Knepley   PetscFunctionBegin;
8322764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
833de716cbcSToby Isaac   if (obj) PetscValidFunction(obj, 2);
8342764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
8352764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
8362764a2aaSMatthew G. Knepley   prob->obj[f] = obj;
8372764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
8382764a2aaSMatthew G. Knepley }
8392764a2aaSMatthew G. Knepley 
840194d53e6SMatthew G. Knepley /*@C
841194d53e6SMatthew G. Knepley   PetscDSGetResidual - Get the pointwise residual function for a given test field
842194d53e6SMatthew G. Knepley 
843194d53e6SMatthew G. Knepley   Not collective
844194d53e6SMatthew G. Knepley 
845194d53e6SMatthew G. Knepley   Input Parameters:
846194d53e6SMatthew G. Knepley + prob - The PetscDS
847194d53e6SMatthew G. Knepley - f    - The test field number
848194d53e6SMatthew G. Knepley 
849194d53e6SMatthew G. Knepley   Output Parameters:
850194d53e6SMatthew G. Knepley + f0 - integrand for the test function term
851194d53e6SMatthew G. Knepley - f1 - integrand for the test function gradient term
852194d53e6SMatthew G. Knepley 
853194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
854194d53e6SMatthew G. Knepley 
855194d53e6SMatthew G. Knepley   \int_\Omega \phi f_0(u, u_t, \nabla u, x, t) + \nabla\phi \cdot {\vec f}_1(u, u_t, \nabla u, x, t)
856194d53e6SMatthew G. Knepley 
857194d53e6SMatthew G. Knepley The calling sequence for the callbacks f0 and f1 is given by:
858194d53e6SMatthew G. Knepley 
85930b9ff8bSMatthew G. Knepley $ f0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
860194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
861194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
86230b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], PetscScalar f0[])
863194d53e6SMatthew G. Knepley 
864194d53e6SMatthew G. Knepley + dim - the spatial dimension
865194d53e6SMatthew G. Knepley . Nf - the number of fields
866194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
867194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
868194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
869194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
870194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
871194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
872194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
873194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
874194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
875194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
876194d53e6SMatthew G. Knepley . t - current time
877194d53e6SMatthew G. Knepley . x - coordinates of the current point
87897b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
87997b6e6e8SMatthew G. Knepley . constants - constant parameters
880194d53e6SMatthew G. Knepley - f0 - output values at the current point
881194d53e6SMatthew G. Knepley 
882194d53e6SMatthew G. Knepley   Level: intermediate
883194d53e6SMatthew G. Knepley 
884194d53e6SMatthew G. Knepley .seealso: PetscDSSetResidual()
885194d53e6SMatthew G. Knepley @*/
8862764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetResidual(PetscDS prob, PetscInt f,
88730b9ff8bSMatthew G. Knepley                                   void (**f0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
888194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
889194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
89097b6e6e8SMatthew G. Knepley                                               PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]),
89130b9ff8bSMatthew G. Knepley                                   void (**f1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
892194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
893194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
89497b6e6e8SMatthew G. Knepley                                               PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]))
8952764a2aaSMatthew G. Knepley {
8962764a2aaSMatthew G. Knepley   PetscFunctionBegin;
8972764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
8982764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
8992764a2aaSMatthew G. Knepley   if (f0) {PetscValidPointer(f0, 3); *f0 = prob->f[f*2+0];}
9002764a2aaSMatthew G. Knepley   if (f1) {PetscValidPointer(f1, 4); *f1 = prob->f[f*2+1];}
9012764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
9022764a2aaSMatthew G. Knepley }
9032764a2aaSMatthew G. Knepley 
904194d53e6SMatthew G. Knepley /*@C
905194d53e6SMatthew G. Knepley   PetscDSSetResidual - Set the pointwise residual function for a given test field
906194d53e6SMatthew G. Knepley 
907194d53e6SMatthew G. Knepley   Not collective
908194d53e6SMatthew G. Knepley 
909194d53e6SMatthew G. Knepley   Input Parameters:
910194d53e6SMatthew G. Knepley + prob - The PetscDS
911194d53e6SMatthew G. Knepley . f    - The test field number
912194d53e6SMatthew G. Knepley . f0 - integrand for the test function term
913194d53e6SMatthew G. Knepley - f1 - integrand for the test function gradient term
914194d53e6SMatthew G. Knepley 
915194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
916194d53e6SMatthew G. Knepley 
917194d53e6SMatthew G. Knepley   \int_\Omega \phi f_0(u, u_t, \nabla u, x, t) + \nabla\phi \cdot {\vec f}_1(u, u_t, \nabla u, x, t)
918194d53e6SMatthew G. Knepley 
919194d53e6SMatthew G. Knepley The calling sequence for the callbacks f0 and f1 is given by:
920194d53e6SMatthew G. Knepley 
92130b9ff8bSMatthew G. Knepley $ f0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
922194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
923194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
92430b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], PetscScalar f0[])
925194d53e6SMatthew G. Knepley 
926194d53e6SMatthew G. Knepley + dim - the spatial dimension
927194d53e6SMatthew G. Knepley . Nf - the number of fields
928194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
929194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
930194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
931194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
932194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
933194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
934194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
935194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
936194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
937194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
938194d53e6SMatthew G. Knepley . t - current time
939194d53e6SMatthew G. Knepley . x - coordinates of the current point
94097b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
94197b6e6e8SMatthew G. Knepley . constants - constant parameters
942194d53e6SMatthew G. Knepley - f0 - output values at the current point
943194d53e6SMatthew G. Knepley 
944194d53e6SMatthew G. Knepley   Level: intermediate
945194d53e6SMatthew G. Knepley 
946194d53e6SMatthew G. Knepley .seealso: PetscDSGetResidual()
947194d53e6SMatthew G. Knepley @*/
9482764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetResidual(PetscDS prob, PetscInt f,
94930b9ff8bSMatthew G. Knepley                                   void (*f0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
950194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
951194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
95297b6e6e8SMatthew G. Knepley                                              PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]),
95330b9ff8bSMatthew G. Knepley                                   void (*f1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
954194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
955194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
95697b6e6e8SMatthew G. Knepley                                              PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]))
9572764a2aaSMatthew G. Knepley {
9582764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
9592764a2aaSMatthew G. Knepley 
9602764a2aaSMatthew G. Knepley   PetscFunctionBegin;
9612764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
962f866a1d0SMatthew G. Knepley   if (f0) PetscValidFunction(f0, 3);
963f866a1d0SMatthew G. Knepley   if (f1) PetscValidFunction(f1, 4);
9642764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
9652764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
9662764a2aaSMatthew G. Knepley   prob->f[f*2+0] = f0;
9672764a2aaSMatthew G. Knepley   prob->f[f*2+1] = f1;
9682764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
9692764a2aaSMatthew G. Knepley }
9702764a2aaSMatthew G. Knepley 
9713e75805dSMatthew G. Knepley /*@C
9723e75805dSMatthew G. Knepley   PetscDSHasJacobian - Signals that Jacobian functions have been set
9733e75805dSMatthew G. Knepley 
9743e75805dSMatthew G. Knepley   Not collective
9753e75805dSMatthew G. Knepley 
9763e75805dSMatthew G. Knepley   Input Parameter:
9773e75805dSMatthew G. Knepley . prob - The PetscDS
9783e75805dSMatthew G. Knepley 
9793e75805dSMatthew G. Knepley   Output Parameter:
9803e75805dSMatthew G. Knepley . hasJac - flag that pointwise function for the Jacobian has been set
9813e75805dSMatthew G. Knepley 
9823e75805dSMatthew G. Knepley   Level: intermediate
9833e75805dSMatthew G. Knepley 
9843e75805dSMatthew G. Knepley .seealso: PetscDSGetJacobianPreconditioner(), PetscDSSetJacobianPreconditioner(), PetscDSGetJacobian()
9853e75805dSMatthew G. Knepley @*/
9863e75805dSMatthew G. Knepley PetscErrorCode PetscDSHasJacobian(PetscDS prob, PetscBool *hasJac)
9873e75805dSMatthew G. Knepley {
9883e75805dSMatthew G. Knepley   PetscInt f, g, h;
9893e75805dSMatthew G. Knepley 
9903e75805dSMatthew G. Knepley   PetscFunctionBegin;
9913e75805dSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
9923e75805dSMatthew G. Knepley   *hasJac = PETSC_FALSE;
9933e75805dSMatthew G. Knepley   for (f = 0; f < prob->Nf; ++f) {
9943e75805dSMatthew G. Knepley     for (g = 0; g < prob->Nf; ++g) {
9953e75805dSMatthew G. Knepley       for (h = 0; h < 4; ++h) {
9963e75805dSMatthew G. Knepley         if (prob->g[(f*prob->Nf + g)*4+h]) *hasJac = PETSC_TRUE;
9973e75805dSMatthew G. Knepley       }
9983e75805dSMatthew G. Knepley     }
9993e75805dSMatthew G. Knepley   }
10003e75805dSMatthew G. Knepley   PetscFunctionReturn(0);
10013e75805dSMatthew G. Knepley }
10023e75805dSMatthew G. Knepley 
1003194d53e6SMatthew G. Knepley /*@C
1004194d53e6SMatthew G. Knepley   PetscDSGetJacobian - Get the pointwise Jacobian function for given test and basis field
1005194d53e6SMatthew G. Knepley 
1006194d53e6SMatthew G. Knepley   Not collective
1007194d53e6SMatthew G. Knepley 
1008194d53e6SMatthew G. Knepley   Input Parameters:
1009194d53e6SMatthew G. Knepley + prob - The PetscDS
1010194d53e6SMatthew G. Knepley . f    - The test field number
1011194d53e6SMatthew G. Knepley - g    - The field number
1012194d53e6SMatthew G. Knepley 
1013194d53e6SMatthew G. Knepley   Output Parameters:
1014194d53e6SMatthew G. Knepley + g0 - integrand for the test and basis function term
1015194d53e6SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1016194d53e6SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1017194d53e6SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1018194d53e6SMatthew G. Knepley 
1019194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1020194d53e6SMatthew G. Knepley 
1021194d53e6SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1022194d53e6SMatthew G. Knepley 
1023194d53e6SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1024194d53e6SMatthew G. Knepley 
102530b9ff8bSMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1026194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1027194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
102830b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal u_tShift, const PetscReal x[], PetscScalar g0[])
1029194d53e6SMatthew G. Knepley 
1030194d53e6SMatthew G. Knepley + dim - the spatial dimension
1031194d53e6SMatthew G. Knepley . Nf - the number of fields
1032194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1033194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1034194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1035194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1036194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1037194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1038194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1039194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1040194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1041194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1042194d53e6SMatthew G. Knepley . t - current time
10432aa1fc23SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1044194d53e6SMatthew G. Knepley . x - coordinates of the current point
104597b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
104697b6e6e8SMatthew G. Knepley . constants - constant parameters
1047194d53e6SMatthew G. Knepley - g0 - output values at the current point
1048194d53e6SMatthew G. Knepley 
1049194d53e6SMatthew G. Knepley   Level: intermediate
1050194d53e6SMatthew G. Knepley 
1051194d53e6SMatthew G. Knepley .seealso: PetscDSSetJacobian()
1052194d53e6SMatthew G. Knepley @*/
10532764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetJacobian(PetscDS prob, PetscInt f, PetscInt g,
105430b9ff8bSMatthew G. Knepley                                   void (**g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1055194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1056194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
105797b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
105830b9ff8bSMatthew G. Knepley                                   void (**g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1059194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1060194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
106197b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
106230b9ff8bSMatthew G. Knepley                                   void (**g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1063194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1064194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
106597b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
106630b9ff8bSMatthew G. Knepley                                   void (**g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1067194d53e6SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1068194d53e6SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
106997b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
10702764a2aaSMatthew G. Knepley {
10712764a2aaSMatthew G. Knepley   PetscFunctionBegin;
10722764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
10732764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
10742764a2aaSMatthew G. Knepley   if ((g < 0) || (g >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", g, prob->Nf);
10752764a2aaSMatthew G. Knepley   if (g0) {PetscValidPointer(g0, 4); *g0 = prob->g[(f*prob->Nf + g)*4+0];}
10762764a2aaSMatthew G. Knepley   if (g1) {PetscValidPointer(g1, 5); *g1 = prob->g[(f*prob->Nf + g)*4+1];}
10772764a2aaSMatthew G. Knepley   if (g2) {PetscValidPointer(g2, 6); *g2 = prob->g[(f*prob->Nf + g)*4+2];}
10782764a2aaSMatthew G. Knepley   if (g3) {PetscValidPointer(g3, 7); *g3 = prob->g[(f*prob->Nf + g)*4+3];}
10792764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
10802764a2aaSMatthew G. Knepley }
10812764a2aaSMatthew G. Knepley 
1082194d53e6SMatthew G. Knepley /*@C
1083194d53e6SMatthew G. Knepley   PetscDSSetJacobian - Set the pointwise Jacobian function for given test and basis fields
1084194d53e6SMatthew G. Knepley 
1085194d53e6SMatthew G. Knepley   Not collective
1086194d53e6SMatthew G. Knepley 
1087194d53e6SMatthew G. Knepley   Input Parameters:
1088194d53e6SMatthew G. Knepley + prob - The PetscDS
1089194d53e6SMatthew G. Knepley . f    - The test field number
1090194d53e6SMatthew G. Knepley . g    - The field number
1091194d53e6SMatthew G. Knepley . g0 - integrand for the test and basis function term
1092194d53e6SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1093194d53e6SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1094194d53e6SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1095194d53e6SMatthew G. Knepley 
1096194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1097194d53e6SMatthew G. Knepley 
1098194d53e6SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1099194d53e6SMatthew G. Knepley 
1100194d53e6SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1101194d53e6SMatthew G. Knepley 
110230b9ff8bSMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1103194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1104194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
110530b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], PetscScalar g0[])
1106194d53e6SMatthew G. Knepley 
1107194d53e6SMatthew G. Knepley + dim - the spatial dimension
1108194d53e6SMatthew G. Knepley . Nf - the number of fields
1109194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1110194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1111194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1112194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1113194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1114194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1115194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1116194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1117194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1118194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1119194d53e6SMatthew G. Knepley . t - current time
11202aa1fc23SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1121194d53e6SMatthew G. Knepley . x - coordinates of the current point
112297b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
112397b6e6e8SMatthew G. Knepley . constants - constant parameters
1124194d53e6SMatthew G. Knepley - g0 - output values at the current point
1125194d53e6SMatthew G. Knepley 
1126194d53e6SMatthew G. Knepley   Level: intermediate
1127194d53e6SMatthew G. Knepley 
1128194d53e6SMatthew G. Knepley .seealso: PetscDSGetJacobian()
1129194d53e6SMatthew G. Knepley @*/
11302764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetJacobian(PetscDS prob, PetscInt f, PetscInt g,
113130b9ff8bSMatthew G. Knepley                                   void (*g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1132194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1133194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
113497b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
113530b9ff8bSMatthew G. Knepley                                   void (*g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1136194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1137194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
113897b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
113930b9ff8bSMatthew G. Knepley                                   void (*g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1140194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1141194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
114297b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
114330b9ff8bSMatthew G. Knepley                                   void (*g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1144194d53e6SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1145194d53e6SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
114697b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
11472764a2aaSMatthew G. Knepley {
11482764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
11492764a2aaSMatthew G. Knepley 
11502764a2aaSMatthew G. Knepley   PetscFunctionBegin;
11512764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
11522764a2aaSMatthew G. Knepley   if (g0) PetscValidFunction(g0, 4);
11532764a2aaSMatthew G. Knepley   if (g1) PetscValidFunction(g1, 5);
11542764a2aaSMatthew G. Knepley   if (g2) PetscValidFunction(g2, 6);
11552764a2aaSMatthew G. Knepley   if (g3) PetscValidFunction(g3, 7);
11562764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
11572764a2aaSMatthew G. Knepley   if (g < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", g);
11582764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, PetscMax(f, g)+1);CHKERRQ(ierr);
11592764a2aaSMatthew G. Knepley   prob->g[(f*prob->Nf + g)*4+0] = g0;
11602764a2aaSMatthew G. Knepley   prob->g[(f*prob->Nf + g)*4+1] = g1;
11612764a2aaSMatthew G. Knepley   prob->g[(f*prob->Nf + g)*4+2] = g2;
11622764a2aaSMatthew G. Knepley   prob->g[(f*prob->Nf + g)*4+3] = g3;
11632764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
11642764a2aaSMatthew G. Knepley }
11652764a2aaSMatthew G. Knepley 
1166475e0ac9SMatthew G. Knepley /*@C
1167475e0ac9SMatthew G. Knepley   PetscDSHasJacobianPreconditioner - Signals that a Jacobian preconditioner matrix has been set
1168475e0ac9SMatthew G. Knepley 
1169475e0ac9SMatthew G. Knepley   Not collective
1170475e0ac9SMatthew G. Knepley 
1171475e0ac9SMatthew G. Knepley   Input Parameter:
1172475e0ac9SMatthew G. Knepley . prob - The PetscDS
1173475e0ac9SMatthew G. Knepley 
1174475e0ac9SMatthew G. Knepley   Output Parameter:
1175475e0ac9SMatthew G. Knepley . hasJacPre - flag that pointwise function for Jacobian preconditioner matrix has been set
1176475e0ac9SMatthew G. Knepley 
1177475e0ac9SMatthew G. Knepley   Level: intermediate
1178475e0ac9SMatthew G. Knepley 
1179475e0ac9SMatthew G. Knepley .seealso: PetscDSGetJacobianPreconditioner(), PetscDSSetJacobianPreconditioner(), PetscDSGetJacobian()
1180475e0ac9SMatthew G. Knepley @*/
1181475e0ac9SMatthew G. Knepley PetscErrorCode PetscDSHasJacobianPreconditioner(PetscDS prob, PetscBool *hasJacPre)
1182475e0ac9SMatthew G. Knepley {
1183475e0ac9SMatthew G. Knepley   PetscInt f, g, h;
1184475e0ac9SMatthew G. Knepley 
1185475e0ac9SMatthew G. Knepley   PetscFunctionBegin;
1186475e0ac9SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1187475e0ac9SMatthew G. Knepley   *hasJacPre = PETSC_FALSE;
1188475e0ac9SMatthew G. Knepley   for (f = 0; f < prob->Nf; ++f) {
1189475e0ac9SMatthew G. Knepley     for (g = 0; g < prob->Nf; ++g) {
1190475e0ac9SMatthew G. Knepley       for (h = 0; h < 4; ++h) {
1191475e0ac9SMatthew G. Knepley         if (prob->gp[(f*prob->Nf + g)*4+h]) *hasJacPre = PETSC_TRUE;
1192475e0ac9SMatthew G. Knepley       }
1193475e0ac9SMatthew G. Knepley     }
1194475e0ac9SMatthew G. Knepley   }
1195475e0ac9SMatthew G. Knepley   PetscFunctionReturn(0);
1196475e0ac9SMatthew G. Knepley }
1197475e0ac9SMatthew G. Knepley 
1198475e0ac9SMatthew G. Knepley /*@C
1199475e0ac9SMatthew G. Knepley   PetscDSGetJacobianPreconditioner - Get the pointwise Jacobian preconditioner function for given test and basis field. If this is missing, the system matrix is used to build the preconditioner.
1200475e0ac9SMatthew G. Knepley 
1201475e0ac9SMatthew G. Knepley   Not collective
1202475e0ac9SMatthew G. Knepley 
1203475e0ac9SMatthew G. Knepley   Input Parameters:
1204475e0ac9SMatthew G. Knepley + prob - The PetscDS
1205475e0ac9SMatthew G. Knepley . f    - The test field number
1206475e0ac9SMatthew G. Knepley - g    - The field number
1207475e0ac9SMatthew G. Knepley 
1208475e0ac9SMatthew G. Knepley   Output Parameters:
1209475e0ac9SMatthew G. Knepley + g0 - integrand for the test and basis function term
1210475e0ac9SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1211475e0ac9SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1212475e0ac9SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1213475e0ac9SMatthew G. Knepley 
1214475e0ac9SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1215475e0ac9SMatthew G. Knepley 
1216475e0ac9SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1217475e0ac9SMatthew G. Knepley 
1218475e0ac9SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1219475e0ac9SMatthew G. Knepley 
1220475e0ac9SMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1221475e0ac9SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1222475e0ac9SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
1223475e0ac9SMatthew G. Knepley $    PetscReal t, const PetscReal u_tShift, const PetscReal x[], PetscScalar g0[])
1224475e0ac9SMatthew G. Knepley 
1225475e0ac9SMatthew G. Knepley + dim - the spatial dimension
1226475e0ac9SMatthew G. Knepley . Nf - the number of fields
1227475e0ac9SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1228475e0ac9SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1229475e0ac9SMatthew G. Knepley . u - each field evaluated at the current point
1230475e0ac9SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1231475e0ac9SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1232475e0ac9SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1233475e0ac9SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1234475e0ac9SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1235475e0ac9SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1236475e0ac9SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1237475e0ac9SMatthew G. Knepley . t - current time
1238475e0ac9SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1239475e0ac9SMatthew G. Knepley . x - coordinates of the current point
124097b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
124197b6e6e8SMatthew G. Knepley . constants - constant parameters
1242475e0ac9SMatthew G. Knepley - g0 - output values at the current point
1243475e0ac9SMatthew G. Knepley 
1244475e0ac9SMatthew G. Knepley   Level: intermediate
1245475e0ac9SMatthew G. Knepley 
1246475e0ac9SMatthew G. Knepley .seealso: PetscDSSetJacobianPreconditioner(), PetscDSGetJacobian()
1247475e0ac9SMatthew G. Knepley @*/
1248475e0ac9SMatthew G. Knepley PetscErrorCode PetscDSGetJacobianPreconditioner(PetscDS prob, PetscInt f, PetscInt g,
1249475e0ac9SMatthew G. Knepley                                   void (**g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1250475e0ac9SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1251475e0ac9SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
125297b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
1253475e0ac9SMatthew G. Knepley                                   void (**g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1254475e0ac9SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1255475e0ac9SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
125697b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
1257475e0ac9SMatthew G. Knepley                                   void (**g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1258475e0ac9SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1259475e0ac9SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
126097b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
1261475e0ac9SMatthew G. Knepley                                   void (**g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1262475e0ac9SMatthew G. Knepley                                               const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1263475e0ac9SMatthew G. Knepley                                               const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
126497b6e6e8SMatthew G. Knepley                                               PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
1265475e0ac9SMatthew G. Knepley {
1266475e0ac9SMatthew G. Knepley   PetscFunctionBegin;
1267475e0ac9SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1268475e0ac9SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
1269475e0ac9SMatthew G. Knepley   if ((g < 0) || (g >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", g, prob->Nf);
1270475e0ac9SMatthew G. Knepley   if (g0) {PetscValidPointer(g0, 4); *g0 = prob->gp[(f*prob->Nf + g)*4+0];}
1271475e0ac9SMatthew G. Knepley   if (g1) {PetscValidPointer(g1, 5); *g1 = prob->gp[(f*prob->Nf + g)*4+1];}
1272475e0ac9SMatthew G. Knepley   if (g2) {PetscValidPointer(g2, 6); *g2 = prob->gp[(f*prob->Nf + g)*4+2];}
1273475e0ac9SMatthew G. Knepley   if (g3) {PetscValidPointer(g3, 7); *g3 = prob->gp[(f*prob->Nf + g)*4+3];}
1274475e0ac9SMatthew G. Knepley   PetscFunctionReturn(0);
1275475e0ac9SMatthew G. Knepley }
1276475e0ac9SMatthew G. Knepley 
1277475e0ac9SMatthew G. Knepley /*@C
1278475e0ac9SMatthew G. Knepley   PetscDSSetJacobianPreconditioner - Set the pointwise Jacobian preconditioner function for given test and basis fields. If this is missing, the system matrix is used to build the preconditioner.
1279475e0ac9SMatthew G. Knepley 
1280475e0ac9SMatthew G. Knepley   Not collective
1281475e0ac9SMatthew G. Knepley 
1282475e0ac9SMatthew G. Knepley   Input Parameters:
1283475e0ac9SMatthew G. Knepley + prob - The PetscDS
1284475e0ac9SMatthew G. Knepley . f    - The test field number
1285475e0ac9SMatthew G. Knepley . g    - The field number
1286475e0ac9SMatthew G. Knepley . g0 - integrand for the test and basis function term
1287475e0ac9SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1288475e0ac9SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1289475e0ac9SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1290475e0ac9SMatthew G. Knepley 
1291475e0ac9SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1292475e0ac9SMatthew G. Knepley 
1293475e0ac9SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1294475e0ac9SMatthew G. Knepley 
1295475e0ac9SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1296475e0ac9SMatthew G. Knepley 
1297475e0ac9SMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1298475e0ac9SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1299475e0ac9SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
1300475e0ac9SMatthew G. Knepley $    PetscReal t, const PetscReal x[], PetscScalar g0[])
1301475e0ac9SMatthew G. Knepley 
1302475e0ac9SMatthew G. Knepley + dim - the spatial dimension
1303475e0ac9SMatthew G. Knepley . Nf - the number of fields
1304475e0ac9SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1305475e0ac9SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1306475e0ac9SMatthew G. Knepley . u - each field evaluated at the current point
1307475e0ac9SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1308475e0ac9SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1309475e0ac9SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1310475e0ac9SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1311475e0ac9SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1312475e0ac9SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1313475e0ac9SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1314475e0ac9SMatthew G. Knepley . t - current time
1315475e0ac9SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1316475e0ac9SMatthew G. Knepley . x - coordinates of the current point
131797b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
131897b6e6e8SMatthew G. Knepley . constants - constant parameters
1319475e0ac9SMatthew G. Knepley - g0 - output values at the current point
1320475e0ac9SMatthew G. Knepley 
1321475e0ac9SMatthew G. Knepley   Level: intermediate
1322475e0ac9SMatthew G. Knepley 
1323475e0ac9SMatthew G. Knepley .seealso: PetscDSGetJacobianPreconditioner(), PetscDSSetJacobian()
1324475e0ac9SMatthew G. Knepley @*/
1325475e0ac9SMatthew G. Knepley PetscErrorCode PetscDSSetJacobianPreconditioner(PetscDS prob, PetscInt f, PetscInt g,
1326475e0ac9SMatthew G. Knepley                                   void (*g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1327475e0ac9SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1328475e0ac9SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
132997b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
1330475e0ac9SMatthew G. Knepley                                   void (*g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1331475e0ac9SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1332475e0ac9SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
133397b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
1334475e0ac9SMatthew G. Knepley                                   void (*g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1335475e0ac9SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1336475e0ac9SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
133797b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
1338475e0ac9SMatthew G. Knepley                                   void (*g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1339475e0ac9SMatthew G. Knepley                                              const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1340475e0ac9SMatthew G. Knepley                                              const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
134197b6e6e8SMatthew G. Knepley                                              PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
1342475e0ac9SMatthew G. Knepley {
1343475e0ac9SMatthew G. Knepley   PetscErrorCode ierr;
1344475e0ac9SMatthew G. Knepley 
1345475e0ac9SMatthew G. Knepley   PetscFunctionBegin;
1346475e0ac9SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1347475e0ac9SMatthew G. Knepley   if (g0) PetscValidFunction(g0, 4);
1348475e0ac9SMatthew G. Knepley   if (g1) PetscValidFunction(g1, 5);
1349475e0ac9SMatthew G. Knepley   if (g2) PetscValidFunction(g2, 6);
1350475e0ac9SMatthew G. Knepley   if (g3) PetscValidFunction(g3, 7);
1351475e0ac9SMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
1352475e0ac9SMatthew G. Knepley   if (g < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", g);
1353475e0ac9SMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, PetscMax(f, g)+1);CHKERRQ(ierr);
1354475e0ac9SMatthew G. Knepley   prob->gp[(f*prob->Nf + g)*4+0] = g0;
1355475e0ac9SMatthew G. Knepley   prob->gp[(f*prob->Nf + g)*4+1] = g1;
1356475e0ac9SMatthew G. Knepley   prob->gp[(f*prob->Nf + g)*4+2] = g2;
1357475e0ac9SMatthew G. Knepley   prob->gp[(f*prob->Nf + g)*4+3] = g3;
1358475e0ac9SMatthew G. Knepley   PetscFunctionReturn(0);
1359475e0ac9SMatthew G. Knepley }
1360475e0ac9SMatthew G. Knepley 
1361b7e05686SMatthew G. Knepley /*@C
1362b7e05686SMatthew G. Knepley   PetscDSHasDynamicJacobian - Signals that a dynamic Jacobian, dF/du_t, has been set
1363b7e05686SMatthew G. Knepley 
1364b7e05686SMatthew G. Knepley   Not collective
1365b7e05686SMatthew G. Knepley 
1366b7e05686SMatthew G. Knepley   Input Parameter:
1367b7e05686SMatthew G. Knepley . prob - The PetscDS
1368b7e05686SMatthew G. Knepley 
1369b7e05686SMatthew G. Knepley   Output Parameter:
1370b7e05686SMatthew G. Knepley . hasDynJac - flag that pointwise function for dynamic Jacobian has been set
1371b7e05686SMatthew G. Knepley 
1372b7e05686SMatthew G. Knepley   Level: intermediate
1373b7e05686SMatthew G. Knepley 
1374b7e05686SMatthew G. Knepley .seealso: PetscDSGetDynamicJacobian(), PetscDSSetDynamicJacobian(), PetscDSGetJacobian()
1375b7e05686SMatthew G. Knepley @*/
1376b7e05686SMatthew G. Knepley PetscErrorCode PetscDSHasDynamicJacobian(PetscDS prob, PetscBool *hasDynJac)
1377b7e05686SMatthew G. Knepley {
1378b7e05686SMatthew G. Knepley   PetscInt f, g, h;
1379b7e05686SMatthew G. Knepley 
1380b7e05686SMatthew G. Knepley   PetscFunctionBegin;
1381b7e05686SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1382b7e05686SMatthew G. Knepley   *hasDynJac = PETSC_FALSE;
1383b7e05686SMatthew G. Knepley   for (f = 0; f < prob->Nf; ++f) {
1384b7e05686SMatthew G. Knepley     for (g = 0; g < prob->Nf; ++g) {
1385b7e05686SMatthew G. Knepley       for (h = 0; h < 4; ++h) {
1386b7e05686SMatthew G. Knepley         if (prob->gt[(f*prob->Nf + g)*4+h]) *hasDynJac = PETSC_TRUE;
1387b7e05686SMatthew G. Knepley       }
1388b7e05686SMatthew G. Knepley     }
1389b7e05686SMatthew G. Knepley   }
1390b7e05686SMatthew G. Knepley   PetscFunctionReturn(0);
1391b7e05686SMatthew G. Knepley }
1392b7e05686SMatthew G. Knepley 
1393b7e05686SMatthew G. Knepley /*@C
1394b7e05686SMatthew G. Knepley   PetscDSGetDynamicJacobian - Get the pointwise dynamic Jacobian, dF/du_t, function for given test and basis field
1395b7e05686SMatthew G. Knepley 
1396b7e05686SMatthew G. Knepley   Not collective
1397b7e05686SMatthew G. Knepley 
1398b7e05686SMatthew G. Knepley   Input Parameters:
1399b7e05686SMatthew G. Knepley + prob - The PetscDS
1400b7e05686SMatthew G. Knepley . f    - The test field number
1401b7e05686SMatthew G. Knepley - g    - The field number
1402b7e05686SMatthew G. Knepley 
1403b7e05686SMatthew G. Knepley   Output Parameters:
1404b7e05686SMatthew G. Knepley + g0 - integrand for the test and basis function term
1405b7e05686SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1406b7e05686SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1407b7e05686SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1408b7e05686SMatthew G. Knepley 
1409b7e05686SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1410b7e05686SMatthew G. Knepley 
1411b7e05686SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1412b7e05686SMatthew G. Knepley 
1413b7e05686SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1414b7e05686SMatthew G. Knepley 
1415b7e05686SMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1416b7e05686SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1417b7e05686SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
1418b7e05686SMatthew G. Knepley $    PetscReal t, const PetscReal u_tShift, const PetscReal x[], PetscScalar g0[])
1419b7e05686SMatthew G. Knepley 
1420b7e05686SMatthew G. Knepley + dim - the spatial dimension
1421b7e05686SMatthew G. Knepley . Nf - the number of fields
1422b7e05686SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1423b7e05686SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1424b7e05686SMatthew G. Knepley . u - each field evaluated at the current point
1425b7e05686SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1426b7e05686SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1427b7e05686SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1428b7e05686SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1429b7e05686SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1430b7e05686SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1431b7e05686SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1432b7e05686SMatthew G. Knepley . t - current time
1433b7e05686SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1434b7e05686SMatthew G. Knepley . x - coordinates of the current point
143597b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
143697b6e6e8SMatthew G. Knepley . constants - constant parameters
1437b7e05686SMatthew G. Knepley - g0 - output values at the current point
1438b7e05686SMatthew G. Knepley 
1439b7e05686SMatthew G. Knepley   Level: intermediate
1440b7e05686SMatthew G. Knepley 
1441b7e05686SMatthew G. Knepley .seealso: PetscDSSetJacobian()
1442b7e05686SMatthew G. Knepley @*/
1443b7e05686SMatthew G. Knepley PetscErrorCode PetscDSGetDynamicJacobian(PetscDS prob, PetscInt f, PetscInt g,
1444b7e05686SMatthew G. Knepley                                          void (**g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1445b7e05686SMatthew G. Knepley                                                      const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1446b7e05686SMatthew G. Knepley                                                      const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
144797b6e6e8SMatthew G. Knepley                                                      PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
1448b7e05686SMatthew G. Knepley                                          void (**g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1449b7e05686SMatthew G. Knepley                                                      const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1450b7e05686SMatthew G. Knepley                                                      const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
145197b6e6e8SMatthew G. Knepley                                                      PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
1452b7e05686SMatthew G. Knepley                                          void (**g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1453b7e05686SMatthew G. Knepley                                                      const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1454b7e05686SMatthew G. Knepley                                                      const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
145597b6e6e8SMatthew G. Knepley                                                      PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
1456b7e05686SMatthew G. Knepley                                          void (**g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1457b7e05686SMatthew G. Knepley                                                      const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1458b7e05686SMatthew G. Knepley                                                      const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
145997b6e6e8SMatthew G. Knepley                                                      PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
1460b7e05686SMatthew G. Knepley {
1461b7e05686SMatthew G. Knepley   PetscFunctionBegin;
1462b7e05686SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1463b7e05686SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
1464b7e05686SMatthew G. Knepley   if ((g < 0) || (g >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", g, prob->Nf);
1465b7e05686SMatthew G. Knepley   if (g0) {PetscValidPointer(g0, 4); *g0 = prob->gt[(f*prob->Nf + g)*4+0];}
1466b7e05686SMatthew G. Knepley   if (g1) {PetscValidPointer(g1, 5); *g1 = prob->gt[(f*prob->Nf + g)*4+1];}
1467b7e05686SMatthew G. Knepley   if (g2) {PetscValidPointer(g2, 6); *g2 = prob->gt[(f*prob->Nf + g)*4+2];}
1468b7e05686SMatthew G. Knepley   if (g3) {PetscValidPointer(g3, 7); *g3 = prob->gt[(f*prob->Nf + g)*4+3];}
1469b7e05686SMatthew G. Knepley   PetscFunctionReturn(0);
1470b7e05686SMatthew G. Knepley }
1471b7e05686SMatthew G. Knepley 
1472b7e05686SMatthew G. Knepley /*@C
1473b7e05686SMatthew G. Knepley   PetscDSSetDynamicJacobian - Set the pointwise dynamic Jacobian, dF/du_t, function for given test and basis fields
1474b7e05686SMatthew G. Knepley 
1475b7e05686SMatthew G. Knepley   Not collective
1476b7e05686SMatthew G. Knepley 
1477b7e05686SMatthew G. Knepley   Input Parameters:
1478b7e05686SMatthew G. Knepley + prob - The PetscDS
1479b7e05686SMatthew G. Knepley . f    - The test field number
1480b7e05686SMatthew G. Knepley . g    - The field number
1481b7e05686SMatthew G. Knepley . g0 - integrand for the test and basis function term
1482b7e05686SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1483b7e05686SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1484b7e05686SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1485b7e05686SMatthew G. Knepley 
1486b7e05686SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1487b7e05686SMatthew G. Knepley 
1488b7e05686SMatthew G. Knepley   \int_\Omega \phi g_0(u, u_t, \nabla u, x, t) \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \nabla \psi
1489b7e05686SMatthew G. Knepley 
1490b7e05686SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1491b7e05686SMatthew G. Knepley 
1492b7e05686SMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1493b7e05686SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1494b7e05686SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
1495b7e05686SMatthew G. Knepley $    PetscReal t, const PetscReal x[], PetscScalar g0[])
1496b7e05686SMatthew G. Knepley 
1497b7e05686SMatthew G. Knepley + dim - the spatial dimension
1498b7e05686SMatthew G. Knepley . Nf - the number of fields
1499b7e05686SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1500b7e05686SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1501b7e05686SMatthew G. Knepley . u - each field evaluated at the current point
1502b7e05686SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1503b7e05686SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1504b7e05686SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1505b7e05686SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1506b7e05686SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1507b7e05686SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1508b7e05686SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1509b7e05686SMatthew G. Knepley . t - current time
1510b7e05686SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1511b7e05686SMatthew G. Knepley . x - coordinates of the current point
151297b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
151397b6e6e8SMatthew G. Knepley . constants - constant parameters
1514b7e05686SMatthew G. Knepley - g0 - output values at the current point
1515b7e05686SMatthew G. Knepley 
1516b7e05686SMatthew G. Knepley   Level: intermediate
1517b7e05686SMatthew G. Knepley 
1518b7e05686SMatthew G. Knepley .seealso: PetscDSGetJacobian()
1519b7e05686SMatthew G. Knepley @*/
1520b7e05686SMatthew G. Knepley PetscErrorCode PetscDSSetDynamicJacobian(PetscDS prob, PetscInt f, PetscInt g,
1521b7e05686SMatthew G. Knepley                                          void (*g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1522b7e05686SMatthew G. Knepley                                                     const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1523b7e05686SMatthew G. Knepley                                                     const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
152497b6e6e8SMatthew G. Knepley                                                     PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
1525b7e05686SMatthew G. Knepley                                          void (*g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1526b7e05686SMatthew G. Knepley                                                     const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1527b7e05686SMatthew G. Knepley                                                     const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
152897b6e6e8SMatthew G. Knepley                                                     PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
1529b7e05686SMatthew G. Knepley                                          void (*g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1530b7e05686SMatthew G. Knepley                                                     const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1531b7e05686SMatthew G. Knepley                                                     const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
153297b6e6e8SMatthew G. Knepley                                                     PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
1533b7e05686SMatthew G. Knepley                                          void (*g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1534b7e05686SMatthew G. Knepley                                                     const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1535b7e05686SMatthew G. Knepley                                                     const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
153697b6e6e8SMatthew G. Knepley                                                     PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
1537b7e05686SMatthew G. Knepley {
1538b7e05686SMatthew G. Knepley   PetscErrorCode ierr;
1539b7e05686SMatthew G. Knepley 
1540b7e05686SMatthew G. Knepley   PetscFunctionBegin;
1541b7e05686SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1542b7e05686SMatthew G. Knepley   if (g0) PetscValidFunction(g0, 4);
1543b7e05686SMatthew G. Knepley   if (g1) PetscValidFunction(g1, 5);
1544b7e05686SMatthew G. Knepley   if (g2) PetscValidFunction(g2, 6);
1545b7e05686SMatthew G. Knepley   if (g3) PetscValidFunction(g3, 7);
1546b7e05686SMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
1547b7e05686SMatthew G. Knepley   if (g < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", g);
1548b7e05686SMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, PetscMax(f, g)+1);CHKERRQ(ierr);
1549b7e05686SMatthew G. Knepley   prob->gt[(f*prob->Nf + g)*4+0] = g0;
1550b7e05686SMatthew G. Knepley   prob->gt[(f*prob->Nf + g)*4+1] = g1;
1551b7e05686SMatthew G. Knepley   prob->gt[(f*prob->Nf + g)*4+2] = g2;
1552b7e05686SMatthew G. Knepley   prob->gt[(f*prob->Nf + g)*4+3] = g3;
1553b7e05686SMatthew G. Knepley   PetscFunctionReturn(0);
1554b7e05686SMatthew G. Knepley }
1555b7e05686SMatthew G. Knepley 
15560c2f2876SMatthew G. Knepley /*@C
15570c2f2876SMatthew G. Knepley   PetscDSGetRiemannSolver - Returns the Riemann solver for the given field
15580c2f2876SMatthew G. Knepley 
15590c2f2876SMatthew G. Knepley   Not collective
15600c2f2876SMatthew G. Knepley 
15610c2f2876SMatthew G. Knepley   Input Arguments:
15620c2f2876SMatthew G. Knepley + prob - The PetscDS object
15630c2f2876SMatthew G. Knepley - f    - The field number
15640c2f2876SMatthew G. Knepley 
15650c2f2876SMatthew G. Knepley   Output Argument:
15660c2f2876SMatthew G. Knepley . r    - Riemann solver
15670c2f2876SMatthew G. Knepley 
15680c2f2876SMatthew G. Knepley   Calling sequence for r:
15690c2f2876SMatthew G. Knepley 
15705db36cf9SMatthew G. Knepley $ r(PetscInt dim, PetscInt Nf, const PetscReal x[], const PetscReal n[], const PetscScalar uL[], const PetscScalar uR[], PetscScalar flux[], void *ctx)
15710c2f2876SMatthew G. Knepley 
15725db36cf9SMatthew G. Knepley + dim  - The spatial dimension
15735db36cf9SMatthew G. Knepley . Nf   - The number of fields
15745db36cf9SMatthew G. Knepley . x    - The coordinates at a point on the interface
15750c2f2876SMatthew G. Knepley . n    - The normal vector to the interface
15760c2f2876SMatthew G. Knepley . uL   - The state vector to the left of the interface
15770c2f2876SMatthew G. Knepley . uR   - The state vector to the right of the interface
15780c2f2876SMatthew G. Knepley . flux - output array of flux through the interface
157997b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
158097b6e6e8SMatthew G. Knepley . constants - constant parameters
15810c2f2876SMatthew G. Knepley - ctx  - optional user context
15820c2f2876SMatthew G. Knepley 
15830c2f2876SMatthew G. Knepley   Level: intermediate
15840c2f2876SMatthew G. Knepley 
15850c2f2876SMatthew G. Knepley .seealso: PetscDSSetRiemannSolver()
15860c2f2876SMatthew G. Knepley @*/
15870c2f2876SMatthew G. Knepley PetscErrorCode PetscDSGetRiemannSolver(PetscDS prob, PetscInt f,
158897b6e6e8SMatthew G. Knepley                                        void (**r)(PetscInt dim, PetscInt Nf, const PetscReal x[], const PetscReal n[], const PetscScalar uL[], const PetscScalar uR[], PetscInt numConstants, const PetscScalar constants[], PetscScalar flux[], void *ctx))
15890c2f2876SMatthew G. Knepley {
15900c2f2876SMatthew G. Knepley   PetscFunctionBegin;
15910c2f2876SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
15920c2f2876SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
15930c2f2876SMatthew G. Knepley   PetscValidPointer(r, 3);
15940c2f2876SMatthew G. Knepley   *r = prob->r[f];
15950c2f2876SMatthew G. Knepley   PetscFunctionReturn(0);
15960c2f2876SMatthew G. Knepley }
15970c2f2876SMatthew G. Knepley 
15980c2f2876SMatthew G. Knepley /*@C
15990c2f2876SMatthew G. Knepley   PetscDSSetRiemannSolver - Sets the Riemann solver for the given field
16000c2f2876SMatthew G. Knepley 
16010c2f2876SMatthew G. Knepley   Not collective
16020c2f2876SMatthew G. Knepley 
16030c2f2876SMatthew G. Knepley   Input Arguments:
16040c2f2876SMatthew G. Knepley + prob - The PetscDS object
16050c2f2876SMatthew G. Knepley . f    - The field number
16060c2f2876SMatthew G. Knepley - r    - Riemann solver
16070c2f2876SMatthew G. Knepley 
16080c2f2876SMatthew G. Knepley   Calling sequence for r:
16090c2f2876SMatthew G. Knepley 
16105db36cf9SMatthew G. Knepley $ r(PetscInt dim, PetscInt Nf, const PetscReal x[], const PetscReal n[], const PetscScalar uL[], const PetscScalar uR[], PetscScalar flux[], void *ctx)
16110c2f2876SMatthew G. Knepley 
16125db36cf9SMatthew G. Knepley + dim  - The spatial dimension
16135db36cf9SMatthew G. Knepley . Nf   - The number of fields
16145db36cf9SMatthew G. Knepley . x    - The coordinates at a point on the interface
16150c2f2876SMatthew G. Knepley . n    - The normal vector to the interface
16160c2f2876SMatthew G. Knepley . uL   - The state vector to the left of the interface
16170c2f2876SMatthew G. Knepley . uR   - The state vector to the right of the interface
16180c2f2876SMatthew G. Knepley . flux - output array of flux through the interface
161997b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
162097b6e6e8SMatthew G. Knepley . constants - constant parameters
16210c2f2876SMatthew G. Knepley - ctx  - optional user context
16220c2f2876SMatthew G. Knepley 
16230c2f2876SMatthew G. Knepley   Level: intermediate
16240c2f2876SMatthew G. Knepley 
16250c2f2876SMatthew G. Knepley .seealso: PetscDSGetRiemannSolver()
16260c2f2876SMatthew G. Knepley @*/
16270c2f2876SMatthew G. Knepley PetscErrorCode PetscDSSetRiemannSolver(PetscDS prob, PetscInt f,
162897b6e6e8SMatthew G. Knepley                                        void (*r)(PetscInt dim, PetscInt Nf, const PetscReal x[], const PetscReal n[], const PetscScalar uL[], const PetscScalar uR[], PetscInt numConstants, const PetscScalar constants[], PetscScalar flux[], void *ctx))
16290c2f2876SMatthew G. Knepley {
16300c2f2876SMatthew G. Knepley   PetscErrorCode ierr;
16310c2f2876SMatthew G. Knepley 
16320c2f2876SMatthew G. Knepley   PetscFunctionBegin;
16330c2f2876SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
1634de716cbcSToby Isaac   if (r) PetscValidFunction(r, 3);
16350c2f2876SMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
16360c2f2876SMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
16370c2f2876SMatthew G. Knepley   prob->r[f] = r;
16380c2f2876SMatthew G. Knepley   PetscFunctionReturn(0);
16390c2f2876SMatthew G. Knepley }
16400c2f2876SMatthew G. Knepley 
16410c2f2876SMatthew G. Knepley PetscErrorCode PetscDSGetContext(PetscDS prob, PetscInt f, void **ctx)
16420c2f2876SMatthew G. Knepley {
16430c2f2876SMatthew G. Knepley   PetscFunctionBegin;
16440c2f2876SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
16450c2f2876SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
16460c2f2876SMatthew G. Knepley   PetscValidPointer(ctx, 3);
16470c2f2876SMatthew G. Knepley   *ctx = prob->ctx[f];
16480c2f2876SMatthew G. Knepley   PetscFunctionReturn(0);
16490c2f2876SMatthew G. Knepley }
16500c2f2876SMatthew G. Knepley 
16510c2f2876SMatthew G. Knepley PetscErrorCode PetscDSSetContext(PetscDS prob, PetscInt f, void *ctx)
16520c2f2876SMatthew G. Knepley {
16530c2f2876SMatthew G. Knepley   PetscErrorCode ierr;
16540c2f2876SMatthew G. Knepley 
16550c2f2876SMatthew G. Knepley   PetscFunctionBegin;
16560c2f2876SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
16570c2f2876SMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
16580c2f2876SMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
16590c2f2876SMatthew G. Knepley   prob->ctx[f] = ctx;
16600c2f2876SMatthew G. Knepley   PetscFunctionReturn(0);
16610c2f2876SMatthew G. Knepley }
16620c2f2876SMatthew G. Knepley 
1663194d53e6SMatthew G. Knepley /*@C
1664194d53e6SMatthew G. Knepley   PetscDSGetBdResidual - Get the pointwise boundary residual function for a given test field
1665194d53e6SMatthew G. Knepley 
1666194d53e6SMatthew G. Knepley   Not collective
1667194d53e6SMatthew G. Knepley 
1668194d53e6SMatthew G. Knepley   Input Parameters:
1669194d53e6SMatthew G. Knepley + prob - The PetscDS
1670194d53e6SMatthew G. Knepley - f    - The test field number
1671194d53e6SMatthew G. Knepley 
1672194d53e6SMatthew G. Knepley   Output Parameters:
1673194d53e6SMatthew G. Knepley + f0 - boundary integrand for the test function term
1674194d53e6SMatthew G. Knepley - f1 - boundary integrand for the test function gradient term
1675194d53e6SMatthew G. Knepley 
1676194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1677194d53e6SMatthew G. Knepley 
1678194d53e6SMatthew G. Knepley   \int_\Gamma \phi {\vec f}_0(u, u_t, \nabla u, x, t) \cdot \hat n + \nabla\phi \cdot {\overleftrightarrow f}_1(u, u_t, \nabla u, x, t) \cdot \hat n
1679194d53e6SMatthew G. Knepley 
1680194d53e6SMatthew G. Knepley The calling sequence for the callbacks f0 and f1 is given by:
1681194d53e6SMatthew G. Knepley 
168230b9ff8bSMatthew G. Knepley $ f0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1683194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1684194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
168530b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], const PetscReal n[], PetscScalar f0[])
1686194d53e6SMatthew G. Knepley 
1687194d53e6SMatthew G. Knepley + dim - the spatial dimension
1688194d53e6SMatthew G. Knepley . Nf - the number of fields
1689194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1690194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1691194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1692194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1693194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1694194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1695194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1696194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1697194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1698194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1699194d53e6SMatthew G. Knepley . t - current time
1700194d53e6SMatthew G. Knepley . x - coordinates of the current point
1701194d53e6SMatthew G. Knepley . n - unit normal at the current point
170297b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
170397b6e6e8SMatthew G. Knepley . constants - constant parameters
1704194d53e6SMatthew G. Knepley - f0 - output values at the current point
1705194d53e6SMatthew G. Knepley 
1706194d53e6SMatthew G. Knepley   Level: intermediate
1707194d53e6SMatthew G. Knepley 
1708194d53e6SMatthew G. Knepley .seealso: PetscDSSetBdResidual()
1709194d53e6SMatthew G. Knepley @*/
17102764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetBdResidual(PetscDS prob, PetscInt f,
171130b9ff8bSMatthew G. Knepley                                     void (**f0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1712194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1713194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
171497b6e6e8SMatthew G. Knepley                                                 PetscReal t, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]),
171530b9ff8bSMatthew G. Knepley                                     void (**f1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1716194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1717194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
171897b6e6e8SMatthew G. Knepley                                                 PetscReal t, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]))
17192764a2aaSMatthew G. Knepley {
17202764a2aaSMatthew G. Knepley   PetscFunctionBegin;
17212764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
17222764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
17232764a2aaSMatthew G. Knepley   if (f0) {PetscValidPointer(f0, 3); *f0 = prob->fBd[f*2+0];}
17242764a2aaSMatthew G. Knepley   if (f1) {PetscValidPointer(f1, 4); *f1 = prob->fBd[f*2+1];}
17252764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
17262764a2aaSMatthew G. Knepley }
17272764a2aaSMatthew G. Knepley 
1728194d53e6SMatthew G. Knepley /*@C
1729194d53e6SMatthew G. Knepley   PetscDSSetBdResidual - Get the pointwise boundary residual function for a given test field
1730194d53e6SMatthew G. Knepley 
1731194d53e6SMatthew G. Knepley   Not collective
1732194d53e6SMatthew G. Knepley 
1733194d53e6SMatthew G. Knepley   Input Parameters:
1734194d53e6SMatthew G. Knepley + prob - The PetscDS
1735194d53e6SMatthew G. Knepley . f    - The test field number
1736194d53e6SMatthew G. Knepley . f0 - boundary integrand for the test function term
1737194d53e6SMatthew G. Knepley - f1 - boundary integrand for the test function gradient term
1738194d53e6SMatthew G. Knepley 
1739194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1740194d53e6SMatthew G. Knepley 
1741194d53e6SMatthew G. Knepley   \int_\Gamma \phi {\vec f}_0(u, u_t, \nabla u, x, t) \cdot \hat n + \nabla\phi \cdot {\overleftrightarrow f}_1(u, u_t, \nabla u, x, t) \cdot \hat n
1742194d53e6SMatthew G. Knepley 
1743194d53e6SMatthew G. Knepley The calling sequence for the callbacks f0 and f1 is given by:
1744194d53e6SMatthew G. Knepley 
174530b9ff8bSMatthew G. Knepley $ f0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1746194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1747194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
174830b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], const PetscReal n[], PetscScalar f0[])
1749194d53e6SMatthew G. Knepley 
1750194d53e6SMatthew G. Knepley + dim - the spatial dimension
1751194d53e6SMatthew G. Knepley . Nf - the number of fields
1752194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1753194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1754194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1755194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1756194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1757194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1758194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1759194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1760194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1761194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1762194d53e6SMatthew G. Knepley . t - current time
1763194d53e6SMatthew G. Knepley . x - coordinates of the current point
1764194d53e6SMatthew G. Knepley . n - unit normal at the current point
176597b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
176697b6e6e8SMatthew G. Knepley . constants - constant parameters
1767194d53e6SMatthew G. Knepley - f0 - output values at the current point
1768194d53e6SMatthew G. Knepley 
1769194d53e6SMatthew G. Knepley   Level: intermediate
1770194d53e6SMatthew G. Knepley 
1771194d53e6SMatthew G. Knepley .seealso: PetscDSGetBdResidual()
1772194d53e6SMatthew G. Knepley @*/
17732764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetBdResidual(PetscDS prob, PetscInt f,
177430b9ff8bSMatthew G. Knepley                                     void (*f0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1775194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1776194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
177797b6e6e8SMatthew G. Knepley                                                PetscReal t, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]),
177830b9ff8bSMatthew G. Knepley                                     void (*f1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1779194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1780194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
178197b6e6e8SMatthew G. Knepley                                                PetscReal t, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]))
17822764a2aaSMatthew G. Knepley {
17832764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
17842764a2aaSMatthew G. Knepley 
17852764a2aaSMatthew G. Knepley   PetscFunctionBegin;
17862764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
17872764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
17882764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, f+1);CHKERRQ(ierr);
17892764a2aaSMatthew G. Knepley   if (f0) {PetscValidFunction(f0, 3); prob->fBd[f*2+0] = f0;}
17902764a2aaSMatthew G. Knepley   if (f1) {PetscValidFunction(f1, 4); prob->fBd[f*2+1] = f1;}
17912764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
17922764a2aaSMatthew G. Knepley }
17932764a2aaSMatthew G. Knepley 
1794194d53e6SMatthew G. Knepley /*@C
1795194d53e6SMatthew G. Knepley   PetscDSGetBdJacobian - Get the pointwise boundary Jacobian function for given test and basis field
1796194d53e6SMatthew G. Knepley 
1797194d53e6SMatthew G. Knepley   Not collective
1798194d53e6SMatthew G. Knepley 
1799194d53e6SMatthew G. Knepley   Input Parameters:
1800194d53e6SMatthew G. Knepley + prob - The PetscDS
1801194d53e6SMatthew G. Knepley . f    - The test field number
1802194d53e6SMatthew G. Knepley - g    - The field number
1803194d53e6SMatthew G. Knepley 
1804194d53e6SMatthew G. Knepley   Output Parameters:
1805194d53e6SMatthew G. Knepley + g0 - integrand for the test and basis function term
1806194d53e6SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1807194d53e6SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1808194d53e6SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1809194d53e6SMatthew G. Knepley 
1810194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1811194d53e6SMatthew G. Knepley 
1812194d53e6SMatthew G. Knepley   \int_\Gamma \phi {\vec g}_0(u, u_t, \nabla u, x, t) \cdot \hat n \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \cdot \hat n \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \cdot \hat n \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \hat n \cdot \nabla \psi
1813194d53e6SMatthew G. Knepley 
1814194d53e6SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1815194d53e6SMatthew G. Knepley 
181630b9ff8bSMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1817194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1818194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
181930b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], const PetscReal n[], PetscScalar g0[])
1820194d53e6SMatthew G. Knepley 
1821194d53e6SMatthew G. Knepley + dim - the spatial dimension
1822194d53e6SMatthew G. Knepley . Nf - the number of fields
1823194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1824194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1825194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1826194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1827194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1828194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1829194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1830194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1831194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1832194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1833194d53e6SMatthew G. Knepley . t - current time
18342aa1fc23SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1835194d53e6SMatthew G. Knepley . x - coordinates of the current point
1836194d53e6SMatthew G. Knepley . n - normal at the current point
183797b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
183897b6e6e8SMatthew G. Knepley . constants - constant parameters
1839194d53e6SMatthew G. Knepley - g0 - output values at the current point
1840194d53e6SMatthew G. Knepley 
1841194d53e6SMatthew G. Knepley   Level: intermediate
1842194d53e6SMatthew G. Knepley 
1843194d53e6SMatthew G. Knepley .seealso: PetscDSSetBdJacobian()
1844194d53e6SMatthew G. Knepley @*/
18452764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetBdJacobian(PetscDS prob, PetscInt f, PetscInt g,
184630b9ff8bSMatthew G. Knepley                                     void (**g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1847194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1848194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
184997b6e6e8SMatthew G. Knepley                                                 PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
185030b9ff8bSMatthew G. Knepley                                     void (**g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1851194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1852194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
185397b6e6e8SMatthew G. Knepley                                                 PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
185430b9ff8bSMatthew G. Knepley                                     void (**g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1855194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1856194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
185797b6e6e8SMatthew G. Knepley                                                 PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
185830b9ff8bSMatthew G. Knepley                                     void (**g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1859194d53e6SMatthew G. Knepley                                                 const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1860194d53e6SMatthew G. Knepley                                                 const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
186197b6e6e8SMatthew G. Knepley                                                 PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
18622764a2aaSMatthew G. Knepley {
18632764a2aaSMatthew G. Knepley   PetscFunctionBegin;
18642764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
18652764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
18662764a2aaSMatthew G. Knepley   if ((g < 0) || (g >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", g, prob->Nf);
18672764a2aaSMatthew G. Knepley   if (g0) {PetscValidPointer(g0, 4); *g0 = prob->gBd[(f*prob->Nf + g)*4+0];}
18682764a2aaSMatthew G. Knepley   if (g1) {PetscValidPointer(g1, 5); *g1 = prob->gBd[(f*prob->Nf + g)*4+1];}
18692764a2aaSMatthew G. Knepley   if (g2) {PetscValidPointer(g2, 6); *g2 = prob->gBd[(f*prob->Nf + g)*4+2];}
18702764a2aaSMatthew G. Knepley   if (g3) {PetscValidPointer(g3, 7); *g3 = prob->gBd[(f*prob->Nf + g)*4+3];}
18712764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
18722764a2aaSMatthew G. Knepley }
18732764a2aaSMatthew G. Knepley 
1874194d53e6SMatthew G. Knepley /*@C
1875194d53e6SMatthew G. Knepley   PetscDSSetBdJacobian - Set the pointwise boundary Jacobian function for given test and basis field
1876194d53e6SMatthew G. Knepley 
1877194d53e6SMatthew G. Knepley   Not collective
1878194d53e6SMatthew G. Knepley 
1879194d53e6SMatthew G. Knepley   Input Parameters:
1880194d53e6SMatthew G. Knepley + prob - The PetscDS
1881194d53e6SMatthew G. Knepley . f    - The test field number
1882194d53e6SMatthew G. Knepley . g    - The field number
1883194d53e6SMatthew G. Knepley . g0 - integrand for the test and basis function term
1884194d53e6SMatthew G. Knepley . g1 - integrand for the test function and basis function gradient term
1885194d53e6SMatthew G. Knepley . g2 - integrand for the test function gradient and basis function term
1886194d53e6SMatthew G. Knepley - g3 - integrand for the test function gradient and basis function gradient term
1887194d53e6SMatthew G. Knepley 
1888194d53e6SMatthew G. Knepley   Note: We are using a first order FEM model for the weak form:
1889194d53e6SMatthew G. Knepley 
1890194d53e6SMatthew G. Knepley   \int_\Gamma \phi {\vec g}_0(u, u_t, \nabla u, x, t) \cdot \hat n \psi + \phi {\vec g}_1(u, u_t, \nabla u, x, t) \cdot \hat n \nabla \psi + \nabla\phi \cdot {\vec g}_2(u, u_t, \nabla u, x, t) \cdot \hat n \psi + \nabla\phi \cdot {\overleftrightarrow g}_3(u, u_t, \nabla u, x, t) \cdot \hat n \cdot \nabla \psi
1891194d53e6SMatthew G. Knepley 
1892194d53e6SMatthew G. Knepley The calling sequence for the callbacks g0, g1, g2 and g3 is given by:
1893194d53e6SMatthew G. Knepley 
189430b9ff8bSMatthew G. Knepley $ g0(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1895194d53e6SMatthew G. Knepley $    const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1896194d53e6SMatthew G. Knepley $    const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
189730b9ff8bSMatthew G. Knepley $    PetscReal t, const PetscReal x[], const PetscReal n[], PetscScalar g0[])
1898194d53e6SMatthew G. Knepley 
1899194d53e6SMatthew G. Knepley + dim - the spatial dimension
1900194d53e6SMatthew G. Knepley . Nf - the number of fields
1901194d53e6SMatthew G. Knepley . uOff - the offset into u[] and u_t[] for each field
1902194d53e6SMatthew G. Knepley . uOff_x - the offset into u_x[] for each field
1903194d53e6SMatthew G. Knepley . u - each field evaluated at the current point
1904194d53e6SMatthew G. Knepley . u_t - the time derivative of each field evaluated at the current point
1905194d53e6SMatthew G. Knepley . u_x - the gradient of each field evaluated at the current point
1906194d53e6SMatthew G. Knepley . aOff - the offset into a[] and a_t[] for each auxiliary field
1907194d53e6SMatthew G. Knepley . aOff_x - the offset into a_x[] for each auxiliary field
1908194d53e6SMatthew G. Knepley . a - each auxiliary field evaluated at the current point
1909194d53e6SMatthew G. Knepley . a_t - the time derivative of each auxiliary field evaluated at the current point
1910194d53e6SMatthew G. Knepley . a_x - the gradient of auxiliary each field evaluated at the current point
1911194d53e6SMatthew G. Knepley . t - current time
19122aa1fc23SMatthew G. Knepley . u_tShift - the multiplier a for dF/dU_t
1913194d53e6SMatthew G. Knepley . x - coordinates of the current point
1914194d53e6SMatthew G. Knepley . n - normal at the current point
191597b6e6e8SMatthew G. Knepley . numConstants - number of constant parameters
191697b6e6e8SMatthew G. Knepley . constants - constant parameters
1917194d53e6SMatthew G. Knepley - g0 - output values at the current point
1918194d53e6SMatthew G. Knepley 
1919194d53e6SMatthew G. Knepley   Level: intermediate
1920194d53e6SMatthew G. Knepley 
1921194d53e6SMatthew G. Knepley .seealso: PetscDSGetBdJacobian()
1922194d53e6SMatthew G. Knepley @*/
19232764a2aaSMatthew G. Knepley PetscErrorCode PetscDSSetBdJacobian(PetscDS prob, PetscInt f, PetscInt g,
192430b9ff8bSMatthew G. Knepley                                     void (*g0)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1925194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1926194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
192797b6e6e8SMatthew G. Knepley                                                PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]),
192830b9ff8bSMatthew G. Knepley                                     void (*g1)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1929194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1930194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
193197b6e6e8SMatthew G. Knepley                                                PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]),
193230b9ff8bSMatthew G. Knepley                                     void (*g2)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1933194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1934194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
193597b6e6e8SMatthew G. Knepley                                                PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]),
193630b9ff8bSMatthew G. Knepley                                     void (*g3)(PetscInt dim, PetscInt Nf, PetscInt NfAux,
1937194d53e6SMatthew G. Knepley                                                const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[],
1938194d53e6SMatthew G. Knepley                                                const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[],
193997b6e6e8SMatthew G. Knepley                                                PetscReal t, PetscReal u_tShift, const PetscReal x[], const PetscReal n[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]))
19402764a2aaSMatthew G. Knepley {
19412764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
19422764a2aaSMatthew G. Knepley 
19432764a2aaSMatthew G. Knepley   PetscFunctionBegin;
19442764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
19452764a2aaSMatthew G. Knepley   if (g0) PetscValidFunction(g0, 4);
19462764a2aaSMatthew G. Knepley   if (g1) PetscValidFunction(g1, 5);
19472764a2aaSMatthew G. Knepley   if (g2) PetscValidFunction(g2, 6);
19482764a2aaSMatthew G. Knepley   if (g3) PetscValidFunction(g3, 7);
19492764a2aaSMatthew G. Knepley   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
19502764a2aaSMatthew G. Knepley   if (g < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", g);
19512764a2aaSMatthew G. Knepley   ierr = PetscDSEnlarge_Static(prob, PetscMax(f, g)+1);CHKERRQ(ierr);
19522764a2aaSMatthew G. Knepley   prob->gBd[(f*prob->Nf + g)*4+0] = g0;
19532764a2aaSMatthew G. Knepley   prob->gBd[(f*prob->Nf + g)*4+1] = g1;
19542764a2aaSMatthew G. Knepley   prob->gBd[(f*prob->Nf + g)*4+2] = g2;
19552764a2aaSMatthew G. Knepley   prob->gBd[(f*prob->Nf + g)*4+3] = g3;
19562764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
19572764a2aaSMatthew G. Knepley }
19582764a2aaSMatthew G. Knepley 
1959*0d3e9b51SMatthew G. Knepley /*@C
196097b6e6e8SMatthew G. Knepley   PetscDSGetConstants - Returns the array of constants passed to point functions
196197b6e6e8SMatthew G. Knepley 
196297b6e6e8SMatthew G. Knepley   Not collective
196397b6e6e8SMatthew G. Knepley 
196497b6e6e8SMatthew G. Knepley   Input Parameter:
196597b6e6e8SMatthew G. Knepley . prob - The PetscDS object
196697b6e6e8SMatthew G. Knepley 
196797b6e6e8SMatthew G. Knepley   Output Parameters:
196897b6e6e8SMatthew G. Knepley + numConstants - The number of constants
196997b6e6e8SMatthew G. Knepley - constants    - The array of constants, NULL if there are none
197097b6e6e8SMatthew G. Knepley 
197197b6e6e8SMatthew G. Knepley   Level: intermediate
197297b6e6e8SMatthew G. Knepley 
197397b6e6e8SMatthew G. Knepley .seealso: PetscDSSetConstants(), PetscDSCreate()
197497b6e6e8SMatthew G. Knepley @*/
197597b6e6e8SMatthew G. Knepley PetscErrorCode PetscDSGetConstants(PetscDS prob, PetscInt *numConstants, const PetscScalar *constants[])
197697b6e6e8SMatthew G. Knepley {
197797b6e6e8SMatthew G. Knepley   PetscFunctionBegin;
197897b6e6e8SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
197997b6e6e8SMatthew G. Knepley   if (numConstants) {PetscValidPointer(numConstants, 2); *numConstants = prob->numConstants;}
198097b6e6e8SMatthew G. Knepley   if (constants)    {PetscValidPointer(constants, 3);    *constants    = prob->constants;}
198197b6e6e8SMatthew G. Knepley   PetscFunctionReturn(0);
198297b6e6e8SMatthew G. Knepley }
198397b6e6e8SMatthew G. Knepley 
1984*0d3e9b51SMatthew G. Knepley /*@C
198597b6e6e8SMatthew G. Knepley   PetscDSSetConstants - Set the array of constants passed to point functions
198697b6e6e8SMatthew G. Knepley 
198797b6e6e8SMatthew G. Knepley   Not collective
198897b6e6e8SMatthew G. Knepley 
198997b6e6e8SMatthew G. Knepley   Input Parameters:
199097b6e6e8SMatthew G. Knepley + prob         - The PetscDS object
199197b6e6e8SMatthew G. Knepley . numConstants - The number of constants
199297b6e6e8SMatthew G. Knepley - constants    - The array of constants, NULL if there are none
199397b6e6e8SMatthew G. Knepley 
199497b6e6e8SMatthew G. Knepley   Level: intermediate
199597b6e6e8SMatthew G. Knepley 
199697b6e6e8SMatthew G. Knepley .seealso: PetscDSGetConstants(), PetscDSCreate()
199797b6e6e8SMatthew G. Knepley @*/
199897b6e6e8SMatthew G. Knepley PetscErrorCode PetscDSSetConstants(PetscDS prob, PetscInt numConstants, PetscScalar constants[])
199997b6e6e8SMatthew G. Knepley {
200097b6e6e8SMatthew G. Knepley   PetscErrorCode ierr;
200197b6e6e8SMatthew G. Knepley 
200297b6e6e8SMatthew G. Knepley   PetscFunctionBegin;
200397b6e6e8SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
200497b6e6e8SMatthew G. Knepley   if (numConstants != prob->numConstants) {
200597b6e6e8SMatthew G. Knepley     ierr = PetscFree(prob->constants);CHKERRQ(ierr);
200697b6e6e8SMatthew G. Knepley     prob->constants = NULL;
200797b6e6e8SMatthew G. Knepley   }
200897b6e6e8SMatthew G. Knepley   prob->numConstants = numConstants;
200997b6e6e8SMatthew G. Knepley   if (prob->numConstants) {
201097b6e6e8SMatthew G. Knepley     PetscValidPointer(constants, 3);
201197b6e6e8SMatthew G. Knepley     ierr = PetscMalloc1(prob->numConstants, &prob->constants);CHKERRQ(ierr);
201297b6e6e8SMatthew G. Knepley     ierr = PetscMemcpy(prob->constants, constants, prob->numConstants * sizeof(PetscScalar));CHKERRQ(ierr);
201397b6e6e8SMatthew G. Knepley   }
201497b6e6e8SMatthew G. Knepley   PetscFunctionReturn(0);
201597b6e6e8SMatthew G. Knepley }
201697b6e6e8SMatthew G. Knepley 
20174cd1e086SMatthew G. Knepley /*@
20184cd1e086SMatthew G. Knepley   PetscDSGetFieldIndex - Returns the index of the given field
20194cd1e086SMatthew G. Knepley 
20204cd1e086SMatthew G. Knepley   Not collective
20214cd1e086SMatthew G. Knepley 
20224cd1e086SMatthew G. Knepley   Input Parameters:
20234cd1e086SMatthew G. Knepley + prob - The PetscDS object
20244cd1e086SMatthew G. Knepley - disc - The discretization object
20254cd1e086SMatthew G. Knepley 
20264cd1e086SMatthew G. Knepley   Output Parameter:
20274cd1e086SMatthew G. Knepley . f - The field number
20284cd1e086SMatthew G. Knepley 
20294cd1e086SMatthew G. Knepley   Level: beginner
20304cd1e086SMatthew G. Knepley 
2031f744cafaSSander Arens .seealso: PetscGetDiscretization(), PetscDSGetNumFields(), PetscDSCreate()
20324cd1e086SMatthew G. Knepley @*/
20334cd1e086SMatthew G. Knepley PetscErrorCode PetscDSGetFieldIndex(PetscDS prob, PetscObject disc, PetscInt *f)
20344cd1e086SMatthew G. Knepley {
20354cd1e086SMatthew G. Knepley   PetscInt g;
20364cd1e086SMatthew G. Knepley 
20374cd1e086SMatthew G. Knepley   PetscFunctionBegin;
20384cd1e086SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
20394cd1e086SMatthew G. Knepley   PetscValidPointer(f, 3);
20404cd1e086SMatthew G. Knepley   *f = -1;
20414cd1e086SMatthew G. Knepley   for (g = 0; g < prob->Nf; ++g) {if (disc == prob->disc[g]) break;}
20424cd1e086SMatthew G. Knepley   if (g == prob->Nf) SETERRQ(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_WRONG, "Field not found in PetscDS.");
20434cd1e086SMatthew G. Knepley   *f = g;
20444cd1e086SMatthew G. Knepley   PetscFunctionReturn(0);
20454cd1e086SMatthew G. Knepley }
20464cd1e086SMatthew G. Knepley 
20474cd1e086SMatthew G. Knepley /*@
20484cd1e086SMatthew G. Knepley   PetscDSGetFieldSize - Returns the size of the given field in the full space basis
20494cd1e086SMatthew G. Knepley 
20504cd1e086SMatthew G. Knepley   Not collective
20514cd1e086SMatthew G. Knepley 
20524cd1e086SMatthew G. Knepley   Input Parameters:
20534cd1e086SMatthew G. Knepley + prob - The PetscDS object
20544cd1e086SMatthew G. Knepley - f - The field number
20554cd1e086SMatthew G. Knepley 
20564cd1e086SMatthew G. Knepley   Output Parameter:
20574cd1e086SMatthew G. Knepley . size - The size
20584cd1e086SMatthew G. Knepley 
20594cd1e086SMatthew G. Knepley   Level: beginner
20604cd1e086SMatthew G. Knepley 
2061f744cafaSSander Arens .seealso: PetscDSGetFieldOffset(), PetscDSGetNumFields(), PetscDSCreate()
20624cd1e086SMatthew G. Knepley @*/
20634cd1e086SMatthew G. Knepley PetscErrorCode PetscDSGetFieldSize(PetscDS prob, PetscInt f, PetscInt *size)
20644cd1e086SMatthew G. Knepley {
20652166fd64SMatthew G. Knepley   PetscErrorCode ierr;
20662166fd64SMatthew G. Knepley 
20674cd1e086SMatthew G. Knepley   PetscFunctionBegin;
20684cd1e086SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
20694cd1e086SMatthew G. Knepley   PetscValidPointer(size, 3);
20704cd1e086SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
20712166fd64SMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
2072d4742ddaSMatthew G. Knepley   *size = prob->Nb[f];
20734cd1e086SMatthew G. Knepley   PetscFunctionReturn(0);
20744cd1e086SMatthew G. Knepley }
20754cd1e086SMatthew G. Knepley 
2076bc4ae4beSMatthew G. Knepley /*@
2077bc4ae4beSMatthew G. Knepley   PetscDSGetFieldOffset - Returns the offset of the given field in the full space basis
2078bc4ae4beSMatthew G. Knepley 
2079bc4ae4beSMatthew G. Knepley   Not collective
2080bc4ae4beSMatthew G. Knepley 
2081bc4ae4beSMatthew G. Knepley   Input Parameters:
2082bc4ae4beSMatthew G. Knepley + prob - The PetscDS object
2083bc4ae4beSMatthew G. Knepley - f - The field number
2084bc4ae4beSMatthew G. Knepley 
2085bc4ae4beSMatthew G. Knepley   Output Parameter:
2086bc4ae4beSMatthew G. Knepley . off - The offset
2087bc4ae4beSMatthew G. Knepley 
2088bc4ae4beSMatthew G. Knepley   Level: beginner
2089bc4ae4beSMatthew G. Knepley 
2090f744cafaSSander Arens .seealso: PetscDSGetFieldSize(), PetscDSGetNumFields(), PetscDSCreate()
2091bc4ae4beSMatthew G. Knepley @*/
20922764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetFieldOffset(PetscDS prob, PetscInt f, PetscInt *off)
20932764a2aaSMatthew G. Knepley {
20944cd1e086SMatthew G. Knepley   PetscInt       size, g;
20952764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
20962764a2aaSMatthew G. Knepley 
20972764a2aaSMatthew G. Knepley   PetscFunctionBegin;
20982764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
20992764a2aaSMatthew G. Knepley   PetscValidPointer(off, 3);
21002764a2aaSMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
21012764a2aaSMatthew G. Knepley   *off = 0;
21022764a2aaSMatthew G. Knepley   for (g = 0; g < f; ++g) {
21034cd1e086SMatthew G. Knepley     ierr = PetscDSGetFieldSize(prob, g, &size);CHKERRQ(ierr);
21044cd1e086SMatthew G. Knepley     *off += size;
21052764a2aaSMatthew G. Knepley   }
21062764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
21072764a2aaSMatthew G. Knepley }
21082764a2aaSMatthew G. Knepley 
2109bc4ae4beSMatthew G. Knepley /*@
211047e57110SSander Arens   PetscDSGetDimensions - Returns the size of the approximation space for each field on an evaluation point
2111bc4ae4beSMatthew G. Knepley 
2112bc4ae4beSMatthew G. Knepley   Not collective
2113bc4ae4beSMatthew G. Knepley 
211447e57110SSander Arens   Input Parameter:
211547e57110SSander Arens . prob - The PetscDS object
2116bc4ae4beSMatthew G. Knepley 
2117bc4ae4beSMatthew G. Knepley   Output Parameter:
211847e57110SSander Arens . dimensions - The number of dimensions
2119bc4ae4beSMatthew G. Knepley 
2120bc4ae4beSMatthew G. Knepley   Level: beginner
2121bc4ae4beSMatthew G. Knepley 
212247e57110SSander Arens .seealso: PetscDSGetComponentOffsets(), PetscDSGetNumFields(), PetscDSCreate()
2123bc4ae4beSMatthew G. Knepley @*/
212447e57110SSander Arens PetscErrorCode PetscDSGetDimensions(PetscDS prob, PetscInt *dimensions[])
21252764a2aaSMatthew G. Knepley {
21262764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
21272764a2aaSMatthew G. Knepley 
21282764a2aaSMatthew G. Knepley   PetscFunctionBegin;
21292764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
213047e57110SSander Arens   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
213147e57110SSander Arens   PetscValidPointer(dimensions, 2);
213247e57110SSander Arens   *dimensions = prob->Nb;
213347e57110SSander Arens   PetscFunctionReturn(0);
21346ce16762SMatthew G. Knepley }
213547e57110SSander Arens 
213647e57110SSander Arens /*@
213747e57110SSander Arens   PetscDSGetComponents - Returns the number of components for each field on an evaluation point
213847e57110SSander Arens 
213947e57110SSander Arens   Not collective
214047e57110SSander Arens 
214147e57110SSander Arens   Input Parameter:
214247e57110SSander Arens . prob - The PetscDS object
214347e57110SSander Arens 
214447e57110SSander Arens   Output Parameter:
214547e57110SSander Arens . components - The number of components
214647e57110SSander Arens 
214747e57110SSander Arens   Level: beginner
214847e57110SSander Arens 
214947e57110SSander Arens .seealso: PetscDSGetComponentOffsets(), PetscDSGetNumFields(), PetscDSCreate()
215047e57110SSander Arens @*/
215147e57110SSander Arens PetscErrorCode PetscDSGetComponents(PetscDS prob, PetscInt *components[])
215247e57110SSander Arens {
215347e57110SSander Arens   PetscErrorCode ierr;
215447e57110SSander Arens 
215547e57110SSander Arens   PetscFunctionBegin;
215647e57110SSander Arens   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
215747e57110SSander Arens   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
215847e57110SSander Arens   PetscValidPointer(components, 2);
215947e57110SSander Arens   *components = prob->Nc;
21606ce16762SMatthew G. Knepley   PetscFunctionReturn(0);
21616ce16762SMatthew G. Knepley }
21626ce16762SMatthew G. Knepley 
21636ce16762SMatthew G. Knepley /*@
21646ce16762SMatthew G. Knepley   PetscDSGetComponentOffset - Returns the offset of the given field on an evaluation point
21656ce16762SMatthew G. Knepley 
21666ce16762SMatthew G. Knepley   Not collective
21676ce16762SMatthew G. Knepley 
21686ce16762SMatthew G. Knepley   Input Parameters:
21696ce16762SMatthew G. Knepley + prob - The PetscDS object
21706ce16762SMatthew G. Knepley - f - The field number
21716ce16762SMatthew G. Knepley 
21726ce16762SMatthew G. Knepley   Output Parameter:
21736ce16762SMatthew G. Knepley . off - The offset
21746ce16762SMatthew G. Knepley 
21756ce16762SMatthew G. Knepley   Level: beginner
21766ce16762SMatthew G. Knepley 
2177f744cafaSSander Arens .seealso: PetscDSGetNumFields(), PetscDSCreate()
21786ce16762SMatthew G. Knepley @*/
21796ce16762SMatthew G. Knepley PetscErrorCode PetscDSGetComponentOffset(PetscDS prob, PetscInt f, PetscInt *off)
21806ce16762SMatthew G. Knepley {
21816ce16762SMatthew G. Knepley   PetscFunctionBegin;
21826ce16762SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
21836ce16762SMatthew G. Knepley   PetscValidPointer(off, 3);
21846ce16762SMatthew G. Knepley   if ((f < 0) || (f >= prob->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, prob->Nf);
218547e57110SSander Arens   *off = prob->off[f];
21862764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
21872764a2aaSMatthew G. Knepley }
21882764a2aaSMatthew G. Knepley 
2189194d53e6SMatthew G. Knepley /*@
2190194d53e6SMatthew G. Knepley   PetscDSGetComponentOffsets - Returns the offset of each field on an evaluation point
2191194d53e6SMatthew G. Knepley 
2192194d53e6SMatthew G. Knepley   Not collective
2193194d53e6SMatthew G. Knepley 
2194194d53e6SMatthew G. Knepley   Input Parameter:
2195194d53e6SMatthew G. Knepley . prob - The PetscDS object
2196194d53e6SMatthew G. Knepley 
2197194d53e6SMatthew G. Knepley   Output Parameter:
2198194d53e6SMatthew G. Knepley . offsets - The offsets
2199194d53e6SMatthew G. Knepley 
2200194d53e6SMatthew G. Knepley   Level: beginner
2201194d53e6SMatthew G. Knepley 
2202f744cafaSSander Arens .seealso: PetscDSGetNumFields(), PetscDSCreate()
2203194d53e6SMatthew G. Knepley @*/
2204194d53e6SMatthew G. Knepley PetscErrorCode PetscDSGetComponentOffsets(PetscDS prob, PetscInt *offsets[])
2205194d53e6SMatthew G. Knepley {
2206194d53e6SMatthew G. Knepley   PetscFunctionBegin;
2207194d53e6SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
2208194d53e6SMatthew G. Knepley   PetscValidPointer(offsets, 2);
2209194d53e6SMatthew G. Knepley   *offsets = prob->off;
2210194d53e6SMatthew G. Knepley   PetscFunctionReturn(0);
2211194d53e6SMatthew G. Knepley }
2212194d53e6SMatthew G. Knepley 
2213194d53e6SMatthew G. Knepley /*@
2214194d53e6SMatthew G. Knepley   PetscDSGetComponentDerivativeOffsets - Returns the offset of each field derivative on an evaluation point
2215194d53e6SMatthew G. Knepley 
2216194d53e6SMatthew G. Knepley   Not collective
2217194d53e6SMatthew G. Knepley 
2218194d53e6SMatthew G. Knepley   Input Parameter:
2219194d53e6SMatthew G. Knepley . prob - The PetscDS object
2220194d53e6SMatthew G. Knepley 
2221194d53e6SMatthew G. Knepley   Output Parameter:
2222194d53e6SMatthew G. Knepley . offsets - The offsets
2223194d53e6SMatthew G. Knepley 
2224194d53e6SMatthew G. Knepley   Level: beginner
2225194d53e6SMatthew G. Knepley 
2226f744cafaSSander Arens .seealso: PetscDSGetNumFields(), PetscDSCreate()
2227194d53e6SMatthew G. Knepley @*/
2228194d53e6SMatthew G. Knepley PetscErrorCode PetscDSGetComponentDerivativeOffsets(PetscDS prob, PetscInt *offsets[])
2229194d53e6SMatthew G. Knepley {
2230194d53e6SMatthew G. Knepley   PetscFunctionBegin;
2231194d53e6SMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
2232194d53e6SMatthew G. Knepley   PetscValidPointer(offsets, 2);
2233194d53e6SMatthew G. Knepley   *offsets = prob->offDer;
2234194d53e6SMatthew G. Knepley   PetscFunctionReturn(0);
2235194d53e6SMatthew G. Knepley }
2236194d53e6SMatthew G. Knepley 
223768c9edb9SMatthew G. Knepley /*@C
223868c9edb9SMatthew G. Knepley   PetscDSGetTabulation - Return the basis tabulation at quadrature points for the volume discretization
223968c9edb9SMatthew G. Knepley 
224068c9edb9SMatthew G. Knepley   Not collective
224168c9edb9SMatthew G. Knepley 
224268c9edb9SMatthew G. Knepley   Input Parameter:
224368c9edb9SMatthew G. Knepley . prob - The PetscDS object
224468c9edb9SMatthew G. Knepley 
224568c9edb9SMatthew G. Knepley   Output Parameters:
224668c9edb9SMatthew G. Knepley + basis - The basis function tabulation at quadrature points
224768c9edb9SMatthew G. Knepley - basisDer - The basis function derivative tabulation at quadrature points
224868c9edb9SMatthew G. Knepley 
224968c9edb9SMatthew G. Knepley   Level: intermediate
225068c9edb9SMatthew G. Knepley 
2251f744cafaSSander Arens .seealso: PetscDSCreate()
225268c9edb9SMatthew G. Knepley @*/
22532764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetTabulation(PetscDS prob, PetscReal ***basis, PetscReal ***basisDer)
22542764a2aaSMatthew G. Knepley {
22552764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
22562764a2aaSMatthew G. Knepley 
22572764a2aaSMatthew G. Knepley   PetscFunctionBegin;
22582764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
22592764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
22602764a2aaSMatthew G. Knepley   if (basis)    {PetscValidPointer(basis, 2);    *basis    = prob->basis;}
22612764a2aaSMatthew G. Knepley   if (basisDer) {PetscValidPointer(basisDer, 3); *basisDer = prob->basisDer;}
22622764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
22632764a2aaSMatthew G. Knepley }
22642764a2aaSMatthew G. Knepley 
226568c9edb9SMatthew G. Knepley /*@C
22664d0b9603SSander Arens   PetscDSGetFaceTabulation - Return the basis tabulation at quadrature points on the faces
226768c9edb9SMatthew G. Knepley 
226868c9edb9SMatthew G. Knepley   Not collective
226968c9edb9SMatthew G. Knepley 
227068c9edb9SMatthew G. Knepley   Input Parameter:
227168c9edb9SMatthew G. Knepley . prob - The PetscDS object
227268c9edb9SMatthew G. Knepley 
227368c9edb9SMatthew G. Knepley   Output Parameters:
22744d0b9603SSander Arens + basisFace - The basis function tabulation at quadrature points
22754d0b9603SSander Arens - basisDerFace - The basis function derivative tabulation at quadrature points
227668c9edb9SMatthew G. Knepley 
227768c9edb9SMatthew G. Knepley   Level: intermediate
227868c9edb9SMatthew G. Knepley 
227968c9edb9SMatthew G. Knepley .seealso: PetscDSGetTabulation(), PetscDSCreate()
228068c9edb9SMatthew G. Knepley @*/
22814d0b9603SSander Arens PetscErrorCode PetscDSGetFaceTabulation(PetscDS prob, PetscReal ***basis, PetscReal ***basisDer)
22822764a2aaSMatthew G. Knepley {
22832764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
22842764a2aaSMatthew G. Knepley 
22852764a2aaSMatthew G. Knepley   PetscFunctionBegin;
22862764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
22872764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
22884d0b9603SSander Arens   if (basis)    {PetscValidPointer(basis, 2);    *basis    = prob->basisFace;}
22894d0b9603SSander Arens   if (basisDer) {PetscValidPointer(basisDer, 3); *basisDer = prob->basisDerFace;}
22902764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
22912764a2aaSMatthew G. Knepley }
22922764a2aaSMatthew G. Knepley 
22932764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetEvaluationArrays(PetscDS prob, PetscScalar **u, PetscScalar **u_t, PetscScalar **u_x)
22942764a2aaSMatthew G. Knepley {
22952764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
22962764a2aaSMatthew G. Knepley 
22972764a2aaSMatthew G. Knepley   PetscFunctionBegin;
22982764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
22992764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
23002764a2aaSMatthew G. Knepley   if (u)   {PetscValidPointer(u, 2);   *u   = prob->u;}
23012764a2aaSMatthew G. Knepley   if (u_t) {PetscValidPointer(u_t, 3); *u_t = prob->u_t;}
23022764a2aaSMatthew G. Knepley   if (u_x) {PetscValidPointer(u_x, 4); *u_x = prob->u_x;}
23032764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
23042764a2aaSMatthew G. Knepley }
23052764a2aaSMatthew G. Knepley 
23062764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetWeakFormArrays(PetscDS prob, PetscScalar **f0, PetscScalar **f1, PetscScalar **g0, PetscScalar **g1, PetscScalar **g2, PetscScalar **g3)
23072764a2aaSMatthew G. Knepley {
23082764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
23092764a2aaSMatthew G. Knepley 
23102764a2aaSMatthew G. Knepley   PetscFunctionBegin;
23112764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
23122764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
23132764a2aaSMatthew G. Knepley   if (f0) {PetscValidPointer(f0, 2); *f0 = prob->f0;}
23142764a2aaSMatthew G. Knepley   if (f1) {PetscValidPointer(f1, 3); *f1 = prob->f1;}
23152764a2aaSMatthew G. Knepley   if (g0) {PetscValidPointer(g0, 4); *g0 = prob->g0;}
23162764a2aaSMatthew G. Knepley   if (g1) {PetscValidPointer(g1, 5); *g1 = prob->g1;}
23172764a2aaSMatthew G. Knepley   if (g2) {PetscValidPointer(g2, 6); *g2 = prob->g2;}
23182764a2aaSMatthew G. Knepley   if (g3) {PetscValidPointer(g3, 7); *g3 = prob->g3;}
23192764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
23202764a2aaSMatthew G. Knepley }
23212764a2aaSMatthew G. Knepley 
23222764a2aaSMatthew G. Knepley PetscErrorCode PetscDSGetRefCoordArrays(PetscDS prob, PetscReal **x, PetscScalar **refSpaceDer)
23232764a2aaSMatthew G. Knepley {
23242764a2aaSMatthew G. Knepley   PetscErrorCode ierr;
23252764a2aaSMatthew G. Knepley 
23262764a2aaSMatthew G. Knepley   PetscFunctionBegin;
23272764a2aaSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
23282764a2aaSMatthew G. Knepley   ierr = PetscDSSetUp(prob);CHKERRQ(ierr);
23292764a2aaSMatthew G. Knepley   if (x)           {PetscValidPointer(x, 2);           *x           = prob->x;}
23302764a2aaSMatthew G. Knepley   if (refSpaceDer) {PetscValidPointer(refSpaceDer, 3); *refSpaceDer = prob->refSpaceDer;}
23312764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
23322764a2aaSMatthew G. Knepley }
23332764a2aaSMatthew G. Knepley 
233458ebd649SToby Isaac /*@C
233558ebd649SToby Isaac   PetscDSAddBoundary - Add a boundary condition to the model
233658ebd649SToby Isaac 
233758ebd649SToby Isaac   Input Parameters:
233858ebd649SToby Isaac + ds          - The PetscDS object
23392d47a189SJulian Andrej . type        - The type of condition, e.g. DM_BC_ESSENTIAL/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
234058ebd649SToby Isaac . name        - The BC name
234158ebd649SToby Isaac . labelname   - The label defining constrained points
234258ebd649SToby Isaac . field       - The field to constrain
234358ebd649SToby Isaac . numcomps    - The number of constrained field components
234458ebd649SToby Isaac . comps       - An array of constrained component numbers
234558ebd649SToby Isaac . bcFunc      - A pointwise function giving boundary values
234658ebd649SToby Isaac . numids      - The number of DMLabel ids for constrained points
234758ebd649SToby Isaac . ids         - An array of ids for constrained points
234858ebd649SToby Isaac - ctx         - An optional user context for bcFunc
234958ebd649SToby Isaac 
235058ebd649SToby Isaac   Options Database Keys:
235158ebd649SToby Isaac + -bc_<boundary name> <num> - Overrides the boundary ids
235258ebd649SToby Isaac - -bc_<boundary name>_comp <num> - Overrides the boundary components
235358ebd649SToby Isaac 
235458ebd649SToby Isaac   Level: developer
235558ebd649SToby Isaac 
235658ebd649SToby Isaac .seealso: PetscDSGetBoundary()
235758ebd649SToby Isaac @*/
2358a30ec4eaSSatish Balay PetscErrorCode PetscDSAddBoundary(PetscDS ds, DMBoundaryConditionType type, const char name[], const char labelname[], PetscInt field, PetscInt numcomps, const PetscInt *comps, void (*bcFunc)(void), PetscInt numids, const PetscInt *ids, void *ctx)
235958ebd649SToby Isaac {
236058ebd649SToby Isaac   DSBoundary     b;
236158ebd649SToby Isaac   PetscErrorCode ierr;
236258ebd649SToby Isaac 
236358ebd649SToby Isaac   PetscFunctionBegin;
236458ebd649SToby Isaac   PetscValidHeaderSpecific(ds, PETSCDS_CLASSID, 1);
236558ebd649SToby Isaac   ierr = PetscNew(&b);CHKERRQ(ierr);
236658ebd649SToby Isaac   ierr = PetscStrallocpy(name, (char **) &b->name);CHKERRQ(ierr);
236758ebd649SToby Isaac   ierr = PetscStrallocpy(labelname, (char **) &b->labelname);CHKERRQ(ierr);
236858ebd649SToby Isaac   ierr = PetscMalloc1(numcomps, &b->comps);CHKERRQ(ierr);
236958ebd649SToby Isaac   if (numcomps) {ierr = PetscMemcpy(b->comps, comps, numcomps*sizeof(PetscInt));CHKERRQ(ierr);}
237058ebd649SToby Isaac   ierr = PetscMalloc1(numids, &b->ids);CHKERRQ(ierr);
237158ebd649SToby Isaac   if (numids) {ierr = PetscMemcpy(b->ids, ids, numids*sizeof(PetscInt));CHKERRQ(ierr);}
2372f971fd6bSMatthew G. Knepley   b->type            = type;
237358ebd649SToby Isaac   b->field           = field;
237458ebd649SToby Isaac   b->numcomps        = numcomps;
237558ebd649SToby Isaac   b->func            = bcFunc;
237658ebd649SToby Isaac   b->numids          = numids;
237758ebd649SToby Isaac   b->ctx             = ctx;
237858ebd649SToby Isaac   b->next            = ds->boundary;
237958ebd649SToby Isaac   ds->boundary       = b;
238058ebd649SToby Isaac   PetscFunctionReturn(0);
238158ebd649SToby Isaac }
238258ebd649SToby Isaac 
238358ebd649SToby Isaac /*@
238458ebd649SToby Isaac   PetscDSGetNumBoundary - Get the number of registered BC
238558ebd649SToby Isaac 
238658ebd649SToby Isaac   Input Parameters:
238758ebd649SToby Isaac . ds - The PetscDS object
238858ebd649SToby Isaac 
238958ebd649SToby Isaac   Output Parameters:
239058ebd649SToby Isaac . numBd - The number of BC
239158ebd649SToby Isaac 
239258ebd649SToby Isaac   Level: intermediate
239358ebd649SToby Isaac 
239458ebd649SToby Isaac .seealso: PetscDSAddBoundary(), PetscDSGetBoundary()
239558ebd649SToby Isaac @*/
239658ebd649SToby Isaac PetscErrorCode PetscDSGetNumBoundary(PetscDS ds, PetscInt *numBd)
239758ebd649SToby Isaac {
239858ebd649SToby Isaac   DSBoundary b = ds->boundary;
239958ebd649SToby Isaac 
240058ebd649SToby Isaac   PetscFunctionBegin;
240158ebd649SToby Isaac   PetscValidHeaderSpecific(ds, PETSCDS_CLASSID, 1);
240258ebd649SToby Isaac   PetscValidPointer(numBd, 2);
240358ebd649SToby Isaac   *numBd = 0;
240458ebd649SToby Isaac   while (b) {++(*numBd); b = b->next;}
240558ebd649SToby Isaac   PetscFunctionReturn(0);
240658ebd649SToby Isaac }
240758ebd649SToby Isaac 
240858ebd649SToby Isaac /*@C
240958ebd649SToby Isaac   PetscDSGetBoundary - Add a boundary condition to the model
241058ebd649SToby Isaac 
241158ebd649SToby Isaac   Input Parameters:
241258ebd649SToby Isaac + ds          - The PetscDS object
241358ebd649SToby Isaac - bd          - The BC number
241458ebd649SToby Isaac 
241558ebd649SToby Isaac   Output Parameters:
24162d47a189SJulian Andrej + type        - The type of condition, e.g. DM_BC_ESSENTIAL/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
241758ebd649SToby Isaac . name        - The BC name
241858ebd649SToby Isaac . labelname   - The label defining constrained points
241958ebd649SToby Isaac . field       - The field to constrain
242058ebd649SToby Isaac . numcomps    - The number of constrained field components
242158ebd649SToby Isaac . comps       - An array of constrained component numbers
242258ebd649SToby Isaac . bcFunc      - A pointwise function giving boundary values
242358ebd649SToby Isaac . numids      - The number of DMLabel ids for constrained points
242458ebd649SToby Isaac . ids         - An array of ids for constrained points
242558ebd649SToby Isaac - ctx         - An optional user context for bcFunc
242658ebd649SToby Isaac 
242758ebd649SToby Isaac   Options Database Keys:
242858ebd649SToby Isaac + -bc_<boundary name> <num> - Overrides the boundary ids
242958ebd649SToby Isaac - -bc_<boundary name>_comp <num> - Overrides the boundary components
243058ebd649SToby Isaac 
243158ebd649SToby Isaac   Level: developer
243258ebd649SToby Isaac 
243358ebd649SToby Isaac .seealso: PetscDSAddBoundary()
243458ebd649SToby Isaac @*/
2435a30ec4eaSSatish Balay PetscErrorCode PetscDSGetBoundary(PetscDS ds, PetscInt bd, DMBoundaryConditionType *type, const char **name, const char **labelname, PetscInt *field, PetscInt *numcomps, const PetscInt **comps, void (**func)(void), PetscInt *numids, const PetscInt **ids, void **ctx)
243658ebd649SToby Isaac {
243758ebd649SToby Isaac   DSBoundary b    = ds->boundary;
243858ebd649SToby Isaac   PetscInt   n    = 0;
243958ebd649SToby Isaac 
244058ebd649SToby Isaac   PetscFunctionBegin;
244158ebd649SToby Isaac   PetscValidHeaderSpecific(ds, PETSCDS_CLASSID, 1);
244258ebd649SToby Isaac   while (b) {
244358ebd649SToby Isaac     if (n == bd) break;
244458ebd649SToby Isaac     b = b->next;
244558ebd649SToby Isaac     ++n;
244658ebd649SToby Isaac   }
244758ebd649SToby Isaac   if (!b) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Boundary %d is not in [0, %d)", bd, n);
2448f971fd6bSMatthew G. Knepley   if (type) {
2449f971fd6bSMatthew G. Knepley     PetscValidPointer(type, 3);
2450f971fd6bSMatthew G. Knepley     *type = b->type;
245158ebd649SToby Isaac   }
245258ebd649SToby Isaac   if (name) {
245358ebd649SToby Isaac     PetscValidPointer(name, 4);
245458ebd649SToby Isaac     *name = b->name;
245558ebd649SToby Isaac   }
245658ebd649SToby Isaac   if (labelname) {
245758ebd649SToby Isaac     PetscValidPointer(labelname, 5);
245858ebd649SToby Isaac     *labelname = b->labelname;
245958ebd649SToby Isaac   }
246058ebd649SToby Isaac   if (field) {
246158ebd649SToby Isaac     PetscValidPointer(field, 6);
246258ebd649SToby Isaac     *field = b->field;
246358ebd649SToby Isaac   }
246458ebd649SToby Isaac   if (numcomps) {
246558ebd649SToby Isaac     PetscValidPointer(numcomps, 7);
246658ebd649SToby Isaac     *numcomps = b->numcomps;
246758ebd649SToby Isaac   }
246858ebd649SToby Isaac   if (comps) {
246958ebd649SToby Isaac     PetscValidPointer(comps, 8);
247058ebd649SToby Isaac     *comps = b->comps;
247158ebd649SToby Isaac   }
247258ebd649SToby Isaac   if (func) {
247358ebd649SToby Isaac     PetscValidPointer(func, 9);
247458ebd649SToby Isaac     *func = b->func;
247558ebd649SToby Isaac   }
247658ebd649SToby Isaac   if (numids) {
247758ebd649SToby Isaac     PetscValidPointer(numids, 10);
247858ebd649SToby Isaac     *numids = b->numids;
247958ebd649SToby Isaac   }
248058ebd649SToby Isaac   if (ids) {
248158ebd649SToby Isaac     PetscValidPointer(ids, 11);
248258ebd649SToby Isaac     *ids = b->ids;
248358ebd649SToby Isaac   }
248458ebd649SToby Isaac   if (ctx) {
248558ebd649SToby Isaac     PetscValidPointer(ctx, 12);
248658ebd649SToby Isaac     *ctx = b->ctx;
248758ebd649SToby Isaac   }
248858ebd649SToby Isaac   PetscFunctionReturn(0);
248958ebd649SToby Isaac }
249058ebd649SToby Isaac 
2491dff059c6SToby Isaac PetscErrorCode PetscDSCopyBoundary(PetscDS probA, PetscDS probB)
2492dff059c6SToby Isaac {
2493dff059c6SToby Isaac   DSBoundary     b, next, *lastnext;
2494dff059c6SToby Isaac   PetscErrorCode ierr;
2495dff059c6SToby Isaac 
2496dff059c6SToby Isaac   PetscFunctionBegin;
2497dff059c6SToby Isaac   PetscValidHeaderSpecific(probA, PETSCDS_CLASSID, 1);
2498dff059c6SToby Isaac   PetscValidHeaderSpecific(probB, PETSCDS_CLASSID, 2);
2499dff059c6SToby Isaac   if (probA == probB) PetscFunctionReturn(0);
2500dff059c6SToby Isaac   next = probB->boundary;
2501dff059c6SToby Isaac   while (next) {
2502dff059c6SToby Isaac     DSBoundary b = next;
2503dff059c6SToby Isaac 
2504dff059c6SToby Isaac     next = b->next;
2505dff059c6SToby Isaac     ierr = PetscFree(b->comps);CHKERRQ(ierr);
2506dff059c6SToby Isaac     ierr = PetscFree(b->ids);CHKERRQ(ierr);
2507dff059c6SToby Isaac     ierr = PetscFree(b->name);CHKERRQ(ierr);
2508dff059c6SToby Isaac     ierr = PetscFree(b->labelname);CHKERRQ(ierr);
2509dff059c6SToby Isaac     ierr = PetscFree(b);CHKERRQ(ierr);
2510dff059c6SToby Isaac   }
2511dff059c6SToby Isaac   lastnext = &(probB->boundary);
2512dff059c6SToby Isaac   for (b = probA->boundary; b; b = b->next) {
2513dff059c6SToby Isaac     DSBoundary bNew;
2514dff059c6SToby Isaac 
2515459726d8SSatish Balay     ierr = PetscNew(&bNew);CHKERRQ(ierr);
2516dff059c6SToby Isaac     bNew->numcomps = b->numcomps;
2517dff059c6SToby Isaac     ierr = PetscMalloc1(bNew->numcomps, &bNew->comps);CHKERRQ(ierr);
2518dff059c6SToby Isaac     ierr = PetscMemcpy(bNew->comps, b->comps, bNew->numcomps*sizeof(PetscInt));CHKERRQ(ierr);
2519dff059c6SToby Isaac     bNew->numids = b->numids;
2520dff059c6SToby Isaac     ierr = PetscMalloc1(bNew->numids, &bNew->ids);CHKERRQ(ierr);
2521dff059c6SToby Isaac     ierr = PetscMemcpy(bNew->ids, b->ids, bNew->numids*sizeof(PetscInt));CHKERRQ(ierr);
2522dff059c6SToby Isaac     ierr = PetscStrallocpy(b->labelname,(char **) &(bNew->labelname));CHKERRQ(ierr);
2523dff059c6SToby Isaac     ierr = PetscStrallocpy(b->name,(char **) &(bNew->name));CHKERRQ(ierr);
2524dff059c6SToby Isaac     bNew->ctx   = b->ctx;
2525f971fd6bSMatthew G. Knepley     bNew->type  = b->type;
2526dff059c6SToby Isaac     bNew->field = b->field;
2527dff059c6SToby Isaac     bNew->func  = b->func;
2528dff059c6SToby Isaac 
2529dff059c6SToby Isaac     *lastnext = bNew;
2530dff059c6SToby Isaac     lastnext = &(bNew->next);
2531dff059c6SToby Isaac   }
2532dff059c6SToby Isaac   PetscFunctionReturn(0);
2533dff059c6SToby Isaac }
2534dff059c6SToby Isaac 
2535da51fcedSMatthew G. Knepley /*@
2536da51fcedSMatthew G. Knepley   PetscDSCopyEquations - Copy all pointwise function pointers to the new problem
2537da51fcedSMatthew G. Knepley 
2538da51fcedSMatthew G. Knepley   Not collective
2539da51fcedSMatthew G. Knepley 
2540da51fcedSMatthew G. Knepley   Input Parameter:
2541da51fcedSMatthew G. Knepley . prob - The PetscDS object
2542da51fcedSMatthew G. Knepley 
2543da51fcedSMatthew G. Knepley   Output Parameter:
2544da51fcedSMatthew G. Knepley . newprob - The PetscDS copy
2545da51fcedSMatthew G. Knepley 
2546da51fcedSMatthew G. Knepley   Level: intermediate
2547da51fcedSMatthew G. Knepley 
2548da51fcedSMatthew G. Knepley .seealso: PetscDSSetResidual(), PetscDSSetJacobian(), PetscDSSetRiemannSolver(), PetscDSSetBdResidual(), PetscDSSetBdJacobian(), PetscDSCreate()
2549da51fcedSMatthew G. Knepley @*/
2550da51fcedSMatthew G. Knepley PetscErrorCode PetscDSCopyEquations(PetscDS prob, PetscDS newprob)
2551da51fcedSMatthew G. Knepley {
2552da51fcedSMatthew G. Knepley   PetscInt       Nf, Ng, f, g;
2553da51fcedSMatthew G. Knepley   PetscErrorCode ierr;
2554da51fcedSMatthew G. Knepley 
2555da51fcedSMatthew G. Knepley   PetscFunctionBegin;
2556da51fcedSMatthew G. Knepley   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
2557da51fcedSMatthew G. Knepley   PetscValidHeaderSpecific(newprob, PETSCDS_CLASSID, 2);
2558da51fcedSMatthew G. Knepley   ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr);
2559da51fcedSMatthew G. Knepley   ierr = PetscDSGetNumFields(newprob, &Ng);CHKERRQ(ierr);
2560da51fcedSMatthew G. Knepley   if (Nf != Ng) SETERRQ2(PetscObjectComm((PetscObject) prob), PETSC_ERR_ARG_SIZ, "Number of fields must match %D != %D", Nf, Ng);CHKERRQ(ierr);
2561da51fcedSMatthew G. Knepley   for (f = 0; f < Nf; ++f) {
2562da51fcedSMatthew G. Knepley     PetscPointFunc   obj;
2563da51fcedSMatthew G. Knepley     PetscPointFunc   f0, f1;
2564da51fcedSMatthew G. Knepley     PetscPointJac    g0, g1, g2, g3;
2565da51fcedSMatthew G. Knepley     PetscBdPointFunc f0Bd, f1Bd;
2566da51fcedSMatthew G. Knepley     PetscBdPointJac  g0Bd, g1Bd, g2Bd, g3Bd;
2567da51fcedSMatthew G. Knepley     PetscRiemannFunc r;
2568da51fcedSMatthew G. Knepley 
2569da51fcedSMatthew G. Knepley     ierr = PetscDSGetObjective(prob, f, &obj);CHKERRQ(ierr);
2570da51fcedSMatthew G. Knepley     ierr = PetscDSGetResidual(prob, f, &f0, &f1);CHKERRQ(ierr);
2571da51fcedSMatthew G. Knepley     ierr = PetscDSGetBdResidual(prob, f, &f0Bd, &f1Bd);CHKERRQ(ierr);
2572da51fcedSMatthew G. Knepley     ierr = PetscDSGetRiemannSolver(prob, f, &r);CHKERRQ(ierr);
2573da51fcedSMatthew G. Knepley     ierr = PetscDSSetObjective(newprob, f, obj);CHKERRQ(ierr);
2574da51fcedSMatthew G. Knepley     ierr = PetscDSSetResidual(newprob, f, f0, f1);CHKERRQ(ierr);
2575da51fcedSMatthew G. Knepley     ierr = PetscDSSetBdResidual(newprob, f, f0Bd, f1Bd);CHKERRQ(ierr);
2576da51fcedSMatthew G. Knepley     ierr = PetscDSSetRiemannSolver(newprob, f, r);CHKERRQ(ierr);
2577da51fcedSMatthew G. Knepley     for (g = 0; g < Nf; ++g) {
2578da51fcedSMatthew G. Knepley       ierr = PetscDSGetJacobian(prob, f, g, &g0, &g1, &g2, &g3);CHKERRQ(ierr);
2579da51fcedSMatthew G. Knepley       ierr = PetscDSGetBdJacobian(prob, f, g, &g0Bd, &g1Bd, &g2Bd, &g3Bd);CHKERRQ(ierr);
2580da51fcedSMatthew G. Knepley       ierr = PetscDSSetJacobian(newprob, f, g, g0, g1, g2, g3);CHKERRQ(ierr);
2581da51fcedSMatthew G. Knepley       ierr = PetscDSSetBdJacobian(newprob, f, g, g0Bd, g1Bd, g2Bd, g3Bd);CHKERRQ(ierr);
2582da51fcedSMatthew G. Knepley     }
2583da51fcedSMatthew G. Knepley   }
2584da51fcedSMatthew G. Knepley   PetscFunctionReturn(0);
2585da51fcedSMatthew G. Knepley }
2586da51fcedSMatthew G. Knepley 
2587bc4ae4beSMatthew G. Knepley static PetscErrorCode PetscDSDestroy_Basic(PetscDS prob)
25882764a2aaSMatthew G. Knepley {
2589931fb3b8SToby Isaac   PetscErrorCode      ierr;
2590931fb3b8SToby Isaac 
25912764a2aaSMatthew G. Knepley   PetscFunctionBegin;
2592931fb3b8SToby Isaac   ierr = PetscFree(prob->data);CHKERRQ(ierr);
25932764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
25942764a2aaSMatthew G. Knepley }
25952764a2aaSMatthew G. Knepley 
2596bc4ae4beSMatthew G. Knepley static PetscErrorCode PetscDSInitialize_Basic(PetscDS prob)
25972764a2aaSMatthew G. Knepley {
25982764a2aaSMatthew G. Knepley   PetscFunctionBegin;
25992764a2aaSMatthew G. Knepley   prob->ops->setfromoptions = NULL;
26002764a2aaSMatthew G. Knepley   prob->ops->setup          = NULL;
26012764a2aaSMatthew G. Knepley   prob->ops->view           = NULL;
26022764a2aaSMatthew G. Knepley   prob->ops->destroy        = PetscDSDestroy_Basic;
26032764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
26042764a2aaSMatthew G. Knepley }
26052764a2aaSMatthew G. Knepley 
26062764a2aaSMatthew G. Knepley /*MC
26072764a2aaSMatthew G. Knepley   PETSCDSBASIC = "basic" - A discrete system with pointwise residual and boundary residual functions
26082764a2aaSMatthew G. Knepley 
26092764a2aaSMatthew G. Knepley   Level: intermediate
26102764a2aaSMatthew G. Knepley 
26112764a2aaSMatthew G. Knepley .seealso: PetscDSType, PetscDSCreate(), PetscDSSetType()
26122764a2aaSMatthew G. Knepley M*/
26132764a2aaSMatthew G. Knepley 
26142764a2aaSMatthew G. Knepley PETSC_EXTERN PetscErrorCode PetscDSCreate_Basic(PetscDS prob)
26152764a2aaSMatthew G. Knepley {
26162764a2aaSMatthew G. Knepley   PetscDS_Basic *b;
26172764a2aaSMatthew G. Knepley   PetscErrorCode      ierr;
26182764a2aaSMatthew G. Knepley 
26192764a2aaSMatthew G. Knepley   PetscFunctionBegin;
2620931fb3b8SToby Isaac   PetscValidHeaderSpecific(prob, PETSCDS_CLASSID, 1);
26212764a2aaSMatthew G. Knepley   ierr       = PetscNewLog(prob, &b);CHKERRQ(ierr);
26222764a2aaSMatthew G. Knepley   prob->data = b;
26232764a2aaSMatthew G. Knepley 
26242764a2aaSMatthew G. Knepley   ierr = PetscDSInitialize_Basic(prob);CHKERRQ(ierr);
26252764a2aaSMatthew G. Knepley   PetscFunctionReturn(0);
26262764a2aaSMatthew G. Knepley }
2627