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