FieldBCOps.H

Go to the documentation of this file.

#ifndef FIELDBCOPS_H
#define FIELDBCOPS_H
 
#include "amr-wind/core/FieldDescTypes.H"
#include "amr-wind/core/Field.H"
 
#include "AMReX_Gpu.H"
#include "AMReX_FArrayBox.H"
#include "AMReX_Geometry.H"
#include "AMReX_PhysBCFunct.H"
 
namespace amr_wind {
 
class Field;
 
class FieldBCIface
{
public:
    virtual ~FieldBCIface() = default;
 
    virtual void operator()(Field& field, const FieldState rho_state) = 0;
};
 
struct FieldBCNoOp
{
    using FunctorType = FieldBCNoOp;
 
    AMREX_GPU_HOST
    constexpr FieldBCNoOp() = default;
 
    AMREX_GPU_HOST
    constexpr explicit FieldBCNoOp(const Field& /*unused*/) {}
 
    FunctorType operator()(const int /*face_dir*/ = -1) const { return *this; }
 
    AMREX_GPU_DEVICE
    void operator()(
        const amrex::IntVect& /* iv */,
        amrex::Array4<amrex::Real> const& /* field */,
        const int /* dcomp */,
        const int /* numcomp */,
        amrex::GeometryData const& /* geom */,
        const amrex::Real /* time */,
        const amrex::BCRec* /* bcr */,
        const int /* bcomp */,
        const int /* orig_comp */) const
    {}
 
    AMREX_GPU_DEVICE
    inline void set_inflow(
        const amrex::IntVect& /*unused*/,
        amrex::Array4<amrex::Real> const& /*unused*/,
        amrex::GeometryData const& /*unused*/,
        const amrex::Real /*unused*/,
        amrex::Orientation /*unused*/,
        const int /*unused*/,
        const int /*unused*/,
        const int /*unused*/) const
    {}
};
 
struct ConstDirichlet
{
    using DeviceType = ConstDirichlet;
 
    int m_ncomp;
    amrex::GpuArray<const amrex::Real*, AMREX_SPACEDIM * 2> m_bcv;
 
    AMREX_GPU_HOST
    explicit ConstDirichlet(const Field& fld)
        : m_ncomp(fld.num_comp()), m_bcv(fld.bc_values_device())
    {}
 
    AMREX_GPU_HOST
    DeviceType device_instance() const { return *this; }
 
    AMREX_GPU_DEVICE
    inline void operator()(
        const amrex::IntVect& iv,
        amrex::Array4<amrex::Real> const& field,
        amrex::GeometryData const& /*unused*/,
        const amrex::Real /*unused*/,
        amrex::Orientation ori,
        const int comp,
        const int dcomp,
        const int orig_comp) const
    {
        field(iv[0], iv[1], iv[2], dcomp + comp) = m_bcv[ori][orig_comp + comp];
    }
};
 
template <typename InflowOp, typename WallOp>
struct DirichletOp
{
    using InflowOpType = InflowOp;
    using WallOpType = WallOp;
 
    int m_ncomp;
    amrex::GpuArray<BC, AMREX_SPACEDIM * 2> m_bc_type;
    const InflowOpType m_inflow_op;
    const WallOpType m_wall_op;
    const int m_face_dir;
 
    AMREX_GPU_HOST
    constexpr explicit DirichletOp(const Field& fld, const int face_dir)
        : m_ncomp(fld.num_comp())
        , m_bc_type(fld.bc_type())
        , m_inflow_op(fld)
        , m_wall_op(fld)
        , m_face_dir(face_dir)
    {}
 
    AMREX_GPU_HOST
    DirichletOp(
        const Field& fld,
        const InflowOpType& inflow_op,
        const WallOpType& wall_op,
        const int face_dir)
        : m_ncomp(fld.num_comp())
        , m_bc_type(fld.bc_type())
        , m_inflow_op(inflow_op)
        , m_wall_op(wall_op)
        , m_face_dir{face_dir}
    {}
 
    AMREX_GPU_DEVICE
    inline void operator()(
        const amrex::IntVect& iv,
        amrex::Array4<amrex::Real> const& field,
        const int dcomp,
        const int numcomp,
        amrex::GeometryData const& geom,
        const amrex::Real time,
        const amrex::BCRec* bcr,
        const int bcomp,
        const int orig_comp) const
    {
        // do something for external Dirichlet (amrex::BCType::ext_dir)
        const amrex::Box& domain_box = geom.Domain();
 
        for (int n = 0; n < numcomp; ++n) {
            const amrex::BCRec& bc = bcr[bcomp + n];
 
            for (amrex::OrientationIter oit; oit != nullptr; ++oit) {
                auto ori = oit();
                const int idir = ori.coordDir();
 
                // Check if this boundary is a dirichlet or mixed type
                const auto bctyp = (ori.isLow() ? bc.lo(idir) : bc.hi(idir));
                if ((bctyp != amrex::BCType::ext_dir) &&
                    (bctyp != amrex::BCType::direction_dependent)) {
                    continue;
                }
 
                // Check if the point in question is on the boundary
                const int big_end =
                    domain_box.bigEnd(idir) + ((idir == m_face_dir) ? 1 : 0);
                const bool is_bndry =
                    (ori.isLow() ? (iv[idir] < domain_box.smallEnd(idir))
                                 : (iv[idir] > big_end));
                if (!is_bndry) {
                    continue;
                }
 
                if ((m_bc_type[ori] == BC::mass_inflow) ||
                    (m_bc_type[ori] == BC::mass_inflow_outflow)) {
                    m_inflow_op(
                        iv, field, geom, time, ori, n, dcomp, orig_comp);
                } else {
                    m_wall_op(iv, field, geom, time, ori, n, dcomp, orig_comp);
                }
            }
        }
    }
 
    AMREX_GPU_DEVICE
    inline void set_inflow(
        const amrex::IntVect& iv,
        amrex::Array4<amrex::Real> const& field,
        amrex::GeometryData const& geom,
        const amrex::Real time,
        amrex::Orientation ori,
        const int comp,
        const int dcomp,
        const int orig_comp) const
    {
        m_inflow_op(iv, field, geom, time, ori, comp, dcomp, orig_comp);
    }
};
 
struct FieldBCDirichlet
{
    using FunctorType = DirichletOp<ConstDirichlet, ConstDirichlet>;
 
    explicit FieldBCDirichlet(const Field& fld) : m_field(fld) {}
 
    inline FunctorType operator()(const int face_dir = -1) const
    {
        return FunctorType(m_field, face_dir);
    }
 
    const Field& m_field;
};
 
template <typename InflowOp, typename WallOp>
struct BCOpCreator
{
    using InflowOpType = typename InflowOp::DeviceType;
    using WallOpType = typename WallOp::DeviceType;
    using FunctorType = DirichletOp<InflowOpType, WallOpType>;
 
    explicit BCOpCreator(const Field& fld)
        : m_field(fld), m_inflow_op(fld), m_wall_op(fld)
    {}
 
    explicit BCOpCreator(
        const Field& fld, const InflowOp& inflow_op, const WallOp& wall_op)
        : m_field(fld), m_inflow_op(inflow_op), m_wall_op(wall_op)
    {}
 
    inline FunctorType operator()(const int face_dir = -1) const
    {
        return FunctorType{
            m_field, m_inflow_op.device_instance(), m_wall_op.device_instance(),
            face_dir};
    }
 
    const Field& m_field;
    InflowOp m_inflow_op;
    WallOp m_wall_op;
};
 
} // namespace amr_wind
 
#endif /* FIELDBCOPS_H */