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