1 #ifndef ERF_PBLMODELS_H_
2 #define ERF_PBLMODELS_H_
25 const amrex::MultiFab&
yvel,
26 const amrex::MultiFab& cons_in,
27 amrex::MultiFab& eddyViscosity,
28 const amrex::Geometry& geom,
30 std::unique_ptr<SurfaceLayer>& SurfLayer,
31 bool use_terrain_fitted_coords,
34 const amrex::BCRec* bc_ptr,
36 const std::unique_ptr<amrex::MultiFab>& z_phys_nd,
58 const amrex::MultiFab&
yvel,
59 const amrex::MultiFab& cons_in,
60 amrex::MultiFab& eddyViscosity,
61 const amrex::Geometry& geom,
63 std::unique_ptr<SurfaceLayer>& SurfLayer,
64 bool use_terrain_fitted_coords,
67 const amrex::BCRec* bc_ptr,
69 const std::unique_ptr<amrex::MultiFab>& z_phys_nd,
91 const amrex::MultiFab&
yvel,
92 const amrex::MultiFab& cons_in,
93 amrex::MultiFab& eddyViscosity,
94 const amrex::Geometry& geom,
96 std::unique_ptr<SurfaceLayer>& SurfLayer,
97 bool use_terrain_fitted_coords,
100 const amrex::BCRec* bc_ptr,
102 const std::unique_ptr<amrex::MultiFab>& z_phys_nd,
123 const amrex::MultiFab&
yvel,
124 const amrex::MultiFab& cons_in,
125 amrex::MultiFab& eddyViscosity,
126 const amrex::Geometry& geom,
128 std::unique_ptr<SurfaceLayer>& SurfLayer,
129 bool use_terrain_fitted_coords,
132 const amrex::BCRec* bc_ptr,
134 const std::unique_ptr<amrex::MultiFab>& z_phys_nd,
148 const amrex::Array4<const amrex::Real>& uvel,
149 const amrex::Array4<const amrex::Real>& vvel,
150 const amrex::Array4<const amrex::Real>& cell_data,
153 const amrex::Real& dz_inv,
154 const bool c_ext_dir_on_zlo,
155 const bool c_ext_dir_on_zhi,
156 const bool u_ext_dir_on_zlo,
157 const bool u_ext_dir_on_zhi,
158 const bool v_ext_dir_on_zlo,
159 const bool v_ext_dir_on_zhi,
160 amrex::Real& dthetadz,
165 if ( k==izmax && c_ext_dir_on_zhi ) {
166 dthetadz = (1.0/3.0)*( -
GetThetav(i,j,k-1,cell_data,moisture_indices)
167 - 3.0 *
GetThetav(i,j,k ,cell_data,moisture_indices)
168 + 4.0 *
GetThetav(i,j,k+1,cell_data,moisture_indices) )*dz_inv;
169 }
else if ( k==izmin && c_ext_dir_on_zlo ) {
170 dthetadz = (1.0/3.0)*(
GetThetav(i,j,k+1,cell_data,moisture_indices)
171 + 3.0 *
GetThetav(i,j,k ,cell_data,moisture_indices)
172 - 4.0 *
GetThetav(i,j,k-1,cell_data,moisture_indices) )*dz_inv;
174 dthetadz = 0.5*(
GetThetav(i,j,k+1,cell_data,moisture_indices)
175 -
GetThetav(i,j,k-1,cell_data,moisture_indices) )*dz_inv;
178 if ( k==izmax && u_ext_dir_on_zhi ) {
179 dudz = (1.0/6.0)*( (-uvel(i ,j,k-1) - 3.0 * uvel(i ,j,k ) + 4.0 * uvel(i ,j,k+1))
180 + (-uvel(i+1,j,k-1) - 3.0 * uvel(i+1,j,k ) + 4.0 * uvel(i+1,j,k+1)) )*dz_inv;
181 }
else if ( k==izmin && u_ext_dir_on_zlo ) {
182 dudz = (1.0/6.0)*( (uvel(i ,j,k+1) + 3.0 * uvel(i ,j,k ) - 4.0 * uvel(i ,j,k-1))
183 + (uvel(i+1,j,k+1) + 3.0 * uvel(i+1,j,k ) - 4.0 * uvel(i+1,j,k-1)) )*dz_inv;
185 dudz = 0.25*( uvel(i,j,k+1) - uvel(i,j,k-1) + uvel(i+1,j,k+1) - uvel(i+1,j,k-1) )*dz_inv;
188 if ( k==izmax && v_ext_dir_on_zhi ) {
189 dvdz = (1.0/6.0)*( (-vvel(i,j ,k-1) - 3.0 * vvel(i,j ,k ) + 4.0 * vvel(i,j ,k+1))
190 + (-vvel(i,j+1,k-1) - 3.0 * vvel(i,j+1,k ) + 4.0 * vvel(i,j+1,k+1)) )*dz_inv;
191 }
else if ( k==izmin && v_ext_dir_on_zlo ) {
192 dvdz = (1.0/6.0)*( (vvel(i,j ,k+1) + 3.0 * vvel(i,j ,k ) - 4.0 * vvel(i,j ,k-1))
193 + (vvel(i,j+1,k+1) + 3.0 * vvel(i,j+1,k ) - 4.0 * vvel(i,j+1,k-1)) )*dz_inv;
195 dvdz = 0.25*( vvel(i,j,k+1) - vvel(i,j,k-1) + vvel(i,j+1,k+1) - vvel(i,j+1,k-1) )*dz_inv;
216 const amrex::Array4<const amrex::Real>& uvel,
217 const amrex::Array4<const amrex::Real>& vvel,
218 const amrex::Array4<const amrex::Real>& cell_data,
219 const amrex::Array4<const amrex::Real>& cell_prim,
220 const amrex::Array4<const amrex::Real>& K_turb,
221 const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& cellSizeInv,
222 const amrex::Box& domain,
223 amrex::Real pbl_mynn_B1_l,
224 const amrex::Real theta_mean,
226 bool c_ext_dir_on_zlo,
227 bool c_ext_dir_on_zhi,
228 bool u_ext_dir_on_zlo,
229 bool u_ext_dir_on_zhi,
230 bool v_ext_dir_on_zlo,
231 bool v_ext_dir_on_zhi,
232 const amrex::Real met_h_zeta=1.0)
235 amrex::Real dthetadz, dudz, dvdz;
236 amrex::Real source_term = 0.0;
238 amrex::Real dz_inv = cellSizeInv[2];
239 int izmin = domain.smallEnd(2);
240 int izmax = domain.bigEnd(2);
243 uvel, vvel, cell_data, izmin, izmax, dz_inv/met_h_zeta,
244 c_ext_dir_on_zlo, c_ext_dir_on_zhi,
245 u_ext_dir_on_zlo, u_ext_dir_on_zhi,
246 v_ext_dir_on_zlo, v_ext_dir_on_zhi,
247 dthetadz, dudz, dvdz,
262 source_term += K_turb(i,j,k,
EddyDiff::Mom_v) * (dudz*dudz + dvdz*dvdz);
268 amrex::Real qke = 2.0 * cell_prim(i,j,k,
PrimKE_comp);
269 if (std::abs(qke) > 0.0) {
270 source_term -= cell_data(i,j,k,
Rho_comp) * std::pow(qke,1.5) /
constexpr amrex::Real CONST_GRAV
Definition: ERF_Constants.H:21
TurbChoice turbChoice
Definition: ERF_DiffSetup.H:2
#define Rho_comp
Definition: ERF_IndexDefines.H:36
#define PrimKE_comp
Definition: ERF_IndexDefines.H:51
AMREX_GPU_DEVICE AMREX_FORCE_INLINE amrex::Real GetThetav(int i, int j, int k, const amrex::Array4< amrex::Real const > &cell_data, const MoistureComponentIndices &moisture_indices)
Definition: ERF_MoistUtils.H:72
AMREX_GPU_DEVICE AMREX_FORCE_INLINE amrex::Real ComputeQKESourceTerms(int i, int j, int k, const amrex::Array4< const amrex::Real > &uvel, const amrex::Array4< const amrex::Real > &vvel, const amrex::Array4< const amrex::Real > &cell_data, const amrex::Array4< const amrex::Real > &cell_prim, const amrex::Array4< const amrex::Real > &K_turb, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &cellSizeInv, const amrex::Box &domain, amrex::Real pbl_mynn_B1_l, const amrex::Real theta_mean, const MoistureComponentIndices &moisture_indices, bool c_ext_dir_on_zlo, bool c_ext_dir_on_zhi, bool u_ext_dir_on_zlo, bool u_ext_dir_on_zhi, bool v_ext_dir_on_zlo, bool v_ext_dir_on_zhi, const amrex::Real met_h_zeta=1.0)
Definition: ERF_PBLModels.H:215
AMREX_GPU_DEVICE AMREX_FORCE_INLINE void ComputeVerticalDerivativesPBL(int i, int j, int k, const amrex::Array4< const amrex::Real > &uvel, const amrex::Array4< const amrex::Real > &vvel, const amrex::Array4< const amrex::Real > &cell_data, const int izmin, const int izmax, const amrex::Real &dz_inv, const bool c_ext_dir_on_zlo, const bool c_ext_dir_on_zhi, const bool u_ext_dir_on_zlo, const bool u_ext_dir_on_zhi, const bool v_ext_dir_on_zlo, const bool v_ext_dir_on_zhi, amrex::Real &dthetadz, amrex::Real &dudz, amrex::Real &dvdz, const MoistureComponentIndices &moisture_indices)
Definition: ERF_PBLModels.H:147
void ComputeDiffusivityMYNN25(const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &cons_in, amrex::MultiFab &eddyViscosity, const amrex::Geometry &geom, const TurbChoice &turbChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, bool use_terrain_fitted_coords, bool use_moisture, int level, const amrex::BCRec *bc_ptr, bool, const std::unique_ptr< amrex::MultiFab > &z_phys_nd, const MoistureComponentIndices &moisture_indices)
void ComputeDiffusivityYSU(const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &cons_in, amrex::MultiFab &eddyViscosity, const amrex::Geometry &geom, const TurbChoice &turbChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, bool use_terrain_fitted_coords, bool use_moisture, int level, const amrex::BCRec *bc_ptr, bool, const std::unique_ptr< amrex::MultiFab > &z_phys_nd, const MoistureComponentIndices &moisture_indices)
void ComputeDiffusivityMYNNEDMF(const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &cons_in, amrex::MultiFab &eddyViscosity, const amrex::Geometry &geom, const TurbChoice &turbChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, bool use_terrain_fitted_coords, bool use_moisture, int level, const amrex::BCRec *bc_ptr, bool, const std::unique_ptr< amrex::MultiFab > &z_phys_nd, const MoistureComponentIndices &moisture_indices)
void ComputeDiffusivityMRF(const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &cons_in, amrex::MultiFab &eddyViscosity, const amrex::Geometry &geom, const TurbChoice &turbChoice, std::unique_ptr< SurfaceLayer > &SurfLayer, bool use_terrain_fitted_coords, bool use_moisture, int level, const amrex::BCRec *bc_ptr, bool, const std::unique_ptr< amrex::MultiFab > &z_phys_nd, const MoistureComponentIndices &moisture_indices)
@ Theta_v
Definition: ERF_IndexDefines.H:176
@ Turb_lengthscale
Definition: ERF_IndexDefines.H:180
@ Mom_v
Definition: ERF_IndexDefines.H:175
@ xvel
Definition: ERF_IndexDefines.H:141
@ yvel
Definition: ERF_IndexDefines.H:142
Definition: ERF_DataStruct.H:99
Definition: ERF_TurbStruct.H:41