1 #ifndef ERF_TERRAIN_IF_H_
2 #define ERF_TERRAIN_IF_H_
4 #include <AMReX_Array.H>
5 #include <AMReX_EB2_IF_Base.H>
17 TerrainIF (amrex::FArrayBox
const& a_z_terrain, amrex::Geometry
const& a_geom,
18 amrex::Gpu::DeviceVector<amrex::Real>& a_dz_stretched)
19 :
terr_arr(a_z_terrain.const_array()),
20 dz_s(a_dz_stretched.data()),
21 dx(a_geom.CellSize(0)),
22 dy(a_geom.CellSize(1)),
23 dz(a_geom.CellSize(2)),
29 i_hi(static_cast<int>(std::round((a_geom.ProbHi(0)-a_geom.ProbLo(0))/a_geom.CellSize(0)))),
30 j_hi(static_cast<int>(std::round((a_geom.ProbHi(1)-a_geom.ProbLo(1))/a_geom.CellSize(1))))
33 AMREX_GPU_HOST_DEVICE
inline
34 amrex::Real
operator() (AMREX_D_DECL(amrex::Real
x, amrex::Real
y, amrex::Real
z))
48 amrex::Real terr_z{0.};
51 const int k1 =
static_cast<int>(std::floor((
z-
prob_lo_z) /
dz));
53 const amrex::Real remainder_z = (
z - z1)/
dz;
55 for (
int kk = 0; kk < k1; ++kk) {
56 z_stretched +=
dz_s[kk];
58 z_stretched += remainder_z *
dz_s[k1];
67 }
else if (x <= prob_lo_x && y >=
prob_hi_y) {
104 const amrex::Real denom =
dx*
dy;
105 const amrex::Real w11 = (x2-
x)*(y2-
y)/denom;
106 const amrex::Real w12 = (x2-
x)*(
y-y1)/denom;
107 const amrex::Real w21 = (
x-x1)*(y2-
y)/denom;
108 const amrex::Real w22 = (
x-x1)*(
y-y1)/denom;
111 return -(z_stretched - terr_z);
114 AMREX_GPU_HOST_DEVICE
115 inline amrex::Real
operator() (
const amrex::RealArray& p)
const noexcept
117 return this->
operator() (AMREX_D_DECL(p[0], p[1], p[2]));
Definition: ERF_EBIFTerrain.H:14
amrex::Real prob_lo_x
Definition: ERF_EBIFTerrain.H:124
amrex::Real dy
Definition: ERF_EBIFTerrain.H:123
int j_hi
Definition: ERF_EBIFTerrain.H:126
TerrainIF(amrex::FArrayBox const &a_z_terrain, amrex::Geometry const &a_geom, amrex::Gpu::DeviceVector< amrex::Real > &a_dz_stretched)
Definition: ERF_EBIFTerrain.H:17
amrex::Array4< amrex::Real const > terr_arr
Definition: ERF_EBIFTerrain.H:121
amrex::Real prob_hi_x
Definition: ERF_EBIFTerrain.H:125
amrex::Real const * dz_s
Definition: ERF_EBIFTerrain.H:122
amrex::Real prob_lo_y
Definition: ERF_EBIFTerrain.H:124
amrex::Real dx
Definition: ERF_EBIFTerrain.H:123
amrex::Real prob_lo_z
Definition: ERF_EBIFTerrain.H:124
int i_hi
Definition: ERF_EBIFTerrain.H:126
AMREX_GPU_HOST_DEVICE amrex::Real operator()(AMREX_D_DECL(amrex::Real x, amrex::Real y, amrex::Real z)) const noexcept
Definition: ERF_EBIFTerrain.H:34
amrex::Real dz
Definition: ERF_EBIFTerrain.H:123
amrex::Real prob_hi_y
Definition: ERF_EBIFTerrain.H:125