xref: /libCEED/examples/fluids/navierstokes.c (revision 26aa2eee9110c2e2e9a928983954a91a28fd4e25)
1ccaff030SJeremy L Thompson // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
2ccaff030SJeremy L Thompson // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
3ccaff030SJeremy L Thompson // reserved. See files LICENSE and NOTICE for details.
4ccaff030SJeremy L Thompson //
5ccaff030SJeremy L Thompson // This file is part of CEED, a collection of benchmarks, miniapps, software
6ccaff030SJeremy L Thompson // libraries and APIs for efficient high-order finite element and spectral
7ccaff030SJeremy L Thompson // element discretizations for exascale applications. For more information and
8ccaff030SJeremy L Thompson // source code availability see http://github.com/ceed.
9ccaff030SJeremy L Thompson //
10ccaff030SJeremy L Thompson // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11ccaff030SJeremy L Thompson // a collaborative effort of two U.S. Department of Energy organizations (Office
12ccaff030SJeremy L Thompson // of Science and the National Nuclear Security Administration) responsible for
13ccaff030SJeremy L Thompson // the planning and preparation of a capable exascale ecosystem, including
14ccaff030SJeremy L Thompson // software, applications, hardware, advanced system engineering and early
15ccaff030SJeremy L Thompson // testbed platforms, in support of the nation's exascale computing imperative.
16ccaff030SJeremy L Thompson 
17ccaff030SJeremy L Thompson //                        libCEED + PETSc Example: Navier-Stokes
18ccaff030SJeremy L Thompson //
19ccaff030SJeremy L Thompson // This example demonstrates a simple usage of libCEED with PETSc to solve a
20ccaff030SJeremy L Thompson // Navier-Stokes problem.
21ccaff030SJeremy L Thompson //
22ccaff030SJeremy L Thompson // The code is intentionally "raw", using only low-level communication
23ccaff030SJeremy L Thompson // primitives.
24ccaff030SJeremy L Thompson //
25ccaff030SJeremy L Thompson // Build with:
26ccaff030SJeremy L Thompson //
27ccaff030SJeremy L Thompson //     make [PETSC_DIR=</path/to/petsc>] [CEED_DIR=</path/to/libceed>] navierstokes
28ccaff030SJeremy L Thompson //
29ccaff030SJeremy L Thompson // Sample runs:
30ccaff030SJeremy L Thompson //
3184d34d69SLeila Ghaffari //     ./navierstokes -ceed /cpu/self -problem density_current -degree 1
3228688798Sjeremylt //     ./navierstokes -ceed /gpu/cuda -problem advection -degree 1
33ccaff030SJeremy L Thompson //
34dc8efd83SLeila Ghaffari //TESTARGS(name="dc_explicit") -ceed {ceed_resource} -test -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -thetaC -35. -ts_dt 1e-3 -compare_final_state_atol 1E-11 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-dc-explicit.bin
35dc8efd83SLeila Ghaffari //TESTARGS(name="dc_implicit_stab_none") -ceed {ceed_resource} -test -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -thetaC -35. -ksp_atol 1e-4 -ksp_rtol 1e-3 -ksp_type bcgs -snes_atol 1e-3 -snes_lag_jacobian 100 -snes_lag_jacobian_persists -snes_mf_operator -ts_dt 1e-3 -implicit -ts_type alpha -compare_final_state_atol 5E-4 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-dc-implicit-stab-none.bin
36dc8efd83SLeila Ghaffari //TESTARGS(name="adv_rotation_explicit_strong") -ceed {ceed_resource} -test -problem advection -strong_form 1 -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ts_dt 1e-3 -compare_final_state_atol 1E-11 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv-rotation-explicit-strong.bin
37dc8efd83SLeila Ghaffari //TESTARGS(name="adv_rotation_implicit_stab_supg") -ceed {ceed_resource} -test -problem advection -CtauS .3 -stab supg -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ksp_atol 1e-4 -ksp_rtol 1e-3 -ksp_type bcgs -snes_atol 1e-3 -snes_lag_jacobian 100 -snes_lag_jacobian_persists -snes_mf_operator -ts_dt 1e-3 -implicit -ts_type alpha -compare_final_state_atol 5E-4 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv-rotation-implicit-stab-supg.bin
38dc8efd83SLeila Ghaffari //TESTARGS(name="adv_translation_implicit_stab_su") -ceed {ceed_resource} -test -problem advection -CtauS .3 -stab su -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ksp_atol 1e-4 -ksp_rtol 1e-3 -ksp_type bcgs -snes_atol 1e-3 -snes_lag_jacobian 100 -snes_lag_jacobian_persists -snes_mf_operator -ts_dt 1e-3 -implicit -ts_type alpha -problem_advection_wind translation -problem_advection_wind_translation .53,-1.33,-2.65 -compare_final_state_atol 5E-4 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv-translation-implicit-stab-su.bin
39dc8efd83SLeila Ghaffari //TESTARGS(name="adv2d_rotation_explicit_strong") -ceed {ceed_resource} -test -problem advection2d -strong_form 1 -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ts_dt 1e-3 -compare_final_state_atol 1E-11 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv2d-rotation-explicit-strong.bin
40dc8efd83SLeila Ghaffari //TESTARGS(name="adv2d_rotation_implicit_stab_supg") -ceed {ceed_resource} -test -problem advection2d -CtauS .3 -stab supg -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ksp_atol 1e-4 -ksp_rtol 1e-3 -ksp_type bcgs -snes_atol 1e-3 -snes_lag_jacobian 100 -snes_lag_jacobian_persists -snes_mf_operator -ts_dt 1e-3 -implicit -ts_type alpha -compare_final_state_atol 5E-4 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv2d-rotation-implicit-stab-supg.bin
41dc8efd83SLeila Ghaffari //TESTARGS(name="adv2d_translation_implicit_stab_su") -ceed {ceed_resource} -test -problem advection2d -CtauS .3 -stab su -degree 3 -dm_plex_box_faces 1,1,2 -units_kilogram 1e-9 -lx 125 -ly 125 -lz 250 -center 62.5,62.5,187.5 -rc 100. -ksp_atol 1e-4 -ksp_rtol 1e-3 -ksp_type bcgs -snes_atol 1e-3 -snes_lag_jacobian 100 -snes_lag_jacobian_persists -snes_mf_operator -ts_dt 1e-3 -implicit -ts_type alpha -problem_advection_wind translation -problem_advection_wind_translation .53,-1.33,0 -compare_final_state_atol 5E-4 -compare_final_state_filename examples/fluids/tests-output/fluids-navierstokes-adv2d-translation-implicit-stab-su.bin
42ccaff030SJeremy L Thompson 
43ccaff030SJeremy L Thompson /// @file
44ccaff030SJeremy L Thompson /// Navier-Stokes example using PETSc
45ccaff030SJeremy L Thompson 
46ccaff030SJeremy L Thompson const char help[] = "Solve Navier-Stokes using PETSc and libCEED\n";
47ccaff030SJeremy L Thompson 
48ccaff030SJeremy L Thompson #include <ceed.h>
493d576824SJeremy L Thompson #include <petscdmplex.h>
50ccaff030SJeremy L Thompson #include <petscsys.h>
513d576824SJeremy L Thompson #include <petscts.h>
523d576824SJeremy L Thompson #include <stdbool.h>
53ccaff030SJeremy L Thompson #include "advection.h"
54ccaff030SJeremy L Thompson #include "advection2d.h"
553d576824SJeremy L Thompson #include "common.h"
56eb088987SLeila Ghaffari #include "euler-vortex.h"
57ccaff030SJeremy L Thompson #include "densitycurrent.h"
583d576824SJeremy L Thompson #include "setup-boundary.h"
59ccaff030SJeremy L Thompson 
6084d34d69SLeila Ghaffari #if PETSC_VERSION_LT(3,14,0)
6184d34d69SLeila Ghaffari #  define DMPlexGetClosureIndices(a,b,c,d,e,f,g,h,i) DMPlexGetClosureIndices(a,b,c,d,f,g,i)
6284d34d69SLeila Ghaffari #  define DMPlexRestoreClosureIndices(a,b,c,d,e,f,g,h,i) DMPlexRestoreClosureIndices(a,b,c,d,f,g,i)
6384d34d69SLeila Ghaffari #endif
6484d34d69SLeila Ghaffari 
653ab4fca6SValeria Barra #if PETSC_VERSION_LT(3,14,0)
663ab4fca6SValeria Barra #  define DMAddBoundary(a,b,c,d,e,f,g,h,i,j,k,l) DMAddBoundary(a,b,c,d,e,f,g,h,j,k,l)
673ab4fca6SValeria Barra #endif
683ab4fca6SValeria Barra 
6984d34d69SLeila Ghaffari // MemType Options
7084d34d69SLeila Ghaffari static const char *const memTypes[] = {
7184d34d69SLeila Ghaffari   "host",
7284d34d69SLeila Ghaffari   "device",
7384d34d69SLeila Ghaffari   "memType", "CEED_MEM_", NULL
7484d34d69SLeila Ghaffari };
7584d34d69SLeila Ghaffari 
76ccaff030SJeremy L Thompson // Problem Options
77ccaff030SJeremy L Thompson typedef enum {
78ccaff030SJeremy L Thompson   NS_DENSITY_CURRENT = 0,
79ccaff030SJeremy L Thompson   NS_ADVECTION = 1,
80ccaff030SJeremy L Thompson   NS_ADVECTION2D = 2,
81eb088987SLeila Ghaffari   NS_EULER_VORTEX = 3,
82ccaff030SJeremy L Thompson } problemType;
83ccaff030SJeremy L Thompson static const char *const problemTypes[] = {
84ccaff030SJeremy L Thompson   "density_current",
85ccaff030SJeremy L Thompson   "advection",
86ccaff030SJeremy L Thompson   "advection2d",
87e5ed8c30SLeila Ghaffari   "euler_vortex",
8884d34d69SLeila Ghaffari   "problemType", "NS_", NULL
89ccaff030SJeremy L Thompson };
90ccaff030SJeremy L Thompson 
911184866aSLeila Ghaffari // Wind Options for Advection
921184866aSLeila Ghaffari typedef enum {
931184866aSLeila Ghaffari   ADVECTION_WIND_ROTATION = 0,
941184866aSLeila Ghaffari   ADVECTION_WIND_TRANSLATION = 1,
951184866aSLeila Ghaffari } WindType;
961184866aSLeila Ghaffari static const char *const WindTypes[] = {
971184866aSLeila Ghaffari   "rotation",
981184866aSLeila Ghaffari   "translation",
991184866aSLeila Ghaffari   "WindType", "ADVECTION_WIND_", NULL
1001184866aSLeila Ghaffari };
1011184866aSLeila Ghaffari 
102ccaff030SJeremy L Thompson typedef enum {
103ccaff030SJeremy L Thompson   STAB_NONE = 0,
104ccaff030SJeremy L Thompson   STAB_SU = 1,   // Streamline Upwind
105ccaff030SJeremy L Thompson   STAB_SUPG = 2, // Streamline Upwind Petrov-Galerkin
106ccaff030SJeremy L Thompson } StabilizationType;
107ccaff030SJeremy L Thompson static const char *const StabilizationTypes[] = {
10884d34d69SLeila Ghaffari   "none",
109ccaff030SJeremy L Thompson   "SU",
110ccaff030SJeremy L Thompson   "SUPG",
111ccaff030SJeremy L Thompson   "StabilizationType", "STAB_", NULL
112ccaff030SJeremy L Thompson };
113ccaff030SJeremy L Thompson 
114ccaff030SJeremy L Thompson // Problem specific data
115ccaff030SJeremy L Thompson typedef struct {
1168b982baeSLeila Ghaffari   CeedInt dim, qdatasizeVol, qdatasizeSur;
117ebb4b9bdSLeila Ghaffari   CeedQFunctionUser setupVol, setupSur, ics, applyVol_rhs, applyVol_ifunction,
118e2043960SLeila Ghaffari                     applySur;
119ccaff030SJeremy L Thompson   PetscErrorCode (*bc)(PetscInt, PetscReal, const PetscReal[], PetscInt,
120ccaff030SJeremy L Thompson                        PetscScalar[], void *);
1217659d40cSLeila Ghaffari   const char *setupVol_loc, *setupSur_loc, *ics_loc, *applyVol_rhs_loc,
122e2043960SLeila Ghaffari         *applyVol_ifunction_loc, *applySur_loc;
123ccaff030SJeremy L Thompson   const bool non_zero_time;
124ccaff030SJeremy L Thompson } problemData;
125ccaff030SJeremy L Thompson 
126ccaff030SJeremy L Thompson problemData problemOptions[] = {
127ccaff030SJeremy L Thompson   [NS_DENSITY_CURRENT] = {
128ccaff030SJeremy L Thompson     .dim                       = 3,
129ea6e0f84SLeila Ghaffari     .qdatasizeVol              = 10,
1308b982baeSLeila Ghaffari     .qdatasizeSur              = 4,
131b0137797SLeila Ghaffari     .setupVol                  = Setup,
132b0137797SLeila Ghaffari     .setupVol_loc              = Setup_loc,
133356fbf4bSLeila Ghaffari     .setupSur                  = SetupBoundary,
134356fbf4bSLeila Ghaffari     .setupSur_loc              = SetupBoundary_loc,
135ccaff030SJeremy L Thompson     .ics                       = ICsDC,
136ccaff030SJeremy L Thompson     .ics_loc                   = ICsDC_loc,
137c96c872fSLeila Ghaffari     .applyVol_rhs              = DC,
138c96c872fSLeila Ghaffari     .applyVol_rhs_loc          = DC_loc,
139c96c872fSLeila Ghaffari     .applyVol_ifunction        = IFunction_DC,
140c96c872fSLeila Ghaffari     .applyVol_ifunction_loc    = IFunction_DC_loc,
141ccaff030SJeremy L Thompson     .bc                        = Exact_DC,
14284d34d69SLeila Ghaffari     .non_zero_time             = PETSC_FALSE,
143ccaff030SJeremy L Thompson   },
144ccaff030SJeremy L Thompson   [NS_ADVECTION] = {
145ccaff030SJeremy L Thompson     .dim                       = 3,
146ea6e0f84SLeila Ghaffari     .qdatasizeVol              = 10,
1478b982baeSLeila Ghaffari     .qdatasizeSur              = 4,
148b0137797SLeila Ghaffari     .setupVol                  = Setup,
149b0137797SLeila Ghaffari     .setupVol_loc              = Setup_loc,
150356fbf4bSLeila Ghaffari     .setupSur                  = SetupBoundary,
151356fbf4bSLeila Ghaffari     .setupSur_loc              = SetupBoundary_loc,
152ccaff030SJeremy L Thompson     .ics                       = ICsAdvection,
153ccaff030SJeremy L Thompson     .ics_loc                   = ICsAdvection_loc,
154c96c872fSLeila Ghaffari     .applyVol_rhs              = Advection,
155c96c872fSLeila Ghaffari     .applyVol_rhs_loc          = Advection_loc,
156c96c872fSLeila Ghaffari     .applyVol_ifunction        = IFunction_Advection,
157c96c872fSLeila Ghaffari     .applyVol_ifunction_loc    = IFunction_Advection_loc,
1587659d40cSLeila Ghaffari     .applySur                  = Advection_Sur,
1597659d40cSLeila Ghaffari     .applySur_loc              = Advection_Sur_loc,
160ccaff030SJeremy L Thompson     .bc                        = Exact_Advection,
16184d34d69SLeila Ghaffari     .non_zero_time             = PETSC_FALSE,
162ccaff030SJeremy L Thompson   },
163ccaff030SJeremy L Thompson   [NS_ADVECTION2D] = {
164ccaff030SJeremy L Thompson     .dim                       = 2,
165ea6e0f84SLeila Ghaffari     .qdatasizeVol              = 5,
1668b982baeSLeila Ghaffari     .qdatasizeSur              = 3,
167c96c872fSLeila Ghaffari     .setupVol                  = Setup2d,
168c96c872fSLeila Ghaffari     .setupVol_loc              = Setup2d_loc,
169b0137797SLeila Ghaffari     .setupSur                  = SetupBoundary2d,
170b0137797SLeila Ghaffari     .setupSur_loc              = SetupBoundary2d_loc,
171ccaff030SJeremy L Thompson     .ics                       = ICsAdvection2d,
172ccaff030SJeremy L Thompson     .ics_loc                   = ICsAdvection2d_loc,
173c96c872fSLeila Ghaffari     .applyVol_rhs              = Advection2d,
174c96c872fSLeila Ghaffari     .applyVol_rhs_loc          = Advection2d_loc,
175c96c872fSLeila Ghaffari     .applyVol_ifunction        = IFunction_Advection2d,
176c96c872fSLeila Ghaffari     .applyVol_ifunction_loc    = IFunction_Advection2d_loc,
1777659d40cSLeila Ghaffari     .applySur                  = Advection2d_Sur,
1787659d40cSLeila Ghaffari     .applySur_loc              = Advection2d_Sur_loc,
179ccaff030SJeremy L Thompson     .bc                        = Exact_Advection2d,
18084d34d69SLeila Ghaffari     .non_zero_time             = PETSC_TRUE,
181ccaff030SJeremy L Thompson   },
182eb088987SLeila Ghaffari   [NS_EULER_VORTEX] = {
183eb088987SLeila Ghaffari     .dim                       = 3,
184eb088987SLeila Ghaffari     .qdatasizeVol              = 10,
185eb088987SLeila Ghaffari     .qdatasizeSur              = 4,
186eb088987SLeila Ghaffari     .setupVol                  = Setup,
187eb088987SLeila Ghaffari     .setupVol_loc              = Setup_loc,
188eb088987SLeila Ghaffari     .setupSur                  = SetupBoundary,
189eb088987SLeila Ghaffari     .setupSur_loc              = SetupBoundary_loc,
190eb088987SLeila Ghaffari     .ics                       = ICsEuler,
191eb088987SLeila Ghaffari     .ics_loc                   = ICsEuler_loc,
192eb088987SLeila Ghaffari     .applyVol_rhs              = Euler,
193eb088987SLeila Ghaffari     .applyVol_rhs_loc          = Euler_loc,
194b6df2e11SLeila Ghaffari     .applySur                  = Euler_Sur,
195b6df2e11SLeila Ghaffari     .applySur_loc              = Euler_Sur_loc,
196eb088987SLeila Ghaffari     .bc                        = Exact_Euler,
19747ff3025SLeila Ghaffari     .non_zero_time             = PETSC_TRUE,
198eb088987SLeila Ghaffari   },
199ccaff030SJeremy L Thompson };
200ccaff030SJeremy L Thompson 
201ccaff030SJeremy L Thompson // PETSc user data
202ccaff030SJeremy L Thompson typedef struct User_ *User;
203ccaff030SJeremy L Thompson typedef struct Units_ *Units;
204ccaff030SJeremy L Thompson 
205ccaff030SJeremy L Thompson struct User_ {
206ccaff030SJeremy L Thompson   MPI_Comm comm;
207ccaff030SJeremy L Thompson   PetscInt outputfreq;
208ccaff030SJeremy L Thompson   DM dm;
209ccaff030SJeremy L Thompson   DM dmviz;
210ccaff030SJeremy L Thompson   Mat interpviz;
211ccaff030SJeremy L Thompson   Ceed ceed;
212ccaff030SJeremy L Thompson   Units units;
213ccaff030SJeremy L Thompson   CeedVector qceed, qdotceed, gceed;
2141e150236SLeila Ghaffari   CeedOperator op_rhs_vol, op_rhs, op_ifunction_vol, op_ifunction;
215ccaff030SJeremy L Thompson   Vec M;
216d99129b9SLeila Ghaffari   char outputdir[PETSC_MAX_PATH_LEN];
217ccaff030SJeremy L Thompson   PetscInt contsteps;
2185bf10c6fSLeila Ghaffari   EulerContext ctxEulerData;
219ccaff030SJeremy L Thompson };
220ccaff030SJeremy L Thompson 
221ccaff030SJeremy L Thompson struct Units_ {
222ccaff030SJeremy L Thompson   // fundamental units
223ccaff030SJeremy L Thompson   PetscScalar meter;
224ccaff030SJeremy L Thompson   PetscScalar kilogram;
225ccaff030SJeremy L Thompson   PetscScalar second;
226ccaff030SJeremy L Thompson   PetscScalar Kelvin;
227ccaff030SJeremy L Thompson   // derived units
228ccaff030SJeremy L Thompson   PetscScalar Pascal;
229ccaff030SJeremy L Thompson   PetscScalar JperkgK;
230ccaff030SJeremy L Thompson   PetscScalar mpersquareds;
231ccaff030SJeremy L Thompson   PetscScalar WpermK;
232ccaff030SJeremy L Thompson   PetscScalar kgpercubicm;
233ccaff030SJeremy L Thompson   PetscScalar kgpersquaredms;
234ccaff030SJeremy L Thompson   PetscScalar Joulepercubicm;
23516c0476cSLeila Ghaffari   PetscScalar Joule;
236ccaff030SJeremy L Thompson };
237ccaff030SJeremy L Thompson 
238ccaff030SJeremy L Thompson typedef struct SimpleBC_ *SimpleBC;
239ccaff030SJeremy L Thompson struct SimpleBC_ {
2407659d40cSLeila Ghaffari   PetscInt nwall, nslip[3];
2417659d40cSLeila Ghaffari   PetscInt walls[6], slips[3][6];
24284d34d69SLeila Ghaffari   PetscBool userbc;
243ccaff030SJeremy L Thompson };
244ccaff030SJeremy L Thompson 
245ccaff030SJeremy L Thompson // Essential BC dofs are encoded in closure indices as -(i+1).
246ccaff030SJeremy L Thompson static PetscInt Involute(PetscInt i) {
247ccaff030SJeremy L Thompson   return i >= 0 ? i : -(i+1);
248ccaff030SJeremy L Thompson }
249ccaff030SJeremy L Thompson 
250ccaff030SJeremy L Thompson // Utility function to create local CEED restriction
251ccaff030SJeremy L Thompson static PetscErrorCode CreateRestrictionFromPlex(Ceed ceed, DM dm, CeedInt P,
25284d34d69SLeila Ghaffari     CeedInt height, DMLabel domainLabel, CeedInt value,
253ccaff030SJeremy L Thompson     CeedElemRestriction *Erestrict) {
254ccaff030SJeremy L Thompson 
255ccaff030SJeremy L Thompson   PetscSection section;
2561184866aSLeila Ghaffari   PetscInt p, Nelem, Ndof, *erestrict, eoffset, nfields, dim, depth;
2570c6c0b13SLeila Ghaffari   DMLabel depthLabel;
2580c6c0b13SLeila Ghaffari   IS depthIS, iterIS;
25984d34d69SLeila Ghaffari   Vec Uloc;
2600c6c0b13SLeila Ghaffari   const PetscInt *iterIndices;
261ccaff030SJeremy L Thompson   PetscErrorCode ierr;
262ccaff030SJeremy L Thompson 
263ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
264ccaff030SJeremy L Thompson   ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr);
265da51bdd9SLeila Ghaffari   dim -= height;
266ccaff030SJeremy L Thompson   ierr = DMGetLocalSection(dm, &section); CHKERRQ(ierr);
267ccaff030SJeremy L Thompson   ierr = PetscSectionGetNumFields(section, &nfields); CHKERRQ(ierr);
268ccaff030SJeremy L Thompson   PetscInt ncomp[nfields], fieldoff[nfields+1];
269ccaff030SJeremy L Thompson   fieldoff[0] = 0;
270ccaff030SJeremy L Thompson   for (PetscInt f=0; f<nfields; f++) {
271ccaff030SJeremy L Thompson     ierr = PetscSectionGetFieldComponents(section, f, &ncomp[f]); CHKERRQ(ierr);
272ccaff030SJeremy L Thompson     fieldoff[f+1] = fieldoff[f] + ncomp[f];
273ccaff030SJeremy L Thompson   }
274ccaff030SJeremy L Thompson 
2750c6c0b13SLeila Ghaffari   ierr = DMPlexGetDepth(dm, &depth); CHKERRQ(ierr);
2760c6c0b13SLeila Ghaffari   ierr = DMPlexGetDepthLabel(dm, &depthLabel); CHKERRQ(ierr);
2770c6c0b13SLeila Ghaffari   ierr = DMLabelGetStratumIS(depthLabel, depth - height, &depthIS); CHKERRQ(ierr);
2780c6c0b13SLeila Ghaffari   if (domainLabel) {
2790c6c0b13SLeila Ghaffari     IS domainIS;
2800c6c0b13SLeila Ghaffari     ierr = DMLabelGetStratumIS(domainLabel, value, &domainIS); CHKERRQ(ierr);
2811119eeeeSJed Brown     if (domainIS) { // domainIS is non-empty
2820c6c0b13SLeila Ghaffari       ierr = ISIntersect(depthIS, domainIS, &iterIS); CHKERRQ(ierr);
2830c6c0b13SLeila Ghaffari       ierr = ISDestroy(&domainIS); CHKERRQ(ierr);
2841119eeeeSJed Brown     } else { // domainIS is NULL (empty)
2851119eeeeSJed Brown       iterIS = NULL;
2861119eeeeSJed Brown     }
2870c6c0b13SLeila Ghaffari     ierr = ISDestroy(&depthIS); CHKERRQ(ierr);
2880c6c0b13SLeila Ghaffari   } else {
2890c6c0b13SLeila Ghaffari     iterIS = depthIS;
2900c6c0b13SLeila Ghaffari   }
2911119eeeeSJed Brown   if (iterIS) {
2920c6c0b13SLeila Ghaffari     ierr = ISGetLocalSize(iterIS, &Nelem); CHKERRQ(ierr);
2930c6c0b13SLeila Ghaffari     ierr = ISGetIndices(iterIS, &iterIndices); CHKERRQ(ierr);
2941119eeeeSJed Brown   } else {
2951119eeeeSJed Brown     Nelem = 0;
2961119eeeeSJed Brown     iterIndices = NULL;
2971119eeeeSJed Brown   }
298ccaff030SJeremy L Thompson   ierr = PetscMalloc1(Nelem*PetscPowInt(P, dim), &erestrict); CHKERRQ(ierr);
2990c6c0b13SLeila Ghaffari   for (p=0,eoffset=0; p<Nelem; p++) {
3000c6c0b13SLeila Ghaffari     PetscInt c = iterIndices[p];
301ccaff030SJeremy L Thompson     PetscInt numindices, *indices, nnodes;
30284d34d69SLeila Ghaffari     ierr = DMPlexGetClosureIndices(dm, section, section, c, PETSC_TRUE,
30384d34d69SLeila Ghaffari                                    &numindices, &indices, NULL, NULL);
30484d34d69SLeila Ghaffari     CHKERRQ(ierr);
30532b5ec5fSJed Brown     bool flip = false;
30632b5ec5fSJed Brown     if (height > 0) {
30732b5ec5fSJed Brown       PetscInt numCells, numFaces, start = -1;
30832b5ec5fSJed Brown       const PetscInt *orients, *faces, *cells;
30932b5ec5fSJed Brown       ierr = DMPlexGetSupport(dm, c, &cells); CHKERRQ(ierr);
31032b5ec5fSJed Brown       ierr = DMPlexGetSupportSize(dm, c, &numCells); CHKERRQ(ierr);
311ebb4b9bdSLeila Ghaffari       if (numCells != 1) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP,
312f259b054Svaleriabarra                                     "Expected one cell in support of exterior face, but got %D cells",
313f259b054Svaleriabarra                                     numCells);
31432b5ec5fSJed Brown       ierr = DMPlexGetCone(dm, cells[0], &faces); CHKERRQ(ierr);
31532b5ec5fSJed Brown       ierr = DMPlexGetConeSize(dm, cells[0], &numFaces); CHKERRQ(ierr);
31632b5ec5fSJed Brown       for (PetscInt i=0; i<numFaces; i++) {if (faces[i] == c) start = i;}
317ebb4b9bdSLeila Ghaffari       if (start < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_CORRUPT,
318f259b054Svaleriabarra                                 "Could not find face %D in cone of its support",
319f259b054Svaleriabarra                                 c);
32032b5ec5fSJed Brown       ierr = DMPlexGetConeOrientation(dm, cells[0], &orients); CHKERRQ(ierr);
32132b5ec5fSJed Brown       if (orients[start] < 0) flip = true;
32232b5ec5fSJed Brown     }
32384d34d69SLeila Ghaffari     if (numindices % fieldoff[nfields]) SETERRQ1(PETSC_COMM_SELF,
32484d34d69SLeila Ghaffari           PETSC_ERR_ARG_INCOMP, "Number of closure indices not compatible with Cell %D",
32584d34d69SLeila Ghaffari           c);
326ccaff030SJeremy L Thompson     nnodes = numindices / fieldoff[nfields];
327ccaff030SJeremy L Thompson     for (PetscInt i=0; i<nnodes; i++) {
32832b5ec5fSJed Brown       PetscInt ii = i;
32932b5ec5fSJed Brown       if (flip) {
33032b5ec5fSJed Brown         if (P == nnodes) ii = nnodes - 1 - i;
33132b5ec5fSJed Brown         else if (P*P == nnodes) {
33232b5ec5fSJed Brown           PetscInt row = i / P, col = i % P;
33332b5ec5fSJed Brown           ii = row + col * P;
334ebb4b9bdSLeila Ghaffari         } else SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP,
335f259b054Svaleriabarra                           "No support for flipping point with %D nodes != P (%D) or P^2",
336f259b054Svaleriabarra                           nnodes, P);
33732b5ec5fSJed Brown       }
338ccaff030SJeremy L Thompson       // Check that indices are blocked by node and thus can be coalesced as a single field with
339ccaff030SJeremy L Thompson       // fieldoff[nfields] = sum(ncomp) components.
340ccaff030SJeremy L Thompson       for (PetscInt f=0; f<nfields; f++) {
341ccaff030SJeremy L Thompson         for (PetscInt j=0; j<ncomp[f]; j++) {
34232b5ec5fSJed Brown           if (Involute(indices[fieldoff[f]*nnodes + ii*ncomp[f] + j])
34332b5ec5fSJed Brown               != Involute(indices[ii*ncomp[0]]) + fieldoff[f] + j)
344ccaff030SJeremy L Thompson             SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP,
345ccaff030SJeremy L Thompson                      "Cell %D closure indices not interlaced for node %D field %D component %D",
34632b5ec5fSJed Brown                      c, ii, f, j);
347ccaff030SJeremy L Thompson         }
348ccaff030SJeremy L Thompson       }
349ccaff030SJeremy L Thompson       // Essential boundary conditions are encoded as -(loc+1), but we don't care so we decode.
35032b5ec5fSJed Brown       PetscInt loc = Involute(indices[ii*ncomp[0]]);
3516f55dfd5Svaleriabarra       erestrict[eoffset++] = loc;
352ccaff030SJeremy L Thompson     }
35384d34d69SLeila Ghaffari     ierr = DMPlexRestoreClosureIndices(dm, section, section, c, PETSC_TRUE,
35484d34d69SLeila Ghaffari                                        &numindices, &indices, NULL, NULL);
35584d34d69SLeila Ghaffari     CHKERRQ(ierr);
356ccaff030SJeremy L Thompson   }
3570c6c0b13SLeila Ghaffari   if (eoffset != Nelem*PetscPowInt(P, dim))
3580c6c0b13SLeila Ghaffari     SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_LIB,
3590c6c0b13SLeila Ghaffari              "ElemRestriction of size (%D,%D) initialized %D nodes", Nelem,
360ccaff030SJeremy L Thompson              PetscPowInt(P, dim),eoffset);
3611119eeeeSJed Brown   if (iterIS) {
3620c6c0b13SLeila Ghaffari     ierr = ISRestoreIndices(iterIS, &iterIndices); CHKERRQ(ierr);
3631119eeeeSJed Brown   }
3640c6c0b13SLeila Ghaffari   ierr = ISDestroy(&iterIS); CHKERRQ(ierr);
3650c6c0b13SLeila Ghaffari 
366ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &Uloc); CHKERRQ(ierr);
367ccaff030SJeremy L Thompson   ierr = VecGetLocalSize(Uloc, &Ndof); CHKERRQ(ierr);
368ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &Uloc); CHKERRQ(ierr);
3696f55dfd5Svaleriabarra   CeedElemRestrictionCreate(ceed, Nelem, PetscPowInt(P, dim), fieldoff[nfields],
3706f55dfd5Svaleriabarra                             1, Ndof, CEED_MEM_HOST, CEED_COPY_VALUES, erestrict,
3716f55dfd5Svaleriabarra                             Erestrict);
372ccaff030SJeremy L Thompson   ierr = PetscFree(erestrict); CHKERRQ(ierr);
373ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
374ccaff030SJeremy L Thompson }
375ccaff030SJeremy L Thompson 
376c96c872fSLeila Ghaffari // Utility function to get Ceed Restriction for each domain
3771e150236SLeila Ghaffari static PetscErrorCode GetRestrictionForDomain(Ceed ceed, DM dm, CeedInt height,
3781e150236SLeila Ghaffari     DMLabel domainLabel, PetscInt value, CeedInt P, CeedInt Q, CeedInt qdatasize,
3791e150236SLeila Ghaffari     CeedElemRestriction *restrictq, CeedElemRestriction *restrictx,
3801e150236SLeila Ghaffari     CeedElemRestriction *restrictqdi) {
381c96c872fSLeila Ghaffari 
382c96c872fSLeila Ghaffari   DM dmcoord;
3831e150236SLeila Ghaffari   CeedInt dim, localNelem;
3841e150236SLeila Ghaffari   CeedInt Qdim;
385c96c872fSLeila Ghaffari   PetscErrorCode ierr;
386c96c872fSLeila Ghaffari 
387c96c872fSLeila Ghaffari   PetscFunctionBeginUser;
3881e150236SLeila Ghaffari   ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr);
3891e150236SLeila Ghaffari   dim -= height;
3901e150236SLeila Ghaffari   Qdim = CeedIntPow(Q, dim);
391c96c872fSLeila Ghaffari   ierr = DMGetCoordinateDM(dm, &dmcoord); CHKERRQ(ierr);
392c96c872fSLeila Ghaffari   ierr = DMPlexSetClosurePermutationTensor(dmcoord, PETSC_DETERMINE, NULL);
393c96c872fSLeila Ghaffari   CHKERRQ(ierr);
394ebb4b9bdSLeila Ghaffari   ierr = CreateRestrictionFromPlex(ceed, dm, P, height, domainLabel, value,
395ebb4b9bdSLeila Ghaffari                                    restrictq);
396c96c872fSLeila Ghaffari   CHKERRQ(ierr);
397ebb4b9bdSLeila Ghaffari   ierr = CreateRestrictionFromPlex(ceed, dmcoord, 2, height, domainLabel, value,
398ebb4b9bdSLeila Ghaffari                                    restrictx);
399c96c872fSLeila Ghaffari   CHKERRQ(ierr);
400c96c872fSLeila Ghaffari   CeedElemRestrictionGetNumElements(*restrictq, &localNelem);
401c96c872fSLeila Ghaffari   CeedElemRestrictionCreateStrided(ceed, localNelem, Qdim,
402c96c872fSLeila Ghaffari                                    qdatasize, qdatasize*localNelem*Qdim,
403c96c872fSLeila Ghaffari                                    CEED_STRIDES_BACKEND, restrictqdi);
404c96c872fSLeila Ghaffari   PetscFunctionReturn(0);
405c96c872fSLeila Ghaffari }
406c96c872fSLeila Ghaffari 
4071e150236SLeila Ghaffari // Utility function to create CEED Composite Operator for the entire domain
4087659d40cSLeila Ghaffari static PetscErrorCode CreateOperatorForDomain(Ceed ceed, DM dm, SimpleBC bc,
4094438636fSLeila Ghaffari     problemType problemChoice, WindType wind_type, CeedOperator op_applyVol,
410e2043960SLeila Ghaffari     CeedQFunction qf_applySur, CeedQFunction qf_setupSur,CeedInt height,
4114438636fSLeila Ghaffari     CeedInt numP_Sur, CeedInt numQ_Sur, CeedInt qdatasizeSur, CeedInt NqptsSur,
4124438636fSLeila Ghaffari     CeedBasis basisxSur, CeedBasis basisqSur, CeedOperator *op_apply) {
413ca3ac6ddSLeila Ghaffari 
4147659d40cSLeila Ghaffari   CeedInt dim, nFace;
4154438636fSLeila Ghaffari   PetscInt lsize;
4161e150236SLeila Ghaffari   Vec Xloc;
4174438636fSLeila Ghaffari   CeedVector xcorners;
418ca3ac6ddSLeila Ghaffari   DMLabel domainLabel;
4191e150236SLeila Ghaffari   PetscScalar *x;
420ca3ac6ddSLeila Ghaffari   PetscErrorCode ierr;
421ca3ac6ddSLeila Ghaffari 
422ca3ac6ddSLeila Ghaffari   PetscFunctionBeginUser;
423ca3ac6ddSLeila Ghaffari   // Composite Operaters
424ca3ac6ddSLeila Ghaffari   CeedCompositeOperatorCreate(ceed, op_apply);
4254438636fSLeila Ghaffari   // --Apply a Sub-Operator for the volume
4264438636fSLeila Ghaffari   CeedCompositeOperatorAddSub(*op_apply, op_applyVol);
427ca3ac6ddSLeila Ghaffari 
4284438636fSLeila Ghaffari   // Required data for in/outflow BCs
4291e150236SLeila Ghaffari   ierr = DMGetCoordinatesLocal(dm, &Xloc); CHKERRQ(ierr);
4301e150236SLeila Ghaffari   ierr = VecGetLocalSize(Xloc, &lsize); CHKERRQ(ierr);
4311e150236SLeila Ghaffari   ierr = CeedVectorCreate(ceed, lsize, &xcorners); CHKERRQ(ierr);
4321e150236SLeila Ghaffari   ierr = VecGetArray(Xloc, &x); CHKERRQ(ierr);
4331e150236SLeila Ghaffari   CeedVectorSetArray(xcorners, CEED_MEM_HOST, CEED_USE_POINTER, x);
434ca3ac6ddSLeila Ghaffari   ierr = DMGetLabel(dm, "Face Sets", &domainLabel); CHKERRQ(ierr);
4357659d40cSLeila Ghaffari   ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr);
4364438636fSLeila Ghaffari 
437*26aa2eeeSLeila Ghaffari   if (wind_type == ADVECTION_WIND_TRANSLATION
438*26aa2eeeSLeila Ghaffari       || problemChoice == NS_EULER_VORTEX) {
4394438636fSLeila Ghaffari     // Ignore wall and slip BCs
440b6df2e11SLeila Ghaffari     bc->nwall = bc->nslip[0] = bc->nslip[1] = 0;
441b6df2e11SLeila Ghaffari     if (wind_type == ADVECTION_WIND_TRANSLATION) bc->nslip[2] = 0;
4424438636fSLeila Ghaffari 
4434438636fSLeila Ghaffari     // Set number of faces
4447659d40cSLeila Ghaffari     if (dim == 2) nFace = 4;
4457659d40cSLeila Ghaffari     if (dim == 3) nFace = 6;
4469fe13df9SLeila Ghaffari 
4477659d40cSLeila Ghaffari     // Create CEED Operator for each boundary face
4484438636fSLeila Ghaffari     PetscInt localNelemSur[6];
4494438636fSLeila Ghaffari     CeedVector qdataSur[6];
4504438636fSLeila Ghaffari     CeedOperator op_setupSur[6], op_applySur[6];
4514438636fSLeila Ghaffari     CeedElemRestriction restrictxSur[6], restrictqSur[6], restrictqdiSur[6];
45286ba6f09SLeila Ghaffari 
4537659d40cSLeila Ghaffari     for (CeedInt i=0; i<nFace; i++) {
4547659d40cSLeila Ghaffari       ierr = GetRestrictionForDomain(ceed, dm, height, domainLabel, i+1, numP_Sur,
455f259b054Svaleriabarra                                      numQ_Sur, qdatasizeSur, &restrictqSur[i],
456f259b054Svaleriabarra                                      &restrictxSur[i], &restrictqdiSur[i]);
457f259b054Svaleriabarra       CHKERRQ(ierr);
4587659d40cSLeila Ghaffari       // Create the CEED vectors that will be needed in Boundary setup
4597659d40cSLeila Ghaffari       CeedElemRestrictionGetNumElements(restrictqSur[i], &localNelemSur[i]);
460f259b054Svaleriabarra       CeedVectorCreate(ceed, qdatasizeSur*localNelemSur[i]*NqptsSur,
461f259b054Svaleriabarra                        &qdataSur[i]);
4627659d40cSLeila Ghaffari       // Create the operator that builds the quadrature data for the Boundary operator
4637659d40cSLeila Ghaffari       CeedOperatorCreate(ceed, qf_setupSur, NULL, NULL, &op_setupSur[i]);
464ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "dx", restrictxSur[i], basisxSur,
465ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4667659d40cSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "weight", CEED_ELEMRESTRICTION_NONE,
467ca3ac6ddSLeila Ghaffari                            basisxSur, CEED_VECTOR_NONE);
4687659d40cSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "qdataSur", restrictqdiSur[i],
469ca3ac6ddSLeila Ghaffari                            CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
4707659d40cSLeila Ghaffari       // Create Boundary operator
4717659d40cSLeila Ghaffari       CeedOperatorCreate(ceed, qf_applySur, NULL, NULL, &op_applySur[i]);
472ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "q", restrictqSur[i], basisqSur,
473ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4747659d40cSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "qdataSur", restrictqdiSur[i],
4757659d40cSLeila Ghaffari                            CEED_BASIS_COLLOCATED, qdataSur[i]);
476f259b054Svaleriabarra       CeedOperatorSetField(op_applySur[i], "x", restrictxSur[i], basisxSur,
477f259b054Svaleriabarra                            xcorners);
478ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "v", restrictqSur[i], basisqSur,
479ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4807659d40cSLeila Ghaffari       // Apply CEED operator for Boundary setup
481ebb4b9bdSLeila Ghaffari       CeedOperatorApply(op_setupSur[i], xcorners, qdataSur[i],
482ebb4b9bdSLeila Ghaffari                         CEED_REQUEST_IMMEDIATE);
4834438636fSLeila Ghaffari       // --Apply Sub-Operator for the Boundary
4847659d40cSLeila Ghaffari       CeedCompositeOperatorAddSub(*op_apply, op_applySur[i]);
4859fe13df9SLeila Ghaffari     }
4861e150236SLeila Ghaffari     CeedVectorDestroy(&xcorners);
487ca3ac6ddSLeila Ghaffari   }
488ca3ac6ddSLeila Ghaffari   PetscFunctionReturn(0);
489ca3ac6ddSLeila Ghaffari }
490ca3ac6ddSLeila Ghaffari 
491ccaff030SJeremy L Thompson static int CreateVectorFromPetscVec(Ceed ceed, Vec p, CeedVector *v) {
492ccaff030SJeremy L Thompson   PetscErrorCode ierr;
493ccaff030SJeremy L Thompson   PetscInt m;
494ccaff030SJeremy L Thompson 
495ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
496ccaff030SJeremy L Thompson   ierr = VecGetLocalSize(p, &m); CHKERRQ(ierr);
497ccaff030SJeremy L Thompson   ierr = CeedVectorCreate(ceed, m, v); CHKERRQ(ierr);
498ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
499ccaff030SJeremy L Thompson }
500ccaff030SJeremy L Thompson 
501ccaff030SJeremy L Thompson static int VectorPlacePetscVec(CeedVector c, Vec p) {
502ccaff030SJeremy L Thompson   PetscErrorCode ierr;
503ccaff030SJeremy L Thompson   PetscInt mceed, mpetsc;
504ccaff030SJeremy L Thompson   PetscScalar *a;
505ccaff030SJeremy L Thompson 
506ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
507ccaff030SJeremy L Thompson   ierr = CeedVectorGetLength(c, &mceed); CHKERRQ(ierr);
508ccaff030SJeremy L Thompson   ierr = VecGetLocalSize(p, &mpetsc); CHKERRQ(ierr);
509ccaff030SJeremy L Thompson   if (mceed != mpetsc) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP,
51084d34d69SLeila Ghaffari                                   "Cannot place PETSc Vec of length %D in CeedVector of length %D",
51184d34d69SLeila Ghaffari                                   mpetsc, mceed);
512ccaff030SJeremy L Thompson   ierr = VecGetArray(p, &a); CHKERRQ(ierr);
513ccaff030SJeremy L Thompson   CeedVectorSetArray(c, CEED_MEM_HOST, CEED_USE_POINTER, a);
514ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
515ccaff030SJeremy L Thompson }
516ccaff030SJeremy L Thompson 
517ccaff030SJeremy L Thompson static PetscErrorCode DMPlexInsertBoundaryValues_NS(DM dm,
518ccaff030SJeremy L Thompson     PetscBool insertEssential, Vec Qloc, PetscReal time, Vec faceGeomFVM,
519ccaff030SJeremy L Thompson     Vec cellGeomFVM, Vec gradFVM) {
520ccaff030SJeremy L Thompson   PetscErrorCode ierr;
521ccaff030SJeremy L Thompson   Vec Qbc;
522ccaff030SJeremy L Thompson 
523ccaff030SJeremy L Thompson   PetscFunctionBegin;
524ccaff030SJeremy L Thompson   ierr = DMGetNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
525ccaff030SJeremy L Thompson   ierr = VecAXPY(Qloc, 1., Qbc); CHKERRQ(ierr);
526ccaff030SJeremy L Thompson   ierr = DMRestoreNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
527ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
528ccaff030SJeremy L Thompson }
529ccaff030SJeremy L Thompson 
530ccaff030SJeremy L Thompson // This is the RHS of the ODE, given as u_t = G(t,u)
531ccaff030SJeremy L Thompson // This function takes in a state vector Q and writes into G
532ccaff030SJeremy L Thompson static PetscErrorCode RHS_NS(TS ts, PetscReal t, Vec Q, Vec G, void *userData) {
533ccaff030SJeremy L Thompson   PetscErrorCode ierr;
534ccaff030SJeremy L Thompson   User user = *(User *)userData;
535ccaff030SJeremy L Thompson   PetscScalar *q, *g;
536ccaff030SJeremy L Thompson   Vec Qloc, Gloc;
537ccaff030SJeremy L Thompson 
538ccaff030SJeremy L Thompson   // Global-to-local
539ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
5405bf10c6fSLeila Ghaffari   user->ctxEulerData->currentTime = t;
541ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
542ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
543ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
544ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
545ccaff030SJeremy L Thompson   ierr = DMPlexInsertBoundaryValues(user->dm, PETSC_TRUE, Qloc, 0.0,
546ccaff030SJeremy L Thompson                                     NULL, NULL, NULL); CHKERRQ(ierr);
547ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Gloc); CHKERRQ(ierr);
548ccaff030SJeremy L Thompson 
549ccaff030SJeremy L Thompson   // Ceed Vectors
550ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qloc, (const PetscScalar **)&q); CHKERRQ(ierr);
551ccaff030SJeremy L Thompson   ierr = VecGetArray(Gloc, &g); CHKERRQ(ierr);
552ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qceed, CEED_MEM_HOST, CEED_USE_POINTER, q);
553ccaff030SJeremy L Thompson   CeedVectorSetArray(user->gceed, CEED_MEM_HOST, CEED_USE_POINTER, g);
554ccaff030SJeremy L Thompson 
555ccaff030SJeremy L Thompson   // Apply CEED operator
556ccaff030SJeremy L Thompson   CeedOperatorApply(user->op_rhs, user->qceed, user->gceed,
557ccaff030SJeremy L Thompson                     CEED_REQUEST_IMMEDIATE);
558ccaff030SJeremy L Thompson 
559ccaff030SJeremy L Thompson   // Restore vectors
560ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qloc, (const PetscScalar **)&q); CHKERRQ(ierr);
561ccaff030SJeremy L Thompson   ierr = VecRestoreArray(Gloc, &g); CHKERRQ(ierr);
562ccaff030SJeremy L Thompson 
563ccaff030SJeremy L Thompson   ierr = VecZeroEntries(G); CHKERRQ(ierr);
564ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(user->dm, Gloc, ADD_VALUES, G); CHKERRQ(ierr);
565ccaff030SJeremy L Thompson 
566ccaff030SJeremy L Thompson   // Inverse of the lumped mass matrix
567ccaff030SJeremy L Thompson   ierr = VecPointwiseMult(G, G, user->M); // M is Minv
568ccaff030SJeremy L Thompson   CHKERRQ(ierr);
569ccaff030SJeremy L Thompson 
570ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
571ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
572ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
573ccaff030SJeremy L Thompson }
574ccaff030SJeremy L Thompson 
575ccaff030SJeremy L Thompson static PetscErrorCode IFunction_NS(TS ts, PetscReal t, Vec Q, Vec Qdot, Vec G,
576ccaff030SJeremy L Thompson                                    void *userData) {
577ccaff030SJeremy L Thompson   PetscErrorCode ierr;
578ccaff030SJeremy L Thompson   User user = *(User *)userData;
579ccaff030SJeremy L Thompson   const PetscScalar *q, *qdot;
580ccaff030SJeremy L Thompson   PetscScalar *g;
581ccaff030SJeremy L Thompson   Vec Qloc, Qdotloc, Gloc;
582ccaff030SJeremy L Thompson 
583ccaff030SJeremy L Thompson   // Global-to-local
584ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
585ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
586ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qdotloc); CHKERRQ(ierr);
587ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
588ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
589ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
590ccaff030SJeremy L Thompson   ierr = DMPlexInsertBoundaryValues(user->dm, PETSC_TRUE, Qloc, 0.0,
591ccaff030SJeremy L Thompson                                     NULL, NULL, NULL); CHKERRQ(ierr);
592ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qdotloc); CHKERRQ(ierr);
593ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Qdot, INSERT_VALUES, Qdotloc); CHKERRQ(ierr);
594ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Gloc); CHKERRQ(ierr);
595ccaff030SJeremy L Thompson 
596ccaff030SJeremy L Thompson   // Ceed Vectors
597ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qloc, &q); CHKERRQ(ierr);
598ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qdotloc, &qdot); CHKERRQ(ierr);
599ccaff030SJeremy L Thompson   ierr = VecGetArray(Gloc, &g); CHKERRQ(ierr);
600ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qceed, CEED_MEM_HOST, CEED_USE_POINTER,
601ccaff030SJeremy L Thompson                      (PetscScalar *)q);
602ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qdotceed, CEED_MEM_HOST, CEED_USE_POINTER,
603ccaff030SJeremy L Thompson                      (PetscScalar *)qdot);
604ccaff030SJeremy L Thompson   CeedVectorSetArray(user->gceed, CEED_MEM_HOST, CEED_USE_POINTER, g);
605ccaff030SJeremy L Thompson 
606ccaff030SJeremy L Thompson   // Apply CEED operator
607ccaff030SJeremy L Thompson   CeedOperatorApply(user->op_ifunction, user->qceed, user->gceed,
608ccaff030SJeremy L Thompson                     CEED_REQUEST_IMMEDIATE);
609ccaff030SJeremy L Thompson 
610ccaff030SJeremy L Thompson   // Restore vectors
611ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qloc, &q); CHKERRQ(ierr);
612ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qdotloc, &qdot); CHKERRQ(ierr);
613ccaff030SJeremy L Thompson   ierr = VecRestoreArray(Gloc, &g); CHKERRQ(ierr);
614ccaff030SJeremy L Thompson 
615ccaff030SJeremy L Thompson   ierr = VecZeroEntries(G); CHKERRQ(ierr);
616ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(user->dm, Gloc, ADD_VALUES, G); CHKERRQ(ierr);
617ccaff030SJeremy L Thompson 
618ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
619ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qdotloc); CHKERRQ(ierr);
620ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
621ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
622ccaff030SJeremy L Thompson }
623ccaff030SJeremy L Thompson 
624ccaff030SJeremy L Thompson // User provided TS Monitor
625ccaff030SJeremy L Thompson static PetscErrorCode TSMonitor_NS(TS ts, PetscInt stepno, PetscReal time,
626ccaff030SJeremy L Thompson                                    Vec Q, void *ctx) {
627ccaff030SJeremy L Thompson   User user = ctx;
628ccaff030SJeremy L Thompson   Vec Qloc;
629ccaff030SJeremy L Thompson   char filepath[PETSC_MAX_PATH_LEN];
630ccaff030SJeremy L Thompson   PetscViewer viewer;
631ccaff030SJeremy L Thompson   PetscErrorCode ierr;
632ccaff030SJeremy L Thompson 
633ccaff030SJeremy L Thompson   // Set up output
634ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
635ccaff030SJeremy L Thompson   // Print every 'outputfreq' steps
636ccaff030SJeremy L Thompson   if (stepno % user->outputfreq != 0)
637ccaff030SJeremy L Thompson     PetscFunctionReturn(0);
638ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
639ccaff030SJeremy L Thompson   ierr = PetscObjectSetName((PetscObject)Qloc, "StateVec"); CHKERRQ(ierr);
640ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
641ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
642ccaff030SJeremy L Thompson 
643ccaff030SJeremy L Thompson   // Output
644ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-%03D.vtu",
645d99129b9SLeila Ghaffari                        user->outputdir, stepno + user->contsteps);
646ccaff030SJeremy L Thompson   CHKERRQ(ierr);
647ccaff030SJeremy L Thompson   ierr = PetscViewerVTKOpen(PetscObjectComm((PetscObject)Q), filepath,
648ccaff030SJeremy L Thompson                             FILE_MODE_WRITE, &viewer); CHKERRQ(ierr);
649ccaff030SJeremy L Thompson   ierr = VecView(Qloc, viewer); CHKERRQ(ierr);
6509d801c56SJed Brown   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
651ccaff030SJeremy L Thompson   if (user->dmviz) {
652ccaff030SJeremy L Thompson     Vec Qrefined, Qrefined_loc;
653ccaff030SJeremy L Thompson     char filepath_refined[PETSC_MAX_PATH_LEN];
654ccaff030SJeremy L Thompson     PetscViewer viewer_refined;
655ccaff030SJeremy L Thompson 
656ccaff030SJeremy L Thompson     ierr = DMGetGlobalVector(user->dmviz, &Qrefined); CHKERRQ(ierr);
657ccaff030SJeremy L Thompson     ierr = DMGetLocalVector(user->dmviz, &Qrefined_loc); CHKERRQ(ierr);
658ccaff030SJeremy L Thompson     ierr = PetscObjectSetName((PetscObject)Qrefined_loc, "Refined");
659ccaff030SJeremy L Thompson     CHKERRQ(ierr);
660ccaff030SJeremy L Thompson     ierr = MatInterpolate(user->interpviz, Q, Qrefined); CHKERRQ(ierr);
661ccaff030SJeremy L Thompson     ierr = VecZeroEntries(Qrefined_loc); CHKERRQ(ierr);
662ccaff030SJeremy L Thompson     ierr = DMGlobalToLocal(user->dmviz, Qrefined, INSERT_VALUES, Qrefined_loc);
663ccaff030SJeremy L Thompson     CHKERRQ(ierr);
664ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath_refined, sizeof filepath_refined,
665ccaff030SJeremy L Thompson                          "%s/nsrefined-%03D.vtu",
666d99129b9SLeila Ghaffari                          user->outputdir, stepno + user->contsteps);
667ccaff030SJeremy L Thompson     CHKERRQ(ierr);
668ccaff030SJeremy L Thompson     ierr = PetscViewerVTKOpen(PetscObjectComm((PetscObject)Qrefined),
669ccaff030SJeremy L Thompson                               filepath_refined,
670ccaff030SJeremy L Thompson                               FILE_MODE_WRITE, &viewer_refined); CHKERRQ(ierr);
671ccaff030SJeremy L Thompson     ierr = VecView(Qrefined_loc, viewer_refined); CHKERRQ(ierr);
672ccaff030SJeremy L Thompson     ierr = DMRestoreLocalVector(user->dmviz, &Qrefined_loc); CHKERRQ(ierr);
673ccaff030SJeremy L Thompson     ierr = DMRestoreGlobalVector(user->dmviz, &Qrefined); CHKERRQ(ierr);
674ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer_refined); CHKERRQ(ierr);
675ccaff030SJeremy L Thompson   }
676ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
677ccaff030SJeremy L Thompson 
678ccaff030SJeremy L Thompson   // Save data in a binary file for continuation of simulations
679ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-solution.bin",
680d99129b9SLeila Ghaffari                        user->outputdir); CHKERRQ(ierr);
681ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryOpen(user->comm, filepath, FILE_MODE_WRITE, &viewer);
682ccaff030SJeremy L Thompson   CHKERRQ(ierr);
683ccaff030SJeremy L Thompson   ierr = VecView(Q, viewer); CHKERRQ(ierr);
684ccaff030SJeremy L Thompson   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
685ccaff030SJeremy L Thompson 
686ccaff030SJeremy L Thompson   // Save time stamp
687ccaff030SJeremy L Thompson   // Dimensionalize time back
688ccaff030SJeremy L Thompson   time /= user->units->second;
689ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-time.bin",
690d99129b9SLeila Ghaffari                        user->outputdir); CHKERRQ(ierr);
691ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryOpen(user->comm, filepath, FILE_MODE_WRITE, &viewer);
692ccaff030SJeremy L Thompson   CHKERRQ(ierr);
693ccaff030SJeremy L Thompson   #if PETSC_VERSION_GE(3,13,0)
694ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryWrite(viewer, &time, 1, PETSC_REAL);
695ccaff030SJeremy L Thompson   #else
696ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryWrite(viewer, &time, 1, PETSC_REAL, true);
697ccaff030SJeremy L Thompson   #endif
698ccaff030SJeremy L Thompson   CHKERRQ(ierr);
699ccaff030SJeremy L Thompson   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
700ccaff030SJeremy L Thompson 
701ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
702ccaff030SJeremy L Thompson }
703ccaff030SJeremy L Thompson 
70484d34d69SLeila Ghaffari static PetscErrorCode ICs_FixMultiplicity(CeedOperator op_ics,
705ccaff030SJeremy L Thompson     CeedVector xcorners, CeedVector q0ceed, DM dm, Vec Qloc, Vec Q,
706777ff853SJeremy L Thompson     CeedElemRestriction restrictq, CeedQFunctionContext ctxSetup, CeedScalar time) {
707ccaff030SJeremy L Thompson   PetscErrorCode ierr;
708ccaff030SJeremy L Thompson   CeedVector multlvec;
709ccaff030SJeremy L Thompson   Vec Multiplicity, MultiplicityLoc;
710ccaff030SJeremy L Thompson 
711777ff853SJeremy L Thompson   SetupContext ctxSetupData;
712777ff853SJeremy L Thompson   CeedQFunctionContextGetData(ctxSetup, CEED_MEM_HOST, (void **)&ctxSetupData);
713777ff853SJeremy L Thompson   ctxSetupData->time = time;
714777ff853SJeremy L Thompson   CeedQFunctionContextRestoreData(ctxSetup, (void **)&ctxSetupData);
715777ff853SJeremy L Thompson 
716ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
717ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(q0ceed, Qloc); CHKERRQ(ierr);
718ccaff030SJeremy L Thompson   CeedOperatorApply(op_ics, xcorners, q0ceed, CEED_REQUEST_IMMEDIATE);
719ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Q); CHKERRQ(ierr);
720ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, Qloc, ADD_VALUES, Q); CHKERRQ(ierr);
721ccaff030SJeremy L Thompson 
722ccaff030SJeremy L Thompson   // Fix multiplicity for output of ICs
723ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &MultiplicityLoc); CHKERRQ(ierr);
724ccaff030SJeremy L Thompson   CeedElemRestrictionCreateVector(restrictq, &multlvec, NULL);
725ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(multlvec, MultiplicityLoc); CHKERRQ(ierr);
726ccaff030SJeremy L Thompson   CeedElemRestrictionGetMultiplicity(restrictq, multlvec);
727ccaff030SJeremy L Thompson   CeedVectorDestroy(&multlvec);
728ccaff030SJeremy L Thompson   ierr = DMGetGlobalVector(dm, &Multiplicity); CHKERRQ(ierr);
729ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Multiplicity); CHKERRQ(ierr);
730ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, MultiplicityLoc, ADD_VALUES, Multiplicity);
731ccaff030SJeremy L Thompson   CHKERRQ(ierr);
732ccaff030SJeremy L Thompson   ierr = VecPointwiseDivide(Q, Q, Multiplicity); CHKERRQ(ierr);
733ccaff030SJeremy L Thompson   ierr = VecPointwiseDivide(Qloc, Qloc, MultiplicityLoc); CHKERRQ(ierr);
734ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &MultiplicityLoc); CHKERRQ(ierr);
735ccaff030SJeremy L Thompson   ierr = DMRestoreGlobalVector(dm, &Multiplicity); CHKERRQ(ierr);
736ccaff030SJeremy L Thompson 
737ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
738ccaff030SJeremy L Thompson }
739ccaff030SJeremy L Thompson 
740ccaff030SJeremy L Thompson static PetscErrorCode ComputeLumpedMassMatrix(Ceed ceed, DM dm,
741ccaff030SJeremy L Thompson     CeedElemRestriction restrictq, CeedBasis basisq,
742ccaff030SJeremy L Thompson     CeedElemRestriction restrictqdi, CeedVector qdata, Vec M) {
743ccaff030SJeremy L Thompson   PetscErrorCode ierr;
744ccaff030SJeremy L Thompson   CeedQFunction qf_mass;
745ccaff030SJeremy L Thompson   CeedOperator op_mass;
746ccaff030SJeremy L Thompson   CeedVector mceed;
747ccaff030SJeremy L Thompson   Vec Mloc;
748ccaff030SJeremy L Thompson   CeedInt ncompq, qdatasize;
749ccaff030SJeremy L Thompson 
750ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
751ccaff030SJeremy L Thompson   CeedElemRestrictionGetNumComponents(restrictq, &ncompq);
752ccaff030SJeremy L Thompson   CeedElemRestrictionGetNumComponents(restrictqdi, &qdatasize);
753ccaff030SJeremy L Thompson   // Create the Q-function that defines the action of the mass operator
754ccaff030SJeremy L Thompson   CeedQFunctionCreateInterior(ceed, 1, Mass, Mass_loc, &qf_mass);
755ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_mass, "q", ncompq, CEED_EVAL_INTERP);
756ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_mass, "qdata", qdatasize, CEED_EVAL_NONE);
757ccaff030SJeremy L Thompson   CeedQFunctionAddOutput(qf_mass, "v", ncompq, CEED_EVAL_INTERP);
758ccaff030SJeremy L Thompson 
759ccaff030SJeremy L Thompson   // Create the mass operator
760ccaff030SJeremy L Thompson   CeedOperatorCreate(ceed, qf_mass, NULL, NULL, &op_mass);
761ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
762ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "qdata", restrictqdi,
763ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, qdata);
764ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
765ccaff030SJeremy L Thompson 
766ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &Mloc); CHKERRQ(ierr);
767ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Mloc); CHKERRQ(ierr);
768ccaff030SJeremy L Thompson   CeedElemRestrictionCreateVector(restrictq, &mceed, NULL);
769ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(mceed, Mloc); CHKERRQ(ierr);
770ccaff030SJeremy L Thompson 
771ccaff030SJeremy L Thompson   {
772ccaff030SJeremy L Thompson     // Compute a lumped mass matrix
773ccaff030SJeremy L Thompson     CeedVector onesvec;
774ccaff030SJeremy L Thompson     CeedElemRestrictionCreateVector(restrictq, &onesvec, NULL);
775ccaff030SJeremy L Thompson     CeedVectorSetValue(onesvec, 1.0);
776ccaff030SJeremy L Thompson     CeedOperatorApply(op_mass, onesvec, mceed, CEED_REQUEST_IMMEDIATE);
777ccaff030SJeremy L Thompson     CeedVectorDestroy(&onesvec);
778ccaff030SJeremy L Thompson     CeedOperatorDestroy(&op_mass);
779ccaff030SJeremy L Thompson     CeedVectorDestroy(&mceed);
780ccaff030SJeremy L Thompson   }
781ccaff030SJeremy L Thompson   CeedQFunctionDestroy(&qf_mass);
782ccaff030SJeremy L Thompson 
783ccaff030SJeremy L Thompson   ierr = VecZeroEntries(M); CHKERRQ(ierr);
784ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, Mloc, ADD_VALUES, M); CHKERRQ(ierr);
785ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &Mloc); CHKERRQ(ierr);
786ccaff030SJeremy L Thompson 
787ccaff030SJeremy L Thompson   // Invert diagonally lumped mass vector for RHS function
788ccaff030SJeremy L Thompson   ierr = VecReciprocal(M); CHKERRQ(ierr);
789ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
790ccaff030SJeremy L Thompson }
791ccaff030SJeremy L Thompson 
79284d34d69SLeila Ghaffari static PetscErrorCode SetUpDM(DM dm, problemData *problem, PetscInt degree,
7935bf10c6fSLeila Ghaffari                               SimpleBC bc, void *ctxSetupData, void *ctxMMS) {
794ccaff030SJeremy L Thompson   PetscErrorCode ierr;
795ccaff030SJeremy L Thompson 
796ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
797ccaff030SJeremy L Thompson   {
798ccaff030SJeremy L Thompson     // Configure the finite element space and boundary conditions
799ccaff030SJeremy L Thompson     PetscFE fe;
800ccaff030SJeremy L Thompson     PetscInt ncompq = 5;
801ff6701fcSJed Brown     ierr = PetscFECreateLagrange(PETSC_COMM_SELF, problem->dim, ncompq,
802ff6701fcSJed Brown                                  PETSC_FALSE, degree, PETSC_DECIDE,
80332ed2d11SJed Brown                                  &fe); CHKERRQ(ierr);
804ccaff030SJeremy L Thompson     ierr = PetscObjectSetName((PetscObject)fe, "Q"); CHKERRQ(ierr);
805ccaff030SJeremy L Thompson     ierr = DMAddField(dm, NULL,(PetscObject)fe); CHKERRQ(ierr);
806ccaff030SJeremy L Thompson     ierr = DMCreateDS(dm); CHKERRQ(ierr);
80707af6069Svaleriabarra     {
80807af6069Svaleriabarra       PetscInt comps[1] = {1};
80907af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipx", "Face Sets", 0,
8103ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[0],
811777ff853SJeremy L Thompson                            bc->slips[0], ctxSetupData); CHKERRQ(ierr);
81207af6069Svaleriabarra       comps[0] = 2;
81307af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipy", "Face Sets", 0,
8143ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[1],
815777ff853SJeremy L Thompson                            bc->slips[1], ctxSetupData); CHKERRQ(ierr);
81607af6069Svaleriabarra       comps[0] = 3;
81707af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipz", "Face Sets", 0,
8183ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[2],
819777ff853SJeremy L Thompson                            bc->slips[2], ctxSetupData); CHKERRQ(ierr);
82007af6069Svaleriabarra     }
82184d34d69SLeila Ghaffari     if (bc->userbc == PETSC_TRUE) {
82284d34d69SLeila Ghaffari       for (PetscInt c = 0; c < 3; c++) {
82384d34d69SLeila Ghaffari         for (PetscInt s = 0; s < bc->nslip[c]; s++) {
82484d34d69SLeila Ghaffari           for (PetscInt w = 0; w < bc->nwall; w++) {
82584d34d69SLeila Ghaffari             if (bc->slips[c][s] == bc->walls[w])
82684d34d69SLeila Ghaffari               SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG,
827f259b054Svaleriabarra                        "Boundary condition already set on face %D!\n",
828f259b054Svaleriabarra                        bc->walls[w]);
82984d34d69SLeila Ghaffari           }
83084d34d69SLeila Ghaffari         }
83184d34d69SLeila Ghaffari       }
83284d34d69SLeila Ghaffari     }
83384d34d69SLeila Ghaffari     // Wall boundary conditions are zero energy density and zero flux for
83484d34d69SLeila Ghaffari     //   velocity in advection/advection2d, and zero velocity and zero flux
83584d34d69SLeila Ghaffari     //   for mass density and energy density in density_current
83684d34d69SLeila Ghaffari     {
83784d34d69SLeila Ghaffari       if (problem->bc == Exact_Advection || problem->bc == Exact_Advection2d) {
83884d34d69SLeila Ghaffari         PetscInt comps[1] = {4};
83984d34d69SLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", "Face Sets", 0,
8403ab4fca6SValeria Barra                              1, comps, (void(*)(void))problem->bc, NULL,
8410da62991SLeila Ghaffari                              bc->nwall, bc->walls, ctxSetupData); CHKERRQ(ierr);
842753d00aaSLeila Ghaffari       } else if (problem->bc == Exact_DC) {
84384d34d69SLeila Ghaffari         PetscInt comps[3] = {1, 2, 3};
84484d34d69SLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", "Face Sets", 0,
8453ab4fca6SValeria Barra                              3, comps, (void(*)(void))problem->bc, NULL,
846777ff853SJeremy L Thompson                              bc->nwall, bc->walls, ctxSetupData); CHKERRQ(ierr);
847753d00aaSLeila Ghaffari       } else if (problem->bc == Exact_Euler) {
8484a452432SLeila Ghaffari         PetscInt comps[3] = {1, 2, 3};
8494a452432SLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", "Face Sets", 0,
8504a452432SLeila Ghaffari                              3, comps, (void(*)(void))problem->bc, NULL,
8514a452432SLeila Ghaffari                              bc->nwall, bc->walls, ctxMMS); CHKERRQ(ierr);
85284d34d69SLeila Ghaffari       } else
85384d34d69SLeila Ghaffari         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_NULL,
85484d34d69SLeila Ghaffari                 "Undefined boundary conditions for this problem");
85584d34d69SLeila Ghaffari     }
856ccaff030SJeremy L Thompson     ierr = DMPlexSetClosurePermutationTensor(dm, PETSC_DETERMINE, NULL);
857ccaff030SJeremy L Thompson     CHKERRQ(ierr);
858ccaff030SJeremy L Thompson     ierr = PetscFEDestroy(&fe); CHKERRQ(ierr);
859ccaff030SJeremy L Thompson   }
860ccaff030SJeremy L Thompson   {
861ccaff030SJeremy L Thompson     // Empty name for conserved field (because there is only one field)
862ccaff030SJeremy L Thompson     PetscSection section;
863ccaff030SJeremy L Thompson     ierr = DMGetLocalSection(dm, &section); CHKERRQ(ierr);
864ccaff030SJeremy L Thompson     ierr = PetscSectionSetFieldName(section, 0, ""); CHKERRQ(ierr);
865ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 0, "Density");
866ccaff030SJeremy L Thompson     CHKERRQ(ierr);
867ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 1, "MomentumX");
868ccaff030SJeremy L Thompson     CHKERRQ(ierr);
869ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 2, "MomentumY");
870ccaff030SJeremy L Thompson     CHKERRQ(ierr);
871ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 3, "MomentumZ");
872ccaff030SJeremy L Thompson     CHKERRQ(ierr);
873ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 4, "EnergyDensity");
874ccaff030SJeremy L Thompson     CHKERRQ(ierr);
875ccaff030SJeremy L Thompson   }
876ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
877ccaff030SJeremy L Thompson }
878ccaff030SJeremy L Thompson 
879ccaff030SJeremy L Thompson int main(int argc, char **argv) {
880ccaff030SJeremy L Thompson   PetscInt ierr;
881ccaff030SJeremy L Thompson   MPI_Comm comm;
88284d34d69SLeila Ghaffari   DM dm, dmcoord, dmviz;
883ccaff030SJeremy L Thompson   Mat interpviz;
884ccaff030SJeremy L Thompson   TS ts;
885ccaff030SJeremy L Thompson   TSAdapt adapt;
886ccaff030SJeremy L Thompson   User user;
887ccaff030SJeremy L Thompson   Units units;
8885bf10c6fSLeila Ghaffari   EulerContext ctxEulerData;
889ccaff030SJeremy L Thompson   char ceedresource[4096] = "/cpu/self";
89084d34d69SLeila Ghaffari   PetscInt localNelemVol, lnodes, gnodes, steps;
891ccaff030SJeremy L Thompson   const PetscInt ncompq = 5;
892ccaff030SJeremy L Thompson   PetscMPIInt rank;
893ccaff030SJeremy L Thompson   PetscScalar ftime;
894ccaff030SJeremy L Thompson   Vec Q, Qloc, Xloc;
895ccaff030SJeremy L Thompson   Ceed ceed;
89684d34d69SLeila Ghaffari   CeedInt numP, numQ;
897cfa64770SLeila Ghaffari   CeedVector xcorners, qdata, q0ceed;
89884d34d69SLeila Ghaffari   CeedBasis basisx, basisxc, basisq;
89984d34d69SLeila Ghaffari   CeedElemRestriction restrictx, restrictq, restrictqdi;
900cfa64770SLeila Ghaffari   CeedQFunction qf_setupVol, qf_ics, qf_rhsVol, qf_ifunctionVol;
9015bf10c6fSLeila Ghaffari   CeedQFunctionContext ctxSetup, ctxNS, ctxAdvection2d, ctxSurface, ctxEuler;
902cfa64770SLeila Ghaffari   CeedOperator op_setupVol, op_ics;
903ccaff030SJeremy L Thompson   CeedScalar Rd;
90484d34d69SLeila Ghaffari   CeedMemType memtyperequested;
905ccaff030SJeremy L Thompson   PetscScalar WpermK, Pascal, JperkgK, mpersquareds, kgpercubicm,
90616c0476cSLeila Ghaffari               kgpersquaredms, Joulepercubicm, Joule;
907ccaff030SJeremy L Thompson   problemType problemChoice;
908ccaff030SJeremy L Thompson   problemData *problem = NULL;
9091184866aSLeila Ghaffari   WindType wind_type;
910ccaff030SJeremy L Thompson   StabilizationType stab;
91184d34d69SLeila Ghaffari   PetscBool implicit;
912cb3e2689Svaleriabarra   PetscInt    viz_refine = 0;
913ccaff030SJeremy L Thompson   struct SimpleBC_ bc = {
91484d34d69SLeila Ghaffari     .nslip = {2, 2, 2},
91584d34d69SLeila Ghaffari     .slips = {{5, 6}, {3, 4}, {1, 2}}
916ccaff030SJeremy L Thompson   };
917ccaff030SJeremy L Thompson   double start, cpu_time_used;
918dc8efd83SLeila Ghaffari   // Test variables
919dc8efd83SLeila Ghaffari   PetscBool test;
920dc8efd83SLeila Ghaffari   PetscScalar testtol = 0.;
921dc8efd83SLeila Ghaffari   char filepath[PETSC_MAX_PATH_LEN];
92284d34d69SLeila Ghaffari   // Check PETSc CUDA support
92384d34d69SLeila Ghaffari   PetscBool petschavecuda, setmemtyperequest = PETSC_FALSE;
92484d34d69SLeila Ghaffari   // *INDENT-OFF*
92584d34d69SLeila Ghaffari   #ifdef PETSC_HAVE_CUDA
92684d34d69SLeila Ghaffari   petschavecuda = PETSC_TRUE;
92784d34d69SLeila Ghaffari   #else
92884d34d69SLeila Ghaffari   petschavecuda = PETSC_FALSE;
92984d34d69SLeila Ghaffari   #endif
93084d34d69SLeila Ghaffari   // *INDENT-ON*
931ccaff030SJeremy L Thompson 
932ccaff030SJeremy L Thompson   // Create the libCEED contexts
933ccaff030SJeremy L Thompson   PetscScalar meter          = 1e-2;     // 1 meter in scaled length units
934ccaff030SJeremy L Thompson   PetscScalar second         = 1e-2;     // 1 second in scaled time units
935ccaff030SJeremy L Thompson   PetscScalar kilogram       = 1e-6;     // 1 kilogram in scaled mass units
936ccaff030SJeremy L Thompson   PetscScalar Kelvin         = 1;        // 1 Kelvin in scaled temperature units
937ccaff030SJeremy L Thompson   CeedScalar theta0          = 300.;     // K
938ccaff030SJeremy L Thompson   CeedScalar thetaC          = -15.;     // K
939ccaff030SJeremy L Thompson   CeedScalar P0              = 1.e5;     // Pa
94016c0476cSLeila Ghaffari   CeedScalar E_wind          = 1.e6;     // J
941ccaff030SJeremy L Thompson   CeedScalar N               = 0.01;     // 1/s
942ccaff030SJeremy L Thompson   CeedScalar cv              = 717.;     // J/(kg K)
943ccaff030SJeremy L Thompson   CeedScalar cp              = 1004.;    // J/(kg K)
944011314a7SLeila Ghaffari   CeedScalar vortex_strength = 5.;       // -
94535e1ec25SLeila Ghaffari   CeedScalar T_inlet         = 1.;       // K
94635e1ec25SLeila Ghaffari   CeedScalar P_inlet         = 1.;       // Pa
947ccaff030SJeremy L Thompson   CeedScalar g               = 9.81;     // m/s^2
948ccaff030SJeremy L Thompson   CeedScalar lambda          = -2./3.;   // -
949ccaff030SJeremy L Thompson   CeedScalar mu              = 75.;      // Pa s, dynamic viscosity
950ccaff030SJeremy L Thompson   // mu = 75 is not physical for air, but is good for numerical stability
951ccaff030SJeremy L Thompson   CeedScalar k               = 0.02638;  // W/(m K)
952ccaff030SJeremy L Thompson   CeedScalar CtauS           = 0.;       // dimensionless
953ccaff030SJeremy L Thompson   CeedScalar strong_form     = 0.;       // [0,1]
954ccaff030SJeremy L Thompson   PetscScalar lx             = 8000.;    // m
955ccaff030SJeremy L Thompson   PetscScalar ly             = 8000.;    // m
956ccaff030SJeremy L Thompson   PetscScalar lz             = 4000.;    // m
957ccaff030SJeremy L Thompson   CeedScalar rc              = 1000.;    // m (Radius of bubble)
958ccaff030SJeremy L Thompson   PetscScalar resx           = 1000.;    // m (resolution in x)
959ccaff030SJeremy L Thompson   PetscScalar resy           = 1000.;    // m (resolution in y)
960ccaff030SJeremy L Thompson   PetscScalar resz           = 1000.;    // m (resolution in z)
961ccaff030SJeremy L Thompson   PetscInt outputfreq        = 10;       // -
962ccaff030SJeremy L Thompson   PetscInt contsteps         = 0;        // -
96384d34d69SLeila Ghaffari   PetscInt degree            = 1;        // -
96484d34d69SLeila Ghaffari   PetscInt qextra            = 2;        // -
965ea6e0f84SLeila Ghaffari   PetscInt qextraSur         = 2;        // -
966e8107fdaSLeila Ghaffari   PetscReal center[3], dc_axis[3] = {0, 0, 0}, wind[3] = {1., 0, 0},
9676bd16babSLeila Ghaffari                                     etv_mean_velocity[3] = {1., 1., 0};
968ccaff030SJeremy L Thompson   ierr = PetscInitialize(&argc, &argv, NULL, help);
969ccaff030SJeremy L Thompson   if (ierr) return ierr;
970ccaff030SJeremy L Thompson 
971ccaff030SJeremy L Thompson   // Allocate PETSc context
972ccaff030SJeremy L Thompson   ierr = PetscCalloc1(1, &user); CHKERRQ(ierr);
973ccaff030SJeremy L Thompson   ierr = PetscMalloc1(1, &units); CHKERRQ(ierr);
9745bf10c6fSLeila Ghaffari   ierr = PetscMalloc1(1, &ctxEulerData); CHKERRQ(ierr);
975ccaff030SJeremy L Thompson 
976ccaff030SJeremy L Thompson   // Parse command line options
977ccaff030SJeremy L Thompson   comm = PETSC_COMM_WORLD;
978ccaff030SJeremy L Thompson   ierr = PetscOptionsBegin(comm, NULL, "Navier-Stokes in PETSc with libCEED",
979ccaff030SJeremy L Thompson                            NULL); CHKERRQ(ierr);
980ccaff030SJeremy L Thompson   ierr = PetscOptionsString("-ceed", "CEED resource specifier",
981ccaff030SJeremy L Thompson                             NULL, ceedresource, ceedresource,
982ccaff030SJeremy L Thompson                             sizeof(ceedresource), NULL); CHKERRQ(ierr);
983dc8efd83SLeila Ghaffari   ierr = PetscOptionsBool("-test", "Run in test mode",
984dc8efd83SLeila Ghaffari                           NULL, test=PETSC_FALSE, &test, NULL); CHKERRQ(ierr);
985dc8efd83SLeila Ghaffari   ierr = PetscOptionsScalar("-compare_final_state_atol",
986dc8efd83SLeila Ghaffari                             "Test absolute tolerance",
987dc8efd83SLeila Ghaffari                             NULL, testtol, &testtol, NULL); CHKERRQ(ierr);
988dc8efd83SLeila Ghaffari   ierr = PetscOptionsString("-compare_final_state_filename", "Test filename",
989dc8efd83SLeila Ghaffari                             NULL, filepath, filepath,
990dc8efd83SLeila Ghaffari                             sizeof(filepath), NULL); CHKERRQ(ierr);
991ccaff030SJeremy L Thompson   problemChoice = NS_DENSITY_CURRENT;
992ccaff030SJeremy L Thompson   ierr = PetscOptionsEnum("-problem", "Problem to solve", NULL,
993ccaff030SJeremy L Thompson                           problemTypes, (PetscEnum)problemChoice,
994ccaff030SJeremy L Thompson                           (PetscEnum *)&problemChoice, NULL); CHKERRQ(ierr);
995ccaff030SJeremy L Thompson   problem = &problemOptions[problemChoice];
9961184866aSLeila Ghaffari   ierr = PetscOptionsEnum("-problem_advection_wind", "Wind type in Advection",
997f259b054Svaleriabarra                           NULL, WindTypes,
998f259b054Svaleriabarra                           (PetscEnum)(wind_type = ADVECTION_WIND_ROTATION),
9991184866aSLeila Ghaffari                           (PetscEnum *)&wind_type, NULL); CHKERRQ(ierr);
10001184866aSLeila Ghaffari   PetscInt n = problem->dim;
100182c09b01SLeila Ghaffari   PetscBool userWind;
1002ebb4b9bdSLeila Ghaffari   ierr = PetscOptionsRealArray("-problem_advection_wind_translation",
1003ebb4b9bdSLeila Ghaffari                                "Constant wind vector",
100482c09b01SLeila Ghaffari                                NULL, wind, &n, &userWind); CHKERRQ(ierr);
100582c09b01SLeila Ghaffari   if (wind_type == ADVECTION_WIND_ROTATION && userWind) {
1006ebb4b9bdSLeila Ghaffari     ierr = PetscPrintf(comm,
1007ebb4b9bdSLeila Ghaffari                        "Warning! Use -problem_advection_wind_translation only with -problem_advection_wind translation\n");
100882c09b01SLeila Ghaffari     CHKERRQ(ierr);
10091184866aSLeila Ghaffari   }
1010b9c3f7b3SLeila Ghaffari   if (wind_type == ADVECTION_WIND_TRANSLATION
1011b9c3f7b3SLeila Ghaffari       && (problemChoice == NS_DENSITY_CURRENT ||
1012b9c3f7b3SLeila Ghaffari           problemChoice == NS_EULER_VORTEX)) {
101367babd9cSLeila Ghaffari     SETERRQ(comm, PETSC_ERR_ARG_INCOMP,
1014b9c3f7b3SLeila Ghaffari             "-problem_advection_wind translation is not defined for -problem density_current or -problem euler_vortex");
101567babd9cSLeila Ghaffari   }
1016e8107fdaSLeila Ghaffari   ierr = PetscOptionsRealArray("-problem_euler_mean_velocity",
1017e8107fdaSLeila Ghaffari                                "Mean velocity vector",
1018e8107fdaSLeila Ghaffari                                NULL, etv_mean_velocity, &n, NULL);
1019e8107fdaSLeila Ghaffari   CHKERRQ(ierr);
1020ccaff030SJeremy L Thompson   ierr = PetscOptionsEnum("-stab", "Stabilization method", NULL,
1021ccaff030SJeremy L Thompson                           StabilizationTypes, (PetscEnum)(stab = STAB_NONE),
1022ccaff030SJeremy L Thompson                           (PetscEnum *)&stab, NULL); CHKERRQ(ierr);
1023ccaff030SJeremy L Thompson   ierr = PetscOptionsBool("-implicit", "Use implicit (IFunction) formulation",
1024ccaff030SJeremy L Thompson                           NULL, implicit=PETSC_FALSE, &implicit, NULL);
1025ccaff030SJeremy L Thompson   CHKERRQ(ierr);
102684d34d69SLeila Ghaffari   if (!implicit && stab != STAB_NONE) {
102784d34d69SLeila Ghaffari     ierr = PetscPrintf(comm, "Warning! Use -stab only with -implicit\n");
102884d34d69SLeila Ghaffari     CHKERRQ(ierr);
102984d34d69SLeila Ghaffari   }
1030ccaff030SJeremy L Thompson   {
10317573aee6SJed Brown     PetscInt len;
10327573aee6SJed Brown     PetscBool flg;
1033ccaff030SJeremy L Thompson     ierr = PetscOptionsIntArray("-bc_wall",
1034ccaff030SJeremy L Thompson                                 "Use wall boundary conditions on this list of faces",
1035ccaff030SJeremy L Thompson                                 NULL, bc.walls,
1036ccaff030SJeremy L Thompson                                 (len = sizeof(bc.walls) / sizeof(bc.walls[0]),
1037ccaff030SJeremy L Thompson                                  &len), &flg); CHKERRQ(ierr);
10387573aee6SJed Brown     if (flg) {
10397573aee6SJed Brown       bc.nwall = len;
10407573aee6SJed Brown       // Using a no-slip wall disables automatic slip walls (they must be set explicitly)
10417573aee6SJed Brown       bc.nslip[0] = bc.nslip[1] = bc.nslip[2] = 0;
10427573aee6SJed Brown     }
1043ccaff030SJeremy L Thompson     for (PetscInt j=0; j<3; j++) {
1044ccaff030SJeremy L Thompson       const char *flags[3] = {"-bc_slip_x", "-bc_slip_y", "-bc_slip_z"};
1045ccaff030SJeremy L Thompson       ierr = PetscOptionsIntArray(flags[j],
1046ccaff030SJeremy L Thompson                                   "Use slip boundary conditions on this list of faces",
1047ccaff030SJeremy L Thompson                                   NULL, bc.slips[j],
1048ccaff030SJeremy L Thompson                                   (len = sizeof(bc.slips[j]) / sizeof(bc.slips[j][0]),
1049ccaff030SJeremy L Thompson                                    &len), &flg);
1050ccaff030SJeremy L Thompson       CHKERRQ(ierr);
105184d34d69SLeila Ghaffari       if (flg) {
105284d34d69SLeila Ghaffari         bc.nslip[j] = len;
105384d34d69SLeila Ghaffari         bc.userbc = PETSC_TRUE;
105484d34d69SLeila Ghaffari       }
1055ccaff030SJeremy L Thompson     }
1056ccaff030SJeremy L Thompson   }
1057cb3e2689Svaleriabarra   ierr = PetscOptionsInt("-viz_refine",
1058cb3e2689Svaleriabarra                          "Regular refinement levels for visualization",
1059cb3e2689Svaleriabarra                          NULL, viz_refine, &viz_refine, NULL);
1060ccaff030SJeremy L Thompson   CHKERRQ(ierr);
1061ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_meter", "1 meter in scaled length units",
1062ccaff030SJeremy L Thompson                             NULL, meter, &meter, NULL); CHKERRQ(ierr);
1063ccaff030SJeremy L Thompson   meter = fabs(meter);
1064ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_second","1 second in scaled time units",
1065ccaff030SJeremy L Thompson                             NULL, second, &second, NULL); CHKERRQ(ierr);
1066ccaff030SJeremy L Thompson   second = fabs(second);
1067ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_kilogram","1 kilogram in scaled mass units",
1068ccaff030SJeremy L Thompson                             NULL, kilogram, &kilogram, NULL); CHKERRQ(ierr);
1069ccaff030SJeremy L Thompson   kilogram = fabs(kilogram);
1070ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_Kelvin",
1071ccaff030SJeremy L Thompson                             "1 Kelvin in scaled temperature units",
1072ccaff030SJeremy L Thompson                             NULL, Kelvin, &Kelvin, NULL); CHKERRQ(ierr);
1073ccaff030SJeremy L Thompson   Kelvin = fabs(Kelvin);
1074ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-theta0", "Reference potential temperature",
1075ccaff030SJeremy L Thompson                             NULL, theta0, &theta0, NULL); CHKERRQ(ierr);
1076ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-thetaC", "Perturbation of potential temperature",
1077ccaff030SJeremy L Thompson                             NULL, thetaC, &thetaC, NULL); CHKERRQ(ierr);
1078ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-P0", "Atmospheric pressure",
1079ccaff030SJeremy L Thompson                             NULL, P0, &P0, NULL); CHKERRQ(ierr);
108016c0476cSLeila Ghaffari   ierr = PetscOptionsScalar("-E_wind", "Total energy of inflow wind",
108116c0476cSLeila Ghaffari                             NULL, E_wind, &E_wind, NULL); CHKERRQ(ierr);
1082ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-N", "Brunt-Vaisala frequency",
1083ccaff030SJeremy L Thompson                             NULL, N, &N, NULL); CHKERRQ(ierr);
1084ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-cv", "Heat capacity at constant volume",
1085ccaff030SJeremy L Thompson                             NULL, cv, &cv, NULL); CHKERRQ(ierr);
1086ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-cp", "Heat capacity at constant pressure",
1087ccaff030SJeremy L Thompson                             NULL, cp, &cp, NULL); CHKERRQ(ierr);
1088011314a7SLeila Ghaffari   PetscBool userVortex;
1089011314a7SLeila Ghaffari   ierr = PetscOptionsScalar("-vortex_strength", "Strength of Vortex",
1090011314a7SLeila Ghaffari                             NULL, vortex_strength, &vortex_strength, &userVortex);
1091011314a7SLeila Ghaffari   CHKERRQ(ierr);
1092011314a7SLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userVortex) {
1093011314a7SLeila Ghaffari     ierr = PetscPrintf(comm,
1094011314a7SLeila Ghaffari                        "Warning! Use -vortex_strength only with -problem euler_vortex\n");
1095011314a7SLeila Ghaffari     CHKERRQ(ierr);
1096011314a7SLeila Ghaffari   }
1097c8cb9b1fSLeila Ghaffari   PetscBool userTinlet;
1098c8cb9b1fSLeila Ghaffari   ierr = PetscOptionsScalar("-T_inlet", "Incoming Temperature",
1099c8cb9b1fSLeila Ghaffari                             NULL, T_inlet, &T_inlet, &userTinlet);
1100c8cb9b1fSLeila Ghaffari   CHKERRQ(ierr);
1101c8cb9b1fSLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userTinlet) {
1102c8cb9b1fSLeila Ghaffari     ierr = PetscPrintf(comm,
1103c8cb9b1fSLeila Ghaffari                        "Warning! Use -T_inlet only with -problem euler_vortex\n");
1104c8cb9b1fSLeila Ghaffari     CHKERRQ(ierr);
1105c8cb9b1fSLeila Ghaffari   }
1106c8cb9b1fSLeila Ghaffari   PetscBool userPinlet;
1107c8cb9b1fSLeila Ghaffari   ierr = PetscOptionsScalar("-P_inlet", "Incoming Pressure",
1108c8cb9b1fSLeila Ghaffari                             NULL, P_inlet, &P_inlet, &userPinlet);
1109c8cb9b1fSLeila Ghaffari   CHKERRQ(ierr);
1110c8cb9b1fSLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userPinlet) {
1111c8cb9b1fSLeila Ghaffari     ierr = PetscPrintf(comm,
1112c8cb9b1fSLeila Ghaffari                        "Warning! Use -P_inlet only with -problem euler_vortex\n");
1113c8cb9b1fSLeila Ghaffari     CHKERRQ(ierr);
1114c8cb9b1fSLeila Ghaffari   }
1115ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-g", "Gravitational acceleration",
1116ccaff030SJeremy L Thompson                             NULL, g, &g, NULL); CHKERRQ(ierr);
1117ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lambda",
1118ccaff030SJeremy L Thompson                             "Stokes hypothesis second viscosity coefficient",
1119ccaff030SJeremy L Thompson                             NULL, lambda, &lambda, NULL); CHKERRQ(ierr);
1120ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-mu", "Shear dynamic viscosity coefficient",
1121ccaff030SJeremy L Thompson                             NULL, mu, &mu, NULL); CHKERRQ(ierr);
1122ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-k", "Thermal conductivity",
1123ccaff030SJeremy L Thompson                             NULL, k, &k, NULL); CHKERRQ(ierr);
1124ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-CtauS",
1125ccaff030SJeremy L Thompson                             "Scale coefficient for tau (nondimensional)",
1126ccaff030SJeremy L Thompson                             NULL, CtauS, &CtauS, NULL); CHKERRQ(ierr);
112784d34d69SLeila Ghaffari   if (stab == STAB_NONE && CtauS != 0) {
112884d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
112984d34d69SLeila Ghaffari                        "Warning! Use -CtauS only with -stab su or -stab supg\n");
113084d34d69SLeila Ghaffari     CHKERRQ(ierr);
113184d34d69SLeila Ghaffari   }
1132ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-strong_form",
1133ccaff030SJeremy L Thompson                             "Strong (1) or weak/integrated by parts (0) advection residual",
1134ccaff030SJeremy L Thompson                             NULL, strong_form, &strong_form, NULL);
1135ccaff030SJeremy L Thompson   CHKERRQ(ierr);
113684d34d69SLeila Ghaffari   if (problemChoice == NS_DENSITY_CURRENT && (CtauS != 0 || strong_form != 0)) {
113784d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
113884d34d69SLeila Ghaffari                        "Warning! Problem density_current does not support -CtauS or -strong_form\n");
113984d34d69SLeila Ghaffari     CHKERRQ(ierr);
114084d34d69SLeila Ghaffari   }
1141ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lx", "Length scale in x direction",
1142ccaff030SJeremy L Thompson                             NULL, lx, &lx, NULL); CHKERRQ(ierr);
1143ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-ly", "Length scale in y direction",
1144ccaff030SJeremy L Thompson                             NULL, ly, &ly, NULL); CHKERRQ(ierr);
1145ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lz", "Length scale in z direction",
1146ccaff030SJeremy L Thompson                             NULL, lz, &lz, NULL); CHKERRQ(ierr);
1147ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-rc", "Characteristic radius of thermal bubble",
1148ccaff030SJeremy L Thompson                             NULL, rc, &rc, NULL); CHKERRQ(ierr);
1149ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resx","Target resolution in x",
1150ccaff030SJeremy L Thompson                             NULL, resx, &resx, NULL); CHKERRQ(ierr);
1151ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resy","Target resolution in y",
1152ccaff030SJeremy L Thompson                             NULL, resy, &resy, NULL); CHKERRQ(ierr);
1153ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resz","Target resolution in z",
1154ccaff030SJeremy L Thompson                             NULL, resz, &resz, NULL); CHKERRQ(ierr);
115582c09b01SLeila Ghaffari   n = problem->dim;
1156ccaff030SJeremy L Thompson   center[0] = 0.5 * lx;
1157ccaff030SJeremy L Thompson   center[1] = 0.5 * ly;
1158ccaff030SJeremy L Thompson   center[2] = 0.5 * lz;
1159ccaff030SJeremy L Thompson   ierr = PetscOptionsRealArray("-center", "Location of bubble center",
1160ccaff030SJeremy L Thompson                                NULL, center, &n, NULL); CHKERRQ(ierr);
1161ccaff030SJeremy L Thompson   n = problem->dim;
1162ccaff030SJeremy L Thompson   ierr = PetscOptionsRealArray("-dc_axis",
1163ccaff030SJeremy L Thompson                                "Axis of density current cylindrical anomaly, or {0,0,0} for spherically symmetric",
1164ccaff030SJeremy L Thompson                                NULL, dc_axis, &n, NULL); CHKERRQ(ierr);
1165ccaff030SJeremy L Thompson   {
1166ccaff030SJeremy L Thompson     PetscReal norm = PetscSqrtReal(PetscSqr(dc_axis[0]) +
1167ccaff030SJeremy L Thompson                                    PetscSqr(dc_axis[1]) + PetscSqr(dc_axis[2]));
1168ccaff030SJeremy L Thompson     if (norm > 0) {
1169ccaff030SJeremy L Thompson       for (int i=0; i<3; i++) dc_axis[i] /= norm;
1170ccaff030SJeremy L Thompson     }
1171ccaff030SJeremy L Thompson   }
1172ccaff030SJeremy L Thompson   ierr = PetscOptionsInt("-output_freq",
1173ccaff030SJeremy L Thompson                          "Frequency of output, in number of steps",
1174ccaff030SJeremy L Thompson                          NULL, outputfreq, &outputfreq, NULL); CHKERRQ(ierr);
1175ccaff030SJeremy L Thompson   ierr = PetscOptionsInt("-continue", "Continue from previous solution",
1176ccaff030SJeremy L Thompson                          NULL, contsteps, &contsteps, NULL); CHKERRQ(ierr);
117784d34d69SLeila Ghaffari   ierr = PetscOptionsInt("-degree", "Polynomial degree of finite elements",
117884d34d69SLeila Ghaffari                          NULL, degree, &degree, NULL); CHKERRQ(ierr);
117984d34d69SLeila Ghaffari   ierr = PetscOptionsInt("-qextra", "Number of extra quadrature points",
118084d34d69SLeila Ghaffari                          NULL, qextra, &qextra, NULL); CHKERRQ(ierr);
118181f92cf0SLeila Ghaffari   PetscBool userQextraSur;
1182ebb4b9bdSLeila Ghaffari   ierr = PetscOptionsInt("-qextra_boundary",
1183ebb4b9bdSLeila Ghaffari                          "Number of extra quadrature points on in/outflow faces",
1184f259b054Svaleriabarra                          NULL, qextraSur, &qextraSur, &userQextraSur);
1185f259b054Svaleriabarra   CHKERRQ(ierr);
1186ebb4b9bdSLeila Ghaffari   if ((wind_type == ADVECTION_WIND_ROTATION
1187ebb4b9bdSLeila Ghaffari        || problemChoice == NS_DENSITY_CURRENT) && userQextraSur) {
1188ebb4b9bdSLeila Ghaffari     ierr = PetscPrintf(comm,
1189ebb4b9bdSLeila Ghaffari                        "Warning! Use -qextra_boundary only with -problem_advection_wind translation\n");
119081f92cf0SLeila Ghaffari     CHKERRQ(ierr);
119181f92cf0SLeila Ghaffari   }
1192d99129b9SLeila Ghaffari   ierr = PetscStrncpy(user->outputdir, ".", 2); CHKERRQ(ierr);
1193d99129b9SLeila Ghaffari   ierr = PetscOptionsString("-output_dir", "Output directory",
1194d99129b9SLeila Ghaffari                             NULL, user->outputdir, user->outputdir,
1195d99129b9SLeila Ghaffari                             sizeof(user->outputdir), NULL); CHKERRQ(ierr);
119684d34d69SLeila Ghaffari   memtyperequested = petschavecuda ? CEED_MEM_DEVICE : CEED_MEM_HOST;
119784d34d69SLeila Ghaffari   ierr = PetscOptionsEnum("-memtype",
119884d34d69SLeila Ghaffari                           "CEED MemType requested", NULL,
119984d34d69SLeila Ghaffari                           memTypes, (PetscEnum)memtyperequested,
120084d34d69SLeila Ghaffari                           (PetscEnum *)&memtyperequested, &setmemtyperequest);
120184d34d69SLeila Ghaffari   CHKERRQ(ierr);
1202ccaff030SJeremy L Thompson   ierr = PetscOptionsEnd(); CHKERRQ(ierr);
1203ccaff030SJeremy L Thompson 
1204ccaff030SJeremy L Thompson   // Define derived units
1205ccaff030SJeremy L Thompson   Pascal = kilogram / (meter * PetscSqr(second));
1206ccaff030SJeremy L Thompson   JperkgK =  PetscSqr(meter) / (PetscSqr(second) * Kelvin);
1207ccaff030SJeremy L Thompson   mpersquareds = meter / PetscSqr(second);
1208ccaff030SJeremy L Thompson   WpermK = kilogram * meter / (pow(second,3) * Kelvin);
1209ccaff030SJeremy L Thompson   kgpercubicm = kilogram / pow(meter,3);
1210ccaff030SJeremy L Thompson   kgpersquaredms = kilogram / (PetscSqr(meter) * second);
1211ccaff030SJeremy L Thompson   Joulepercubicm = kilogram / (meter * PetscSqr(second));
121216c0476cSLeila Ghaffari   Joule = kilogram * PetscSqr(meter) / PetscSqr(second);
1213ccaff030SJeremy L Thompson 
1214ccaff030SJeremy L Thompson   // Scale variables to desired units
1215ccaff030SJeremy L Thompson   theta0 *= Kelvin;
1216ccaff030SJeremy L Thompson   thetaC *= Kelvin;
1217ccaff030SJeremy L Thompson   P0 *= Pascal;
121816c0476cSLeila Ghaffari   E_wind *= Joule;
1219ccaff030SJeremy L Thompson   N *= (1./second);
1220ccaff030SJeremy L Thompson   cv *= JperkgK;
1221ccaff030SJeremy L Thompson   cp *= JperkgK;
1222ccaff030SJeremy L Thompson   Rd = cp - cv;
1223c8cb9b1fSLeila Ghaffari   T_inlet *= Kelvin;
1224c8cb9b1fSLeila Ghaffari   P_inlet *= Pascal;
1225ccaff030SJeremy L Thompson   g *= mpersquareds;
1226ccaff030SJeremy L Thompson   mu *= Pascal * second;
1227ccaff030SJeremy L Thompson   k *= WpermK;
1228ccaff030SJeremy L Thompson   lx = fabs(lx) * meter;
1229ccaff030SJeremy L Thompson   ly = fabs(ly) * meter;
1230ccaff030SJeremy L Thompson   lz = fabs(lz) * meter;
1231ccaff030SJeremy L Thompson   rc = fabs(rc) * meter;
1232ccaff030SJeremy L Thompson   resx = fabs(resx) * meter;
1233ccaff030SJeremy L Thompson   resy = fabs(resy) * meter;
1234ccaff030SJeremy L Thompson   resz = fabs(resz) * meter;
1235ccaff030SJeremy L Thompson   for (int i=0; i<3; i++) center[i] *= meter;
1236ccaff030SJeremy L Thompson 
1237ccaff030SJeremy L Thompson   const CeedInt dim = problem->dim, ncompx = problem->dim,
1238cfa64770SLeila Ghaffari                 qdatasizeVol = problem->qdatasizeVol;
1239ccaff030SJeremy L Thompson   // Set up the libCEED context
1240777ff853SJeremy L Thompson   struct SetupContext_ ctxSetupData = {
1241ccaff030SJeremy L Thompson     .theta0 = theta0,
1242ccaff030SJeremy L Thompson     .thetaC = thetaC,
1243ccaff030SJeremy L Thompson     .P0 = P0,
1244ccaff030SJeremy L Thompson     .N = N,
1245ccaff030SJeremy L Thompson     .cv = cv,
1246ccaff030SJeremy L Thompson     .cp = cp,
1247ccaff030SJeremy L Thompson     .Rd = Rd,
1248ccaff030SJeremy L Thompson     .g = g,
1249ccaff030SJeremy L Thompson     .rc = rc,
1250ccaff030SJeremy L Thompson     .lx = lx,
1251ccaff030SJeremy L Thompson     .ly = ly,
1252ccaff030SJeremy L Thompson     .lz = lz,
1253ccaff030SJeremy L Thompson     .center[0] = center[0],
1254ccaff030SJeremy L Thompson     .center[1] = center[1],
1255ccaff030SJeremy L Thompson     .center[2] = center[2],
1256ccaff030SJeremy L Thompson     .dc_axis[0] = dc_axis[0],
1257ccaff030SJeremy L Thompson     .dc_axis[1] = dc_axis[1],
1258ccaff030SJeremy L Thompson     .dc_axis[2] = dc_axis[2],
12591184866aSLeila Ghaffari     .wind[0] = wind[0],
12601184866aSLeila Ghaffari     .wind[1] = wind[1],
12611184866aSLeila Ghaffari     .wind[2] = wind[2],
12623a9f53b0SLeila Ghaffari     .time = 0,
1263011314a7SLeila Ghaffari     .vortex_strength = vortex_strength,
12641184866aSLeila Ghaffari     .wind_type = wind_type,
1265ccaff030SJeremy L Thompson   };
1266ccaff030SJeremy L Thompson 
126784d34d69SLeila Ghaffari   // Create the mesh
1268ccaff030SJeremy L Thompson   {
1269ccaff030SJeremy L Thompson     const PetscReal scale[3] = {lx, ly, lz};
1270ccaff030SJeremy L Thompson     ierr = DMPlexCreateBoxMesh(comm, dim, PETSC_FALSE, NULL, NULL, scale,
12712561194eSLeila Ghaffari                                NULL, PETSC_TRUE, &dm);
1272ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1273ccaff030SJeremy L Thompson   }
127484d34d69SLeila Ghaffari 
127584d34d69SLeila Ghaffari   // Distribute the mesh over processes
127684d34d69SLeila Ghaffari   {
1277ccaff030SJeremy L Thompson     DM               dmDist = NULL;
1278ccaff030SJeremy L Thompson     PetscPartitioner part;
1279ccaff030SJeremy L Thompson 
1280ccaff030SJeremy L Thompson     ierr = DMPlexGetPartitioner(dm, &part); CHKERRQ(ierr);
1281ccaff030SJeremy L Thompson     ierr = PetscPartitionerSetFromOptions(part); CHKERRQ(ierr);
1282ccaff030SJeremy L Thompson     ierr = DMPlexDistribute(dm, 0, NULL, &dmDist); CHKERRQ(ierr);
1283ccaff030SJeremy L Thompson     if (dmDist) {
1284ccaff030SJeremy L Thompson       ierr = DMDestroy(&dm); CHKERRQ(ierr);
1285ccaff030SJeremy L Thompson       dm  = dmDist;
1286ccaff030SJeremy L Thompson     }
1287ccaff030SJeremy L Thompson   }
1288ccaff030SJeremy L Thompson   ierr = DMViewFromOptions(dm, NULL, "-dm_view"); CHKERRQ(ierr);
1289ccaff030SJeremy L Thompson 
129084d34d69SLeila Ghaffari   // Setup DM
1291ccaff030SJeremy L Thompson   ierr = DMLocalizeCoordinates(dm); CHKERRQ(ierr);
1292ccaff030SJeremy L Thompson   ierr = DMSetFromOptions(dm); CHKERRQ(ierr);
12935bf10c6fSLeila Ghaffari   ierr = SetUpDM(dm, problem, degree, &bc, &ctxSetupData, ctxEulerData);
12945bf10c6fSLeila Ghaffari   CHKERRQ(ierr);
129584d34d69SLeila Ghaffari 
129684d34d69SLeila Ghaffari   // Refine DM for high-order viz
1297ccaff030SJeremy L Thompson   dmviz = NULL;
1298ccaff030SJeremy L Thompson   interpviz = NULL;
1299ccaff030SJeremy L Thompson   if (viz_refine) {
1300ff6701fcSJed Brown     DM dmhierarchy[viz_refine+1];
1301ff6701fcSJed Brown 
1302ccaff030SJeremy L Thompson     ierr = DMPlexSetRefinementUniform(dm, PETSC_TRUE); CHKERRQ(ierr);
1303ff6701fcSJed Brown     dmhierarchy[0] = dm;
130484d34d69SLeila Ghaffari     for (PetscInt i = 0, d = degree; i < viz_refine; i++) {
1305ff6701fcSJed Brown       Mat interp_next;
1306ff6701fcSJed Brown 
1307ff6701fcSJed Brown       ierr = DMRefine(dmhierarchy[i], MPI_COMM_NULL, &dmhierarchy[i+1]);
1308ccaff030SJeremy L Thompson       CHKERRQ(ierr);
1309bc6a41f7SJed Brown       ierr = DMClearDS(dmhierarchy[i+1]); CHKERRQ(ierr);
1310bc6a41f7SJed Brown       ierr = DMClearFields(dmhierarchy[i+1]); CHKERRQ(ierr);
1311ff6701fcSJed Brown       ierr = DMSetCoarseDM(dmhierarchy[i+1], dmhierarchy[i]); CHKERRQ(ierr);
1312ff6701fcSJed Brown       d = (d + 1) / 2;
1313ff6701fcSJed Brown       if (i + 1 == viz_refine) d = 1;
13145bf10c6fSLeila Ghaffari       ierr = SetUpDM(dmhierarchy[i+1], problem, d, &bc, &ctxSetupData,
13155bf10c6fSLeila Ghaffari                      ctxEulerData); CHKERRQ(ierr);
1316ff6701fcSJed Brown       ierr = DMCreateInterpolation(dmhierarchy[i], dmhierarchy[i+1],
1317ff6701fcSJed Brown                                    &interp_next, NULL); CHKERRQ(ierr);
1318ff6701fcSJed Brown       if (!i) interpviz = interp_next;
1319ff6701fcSJed Brown       else {
1320ff6701fcSJed Brown         Mat C;
1321ff6701fcSJed Brown         ierr = MatMatMult(interp_next, interpviz, MAT_INITIAL_MATRIX,
1322ff6701fcSJed Brown                           PETSC_DECIDE, &C); CHKERRQ(ierr);
1323ff6701fcSJed Brown         ierr = MatDestroy(&interp_next); CHKERRQ(ierr);
1324ff6701fcSJed Brown         ierr = MatDestroy(&interpviz); CHKERRQ(ierr);
1325ff6701fcSJed Brown         interpviz = C;
1326ff6701fcSJed Brown       }
1327ff6701fcSJed Brown     }
1328cb3e2689Svaleriabarra     for (PetscInt i=1; i<viz_refine; i++) {
1329ff6701fcSJed Brown       ierr = DMDestroy(&dmhierarchy[i]); CHKERRQ(ierr);
1330cb3e2689Svaleriabarra     }
1331ff6701fcSJed Brown     dmviz = dmhierarchy[viz_refine];
1332ccaff030SJeremy L Thompson   }
1333ccaff030SJeremy L Thompson   ierr = DMCreateGlobalVector(dm, &Q); CHKERRQ(ierr);
1334ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &Qloc); CHKERRQ(ierr);
1335ccaff030SJeremy L Thompson   ierr = VecGetSize(Qloc, &lnodes); CHKERRQ(ierr);
1336ccaff030SJeremy L Thompson   lnodes /= ncompq;
1337ccaff030SJeremy L Thompson 
133884d34d69SLeila Ghaffari   // Initialize CEED
133984d34d69SLeila Ghaffari   CeedInit(ceedresource, &ceed);
134084d34d69SLeila Ghaffari   // Set memtype
134184d34d69SLeila Ghaffari   CeedMemType memtypebackend;
134284d34d69SLeila Ghaffari   CeedGetPreferredMemType(ceed, &memtypebackend);
134384d34d69SLeila Ghaffari   // Check memtype compatibility
134484d34d69SLeila Ghaffari   if (!setmemtyperequest)
134584d34d69SLeila Ghaffari     memtyperequested = memtypebackend;
134684d34d69SLeila Ghaffari   else if (!petschavecuda && memtyperequested == CEED_MEM_DEVICE)
134784d34d69SLeila Ghaffari     SETERRQ1(PETSC_COMM_WORLD, PETSC_ERR_SUP_SYS,
134884d34d69SLeila Ghaffari              "PETSc was not built with CUDA. "
134984d34d69SLeila Ghaffari              "Requested MemType CEED_MEM_DEVICE is not supported.", NULL);
135084d34d69SLeila Ghaffari 
135184d34d69SLeila Ghaffari   // Set number of 1D nodes and quadrature points
135284d34d69SLeila Ghaffari   numP = degree + 1;
135384d34d69SLeila Ghaffari   numQ = numP + qextra;
135484d34d69SLeila Ghaffari 
135584d34d69SLeila Ghaffari   // Print summary
1356dc8efd83SLeila Ghaffari   if (!test) {
1357ccaff030SJeremy L Thompson     CeedInt gdofs, odofs;
1358ccaff030SJeremy L Thompson     int comm_size;
1359ccaff030SJeremy L Thompson     char box_faces_str[PETSC_MAX_PATH_LEN] = "NONE";
1360ccaff030SJeremy L Thompson     ierr = VecGetSize(Q, &gdofs); CHKERRQ(ierr);
1361ccaff030SJeremy L Thompson     ierr = VecGetLocalSize(Q, &odofs); CHKERRQ(ierr);
136284d34d69SLeila Ghaffari     gnodes = gdofs/ncompq;
1363ccaff030SJeremy L Thompson     ierr = MPI_Comm_size(comm, &comm_size); CHKERRQ(ierr);
1364ccaff030SJeremy L Thompson     ierr = PetscOptionsGetString(NULL, NULL, "-dm_plex_box_faces", box_faces_str,
1365ccaff030SJeremy L Thompson                                  sizeof(box_faces_str), NULL); CHKERRQ(ierr);
136684d34d69SLeila Ghaffari     const char *usedresource;
136784d34d69SLeila Ghaffari     CeedGetResource(ceed, &usedresource);
1368ccaff030SJeremy L Thompson 
136984d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
137084d34d69SLeila Ghaffari                        "\n-- Navier-Stokes solver - libCEED + PETSc --\n"
137184d34d69SLeila Ghaffari                        "  rank(s)                              : %d\n"
137284d34d69SLeila Ghaffari                        "  Problem:\n"
137384d34d69SLeila Ghaffari                        "    Problem Name                       : %s\n"
137484d34d69SLeila Ghaffari                        "    Stabilization                      : %s\n"
137584d34d69SLeila Ghaffari                        "  PETSc:\n"
137684d34d69SLeila Ghaffari                        "    Box Faces                          : %s\n"
137784d34d69SLeila Ghaffari                        "  libCEED:\n"
137884d34d69SLeila Ghaffari                        "    libCEED Backend                    : %s\n"
137984d34d69SLeila Ghaffari                        "    libCEED Backend MemType            : %s\n"
138084d34d69SLeila Ghaffari                        "    libCEED User Requested MemType     : %s\n"
138184d34d69SLeila Ghaffari                        "  Mesh:\n"
138284d34d69SLeila Ghaffari                        "    Number of 1D Basis Nodes (P)       : %d\n"
138384d34d69SLeila Ghaffari                        "    Number of 1D Quadrature Points (Q) : %d\n"
138484d34d69SLeila Ghaffari                        "    Global DoFs                        : %D\n"
138584d34d69SLeila Ghaffari                        "    Owned DoFs                         : %D\n"
138684d34d69SLeila Ghaffari                        "    DoFs per node                      : %D\n"
138784d34d69SLeila Ghaffari                        "    Global nodes                       : %D\n"
138884d34d69SLeila Ghaffari                        "    Owned nodes                        : %D\n",
138984d34d69SLeila Ghaffari                        comm_size, problemTypes[problemChoice],
139084d34d69SLeila Ghaffari                        StabilizationTypes[stab], box_faces_str, usedresource,
139184d34d69SLeila Ghaffari                        CeedMemTypes[memtypebackend],
139284d34d69SLeila Ghaffari                        (setmemtyperequest) ?
139384d34d69SLeila Ghaffari                        CeedMemTypes[memtyperequested] : "none",
139484d34d69SLeila Ghaffari                        numP, numQ, gdofs, odofs, ncompq, gnodes, lnodes);
139584d34d69SLeila Ghaffari     CHKERRQ(ierr);
13960c6c0b13SLeila Ghaffari   }
13970c6c0b13SLeila Ghaffari 
1398ccaff030SJeremy L Thompson   // Set up global mass vector
1399ccaff030SJeremy L Thompson   ierr = VecDuplicate(Q, &user->M); CHKERRQ(ierr);
1400ccaff030SJeremy L Thompson 
140184d34d69SLeila Ghaffari   // Set up libCEED
1402ccaff030SJeremy L Thompson   // CEED Bases
1403ccaff030SJeremy L Thompson   CeedInit(ceedresource, &ceed);
140484d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompq, numP, numQ, CEED_GAUSS,
140584d34d69SLeila Ghaffari                                   &basisq);
140684d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompx, 2, numQ, CEED_GAUSS,
140784d34d69SLeila Ghaffari                                   &basisx);
140884d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompx, 2, numP,
140984d34d69SLeila Ghaffari                                   CEED_GAUSS_LOBATTO, &basisxc);
1410ccaff030SJeremy L Thompson   ierr = DMGetCoordinateDM(dm, &dmcoord); CHKERRQ(ierr);
1411ccaff030SJeremy L Thompson   ierr = DMPlexSetClosurePermutationTensor(dmcoord, PETSC_DETERMINE, NULL);
1412ccaff030SJeremy L Thompson   CHKERRQ(ierr);
1413ccaff030SJeremy L Thompson 
1414ccaff030SJeremy L Thompson   // CEED Restrictions
14151e150236SLeila Ghaffari   ierr = GetRestrictionForDomain(ceed, dm, 0, 0, 0, numP, numQ,
141684d34d69SLeila Ghaffari                                  qdatasizeVol, &restrictq, &restrictx,
141784d34d69SLeila Ghaffari                                  &restrictqdi); CHKERRQ(ierr);
1418ccaff030SJeremy L Thompson 
1419ccaff030SJeremy L Thompson   ierr = DMGetCoordinatesLocal(dm, &Xloc); CHKERRQ(ierr);
1420ccaff030SJeremy L Thompson   ierr = CreateVectorFromPetscVec(ceed, Xloc, &xcorners); CHKERRQ(ierr);
1421ccaff030SJeremy L Thompson 
1422ccaff030SJeremy L Thompson   // Create the CEED vectors that will be needed in setup
1423bd910870SLeila Ghaffari   CeedInt NqptsVol;
142484d34d69SLeila Ghaffari   CeedBasisGetNumQuadraturePoints(basisq, &NqptsVol);
142584d34d69SLeila Ghaffari   CeedElemRestrictionGetNumElements(restrictq, &localNelemVol);
14268b982baeSLeila Ghaffari   CeedVectorCreate(ceed, qdatasizeVol*localNelemVol*NqptsVol, &qdata);
142784d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &q0ceed, NULL);
1428ccaff030SJeremy L Thompson 
1429ccaff030SJeremy L Thompson   // Create the Q-function that builds the quadrature data for the NS operator
1430ea6e0f84SLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setupVol, problem->setupVol_loc,
1431ea6e0f84SLeila Ghaffari                               &qf_setupVol);
1432ea6e0f84SLeila Ghaffari   CeedQFunctionAddInput(qf_setupVol, "dx", ncompx*dim, CEED_EVAL_GRAD);
1433ea6e0f84SLeila Ghaffari   CeedQFunctionAddInput(qf_setupVol, "weight", 1, CEED_EVAL_WEIGHT);
14348b982baeSLeila Ghaffari   CeedQFunctionAddOutput(qf_setupVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1435ccaff030SJeremy L Thompson 
1436ccaff030SJeremy L Thompson   // Create the Q-function that sets the ICs of the operator
1437ccaff030SJeremy L Thompson   CeedQFunctionCreateInterior(ceed, 1, problem->ics, problem->ics_loc, &qf_ics);
1438ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_ics, "x", ncompx, CEED_EVAL_INTERP);
1439ccaff030SJeremy L Thompson   CeedQFunctionAddOutput(qf_ics, "q0", ncompq, CEED_EVAL_NONE);
1440ccaff030SJeremy L Thompson 
1441ea6e0f84SLeila Ghaffari   qf_rhsVol = NULL;
1442ea6e0f84SLeila Ghaffari   if (problem->applyVol_rhs) { // Create the Q-function that defines the action of the RHS operator
1443ea6e0f84SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applyVol_rhs,
1444ea6e0f84SLeila Ghaffari                                 problem->applyVol_rhs_loc, &qf_rhsVol);
1445ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "q", ncompq, CEED_EVAL_INTERP);
1446ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "dq", ncompq*dim, CEED_EVAL_GRAD);
14478b982baeSLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1448ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "x", ncompx, CEED_EVAL_INTERP);
1449ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_rhsVol, "v", ncompq, CEED_EVAL_INTERP);
1450ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_rhsVol, "dv", ncompq*dim, CEED_EVAL_GRAD);
1451ccaff030SJeremy L Thompson   }
1452ccaff030SJeremy L Thompson 
1453ea6e0f84SLeila Ghaffari   qf_ifunctionVol = NULL;
1454ea6e0f84SLeila Ghaffari   if (problem->applyVol_ifunction) { // Create the Q-function that defines the action of the IFunction
1455ea6e0f84SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applyVol_ifunction,
1456ea6e0f84SLeila Ghaffari                                 problem->applyVol_ifunction_loc, &qf_ifunctionVol);
1457ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "q", ncompq, CEED_EVAL_INTERP);
1458ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "dq", ncompq*dim, CEED_EVAL_GRAD);
1459ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "qdot", ncompq, CEED_EVAL_INTERP);
14608b982baeSLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1461ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "x", ncompx, CEED_EVAL_INTERP);
1462ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_ifunctionVol, "v", ncompq, CEED_EVAL_INTERP);
1463ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_ifunctionVol, "dv", ncompq*dim, CEED_EVAL_GRAD);
1464ccaff030SJeremy L Thompson   }
1465ccaff030SJeremy L Thompson 
1466ccaff030SJeremy L Thompson   // Create the operator that builds the quadrature data for the NS operator
1467ea6e0f84SLeila Ghaffari   CeedOperatorCreate(ceed, qf_setupVol, NULL, NULL, &op_setupVol);
146884d34d69SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "dx", restrictx, basisx, CEED_VECTOR_ACTIVE);
1469ea6e0f84SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "weight", CEED_ELEMRESTRICTION_NONE,
147084d34d69SLeila Ghaffari                        basisx, CEED_VECTOR_NONE);
147184d34d69SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "qdata", restrictqdi,
1472ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
1473ccaff030SJeremy L Thompson 
1474ccaff030SJeremy L Thompson   // Create the operator that sets the ICs
1475ccaff030SJeremy L Thompson   CeedOperatorCreate(ceed, qf_ics, NULL, NULL, &op_ics);
147684d34d69SLeila Ghaffari   CeedOperatorSetField(op_ics, "x", restrictx, basisxc, CEED_VECTOR_ACTIVE);
147784d34d69SLeila Ghaffari   CeedOperatorSetField(op_ics, "q0", restrictq,
1478ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
1479ccaff030SJeremy L Thompson 
148084d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->qceed, NULL);
148184d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->qdotceed, NULL);
148284d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->gceed, NULL);
1483ccaff030SJeremy L Thompson 
1484ea6e0f84SLeila Ghaffari   if (qf_rhsVol) { // Create the RHS physics operator
1485ccaff030SJeremy L Thompson     CeedOperator op;
1486ea6e0f84SLeila Ghaffari     CeedOperatorCreate(ceed, qf_rhsVol, NULL, NULL, &op);
148784d34d69SLeila Ghaffari     CeedOperatorSetField(op, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
148884d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dq", restrictq, basisq, CEED_VECTOR_ACTIVE);
148984d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdata", restrictqdi,
14908b982baeSLeila Ghaffari                          CEED_BASIS_COLLOCATED, qdata);
149184d34d69SLeila Ghaffari     CeedOperatorSetField(op, "x", restrictx, basisx, xcorners);
149284d34d69SLeila Ghaffari     CeedOperatorSetField(op, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
149384d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dv", restrictq, basisq, CEED_VECTOR_ACTIVE);
1494d3630711SJed Brown     user->op_rhs_vol = op;
1495ccaff030SJeremy L Thompson   }
1496ccaff030SJeremy L Thompson 
1497ea6e0f84SLeila Ghaffari   if (qf_ifunctionVol) { // Create the IFunction operator
1498ccaff030SJeremy L Thompson     CeedOperator op;
1499ea6e0f84SLeila Ghaffari     CeedOperatorCreate(ceed, qf_ifunctionVol, NULL, NULL, &op);
150084d34d69SLeila Ghaffari     CeedOperatorSetField(op, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
150184d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dq", restrictq, basisq, CEED_VECTOR_ACTIVE);
150284d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdot", restrictq, basisq, user->qdotceed);
150384d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdata", restrictqdi,
15048b982baeSLeila Ghaffari                          CEED_BASIS_COLLOCATED, qdata);
150584d34d69SLeila Ghaffari     CeedOperatorSetField(op, "x", restrictx, basisx, xcorners);
150684d34d69SLeila Ghaffari     CeedOperatorSetField(op, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
150784d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dv", restrictq, basisq, CEED_VECTOR_ACTIVE);
1508d3630711SJed Brown     user->op_ifunction_vol = op;
1509ccaff030SJeremy L Thompson   }
1510ccaff030SJeremy L Thompson 
15116a0edaf9SLeila Ghaffari   // Set up CEED for the boundaries
15126a0edaf9SLeila Ghaffari   CeedInt height = 1;
15136a0edaf9SLeila Ghaffari   CeedInt dimSur = dim - height;
15141e150236SLeila Ghaffari   CeedInt numP_Sur = degree + 1;
15151e150236SLeila Ghaffari   CeedInt numQ_Sur = numP_Sur + qextraSur;
1516cfa64770SLeila Ghaffari   const CeedInt qdatasizeSur = problem->qdatasizeSur;
1517cfa64770SLeila Ghaffari   CeedBasis basisxSur, basisxcSur, basisqSur;
1518cfa64770SLeila Ghaffari   CeedInt NqptsSur;
1519e2043960SLeila Ghaffari   CeedQFunction qf_setupSur, qf_applySur;
1520cfa64770SLeila Ghaffari 
1521cfa64770SLeila Ghaffari   // CEED bases for the boundaries
1522ebb4b9bdSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompq, numP_Sur, numQ_Sur,
1523ebb4b9bdSLeila Ghaffari                                   CEED_GAUSS,
15246a0edaf9SLeila Ghaffari                                   &basisqSur);
15256a0edaf9SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompx, 2, numQ_Sur, CEED_GAUSS,
15266a0edaf9SLeila Ghaffari                                   &basisxSur);
15276a0edaf9SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompx, 2, numP_Sur,
15286a0edaf9SLeila Ghaffari                                   CEED_GAUSS_LOBATTO, &basisxcSur);
15296a0edaf9SLeila Ghaffari   CeedBasisGetNumQuadraturePoints(basisqSur, &NqptsSur);
15306a0edaf9SLeila Ghaffari 
1531cfa64770SLeila Ghaffari   // Create the Q-function that builds the quadrature data for the Surface operator
15326a0edaf9SLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setupSur, problem->setupSur_loc,
15336a0edaf9SLeila Ghaffari                               &qf_setupSur);
15346a0edaf9SLeila Ghaffari   CeedQFunctionAddInput(qf_setupSur, "dx", ncompx*dimSur, CEED_EVAL_GRAD);
15356a0edaf9SLeila Ghaffari   CeedQFunctionAddInput(qf_setupSur, "weight", 1, CEED_EVAL_WEIGHT);
15366a0edaf9SLeila Ghaffari   CeedQFunctionAddOutput(qf_setupSur, "qdataSur", qdatasizeSur, CEED_EVAL_NONE);
15376a0edaf9SLeila Ghaffari 
15387659d40cSLeila Ghaffari   // Creat Q-Function for Boundaries
153986ba6f09SLeila Ghaffari   // -- Defined for Advection(2d) test cases
15407ed8b9c7SLeila Ghaffari   qf_applySur = NULL;
15417659d40cSLeila Ghaffari   if (problem->applySur) {
15427659d40cSLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applySur,
15437ed8b9c7SLeila Ghaffari                                 problem->applySur_loc, &qf_applySur);
15447ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "q", ncompq, CEED_EVAL_INTERP);
15457ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "qdataSur", qdatasizeSur, CEED_EVAL_NONE);
15467ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "x", ncompx, CEED_EVAL_INTERP);
15477ed8b9c7SLeila Ghaffari     CeedQFunctionAddOutput(qf_applySur, "v", ncompq, CEED_EVAL_INTERP);
15489fe13df9SLeila Ghaffari   }
15499fe13df9SLeila Ghaffari 
15509fe13df9SLeila Ghaffari   // Create CEED Operator for the whole domain
15519fe13df9SLeila Ghaffari   if (!implicit)
15524438636fSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, &bc, problemChoice, wind_type,
15534438636fSLeila Ghaffari                                    user->op_rhs_vol, qf_applySur,
1554e2043960SLeila Ghaffari                                    qf_setupSur, height, numP_Sur, numQ_Sur,
15554438636fSLeila Ghaffari                                    qdatasizeSur, NqptsSur, basisxSur,
15564438636fSLeila Ghaffari                                    basisqSur, &user->op_rhs);
15574438636fSLeila Ghaffari   CHKERRQ(ierr);
15589fe13df9SLeila Ghaffari   if (implicit)
15594438636fSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, &bc, problemChoice, wind_type,
15604438636fSLeila Ghaffari                                    user->op_ifunction_vol, qf_applySur,
1561e2043960SLeila Ghaffari                                    qf_setupSur, height, numP_Sur, numQ_Sur,
15624438636fSLeila Ghaffari                                    qdatasizeSur, NqptsSur, basisxSur,
15634438636fSLeila Ghaffari                                    basisqSur, &user->op_ifunction);
15644438636fSLeila Ghaffari   CHKERRQ(ierr);
15657659d40cSLeila Ghaffari   // Set up contex for QFunctions
1566777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxSetup);
1567777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxSetup, CEED_MEM_HOST, CEED_USE_POINTER,
1568777ff853SJeremy L Thompson                               sizeof ctxSetupData, &ctxSetupData);
15695bf10c6fSLeila Ghaffari   if (qf_ics && problemChoice != NS_EULER_VORTEX)
1570777ff853SJeremy L Thompson     CeedQFunctionSetContext(qf_ics, ctxSetup);
1571777ff853SJeremy L Thompson 
15725bf10c6fSLeila Ghaffari   CeedScalar ctxNSData[8] = {lambda, mu, k, cv, cp, g, Rd};
1573777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxNS);
1574777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxNS, CEED_MEM_HOST, CEED_USE_POINTER,
1575777ff853SJeremy L Thompson                               sizeof ctxNSData, &ctxNSData);
1576777ff853SJeremy L Thompson 
15770da62991SLeila Ghaffari   struct Advection2dContext_ ctxAdvection2dData = {
1578ccaff030SJeremy L Thompson     .CtauS = CtauS,
1579ccaff030SJeremy L Thompson     .strong_form = strong_form,
1580ccaff030SJeremy L Thompson     .stabilization = stab,
1581ccaff030SJeremy L Thompson   };
1582777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxAdvection2d);
1583777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxAdvection2d, CEED_MEM_HOST, CEED_USE_POINTER,
1584777ff853SJeremy L Thompson                               sizeof ctxAdvection2dData, &ctxAdvection2dData);
1585777ff853SJeremy L Thompson 
1586777ff853SJeremy L Thompson   struct SurfaceContext_ ctxSurfaceData = {
158716c0476cSLeila Ghaffari     .E_wind = E_wind,
15887659d40cSLeila Ghaffari     .strong_form = strong_form,
1589c8cb9b1fSLeila Ghaffari     .T_inlet = T_inlet,
1590c8cb9b1fSLeila Ghaffari     .P_inlet = P_inlet,
1591c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[0] = etv_mean_velocity[0],
1592c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[1] = etv_mean_velocity[1],
1593c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[2] = etv_mean_velocity[2],
1594e5ed8c30SLeila Ghaffari     .implicit = implicit,
1595e5ed8c30SLeila Ghaffari   };
1596777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxSurface);
1597777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxSurface, CEED_MEM_HOST, CEED_USE_POINTER,
1598777ff853SJeremy L Thompson                               sizeof ctxSurfaceData, &ctxSurfaceData);
1599777ff853SJeremy L Thompson 
16005bf10c6fSLeila Ghaffari   // Set up ctxEulerData structure
16015bf10c6fSLeila Ghaffari   ctxEulerData->time = 0.;
16025bf10c6fSLeila Ghaffari   ctxEulerData->currentTime = 0.;
16035bf10c6fSLeila Ghaffari   ctxEulerData->center[0] = center[0];
16045bf10c6fSLeila Ghaffari   ctxEulerData->center[1] = center[1];
16055bf10c6fSLeila Ghaffari   ctxEulerData->center[2] = center[2];
16065bf10c6fSLeila Ghaffari   ctxEulerData->vortex_strength = vortex_strength;
16070da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[0] = etv_mean_velocity[0];
16080da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[1] = etv_mean_velocity[1];
16090da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[2] = etv_mean_velocity[2];
16105bf10c6fSLeila Ghaffari   user->ctxEulerData = ctxEulerData;
16115bf10c6fSLeila Ghaffari 
16125bf10c6fSLeila Ghaffari   CeedQFunctionContextCreate(ceed, &ctxEuler);
16135bf10c6fSLeila Ghaffari   CeedQFunctionContextSetData(ctxEuler, CEED_MEM_HOST, CEED_USE_POINTER,
16140da62991SLeila Ghaffari                               sizeof *ctxEulerData, ctxEulerData);
16155bf10c6fSLeila Ghaffari 
1616ccaff030SJeremy L Thompson   switch (problemChoice) {
1617ccaff030SJeremy L Thompson   case NS_DENSITY_CURRENT:
16185bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxNS);
16195bf10c6fSLeila Ghaffari     if (qf_ifunctionVol) CeedQFunctionSetContext(qf_ifunctionVol, ctxNS);
1620ccaff030SJeremy L Thompson     break;
1621ccaff030SJeremy L Thompson   case NS_ADVECTION:
1622ccaff030SJeremy L Thompson   case NS_ADVECTION2D:
16235bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxAdvection2d);
16245bf10c6fSLeila Ghaffari     if (qf_ifunctionVol) CeedQFunctionSetContext(qf_ifunctionVol, ctxAdvection2d);
16255bf10c6fSLeila Ghaffari     if (qf_applySur) CeedQFunctionSetContext(qf_applySur, ctxSurface);
1626e5ed8c30SLeila Ghaffari   case NS_EULER_VORTEX:
16270da62991SLeila Ghaffari     if (qf_ics) CeedQFunctionSetContext(qf_ics, ctxEuler);
16285bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxEuler);
1629c8cb9b1fSLeila Ghaffari     if (qf_applySur) CeedQFunctionSetContext(qf_applySur, ctxSurface);
1630ccaff030SJeremy L Thompson   }
1631ccaff030SJeremy L Thompson 
1632ccaff030SJeremy L Thompson   // Set up PETSc context
1633ccaff030SJeremy L Thompson   // Set up units structure
1634ccaff030SJeremy L Thompson   units->meter = meter;
1635ccaff030SJeremy L Thompson   units->kilogram = kilogram;
1636ccaff030SJeremy L Thompson   units->second = second;
1637ccaff030SJeremy L Thompson   units->Kelvin = Kelvin;
1638ccaff030SJeremy L Thompson   units->Pascal = Pascal;
1639ccaff030SJeremy L Thompson   units->JperkgK = JperkgK;
1640ccaff030SJeremy L Thompson   units->mpersquareds = mpersquareds;
1641ccaff030SJeremy L Thompson   units->WpermK = WpermK;
1642ccaff030SJeremy L Thompson   units->kgpercubicm = kgpercubicm;
1643ccaff030SJeremy L Thompson   units->kgpersquaredms = kgpersquaredms;
1644ccaff030SJeremy L Thompson   units->Joulepercubicm = Joulepercubicm;
164516c0476cSLeila Ghaffari   units->Joule = Joule;
1646ccaff030SJeremy L Thompson 
1647ccaff030SJeremy L Thompson   // Set up user structure
1648ccaff030SJeremy L Thompson   user->comm = comm;
1649ccaff030SJeremy L Thompson   user->outputfreq = outputfreq;
1650ccaff030SJeremy L Thompson   user->contsteps = contsteps;
1651ccaff030SJeremy L Thompson   user->units = units;
1652ccaff030SJeremy L Thompson   user->dm = dm;
1653ccaff030SJeremy L Thompson   user->dmviz = dmviz;
1654ccaff030SJeremy L Thompson   user->interpviz = interpviz;
1655ccaff030SJeremy L Thompson   user->ceed = ceed;
1656ccaff030SJeremy L Thompson 
16578b982baeSLeila Ghaffari   // Calculate qdata and ICs
1658ccaff030SJeremy L Thompson   // Set up state global and local vectors
1659ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Q); CHKERRQ(ierr);
1660ccaff030SJeremy L Thompson 
1661cfa64770SLeila Ghaffari   ierr = VectorPlacePetscVec(q0ceed, Qloc); CHKERRQ(ierr);
1662ccaff030SJeremy L Thompson 
1663ccaff030SJeremy L Thompson   // Apply Setup Ceed Operators
1664ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(xcorners, Xloc); CHKERRQ(ierr);
16658b982baeSLeila Ghaffari   CeedOperatorApply(op_setupVol, xcorners, qdata, CEED_REQUEST_IMMEDIATE);
166684d34d69SLeila Ghaffari   ierr = ComputeLumpedMassMatrix(ceed, dm, restrictq, basisq, restrictqdi, qdata,
1667ccaff030SJeremy L Thompson                                  user->M); CHKERRQ(ierr);
1668ccaff030SJeremy L Thompson 
166984d34d69SLeila Ghaffari   ierr = ICs_FixMultiplicity(op_ics, xcorners, q0ceed, dm, Qloc, Q, restrictq,
1670777ff853SJeremy L Thompson                              ctxSetup, 0.0); CHKERRQ(ierr);
1671ccaff030SJeremy L Thompson   if (1) { // Record boundary values from initial condition and override DMPlexInsertBoundaryValues()
1672ccaff030SJeremy L Thompson     // We use this for the main simulation DM because the reference DMPlexInsertBoundaryValues() is very slow.  If we
1673ccaff030SJeremy L Thompson     // disable this, we should still get the same results due to the problem->bc function, but with potentially much
1674ccaff030SJeremy L Thompson     // slower execution.
1675ccaff030SJeremy L Thompson     Vec Qbc;
1676ccaff030SJeremy L Thompson     ierr = DMGetNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
1677ccaff030SJeremy L Thompson     ierr = VecCopy(Qloc, Qbc); CHKERRQ(ierr);
1678ccaff030SJeremy L Thompson     ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
1679ccaff030SJeremy L Thompson     ierr = DMGlobalToLocal(dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
1680ccaff030SJeremy L Thompson     ierr = VecAXPY(Qbc, -1., Qloc); CHKERRQ(ierr);
1681ccaff030SJeremy L Thompson     ierr = DMRestoreNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
1682ccaff030SJeremy L Thompson     ierr = PetscObjectComposeFunction((PetscObject)dm,
168384d34d69SLeila Ghaffari                                       "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_NS);
168484d34d69SLeila Ghaffari     CHKERRQ(ierr);
1685ccaff030SJeremy L Thompson   }
1686ccaff030SJeremy L Thompson 
1687ccaff030SJeremy L Thompson   MPI_Comm_rank(comm, &rank);
1688d99129b9SLeila Ghaffari   if (!rank) {ierr = PetscMkdir(user->outputdir); CHKERRQ(ierr);}
1689ccaff030SJeremy L Thompson   // Gather initial Q values
1690ccaff030SJeremy L Thompson   // In case of continuation of simulation, set up initial values from binary file
1691ccaff030SJeremy L Thompson   if (contsteps) { // continue from existent solution
1692ccaff030SJeremy L Thompson     PetscViewer viewer;
1693ccaff030SJeremy L Thompson     char filepath[PETSC_MAX_PATH_LEN];
1694ccaff030SJeremy L Thompson     // Read input
1695ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-solution.bin",
1696d99129b9SLeila Ghaffari                          user->outputdir);
1697ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1698ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
1699ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1700ccaff030SJeremy L Thompson     ierr = VecLoad(Q, viewer); CHKERRQ(ierr);
1701ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
1702ccaff030SJeremy L Thompson   }
1703ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &Qloc); CHKERRQ(ierr);
1704ccaff030SJeremy L Thompson 
1705ccaff030SJeremy L Thompson   // Create and setup TS
1706ccaff030SJeremy L Thompson   ierr = TSCreate(comm, &ts); CHKERRQ(ierr);
1707ccaff030SJeremy L Thompson   ierr = TSSetDM(ts, dm); CHKERRQ(ierr);
1708ccaff030SJeremy L Thompson   if (implicit) {
1709ccaff030SJeremy L Thompson     ierr = TSSetType(ts, TSBDF); CHKERRQ(ierr);
1710ccaff030SJeremy L Thompson     if (user->op_ifunction) {
1711ccaff030SJeremy L Thompson       ierr = TSSetIFunction(ts, NULL, IFunction_NS, &user); CHKERRQ(ierr);
1712ccaff030SJeremy L Thompson     } else {                    // Implicit integrators can fall back to using an RHSFunction
1713ccaff030SJeremy L Thompson       ierr = TSSetRHSFunction(ts, NULL, RHS_NS, &user); CHKERRQ(ierr);
1714ccaff030SJeremy L Thompson     }
1715ccaff030SJeremy L Thompson   } else {
1716ccaff030SJeremy L Thompson     if (!user->op_rhs) SETERRQ(comm, PETSC_ERR_ARG_NULL,
1717ccaff030SJeremy L Thompson                                  "Problem does not provide RHSFunction");
1718ccaff030SJeremy L Thompson     ierr = TSSetType(ts, TSRK); CHKERRQ(ierr);
1719ccaff030SJeremy L Thompson     ierr = TSRKSetType(ts, TSRK5F); CHKERRQ(ierr);
1720ccaff030SJeremy L Thompson     ierr = TSSetRHSFunction(ts, NULL, RHS_NS, &user); CHKERRQ(ierr);
1721ccaff030SJeremy L Thompson   }
1722ccaff030SJeremy L Thompson   ierr = TSSetMaxTime(ts, 500. * units->second); CHKERRQ(ierr);
1723ccaff030SJeremy L Thompson   ierr = TSSetExactFinalTime(ts, TS_EXACTFINALTIME_STEPOVER); CHKERRQ(ierr);
1724ccaff030SJeremy L Thompson   ierr = TSSetTimeStep(ts, 1.e-2 * units->second); CHKERRQ(ierr);
1725dc8efd83SLeila Ghaffari   if (test) {ierr = TSSetMaxSteps(ts, 10); CHKERRQ(ierr);}
1726ccaff030SJeremy L Thompson   ierr = TSGetAdapt(ts, &adapt); CHKERRQ(ierr);
1727ccaff030SJeremy L Thompson   ierr = TSAdaptSetStepLimits(adapt,
1728ccaff030SJeremy L Thompson                               1.e-12 * units->second,
1729ccaff030SJeremy L Thompson                               1.e2 * units->second); CHKERRQ(ierr);
1730ccaff030SJeremy L Thompson   ierr = TSSetFromOptions(ts); CHKERRQ(ierr);
1731ccaff030SJeremy L Thompson   if (!contsteps) { // print initial condition
1732dc8efd83SLeila Ghaffari     if (!test) {
1733ccaff030SJeremy L Thompson       ierr = TSMonitor_NS(ts, 0, 0., Q, user); CHKERRQ(ierr);
1734ccaff030SJeremy L Thompson     }
1735ccaff030SJeremy L Thompson   } else { // continue from time of last output
1736ccaff030SJeremy L Thompson     PetscReal time;
1737ccaff030SJeremy L Thompson     PetscInt count;
1738ccaff030SJeremy L Thompson     PetscViewer viewer;
1739ccaff030SJeremy L Thompson     char filepath[PETSC_MAX_PATH_LEN];
1740ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-time.bin",
1741d99129b9SLeila Ghaffari                          user->outputdir); CHKERRQ(ierr);
1742ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
1743ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1744ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryRead(viewer, &time, 1, &count, PETSC_REAL);
1745ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1746ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
1747ccaff030SJeremy L Thompson     ierr = TSSetTime(ts, time * user->units->second); CHKERRQ(ierr);
1748ccaff030SJeremy L Thompson   }
1749dc8efd83SLeila Ghaffari   if (!test) {
1750ccaff030SJeremy L Thompson     ierr = TSMonitorSet(ts, TSMonitor_NS, user, NULL); CHKERRQ(ierr);
1751ccaff030SJeremy L Thompson   }
1752ccaff030SJeremy L Thompson 
1753ccaff030SJeremy L Thompson   // Solve
1754ccaff030SJeremy L Thompson   start = MPI_Wtime();
1755ccaff030SJeremy L Thompson   ierr = PetscBarrier((PetscObject)ts); CHKERRQ(ierr);
1756ccaff030SJeremy L Thompson   ierr = TSSolve(ts, Q); CHKERRQ(ierr);
1757ccaff030SJeremy L Thompson   cpu_time_used = MPI_Wtime() - start;
1758ccaff030SJeremy L Thompson   ierr = TSGetSolveTime(ts, &ftime); CHKERRQ(ierr);
1759ccaff030SJeremy L Thompson   ierr = MPI_Allreduce(MPI_IN_PLACE, &cpu_time_used, 1, MPI_DOUBLE, MPI_MIN,
1760ccaff030SJeremy L Thompson                        comm); CHKERRQ(ierr);
1761dc8efd83SLeila Ghaffari   if (!test) {
1762ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
176384d34d69SLeila Ghaffari                        "Time taken for solution (sec): %g\n",
1764ccaff030SJeremy L Thompson                        (double)cpu_time_used); CHKERRQ(ierr);
1765ccaff030SJeremy L Thompson   }
1766ccaff030SJeremy L Thompson 
1767ccaff030SJeremy L Thompson   // Get error
1768dc8efd83SLeila Ghaffari   if (problem->non_zero_time && !test) {
1769ccaff030SJeremy L Thompson     Vec Qexact, Qexactloc;
1770ccaff030SJeremy L Thompson     PetscReal norm;
1771ccaff030SJeremy L Thompson     ierr = DMCreateGlobalVector(dm, &Qexact); CHKERRQ(ierr);
1772ccaff030SJeremy L Thompson     ierr = DMGetLocalVector(dm, &Qexactloc); CHKERRQ(ierr);
1773ccaff030SJeremy L Thompson     ierr = VecGetSize(Qexactloc, &lnodes); CHKERRQ(ierr);
1774ccaff030SJeremy L Thompson 
177584d34d69SLeila Ghaffari     ierr = ICs_FixMultiplicity(op_ics, xcorners, q0ceed, dm, Qexactloc, Qexact,
1776777ff853SJeremy L Thompson                                restrictq, ctxSetup, ftime); CHKERRQ(ierr);
1777ccaff030SJeremy L Thompson 
1778ccaff030SJeremy L Thompson     ierr = VecAXPY(Q, -1.0, Qexact);  CHKERRQ(ierr);
1779ccaff030SJeremy L Thompson     ierr = VecNorm(Q, NORM_MAX, &norm); CHKERRQ(ierr);
1780cfa64770SLeila Ghaffari     CeedVectorDestroy(&q0ceed);
1781ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
1782ccaff030SJeremy L Thompson                        "Max Error: %g\n",
1783ccaff030SJeremy L Thompson                        (double)norm); CHKERRQ(ierr);
178484d34d69SLeila Ghaffari     // Clean up vectors
178584d34d69SLeila Ghaffari     ierr = DMRestoreLocalVector(dm, &Qexactloc); CHKERRQ(ierr);
178684d34d69SLeila Ghaffari     ierr = VecDestroy(&Qexact); CHKERRQ(ierr);
1787ccaff030SJeremy L Thompson   }
1788ccaff030SJeremy L Thompson 
1789ccaff030SJeremy L Thompson   // Output Statistics
1790ccaff030SJeremy L Thompson   ierr = TSGetStepNumber(ts, &steps); CHKERRQ(ierr);
1791dc8efd83SLeila Ghaffari   if (!test) {
1792ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
1793ccaff030SJeremy L Thompson                        "Time integrator took %D time steps to reach final time %g\n",
1794ccaff030SJeremy L Thompson                        steps, (double)ftime); CHKERRQ(ierr);
1795ccaff030SJeremy L Thompson   }
179684d34d69SLeila Ghaffari   // Output numerical values from command line
179784d34d69SLeila Ghaffari   ierr = VecViewFromOptions(Q, NULL, "-vec_view"); CHKERRQ(ierr);
179884d34d69SLeila Ghaffari 
1799335ea220SLeila Ghaffari   // Compare reference solution values with current test run for CI
1800dc8efd83SLeila Ghaffari   if (test) {
180184d34d69SLeila Ghaffari     PetscViewer viewer;
180284d34d69SLeila Ghaffari     // Read reference file
180384d34d69SLeila Ghaffari     Vec Qref;
180484d34d69SLeila Ghaffari     PetscReal error, Qrefnorm;
180584d34d69SLeila Ghaffari     ierr = VecDuplicate(Q, &Qref); CHKERRQ(ierr);
1806dc8efd83SLeila Ghaffari     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
180784d34d69SLeila Ghaffari     CHKERRQ(ierr);
180884d34d69SLeila Ghaffari     ierr = VecLoad(Qref, viewer); CHKERRQ(ierr);
180984d34d69SLeila Ghaffari     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
181084d34d69SLeila Ghaffari 
181184d34d69SLeila Ghaffari     // Compute error with respect to reference solution
181284d34d69SLeila Ghaffari     ierr = VecAXPY(Q, -1.0, Qref);  CHKERRQ(ierr);
181384d34d69SLeila Ghaffari     ierr = VecNorm(Qref, NORM_MAX, &Qrefnorm); CHKERRQ(ierr);
181484d34d69SLeila Ghaffari     ierr = VecScale(Q, 1./Qrefnorm); CHKERRQ(ierr);
181584d34d69SLeila Ghaffari     ierr = VecNorm(Q, NORM_MAX, &error); CHKERRQ(ierr);
181684d34d69SLeila Ghaffari     ierr = VecDestroy(&Qref); CHKERRQ(ierr);
181784d34d69SLeila Ghaffari     // Check error
1818dc8efd83SLeila Ghaffari     if (error > testtol) {
181984d34d69SLeila Ghaffari       ierr = PetscPrintf(PETSC_COMM_WORLD,
182084d34d69SLeila Ghaffari                          "Test failed with error norm %g\n",
182184d34d69SLeila Ghaffari                          (double)error); CHKERRQ(ierr);
182284d34d69SLeila Ghaffari     }
182384d34d69SLeila Ghaffari     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
182484d34d69SLeila Ghaffari   }
18259cf88b28Svaleriabarra 
1826ccaff030SJeremy L Thompson   // Clean up libCEED
18278b982baeSLeila Ghaffari   CeedVectorDestroy(&qdata);
1828ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->qceed);
1829ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->qdotceed);
1830ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->gceed);
1831ccaff030SJeremy L Thompson   CeedVectorDestroy(&xcorners);
183284d34d69SLeila Ghaffari   CeedBasisDestroy(&basisq);
183384d34d69SLeila Ghaffari   CeedBasisDestroy(&basisx);
183484d34d69SLeila Ghaffari   CeedBasisDestroy(&basisxc);
183584d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictq);
183684d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictx);
183784d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictqdi);
1838ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_setupVol);
1839ccaff030SJeremy L Thompson   CeedQFunctionDestroy(&qf_ics);
1840ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_rhsVol);
1841ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_ifunctionVol);
1842777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxSetup);
1843777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxNS);
1844777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxAdvection2d);
1845777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxSurface);
18460da62991SLeila Ghaffari   CeedQFunctionContextDestroy(&ctxEuler);
1847ea6e0f84SLeila Ghaffari   CeedOperatorDestroy(&op_setupVol);
1848ccaff030SJeremy L Thompson   CeedOperatorDestroy(&op_ics);
18496a0edaf9SLeila Ghaffari   CeedOperatorDestroy(&user->op_rhs_vol);
18506a0edaf9SLeila Ghaffari   CeedOperatorDestroy(&user->op_ifunction_vol);
18516a0edaf9SLeila Ghaffari   CeedDestroy(&ceed);
18526a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisqSur);
18536a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisxSur);
18546a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisxcSur);
18556a0edaf9SLeila Ghaffari   CeedQFunctionDestroy(&qf_setupSur);
18567ed8b9c7SLeila Ghaffari   CeedQFunctionDestroy(&qf_applySur);
1857ccaff030SJeremy L Thompson   CeedOperatorDestroy(&user->op_rhs);
1858ccaff030SJeremy L Thompson   CeedOperatorDestroy(&user->op_ifunction);
1859ccaff030SJeremy L Thompson 
1860ccaff030SJeremy L Thompson   // Clean up PETSc
1861ccaff030SJeremy L Thompson   ierr = VecDestroy(&Q); CHKERRQ(ierr);
1862ccaff030SJeremy L Thompson   ierr = VecDestroy(&user->M); CHKERRQ(ierr);
1863ccaff030SJeremy L Thompson   ierr = MatDestroy(&interpviz); CHKERRQ(ierr);
1864ccaff030SJeremy L Thompson   ierr = DMDestroy(&dmviz); CHKERRQ(ierr);
1865ccaff030SJeremy L Thompson   ierr = TSDestroy(&ts); CHKERRQ(ierr);
1866ccaff030SJeremy L Thompson   ierr = DMDestroy(&dm); CHKERRQ(ierr);
1867ccaff030SJeremy L Thompson   ierr = PetscFree(units); CHKERRQ(ierr);
1868ccaff030SJeremy L Thompson   ierr = PetscFree(user); CHKERRQ(ierr);
18695bf10c6fSLeila Ghaffari   ierr = PetscFree(ctxEulerData); CHKERRQ(ierr);
1870ccaff030SJeremy L Thompson   return PetscFinalize();
1871ccaff030SJeremy L Thompson }
1872