xref: /libCEED/examples/fluids/navierstokes.c (revision c8cb9b1f68669099d8f2dbe77ff8d748b826da66)
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 
437b6df2e11SLeila Ghaffari   if (wind_type == ADVECTION_WIND_TRANSLATION || problemChoice == NS_EULER_VORTEX) {
4384438636fSLeila Ghaffari     // Ignore wall and slip BCs
439b6df2e11SLeila Ghaffari     bc->nwall = bc->nslip[0] = bc->nslip[1] = 0;
440b6df2e11SLeila Ghaffari     if (wind_type == ADVECTION_WIND_TRANSLATION) bc->nslip[2] = 0;
4414438636fSLeila Ghaffari 
4424438636fSLeila Ghaffari     // Set number of faces
4437659d40cSLeila Ghaffari     if (dim == 2) nFace = 4;
4447659d40cSLeila Ghaffari     if (dim == 3) nFace = 6;
4459fe13df9SLeila Ghaffari 
4467659d40cSLeila Ghaffari     // Create CEED Operator for each boundary face
4474438636fSLeila Ghaffari     PetscInt localNelemSur[6];
4484438636fSLeila Ghaffari     CeedVector qdataSur[6];
4494438636fSLeila Ghaffari     CeedOperator op_setupSur[6], op_applySur[6];
4504438636fSLeila Ghaffari     CeedElemRestriction restrictxSur[6], restrictqSur[6], restrictqdiSur[6];
45186ba6f09SLeila Ghaffari 
4527659d40cSLeila Ghaffari     for (CeedInt i=0; i<nFace; i++) {
4537659d40cSLeila Ghaffari       ierr = GetRestrictionForDomain(ceed, dm, height, domainLabel, i+1, numP_Sur,
454f259b054Svaleriabarra                                      numQ_Sur, qdatasizeSur, &restrictqSur[i],
455f259b054Svaleriabarra                                      &restrictxSur[i], &restrictqdiSur[i]);
456f259b054Svaleriabarra       CHKERRQ(ierr);
4577659d40cSLeila Ghaffari       // Create the CEED vectors that will be needed in Boundary setup
4587659d40cSLeila Ghaffari       CeedElemRestrictionGetNumElements(restrictqSur[i], &localNelemSur[i]);
459f259b054Svaleriabarra       CeedVectorCreate(ceed, qdatasizeSur*localNelemSur[i]*NqptsSur,
460f259b054Svaleriabarra                        &qdataSur[i]);
4617659d40cSLeila Ghaffari       // Create the operator that builds the quadrature data for the Boundary operator
4627659d40cSLeila Ghaffari       CeedOperatorCreate(ceed, qf_setupSur, NULL, NULL, &op_setupSur[i]);
463ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "dx", restrictxSur[i], basisxSur,
464ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4657659d40cSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "weight", CEED_ELEMRESTRICTION_NONE,
466ca3ac6ddSLeila Ghaffari                            basisxSur, CEED_VECTOR_NONE);
4677659d40cSLeila Ghaffari       CeedOperatorSetField(op_setupSur[i], "qdataSur", restrictqdiSur[i],
468ca3ac6ddSLeila Ghaffari                            CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
4697659d40cSLeila Ghaffari       // Create Boundary operator
4707659d40cSLeila Ghaffari       CeedOperatorCreate(ceed, qf_applySur, NULL, NULL, &op_applySur[i]);
471ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "q", restrictqSur[i], basisqSur,
472ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4737659d40cSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "qdataSur", restrictqdiSur[i],
4747659d40cSLeila Ghaffari                            CEED_BASIS_COLLOCATED, qdataSur[i]);
475f259b054Svaleriabarra       CeedOperatorSetField(op_applySur[i], "x", restrictxSur[i], basisxSur,
476f259b054Svaleriabarra                            xcorners);
477ebb4b9bdSLeila Ghaffari       CeedOperatorSetField(op_applySur[i], "v", restrictqSur[i], basisqSur,
478ebb4b9bdSLeila Ghaffari                            CEED_VECTOR_ACTIVE);
4797659d40cSLeila Ghaffari       // Apply CEED operator for Boundary setup
480ebb4b9bdSLeila Ghaffari       CeedOperatorApply(op_setupSur[i], xcorners, qdataSur[i],
481ebb4b9bdSLeila Ghaffari                         CEED_REQUEST_IMMEDIATE);
4824438636fSLeila Ghaffari       // --Apply Sub-Operator for the Boundary
4837659d40cSLeila Ghaffari       CeedCompositeOperatorAddSub(*op_apply, op_applySur[i]);
4849fe13df9SLeila Ghaffari     }
4851e150236SLeila Ghaffari     CeedVectorDestroy(&xcorners);
486ca3ac6ddSLeila Ghaffari   }
487ca3ac6ddSLeila Ghaffari   PetscFunctionReturn(0);
488ca3ac6ddSLeila Ghaffari }
489ca3ac6ddSLeila Ghaffari 
490ccaff030SJeremy L Thompson static int CreateVectorFromPetscVec(Ceed ceed, Vec p, CeedVector *v) {
491ccaff030SJeremy L Thompson   PetscErrorCode ierr;
492ccaff030SJeremy L Thompson   PetscInt m;
493ccaff030SJeremy L Thompson 
494ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
495ccaff030SJeremy L Thompson   ierr = VecGetLocalSize(p, &m); CHKERRQ(ierr);
496ccaff030SJeremy L Thompson   ierr = CeedVectorCreate(ceed, m, v); CHKERRQ(ierr);
497ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
498ccaff030SJeremy L Thompson }
499ccaff030SJeremy L Thompson 
500ccaff030SJeremy L Thompson static int VectorPlacePetscVec(CeedVector c, Vec p) {
501ccaff030SJeremy L Thompson   PetscErrorCode ierr;
502ccaff030SJeremy L Thompson   PetscInt mceed, mpetsc;
503ccaff030SJeremy L Thompson   PetscScalar *a;
504ccaff030SJeremy L Thompson 
505ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
506ccaff030SJeremy L Thompson   ierr = CeedVectorGetLength(c, &mceed); CHKERRQ(ierr);
507ccaff030SJeremy L Thompson   ierr = VecGetLocalSize(p, &mpetsc); CHKERRQ(ierr);
508ccaff030SJeremy L Thompson   if (mceed != mpetsc) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP,
50984d34d69SLeila Ghaffari                                   "Cannot place PETSc Vec of length %D in CeedVector of length %D",
51084d34d69SLeila Ghaffari                                   mpetsc, mceed);
511ccaff030SJeremy L Thompson   ierr = VecGetArray(p, &a); CHKERRQ(ierr);
512ccaff030SJeremy L Thompson   CeedVectorSetArray(c, CEED_MEM_HOST, CEED_USE_POINTER, a);
513ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
514ccaff030SJeremy L Thompson }
515ccaff030SJeremy L Thompson 
516ccaff030SJeremy L Thompson static PetscErrorCode DMPlexInsertBoundaryValues_NS(DM dm,
517ccaff030SJeremy L Thompson     PetscBool insertEssential, Vec Qloc, PetscReal time, Vec faceGeomFVM,
518ccaff030SJeremy L Thompson     Vec cellGeomFVM, Vec gradFVM) {
519ccaff030SJeremy L Thompson   PetscErrorCode ierr;
520ccaff030SJeremy L Thompson   Vec Qbc;
521ccaff030SJeremy L Thompson 
522ccaff030SJeremy L Thompson   PetscFunctionBegin;
523ccaff030SJeremy L Thompson   ierr = DMGetNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
524ccaff030SJeremy L Thompson   ierr = VecAXPY(Qloc, 1., Qbc); CHKERRQ(ierr);
525ccaff030SJeremy L Thompson   ierr = DMRestoreNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
526ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
527ccaff030SJeremy L Thompson }
528ccaff030SJeremy L Thompson 
529ccaff030SJeremy L Thompson // This is the RHS of the ODE, given as u_t = G(t,u)
530ccaff030SJeremy L Thompson // This function takes in a state vector Q and writes into G
531ccaff030SJeremy L Thompson static PetscErrorCode RHS_NS(TS ts, PetscReal t, Vec Q, Vec G, void *userData) {
532ccaff030SJeremy L Thompson   PetscErrorCode ierr;
533ccaff030SJeremy L Thompson   User user = *(User *)userData;
534ccaff030SJeremy L Thompson   PetscScalar *q, *g;
535ccaff030SJeremy L Thompson   Vec Qloc, Gloc;
536ccaff030SJeremy L Thompson 
537ccaff030SJeremy L Thompson   // Global-to-local
538ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
5395bf10c6fSLeila Ghaffari   user->ctxEulerData->currentTime = t;
540ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
541ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
542ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
543ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
544ccaff030SJeremy L Thompson   ierr = DMPlexInsertBoundaryValues(user->dm, PETSC_TRUE, Qloc, 0.0,
545ccaff030SJeremy L Thompson                                     NULL, NULL, NULL); CHKERRQ(ierr);
546ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Gloc); CHKERRQ(ierr);
547ccaff030SJeremy L Thompson 
548ccaff030SJeremy L Thompson   // Ceed Vectors
549ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qloc, (const PetscScalar **)&q); CHKERRQ(ierr);
550ccaff030SJeremy L Thompson   ierr = VecGetArray(Gloc, &g); CHKERRQ(ierr);
551ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qceed, CEED_MEM_HOST, CEED_USE_POINTER, q);
552ccaff030SJeremy L Thompson   CeedVectorSetArray(user->gceed, CEED_MEM_HOST, CEED_USE_POINTER, g);
553ccaff030SJeremy L Thompson 
554ccaff030SJeremy L Thompson   // Apply CEED operator
555ccaff030SJeremy L Thompson   CeedOperatorApply(user->op_rhs, user->qceed, user->gceed,
556ccaff030SJeremy L Thompson                     CEED_REQUEST_IMMEDIATE);
557ccaff030SJeremy L Thompson 
558ccaff030SJeremy L Thompson   // Restore vectors
559ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qloc, (const PetscScalar **)&q); CHKERRQ(ierr);
560ccaff030SJeremy L Thompson   ierr = VecRestoreArray(Gloc, &g); CHKERRQ(ierr);
561ccaff030SJeremy L Thompson 
562ccaff030SJeremy L Thompson   ierr = VecZeroEntries(G); CHKERRQ(ierr);
563ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(user->dm, Gloc, ADD_VALUES, G); CHKERRQ(ierr);
564ccaff030SJeremy L Thompson 
565ccaff030SJeremy L Thompson   // Inverse of the lumped mass matrix
566ccaff030SJeremy L Thompson   ierr = VecPointwiseMult(G, G, user->M); // M is Minv
567ccaff030SJeremy L Thompson   CHKERRQ(ierr);
568ccaff030SJeremy L Thompson 
569ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
570ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
571ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
572ccaff030SJeremy L Thompson }
573ccaff030SJeremy L Thompson 
574ccaff030SJeremy L Thompson static PetscErrorCode IFunction_NS(TS ts, PetscReal t, Vec Q, Vec Qdot, Vec G,
575ccaff030SJeremy L Thompson                                    void *userData) {
576ccaff030SJeremy L Thompson   PetscErrorCode ierr;
577ccaff030SJeremy L Thompson   User user = *(User *)userData;
578ccaff030SJeremy L Thompson   const PetscScalar *q, *qdot;
579ccaff030SJeremy L Thompson   PetscScalar *g;
580ccaff030SJeremy L Thompson   Vec Qloc, Qdotloc, Gloc;
581ccaff030SJeremy L Thompson 
582ccaff030SJeremy L Thompson   // Global-to-local
583ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
584ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
585ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qdotloc); CHKERRQ(ierr);
586ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
587ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
588ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
589ccaff030SJeremy L Thompson   ierr = DMPlexInsertBoundaryValues(user->dm, PETSC_TRUE, Qloc, 0.0,
590ccaff030SJeremy L Thompson                                     NULL, NULL, NULL); CHKERRQ(ierr);
591ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qdotloc); CHKERRQ(ierr);
592ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Qdot, INSERT_VALUES, Qdotloc); CHKERRQ(ierr);
593ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Gloc); CHKERRQ(ierr);
594ccaff030SJeremy L Thompson 
595ccaff030SJeremy L Thompson   // Ceed Vectors
596ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qloc, &q); CHKERRQ(ierr);
597ccaff030SJeremy L Thompson   ierr = VecGetArrayRead(Qdotloc, &qdot); CHKERRQ(ierr);
598ccaff030SJeremy L Thompson   ierr = VecGetArray(Gloc, &g); CHKERRQ(ierr);
599ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qceed, CEED_MEM_HOST, CEED_USE_POINTER,
600ccaff030SJeremy L Thompson                      (PetscScalar *)q);
601ccaff030SJeremy L Thompson   CeedVectorSetArray(user->qdotceed, CEED_MEM_HOST, CEED_USE_POINTER,
602ccaff030SJeremy L Thompson                      (PetscScalar *)qdot);
603ccaff030SJeremy L Thompson   CeedVectorSetArray(user->gceed, CEED_MEM_HOST, CEED_USE_POINTER, g);
604ccaff030SJeremy L Thompson 
605ccaff030SJeremy L Thompson   // Apply CEED operator
606ccaff030SJeremy L Thompson   CeedOperatorApply(user->op_ifunction, user->qceed, user->gceed,
607ccaff030SJeremy L Thompson                     CEED_REQUEST_IMMEDIATE);
608ccaff030SJeremy L Thompson 
609ccaff030SJeremy L Thompson   // Restore vectors
610ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qloc, &q); CHKERRQ(ierr);
611ccaff030SJeremy L Thompson   ierr = VecRestoreArrayRead(Qdotloc, &qdot); CHKERRQ(ierr);
612ccaff030SJeremy L Thompson   ierr = VecRestoreArray(Gloc, &g); CHKERRQ(ierr);
613ccaff030SJeremy L Thompson 
614ccaff030SJeremy L Thompson   ierr = VecZeroEntries(G); CHKERRQ(ierr);
615ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(user->dm, Gloc, ADD_VALUES, G); CHKERRQ(ierr);
616ccaff030SJeremy L Thompson 
617ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
618ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qdotloc); CHKERRQ(ierr);
619ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Gloc); CHKERRQ(ierr);
620ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
621ccaff030SJeremy L Thompson }
622ccaff030SJeremy L Thompson 
623ccaff030SJeremy L Thompson // User provided TS Monitor
624ccaff030SJeremy L Thompson static PetscErrorCode TSMonitor_NS(TS ts, PetscInt stepno, PetscReal time,
625ccaff030SJeremy L Thompson                                    Vec Q, void *ctx) {
626ccaff030SJeremy L Thompson   User user = ctx;
627ccaff030SJeremy L Thompson   Vec Qloc;
628ccaff030SJeremy L Thompson   char filepath[PETSC_MAX_PATH_LEN];
629ccaff030SJeremy L Thompson   PetscViewer viewer;
630ccaff030SJeremy L Thompson   PetscErrorCode ierr;
631ccaff030SJeremy L Thompson 
632ccaff030SJeremy L Thompson   // Set up output
633ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
634ccaff030SJeremy L Thompson   // Print every 'outputfreq' steps
635ccaff030SJeremy L Thompson   if (stepno % user->outputfreq != 0)
636ccaff030SJeremy L Thompson     PetscFunctionReturn(0);
637ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
638ccaff030SJeremy L Thompson   ierr = PetscObjectSetName((PetscObject)Qloc, "StateVec"); CHKERRQ(ierr);
639ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
640ccaff030SJeremy L Thompson   ierr = DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
641ccaff030SJeremy L Thompson 
642ccaff030SJeremy L Thompson   // Output
643ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-%03D.vtu",
644d99129b9SLeila Ghaffari                        user->outputdir, stepno + user->contsteps);
645ccaff030SJeremy L Thompson   CHKERRQ(ierr);
646ccaff030SJeremy L Thompson   ierr = PetscViewerVTKOpen(PetscObjectComm((PetscObject)Q), filepath,
647ccaff030SJeremy L Thompson                             FILE_MODE_WRITE, &viewer); CHKERRQ(ierr);
648ccaff030SJeremy L Thompson   ierr = VecView(Qloc, viewer); CHKERRQ(ierr);
6499d801c56SJed Brown   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
650ccaff030SJeremy L Thompson   if (user->dmviz) {
651ccaff030SJeremy L Thompson     Vec Qrefined, Qrefined_loc;
652ccaff030SJeremy L Thompson     char filepath_refined[PETSC_MAX_PATH_LEN];
653ccaff030SJeremy L Thompson     PetscViewer viewer_refined;
654ccaff030SJeremy L Thompson 
655ccaff030SJeremy L Thompson     ierr = DMGetGlobalVector(user->dmviz, &Qrefined); CHKERRQ(ierr);
656ccaff030SJeremy L Thompson     ierr = DMGetLocalVector(user->dmviz, &Qrefined_loc); CHKERRQ(ierr);
657ccaff030SJeremy L Thompson     ierr = PetscObjectSetName((PetscObject)Qrefined_loc, "Refined");
658ccaff030SJeremy L Thompson     CHKERRQ(ierr);
659ccaff030SJeremy L Thompson     ierr = MatInterpolate(user->interpviz, Q, Qrefined); CHKERRQ(ierr);
660ccaff030SJeremy L Thompson     ierr = VecZeroEntries(Qrefined_loc); CHKERRQ(ierr);
661ccaff030SJeremy L Thompson     ierr = DMGlobalToLocal(user->dmviz, Qrefined, INSERT_VALUES, Qrefined_loc);
662ccaff030SJeremy L Thompson     CHKERRQ(ierr);
663ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath_refined, sizeof filepath_refined,
664ccaff030SJeremy L Thompson                          "%s/nsrefined-%03D.vtu",
665d99129b9SLeila Ghaffari                          user->outputdir, stepno + user->contsteps);
666ccaff030SJeremy L Thompson     CHKERRQ(ierr);
667ccaff030SJeremy L Thompson     ierr = PetscViewerVTKOpen(PetscObjectComm((PetscObject)Qrefined),
668ccaff030SJeremy L Thompson                               filepath_refined,
669ccaff030SJeremy L Thompson                               FILE_MODE_WRITE, &viewer_refined); CHKERRQ(ierr);
670ccaff030SJeremy L Thompson     ierr = VecView(Qrefined_loc, viewer_refined); CHKERRQ(ierr);
671ccaff030SJeremy L Thompson     ierr = DMRestoreLocalVector(user->dmviz, &Qrefined_loc); CHKERRQ(ierr);
672ccaff030SJeremy L Thompson     ierr = DMRestoreGlobalVector(user->dmviz, &Qrefined); CHKERRQ(ierr);
673ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer_refined); CHKERRQ(ierr);
674ccaff030SJeremy L Thompson   }
675ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(user->dm, &Qloc); CHKERRQ(ierr);
676ccaff030SJeremy L Thompson 
677ccaff030SJeremy L Thompson   // Save data in a binary file for continuation of simulations
678ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-solution.bin",
679d99129b9SLeila Ghaffari                        user->outputdir); CHKERRQ(ierr);
680ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryOpen(user->comm, filepath, FILE_MODE_WRITE, &viewer);
681ccaff030SJeremy L Thompson   CHKERRQ(ierr);
682ccaff030SJeremy L Thompson   ierr = VecView(Q, viewer); CHKERRQ(ierr);
683ccaff030SJeremy L Thompson   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
684ccaff030SJeremy L Thompson 
685ccaff030SJeremy L Thompson   // Save time stamp
686ccaff030SJeremy L Thompson   // Dimensionalize time back
687ccaff030SJeremy L Thompson   time /= user->units->second;
688ccaff030SJeremy L Thompson   ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-time.bin",
689d99129b9SLeila Ghaffari                        user->outputdir); CHKERRQ(ierr);
690ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryOpen(user->comm, filepath, FILE_MODE_WRITE, &viewer);
691ccaff030SJeremy L Thompson   CHKERRQ(ierr);
692ccaff030SJeremy L Thompson   #if PETSC_VERSION_GE(3,13,0)
693ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryWrite(viewer, &time, 1, PETSC_REAL);
694ccaff030SJeremy L Thompson   #else
695ccaff030SJeremy L Thompson   ierr = PetscViewerBinaryWrite(viewer, &time, 1, PETSC_REAL, true);
696ccaff030SJeremy L Thompson   #endif
697ccaff030SJeremy L Thompson   CHKERRQ(ierr);
698ccaff030SJeremy L Thompson   ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
699ccaff030SJeremy L Thompson 
700ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
701ccaff030SJeremy L Thompson }
702ccaff030SJeremy L Thompson 
70384d34d69SLeila Ghaffari static PetscErrorCode ICs_FixMultiplicity(CeedOperator op_ics,
704ccaff030SJeremy L Thompson     CeedVector xcorners, CeedVector q0ceed, DM dm, Vec Qloc, Vec Q,
705777ff853SJeremy L Thompson     CeedElemRestriction restrictq, CeedQFunctionContext ctxSetup, CeedScalar time) {
706ccaff030SJeremy L Thompson   PetscErrorCode ierr;
707ccaff030SJeremy L Thompson   CeedVector multlvec;
708ccaff030SJeremy L Thompson   Vec Multiplicity, MultiplicityLoc;
709ccaff030SJeremy L Thompson 
710777ff853SJeremy L Thompson   SetupContext ctxSetupData;
711777ff853SJeremy L Thompson   CeedQFunctionContextGetData(ctxSetup, CEED_MEM_HOST, (void **)&ctxSetupData);
712777ff853SJeremy L Thompson   ctxSetupData->time = time;
713777ff853SJeremy L Thompson   CeedQFunctionContextRestoreData(ctxSetup, (void **)&ctxSetupData);
714777ff853SJeremy L Thompson 
715ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
716ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(q0ceed, Qloc); CHKERRQ(ierr);
717ccaff030SJeremy L Thompson   CeedOperatorApply(op_ics, xcorners, q0ceed, CEED_REQUEST_IMMEDIATE);
718ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Q); CHKERRQ(ierr);
719ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, Qloc, ADD_VALUES, Q); CHKERRQ(ierr);
720ccaff030SJeremy L Thompson 
721ccaff030SJeremy L Thompson   // Fix multiplicity for output of ICs
722ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &MultiplicityLoc); CHKERRQ(ierr);
723ccaff030SJeremy L Thompson   CeedElemRestrictionCreateVector(restrictq, &multlvec, NULL);
724ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(multlvec, MultiplicityLoc); CHKERRQ(ierr);
725ccaff030SJeremy L Thompson   CeedElemRestrictionGetMultiplicity(restrictq, multlvec);
726ccaff030SJeremy L Thompson   CeedVectorDestroy(&multlvec);
727ccaff030SJeremy L Thompson   ierr = DMGetGlobalVector(dm, &Multiplicity); CHKERRQ(ierr);
728ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Multiplicity); CHKERRQ(ierr);
729ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, MultiplicityLoc, ADD_VALUES, Multiplicity);
730ccaff030SJeremy L Thompson   CHKERRQ(ierr);
731ccaff030SJeremy L Thompson   ierr = VecPointwiseDivide(Q, Q, Multiplicity); CHKERRQ(ierr);
732ccaff030SJeremy L Thompson   ierr = VecPointwiseDivide(Qloc, Qloc, MultiplicityLoc); CHKERRQ(ierr);
733ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &MultiplicityLoc); CHKERRQ(ierr);
734ccaff030SJeremy L Thompson   ierr = DMRestoreGlobalVector(dm, &Multiplicity); CHKERRQ(ierr);
735ccaff030SJeremy L Thompson 
736ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
737ccaff030SJeremy L Thompson }
738ccaff030SJeremy L Thompson 
739ccaff030SJeremy L Thompson static PetscErrorCode ComputeLumpedMassMatrix(Ceed ceed, DM dm,
740ccaff030SJeremy L Thompson     CeedElemRestriction restrictq, CeedBasis basisq,
741ccaff030SJeremy L Thompson     CeedElemRestriction restrictqdi, CeedVector qdata, Vec M) {
742ccaff030SJeremy L Thompson   PetscErrorCode ierr;
743ccaff030SJeremy L Thompson   CeedQFunction qf_mass;
744ccaff030SJeremy L Thompson   CeedOperator op_mass;
745ccaff030SJeremy L Thompson   CeedVector mceed;
746ccaff030SJeremy L Thompson   Vec Mloc;
747ccaff030SJeremy L Thompson   CeedInt ncompq, qdatasize;
748ccaff030SJeremy L Thompson 
749ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
750ccaff030SJeremy L Thompson   CeedElemRestrictionGetNumComponents(restrictq, &ncompq);
751ccaff030SJeremy L Thompson   CeedElemRestrictionGetNumComponents(restrictqdi, &qdatasize);
752ccaff030SJeremy L Thompson   // Create the Q-function that defines the action of the mass operator
753ccaff030SJeremy L Thompson   CeedQFunctionCreateInterior(ceed, 1, Mass, Mass_loc, &qf_mass);
754ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_mass, "q", ncompq, CEED_EVAL_INTERP);
755ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_mass, "qdata", qdatasize, CEED_EVAL_NONE);
756ccaff030SJeremy L Thompson   CeedQFunctionAddOutput(qf_mass, "v", ncompq, CEED_EVAL_INTERP);
757ccaff030SJeremy L Thompson 
758ccaff030SJeremy L Thompson   // Create the mass operator
759ccaff030SJeremy L Thompson   CeedOperatorCreate(ceed, qf_mass, NULL, NULL, &op_mass);
760ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
761ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "qdata", restrictqdi,
762ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, qdata);
763ccaff030SJeremy L Thompson   CeedOperatorSetField(op_mass, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
764ccaff030SJeremy L Thompson 
765ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &Mloc); CHKERRQ(ierr);
766ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Mloc); CHKERRQ(ierr);
767ccaff030SJeremy L Thompson   CeedElemRestrictionCreateVector(restrictq, &mceed, NULL);
768ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(mceed, Mloc); CHKERRQ(ierr);
769ccaff030SJeremy L Thompson 
770ccaff030SJeremy L Thompson   {
771ccaff030SJeremy L Thompson     // Compute a lumped mass matrix
772ccaff030SJeremy L Thompson     CeedVector onesvec;
773ccaff030SJeremy L Thompson     CeedElemRestrictionCreateVector(restrictq, &onesvec, NULL);
774ccaff030SJeremy L Thompson     CeedVectorSetValue(onesvec, 1.0);
775ccaff030SJeremy L Thompson     CeedOperatorApply(op_mass, onesvec, mceed, CEED_REQUEST_IMMEDIATE);
776ccaff030SJeremy L Thompson     CeedVectorDestroy(&onesvec);
777ccaff030SJeremy L Thompson     CeedOperatorDestroy(&op_mass);
778ccaff030SJeremy L Thompson     CeedVectorDestroy(&mceed);
779ccaff030SJeremy L Thompson   }
780ccaff030SJeremy L Thompson   CeedQFunctionDestroy(&qf_mass);
781ccaff030SJeremy L Thompson 
782ccaff030SJeremy L Thompson   ierr = VecZeroEntries(M); CHKERRQ(ierr);
783ccaff030SJeremy L Thompson   ierr = DMLocalToGlobal(dm, Mloc, ADD_VALUES, M); CHKERRQ(ierr);
784ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &Mloc); CHKERRQ(ierr);
785ccaff030SJeremy L Thompson 
786ccaff030SJeremy L Thompson   // Invert diagonally lumped mass vector for RHS function
787ccaff030SJeremy L Thompson   ierr = VecReciprocal(M); CHKERRQ(ierr);
788ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
789ccaff030SJeremy L Thompson }
790ccaff030SJeremy L Thompson 
79184d34d69SLeila Ghaffari static PetscErrorCode SetUpDM(DM dm, problemData *problem, PetscInt degree,
7925bf10c6fSLeila Ghaffari                               SimpleBC bc, void *ctxSetupData, void *ctxMMS) {
793ccaff030SJeremy L Thompson   PetscErrorCode ierr;
794ccaff030SJeremy L Thompson 
795ccaff030SJeremy L Thompson   PetscFunctionBeginUser;
796ccaff030SJeremy L Thompson   {
797ccaff030SJeremy L Thompson     // Configure the finite element space and boundary conditions
798ccaff030SJeremy L Thompson     PetscFE fe;
799ccaff030SJeremy L Thompson     PetscInt ncompq = 5;
800ff6701fcSJed Brown     ierr = PetscFECreateLagrange(PETSC_COMM_SELF, problem->dim, ncompq,
801ff6701fcSJed Brown                                  PETSC_FALSE, degree, PETSC_DECIDE,
80232ed2d11SJed Brown                                  &fe); CHKERRQ(ierr);
803ccaff030SJeremy L Thompson     ierr = PetscObjectSetName((PetscObject)fe, "Q"); CHKERRQ(ierr);
804ccaff030SJeremy L Thompson     ierr = DMAddField(dm, NULL,(PetscObject)fe); CHKERRQ(ierr);
805ccaff030SJeremy L Thompson     ierr = DMCreateDS(dm); CHKERRQ(ierr);
80607af6069Svaleriabarra     {
80707af6069Svaleriabarra       PetscInt comps[1] = {1};
80807af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipx", "Face Sets", 0,
8093ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[0],
810777ff853SJeremy L Thompson                            bc->slips[0], ctxSetupData); CHKERRQ(ierr);
81107af6069Svaleriabarra       comps[0] = 2;
81207af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipy", "Face Sets", 0,
8133ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[1],
814777ff853SJeremy L Thompson                            bc->slips[1], ctxSetupData); CHKERRQ(ierr);
81507af6069Svaleriabarra       comps[0] = 3;
81607af6069Svaleriabarra       ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "slipz", "Face Sets", 0,
8173ab4fca6SValeria Barra                            1, comps, (void(*)(void))NULL, NULL, bc->nslip[2],
818777ff853SJeremy L Thompson                            bc->slips[2], ctxSetupData); CHKERRQ(ierr);
81907af6069Svaleriabarra     }
82084d34d69SLeila Ghaffari     if (bc->userbc == PETSC_TRUE) {
82184d34d69SLeila Ghaffari       for (PetscInt c = 0; c < 3; c++) {
82284d34d69SLeila Ghaffari         for (PetscInt s = 0; s < bc->nslip[c]; s++) {
82384d34d69SLeila Ghaffari           for (PetscInt w = 0; w < bc->nwall; w++) {
82484d34d69SLeila Ghaffari             if (bc->slips[c][s] == bc->walls[w])
82584d34d69SLeila Ghaffari               SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG,
826f259b054Svaleriabarra                        "Boundary condition already set on face %D!\n",
827f259b054Svaleriabarra                        bc->walls[w]);
82884d34d69SLeila Ghaffari           }
82984d34d69SLeila Ghaffari         }
83084d34d69SLeila Ghaffari       }
83184d34d69SLeila Ghaffari     }
83284d34d69SLeila Ghaffari     // Wall boundary conditions are zero energy density and zero flux for
83384d34d69SLeila Ghaffari     //   velocity in advection/advection2d, and zero velocity and zero flux
83484d34d69SLeila Ghaffari     //   for mass density and energy density in density_current
83584d34d69SLeila Ghaffari     {
83684d34d69SLeila Ghaffari       if (problem->bc == Exact_Advection || problem->bc == Exact_Advection2d) {
83784d34d69SLeila Ghaffari         PetscInt comps[1] = {4};
83884d34d69SLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", "Face Sets", 0,
8393ab4fca6SValeria Barra                              1, comps, (void(*)(void))problem->bc, NULL,
8400da62991SLeila Ghaffari                              bc->nwall, bc->walls, ctxSetupData); CHKERRQ(ierr);
841753d00aaSLeila Ghaffari       } else if (problem->bc == Exact_DC) {
84284d34d69SLeila Ghaffari         PetscInt comps[3] = {1, 2, 3};
84384d34d69SLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", "Face Sets", 0,
8443ab4fca6SValeria Barra                              3, comps, (void(*)(void))problem->bc, NULL,
845777ff853SJeremy L Thompson                              bc->nwall, bc->walls, ctxSetupData); CHKERRQ(ierr);
846753d00aaSLeila Ghaffari       } else if (problem->bc == Exact_Euler) {
84731effe6dSLeila Ghaffari         PetscInt bcMMS[4] = {3, 4, 5, 6};
848753d00aaSLeila Ghaffari         ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "mms", "Face Sets", 0,
8498896b5a1SLeila Ghaffari                              0, NULL, (void(*)(void))problem->bc, NULL,
85031effe6dSLeila Ghaffari                              4, bcMMS, ctxMMS); CHKERRQ(ierr);
85184d34d69SLeila Ghaffari       } else
85284d34d69SLeila Ghaffari         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_NULL,
85384d34d69SLeila Ghaffari                 "Undefined boundary conditions for this problem");
85484d34d69SLeila Ghaffari     }
855ccaff030SJeremy L Thompson     ierr = DMPlexSetClosurePermutationTensor(dm, PETSC_DETERMINE, NULL);
856ccaff030SJeremy L Thompson     CHKERRQ(ierr);
857ccaff030SJeremy L Thompson     ierr = PetscFEDestroy(&fe); CHKERRQ(ierr);
858ccaff030SJeremy L Thompson   }
859ccaff030SJeremy L Thompson   {
860ccaff030SJeremy L Thompson     // Empty name for conserved field (because there is only one field)
861ccaff030SJeremy L Thompson     PetscSection section;
862ccaff030SJeremy L Thompson     ierr = DMGetLocalSection(dm, &section); CHKERRQ(ierr);
863ccaff030SJeremy L Thompson     ierr = PetscSectionSetFieldName(section, 0, ""); CHKERRQ(ierr);
864ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 0, "Density");
865ccaff030SJeremy L Thompson     CHKERRQ(ierr);
866ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 1, "MomentumX");
867ccaff030SJeremy L Thompson     CHKERRQ(ierr);
868ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 2, "MomentumY");
869ccaff030SJeremy L Thompson     CHKERRQ(ierr);
870ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 3, "MomentumZ");
871ccaff030SJeremy L Thompson     CHKERRQ(ierr);
872ccaff030SJeremy L Thompson     ierr = PetscSectionSetComponentName(section, 0, 4, "EnergyDensity");
873ccaff030SJeremy L Thompson     CHKERRQ(ierr);
874ccaff030SJeremy L Thompson   }
875ccaff030SJeremy L Thompson   PetscFunctionReturn(0);
876ccaff030SJeremy L Thompson }
877ccaff030SJeremy L Thompson 
878ccaff030SJeremy L Thompson int main(int argc, char **argv) {
879ccaff030SJeremy L Thompson   PetscInt ierr;
880ccaff030SJeremy L Thompson   MPI_Comm comm;
88184d34d69SLeila Ghaffari   DM dm, dmcoord, dmviz;
882ccaff030SJeremy L Thompson   Mat interpviz;
883ccaff030SJeremy L Thompson   TS ts;
884ccaff030SJeremy L Thompson   TSAdapt adapt;
885ccaff030SJeremy L Thompson   User user;
886ccaff030SJeremy L Thompson   Units units;
8875bf10c6fSLeila Ghaffari   EulerContext ctxEulerData;
888ccaff030SJeremy L Thompson   char ceedresource[4096] = "/cpu/self";
88984d34d69SLeila Ghaffari   PetscInt localNelemVol, lnodes, gnodes, steps;
890ccaff030SJeremy L Thompson   const PetscInt ncompq = 5;
891ccaff030SJeremy L Thompson   PetscMPIInt rank;
892ccaff030SJeremy L Thompson   PetscScalar ftime;
893ccaff030SJeremy L Thompson   Vec Q, Qloc, Xloc;
894ccaff030SJeremy L Thompson   Ceed ceed;
89584d34d69SLeila Ghaffari   CeedInt numP, numQ;
896cfa64770SLeila Ghaffari   CeedVector xcorners, qdata, q0ceed;
89784d34d69SLeila Ghaffari   CeedBasis basisx, basisxc, basisq;
89884d34d69SLeila Ghaffari   CeedElemRestriction restrictx, restrictq, restrictqdi;
899cfa64770SLeila Ghaffari   CeedQFunction qf_setupVol, qf_ics, qf_rhsVol, qf_ifunctionVol;
9005bf10c6fSLeila Ghaffari   CeedQFunctionContext ctxSetup, ctxNS, ctxAdvection2d, ctxSurface, ctxEuler;
901cfa64770SLeila Ghaffari   CeedOperator op_setupVol, op_ics;
902ccaff030SJeremy L Thompson   CeedScalar Rd;
90384d34d69SLeila Ghaffari   CeedMemType memtyperequested;
904ccaff030SJeremy L Thompson   PetscScalar WpermK, Pascal, JperkgK, mpersquareds, kgpercubicm,
90516c0476cSLeila Ghaffari               kgpersquaredms, Joulepercubicm, Joule;
906ccaff030SJeremy L Thompson   problemType problemChoice;
907ccaff030SJeremy L Thompson   problemData *problem = NULL;
9081184866aSLeila Ghaffari   WindType wind_type;
909ccaff030SJeremy L Thompson   StabilizationType stab;
91084d34d69SLeila Ghaffari   PetscBool implicit;
911cb3e2689Svaleriabarra   PetscInt    viz_refine = 0;
912ccaff030SJeremy L Thompson   struct SimpleBC_ bc = {
91384d34d69SLeila Ghaffari     .nslip = {2, 2, 2},
91484d34d69SLeila Ghaffari     .slips = {{5, 6}, {3, 4}, {1, 2}}
915ccaff030SJeremy L Thompson   };
916ccaff030SJeremy L Thompson   double start, cpu_time_used;
917dc8efd83SLeila Ghaffari   // Test variables
918dc8efd83SLeila Ghaffari   PetscBool test;
919dc8efd83SLeila Ghaffari   PetscScalar testtol = 0.;
920dc8efd83SLeila Ghaffari   char filepath[PETSC_MAX_PATH_LEN];
92184d34d69SLeila Ghaffari   // Check PETSc CUDA support
92284d34d69SLeila Ghaffari   PetscBool petschavecuda, setmemtyperequest = PETSC_FALSE;
92384d34d69SLeila Ghaffari   // *INDENT-OFF*
92484d34d69SLeila Ghaffari   #ifdef PETSC_HAVE_CUDA
92584d34d69SLeila Ghaffari   petschavecuda = PETSC_TRUE;
92684d34d69SLeila Ghaffari   #else
92784d34d69SLeila Ghaffari   petschavecuda = PETSC_FALSE;
92884d34d69SLeila Ghaffari   #endif
92984d34d69SLeila Ghaffari   // *INDENT-ON*
930ccaff030SJeremy L Thompson 
931ccaff030SJeremy L Thompson   // Create the libCEED contexts
932ccaff030SJeremy L Thompson   PetscScalar meter          = 1e-2;     // 1 meter in scaled length units
933ccaff030SJeremy L Thompson   PetscScalar second         = 1e-2;     // 1 second in scaled time units
934ccaff030SJeremy L Thompson   PetscScalar kilogram       = 1e-6;     // 1 kilogram in scaled mass units
935ccaff030SJeremy L Thompson   PetscScalar Kelvin         = 1;        // 1 Kelvin in scaled temperature units
936ccaff030SJeremy L Thompson   CeedScalar theta0          = 300.;     // K
937ccaff030SJeremy L Thompson   CeedScalar thetaC          = -15.;     // K
938ccaff030SJeremy L Thompson   CeedScalar P0              = 1.e5;     // Pa
93916c0476cSLeila Ghaffari   CeedScalar E_wind          = 1.e6;     // J
940ccaff030SJeremy L Thompson   CeedScalar N               = 0.01;     // 1/s
941ccaff030SJeremy L Thompson   CeedScalar cv              = 717.;     // J/(kg K)
942ccaff030SJeremy L Thompson   CeedScalar cp              = 1004.;    // J/(kg K)
943011314a7SLeila Ghaffari   CeedScalar vortex_strength = 5.;       // -
944*c8cb9b1fSLeila Ghaffari   CeedScalar T_inlet         = 300.;     // K
945*c8cb9b1fSLeila Ghaffari   CeedScalar P_inlet         = 1.e5;     // Pa
946ccaff030SJeremy L Thompson   CeedScalar g               = 9.81;     // m/s^2
947ccaff030SJeremy L Thompson   CeedScalar lambda          = -2./3.;   // -
948ccaff030SJeremy L Thompson   CeedScalar mu              = 75.;      // Pa s, dynamic viscosity
949ccaff030SJeremy L Thompson   // mu = 75 is not physical for air, but is good for numerical stability
950ccaff030SJeremy L Thompson   CeedScalar k               = 0.02638;  // W/(m K)
951ccaff030SJeremy L Thompson   CeedScalar CtauS           = 0.;       // dimensionless
952ccaff030SJeremy L Thompson   CeedScalar strong_form     = 0.;       // [0,1]
953ccaff030SJeremy L Thompson   PetscScalar lx             = 8000.;    // m
954ccaff030SJeremy L Thompson   PetscScalar ly             = 8000.;    // m
955ccaff030SJeremy L Thompson   PetscScalar lz             = 4000.;    // m
956ccaff030SJeremy L Thompson   CeedScalar rc              = 1000.;    // m (Radius of bubble)
957ccaff030SJeremy L Thompson   PetscScalar resx           = 1000.;    // m (resolution in x)
958ccaff030SJeremy L Thompson   PetscScalar resy           = 1000.;    // m (resolution in y)
959ccaff030SJeremy L Thompson   PetscScalar resz           = 1000.;    // m (resolution in z)
960ccaff030SJeremy L Thompson   PetscInt outputfreq        = 10;       // -
961ccaff030SJeremy L Thompson   PetscInt contsteps         = 0;        // -
96284d34d69SLeila Ghaffari   PetscInt degree            = 1;        // -
96384d34d69SLeila Ghaffari   PetscInt qextra            = 2;        // -
964ea6e0f84SLeila Ghaffari   PetscInt qextraSur         = 2;        // -
965e8107fdaSLeila Ghaffari   PetscReal center[3], dc_axis[3] = {0, 0, 0}, wind[3] = {1., 0, 0},
9666bd16babSLeila Ghaffari                                     etv_mean_velocity[3] = {1., 1., 0};
967ccaff030SJeremy L Thompson   ierr = PetscInitialize(&argc, &argv, NULL, help);
968ccaff030SJeremy L Thompson   if (ierr) return ierr;
969ccaff030SJeremy L Thompson 
970ccaff030SJeremy L Thompson   // Allocate PETSc context
971ccaff030SJeremy L Thompson   ierr = PetscCalloc1(1, &user); CHKERRQ(ierr);
972ccaff030SJeremy L Thompson   ierr = PetscMalloc1(1, &units); CHKERRQ(ierr);
9735bf10c6fSLeila Ghaffari   ierr = PetscMalloc1(1, &ctxEulerData); CHKERRQ(ierr);
974ccaff030SJeremy L Thompson 
975ccaff030SJeremy L Thompson   // Parse command line options
976ccaff030SJeremy L Thompson   comm = PETSC_COMM_WORLD;
977ccaff030SJeremy L Thompson   ierr = PetscOptionsBegin(comm, NULL, "Navier-Stokes in PETSc with libCEED",
978ccaff030SJeremy L Thompson                            NULL); CHKERRQ(ierr);
979ccaff030SJeremy L Thompson   ierr = PetscOptionsString("-ceed", "CEED resource specifier",
980ccaff030SJeremy L Thompson                             NULL, ceedresource, ceedresource,
981ccaff030SJeremy L Thompson                             sizeof(ceedresource), NULL); CHKERRQ(ierr);
982dc8efd83SLeila Ghaffari   ierr = PetscOptionsBool("-test", "Run in test mode",
983dc8efd83SLeila Ghaffari                           NULL, test=PETSC_FALSE, &test, NULL); CHKERRQ(ierr);
984dc8efd83SLeila Ghaffari   ierr = PetscOptionsScalar("-compare_final_state_atol",
985dc8efd83SLeila Ghaffari                             "Test absolute tolerance",
986dc8efd83SLeila Ghaffari                             NULL, testtol, &testtol, NULL); CHKERRQ(ierr);
987dc8efd83SLeila Ghaffari   ierr = PetscOptionsString("-compare_final_state_filename", "Test filename",
988dc8efd83SLeila Ghaffari                             NULL, filepath, filepath,
989dc8efd83SLeila Ghaffari                             sizeof(filepath), NULL); CHKERRQ(ierr);
990ccaff030SJeremy L Thompson   problemChoice = NS_DENSITY_CURRENT;
991ccaff030SJeremy L Thompson   ierr = PetscOptionsEnum("-problem", "Problem to solve", NULL,
992ccaff030SJeremy L Thompson                           problemTypes, (PetscEnum)problemChoice,
993ccaff030SJeremy L Thompson                           (PetscEnum *)&problemChoice, NULL); CHKERRQ(ierr);
994ccaff030SJeremy L Thompson   problem = &problemOptions[problemChoice];
9951184866aSLeila Ghaffari   ierr = PetscOptionsEnum("-problem_advection_wind", "Wind type in Advection",
996f259b054Svaleriabarra                           NULL, WindTypes,
997f259b054Svaleriabarra                           (PetscEnum)(wind_type = ADVECTION_WIND_ROTATION),
9981184866aSLeila Ghaffari                           (PetscEnum *)&wind_type, NULL); CHKERRQ(ierr);
9991184866aSLeila Ghaffari   PetscInt n = problem->dim;
100082c09b01SLeila Ghaffari   PetscBool userWind;
1001ebb4b9bdSLeila Ghaffari   ierr = PetscOptionsRealArray("-problem_advection_wind_translation",
1002ebb4b9bdSLeila Ghaffari                                "Constant wind vector",
100382c09b01SLeila Ghaffari                                NULL, wind, &n, &userWind); CHKERRQ(ierr);
100482c09b01SLeila Ghaffari   if (wind_type == ADVECTION_WIND_ROTATION && userWind) {
1005ebb4b9bdSLeila Ghaffari     ierr = PetscPrintf(comm,
1006ebb4b9bdSLeila Ghaffari                        "Warning! Use -problem_advection_wind_translation only with -problem_advection_wind translation\n");
100782c09b01SLeila Ghaffari     CHKERRQ(ierr);
10081184866aSLeila Ghaffari   }
1009b9c3f7b3SLeila Ghaffari   if (wind_type == ADVECTION_WIND_TRANSLATION
1010b9c3f7b3SLeila Ghaffari       && (problemChoice == NS_DENSITY_CURRENT ||
1011b9c3f7b3SLeila Ghaffari           problemChoice == NS_EULER_VORTEX)) {
101267babd9cSLeila Ghaffari     SETERRQ(comm, PETSC_ERR_ARG_INCOMP,
1013b9c3f7b3SLeila Ghaffari             "-problem_advection_wind translation is not defined for -problem density_current or -problem euler_vortex");
101467babd9cSLeila Ghaffari   }
1015e8107fdaSLeila Ghaffari   ierr = PetscOptionsRealArray("-problem_euler_mean_velocity",
1016e8107fdaSLeila Ghaffari                                "Mean velocity vector",
1017e8107fdaSLeila Ghaffari                                NULL, etv_mean_velocity, &n, NULL);
1018e8107fdaSLeila Ghaffari   CHKERRQ(ierr);
1019ccaff030SJeremy L Thompson   ierr = PetscOptionsEnum("-stab", "Stabilization method", NULL,
1020ccaff030SJeremy L Thompson                           StabilizationTypes, (PetscEnum)(stab = STAB_NONE),
1021ccaff030SJeremy L Thompson                           (PetscEnum *)&stab, NULL); CHKERRQ(ierr);
1022ccaff030SJeremy L Thompson   ierr = PetscOptionsBool("-implicit", "Use implicit (IFunction) formulation",
1023ccaff030SJeremy L Thompson                           NULL, implicit=PETSC_FALSE, &implicit, NULL);
1024ccaff030SJeremy L Thompson   CHKERRQ(ierr);
102584d34d69SLeila Ghaffari   if (!implicit && stab != STAB_NONE) {
102684d34d69SLeila Ghaffari     ierr = PetscPrintf(comm, "Warning! Use -stab only with -implicit\n");
102784d34d69SLeila Ghaffari     CHKERRQ(ierr);
102884d34d69SLeila Ghaffari   }
1029ccaff030SJeremy L Thompson   {
10307573aee6SJed Brown     PetscInt len;
10317573aee6SJed Brown     PetscBool flg;
1032ccaff030SJeremy L Thompson     ierr = PetscOptionsIntArray("-bc_wall",
1033ccaff030SJeremy L Thompson                                 "Use wall boundary conditions on this list of faces",
1034ccaff030SJeremy L Thompson                                 NULL, bc.walls,
1035ccaff030SJeremy L Thompson                                 (len = sizeof(bc.walls) / sizeof(bc.walls[0]),
1036ccaff030SJeremy L Thompson                                  &len), &flg); CHKERRQ(ierr);
10377573aee6SJed Brown     if (flg) {
10387573aee6SJed Brown       bc.nwall = len;
10397573aee6SJed Brown       // Using a no-slip wall disables automatic slip walls (they must be set explicitly)
10407573aee6SJed Brown       bc.nslip[0] = bc.nslip[1] = bc.nslip[2] = 0;
10417573aee6SJed Brown     }
1042ccaff030SJeremy L Thompson     for (PetscInt j=0; j<3; j++) {
1043ccaff030SJeremy L Thompson       const char *flags[3] = {"-bc_slip_x", "-bc_slip_y", "-bc_slip_z"};
1044ccaff030SJeremy L Thompson       ierr = PetscOptionsIntArray(flags[j],
1045ccaff030SJeremy L Thompson                                   "Use slip boundary conditions on this list of faces",
1046ccaff030SJeremy L Thompson                                   NULL, bc.slips[j],
1047ccaff030SJeremy L Thompson                                   (len = sizeof(bc.slips[j]) / sizeof(bc.slips[j][0]),
1048ccaff030SJeremy L Thompson                                    &len), &flg);
1049ccaff030SJeremy L Thompson       CHKERRQ(ierr);
105084d34d69SLeila Ghaffari       if (flg) {
105184d34d69SLeila Ghaffari         bc.nslip[j] = len;
105284d34d69SLeila Ghaffari         bc.userbc = PETSC_TRUE;
105384d34d69SLeila Ghaffari       }
1054ccaff030SJeremy L Thompson     }
1055ccaff030SJeremy L Thompson   }
1056cb3e2689Svaleriabarra   ierr = PetscOptionsInt("-viz_refine",
1057cb3e2689Svaleriabarra                          "Regular refinement levels for visualization",
1058cb3e2689Svaleriabarra                          NULL, viz_refine, &viz_refine, NULL);
1059ccaff030SJeremy L Thompson   CHKERRQ(ierr);
1060ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_meter", "1 meter in scaled length units",
1061ccaff030SJeremy L Thompson                             NULL, meter, &meter, NULL); CHKERRQ(ierr);
1062ccaff030SJeremy L Thompson   meter = fabs(meter);
1063ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_second","1 second in scaled time units",
1064ccaff030SJeremy L Thompson                             NULL, second, &second, NULL); CHKERRQ(ierr);
1065ccaff030SJeremy L Thompson   second = fabs(second);
1066ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_kilogram","1 kilogram in scaled mass units",
1067ccaff030SJeremy L Thompson                             NULL, kilogram, &kilogram, NULL); CHKERRQ(ierr);
1068ccaff030SJeremy L Thompson   kilogram = fabs(kilogram);
1069ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-units_Kelvin",
1070ccaff030SJeremy L Thompson                             "1 Kelvin in scaled temperature units",
1071ccaff030SJeremy L Thompson                             NULL, Kelvin, &Kelvin, NULL); CHKERRQ(ierr);
1072ccaff030SJeremy L Thompson   Kelvin = fabs(Kelvin);
1073ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-theta0", "Reference potential temperature",
1074ccaff030SJeremy L Thompson                             NULL, theta0, &theta0, NULL); CHKERRQ(ierr);
1075ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-thetaC", "Perturbation of potential temperature",
1076ccaff030SJeremy L Thompson                             NULL, thetaC, &thetaC, NULL); CHKERRQ(ierr);
1077ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-P0", "Atmospheric pressure",
1078ccaff030SJeremy L Thompson                             NULL, P0, &P0, NULL); CHKERRQ(ierr);
107916c0476cSLeila Ghaffari   ierr = PetscOptionsScalar("-E_wind", "Total energy of inflow wind",
108016c0476cSLeila Ghaffari                             NULL, E_wind, &E_wind, NULL); CHKERRQ(ierr);
1081ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-N", "Brunt-Vaisala frequency",
1082ccaff030SJeremy L Thompson                             NULL, N, &N, NULL); CHKERRQ(ierr);
1083ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-cv", "Heat capacity at constant volume",
1084ccaff030SJeremy L Thompson                             NULL, cv, &cv, NULL); CHKERRQ(ierr);
1085ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-cp", "Heat capacity at constant pressure",
1086ccaff030SJeremy L Thompson                             NULL, cp, &cp, NULL); CHKERRQ(ierr);
1087011314a7SLeila Ghaffari   PetscBool userVortex;
1088011314a7SLeila Ghaffari   ierr = PetscOptionsScalar("-vortex_strength", "Strength of Vortex",
1089011314a7SLeila Ghaffari                             NULL, vortex_strength, &vortex_strength, &userVortex);
1090011314a7SLeila Ghaffari   CHKERRQ(ierr);
1091011314a7SLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userVortex) {
1092011314a7SLeila Ghaffari     ierr = PetscPrintf(comm,
1093011314a7SLeila Ghaffari                        "Warning! Use -vortex_strength only with -problem euler_vortex\n");
1094011314a7SLeila Ghaffari     CHKERRQ(ierr);
1095011314a7SLeila Ghaffari   }
1096*c8cb9b1fSLeila Ghaffari   PetscBool userTinlet;
1097*c8cb9b1fSLeila Ghaffari   ierr = PetscOptionsScalar("-T_inlet", "Incoming Temperature",
1098*c8cb9b1fSLeila Ghaffari                             NULL, T_inlet, &T_inlet, &userTinlet);
1099*c8cb9b1fSLeila Ghaffari   CHKERRQ(ierr);
1100*c8cb9b1fSLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userTinlet) {
1101*c8cb9b1fSLeila Ghaffari     ierr = PetscPrintf(comm,
1102*c8cb9b1fSLeila Ghaffari                        "Warning! Use -T_inlet only with -problem euler_vortex\n");
1103*c8cb9b1fSLeila Ghaffari     CHKERRQ(ierr);
1104*c8cb9b1fSLeila Ghaffari   }
1105*c8cb9b1fSLeila Ghaffari   PetscBool userPinlet;
1106*c8cb9b1fSLeila Ghaffari   ierr = PetscOptionsScalar("-P_inlet", "Incoming Pressure",
1107*c8cb9b1fSLeila Ghaffari                             NULL, P_inlet, &P_inlet, &userPinlet);
1108*c8cb9b1fSLeila Ghaffari   CHKERRQ(ierr);
1109*c8cb9b1fSLeila Ghaffari   if (problemChoice != NS_EULER_VORTEX && userPinlet) {
1110*c8cb9b1fSLeila Ghaffari     ierr = PetscPrintf(comm,
1111*c8cb9b1fSLeila Ghaffari                        "Warning! Use -P_inlet only with -problem euler_vortex\n");
1112*c8cb9b1fSLeila Ghaffari     CHKERRQ(ierr);
1113*c8cb9b1fSLeila Ghaffari   }
1114ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-g", "Gravitational acceleration",
1115ccaff030SJeremy L Thompson                             NULL, g, &g, NULL); CHKERRQ(ierr);
1116ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lambda",
1117ccaff030SJeremy L Thompson                             "Stokes hypothesis second viscosity coefficient",
1118ccaff030SJeremy L Thompson                             NULL, lambda, &lambda, NULL); CHKERRQ(ierr);
1119ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-mu", "Shear dynamic viscosity coefficient",
1120ccaff030SJeremy L Thompson                             NULL, mu, &mu, NULL); CHKERRQ(ierr);
1121ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-k", "Thermal conductivity",
1122ccaff030SJeremy L Thompson                             NULL, k, &k, NULL); CHKERRQ(ierr);
1123ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-CtauS",
1124ccaff030SJeremy L Thompson                             "Scale coefficient for tau (nondimensional)",
1125ccaff030SJeremy L Thompson                             NULL, CtauS, &CtauS, NULL); CHKERRQ(ierr);
112684d34d69SLeila Ghaffari   if (stab == STAB_NONE && CtauS != 0) {
112784d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
112884d34d69SLeila Ghaffari                        "Warning! Use -CtauS only with -stab su or -stab supg\n");
112984d34d69SLeila Ghaffari     CHKERRQ(ierr);
113084d34d69SLeila Ghaffari   }
1131ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-strong_form",
1132ccaff030SJeremy L Thompson                             "Strong (1) or weak/integrated by parts (0) advection residual",
1133ccaff030SJeremy L Thompson                             NULL, strong_form, &strong_form, NULL);
1134ccaff030SJeremy L Thompson   CHKERRQ(ierr);
113584d34d69SLeila Ghaffari   if (problemChoice == NS_DENSITY_CURRENT && (CtauS != 0 || strong_form != 0)) {
113684d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
113784d34d69SLeila Ghaffari                        "Warning! Problem density_current does not support -CtauS or -strong_form\n");
113884d34d69SLeila Ghaffari     CHKERRQ(ierr);
113984d34d69SLeila Ghaffari   }
1140ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lx", "Length scale in x direction",
1141ccaff030SJeremy L Thompson                             NULL, lx, &lx, NULL); CHKERRQ(ierr);
1142ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-ly", "Length scale in y direction",
1143ccaff030SJeremy L Thompson                             NULL, ly, &ly, NULL); CHKERRQ(ierr);
1144ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-lz", "Length scale in z direction",
1145ccaff030SJeremy L Thompson                             NULL, lz, &lz, NULL); CHKERRQ(ierr);
1146ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-rc", "Characteristic radius of thermal bubble",
1147ccaff030SJeremy L Thompson                             NULL, rc, &rc, NULL); CHKERRQ(ierr);
1148ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resx","Target resolution in x",
1149ccaff030SJeremy L Thompson                             NULL, resx, &resx, NULL); CHKERRQ(ierr);
1150ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resy","Target resolution in y",
1151ccaff030SJeremy L Thompson                             NULL, resy, &resy, NULL); CHKERRQ(ierr);
1152ccaff030SJeremy L Thompson   ierr = PetscOptionsScalar("-resz","Target resolution in z",
1153ccaff030SJeremy L Thompson                             NULL, resz, &resz, NULL); CHKERRQ(ierr);
115482c09b01SLeila Ghaffari   n = problem->dim;
1155ccaff030SJeremy L Thompson   center[0] = 0.5 * lx;
1156ccaff030SJeremy L Thompson   center[1] = 0.5 * ly;
1157ccaff030SJeremy L Thompson   center[2] = 0.5 * lz;
1158ccaff030SJeremy L Thompson   ierr = PetscOptionsRealArray("-center", "Location of bubble center",
1159ccaff030SJeremy L Thompson                                NULL, center, &n, NULL); CHKERRQ(ierr);
1160ccaff030SJeremy L Thompson   n = problem->dim;
1161ccaff030SJeremy L Thompson   ierr = PetscOptionsRealArray("-dc_axis",
1162ccaff030SJeremy L Thompson                                "Axis of density current cylindrical anomaly, or {0,0,0} for spherically symmetric",
1163ccaff030SJeremy L Thompson                                NULL, dc_axis, &n, NULL); CHKERRQ(ierr);
1164ccaff030SJeremy L Thompson   {
1165ccaff030SJeremy L Thompson     PetscReal norm = PetscSqrtReal(PetscSqr(dc_axis[0]) +
1166ccaff030SJeremy L Thompson                                    PetscSqr(dc_axis[1]) + PetscSqr(dc_axis[2]));
1167ccaff030SJeremy L Thompson     if (norm > 0) {
1168ccaff030SJeremy L Thompson       for (int i=0; i<3; i++) dc_axis[i] /= norm;
1169ccaff030SJeremy L Thompson     }
1170ccaff030SJeremy L Thompson   }
1171ccaff030SJeremy L Thompson   ierr = PetscOptionsInt("-output_freq",
1172ccaff030SJeremy L Thompson                          "Frequency of output, in number of steps",
1173ccaff030SJeremy L Thompson                          NULL, outputfreq, &outputfreq, NULL); CHKERRQ(ierr);
1174ccaff030SJeremy L Thompson   ierr = PetscOptionsInt("-continue", "Continue from previous solution",
1175ccaff030SJeremy L Thompson                          NULL, contsteps, &contsteps, NULL); CHKERRQ(ierr);
117684d34d69SLeila Ghaffari   ierr = PetscOptionsInt("-degree", "Polynomial degree of finite elements",
117784d34d69SLeila Ghaffari                          NULL, degree, &degree, NULL); CHKERRQ(ierr);
117884d34d69SLeila Ghaffari   ierr = PetscOptionsInt("-qextra", "Number of extra quadrature points",
117984d34d69SLeila Ghaffari                          NULL, qextra, &qextra, NULL); CHKERRQ(ierr);
118081f92cf0SLeila Ghaffari   PetscBool userQextraSur;
1181ebb4b9bdSLeila Ghaffari   ierr = PetscOptionsInt("-qextra_boundary",
1182ebb4b9bdSLeila Ghaffari                          "Number of extra quadrature points on in/outflow faces",
1183f259b054Svaleriabarra                          NULL, qextraSur, &qextraSur, &userQextraSur);
1184f259b054Svaleriabarra   CHKERRQ(ierr);
1185ebb4b9bdSLeila Ghaffari   if ((wind_type == ADVECTION_WIND_ROTATION
1186ebb4b9bdSLeila Ghaffari        || problemChoice == NS_DENSITY_CURRENT) && userQextraSur) {
1187ebb4b9bdSLeila Ghaffari     ierr = PetscPrintf(comm,
1188ebb4b9bdSLeila Ghaffari                        "Warning! Use -qextra_boundary only with -problem_advection_wind translation\n");
118981f92cf0SLeila Ghaffari     CHKERRQ(ierr);
119081f92cf0SLeila Ghaffari   }
1191d99129b9SLeila Ghaffari   ierr = PetscStrncpy(user->outputdir, ".", 2); CHKERRQ(ierr);
1192d99129b9SLeila Ghaffari   ierr = PetscOptionsString("-output_dir", "Output directory",
1193d99129b9SLeila Ghaffari                             NULL, user->outputdir, user->outputdir,
1194d99129b9SLeila Ghaffari                             sizeof(user->outputdir), NULL); CHKERRQ(ierr);
119584d34d69SLeila Ghaffari   memtyperequested = petschavecuda ? CEED_MEM_DEVICE : CEED_MEM_HOST;
119684d34d69SLeila Ghaffari   ierr = PetscOptionsEnum("-memtype",
119784d34d69SLeila Ghaffari                           "CEED MemType requested", NULL,
119884d34d69SLeila Ghaffari                           memTypes, (PetscEnum)memtyperequested,
119984d34d69SLeila Ghaffari                           (PetscEnum *)&memtyperequested, &setmemtyperequest);
120084d34d69SLeila Ghaffari   CHKERRQ(ierr);
1201ccaff030SJeremy L Thompson   ierr = PetscOptionsEnd(); CHKERRQ(ierr);
1202ccaff030SJeremy L Thompson 
1203ccaff030SJeremy L Thompson   // Define derived units
1204ccaff030SJeremy L Thompson   Pascal = kilogram / (meter * PetscSqr(second));
1205ccaff030SJeremy L Thompson   JperkgK =  PetscSqr(meter) / (PetscSqr(second) * Kelvin);
1206ccaff030SJeremy L Thompson   mpersquareds = meter / PetscSqr(second);
1207ccaff030SJeremy L Thompson   WpermK = kilogram * meter / (pow(second,3) * Kelvin);
1208ccaff030SJeremy L Thompson   kgpercubicm = kilogram / pow(meter,3);
1209ccaff030SJeremy L Thompson   kgpersquaredms = kilogram / (PetscSqr(meter) * second);
1210ccaff030SJeremy L Thompson   Joulepercubicm = kilogram / (meter * PetscSqr(second));
121116c0476cSLeila Ghaffari   Joule = kilogram * PetscSqr(meter) / PetscSqr(second);
1212ccaff030SJeremy L Thompson 
1213ccaff030SJeremy L Thompson   // Scale variables to desired units
1214ccaff030SJeremy L Thompson   theta0 *= Kelvin;
1215ccaff030SJeremy L Thompson   thetaC *= Kelvin;
1216ccaff030SJeremy L Thompson   P0 *= Pascal;
121716c0476cSLeila Ghaffari   E_wind *= Joule;
1218ccaff030SJeremy L Thompson   N *= (1./second);
1219ccaff030SJeremy L Thompson   cv *= JperkgK;
1220ccaff030SJeremy L Thompson   cp *= JperkgK;
1221ccaff030SJeremy L Thompson   Rd = cp - cv;
1222*c8cb9b1fSLeila Ghaffari   T_inlet *= Kelvin;
1223*c8cb9b1fSLeila Ghaffari   P_inlet *= Pascal;
1224ccaff030SJeremy L Thompson   g *= mpersquareds;
1225ccaff030SJeremy L Thompson   mu *= Pascal * second;
1226ccaff030SJeremy L Thompson   k *= WpermK;
1227ccaff030SJeremy L Thompson   lx = fabs(lx) * meter;
1228ccaff030SJeremy L Thompson   ly = fabs(ly) * meter;
1229ccaff030SJeremy L Thompson   lz = fabs(lz) * meter;
1230ccaff030SJeremy L Thompson   rc = fabs(rc) * meter;
1231ccaff030SJeremy L Thompson   resx = fabs(resx) * meter;
1232ccaff030SJeremy L Thompson   resy = fabs(resy) * meter;
1233ccaff030SJeremy L Thompson   resz = fabs(resz) * meter;
1234ccaff030SJeremy L Thompson   for (int i=0; i<3; i++) center[i] *= meter;
1235ccaff030SJeremy L Thompson 
1236ccaff030SJeremy L Thompson   const CeedInt dim = problem->dim, ncompx = problem->dim,
1237cfa64770SLeila Ghaffari                 qdatasizeVol = problem->qdatasizeVol;
1238ccaff030SJeremy L Thompson   // Set up the libCEED context
1239777ff853SJeremy L Thompson   struct SetupContext_ ctxSetupData = {
1240ccaff030SJeremy L Thompson     .theta0 = theta0,
1241ccaff030SJeremy L Thompson     .thetaC = thetaC,
1242ccaff030SJeremy L Thompson     .P0 = P0,
1243ccaff030SJeremy L Thompson     .N = N,
1244ccaff030SJeremy L Thompson     .cv = cv,
1245ccaff030SJeremy L Thompson     .cp = cp,
1246ccaff030SJeremy L Thompson     .Rd = Rd,
1247ccaff030SJeremy L Thompson     .g = g,
1248ccaff030SJeremy L Thompson     .rc = rc,
1249ccaff030SJeremy L Thompson     .lx = lx,
1250ccaff030SJeremy L Thompson     .ly = ly,
1251ccaff030SJeremy L Thompson     .lz = lz,
1252ccaff030SJeremy L Thompson     .center[0] = center[0],
1253ccaff030SJeremy L Thompson     .center[1] = center[1],
1254ccaff030SJeremy L Thompson     .center[2] = center[2],
1255ccaff030SJeremy L Thompson     .dc_axis[0] = dc_axis[0],
1256ccaff030SJeremy L Thompson     .dc_axis[1] = dc_axis[1],
1257ccaff030SJeremy L Thompson     .dc_axis[2] = dc_axis[2],
12581184866aSLeila Ghaffari     .wind[0] = wind[0],
12591184866aSLeila Ghaffari     .wind[1] = wind[1],
12601184866aSLeila Ghaffari     .wind[2] = wind[2],
12613a9f53b0SLeila Ghaffari     .time = 0,
1262011314a7SLeila Ghaffari     .vortex_strength = vortex_strength,
12631184866aSLeila Ghaffari     .wind_type = wind_type,
1264ccaff030SJeremy L Thompson   };
1265ccaff030SJeremy L Thompson 
126684d34d69SLeila Ghaffari   // Create the mesh
1267ccaff030SJeremy L Thompson   {
1268ccaff030SJeremy L Thompson     const PetscReal scale[3] = {lx, ly, lz};
1269ccaff030SJeremy L Thompson     ierr = DMPlexCreateBoxMesh(comm, dim, PETSC_FALSE, NULL, NULL, scale,
12702561194eSLeila Ghaffari                                NULL, PETSC_TRUE, &dm);
1271ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1272ccaff030SJeremy L Thompson   }
127384d34d69SLeila Ghaffari 
127484d34d69SLeila Ghaffari   // Distribute the mesh over processes
127584d34d69SLeila Ghaffari   {
1276ccaff030SJeremy L Thompson     DM               dmDist = NULL;
1277ccaff030SJeremy L Thompson     PetscPartitioner part;
1278ccaff030SJeremy L Thompson 
1279ccaff030SJeremy L Thompson     ierr = DMPlexGetPartitioner(dm, &part); CHKERRQ(ierr);
1280ccaff030SJeremy L Thompson     ierr = PetscPartitionerSetFromOptions(part); CHKERRQ(ierr);
1281ccaff030SJeremy L Thompson     ierr = DMPlexDistribute(dm, 0, NULL, &dmDist); CHKERRQ(ierr);
1282ccaff030SJeremy L Thompson     if (dmDist) {
1283ccaff030SJeremy L Thompson       ierr = DMDestroy(&dm); CHKERRQ(ierr);
1284ccaff030SJeremy L Thompson       dm  = dmDist;
1285ccaff030SJeremy L Thompson     }
1286ccaff030SJeremy L Thompson   }
1287ccaff030SJeremy L Thompson   ierr = DMViewFromOptions(dm, NULL, "-dm_view"); CHKERRQ(ierr);
1288ccaff030SJeremy L Thompson 
128984d34d69SLeila Ghaffari   // Setup DM
1290ccaff030SJeremy L Thompson   ierr = DMLocalizeCoordinates(dm); CHKERRQ(ierr);
1291ccaff030SJeremy L Thompson   ierr = DMSetFromOptions(dm); CHKERRQ(ierr);
12925bf10c6fSLeila Ghaffari   ierr = SetUpDM(dm, problem, degree, &bc, &ctxSetupData, ctxEulerData);
12935bf10c6fSLeila Ghaffari   CHKERRQ(ierr);
129484d34d69SLeila Ghaffari 
129584d34d69SLeila Ghaffari   // Refine DM for high-order viz
1296ccaff030SJeremy L Thompson   dmviz = NULL;
1297ccaff030SJeremy L Thompson   interpviz = NULL;
1298ccaff030SJeremy L Thompson   if (viz_refine) {
1299ff6701fcSJed Brown     DM dmhierarchy[viz_refine+1];
1300ff6701fcSJed Brown 
1301ccaff030SJeremy L Thompson     ierr = DMPlexSetRefinementUniform(dm, PETSC_TRUE); CHKERRQ(ierr);
1302ff6701fcSJed Brown     dmhierarchy[0] = dm;
130384d34d69SLeila Ghaffari     for (PetscInt i = 0, d = degree; i < viz_refine; i++) {
1304ff6701fcSJed Brown       Mat interp_next;
1305ff6701fcSJed Brown 
1306ff6701fcSJed Brown       ierr = DMRefine(dmhierarchy[i], MPI_COMM_NULL, &dmhierarchy[i+1]);
1307ccaff030SJeremy L Thompson       CHKERRQ(ierr);
1308bc6a41f7SJed Brown       ierr = DMClearDS(dmhierarchy[i+1]); CHKERRQ(ierr);
1309bc6a41f7SJed Brown       ierr = DMClearFields(dmhierarchy[i+1]); CHKERRQ(ierr);
1310ff6701fcSJed Brown       ierr = DMSetCoarseDM(dmhierarchy[i+1], dmhierarchy[i]); CHKERRQ(ierr);
1311ff6701fcSJed Brown       d = (d + 1) / 2;
1312ff6701fcSJed Brown       if (i + 1 == viz_refine) d = 1;
13135bf10c6fSLeila Ghaffari       ierr = SetUpDM(dmhierarchy[i+1], problem, d, &bc, &ctxSetupData,
13145bf10c6fSLeila Ghaffari                      ctxEulerData); CHKERRQ(ierr);
1315ff6701fcSJed Brown       ierr = DMCreateInterpolation(dmhierarchy[i], dmhierarchy[i+1],
1316ff6701fcSJed Brown                                    &interp_next, NULL); CHKERRQ(ierr);
1317ff6701fcSJed Brown       if (!i) interpviz = interp_next;
1318ff6701fcSJed Brown       else {
1319ff6701fcSJed Brown         Mat C;
1320ff6701fcSJed Brown         ierr = MatMatMult(interp_next, interpviz, MAT_INITIAL_MATRIX,
1321ff6701fcSJed Brown                           PETSC_DECIDE, &C); CHKERRQ(ierr);
1322ff6701fcSJed Brown         ierr = MatDestroy(&interp_next); CHKERRQ(ierr);
1323ff6701fcSJed Brown         ierr = MatDestroy(&interpviz); CHKERRQ(ierr);
1324ff6701fcSJed Brown         interpviz = C;
1325ff6701fcSJed Brown       }
1326ff6701fcSJed Brown     }
1327cb3e2689Svaleriabarra     for (PetscInt i=1; i<viz_refine; i++) {
1328ff6701fcSJed Brown       ierr = DMDestroy(&dmhierarchy[i]); CHKERRQ(ierr);
1329cb3e2689Svaleriabarra     }
1330ff6701fcSJed Brown     dmviz = dmhierarchy[viz_refine];
1331ccaff030SJeremy L Thompson   }
1332ccaff030SJeremy L Thompson   ierr = DMCreateGlobalVector(dm, &Q); CHKERRQ(ierr);
1333ccaff030SJeremy L Thompson   ierr = DMGetLocalVector(dm, &Qloc); CHKERRQ(ierr);
1334ccaff030SJeremy L Thompson   ierr = VecGetSize(Qloc, &lnodes); CHKERRQ(ierr);
1335ccaff030SJeremy L Thompson   lnodes /= ncompq;
1336ccaff030SJeremy L Thompson 
133784d34d69SLeila Ghaffari   // Initialize CEED
133884d34d69SLeila Ghaffari   CeedInit(ceedresource, &ceed);
133984d34d69SLeila Ghaffari   // Set memtype
134084d34d69SLeila Ghaffari   CeedMemType memtypebackend;
134184d34d69SLeila Ghaffari   CeedGetPreferredMemType(ceed, &memtypebackend);
134284d34d69SLeila Ghaffari   // Check memtype compatibility
134384d34d69SLeila Ghaffari   if (!setmemtyperequest)
134484d34d69SLeila Ghaffari     memtyperequested = memtypebackend;
134584d34d69SLeila Ghaffari   else if (!petschavecuda && memtyperequested == CEED_MEM_DEVICE)
134684d34d69SLeila Ghaffari     SETERRQ1(PETSC_COMM_WORLD, PETSC_ERR_SUP_SYS,
134784d34d69SLeila Ghaffari              "PETSc was not built with CUDA. "
134884d34d69SLeila Ghaffari              "Requested MemType CEED_MEM_DEVICE is not supported.", NULL);
134984d34d69SLeila Ghaffari 
135084d34d69SLeila Ghaffari   // Set number of 1D nodes and quadrature points
135184d34d69SLeila Ghaffari   numP = degree + 1;
135284d34d69SLeila Ghaffari   numQ = numP + qextra;
135384d34d69SLeila Ghaffari 
135484d34d69SLeila Ghaffari   // Print summary
1355dc8efd83SLeila Ghaffari   if (!test) {
1356ccaff030SJeremy L Thompson     CeedInt gdofs, odofs;
1357ccaff030SJeremy L Thompson     int comm_size;
1358ccaff030SJeremy L Thompson     char box_faces_str[PETSC_MAX_PATH_LEN] = "NONE";
1359ccaff030SJeremy L Thompson     ierr = VecGetSize(Q, &gdofs); CHKERRQ(ierr);
1360ccaff030SJeremy L Thompson     ierr = VecGetLocalSize(Q, &odofs); CHKERRQ(ierr);
136184d34d69SLeila Ghaffari     gnodes = gdofs/ncompq;
1362ccaff030SJeremy L Thompson     ierr = MPI_Comm_size(comm, &comm_size); CHKERRQ(ierr);
1363ccaff030SJeremy L Thompson     ierr = PetscOptionsGetString(NULL, NULL, "-dm_plex_box_faces", box_faces_str,
1364ccaff030SJeremy L Thompson                                  sizeof(box_faces_str), NULL); CHKERRQ(ierr);
136584d34d69SLeila Ghaffari     const char *usedresource;
136684d34d69SLeila Ghaffari     CeedGetResource(ceed, &usedresource);
1367ccaff030SJeremy L Thompson 
136884d34d69SLeila Ghaffari     ierr = PetscPrintf(comm,
136984d34d69SLeila Ghaffari                        "\n-- Navier-Stokes solver - libCEED + PETSc --\n"
137084d34d69SLeila Ghaffari                        "  rank(s)                              : %d\n"
137184d34d69SLeila Ghaffari                        "  Problem:\n"
137284d34d69SLeila Ghaffari                        "    Problem Name                       : %s\n"
137384d34d69SLeila Ghaffari                        "    Stabilization                      : %s\n"
137484d34d69SLeila Ghaffari                        "  PETSc:\n"
137584d34d69SLeila Ghaffari                        "    Box Faces                          : %s\n"
137684d34d69SLeila Ghaffari                        "  libCEED:\n"
137784d34d69SLeila Ghaffari                        "    libCEED Backend                    : %s\n"
137884d34d69SLeila Ghaffari                        "    libCEED Backend MemType            : %s\n"
137984d34d69SLeila Ghaffari                        "    libCEED User Requested MemType     : %s\n"
138084d34d69SLeila Ghaffari                        "  Mesh:\n"
138184d34d69SLeila Ghaffari                        "    Number of 1D Basis Nodes (P)       : %d\n"
138284d34d69SLeila Ghaffari                        "    Number of 1D Quadrature Points (Q) : %d\n"
138384d34d69SLeila Ghaffari                        "    Global DoFs                        : %D\n"
138484d34d69SLeila Ghaffari                        "    Owned DoFs                         : %D\n"
138584d34d69SLeila Ghaffari                        "    DoFs per node                      : %D\n"
138684d34d69SLeila Ghaffari                        "    Global nodes                       : %D\n"
138784d34d69SLeila Ghaffari                        "    Owned nodes                        : %D\n",
138884d34d69SLeila Ghaffari                        comm_size, problemTypes[problemChoice],
138984d34d69SLeila Ghaffari                        StabilizationTypes[stab], box_faces_str, usedresource,
139084d34d69SLeila Ghaffari                        CeedMemTypes[memtypebackend],
139184d34d69SLeila Ghaffari                        (setmemtyperequest) ?
139284d34d69SLeila Ghaffari                        CeedMemTypes[memtyperequested] : "none",
139384d34d69SLeila Ghaffari                        numP, numQ, gdofs, odofs, ncompq, gnodes, lnodes);
139484d34d69SLeila Ghaffari     CHKERRQ(ierr);
13950c6c0b13SLeila Ghaffari   }
13960c6c0b13SLeila Ghaffari 
1397ccaff030SJeremy L Thompson   // Set up global mass vector
1398ccaff030SJeremy L Thompson   ierr = VecDuplicate(Q, &user->M); CHKERRQ(ierr);
1399ccaff030SJeremy L Thompson 
140084d34d69SLeila Ghaffari   // Set up libCEED
1401ccaff030SJeremy L Thompson   // CEED Bases
1402ccaff030SJeremy L Thompson   CeedInit(ceedresource, &ceed);
140384d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompq, numP, numQ, CEED_GAUSS,
140484d34d69SLeila Ghaffari                                   &basisq);
140584d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompx, 2, numQ, CEED_GAUSS,
140684d34d69SLeila Ghaffari                                   &basisx);
140784d34d69SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, ncompx, 2, numP,
140884d34d69SLeila Ghaffari                                   CEED_GAUSS_LOBATTO, &basisxc);
1409ccaff030SJeremy L Thompson   ierr = DMGetCoordinateDM(dm, &dmcoord); CHKERRQ(ierr);
1410ccaff030SJeremy L Thompson   ierr = DMPlexSetClosurePermutationTensor(dmcoord, PETSC_DETERMINE, NULL);
1411ccaff030SJeremy L Thompson   CHKERRQ(ierr);
1412ccaff030SJeremy L Thompson 
1413ccaff030SJeremy L Thompson   // CEED Restrictions
14141e150236SLeila Ghaffari   ierr = GetRestrictionForDomain(ceed, dm, 0, 0, 0, numP, numQ,
141584d34d69SLeila Ghaffari                                  qdatasizeVol, &restrictq, &restrictx,
141684d34d69SLeila Ghaffari                                  &restrictqdi); CHKERRQ(ierr);
1417ccaff030SJeremy L Thompson 
1418ccaff030SJeremy L Thompson   ierr = DMGetCoordinatesLocal(dm, &Xloc); CHKERRQ(ierr);
1419ccaff030SJeremy L Thompson   ierr = CreateVectorFromPetscVec(ceed, Xloc, &xcorners); CHKERRQ(ierr);
1420ccaff030SJeremy L Thompson 
1421ccaff030SJeremy L Thompson   // Create the CEED vectors that will be needed in setup
1422bd910870SLeila Ghaffari   CeedInt NqptsVol;
142384d34d69SLeila Ghaffari   CeedBasisGetNumQuadraturePoints(basisq, &NqptsVol);
142484d34d69SLeila Ghaffari   CeedElemRestrictionGetNumElements(restrictq, &localNelemVol);
14258b982baeSLeila Ghaffari   CeedVectorCreate(ceed, qdatasizeVol*localNelemVol*NqptsVol, &qdata);
142684d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &q0ceed, NULL);
1427ccaff030SJeremy L Thompson 
1428ccaff030SJeremy L Thompson   // Create the Q-function that builds the quadrature data for the NS operator
1429ea6e0f84SLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setupVol, problem->setupVol_loc,
1430ea6e0f84SLeila Ghaffari                               &qf_setupVol);
1431ea6e0f84SLeila Ghaffari   CeedQFunctionAddInput(qf_setupVol, "dx", ncompx*dim, CEED_EVAL_GRAD);
1432ea6e0f84SLeila Ghaffari   CeedQFunctionAddInput(qf_setupVol, "weight", 1, CEED_EVAL_WEIGHT);
14338b982baeSLeila Ghaffari   CeedQFunctionAddOutput(qf_setupVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1434ccaff030SJeremy L Thompson 
1435ccaff030SJeremy L Thompson   // Create the Q-function that sets the ICs of the operator
1436ccaff030SJeremy L Thompson   CeedQFunctionCreateInterior(ceed, 1, problem->ics, problem->ics_loc, &qf_ics);
1437ccaff030SJeremy L Thompson   CeedQFunctionAddInput(qf_ics, "x", ncompx, CEED_EVAL_INTERP);
1438ccaff030SJeremy L Thompson   CeedQFunctionAddOutput(qf_ics, "q0", ncompq, CEED_EVAL_NONE);
1439ccaff030SJeremy L Thompson 
1440ea6e0f84SLeila Ghaffari   qf_rhsVol = NULL;
1441ea6e0f84SLeila Ghaffari   if (problem->applyVol_rhs) { // Create the Q-function that defines the action of the RHS operator
1442ea6e0f84SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applyVol_rhs,
1443ea6e0f84SLeila Ghaffari                                 problem->applyVol_rhs_loc, &qf_rhsVol);
1444ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "q", ncompq, CEED_EVAL_INTERP);
1445ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "dq", ncompq*dim, CEED_EVAL_GRAD);
14468b982baeSLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1447ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_rhsVol, "x", ncompx, CEED_EVAL_INTERP);
1448ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_rhsVol, "v", ncompq, CEED_EVAL_INTERP);
1449ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_rhsVol, "dv", ncompq*dim, CEED_EVAL_GRAD);
1450ccaff030SJeremy L Thompson   }
1451ccaff030SJeremy L Thompson 
1452ea6e0f84SLeila Ghaffari   qf_ifunctionVol = NULL;
1453ea6e0f84SLeila Ghaffari   if (problem->applyVol_ifunction) { // Create the Q-function that defines the action of the IFunction
1454ea6e0f84SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applyVol_ifunction,
1455ea6e0f84SLeila Ghaffari                                 problem->applyVol_ifunction_loc, &qf_ifunctionVol);
1456ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "q", ncompq, CEED_EVAL_INTERP);
1457ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "dq", ncompq*dim, CEED_EVAL_GRAD);
1458ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "qdot", ncompq, CEED_EVAL_INTERP);
14598b982baeSLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "qdata", qdatasizeVol, CEED_EVAL_NONE);
1460ea6e0f84SLeila Ghaffari     CeedQFunctionAddInput(qf_ifunctionVol, "x", ncompx, CEED_EVAL_INTERP);
1461ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_ifunctionVol, "v", ncompq, CEED_EVAL_INTERP);
1462ea6e0f84SLeila Ghaffari     CeedQFunctionAddOutput(qf_ifunctionVol, "dv", ncompq*dim, CEED_EVAL_GRAD);
1463ccaff030SJeremy L Thompson   }
1464ccaff030SJeremy L Thompson 
1465ccaff030SJeremy L Thompson   // Create the operator that builds the quadrature data for the NS operator
1466ea6e0f84SLeila Ghaffari   CeedOperatorCreate(ceed, qf_setupVol, NULL, NULL, &op_setupVol);
146784d34d69SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "dx", restrictx, basisx, CEED_VECTOR_ACTIVE);
1468ea6e0f84SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "weight", CEED_ELEMRESTRICTION_NONE,
146984d34d69SLeila Ghaffari                        basisx, CEED_VECTOR_NONE);
147084d34d69SLeila Ghaffari   CeedOperatorSetField(op_setupVol, "qdata", restrictqdi,
1471ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
1472ccaff030SJeremy L Thompson 
1473ccaff030SJeremy L Thompson   // Create the operator that sets the ICs
1474ccaff030SJeremy L Thompson   CeedOperatorCreate(ceed, qf_ics, NULL, NULL, &op_ics);
147584d34d69SLeila Ghaffari   CeedOperatorSetField(op_ics, "x", restrictx, basisxc, CEED_VECTOR_ACTIVE);
147684d34d69SLeila Ghaffari   CeedOperatorSetField(op_ics, "q0", restrictq,
1477ccaff030SJeremy L Thompson                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
1478ccaff030SJeremy L Thompson 
147984d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->qceed, NULL);
148084d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->qdotceed, NULL);
148184d34d69SLeila Ghaffari   CeedElemRestrictionCreateVector(restrictq, &user->gceed, NULL);
1482ccaff030SJeremy L Thompson 
1483ea6e0f84SLeila Ghaffari   if (qf_rhsVol) { // Create the RHS physics operator
1484ccaff030SJeremy L Thompson     CeedOperator op;
1485ea6e0f84SLeila Ghaffari     CeedOperatorCreate(ceed, qf_rhsVol, NULL, NULL, &op);
148684d34d69SLeila Ghaffari     CeedOperatorSetField(op, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
148784d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dq", restrictq, basisq, CEED_VECTOR_ACTIVE);
148884d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdata", restrictqdi,
14898b982baeSLeila Ghaffari                          CEED_BASIS_COLLOCATED, qdata);
149084d34d69SLeila Ghaffari     CeedOperatorSetField(op, "x", restrictx, basisx, xcorners);
149184d34d69SLeila Ghaffari     CeedOperatorSetField(op, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
149284d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dv", restrictq, basisq, CEED_VECTOR_ACTIVE);
1493d3630711SJed Brown     user->op_rhs_vol = op;
1494ccaff030SJeremy L Thompson   }
1495ccaff030SJeremy L Thompson 
1496ea6e0f84SLeila Ghaffari   if (qf_ifunctionVol) { // Create the IFunction operator
1497ccaff030SJeremy L Thompson     CeedOperator op;
1498ea6e0f84SLeila Ghaffari     CeedOperatorCreate(ceed, qf_ifunctionVol, NULL, NULL, &op);
149984d34d69SLeila Ghaffari     CeedOperatorSetField(op, "q", restrictq, basisq, CEED_VECTOR_ACTIVE);
150084d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dq", restrictq, basisq, CEED_VECTOR_ACTIVE);
150184d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdot", restrictq, basisq, user->qdotceed);
150284d34d69SLeila Ghaffari     CeedOperatorSetField(op, "qdata", restrictqdi,
15038b982baeSLeila Ghaffari                          CEED_BASIS_COLLOCATED, qdata);
150484d34d69SLeila Ghaffari     CeedOperatorSetField(op, "x", restrictx, basisx, xcorners);
150584d34d69SLeila Ghaffari     CeedOperatorSetField(op, "v", restrictq, basisq, CEED_VECTOR_ACTIVE);
150684d34d69SLeila Ghaffari     CeedOperatorSetField(op, "dv", restrictq, basisq, CEED_VECTOR_ACTIVE);
1507d3630711SJed Brown     user->op_ifunction_vol = op;
1508ccaff030SJeremy L Thompson   }
1509ccaff030SJeremy L Thompson 
15106a0edaf9SLeila Ghaffari   // Set up CEED for the boundaries
15116a0edaf9SLeila Ghaffari   CeedInt height = 1;
15126a0edaf9SLeila Ghaffari   CeedInt dimSur = dim - height;
15131e150236SLeila Ghaffari   CeedInt numP_Sur = degree + 1;
15141e150236SLeila Ghaffari   CeedInt numQ_Sur = numP_Sur + qextraSur;
1515cfa64770SLeila Ghaffari   const CeedInt qdatasizeSur = problem->qdatasizeSur;
1516cfa64770SLeila Ghaffari   CeedBasis basisxSur, basisxcSur, basisqSur;
1517cfa64770SLeila Ghaffari   CeedInt NqptsSur;
1518e2043960SLeila Ghaffari   CeedQFunction qf_setupSur, qf_applySur;
1519cfa64770SLeila Ghaffari 
1520cfa64770SLeila Ghaffari   // CEED bases for the boundaries
1521ebb4b9bdSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompq, numP_Sur, numQ_Sur,
1522ebb4b9bdSLeila Ghaffari                                   CEED_GAUSS,
15236a0edaf9SLeila Ghaffari                                   &basisqSur);
15246a0edaf9SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompx, 2, numQ_Sur, CEED_GAUSS,
15256a0edaf9SLeila Ghaffari                                   &basisxSur);
15266a0edaf9SLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dimSur, ncompx, 2, numP_Sur,
15276a0edaf9SLeila Ghaffari                                   CEED_GAUSS_LOBATTO, &basisxcSur);
15286a0edaf9SLeila Ghaffari   CeedBasisGetNumQuadraturePoints(basisqSur, &NqptsSur);
15296a0edaf9SLeila Ghaffari 
1530cfa64770SLeila Ghaffari   // Create the Q-function that builds the quadrature data for the Surface operator
15316a0edaf9SLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setupSur, problem->setupSur_loc,
15326a0edaf9SLeila Ghaffari                               &qf_setupSur);
15336a0edaf9SLeila Ghaffari   CeedQFunctionAddInput(qf_setupSur, "dx", ncompx*dimSur, CEED_EVAL_GRAD);
15346a0edaf9SLeila Ghaffari   CeedQFunctionAddInput(qf_setupSur, "weight", 1, CEED_EVAL_WEIGHT);
15356a0edaf9SLeila Ghaffari   CeedQFunctionAddOutput(qf_setupSur, "qdataSur", qdatasizeSur, CEED_EVAL_NONE);
15366a0edaf9SLeila Ghaffari 
15377659d40cSLeila Ghaffari   // Creat Q-Function for Boundaries
153886ba6f09SLeila Ghaffari   // -- Defined for Advection(2d) test cases
15397ed8b9c7SLeila Ghaffari   qf_applySur = NULL;
15407659d40cSLeila Ghaffari   if (problem->applySur) {
15417659d40cSLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->applySur,
15427ed8b9c7SLeila Ghaffari                                 problem->applySur_loc, &qf_applySur);
15437ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "q", ncompq, CEED_EVAL_INTERP);
15447ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "qdataSur", qdatasizeSur, CEED_EVAL_NONE);
15457ed8b9c7SLeila Ghaffari     CeedQFunctionAddInput(qf_applySur, "x", ncompx, CEED_EVAL_INTERP);
15467ed8b9c7SLeila Ghaffari     CeedQFunctionAddOutput(qf_applySur, "v", ncompq, CEED_EVAL_INTERP);
15479fe13df9SLeila Ghaffari   }
15489fe13df9SLeila Ghaffari 
15499fe13df9SLeila Ghaffari   // Create CEED Operator for the whole domain
15509fe13df9SLeila Ghaffari   if (!implicit)
15514438636fSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, &bc, problemChoice, wind_type,
15524438636fSLeila Ghaffari                                    user->op_rhs_vol, qf_applySur,
1553e2043960SLeila Ghaffari                                    qf_setupSur, height, numP_Sur, numQ_Sur,
15544438636fSLeila Ghaffari                                    qdatasizeSur, NqptsSur, basisxSur,
15554438636fSLeila Ghaffari                                    basisqSur, &user->op_rhs);
15564438636fSLeila Ghaffari   CHKERRQ(ierr);
15579fe13df9SLeila Ghaffari   if (implicit)
15584438636fSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, &bc, problemChoice, wind_type,
15594438636fSLeila Ghaffari                                    user->op_ifunction_vol, qf_applySur,
1560e2043960SLeila Ghaffari                                    qf_setupSur, height, numP_Sur, numQ_Sur,
15614438636fSLeila Ghaffari                                    qdatasizeSur, NqptsSur, basisxSur,
15624438636fSLeila Ghaffari                                    basisqSur, &user->op_ifunction);
15634438636fSLeila Ghaffari   CHKERRQ(ierr);
15647659d40cSLeila Ghaffari   // Set up contex for QFunctions
1565777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxSetup);
1566777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxSetup, CEED_MEM_HOST, CEED_USE_POINTER,
1567777ff853SJeremy L Thompson                               sizeof ctxSetupData, &ctxSetupData);
15685bf10c6fSLeila Ghaffari   if (qf_ics && problemChoice != NS_EULER_VORTEX)
1569777ff853SJeremy L Thompson     CeedQFunctionSetContext(qf_ics, ctxSetup);
1570777ff853SJeremy L Thompson 
15715bf10c6fSLeila Ghaffari   CeedScalar ctxNSData[8] = {lambda, mu, k, cv, cp, g, Rd};
1572777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxNS);
1573777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxNS, CEED_MEM_HOST, CEED_USE_POINTER,
1574777ff853SJeremy L Thompson                               sizeof ctxNSData, &ctxNSData);
1575777ff853SJeremy L Thompson 
15760da62991SLeila Ghaffari   struct Advection2dContext_ ctxAdvection2dData = {
1577ccaff030SJeremy L Thompson     .CtauS = CtauS,
1578ccaff030SJeremy L Thompson     .strong_form = strong_form,
1579ccaff030SJeremy L Thompson     .stabilization = stab,
1580ccaff030SJeremy L Thompson   };
1581777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxAdvection2d);
1582777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxAdvection2d, CEED_MEM_HOST, CEED_USE_POINTER,
1583777ff853SJeremy L Thompson                               sizeof ctxAdvection2dData, &ctxAdvection2dData);
1584777ff853SJeremy L Thompson 
1585777ff853SJeremy L Thompson   struct SurfaceContext_ ctxSurfaceData = {
158616c0476cSLeila Ghaffari     .E_wind = E_wind,
15877659d40cSLeila Ghaffari     .strong_form = strong_form,
1588*c8cb9b1fSLeila Ghaffari     .T_inlet = T_inlet,
1589*c8cb9b1fSLeila Ghaffari     .P_inlet = P_inlet,
1590*c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[0] = etv_mean_velocity[0],
1591*c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[1] = etv_mean_velocity[1],
1592*c8cb9b1fSLeila Ghaffari     .etv_mean_velocity[2] = etv_mean_velocity[2],
1593e5ed8c30SLeila Ghaffari     .implicit = implicit,
1594e5ed8c30SLeila Ghaffari   };
1595777ff853SJeremy L Thompson   CeedQFunctionContextCreate(ceed, &ctxSurface);
1596777ff853SJeremy L Thompson   CeedQFunctionContextSetData(ctxSurface, CEED_MEM_HOST, CEED_USE_POINTER,
1597777ff853SJeremy L Thompson                               sizeof ctxSurfaceData, &ctxSurfaceData);
1598777ff853SJeremy L Thompson 
15995bf10c6fSLeila Ghaffari   // Set up ctxEulerData structure
16005bf10c6fSLeila Ghaffari   ctxEulerData->time = 0.;
16015bf10c6fSLeila Ghaffari   ctxEulerData->currentTime = 0.;
16025bf10c6fSLeila Ghaffari   ctxEulerData->center[0] = center[0];
16035bf10c6fSLeila Ghaffari   ctxEulerData->center[1] = center[1];
16045bf10c6fSLeila Ghaffari   ctxEulerData->center[2] = center[2];
16055bf10c6fSLeila Ghaffari   ctxEulerData->vortex_strength = vortex_strength;
16060da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[0] = etv_mean_velocity[0];
16070da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[1] = etv_mean_velocity[1];
16080da62991SLeila Ghaffari   ctxEulerData->etv_mean_velocity[2] = etv_mean_velocity[2];
16095bf10c6fSLeila Ghaffari   user->ctxEulerData = ctxEulerData;
16105bf10c6fSLeila Ghaffari 
16115bf10c6fSLeila Ghaffari   CeedQFunctionContextCreate(ceed, &ctxEuler);
16125bf10c6fSLeila Ghaffari   CeedQFunctionContextSetData(ctxEuler, CEED_MEM_HOST, CEED_USE_POINTER,
16130da62991SLeila Ghaffari                               sizeof *ctxEulerData, ctxEulerData);
16145bf10c6fSLeila Ghaffari 
1615ccaff030SJeremy L Thompson   switch (problemChoice) {
1616ccaff030SJeremy L Thompson   case NS_DENSITY_CURRENT:
16175bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxNS);
16185bf10c6fSLeila Ghaffari     if (qf_ifunctionVol) CeedQFunctionSetContext(qf_ifunctionVol, ctxNS);
1619ccaff030SJeremy L Thompson     break;
1620ccaff030SJeremy L Thompson   case NS_ADVECTION:
1621ccaff030SJeremy L Thompson   case NS_ADVECTION2D:
16225bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxAdvection2d);
16235bf10c6fSLeila Ghaffari     if (qf_ifunctionVol) CeedQFunctionSetContext(qf_ifunctionVol, ctxAdvection2d);
16245bf10c6fSLeila Ghaffari     if (qf_applySur) CeedQFunctionSetContext(qf_applySur, ctxSurface);
1625e5ed8c30SLeila Ghaffari   case NS_EULER_VORTEX:
16260da62991SLeila Ghaffari     if (qf_ics) CeedQFunctionSetContext(qf_ics, ctxEuler);
16275bf10c6fSLeila Ghaffari     if (qf_rhsVol) CeedQFunctionSetContext(qf_rhsVol, ctxEuler);
1628*c8cb9b1fSLeila Ghaffari     if (qf_applySur) CeedQFunctionSetContext(qf_applySur, ctxSurface);
1629ccaff030SJeremy L Thompson   }
1630ccaff030SJeremy L Thompson 
1631ccaff030SJeremy L Thompson   // Set up PETSc context
1632ccaff030SJeremy L Thompson   // Set up units structure
1633ccaff030SJeremy L Thompson   units->meter = meter;
1634ccaff030SJeremy L Thompson   units->kilogram = kilogram;
1635ccaff030SJeremy L Thompson   units->second = second;
1636ccaff030SJeremy L Thompson   units->Kelvin = Kelvin;
1637ccaff030SJeremy L Thompson   units->Pascal = Pascal;
1638ccaff030SJeremy L Thompson   units->JperkgK = JperkgK;
1639ccaff030SJeremy L Thompson   units->mpersquareds = mpersquareds;
1640ccaff030SJeremy L Thompson   units->WpermK = WpermK;
1641ccaff030SJeremy L Thompson   units->kgpercubicm = kgpercubicm;
1642ccaff030SJeremy L Thompson   units->kgpersquaredms = kgpersquaredms;
1643ccaff030SJeremy L Thompson   units->Joulepercubicm = Joulepercubicm;
164416c0476cSLeila Ghaffari   units->Joule = Joule;
1645ccaff030SJeremy L Thompson 
1646ccaff030SJeremy L Thompson   // Set up user structure
1647ccaff030SJeremy L Thompson   user->comm = comm;
1648ccaff030SJeremy L Thompson   user->outputfreq = outputfreq;
1649ccaff030SJeremy L Thompson   user->contsteps = contsteps;
1650ccaff030SJeremy L Thompson   user->units = units;
1651ccaff030SJeremy L Thompson   user->dm = dm;
1652ccaff030SJeremy L Thompson   user->dmviz = dmviz;
1653ccaff030SJeremy L Thompson   user->interpviz = interpviz;
1654ccaff030SJeremy L Thompson   user->ceed = ceed;
1655ccaff030SJeremy L Thompson 
16568b982baeSLeila Ghaffari   // Calculate qdata and ICs
1657ccaff030SJeremy L Thompson   // Set up state global and local vectors
1658ccaff030SJeremy L Thompson   ierr = VecZeroEntries(Q); CHKERRQ(ierr);
1659ccaff030SJeremy L Thompson 
1660cfa64770SLeila Ghaffari   ierr = VectorPlacePetscVec(q0ceed, Qloc); CHKERRQ(ierr);
1661ccaff030SJeremy L Thompson 
1662ccaff030SJeremy L Thompson   // Apply Setup Ceed Operators
1663ccaff030SJeremy L Thompson   ierr = VectorPlacePetscVec(xcorners, Xloc); CHKERRQ(ierr);
16648b982baeSLeila Ghaffari   CeedOperatorApply(op_setupVol, xcorners, qdata, CEED_REQUEST_IMMEDIATE);
166584d34d69SLeila Ghaffari   ierr = ComputeLumpedMassMatrix(ceed, dm, restrictq, basisq, restrictqdi, qdata,
1666ccaff030SJeremy L Thompson                                  user->M); CHKERRQ(ierr);
1667ccaff030SJeremy L Thompson 
166884d34d69SLeila Ghaffari   ierr = ICs_FixMultiplicity(op_ics, xcorners, q0ceed, dm, Qloc, Q, restrictq,
1669777ff853SJeremy L Thompson                              ctxSetup, 0.0); CHKERRQ(ierr);
1670ccaff030SJeremy L Thompson   if (1) { // Record boundary values from initial condition and override DMPlexInsertBoundaryValues()
1671ccaff030SJeremy L Thompson     // We use this for the main simulation DM because the reference DMPlexInsertBoundaryValues() is very slow.  If we
1672ccaff030SJeremy L Thompson     // disable this, we should still get the same results due to the problem->bc function, but with potentially much
1673ccaff030SJeremy L Thompson     // slower execution.
1674ccaff030SJeremy L Thompson     Vec Qbc;
1675ccaff030SJeremy L Thompson     ierr = DMGetNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
1676ccaff030SJeremy L Thompson     ierr = VecCopy(Qloc, Qbc); CHKERRQ(ierr);
1677ccaff030SJeremy L Thompson     ierr = VecZeroEntries(Qloc); CHKERRQ(ierr);
1678ccaff030SJeremy L Thompson     ierr = DMGlobalToLocal(dm, Q, INSERT_VALUES, Qloc); CHKERRQ(ierr);
1679ccaff030SJeremy L Thompson     ierr = VecAXPY(Qbc, -1., Qloc); CHKERRQ(ierr);
1680ccaff030SJeremy L Thompson     ierr = DMRestoreNamedLocalVector(dm, "Qbc", &Qbc); CHKERRQ(ierr);
1681ccaff030SJeremy L Thompson     ierr = PetscObjectComposeFunction((PetscObject)dm,
168284d34d69SLeila Ghaffari                                       "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_NS);
168384d34d69SLeila Ghaffari     CHKERRQ(ierr);
1684ccaff030SJeremy L Thompson   }
1685ccaff030SJeremy L Thompson 
1686ccaff030SJeremy L Thompson   MPI_Comm_rank(comm, &rank);
1687d99129b9SLeila Ghaffari   if (!rank) {ierr = PetscMkdir(user->outputdir); CHKERRQ(ierr);}
1688ccaff030SJeremy L Thompson   // Gather initial Q values
1689ccaff030SJeremy L Thompson   // In case of continuation of simulation, set up initial values from binary file
1690ccaff030SJeremy L Thompson   if (contsteps) { // continue from existent solution
1691ccaff030SJeremy L Thompson     PetscViewer viewer;
1692ccaff030SJeremy L Thompson     char filepath[PETSC_MAX_PATH_LEN];
1693ccaff030SJeremy L Thompson     // Read input
1694ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-solution.bin",
1695d99129b9SLeila Ghaffari                          user->outputdir);
1696ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1697ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
1698ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1699ccaff030SJeremy L Thompson     ierr = VecLoad(Q, viewer); CHKERRQ(ierr);
1700ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
1701ccaff030SJeremy L Thompson   }
1702ccaff030SJeremy L Thompson   ierr = DMRestoreLocalVector(dm, &Qloc); CHKERRQ(ierr);
1703ccaff030SJeremy L Thompson 
1704ccaff030SJeremy L Thompson   // Create and setup TS
1705ccaff030SJeremy L Thompson   ierr = TSCreate(comm, &ts); CHKERRQ(ierr);
1706ccaff030SJeremy L Thompson   ierr = TSSetDM(ts, dm); CHKERRQ(ierr);
1707ccaff030SJeremy L Thompson   if (implicit) {
1708ccaff030SJeremy L Thompson     ierr = TSSetType(ts, TSBDF); CHKERRQ(ierr);
1709ccaff030SJeremy L Thompson     if (user->op_ifunction) {
1710ccaff030SJeremy L Thompson       ierr = TSSetIFunction(ts, NULL, IFunction_NS, &user); CHKERRQ(ierr);
1711ccaff030SJeremy L Thompson     } else {                    // Implicit integrators can fall back to using an RHSFunction
1712ccaff030SJeremy L Thompson       ierr = TSSetRHSFunction(ts, NULL, RHS_NS, &user); CHKERRQ(ierr);
1713ccaff030SJeremy L Thompson     }
1714ccaff030SJeremy L Thompson   } else {
1715ccaff030SJeremy L Thompson     if (!user->op_rhs) SETERRQ(comm, PETSC_ERR_ARG_NULL,
1716ccaff030SJeremy L Thompson                                  "Problem does not provide RHSFunction");
1717ccaff030SJeremy L Thompson     ierr = TSSetType(ts, TSRK); CHKERRQ(ierr);
1718ccaff030SJeremy L Thompson     ierr = TSRKSetType(ts, TSRK5F); CHKERRQ(ierr);
1719ccaff030SJeremy L Thompson     ierr = TSSetRHSFunction(ts, NULL, RHS_NS, &user); CHKERRQ(ierr);
1720ccaff030SJeremy L Thompson   }
1721ccaff030SJeremy L Thompson   ierr = TSSetMaxTime(ts, 500. * units->second); CHKERRQ(ierr);
1722ccaff030SJeremy L Thompson   ierr = TSSetExactFinalTime(ts, TS_EXACTFINALTIME_STEPOVER); CHKERRQ(ierr);
1723ccaff030SJeremy L Thompson   ierr = TSSetTimeStep(ts, 1.e-2 * units->second); CHKERRQ(ierr);
1724dc8efd83SLeila Ghaffari   if (test) {ierr = TSSetMaxSteps(ts, 10); CHKERRQ(ierr);}
1725ccaff030SJeremy L Thompson   ierr = TSGetAdapt(ts, &adapt); CHKERRQ(ierr);
1726ccaff030SJeremy L Thompson   ierr = TSAdaptSetStepLimits(adapt,
1727ccaff030SJeremy L Thompson                               1.e-12 * units->second,
1728ccaff030SJeremy L Thompson                               1.e2 * units->second); CHKERRQ(ierr);
1729ccaff030SJeremy L Thompson   ierr = TSSetFromOptions(ts); CHKERRQ(ierr);
1730ccaff030SJeremy L Thompson   if (!contsteps) { // print initial condition
1731dc8efd83SLeila Ghaffari     if (!test) {
1732ccaff030SJeremy L Thompson       ierr = TSMonitor_NS(ts, 0, 0., Q, user); CHKERRQ(ierr);
1733ccaff030SJeremy L Thompson     }
1734ccaff030SJeremy L Thompson   } else { // continue from time of last output
1735ccaff030SJeremy L Thompson     PetscReal time;
1736ccaff030SJeremy L Thompson     PetscInt count;
1737ccaff030SJeremy L Thompson     PetscViewer viewer;
1738ccaff030SJeremy L Thompson     char filepath[PETSC_MAX_PATH_LEN];
1739ccaff030SJeremy L Thompson     ierr = PetscSNPrintf(filepath, sizeof filepath, "%s/ns-time.bin",
1740d99129b9SLeila Ghaffari                          user->outputdir); CHKERRQ(ierr);
1741ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
1742ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1743ccaff030SJeremy L Thompson     ierr = PetscViewerBinaryRead(viewer, &time, 1, &count, PETSC_REAL);
1744ccaff030SJeremy L Thompson     CHKERRQ(ierr);
1745ccaff030SJeremy L Thompson     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
1746ccaff030SJeremy L Thompson     ierr = TSSetTime(ts, time * user->units->second); CHKERRQ(ierr);
1747ccaff030SJeremy L Thompson   }
1748dc8efd83SLeila Ghaffari   if (!test) {
1749ccaff030SJeremy L Thompson     ierr = TSMonitorSet(ts, TSMonitor_NS, user, NULL); CHKERRQ(ierr);
1750ccaff030SJeremy L Thompson   }
1751ccaff030SJeremy L Thompson 
1752ccaff030SJeremy L Thompson   // Solve
1753ccaff030SJeremy L Thompson   start = MPI_Wtime();
1754ccaff030SJeremy L Thompson   ierr = PetscBarrier((PetscObject)ts); CHKERRQ(ierr);
1755ccaff030SJeremy L Thompson   ierr = TSSolve(ts, Q); CHKERRQ(ierr);
1756ccaff030SJeremy L Thompson   cpu_time_used = MPI_Wtime() - start;
1757ccaff030SJeremy L Thompson   ierr = TSGetSolveTime(ts, &ftime); CHKERRQ(ierr);
1758ccaff030SJeremy L Thompson   ierr = MPI_Allreduce(MPI_IN_PLACE, &cpu_time_used, 1, MPI_DOUBLE, MPI_MIN,
1759ccaff030SJeremy L Thompson                        comm); CHKERRQ(ierr);
1760dc8efd83SLeila Ghaffari   if (!test) {
1761ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
176284d34d69SLeila Ghaffari                        "Time taken for solution (sec): %g\n",
1763ccaff030SJeremy L Thompson                        (double)cpu_time_used); CHKERRQ(ierr);
1764ccaff030SJeremy L Thompson   }
1765ccaff030SJeremy L Thompson 
1766ccaff030SJeremy L Thompson   // Get error
1767dc8efd83SLeila Ghaffari   if (problem->non_zero_time && !test) {
1768ccaff030SJeremy L Thompson     Vec Qexact, Qexactloc;
1769ccaff030SJeremy L Thompson     PetscReal norm;
1770ccaff030SJeremy L Thompson     ierr = DMCreateGlobalVector(dm, &Qexact); CHKERRQ(ierr);
1771ccaff030SJeremy L Thompson     ierr = DMGetLocalVector(dm, &Qexactloc); CHKERRQ(ierr);
1772ccaff030SJeremy L Thompson     ierr = VecGetSize(Qexactloc, &lnodes); CHKERRQ(ierr);
1773ccaff030SJeremy L Thompson 
177484d34d69SLeila Ghaffari     ierr = ICs_FixMultiplicity(op_ics, xcorners, q0ceed, dm, Qexactloc, Qexact,
1775777ff853SJeremy L Thompson                                restrictq, ctxSetup, ftime); CHKERRQ(ierr);
1776ccaff030SJeremy L Thompson 
1777ccaff030SJeremy L Thompson     ierr = VecAXPY(Q, -1.0, Qexact);  CHKERRQ(ierr);
1778ccaff030SJeremy L Thompson     ierr = VecNorm(Q, NORM_MAX, &norm); CHKERRQ(ierr);
1779cfa64770SLeila Ghaffari     CeedVectorDestroy(&q0ceed);
1780ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
1781ccaff030SJeremy L Thompson                        "Max Error: %g\n",
1782ccaff030SJeremy L Thompson                        (double)norm); CHKERRQ(ierr);
178384d34d69SLeila Ghaffari     // Clean up vectors
178484d34d69SLeila Ghaffari     ierr = DMRestoreLocalVector(dm, &Qexactloc); CHKERRQ(ierr);
178584d34d69SLeila Ghaffari     ierr = VecDestroy(&Qexact); CHKERRQ(ierr);
1786ccaff030SJeremy L Thompson   }
1787ccaff030SJeremy L Thompson 
1788ccaff030SJeremy L Thompson   // Output Statistics
1789ccaff030SJeremy L Thompson   ierr = TSGetStepNumber(ts, &steps); CHKERRQ(ierr);
1790dc8efd83SLeila Ghaffari   if (!test) {
1791ccaff030SJeremy L Thompson     ierr = PetscPrintf(PETSC_COMM_WORLD,
1792ccaff030SJeremy L Thompson                        "Time integrator took %D time steps to reach final time %g\n",
1793ccaff030SJeremy L Thompson                        steps, (double)ftime); CHKERRQ(ierr);
1794ccaff030SJeremy L Thompson   }
179584d34d69SLeila Ghaffari   // Output numerical values from command line
179684d34d69SLeila Ghaffari   ierr = VecViewFromOptions(Q, NULL, "-vec_view"); CHKERRQ(ierr);
179784d34d69SLeila Ghaffari 
1798335ea220SLeila Ghaffari   // Compare reference solution values with current test run for CI
1799dc8efd83SLeila Ghaffari   if (test) {
180084d34d69SLeila Ghaffari     PetscViewer viewer;
180184d34d69SLeila Ghaffari     // Read reference file
180284d34d69SLeila Ghaffari     Vec Qref;
180384d34d69SLeila Ghaffari     PetscReal error, Qrefnorm;
180484d34d69SLeila Ghaffari     ierr = VecDuplicate(Q, &Qref); CHKERRQ(ierr);
1805dc8efd83SLeila Ghaffari     ierr = PetscViewerBinaryOpen(comm, filepath, FILE_MODE_READ, &viewer);
180684d34d69SLeila Ghaffari     CHKERRQ(ierr);
180784d34d69SLeila Ghaffari     ierr = VecLoad(Qref, viewer); CHKERRQ(ierr);
180884d34d69SLeila Ghaffari     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
180984d34d69SLeila Ghaffari 
181084d34d69SLeila Ghaffari     // Compute error with respect to reference solution
181184d34d69SLeila Ghaffari     ierr = VecAXPY(Q, -1.0, Qref);  CHKERRQ(ierr);
181284d34d69SLeila Ghaffari     ierr = VecNorm(Qref, NORM_MAX, &Qrefnorm); CHKERRQ(ierr);
181384d34d69SLeila Ghaffari     ierr = VecScale(Q, 1./Qrefnorm); CHKERRQ(ierr);
181484d34d69SLeila Ghaffari     ierr = VecNorm(Q, NORM_MAX, &error); CHKERRQ(ierr);
181584d34d69SLeila Ghaffari     ierr = VecDestroy(&Qref); CHKERRQ(ierr);
181684d34d69SLeila Ghaffari     // Check error
1817dc8efd83SLeila Ghaffari     if (error > testtol) {
181884d34d69SLeila Ghaffari       ierr = PetscPrintf(PETSC_COMM_WORLD,
181984d34d69SLeila Ghaffari                          "Test failed with error norm %g\n",
182084d34d69SLeila Ghaffari                          (double)error); CHKERRQ(ierr);
182184d34d69SLeila Ghaffari     }
182284d34d69SLeila Ghaffari     ierr = PetscViewerDestroy(&viewer); CHKERRQ(ierr);
182384d34d69SLeila Ghaffari   }
18249cf88b28Svaleriabarra 
1825ccaff030SJeremy L Thompson   // Clean up libCEED
18268b982baeSLeila Ghaffari   CeedVectorDestroy(&qdata);
1827ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->qceed);
1828ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->qdotceed);
1829ccaff030SJeremy L Thompson   CeedVectorDestroy(&user->gceed);
1830ccaff030SJeremy L Thompson   CeedVectorDestroy(&xcorners);
183184d34d69SLeila Ghaffari   CeedBasisDestroy(&basisq);
183284d34d69SLeila Ghaffari   CeedBasisDestroy(&basisx);
183384d34d69SLeila Ghaffari   CeedBasisDestroy(&basisxc);
183484d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictq);
183584d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictx);
183684d34d69SLeila Ghaffari   CeedElemRestrictionDestroy(&restrictqdi);
1837ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_setupVol);
1838ccaff030SJeremy L Thompson   CeedQFunctionDestroy(&qf_ics);
1839ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_rhsVol);
1840ea6e0f84SLeila Ghaffari   CeedQFunctionDestroy(&qf_ifunctionVol);
1841777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxSetup);
1842777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxNS);
1843777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxAdvection2d);
1844777ff853SJeremy L Thompson   CeedQFunctionContextDestroy(&ctxSurface);
18450da62991SLeila Ghaffari   CeedQFunctionContextDestroy(&ctxEuler);
1846ea6e0f84SLeila Ghaffari   CeedOperatorDestroy(&op_setupVol);
1847ccaff030SJeremy L Thompson   CeedOperatorDestroy(&op_ics);
18486a0edaf9SLeila Ghaffari   CeedOperatorDestroy(&user->op_rhs_vol);
18496a0edaf9SLeila Ghaffari   CeedOperatorDestroy(&user->op_ifunction_vol);
18506a0edaf9SLeila Ghaffari   CeedDestroy(&ceed);
18516a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisqSur);
18526a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisxSur);
18536a0edaf9SLeila Ghaffari   CeedBasisDestroy(&basisxcSur);
18546a0edaf9SLeila Ghaffari   CeedQFunctionDestroy(&qf_setupSur);
18557ed8b9c7SLeila Ghaffari   CeedQFunctionDestroy(&qf_applySur);
1856ccaff030SJeremy L Thompson   CeedOperatorDestroy(&user->op_rhs);
1857ccaff030SJeremy L Thompson   CeedOperatorDestroy(&user->op_ifunction);
1858ccaff030SJeremy L Thompson 
1859ccaff030SJeremy L Thompson   // Clean up PETSc
1860ccaff030SJeremy L Thompson   ierr = VecDestroy(&Q); CHKERRQ(ierr);
1861ccaff030SJeremy L Thompson   ierr = VecDestroy(&user->M); CHKERRQ(ierr);
1862ccaff030SJeremy L Thompson   ierr = MatDestroy(&interpviz); CHKERRQ(ierr);
1863ccaff030SJeremy L Thompson   ierr = DMDestroy(&dmviz); CHKERRQ(ierr);
1864ccaff030SJeremy L Thompson   ierr = TSDestroy(&ts); CHKERRQ(ierr);
1865ccaff030SJeremy L Thompson   ierr = DMDestroy(&dm); CHKERRQ(ierr);
1866ccaff030SJeremy L Thompson   ierr = PetscFree(units); CHKERRQ(ierr);
1867ccaff030SJeremy L Thompson   ierr = PetscFree(user); CHKERRQ(ierr);
18685bf10c6fSLeila Ghaffari   ierr = PetscFree(ctxEulerData); CHKERRQ(ierr);
1869ccaff030SJeremy L Thompson   return PetscFinalize();
1870ccaff030SJeremy L Thompson }
1871