#include <ERF_ShocInterface.H>

Collaboration diagram for SHOCInterface::SHOCPreprocess:

[legend]

Public Member Functions
	SHOCPreprocess ()=default

KOKKOS_INLINE_FUNCTION void	operator() (const Kokkos::TeamPolicy< KT::ExeSpace >::member_type &team) const

void	set_variables (const int ncol_, const int nlev_, const Real z_surf_, const view_2d_const &T_mid_, const view_2d_const &p_mid_, const view_2d_const &p_int_, const view_2d_const &pseudo_density_, const view_2d_const &omega_, const view_1d_const &phis_, const view_1d_const &surf_sens_flux_, const view_1d_const &surf_evap_, const sview_2d_const &surf_mom_flux_, const view_3d_strided &qtracers_, const view_2d &qv_, const view_2d_const &qc_, const view_2d &qc_copy_, const view_2d &tke_, const view_2d &tke_copy_, const view_2d &z_mid_, const view_2d &z_int_, const view_2d &dse_, const view_2d &rrho_, const view_2d &rrho_i_, const view_2d &thv_, const view_2d &dz_, const view_2d &zt_grid_, const view_2d &zi_grid_, const view_1d &wpthlp_sfc_, const view_1d &wprtp_sfc_, const view_1d &upwp_sfc_, const view_1d &vpwp_sfc_, const view_2d &wtracer_sfc_, const view_2d &wm_zt_, const view_2d &inv_exner_, const view_2d &thlm_, const view_2d &qw_, const view_2d &cldfrac_liq_, const view_2d &cldfrac_liq_prev_)

Public Attributes
int	ncol

int	nlev

Real	z_surf

view_2d_const	T_mid

view_2d_const	p_mid

view_2d_const	p_int

view_2d_const	pseudo_density

view_2d_const	omega

view_1d_const	phis

view_1d_const	surf_sens_flux

view_1d_const	surf_evap

sview_2d_const	surf_mom_flux

view_3d_strided	qtracers

view_2d	qv

view_2d_const	qc

view_2d	qc_copy

view_2d	z_mid

view_2d	z_int

view_2d	shoc_s

view_2d	tke

view_2d	tke_copy

view_2d	rrho

view_2d	rrho_i

view_2d	thv

view_2d	dz

view_2d	zt_grid

view_2d	zi_grid

view_1d	wpthlp_sfc

view_1d	wprtp_sfc

view_1d	upwp_sfc

view_1d	vpwp_sfc

view_2d	wtracer_sfc

view_2d	wm_zt

view_2d	inv_exner

view_2d	thlm

view_2d	qw

view_2d	cloud_frac

view_2d	cldfrac_liq

view_2d	cldfrac_liq_prev

Constructor & Destructor Documentation

◆ SHOCPreprocess()

SHOCInterface::SHOCPreprocess::SHOCPreprocess ( )

default

Member Function Documentation

◆ operator()()

KOKKOS_INLINE_FUNCTION void SHOCInterface::SHOCPreprocess::operator() ( const Kokkos::TeamPolicy< KT::ExeSpace >::member_type & team ) const

inline

         {
             const int i = team.league_rank();
  
             const Real zvir    = C::ZVIR;
             const Real cpair   = C::Cpair;
             const Real ggr     = C::gravit;
             const Real inv_ggr = 1/ggr;
             const Real mintke  = SC::mintke;
  
             const int nlev_packs = ekat::npack<Spack>(nlev);
  
             Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (const Int& k)
             {
                 cldfrac_liq_prev(i,k)=cldfrac_liq(i,k);
  
                 // Inverse of Exner. In non-rel builds, assert that exner != 0 when in range before computing.
                 const Spack exner = PF::exner_function(p_mid(i,k));
                 const Smask nonzero = (exner != 0);
                 EKAT_KERNEL_ASSERT((nonzero || !(ekat::range<IntSmallPack>(k*Spack::n) < nlev)).all());
                 inv_exner(i,k).set(nonzero, 1/exner);
  
                 tke(i,k) = ekat::max(mintke, tke(i,k));
  
                 // Tracers are updated as a group. The tracers tke and qc act as separate inputs to shoc_main()
                 // and are therefore updated differently to the tracers group's monolithic field. Here, we make
                 // a copy if each of these tracers and pass to shoc_main() so that changes to the tracer group
                 // does not alter tke or qc  values. Then during post processing, we copy back correct values of
                 // tke and qc to tracer group in postprocessing.
                 // TODO: remove *_copy views once SHOC can request a subset of tracers.
                 tke_copy(i,k) = tke(i,k);
                 qc_copy(i,k)  = qc(i,k);
  
                 qw(i,k) = qv(i,k) + qc(i,k);
  
                 // Temperature
                 // NOTE: theta_v (thv) is intentionally different from one in HOMME
                 const auto theta_zt = PF::calculate_theta_from_T(T_mid(i,k),p_mid(i,k));
                 thlm(i,k) = PF::calculate_thetal_from_theta(theta_zt,T_mid(i,k),qc(i,k));
                 thv(i,k)  = theta_zt*(1 + zvir*qv(i,k) - qc(i,k));
  
                 // Vertical layer thickness
                 dz(i,k) = PF::calculate_dz(pseudo_density(i,k), p_mid(i,k), T_mid(i,k), qv(i,k));
  
                 rrho(i,k) = inv_ggr*(pseudo_density(i,k)/dz(i,k));
                 wm_zt(i,k) = -1*omega(i,k)/(rrho(i,k)*ggr);
             });
             team.team_barrier();
  
             // Compute vertical layer heights
             const auto dz_s    = ekat::subview(dz,    i);
             const auto z_int_s = ekat::subview(z_int, i);
             const auto z_mid_s = ekat::subview(z_mid, i);
             PF::calculate_z_int(team,nlev,dz_s,z_surf,z_int_s);
             team.team_barrier();
             PF::calculate_z_mid(team,nlev,z_int_s,z_mid_s);
             team.team_barrier();
  
             const int nlevi_v = nlev/Spack::n;
             const int nlevi_p = nlev%Spack::n;
             Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (const Int& k)
             {
                 zt_grid(i,k) = z_mid(i,k) - z_int(i, nlevi_v)[nlevi_p];
                 zi_grid(i,k) = z_int(i,k) - z_int(i, nlevi_v)[nlevi_p];
  
                 // Dry static energy
                 shoc_s(i,k) = PF::calculate_dse(T_mid(i,k),z_mid(i,k),phis(i));
  
                 if (k+1 == nlev_packs) zi_grid(i,nlevi_v)[nlevi_p] = 0;
             });
             team.team_barrier();
  
             const auto zt_grid_s = ekat::subview(zt_grid, i);
             const auto zi_grid_s = ekat::subview(zi_grid, i);
             const auto rrho_s    = ekat::subview(rrho, i);
             const auto rrho_i_s  = ekat::subview(rrho_i, i);
             SHF::linear_interp(team,zt_grid_s,zi_grid_s,rrho_s,rrho_i_s,nlev,nlev+1,0);
             team.team_barrier();
  
             const auto exner_int = PF::exner_function(p_int(i,nlevi_v)[nlevi_p]);
             const auto inv_exner_int_surf = 1/exner_int;
  
             wpthlp_sfc(i) = (surf_sens_flux(i)/(cpair*rrho_i(i,nlevi_v)[nlevi_p]))*inv_exner_int_surf;
             wprtp_sfc(i)  = surf_evap(i)/rrho_i(i,nlevi_v)[nlevi_p];
             upwp_sfc(i) = surf_mom_flux(i,0)/rrho_i(i,nlevi_v)[nlevi_p];
             vpwp_sfc(i) = surf_mom_flux(i,1)/rrho_i(i,nlevi_v)[nlevi_p];
         } // operator

◆ set_variables()

void SHOCInterface::SHOCPreprocess::set_variables	(	const int	ncol_,
		const int	nlev_,
		const Real	z_surf_,
		const view_2d_const &	T_mid_,
		const view_2d_const &	p_mid_,
		const view_2d_const &	p_int_,
		const view_2d_const &	pseudo_density_,
		const view_2d_const &	omega_,
		const view_1d_const &	phis_,
		const view_1d_const &	surf_sens_flux_,
		const view_1d_const &	surf_evap_,
		const sview_2d_const &	surf_mom_flux_,
		const view_3d_strided &	qtracers_,
		const view_2d &	qv_,
		const view_2d_const &	qc_,
		const view_2d &	qc_copy_,
		const view_2d &	tke_,
		const view_2d &	tke_copy_,
		const view_2d &	z_mid_,
		const view_2d &	z_int_,
		const view_2d &	dse_,
		const view_2d &	rrho_,
		const view_2d &	rrho_i_,
		const view_2d &	thv_,
		const view_2d &	dz_,
		const view_2d &	zt_grid_,
		const view_2d &	zi_grid_,
		const view_1d &	wpthlp_sfc_,
		const view_1d &	wprtp_sfc_,
		const view_1d &	upwp_sfc_,
		const view_1d &	vpwp_sfc_,
		const view_2d &	wtracer_sfc_,
		const view_2d &	wm_zt_,
		const view_2d &	inv_exner_,
		const view_2d &	thlm_,
		const view_2d &	qw_,
		const view_2d &	cldfrac_liq_,
		const view_2d &	cldfrac_liq_prev_
	)

inline

         {
             ncol = ncol_;
             nlev = nlev_;
             z_surf = z_surf_;
             // IN
             T_mid = T_mid_;
             p_mid = p_mid_;
             p_int = p_int_;
             pseudo_density = pseudo_density_;
             omega = omega_;
             phis = phis_;
             surf_sens_flux = surf_sens_flux_;
             surf_evap = surf_evap_;
             surf_mom_flux = surf_mom_flux_;
             qv = qv_;
             // OUT
             qtracers = qtracers_;
             qc = qc_;
             qc_copy = qc_copy_;
             shoc_s = dse_;
             tke = tke_;
             tke_copy = tke_copy_;
             z_mid = z_mid_;
             z_int = z_int_;
             rrho = rrho_;
             rrho_i = rrho_i_;
             thv = thv_;
             dz = dz_;
             zt_grid = zt_grid_;
             zi_grid = zi_grid_;
             wpthlp_sfc = wpthlp_sfc_;
             wprtp_sfc = wprtp_sfc_;
             upwp_sfc = upwp_sfc_;
             vpwp_sfc = vpwp_sfc_;
             wtracer_sfc = wtracer_sfc_;
             wm_zt = wm_zt_;
             inv_exner = inv_exner_;
             thlm = thlm_;
             qw = qw_;
             cldfrac_liq=cldfrac_liq_;
             cldfrac_liq_prev=cldfrac_liq_prev_;
         } // set_variables