Radiation Class Reference

#include <ERF_Radiation.H>

Collaboration diagram for Radiation:

Public Member Functions
	Radiation (const int &lev, SolverChoice &sc)
	~Radiation ()=default
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)
void	alloc_buffers ()
void	dealloc_buffers ()
void	mf_to_yakl_buffers ()
void	yakl_buffers_to_mf ()
void	write_rrtmgp_fluxes ()
void	initialize_impl ()
void	run_impl ()
void	finalize_impl ()
void	rad_run_impl ()

Private Attributes
int	m_lev
int	m_step
amrex::Real	m_time
amrex::Real	m_dt
amrex::Geometry	m_geom
amrex::BoxArray	m_ba
bool	m_update_rad = false
bool	m_rad_write_fluxes = false
bool	m_moist = false
bool	m_ice = false
amrex::MultiFab *	m_cons_in = nullptr
amrex::MultiFab *	m_qheating_rates = nullptr
amrex::MultiFab *	m_z_phys = nullptr
amrex::MultiFab *	m_lat = nullptr
amrex::MultiFab *	m_lon = nullptr
amrex::Real	m_lat_cons = 39.809860
amrex::Real	m_lon_cons = -98.555183
amrex::MultiFab *	m_lsm_fluxes = nullptr
amrex::MultiFab *	m_lsm_zenith = nullptr
std::string	rrtmgp_file_path = "."
std::string	rrtmgp_coeffs_sw = "rrtmgp-data-sw-g224-2018-12-04.nc"
std::string	rrtmgp_coeffs_lw = "rrtmgp-data-lw-g224-2018-12-04.nc"
std::string	rrtmgp_cloud_optics_sw = "rrtmgp-cloud-optics-coeffs-sw.nc"
std::string	rrtmgp_cloud_optics_lw = "rrtmgp-cloud-optics-coeffs-lw.nc"
std::string	rrtmgp_coeffs_file_sw
std::string	rrtmgp_coeffs_file_lw
std::string	rrtmgp_cloud_optics_file_sw
std::string	rrtmgp_cloud_optics_file_lw
int	m_ngas = 8
const std::vector< std::string >	m_gas_names
const std::vector< amrex::Real >	m_mol_weight_gas
amrex::Real	m_co2vmr = 388.717e-6
amrex::Vector< amrex::Real >	m_o3vmr
amrex::Real	m_n2ovmr = 323.141e-9
amrex::Real	m_covmr = 1.0e-7
amrex::Real	m_ch4vmr = 1807.851e-9
amrex::Real	m_o2vmr = 0.209448
amrex::Real	m_n2vmr = 0.7906
int	m_o3_size
real1d	m_gas_mol_weights
string1dv	gas_names_yakl_offset
GasConcs	m_gas_concs
int	m_ncol
int	m_nlay
amrex::Vector< int >	m_col_offsets
bool	m_do_aerosol_rad = true
bool	m_extra_clnsky_diag = false
bool	m_extra_clnclrsky_diag = false
int	m_orbital_year = -9999
int	m_orbital_mon = -9999
int	m_orbital_day = -9999
int	m_orbital_sec = -9999
bool	m_fixed_orbital_year = false
amrex::Real	m_orbital_eccen = -9999.
amrex::Real	m_orbital_obliq = -9999.
amrex::Real	m_orbital_mvelp = -9999.
amrex::Real	m_fixed_total_solar_irradiance = -9999.
amrex::Real	m_fixed_solar_zenith_angle = -9999.
int	m_nswbands = 14
int	m_nlwbands = 16
int	m_nswgpts = 112
int	m_nlwgpts = 128
int	m_rad_freq_in_steps = 1
bool	m_do_subcol_sampling = true
real1d	o3_lay
real1d	cosine_zenith
real1d	mu0
real1d	sfc_alb_dir_vis
real1d	sfc_alb_dir_nir
real1d	sfc_alb_dif_vis
real1d	sfc_alb_dif_nir
real1d	sfc_flux_dir_vis
real1d	sfc_flux_dir_nir
real1d	sfc_flux_dif_vis
real1d	sfc_flux_dif_nir
real1d	lat
real1d	lon
real2d	r_lay
real2d	p_lay
real2d	t_lay
real2d	z_del
real2d	p_del
real2d	qv_lay
real2d	qc_lay
real2d	qi_lay
real2d	cldfrac_tot
real2d	eff_radius_qc
real2d	eff_radius_qi
real2d	tmp2d
real2d	lwp
real2d	iwp
real2d	sw_heating
real2d	lw_heating
real2d	d_tint
real2d	p_lev
real2d	t_lev
real2d	sw_flux_up
real2d	sw_flux_dn
real2d	sw_flux_dn_dir
real2d	lw_flux_up
real2d	lw_flux_dn
real2d	sw_clnclrsky_flux_up
real2d	sw_clnclrsky_flux_dn
real2d	sw_clnclrsky_flux_dn_dir
real2d	sw_clrsky_flux_up
real2d	sw_clrsky_flux_dn
real2d	sw_clrsky_flux_dn_dir
real2d	sw_clnsky_flux_up
real2d	sw_clnsky_flux_dn
real2d	sw_clnsky_flux_dn_dir
real2d	lw_clnclrsky_flux_up
real2d	lw_clnclrsky_flux_dn
real2d	lw_clrsky_flux_up
real2d	lw_clrsky_flux_dn
real2d	lw_clnsky_flux_up
real2d	lw_clnsky_flux_dn
real3d	sw_bnd_flux_up
real3d	sw_bnd_flux_dn
real3d	sw_bnd_flux_dir
real3d	sw_bnd_flux_dif
real3d	lw_bnd_flux_up
real3d	lw_bnd_flux_dn
real2d	sfc_alb_dir
real2d	sfc_alb_dif
real3d	aero_tau_sw
real3d	aero_ssa_sw
real3d	aero_g_sw
real3d	aero_tau_lw
real3d	cld_tau_sw_bnd
real3d	cld_tau_lw_bnd
real3d	cld_tau_sw_gpt
real3d	cld_tau_lw_gpt

Constructor & Destructor Documentation

Radiation()

Radiation::Radiation	(	const int &	lev,
		SolverChoice &	sc
	)

{
    // Initialize YAKL
    if (!yakl::isInitialized()) { yakl::init(); }
 
    // Check if we have a valid moisture model
    if (sc.moisture_type != MoistureType::None) { m_moist = true; }
 
    // Check if we have a moisture model with ice
    if (sc.moisture_type == MoistureType::SAM)  { m_ice = true; }
 
    // Construct parser object for following reads
    ParmParse pp("erf");
 
    // Radiation timestep, as a number of atm steps
    pp.query("rad_freq_in_steps", m_rad_freq_in_steps);
 
    // Flag to write fluxes to plt file
    pp.query("rad_write_fluxes", m_rad_write_fluxes);
 
    // Do MCICA subcolumn sampling
    pp.query("rad_do_subcol_sampling", m_do_subcol_sampling);
 
    // Determine orbital year. If orbital_year is negative, use current year
    // from timestamp for orbital year; if positive, use provided orbital year
    // for duration of simulation.
    m_fixed_orbital_year = pp.query("rad_orbital_year", m_orbital_year);
 
    // Get orbital parameters from inputs file
    pp.query("rad_orbital_eccentricity", m_orbital_eccen);
    pp.query("rad_orbital_obliquity"   , m_orbital_obliq);
    pp.query("rad_orbital_mvelp"       , m_orbital_mvelp);
 
    // Get a constant lat/lon for idealized simulations
    pp.query("rad_cons_lat", m_lat_cons);
    pp.query("rad_cons_lon", m_lon_cons);
 
    // Value for prescribing an invariant solar constant (i.e. total solar irradiance at
    // TOA).  Used for idealized experiments such as RCE. Disabled when value is less than 0.
    pp.query("fixed_total_solar_irradiance", m_fixed_total_solar_irradiance);
 
    // Determine whether or not we are using a fixed solar zenith angle (positive value)
    pp.query("fixed_solar_zenith_angle", m_fixed_solar_zenith_angle);
 
    // Get prescribed surface values of greenhouse gases
    pp.query("co2vmr", m_co2vmr);
    pp.queryarr("o3vmr" , m_o3vmr );
    pp.query("n2ovmr", m_n2ovmr);
    pp.query("covmr" , m_covmr );
    pp.query("ch4vmr", m_ch4vmr);
    pp.query("o2vmr" , m_o2vmr );
    pp.query("n2vmr" , m_n2vmr );
 
    // Required aerosol optical properties from SPA
    pp.query("rad_do_aerosol", m_do_aerosol_rad);
 
    // Whether we do extra clean/clear sky calculations
    pp.query("rad_extra_clnclrsky_diag", m_extra_clnclrsky_diag);
    pp.query("rad_extra_clnsky_diag"   , m_extra_clnsky_diag);
 
    // Parse the band and gauss pt sizes
    pp.query("nswbands", m_nswbands);
    pp.query("nlwbands", m_nlwbands);
    pp.query("nswgpts" , m_nswgpts );
    pp.query("nlwgpts" , m_nlwgpts );
 
    // Parse path and file names
    pp.query("rrtmgp_file_path"      , rrtmgp_file_path);
    pp.query("rrtmgp_coeffs_sw"      , rrtmgp_coeffs_sw  );
    pp.query("rrtmgp_coeffs_lw"      , rrtmgp_coeffs_lw  );
    pp.query("rrtmgp_cloud_optics_sw", rrtmgp_cloud_optics_sw);
    pp.query("rrtmgp_cloud_optics_lw", rrtmgp_cloud_optics_lw);
 
    // Append file names to path
    rrtmgp_coeffs_file_sw       = rrtmgp_file_path + "/" + rrtmgp_coeffs_sw;
    rrtmgp_coeffs_file_lw       = rrtmgp_file_path + "/" + rrtmgp_coeffs_lw;
    rrtmgp_cloud_optics_file_sw = rrtmgp_file_path + "/" + rrtmgp_cloud_optics_sw;
    rrtmgp_cloud_optics_file_lw = rrtmgp_file_path + "/" + rrtmgp_cloud_optics_lw;
 
    // Output for user
    if (lev == 0) {
        Print() << "Radiation interface constructed:\n";
        Print() << "========================================================\n";
        Print() << "Coeff SW file: " << rrtmgp_coeffs_file_sw << "\n";
        Print() << "Coeff LW file: " << rrtmgp_coeffs_file_lw << "\n";
        Print() << "Cloud SW file: " << rrtmgp_cloud_optics_file_sw << "\n";
        Print() << "Cloud LW file: " << rrtmgp_cloud_optics_file_lw << "\n";
        Print() << "========================================================\n";
    }
}

Here is the call graph for this function:

~Radiation()

Radiation::~Radiation ( )

default

Member Function Documentation

alloc_buffers()

void Radiation::alloc_buffers

(

)

{
    // 1d size (m_ngas)
    m_gas_mol_weights = real1d("m_gas_mol_weights", m_ngas);
    realHost1d m_gas_mol_weights_h("m_gas_mol_weights_h", m_ngas);
    gas_names_yakl_offset.clear();
    parallel_for(m_ngas, YAKL_LAMBDA (int igas)
    {
        m_gas_mol_weights_h(igas)   = m_mol_weight_gas[igas-1];
        gas_names_yakl_offset.push_back(m_gas_names[igas-1]);
    });
    m_gas_mol_weights_h.deep_copy_to(m_gas_mol_weights);
 
    // 1d size (1 or nlay)
    m_o3_size = m_o3vmr.size();
    AMREX_ASSERT_WITH_MESSAGE(((m_o3_size==1) || (m_o3_size==m_nlay)), "O3 VMR array must be length 1 or nlay");
    o3_lay = real1d("o3_lay", m_o3_size);
    realHost1d o3_lay_h("o3_lay_h", m_o3_size);
    parallel_for(m_o3_size, YAKL_LAMBDA (int io3)
    {
        o3_lay_h(io3) = m_o3vmr[io3-1];
    });
    o3_lay_h.deep_copy_to(o3_lay);
 
    // 1d size (ncol)
    cosine_zenith    = real1d("cosine_zenith"   , m_ncol);
    mu0              = real1d("mu0"             , m_ncol);
    sfc_alb_dir_vis  = real1d("sfc_alb_dir_vis" , m_ncol);
    sfc_alb_dir_nir  = real1d("sfc_alb_dir_nir" , m_ncol);
    sfc_alb_dif_vis  = real1d("sfc_alb_dif_vis" , m_ncol);
    sfc_alb_dif_nir  = real1d("sfc_alb_dif_nir" , m_ncol);
    sfc_flux_dir_vis = real1d("sfc_flux_dir_vis", m_ncol);
    sfc_flux_dir_nir = real1d("sfc_flux_dir_nir", m_ncol);
    sfc_flux_dif_vis = real1d("sfc_flux_dif_vis", m_ncol);
    sfc_flux_dif_nir = real1d("sfc_flux_dif_nir", m_ncol);
    lat              = real1d("lat"             , m_ncol);
    lon              = real1d("lon"             , m_ncol);;
 
    // 2d size (ncol, nlay)
    r_lay         = real2d("r_lay"        , m_ncol, m_nlay);
    p_lay         = real2d("p_lay"        , m_ncol, m_nlay);
    t_lay         = real2d("t_lay"        , m_ncol, m_nlay);
    z_del         = real2d("z_del"        , m_ncol, m_nlay);
    p_del         = real2d("p_del"        , m_ncol, m_nlay);
    qv_lay        = real2d("qv"           , m_ncol, m_nlay);
    qc_lay        = real2d("qc"           , m_ncol, m_nlay);
    qi_lay        = real2d("qi"           , m_ncol, m_nlay);
    cldfrac_tot   = real2d("cldfrac_tot"  , m_ncol, m_nlay);
    eff_radius_qc = real2d("eff_radius_qc", m_ncol, m_nlay);
    eff_radius_qi = real2d("eff_radius_qi", m_ncol, m_nlay);
    tmp2d         = real2d("tmp2d"        , m_ncol, m_nlay);
    lwp           = real2d("lwp"          , m_ncol, m_nlay);
    iwp           = real2d("iwp"          , m_ncol, m_nlay);
    sw_heating    = real2d("sw_heating"   , m_ncol, m_nlay);
    lw_heating    = real2d("lw_heating"   , m_ncol, m_nlay);
 
    // 2d size (ncol, nlay+1)
    d_tint                   = real2d("d_tint"                  , m_ncol, m_nlay+1);
    p_lev                    = real2d("p_lev"                   , m_ncol, m_nlay+1);
    t_lev                    = real2d("t_lev"                   , m_ncol, m_nlay+1);
    sw_flux_up               = real2d("sw_flux_up"              , m_ncol, m_nlay+1);
    sw_flux_dn               = real2d("sw_flux_dn"              , m_ncol, m_nlay+1);
    sw_flux_dn_dir           = real2d("sw_flux_dn_dir"          , m_ncol, m_nlay+1);
    lw_flux_up               = real2d("sw_flux_up"              , m_ncol, m_nlay+1);
    lw_flux_dn               = real2d("sw_flux_dn"              , m_ncol, m_nlay+1);
    sw_clnclrsky_flux_up     = real2d("sw_clnclrsky_flux_up"    , m_ncol, m_nlay+1);
    sw_clnclrsky_flux_dn     = real2d("sw_clnclrsky_flux_dn"    , m_ncol, m_nlay+1);
    sw_clnclrsky_flux_dn_dir = real2d("sw_clnclrsky_flux_dn_dir", m_ncol, m_nlay+1);
    sw_clrsky_flux_up        = real2d("sw_clrsky_flux_up"       , m_ncol, m_nlay+1);
    sw_clrsky_flux_dn        = real2d("sw_clrsky_flux_dn"       , m_ncol, m_nlay+1);
    sw_clrsky_flux_dn_dir    = real2d("sw_clrsky_flux_dn_dir"   , m_ncol, m_nlay+1);
    sw_clnsky_flux_up        = real2d("sw_clnsky_flux_up"       , m_ncol, m_nlay+1);
    sw_clnsky_flux_dn        = real2d("sw_clnsky_flux_dn"       , m_ncol, m_nlay+1);
    sw_clnsky_flux_dn_dir    = real2d("sw_clnsky_flux_dn_dir"   , m_ncol, m_nlay+1);
    lw_clnclrsky_flux_up     = real2d("lw_clnclrsky_flux_up"    , m_ncol, m_nlay+1);
    lw_clnclrsky_flux_dn     = real2d("lw_clnclrsky_flux_dn"    , m_ncol, m_nlay+1);
    lw_clrsky_flux_up        = real2d("lw_clrsky_flux_up"       , m_ncol, m_nlay+1);
    lw_clrsky_flux_dn        = real2d("lw_clrsky_flux_dn"       , m_ncol, m_nlay+1);
    lw_clnsky_flux_up        = real2d("lw_clnsky_flux_up"       , m_ncol, m_nlay+1);
    lw_clnsky_flux_dn        = real2d("lw_clnsky_flux_dn"       , m_ncol, m_nlay+1);
 
    // 3d size (ncol, nlay+1, nswbands)
    sw_bnd_flux_up  = real3d("sw_bnd_flux_up" , m_ncol, m_nlay+1, m_nswbands);
    sw_bnd_flux_dn  = real3d("sw_bnd_flux_dn" , m_ncol, m_nlay+1, m_nswbands);
    sw_bnd_flux_dir = real3d("sw_bnd_flux_dir", m_ncol, m_nlay+1, m_nswbands);
    sw_bnd_flux_dif = real3d("sw_bnd_flux_dif", m_ncol, m_nlay+1, m_nswbands);
 
    // 3d size (ncol, nlay+1, nlwbands)
    lw_bnd_flux_up = real3d("lw_bnd_flux_up" , m_ncol, m_nlay+1, m_nlwbands);
    lw_bnd_flux_dn = real3d("lw_bnd_flux_up" , m_ncol, m_nlay+1, m_nlwbands);
 
    // 2d size (ncol, nswbands)
    sfc_alb_dir = real2d("sfc_alb_dir", m_ncol, m_nswbands);
    sfc_alb_dif = real2d("sfc_alb_dif", m_ncol, m_nswbands);
 
    // 3d size (ncol, nlay, n[sw,lw]bands)
    aero_tau_sw = real3d("aero_tau_sw", m_ncol, m_nlay, m_nswbands);
    aero_ssa_sw = real3d("aero_ssa_sw", m_ncol, m_nlay, m_nswbands);
    aero_g_sw   = real3d("aero_g_sw"  , m_ncol, m_nlay, m_nswbands);
    aero_tau_lw = real3d("aero_tau_lw", m_ncol, m_nlay, m_nlwbands);
 
    // 3d size (ncol, nlay, n[sw,lw]bnds)
    cld_tau_sw_bnd = real3d("cld_tau_sw_bnd", m_ncol, m_nlay, m_nswbands);
    cld_tau_lw_bnd = real3d("cld_tau_lw_bnd", m_ncol, m_nlay, m_nlwbands);
 
    // 3d size (ncol, nlay, n[sw,lw]gpts)
    cld_tau_sw_gpt = real3d("cld_tau_sw_gpt", m_ncol, m_nlay, m_nswgpts);
    cld_tau_lw_gpt = real3d("cld_tau_lw_gpt", m_ncol, m_nlay, m_nlwgpts);
}

dealloc_buffers()

void Radiation::dealloc_buffers

(

)

{
    // 1d size (m_ngas)
    m_gas_mol_weights.deallocate();
 
    // 1d size (1 or nlay)
    o3_lay.deallocate();
 
    // 1d size (ncol)
    cosine_zenith.deallocate();
    mu0.deallocate();
    sfc_alb_dir_vis.deallocate();
    sfc_alb_dir_nir.deallocate();
    sfc_alb_dif_vis.deallocate();
    sfc_alb_dif_nir.deallocate();
    sfc_flux_dir_vis.deallocate();
    sfc_flux_dir_nir.deallocate();
    sfc_flux_dif_vis.deallocate();
    sfc_flux_dif_nir.deallocate();
    lat.deallocate();
    lon.deallocate();
 
    // 2d size (ncol, nlay)
    r_lay.deallocate();
    p_lay.deallocate();
    t_lay.deallocate();
    z_del.deallocate();
    p_del.deallocate();
    qv_lay.deallocate();
    qc_lay.deallocate();
    qi_lay.deallocate();
    cldfrac_tot.deallocate();
    eff_radius_qc.deallocate();
    eff_radius_qi.deallocate();
    tmp2d.deallocate();
    lwp.deallocate();
    iwp.deallocate();
 
    sw_heating.deallocate();
    lw_heating.deallocate();
 
    // 2d size (ncol, nlay+1)
    d_tint.deallocate();
    p_lev.deallocate();
    t_lev.deallocate();
 
    sw_flux_up.deallocate();
    sw_flux_dn.deallocate();
    sw_flux_dn_dir.deallocate();
    lw_flux_up.deallocate();
    lw_flux_dn.deallocate();
    sw_clnclrsky_flux_up.deallocate();
    sw_clnclrsky_flux_dn.deallocate();
    sw_clnclrsky_flux_dn_dir.deallocate();
    sw_clrsky_flux_up.deallocate();
    sw_clrsky_flux_dn.deallocate();
    sw_clrsky_flux_dn_dir.deallocate();
    sw_clnsky_flux_up.deallocate();
    sw_clnsky_flux_dn.deallocate();
    sw_clnsky_flux_dn_dir.deallocate();
    lw_clnclrsky_flux_up.deallocate();
    lw_clnclrsky_flux_dn.deallocate();
    lw_clrsky_flux_up.deallocate();
    lw_clrsky_flux_dn.deallocate();
    lw_clnsky_flux_up.deallocate();
    lw_clnsky_flux_dn.deallocate();
 
    // 3d size (ncol, nlay+1, nswbands)
    sw_bnd_flux_up.deallocate();
    sw_bnd_flux_dn.deallocate();
    sw_bnd_flux_dir.deallocate();
    sw_bnd_flux_dif.deallocate();
 
    // 3d size (ncol, nlay+1, nlwbands)
    lw_bnd_flux_up.deallocate();
    lw_bnd_flux_dn.deallocate();
 
    // 2d size (ncol, nswbands)
    sfc_alb_dir.deallocate();
    sfc_alb_dif.deallocate();
 
    // 3d size (ncol, nlay, n[sw,lw]bands)
    aero_tau_sw.deallocate();
    aero_ssa_sw.deallocate();
    aero_g_sw.deallocate();
    aero_tau_lw.deallocate();
 
    // 3d size (ncol, nlay, n[sw,lw]bnds)
    cld_tau_sw_bnd.deallocate();
    cld_tau_lw_bnd.deallocate();
 
    // 3d size (ncol, nlay, n[sw,lw]gpts)
    cld_tau_sw_gpt.deallocate();
    cld_tau_lw_gpt.deallocate();
}

finalize_impl()

void Radiation::finalize_impl

(

)

{
    // Finish rrtmgp
    m_gas_concs.reset();
    rrtmgp::rrtmgp_finalize();
 
    // Fill the AMReX MFs from YAKL Arrays
    yakl_buffers_to_mf();
 
    // Write fluxes if requested
    if (m_rad_write_fluxes) { write_rrtmgp_fluxes(); }
 
    // Deallocate the buffer arrays
    dealloc_buffers();
}

Referenced by rad_run_impl().

Here is the call graph for this function:

Here is the caller graph for this function:

initialize_impl()

void Radiation::initialize_impl

(

)

{
    // Call API to initialize
    m_gas_concs.init(gas_names_yakl_offset, m_ncol, m_nlay);
    rrtmgp::rrtmgp_initialize(m_gas_concs,
                              rrtmgp_coeffs_file_sw      , rrtmgp_coeffs_file_lw      ,
                              rrtmgp_cloud_optics_file_sw, rrtmgp_cloud_optics_file_lw);
}

Referenced by rad_run_impl().

Here is the call graph for this function:

Here is the caller graph for this function:

mf_to_yakl_buffers()

void Radiation::mf_to_yakl_buffers

(

)

{
    bool moist = m_moist;
    bool ice   = m_ice;
    int  ncol  = m_ncol;
    int  nlay  = m_nlay;
    Real dz    = m_geom.CellSize(2);
    Real cons_lat = m_lat_cons;
    Real cons_lon = m_lon_cons;
    for (MFIter mfi(*m_cons_in); mfi.isValid(); ++mfi) {
        const auto& vbx  = mfi.validbox();
        const int nx     = vbx.length(0);
        const int imin   = vbx.smallEnd(0);
        const int jmin   = vbx.smallEnd(1);
        const int offset = m_col_offsets[mfi.index()];
        const Array4<const Real>& cons_arr = m_cons_in->const_array(mfi);
        const Array4<const Real>& z_arr    = (m_z_phys) ? m_z_phys->const_array(mfi) :
                                                          Array4<const Real>{};
        const Array4<const Real>& lat_arr  = (m_lat)    ? m_lat->const_array(mfi) :
                                                          Array4<const Real>{};
        const Array4<const Real>& lon_arr  = (m_lon)    ? m_lon->const_array(mfi) :
                                                          Array4<const Real>{};
        ParallelFor(vbx, [=] AMREX_GPU_DEVICE (int i, int j, int k)
        {
            // map [i,j,k] 0-based to [icol, ilay] 1-based
            const int icol = (j-jmin)*nx + (i-imin) + 1 + offset;
            const int ilay = k+1;
 
            // EOS input (at CC)
            Real r  = cons_arr(i,j,k,Rho_comp);
            Real rt = cons_arr(i,j,k,RhoTheta_comp);
            Real qv = (moist) ? cons_arr(i,j,k,RhoQ1_comp)/r : 0.0;
            Real qc = (moist) ? cons_arr(i,j,k,RhoQ2_comp)/r : 0.0;
            Real qi = (ice)   ? cons_arr(i,j,k,RhoQ3_comp)/r : 0.0;
 
            // EOS avg to z-face
            Real r_lo  = cons_arr(i,j,k-1,Rho_comp);
            Real rt_lo = cons_arr(i,j,k-1,RhoTheta_comp);
            Real qv_lo = (moist) ? cons_arr(i,j,k-1,RhoQ1_comp)/r_lo : 0.0;
            Real r_avg  = 0.5 * (r  + r_lo);
            Real rt_avg = 0.5 * (rt + rt_lo);
            Real qv_avg = 0.5 * (qv + qv_lo);
 
            // Buffers at CC
            r_lay(icol,ilay) = r;
            p_lay(icol,ilay) = getPgivenRTh(rt, qv);
            t_lay(icol,ilay) = getTgivenRandRTh(r, rt, qv);
            z_del(icol,ilay) = (z_arr) ? 0.25 * ( (z_arr(i  ,j  ,k+1) - z_arr(i  ,j  ,k))
                                                + (z_arr(i+1,j  ,k+1) - z_arr(i+1,j  ,k))
                                                + (z_arr(i  ,j+1,k+1) - z_arr(i  ,j+1,k))
                                                + (z_arr(i+1,j  ,k+1) - z_arr(i+1,j  ,k)) ) : dz;
            qv_lay(icol,ilay) = qv;
            qc_lay(icol,ilay) = qc;
            qi_lay(icol,ilay) = qi;
            cldfrac_tot(icol,ilay) = ((qc+qi)>1.0e-5) ? 1. : 0.;
 
            // NOTE: These are populated in 'mixing_ratio_to_cloud_mass'
            lwp(icol,ilay) = 0.0;
            iwp(icol,ilay) = 0.0;
 
            // NOTE: These would be populated from P3 (we use the constants in p3_main_impl.hpp)
            eff_radius_qc(icol,ilay) = (qc>1.0e-12) ? 10.0e-6 : 0.0;
            eff_radius_qi(icol,ilay) = (qi>1.0e-12) ? 25.0e-6 : 0.0;
 
            // Buffers on z-faces (nlay+1)
            p_lev(icol,ilay) = getPgivenRTh(rt_avg, qv_avg);
            t_lev(icol,ilay) = getTgivenRandRTh(r_avg, rt_avg, qv_avg);
            if (ilay==nlay) {
                Real r_hi  = cons_arr(i,j,k+1,Rho_comp);
                Real rt_hi = cons_arr(i,j,k+1,RhoTheta_comp);
                Real qv_hi = (moist) ? cons_arr(i,j,k+1,RhoQ1_comp)/r_hi : 0.0;
                r_avg  = 0.5 * (r  + r_hi);
                rt_avg = 0.5 * (rt + rt_hi);
                qv_avg = 0.5 * (qv + qv_hi);
                p_lev(icol,ilay+1) = getPgivenRTh(rt_avg, qv_avg);
                t_lev(icol,ilay+1) = getTgivenRandRTh(r_avg, rt_avg, qv_avg);
            }
 
            // 1D data structures
            if (k==0) {
                lat(icol) = (m_lat) ? lat_arr(i,j,0) : cons_lat;
                lon(icol) = (m_lon) ? lon_arr(i,j,0) : cons_lon;
            }
 
        });
    }
 
    // Separate YAKL kernel for derived quantities
    parallel_for(SimpleBounds<2>(ncol, nlay), YAKL_LAMBDA (int icol, int ilay)
    {
        p_del(icol,ilay)  = p_lev(icol,ilay+1) - p_lev(icol,ilay);
    });
 
    // TODO: Fill properly
    // No LSM, so follow EAMXX dummy atmos and set constants
    yakl::memset(mu0, 0.86);
    yakl::memset(sfc_alb_dir_vis, 0.06);
    yakl::memset(sfc_alb_dir_nir, 0.06);
    yakl::memset(sfc_alb_dif_vis, 0.06);
    yakl::memset(sfc_alb_dif_nir, 0.06);
 
    // TODO: Fill properly
    yakl::memset(aero_tau_sw, 0.0);
    yakl::memset(aero_ssa_sw, 0.0);
    yakl::memset(aero_g_sw  , 0.0);
    yakl::memset(aero_tau_lw, 0.0);
}

Here is the call graph for this function:

rad_run_impl()

void Radiation::rad_run_impl ( )

inline

    {
        if (m_update_rad) {
            amrex::Print() << "Advancing radiation at level: " << m_lev << " ...";
            this->initialize_impl();
            this->run_impl();
            this->finalize_impl();
            amrex::Print() << "DONE\n";
        }
    }

Here is the call graph for this function:

run_impl()

void Radiation::run_impl

(

)

{
    // Local copies
    const auto ncol     = m_ncol;
    const auto nlay     = m_nlay;
    const auto nswbands = m_nswbands;
 
    // Compute orbital parameters; these are used both for computing
    // the solar zenith angle and also for computing total solar
    // irradiance scaling (tsi_scaling).
    real obliqr, lambm0, mvelpp;
    int  orbital_year = m_orbital_year;
    real eccen        = m_orbital_eccen;
    real obliq        = m_orbital_obliq;
    real mvelp        = m_orbital_mvelp;
    if (eccen >= 0 && obliq >= 0 && mvelp >= 0) {
      // fixed orbital parameters forced with orbital_year == ORB_UNDEF_INT
      orbital_year = ORB_UNDEF_INT;
    }
    orbital_params(orbital_year, eccen, obliq,
                   mvelp, obliqr, lambm0, mvelpp);
 
    // Use the orbital parameters to calculate the solar declination and eccentricity factor
    real delta, eccf;
    // TODO: Generalize the days per month.
    // Want day + fraction; calday 1 == Jan 1 0Z
    static constexpr real dpm = (365.0/12.0);
    real calday = (m_orbital_mon-1.0)*dpm + std::min(m_orbital_day-1.0,dpm-1.0) + m_orbital_sec/86400.0;
    orbital_decl(calday, eccen, mvelpp, lambm0, obliqr, delta, eccf);
 
    // Overwrite eccf if using a fixed solar constant.
    auto fixed_total_solar_irradiance = m_fixed_total_solar_irradiance;
    if (fixed_total_solar_irradiance >= 0){
       eccf = fixed_total_solar_irradiance/1360.9;
    }
 
    // Precompute volume mixing ratio (VMR) for all gases
    //
    // H2O is obtained from qv.
    // All other comps are set to constants for now
    for (int igas(0); igas < m_ngas; ++igas) {
      auto name = m_gas_names[igas];
      auto gas_mol_weight = m_mol_weight_gas[igas];
      if (name == "H2O") {
          parallel_for(SimpleBounds<2>(ncol, nlay), YAKL_LAMBDA (int icol, int ilay)
          {
              tmp2d(icol,ilay) = qv_lay(icol,ilay) * mwdair/ gas_mol_weight;
          });
      } else if (name == "CO2") {
          yakl::memset(tmp2d, m_co2vmr);
      } else if (name == "O3")  {
          if (m_o3_size==1) {
              yakl::memset(tmp2d, m_o3vmr[0] );
          } else {
              parallel_for(SimpleBounds<2>(ncol, nlay), YAKL_LAMBDA (int icol, int ilay)
              {
                  tmp2d(icol,ilay) = o3_lay(ilay);
              });
          }
      } else if (name == "N2O") {
          yakl::memset(tmp2d, m_n2ovmr);
      } else if (name == "CO")  {
          yakl::memset(tmp2d, m_covmr );
      } else if (name == "CH4") {
          yakl::memset(tmp2d, m_ch4vmr);
      } else if (name == "O2") {
          yakl::memset(tmp2d, m_o2vmr );
      } else if (name == "N2") {
          yakl::memset(tmp2d, m_n2vmr );
      } else {
          Abort("Radiation: Unknown gas component.");
      }
 
      // Populate GasConcs object
      m_gas_concs.set_vmr(name, tmp2d);
    }
 
 
    // TODO: No LSM so leaving comment for code
    // Calculate T_int from longwave flux up from the surface, assuming
    // blackbody emission with emissivity of 1.
 
 
    // Determine the cosine zenith angle.
    // This must be done on HOST and copied to device.
    realHost1d h_mu0("h_mu0", ncol);
    if (m_fixed_solar_zenith_angle > 0) {
        yakl::memset(h_mu0, m_fixed_solar_zenith_angle);
    } else {
        realHost1d h_lat("h_lat", ncol);
        realHost1d h_lon("h_lon", ncol);
        lat.deep_copy_to(h_lat);
        lon.deep_copy_to(h_lon);
        parallel_for(ncol, YAKL_LAMBDA (int icol)
        {
            // Convert lat/lon to radians
            real dt      = real(m_dt);
            real lat_col = h_lat(icol)*PI/180.0;
            real lon_col = h_lon(icol)*PI/180.0;
            real lcalday = calday;
            real ldelta  = delta;
            h_mu0(icol)  = orbital_cos_zenith(lcalday, lat_col, lon_col, ldelta, m_rad_freq_in_steps * dt);
        });
    }
    h_mu0.deep_copy_to(mu0);
 
    // Compute layer cloud mass (per unit area), populates lwp/iwp
    rrtmgp::mixing_ratio_to_cloud_mass(qc_lay, cldfrac_tot, p_del, lwp);
    rrtmgp::mixing_ratio_to_cloud_mass(qi_lay, cldfrac_tot, p_del, iwp);
 
    // Convert to g/m2 (needed by RRTMGP)
    parallel_for(SimpleBounds<2>(ncol, nlay), YAKL_LAMBDA (int icol, int ilay)
    {
        lwp(icol,ilay) *= 1.e3;
        iwp(icol,ilay) *= 1.e3;
    });
 
    // Compute band-by-band surface_albedos. This is needed since
    // the AD passes broadband albedos, but rrtmgp require band-by-band.
    rrtmgp::compute_band_by_band_surface_albedos(ncol, nswbands,
                                                 sfc_alb_dir_vis, sfc_alb_dir_nir,
                                                 sfc_alb_dif_vis, sfc_alb_dif_nir,
                                                 sfc_alb_dir    , sfc_alb_dif);
 
    // Run RRTMGP driver
    rrtmgp::rrtmgp_main(ncol, m_nlay,
                        p_lay, t_lay, p_lev, t_lev,
                        m_gas_concs,
                        sfc_alb_dir, sfc_alb_dif, mu0,
                        lwp, iwp, eff_radius_qc, eff_radius_qi, cldfrac_tot,
                        aero_tau_sw, aero_ssa_sw, aero_g_sw, aero_tau_lw,
                        cld_tau_sw_bnd, cld_tau_lw_bnd,
                        cld_tau_sw_gpt, cld_tau_lw_gpt,
                        sw_flux_up, sw_flux_dn, sw_flux_dn_dir, lw_flux_up, lw_flux_dn,
                        sw_clnclrsky_flux_up, sw_clnclrsky_flux_dn, sw_clnclrsky_flux_dn_dir,
                        sw_clrsky_flux_up, sw_clrsky_flux_dn, sw_clrsky_flux_dn_dir,
                        sw_clnsky_flux_up, sw_clnsky_flux_dn, sw_clnsky_flux_dn_dir,
                        lw_clnclrsky_flux_up, lw_clnclrsky_flux_dn,
                        lw_clrsky_flux_up, lw_clrsky_flux_dn,
                        lw_clnsky_flux_up, lw_clnsky_flux_dn,
                        sw_bnd_flux_up, sw_bnd_flux_dn, sw_bnd_flux_dir, lw_bnd_flux_up, lw_bnd_flux_dn,
                        eccf, m_extra_clnclrsky_diag, m_extra_clnsky_diag);
 
#if 0
    // UNIT TEST
    //================================================================================
    yakl::memset(mu0, 0.86);
    yakl::memset(sfc_alb_dir_vis, 0.06);
    yakl::memset(sfc_alb_dir_nir, 0.06);
    yakl::memset(sfc_alb_dif_vis, 0.06);
    yakl::memset(sfc_alb_dif_nir, 0.06);
 
    // Generate some fake liquid and ice water data. We pick values to be midway between
    // the min and max of the valid lookup table values for effective radii
    real rel_val = 0.5 * (rrtmgp::cloud_optics_sw.get_min_radius_liq()
                        + rrtmgp::cloud_optics_sw.get_max_radius_liq());
    real rei_val = 0.5 * (rrtmgp::cloud_optics_sw.get_min_radius_ice()
                        + rrtmgp::cloud_optics_sw.get_max_radius_ice());
 
    // Restrict clouds to troposphere (> 100 hPa = 100*100 Pa) and not very close to the ground (< 900 hPa), and
    // put them in 2/3 of the columns since that's roughly the total cloudiness of earth.
    // Set sane values for liquid and ice water path.
    // NOTE: these "sane" values are in g/m2!
    parallel_for( SimpleBounds<2>(nlay,ncol) , YAKL_LAMBDA (int ilay, int icol)
    {
        cldfrac_tot(icol,ilay) = (p_lay(icol,ilay) > 100._wp * 100._wp) &&
                                 (p_lay(icol,ilay) < 900._wp * 100._wp) &&
                                 (icol%3 != 0);
        // Ice and liquid will overlap in a few layers
        lwp(icol,ilay) = (cldfrac_tot(icol,ilay) && t_lay(icol,ilay) > 263._wp) ? 10._wp : 0._wp;
        iwp(icol,ilay) = (cldfrac_tot(icol,ilay) && t_lay(icol,ilay) < 273._wp) ? 10._wp : 0._wp;
        eff_radius_qc(icol,ilay) = (lwp(icol,ilay) > 0._wp) ? rel_val : 0._wp;
        eff_radius_qi(icol,ilay) = (iwp(icol,ilay) > 0._wp) ? rei_val : 0._wp;
    });
 
    rrtmgp::compute_band_by_band_surface_albedos(ncol, nswbands,
                                                 sfc_alb_dir_vis, sfc_alb_dir_nir,
                                                 sfc_alb_dif_vis, sfc_alb_dif_nir,
                                                 sfc_alb_dir    , sfc_alb_dif);
 
    yakl::memset(aero_tau_sw, 0.0);
    yakl::memset(aero_ssa_sw, 0.0);
    yakl::memset(aero_g_sw  , 0.0);
    yakl::memset(aero_tau_lw, 0.0);
 
    rrtmgp::rrtmgp_main(ncol, m_nlay,
                        p_lay, t_lay, p_lev, t_lev,
                        m_gas_concs,
                        sfc_alb_dir, sfc_alb_dif, mu0,
                        lwp, iwp, eff_radius_qc, eff_radius_qi, cldfrac_tot,
                        aero_tau_sw, aero_ssa_sw, aero_g_sw, aero_tau_lw,
                        cld_tau_sw_bnd, cld_tau_lw_bnd,
                        cld_tau_sw_gpt, cld_tau_lw_gpt,
                        sw_flux_up, sw_flux_dn, sw_flux_dn_dir, lw_flux_up, lw_flux_dn,
                        sw_clnclrsky_flux_up, sw_clnclrsky_flux_dn, sw_clnclrsky_flux_dn_dir,
                        sw_clrsky_flux_up, sw_clrsky_flux_dn, sw_clrsky_flux_dn_dir,
                        sw_clnsky_flux_up, sw_clnsky_flux_dn, sw_clnsky_flux_dn_dir,
                        lw_clnclrsky_flux_up, lw_clnclrsky_flux_dn,
                        lw_clrsky_flux_up, lw_clrsky_flux_dn,
                        lw_clnsky_flux_up, lw_clnsky_flux_dn,
                        sw_bnd_flux_up, sw_bnd_flux_dn, sw_bnd_flux_dir, lw_bnd_flux_up, lw_bnd_flux_dn,
                        1.0);
    //================================================================================
#endif
 
    // Update heating tendency
    rrtmgp::compute_heating_rate(sw_flux_up, sw_flux_dn, r_lay, z_del, sw_heating);
    rrtmgp::compute_heating_rate(lw_flux_up, lw_flux_dn, r_lay, z_del, lw_heating);
 
    // Compute surface fluxes
    const int kbot = nlay + 1; // Should this be 1 for our layout?
    parallel_for(SimpleBounds<3>(ncol, nlay+1, nswbands), YAKL_LAMBDA (int icol, int ilay, int ibnd)
    {
        sw_bnd_flux_dif(icol,ilay,ibnd) = sw_bnd_flux_dn(icol,ilay,ibnd) - sw_bnd_flux_dir(icol,ilay,ibnd);
    });
    rrtmgp::compute_broadband_surface_fluxes(ncol, kbot, nswbands,
                                             sw_bnd_flux_dir , sw_bnd_flux_dif ,
                                             sfc_flux_dir_vis, sfc_flux_dir_nir,
                                             sfc_flux_dif_vis, sfc_flux_dif_nir);
}

Referenced by rad_run_impl().

Here is the call graph for this function:

Here is the caller graph for this function:

set_grids()

void Radiation::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
	)

{
    // Set data members that may change
    m_lev            = level;
    m_step           = step;
    m_time           = time;
    m_dt             = dt;
    m_geom           = geom;
    m_cons_in        = cons_in;
    m_lsm_fluxes     = lsm_fluxes;
    m_qheating_rates = qheating_rates;
    m_z_phys         = z_phys;
    m_lat            = lat;
    m_lon            = lon;
 
    // Update the day and month
    time_t timestamp = time_t(int(time));
    struct tm *timeinfo = localtime(&timestamp);
    if (m_fixed_orbital_year) {
        m_orbital_mon  = timeinfo->tm_mon + 1;
        m_orbital_day  = timeinfo->tm_mday;
        m_orbital_sec  = timeinfo->tm_hour*3600 + timeinfo->tm_min*60 + timeinfo->tm_sec;
    } else {
        m_orbital_year = timeinfo->tm_year + 1900;
        m_orbital_mon  = timeinfo->tm_mon  + 1;
        m_orbital_day  = timeinfo->tm_mday;
        m_orbital_sec  = timeinfo->tm_hour*3600 + timeinfo->tm_min*60 + timeinfo->tm_sec;
    }
 
    // Only allocate and proceed if we are going to update radiation
    m_update_rad = false;
    if (m_rad_freq_in_steps > 0) { m_update_rad = ( (m_step == 0) || (m_step % m_rad_freq_in_steps == 0) ); }
 
    if (m_update_rad) {
        // Reset vector of offsets for columnar data
        m_nlay = geom.Domain().length(2);
 
        m_ncol = 0;
        m_col_offsets.clear();
        m_col_offsets.resize(int(ba.size()));
        for (MFIter mfi(*m_cons_in); mfi.isValid(); ++mfi) {
            const auto& vbx = mfi.validbox();
            int nx = vbx.length(0);
            int ny = vbx.length(1);
            m_col_offsets[mfi.index()] = m_ncol;
            m_ncol += nx * ny;
        }
 
        // Allocate the buffer arrays
        alloc_buffers();
 
        // Fill the YAKL Arrays from AMReX MFs
        mf_to_yakl_buffers();
    }
}

write_rrtmgp_fluxes()

void Radiation::write_rrtmgp_fluxes

(

)

{
    int n_fluxes = 5;
    MultiFab mf_flux(m_cons_in->boxArray(), m_cons_in->DistributionMap(), n_fluxes, 0);
 
   for (MFIter mfi(mf_flux); mfi.isValid(); ++mfi) {
        const auto& vbx      = mfi.validbox();
        const int nx         = vbx.length(0);
        const int imin       = vbx.smallEnd(0);
        const int jmin       = vbx.smallEnd(1);
        const int offset     = m_col_offsets[mfi.index()];
        const Array4<Real>& dst_arr = mf_flux.array(mfi);
        ParallelFor(vbx, [=] AMREX_GPU_DEVICE (int i, int j, int k)
        {
            // map [i,j,k] 0-based to [icol, ilay] 1-based
            const int icol = (j-jmin)*nx + (i-imin) + 1 + offset;
            const int ilay = k+1;
 
            // SW and LW fluxes
            dst_arr(i,j,k,0) = sw_flux_up(icol,ilay);
            dst_arr(i,j,k,1) = sw_flux_dn(icol,ilay);
            dst_arr(i,j,k,2) = sw_flux_dn_dir(icol,ilay);
            dst_arr(i,j,k,3) = lw_flux_up(icol,ilay);
            dst_arr(i,j,k,4) = lw_flux_dn(icol,ilay);
        });
   }
 
 
   std::string plotfilename = amrex::Concatenate("plt_rad", m_step, 5);
   Vector<std::string> flux_names = {"sw_flux_up", "sw_flux_dn", "sw_flux_dir",
                                     "lw_flux_up", "lw_flux_dn"};
   WriteSingleLevelPlotfile(plotfilename, mf_flux, flux_names, m_geom, m_time, m_step);
}

Here is the call graph for this function:

yakl_buffers_to_mf()

void Radiation::yakl_buffers_to_mf

(

)

{
    for (MFIter mfi(*m_cons_in); mfi.isValid(); ++mfi) {
        const auto& vbx      = mfi.validbox();
        const auto& sbx      = makeSlab(vbx,2,vbx.smallEnd(2));
        const int nx         = vbx.length(0);
        const int imin       = vbx.smallEnd(0);
        const int jmin       = vbx.smallEnd(1);
        const int offset     = m_col_offsets[mfi.index()];
        const Array4<Real>& q_arr = m_qheating_rates->array(mfi);
        ParallelFor(vbx, [=] AMREX_GPU_DEVICE (int i, int j, int k)
        {
            // map [i,j,k] 0-based to [icol, ilay] 1-based
            const int icol = (j-jmin)*nx + (i-imin) + 1 + offset;
            const int ilay = k+1;
 
            // Temperature heating rate for SW and LW
            q_arr(i,j,k,0) = sw_heating(icol,ilay);
            q_arr(i,j,k,1) = lw_heating(icol,ilay);
 
            // Convert the rates for theta_d
            Real exner = getExnergivenP(Real(p_lay(icol,ilay)), R_d/Cp_d);
            q_arr(i,j,k,0) *= exner;
            q_arr(i,j,k,1) *= exner;
 
            /*
            if (i==0 && j==0) {
                AllPrint() << "Qsrcs: " << IntVect(i,j,k) << ' '
                           << IntVect(icol,0,ilay) << ' '
                           << q_arr(i,j,k,0) << ' '
                           << q_arr(i,j,k,1) << ' '
                           << sw_heating(icol,ilay) << ' '
                           << lw_heating(icol,ilay) << ' '
                           << exner << "\n";
            }
            */
 
        });
        if (m_lsm_fluxes) {
            const Array4<Real>& lsm_arr =  m_lsm_fluxes->array(mfi);
            ParallelFor(sbx, [=] AMREX_GPU_DEVICE (int i, int j, int k)
            {
                // map [i,j,k] 0-based to [icol, ilay] 1-based
                const int icol = (j-jmin)*nx + (i-imin) + 1 + offset;
 
                // SW fluxes for LSM
                lsm_arr(i,j,k,0) = sfc_flux_dir_vis(icol);
                lsm_arr(i,j,k,1) = sfc_flux_dir_nir(icol);
                lsm_arr(i,j,k,2) = sfc_flux_dif_vis(icol);
                lsm_arr(i,j,k,3) = sfc_flux_dif_nir(icol);
 
                // Net SW flux for LSM
                lsm_arr(i,j,k,4) = sfc_flux_dir_vis(icol) + sfc_flux_dir_nir(icol)
                                 + sfc_flux_dif_vis(icol) + sfc_flux_dif_nir(icol);
 
                // LW flux for LSM (at bottom surface)
                lsm_arr(i,j,k,5) = lw_flux_dn(icol,1);
            });
        }
    }
}

Here is the call graph for this function:

Member Data Documentation

aero_g_sw

real3d Radiation::aero_g_sw

private

aero_ssa_sw

real3d Radiation::aero_ssa_sw

private

aero_tau_lw

real3d Radiation::aero_tau_lw

private

aero_tau_sw

real3d Radiation::aero_tau_sw

private

cld_tau_lw_bnd

real3d Radiation::cld_tau_lw_bnd

private

cld_tau_lw_gpt

real3d Radiation::cld_tau_lw_gpt

private

cld_tau_sw_bnd

real3d Radiation::cld_tau_sw_bnd

private

cld_tau_sw_gpt

real3d Radiation::cld_tau_sw_gpt

private

cldfrac_tot

real2d Radiation::cldfrac_tot

private

cosine_zenith

real1d Radiation::cosine_zenith

private

d_tint

real2d Radiation::d_tint

private

eff_radius_qc

real2d Radiation::eff_radius_qc

private

eff_radius_qi

real2d Radiation::eff_radius_qi

private

gas_names_yakl_offset

string1dv Radiation::gas_names_yakl_offset

private

iwp

real2d Radiation::iwp

private

lat

real1d Radiation::lat

private

lon

real1d Radiation::lon

private

lw_bnd_flux_dn

real3d Radiation::lw_bnd_flux_dn

private

lw_bnd_flux_up

real3d Radiation::lw_bnd_flux_up

private

lw_clnclrsky_flux_dn

real2d Radiation::lw_clnclrsky_flux_dn

private

lw_clnclrsky_flux_up

real2d Radiation::lw_clnclrsky_flux_up

private

lw_clnsky_flux_dn

real2d Radiation::lw_clnsky_flux_dn

private

lw_clnsky_flux_up

real2d Radiation::lw_clnsky_flux_up

private

lw_clrsky_flux_dn

real2d Radiation::lw_clrsky_flux_dn

private

lw_clrsky_flux_up

real2d Radiation::lw_clrsky_flux_up

private

lw_flux_dn

real2d Radiation::lw_flux_dn

private

lw_flux_up

real2d Radiation::lw_flux_up

private

lw_heating

real2d Radiation::lw_heating

private

lwp

real2d Radiation::lwp

private

m_ba

amrex::BoxArray Radiation::m_ba

private

m_ch4vmr

amrex::Real Radiation::m_ch4vmr = 1807.851e-9

private

m_co2vmr

amrex::Real Radiation::m_co2vmr = 388.717e-6

private

m_col_offsets

amrex::Vector<int> Radiation::m_col_offsets

private

m_cons_in

amrex::MultiFab* Radiation::m_cons_in = nullptr

private

m_covmr

amrex::Real Radiation::m_covmr = 1.0e-7

private

m_do_aerosol_rad

bool Radiation::m_do_aerosol_rad = true

private

m_do_subcol_sampling

bool Radiation::m_do_subcol_sampling = true

private

m_dt

amrex::Real Radiation::m_dt

private

m_extra_clnclrsky_diag

bool Radiation::m_extra_clnclrsky_diag = false

private

m_extra_clnsky_diag

bool Radiation::m_extra_clnsky_diag = false

private

m_fixed_orbital_year

bool Radiation::m_fixed_orbital_year = false

private

m_fixed_solar_zenith_angle

amrex::Real Radiation::m_fixed_solar_zenith_angle = -9999.

private

m_fixed_total_solar_irradiance

amrex::Real Radiation::m_fixed_total_solar_irradiance = -9999.

private

m_gas_concs

GasConcs Radiation::m_gas_concs

private

m_gas_mol_weights

real1d Radiation::m_gas_mol_weights

private

m_gas_names

const std::vector<std::string> Radiation::m_gas_names

private

Initial value:

= {"H2O", "CO2", "O3", "N2O",
                                                  "CO" , "CH4", "O2", "N2" }

m_geom

amrex::Geometry Radiation::m_geom

private

m_ice

bool Radiation::m_ice = false

private

m_lat

amrex::MultiFab* Radiation::m_lat = nullptr

private

m_lat_cons

amrex::Real Radiation::m_lat_cons = 39.809860

private

m_lev

int Radiation::m_lev

private

Referenced by rad_run_impl().

m_lon

amrex::MultiFab* Radiation::m_lon = nullptr

private

m_lon_cons

amrex::Real Radiation::m_lon_cons = -98.555183

private

m_lsm_fluxes

amrex::MultiFab* Radiation::m_lsm_fluxes = nullptr

private

m_lsm_zenith

amrex::MultiFab* Radiation::m_lsm_zenith = nullptr

private

m_moist

bool Radiation::m_moist = false

private

m_mol_weight_gas

const std::vector<amrex::Real> Radiation::m_mol_weight_gas

private

Initial value:

= {18.01528, 44.00950, 47.9982, 44.0128,
                                                       28.01010, 16.04246, 31.9980, 28.0134}

m_n2ovmr

amrex::Real Radiation::m_n2ovmr = 323.141e-9

private

m_n2vmr

amrex::Real Radiation::m_n2vmr = 0.7906

private

m_ncol

int Radiation::m_ncol

private

m_ngas

int Radiation::m_ngas = 8

private

m_nlay

int Radiation::m_nlay

private

m_nlwbands

int Radiation::m_nlwbands = 16

private

m_nlwgpts

int Radiation::m_nlwgpts = 128

private

m_nswbands

int Radiation::m_nswbands = 14

private

m_nswgpts

int Radiation::m_nswgpts = 112

private

m_o2vmr

amrex::Real Radiation::m_o2vmr = 0.209448

private

m_o3_size

int Radiation::m_o3_size

private

m_o3vmr

amrex::Vector<amrex::Real> Radiation::m_o3vmr

private

m_orbital_day

int Radiation::m_orbital_day = -9999

private

m_orbital_eccen

amrex::Real Radiation::m_orbital_eccen = -9999.

private

m_orbital_mon

int Radiation::m_orbital_mon = -9999

private

m_orbital_mvelp

amrex::Real Radiation::m_orbital_mvelp = -9999.

private

m_orbital_obliq

amrex::Real Radiation::m_orbital_obliq = -9999.

private

m_orbital_sec

int Radiation::m_orbital_sec = -9999

private

m_orbital_year

int Radiation::m_orbital_year = -9999

private

m_qheating_rates

amrex::MultiFab* Radiation::m_qheating_rates = nullptr

private

m_rad_freq_in_steps

int Radiation::m_rad_freq_in_steps = 1

private

m_rad_write_fluxes

bool Radiation::m_rad_write_fluxes = false

private

m_step

int Radiation::m_step

private

m_time

amrex::Real Radiation::m_time

private

m_update_rad

bool Radiation::m_update_rad = false

private

Referenced by rad_run_impl().

m_z_phys

amrex::MultiFab* Radiation::m_z_phys = nullptr

private

mu0

real1d Radiation::mu0

private

o3_lay

real1d Radiation::o3_lay

private

p_del

real2d Radiation::p_del

private

p_lay

real2d Radiation::p_lay

private

p_lev

real2d Radiation::p_lev

private

qc_lay

real2d Radiation::qc_lay

private

qi_lay

real2d Radiation::qi_lay

private

qv_lay

real2d Radiation::qv_lay

private

r_lay

real2d Radiation::r_lay

private

rrtmgp_cloud_optics_file_lw

std::string Radiation::rrtmgp_cloud_optics_file_lw

private

rrtmgp_cloud_optics_file_sw

std::string Radiation::rrtmgp_cloud_optics_file_sw

private

rrtmgp_cloud_optics_lw

std::string Radiation::rrtmgp_cloud_optics_lw = "rrtmgp-cloud-optics-coeffs-lw.nc"

private

rrtmgp_cloud_optics_sw

std::string Radiation::rrtmgp_cloud_optics_sw = "rrtmgp-cloud-optics-coeffs-sw.nc"

private

rrtmgp_coeffs_file_lw

std::string Radiation::rrtmgp_coeffs_file_lw

private

rrtmgp_coeffs_file_sw

std::string Radiation::rrtmgp_coeffs_file_sw

private

rrtmgp_coeffs_lw

std::string Radiation::rrtmgp_coeffs_lw = "rrtmgp-data-lw-g224-2018-12-04.nc"

private

rrtmgp_coeffs_sw

std::string Radiation::rrtmgp_coeffs_sw = "rrtmgp-data-sw-g224-2018-12-04.nc"

private

rrtmgp_file_path

std::string Radiation::rrtmgp_file_path = "."

private

sfc_alb_dif

real2d Radiation::sfc_alb_dif

private

sfc_alb_dif_nir

real1d Radiation::sfc_alb_dif_nir

private

sfc_alb_dif_vis

real1d Radiation::sfc_alb_dif_vis

private

sfc_alb_dir

real2d Radiation::sfc_alb_dir

private

sfc_alb_dir_nir

real1d Radiation::sfc_alb_dir_nir

private

sfc_alb_dir_vis

real1d Radiation::sfc_alb_dir_vis

private

sfc_flux_dif_nir

real1d Radiation::sfc_flux_dif_nir

private

sfc_flux_dif_vis

real1d Radiation::sfc_flux_dif_vis

private

sfc_flux_dir_nir

real1d Radiation::sfc_flux_dir_nir

private

sfc_flux_dir_vis

real1d Radiation::sfc_flux_dir_vis

private

sw_bnd_flux_dif

real3d Radiation::sw_bnd_flux_dif

private

sw_bnd_flux_dir

real3d Radiation::sw_bnd_flux_dir

private

sw_bnd_flux_dn

real3d Radiation::sw_bnd_flux_dn

private

sw_bnd_flux_up

real3d Radiation::sw_bnd_flux_up

private

sw_clnclrsky_flux_dn

real2d Radiation::sw_clnclrsky_flux_dn

private

sw_clnclrsky_flux_dn_dir

real2d Radiation::sw_clnclrsky_flux_dn_dir

private

sw_clnclrsky_flux_up

real2d Radiation::sw_clnclrsky_flux_up

private

sw_clnsky_flux_dn

real2d Radiation::sw_clnsky_flux_dn

private

sw_clnsky_flux_dn_dir

real2d Radiation::sw_clnsky_flux_dn_dir

private

sw_clnsky_flux_up

real2d Radiation::sw_clnsky_flux_up

private

sw_clrsky_flux_dn

real2d Radiation::sw_clrsky_flux_dn

private

sw_clrsky_flux_dn_dir

real2d Radiation::sw_clrsky_flux_dn_dir

private

sw_clrsky_flux_up

real2d Radiation::sw_clrsky_flux_up

private

sw_flux_dn

real2d Radiation::sw_flux_dn

private

sw_flux_dn_dir

real2d Radiation::sw_flux_dn_dir

private

sw_flux_up

real2d Radiation::sw_flux_up

private

sw_heating

real2d Radiation::sw_heating

private

t_lay

real2d Radiation::t_lay

private

t_lev

real2d Radiation::t_lev

private

tmp2d

real2d Radiation::tmp2d

private

z_del

real2d Radiation::z_del

private

---

The documentation for this class was generated from the following files:

• Source/Radiation/ERF_Radiation.H
• Source/Radiation/ERF_Radiation.cpp