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() <<
"Radiation advancing level " <<
m_lev <<
" at (YY-MM-DD SS) " <<
m_orbital_year <<
'-'
172 amrex::Print() <<
"DONE\n";
178 amrex::Vector<std::string>
186 amrex::Vector<std::string>
241 "sfc_alb_dir_vis",
"sfc_alb_dir_nir",
242 "sfc_alb_dif_vis",
"sfc_alb_dif_nir"};
287 const std::vector<std::string>
m_gas_names = {
"H2O",
"CO2",
"O3",
"N2O",
288 "CO" ,
"CH4",
"O2",
"N2" };
290 28.01010, 16.04246, 31.9980, 28.0134};
307 GasConcsK<amrex::Real, layout_t, KokkosDefaultDevice>
m_gas_concs;
Kokkos::View< RealT *, KokkosDefaultDevice > real1d_k
Definition: ERF_Kokkos.H:18
Kokkos::View< RealT ***, layout_t, KokkosDefaultDevice > real3d_k
Definition: ERF_Kokkos.H:20
Kokkos::View< RealT **, layout_t, KokkosDefaultDevice > real2d_k
Definition: ERF_Kokkos.H:19
amrex::Real Real
Definition: ERF_ShocInterface.H:19
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:261
real2d_k lw_flux_up
Definition: ERF_Radiation.H:408
virtual amrex::Vector< std::string > get_lsm_output_varnames() override
Definition: ERF_Radiation.H:187
real3d_k aero_tau_sw
Definition: ERF_Radiation.H:444
int m_o3_size
Definition: ERF_Radiation.H:303
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:280
std::string rrtmgp_coeffs_sw
Definition: ERF_Radiation.H:276
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:358
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:380
real2d_k eff_radius_qi
Definition: ERF_Radiation.H:393
void dealloc_buffers()
Definition: ERF_Radiation.cpp:325
real1d_k m_gas_mol_weights
Definition: ERF_Radiation.H:304
amrex::Real m_lon_cons
Definition: ERF_Radiation.H:265
void initialize_impl()
Definition: ERF_Radiation.cpp:975
int m_nswbands
Definition: ERF_Radiation.H:352
bool m_do_aerosol_rad
Definition: ERF_Radiation.H:320
std::string rrtmgp_coeffs_lw
Definition: ERF_Radiation.H:277
bool m_moist
Definition: ERF_Radiation.H:233
std::string rrtmgp_cloud_optics_file_lw
Definition: ERF_Radiation.H:283
amrex::MultiFab * m_cons_in
Definition: ERF_Radiation.H:251
amrex::MultiFab datalog_mf
Definition: ERF_Radiation.H:272
Radiation(const int &lev, SolverChoice &sc)
Definition: ERF_Radiation.cpp:20
real2d_k sw_heating
Definition: ERF_Radiation.H:396
bool m_lsm
Definition: ERF_Radiation.H:237
bool m_do_subcol_sampling
Definition: ERF_Radiation.H:361
real3d_k lw_bnd_flux_dn
Definition: ERF_Radiation.H:434
bool m_rad_write_fluxes
Definition: ERF_Radiation.H:228
real3d_k sw_bnd_flux_up
Definition: ERF_Radiation.H:427
real2d_k qv_lay
Definition: ERF_Radiation.H:388
std::string rrtmgp_cloud_optics_file_sw
Definition: ERF_Radiation.H:282
amrex::Vector< std::string > m_lsm_output_names
Definition: ERF_Radiation.H:245
void run_impl()
Definition: ERF_Radiation.cpp:986
real2d_k sw_clnclrsky_flux_dn_dir
Definition: ERF_Radiation.H:412
real1d_k sfc_flux_dif_vis
Definition: ERF_Radiation.H:374
int m_step
Definition: ERF_Radiation.H:210
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:376
void populateDatalogMF()
Definition: ERF_Radiation.cpp:751
real2d_k qi_lay
Definition: ERF_Radiation.H:390
real2d_k sw_clrsky_flux_up
Definition: ERF_Radiation.H:413
amrex::Real m_orbital_mvelp
Definition: ERF_Radiation.H:340
real2d_k t_lev
Definition: ERF_Radiation.H:404
amrex::Vector< amrex::Real > m_o3vmr
Definition: ERF_Radiation.H:294
real1d_k o3_lay
Definition: ERF_Radiation.H:364
std::string rrtmgp_cloud_optics_sw
Definition: ERF_Radiation.H:278
real1d_k sfc_alb_dif_vis
Definition: ERF_Radiation.H:370
int m_nlwgpts
Definition: ERF_Radiation.H:355
real1d_k mu0
Definition: ERF_Radiation.H:367
real2d_k cldfrac_tot
Definition: ERF_Radiation.H:391
real3d_k cld_tau_lw_bnd
Definition: ERF_Radiation.H:451
real1d_k sfc_flux_dir_nir
Definition: ERF_Radiation.H:373
amrex::MultiFab * m_z_phys
Definition: ERF_Radiation.H:257
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:279
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:293
bool m_extra_clnsky_diag
Definition: ERF_Radiation.H:323
real2d_k emis_sfc
Definition: ERF_Radiation.H:441
real1d_k sfc_alb_dif_nir
Definition: ERF_Radiation.H:371
amrex::MultiFab * m_lsm_fluxes
Definition: ERF_Radiation.H:268
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:424
amrex::BoxArray m_ba
Definition: ERF_Radiation.H:222
real2d_k p_lay
Definition: ERF_Radiation.H:384
real2d_k r_lay
Definition: ERF_Radiation.H:383
int m_ncol
Definition: ERF_Radiation.H:313
real2d_k sw_flux_dn_dir
Definition: ERF_Radiation.H:407
const std::vector< amrex::Real > m_mol_weight_gas
Definition: ERF_Radiation.H:289
amrex::Real m_lat_cons
Definition: ERF_Radiation.H:264
amrex::Real m_fixed_total_solar_irradiance
Definition: ERF_Radiation.H:345
void kokkos_buffers_to_mf(amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs)
Definition: ERF_Radiation.cpp:610
std::vector< std::string > gas_names_offset
Definition: ERF_Radiation.H:305
real2d_k sw_clrsky_flux_dn
Definition: ERF_Radiation.H:414
amrex::Real m_dt
Definition: ERF_Radiation.H:216
virtual amrex::Vector< std::string > get_lsm_input_varnames() override
Definition: ERF_Radiation.H:179
int m_orbital_mon
Definition: ERF_Radiation.H:330
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:298
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:307
amrex::MultiFab * m_qheating_rates
Definition: ERF_Radiation.H:254
real2d_k qc_lay
Definition: ERF_Radiation.H:389
virtual void WriteDataLog(const amrex::Real &time) override
Definition: ERF_Radiation.cpp:840
real2d_k lw_clrsky_flux_up
Definition: ERF_Radiation.H:421
amrex::Real m_n2ovmr
Definition: ERF_Radiation.H:295
amrex::Real m_rad_t_sfc
Definition: ERF_Radiation.H:248
std::string rrtmgp_file_path
Definition: ERF_Radiation.H:275
real2d_k sw_flux_dn
Definition: ERF_Radiation.H:406
amrex::MultiFab * m_lsm_zenith
Definition: ERF_Radiation.H:269
int m_nlwbands
Definition: ERF_Radiation.H:353
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:422
amrex::Real m_orbital_eccen
Definition: ERF_Radiation.H:338
void finalize_impl(amrex::Vector< amrex::MultiFab * > &lsm_output_ptrs)
Definition: ERF_Radiation.cpp:1242
amrex::Real m_covmr
Definition: ERF_Radiation.H:296
int m_orbital_sec
Definition: ERF_Radiation.H:332
real2d_k z_del
Definition: ERF_Radiation.H:386
real2d_k p_del
Definition: ERF_Radiation.H:387
real3d_k aero_tau_lw
Definition: ERF_Radiation.H:447
real2d_k lw_flux_dn
Definition: ERF_Radiation.H:409
~Radiation()
Definition: ERF_Radiation.H:55
void mf_to_kokkos_buffers(amrex::Vector< amrex::MultiFab * > &lsm_input_ptrs)
Definition: ERF_Radiation.cpp:431
real2d_k sw_clrsky_flux_dn_dir
Definition: ERF_Radiation.H:415
void write_rrtmgp_fluxes()
Definition: ERF_Radiation.cpp:710
std::string rrtmgp_coeffs_file_lw
Definition: ERF_Radiation.H:281
int m_lev
Definition: ERF_Radiation.H:207
real1d_k sfc_flux_dif_nir
Definition: ERF_Radiation.H:375
int m_ngas
Definition: ERF_Radiation.H:286
real2d_k lw_heating
Definition: ERF_Radiation.H:397
real1d_k sfc_alb_dir_nir
Definition: ERF_Radiation.H:369
real2d_k lw_clnsky_flux_up
Definition: ERF_Radiation.H:423
amrex::Real m_ch4vmr
Definition: ERF_Radiation.H:297
bool m_update_rad
Definition: ERF_Radiation.H:225
real2d_k sfc_alb_dir
Definition: ERF_Radiation.H:437
bool m_ice
Definition: ERF_Radiation.H:234
amrex::Real m_fixed_solar_zenith_angle
Definition: ERF_Radiation.H:349
real1d_k lon
Definition: ERF_Radiation.H:377
int m_orbital_day
Definition: ERF_Radiation.H:331
real1d_k sfc_emis
Definition: ERF_Radiation.H:378
amrex::Real m_orbital_obliq
Definition: ERF_Radiation.H:339
amrex::Real m_n2vmr
Definition: ERF_Radiation.H:299
real3d_k cld_tau_sw_bnd
Definition: ERF_Radiation.H:450
bool m_fixed_orbital_year
Definition: ERF_Radiation.H:337
real2d_k sw_clnsky_flux_up
Definition: ERF_Radiation.H:416
int m_nlay
Definition: ERF_Radiation.H:314
real3d_k lw_bnd_flux_up
Definition: ERF_Radiation.H:433
int m_nswgpts
Definition: ERF_Radiation.H:354
bool m_first_step
Definition: ERF_Radiation.H:230
real1d_k sfc_alb_dir_vis
Definition: ERF_Radiation.H:368
const std::vector< std::string > m_gas_names
Definition: ERF_Radiation.H:287
real2d_k sw_clrsky_heating
Definition: ERF_Radiation.H:398
bool m_extra_clnclrsky_diag
Definition: ERF_Radiation.H:324
real2d_k t_lay
Definition: ERF_Radiation.H:385
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:134
real1d_k sfc_flux_dir_vis
Definition: ERF_Radiation.H:372
real2d_k iwp
Definition: ERF_Radiation.H:395
amrex::Geometry m_geom
Definition: ERF_Radiation.H:219
amrex::MultiFab * m_lat
Definition: ERF_Radiation.H:260
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:392
amrex::Vector< int > m_col_offsets
Definition: ERF_Radiation.H:317
real1d_k t_sfc
Definition: ERF_Radiation.H:379
void alloc_buffers()
Definition: ERF_Radiation.cpp:201
int m_orbital_year
Definition: ERF_Radiation.H:329
real2d_k sw_clnclrsky_flux_dn
Definition: ERF_Radiation.H:411
amrex::Vector< std::string > m_lsm_input_names
Definition: ERF_Radiation.H:240
amrex::Real m_time
Definition: ERF_Radiation.H:213
Definition: ERF_DataStruct.H:123