helmholtz_3d_hf_shift.h

#ifndef NIHU_HELMHOLTZ_3D_HF_SHIFT_H_INCLUDED
#define NIHU_HELMHOLTZ_3D_HF_SHIFT_H_INCLUDED
 
#include "helmholtz_3d_hf_level_data.h"
 
#define L2L_SHIFT_FIRST 1
 
namespace NiHu
{
namespace fmm
{
 
class helmholtz_3d_hf_fmm_upshift
{
    using cvector_t = Eigen::Matrix<std::complex<double>, Eigen::Dynamic, 1>;
 
public:
    helmholtz_3d_hf_fmm_upshift()
        : m_level_data(nullptr)
    {
    }
 
    helmholtz_3d_hf_fmm_upshift(cvector_t const &shift, helmholtz_3d_hf_level_data const &ld)
        : m_shift(shift), m_level_data(&ld)
    {
    }
 
    cvector_t operator*(cvector_t const &rhs) const
    {
        return m_shift.array() * m_level_data->interp_up(rhs).array();
    }
 
private:
    cvector_t m_shift;
    helmholtz_3d_hf_level_data const *m_level_data;
};
 
class helmholtz_3d_hf_fmm_downshift
{
    using cvector_t = Eigen::Matrix<std::complex<double>, Eigen::Dynamic, 1>;
 
public:
    helmholtz_3d_hf_fmm_downshift()
        : m_level_data(nullptr)
    {
    }
 
    helmholtz_3d_hf_fmm_downshift(cvector_t const &shift, helmholtz_3d_hf_level_data const &ld)
        : m_shift(shift), m_level_data(&ld)
    {
    }
 
    cvector_t operator*(cvector_t const &rhs) const
    {
#if L2L_SHIFT_FIRST
        return m_level_data->interp_down(m_shift.array() * rhs.array());
#else
        return m_shift.array() * m_level_data->interp_down(rhs).array();
#endif
    }
 
private:
    cvector_t m_shift;
    helmholtz_3d_hf_level_data const *m_level_data;
};
 
}
}
 
#endif // NIHU_HELMHOLTZ_3D_HF_SHIFT_H_INCLUDED