1 #ifndef ERF_RADIATION_H
2 #define ERF_RADIATION_H
23 #include <mo_gas_concentrations.h>
25 #include <Kokkos_Core.hpp>
27 #include <AMReX_ParmParse.H>
28 #include <AMReX_FArrayBox.H>
29 #include <AMReX_Geometry.H>
30 #include <AMReX_TableData.H>
31 #include <AMReX_MultiFabUtil.H>
32 #include <AMReX_PlotFileUtil.H>
57 virtual void Init (
const amrex::Geometry& geom,
58 const amrex::BoxArray& ba,
59 amrex::MultiFab* cons_in)
override {};
61 virtual void Run (
int& level,
64 const amrex::Real& dt,
65 const amrex::BoxArray& ba,
66 amrex::Geometry& geom,
67 amrex::MultiFab* cons_in,
68 amrex::MultiFab* lsm_fluxes,
69 amrex::MultiFab* lsm_zenith,
70 amrex::MultiFab* qheating_rates,
71 amrex::MultiFab* z_phys,
73 amrex::MultiFab*
lon)
override
75 set_grids(level, step, time, dt, ba, geom,
76 cons_in, lsm_fluxes, lsm_zenith,
77 qheating_rates, z_phys,
lat,
lon);
86 const amrex::Real& dt,
87 const amrex::BoxArray& ba,
88 amrex::Geometry& geom,
89 amrex::MultiFab* cons_in,
90 amrex::MultiFab* lsm_fluxes,
91 amrex::MultiFab* lsm_zenith,
92 amrex::MultiFab* qheating_rates,
93 amrex::MultiFab* z_phys,
95 amrex::MultiFab*
lon);
97 template<
typename LandSurfaceModelType>
100 if constexpr(std::is_same_v<LandSurfaceModelType, SLM>)
156 Kokkos::deep_copy(
lw_src, 0.0);
197 amrex::Print() <<
"Advancing radiation at level: " <<
m_lev <<
" ...";
201 amrex::Print() <<
"DONE\n";
207 virtual void WriteDataLog (
const amrex::Real &time)
override;
284 const std::vector<std::string>
m_gas_names = {
"H2O",
"CO2",
"O3",
"N2O",
285 "CO" ,
"CH4",
"O2",
"N2" };
287 28.01010, 16.04246, 31.9980, 28.0134};
Kokkos::View< RealT *, RadDefaultDevice > real1d_k
Definition: ERF_Kokkos_Radiation.H:17
Kokkos::View< RealT ***, layout_t, RadDefaultDevice > real3d_k
Definition: ERF_Kokkos_Radiation.H:19
Kokkos::View< RealT **, layout_t, RadDefaultDevice > real2d_k
Definition: ERF_Kokkos_Radiation.H:18
Definition: ERF_RadiationInterface.H:14
Definition: ERF_Radiation.H:45
real3d_k sw_bnd_flux_dn
Definition: ERF_Radiation.H:428
amrex::MultiFab * m_lon
Definition: ERF_Radiation.H:258
real2d_k lw_flux_up
Definition: ERF_Radiation.H:408
real3d_k aero_tau_sw
Definition: ERF_Radiation.H:444
int m_o3_size
Definition: ERF_Radiation.H:300
real3d_k sw_bnd_flux_dir
Definition: ERF_Radiation.H:429
real2d_k sw_clnsky_flux_dn
Definition: ERF_Radiation.H:417
std::string rrtmgp_coeffs_file_sw
Definition: ERF_Radiation.H:277
std::string rrtmgp_coeffs_sw
Definition: ERF_Radiation.H:273
real2d_k d_tint
Definition: ERF_Radiation.H:402
real2d_k lw_clnclrsky_flux_dn
Definition: ERF_Radiation.H:420
int m_rad_freq_in_steps
Definition: ERF_Radiation.H:356
real2d_k lwp
Definition: ERF_Radiation.H:394
real3d_k cld_tau_lw_gpt
Definition: ERF_Radiation.H:455
real1d_k lw_src
Definition: ERF_Radiation.H:379
real2d_k eff_radius_qi
Definition: ERF_Radiation.H:392
void dealloc_buffers()
Definition: ERF_Radiation.cpp:315
real1d_k m_gas_mol_weights
Definition: ERF_Radiation.H:301
amrex::Real m_lon_cons
Definition: ERF_Radiation.H:262
void initialize_impl()
Definition: ERF_Radiation.cpp:833
int m_nswbands
Definition: ERF_Radiation.H:350
bool m_do_aerosol_rad
Definition: ERF_Radiation.H:317
std::string rrtmgp_coeffs_lw
Definition: ERF_Radiation.H:274
bool m_moist
Definition: ERF_Radiation.H:241
std::string rrtmgp_cloud_optics_file_lw
Definition: ERF_Radiation.H:280
amrex::MultiFab * m_cons_in
Definition: ERF_Radiation.H:248
amrex::MultiFab datalog_mf
Definition: ERF_Radiation.H:269
Radiation(const int &lev, SolverChoice &sc)
Definition: ERF_Radiation.cpp:19
real2d_k sw_heating
Definition: ERF_Radiation.H:396
bool m_lsm
Definition: ERF_Radiation.H:245
bool m_do_subcol_sampling
Definition: ERF_Radiation.H:359
real3d_k lw_bnd_flux_dn
Definition: ERF_Radiation.H:434
bool m_rad_write_fluxes
Definition: ERF_Radiation.H:236
void set_lsm_inputs(LandSurfaceModelType *model)
Definition: ERF_Radiation.H:98
real3d_k sw_bnd_flux_up
Definition: ERF_Radiation.H:427
virtual void Run(int &level, int &step, amrex::Real &time, const amrex::Real &dt, const amrex::BoxArray &ba, amrex::Geometry &geom, amrex::MultiFab *cons_in, amrex::MultiFab *lsm_fluxes, amrex::MultiFab *lsm_zenith, amrex::MultiFab *qheating_rates, amrex::MultiFab *z_phys, amrex::MultiFab *lat, amrex::MultiFab *lon) override
Definition: ERF_Radiation.H:61
real2d_k qv_lay
Definition: ERF_Radiation.H:387
std::string rrtmgp_cloud_optics_file_sw
Definition: ERF_Radiation.H:279
void run_impl()
Definition: ERF_Radiation.cpp:844
real2d_k sw_clnclrsky_flux_dn_dir
Definition: ERF_Radiation.H:412
real1d_k sfc_flux_dif_vis
Definition: ERF_Radiation.H:373
int m_step
Definition: ERF_Radiation.H:218
real2d_k lw_clnclrsky_flux_up
Definition: ERF_Radiation.H:419
real3d_k cld_tau_sw_gpt
Definition: ERF_Radiation.H:454
real1d_k lat
Definition: ERF_Radiation.H:375
void populateDatalogMF()
Definition: ERF_Radiation.cpp:638
real2d_k qi_lay
Definition: ERF_Radiation.H:389
real2d_k sw_clrsky_flux_up
Definition: ERF_Radiation.H:413
amrex::Real m_orbital_mvelp
Definition: ERF_Radiation.H:337
real2d_k t_lev
Definition: ERF_Radiation.H:404
amrex::Vector< amrex::Real > m_o3vmr
Definition: ERF_Radiation.H:291
void mf_to_kokkos_buffers()
Definition: ERF_Radiation.cpp:423
real1d_k o3_lay
Definition: ERF_Radiation.H:362
std::string rrtmgp_cloud_optics_sw
Definition: ERF_Radiation.H:275
void finalize_impl()
Definition: ERF_Radiation.cpp:1092
real1d_k sfc_alb_dif_vis
Definition: ERF_Radiation.H:369
int m_nlwgpts
Definition: ERF_Radiation.H:353
real1d_k mu0
Definition: ERF_Radiation.H:366
real2d_k cldfrac_tot
Definition: ERF_Radiation.H:390
real3d_k cld_tau_lw_bnd
Definition: ERF_Radiation.H:451
real1d_k sfc_flux_dir_nir
Definition: ERF_Radiation.H:372
amrex::MultiFab * m_z_phys
Definition: ERF_Radiation.H:254
real2d_k sw_clnclrsky_flux_up
Definition: ERF_Radiation.H:410
real3d_k aero_g_sw
Definition: ERF_Radiation.H:446
real2d_k sw_flux_up
Definition: ERF_Radiation.H:405
std::string rrtmgp_cloud_optics_lw
Definition: ERF_Radiation.H:276
real3d_k sw_bnd_flux_dif
Definition: ERF_Radiation.H:430
real2d_k sfc_alb_dif
Definition: ERF_Radiation.H:438
amrex::Real m_co2vmr
Definition: ERF_Radiation.H:290
bool m_extra_clnsky_diag
Definition: ERF_Radiation.H:320
real2d_k emis_sfc
Definition: ERF_Radiation.H:441
real1d_k sfc_alb_dif_nir
Definition: ERF_Radiation.H:370
amrex::MultiFab * m_lsm_fluxes
Definition: ERF_Radiation.H:265
virtual void Init(const amrex::Geometry &geom, const amrex::BoxArray &ba, amrex::MultiFab *cons_in) override
Definition: ERF_Radiation.H:57
real2d_k lw_clnsky_flux_dn
Definition: ERF_Radiation.H:424
amrex::BoxArray m_ba
Definition: ERF_Radiation.H:230
real2d_k p_lay
Definition: ERF_Radiation.H:383
real2d_k r_lay
Definition: ERF_Radiation.H:382
int m_ncol
Definition: ERF_Radiation.H:310
real2d_k sw_flux_dn_dir
Definition: ERF_Radiation.H:407
const std::vector< amrex::Real > m_mol_weight_gas
Definition: ERF_Radiation.H:286
amrex::Real m_lat_cons
Definition: ERF_Radiation.H:261
amrex::Real m_fixed_total_solar_irradiance
Definition: ERF_Radiation.H:342
std::vector< std::string > gas_names_offset
Definition: ERF_Radiation.H:302
real2d_k sw_clrsky_flux_dn
Definition: ERF_Radiation.H:414
amrex::Real m_dt
Definition: ERF_Radiation.H:224
int m_orbital_mon
Definition: ERF_Radiation.H:327
real3d_k aero_ssa_sw
Definition: ERF_Radiation.H:445
real2d_k sw_clnsky_flux_dn_dir
Definition: ERF_Radiation.H:418
amrex::Real m_o2vmr
Definition: ERF_Radiation.H:295
amrex::MultiFab * m_qheating_rates
Definition: ERF_Radiation.H:251
real2d_k qc_lay
Definition: ERF_Radiation.H:388
virtual void WriteDataLog(const amrex::Real &time) override
Definition: ERF_Radiation.cpp:698
real2d_k lw_clrsky_flux_up
Definition: ERF_Radiation.H:421
amrex::Real m_n2ovmr
Definition: ERF_Radiation.H:292
std::string rrtmgp_file_path
Definition: ERF_Radiation.H:272
real2d_k sw_flux_dn
Definition: ERF_Radiation.H:406
amrex::MultiFab * m_lsm_zenith
Definition: ERF_Radiation.H:266
int m_nlwbands
Definition: ERF_Radiation.H:351
real2d_k tmp2d
Definition: ERF_Radiation.H:393
real2d_k lw_clrsky_flux_dn
Definition: ERF_Radiation.H:422
GasConcsK< amrex::Real, layout_t, RadDefaultDevice > m_gas_concs
Definition: ERF_Radiation.H:304
amrex::Real m_orbital_eccen
Definition: ERF_Radiation.H:335
amrex::Real m_covmr
Definition: ERF_Radiation.H:293
int m_orbital_sec
Definition: ERF_Radiation.H:329
real2d_k z_del
Definition: ERF_Radiation.H:385
real2d_k p_del
Definition: ERF_Radiation.H:386
real3d_k aero_tau_lw
Definition: ERF_Radiation.H:447
real2d_k lw_flux_dn
Definition: ERF_Radiation.H:409
void kokkos_buffers_to_mf()
Definition: ERF_Radiation.cpp:542
real2d_k sw_clrsky_flux_dn_dir
Definition: ERF_Radiation.H:415
void write_rrtmgp_fluxes()
Definition: ERF_Radiation.cpp:604
std::string rrtmgp_coeffs_file_lw
Definition: ERF_Radiation.H:278
int m_lev
Definition: ERF_Radiation.H:215
real1d_k sfc_flux_dif_nir
Definition: ERF_Radiation.H:374
int m_ngas
Definition: ERF_Radiation.H:283
real2d_k lw_heating
Definition: ERF_Radiation.H:397
real1d_k sfc_alb_dir_nir
Definition: ERF_Radiation.H:368
real2d_k lw_clnsky_flux_up
Definition: ERF_Radiation.H:423
void set_grids(int &level, int &step, amrex::Real &time, const amrex::Real &dt, const amrex::BoxArray &ba, amrex::Geometry &geom, amrex::MultiFab *cons_in, amrex::MultiFab *lsm_fluxes, amrex::MultiFab *lsm_zenith, amrex::MultiFab *qheating_rates, amrex::MultiFab *z_phys, amrex::MultiFab *lat, amrex::MultiFab *lon)
Definition: ERF_Radiation.cpp:115
amrex::Real m_ch4vmr
Definition: ERF_Radiation.H:294
bool m_update_rad
Definition: ERF_Radiation.H:233
real2d_k sfc_alb_dir
Definition: ERF_Radiation.H:437
bool m_ice
Definition: ERF_Radiation.H:242
amrex::Real m_fixed_solar_zenith_angle
Definition: ERF_Radiation.H:346
real1d_k lon
Definition: ERF_Radiation.H:376
int m_orbital_day
Definition: ERF_Radiation.H:328
real1d_k sfc_emis
Definition: ERF_Radiation.H:377
amrex::Real m_orbital_obliq
Definition: ERF_Radiation.H:336
amrex::Real m_n2vmr
Definition: ERF_Radiation.H:296
real3d_k cld_tau_sw_bnd
Definition: ERF_Radiation.H:450
bool m_fixed_orbital_year
Definition: ERF_Radiation.H:334
real2d_k sw_clnsky_flux_up
Definition: ERF_Radiation.H:416
int m_nlay
Definition: ERF_Radiation.H:311
real3d_k lw_bnd_flux_up
Definition: ERF_Radiation.H:433
int m_nswgpts
Definition: ERF_Radiation.H:352
bool m_first_step
Definition: ERF_Radiation.H:238
real1d_k sfc_alb_dir_vis
Definition: ERF_Radiation.H:367
const std::vector< std::string > m_gas_names
Definition: ERF_Radiation.H:284
real2d_k sw_clrsky_heating
Definition: ERF_Radiation.H:398
bool m_extra_clnclrsky_diag
Definition: ERF_Radiation.H:321
real2d_k t_lay
Definition: ERF_Radiation.H:384
void rad_run_impl()
Definition: ERF_Radiation.H:194
real1d_k sfc_flux_dir_vis
Definition: ERF_Radiation.H:371
real2d_k iwp
Definition: ERF_Radiation.H:395
amrex::Geometry m_geom
Definition: ERF_Radiation.H:227
amrex::MultiFab * m_lat
Definition: ERF_Radiation.H:257
real2d_k p_lev
Definition: ERF_Radiation.H:403
real2d_k lw_clrsky_heating
Definition: ERF_Radiation.H:399
real2d_k eff_radius_qc
Definition: ERF_Radiation.H:391
real1d_k cosine_zenith
Definition: ERF_Radiation.H:365
amrex::Vector< int > m_col_offsets
Definition: ERF_Radiation.H:314
real1d_k t_sfc
Definition: ERF_Radiation.H:378
void alloc_buffers()
Definition: ERF_Radiation.cpp:193
int m_orbital_year
Definition: ERF_Radiation.H:326
real2d_k sw_clnclrsky_flux_dn
Definition: ERF_Radiation.H:411
amrex::Real m_time
Definition: ERF_Radiation.H:221
Definition: ERF_DataStruct.H:123