TerrainIF Class Reference

#include <ERF_EBIFTerrain.H>

Inheritance diagram for TerrainIF:

Collaboration diagram for TerrainIF:

Public Member Functions
	TerrainIF (amrex::FArrayBox const &a_z_terrain, amrex::Geometry const &a_geom, amrex::Gpu::DeviceVector< amrex::Real > &a_dz_stretched)
AMREX_GPU_HOST_DEVICE amrex::Real	operator() (AMREX_D_DECL(amrex::Real x, amrex::Real y, amrex::Real z)) const noexcept
AMREX_GPU_HOST_DEVICE amrex::Real	operator() (const amrex::RealArray &p) const noexcept

Protected Attributes
amrex::Array4< amrex::Real const >	terr_arr
amrex::Real const *	dz_s
amrex::Real	dx
amrex::Real	dy
amrex::Real	dz
amrex::Real	prob_lo_x
amrex::Real	prob_lo_y
amrex::Real	prob_lo_z
amrex::Real	prob_hi_x
amrex::Real	prob_hi_y
int	i_hi
int	j_hi

Constructor & Destructor Documentation

TerrainIF()

TerrainIF::TerrainIF ( amrex::FArrayBox const &  a_z_terrain, amrex::Geometry const &  a_geom, amrex::Gpu::DeviceVector< amrex::Real > &  a_dz_stretched  )

inline

        : terr_arr(a_z_terrain.const_array()),
          dz_s(a_dz_stretched.data()),
          dx(a_geom.CellSize(0)),
          dy(a_geom.CellSize(1)),
          dz(a_geom.CellSize(2)),
          prob_lo_x(a_geom.ProbLo(0)),
          prob_lo_y(a_geom.ProbLo(1)),
          prob_lo_z(a_geom.ProbLo(2)),
          prob_hi_x(a_geom.ProbHi(0)),
          prob_hi_y(a_geom.ProbHi(1)),
          i_hi(static_cast<int>(std::round((a_geom.ProbHi(0)-a_geom.ProbLo(0))/a_geom.CellSize(0)))),
          j_hi(static_cast<int>(std::round((a_geom.ProbHi(1)-a_geom.ProbLo(1))/a_geom.CellSize(1))))
        {}

Member Function Documentation

operator()() [1/2]

AMREX_GPU_HOST_DEVICE amrex::Real TerrainIF::operator() ( AMREX_D_DECL(amrex::Real x, amrex::Real y, amrex::Real z)  ) const

inlinenoexcept

    {
        int i1{};
        int j1{};
        int i2{};
        int j2{};
 
        amrex::Real x1{};
        amrex::Real x2{};
        amrex::Real y1{};
        amrex::Real y2{};
        amrex::Real w1{};
        amrex::Real w2{};
        amrex::Real terr_z{0.};
 
        // z_stretched
        const int k1 = static_cast<int>(std::floor((z-prob_lo_z) / dz));
        const amrex::Real z1 = prob_lo_z + k1*dz;
        const amrex::Real remainder_z = (z - z1)/dz;
        amrex::Real z_stretched = prob_lo_z;
        for (int kk = 0; kk < k1; ++kk) {
            z_stretched += dz_s[kk];
        }
        z_stretched += remainder_z * dz_s[k1];
 
        // Interpolation (nine subregions of x-y plane)
        if        (x <= prob_lo_x && y <= prob_lo_y) {
            terr_z = terr_arr(0   ,0   ,0);
        } else if (x >= prob_hi_x && y <= prob_lo_y) {
            terr_z = terr_arr(i_hi,0   ,0);
        } else if (x >= prob_hi_x && y >= prob_hi_y) {
            terr_z = terr_arr(i_hi,j_hi,0);
        } else if (x <= prob_lo_x && y >= prob_hi_y) {
            terr_z = terr_arr(0   ,j_hi,0);
        } else if (x > prob_lo_x && x < prob_hi_x && (y <= prob_lo_y || y >= prob_hi_y) ) {
            i1 = static_cast<int>(std::floor((x-prob_lo_x) / dx));
            i2 = i1+1;
            x1 = prob_lo_x + i1*dx;
            x2 = x1 + dx;
            w1 = (x2-x)/dx;
            w2 = (x-x1)/dx;
            if        (y <= prob_lo_y){
                terr_z = w1*terr_arr(i1,0   ,0) + w2*terr_arr(i2,0   ,0);
            } else if (y >= prob_hi_y) {
                terr_z = w1*terr_arr(i1,j_hi,0) + w2*terr_arr(i2,j_hi,0);
            }
        } else if (y > prob_lo_y && y < prob_hi_y && (x <= prob_lo_x || x >= prob_hi_x) ) {
            j1 = static_cast<int>(std::floor((y-prob_lo_y) / dy));
            j2 = j1+1;
            y1 = prob_lo_y + j1*dy;
            y2 = y1 + dy;
            w1 = (y2-y)/dy;
            w2 = (y-y1)/dy;
            if        (x <= prob_lo_x){
                terr_z = w1*terr_arr(0   ,j1,0) + w2*terr_arr(0   ,j2,0);
            } else if (x >= prob_hi_x) {
                terr_z = w1*terr_arr(i_hi,j1,0) + w2*terr_arr(i_hi,j2,0);
            }
        } else {
            // Do bilinear interpolation of the terrain surface
            i1 = static_cast<int>(std::floor((x-prob_lo_x) / dx));
            i2 = i1+1;
            j1 = static_cast<int>(std::floor((y-prob_lo_y) / dy));
            j2 = j1+1;
            x1 = prob_lo_x + i1*dx;
            x2 = x1 + dx;
            y1 = prob_lo_y + j1*dy;
            y2 = y1 + dy;
 
            const amrex::Real denom = dx*dy;
            const amrex::Real w11 = (x2-x)*(y2-y)/denom;
            const amrex::Real w12 = (x2-x)*(y-y1)/denom;
            const amrex::Real w21 = (x-x1)*(y2-y)/denom;
            const amrex::Real w22 = (x-x1)*(y-y1)/denom;
            terr_z = w11*terr_arr(i1,j1,0) + w12*terr_arr(i1,j2,0) + w21*terr_arr(i2,j1,0) + w22*terr_arr(i2,j2,0);
        }
        return -(z_stretched - terr_z);
    }

Referenced by operator()().

Here is the caller graph for this function:

operator()() [2/2]

AMREX_GPU_HOST_DEVICE amrex::Real TerrainIF::operator() ( const amrex::RealArray &  p ) const

inlinenoexcept

    {
        return this->operator() (AMREX_D_DECL(p[0], p[1], p[2]));
    }

Here is the call graph for this function:

Member Data Documentation

dx

amrex::Real TerrainIF::dx

protected

Referenced by operator()().

dy

amrex::Real TerrainIF::dy

protected

Referenced by operator()().

dz

amrex::Real TerrainIF::dz

protected

Referenced by operator()().

dz_s

amrex::Real const* TerrainIF::dz_s

protected

Referenced by operator()().

i_hi

int TerrainIF::i_hi

protected

Referenced by operator()().

j_hi

int TerrainIF::j_hi

protected

Referenced by operator()().

prob_hi_x

amrex::Real TerrainIF::prob_hi_x

protected

Referenced by operator()().

prob_hi_y

amrex::Real TerrainIF::prob_hi_y

protected

Referenced by operator()().

prob_lo_x

amrex::Real TerrainIF::prob_lo_x

protected

Referenced by operator()().

prob_lo_y

amrex::Real TerrainIF::prob_lo_y

protected

Referenced by operator()().

prob_lo_z

amrex::Real TerrainIF::prob_lo_z

protected

Referenced by operator()().

terr_arr

amrex::Array4<amrex::Real const> TerrainIF::terr_arr

protected

Referenced by operator()().

---

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

• Source/EB/ERF_EBIFTerrain.H