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