ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_TI_slow_rhs_pre.H
Go to the documentation of this file.
1 #include "ERF_SrcHeaders.H"
2 
3 /**
4  * Wrapper for calling the routine that creates the slow RHS
5  */
6  auto slow_rhs_fun_pre = [&,zero_d=zero,one_d=one](Vector<MultiFab>& S_rhs,
7  Vector<MultiFab>& S_old,
8  Vector<MultiFab>& S_data,
9  const double old_step_time,
10  const double old_stage_time,
11  const double new_stage_time,
12  const int nrk)
13  {
14  BL_PROFILE("slow_rhs_fun_pre");
15  //
16  // Define primitive variables for all later RK stages
17  // (We have already done this for the first RK step)
18  // Note that it is essential this happen before the call to make_mom_sources
19  // because some of the buoyancy routines use the primitive variables
20  //
21  if (nrk > 0) {
22  int ng_cons = S_data[IntVars::cons].nGrow();
23  cons_to_prim(S_data[IntVars::cons], S_prim, ng_cons);
25  }
26 
27  if (verbose) Print() << std::setprecision(timeprecision)
28  << "Making slow rhs at time " << old_stage_time
29  << " for fast variables advancing from " << old_step_time
30  << " to " << new_stage_time << std::endl;
31 
32  double slow_dt = new_stage_time - old_step_time;
33 
34  const GpuArray<Real, AMREX_SPACEDIM> dxInv = fine_geom.InvCellSizeArray();
35 
36  // *************************************************************************
37  // Set up flux registers if using two_way coupling
38  // *************************************************************************
39  YAFluxRegister* fr_as_crse = nullptr;
40  YAFluxRegister* fr_as_fine = nullptr;
41  if (solverChoice.coupling_type == CouplingType::TwoWay && finest_level > 0) {
42  if (level < finest_level) {
43  fr_as_crse = getAdvFluxReg(level+1);
44  fr_as_crse->reset();
45  }
46  if (level > 0) {
47  fr_as_fine = getAdvFluxReg(level);
48  }
49  }
50 
51  // *************************************************************************
52  // Get multifab pointers
53  // *************************************************************************
54 
55  // Canopy data for mom sources
56  MultiFab* forest_drag = (solverChoice.do_forest_drag) ?
57  m_forest_drag[level]->get_drag_field() : nullptr;
58  MultiFab* frontal_area = (solverChoice.do_forest_drag) ?
59  m_forest_drag[level]->get_frontal_area() : nullptr;
60 
61  // Immersed Forcing
62  MultiFab* terrain_blank = (solverChoice.terrain_type == TerrainType::ImmersedForcing ||
63  solverChoice.buildings_type == BuildingsType::ImmersedForcing) ?
64  terrain_blanking[level].get() : nullptr;
65  MultiFab* terrain_blank_xface = (solverChoice.terrain_type == TerrainType::ImmersedForcing ||
66  solverChoice.buildings_type == BuildingsType::ImmersedForcing) ?
67  terrain_blanking_xface[level].get() : nullptr;
68  MultiFab* terrain_blank_yface = (solverChoice.terrain_type == TerrainType::ImmersedForcing ||
69  solverChoice.buildings_type == BuildingsType::ImmersedForcing) ?
70  terrain_blanking_yface[level].get() : nullptr;
71  MultiFab* terrain_blank_zface = (solverChoice.terrain_type == TerrainType::ImmersedForcing ||
72  solverChoice.buildings_type == BuildingsType::ImmersedForcing) ?
73  terrain_blanking_zface[level].get() : nullptr;
74 
75  // Update the total moisture variable *before* computing sources since this is used in
76  // the buoyancy calculation
77  if (solverChoice.moisture_type != MoistureType::None) {
78  int n_qstate_into_total = micro->Get_Qstate_Moist_Size() - micro->Get_Qstate_Moist_NumConc_Size();
79  make_qt(S_data[IntVars::cons], qt, n_qstate_into_total);
80  }
81 
82  MultiFab p0_to_use, base_to_use;
83  MultiFab *zpn_to_use, *zpc_to_use, *ax_to_use, *ay_to_use, *az_to_use, *dJ_to_use;
84 
85  // Moving terrain
86  std::unique_ptr<MultiFab> z_t_pert;
87  if ( solverChoice.terrain_type == TerrainType::MovingFittedMesh )
88  {
89  z_t_pert = std::make_unique<MultiFab>(S_data[IntVars::zmom].boxArray(), S_data[IntVars::zmom].DistributionMap(), 1, 1);
90  update_terrain_stage(level, old_step_time, old_stage_time, new_stage_time, slow_dt);
91 
92  p0_to_use = MultiFab(base_state_new[level], make_alias, BaseState::p0_comp, 1);
93  base_to_use = MultiFab(base_state_new[level], make_alias, 0, BaseState::num_comps);
94  zpn_to_use = z_phys_nd_src[level].get();
95  zpc_to_use = z_phys_cc_src[level].get();
96  ax_to_use = ax_src[level].get();
97  ay_to_use = ay_src[level].get();
98  az_to_use = az_src[level].get();
99  dJ_to_use = detJ_cc_src[level].get();
100 
101  } else {
102  p0_to_use = MultiFab(base_state[level], make_alias, BaseState::p0_comp, 1);
103  base_to_use = MultiFab(base_state[level], make_alias, 0, BaseState::num_comps);
104  zpn_to_use = z_phys_nd[level].get();
105  zpc_to_use = z_phys_cc[level].get();
106  ax_to_use = ax[level].get();
107  ay_to_use = ay[level].get();
108  az_to_use = az[level].get();
109  dJ_to_use = detJ_cc[level].get();
110  }
111 
112  // Get planar averages from persistent storage for immersed forcing
113  bool l_use_IF = (solverChoice.terrain_type == TerrainType::ImmersedForcing ||
114  solverChoice.buildings_type == BuildingsType::ImmersedForcing);
115  Table1D<Real> r_avg_to_pass = l_use_IF ? r_plane_avg[level].table() : Table1D<Real>();
116  Table1D<Real> t_avg_to_pass = l_use_IF ? t_plane_avg[level].table() : Table1D<Real>();
117 
118  // *****************************************************************************
119  // Construct the source terms for the cell-centered (conserved) variables
120  // *****************************************************************************
121  make_sources(level, nrk, slow_dt, old_stage_time,
122  S_data, S_prim, cc_src, base_state[level], zpc_to_use,
123  xvel_new, yvel_new, zvel_new,
124  qheating_rates[level].get(),
125  terrain_blank, fine_geom, solverChoice,
126  mapfac[level],
127  rhotheta_src[level].get(), rhoqt_src[level].get(),
128  dptr_wbar_sub, d_rayleigh_ptrs_at_lev, d_sinesq_at_lev,
129  turbPert,
130  r_avg_to_pass, t_avg_to_pass,
131  true);
132 
133  // Heat flux from the building faces (immersed-boundary surface energy
134  // balance), added to the freshly rebuilt source; no-op unless enabled.
135  if (ibseb_params.enable && ibseb_params.couple_heat &&
136  level < static_cast<int>(m_ibseb.size()) && m_ibseb[level]) {
137  m_ibseb[level]->add_heat_flux_to_source(cc_src, S_data[IntVars::cons], fine_geom,
138  solverChoice.c_p, solverChoice.rdOcp);
139  }
140 
141  // *****************************************************************************
142  // Add nudging terms on moist variables if we have a non-zero relaxation region
143  // *****************************************************************************
144 #if defined(ERF_USE_NETCDF)
145  if ( solverChoice.use_real_bcs && (level==0) &&
146  (solverChoice.moisture_type != MoistureType::None) )
147  {
148  Real bdy_factor = solverChoice.bdy_nudge_factor;
149  Real l_rdOcp = solverChoice.rdOcp;
150  Real l_c_p = solverChoice.c_p;
151  int moist_nudge_type = solverChoice.bdy_moist_nudge_type;
152  int num_q = micro->Get_Qstate_Moist_Size() - micro->Get_Qstate_Moist_NumConc_Size();
153  AMREX_ALWAYS_ASSERT( (solverChoice.bdy_moist_nudge_type != 2) ||
154  (solverChoice.moisture_type != MoistureType::Morrison_NoIce &&
155  solverChoice.moisture_type != MoistureType::SAM_NoIce) );
156  add_moist_nudging_terms(S_data[IntVars::cons], cc_src, num_q, slow_dt,
157  start_time+old_stage_time,
158  start_bdy_time, final_bdy_time, bdy_time_interval,
159  bdy_factor, real_width, geom[level],
160  bdy_data_xlo, bdy_data_xhi, bdy_data_ylo, bdy_data_yhi,
161  m_r2d, l_c_p, l_rdOcp,
162  solverChoice.moisture_indices,
163  solverChoice.use_wrf_bdy_density, moist_nudge_type);
164  }
165 #endif
166 
167  // Accumulate fixed-leaf-temperature canopy exchange in the staged
168  // source storage. The existing slow-pre and slow-post paths then
169  // apply rho-theta and rho-qv with their normal RK/anelastic weighting.
170  if (solverChoice.do_forest_drag && solverChoice.forest_biophysics &&
171  solverChoice.forest_biophysics_heat && frontal_area) {
173  S_data[IntVars::cons],
174  xvel_new, yvel_new, zvel_new,
175  frontal_area, base_to_use,
176  solverChoice);
177  }
178 
179  // *****************************************************************************
180  // Define the pressure gradient
181  // *****************************************************************************
182  make_gradp_pert(level, solverChoice, fine_geom, S_data,
183  p0_to_use, *zpn_to_use, *zpc_to_use,
184  mapfac[level],
185  get_eb(level), gradp[level]);
186 
187  // *****************************************************************************
188  // Define the buoyancy forcing term in the z-direction
189  // *****************************************************************************
190  int num_q = micro->Get_Qstate_Moist_Size() - micro->Get_Qstate_Moist_NumConc_Size();
191  make_buoyancy(level, S_data, S_prim, qt, buoyancy, fine_geom, solverChoice, base_to_use,
192  num_q, get_eb(level), solverChoice.anelastic[level]);
193 
194  // *****************************************************************************
195  // Make remaining (not gradp or buoyancy) momentum sources
196  // *****************************************************************************
197  make_mom_sources(old_stage_time, slow_dt,
198  S_data, zpn_to_use, zpc_to_use, stretched_dz_h[level],
199  xvel_new, yvel_new, zvel_new,
200  xmom_src, ymom_src, zmom_src,
201  base_to_use, forest_drag, terrain_blank,
202  terrain_blank_xface, terrain_blank_yface, terrain_blank_zface,
203  cosPhi_m[level].get(), sinPhi_m[level].get(), fine_geom, solverChoice,
204  mapfac[level],
205  (solverChoice.have_geo_wind_profile) ? d_u_geos[level].data(): nullptr,
206  (solverChoice.have_geo_wind_profile) ? d_v_geos[level].data(): nullptr,
207  dptr_wbar_sub, d_rayleigh_ptrs_at_lev, d_sinesq_at_lev, d_sinesq_stag_at_lev,
208  d_sponge_ptrs_at_lev,
209  (solverChoice.hindcast_lateral_forcing? &forecast_state_interp[level] : nullptr),
210  input_sounding_data, lsf, lsf_data[level],
211  get_eb(level), true);
212 
213  // *****************************************************************************
214  // Add body sources if doing flow around a body
215  // *****************************************************************************
216  add_thin_body_sources(xmom_src, ymom_src, zmom_src,
217  xflux_imask[level], yflux_imask[level], zflux_imask[level],
218  thin_xforce[level], thin_yforce[level], thin_zforce[level]);
219 
220  // *****************************************************************************
221  // Define RHS for rho, rho_theta and momenta
222  // *****************************************************************************
223  erf_slow_rhs_pre(level, finest_level, nrk, slow_dt, S_rhs, S_old, S_data,
224  S_prim, qt, avg_xmom[level], avg_ymom[level], avg_zmom[level],
225  xvel_new, yvel_new, zvel_new,
226  z_t_rk[level], cc_src, xmom_src, ymom_src, zmom_src, buoyancy,
227  (level > 0) ? &zmom_crse_rhs[level] : nullptr,
228  Tau[level], Tau_corr[level], Tau_EB[level],
229  SmnSmn, eddyDiffs, Hfx1, Hfx2, Hfx3, Q1fx1, Q1fx2, Q1fx3, Q2fx3, Diss, Hfx3_EB,
230  fine_geom, solverChoice, m_SurfaceLayer, domain_bcs_type_d, domain_bcs_type,
231  *zpn_to_use, *zpc_to_use, *ax_to_use, *ay_to_use, *az_to_use, *dJ_to_use,
232  stretched_dz_d[level], gradp[level],
233  mapfac[level], get_eb(level),
234 #ifdef ERF_USE_EAMXX_SHOC
235  eamxx_shoc_interface[level].get(),
236 #endif
237  native_shoc_driver[level].get(),
238  fr_as_crse, fr_as_fine, &base_to_use,
239  cloud_chamber_config.active ? &cloud_chamber_config : nullptr,
240  cloud_chamber_budget.get());
241 
242  if ((solverChoice.vert_implicit_fac[level][nrk] > zero_d) && solverChoice.implicit_before_substep) {
243  const Real stage_dt = static_cast<Real>(slow_dt);
244  /**
245  * @brief Temporary MultiFab for conserved variable updates.
246  */
247  MultiFab scratch(S_data[IntVars::cons].boxArray(),S_data[IntVars::cons].DistributionMap(), 2,
248  S_data[IntVars::cons].nGrowVect());
249  MultiFab::Copy(scratch, S_old[IntVars::cons], 0, 0, 2, S_data[IntVars::cons].nGrowVect()); // scratch := S_old (for rho, rhotheta)
250  MultiFab::Saxpy(scratch, stage_dt, S_rhs[IntVars::cons], 0, 0, 2, 0); // scratch := S_old + stage_dt*Src (for rho, rhotheta)
251  scratch.FillBoundary(geom[level].periodicity());
252 
253  /**
254  * @brief Temporary MultiFab for x-momentum updates.
255  */
256  MultiFab scratch_xmom(S_data[IntVars::xmom].boxArray(),
257  S_data[IntVars::xmom].DistributionMap(), 1,
258  S_data[IntVars::xmom].nGrowVect());
259  /**
260  * @brief Temporary MultiFab for y-momentum updates.
261  */
262  MultiFab scratch_ymom(S_data[IntVars::ymom].boxArray(),
263  S_data[IntVars::ymom].DistributionMap(), 1,
264  S_data[IntVars::ymom].nGrowVect());
265 #ifdef ERF_IMPLICIT_W
266  /**
267  * @brief Temporary MultiFab for z-momentum updates.
268  */
269  MultiFab scratch_zmom(S_data[IntVars::zmom].boxArray(),
270  S_data[IntVars::zmom].DistributionMap(), 1,
271  S_data[IntVars::zmom].nGrowVect());
272 #endif
273  if (solverChoice.implicit_momentum_diffusion) {
274  MultiFab::Copy(scratch_xmom, S_old[IntVars::xmom], 0, 0, 1, S_data[IntVars::xmom].nGrowVect()); // scratch := S_old
275  MultiFab::Saxpy(scratch_xmom, stage_dt, S_rhs[IntVars::xmom], 0, 0, 1, 0); // scratch := S_old + stage_dt*Src
276  scratch_xmom.FillBoundary(geom[level].periodicity());
277 
278  MultiFab::Copy(scratch_ymom, S_old[IntVars::ymom], 0, 0, 1, S_data[IntVars::ymom].nGrowVect()); // scratch := S_old
279  MultiFab::Saxpy(scratch_ymom, stage_dt, S_rhs[IntVars::ymom], 0, 0, 1, 0); // scratch := S_old + stage_dt*Src
280  scratch_ymom.FillBoundary(geom[level].periodicity());
281 #ifdef ERF_IMPLICIT_W
282  MultiFab::Copy(scratch_zmom, S_old[IntVars::zmom], 0, 0, 1, S_data[IntVars::zmom].nGrowVect()); // scratch := S_old
283  MultiFab::Saxpy(scratch_zmom, stage_dt, S_rhs[IntVars::zmom], 0, 0, 1, 0); // scratch := S_old + stage_dt*Src
284  scratch_zmom.FillBoundary(geom[level].periodicity());
285 #endif
286  }
287 
288 #include "ERF_ImplicitPre.H"
289 
290  MultiFab::Saxpy(scratch, -one_d, S_old[IntVars::cons], 1, 1, 1, 0); // scratch := (S_new - S_old) (for rhotheta only)
291  scratch.mult(one_d / stage_dt); // scratch := (S_new - S_old) / stage_dt
292  MultiFab::Copy(S_rhs[IntVars::cons], scratch, 1, 1, 1, 0); // slow_rhs := (S_new - S_old) / stage_dt (for rhotheta only)
293 
294  if (solverChoice.implicit_momentum_diffusion) {
295  MultiFab::Saxpy(scratch_xmom, -one_d, S_old[IntVars::xmom], 0, 0, 1, 0); // scratch := (S_new - S_old)
296  scratch_xmom.mult(one_d / stage_dt); // scratch := (S_new - S_old) / stage_dt
297  MultiFab::Copy(S_rhs[IntVars::xmom], scratch_xmom, 0, 0, 1, 0); // slow_rhs := (S_new - S_old) / stage_dt
298 
299  MultiFab::Saxpy(scratch_ymom, -one_d, S_old[IntVars::ymom], 0, 0, 1, 0); // scratch := (S_new - S_old)
300  scratch_ymom.mult(one_d / stage_dt); // scratch := (S_new - S_old) / stage_dt
301  MultiFab::Copy(S_rhs[IntVars::ymom], scratch_ymom, 0, 0, 1, 0); // slow_rhs := (S_new - S_old) / stage_dt
302 #ifdef ERF_IMPLICIT_W
303  MultiFab::Saxpy(scratch_zmom, -one_d, S_old[IntVars::zmom], 0, 0, 1, 0); // scratch := (S_new - S_old)
304  scratch_zmom.mult(one_d / stage_dt); // scratch := (S_new - S_old) / stage_dt
305  MultiFab::Copy(S_rhs[IntVars::zmom], scratch_zmom, 0, 0, 1, 0); // slow_rhs := (S_new - S_old) / stage_dt
306 #endif
307  }
308  }
309 
310 #ifdef ERF_USE_EAMXX_SHOC
311  if (solverChoice.turbChoice[level].uses_eamxx_shoc() && eamxx_shoc_interface[level]) {
312  eamxx_shoc_interface[level]->add_fast_tend(S_rhs);
313  }
314 #endif
315  if (solverChoice.turbChoice[level].uses_native_shoc() && native_shoc_driver[level]) {
316  // Native SHOC applies its increment directly to state; ERF no
317  // longer adds a separate fast RHS contribution.
318  }
319 
320  // *****************************************************************************
321  // Update for moving terrain
322  // *****************************************************************************
323  if ( solverChoice.terrain_type == TerrainType::MovingFittedMesh )
324  {
325  MultiFab r_hse_new (base_state_new[level], make_alias, BaseState::r0_comp, 1);
326  MultiFab p_hse_new (base_state_new[level], make_alias, BaseState::p0_comp, 1);
327  MultiFab pi_hse_new (base_state_new[level], make_alias, BaseState::pi0_comp, 1);
328  MultiFab th_hse_new (base_state_new[level], make_alias, BaseState::th0_comp, 1);
329 
330  MultiFab* r0_new = &r_hse_new;
331  MultiFab* p0_new = &p_hse_new;
332  MultiFab* pi0_new = &pi_hse_new;
333  MultiFab* th0_new = &th_hse_new;
334 
335  // We define and evolve (rho theta)_0 in order to re-create p_0 in a way that is consistent
336  // with our update of (rho theta) but does NOT maintain dp_0 / dz = -rho_0 g. This is why
337  // we no longer discretize the vertical pressure gradient in perturbational form.
338  MultiFab rt0(p0->boxArray(),p0->DistributionMap(),1,1);
339  MultiFab rt0_new(p0->boxArray(),p0->DistributionMap(),1,1);
340  MultiFab r0_temp(p0->boxArray(),p0->DistributionMap(),1,1);
341 
342  // Remember this does NOT maintain dp_0 / dz = -rho_0 g, so we can no longer
343  // discretize the vertical pressure gradient in perturbational form.
344  AMREX_ALWAYS_ASSERT(solverChoice.advChoice.dycore_horiz_adv_type == AdvType::Centered_2nd);
345  AMREX_ALWAYS_ASSERT(solverChoice.advChoice.dycore_vert_adv_type == AdvType::Centered_2nd);
346 
347  double dt_base = new_stage_time - old_step_time;
348 
349  const Real l_rdOcp = solverChoice.rdOcp;
350 
351 #ifdef _OPENMP
352 #pragma omp parallel if (amrex::Gpu::notInLaunchRegion())
353 #endif
354  for ( MFIter mfi(*p0,TilingIfNotGPU()); mfi.isValid(); ++mfi)
355  {
356  const Array4<Real > rt0_arr = rt0.array(mfi);
357  const Array4<Real > rt0_tmp_arr = rt0_new.array(mfi);
358 
359  const Array4<Real const> r0_arr = r0->const_array(mfi);
360  const Array4<Real > r0_new_arr = r0_new->array(mfi);
361  const Array4<Real > r0_tmp_arr = r0_temp.array(mfi);
362 
363  const Array4<Real const> p0_arr = p0->const_array(mfi);
364  const Array4<Real > p0_new_arr = p0_new->array(mfi);
365  const Array4<Real > pi0_new_arr = pi0_new->array(mfi);
366  const Array4<Real > th0_new_arr = th0_new->array(mfi);
367 
368  const Array4<Real >& z_t_arr = z_t_rk[level]->array(mfi);
369 
370  const Array4<Real const>& dJ_old_arr = detJ_cc[level]->const_array(mfi);
371  const Array4<Real const>& dJ_new_arr = detJ_cc_new[level]->const_array(mfi);
372  const Array4<Real const>& dJ_src_arr = detJ_cc_src[level]->const_array(mfi);
373 
374  Box gbx = mfi.growntilebox({1,1,1});
375  amrex::ParallelFor(gbx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
376  {
377  rt0_arr(i,j,k) = getRhoThetagivenP(p0_arr(i,j,k));
378  rt0_tmp_arr(i,j,k) = getRhoThetagivenP(p0_new_arr(i,j,k));
379  r0_tmp_arr(i,j,k) = r0_new_arr(i,j,k);
380  });
381 
382  Box gbx2 = mfi.growntilebox({1,1,0});
383  amrex::ParallelFor(gbx2, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
384  {
385  Real zflux_r_lo = -z_t_arr(i,j,k ) * myhalf * (r0_tmp_arr(i,j,k) + r0_tmp_arr(i,j,k-1));
386  Real zflux_r_hi = -z_t_arr(i,j,k+1) * myhalf * (r0_tmp_arr(i,j,k) + r0_tmp_arr(i,j,k+1));
387 
388  Real zflux_rt_lo = zflux_r_lo * myhalf * (rt0_tmp_arr(i,j,k)/r0_tmp_arr(i,j,k) + rt0_tmp_arr(i,j,k-1)/r0_tmp_arr(i,j,k-1));
389  Real zflux_rt_hi = zflux_r_hi * myhalf * (rt0_tmp_arr(i,j,k)/r0_tmp_arr(i,j,k) + rt0_tmp_arr(i,j,k+1)/r0_tmp_arr(i,j,k+1));
390 
391  Real invdetJ = one / dJ_src_arr(i,j,k);
392 
393  Real src_r = - invdetJ * ( zflux_r_hi - zflux_r_lo ) * dxInv[2];
394  Real src_rt = - invdetJ * ( zflux_rt_hi - zflux_rt_lo ) * dxInv[2];
395 
396  Real dt_b = static_cast<Real>(dt_base);
397  Real rho0_new = dJ_old_arr(i,j,k) * r0_arr(i,j,k) + dt_b * dJ_src_arr(i,j,k) * src_r;
398  Real rt0_tmp_new = dJ_old_arr(i,j,k) * rt0_arr(i,j,k) + dt_b * dJ_src_arr(i,j,k) * src_rt;
399 
400  r0_new_arr(i,j,k) = rho0_new / dJ_new_arr(i,j,k);
401  rt0_tmp_new /= dJ_new_arr(i,j,k);
402 
403  p0_new_arr(i,j,k) = getPgivenRTh(rt0_tmp_new);
404  pi0_new_arr(i,j,k) = getExnergivenRTh(rt0_tmp_new, l_rdOcp);
405  th0_new_arr(i,j,k) = rt0_tmp_new / r0_new_arr(i,j,k);
406  });
407  } // MFIter
408  r0_new->FillBoundary(fine_geom.periodicity());
409  p0_new->FillBoundary(fine_geom.periodicity());
410  th0_new->FillBoundary(fine_geom.periodicity());
411  }
412 
413 #ifdef ERF_USE_NETCDF
414  // Populate RHS for relaxation zones if using real bcs
415  if (solverChoice.use_real_bcs && (level == 0)) {
416  const Real bdy_factor = solverChoice.bdy_nudge_factor;
417  if (real_width>0) {
418  //
419  // Note that old_stage_time is elapsed time, but (start_time+old_stage_time) is total time
420  // start_bdy_time and final_bdy_time are total time
421  //
422  double total_time = start_time + old_stage_time;
423  Real l_rdOcp = solverChoice.rdOcp;
424  Real l_c_p = solverChoice.c_p;
425  realbdy_compute_interior_ghost_rhs(total_time, slow_dt,
426  start_bdy_time, final_bdy_time, bdy_time_interval,
427  bdy_factor, real_width, fine_geom,
428  S_rhs, S_data,
429  bdy_data_xlo, bdy_data_xhi,
430  bdy_data_ylo, bdy_data_yhi,
431  m_r2d, l_c_p, l_rdOcp,
432  solverChoice.use_wrf_bdy_density,
433  solverChoice.bdy_rho_nudge_factor);
434  }
435  }
436 #endif
437  }; // end slow_rhs_fun_pre
void add_thin_body_sources(MultiFab &xmom_src, MultiFab &ymom_src, MultiFab &zmom_src, std::unique_ptr< iMultiFab > &xflux_imask_lev, std::unique_ptr< iMultiFab > &yflux_imask_lev, std::unique_ptr< iMultiFab > &zflux_imask_lev, std::unique_ptr< MultiFab > &thin_xforce_lev, std::unique_ptr< MultiFab > &thin_yforce_lev, std::unique_ptr< MultiFab > &thin_zforce_lev)
Definition: ERF_AddThinBodySources.cpp:27
AMREX_INLINE void AddCanopyBiophysicsHeatSources(amrex::MultiFab &cell_source, const amrex::MultiFab &S_data, const amrex::MultiFab &xvel, const amrex::MultiFab &yvel, const amrex::MultiFab &zvel, const amrex::MultiFab *frontal_area, const amrex::MultiFab &base_state, const SolverChoice &solver_choice)
Definition: ERF_CanopyBiophysics.H:28
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real getRhoThetagivenP(const amrex::Real p, const amrex::Real qv=amrex::Real(0))
Definition: ERF_EOS.H:172
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real getExnergivenRTh(const amrex::Real rhotheta, const amrex::Real rdOcp, const amrex::Real qv=amrex::Real(0))
Definition: ERF_EOS.H:156
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real getPgivenRTh(const amrex::Real rhotheta, const amrex::Real qv=amrex::Real(0))
Definition: ERF_EOS.H:81
@ Centered_2nd
amrex::GpuArray< Real, AMREX_SPACEDIM > dxInv
Definition: ERF_InitCustomPertVels_ParticleTests.H:17
AMREX_ALWAYS_ASSERT(bx.length()[2]==khi+1)
pp get("wavelength", wavelength)
void realbdy_compute_interior_ghost_rhs(const double &time, const double &delta_t_d, const double &start_bdy_time, const double &final_bdy_time, const double &bdy_time_interval, const Real &nudge_factor, int width, const Geometry &geom, Vector< MultiFab > &S_rhs, Vector< MultiFab > &S_cur_data, Vector< Vector< FArrayBox >> &bdy_data_xlo, Vector< Vector< FArrayBox >> &bdy_data_xhi, Vector< Vector< FArrayBox >> &bdy_data_ylo, Vector< Vector< FArrayBox >> &bdy_data_yhi, std::unique_ptr< ReadBndryPlanes > &m_r2d, const Real &c_p, const Real &rdOcp, const bool use_wrf_bdy_density, const Real &bdy_rho_nudge_factor)
Definition: ERF_InteriorGhostCells.cpp:161
void make_buoyancy(int lev, const Vector< MultiFab > &S_data, const MultiFab &S_prim, const MultiFab &qt, MultiFab &buoyancy, const Geometry geom, const SolverChoice &solverChoice, const MultiFab &base_state, const int n_qstate, const eb_ &ebfact, const int anelastic)
Definition: ERF_MakeBuoyancy.cpp:32
void make_gradp_pert(int level, const SolverChoice &solverChoice, const Geometry &geom, Vector< MultiFab > &S_data, const MultiFab &p0, const MultiFab &z_phys_nd, const MultiFab &z_phys_cc, Vector< std::unique_ptr< MultiFab >> &mapfac, const eb_ &ebfact, Vector< MultiFab > &gradp)
Definition: ERF_MakeGradP.cpp:28
void make_mom_sources(double time_d, double dt, const Vector< MultiFab > &S_data, const MultiFab *z_phys_nd, const MultiFab *z_phys_cc, Vector< Real > &stretched_dz_h, const MultiFab &xvel, const MultiFab &yvel, const MultiFab &wvel, MultiFab &xmom_src, MultiFab &ymom_src, MultiFab &zmom_src, const MultiFab &base_state, MultiFab *forest_drag, MultiFab *terrain_blank, MultiFab *terrain_blank_xface, MultiFab *terrain_blank_yface, MultiFab *terrain_blank_zface, MultiFab *cosPhi_mf, MultiFab *sinPhi_mf, const Geometry geom, const SolverChoice &solverChoice, Vector< std::unique_ptr< MultiFab >> &, const Real *dptr_u_geos, const Real *dptr_v_geos, const Real *dptr_wbar_sub, const Vector< Real * > d_rayleigh_ptrs_at_lev, const amrex::Real *d_sinesq_at_lev, const amrex::Real *d_sinesq_stag_at_lev, const Vector< Real * > d_sponge_ptrs_at_lev, const Vector< MultiFab > *forecast_state_at_lev, InputSoundingData &input_sounding_data, LargeScaleForcingData &lsf_data, std::unique_ptr< amrex::MultiFab > &lsf_tendencies, const eb_ &ebfact, bool is_slow_step)
Definition: ERF_MakeMomSources.cpp:38
void make_sources(int level, int, double dt, double time_d, const Vector< MultiFab > &S_data, const MultiFab &S_prim, MultiFab &source, const MultiFab &base_state, const MultiFab *z_phys_cc, const MultiFab &xvel, const MultiFab &yvel, const MultiFab &zvel, const MultiFab *qheating_rates, MultiFab *terrain_blank, const Geometry geom, const SolverChoice &solverChoice, Vector< std::unique_ptr< MultiFab >> &mapfac, const MultiFab *rhotheta_src, const MultiFab *rhoqt_src, const Real *dptr_wbar_sub, const Vector< Real * > d_rayleigh_ptrs_at_lev, const Real *d_sinesq_at_lev, TurbulentPerturbation &turbPert, const Table1D< Real > r_plane_avg, const Table1D< Real > t_plane_avg, bool is_slow_step)
Definition: ERF_MakeSources.cpp:36
ParallelFor(fab_box, [=] AMREX_GPU_DEVICE(int i, int j, int k) { qrcuten_arr(i, j, k)=Real(0);qscuten_arr(i, j, k)=Real(0);qicuten_arr(i, j, k)=Real(0);})
constexpr amrex::Real one
Definition: ERF_NumericalConstants.H:30
constexpr amrex::Real zero
Definition: ERF_NumericalConstants.H:29
constexpr amrex::Real myhalf
Definition: ERF_NumericalConstants.H:34
amrex::Real Real
Definition: ERF_ShocInterface.H:19
void erf_slow_rhs_pre(int level, int finest_level, int nrk, double dt, Vector< MultiFab > &S_rhs, Vector< MultiFab > &S_old, Vector< MultiFab > &S_data, const MultiFab &S_prim, const MultiFab &qt, MultiFab &avg_xmom, MultiFab &avg_ymom, MultiFab &avg_zmom, const MultiFab &xvel, const MultiFab &yvel, const MultiFab &zvel, std::unique_ptr< MultiFab > &z_t_mf, const MultiFab &cc_src, const MultiFab &xmom_src, const MultiFab &ymom_src, const MultiFab &zmom_src, const MultiFab &buoyancy, const MultiFab *zmom_crse_rhs, Vector< std::unique_ptr< MultiFab >> &Tau_lev, Vector< std::unique_ptr< MultiFab >> &Tau_corr_lev, Vector< Vector< std::unique_ptr< MultiFab >>> &Tau_EB, MultiFab *SmnSmn, MultiFab *eddyDiffs, MultiFab *Hfx1, MultiFab *Hfx2, MultiFab *Hfx3, MultiFab *Q1fx1, MultiFab *Q1fx2, MultiFab *Q1fx3, MultiFab *Q2fx3, MultiFab *Diss, MultiFab *Hfx3_EB, const Geometry geom, const SolverChoice &solverChoice, const amrex::Vector< std::unique_ptr< SurfaceLayer >> &SurfLayer, const Gpu::DeviceVector< BCRec > &domain_bcs_type_d, const Vector< BCRec > &domain_bcs_type_h, const MultiFab &z_phys_nd, const MultiFab &z_phys_cc, const MultiFab &ax, const MultiFab &ay, const MultiFab &az, const MultiFab &detJ, Gpu::DeviceVector< Real > &stretched_dz_d, Vector< MultiFab > &gradp, Vector< std::unique_ptr< MultiFab >> &mapfac, const eb_ &ebfact, ShocDriver *native_shoc_lev, YAFluxRegister *fr_as_crse, YAFluxRegister *fr_as_fine, const MultiFab *cloud_chamber_base_state, const erf_cloud_chamber::Config *cloud_chamber_config, CloudChamberBudget *cloud_budget)
Definition: ERF_SlowRhsPre.cpp:69
auto slow_rhs_fun_pre
Definition: ERF_TI_slow_rhs_pre.H:6
auto make_pi_stage
Definition: ERF_TI_utils.H:4
auto update_terrain_stage
Definition: ERF_TI_utils.H:125
void cons_to_prim(const MultiFab &cons_state, MultiFab &S_prim, int ng)
Definition: ERF_Utils.cpp:13
void make_qt(const MultiFab &cons_state, MultiFab &qt, int n_qstate_into_total)
Definition: ERF_Utils.cpp:108
@ num_comps
Definition: ERF_IndexDefines.H:81
@ pi0_comp
Definition: ERF_IndexDefines.H:78
@ p0_comp
Definition: ERF_IndexDefines.H:77
@ th0_comp
Definition: ERF_IndexDefines.H:79
@ r0_comp
Definition: ERF_IndexDefines.H:76
@ ymom
Definition: ERF_IndexDefines.H:234
@ cons
Definition: ERF_IndexDefines.H:232
@ zmom
Definition: ERF_IndexDefines.H:235
@ xmom
Definition: ERF_IndexDefines.H:233
@ qt
Definition: ERF_Kessler.H:30
real(c_double), parameter p0
Definition: ERF_module_model_constants.F90:40
real(kind=kind_phys), parameter, private r0
Definition: ERF_module_mp_wdm6.F90:75