ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_TI_slow_headers.H
Go to the documentation of this file.
1 #ifndef ERF_SLOW_INTEGRATION_H_
2 #define ERF_SLOW_INTEGRATION_H_
3 
4 #include "AMReX_MultiFab.H"
5 #include "AMReX_BCRec.H"
6 #include "AMReX_YAFluxRegister.H"
7 #include "AMReX_MultiCutFab.H"
8 #include "AMReX_EBMultiFabUtil.H"
9 
10 #include "ERF_DataStruct.H"
11 #include "ERF_IndexDefines.H"
12 #include "ERF_SurfaceLayer.H"
13 #include "ERF_Advection.H"
14 #include "ERF_Diffusion.H"
15 #include "ERF_EddyViscosity.H"
16 #include "ERF_PlaneAverage.H"
17 #include "ERF_TerrainMetrics.H"
18 #include "ERF_TileNoZ.H"
19 #include "ERF_EB.H"
20 
21 void erf_make_tau_terms (int level, int nrk,
22  const amrex::Vector<amrex::BCRec>& domain_bcs_type,
23  const amrex::MultiFab& z_phys_nd,
24  amrex::Vector<amrex::MultiFab>& S_data,
25  const amrex::MultiFab& xvel,
26  const amrex::MultiFab& yvel,
27  const amrex::MultiFab& zvel,
28  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& Tau_lev,
29  amrex::MultiFab* SmnSmn,
30  amrex::MultiFab* eddyDiffs,
31  const amrex::Geometry geom,
32  const SolverChoice& solverChoice,
33  std::unique_ptr<SurfaceLayer>& SurfLayer,
34  amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
35  const amrex::MultiFab& detJ,
36  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& mapfac);
37 
38 /**
39  * Function for computing the slow RHS for the evolution equations for the density, potential temperature and momentum.
40  *
41  */
42 void erf_slow_rhs_pre (int level, int finest_level, int nrk,
43  amrex::Real dt,
44  amrex::Vector<amrex::MultiFab>& S_rhs,
45  amrex::Vector<amrex::MultiFab>& S_old,
46  amrex::Vector<amrex::MultiFab>& S_data,
47  const amrex::MultiFab & S_prim,
48  const amrex::MultiFab & qt,
49  amrex::MultiFab& avg_xmom,
50  amrex::MultiFab& avg_ymom,
51  amrex::MultiFab& avg_zmom,
52  const amrex::MultiFab& xvel,
53  const amrex::MultiFab& yvel,
54  const amrex::MultiFab& zvel,
55  std::unique_ptr<amrex::MultiFab>& z_t,
56  const amrex::MultiFab& cc_src,
57  const amrex::MultiFab& xmom_src,
58  const amrex::MultiFab& ymom_src,
59  const amrex::MultiFab& zmom_src,
60  const amrex::MultiFab& buoyancy,
61  const amrex::MultiFab* zmom_crse_rhs,
62  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& Tau_lev,
63  amrex::MultiFab* SmnSmn,
64  amrex::MultiFab* eddyDiffs,
65  amrex::MultiFab* Hfx1,
66  amrex::MultiFab* Hfx2,
67  amrex::MultiFab* Hfx3,
68  amrex::MultiFab* Q1fx1,
69  amrex::MultiFab* Q1fx2,
70  amrex::MultiFab* Q1fx3,
71  amrex::MultiFab* Q2fx3,
72  amrex::MultiFab* Diss,
73  const amrex::Geometry geom,
74  const SolverChoice& solverChoice,
75  std::unique_ptr<SurfaceLayer>& SurfLayer,
76  const amrex::Gpu::DeviceVector<amrex::BCRec>& domain_bcs_type_d,
77  const amrex::Vector<amrex::BCRec>& domain_bcs_type,
78  const amrex::MultiFab& z_phys_nd,
79  const amrex::MultiFab& z_phys_cc,
80  const amrex::MultiFab& ax,
81  const amrex::MultiFab& ay,
82  const amrex::MultiFab& az,
83  const amrex::MultiFab& dJ,
84  amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
85  amrex::Vector<amrex::MultiFab>& gradp,
86  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& mapfac,
87  const eb_& ebfact,
88 #ifdef ERF_USE_SHOC
89  std::unique_ptr<SHOCInterface>& shoc_lev,
90 #endif
91  amrex::YAFluxRegister* fr_as_crse,
92  amrex::YAFluxRegister* fr_as_fine);
93 
94 /**
95  * Function for computing the slow RHS for the evolution equations for the scalars other than density or potential temperature
96  *
97  */
98 void erf_slow_rhs_post (int level, int finest_level, int nrk,
99  amrex::Real dt,
100  int n_qstate,
101  amrex::Vector<amrex::MultiFab>& S_rhs,
102  amrex::Vector<amrex::MultiFab>& S_old,
103  amrex::Vector<amrex::MultiFab>& S_new,
104  amrex::Vector<amrex::MultiFab>& S_data,
105  const amrex::MultiFab& S_prim,
106  amrex::MultiFab& avg_xmom,
107  amrex::MultiFab& avg_ymom,
108  amrex::MultiFab& avg_zmom,
109  const amrex::MultiFab& xvel,
110  const amrex::MultiFab& yvel,
111  const amrex::MultiFab& zvel,
112  const amrex::MultiFab& source,
113  const amrex::MultiFab* SmnSmn,
114  const amrex::MultiFab* eddyDiffs,
115  amrex::MultiFab* Hfx1,
116  amrex::MultiFab* Hfx2,
117  amrex::MultiFab* Hfx3,
118  amrex::MultiFab* Q1fx1,
119  amrex::MultiFab* Q1fx2,
120  amrex::MultiFab* Q1fx3,
121  amrex::MultiFab* Q2fx3,
122  amrex::MultiFab* Diss,
123  const amrex::Geometry geom,
124  const SolverChoice& solverChoice,
125  std::unique_ptr<SurfaceLayer>& SurfLayer,
126  const amrex::Gpu::DeviceVector<amrex::BCRec>& domain_bcs_type_d,
127  const amrex::Vector<amrex::BCRec>& domain_bcs_type,
128  std::unique_ptr<amrex::MultiFab>& z_phys_nd,
129  std::unique_ptr<amrex::MultiFab>& z_phys_cc,
130  std::unique_ptr<amrex::MultiFab>& ax,
131  std::unique_ptr<amrex::MultiFab>& ay,
132  std::unique_ptr<amrex::MultiFab>& az,
133  std::unique_ptr<amrex::MultiFab>& dJ_old,
134  amrex::MultiFab* dJ_new,
135  amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
136  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& mapfac,
137  amrex::EBFArrayBoxFactory const& ebfact,
138 #if defined(ERF_USE_NETCDF)
139  const bool& moist_zero,
140  const amrex::Real& bdy_time_interval,
141  const amrex::Real& new_stage_time,
142  const amrex::Real& stop_time,
143  int width,
144  int set_width,
145  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xlo,
146  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xhi,
147  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_ylo,
148  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_yhi,
149 #endif
150 #ifdef ERF_USE_SHOC
151  std::unique_ptr<SHOCInterface>& shoc_lev,
152 #endif
153  amrex::YAFluxRegister* fr_as_crse,
154  amrex::YAFluxRegister* fr_as_fine);
155 
156 #endif
amrex::Real Real
Definition: ERF_ShocInterface.H:19
void erf_make_tau_terms(int level, int nrk, const amrex::Vector< amrex::BCRec > &domain_bcs_type, const amrex::MultiFab &z_phys_nd, amrex::Vector< amrex::MultiFab > &S_data, const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &zvel, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &Tau_lev, amrex::MultiFab *SmnSmn, amrex::MultiFab *eddyDiffs, const amrex::Geometry geom, const SolverChoice &solverChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, const amrex::MultiFab &detJ, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &mapfac)
void erf_slow_rhs_post(int level, int finest_level, int nrk, amrex::Real dt, int n_qstate, amrex::Vector< amrex::MultiFab > &S_rhs, amrex::Vector< amrex::MultiFab > &S_old, amrex::Vector< amrex::MultiFab > &S_new, amrex::Vector< amrex::MultiFab > &S_data, const amrex::MultiFab &S_prim, amrex::MultiFab &avg_xmom, amrex::MultiFab &avg_ymom, amrex::MultiFab &avg_zmom, const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &zvel, const amrex::MultiFab &source, const amrex::MultiFab *SmnSmn, const amrex::MultiFab *eddyDiffs, amrex::MultiFab *Hfx1, amrex::MultiFab *Hfx2, amrex::MultiFab *Hfx3, amrex::MultiFab *Q1fx1, amrex::MultiFab *Q1fx2, amrex::MultiFab *Q1fx3, amrex::MultiFab *Q2fx3, amrex::MultiFab *Diss, const amrex::Geometry geom, const SolverChoice &solverChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, const amrex::Gpu::DeviceVector< amrex::BCRec > &domain_bcs_type_d, const amrex::Vector< amrex::BCRec > &domain_bcs_type, std::unique_ptr< amrex::MultiFab > &z_phys_nd, std::unique_ptr< amrex::MultiFab > &z_phys_cc, std::unique_ptr< amrex::MultiFab > &ax, std::unique_ptr< amrex::MultiFab > &ay, std::unique_ptr< amrex::MultiFab > &az, std::unique_ptr< amrex::MultiFab > &dJ_old, amrex::MultiFab *dJ_new, amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &mapfac, amrex::EBFArrayBoxFactory const &ebfact, amrex::YAFluxRegister *fr_as_crse, amrex::YAFluxRegister *fr_as_fine)
void erf_slow_rhs_pre(int level, int finest_level, int nrk, amrex::Real dt, amrex::Vector< amrex::MultiFab > &S_rhs, amrex::Vector< amrex::MultiFab > &S_old, amrex::Vector< amrex::MultiFab > &S_data, const amrex::MultiFab &S_prim, const amrex::MultiFab &qt, amrex::MultiFab &avg_xmom, amrex::MultiFab &avg_ymom, amrex::MultiFab &avg_zmom, const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &zvel, std::unique_ptr< amrex::MultiFab > &z_t, const amrex::MultiFab &cc_src, const amrex::MultiFab &xmom_src, const amrex::MultiFab &ymom_src, const amrex::MultiFab &zmom_src, const amrex::MultiFab &buoyancy, const amrex::MultiFab *zmom_crse_rhs, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &Tau_lev, amrex::MultiFab *SmnSmn, amrex::MultiFab *eddyDiffs, amrex::MultiFab *Hfx1, amrex::MultiFab *Hfx2, amrex::MultiFab *Hfx3, amrex::MultiFab *Q1fx1, amrex::MultiFab *Q1fx2, amrex::MultiFab *Q1fx3, amrex::MultiFab *Q2fx3, amrex::MultiFab *Diss, const amrex::Geometry geom, const SolverChoice &solverChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, const amrex::Gpu::DeviceVector< amrex::BCRec > &domain_bcs_type_d, const amrex::Vector< amrex::BCRec > &domain_bcs_type, const amrex::MultiFab &z_phys_nd, const amrex::MultiFab &z_phys_cc, const amrex::MultiFab &ax, const amrex::MultiFab &ay, const amrex::MultiFab &az, const amrex::MultiFab &dJ, amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, amrex::Vector< amrex::MultiFab > &gradp, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &mapfac, const eb_ &ebfact, amrex::YAFluxRegister *fr_as_crse, amrex::YAFluxRegister *fr_as_fine)
Definition: ERF_EB.H:13
@ qt
Definition: ERF_Kessler.H:27
@ xvel
Definition: ERF_IndexDefines.H:141
@ zvel
Definition: ERF_IndexDefines.H:143
@ yvel
Definition: ERF_IndexDefines.H:142
Definition: ERF_DataStruct.H:123