#include <ERF_ShocInterface.H>

Collaboration diagram for SHOCInterface::SHOCPostprocess:

[legend]

Public Member Functions
	SHOCPostprocess ()=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 view_2d_const &rrho_, const view_2d &qv_, const view_2d_const &qw_, const view_2d &qc_, const view_2d_const &qc_copy_, const view_2d &tke_, const view_2d_const &tke_copy_, const view_3d_strided &qtracers_, const view_2d_const &qc2_, const view_2d &cldfrac_liq_, const view_2d &inv_qc_relvar_, const view_2d &T_mid_, const view_2d_const &dse_, const view_2d_const &z_mid_, const view_1d_const phis_)

void	set_mass_and_energy_fluxes (const view_1d_const &surf_evap_, const view_1d_const &surf_sens_flux_, const view_1d &vapor_flux_, const view_1d &water_flux_, const view_1d &ice_flux_, const view_1d &heat_flux_)

Public Attributes
int	ncol

int	nlev

view_2d_const	rrho

view_2d	qv

view_2d	qc

view_2d	tke

view_2d_const	tke_copy

view_2d_const	qc_copy

view_2d_const	qw

view_3d_strided	qtracers

view_2d_const	qc2

view_2d	cldfrac_liq

view_2d	inv_qc_relvar

view_2d	T_mid

view_2d_const	dse

view_2d_const	z_mid

view_1d_const	phis

bool	compute_mass_and_energy_fluxes = false

view_1d_const	surf_evap

view_1d_const	surf_sens_flux

view_1d	vapor_flux

view_1d	water_flux

view_1d	ice_flux

view_1d	heat_flux

Constructor & Destructor Documentation

◆ SHOCPostprocess()

SHOCInterface::SHOCPostprocess::SHOCPostprocess ( )

default

Member Function Documentation

◆ operator()()

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

inline

         {
             const int i = team.league_rank();
  
             const Real inv_qc_relvar_max = 10;
             const Real inv_qc_relvar_min = 0.001;
  
             const int nlev_packs = ekat::npack<Spack>(nlev);
             Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (const Int& k)
             {
                 // See comment in SHOCPreprocess::operator() about the necessity of *_copy views
                 tke(i,k) = tke_copy(i,k);
                 qc(i,k)  = qc_copy(i,k);
  
                 qv(i,k) = qw(i,k) - qc(i,k);
  
                 cldfrac_liq(i,k) = ekat::min(cldfrac_liq(i,k), 1);
  
                 //P3 uses inv_qc_relvar, P3 is using dry mmrs, but
                 //wet<->dry conversion is a constant factor that cancels out in mean(qc)^2/mean(qc'*qc').
                 inv_qc_relvar(i,k) = 1;
                 const auto condition = (qc(i,k) != 0 && qc2(i,k) != 0);
                 if (condition.any()) {
                     inv_qc_relvar(i,k).set(condition,
                                            ekat::min(inv_qc_relvar_max,
                                                      ekat::max(inv_qc_relvar_min,
                                                                ekat::square(qc(i,k))/qc2(i,k))));
                 }
  
                 // Temperature
                 const Spack dse_ik(dse(i,k));
                 const Spack z_mid_ik(z_mid(i,k));
                 const Real  phis_i(phis(i));
                 T_mid(i,k) = PF::calculate_temperature_from_dse(dse_ik,z_mid_ik,phis_i);
  
             });
  
             // If necessary, set appropriate boundary fluxes for energy and mass conservation checks.
             // Any boundary fluxes not included in SHOC interface are set to 0.
             if (compute_mass_and_energy_fluxes) {
                 vapor_flux(i) = surf_evap(i);
                 water_flux(i) = 0.0;
                 ice_flux(i)   = 0.0;
                 heat_flux(i)  = surf_sens_flux(i);
             }
         } // operator

◆ set_mass_and_energy_fluxes()

void SHOCInterface::SHOCPostprocess::set_mass_and_energy_fluxes	(	const view_1d_const &	surf_evap_,
		const view_1d_const &	surf_sens_flux_,
		const view_1d &	vapor_flux_,
		const view_1d &	water_flux_,
		const view_1d &	ice_flux_,
		const view_1d &	heat_flux_
	)

inline

         {
             compute_mass_and_energy_fluxes = true;
             surf_evap = surf_evap_;
             surf_sens_flux = surf_sens_flux_;
             vapor_flux = vapor_flux_;
             water_flux = water_flux_;
             ice_flux = ice_flux_;
             heat_flux = heat_flux_;
         }

◆ set_variables()

void SHOCInterface::SHOCPostprocess::set_variables	(	const int	ncol_,
		const int	nlev_,
		const view_2d_const &	rrho_,
		const view_2d &	qv_,
		const view_2d_const &	qw_,
		const view_2d &	qc_,
		const view_2d_const &	qc_copy_,
		const view_2d &	tke_,
		const view_2d_const &	tke_copy_,
		const view_3d_strided &	qtracers_,
		const view_2d_const &	qc2_,
		const view_2d &	cldfrac_liq_,
		const view_2d &	inv_qc_relvar_,
		const view_2d &	T_mid_,
		const view_2d_const &	dse_,
		const view_2d_const &	z_mid_,
		const view_1d_const	phis_
	)

inline

         {
             ncol = ncol_;
             nlev = nlev_;
             rrho = rrho_;
             qv = qv_;
             qw = qw_;
             qc = qc_;
             qc_copy = qc_copy_;
             tke = tke_;
             tke_copy = tke_copy_;
             qtracers = qtracers_;
             qc2 = qc2_;
             cldfrac_liq = cldfrac_liq_;
             inv_qc_relvar = inv_qc_relvar_;
             T_mid = T_mid_;
             dse = dse_;
             z_mid = z_mid_;
             phis = phis_;
         } // set_variables