ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_Utils.H
Go to the documentation of this file.
1 #ifndef ERF_UTILS_H_
2 #define ERF_UTILS_H_
3 
4 #include "AMReX.H"
5 #include "AMReX_MultiFab.H"
6 #include "AMReX_BCRec.H"
7 #include "ERF_Constants.H"
8 #include "ERF_DataStruct.H"
9 #include "ERF_IndexDefines.H"
10 #include "ERF_SurfaceLayer.H"
11 #include "ERF_FillPatcher.H"
12 #include "ERF_ReadBndryPlanes.H"
13 
14 /*
15  * Rebalance state to satisfy both EOS and HSE one column at a time
16  */
17 void
18 rebalance_columns(amrex::MultiFab& rho, const amrex::MultiFab& theta,
19  const amrex::MultiFab& qt, const amrex::MultiFab& qv,
20  const amrex::MultiFab* z_phys, const amrex::Geometry& geom,
21  bool use_existing_sfc_density);
22 
23 /*
24  * Create a new BoxArray in which every grid touches the bottom boundary
25  */
26 void ChopGrids2D (amrex::BoxArray& ba, const amrex::Box& domain, int target_size);
27 
28 /*
29  * Create a new BoxArray with exactly the number of boxes as MPI ranks.
30  * This is designed to work only on the base level which covers the entire domain.
31  */
32 amrex::BoxArray
33 ERFPostProcessBaseGrids (const amrex::Box& domain, bool decompose_in_z);
34 
35 void
36 cons_to_prim(const amrex::MultiFab& cons_state, amrex::MultiFab& S_prim, int ng);
37 
38 void
39 make_qt(const amrex::MultiFab& cons_state, amrex::MultiFab& qt, int n_qstate_into_total);
40 
41 /*
42  * Create the Jacobian for the metric transformation when use_terrain is true
43  */
44 void make_J (const amrex::Geometry& geom,
45  amrex::MultiFab& z_phys_nd,
46  amrex::MultiFab& detJ_cc);
47 
48 void make_areas (const amrex::Geometry& geom,
49  amrex::MultiFab& z_phys_nd,
50  amrex::MultiFab& ax,
51  amrex::MultiFab& ay,
52  amrex::MultiFab& az);
53 
54 /*
55  * Average z_phys_nd on nodes to cell centers
56  */
57 void make_zcc (const amrex::Geometry& geom,
58  amrex::MultiFab& z_phys_nd,
59  amrex::MultiFab& z_phys_cc);
60 
61 /*
62  * Convert momentum to velocity by dividing by density averaged onto faces
63  */
64 void MomentumToVelocity (amrex::MultiFab& xvel_out,
65  amrex::MultiFab& yvel_out,
66  amrex::MultiFab& zvel_out,
67  const amrex::MultiFab& cons_in,
68  const amrex::MultiFab& xmom_in,
69  const amrex::MultiFab& ymom_in,
70  const amrex::MultiFab& zmom_in,
71  const amrex::Box& domain,
72  const amrex::Vector<amrex::BCRec>& domain_bcs_type_h,
73  const amrex::MultiFab* c_vfrac = nullptr // optional
74  );
75 
76 /*
77  * Convert velocity to momentum by multiplying by density averaged onto faces
78  */
79 void VelocityToMomentum (const amrex::MultiFab& xvel_in,
80  const amrex::IntVect & xvel_ngrow,
81  const amrex::MultiFab& yvel_in,
82  const amrex::IntVect & yvel_ngrow,
83  const amrex::MultiFab& zvel_in,
84  const amrex::IntVect & zvel_ngrow,
85  const amrex::MultiFab& cons_in,
86  amrex::MultiFab& xmom_out,
87  amrex::MultiFab& ymom_out,
88  amrex::MultiFab& zmom_out,
89  const amrex::Box& domain,
90  const amrex::Vector<amrex::BCRec>& domain_bcs_type_h,
91  const amrex::MultiFab* c_vfrac = nullptr // optional
92  );
93 
94 /*
95  * Convert (den_div u) to (den_mlt u) by multiplying by (den_mlt/den_div)
96  */
97 void
98 ConvertForProjection (const amrex::MultiFab& den_div, const amrex::MultiFab& den_mlt,
99  amrex::MultiFab& xmom, amrex::MultiFab& ymom, amrex::MultiFab& zmom,
100  const amrex::Box& domain, const amrex::Vector<amrex::BCRec>& domain_bcs_type_h);
101 
102 /*
103  * Compute lateral boundary mass influx/outflux from base-state-density-weighted
104  * boundary normal velocities.
105  */
106 void
107 compute_influx_outflux_bdy (amrex::FArrayBox& bdy_data_xlo, amrex::FArrayBox& bdy_data_xhi,
108  amrex::FArrayBox& bdy_data_ylo, amrex::FArrayBox& bdy_data_yhi,
109  amrex::Array<amrex::MultiFab*, AMREX_SPACEDIM>& area_vec,
110  const amrex::Geometry& geom,
111  amrex::Real& influx, amrex::Real& outflux,
112  const int n);
113 
114 /*
115  * \brief Enforces solvability on lateral boundary data by scaling outflow to
116  * match with inflow.
117  */
118 void enforceInOutSolvability_bdy (const amrex::MultiFab& rho0,
119  amrex::FArrayBox& bdy_data_xlo,
120  amrex::FArrayBox& bdy_data_xhi,
121  amrex::FArrayBox& bdy_data_ylo,
122  amrex::FArrayBox& bdy_data_yhi,
123  amrex::Array<amrex::MultiFab*, AMREX_SPACEDIM>& area_vec,
124  const amrex::Geometry& geom,
125  const amrex::Vector<amrex::BCRec>& domain_bcs_type_h);
126 
127 /*
128  * \brief Enforces solvability by scaling outflow to match with inflow.
129  *
130  */
131 void enforceInOutSolvability (int lev,
132  amrex::Array<amrex::MultiFab*, AMREX_SPACEDIM>& vels_vec,
133  amrex::Array<amrex::MultiFab*, AMREX_SPACEDIM>& area_vec,
134  const amrex::Geometry& geom);
135 
136 /*
137  * Compute boxes for looping over interior/exterior ghost cells
138  * for use by fillpatch, erf_slow_rhs_pre, and erf_slow_rhs_post
139  */
140 void realbdy_interior_bxs_xy (const amrex::Box& bx,
141  const amrex::Box& domain,
142  const int& width,
143  amrex::Box& bx_xlo,
144  amrex::Box& bx_xhi,
145  amrex::Box& bx_ylo,
146  amrex::Box& bx_yhi,
147  const amrex::IntVect& ng_vect=amrex::IntVect(0,0,0),
148  const bool get_int_ng=false);
149 
150 /*
151  * Compute boxes for looping over set region cells
152  * for use by fillpatch, erf_slow_rhs_pre, and erf_slow_rhs_post
153  */
154 void realbdy_bc_bxs_xy (const amrex::Box& bx,
155  const amrex::Box& domain,
156  const int& set_width,
157  amrex::Box& bx_xlo,
158  amrex::Box& bx_xhi,
159  amrex::Box& bx_ylo,
160  amrex::Box& bx_yhi,
161  const amrex::IntVect& ng_vect=amrex::IntVect(0,0,0));
162 
163 /*
164  * Compute relaxation region RHS with wrfbdy
165  */
166 void realbdy_compute_interior_ghost_rhs (const amrex::Real& total_time,
167  const amrex::Real& delta_t,
168  const amrex::Real& start_bdy_time,
169  const amrex::Real& final_bdy_time,
170  const amrex::Real& bdy_time_interval,
171  const amrex::Real& nudge_factor,
172  int width,
173  const amrex::Geometry& geom,
174  amrex::Vector<amrex::MultiFab>& S_rhs,
175  amrex::Vector<amrex::MultiFab>& S_cur_data,
176  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xlo,
177  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xhi,
178  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_ylo,
179  amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_yhi,
180  std::unique_ptr<ReadBndryPlanes>& m_r2d,
181  const amrex::Real& c_p,
182  const amrex::Real& rdOcp);
183 
184 /*
185  * Compute relaxation region RHS at fine-crse interface
186  */
187 void
188 fine_compute_interior_ghost_rhs (const amrex::Real& time,
189  const amrex::Real& delta_t,
190  const int& width,
191  const int& set_width,
192  const amrex::Geometry& geom,
193  ERFFillPatcher* FPr_c,
194  ERFFillPatcher* FPr_u,
195  ERFFillPatcher* FPr_v,
196  ERFFillPatcher* FPr_w,
197  amrex::Vector<amrex::BCRec>& domain_bcs_type,
198  amrex::Vector<amrex::MultiFab>& S_rhs_f,
199  amrex::Vector<amrex::MultiFab>& S_data_f);
200 
201 /*
202  * Accumulate time averaged velocity fields
203  */
204 void
205 Time_Avg_Vel_atCC (const amrex::Real& dt,
206  amrex::Real& t_avg_cnt,
207  amrex::MultiFab* vel_t_avg,
208  amrex::MultiFab& xvel,
209  amrex::MultiFab& yvel,
210  amrex::MultiFab& zvel);
211 
212 /**
213  * Compute the nudging relaxation
214  *
215  * @param[in] delta_t time step
216  * @param[in] icomp component offset
217  * @param[in] num_var number of variables to loop
218  * @param[in] width width of wrf bdy file
219  * @param[in] dom_lo low bound of domain
220  * @param[in] dom_hi high bound of domain
221  * @param[in] F1 drift relaxation parameter
222  * @param[in] bx_xlo box for low x relaxation
223  * @param[in] bx_xhi box for high x relaxation
224  * @param[in] bx_ylo box for low y relaxation
225  * @param[in] bx_yhi box for high y relaxation
226  * @param[in] arr_xlo array for low x relaxation
227  * @param[in] arr_xhi array for high x relaxation
228  * @param[in] arr_ylo array for low y relaxation
229  * @param[in] arr_yhi array for high y relaxation
230  * @param[in] data_arr data array
231  * @param[out] rhs_arr RHS array
232  */
233 AMREX_GPU_HOST
234 AMREX_FORCE_INLINE
235 void
236 realbdy_compute_relaxation (const int& icomp,
237  const int& num_var,
238  const int& width,
239  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>& dx,
240  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>& ProbLo,
241  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM>& ProbHi,
242  const amrex::Real& F1,
243  const amrex::Box& bx_xlo,
244  const amrex::Box& bx_xhi,
245  const amrex::Box& bx_ylo,
246  const amrex::Box& bx_yhi,
247  const amrex::Array4<const amrex::Real>& arr_xlo,
248  const amrex::Array4<const amrex::Real>& arr_xhi,
249  const amrex::Array4<const amrex::Real>& arr_ylo,
250  const amrex::Array4<const amrex::Real>& arr_yhi,
251  const amrex::Array4<const amrex::Real>& data_arr,
252  const amrex::Array4<amrex::Real>& rhs_arr,
253  const amrex::Real& c_p,
254  const amrex::Real& rdOcp,
255  const int bdy_moist_nudge_type=0)
256 {
257  amrex::IntVect iv = bx_xlo.type();
258  amrex::Real ioff = (iv[0]==1) ? zero : myhalf;
259  amrex::Real joff = (iv[1]==1) ? zero : myhalf;
260 
261  // These are defined in ERF_Constants.H
262  amrex::Real l_rdOcp = rdOcp;
263  amrex::Real cond_fac = lcond / c_p; // condensation
264  amrex::Real sub_fac = lsub / c_p; // sublimation
265 
266  int nq = num_var;
267 
268  amrex::ParallelFor( bx_xlo, bx_xhi,
269  [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
270  {
271  // Corners with x boxes
272  amrex::Real x = ProbLo[0] + (i + ioff) * dx[0];
273  amrex::Real y = ProbLo[1] + (j + joff) * dx[1];
274  amrex::Real x_end = ProbLo[0] + width * dx[0];
275  amrex::Real y_end = ProbLo[1] + width * dx[1];
276  amrex::Real y_strt = ProbHi[1] - width * dx[1];
277  amrex::Real xi = (x_end - x) / (x_end - ProbLo[0]);
278  amrex::Real eta_lo = (y < y_end ) ? (y_end - y) / (y_end - ProbLo[1]) : zero;
279  amrex::Real eta_hi = (y > y_strt) ? (y - y_strt) / (ProbHi[1] - y_strt) : zero;
280  amrex::Real eta = std::max(eta_lo,eta_hi);
281  amrex::Real Factor = std::max(xi*xi,eta*eta) * F1;
282 
283  if (icomp == RhoQ1_comp)
284  {
285  amrex::Real exner_pi = getExnergivenRTh(data_arr(i,j,k,RhoTheta_comp),l_rdOcp,
286  data_arr(i,j,k,RhoQ1_comp)/data_arr(i,j,k,Rho_comp));
287 
288  // qv
289  amrex::Real delta = arr_xlo(i,j,k,0) - data_arr(i,j,k,icomp);
290  rhs_arr(i,j,k,icomp) += Factor*delta;
291 
292  if (bdy_moist_nudge_type==2) {
293  // qc and heat source term due to evaporation
294  delta = -data_arr(i,j,k,icomp+1); // This effectively nudges to 0
295  rhs_arr(i,j,k,icomp+1 ) += Factor*delta;
296  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * cond_fac / exner_pi;
297 
298  for (int n = 2; n < nq; n++) {
299  delta = -data_arr(i,j,k,icomp+n); // This effectively nudges to 0
300  rhs_arr(i,j,k,icomp+n) += Factor*delta;
301  if (nq > 3 && n == 2) { // nq > 3 guarantees that component 2 is ice
302  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * sub_fac / exner_pi;
303  }
304  } // n
305  } // moist_nudge_type
306  } else {
307  for (int n = 0; n < nq; n++) {
308  amrex::Real delta = arr_xlo(i,j,k,n) - data_arr(i,j,k,n+icomp);
309  rhs_arr(i,j,k,n+icomp) += Factor*delta;
310  }
311  }
312  },
313  [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
314  {
315  // Corners with x boxes
316  amrex::Real x = ProbLo[0] + (i + ioff) * dx[0];
317  amrex::Real y = ProbLo[1] + (j + joff) * dx[1];
318  amrex::Real x_strt = ProbHi[0] - width * dx[0];
319  amrex::Real y_strt = ProbHi[1] - width * dx[1];
320  amrex::Real y_end = ProbLo[1] + width * dx[1];
321  amrex::Real xi = (x - x_strt) / (ProbHi[0] - x_strt);
322  amrex::Real eta_lo = (y < y_end ) ? (y_end - y) / (y_end - ProbLo[1]) : zero;
323  amrex::Real eta_hi = (y > y_strt) ? (y - y_strt) / (ProbHi[1] - y_strt) : zero;
324  amrex::Real eta = std::max(eta_lo,eta_hi);
325  amrex::Real Factor = std::max(xi*xi,eta*eta) * F1;
326 
327  if (icomp == RhoQ1_comp)
328  {
329  amrex::Real exner_pi = getExnergivenRTh(data_arr(i,j,k,RhoTheta_comp),l_rdOcp,
330  data_arr(i,j,k,RhoQ1_comp)/data_arr(i,j,k,Rho_comp));
331 
332  // qv
333  amrex::Real delta = arr_xhi(i,j,k,0) - data_arr(i,j,k,icomp);
334  rhs_arr(i,j,k,icomp) += Factor*delta;
335 
336  if (bdy_moist_nudge_type==2) {
337  // qc and heat source term due to evaporation
338  delta = -data_arr(i,j,k,icomp+1); // This effectively nudges to 0
339  rhs_arr(i,j,k,icomp+1 ) += Factor*delta;
340  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * cond_fac / exner_pi;
341 
342  for (int n = 2; n < nq; n++) {
343  delta = -data_arr(i,j,k,icomp+n); // This effectively nudges to 0
344  rhs_arr(i,j,k,icomp+n) += Factor*delta;
345  if (nq > 3 && n == 2) { // nq > 3 guarantees that component 2 is ice
346  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * sub_fac / exner_pi;
347  }
348  } // n
349  } // moist_nudge_type
350  } else {
351  for (int n = 0; n < nq; n++) {
352  amrex::Real delta = arr_xhi(i,j,k,n) - data_arr(i,j,k,n+icomp);
353  rhs_arr(i,j,k,n+icomp) += Factor*delta;
354  }
355  }
356  });
357 
358  amrex::ParallelFor( bx_ylo, bx_yhi,
359  [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
360  {
361  // No corners for y boxes
362  amrex::Real y = ProbLo[1] + (j + joff) * dx[1];
363  amrex::Real y_end = ProbLo[1] + width * dx[1];
364  amrex::Real eta = (y_end - y) / (y_end - ProbLo[1]);
365  amrex::Real Factor = eta*eta * F1;
366 
367  if (icomp == RhoQ1_comp)
368  {
369  amrex::Real exner_pi = getExnergivenRTh(data_arr(i,j,k,RhoTheta_comp),l_rdOcp,
370  data_arr(i,j,k,RhoQ1_comp)/data_arr(i,j,k,Rho_comp));
371  // qv
372  amrex::Real delta = arr_ylo(i,j,k,0) - data_arr(i,j,k,icomp);
373  rhs_arr(i,j,k,icomp) += Factor*delta;
374 
375  if (bdy_moist_nudge_type==2) {
376  // qc and heat source term due to evaporation
377  delta = -data_arr(i,j,k,icomp+1); // This effectively nudges to 0
378  rhs_arr(i,j,k,icomp+1 ) += Factor*delta;
379  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * cond_fac / exner_pi;
380 
381  for (int n = 2; n < nq; n++) {
382  delta = -data_arr(i,j,k,icomp+n); // This effectively nudges to 0
383  rhs_arr(i,j,k,icomp+n) += Factor*delta;
384  if (nq > 3 && n == 2) { // nq > 3 guarantees that component 2 is ice
385  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * sub_fac / exner_pi;
386  }
387  } // n
388  } // moist_nudge_type
389  } else {
390  for (int n = 0; n < nq; n++) {
391  amrex::Real delta = arr_ylo(i,j,k,n) - data_arr(i,j,k,n+icomp);
392  rhs_arr(i,j,k,n+icomp) += Factor*delta;
393  }
394  }
395  },
396  [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
397  {
398  // No corners for y boxes
399  amrex::Real y = ProbLo[1] + (j + joff) * dx[1];
400  amrex::Real y_strt = ProbHi[1] - width * dx[1];
401  amrex::Real eta = (y - y_strt) / (ProbHi[1] - y_strt);
402  amrex::Real Factor = eta*eta * F1;
403 
404  if (icomp == RhoQ1_comp)
405  {
406  amrex::Real exner_pi = getExnergivenRTh(data_arr(i,j,k,RhoTheta_comp),l_rdOcp,
407  data_arr(i,j,k,RhoQ1_comp)/data_arr(i,j,k,Rho_comp));
408  // qv
409  amrex::Real delta = arr_yhi(i,j,k,0) - data_arr(i,j,k,icomp);
410  rhs_arr(i,j,k,icomp) += Factor*delta;
411 
412  if (bdy_moist_nudge_type==2) {
413  // qc and heat source term due to evaporation
414  delta = -data_arr(i,j,k,icomp+1); // This effectively nudges to 0
415  rhs_arr(i,j,k,icomp+1 ) += Factor*delta;
416  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * cond_fac / exner_pi;
417 
418  for (int n = 2; n < nq; n++) {
419  delta = -data_arr(i,j,k,icomp+n); // This effectively nudges to 0
420  rhs_arr(i,j,k,icomp+n) += Factor*delta;
421  if (nq > 3 && n == 2) { // nq > 3 guarantees that component 2 is ice
422  rhs_arr(i,j,k,RhoTheta_comp) += Factor * delta * sub_fac / exner_pi;
423  }
424  } // n
425  } // moist_nudge_type
426  } else {
427  for (int n = 0; n < nq; n++) {
428  amrex::Real delta = arr_yhi(i,j,k,n) - data_arr(i,j,k,n+icomp);
429  rhs_arr(i,j,k,n+icomp) += Factor*delta;
430  }
431  }
432  });
433 }
434 
435 /*
436  * Effectively a Multiply for a MultiFab and an iMultiFab mask
437  */
438 AMREX_GPU_HOST
439 AMREX_FORCE_INLINE
440 void
441 ApplyMask (amrex::MultiFab& dst,
442  const amrex::iMultiFab& imask,
443  const int nghost = 0)
444 {
445  for (amrex::MFIter mfi(dst,amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi)
446  {
447  const amrex::Box& bx = mfi.growntilebox(nghost);
448  if (bx.ok())
449  {
450  auto dstFab = dst.array(mfi);
451  const auto maskFab = imask.const_array(mfi);
452  ParallelFor(bx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
453  {
454  dstFab(i,j,k) *= maskFab(i,j,k);
455  });
456  }
457  }
458 }
459 
460 AMREX_GPU_HOST
461 AMREX_FORCE_INLINE
462 void
463 ApplyInvertedMask (amrex::MultiFab& dst,
464  const amrex::iMultiFab& imask,
465  const int nghost = 0)
466 {
467  for (amrex::MFIter mfi(dst,amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi)
468  {
469  const amrex::Box& bx = mfi.growntilebox(nghost);
470  if (bx.ok())
471  {
472  auto dstFab = dst.array(mfi);
473  const auto maskFab = imask.const_array(mfi);
474  ParallelFor(bx, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
475  {
476  dstFab(i,j,k) *= (1-maskFab(i,j,k));
477  });
478  }
479  }
480 }
481 
482 void
483 thinbody_wall_dist (std::unique_ptr<amrex::MultiFab>& wdist,
484  amrex::Vector<amrex::IntVect>& xfaces,
485  amrex::Vector<amrex::IntVect>& yfaces,
486  amrex::Vector<amrex::IntVect>& zfaces,
487  const amrex::Geometry& geomdata,
488  std::unique_ptr<amrex::MultiFab>& z_phys_cc);
489 
490 
491 void WeatherDataInterpolation(const amrex::Real time);
492 
493 AMREX_GPU_HOST_DEVICE
494 AMREX_FORCE_INLINE
495 amrex::Real
496 quad_interp_1d(amrex::Real z,
497  amrex::Real z0, amrex::Real p0,
498  amrex::Real z1, amrex::Real p1,
499  amrex::Real z2, amrex::Real p2)
500 {
501  return p0 * ( (z - z1) * (z - z2) ) / ( (z0 - z1) * (z0 - z2) )
502  + p1 * ( (z - z0) * (z - z2) ) / ( (z1 - z0) * (z1 - z2) )
503  + p2 * ( (z - z0) * (z - z1) ) / ( (z2 - z0) * (z2 - z1) );
504 }
505 
506 #endif
ERF_Constants.H
lsub
constexpr amrex::Real lsub
Definition: ERF_Constants.H:100
zero
constexpr amrex::Real zero
Definition: ERF_Constants.H:8
lcond
constexpr amrex::Real lcond
Definition: ERF_Constants.H:98
myhalf
constexpr amrex::Real myhalf
Definition: ERF_Constants.H:13
ERF_DataStruct.H
Coord::y
@ y
Coord::x
@ x
Coord::z
@ z
getExnergivenRTh
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
ERF_FillPatcher.H
ERF_IndexDefines.H
Rho_comp
#define Rho_comp
Definition: ERF_IndexDefines.H:36
RhoTheta_comp
#define RhoTheta_comp
Definition: ERF_IndexDefines.H:37
RhoQ1_comp
#define RhoQ1_comp
Definition: ERF_IndexDefines.H:42
z0
Real z0
Definition: ERF_InitCustomPertVels_ScalarAdvDiff.H:8
dx
const Real dx
Definition: ERF_InitCustomPert_ABL.H:23
rdOcp
const Real rdOcp
Definition: ERF_InitCustomPert_Bomex.H:16
rho
rho
Definition: ERF_InitCustomPert_Bubble.H:106
ParallelFor
ParallelFor(grown_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);})
ERF_ReadBndryPlanes.H
Real
amrex::Real Real
Definition: ERF_ShocInterface.H:19
ERF_SurfaceLayer.H
MomentumToVelocity
void MomentumToVelocity(amrex::MultiFab &xvel_out, amrex::MultiFab &yvel_out, amrex::MultiFab &zvel_out, const amrex::MultiFab &cons_in, const amrex::MultiFab &xmom_in, const amrex::MultiFab &ymom_in, const amrex::MultiFab &zmom_in, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h, const amrex::MultiFab *c_vfrac=nullptr)
compute_influx_outflux_bdy
void compute_influx_outflux_bdy(amrex::FArrayBox &bdy_data_xlo, amrex::FArrayBox &bdy_data_xhi, amrex::FArrayBox &bdy_data_ylo, amrex::FArrayBox &bdy_data_yhi, amrex::Array< amrex::MultiFab *, AMREX_SPACEDIM > &area_vec, const amrex::Geometry &geom, amrex::Real &influx, amrex::Real &outflux, const int n)
VelocityToMomentum
void VelocityToMomentum(const amrex::MultiFab &xvel_in, const amrex::IntVect &xvel_ngrow, const amrex::MultiFab &yvel_in, const amrex::IntVect &yvel_ngrow, const amrex::MultiFab &zvel_in, const amrex::IntVect &zvel_ngrow, const amrex::MultiFab &cons_in, amrex::MultiFab &xmom_out, amrex::MultiFab &ymom_out, amrex::MultiFab &zmom_out, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h, const amrex::MultiFab *c_vfrac=nullptr)
cons_to_prim
void cons_to_prim(const amrex::MultiFab &cons_state, amrex::MultiFab &S_prim, int ng)
WeatherDataInterpolation
void WeatherDataInterpolation(const amrex::Real time)
fine_compute_interior_ghost_rhs
void fine_compute_interior_ghost_rhs(const amrex::Real &time, const amrex::Real &delta_t, const int &width, const int &set_width, const amrex::Geometry &geom, ERFFillPatcher *FPr_c, ERFFillPatcher *FPr_u, ERFFillPatcher *FPr_v, ERFFillPatcher *FPr_w, amrex::Vector< amrex::BCRec > &domain_bcs_type, amrex::Vector< amrex::MultiFab > &S_rhs_f, amrex::Vector< amrex::MultiFab > &S_data_f)
realbdy_compute_interior_ghost_rhs
void realbdy_compute_interior_ghost_rhs(const amrex::Real &total_time, const amrex::Real &delta_t, const amrex::Real &start_bdy_time, const amrex::Real &final_bdy_time, const amrex::Real &bdy_time_interval, const amrex::Real &nudge_factor, int width, const amrex::Geometry &geom, amrex::Vector< amrex::MultiFab > &S_rhs, amrex::Vector< amrex::MultiFab > &S_cur_data, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_xlo, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_xhi, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_ylo, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_yhi, std::unique_ptr< ReadBndryPlanes > &m_r2d, const amrex::Real &c_p, const amrex::Real &rdOcp)
enforceInOutSolvability_bdy
void enforceInOutSolvability_bdy(const amrex::MultiFab &rho0, amrex::FArrayBox &bdy_data_xlo, amrex::FArrayBox &bdy_data_xhi, amrex::FArrayBox &bdy_data_ylo, amrex::FArrayBox &bdy_data_yhi, amrex::Array< amrex::MultiFab *, AMREX_SPACEDIM > &area_vec, const amrex::Geometry &geom, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h)
quad_interp_1d
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real quad_interp_1d(amrex::Real z, amrex::Real z0, amrex::Real p0, amrex::Real z1, amrex::Real p1, amrex::Real z2, amrex::Real p2)
Definition: ERF_Utils.H:496
thinbody_wall_dist
void thinbody_wall_dist(std::unique_ptr< amrex::MultiFab > &wdist, amrex::Vector< amrex::IntVect > &xfaces, amrex::Vector< amrex::IntVect > &yfaces, amrex::Vector< amrex::IntVect > &zfaces, const amrex::Geometry &geomdata, std::unique_ptr< amrex::MultiFab > &z_phys_cc)
ERFPostProcessBaseGrids
amrex::BoxArray ERFPostProcessBaseGrids(const amrex::Box &domain, bool decompose_in_z)
make_zcc
void make_zcc(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &z_phys_cc)
realbdy_bc_bxs_xy
void realbdy_bc_bxs_xy(const amrex::Box &bx, const amrex::Box &domain, const int &set_width, amrex::Box &bx_xlo, amrex::Box &bx_xhi, amrex::Box &bx_ylo, amrex::Box &bx_yhi, const amrex::IntVect &ng_vect=amrex::IntVect(0, 0, 0))
realbdy_interior_bxs_xy
void realbdy_interior_bxs_xy(const amrex::Box &bx, const amrex::Box &domain, const int &width, amrex::Box &bx_xlo, amrex::Box &bx_xhi, amrex::Box &bx_ylo, amrex::Box &bx_yhi, const amrex::IntVect &ng_vect=amrex::IntVect(0, 0, 0), const bool get_int_ng=false)
realbdy_compute_relaxation
AMREX_GPU_HOST AMREX_FORCE_INLINE void realbdy_compute_relaxation(const int &icomp, const int &num_var, const int &width, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dx, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &ProbLo, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &ProbHi, const amrex::Real &F1, const amrex::Box &bx_xlo, const amrex::Box &bx_xhi, const amrex::Box &bx_ylo, const amrex::Box &bx_yhi, const amrex::Array4< const amrex::Real > &arr_xlo, const amrex::Array4< const amrex::Real > &arr_xhi, const amrex::Array4< const amrex::Real > &arr_ylo, const amrex::Array4< const amrex::Real > &arr_yhi, const amrex::Array4< const amrex::Real > &data_arr, const amrex::Array4< amrex::Real > &rhs_arr, const amrex::Real &c_p, const amrex::Real &rdOcp, const int bdy_moist_nudge_type=0)
Definition: ERF_Utils.H:236
Time_Avg_Vel_atCC
void Time_Avg_Vel_atCC(const amrex::Real &dt, amrex::Real &t_avg_cnt, amrex::MultiFab *vel_t_avg, amrex::MultiFab &xvel, amrex::MultiFab &yvel, amrex::MultiFab &zvel)
ApplyInvertedMask
AMREX_GPU_HOST AMREX_FORCE_INLINE void ApplyInvertedMask(amrex::MultiFab &dst, const amrex::iMultiFab &imask, const int nghost=0)
Definition: ERF_Utils.H:463
make_J
void make_J(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &detJ_cc)
ApplyMask
AMREX_GPU_HOST AMREX_FORCE_INLINE void ApplyMask(amrex::MultiFab &dst, const amrex::iMultiFab &imask, const int nghost=0)
Definition: ERF_Utils.H:441
rebalance_columns
void rebalance_columns(amrex::MultiFab &rho, const amrex::MultiFab &theta, const amrex::MultiFab &qt, const amrex::MultiFab &qv, const amrex::MultiFab *z_phys, const amrex::Geometry &geom, bool use_existing_sfc_density)
ConvertForProjection
void ConvertForProjection(const amrex::MultiFab &den_div, const amrex::MultiFab &den_mlt, amrex::MultiFab &xmom, amrex::MultiFab &ymom, amrex::MultiFab &zmom, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h)
ChopGrids2D
void ChopGrids2D(amrex::BoxArray &ba, const amrex::Box &domain, int target_size)
enforceInOutSolvability
void enforceInOutSolvability(int lev, amrex::Array< amrex::MultiFab *, AMREX_SPACEDIM > &vels_vec, amrex::Array< amrex::MultiFab *, AMREX_SPACEDIM > &area_vec, const amrex::Geometry &geom)
make_areas
void make_areas(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &ax, amrex::MultiFab &ay, amrex::MultiFab &az)
make_qt
void make_qt(const amrex::MultiFab &cons_state, amrex::MultiFab &qt, int n_qstate_into_total)
ERFFillPatcher
Definition: ERF_FillPatcher.H:9
IntVars::ymom
@ ymom
Definition: ERF_IndexDefines.H:194
IntVars::zmom
@ zmom
Definition: ERF_IndexDefines.H:195
IntVars::xmom
@ xmom
Definition: ERF_IndexDefines.H:193
LsmVar_MM5::theta
@ theta
Definition: ERF_MM5.H:20
MicVar_Kess::qt
@ qt
Definition: ERF_Kessler.H:28
MicVar_Kess::qv
@ qv
Definition: ERF_Kessler.H:29
MicVar_Morr::ng
@ ng
Definition: ERF_Morrison.H:48
Vars::xvel
@ xvel
Definition: ERF_IndexDefines.H:175
Vars::zvel
@ zvel
Definition: ERF_IndexDefines.H:177
Vars::yvel
@ yvel
Definition: ERF_IndexDefines.H:176
module_model_constants::p0
real(c_double), parameter p0
Definition: ERF_module_model_constants.F90:40