ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF.H
Go to the documentation of this file.
1 #ifndef ERF_H_
2 #define ERF_H_
3 
4 #include <string>
5 #include <limits>
6 #include <memory>
7 
8 #ifdef _OPENMP
9 #include <omp.h>
10 #endif
11 
12 #include <AMReX_AmrCore.H>
13 #include <AMReX_BCRec.H>
14 #include <AMReX_Print.H>
15 
16 #include <AMReX_ParallelDescriptor.H>
17 #include <AMReX_ParmParse.H>
18 #include <AMReX_MultiFabUtil.H>
19 #include <AMReX_FillPatchUtil.H>
20 #include <AMReX_VisMF.H>
21 #include <AMReX_PhysBCFunct.H>
22 #include <AMReX_YAFluxRegister.H>
23 #include <AMReX_ErrorList.H>
24 #include <AMReX_EBFabFactory.H>
25 #include <AMReX_EBMultiFabUtil.H>
26 
27 #include <ERF_EB.H>
28 
29 #ifdef ERF_USE_FFT
30 #include <AMReX_FFT_Poisson.H>
31 #endif
32 
33 #ifdef AMREX_MEM_PROFILING
34 #include <AMReX_MemProfiler.H>
35 #endif
36 
37 #include "ERF_ProbCommon.H"
38 
39 #include <ERF_IndexDefines.H>
40 #include <ERF_DataStruct.H>
41 #include <ERF_TurbPertStruct.H>
42 #include <ERF_InputSoundingData.H>
43 #include <ERF_InputSpongeData.H>
44 #include <ERF_SurfaceLayer.H>
45 #include <ERF_Derive.H>
46 #include <ERF_ReadBndryPlanes.H>
47 #include <ERF_WriteBndryPlanes.H>
48 #include <ERF_MRI.H>
49 #include <ERF_PhysBCFunct.H>
50 #include <ERF_FillPatcher.H>
51 #include <ERF_SampleData.H>
52 #include <ERF_ForestDrag.H>
53 #include <ERF_Provenance.H>
54 
55 #ifdef ERF_USE_PARTICLES
56 #include "ERF_ParticleData.H"
57 #endif
58 
61 #include "ERF_LandSurface.H"
62 
63 #ifdef ERF_USE_WINDFARM
64 #include "ERF_WindFarm.H"
65 #endif
66 
67 #include <ERF_RadiationModels.H>
68 
69 #ifdef ERF_USE_EAMXX_SHOC
70 #include "ERF_ShocInterface.H"
71 #endif
72 
73 #include "ERF_ShocDriver.H"
74 
75 #ifdef ERF_USE_P3
76 #include "ERF_P3Interface.H"
77 #endif
78 
79 #include <iostream>
80 
81 #ifdef AMREX_LAZY
82 #include <AMReX_Lazy.H>
83 #endif
84 
85 #ifndef AMREX_USE_MPI
86 using amrex::MPI_COMM_WORLD;
87 using amrex::MPI_Comm;
88 #endif
89 
90 #ifdef ERF_USE_MULTIBLOCK
91 class MultiBlockContainer;
92 #endif
93 
94 /**
95  * Enum of possible interpolation types between coarse/fine
96 */
97 AMREX_ENUM(StateInterpType,
98  FullState, Perturbational
99 );
100 
101 /**
102  * Enum of possible plotfile types
103 */
104 AMREX_ENUM(PlotFileType,
105  None, Amrex, Netcdf
106 );
107 
108 /**
109  * Main class in ERF code, instantiated from main.cpp
110 */
111 
112 class ERF
113  : public amrex::AmrCore
114 {
115 public:
116 
117  ////////////////
118  // public member functions
119 
120  // constructor - reads in parameters from inputs file
121  // - sizes multilevel arrays and data structures
122  ERF ();
123  ~ERF () override;
124 
125  void ERF_shared ();
126 
127  // Declare a default move constructor so we ensure the destructor is
128  // not called when we return an object of this class by value
129  ERF (ERF&&) noexcept = delete;
130 
131  // Declare a default move assignment operator
132  ERF& operator=(ERF&& other) noexcept = delete;
133 
134  // Delete the copy constructor
135  ERF (const ERF& other) = delete;
136  //
137  // Delete the copy assignment operator
138  ERF& operator=(const ERF& other) = delete;
139 
140  // Advance solution to final time
141  void Evolve ();
142  double EvolveOneStep (double time, double dt_request);
143 
144  void ConfigureDriverAtmosToOceanCoupling (bool use_coupling_driver,
145  bool use_two_way_coupling,
146  bool use_state_contract);
147  void SetDriverAtmosToOceanStateContract (bool use_state_contract);
148  // Driver currently consumes BoxArray/DistributionMapping only. If we later
149  // need exact copy semantics, promote this to expose IndexType/ngrow
150  // explicitly or return the underlying MultiFab pointer/reference instead.
151  void GetOceanToAtmosSurfaceLayout (amrex::BoxArray& ba,
152  amrex::DistributionMapping& dm);
153 
154  void PackAtmosphericStates (amrex::Vector<amrex::MultiFab*>& states, double time);
155  void ApplyOceanSurfaceState (const amrex::Vector<amrex::MultiFab*>& state, double time);
156 
157  // Tag cells for refinement
158  void ErrorEst (int lev, amrex::TagBoxArray& tags, amrex::Real time, int ngrow) override;
159 
160  // Tag cells in a given box
161  void read_box_for_refinement (std::string& ref_prefix, int& lev_for_box, amrex::RealBox& real_box);
162  void update_box_for_refinement (std::string& ref_prefix, int& lev_for_box, amrex::RealBox& real_box, const double time);
163 
164  // Refinement criteria for hurricanes
165  void HurricaneTracker (int lev,
166  double time,
167  const amrex::MultiFab& cc_vel,
168  const amrex::Real velmag_threshold,
169  amrex::TagBoxArray* tags = nullptr);
170 
171  bool FindInitialEye (int lev,
172  const amrex::MultiFab& cc_vel,
173  const amrex::Real velmag_threshold,
174  amrex::Real& eye_x, amrex::Real& eye_y);
175 
176  // Hurricane track line output as VTKPolyline
177  amrex::Vector<std::array<amrex::Real, 2>> hurricane_track_xy;
178  amrex::Vector<std::array<amrex::Real, 2>> hurricane_eye_track_xy,
184  amrex::Vector<amrex::Vector<amrex::MultiFab>> forecast_state_1,
187  amrex::Vector<amrex::MultiFab> surface_state_1,
190 
191  amrex::Vector<amrex::Vector<amrex::MultiFab>> bckgnd_state;
192 
193  std::string MakeVTKFilename(int nstep);
194  std::string MakeVTKFilename_TrackerCircle(int nstep);
195  std::string MakeVTKFilename_EyeTracker_xy(int nstep);
196  std::string MakeFilename_EyeTracker_latlon(int nstep);
197  std::string MakeFilename_EyeTracker_maxvel(int nstep);
198  std::string MakeFilename_EyeTracker_minpressure(int nstep);
199  void WriteVTKPolyline(const std::string& filename,
200  amrex::Vector<std::array<amrex::Real, 2>>& points_xy);
201 
202  void WriteLinePlot(const std::string& filename,
203  amrex::Vector<std::array<amrex::Real, 2>>& points_xy);
204 
206 
208  const amrex::Geometry& geom,
209  const amrex::Vector<amrex::MultiFab>& S_data,
210  const amrex::Real& hurricane_eye_latitude,
211  const amrex::Real& hurricane_eye_longitude);
212 
214  const amrex::Geometry& geom,
215  const amrex::Vector<amrex::MultiFab>& S_data,
216  MoistureType moisture_type);
217 
218  void ComputeGlobalMinLocation(const SolverChoice& sc,
219  const amrex::Geometry& geom,
220  const amrex::Vector<amrex::MultiFab>& S_data,
221  amrex::Real* d_val_min_ptr,
222  int* d_i_min_ptr,
223  int* d_j_min_ptr,
224  amrex::Real& global_val_min,
225  int& global_i_min,
226  int& global_j_min);
227 
228  void HurricaneMaxVelTracker(const amrex::Geometry& geom,
229  const amrex::MultiFab& mf_cc_vel,
230  const double& time);
231 
232  void HurricaneMinPressureTracker(MoistureType moisture_type,
233  const amrex::Geometry& geom,
234  const amrex::MultiFab& mf_cons_var,
235  const double& time);
236 
237  void HurricaneTrackerCircle();
238 
240 
241  // Initialize multilevel data
242  void InitData ();
243 
244  // Initialize multilevel data before MultiBlock
245  void InitData_pre ();
246 
247  // Initialize multilevel data after MultiBlock
248  void InitData_post ();
249 
250  // Interpolate 2D arrays lat_m, lon_m, sinPhi, cosPhi
251  void Interp2DArrays (int lev, const amrex::BoxArray& my_ba2d, const amrex::DistributionMapping& my_dm);
252 
254 
255  // Compute the divergence -- whether EB, no-terrain, flat terrain or general terrain
256  void compute_divergence (int lev, amrex::MultiFab& rhs, amrex::Array<amrex::MultiFab const*,AMREX_SPACEDIM> rho0_u_const,
257  amrex::Geometry const& geom_at_lev);
258 
259  // Project the velocities to be divergence-free -- this is only relevant if anelastic == 1
260  void project_initial_velocity (int lev, double time, double dt);
261  void project_momenta (int lev, double l_time, double l_dt, amrex::Vector<amrex::MultiFab >& vars);
262 
263  // Project the velocities to be divergence-free with a thin body
264  void project_velocity_tb (int lev, double dt, amrex::Vector<amrex::MultiFab >& vars);
265 
266  // Calculate wall distance by solving a Poisson equation
267  void poisson_wall_dist (int lev);
268 
269  void make_subdomains(const amrex::BoxList& ba, amrex::Vector<amrex::BoxArray>& bins);
270 
271 #ifdef ERF_USE_FFT
272  void build_fft_solvers (int lev);
273  void solve_with_fft (int lev, int isub, const amrex::Box& subdomain,
274  amrex::MultiFab& rhs, amrex::MultiFab& p,
275  amrex::Array<amrex::MultiFab,AMREX_SPACEDIM>& fluxes);
276 #endif
277  void solve_with_gmres (int lev, const amrex::Box& subdomain,
278  amrex::MultiFab& rhs, amrex::MultiFab& p,
279  amrex::Array<amrex::MultiFab,AMREX_SPACEDIM>& fluxes,
280  amrex::MultiFab& ax_sub, amrex::MultiFab& ay_sub,
281  amrex::MultiFab& az_sub,
282  amrex::MultiFab& /*dJ_sub*/, amrex::MultiFab& znd_sub);
283 
284  void ImposeBCsOnPhi (int lev, amrex::MultiFab& phi, const amrex::Box& subdomain);
285 
286  // Init (NOT restart or regrid)
287  void init_only (int lev, double time);
288 
289  // Restart
290  void restart ();
291 
292  // Check for NaNs and other bad stuff
293  void check_state_for_nans(amrex::MultiFab const& S);
294  void check_vels_for_nans(amrex::MultiFab const& xvel, amrex::MultiFab const& yvel, amrex::MultiFab const& zvel);
295  void check_for_negative_theta (amrex::MultiFab& S);
296  void check_for_low_temp (amrex::MultiFab& S);
297 
299 
300  void check_mesh_type(int lev);
301 
302  // Is it time to write a plotfile or checkpoint?
303  // Note that "cur_time" is always in double precision
304  bool writeNow (double cur_time, const int nstep, const int plot_int,
305  const double plot_per, const double dt_0, double& last_file_time);
306 
307  // Called after every level 0 timestep
308  void post_timestep (int nstep, double time, double dt_lev);
309 
310  // Do whatever writing needs to be done at intermediate times
311  void WriteAtIntermediateTime(int nstep, double time);
312 
313  // Do whatever writing needs to be done after time-stepping is complete
314  void WriteAtFinalTime();
315 
316  // Diagnostics
317  void sum_integrated_quantities (double time);
318  void sum_derived_quantities (double time);
319  void sum_energy_quantities (double time);
320 
321  void write_1D_profiles (double time);
322  void write_1D_profiles_stag (double time);
323 
324  amrex::Real cloud_fraction (double time);
325 
326  // Fill the physical boundary conditions for cell-centered velocity (diagnostic only)
327  void FillBdyCCVels (amrex::MultiFab& mf_cc_vel, amrex::Geometry& lev_geom);
328 
329  void sample_points (int lev, double time, amrex::IntVect cell, amrex::MultiFab& mf);
330  void sample_lines (int lev, double time, amrex::IntVect cell, amrex::MultiFab& mf);
331 
332  void derive_diag_profiles (double time,
333  amrex::Gpu::HostVector<amrex::Real>& h_avg_u , amrex::Gpu::HostVector<amrex::Real>& h_avg_v,
334  amrex::Gpu::HostVector<amrex::Real>& h_avg_w , amrex::Gpu::HostVector<amrex::Real>& h_avg_rho,
335  amrex::Gpu::HostVector<amrex::Real>& h_avg_th , amrex::Gpu::HostVector<amrex::Real>& h_avg_ksgs,
336  amrex::Gpu::HostVector<amrex::Real>& h_avg_Kmv , amrex::Gpu::HostVector<amrex::Real>& h_avg_Khv,
337  amrex::Gpu::HostVector<amrex::Real>& h_avg_qv , amrex::Gpu::HostVector<amrex::Real>& h_avg_qc,
338  amrex::Gpu::HostVector<amrex::Real>& h_avg_qr ,
339  amrex::Gpu::HostVector<amrex::Real>& h_avg_wqv , amrex::Gpu::HostVector<amrex::Real>& h_avg_wqc,
340  amrex::Gpu::HostVector<amrex::Real>& h_avg_wqr ,
341  amrex::Gpu::HostVector<amrex::Real>& h_avg_qi , amrex::Gpu::HostVector<amrex::Real>& h_avg_qs,
342  amrex::Gpu::HostVector<amrex::Real>& h_avg_qg ,
343  amrex::Gpu::HostVector<amrex::Real>& h_avg_uu , amrex::Gpu::HostVector<amrex::Real>& h_avg_uv,
344  amrex::Gpu::HostVector<amrex::Real>& h_avg_uw,
345  amrex::Gpu::HostVector<amrex::Real>& h_avg_vv , amrex::Gpu::HostVector<amrex::Real>& h_avg_vw,
346  amrex::Gpu::HostVector<amrex::Real>& h_avg_ww,
347  amrex::Gpu::HostVector<amrex::Real>& h_avg_uth , amrex::Gpu::HostVector<amrex::Real>& h_avg_vth,
348  amrex::Gpu::HostVector<amrex::Real>& h_avg_wth, amrex::Gpu::HostVector<amrex::Real>& h_avg_thth,
349  amrex::Gpu::HostVector<amrex::Real>& h_avg_ku, amrex::Gpu::HostVector<amrex::Real>& h_avg_kv,
350  amrex::Gpu::HostVector<amrex::Real>& h_avg_kw,
351  amrex::Gpu::HostVector<amrex::Real>& h_avg_p,
352  amrex::Gpu::HostVector<amrex::Real>& h_avg_pu, amrex::Gpu::HostVector<amrex::Real>& h_avg_pv,
353  amrex::Gpu::HostVector<amrex::Real>& h_avg_pw, amrex::Gpu::HostVector<amrex::Real>& h_avg_wthv);
354  void derive_diag_profiles_stag (double time,
355  amrex::Gpu::HostVector<amrex::Real>& h_avg_u , amrex::Gpu::HostVector<amrex::Real>& h_avg_v,
356  amrex::Gpu::HostVector<amrex::Real>& h_avg_w , amrex::Gpu::HostVector<amrex::Real>& h_avg_rho,
357  amrex::Gpu::HostVector<amrex::Real>& h_avg_th , amrex::Gpu::HostVector<amrex::Real>& h_avg_ksgs,
358  amrex::Gpu::HostVector<amrex::Real>& h_avg_Kmv , amrex::Gpu::HostVector<amrex::Real>& h_avg_Khv,
359  amrex::Gpu::HostVector<amrex::Real>& h_avg_qv , amrex::Gpu::HostVector<amrex::Real>& h_avg_qc,
360  amrex::Gpu::HostVector<amrex::Real>& h_avg_qr ,
361  amrex::Gpu::HostVector<amrex::Real>& h_avg_wqv , amrex::Gpu::HostVector<amrex::Real>& h_avg_wqc,
362  amrex::Gpu::HostVector<amrex::Real>& h_avg_wqr ,
363  amrex::Gpu::HostVector<amrex::Real>& h_avg_qi , amrex::Gpu::HostVector<amrex::Real>& h_avg_qs,
364  amrex::Gpu::HostVector<amrex::Real>& h_avg_qg ,
365  amrex::Gpu::HostVector<amrex::Real>& h_avg_uu , amrex::Gpu::HostVector<amrex::Real>& h_avg_uv,
366  amrex::Gpu::HostVector<amrex::Real>& h_avg_uw,
367  amrex::Gpu::HostVector<amrex::Real>& h_avg_vv , amrex::Gpu::HostVector<amrex::Real>& h_avg_vw,
368  amrex::Gpu::HostVector<amrex::Real>& h_avg_ww,
369  amrex::Gpu::HostVector<amrex::Real>& h_avg_uth , amrex::Gpu::HostVector<amrex::Real>& h_avg_vth,
370  amrex::Gpu::HostVector<amrex::Real>& h_avg_wth, amrex::Gpu::HostVector<amrex::Real>& h_avg_thth,
371  amrex::Gpu::HostVector<amrex::Real>& h_avg_ku, amrex::Gpu::HostVector<amrex::Real>& h_avg_kv,
372  amrex::Gpu::HostVector<amrex::Real>& h_avg_kw,
373  amrex::Gpu::HostVector<amrex::Real>& h_avg_p,
374  amrex::Gpu::HostVector<amrex::Real>& h_avg_pu, amrex::Gpu::HostVector<amrex::Real>& h_avg_pv,
375  amrex::Gpu::HostVector<amrex::Real>& h_avg_pw, amrex::Gpu::HostVector<amrex::Real>& h_avg_wthv);
376 
377  void derive_stress_profiles (amrex::Gpu::HostVector<amrex::Real>& h_avg_tau11, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau12,
378  amrex::Gpu::HostVector<amrex::Real>& h_avg_tau13, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau22,
379  amrex::Gpu::HostVector<amrex::Real>& h_avg_tau23, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau33,
380  amrex::Gpu::HostVector<amrex::Real>& h_avg_hfx3, amrex::Gpu::HostVector<amrex::Real>& h_avg_q1fx3,
381  amrex::Gpu::HostVector<amrex::Real>& h_avg_q2fx3, amrex::Gpu::HostVector<amrex::Real>& h_avg_diss);
382  void derive_stress_profiles_stag (amrex::Gpu::HostVector<amrex::Real>& h_avg_tau11, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau12,
383  amrex::Gpu::HostVector<amrex::Real>& h_avg_tau13, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau22,
384  amrex::Gpu::HostVector<amrex::Real>& h_avg_tau23, amrex::Gpu::HostVector<amrex::Real>& h_avg_tau33,
385  amrex::Gpu::HostVector<amrex::Real>& h_avg_hfx3, amrex::Gpu::HostVector<amrex::Real>& h_avg_q1fx3,
386  amrex::Gpu::HostVector<amrex::Real>& h_avg_q2fx3, amrex::Gpu::HostVector<amrex::Real>& h_avg_diss);
387 
388  // Perform the volume-weighted sum
390  volWgtSumMF (int lev, const amrex::MultiFab& mf, int comp,
391  const amrex::MultiFab& dJ, const amrex::MultiFab& mfx, const amrex::MultiFab& mfy,
392  bool finemask, bool local = true);
393 
394  void
395  volWgtColumnSum (int lev, const amrex::MultiFab& mf, int comp,
396  amrex::MultiFab& mf_2d, const amrex::MultiFab& dJ);
397 
398  // Decide if it is time to take an action
399  static bool is_it_time_for_action (int nstep, double time, double dt,
400  int action_interval, double action_per);
401 
402  // Make a new level using provided BoxArray and DistributionMapping and
403  // fill with interpolated coarse level data.
404  // overrides the pure virtual function in AmrCore
405  void MakeNewLevelFromCoarse (int lev, amrex::Real time, const amrex::BoxArray& ba,
406  const amrex::DistributionMapping& dm) override;
407 
408  // Remake an existing level using provided BoxArray and DistributionMapping and
409  // fill with existing fine and coarse data.
410  // overrides the pure virtual function in AmrCore
411  void RemakeLevel (int lev, amrex::Real time, const amrex::BoxArray& ba,
412  const amrex::DistributionMapping& dm) override;
413 
414  // Delete level data
415  // overrides the pure virtual function in AmrCore
416  void ClearLevel (int lev) override;
417 
418  // Make a new level from scratch using provided BoxArray and DistributionMapping.
419  // Only used during initialization.
420  // overrides the pure virtual function in AmrCore
421  void MakeNewLevelFromScratch (int lev, amrex::Real time, const amrex::BoxArray& ba,
422  const amrex::DistributionMapping& dm) override;
423 
424  // compute dt from CFL considerations
425  double estTimeStep (int lev, long& dt_fast_ratio) const;
426 
427 #ifdef ERF_USE_WW3_COUPLING
428  //amrex::Print() << " About to call send_to_ww3 from ERF.H" << std::endl;
429  void send_to_ww3(int lev);
430  //amrex::Print() << " About to call read_waves from ERF.H" << std::endl;
431  void read_waves(int lev);
432  //void send_to_ww3 (int lev);
433  //void read_waves (int lev);
434 
435  //void send_to_ww3 (int lev);
436 #endif
437 
438  // Interface for advancing the data at one level by one "slow" timestep
439  void advance_dycore (int level,
440  amrex::Vector<amrex::MultiFab>& state_old,
441  amrex::Vector<amrex::MultiFab>& state_new,
442  amrex::MultiFab& xvel_old, amrex::MultiFab& yvel_old, amrex::MultiFab& zvel_old,
443  amrex::MultiFab& xvel_new, amrex::MultiFab& yvel_new, amrex::MultiFab& zvel_new,
444  amrex::MultiFab& source, amrex::MultiFab& xmom_src,
445  amrex::MultiFab& ymom_src, amrex::MultiFab& zmom_src,
446  amrex::MultiFab& buoyancy, amrex::Geometry fine_geom,
447  double dt, double time);
448 
449  void advance_microphysics (int lev,
450  amrex::MultiFab& cons_in,
451  const double& dt_advance,
452  const int& iteration,
453  const double& time);
454 
455  void advance_lsm (int lev,
456  amrex::MultiFab& cons_in,
457  amrex::MultiFab& xvel_in,
458  amrex::MultiFab& yvel_in,
459  const double& time,
460  const double& dt_advance);
461 
462  void advance_radiation (int lev,
463  amrex::MultiFab& cons_in,
464  const double& dt_advance);
465 
466 #ifdef ERF_USE_EAMXX_SHOC
467  void compute_shoc_tendencies (int lev,
468  amrex::MultiFab* cons,
469  amrex::MultiFab* xvel,
470  amrex::MultiFab* yvel,
471  amrex::MultiFab* zvel,
472  amrex::Real* w_subsid,
473  amrex::MultiFab* tau13,
474  amrex::MultiFab* tau23,
475  amrex::MultiFab* hfx3,
476  amrex::MultiFab* qfx3,
477  amrex::MultiFab* eddyDiffs,
478  amrex::MultiFab* z_phys_nd,
479  const double& dt_advance);
480 #endif
481 
482  void compute_native_shoc_tendencies (int lev,
483  amrex::MultiFab* cons,
484  amrex::MultiFab* xvel,
485  amrex::MultiFab* yvel,
486  amrex::MultiFab* zvel,
487  amrex::Real* w_subsid,
488  amrex::MultiFab* tau13,
489  amrex::MultiFab* tau23,
490  amrex::MultiFab* hfx3,
491  amrex::MultiFab* qfx3,
492  amrex::MultiFab* eddyDiffs,
493  amrex::MultiFab* z_phys_nd,
494  const double& dt_advance);
495 
496 #ifdef ERF_USE_P3
497  void compute_p3_tendencies (int lev,
498  amrex::MultiFab& cons_in,
499  amrex::MultiFab& source,
500  const double& dt_advance);
501 #endif
502 
503  void build_fine_mask (int lev, amrex::MultiFab& fine_mask);
504 
505  void MakeHorizontalAverages ();
506  void MakeDiagnosticAverage (amrex::Vector<amrex::Real>& h_havg, amrex::MultiFab& S, int n);
507  void derive_upwp (amrex::Vector<amrex::Real>& h_havg);
508 
509  // Write plotfile to disk
510  void Write3DPlotFile (int which, PlotFileType plotfile_type, amrex::Vector<std::string> plot_var_names);
511  void Write2DPlotFile (int which, PlotFileType plotfile_type, amrex::Vector<std::string> plot_var_names);
512 
513  // Write subvolume of data into a different "plotfile"
514  void WriteSubvolume (int isub, amrex::Vector<std::string> subvol_var_names);
515 
516  void WriteMultiLevelPlotfileWithTerrain (const std::string &plotfilename,
517  int nlevels,
518  const amrex::Vector<const amrex::MultiFab*> &mf,
519  const amrex::Vector<const amrex::MultiFab*> &mf_nd,
520  const amrex::Vector<std::string> &varnames,
521  const amrex::Vector<amrex::Geometry>& my_geom,
522  double time,
523  const amrex::Vector<int> &level_steps,
524  const amrex::Vector<amrex::IntVect>& my_ref_ratio,
525  const std::string &versionName = "HyperCLaw-V1.1",
526  const std::string &levelPrefix = "Level_",
527  const std::string &mfPrefix = "Cell",
528  const amrex::Vector<std::string>& extra_dirs = amrex::Vector<std::string>()) const;
529 
530 
531  void WriteGenericPlotfileHeaderWithTerrain (std::ostream &HeaderFile,
532  int nlevels,
533  const amrex::Vector<amrex::BoxArray> &bArray,
534  const amrex::Vector<std::string> &varnames,
535  const amrex::Vector<amrex::Geometry>& my_geom,
536  double time,
537  const amrex::Vector<int> &level_steps,
538  const amrex::Vector<amrex::IntVect>& my_ref_ratio,
539  const std::string &versionName,
540  const std::string &levelPrefix,
541  const std::string &mfPrefix) const;
542 
543  void erf_enforce_hse (int lev,
544  amrex::MultiFab& dens, amrex::MultiFab& pres, amrex::MultiFab& pi,
545  amrex::MultiFab& th, amrex::MultiFab& qv,
546  std::unique_ptr<amrex::MultiFab>& z_cc);
547 
548 #ifdef ERF_USE_NETCDF
549  //! Write a timestep to 1D vertical column output for coupling
550  void writeToNCColumnFile (int lev,
551  const std::string& colfile_name, amrex::Real xloc, amrex::Real yloc,
552  double time);
553 #endif //ERF_USE_NETCDF
554 
555  void init_from_input_sounding (int lev);
556 
557  void init_immersed_forcing (int lev);
558 
559  void input_sponge (int lev);
560 
561  void init_from_hse (int lev);
562 
563  void init_thin_body (int lev, const amrex::BoxArray& ba, const amrex::DistributionMapping& dm);
564 
565  void FillForecastStateMultiFabs(const int lev,
566  const std::string& filename,
567  const std::unique_ptr<amrex::MultiFab>& z_phys_nd,
568  amrex::Vector<amrex::Vector<amrex::MultiFab>>& forecast_state);
569 
570  void FillSurfaceStateMultiFabs(const int lev,
571  const std::string& filename,
572  amrex::Vector<amrex::MultiFab>& surface_state);
573 
574 
575  void WeatherDataInterpolation(const int nlevs,
576  const double time,
577  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& z_phys_nd,
578  bool regrid_forces_file_read);
579 
580  void SurfaceDataInterpolation(const int nlevs,
581  const double time,
582  amrex::Vector<std::unique_ptr<amrex::MultiFab>>& z_phys_nd,
583  bool regrid_forces_file_read);
584 
586  amrex::MultiFab& mf_cc_pert,
587  amrex::MultiFab& cons_pert,
588  amrex::MultiFab& xvel_pert,
589  amrex::MultiFab& yvel_pert,
590  amrex::MultiFab& zvel_pert);
591 
592  void create_random_perturbations (const int lev,
593  amrex::MultiFab& mf_cc_pert);
594 
595  void apply_gaussian_smoothing_to_perturbations (const int lev,
596  amrex::MultiFab& mf_cc_pert);
597 
599 
600  void PerformDataAssimilation(int da_iter);
601 
602  void init_custom (int lev);
603 
604 #ifdef ERF_USE_MULTIBLOCK
605  // constructor used when ERF is created by a multiblock driver
606  // calls AmrCore constructor -> AmrMesh constructor
607  ERF (const amrex::RealBox& rb, int max_level_in,
608  const amrex::Vector<int>& n_cell_in, int coord,
609  const amrex::Vector<amrex::IntVect>& ref_ratio,
610  const amrex::Array<int,AMREX_SPACEDIM>& is_per,
611  std::string prefix);
612 
613  // Advance a block specified number of time steps
614  void Evolve_MB (int MBstep, int max_block_step);
615 
616  // Set parmparse prefix for MultiBlock
617  void SetParmParsePrefix (std::string name) { pp_prefix = name; }
618 
619  // Set 'this' from multiblock container
620  void SetMultiBlockPointer (MultiBlockContainer *mbc) { m_mbc = mbc; }
621 
622  // Public data copy for MB
623  std::vector<amrex::Box> domain_p;
624 
625  // Primary solution data data containers
626  amrex::Vector<amrex::Vector<amrex::MultiFab> > vars_new;
627  amrex::Vector<amrex::Vector<amrex::MultiFab> > vars_old;
628 
629  // Pressure gradient at each level
630  amrex::Vector<amrex::Vector<amrex::MultiFab> > gradp;
631 
632  // Velocity time averaged field
633  amrex::Vector<std::unique_ptr<amrex::MultiFab>> vel_t_avg;
634  amrex::Vector<double> t_avg_cnt;
635 #endif
636 
637  std::string pp_prefix {"erf"};
638 
639  void fill_from_bndryregs (const amrex::Vector<amrex::MultiFab*>& mfs,
640  double time);
641 
642 #ifdef ERF_USE_NETCDF
643  // Version in which we always set the velocity values but don't set theta
644  void fill_from_realbdy (const amrex::Vector<amrex::MultiFab*>& mfs,
645  double time,
646  bool cons_only,
647  int icomp_cons,
648  int ncomp_cons,
649  amrex::IntVect ngvect_cons,
650  amrex::IntVect ngvect_vels);
651 #endif
652 
653 #ifdef ERF_USE_NETCDF
654  void init_from_wrfinput (int lev, amrex::MultiFab& mf_PSFC);
655  void init_from_metgrid (int lev);
656  void init_from_ncfile (int lev);
657 #endif // ERF_USE_NETCDF
658 
659 #ifdef ERF_USE_WINDFARM
660  void init_windfarm(int lev);
661  void advance_windfarm (const amrex::Geometry& a_geom,
662  const double& dt_advance,
663  amrex::MultiFab& cons_in,
664  amrex::MultiFab& U_old,
665  amrex::MultiFab& V_old,
666  amrex::MultiFab& W_old,
667  amrex::MultiFab& mf_vars_windfarm,
668  const amrex::MultiFab& mf_Nturb,
669  const amrex::MultiFab& mf_SMark,
670  const double& time);
671 #endif
672 
673  void MakeEBGeometry ();
674  void make_eb_box ();
676 
677  // more flexible version of AverageDown() that lets you average down across multiple levels
678  void AverageDownTo (int crse_lev,
679  int scomp,
680  int ncomp,
681  bool do_perturbational_and_momenta = true); // NOLINT
682 
683  /*! \brief Conservation-preserving fine→coarse average of moist state components
684  * (RhoTheta and the contiguous moist q range) using detJ/mfac weighting,
685  * reusing AverageDownTo with perturbational/momenta handling disabled. */
686  void AverageDownMoistStateTo (int crse_lev);
687 
688  // Write checkpoint file to disk
689  void WriteCheckpointFile () const;
690 
691  // Read checkpoint file from disk
692  void ReadCheckpointFile ();
693 
694  // Read only velocities from checkpoint file
695  void ReadVelsOnlyFromCheckpointFile (int lev_to_fill, std::string& chkfile);
696 
697  // read checkpoint file from disk -- called after instantiating m_SurfaceLayer
699 
700  void init_zphys (int lev, double elapsed_time);
701  void remake_zphys (int lev, std::unique_ptr<amrex::MultiFab>& temp_zphys_nd);
702  void update_terrain_arrays (int lev);
703 
707 
708 private:
709 
710  ///////////////////////////
711  // private member functions
712  ///////////////////////////
713 
714  // read in some parameters from inputs file
715  void ReadParameters ();
716  void ParameterSanityChecks ();
717 
718  // set covered coarse cells/faces to be the average of overlying fine cells/faces
719  void AverageDown ();
720 
721  void update_diffusive_arrays (int lev, const amrex::BoxArray& ba, const amrex::DistributionMapping& dm);
722 
723  void Construct_ERFFillPatchers (int lev);
724 
725  void Define_ERFFillPatchers (int lev);
726 
727  void init1DArrays ();
728 
729  void init_bcs ();
730  void init_phys_bcs (bool& rho_read, bool& read_prim_theta);
731 
732  void init_stuff (int lev, const amrex::BoxArray& ba, const amrex::DistributionMapping& dm,
733  amrex::Vector<amrex::MultiFab>& lev_new, amrex::Vector<amrex::MultiFab>& lev_old,
734  amrex::MultiFab& tmp_base_state,
735  std::unique_ptr<amrex::MultiFab>& tmp_zphys_nd);
736 
737  // Initialize the Turbulent perturbation
738  void turbPert_update (const int lev, const double dt);
739  void turbPert_amplitude (const int lev);
740  void turbPert_amplitude_w (const int lev);
741 
742  // Initialize the integrator object
743  void initialize_integrator (int lev, amrex::MultiFab& cons_mf, amrex::MultiFab& vel_mf);
744 
745  // Create the physbcs objects
746  void make_physbcs (int lev);
747 
748  // Initialize the microphysics object
749  void initializeMicrophysics (const int&);
750 
751 #ifdef ERF_USE_WINDFARM
752  // Initialize the windfarm object
753  void initializeWindFarm (const int&);
754 #endif
755 
756  // Compute a vector of new MultiFabs by copying from valid region and filling ghost cells
757  //
758  // NOTE: FillPatch takes in an empty MF, and returns cell-centered + velocities (not momenta)
759  //
760  // This one works only at level = 0 (base state does not change)
761  void FillPatchCrseLevel (int lev, double time,
762  const amrex::Vector<amrex::MultiFab*>& mfs_vel,
763  bool cons_only=false);
764 
765  // This one works only at level > 0 (base state does change)
766  void FillPatchFineLevel (int lev, double time,
767  const amrex::Vector<amrex::MultiFab*>& mfs_vel,
768  const amrex::Vector<amrex::MultiFab*>& mfs_mom,
769  const amrex::MultiFab& old_base_state,
770  const amrex::MultiFab& new_base_state,
771  bool fillset=true, bool cons_only=false);
772 
773  // Compute new multifabs by copying data from valid region and filling ghost cells.
774  // Unlike FillPatch, FillIntermediatePatch will use the supplied multifabs instead of fine level data.
775  // This is to support filling boundary cells at an intermediate time between old/new times
776  // on the fine level when valid data at a specific time is already available (such as
777  // at each RK stage when integrating between initial and final times at a given level).
778  //
779  // NOTE: FillIntermediatePatch takes in updated momenta, and returns both updated velocity and momenta
780  //
781  void FillIntermediatePatch (int lev, double time,
782  const amrex::Vector<amrex::MultiFab*>& mfs_vel,
783  const amrex::Vector<amrex::MultiFab*>& mfs_mom,
784  int ng_cons, int ng_vel, bool cons_only, int icomp_cons, int ncomp_cons);
785 
786  // Fill all multifabs (and all components) in a vector of multifabs corresponding to the
787  // grid variables defined in vars_old and vars_new just as FillCoarsePatch.
788  void FillCoarsePatch (int lev, double time);
789 
790  // advance a level by dt
791  // includes a recursive call for finer levels
792  void timeStep (int lev, double time, int iteration);
793 
794  // advance a single level for a single time step
795  void Advance (int lev, double time, double dt_lev, int iteration, int ncycle);
796 
797  //! Initialize HSE
798  void initHSE ();
799  void initHSE (int lev);
800 
801  //! Initialize Rayleigh damping profiles at a level
802  void initRayleigh_at_level (const int& lev);
803 
804  //! Initialize sponge profiles
805  void initSponge ();
806 
807  //! Set Rayleigh mean profiles from input sounding
808  void setRayleighRefFromSounding (bool restarting);
809 
810  //! Set sponge mean profiles from input sounding
811  void setSpongeRefFromSounding (bool restarting);
812 
813  // a wrapper for estTimeStep()
814  void ComputeDt (int step = -1, double cur_time_d = 0.0);
815 
816  // get plotfile name
817  [[nodiscard]] std::string PlotFileName (int lev) const;
818 
819  // set plotfile variable names
820  static amrex::Vector<std::string> PlotFileVarNames (amrex::Vector<std::string> plot_var_names);
821 
822  // set which variables and derived quantities go into plotfiles
823  void setPlotVariables (const std::string& pp_plot_var_names, amrex::Vector<std::string>& plot_var_names);
824  void setPlotVariables2D (const std::string& pp_plot_var_names, amrex::Vector<std::string>& plot_var_names);
825  // append variables to plot
826  void appendPlotVariables (const std::string& pp_plot_var_names, amrex::Vector<std::string>& plot_var_names);
827 
828  void setSubVolVariables (const std::string& pp_subvol_var_names, amrex::Vector<std::string>& subvol_var_names);
829 
830 #ifdef ERF_USE_NETCDF
831  //! Create 1D vertical column output for coupling
832  void createNCColumnFile (int lev,
833  const std::string& colfile_name, amrex::Real xloc, amrex::Real yloc);
834 
835  // Copy from the NC*fabs into the MultiFabs holding the boundary data
836  void init_from_wrfbdy (amrex::Vector<amrex::FArrayBox*> x_vel_lateral,
837  amrex::Vector<amrex::FArrayBox*> y_vel_lateral,
838  amrex::Vector<amrex::FArrayBox*> z_vel_lateral,
839  amrex::Vector<amrex::FArrayBox*> T_lateral);
840 
841  static double start_bdy_time;
842  static double final_bdy_time;
843 
844  static double start_low_time;
845  static double final_low_time;
846 
847  static double bdy_time_interval;
848  static double low_time_interval;
849 
850  // *** *** FArrayBox's for holding the SURFACE data
851  // amrex::IArrayBox NC_IVGTYP_fab; // Vegetation type (IVGTYP); Discrete numbers;
852  // amrex::FArrayBox NC_z0_fab; // Surface Roughness, z0 = z0 (IVGTYP)
853  // amrex::FArrayBox NC_PSFC_fab; // Surface pressure
854 
855  // TODO: Clarify the relation between SST and TSK
856  // amrex::FArrayBox NC_SST_fab; // Sea Surface Temperature; Defined even for land area
857  // amrex::FArrayBox NC_TSK_fab; // Surface Skin Temperature; Appears to be same as SST...
858 
859  // Vectors (over time) of Vector (over variables) of FArrayBoxs for holding the data read from the wrfbdy NetCDF file
860  amrex::Vector<amrex::Vector<amrex::FArrayBox>> bdy_data_xlo;
861  amrex::Vector<amrex::Vector<amrex::FArrayBox>> bdy_data_xhi;
862  amrex::Vector<amrex::Vector<amrex::FArrayBox>> bdy_data_ylo;
863  amrex::Vector<amrex::Vector<amrex::FArrayBox>> bdy_data_yhi;
864 
865  amrex::Vector<amrex::Vector<amrex::FArrayBox>> low_data_zlo;
866 
867  // Maximum value of terrain read in from wrfinput at level 0
868  amrex::Real z_top = zero;
869 
870 #endif // ERF_USE_NETCDF
871 
872  amrex::Vector<std::unique_ptr<amrex::MultiFab>> lat_m, lon_m; // Latitude and Longitude on the grid
873 
874  amrex::Vector<std::unique_ptr<amrex::MultiFab>> sinPhi_m, cosPhi_m; // Coriolis factors on the grid
875 
876  // Struct for working with the sounding data we take as an input
878 
879  // Struct for working with the sponge data we take as an input
881 
882  // Vector (6 planes) of DeviceVectors (ncell in plane) for Dirichlet BC data
883  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real>> xvel_bc_data;
884  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real>> yvel_bc_data;
885  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real>> zvel_bc_data;
886  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real>> th_bc_data;
887 
888  // Function to read and populate above vectors (if input file exists)
889  void init_Dirichlet_bc_data (const std::string input_file);
890 
891  // Read the file passed to amr.restart and use it as an initial condition for
892  // the current simulation. Supports a different number of components and
893  // ghost cells.
895 
896  // Initialize the new-time data at a level from the initial_data MultiFab
897  void InitializeLevelFromData (int lev, const amrex::MultiFab& initial_data);
898 
899  // utility to skip to next line in Header
900  static void GotoNextLine (std::istream& is);
901 
902  // Single level functions called by advance()
903  void post_update (amrex::MultiFab& state_mf, double time, const amrex::Geometry& geom);
904  void fill_rhs (amrex::MultiFab& rhs_mf, const amrex::MultiFab& state_mf, double time, const amrex::Geometry& geom);
905 
906  ////////////////
907  // private data members
908 
909  std::unique_ptr<ProblemBase> prob = nullptr; // problem-specific functions
910 
911  amrex::Vector<int> num_boxes_at_level; // how many boxes specified at each level by tagging criteria
912  amrex::Vector<int> num_files_at_level; // how many wrfinput files specified at each level
913  amrex::Vector<amrex::Vector<amrex::Box>> boxes_at_level; // the boxes specified at each level by tagging criteria
914 
915  amrex::Vector<int> istep; // which step?
916  amrex::Vector<int> nsubsteps; // how many substeps on each level?
917 
918  // keep track of old time, new time, and time step at each level
919  amrex::Vector<double> t_new;
920  amrex::Vector<double> t_old;
921  amrex::Vector<double> dt;
922  amrex::Vector<long> dt_mri_ratio;
923 
924 #ifndef ERF_USE_MULTIBLOCK
925  // Array of multifabs to store the solution at each level of refinement;
926  // after advancing a level we use "swap".
927  amrex::Vector<amrex::Vector<amrex::MultiFab> > vars_new;
928  amrex::Vector<amrex::Vector<amrex::MultiFab> > vars_old;
929 
930  // Pressure gradient at each level
931  amrex::Vector<amrex::Vector<amrex::MultiFab> > gradp;
932 
933  // Velocity time averaged field
934  amrex::Vector<std::unique_ptr<amrex::MultiFab>> vel_t_avg;
935  amrex::Vector<double> t_avg_cnt;
936 #endif
937  amrex::Vector<std::unique_ptr<MRISplitIntegrator<amrex::Vector<amrex::MultiFab> > > > mri_integrator_mem;
938 
939  // Used for anelastic or when we impose an initial projection
940  amrex::Vector<amrex::MultiFab> pp_inc;
941 
942  // Used only for fast substepping
943  amrex::Vector<amrex::MultiFab> lagged_delta_rt;
944  amrex::Vector<amrex::MultiFab> avg_xmom;
945  amrex::Vector<amrex::MultiFab> avg_ymom;
946  amrex::Vector<amrex::MultiFab> avg_zmom;
947 
948  // Vector over levels of routines to impose physical boundary conditions
949  amrex::Vector<std::unique_ptr<ERFPhysBCFunct_cons>> physbcs_cons;
950  amrex::Vector<std::unique_ptr<ERFPhysBCFunct_u>> physbcs_u;
951  amrex::Vector<std::unique_ptr<ERFPhysBCFunct_v>> physbcs_v;
952  amrex::Vector<std::unique_ptr<ERFPhysBCFunct_w>> physbcs_w;
953  amrex::Vector<std::unique_ptr<ERFPhysBCFunct_base>> physbcs_base;
954 
955  // Store primitive variable for MOST BC
956  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Theta_prim;
957  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Qv_prim;
958  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Qr_prim;
959 
960  // Scratch space for time integrator
961  amrex::Vector<amrex::MultiFab> rU_old;
962  amrex::Vector<amrex::MultiFab> rU_new;
963  amrex::Vector<amrex::MultiFab> rV_old;
964  amrex::Vector<amrex::MultiFab> rV_new;
965  amrex::Vector<amrex::MultiFab> rW_old;
966  amrex::Vector<amrex::MultiFab> rW_new;
967 
968  // amrex::Vector<amrex::MultiFab> xmom_crse_rhs;
969  // amrex::Vector<amrex::MultiFab> ymom_crse_rhs;
970  amrex::Vector<amrex::MultiFab> zmom_crse_rhs;
971 
972  std::unique_ptr<Microphysics> micro;
973  amrex::Vector<amrex::Vector<amrex::MultiFab*>> qmoist; // (lev,ncomp) rain_accum, snow_accum, graup_accum
974 
975  // Variables for wind farm parametrization models
976 
977 #ifdef ERF_USE_WINDFARM
978  std::unique_ptr<WindFarm> windfarm;
979  amrex::Vector<amrex::MultiFab> Nturb;
980  amrex::Vector<amrex::MultiFab> vars_windfarm; // Fitch: Vabs, Vabsdt, dudt, dvdt, dTKEdt
981  // EWP: dudt, dvdt, dTKEdt
982 
983  amrex::Vector<amrex::MultiFab> SMark; // A multifab that holds an integer corresponding
984  // to the number of the wind turbine to sample
985  // velocity upstream of the turbine
986 #endif
987 
989  amrex::Vector<std::string> lsm_data_name; // (ncomp) -- e.g., sw_dn_dir/dif_vis/nir, lw_dn, cos_zen
990  amrex::Vector<amrex::Vector<amrex::MultiFab*>> lsm_data; // (lev,ncomp) Data for LSM coupling
991  amrex::Vector<std::string> lsm_flux_name; // (ncomp) -- e.g., hfx3, q1fx3, tau13, tau23
992  amrex::Vector<amrex::Vector<amrex::MultiFab*>> lsm_flux; // (lev,ncomp) Fluxes from LSM needed by DYCORE
993 
994  amrex::Vector<std::unique_ptr<IRadiation>> rad; // Radiation model at each level
995  amrex::Vector<std::unique_ptr<amrex::MultiFab>> qheating_rates; // radiation heating rate source terms (SW, LW) (DYCORE)
996  amrex::Vector<std::unique_ptr<amrex::MultiFab>> rad_fluxes; // radiation fluxes (SW up/dn, LW up/dn) (OUTPUT ONLY)
997 #ifdef ERF_USE_EAMXX_SHOC
998  amrex::Vector<std::unique_ptr<SHOCInterface>> eamxx_shoc_interface; // EAMxx SHOC model at each level
999 #endif
1000 
1001  amrex::Vector<std::unique_ptr<ShocDriver>> native_shoc_driver; // Native SHOC model at each level
1002 
1003 #ifdef ERF_USE_P3
1004  amrex::Vector<std::unique_ptr<P3Interface>> p3_interface; // P3 model at each level
1005 #endif
1006 
1007  bool plot_rad = false;
1009 
1010  // Fillpatcher classes for coarse-fine boundaries
1011  int cf_width{0};
1013  amrex::Vector<ERFFillPatcher> FPr_c;
1014  amrex::Vector<ERFFillPatcher> FPr_u;
1015  amrex::Vector<ERFFillPatcher> FPr_v;
1016  amrex::Vector<ERFFillPatcher> FPr_w;
1017 
1018  // Diffusive stresses and Smag
1019  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> Tau;
1020  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> Tau_corr; // correction to tau for implicit
1021  amrex::Vector<std::unique_ptr<amrex::MultiFab>> eddyDiffs_lev;
1022  amrex::Vector<std::unique_ptr<amrex::MultiFab>> SmnSmn_lev;
1023 
1024  // Sea Surface Temps, Skin Temperature and Land Masks (lev, ntimes)
1025  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> sst_lev;
1026  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> tsk_lev;
1027  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::iMultiFab>>> lmask_lev;
1028 
1029  // Land grid types and urban fraction
1030  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::iMultiFab>>> land_type_lev;
1031  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::iMultiFab>>> soil_type_lev;
1032  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> urb_frac_lev;
1033 
1034  // Other SFS terms
1035  amrex::Vector<std::unique_ptr<amrex::MultiFab>> SFS_hfx1_lev, SFS_hfx2_lev, SFS_hfx3_lev;
1036  amrex::Vector<std::unique_ptr<amrex::MultiFab>> SFS_diss_lev;
1037  amrex::Vector<std::unique_ptr<amrex::MultiFab>> SFS_q1fx1_lev, SFS_q1fx2_lev, SFS_q1fx3_lev;
1038  amrex::Vector<std::unique_ptr<amrex::MultiFab>> SFS_q2fx3_lev;
1039 
1040  // Diffusive fluxes for EB
1041  // Tau_EB[level][component][grid_type]
1042  amrex::Vector<amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>>> Tau_EB;
1043  amrex::Vector<std::unique_ptr<amrex::MultiFab>> hfx3_EB;
1044  amrex::Vector<std::unique_ptr<amrex::MultiFab>> qfx3_EB;
1045 
1046  // Grid stretching
1047  amrex::Vector<amrex::Vector<amrex::Real>> zlevels_stag; // nominal height levels
1048 
1049  // Data structures for terrain-fitted coordinates
1050  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_phys_nd;
1051  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_phys_cc;
1052 
1053  amrex::Vector<std::unique_ptr<amrex::MultiFab>> detJ_cc;
1054  amrex::Vector<std::unique_ptr<amrex::MultiFab>> ax;
1055  amrex::Vector<std::unique_ptr<amrex::MultiFab>> ay;
1056  amrex::Vector<std::unique_ptr<amrex::MultiFab>> az;
1057 
1058  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_phys_nd_src;
1059  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_phys_cc_src;
1060  amrex::Vector<std::unique_ptr<amrex::MultiFab>> detJ_cc_src;
1061  amrex::Vector<std::unique_ptr<amrex::MultiFab>> ax_src;
1062  amrex::Vector<std::unique_ptr<amrex::MultiFab>> ay_src;
1063  amrex::Vector<std::unique_ptr<amrex::MultiFab>> az_src;
1064 
1065  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_phys_nd_new;
1066  amrex::Vector<std::unique_ptr<amrex::MultiFab>> detJ_cc_new;
1067 
1068  amrex::Vector<std::unique_ptr<amrex::MultiFab>> z_t_rk;
1069 
1070  // Data structures for terrain immersed forcing
1071  amrex::Vector<std::unique_ptr<amrex::MultiFab>> terrain_blanking;
1072 
1073  // Wall distance function
1074  amrex::Vector<std::unique_ptr<amrex::MultiFab>> walldist;
1075 
1076  // Map scale factors -- vector across levels of vector across types
1077  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::MultiFab>>> mapfac;
1078 
1079  // Mask to be used to identify cells covered by finer level (if level > 0)
1080  amrex::Vector<std::unique_ptr<amrex::MultiFab>> fine_mask;
1081 
1082  amrex::Vector<amrex::Vector<amrex::Real>> stretched_dz_h;
1083  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real>> stretched_dz_d;
1084 
1085  amrex::Vector<amrex::MultiFab> base_state;
1086  amrex::Vector<amrex::MultiFab> base_state_new;
1087 
1088  // Wave coupling data
1089  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Hwave;
1090  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Lwave;
1091  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Hwave_onegrid;
1092  amrex::Vector<std::unique_ptr<amrex::MultiFab>> Lwave_onegrid;
1093  bool finished_wave = false;
1094 
1095  // array of flux registers
1096  amrex::Vector<amrex::YAFluxRegister*> advflux_reg;
1097 
1098  // A BCRec is essentially a 2*DIM integer array storing the boundary
1099  // condition type at each lo/hi walls in each direction. We have one BCRec
1100  // for each component of the cell-centered variables and each velocity component.
1101  amrex::Vector <amrex::BCRec> domain_bcs_type;
1102  amrex::Gpu::DeviceVector<amrex::BCRec> domain_bcs_type_d;
1103 
1104  // We store these so that we can print them out in the job_info file
1105  amrex::Array<std::string,2*AMREX_SPACEDIM> domain_bc_type;
1106 
1107  // These hold the Dirichlet values at walls which need them ...
1108  amrex::Array<amrex::Array<amrex::Real, AMREX_SPACEDIM*2>, AMREX_SPACEDIM+NBCVAR_max> m_bc_extdir_vals;
1109 
1110  // These hold the Neumann values at walls which need them ...
1111  amrex::Array<amrex::Array<amrex::Real, AMREX_SPACEDIM*2>, AMREX_SPACEDIM+NBCVAR_max> m_bc_neumann_vals;
1112 
1113  // Per-face flag: non-reflecting inflow (extrapolate RhoTheta instead of prescribing)
1114  amrex::Array<bool, AMREX_SPACEDIM*2> m_bc_nonreflecting = {{false}};
1115 
1116  // These are the "physical" boundary condition types (e.g. "inflow")
1117  amrex::GpuArray<ERF_BC, AMREX_SPACEDIM*2> phys_bc_type;
1118 
1119  // These are the masks for the thin immersed body
1120  amrex::Vector<std::unique_ptr<amrex::iMultiFab>> xflux_imask;
1121  amrex::Vector<std::unique_ptr<amrex::iMultiFab>> yflux_imask;
1122  amrex::Vector<std::unique_ptr<amrex::iMultiFab>> zflux_imask;
1123  //amrex::Vector<std::unique_ptr<amrex::iMultiFab>> overset_imask;
1124 
1125  // These are the body forces that result from the thin immersed body
1126  amrex::Vector<std::unique_ptr<amrex::MultiFab>> thin_xforce;
1127  amrex::Vector<std::unique_ptr<amrex::MultiFab>> thin_yforce;
1128  amrex::Vector<std::unique_ptr<amrex::MultiFab>> thin_zforce;
1129 
1135 
1140  static double last_check_file_time;
1141 
1142  amrex::Vector<int> last_subvol_step;
1143  amrex::Vector<double> last_subvol_time;
1144 
1146 
1147  // Output control
1148  const int datwidth = 14;
1149  const int datprecision = 6;
1150  const int timeprecision = 13; // e.g., 1-yr LES: 31,536,000 s with dt ~ amrex::Real(0.01) ==> min prec = 10
1151  // epoch time for 2030-01-01 00:00:00 is 1,893,456,000 s with dt ~ amrex::Real(0.01) ==> min prec = 12
1152 
1153  ////////////////
1154  // runtime parameters
1155 
1156  // Maximum number of steps
1157  int max_step = -1;
1158 
1159  // Start and stop times
1160  static double start_time;
1161  static double stop_time;
1162 
1163  bool use_datetime = false;
1164  const std::string datetime_format = "%Y-%m-%d %H:%M:%S"; // ISO 8601 standard
1165 
1166  // if >= 0 we restart from a checkpoint
1167  std::string restart_chkfile = "";
1168 
1169  // Host-only metadata shared by all native output paths. It never enters
1170  // device code or changes the model state read from a checkpoint.
1172 
1173  // Time step controls
1178  static double dt_max_initial;
1179  static double dt_max;
1180 
1181  // Fixed dt at each level (only used if positive)
1182  amrex::Vector<double> fixed_dt;
1183  amrex::Vector<double> fixed_fast_dt;
1185 
1186  // how often each level regrids the higher levels of refinement
1187  // (after a level advances that many time steps)
1188  int regrid_int = -1;
1189 
1190  // Should we regrid level 0 after restart if Ngrids < Nprocs or
1191  // max_grid_size has changed?
1193 
1194  // plotfile prefix and frequency
1195  std::string plot3d_file_1 {"plt_1_"};
1196  std::string plot3d_file_2 {"plt_2_"};
1197  std::string plot2d_file_1 {"plt2d_1_"};
1198  std::string plot2d_file_2 {"plt2d_2_"};
1199  std::string subvol_file {"subvol"};
1201  int m_plot3d_int_1 = -1;
1202  int m_plot3d_int_2 = -1;
1203  int m_plot2d_int_1 = -1;
1204  int m_plot2d_int_2 = -1;
1205 
1206  amrex::Vector<int> m_subvol_int;
1207  amrex::Vector<double> m_subvol_per;
1208 
1209  double m_plot3d_per_1 = -1.0;
1210  double m_plot3d_per_2 = -1.0;
1211  double m_plot2d_per_1 = -1.0;
1212  double m_plot2d_per_2 = -1.0;
1213  bool m_plot_face_vels = false;
1214 
1215  bool plot_lsm = false;
1216 
1217  // other sampling output control
1218  int profile_int = -1;
1219  bool destag_profiles = true;
1220 
1221  // Checkpoint type, prefix and frequency
1222  std::string check_file {"chk"};
1223  int m_check_int = -1;
1224  double m_check_per = -1.0;
1225 
1226  amrex::Vector<std::string> subvol3d_var_names;
1227 
1228  amrex::Vector<std::string> plot3d_var_names_1;
1229  amrex::Vector<std::string> plot3d_var_names_2;
1230  amrex::Vector<std::string> plot2d_var_names_1;
1231  amrex::Vector<std::string> plot2d_var_names_2;
1232  const amrex::Vector<std::string> cons_names {"density", "rhotheta", "rhoKE", "rhoadv_0",
1233  "rhoQ1", "rhoQ2", "rhoQ3",
1234  "rhoQ4", "rhoQ5", "rhoQ6",
1235  "rhoQ7", "rhoQ8", "rhoQ9",
1236  "rhoQ10", "rhoQ11"};
1237 
1238  // **************************************************************************************
1239  // NOTE: The order of variable names here **MUST MATCH THE ORDER** in IO/ERF_Plotfile.cpp
1240  // **************************************************************************************
1241  const amrex::Vector<std::string> derived_names {"temp", "theta", "KE", "scalar", "soundspeed",
1242  "reflectivity","max_reflectivity", "precipitable", "mucape",
1243  "vorticity_x","vorticity_y","vorticity_z",
1244  "helicity", "local_helicity",
1245  "magvel", "divU",
1246  "pres_hse", "dens_hse", "theta_hse", "qv_hse",
1247  "pressure", "pert_pres", "pert_dens",
1248  "buoyancy", "eq_pot_temp", "VPD",
1249 #ifdef ERF_USE_WINDFARM
1250  "num_turb", "SMark0", "SMark1",
1251 #endif
1252  "dpdx", "dpdy", "dpdz", "pres_hse_x", "pres_hse_y",
1253  "z_phys", "detJ", "mapfac", "lat_m", "lon_m",
1254  // Time averaged velocity
1255  "u_t_avg", "v_t_avg", "w_t_avg", "umag_t_avg",
1256  // eddy viscosity
1257  "nut",
1258  // eddy diffusivity of momentum
1259  "Kmv","Kmh",
1260  // eddy diffusivity of heat
1261  "Khv","Khh",
1262  // turbulence lengthscale
1263  "Lturb",
1264  // native SHOC diagnostics
1265  "pblh", "shoc_cldfrac", "shoc_ql", "shoc_ql2", "shoc_cond",
1266  "wqls_sec", "wthv_sec", "w_sec",
1267  "thl_sec", "qw_sec", "qwthl_sec", "wthl_sec", "wqw_sec", "w3",
1268  "brunt", "isotropy", "shear_prod", "buoy_prod", "diss_tke",
1269  // wall distance
1270  "walldist",
1271  // dissipation
1272  "diss",
1273  // moisture vars
1274  "moist_density", "qv", "qc", "qi", "qrain", "qsnow", "qgraup",
1275  "qt", "qn", "qp", "qsat", "nc", "ni", "nr", "ns", "ng",
1276  "rain_accum", "snow_accum", "graup_accum",
1277  "rel_humidity", "condensation_rate",
1278  // Terrain IB mask
1279  "terrain_IB_mask",
1280  // EB variables
1281  "volfrac"
1282 #ifdef ERF_COMPUTE_ERROR
1283  // error vars
1284  ,"xvel_err", "yvel_err", "zvel_err", "pp_err"
1285 #endif
1286  ,"qsrc_sw", "qsrc_lw"
1287  };
1288 
1289  // **************************************************************************************
1290  // NOTE: The order of variable names here **MUST MATCH THE ORDER** in IO/ERF_WriteSubVolume.cpp
1291  // **************************************************************************************
1292  const amrex::Vector<std::string> derived_subvol_names {"soundspeed", "temp", "theta", "KE", "scalar"};
1293 
1294  // algorithm choices
1296 
1297  // Turbulent perturbation structure
1299 
1300 #ifdef ERF_USE_PARTICLES
1301  // Particle container with all particle species
1302  ParticleData particleData;
1303 
1304  // variables and functions for tracers particles
1305  bool m_use_tracer_particles; /*!< tracer particles that advect with flow with optional sedimentation */
1306 
1307  /*! Read tracer and hydro particles parameters */
1308  void readTracersParams();
1309 
1310  /*! Initialize tracer and hydro particles */
1311  void initializeTracers ( amrex::ParGDBBase*,
1312  const amrex::Vector<std::unique_ptr<amrex::MultiFab>>&,
1313  const double time);
1314 
1315  /*! Restart tracer and hydro particles */
1316  void restartTracers ( amrex::ParGDBBase*, const std::string& );
1317 
1318  /*! Evolve tracers and hydro particles */
1319  void evolveTracers( int,
1320  double,
1321  amrex::Vector<amrex::Vector<amrex::MultiFab>>&,
1322  const amrex::Vector<std::unique_ptr<amrex::MultiFab>>& );
1323 
1324 #endif
1325 
1326 #ifdef ERF_USE_MULTIBLOCK
1327  MultiBlockContainer *m_mbc = nullptr;
1328 #endif
1329 
1330  static int verbose;
1331  static int mg_verbose;
1332  static bool use_fft;
1333 
1334  static int check_for_nans;
1335 
1336  // Diagnostic output interval
1337  static int sum_interval;
1338  static int pert_interval;
1339  static double sum_per;
1340 
1341  // Write in native AMReX or NetCDF format for each plotfile
1342  static PlotFileType plotfile3d_type_1;
1343  static PlotFileType plotfile3d_type_2;
1344  static PlotFileType plotfile2d_type_1;
1345  static PlotFileType plotfile2d_type_2;
1346 
1349 
1350  static StateInterpType interpolation_type;
1351 
1352  // NetCDF initialization (wrfinput) file
1353  static amrex::Vector<amrex::Vector<std::string>> nc_init_file;
1354  static amrex::Vector<amrex::Vector<int>> have_read_nc_init_file;
1355 
1356  // NetCDF initialization (wrfbdy/met_em) file
1357  static std::string nc_bdy_file;
1358  int real_width{0};
1359  bool real_extrap_w{true};
1360 
1361  // NetCDF initialization of zlo boundary values
1362  static std::string nc_low_file;
1363 
1364  // Options for vertical interpolation of met_em*.nc data.
1367  bool metgrid_debug_dry{false};
1368  bool metgrid_debug_psfc{false};
1369  bool metgrid_debug_msf{false};
1373  bool metgrid_use_sfc{true};
1374  bool metgrid_retain_sfc{false};
1378 
1379  // Options for ERF boundary files for real simulations.
1380  bool write_erfbdy{false}; // User-settable: write erfbdy during initialization
1381  bool use_erfbdy{false}; // Internal: set during init or restart
1382  std::string erfbdy_file{"erfbdy"}; // User-settable: erfbdy file name
1383  int nvars_erfbdy{0}; // Internal: number of variables in erfbdy
1384 
1385  amrex::Vector<amrex::BoxArray> ba1d;
1386  amrex::Vector<amrex::BoxArray> ba2d;
1387 
1388  std::unique_ptr<amrex::MultiFab> wrf_C1H;
1389  std::unique_ptr<amrex::MultiFab> wrf_C2H;
1390  std::unique_ptr<amrex::MultiFab> wrf_MUB;
1391  std::unique_ptr<amrex::MultiFab> wrf_RDNW;
1392  std::unique_ptr<amrex::MultiFab> wrf_PHB; // Base geopotential
1393 
1394  amrex::Vector<std::unique_ptr<amrex::MultiFab>> mf_PSFC;
1395 
1396  // 1D CDF output (for ingestion in AMR-Wind)
1397  static int output_1d_column;
1398  static int column_interval;
1399  static double column_per;
1402  static std::string column_file_name;
1403 
1404  // 2D BndryRegister output (for ingestion in AMR-Wind)
1409 
1410  // 2D BndryRegister input
1412 
1413  static int ng_dens_hse;
1414  static int ng_pres_hse;
1415 
1416  // Custom source terms
1417  amrex::Vector<std::unique_ptr<amrex::MultiFab>> rhotheta_src;
1418  amrex::Vector<std::unique_ptr<amrex::MultiFab>> rhoqt_src;
1419 
1420  amrex::Vector< amrex::Vector<amrex::Real> > h_w_subsid;
1421  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > d_w_subsid;
1422 
1423  amrex::Vector< amrex::Vector<amrex::Real> > h_u_geos;
1424  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > d_u_geos;
1425 
1426  amrex::Vector< amrex::Vector<amrex::Real> > h_v_geos;
1427  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > d_v_geos;
1428 
1429  // Function to read and populate above host vectors (if input file exists)
1430  void init_geo_wind_profile (const std::string input_file,
1431  amrex::Vector<amrex::Real>& u_geos,
1432  amrex::Gpu::DeviceVector<amrex::Real>& u_geos_d,
1433  amrex::Vector<amrex::Real>& v_geos,
1434  amrex::Gpu::DeviceVector<amrex::Real>& v_geos_d,
1435  const amrex::Geometry& lgeom,
1436  const amrex::Vector<amrex::Real>& zlev_stag);
1437 
1438  // This is a vector over levels of vectors across quantities of Vectors
1439  amrex::Vector<amrex::Vector<amrex::Vector<amrex::Real> > > h_rayleigh_ptrs;
1440  amrex::Vector<amrex::Vector<amrex::Vector<amrex::Real> > > h_sponge_ptrs;
1441 
1442  // These are vectors over levels of Vectors
1443  amrex::Vector<amrex::Vector<amrex::Real> > h_sinesq_ptrs;
1444  amrex::Vector<amrex::Vector<amrex::Real> > h_sinesq_stag_ptrs;
1445 
1446  // This is a vector over levels of vectors across quantities of DeviceVectors
1447  amrex::Vector<amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > > d_rayleigh_ptrs;
1448  amrex::Vector<amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > > d_sponge_ptrs;
1449 
1450  // These are vectors over levels of DeviceVectors
1451  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > d_sinesq_ptrs;
1452  amrex::Vector<amrex::Gpu::DeviceVector<amrex::Real> > d_sinesq_stag_ptrs;
1453 
1454  amrex::Vector<amrex::Real> h_havg_density;
1455  amrex::Vector<amrex::Real> h_havg_temperature;
1456  amrex::Vector<amrex::Real> h_havg_pressure;
1457  amrex::Vector<amrex::Real> h_havg_qv;
1458  amrex::Vector<amrex::Real> h_havg_qc;
1459 
1460  amrex::Gpu::DeviceVector<amrex::Real> d_havg_density;
1461  amrex::Gpu::DeviceVector<amrex::Real> d_havg_temperature;
1462  amrex::Gpu::DeviceVector<amrex::Real> d_havg_pressure;
1463  amrex::Gpu::DeviceVector<amrex::Real> d_havg_qv;
1464  amrex::Gpu::DeviceVector<amrex::Real> d_havg_qc;
1465 
1466  void refinement_criteria_setup ();
1467 
1468  std::unique_ptr<WriteBndryPlanes> m_w2d = nullptr;
1469  std::unique_ptr<ReadBndryPlanes> m_r2d = nullptr;
1470  std::unique_ptr<SurfaceLayer> m_SurfaceLayer = nullptr;
1471  amrex::Vector<std::unique_ptr<ForestDrag>> m_forest_drag;
1472 
1473  //
1474  // Holds info for dynamically generated tagging criteria
1475  //
1476  static amrex::Vector<amrex::AMRErrorTag> ref_tags;
1477 
1478  amrex::Vector<amrex::Vector<amrex::BoxArray>> subdomains;
1479 
1480  amrex::Vector<amrex::Real> dz_min;
1481 
1482  static AMREX_FORCE_INLINE
1483  int
1485  {
1486  int ngrow = 0;
1487 
1488  if (sc.use_num_diff)
1489  {
1490  ngrow = 3;
1491  } else {
1492  if (
1499  { ngrow = 3; }
1500  else if (
1507  { ngrow = 3; }
1508  else if (
1517  { ngrow = 3; }
1518  else if (
1527  { ngrow = 4; }
1528  else
1529  {
1530  if (sc.terrain_type == TerrainType::EB){
1531  ngrow = 4;
1532  } else {
1533  ngrow = 2;
1534  }
1535  }
1536  }
1537 
1538  return ngrow;
1539  }
1540 
1541  AMREX_FORCE_INLINE
1542  amrex::YAFluxRegister* getAdvFluxReg (int lev)
1543  {
1544  return advflux_reg[lev];
1545  }
1546 
1547  AMREX_FORCE_INLINE
1548  std::ostream&
1549  DataLog (int i)
1550  {
1551  return *datalog[i];
1552  }
1553 
1554  AMREX_FORCE_INLINE
1555  std::ostream&
1556  DerDataLog (int i)
1557  {
1558  return *der_datalog[i];
1559  }
1560 
1561  AMREX_FORCE_INLINE
1562  int
1563  NumDataLogs () noexcept
1564  {
1565  return static_cast<int>(datalog.size());
1566  }
1567 
1568  AMREX_FORCE_INLINE
1569  int
1570  NumDerDataLogs () noexcept
1571  {
1572  return static_cast<int>(der_datalog.size());
1573  }
1574 
1575 
1576  AMREX_FORCE_INLINE
1577  std::ostream&
1579  {
1580  return *sampleptlog[i];
1581  }
1582 
1583  AMREX_FORCE_INLINE
1584  int
1586  {
1587  return static_cast<int>(sampleptlog.size());
1588  }
1589 
1590  AMREX_FORCE_INLINE
1591  std::ostream&
1593  {
1594  return *samplelinelog[i];
1595  }
1596 
1597  AMREX_FORCE_INLINE
1598  int
1599  NumSampleLineLogs () noexcept
1600  {
1601  return static_cast<int>(samplelinelog.size());
1602  }
1603 
1604  amrex::IntVect&
1605  SamplePoint (int i)
1606  {
1607  return samplepoint[i];
1608  }
1609 
1610  AMREX_FORCE_INLINE
1611  int
1612  NumSamplePoints () noexcept
1613  {
1614  return static_cast<int>(samplepoint.size());
1615  }
1616 
1617  amrex::IntVect&
1618  SampleLine (int i)
1619  {
1620  return sampleline[i];
1621  }
1622 
1623  AMREX_FORCE_INLINE
1624  int
1625  NumSampleLines () noexcept
1626  {
1627  return static_cast<int>(sampleline.size());
1628  }
1629 
1630  static double startCPUTime;
1631  static double previousCPUTimeUsed;
1632 
1633  static double
1635  {
1636  int numCores = amrex::ParallelDescriptor::NProcs();
1637 #ifdef _OPENMP
1638  numCores = numCores * omp_get_max_threads();
1639 #endif
1640 
1641  double T =
1642  numCores * (amrex::ParallelDescriptor::second() - startCPUTime) +
1644 
1645  return T;
1646  }
1647 
1648  void setRecordDataInfo (int i, const std::string& filename) // NOLINT
1649  {
1650  if (amrex::ParallelDescriptor::IOProcessor())
1651  {
1652  datalog[i] = std::make_unique<std::fstream>();
1653  datalog[i]->open(filename.c_str(),std::ios::out|std::ios::app);
1654  if (!datalog[i]->good()) {
1655  amrex::FileOpenFailed(filename);
1656  }
1657  }
1658  amrex::ParallelDescriptor::Barrier("ERF::setRecordDataInfo");
1659  }
1660 
1661  void setRecordDerDataInfo (int i, const std::string& filename) // NOLINT
1662  {
1663  if (amrex::ParallelDescriptor::IOProcessor())
1664  {
1665  der_datalog[i] = std::make_unique<std::fstream>();
1666  der_datalog[i]->open(filename.c_str(),std::ios::out|std::ios::app);
1667  if (!der_datalog[i]->good()) {
1668  amrex::FileOpenFailed(filename);
1669  }
1670  }
1671  amrex::ParallelDescriptor::Barrier("ERF::setRecordDerDataInfo");
1672  }
1673 
1674  void setRecordEnergyDataInfo (int i, const std::string& filename) // NOLINT
1675  {
1676  if (amrex::ParallelDescriptor::IOProcessor())
1677  {
1678  tot_e_datalog[i] = std::make_unique<std::fstream>();
1679  tot_e_datalog[i]->open(filename.c_str(),std::ios::out|std::ios::app);
1680  if (!tot_e_datalog[i]->good()) {
1681  amrex::FileOpenFailed(filename);
1682  }
1683  }
1684  amrex::ParallelDescriptor::Barrier("ERF::setRecordEnergyDataInfo");
1685  }
1686 
1687  void setRecordSamplePointInfo (int i, int lev, amrex::IntVect& cell, const std::string& filename) // NOLINT
1688  {
1689  amrex::MultiFab dummy(grids[lev],dmap[lev],1,0);
1690  for (amrex::MFIter mfi(dummy); mfi.isValid(); ++mfi)
1691  {
1692  const amrex::Box& bx = mfi.validbox();
1693  if (bx.contains(cell)) {
1694  sampleptlog[i] = std::make_unique<std::fstream>();
1695  sampleptlog[i]->open(filename.c_str(),std::ios::out|std::ios::app);
1696  if (!sampleptlog[i]->good()) {
1697  amrex::FileOpenFailed(filename);
1698  }
1699  }
1700  }
1701  amrex::ParallelDescriptor::Barrier("ERF::setRecordSamplePointInfo");
1702  }
1703 
1704  void setRecordSampleLineInfo (int i, int lev, amrex::IntVect& cell, const std::string& filename) // NOLINT
1705  {
1706  amrex::MultiFab dummy(grids[lev],dmap[lev],1,0);
1707  for (amrex::MFIter mfi(dummy); mfi.isValid(); ++mfi)
1708  {
1709  const amrex::Box& bx = mfi.validbox();
1710  if (bx.contains(cell)) {
1711  samplelinelog[i] = std::make_unique<std::fstream>();
1712  samplelinelog[i]->open(filename.c_str(),std::ios::out|std::ios::app);
1713  if (!samplelinelog[i]->good()) {
1714  amrex::FileOpenFailed(filename);
1715  }
1716  }
1717  }
1718  amrex::ParallelDescriptor::Barrier("ERF::setRecordSampleLineInfo");
1719  }
1720 
1721  // Data sampler for line and plane output
1726  std::unique_ptr<LineSampler> line_sampler = nullptr;
1727  std::unique_ptr<PlaneSampler> plane_sampler = nullptr;
1728 
1729  amrex::Vector<std::unique_ptr<std::fstream> > datalog;
1730  amrex::Vector<std::unique_ptr<std::fstream> > der_datalog;
1731  amrex::Vector<std::unique_ptr<std::fstream> > tot_e_datalog;
1732  amrex::Vector<std::string> datalogname;
1733  amrex::Vector<std::string> der_datalogname;
1734  amrex::Vector<std::string> tot_e_datalogname;
1735 
1736  amrex::Vector<std::unique_ptr<std::fstream> > sampleptlog;
1737  amrex::Vector<std::string> sampleptlogname;
1738  amrex::Vector<amrex::IntVect> samplepoint;
1739 
1740  amrex::Vector<std::unique_ptr<std::fstream> > samplelinelog;
1741  amrex::Vector<std::string> samplelinelogname;
1742  amrex::Vector<amrex::IntVect> sampleline;
1743 
1744  //! The filename of the ith datalog file.
1745  [[nodiscard]] std::string DataLogName (int i) const noexcept { return datalogname[i]; }
1746  [[nodiscard]] std::string DerDataLogName (int i) const noexcept { return der_datalogname[i]; }
1747 
1748  //! The filename of the ith sampleptlog file.
1749  [[nodiscard]] std::string SamplePointLogName (int i) const noexcept { return sampleptlogname[i]; }
1750 
1751  //! The filename of the ith samplelinelog file.
1752  [[nodiscard]] std::string SampleLineLogName (int i) const noexcept { return samplelinelogname[i]; }
1753 
1754  // array of EB objects
1755  amrex::Vector<std::unique_ptr<eb_>> eb;
1756 
1757  [[nodiscard]] eb_ const& get_eb (int lev) const noexcept {
1758  AMREX_ASSERT(lev >= 0 && lev < eb.size() && eb[lev] != nullptr);
1759  return *eb[lev];
1760  }
1761 
1762  [[nodiscard]] amrex::EBFArrayBoxFactory const&
1763  EBFactory (int lev) const noexcept {
1764  return *(eb[lev]->get_const_factory());
1765  }
1766 
1767  [[nodiscard]] static int nghost_eb_basic ()
1768  { return 5; }
1769 
1770  // We need 5 for doing StateRedistribution; otherwise 4 would be enough
1771  [[nodiscard]] static int nghost_eb_volume ()
1772  { return 5; }
1773 
1774  [[nodiscard]] static int nghost_eb_full ()
1775  { return 4; }
1776 
1777 #ifdef ERF_USE_FFT
1778  // Vector over levels of Vector over subdomains of solvers
1779  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::FFT::Poisson<amrex::MultiFab>>>> m_3D_poisson;
1780  amrex::Vector<amrex::Vector<std::unique_ptr<amrex::FFT::PoissonHybrid<amrex::MultiFab>>>> m_2D_poisson;
1781 #endif
1782 
1783 public:
1784  void writeJobInfo (const std::string& dir,
1785  erf_provenance::ArtifactType artifact_type,
1786  int artifact_step,
1787  double artifact_time_seconds) const;
1788  static void writeBuildInfo (std::ostream& os);
1789 
1790  static void print_banner(MPI_Comm /*comm*/, std::ostream& /*out*/);
1791  static void print_usage(MPI_Comm /*comm*/, std::ostream& /*out*/);
1792  static void print_error(MPI_Comm /*comm*/, const std::string& msg);
1793  static void print_summary(std::ostream&);
1794  static void print_tpls(std::ostream& /*out*/);
1795 };
1796 
1797 #endif
AMREX_ENUM(StateInterpType, FullState, Perturbational)
constexpr amrex::Real one
Definition: ERF_Constants.H:9
constexpr amrex::Real zero
Definition: ERF_Constants.H:8
@ tau23
Definition: ERF_DataStruct.H:32
@ tau13
Definition: ERF_DataStruct.H:32
Contains the Eulerian microphysics class.
struct @29 out
#define NBCVAR_max
Definition: ERF_IndexDefines.H:29
@ Centered_6th
Real T
Definition: ERF_InitCustomPert_Bubble.H:106
Contains the Lagrangian microphysics class.
std::string name
Definition: ERF_Plotfile2DCatalog.cpp:97
amrex::Real Real
Definition: ERF_ShocInterface.H:19
void MakeHorizontalAverages()
Definition: ERF.cpp:2561
amrex::Vector< amrex::MultiFab > rU_new
Definition: ERF.H:962
static int last_check_file_step
Definition: ERF.H:1134
amrex::Vector< std::unique_ptr< amrex::MultiFab > > walldist
Definition: ERF.H:1074
void ConfigureDriverAtmosToOceanCoupling(bool use_coupling_driver, bool use_two_way_coupling, bool use_state_contract)
Definition: ERF_Coupling.cpp:118
amrex::Vector< amrex::Vector< amrex::Vector< amrex::Real > > > h_sponge_ptrs
Definition: ERF.H:1440
void GetOceanToAtmosSurfaceLayout(amrex::BoxArray &ba, amrex::DistributionMapping &dm)
Definition: ERF_Coupling.cpp:134
bool metgrid_basic_linear
Definition: ERF.H:1371
amrex::Vector< amrex::Vector< amrex::Box > > boxes_at_level
Definition: ERF.H:913
amrex::Vector< std::string > samplelinelogname
Definition: ERF.H:1741
void MakeEBGeometry()
int max_step
Definition: ERF.H:1157
void HurricaneTracker(int lev, double time, const amrex::MultiFab &cc_vel, const amrex::Real velmag_threshold, amrex::TagBoxArray *tags=nullptr)
Definition: ERF_RefineHurricane.cpp:109
bool metgrid_debug_msf
Definition: ERF.H:1369
AMREX_FORCE_INLINE std::ostream & DerDataLog(int i)
Definition: ERF.H:1556
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > mapfac
Definition: ERF.H:1077
void setRayleighRefFromSounding(bool restarting)
Set Rayleigh mean profiles from input sounding.
Definition: ERF_InitRayleigh.cpp:83
void init_only(int lev, double time)
Definition: ERF.cpp:1813
void turbPert_amplitude_w(const int lev)
Definition: ERF_InitTurbPert.cpp:60
amrex::Vector< std::unique_ptr< MRISplitIntegrator< amrex::Vector< amrex::MultiFab > > > > mri_integrator_mem
Definition: ERF.H:937
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > d_sinesq_stag_ptrs
Definition: ERF.H:1452
void check_vels_for_nans(amrex::MultiFab const &xvel, amrex::MultiFab const &yvel, amrex::MultiFab const &zvel)
Definition: ERF.cpp:2853
void Evolve()
Definition: ERF.cpp:136
amrex::Vector< amrex::MultiFab > avg_xmom
Definition: ERF.H:944
amrex::Vector< amrex::MultiFab > pp_inc
Definition: ERF.H:940
void make_eb_box()
amrex::Vector< ERFFillPatcher > FPr_u
Definition: ERF.H:1014
amrex::Vector< amrex::IntVect > sampleline
Definition: ERF.H:1742
amrex::Vector< std::string > subvol3d_var_names
Definition: ERF.H:1226
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_q1fx3_lev
Definition: ERF.H:1037
amrex::Vector< double > t_new
Definition: ERF.H:919
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Hwave_onegrid
Definition: ERF.H:1091
amrex::Vector< amrex::Vector< amrex::MultiFab > > vars_new
Definition: ERF.H:927
void FillCoarsePatch(int lev, double time)
Definition: ERF_FillCoarsePatch.cpp:21
void WriteAtIntermediateTime(int nstep, double time)
Definition: ERF.cpp:233
ERF(ERF &&) noexcept=delete
bool writeNow(double cur_time, const int nstep, const int plot_int, const double plot_per, const double dt_0, double &last_file_time)
Definition: ERF.cpp:2765
static double startCPUTime
Definition: ERF.H:1630
void FillBdyCCVels(amrex::MultiFab &mf_cc_vel, amrex::Geometry &lev_geom)
Definition: ERF_FillBdyCCVels.cpp:11
AMREX_FORCE_INLINE int NumSampleLineLogs() noexcept
Definition: ERF.H:1599
static double last_plot2d_file_time_1
Definition: ERF.H:1138
static void print_tpls(std::ostream &)
Definition: ERF_ConsoleIO.cpp:137
amrex::Vector< amrex::Real > dz_min
Definition: ERF.H:1480
amrex::Vector< amrex::MultiFab > lagged_delta_rt
Definition: ERF.H:943
amrex::Real plane_sampling_per
Definition: ERF.H:1725
std::string DataLogName(int i) const noexcept
The filename of the ith datalog file.
Definition: ERF.H:1745
std::string plot2d_file_2
Definition: ERF.H:1198
void remake_zphys(int lev, std::unique_ptr< amrex::MultiFab > &temp_zphys_nd)
Definition: ERF_MakeNewArrays.cpp:816
double m_plot3d_per_2
Definition: ERF.H:1210
static double bndry_output_planes_per
Definition: ERF.H:1407
amrex::Vector< std::unique_ptr< amrex::MultiFab > > thin_yforce
Definition: ERF.H:1127
void SetDriverAtmosToOceanStateContract(bool use_state_contract)
Definition: ERF_Coupling.cpp:128
void ErrorEst(int lev, amrex::TagBoxArray &tags, amrex::Real time, int ngrow) override
Definition: ERF_Tagging.cpp:25
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_hfx3_lev
Definition: ERF.H:1035
amrex::Vector< amrex::Vector< amrex::Real > > h_w_subsid
Definition: ERF.H:1420
static void print_banner(MPI_Comm, std::ostream &)
Definition: ERF_ConsoleIO.cpp:60
void volWgtColumnSum(int lev, const amrex::MultiFab &mf, int comp, amrex::MultiFab &mf_2d, const amrex::MultiFab &dJ)
Definition: ERF_VolWgtSum.cpp:82
amrex::Vector< std::unique_ptr< ShocDriver > > native_shoc_driver
Definition: ERF.H:1001
amrex::Vector< std::unique_ptr< std::fstream > > datalog
Definition: ERF.H:1729
std::string MakeFilename_EyeTracker_maxvel(int nstep)
Definition: ERF_TrackerOutput.cpp:66
amrex::Vector< ERFFillPatcher > FPr_v
Definition: ERF.H:1015
int cf_set_width
Definition: ERF.H:1012
std::unique_ptr< amrex::MultiFab > wrf_MUB
Definition: ERF.H:1390
void setPlotVariables(const std::string &pp_plot_var_names, amrex::Vector< std::string > &plot_var_names)
Definition: ERF_Plotfile.cpp:14
amrex::Gpu::DeviceVector< amrex::Real > d_havg_temperature
Definition: ERF.H:1461
void post_timestep(int nstep, double time, double dt_lev)
Definition: ERF.cpp:311
std::unique_ptr< amrex::MultiFab > wrf_C1H
Definition: ERF.H:1388
std::string erfbdy_file
Definition: ERF.H:1382
static int last_plot2d_file_step_2
Definition: ERF.H:1133
const int timeprecision
Definition: ERF.H:1150
void setRecordDataInfo(int i, const std::string &filename)
Definition: ERF.H:1648
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_hfx1_lev
Definition: ERF.H:1035
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_eye_track_xy
Definition: ERF.H:178
void SurfaceDataInterpolation(const int nlevs, const double time, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &z_phys_nd, bool regrid_forces_file_read)
Definition: ERF_SurfaceDataInterpolation.cpp:142
amrex::Vector< amrex::BoxArray > ba2d
Definition: ERF.H:1386
amrex::Array< amrex::Array< amrex::Real, AMREX_SPACEDIM *2 >, AMREX_SPACEDIM+NBCVAR_max > m_bc_extdir_vals
Definition: ERF.H:1108
void init_from_input_sounding(int lev)
Definition: ERF_InitFromInputSounding.cpp:53
void erf_enforce_hse(int lev, amrex::MultiFab &dens, amrex::MultiFab &pres, amrex::MultiFab &pi, amrex::MultiFab &th, amrex::MultiFab &qv, std::unique_ptr< amrex::MultiFab > &z_cc)
Definition: ERF_Init1D.cpp:210
amrex::Vector< amrex::Vector< amrex::MultiFab > > gradp
Definition: ERF.H:931
std::string plot3d_file_1
Definition: ERF.H:1195
eb_ const & get_eb(int lev) const noexcept
Definition: ERF.H:1757
amrex::Gpu::DeviceVector< amrex::Real > d_havg_qv
Definition: ERF.H:1463
amrex::Vector< std::unique_ptr< amrex::MultiFab > > qfx3_EB
Definition: ERF.H:1044
static amrex::Real column_loc_y
Definition: ERF.H:1401
static bool plot_file_on_restart
Definition: ERF.H:1145
amrex::Vector< amrex::Vector< amrex::MultiFab > > bckgnd_state
Definition: ERF.H:191
static int mg_verbose
Definition: ERF.H:1331
double m_check_per
Definition: ERF.H:1224
void ReadParameters()
Definition: ERF.cpp:1945
static double sum_per
Definition: ERF.H:1339
static amrex::Vector< std::string > PlotFileVarNames(amrex::Vector< std::string > plot_var_names)
Definition: ERF_Plotfile.cpp:253
amrex::Vector< amrex::Vector< amrex::MultiFab > > forecast_state_interp
Definition: ERF.H:186
bool FindInitialEye(int lev, const amrex::MultiFab &cc_vel, const amrex::Real velmag_threshold, amrex::Real &eye_x, amrex::Real &eye_y)
Definition: ERF_RefineHurricane.cpp:11
void InitializeFromFile()
amrex::Vector< std::unique_ptr< ERFPhysBCFunct_cons > > physbcs_cons
Definition: ERF.H:949
amrex::Vector< double > dt
Definition: ERF.H:921
static double stop_time
Definition: ERF.H:1161
void advance_lsm(int lev, amrex::MultiFab &cons_in, amrex::MultiFab &xvel_in, amrex::MultiFab &yvel_in, const double &time, const double &dt_advance)
Definition: ERF_AdvanceLSM.cpp:5
amrex::Vector< std::unique_ptr< amrex::MultiFab > > mf_PSFC
Definition: ERF.H:1394
ERF()
Definition: ERF_Constructors.cpp:21
void init_Dirichlet_bc_data(const std::string input_file)
Definition: ERF_InitBCs.cpp:735
~ERF() override
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_phys_nd_src
Definition: ERF.H:1058
amrex::Vector< std::unique_ptr< amrex::MultiFab > > detJ_cc
Definition: ERF.H:1053
amrex::Vector< std::string > lsm_flux_name
Definition: ERF.H:991
void WriteMyEBSurface()
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_phys_cc
Definition: ERF.H:1051
static double column_per
Definition: ERF.H:1399
amrex::Vector< std::unique_ptr< eb_ > > eb
Definition: ERF.H:1755
amrex::Vector< std::unique_ptr< amrex::MultiFab > > eddyDiffs_lev
Definition: ERF.H:1021
static SolverChoice solverChoice
Definition: ERF.H:1295
amrex::Vector< std::unique_ptr< amrex::MultiFab > > hfx3_EB
Definition: ERF.H:1043
amrex::Vector< ERFFillPatcher > FPr_c
Definition: ERF.H:1013
void WriteGenericPlotfileHeaderWithTerrain(std::ostream &HeaderFile, int nlevels, const amrex::Vector< amrex::BoxArray > &bArray, const amrex::Vector< std::string > &varnames, const amrex::Vector< amrex::Geometry > &my_geom, double time, const amrex::Vector< int > &level_steps, const amrex::Vector< amrex::IntVect > &my_ref_ratio, const std::string &versionName, const std::string &levelPrefix, const std::string &mfPrefix) const
Definition: ERF_Plotfile.cpp:2024
void update_box_for_refinement(std::string &ref_prefix, int &lev_for_box, amrex::RealBox &real_box, const double time)
Definition: ERF_RefineBox.cpp:336
bool plot_rad
Definition: ERF.H:1007
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_tracker_circle
Definition: ERF.H:182
static bool use_fft
Definition: ERF.H:1332
bool m_plot_face_vels
Definition: ERF.H:1213
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_maxvel_vs_time
Definition: ERF.H:180
void turbPert_update(const int lev, const double dt)
Definition: ERF_InitTurbPert.cpp:12
void sum_derived_quantities(double time)
Definition: ERF_WriteScalarProfiles.cpp:178
amrex::Vector< amrex::MultiFab > base_state_new
Definition: ERF.H:1086
std::string plot3d_file_2
Definition: ERF.H:1196
amrex::Vector< std::unique_ptr< amrex::MultiFab > > az
Definition: ERF.H:1056
int regrid_int
Definition: ERF.H:1188
void derive_diag_profiles(double time, amrex::Gpu::HostVector< amrex::Real > &h_avg_u, amrex::Gpu::HostVector< amrex::Real > &h_avg_v, amrex::Gpu::HostVector< amrex::Real > &h_avg_w, amrex::Gpu::HostVector< amrex::Real > &h_avg_rho, amrex::Gpu::HostVector< amrex::Real > &h_avg_th, amrex::Gpu::HostVector< amrex::Real > &h_avg_ksgs, amrex::Gpu::HostVector< amrex::Real > &h_avg_Kmv, amrex::Gpu::HostVector< amrex::Real > &h_avg_Khv, amrex::Gpu::HostVector< amrex::Real > &h_avg_qv, amrex::Gpu::HostVector< amrex::Real > &h_avg_qc, amrex::Gpu::HostVector< amrex::Real > &h_avg_qr, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqv, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqc, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqr, amrex::Gpu::HostVector< amrex::Real > &h_avg_qi, amrex::Gpu::HostVector< amrex::Real > &h_avg_qs, amrex::Gpu::HostVector< amrex::Real > &h_avg_qg, amrex::Gpu::HostVector< amrex::Real > &h_avg_uu, amrex::Gpu::HostVector< amrex::Real > &h_avg_uv, amrex::Gpu::HostVector< amrex::Real > &h_avg_uw, amrex::Gpu::HostVector< amrex::Real > &h_avg_vv, amrex::Gpu::HostVector< amrex::Real > &h_avg_vw, amrex::Gpu::HostVector< amrex::Real > &h_avg_ww, amrex::Gpu::HostVector< amrex::Real > &h_avg_uth, amrex::Gpu::HostVector< amrex::Real > &h_avg_vth, amrex::Gpu::HostVector< amrex::Real > &h_avg_wth, amrex::Gpu::HostVector< amrex::Real > &h_avg_thth, amrex::Gpu::HostVector< amrex::Real > &h_avg_ku, amrex::Gpu::HostVector< amrex::Real > &h_avg_kv, amrex::Gpu::HostVector< amrex::Real > &h_avg_kw, amrex::Gpu::HostVector< amrex::Real > &h_avg_p, amrex::Gpu::HostVector< amrex::Real > &h_avg_pu, amrex::Gpu::HostVector< amrex::Real > &h_avg_pv, amrex::Gpu::HostVector< amrex::Real > &h_avg_pw, amrex::Gpu::HostVector< amrex::Real > &h_avg_wthv)
Definition: ERF_Write1DProfiles.cpp:190
bool write_erfbdy
Definition: ERF.H:1380
void solve_with_gmres(int lev, const amrex::Box &subdomain, amrex::MultiFab &rhs, amrex::MultiFab &p, amrex::Array< amrex::MultiFab, AMREX_SPACEDIM > &fluxes, amrex::MultiFab &ax_sub, amrex::MultiFab &ay_sub, amrex::MultiFab &az_sub, amrex::MultiFab &, amrex::MultiFab &znd_sub)
Definition: ERF_SolveWithGMRES.cpp:12
amrex::Vector< amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > > d_sponge_ptrs
Definition: ERF.H:1448
amrex::Vector< long > dt_mri_ratio
Definition: ERF.H:922
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > Tau
Definition: ERF.H:1019
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vel_t_avg
Definition: ERF.H:934
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_q2fx3_lev
Definition: ERF.H:1038
void HurricaneEyeTrackerNotInitial(const SolverChoice &sc, const amrex::Geometry &geom, const amrex::Vector< amrex::MultiFab > &S_data, MoistureType moisture_type)
Definition: ERF_HurricaneDiagnostics.cpp:250
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::iMultiFab > > > lmask_lev
Definition: ERF.H:1027
AMREX_FORCE_INLINE int NumSamplePointLogs() noexcept
Definition: ERF.H:1585
amrex::Vector< amrex::Real > h_havg_pressure
Definition: ERF.H:1456
void update_diffusive_arrays(int lev, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm)
Definition: ERF_MakeNewArrays.cpp:529
std::unique_ptr< amrex::MultiFab > wrf_PHB
Definition: ERF.H:1392
static int verbose
Definition: ERF.H:1330
amrex::Vector< amrex::Real > h_havg_qc
Definition: ERF.H:1458
amrex::Vector< std::unique_ptr< ERFPhysBCFunct_w > > physbcs_w
Definition: ERF.H:952
static int nghost_eb_basic()
Definition: ERF.H:1767
void post_update(amrex::MultiFab &state_mf, double time, const amrex::Geometry &geom)
void AverageDownMoistStateTo(int crse_lev)
Conservation-preserving fine→coarse average of moist state components (RhoTheta and the contiguous mo...
Definition: ERF_AverageDown.cpp:243
amrex::Vector< double > last_subvol_time
Definition: ERF.H:1143
void FillForecastStateMultiFabs(const int lev, const std::string &filename, const std::unique_ptr< amrex::MultiFab > &z_phys_nd, amrex::Vector< amrex::Vector< amrex::MultiFab >> &forecast_state)
Definition: ERF_WeatherDataInterpolation.cpp:64
static std::string column_file_name
Definition: ERF.H:1402
amrex::Vector< std::unique_ptr< std::fstream > > samplelinelog
Definition: ERF.H:1740
amrex::Vector< std::unique_ptr< amrex::MultiFab > > terrain_blanking
Definition: ERF.H:1071
std::unique_ptr< Microphysics > micro
Definition: ERF.H:972
int m_plot2d_int_2
Definition: ERF.H:1204
amrex::Vector< amrex::MultiFab > base_state
Definition: ERF.H:1085
void fill_rhs(amrex::MultiFab &rhs_mf, const amrex::MultiFab &state_mf, double time, const amrex::Geometry &geom)
AMREX_FORCE_INLINE amrex::YAFluxRegister * getAdvFluxReg(int lev)
Definition: ERF.H:1542
int m_plot3d_int_1
Definition: ERF.H:1201
amrex::Vector< amrex::Real > h_havg_density
Definition: ERF.H:1454
static double last_plot2d_file_time_2
Definition: ERF.H:1139
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_phys_nd_new
Definition: ERF.H:1065
void derive_diag_profiles_stag(double time, amrex::Gpu::HostVector< amrex::Real > &h_avg_u, amrex::Gpu::HostVector< amrex::Real > &h_avg_v, amrex::Gpu::HostVector< amrex::Real > &h_avg_w, amrex::Gpu::HostVector< amrex::Real > &h_avg_rho, amrex::Gpu::HostVector< amrex::Real > &h_avg_th, amrex::Gpu::HostVector< amrex::Real > &h_avg_ksgs, amrex::Gpu::HostVector< amrex::Real > &h_avg_Kmv, amrex::Gpu::HostVector< amrex::Real > &h_avg_Khv, amrex::Gpu::HostVector< amrex::Real > &h_avg_qv, amrex::Gpu::HostVector< amrex::Real > &h_avg_qc, amrex::Gpu::HostVector< amrex::Real > &h_avg_qr, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqv, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqc, amrex::Gpu::HostVector< amrex::Real > &h_avg_wqr, amrex::Gpu::HostVector< amrex::Real > &h_avg_qi, amrex::Gpu::HostVector< amrex::Real > &h_avg_qs, amrex::Gpu::HostVector< amrex::Real > &h_avg_qg, amrex::Gpu::HostVector< amrex::Real > &h_avg_uu, amrex::Gpu::HostVector< amrex::Real > &h_avg_uv, amrex::Gpu::HostVector< amrex::Real > &h_avg_uw, amrex::Gpu::HostVector< amrex::Real > &h_avg_vv, amrex::Gpu::HostVector< amrex::Real > &h_avg_vw, amrex::Gpu::HostVector< amrex::Real > &h_avg_ww, amrex::Gpu::HostVector< amrex::Real > &h_avg_uth, amrex::Gpu::HostVector< amrex::Real > &h_avg_vth, amrex::Gpu::HostVector< amrex::Real > &h_avg_wth, amrex::Gpu::HostVector< amrex::Real > &h_avg_thth, amrex::Gpu::HostVector< amrex::Real > &h_avg_ku, amrex::Gpu::HostVector< amrex::Real > &h_avg_kv, amrex::Gpu::HostVector< amrex::Real > &h_avg_kw, amrex::Gpu::HostVector< amrex::Real > &h_avg_p, amrex::Gpu::HostVector< amrex::Real > &h_avg_pu, amrex::Gpu::HostVector< amrex::Real > &h_avg_pv, amrex::Gpu::HostVector< amrex::Real > &h_avg_pw, amrex::Gpu::HostVector< amrex::Real > &h_avg_wthv)
Definition: ERF_Write1DProfiles_stag.cpp:296
void create_background_state_for_ensemble(int lev, amrex::MultiFab &mf_cc_pert, amrex::MultiFab &cons_pert, amrex::MultiFab &xvel_pert, amrex::MultiFab &yvel_pert, amrex::MultiFab &zvel_pert)
Definition: ERF_InitForEnsemble.cpp:516
const amrex::Vector< std::string > derived_subvol_names
Definition: ERF.H:1292
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_minpressure_vs_time
Definition: ERF.H:181
bool metgrid_retain_sfc
Definition: ERF.H:1374
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Qv_prim
Definition: ERF.H:957
static int sum_interval
Definition: ERF.H:1337
static int pert_interval
Definition: ERF.H:1338
amrex::Real line_sampling_per
Definition: ERF.H:1724
void restart()
Definition: ERF.cpp:1747
amrex::Vector< double > t_avg_cnt
Definition: ERF.H:935
static int last_plot3d_file_step_2
Definition: ERF.H:1131
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_q1fx2_lev
Definition: ERF.H:1037
amrex::Vector< double > fixed_fast_dt
Definition: ERF.H:1183
amrex::IntVect & SampleLine(int i)
Definition: ERF.H:1618
void build_fine_mask(int lev, amrex::MultiFab &fine_mask)
Definition: ERF_VolWgtSum.cpp:125
void HurricaneMaxVelTracker(const amrex::Geometry &geom, const amrex::MultiFab &mf_cc_vel, const double &time)
Definition: ERF_HurricaneDiagnostics.cpp:501
int file_name_digits
Definition: ERF.H:1347
amrex::Vector< amrex::MultiFab > rV_new
Definition: ERF.H:964
std::string PlotFileName(int lev) const
void initialize_integrator(int lev, amrex::MultiFab &cons_mf, amrex::MultiFab &vel_mf)
Definition: ERF_MakeNewArrays.cpp:898
amrex::Vector< std::unique_ptr< amrex::MultiFab > > thin_zforce
Definition: ERF.H:1128
amrex::Vector< std::string > plot3d_var_names_2
Definition: ERF.H:1229
void sample_points(int lev, double time, amrex::IntVect cell, amrex::MultiFab &mf)
Definition: ERF_WriteScalarProfiles.cpp:533
void WeatherDataInterpolation(const int nlevs, const double time, amrex::Vector< std::unique_ptr< amrex::MultiFab >> &z_phys_nd, bool regrid_forces_file_read)
Definition: ERF_WeatherDataInterpolation.cpp:347
amrex::Vector< amrex::BCRec > domain_bcs_type
Definition: ERF.H:1101
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Qr_prim
Definition: ERF.H:958
void apply_gaussian_smoothing_to_perturbations(const int lev, amrex::MultiFab &mf_cc_pert)
Definition: ERF_InitForEnsemble.cpp:86
std::string pp_prefix
Definition: ERF.H:637
std::string SampleLineLogName(int i) const noexcept
The filename of the ith samplelinelog file.
Definition: ERF.H:1752
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > sst_lev
Definition: ERF.H:1025
amrex::Vector< std::string > plot2d_var_names_1
Definition: ERF.H:1230
amrex::Vector< std::unique_ptr< amrex::MultiFab > > thin_xforce
Definition: ERF.H:1126
AMREX_FORCE_INLINE int NumSamplePoints() noexcept
Definition: ERF.H:1612
bool metgrid_use_sfc
Definition: ERF.H:1373
void sample_lines(int lev, double time, amrex::IntVect cell, amrex::MultiFab &mf)
Definition: ERF_WriteScalarProfiles.cpp:569
void setPlotVariables2D(const std::string &pp_plot_var_names, amrex::Vector< std::string > &plot_var_names)
Definition: ERF_Plotfile2D.cpp:109
void init_custom(int lev)
Definition: ERF_InitCustomPertState.cpp:26
amrex::Vector< std::unique_ptr< IRadiation > > rad
Definition: ERF.H:994
void ComputeAndWriteEnsemblePerturbations()
std::unique_ptr< ProblemBase > prob
Definition: ERF.H:909
amrex::Vector< int > num_files_at_level
Definition: ERF.H:912
static double last_plot3d_file_time_2
Definition: ERF.H:1137
void init_bcs()
Definition: ERF_InitBCs.cpp:299
int profile_int
Definition: ERF.H:1218
bool metgrid_debug_quiescent
Definition: ERF.H:1365
amrex::Vector< std::unique_ptr< ERFPhysBCFunct_u > > physbcs_u
Definition: ERF.H:950
bool destag_profiles
Definition: ERF.H:1219
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > stretched_dz_d
Definition: ERF.H:1083
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > th_bc_data
Definition: ERF.H:886
std::string DerDataLogName(int i) const noexcept
Definition: ERF.H:1746
void compute_max_pressure_gradient_diagnostic(int lev)
Definition: ERF_Diagnostics.cpp:12
void HurricaneEyeTracker(const SolverChoice &solverChoice)
Definition: ERF_HurricaneDiagnostics.cpp:475
void project_momenta(int lev, double l_time, double l_dt, amrex::Vector< amrex::MultiFab > &vars)
Definition: ERF_PoissonSolve.cpp:102
int m_check_int
Definition: ERF.H:1223
void HurricaneEyeTrackerInitial(const SolverChoice &sc, const amrex::Geometry &geom, const amrex::Vector< amrex::MultiFab > &S_data, const amrex::Real &hurricane_eye_latitude, const amrex::Real &hurricane_eye_longitude)
Definition: ERF_HurricaneDiagnostics.cpp:175
void ReadVelsOnlyFromCheckpointFile(int lev_to_fill, std::string &chkfile)
Definition: ERF_Checkpoint.cpp:1232
static double last_plot3d_file_time_1
Definition: ERF.H:1136
double m_plot2d_per_2
Definition: ERF.H:1212
amrex::Vector< amrex::MultiFab > surface_state_1
Definition: ERF.H:187
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_diss_lev
Definition: ERF.H:1036
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > d_sinesq_ptrs
Definition: ERF.H:1451
amrex::Vector< amrex::MultiFab > rU_old
Definition: ERF.H:961
AMREX_FORCE_INLINE int NumSampleLines() noexcept
Definition: ERF.H:1625
amrex::Array< bool, AMREX_SPACEDIM *2 > m_bc_nonreflecting
Definition: ERF.H:1114
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Theta_prim
Definition: ERF.H:956
static double getCPUTime()
Definition: ERF.H:1634
void write_1D_profiles(double time)
Definition: ERF_Write1DProfiles.cpp:17
void init1DArrays()
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_t_rk
Definition: ERF.H:1068
static int check_for_nans
Definition: ERF.H:1334
void make_subdomains(const amrex::BoxList &ba, amrex::Vector< amrex::BoxArray > &bins)
Definition: ERF_MakeSubdomains.cpp:6
void compute_divergence(int lev, amrex::MultiFab &rhs, amrex::Array< amrex::MultiFab const *, AMREX_SPACEDIM > rho0_u_const, amrex::Geometry const &geom_at_lev)
Definition: ERF_ComputeDivergence.cpp:10
std::string MakeVTKFilename(int nstep)
Definition: ERF_TrackerOutput.cpp:11
void initRayleigh_at_level(const int &lev)
Initialize Rayleigh damping profiles at a level.
Definition: ERF_InitRayleigh.cpp:14
void appendPlotVariables(const std::string &pp_plot_var_names, amrex::Vector< std::string > &plot_var_names)
Definition: ERF_Plotfile.cpp:176
void read_box_for_refinement(std::string &ref_prefix, int &lev_for_box, amrex::RealBox &real_box)
Definition: ERF_RefineBox.cpp:6
int nvars_erfbdy
Definition: ERF.H:1383
amrex::Vector< std::unique_ptr< ERFPhysBCFunct_v > > physbcs_v
Definition: ERF.H:951
amrex::Vector< double > t_old
Definition: ERF.H:920
amrex::Vector< std::string > tot_e_datalogname
Definition: ERF.H:1734
void FillIntermediatePatch(int lev, double time, const amrex::Vector< amrex::MultiFab * > &mfs_vel, const amrex::Vector< amrex::MultiFab * > &mfs_mom, int ng_cons, int ng_vel, bool cons_only, int icomp_cons, int ncomp_cons)
Definition: ERF_FillIntermediatePatch.cpp:28
static int output_bndry_planes
Definition: ERF.H:1405
void project_velocity_tb(int lev, double dt, amrex::Vector< amrex::MultiFab > &vars)
Definition: ERF_PoissonSolve_tb.cpp:22
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > d_v_geos
Definition: ERF.H:1427
void update_terrain_arrays(int lev)
Definition: ERF_MakeNewArrays.cpp:881
static std::string nc_bdy_file
Definition: ERF.H:1357
bool metgrid_interp_theta
Definition: ERF.H:1370
static amrex::Vector< amrex::Vector< std::string > > nc_init_file
Definition: ERF.H:1353
static int input_bndry_planes
Definition: ERF.H:1411
static StateInterpType interpolation_type
Definition: ERF.H:1350
amrex::Gpu::DeviceVector< amrex::Real > d_havg_qc
Definition: ERF.H:1464
amrex::Real cloud_fraction(double time)
Definition: ERF_WriteScalarProfiles.cpp:457
void AverageDown()
Definition: ERF_AverageDown.cpp:16
amrex::Vector< amrex::Vector< amrex::Real > > h_v_geos
Definition: ERF.H:1426
bool regrid_level_0_on_restart
Definition: ERF.H:1192
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Lwave_onegrid
Definition: ERF.H:1092
void advance_microphysics(int lev, amrex::MultiFab &cons_in, const double &dt_advance, const int &iteration, const double &time)
Definition: ERF_AdvanceMicrophysics.cpp:5
InputSoundingData input_sounding_data
Definition: ERF.H:877
void write_1D_profiles_stag(double time)
Definition: ERF_Write1DProfiles_stag.cpp:25
void ApplyOceanSurfaceState(const amrex::Vector< amrex::MultiFab * > &state, double time)
Definition: ERF_Coupling.cpp:508
amrex::Vector< amrex::Vector< amrex::Real > > h_sinesq_ptrs
Definition: ERF.H:1443
void create_random_perturbations(const int lev, amrex::MultiFab &mf_cc_pert)
Definition: ERF_InitForEnsemble.cpp:14
void Write2DPlotFile(int which, PlotFileType plotfile_type, amrex::Vector< std::string > plot_var_names)
Definition: ERF_Plotfile2D.cpp:151
amrex::Gpu::DeviceVector< amrex::Real > d_havg_density
Definition: ERF.H:1460
amrex::Vector< std::unique_ptr< amrex::MultiFab > > rhotheta_src
Definition: ERF.H:1417
void init_from_hse(int lev)
Definition: ERF_InitFromHSE.cpp:32
const std::string datetime_format
Definition: ERF.H:1164
amrex::Vector< amrex::YAFluxRegister * > advflux_reg
Definition: ERF.H:1096
void check_mesh_type(int lev)
Definition: ERF.cpp:2953
erf_provenance::ExecutionProvenance execution_provenance
Definition: ERF.H:1171
int metgrid_force_sfc_k
Definition: ERF.H:1377
void setSubVolVariables(const std::string &pp_subvol_var_names, amrex::Vector< std::string > &subvol_var_names)
Definition: ERF_WriteSubvolume.cpp:9
static int ng_pres_hse
Definition: ERF.H:1414
bool real_extrap_w
Definition: ERF.H:1359
static amrex::Real cfl
Definition: ERF.H:1174
amrex::Vector< std::unique_ptr< amrex::MultiFab > > fine_mask
Definition: ERF.H:1080
amrex::Gpu::DeviceVector< amrex::BCRec > domain_bcs_type_d
Definition: ERF.H:1102
amrex::Vector< std::unique_ptr< ForestDrag > > m_forest_drag
Definition: ERF.H:1471
const int datwidth
Definition: ERF.H:1148
amrex::Vector< amrex::BoxArray > ba1d
Definition: ERF.H:1385
InputSpongeData input_sponge_data
Definition: ERF.H:880
amrex::Vector< amrex::Vector< amrex::BoxArray > > subdomains
Definition: ERF.H:1478
void check_state_for_nans(amrex::MultiFab const &S)
Definition: ERF.cpp:2786
std::string restart_chkfile
Definition: ERF.H:1167
bool metgrid_use_below_sfc
Definition: ERF.H:1372
amrex::Vector< std::string > sampleptlogname
Definition: ERF.H:1737
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > xvel_bc_data
Definition: ERF.H:883
void InitData_pre()
Definition: ERF.cpp:521
amrex::IntVect & SamplePoint(int i)
Definition: ERF.H:1605
bool use_datetime
Definition: ERF.H:1163
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_track_xy
Definition: ERF.H:177
amrex::Vector< amrex::Vector< amrex::Real > > h_u_geos
Definition: ERF.H:1423
void InitializeLevelFromData(int lev, const amrex::MultiFab &initial_data)
std::string subvol_file
Definition: ERF.H:1199
int rad_datalog_int
Definition: ERF.H:1008
amrex::Vector< std::string > datalogname
Definition: ERF.H:1732
void initHSE()
Initialize HSE.
Definition: ERF_Init1D.cpp:192
void AverageDownTo(int crse_lev, int scomp, int ncomp, bool do_perturbational_and_momenta=true)
Definition: ERF_AverageDown.cpp:36
static amrex::Real column_loc_x
Definition: ERF.H:1400
static double dt_max
Definition: ERF.H:1179
amrex::Vector< amrex::MultiFab > surface_state_2
Definition: ERF.H:188
amrex::Vector< std::unique_ptr< amrex::MultiFab > > ax
Definition: ERF.H:1054
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_phys_nd
Definition: ERF.H:1050
void MakeDiagnosticAverage(amrex::Vector< amrex::Real > &h_havg, amrex::MultiFab &S, int n)
Definition: ERF.cpp:2667
amrex::Real metgrid_proximity
Definition: ERF.H:1375
amrex::Real volWgtSumMF(int lev, const amrex::MultiFab &mf, int comp, const amrex::MultiFab &dJ, const amrex::MultiFab &mfx, const amrex::MultiFab &mfy, bool finemask, bool local=true)
Definition: ERF_VolWgtSum.cpp:20
void setRecordDerDataInfo(int i, const std::string &filename)
Definition: ERF.H:1661
amrex::Vector< amrex::Real > h_havg_temperature
Definition: ERF.H:1455
amrex::Vector< std::unique_ptr< std::fstream > > sampleptlog
Definition: ERF.H:1736
void poisson_wall_dist(int lev)
Definition: ERF_PoissonWallDist.cpp:22
amrex::Vector< std::array< amrex::Real, 2 > > hurricane_eye_track_latlon
Definition: ERF.H:179
amrex::Vector< std::unique_ptr< amrex::MultiFab > > detJ_cc_src
Definition: ERF.H:1060
amrex::Gpu::DeviceVector< amrex::Real > d_havg_pressure
Definition: ERF.H:1462
static bool is_it_time_for_action(int nstep, double time, double dt, int action_interval, double action_per)
Definition: ERF_WriteScalarProfiles.cpp:659
std::string plot2d_file_1
Definition: ERF.H:1197
void Advance(int lev, double time, double dt_lev, int iteration, int ncycle)
Definition: ERF_Advance.cpp:20
amrex::Vector< std::string > der_datalogname
Definition: ERF.H:1733
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SmnSmn_lev
Definition: ERF.H:1022
const amrex::Vector< std::string > derived_names
Definition: ERF.H:1241
void sum_integrated_quantities(double time)
Definition: ERF_WriteScalarProfiles.cpp:15
std::string MakeVTKFilename_TrackerCircle(int nstep)
Definition: ERF_TrackerOutput.cpp:24
void ComputeGlobalMinLocation(const SolverChoice &sc, const amrex::Geometry &geom, const amrex::Vector< amrex::MultiFab > &S_data, amrex::Real *d_val_min_ptr, int *d_i_min_ptr, int *d_j_min_ptr, amrex::Real &global_val_min, int &global_i_min, int &global_j_min)
Definition: ERF_HurricaneDiagnostics.cpp:36
amrex::Vector< std::unique_ptr< amrex::MultiFab > > ay_src
Definition: ERF.H:1062
void ComputeDt(int step=-1, double cur_time_d=0.0)
Definition: ERF_ComputeTimestep.cpp:11
AMREX_FORCE_INLINE std::ostream & DataLog(int i)
Definition: ERF.H:1549
std::string MakeVTKFilename_EyeTracker_xy(int nstep)
Definition: ERF_TrackerOutput.cpp:38
amrex::Vector< double > m_subvol_per
Definition: ERF.H:1207
amrex::Vector< int > nsubsteps
Definition: ERF.H:916
amrex::Vector< std::unique_ptr< amrex::iMultiFab > > yflux_imask
Definition: ERF.H:1121
amrex::Vector< amrex::MultiFab > rW_new
Definition: ERF.H:966
amrex::Vector< amrex::MultiFab > weather_forecast_data_2
Definition: ERF.H:183
amrex::Vector< std::unique_ptr< amrex::MultiFab > > lon_m
Definition: ERF.H:872
std::unique_ptr< WriteBndryPlanes > m_w2d
Definition: ERF.H:1468
AMREX_FORCE_INLINE std::ostream & SampleLineLog(int i)
Definition: ERF.H:1592
void HurricaneMinPressureTracker(MoistureType moisture_type, const amrex::Geometry &geom, const amrex::MultiFab &mf_cons_var, const double &time)
Definition: ERF_HurricaneDiagnostics.cpp:556
amrex::Vector< amrex::Vector< amrex::MultiFab * > > lsm_flux
Definition: ERF.H:992
double EvolveOneStep(double time, double dt_request)
Definition: ERF_Coupling.cpp:92
bool use_erfbdy
Definition: ERF.H:1381
amrex::Vector< std::string > plot3d_var_names_1
Definition: ERF.H:1228
std::string SamplePointLogName(int i) const noexcept
The filename of the ith sampleptlog file.
Definition: ERF.H:1749
void InitData_post()
Definition: ERF.cpp:553
void refinement_criteria_setup()
Definition: ERF_Tagging.cpp:472
double estTimeStep(int lev, long &dt_fast_ratio) const
Definition: ERF_ComputeTimestep.cpp:60
static int nghost_eb_volume()
Definition: ERF.H:1771
bool metgrid_debug_dry
Definition: ERF.H:1367
static AMREX_FORCE_INLINE int ComputeGhostCells(const SolverChoice &sc)
Definition: ERF.H:1484
static int bndry_output_planes_interval
Definition: ERF.H:1406
void init_geo_wind_profile(const std::string input_file, amrex::Vector< amrex::Real > &u_geos, amrex::Gpu::DeviceVector< amrex::Real > &u_geos_d, amrex::Vector< amrex::Real > &v_geos, amrex::Gpu::DeviceVector< amrex::Real > &v_geos_d, const amrex::Geometry &lgeom, const amrex::Vector< amrex::Real > &zlev_stag)
Definition: ERF_InitGeowind.cpp:10
void ERF_shared()
Definition: ERF_Constructors.cpp:61
amrex::Vector< std::string > plot2d_var_names_2
Definition: ERF.H:1231
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > Tau_corr
Definition: ERF.H:1020
void FillPatchCrseLevel(int lev, double time, const amrex::Vector< amrex::MultiFab * > &mfs_vel, bool cons_only=false)
Definition: ERF_FillPatch.cpp:290
void WriteSubvolume(int isub, amrex::Vector< std::string > subvol_var_names)
Definition: ERF_WriteSubvolume.cpp:145
void sum_energy_quantities(double time)
Definition: ERF_WriteScalarProfiles.cpp:317
static int nghost_eb_full()
Definition: ERF.H:1774
void RemakeLevel(int lev, amrex::Real time, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm) override
Definition: ERF_MakeNewLevel.cpp:562
static void GotoNextLine(std::istream &is)
Definition: ERF_Checkpoint.cpp:25
void MakeNewLevelFromScratch(int lev, amrex::Real time, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm) override
Definition: ERF_MakeNewLevel.cpp:25
void init_stuff(int lev, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm, amrex::Vector< amrex::MultiFab > &lev_new, amrex::Vector< amrex::MultiFab > &lev_old, amrex::MultiFab &tmp_base_state, std::unique_ptr< amrex::MultiFab > &tmp_zphys_nd)
Definition: ERF_MakeNewArrays.cpp:24
amrex::Vector< amrex::MultiFab > zmom_crse_rhs
Definition: ERF.H:970
void advance_radiation(int lev, amrex::MultiFab &cons_in, const double &dt_advance)
Definition: ERF_AdvanceRadiation.cpp:5
bool metgrid_debug_isothermal
Definition: ERF.H:1366
void PerformDataAssimilation(int da_iter)
amrex::Vector< std::string > lsm_data_name
Definition: ERF.H:989
amrex::Vector< std::unique_ptr< amrex::MultiFab > > rhoqt_src
Definition: ERF.H:1418
void initSponge()
Initialize sponge profiles.
Definition: ERF_InitSponge.cpp:35
std::unique_ptr< PlaneSampler > plane_sampler
Definition: ERF.H:1727
void check_for_low_temp(amrex::MultiFab &S)
Definition: ERF.cpp:2880
amrex::Vector< std::unique_ptr< std::fstream > > der_datalog
Definition: ERF.H:1730
static amrex::Vector< amrex::Vector< int > > have_read_nc_init_file
Definition: ERF.H:1354
amrex::Vector< std::unique_ptr< std::fstream > > tot_e_datalog
Definition: ERF.H:1731
amrex::Vector< std::unique_ptr< amrex::MultiFab > > sinPhi_m
Definition: ERF.H:874
static double previousCPUTimeUsed
Definition: ERF.H:1631
void writeJobInfo(const std::string &dir, erf_provenance::ArtifactType artifact_type, int artifact_step, double artifact_time_seconds) const
Definition: ERF_WriteJobInfo.cpp:12
bool m_driver_has_atm2ocn_coupling
Definition: ERF.H:704
double m_plot3d_per_1
Definition: ERF.H:1209
int real_width
Definition: ERF.H:1358
void MakeNewLevelFromCoarse(int lev, amrex::Real time, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm) override
Definition: ERF_MakeNewLevel.cpp:278
amrex::Vector< int > last_subvol_step
Definition: ERF.H:1142
void fill_from_bndryregs(const amrex::Vector< amrex::MultiFab * > &mfs, double time)
Definition: ERF_BoundaryConditionsBndryReg.cpp:13
amrex::Vector< std::unique_ptr< amrex::MultiFab > > ax_src
Definition: ERF.H:1061
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > urb_frac_lev
Definition: ERF.H:1032
amrex::Vector< std::unique_ptr< amrex::MultiFab > > z_phys_cc_src
Definition: ERF.H:1059
void input_sponge(int lev)
Definition: ERF_InitSponge.cpp:17
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > d_u_geos
Definition: ERF.H:1424
bool m_driver_uses_two_way_coupling
Definition: ERF.H:705
void Define_ERFFillPatchers(int lev)
Definition: ERF.cpp:2739
AMREX_FORCE_INLINE int NumDataLogs() noexcept
Definition: ERF.H:1563
static PlotFileType plotfile3d_type_2
Definition: ERF.H:1343
void setRecordEnergyDataInfo(int i, const std::string &filename)
Definition: ERF.H:1674
TurbulentPerturbation turbPert
Definition: ERF.H:1298
amrex::Vector< amrex::MultiFab > rW_old
Definition: ERF.H:965
amrex::Vector< double > fixed_dt
Definition: ERF.H:1182
amrex::Vector< amrex::Vector< amrex::MultiFab > > forecast_state_2
Definition: ERF.H:185
void check_for_negative_theta(amrex::MultiFab &S)
Definition: ERF.cpp:2915
void ImposeBCsOnPhi(int lev, amrex::MultiFab &phi, const amrex::Box &subdomain)
Definition: ERF_ImposeBCsOnPhi.cpp:12
void Interp2DArrays(int lev, const amrex::BoxArray &my_ba2d, const amrex::DistributionMapping &my_dm)
Definition: ERF.cpp:1515
void ClearLevel(int lev) override
Definition: ERF_MakeNewLevel.cpp:840
std::string MakeFilename_EyeTracker_minpressure(int nstep)
Definition: ERF_TrackerOutput.cpp:80
std::unique_ptr< amrex::MultiFab > wrf_RDNW
Definition: ERF.H:1391
std::unique_ptr< SurfaceLayer > m_SurfaceLayer
Definition: ERF.H:1470
static double dt_max_initial
Definition: ERF.H:1178
static PlotFileType plotfile2d_type_2
Definition: ERF.H:1345
int plane_sampling_interval
Definition: ERF.H:1723
static amrex::Vector< amrex::AMRErrorTag > ref_tags
Definition: ERF.H:1476
void make_physbcs(int lev)
Definition: ERF_MakeNewArrays.cpp:920
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > d_w_subsid
Definition: ERF.H:1421
amrex::Vector< ERFFillPatcher > FPr_w
Definition: ERF.H:1016
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_hfx2_lev
Definition: ERF.H:1035
void WriteLinePlot(const std::string &filename, amrex::Vector< std::array< amrex::Real, 2 >> &points_xy)
Definition: ERF_Write1DProfiles.cpp:587
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::iMultiFab > > > soil_type_lev
Definition: ERF.H:1031
amrex::Vector< amrex::Vector< amrex::Real > > zlevels_stag
Definition: ERF.H:1047
AMREX_FORCE_INLINE int NumDerDataLogs() noexcept
Definition: ERF.H:1570
amrex::EBFArrayBoxFactory const & EBFactory(int lev) const noexcept
Definition: ERF.H:1763
amrex::Vector< amrex::Vector< amrex::MultiFab * > > lsm_data
Definition: ERF.H:990
void init_immersed_forcing(int lev)
Definition: ERF_InitImmersedForcing.cpp:15
amrex::Vector< amrex::Vector< amrex::Real > > stretched_dz_h
Definition: ERF.H:1082
bool use_real_time_in_pltname
Definition: ERF.H:1348
int m_plot2d_int_1
Definition: ERF.H:1203
void WriteCheckpointFile() const
Definition: ERF_Checkpoint.cpp:35
static int output_1d_column
Definition: ERF.H:1397
bool metgrid_debug_psfc
Definition: ERF.H:1368
void ReadStormTrackerRestart()
Definition: ERF_HurricaneDiagnostics.cpp:315
amrex::Vector< std::unique_ptr< amrex::MultiFab > > az_src
Definition: ERF.H:1063
static int fixed_mri_dt_ratio
Definition: ERF.H:1184
static amrex::Real init_shrink
Definition: ERF.H:1176
void WriteVTKPolyline(const std::string &filename, amrex::Vector< std::array< amrex::Real, 2 >> &points_xy)
Definition: ERF_TrackerOutput.cpp:94
void Write3DPlotFile(int which, PlotFileType plotfile_type, amrex::Vector< std::string > plot_var_names)
Definition: ERF_Plotfile.cpp:265
void HurricaneTrackerCircle()
Definition: ERF_HurricaneDiagnostics.cpp:149
static double last_check_file_time
Definition: ERF.H:1140
static std::string nc_low_file
Definition: ERF.H:1362
amrex::Vector< amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > > d_rayleigh_ptrs
Definition: ERF.H:1447
void InitData()
Definition: ERF.cpp:512
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Lwave
Definition: ERF.H:1090
amrex::Vector< std::unique_ptr< amrex::MultiFab > > cosPhi_m
Definition: ERF.H:874
void ParameterSanityChecks()
Definition: ERF.cpp:2493
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::iMultiFab > > > land_type_lev
Definition: ERF.H:1030
static double start_time
Definition: ERF.H:1160
void init_zphys(int lev, double elapsed_time)
Definition: ERF_MakeNewArrays.cpp:688
void derive_stress_profiles(amrex::Gpu::HostVector< amrex::Real > &h_avg_tau11, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau12, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau13, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau22, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau23, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau33, amrex::Gpu::HostVector< amrex::Real > &h_avg_hfx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_q1fx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_q2fx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_diss)
Definition: ERF_Write1DProfiles.cpp:488
amrex::Vector< amrex::Vector< amrex::MultiFab > > forecast_state_1
Definition: ERF.H:184
int cf_width
Definition: ERF.H:1011
static int last_plot2d_file_step_1
Definition: ERF.H:1132
amrex::Vector< std::unique_ptr< amrex::iMultiFab > > zflux_imask
Definition: ERF.H:1122
bool m_expand_plotvars_to_unif_rr
Definition: ERF.H:1200
void Construct_ERFFillPatchers(int lev)
Definition: ERF.cpp:2713
amrex::Vector< int > num_boxes_at_level
Definition: ERF.H:911
void compute_native_shoc_tendencies(int lev, amrex::MultiFab *cons, amrex::MultiFab *xvel, amrex::MultiFab *yvel, amrex::MultiFab *zvel, amrex::Real *w_subsid, amrex::MultiFab *tau13, amrex::MultiFab *tau23, amrex::MultiFab *hfx3, amrex::MultiFab *qfx3, amrex::MultiFab *eddyDiffs, amrex::MultiFab *z_phys_nd, const double &dt_advance)
Definition: ERF_ShocCoupling.cpp:6
void init_phys_bcs(bool &rho_read, bool &read_prim_theta)
Definition: ERF_InitBCs.cpp:20
void FillSurfaceStateMultiFabs(const int lev, const std::string &filename, amrex::Vector< amrex::MultiFab > &surface_state)
Definition: ERF_SurfaceDataInterpolation.cpp:19
static amrex::Real sub_cfl
Definition: ERF.H:1175
static void print_error(MPI_Comm, const std::string &msg)
Definition: ERF_ConsoleIO.cpp:43
static int ng_dens_hse
Definition: ERF.H:1413
std::unique_ptr< ReadBndryPlanes > m_r2d
Definition: ERF.H:1469
amrex::Vector< std::unique_ptr< amrex::MultiFab > > ay
Definition: ERF.H:1055
amrex::Vector< amrex::Vector< amrex::MultiFab * > > qmoist
Definition: ERF.H:973
amrex::Vector< std::unique_ptr< amrex::MultiFab > > SFS_q1fx1_lev
Definition: ERF.H:1037
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > zvel_bc_data
Definition: ERF.H:885
void derive_stress_profiles_stag(amrex::Gpu::HostVector< amrex::Real > &h_avg_tau11, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau12, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau13, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau22, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau23, amrex::Gpu::HostVector< amrex::Real > &h_avg_tau33, amrex::Gpu::HostVector< amrex::Real > &h_avg_hfx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_q1fx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_q2fx3, amrex::Gpu::HostVector< amrex::Real > &h_avg_diss)
Definition: ERF_Write1DProfiles_stag.cpp:618
LandSurface lsm
Definition: ERF.H:988
static amrex::Real change_max
Definition: ERF.H:1177
void setRecordSampleLineInfo(int i, int lev, amrex::IntVect &cell, const std::string &filename)
Definition: ERF.H:1704
void setSpongeRefFromSounding(bool restarting)
Set sponge mean profiles from input sounding.
Definition: ERF_InitSponge.cpp:65
amrex::Vector< amrex::MultiFab > avg_ymom
Definition: ERF.H:945
int line_sampling_interval
Definition: ERF.H:1722
amrex::Vector< amrex::Vector< amrex::Real > > h_sinesq_stag_ptrs
Definition: ERF.H:1444
amrex::Vector< std::unique_ptr< amrex::MultiFab > > detJ_cc_new
Definition: ERF.H:1066
void project_initial_velocity(int lev, double time, double dt)
Definition: ERF_PoissonSolve.cpp:31
amrex::Array< std::string, 2 *AMREX_SPACEDIM > domain_bc_type
Definition: ERF.H:1105
amrex::Vector< amrex::Gpu::DeviceVector< amrex::Real > > yvel_bc_data
Definition: ERF.H:884
static void writeBuildInfo(std::ostream &os)
Definition: ERF_WriteJobInfo.cpp:175
amrex::GpuArray< ERF_BC, AMREX_SPACEDIM *2 > phys_bc_type
Definition: ERF.H:1117
std::string check_file
Definition: ERF.H:1222
amrex::Vector< amrex::IntVect > samplepoint
Definition: ERF.H:1738
amrex::Vector< amrex::MultiFab > surface_state_interp
Definition: ERF.H:189
amrex::Vector< amrex::Real > h_havg_qv
Definition: ERF.H:1457
amrex::Vector< amrex::MultiFab > weather_forecast_data_1
Definition: ERF.H:183
void advance_dycore(int level, amrex::Vector< amrex::MultiFab > &state_old, amrex::Vector< amrex::MultiFab > &state_new, amrex::MultiFab &xvel_old, amrex::MultiFab &yvel_old, amrex::MultiFab &zvel_old, amrex::MultiFab &xvel_new, amrex::MultiFab &yvel_new, amrex::MultiFab &zvel_new, amrex::MultiFab &source, amrex::MultiFab &xmom_src, amrex::MultiFab &ymom_src, amrex::MultiFab &zmom_src, amrex::MultiFab &buoyancy, amrex::Geometry fine_geom, double dt, double time)
Definition: ERF_AdvanceDycore.cpp:38
static void print_usage(MPI_Comm, std::ostream &)
Definition: ERF_ConsoleIO.cpp:26
amrex::Vector< amrex::MultiFab > rV_old
Definition: ERF.H:963
amrex::Vector< std::unique_ptr< amrex::MultiFab > > lat_m
Definition: ERF.H:872
void timeStep(int lev, double time, int iteration)
Definition: ERF_TimeStep.cpp:18
amrex::Vector< amrex::Vector< amrex::Vector< amrex::Real > > > h_rayleigh_ptrs
Definition: ERF.H:1439
void make_eb_regular()
static int last_plot3d_file_step_1
Definition: ERF.H:1130
void FillPatchFineLevel(int lev, double time, const amrex::Vector< amrex::MultiFab * > &mfs_vel, const amrex::Vector< amrex::MultiFab * > &mfs_mom, const amrex::MultiFab &old_base_state, const amrex::MultiFab &new_base_state, bool fillset=true, bool cons_only=false)
Definition: ERF_FillPatch.cpp:20
std::unique_ptr< LineSampler > line_sampler
Definition: ERF.H:1726
amrex::Vector< std::unique_ptr< amrex::MultiFab > > Hwave
Definition: ERF.H:1089
amrex::Vector< amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > > Tau_EB
Definition: ERF.H:1042
amrex::Vector< int > istep
Definition: ERF.H:915
const int datprecision
Definition: ERF.H:1149
amrex::Vector< std::unique_ptr< amrex::iMultiFab > > xflux_imask
Definition: ERF.H:1120
void derive_upwp(amrex::Vector< amrex::Real > &h_havg)
static PlotFileType plotfile2d_type_1
Definition: ERF.H:1344
int metgrid_order
Definition: ERF.H:1376
bool finished_wave
Definition: ERF.H:1093
void ReadCheckpointFile()
Definition: ERF_Checkpoint.cpp:520
amrex::Vector< amrex::Vector< amrex::MultiFab > > vars_old
Definition: ERF.H:928
void WriteAtFinalTime()
Definition: ERF.cpp:276
amrex::Vector< std::unique_ptr< ERFPhysBCFunct_base > > physbcs_base
Definition: ERF.H:953
void init_thin_body(int lev, const amrex::BoxArray &ba, const amrex::DistributionMapping &dm)
Definition: ERF_MakeNewLevel.cpp:925
static PlotFileType plotfile3d_type_1
Definition: ERF.H:1342
amrex::Vector< int > m_subvol_int
Definition: ERF.H:1206
AMREX_FORCE_INLINE std::ostream & SamplePointLog(int i)
Definition: ERF.H:1578
std::string MakeFilename_EyeTracker_latlon(int nstep)
Definition: ERF_TrackerOutput.cpp:52
void WriteMultiLevelPlotfileWithTerrain(const std::string &plotfilename, int nlevels, const amrex::Vector< const amrex::MultiFab * > &mf, const amrex::Vector< const amrex::MultiFab * > &mf_nd, const amrex::Vector< std::string > &varnames, const amrex::Vector< amrex::Geometry > &my_geom, double time, const amrex::Vector< int > &level_steps, const amrex::Vector< amrex::IntVect > &my_ref_ratio, const std::string &versionName="HyperCLaw-V1.1", const std::string &levelPrefix="Level_", const std::string &mfPrefix="Cell", const amrex::Vector< std::string > &extra_dirs=amrex::Vector< std::string >()) const
Definition: ERF_Plotfile.cpp:1937
amrex::Array< amrex::Array< amrex::Real, AMREX_SPACEDIM *2 >, AMREX_SPACEDIM+NBCVAR_max > m_bc_neumann_vals
Definition: ERF.H:1111
void PackAtmosphericStates(amrex::Vector< amrex::MultiFab * > &states, double time)
Definition: ERF_Coupling.cpp:169
bool m_driver_atm2ocn_uses_state_contract
Definition: ERF.H:706
amrex::Vector< amrex::Vector< std::unique_ptr< amrex::MultiFab > > > tsk_lev
Definition: ERF.H:1026
void setRecordSamplePointInfo(int i, int lev, amrex::IntVect &cell, const std::string &filename)
Definition: ERF.H:1687
static int column_interval
Definition: ERF.H:1398
static void print_summary(std::ostream &)
int m_plot3d_int_2
Definition: ERF.H:1202
amrex::Vector< std::unique_ptr< amrex::MultiFab > > qheating_rates
Definition: ERF.H:995
void turbPert_amplitude(const int lev)
Definition: ERF_InitTurbPert.cpp:33
amrex::Vector< std::unique_ptr< amrex::MultiFab > > rad_fluxes
Definition: ERF.H:996
void initializeMicrophysics(const int &)
Definition: ERF.cpp:1713
static double bndry_output_planes_start_time
Definition: ERF.H:1408
std::unique_ptr< amrex::MultiFab > wrf_C2H
Definition: ERF.H:1389
bool plot_lsm
Definition: ERF.H:1215
amrex::Vector< amrex::MultiFab > avg_zmom
Definition: ERF.H:946
const amrex::Vector< std::string > cons_names
Definition: ERF.H:1232
void ReadCheckpointFileSurfaceLayer()
Definition: ERF_Checkpoint.cpp:1272
double m_plot2d_per_1
Definition: ERF.H:1211
Definition: ERF_LandSurface.H:18
Definition: ERF_EB.H:13
Definition: ERF_InterpolationUtils.H:16
@ pres
Definition: ERF_Kessler.H:27
@ qv
Definition: ERF_Kessler.H:30
@ xvel
Definition: ERF_IndexDefines.H:176
@ cons
Definition: ERF_IndexDefines.H:175
@ zvel
Definition: ERF_IndexDefines.H:178
@ yvel
Definition: ERF_IndexDefines.H:177
@ p
Definition: ERF_WSM6.H:189
Definition: ERF_ConsoleIO.cpp:12
ArtifactType
Definition: ERF_Provenance.H:34
real(c_double), parameter, private pi
Definition: ERF_module_mp_morr_two_moment.F90:100
integer, private isub
Definition: ERF_module_mp_morr_two_moment.F90:164
real(kind=kind_phys), parameter, private dens
Definition: ERF_module_mp_wsm6.F90:39
AdvType moistscal_horiz_adv_type
Definition: ERF_AdvStruct.H:423
AdvType dycore_vert_adv_type
Definition: ERF_AdvStruct.H:420
AdvType moistscal_vert_adv_type
Definition: ERF_AdvStruct.H:424
AdvType dryscal_horiz_adv_type
Definition: ERF_AdvStruct.H:421
AdvType dycore_horiz_adv_type
Definition: ERF_AdvStruct.H:419
AdvType dryscal_vert_adv_type
Definition: ERF_AdvStruct.H:422
Definition: ERF_InputSoundingData.H:22
Definition: ERF_InputSpongeData.H:19
Definition: ERF_DataStruct.H:141
bool use_num_diff
Definition: ERF_DataStruct.H:1420
AdvChoice advChoice
Definition: ERF_DataStruct.H:1247
static TerrainType terrain_type
Definition: ERF_DataStruct.H:1230
Definition: ERF_TurbPertStruct.H:22
Definition: ERF_Provenance.H:63