AMReX_EB2_IF_Scale.H

Go to the documentation of this file.

#ifndef AMREX_EB2_IF_SCALE_H_
#define AMREX_EB2_IF_SCALE_H_
#include <AMReX_Config.H>
 
#include <AMReX_EB2_IF_Base.H>
#include <AMReX_Array.H>
 
#include <type_traits>
 
// For all implicit functions, >0: body; =0: boundary; <0: fluid
 
namespace amrex::EB2 {
 
template <class F>
class ScaleIF
{
public:
 
    ScaleIF (F a_f, const RealArray& a_scalefactor)
        : m_f(std::move(a_f)),
#if (AMREX_SPACEDIM == 3)
          m_sfinv{1.0_rt/a_scalefactor[0], 1.0_rt/a_scalefactor[1], 1.0_rt/a_scalefactor[2]}
#else
          m_sfinv{1.0_rt/a_scalefactor[0], 1.0_rt/a_scalefactor[1], 0.0_rt}
#endif
        {}
 
    template <class U=F, std::enable_if_t<IsGPUable<U>::value,int> = 0>
    [[nodiscard]] AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
    Real operator() (AMREX_D_DECL(Real x, Real y, Real z)) const noexcept
    {
        return m_f(AMREX_D_DECL(x*m_sfinv.x,
                                y*m_sfinv.y,
                                z*m_sfinv.z));
    }
 
    [[nodiscard]] AMREX_FORCE_INLINE Real operator() (const RealArray& p) const noexcept
    {
        return m_f({AMREX_D_DECL(p[0]*m_sfinv.x,
                                 p[1]*m_sfinv.y,
                                 p[2]*m_sfinv.z)});
    }
 
protected:
 
    F m_f;
    XDim3 m_sfinv;
};
 
template <class F>
struct IsGPUable<ScaleIF<F>, std::enable_if_t<IsGPUable<F>::value>>
    : std::true_type {};
 
template <class F>
constexpr ScaleIF<std::decay_t<F>>
scale (F&&f, const RealArray& scalefactor)
{
    return ScaleIF<std::decay_t<F>>(std::forward<F>(f),scalefactor);
}
 
}
 
#endif