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>
62 Init (
const amrex::Geometry& geom,
63 const amrex::BoxArray& ba,
64 amrex::MultiFab* cons_in)
override
67 int klo = geom.Domain().smallEnd(2);
68 int khi = geom.Domain().bigEnd(2);
71 m_nlay = geom.Domain().length(2);
76 for (amrex::MFIter mfi(*cons_in); mfi.isValid(); ++mfi) {
77 const amrex::Box& vbx = mfi.validbox();
78 AMREX_ALWAYS_ASSERT_WITH_MESSAGE((klo == vbx.smallEnd(2)) &&
79 (khi == vbx.bigEnd(2)),
80 "Vertical decomposition with radiation is not allowed.");
81 int nx = vbx.length(0);
82 int ny = vbx.length(1);
94 const amrex::BoxArray& ba,
95 amrex::Geometry& geom,
96 amrex::MultiFab* cons_in,
97 amrex::MultiFab* lsm_fluxes,
98 amrex::MultiFab* lsm_zenith,
99 amrex::Vector<amrex::MultiFab*>& lsm_input_ptrs,
100 amrex::Vector<amrex::MultiFab*>& lsm_output_ptrs,
101 amrex::MultiFab* qheating_rates,
102 amrex::MultiFab* z_phys,
103 amrex::MultiFab*
lat,
104 amrex::MultiFab*
lon)
override
106 set_grids(level, step, time, dt, ba, geom,
107 cons_in, lsm_fluxes, lsm_zenith,
108 lsm_input_ptrs, qheating_rates,
119 const amrex::BoxArray& ba,
120 amrex::Geometry& geom,
121 amrex::MultiFab* cons_in,
122 amrex::MultiFab* lsm_fluxes,
123 amrex::MultiFab* lsm_zenith,
124 amrex::Vector<amrex::MultiFab*>& lsm_input_ptrs,
125 amrex::MultiFab* qheating_rates,
126 amrex::MultiFab* z_phys,
127 amrex::MultiFab*
lat,
128 amrex::MultiFab*
lon);
160 finalize_impl (amrex::Vector<amrex::MultiFab*>& lsm_output_ptrs);
167 amrex::Print() <<
"Advancing radiation at level: " <<
m_lev <<
" ...";
171 amrex::Print() <<
"DONE\n";
177 amrex::Vector<std::string>
185 amrex::Vector<std::string>
240 "sfc_alb_dir_vis",
"sfc_alb_dir_nir",
241 "sfc_alb_dif_vis",
"sfc_alb_dif_nir"};
286 const std::vector<std::string>
m_gas_names = {
"H2O",
"CO2",
"O3",
"N2O",
287 "CO" ,
"CH4",
"O2",
"N2" };
289 28.01010, 16.04246, 31.9980, 28.0134};
306 GasConcsK<amrex::Real, layout_t, KokkosDefaultDevice>
m_gas_concs;
Kokkos::View< RealT *, KokkosDefaultDevice > real1d_k
Definition: ERF_Kokkos.H:17
Kokkos::View< RealT ***, layout_t, KokkosDefaultDevice > real3d_k
Definition: ERF_Kokkos.H:19
Kokkos::View< RealT **, layout_t, KokkosDefaultDevice > real2d_k
Definition: ERF_Kokkos.H:18
amrex::Real Real
Definition: ERF_ShocInterface.H:16
Definition: ERF_RadiationInterface.H:14
Definition: ERF_Radiation.H:45
real3d_k sw_bnd_flux_dn
Definition: ERF_Radiation.H:430
amrex::MultiFab * m_lon
Definition: ERF_Radiation.H:260
real2d_k lw_flux_up
Definition: ERF_Radiation.H:410
virtual amrex::Vector< std::string > get_lsm_output_varnames() override
Definition: ERF_Radiation.H:186
real3d_k aero_tau_sw
Definition: ERF_Radiation.H:446
int m_o3_size
Definition: ERF_Radiation.H:302
real3d_k sw_bnd_flux_dir
Definition: ERF_Radiation.H:431
real2d_k sw_clnsky_flux_dn
Definition: ERF_Radiation.H:419
std::string rrtmgp_coeffs_file_sw
Definition: ERF_Radiation.H:279
std::string rrtmgp_coeffs_sw
Definition: ERF_Radiation.H:275
real2d_k d_tint
Definition: ERF_Radiation.H:404
real2d_k lw_clnclrsky_flux_dn
Definition: ERF_Radiation.H:422
int m_rad_freq_in_steps
Definition: ERF_Radiation.H:358
real2d_k lwp
Definition: ERF_Radiation.H:396
real3d_k cld_tau_lw_gpt
Definition: ERF_Radiation.H:457
real1d_k lw_src
Definition: ERF_Radiation.H:381
real2d_k eff_radius_qi
Definition: ERF_Radiation.H:394
void dealloc_buffers()
Definition: ERF_Radiation.cpp:310
real1d_k m_gas_mol_weights
Definition: ERF_Radiation.H:303
amrex::Real m_lon_cons
Definition: ERF_Radiation.H:264
void initialize_impl()
Definition: ERF_Radiation.cpp:951
int m_nswbands
Definition: ERF_Radiation.H:352
bool m_do_aerosol_rad
Definition: ERF_Radiation.H:319
std::string rrtmgp_coeffs_lw
Definition: ERF_Radiation.H:276
bool m_moist
Definition: ERF_Radiation.H:232
std::string rrtmgp_cloud_optics_file_lw
Definition: ERF_Radiation.H:282
amrex::MultiFab * m_cons_in
Definition: ERF_Radiation.H:250
amrex::MultiFab datalog_mf
Definition: ERF_Radiation.H:271
Radiation(const int &lev, SolverChoice &sc)
Definition: ERF_Radiation.cpp:19
real2d_k sw_heating
Definition: ERF_Radiation.H:398
bool m_lsm
Definition: ERF_Radiation.H:236
bool m_do_subcol_sampling
Definition: ERF_Radiation.H:361
real3d_k lw_bnd_flux_dn
Definition: ERF_Radiation.H:436
bool m_rad_write_fluxes
Definition: ERF_Radiation.H:227
real3d_k sw_bnd_flux_up
Definition: ERF_Radiation.H:429
real2d_k qv_lay
Definition: ERF_Radiation.H:389
std::string rrtmgp_cloud_optics_file_sw
Definition: ERF_Radiation.H:281
amrex::Vector< std::string > m_lsm_output_names
Definition: ERF_Radiation.H:244
void run_impl()
Definition: ERF_Radiation.cpp:962
real2d_k sw_clnclrsky_flux_dn_dir
Definition: ERF_Radiation.H:414
real1d_k sfc_flux_dif_vis
Definition: ERF_Radiation.H:375
int m_step
Definition: ERF_Radiation.H:209
real2d_k lw_clnclrsky_flux_up
Definition: ERF_Radiation.H:421
real3d_k cld_tau_sw_gpt
Definition: ERF_Radiation.H:456
real1d_k lat
Definition: ERF_Radiation.H:377
void populateDatalogMF()
Definition: ERF_Radiation.cpp:727
real2d_k qi_lay
Definition: ERF_Radiation.H:391
real2d_k sw_clrsky_flux_up
Definition: ERF_Radiation.H:415
amrex::Real m_orbital_mvelp
Definition: ERF_Radiation.H:339
real2d_k t_lev
Definition: ERF_Radiation.H:406
amrex::Vector< amrex::Real > m_o3vmr
Definition: ERF_Radiation.H:293
real1d_k o3_lay
Definition: ERF_Radiation.H:364
std::string rrtmgp_cloud_optics_sw
Definition: ERF_Radiation.H:277
real1d_k sfc_alb_dif_vis
Definition: ERF_Radiation.H:371
int m_nlwgpts
Definition: ERF_Radiation.H:355
real1d_k mu0
Definition: ERF_Radiation.H:368
real2d_k cldfrac_tot
Definition: ERF_Radiation.H:392
real3d_k cld_tau_lw_bnd
Definition: ERF_Radiation.H:453
real1d_k sfc_flux_dir_nir
Definition: ERF_Radiation.H:374
amrex::MultiFab * m_z_phys
Definition: ERF_Radiation.H:256
real2d_k sw_clnclrsky_flux_up
Definition: ERF_Radiation.H:412
real3d_k aero_g_sw
Definition: ERF_Radiation.H:448
real2d_k sw_flux_up
Definition: ERF_Radiation.H:407
std::string rrtmgp_cloud_optics_lw
Definition: ERF_Radiation.H:278
real3d_k sw_bnd_flux_dif
Definition: ERF_Radiation.H:432
real2d_k sfc_alb_dif
Definition: ERF_Radiation.H:440
amrex::Real m_co2vmr
Definition: ERF_Radiation.H:292
bool m_extra_clnsky_diag
Definition: ERF_Radiation.H:322
real2d_k emis_sfc
Definition: ERF_Radiation.H:443
real1d_k sfc_alb_dif_nir
Definition: ERF_Radiation.H:372
amrex::MultiFab * m_lsm_fluxes
Definition: ERF_Radiation.H:267
virtual void Init(const amrex::Geometry &geom, const amrex::BoxArray &ba, amrex::MultiFab *cons_in) override
Definition: ERF_Radiation.H:62
real2d_k lw_clnsky_flux_dn
Definition: ERF_Radiation.H:426
amrex::BoxArray m_ba
Definition: ERF_Radiation.H:221
real2d_k p_lay
Definition: ERF_Radiation.H:385
real2d_k r_lay
Definition: ERF_Radiation.H:384
int m_ncol
Definition: ERF_Radiation.H:312
real2d_k sw_flux_dn_dir
Definition: ERF_Radiation.H:409
const std::vector< amrex::Real > m_mol_weight_gas
Definition: ERF_Radiation.H:288
amrex::Real m_lat_cons
Definition: ERF_Radiation.H:263
amrex::Real m_fixed_total_solar_irradiance
Definition: ERF_Radiation.H:344
void kokkos_buffers_to_mf(amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs)
Definition: ERF_Radiation.cpp:597
std::vector< std::string > gas_names_offset
Definition: ERF_Radiation.H:304
real2d_k sw_clrsky_flux_dn
Definition: ERF_Radiation.H:416
amrex::Real m_dt
Definition: ERF_Radiation.H:215
virtual amrex::Vector< std::string > get_lsm_input_varnames() override
Definition: ERF_Radiation.H:178
int m_orbital_mon
Definition: ERF_Radiation.H:329
real3d_k aero_ssa_sw
Definition: ERF_Radiation.H:447
real2d_k sw_clnsky_flux_dn_dir
Definition: ERF_Radiation.H:420
amrex::Real m_o2vmr
Definition: ERF_Radiation.H:297
void rad_run_impl(amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs)
Definition: ERF_Radiation.H:164
GasConcsK< amrex::Real, layout_t, KokkosDefaultDevice > m_gas_concs
Definition: ERF_Radiation.H:306
amrex::MultiFab * m_qheating_rates
Definition: ERF_Radiation.H:253
real2d_k qc_lay
Definition: ERF_Radiation.H:390
virtual void WriteDataLog(const amrex::Real &time) override
Definition: ERF_Radiation.cpp:816
real2d_k lw_clrsky_flux_up
Definition: ERF_Radiation.H:423
amrex::Real m_n2ovmr
Definition: ERF_Radiation.H:294
amrex::Real m_rad_t_sfc
Definition: ERF_Radiation.H:247
std::string rrtmgp_file_path
Definition: ERF_Radiation.H:274
real2d_k sw_flux_dn
Definition: ERF_Radiation.H:408
amrex::MultiFab * m_lsm_zenith
Definition: ERF_Radiation.H:268
int m_nlwbands
Definition: ERF_Radiation.H:353
real2d_k tmp2d
Definition: ERF_Radiation.H:395
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::Vector< amrex::MultiFab * > &lsm_input_ptrs, amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs, amrex::MultiFab *qheating_rates, amrex::MultiFab *z_phys, amrex::MultiFab *lat, amrex::MultiFab *lon) override
Definition: ERF_Radiation.H:90
real2d_k lw_clrsky_flux_dn
Definition: ERF_Radiation.H:424
amrex::Real m_orbital_eccen
Definition: ERF_Radiation.H:337
void finalize_impl(amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs)
Definition: ERF_Radiation.cpp:1219
amrex::Real m_covmr
Definition: ERF_Radiation.H:295
int m_orbital_sec
Definition: ERF_Radiation.H:331
real2d_k z_del
Definition: ERF_Radiation.H:387
real2d_k p_del
Definition: ERF_Radiation.H:388
real3d_k aero_tau_lw
Definition: ERF_Radiation.H:449
real2d_k lw_flux_dn
Definition: ERF_Radiation.H:411
~Radiation()
Definition: ERF_Radiation.H:55
void mf_to_kokkos_buffers(amrex::Vector< amrex::MultiFab * > &lsm_input_ptrs)
Definition: ERF_Radiation.cpp:418
real2d_k sw_clrsky_flux_dn_dir
Definition: ERF_Radiation.H:417
void write_rrtmgp_fluxes()
Definition: ERF_Radiation.cpp:686
std::string rrtmgp_coeffs_file_lw
Definition: ERF_Radiation.H:280
int m_lev
Definition: ERF_Radiation.H:206
real1d_k sfc_flux_dif_nir
Definition: ERF_Radiation.H:376
int m_ngas
Definition: ERF_Radiation.H:285
real2d_k lw_heating
Definition: ERF_Radiation.H:399
real1d_k sfc_alb_dir_nir
Definition: ERF_Radiation.H:370
real2d_k lw_clnsky_flux_up
Definition: ERF_Radiation.H:425
amrex::Real m_ch4vmr
Definition: ERF_Radiation.H:296
bool m_update_rad
Definition: ERF_Radiation.H:224
real2d_k sfc_alb_dir
Definition: ERF_Radiation.H:439
bool m_ice
Definition: ERF_Radiation.H:233
amrex::Real m_fixed_solar_zenith_angle
Definition: ERF_Radiation.H:348
real1d_k lon
Definition: ERF_Radiation.H:378
int m_orbital_day
Definition: ERF_Radiation.H:330
real1d_k sfc_emis
Definition: ERF_Radiation.H:379
amrex::Real m_orbital_obliq
Definition: ERF_Radiation.H:338
amrex::Real m_n2vmr
Definition: ERF_Radiation.H:298
real3d_k cld_tau_sw_bnd
Definition: ERF_Radiation.H:452
bool m_fixed_orbital_year
Definition: ERF_Radiation.H:336
real2d_k sw_clnsky_flux_up
Definition: ERF_Radiation.H:418
int m_nlay
Definition: ERF_Radiation.H:313
real3d_k lw_bnd_flux_up
Definition: ERF_Radiation.H:435
int m_nswgpts
Definition: ERF_Radiation.H:354
bool m_first_step
Definition: ERF_Radiation.H:229
real1d_k sfc_alb_dir_vis
Definition: ERF_Radiation.H:369
const std::vector< std::string > m_gas_names
Definition: ERF_Radiation.H:286
real2d_k sw_clrsky_heating
Definition: ERF_Radiation.H:400
bool m_extra_clnclrsky_diag
Definition: ERF_Radiation.H:323
real2d_k t_lay
Definition: ERF_Radiation.H:386
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::Vector< amrex::MultiFab * > &lsm_input_ptrs, amrex::MultiFab *qheating_rates, amrex::MultiFab *z_phys, amrex::MultiFab *lat, amrex::MultiFab *lon)
Definition: ERF_Radiation.cpp:117
real1d_k sfc_flux_dir_vis
Definition: ERF_Radiation.H:373
real2d_k iwp
Definition: ERF_Radiation.H:397
amrex::Geometry m_geom
Definition: ERF_Radiation.H:218
amrex::MultiFab * m_lat
Definition: ERF_Radiation.H:259
real2d_k p_lev
Definition: ERF_Radiation.H:405
real2d_k lw_clrsky_heating
Definition: ERF_Radiation.H:401
real2d_k eff_radius_qc
Definition: ERF_Radiation.H:393
real1d_k cosine_zenith
Definition: ERF_Radiation.H:367
amrex::Vector< int > m_col_offsets
Definition: ERF_Radiation.H:316
real1d_k t_sfc
Definition: ERF_Radiation.H:380
void alloc_buffers()
Definition: ERF_Radiation.cpp:184
int m_orbital_year
Definition: ERF_Radiation.H:328
real2d_k sw_clnclrsky_flux_dn
Definition: ERF_Radiation.H:413
amrex::Vector< std::string > m_lsm_input_names
Definition: ERF_Radiation.H:239
amrex::Real m_time
Definition: ERF_Radiation.H:212
Definition: ERF_DataStruct.H:123