Skip to content

Commit

Permalink
Merge branch 'LatLon' of github.com:wiersema1/ERF into LatLon
Browse files Browse the repository at this point in the history
  • Loading branch information
@wiersema1
wiersema1 committed Apr 24, 2024
2 parents d726581 + 28dfa6c commit b5c3c50
Show file tree
Hide file tree
Showing 5 changed files with 109 additions and 89 deletions.
2 changes: 1 addition & 1 deletion Source/ERF_make_new_arrays.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,7 @@ ERF::init_stuff (int lev, const BoxArray& ba, const DistributionMapping& dm,
//*********************************************************
// Radiation heating source terms
//*********************************************************
qheating_rates[lev] = std::make_unique<MultiFab>(ba, dm, 1, ngrow_state);
qheating_rates[lev] = std::make_unique<MultiFab>(ba, dm, 2, ngrow_state);
qheating_rates[lev]->setVal(0.);
#endif
}
Expand Down
107 changes: 60 additions & 47 deletions Source/Radiation/Radiation.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,7 @@ void Radiation::initialize (const MultiFab& cons_in,
for ( MFIter mfi(cons_in, false); mfi.isValid(); ++mfi) {
auto states_array = cons_in.array(mfi);
auto qt_array = (qmoist[0]) ? qmoist[0]->array(mfi) : Array4<Real> {};
auto qv_array = (qmoist[1]) ? qmoist[1]->array(mfi) : Array4<Real> {};
auto qc_array = (qmoist[2]) ? qmoist[2]->array(mfi) : Array4<Real> {};
auto qi_array = (qmoist.size()>=8) ? qmoist[3]->array(mfi) : Array4<Real> {};

Expand All @@ -188,26 +189,32 @@ void Radiation::initialize (const MultiFab& cons_in,
{
auto icol = j*nx+i+1;
auto ilev = k+1;
Real qv = (qv_array) ? qv_array(i,j,k): 0.0;
qt(icol,ilev) = (qt_array) ? qt_array(i,j,k): 0.0;
qc(icol,ilev) = (qc_array) ? qc_array(i,j,k): 0.0;
qi(icol,ilev) = (qi_array) ? qi_array(i,j,k): 0.0;
qn(icol,ilev) = qc(icol,ilev) + qi(icol,ilev);
tmid(icol,ilev) = getTgivenRandRTh(states_array(i,j,k,Rho_comp),states_array(i,j,k,RhoTheta_comp));
pmid(icol,ilev) = getPgivenRTh(states_array(i,j,k,RhoTheta_comp))/1.0e3;
tmid(icol,ilev) = getTgivenRandRTh(states_array(i,j,k,Rho_comp),states_array(i,j,k,RhoTheta_comp),qv);
// NOTE: RRTMGP code expects pressure in mb so we convert it here
pmid(icol,ilev) = getPgivenRTh(states_array(i,j,k,RhoTheta_comp),qv) * 1.0e-2;
});
}

parallel_for(SimpleBounds<2>(ncol, nlev+1), YAKL_LAMBDA (int icol, int ilev)
{
if (ilev == 1) {
pint(icol, 1) = 2.*pmid(icol, 2) - pmid(icol, 1);
tint(icol, 1) = 2.*tmid(icol, 2) - tmid(icol, 1);
//pint(icol, 1) = 2.*pmid(icol, 2) - pmid(icol, 1);
//tint(icol, 1) = 2.*tmid(icol, 2) - tmid(icol, 1);
pint(icol, 1) = -0.5*pmid(icol, 2) + 1.5*pmid(icol, 1);
tint(icol, 1) = -0.5*tmid(icol, 2) + 1.5*tmid(icol, 1);
} else if (ilev <= nlev) {
pint(icol, ilev) = 0.5*(pmid(icol, ilev-1) + pmid(icol, ilev));
tint(icol, ilev) = 0.5*(tmid(icol, ilev-1) + tmid(icol, ilev));
} else {
pint(icol, nlev+1) = 2.*pmid(icol, nlev-1) - pmid(icol, nlev);
tint(icol, nlev+1) = 2.*tmid(icol, nlev-1) - tmid(icol, nlev);
//pint(icol, nlev+1) = 2.*pmid(icol, nlev-1) - pmid(icol, nlev);
//tint(icol, nlev+1) = 2.*tmid(icol, nlev-1) - tmid(icol, nlev);
pint(icol, nlev+1) = -0.5*pmid(icol, nlev-1) + 1.5*pmid(icol, nlev);
tint(icol, nlev+1) = -0.5*tmid(icol, nlev-1) + 1.5*tmid(icol, nlev);
}
});

Expand Down Expand Up @@ -252,42 +259,42 @@ void Radiation::initialize (const MultiFab& cons_in,
// run radiation model
void Radiation::run ()
{
// Temporary variable for heating rate output
real2d hr("hr", ncol, nlev);

// Cosine solar zenith angle for all columns in chunk
real1d coszrs("coszrs", ncol);

// Pointers to fields on the physics buffer
real2d cld("cld", ncol, nlev), cldfsnow("cldfsnow", ncol, nlev),
iclwp("iclwp", ncol, nlev), iciwp("iciwp", ncol, nlev),
icswp("icswp", ncol, nlev), dei("dei", ncol, nlev),
des("des", ncol, nlev), lambdac("lambdac", ncol, nlev),
mu("mu", ncol, nlev), rei("rei", ncol, nlev), rel("rel", ncol, nlev);
real2d cld("cld" , ncol, nlev), cldfsnow("cldfsnow", ncol, nlev),
iclwp("iclwp", ncol, nlev), iciwp("iciwp" , ncol, nlev),
icswp("icswp", ncol, nlev), dei("dei" , ncol, nlev),
des("des" , ncol, nlev), lambdac("lambdac" , ncol, nlev),
mu("mu" , ncol, nlev), rei("rei" , ncol, nlev),
rel("rel" , ncol, nlev);

// Cloud, snow, and aerosol optical properties
real3d cld_tau_gpt_sw("cld_tau_gpt_sw", ncol, nlev, nswgpts),
cld_ssa_gpt_sw("cld_ssa_gpt_sw", ncol, nlev, nswgpts),
cld_asm_gpt_sw("cld_asm_gpt_sw", ncol, nlev, nswgpts);
cld_ssa_gpt_sw("cld_ssa_gpt_sw", ncol, nlev, nswgpts),
cld_asm_gpt_sw("cld_asm_gpt_sw", ncol, nlev, nswgpts);

real3d cld_tau_bnd_sw("cld_tau_bnd_sw", ncol, nlev, nswbands),
cld_ssa_bnd_sw("cld_ssa_bnd_sw", ncol, nlev, nswbands),
cld_asm_bnd_sw("cld_asm_bnd_sw", ncol, nlev, nswbands);
cld_ssa_bnd_sw("cld_ssa_bnd_sw", ncol, nlev, nswbands),
cld_asm_bnd_sw("cld_asm_bnd_sw", ncol, nlev, nswbands);

real3d aer_tau_bnd_sw("aer_tau_bnd_sw", ncol, nlev, nswbands),
aer_ssa_bnd_sw("aer_ssa_bnd_sw", ncol, nlev, nswbands),
aer_asm_bnd_sw("aer_asm_bnd_sw", ncol, nlev, nswbands);
aer_ssa_bnd_sw("aer_ssa_bnd_sw", ncol, nlev, nswbands),
aer_asm_bnd_sw("aer_asm_bnd_sw", ncol, nlev, nswbands);

real3d cld_tau_bnd_lw("cld_tau_bnd_lw", ncol, nlev, nlwbands),
aer_tau_bnd_lw("aer_tau_bnd_lw", ncol, nlev, nlwbands);
aer_tau_bnd_lw("aer_tau_bnd_lw", ncol, nlev, nlwbands);

real3d cld_tau_gpt_lw("cld_tau_gpt_lw", ncol, nlev, nlwgpts);

// NOTE: these are diagnostic only
real3d liq_tau_bnd_sw("liq_tau_bnd_sw", ncol, nlev, nswbands),
ice_tau_bnd_sw("ice_tau_bnd_sw", ncol, nlev, nswbands),
snw_tau_bnd_sw("snw_tau_bnd_sw", ncol, nlev, nswbands);
ice_tau_bnd_sw("ice_tau_bnd_sw", ncol, nlev, nswbands),
snw_tau_bnd_sw("snw_tau_bnd_sw", ncol, nlev, nswbands);
real3d liq_tau_bnd_lw("liq_tau_bnd_lw", ncol, nlev, nlwbands),
ice_tau_bnd_lw("ice_tau_bnd_lw", ncol, nlev, nlwbands),
snw_tau_bnd_lw("snw_tau_bnd_lw", ncol, nlev, nlwbands);
ice_tau_bnd_lw("ice_tau_bnd_lw", ncol, nlev, nlwbands),
snw_tau_bnd_lw("snw_tau_bnd_lw", ncol, nlev, nlwbands);

// Gas volume mixing ratios
real3d gas_vmr("gas_vmr", ngas, ncol, nlev);
Expand Down Expand Up @@ -512,32 +519,20 @@ void Radiation::run ()
}
} // dolw

// Compute heating rate for dtheta/dt
parallel_for(SimpleBounds<2>(ncol, nlev), YAKL_LAMBDA (int icol, int ilev)
{
// TODO: CHECK THIS UNIT CONVERSION!!
hr(icol,ilev) = (qrs(icol,ilev) + qrl(icol,ilev)) / Cp_d * (1.e5 / std::pow(pmid(icol,ilev), R_d/Cp_d));
});

// Populate theta source from hr
// Populate source term for theta dycore variable
for (MFIter mfi(*(qrad_src)); mfi.isValid(); ++mfi) {
auto qrad_src_array = qrad_src->array(mfi);
const auto& box3d = mfi.tilebox();
auto nx = box3d.length(0);
amrex::ParallelFor(box3d, [=] AMREX_GPU_DEVICE (int i, int j, int k)
{
// Map (col,lev) to (i,j,k)
auto icol = j*nx+i+1;
auto ilev = k+1;
qrad_src_array(i,j,k) = hr(icol,ilev);
});
}

// convert radiative heating rates to Q*dp for energy conservation
if (conserve_energy) {
parallel_for(SimpleBounds<2>(ncol, nlev), YAKL_LAMBDA (int icol, int ilev)
{
qrs(icol,ilev) = qrs(icol,ilev)*pdel(icol,ilev);
qrl(icol,ilev) = qrl(icol,ilev)*pdel(icol,ilev);
// SW and LW sources
qrad_src_array(i,j,k,0) = qrs(icol,ilev);
qrad_src_array(i,j,k,1) = qrl(icol,ilev);
});
}
}
Expand Down Expand Up @@ -714,7 +709,7 @@ void Radiation::radiation_driver_lw (int ncol, int nlev,
FluxesByband& fluxes_allsky, const real2d& qrl, const real2d& qrlc)
{
real3d cld_tau_gpt_rad("cld_tau_gpt_rad", ncol, nlev+1, nlwgpts);
real3d aer_tau_bnd_rad("aer_tau_bnd-rad", ncol, nlev+1, nlwgpts);
real3d aer_tau_bnd_rad("aer_tau_bnd_rad", ncol, nlev+1, nlwgpts);

// Surface emissivity needed for longwave
real2d surface_emissivity("surface_emissivity", nlwbands, ncol);
Expand All @@ -734,7 +729,7 @@ void Radiation::radiation_driver_lw (int ncol, int nlev,
yakl::memset(cld_tau_gpt_rad, 0.);
yakl::memset(aer_tau_bnd_rad, 0.);

parallel_for(SimpleBounds<3>(ncol, nlev, nswgpts), YAKL_LAMBDA (int icol, int ilev, int igpt)
parallel_for(SimpleBounds<3>(ncol, nlev, nlwgpts), YAKL_LAMBDA (int icol, int ilev, int igpt)
{
cld_tau_gpt_rad(icol,ilev,igpt) = cld_tau_gpt(icol,ilev,igpt);
aer_tau_bnd_rad(icol,ilev,igpt) = aer_tau_bnd(icol,ilev,igpt);
Expand Down Expand Up @@ -894,11 +889,29 @@ void Radiation::calculate_heating_rate (const real2d& flux_up,
const real2d& pint,
const real2d& heating_rate)
{
// NOTE: The pressure is in [mb] for RRTMGP to use.
// The fluxes are in [W/m^2] and gravity is [m/s^2].
// We need to convert pressure from [mb] -> [Pa] and divide by Cp [J/kg*K]
// The heating rate is {dF/dP * g / Cp} with units [K/s]
real1d heatfac("heatfac",1);
yakl::memset(heatfac, 1.0e-2/Cp_d);
parallel_for(SimpleBounds<2>(ncol, nlev), YAKL_LAMBDA (int icol, int ilev)
{
heating_rate(icol,ilev) = ( (flux_up(icol,ilev+1) - flux_dn(icol,ilev+1))
- (flux_up(icol,ilev ) - flux_dn(icol,ilev )) )
* CONST_GRAV/(pint(icol,ilev+1)-pint(icol,ilev));
heating_rate(icol,ilev) = heatfac(1) * ( (flux_up(icol,ilev+1) - flux_dn(icol,ilev+1))
- (flux_up(icol,ilev ) - flux_dn(icol,ilev )) )
* CONST_GRAV/(pint(icol,ilev+1)-pint(icol,ilev));
/*
if (icol==1) {
amrex::Print() << "HR: " << ilev << ' '
<< heating_rate(icol,ilev) << ' '
<< (flux_up(icol,ilev+1) - flux_dn(icol,ilev+1)) << ' '
<< (flux_up(icol,ilev ) - flux_dn(icol,ilev )) << ' '
<< (flux_up(icol,ilev+1) - flux_dn(icol,ilev+1))
- (flux_up(icol,ilev ) - flux_dn(icol,ilev )) << ' '
<< (pint(icol,ilev+1)-pint(icol,ilev)) << ' '
<< CONST_GRAV << "\n";
}
*/
});
}

Expand Down
26 changes: 14 additions & 12 deletions Source/Radiation/Run_longwave_rrtmgp.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -98,12 +98,13 @@ void Rrtmgp::run_longwave_rrtmgp (int ngas, int ncol, int nlay,

// Do the clearsky calculation before adding in clouds
FluxesByband fluxes_clrsky;
fluxes_clrsky.flux_up = real2d("clrsky_flux_up", ncol, nlay+1); // clrsky_flux_up;
fluxes_clrsky.flux_dn = real2d("clrsky_flux_up", ncol, nlay+1); //clrsky_flux_dn;
fluxes_clrsky.flux_net = real2d("clrsky_flux_up", ncol, nlay+1); //clrsky_flux_net;
fluxes_clrsky.bnd_flux_up = real3d("clrsky_flux_up", ncol, nlay+1, nlwbands); //clrsky_bnd_flux_up;
fluxes_clrsky.bnd_flux_dn = real3d("clrsky_flux_up", ncol, nlay+1, nlwbands); //clrsky_bnd_flux_dn;
fluxes_clrsky.bnd_flux_net = real3d("clrsky_flux_up", ncol, nlay+1, nlwbands); //clrsky_bnd_flux_net;
fluxes_clrsky.flux_up = real2d("clrsky_flux_up" , ncol, nlay+1); // clrsky_flux_up;
fluxes_clrsky.flux_dn = real2d("clrsky_flux_dn" , ncol, nlay+1); //clrsky_flux_dn;
fluxes_clrsky.flux_net = real2d("clrsky_flux_net", ncol, nlay+1); //clrsky_flux_net;
fluxes_clrsky.bnd_flux_up = real3d("clrsky_bnd_flux_up" , ncol, nlay+1, nlwbands); //clrsky_bnd_flux_up;
fluxes_clrsky.bnd_flux_dn = real3d("clrsky_bnd_flux_dn" , ncol, nlay+1, nlwbands); //clrsky_bnd_flux_dn;
fluxes_clrsky.bnd_flux_net = real3d("clrsky_bnd_flux_net", ncol, nlay+1, nlwbands); //clrsky_bnd_flux_net;

rte_lw(max_gauss_pts, gauss_Ds, gauss_wts, combined_optics, top_at_1, lw_sources, emis_sfc, fluxes_clrsky);

// Copy fluxes back out of FluxesByband object
Expand All @@ -125,12 +126,13 @@ void Rrtmgp::run_longwave_rrtmgp (int ngas, int ncol, int nlay,

// Call LW flux driver
FluxesByband fluxes_allsky;
fluxes_allsky.flux_up = real2d("flux_up", ncol, nlay+1); //allsky_flux_up;
fluxes_allsky.flux_dn = real2d("flux_dn", ncol, nlay+1); //allsky_flux_dn;
fluxes_allsky.flux_net = real2d("flux_net", ncol, nlay+1); //allsky_flux_net;
fluxes_allsky.bnd_flux_up = real3d("flux_up", ncol, nlay+1, nlwbands); //allsky_bnd_flux_up;
fluxes_allsky.bnd_flux_dn = real3d("flux_dn", ncol, nlay+1, nlwbands); //allsky_bnd_flux_dn;
fluxes_allsky.bnd_flux_net = real3d("flux_net", ncol, nlay+1, nlwbands); //allsky_bnd_flux_net;
fluxes_allsky.flux_up = real2d("allsky_flux_up" , ncol, nlay+1); //allsky_flux_up;
fluxes_allsky.flux_dn = real2d("allsky_flux_dn" , ncol, nlay+1); //allsky_flux_dn;
fluxes_allsky.flux_net = real2d("allsky_flux_net", ncol, nlay+1); //allsky_flux_net;
fluxes_allsky.bnd_flux_up = real3d("allsky_bnd_flux_up" , ncol, nlay+1, nlwbands); //allsky_bnd_flux_up;
fluxes_allsky.bnd_flux_dn = real3d("allsky_bnd_flux_dn" , ncol, nlay+1, nlwbands); //allsky_bnd_flux_dn;
fluxes_allsky.bnd_flux_net = real3d("allsky_bnd_flux_net", ncol, nlay+1, nlwbands); //allsky_bnd_flux_net;

rte_lw(max_gauss_pts, gauss_Ds, gauss_wts, combined_optics, top_at_1, lw_sources, emis_sfc, fluxes_allsky);

// Copy fluxes back out of FluxesByband object
Expand Down
60 changes: 32 additions & 28 deletions Source/Radiation/Run_shortwave_rrtmgp.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -34,16 +34,18 @@ void Rrtmgp::run_shortwave_rrtmgp (int ngas, int ncol, int nlay,
bool1d top_at_1_g("top_at_1_g",1);
boolHost1d top_at_1_h("top_at_1_h",1);
bool top_at_1;
parallel_for(SimpleBounds<1>(1), YAKL_LAMBDA (int icol) { // HACK: Single loop kernel is not efficient
top_at_1_g(1) = pmid(1, 1) < pmid (1, 2);
});
top_at_1_g.deep_copy_to(top_at_1_h);
top_at_1 = top_at_1_h(1);
real2d toa_flux("toa_flux", ncol, nswgpts);
k_dist_sw.gas_optics(ncol, nlay, &top_at_1, pmid, pint, tmid, gas_concs, combined_optics, toa_flux); // TODO: top_at_1 is not a valid address and this doesn't match longwave call
k_dist_sw.gas_optics(ncol, nlay, top_at_1, pmid, pint, tmid, gas_concs, combined_optics, toa_flux);

// Apply TOA flux scaling
parallel_for(SimpleBounds<2>(nswgpts,ncol), YAKL_LAMBDA (int igpt, int icol) {
toa_flux(icol, igpt) = tsi_scaling * toa_flux(icol,igpt);
top_at_1_g(1) = pmid(1, 1) < pmid (1, 2);
});
top_at_1_g.deep_copy_to(top_at_1_h);
top_at_1 = top_at_1_h(1);

// Add in aerosol
// TODO: should we avoid allocating here?
Expand Down Expand Up @@ -74,25 +76,26 @@ void Rrtmgp::run_shortwave_rrtmgp (int ngas, int ncol, int nlay,

// Do the clearsky calculation before adding in clouds
FluxesByband fluxes_clrsky;
fluxes_clrsky.flux_up = real2d("clrsky_flux_up", ncol, nlay+1); // clrsky_flux_up;
fluxes_clrsky.flux_dn = real2d("clrsky_flux_up", ncol, nlay+1); //clrsky_flux_dn;
fluxes_clrsky.flux_dn_dir = real2d("clrsky_flux_up", ncol, nlay+1); //clrsky_flux_dn_dir;
fluxes_clrsky.flux_net = real2d("clrsky_flux_up", ncol, nlay+1); //clrsky_flux_net;
fluxes_clrsky.bnd_flux_up = real3d("clrsky_flux_up", ncol, nlay+1, nswbands); //clrsky_bnd_flux_up;
fluxes_clrsky.bnd_flux_dn = real3d("clrsky_flux_up", ncol, nlay+1, nswbands); //clrsky_bnd_flux_dn;
fluxes_clrsky.bnd_flux_dn_dir = real3d("clrsky_flux_up", ncol, nlay+1, nswbands); //clrsky_bnd_flux_dn_dir;
fluxes_clrsky.bnd_flux_net = real3d("clrsky_flux_up", ncol, nlay+1, nswbands); //clrsky_bnd_flux_net;
fluxes_clrsky.flux_up = real2d("clrsky_flux_up" , ncol, nlay+1); // clrsky_flux_up;
fluxes_clrsky.flux_dn = real2d("clrsky_flux_nd" , ncol, nlay+1); //clrsky_flux_dn;
fluxes_clrsky.flux_net = real2d("clrsky_flux_net", ncol, nlay+1); //clrsky_flux_net;
fluxes_clrsky.flux_dn_dir = real2d("clrsky_flux_dn_dir", ncol, nlay+1); //clrsky_flux_dn_dir;
fluxes_clrsky.bnd_flux_up = real3d("clrsky_bnd_flux_up" , ncol, nlay+1, nswbands); //clrsky_bnd_flux_up;
fluxes_clrsky.bnd_flux_dn = real3d("clrsky_bnd_flux_dn" , ncol, nlay+1, nswbands); //clrsky_bnd_flux_dn;
fluxes_clrsky.bnd_flux_net = real3d("clrsky_bnd_flux_net", ncol, nlay+1, nswbands); //clrsky_bnd_flux_net;
fluxes_clrsky.bnd_flux_dn_dir = real3d("clrsky_bnd_flux_dn_dir", ncol, nlay+1, nswbands); //clrsky_bnd_flux_dn_dir;

rte_sw(combined_optics, top_at_1, coszrs, toa_flux, albedo_dir, albedo_dif, fluxes_clrsky);

// Copy fluxes back out of FluxesByband object
fluxes_clrsky.flux_up.deep_copy_to(clrsky_flux_up);
fluxes_clrsky.flux_dn.deep_copy_to(clrsky_flux_dn);
fluxes_clrsky.flux_dn_dir.deep_copy_to(clrsky_flux_dn_dir);
fluxes_clrsky.flux_up.deep_copy_to (clrsky_flux_up);
fluxes_clrsky.flux_dn.deep_copy_to (clrsky_flux_dn);
fluxes_clrsky.flux_net.deep_copy_to(clrsky_flux_net);
fluxes_clrsky.bnd_flux_up.deep_copy_to(clrsky_bnd_flux_up);
fluxes_clrsky.bnd_flux_dn.deep_copy_to(clrsky_bnd_flux_dn);
fluxes_clrsky.bnd_flux_dn_dir.deep_copy_to(clrsky_bnd_flux_dn_dir);
fluxes_clrsky.flux_dn_dir.deep_copy_to(clrsky_flux_dn_dir);
fluxes_clrsky.bnd_flux_up.deep_copy_to (clrsky_bnd_flux_up);
fluxes_clrsky.bnd_flux_dn.deep_copy_to (clrsky_bnd_flux_dn);
fluxes_clrsky.bnd_flux_net.deep_copy_to(clrsky_bnd_flux_net);
fluxes_clrsky.bnd_flux_dn_dir.deep_copy_to(clrsky_bnd_flux_dn_dir);

// Add in clouds
OpticalProps2str cloud_optics;
Expand All @@ -110,25 +113,26 @@ void Rrtmgp::run_shortwave_rrtmgp (int ngas, int ncol, int nlay,

// Call SW flux driver
FluxesByband fluxes_allsky;
fluxes_allsky.flux_up = real2d("allsky_flux_up", ncol, nlay+1); // allsky_flux_up;
fluxes_allsky.flux_dn = real2d("allsky_flux_up", ncol, nlay+1); //allsky_flux_dn;
fluxes_allsky.flux_dn_dir = real2d("allsky_flux_up", ncol, nlay+1); //allsky_flux_dn_dir;
fluxes_allsky.flux_net = real2d("allsky_flux_up", ncol, nlay+1); //allsky_flux_net;
fluxes_allsky.bnd_flux_up = real3d("allsky_flux_up", ncol, nlay+1, nswbands); //allsky_bnd_flux_up;
fluxes_allsky.bnd_flux_dn = real3d("allsky_flux_up", ncol, nlay+1, nswbands); //allsky_bnd_flux_dn;
fluxes_allsky.bnd_flux_dn_dir = real3d("allsky_flux_up", ncol, nlay+1, nswbands); //allsky_bnd_flux_dn_dir;
fluxes_allsky.bnd_flux_net = real3d("allsky_flux_up", ncol, nlay+1, nswbands); //allsky_bnd_flux_net;
fluxes_allsky.flux_up = real2d("allsky_flux_up" , ncol, nlay+1); //allsky_flux_up;
fluxes_allsky.flux_dn = real2d("allsky_flux_dn" , ncol, nlay+1); //allsky_flux_dn;
fluxes_allsky.flux_net = real2d("allsky_flux_net", ncol, nlay+1); //allsky_flux_net;
fluxes_allsky.flux_dn_dir = real2d("allsky_flux_dn_dir", ncol, nlay+1); //allsky_flux_dn_dir;
fluxes_allsky.bnd_flux_up = real3d("allsky_bnd_flux_up" , ncol, nlay+1, nswbands); //allsky_bnd_flux_up;
fluxes_allsky.bnd_flux_dn = real3d("allsky_bnd_flux_dn" , ncol, nlay+1, nswbands); //allsky_bnd_flux_dn;
fluxes_allsky.bnd_flux_net = real3d("allsky_bnd_flux_net", ncol, nlay+1, nswbands); //allsky_bnd_flux_net;
fluxes_allsky.bnd_flux_dn_dir = real3d("allsky_bnd_flux_dn_dir", ncol, nlay+1, nswbands); //allsky_bnd_flux_dn_dir;

rte_sw(combined_optics, top_at_1, coszrs, toa_flux, albedo_dir, albedo_dif, fluxes_allsky);

// Copy fluxes back out of FluxesByband object
fluxes_allsky.flux_up.deep_copy_to(allsky_flux_up);
fluxes_allsky.flux_dn.deep_copy_to(allsky_flux_dn);
fluxes_allsky.flux_dn_dir.deep_copy_to(allsky_flux_dn_dir);
fluxes_allsky.flux_net.deep_copy_to(allsky_flux_net);
fluxes_allsky.flux_dn_dir.deep_copy_to(allsky_flux_dn_dir);
fluxes_allsky.bnd_flux_up.deep_copy_to(allsky_bnd_flux_up);
fluxes_allsky.bnd_flux_dn.deep_copy_to(allsky_bnd_flux_dn);
fluxes_allsky.bnd_flux_dn_dir.deep_copy_to(allsky_bnd_flux_dn_dir);
fluxes_allsky.bnd_flux_net.deep_copy_to(allsky_bnd_flux_net);
fluxes_allsky.bnd_flux_dn_dir.deep_copy_to(allsky_bnd_flux_dn_dir);
yakl::fence();
}

Loading

0 comments on commit b5c3c50

Please sign in to comment.