SWE: Fix set context bug after change in the interface

Author: valeriabarra

examples/fluids/shallow-water/shallowwater.c

@@ -385,10 +385,19 @@ int main(int argc, char **argv) {
   // Free edgenodes structure array
   ierr = PetscFree(edgenodes); CHKERRQ(ierr);
 
-  // Setup libCEED's objects
+  // Setup libCEED's objects and contexts
   ierr = PetscMalloc1(1, &ceeddata); CHKERRQ(ierr);
+  CeedQFunctionContextCreate(ceed, &ceeddata->physCtx);
+  CeedQFunctionContextSetData(ceeddata->physCtx, CEED_MEM_HOST,
+                              CEED_USE_POINTER, sizeof physCtxData,
+                              &physCtxData);
+  CeedQFunctionContextCreate(ceed, &ceeddata->problCtx);
+  CeedQFunctionContextSetData(ceeddata->problCtx, CEED_MEM_HOST,
+                              CEED_USE_POINTER, sizeof problCtxData,
+                              &problCtxData);
+
   ierr = SetupLibceed(dm, ceed, degree, qextra, ncompx, ncompq, user,
-                      ceeddata, problem, &physCtxData, &problCtxData);
+                      ceeddata, problem);
   CHKERRQ(ierr);
 
   // Set up PETSc context

examples/fluids/shallow-water/src/setup.c

@@ -651,9 +651,7 @@ PetscErrorCode VectorPlacePetscVec(CeedVector c, Vec p) {
 
 PetscErrorCode SetupLibceed(DM dm, Ceed ceed, CeedInt degree, CeedInt qextra,
                             PetscInt ncompx, PetscInt ncompq, User user,
-                            CeedData data, problemData *problem,
-                            PhysicsContext physCtxData,
-                            ProblemContext problCtxData) {
+                            CeedData data, problemData *problem) {
   int ierr;
   DM dmcoord;
   Vec Xloc;
@@ -831,13 +829,7 @@ PetscErrorCode SetupLibceed(DM dm, Ceed ceed, CeedInt degree, CeedInt qextra,
   user->op_jacobian = op_jacobian;
 
   // Set up the libCEED context
-  CeedQFunctionContextCreate(ceed, &data->physCtx);
-  CeedQFunctionContextSetData(data->physCtx, CEED_MEM_HOST, CEED_USE_POINTER,
-                              sizeof physCtxData, &physCtxData);
   CeedQFunctionSetContext(qf_ics, data->physCtx);
-  CeedQFunctionContextCreate(ceed, &data->problCtx);
-  CeedQFunctionContextSetData(data->problCtx, CEED_MEM_HOST, CEED_USE_POINTER,
-                              sizeof problCtxData, &problCtxData);
   CeedQFunctionSetContext(qf_explicit, data->problCtx);
   CeedQFunctionSetContext(qf_implicit, data->problCtx);
   CeedQFunctionSetContext(qf_jacobian, data->problCtx);

examples/fluids/shallow-water/sw_headers.h

@@ -183,9 +183,7 @@ PetscErrorCode CreateRestrictionPlex(Ceed ceed, DM dm, CeedInt P, CeedInt ncomp,
 // Auxiliary function to set up libCEED objects for a given degree
 PetscErrorCode SetupLibceed(DM dm, Ceed ceed, CeedInt degree, CeedInt qextra,
                             const PetscInt ncompx, PetscInt ncompq, User user,
-                            CeedData data, problemData *problem,
-                            PhysicsContext physCtxData,
-                            ProblemContext problCtxData);
+                            CeedData data, problemData *problem);
 
 // -----------------------------------------------------------------------------
 // RHS (Explicit part in time-stepper) function setup