ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
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::Vector<amrex::MultiFab>& S_scratch,
50  const amrex::MultiFab& xvel,
51  const amrex::MultiFab& yvel,
52  const amrex::MultiFab& zvel,
53  std::unique_ptr<amrex::MultiFab>& z_t,
54  const amrex::MultiFab& cc_src,
55  const amrex::MultiFab& xmom_src,
56  const amrex::MultiFab& ymom_src,
57  const amrex::MultiFab& zmom_src,
58  const amrex::MultiFab& buoyancy,
59  const amrex::MultiFab* zmom_crse_rhs,
60  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& Tau_lev,
61  amrex::MultiFab* SmnSmn,
62  amrex::MultiFab* eddyDiffs,
63  amrex::MultiFab* Hfx1,
64  amrex::MultiFab* Hfx2,
65  amrex::MultiFab* Hfx3,
66  amrex::MultiFab* Q1fx1,
67  amrex::MultiFab* Q1fx2,
68  amrex::MultiFab* Q1fx3,
69  amrex::MultiFab* Q2fx3,
70  amrex::MultiFab* Diss,
71  const amrex::Geometry geom,
72  const SolverChoice& solverChoice,
73  std::unique_ptr<SurfaceLayer>& SurfLayer,
74  const amrex::Gpu::DeviceVector<amrex::BCRec>& domain_bcs_type_d,
75  const amrex::Vector<amrex::BCRec>& domain_bcs_type,
76  const amrex::MultiFab& z_phys_nd,
77  const amrex::MultiFab& z_phys_cc,
78  const amrex::MultiFab& ax,
79  const amrex::MultiFab& ay,
80  const amrex::MultiFab& az,
81  const amrex::MultiFab& dJ,
82  amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
83  amrex::Vector<amrex::MultiFab>& gradp,
84  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& mapfac,
85  const eb_& ebfact,
86  amrex::YAFluxRegister* fr_as_crse,
87  amrex::YAFluxRegister* fr_as_fine);
88 
89 /**
90  * Function for computing the slow RHS for the evolution equations for the scalars other than density or potential temperature
91  *
92  */
93 void erf_slow_rhs_post (int level, int finest_level, int nrk,
94  amrex::Real dt,
95  int n_qstate,
96  amrex::Vector<amrex::MultiFab>& S_rhs,
97  amrex::Vector<amrex::MultiFab>& S_old,
98  amrex::Vector<amrex::MultiFab>& S_new,
99  amrex::Vector<amrex::MultiFab>& S_data,
100  const amrex::MultiFab& S_prim,
101  amrex::Vector<amrex::MultiFab >& S_scratch,
102  const amrex::MultiFab& xvel,
103  const amrex::MultiFab& yvel,
104  const amrex::MultiFab& zvel,
105  const amrex::MultiFab& source,
106  const amrex::MultiFab* SmnSmn,
107  const amrex::MultiFab* eddyDiffs,
108  amrex::MultiFab* Hfx1,
109  amrex::MultiFab* Hfx2,
110  amrex::MultiFab* Hfx3,
111  amrex::MultiFab* Q1fx1,
112  amrex::MultiFab* Q1fx2,
113  amrex::MultiFab* Q1fx3,
114  amrex::MultiFab* Q2fx3,
115  amrex::MultiFab* Diss,
116  const amrex::Geometry geom,
117  const SolverChoice& solverChoice,
118  std::unique_ptr<SurfaceLayer>& SurfLayer,
119  const amrex::Gpu::DeviceVector<amrex::BCRec>& domain_bcs_type_d,
120  const amrex::Vector<amrex::BCRec>& domain_bcs_type,
121  std::unique_ptr<amrex::MultiFab>& z_phys_nd,
122  std::unique_ptr<amrex::MultiFab>& z_phys_cc,
123  std::unique_ptr<amrex::MultiFab>& ax,
124  std::unique_ptr<amrex::MultiFab>& ay,
125  std::unique_ptr<amrex::MultiFab>& az,
126  std::unique_ptr<amrex::MultiFab>& dJ_old,
127  amrex::MultiFab* dJ_new,
128  amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
129  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& mapfac,
130  amrex::EBFArrayBoxFactory const& ebfact,
131 #if defined(ERF_USE_NETCDF)
132  const bool& moist_zero,
133  const amrex::Real& bdy_time_interval,
134  const amrex::Real& start_bdy_time,
135  const amrex::Real& new_stage_time,
136  int width,
137  int set_width,
138  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xlo,
139  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xhi,
140  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_ylo,
141  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_yhi,
142 #endif
143  amrex::YAFluxRegister* fr_as_crse,
144  amrex::YAFluxRegister* fr_as_fine);
145 
146 #endif
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::Vector< amrex::MultiFab > &S_scratch, 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_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_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::Vector< amrex::MultiFab > &S_scratch, 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:99