AMReX_Interp_1D_C.H

Go to the documentation of this file.

#ifndef AMREX_INTERP_1D_C_H_
#define AMREX_INTERP_1D_C_H_
#include <AMReX_Config.H>
 
#include <AMReX_FArrayBox.H>
#include <AMReX_BCRec.H>
#include <AMReX_Vector.H>
#include <AMReX_Array.H>
#include <cmath>
 
namespace amrex {
 
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
pcinterp_interp (Box const& bx,
                 Array4<Real> const& fine, const int fcomp, const int ncomp,
                 Array4<Real const> const& crse, const int ccomp,
                 IntVect const& ratio) noexcept
{
    const auto lo = amrex::lbound(bx);
    const auto hi = amrex::ubound(bx);
 
    for (int n = 0; n < ncomp; ++n) {
        AMREX_PRAGMA_SIMD
        for (int i = lo.x; i <= hi.x; ++i) {
            const int ic = amrex::coarsen(i,ratio[0]);
            fine(i,0,0,n+fcomp) = crse(ic,0,0,n+ccomp);
        }
    }
}
 
template<typename T>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
nodebilin_slopes (Box const& bx, Array4<T> const& slope, Array4<T const> const& u,
                  const int icomp, const int ncomp, IntVect const& ratio) noexcept
{
    const auto lo = amrex::lbound(bx);
    const auto hi = amrex::ubound(bx);
 
    const Real rx = Real(1.)/Real(ratio[0]);
 
    for (int n = 0; n < ncomp; ++n) {
        AMREX_PRAGMA_SIMD
        for (int i = lo.x; i <= hi.x; ++i) {
            T dx0 = u(i+1,0,0,n+icomp) - u(i,0,0,n+icomp);
            slope(i,0,0,n) = rx*dx0;
        }
    }
}
 
template<typename T>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
nodebilin_interp (Box const& bx, Array4<T> const& fine, const int fcomp, const int ncomp,
                  Array4<T const> const& slope, Array4<T const> const& crse,
                  const int ccomp, IntVect const& ratio) noexcept
{
    const auto lo = amrex::lbound(bx);
    const auto hi = amrex::ubound(bx);
    const auto chi = amrex::ubound(slope);
 
    for (int n = 0; n < ncomp; ++n) {
        AMREX_PRAGMA_SIMD
        for (int i = lo.x; i <= hi.x; ++i) {
            const int ic = amrex::min(amrex::coarsen(i,ratio[0]),chi.x);
            const Real fx = i - ic*ratio[0];
            fine(i,0,0,n+fcomp) = crse(ic,0,0,n+ccomp) + fx*slope(ic,0,0,0);
        }
    }
}
 
// Or, compile time error this?
// Remove these functions in a way that gives a good error message?
 
template<typename T>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
facediv_face_interp (int /*ci*/, int /*cj*/, int /*ck*/,
                     int /*nc*/, int /*nf*/, int /*idir*/,
                     Array4<T const> const& /*crse*/,
                     Array4<T> const& /*fine*/,
                     Array4<const int> const& /*mask*/,
                     IntVect const& /*ratio*/) noexcept
{
    amrex::Abort("No 1D version of FaceDiv exists.\n");
}
 
template<typename T>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
facediv_int (int /*ci*/, int /*cj*/, int /*ck*/, int /*nf*/,
             GpuArray<Array4<T>, AMREX_SPACEDIM> const& /*fine*/,
             IntVect const& /*ratio*/,
             GpuArray<Real, AMREX_SPACEDIM> const& /*cellSize*/) noexcept
{
    amrex::Abort("No 1D version of FaceDiv exists.\n");
}
 
template<typename T>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void
face_linear_interp_x (int i, int /*j*/, int /*k*/, int n, Array4<T> const& fine,
                      Array4<T const> const& crse, IntVect const& ratio) noexcept
{
    const int ii = amrex::coarsen(i,ratio[0]);
    if (i-ii*ratio[0] == 0) {
        fine(i,0,0,n) = crse(ii,0,0,n);
    } else {
        Real const w = static_cast<Real>(i-ii*ratio[0]) * (Real(1.)/Real(ratio[0]));
        fine(i,0,0,n) = (Real(1.)-w) * crse(ii,0,0,n) + w * crse(ii+1,0,0,n);
    }
}
 
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
void ccquartic_interp (int i, int /*j*/, int /*k*/, int n,
                       Array4<Real const> const& crse,
                       Array4<Real>       const& fine ) noexcept
 
{
    // Note: there are asserts in CellConservativeQuartic::interp()
    //       to check whether ratio is all equal to 2.
 
    constexpr Array1D<Real, -2, 2> cL = { -0.01171875_rt,  0.0859375_rt, 0.5_rt, -0.0859375_rt, 0.01171875_rt };
 
    int ic = amrex::coarsen(i,2);
    int irx = i - 2*ic; // = abs(i % 2)
 
    Real ftmp = 2.0_rt * ( cL(-2)*crse(ic-2,0,0,n)
                         + cL(-1)*crse(ic-1,0,0,n)
                         + cL( 0)*crse(ic  ,0,0,n)
                         + cL( 1)*crse(ic+1,0,0,n)
                         + cL( 2)*crse(ic+2,0,0,n) );
    if (irx) {
        ftmp = 2.0_rt * crse(ic,0,0,n) - ftmp;
    }
 
    fine(i,0,0,n) = ftmp;
 
}
 
} // namespace amrex
 
#endif