SLM Class Reference

#include <ERF_SLM.H>

Inheritance diagram for SLM:

Collaboration diagram for SLM:

Public Member Functions
	SLM ()
virtual	~SLM ()=default
void	Define (SolverChoice &) override
void	Init (const int &, const amrex::MultiFab &cons_in, const amrex::Geometry &geom, const amrex::Geometry &, amrex::Vector< amrex::BCRec > &, amrex::IntVect &, const amrex::Real &dt, amrex::Vector< amrex::Vector< std::string >> &) override
void	Advance (const double &dt) override
void	ComputeTsurf ()
void	ComputeFluxes ()
void	AdvanceSLM ()
amrex::MultiFab *	Lsm_Data_Ptr (const int &varIdx) override
amrex::MultiFab *	Lsm_Flux_Ptr (const int &varIdx) override
amrex::Geometry	Lsm_Geom () override
int	Lsm_Data_Size () override
std::string	Lsm_DataName (const int &varIdx) override
Public Member Functions inherited from NullSurf
	NullSurf ()
virtual	~NullSurf ()=default
virtual void	Advance_With_State (const int &, amrex::MultiFab &, amrex::MultiFab &, amrex::MultiFab &, amrex::MultiFab *, amrex::MultiFab *, const SurfacePrecipAccumulationSources &, const amrex::Real &, const amrex::Real &, const int &, const bool)
virtual void	Plot_Landfile (const int &)
virtual void	Update_Micro_Vars (amrex::MultiFab &)
virtual void	Update_State_Vars (amrex::MultiFab &)
virtual void	Copy_State_to_Micro (const amrex::MultiFab &)
virtual void	Copy_Micro_to_State (amrex::MultiFab &)
virtual int	Lsm_Flux_Size ()
virtual std::string	Lsm_FluxName (const int &)
virtual int	Lsm_DataIndex (std::string)
virtual int	Lsm_FluxIndex (std::string)
virtual std::unordered_map< std::string, std::string > &	Lsm_WRFInputNames ()
virtual void	Lsm_Set_Lev0_Data_Ptr (const int &, amrex::MultiFab *)
virtual void	Lsm_Set_Lev0_Flux_Ptr (const int &, amrex::MultiFab *)
virtual bool	Get_Update_Status ()
virtual int	Get_LSM_Step () const
virtual void	Set_LSM_Step (int)
virtual void	Write_Lsm_Restart (const std::string &) const
virtual void	Read_Lsm_Restart (const std::string &)

Private Types
using	FabPtr = std::shared_ptr< amrex::MultiFab >

Private Attributes
int	m_lsm_size = 1
amrex::Vector< int >	LsmVarMap
amrex::Vector< std::string >	LsmVarName
amrex::Geometry	m_geom
amrex::Geometry	m_lsm_geom
double	m_dt
int	khi_lsm
amrex::Array< FabPtr, LsmVar_SLM::NumVars >	lsm_fab_vars
amrex::Array< FabPtr, LsmVar_SLM::NumVars >	lsm_fab_flux
int	m_nz_lsm = 30
amrex::Real	m_dz_lsm = amrex::Real(0.1)
amrex::Real	m_cp_soil = amrex::Real(1.26e6)
amrex::Real	m_k_soil = amrex::Real(0.2)
amrex::Real	m_theta_dir = amrex::Real(400.0)
amrex::Real	m_d_soil = m_k_soil / m_cp_soil

Member Typedef Documentation

FabPtr

using SLM::FabPtr = std::shared_ptr<amrex::MultiFab>

private

Constructor & Destructor Documentation

SLM()

SLM::SLM ( )

inline

{}

~SLM()

virtual SLM::~SLM ( )

virtualdefault

Member Function Documentation

Advance()

void SLM::Advance ( const double &  dt )

inlineoverridevirtual

Reimplemented from NullSurf.

    {
        m_dt = dt;
        this->ComputeFluxes();
        this->AdvanceSLM();
        this->ComputeTsurf();
    }

Here is the call graph for this function:

AdvanceSLM()

void SLM::AdvanceSLM

(

)

{
    // Expose for GPU copy
    Real dt = static_cast<Real>(m_dt);
    Real dzInv = m_lsm_geom.InvCellSize(2);
 
    for ( MFIter mfi(*(lsm_fab_vars[LsmVar_SLM::theta])); mfi.isValid(); ++mfi) {
        auto box3d = mfi.tilebox();
 
        auto theta_array = lsm_fab_vars[LsmVar_SLM::theta]->array(mfi);
        auto theta_flux  = lsm_fab_flux[LsmVar_SLM::theta]->array(mfi);
 
        ParallelFor( box3d, [=] AMREX_GPU_DEVICE (int i, int j, int k)
        {
            theta_array(i,j,k) += dt * ( theta_flux(i,j,k+1) - theta_flux(i,j,k) ) * dzInv;
        });
    }
}

Referenced by Advance().

Here is the call graph for this function:

Here is the caller graph for this function:

ComputeFluxes()

void SLM::ComputeFluxes

(

)

{
    // Expose for GPU copy
    int khi = khi_lsm;
    Real Dsoil = m_d_soil;
    Real dzInv = m_lsm_geom.InvCellSize(2);
 
    for ( MFIter mfi(*(lsm_fab_flux[LsmVar_SLM::theta])); mfi.isValid(); ++mfi) {
        auto box3d = mfi.tilebox();
 
        // Do not overwrite the flux at the top (comes from MOST BC)
        if (box3d.bigEnd(2) == khi+1) box3d.setBig(2,khi);
 
        auto theta_array = lsm_fab_vars[LsmVar_SLM::theta]->array(mfi);
        auto theta_flux  = lsm_fab_flux[LsmVar_SLM::theta]->array(mfi);
 
        ParallelFor( box3d, [=] AMREX_GPU_DEVICE (int i, int j, int k)
        {
            theta_flux(i,j,k) = Dsoil * ( theta_array(i,j,k) - theta_array(i,j,k-1) ) * dzInv;
        });
    }
}

Referenced by Advance().

Here is the call graph for this function:

Here is the caller graph for this function:

ComputeTsurf()

void SLM::ComputeTsurf

(

)

{
    // Expose for GPU copy
    int khi = khi_lsm;
 
    for ( MFIter mfi(*(lsm_fab_vars[LsmVar_SLM::theta])); mfi.isValid(); ++mfi) {
        auto box2d = mfi.tilebox(); box2d.makeSlab(2,khi);
 
        auto theta_array = lsm_fab_vars[LsmVar_SLM::theta]->array(mfi);
 
        ParallelFor( box2d, [=] AMREX_GPU_DEVICE (int i, int j, int )
        {
            theta_array(i,j,khi+1) = Real(1.5)*theta_array(i,j,khi) - myhalf*theta_array(i,j,khi-1);
        });
    }
}

Referenced by Advance().

Here is the call graph for this function:

Here is the caller graph for this function:

Define()

void SLM::Define ( SolverChoice &  )

inlineoverridevirtual

Reimplemented from NullSurf.

    {
        // NOTE: We should parse things from sc here,
        //       but they are hard coded because this
        //       is a demonstration for now.
    }

Init()

void SLM::Init ( const int &  , const amrex::MultiFab &  cons_in, const amrex::Geometry &  geom, const amrex::Geometry &  , amrex::Vector< amrex::BCRec > &  , amrex::IntVect &  , const amrex::Real &  dt, amrex::Vector< amrex::Vector< std::string >> &    )

overridevirtual

Reimplemented from NullSurf.

{
    m_dt = dt;
    m_geom = geom;
 
    Box domain = geom.Domain();
    khi_lsm    = domain.smallEnd(2) - 1;
 
    LsmVarMap.resize(m_lsm_size);
    LsmVarMap = {LsmVar_SLM::theta};
 
    LsmVarName.resize(m_lsm_size);
    LsmVarName = {"theta"};
 
    // NOTE: All boxes in ba extend from zlo to zhi, so this transform is valid.
    //       If that were to change, the dm and new ba are no longer valid and
    //       direct copying between lsm data/flux vars cannot be done in a parfor.
 
    // Set box array for lsm data
    IntVect ng(0,0,1);
    BoxArray ba = cons_in.boxArray();
    DistributionMapping dm = cons_in.DistributionMap();
    BoxList bl_lsm = ba.boxList();
    for (auto& b : bl_lsm) {
        b.setBig(2,khi_lsm);                  // First point below the surface
        b.setSmall(2,khi_lsm - m_nz_lsm + 1); // Last point below the surface
    }
    BoxArray ba_lsm(std::move(bl_lsm));
 
    // Set up lsm geometry
    const RealBox& dom_rb = m_geom.ProbDomain();
    const Real*    dom_dx = m_geom.CellSize();
    RealBox lsm_rb = dom_rb;
    Real lsm_dx[AMREX_SPACEDIM] = {AMREX_D_DECL(dom_dx[0],dom_dx[1],m_dz_lsm)};
    Real lsm_z_hi = dom_rb.lo(2);
    Real lsm_z_lo = lsm_z_hi - Real(m_nz_lsm)*lsm_dx[2];
    lsm_rb.setHi(2,lsm_z_hi); lsm_rb.setLo(2,lsm_z_lo);
    m_lsm_geom.define( ba_lsm.minimalBox(), lsm_rb, m_geom.Coord(), m_geom.isPeriodic() );
 
    // Create the data and fluxes
    for (auto ivar = 0; ivar < LsmVar_SLM::NumVars; ++ivar) {
        // State vars are CC
        Real theta_0 = m_theta_dir;
        lsm_fab_vars[ivar] = std::make_shared<MultiFab>(ba_lsm, dm, 1, ng);
        lsm_fab_vars[ivar]->setVal(theta_0);
 
        // Fluxes are nodal in z
        lsm_fab_flux[ivar] = std::make_shared<MultiFab>(convert(ba_lsm, IntVect(0,0,1)), dm, 1, IntVect(0,0,0));
        lsm_fab_flux[ivar]->setVal(0.);
    }
}

Lsm_Data_Ptr()

amrex::MultiFab* SLM::Lsm_Data_Ptr ( const int &  varIdx )

inlineoverridevirtual

Reimplemented from NullSurf.

    {
        int lsmIdx = LsmVarMap[varIdx];
        AMREX_ALWAYS_ASSERT(lsmIdx < SLM::m_lsm_size && lsmIdx>=0);
        return lsm_fab_vars[lsmIdx].get();
    }

Here is the call graph for this function:

Lsm_Data_Size()

int SLM::Lsm_Data_Size ( )

inlineoverridevirtual

Reimplemented from NullSurf.

{ return SLM::m_lsm_size; }

Lsm_DataName()

std::string SLM::Lsm_DataName ( const int &  varIdx )

inlineoverridevirtual

Reimplemented from NullSurf.

    {
        int lsmIdx = LsmVarMap[varIdx];
        AMREX_ALWAYS_ASSERT(lsmIdx < SLM::m_lsm_size && lsmIdx>=0);
        return LsmVarName[lsmIdx];
    }

Here is the call graph for this function:

Lsm_Flux_Ptr()

amrex::MultiFab* SLM::Lsm_Flux_Ptr ( const int &  varIdx )

inlineoverridevirtual

Reimplemented from NullSurf.

    {
        int lsmIdx = LsmVarMap[varIdx];
        AMREX_ALWAYS_ASSERT(lsmIdx < SLM::m_lsm_size && lsmIdx>=0);
        return lsm_fab_flux[lsmIdx].get();
    }

Here is the call graph for this function:

Lsm_Geom()

amrex::Geometry SLM::Lsm_Geom ( )

inlineoverridevirtual

Reimplemented from NullSurf.

{ return m_lsm_geom; }

Member Data Documentation

khi_lsm

int SLM::khi_lsm

private

lsm_fab_flux

amrex::Array<FabPtr, LsmVar_SLM::NumVars> SLM::lsm_fab_flux

private

Referenced by Lsm_Flux_Ptr().

lsm_fab_vars

amrex::Array<FabPtr, LsmVar_SLM::NumVars> SLM::lsm_fab_vars

private

Referenced by Lsm_Data_Ptr().

LsmVarMap

amrex::Vector<int> SLM::LsmVarMap

private

Referenced by Lsm_Data_Ptr(), Lsm_DataName(), and Lsm_Flux_Ptr().

LsmVarName

amrex::Vector<std::string> SLM::LsmVarName

private

Referenced by Lsm_DataName().

m_cp_soil

amrex::Real SLM::m_cp_soil = amrex::Real(1.26e6)

private

m_d_soil

amrex::Real SLM::m_d_soil = m_k_soil / m_cp_soil

private

m_dt

double SLM::m_dt

private

Referenced by Advance().

m_dz_lsm

amrex::Real SLM::m_dz_lsm = amrex::Real(0.1)

private

m_geom

amrex::Geometry SLM::m_geom

private

m_k_soil

amrex::Real SLM::m_k_soil = amrex::Real(0.2)

private

m_lsm_geom

amrex::Geometry SLM::m_lsm_geom

private

Referenced by Lsm_Geom().

m_lsm_size

int SLM::m_lsm_size = 1

private

Referenced by Lsm_Data_Size().

m_nz_lsm

int SLM::m_nz_lsm = 30

private

m_theta_dir

amrex::Real SLM::m_theta_dir = amrex::Real(400.0)

private

---

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

• Source/LandSurfaceModel/SLM/ERF_SLM.H
• Source/LandSurfaceModel/SLM/ERF_SLM.cpp