ERF_WSM6.H File Reference

#include <memory>
#include <string>
#include <vector>
#include <AMReX_FArrayBox.H>
#include <AMReX_Geometry.H>
#include <AMReX_MultiFabUtil.H>
#include "ERF_Constants.H"
#include "ERF_MicrophysicsConstants.H"
#include "ERF_DataStruct.H"
#include "ERF_IndexDefines.H"
#include "ERF_MicrophysicsColumns.H"
#include "ERF_MicrophysicsUtils.H"
#include "ERF_NullMoist.H"
#include "ERF_SurfacePrecipitation.H"

Include dependency graph for ERF_WSM6.H:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	WSM6

Namespaces
	MicVar_WSM6
	WSM6Sed
	WSM6Ind

Enumerations
enum	{   MicVar_WSM6::rho = 0 , MicVar_WSM6::theta , MicVar_WSM6::tabs , MicVar_WSM6::pres ,   MicVar_WSM6::qv , MicVar_WSM6::qc , MicVar_WSM6::qi , MicVar_WSM6::qr ,   MicVar_WSM6::qs , MicVar_WSM6::qg , MicVar_WSM6::rain_accum , MicVar_WSM6::snow_accum ,   MicVar_WSM6::graupel_accum , MicVar_WSM6::precip_accum , MicVar_WSM6::NumVars }
enum	{   WSM6Sed::qi = 0 , WSM6Sed::qr , WSM6Sed::qs , WSM6Sed::qg ,   WSM6Sed::NumComps }
enum	{   WSM6Ind::t = 0 , WSM6Ind::q , WSM6Ind::qc , WSM6Ind::qi ,   WSM6Ind::qr , WSM6Ind::qs , WSM6Ind::qg , WSM6Ind::den ,   WSM6Ind::p , WSM6Ind::delz , WSM6Ind::fallc , WSM6Ind::falkc ,   WSM6Ind::work1c , WSM6Ind::work2c , WSM6Ind::workr , WSM6Ind::worka ,   WSM6Ind::den_tmp , WSM6Ind::delz_tmp , WSM6Ind::pigen , WSM6Ind::pidep ,   WSM6Ind::pcond , WSM6Ind::prevp , WSM6Ind::psevp , WSM6Ind::pgevp ,   WSM6Ind::psdep , WSM6Ind::pgdep , WSM6Ind::praut , WSM6Ind::psaut ,   WSM6Ind::pgaut , WSM6Ind::piacr , WSM6Ind::pracw , WSM6Ind::praci ,   WSM6Ind::pracs , WSM6Ind::psacw , WSM6Ind::psaci , WSM6Ind::psacr ,   WSM6Ind::pgacw , WSM6Ind::pgaci , WSM6Ind::pgacr , WSM6Ind::pgacs ,   WSM6Ind::paacw , WSM6Ind::psmlt , WSM6Ind::pgmlt , WSM6Ind::pseml ,   WSM6Ind::pgeml , WSM6Ind::qsum , WSM6Ind::xl , WSM6Ind::cpm ,   WSM6Ind::work2 , WSM6Ind::denfac , WSM6Ind::xni , WSM6Ind::denqrs1 ,   WSM6Ind::denqrs2 , WSM6Ind::denqrs3 , WSM6Ind::denqci , WSM6Ind::n0sfac ,   WSM6Ind::rhw , WSM6Ind::rhi , WSM6Ind::qsatw , WSM6Ind::qsati ,   WSM6Ind::rslope_r , WSM6Ind::rslope_s , WSM6Ind::rslope_g , WSM6Ind::rslope2_r ,   WSM6Ind::rslope2_s , WSM6Ind::rslope2_g , WSM6Ind::rslope3_r , WSM6Ind::rslope3_s ,   WSM6Ind::rslope3_g , WSM6Ind::rslopeb_r , WSM6Ind::rslopeb_s , WSM6Ind::rslopeb_g ,   WSM6Ind::qrs_tmp_r , WSM6Ind::qrs_tmp_s , WSM6Ind::qrs_tmp_g , WSM6Ind::falk_r ,   WSM6Ind::falk_s , WSM6Ind::falk_g , WSM6Ind::fall_r , WSM6Ind::fall_s ,   WSM6Ind::fall_g , WSM6Ind::work1_r , WSM6Ind::work1_s , WSM6Ind::work1_g ,   WSM6Ind::workdiffw , WSM6Ind::workdiffi , WSM6Ind::NumInds }

Functions
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	wsm6_copy_state_to_micro_cell (const amrex::Array4< const amrex::Real > &states, const amrex::Array4< const amrex::Real > &base, const amrex::Array4< amrex::Real > &rho, const amrex::Array4< amrex::Real > &theta, const amrex::Array4< amrex::Real > &tabs, const amrex::Array4< amrex::Real > &pres, const amrex::Array4< amrex::Real > &qv, const amrex::Array4< amrex::Real > &qc, const amrex::Array4< amrex::Real > &qi, const amrex::Array4< amrex::Real > &qr, const amrex::Array4< amrex::Real > &qs, const amrex::Array4< amrex::Real > &qg, const amrex::Real rdOcp, const bool use_anelastic_reference_pressure, const int i, const int j, const int k) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real	wsm6_theta_from_temperature_and_pressure (const amrex::Real rho, const amrex::Real temperature, const amrex::Real pressure, const amrex::Real qv, const amrex::Real configured_rdOcp, const bool use_anelastic_reference_pressure) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	wsm6_copy_micro_to_state_cell (const amrex::Array4< amrex::Real > &states, const amrex::Array4< amrex::Real > &theta, const amrex::Array4< const amrex::Real > &rho, const amrex::Array4< const amrex::Real > &tabs, const amrex::Array4< const amrex::Real > &pres, const amrex::Array4< const amrex::Real > &qv, const amrex::Array4< const amrex::Real > &qc, const amrex::Array4< const amrex::Real > &qi, const amrex::Array4< const amrex::Real > &qr, const amrex::Array4< const amrex::Real > &qs, const amrex::Array4< const amrex::Real > &qg, const bool use_anelastic_reference_pressure, const amrex::Real rdOcp, const int i, const int j, const int k) noexcept

Function Documentation

wsm6_copy_micro_to_state_cell()

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void wsm6_copy_micro_to_state_cell ( const amrex::Array4< amrex::Real > &  states, const amrex::Array4< amrex::Real > &  theta, const amrex::Array4< const amrex::Real > &  rho, const amrex::Array4< const amrex::Real > &  tabs, const amrex::Array4< const amrex::Real > &  pres, const amrex::Array4< const amrex::Real > &  qv, const amrex::Array4< const amrex::Real > &  qc, const amrex::Array4< const amrex::Real > &  qi, const amrex::Array4< const amrex::Real > &  qr, const amrex::Array4< const amrex::Real > &  qs, const amrex::Array4< const amrex::Real > &  qg, const bool  use_anelastic_reference_pressure, const amrex::Real  rdOcp, const int  i, const int  j, const int  k  )

noexcept

{
    const amrex::Real theta_value = wsm6_theta_from_temperature_and_pressure(
        rho(i,j,k), tabs(i,j,k), pres(i,j,k), qv(i,j,k), rdOcp,
        use_anelastic_reference_pressure);
    theta(i,j,k) = theta_value;
    states(i,j,k,RhoTheta_comp) = rho(i,j,k) * theta_value;
    states(i,j,k,RhoQ1_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qv(i,j,k));
    states(i,j,k,RhoQ2_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qc(i,j,k));
    states(i,j,k,RhoQ3_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qi(i,j,k));
    states(i,j,k,RhoQ4_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qr(i,j,k));
    states(i,j,k,RhoQ5_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qs(i,j,k));
    states(i,j,k,RhoQ6_comp) = rho(i,j,k) * amrex::max(amrex::Real(0), qg(i,j,k));
}

Referenced by WSM6::Copy_Micro_to_State().

Here is the call graph for this function:

Here is the caller graph for this function:

wsm6_copy_state_to_micro_cell()

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void wsm6_copy_state_to_micro_cell ( const amrex::Array4< const amrex::Real > &  states, const amrex::Array4< const amrex::Real > &  base, const amrex::Array4< amrex::Real > &  rho, const amrex::Array4< amrex::Real > &  theta, const amrex::Array4< amrex::Real > &  tabs, const amrex::Array4< amrex::Real > &  pres, const amrex::Array4< amrex::Real > &  qv, const amrex::Array4< amrex::Real > &  qc, const amrex::Array4< amrex::Real > &  qi, const amrex::Array4< amrex::Real > &  qr, const amrex::Array4< amrex::Real > &  qs, const amrex::Array4< amrex::Real > &  qg, const amrex::Real  rdOcp, const bool  use_anelastic_reference_pressure, const int  i, const int  j, const int  k  )

noexcept

{
    const amrex::Real rho_value = states(i,j,k,Rho_comp);
    const amrex::Real rho_theta = states(i,j,k,RhoTheta_comp);
    rho(i,j,k) = rho_value;
    theta(i,j,k) = rho_theta / rho_value;
    qv(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ1_comp) / rho_value);
    qc(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ2_comp) / rho_value);
    qi(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ3_comp) / rho_value);
    qr(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ4_comp) / rho_value);
    qs(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ5_comp) / rho_value);
    qg(i,j,k) = amrex::max(amrex::Real(0.0), states(i,j,k,RhoQ6_comp) / rho_value);
 
    const amrex::Real p0 = use_anelastic_reference_pressure
        ? base(i,j,k,BaseState::p0_comp) : amrex::Real(0.0);
    const MicrophysicsThermoState thermo = diagnose_microphysics_thermo_state(
        rho_value, rho_theta, qv(i,j,k), rdOcp,
        use_anelastic_reference_pressure, p0);
    tabs(i,j,k) = thermo.temperature;
    pres(i,j,k) = thermo.pressure_pa;
}

Here is the call graph for this function:

wsm6_theta_from_temperature_and_pressure()

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::Real wsm6_theta_from_temperature_and_pressure ( const amrex::Real  rho, const amrex::Real  temperature, const amrex::Real  pressure, const amrex::Real  qv, const amrex::Real  configured_rdOcp, const bool  use_anelastic_reference_pressure  )

noexcept

{
    return use_anelastic_reference_pressure
        ? getThgivenTandP(temperature, pressure, configured_rdOcp)
        : getThgivenRandT(rho, temperature, RdoCp, qv);
}

Referenced by wsm6_copy_micro_to_state_cell().

Here is the call graph for this function:

Here is the caller graph for this function: