filter.H

Go to the documentation of this file.

#ifndef FILTER_H
#define FILTER_H
 
#include "src/fvm/fvm_utils.H"
#include "AMReX_Array4.H"
#include "AMReX_Geometry.H"
#include "AMReX_MFIter.H"
#include "AMReX_REAL.H"
 
using namespace amrex::literals;
 
namespace kynema_sgf::fvm {

template <typename FTypeIn, typename FTypeOut>
struct Filter
{
    Filter(FTypeOut& filterphi, const FTypeIn& phi)
        : m_filterphi(filterphi), m_phi(phi)
    {
        AMREX_ALWAYS_ASSERT(filterphi.num_comp() == phi.num_comp());
        AMREX_ALWAYS_ASSERT(m_phi.num_grow() > amrex::IntVect(0));
    }
 
    template <typename Stencil>
    void apply(const int lev, const amrex::MFIter& mfi) const
    {
        const int ncomp = m_phi.num_comp();
        const auto& geom = m_phi.repo().mesh().Geom(lev);
        const auto& filterphi_arr = m_filterphi(lev).array(mfi);
        const auto& phi_arr = m_phi(lev).const_array(mfi);
 
        const auto& bx_in = mfi.tilebox();
        const auto& bx = Stencil::box(bx_in, geom);
        if (bx.isEmpty()) {
            return;
        }
 
        amrex::ParallelFor(bx, [=] AMREX_GPU_DEVICE(int i, int j, int k) {
            for (int icomp = 0; icomp < ncomp; icomp++) {
                amrex::Real fp1 = Stencil::f00;
                amrex::Real f = Stencil::f01;
                amrex::Real fm1 = Stencil::f02;
                const amrex::Real filx =
                    ((fp1 * phi_arr(i + 1, j, k, icomp)) +
                     (f * phi_arr(i, j, k, icomp)) +
                     (fm1 * phi_arr(i - 1, j, k, icomp)));
 
                fp1 = Stencil::f10;
                f = Stencil::f11;
                fm1 = Stencil::f12;
                const amrex::Real fily =
                    ((fp1 * phi_arr(i, j + 1, k, icomp)) +
                     (f * phi_arr(i, j, k, icomp)) +
                     (fm1 * phi_arr(i, j - 1, k, icomp)));
 
                fp1 = Stencil::f20;
                f = Stencil::f21;
                fm1 = Stencil::f22;
                const amrex::Real filz =
                    ((fp1 * phi_arr(i, j, k + 1, icomp)) +
                     (f * phi_arr(i, j, k, icomp)) +
                     (fm1 * phi_arr(i, j, k - 1, icomp)));
                filterphi_arr(i, j, k, icomp) =
                    1.0_rt / 3.0_rt * (filx + fily + filz);
            }
        });
    }
 
    FTypeOut& m_filterphi;
    const FTypeIn& m_phi;
};

template <typename FTypeIn, typename FTypeOut>
void filter(FTypeOut& filterphi, const FTypeIn& phi)
{
    BL_PROFILE("kynema-sgf::fvm::filter");
    Filter<FTypeIn, FTypeOut> filtering(filterphi, phi);
    impl::apply(filtering, phi);
}

template <typename FType>
std::unique_ptr<ScratchField> filter(const FType& phi)
{
    const std::string gname = phi.name() + "_filter";
    auto filterphi = phi.repo().create_scratch_field(gname, phi.num_comp());
    filter(*filterphi, phi);
    return filterphi;
}
 
} // namespace kynema_sgf::fvm
 
#endif /* FILTER_H */