helmholtz_2d_wb_x2x_matrix.cpp

Go to the documentation of this file.

 
#include "helmholtz_2d_wb_x2x_matrix.h"
 
#include <iostream>
 
namespace NiHu
{
namespace fmm
{
 
helmholtz_2d_wb_m2m_matrix::helmholtz_2d_wb_m2m_matrix()
    : m_level_data_to(nullptr)  
    , m_level_data_from(nullptr)
{
}
 
 
helmholtz_2d_wb_m2m_matrix::helmholtz_2d_wb_m2m_matrix(
    helmholtz_2d_wb_level_data const &level_data_to,
    helmholtz_2d_wb_level_data const &level_data_from,
    cvector_t diag_coeffs)
    : m_level_data_to(&level_data_to)
    , m_level_data_from(&level_data_from)
    , m_cm(m_level_data_to->get_expansion_length(), 
        m_level_data_from->get_expansion_length(), 
        diag_coeffs)
{
    if (m_level_data_from->is_high_freq() && m_level_data_to->is_high_freq())
        m_transfer = m_level_data_to->dft(diag_coeffs);
}
 
helmholtz_2d_wb_m2m_matrix::cvector_t
helmholtz_2d_wb_m2m_matrix::operator*(cvector_t const &rhs) const
{
    // low to low
    if (!m_level_data_from->is_high_freq() && !m_level_data_to->is_high_freq())
        return m_cm * rhs;
 
    // low to high
    if (!m_level_data_from->is_high_freq() && m_level_data_to->is_high_freq())
        return m_level_data_to->dft(m_cm * rhs);
 
    // high to high
    return m_level_data_to->interp_up(rhs).array() * m_transfer.array();
}
 
 
helmholtz_2d_wb_l2l_matrix::helmholtz_2d_wb_l2l_matrix()
    : m_level_data_to(nullptr)
    , m_level_data_from(nullptr)
{
}
 
helmholtz_2d_wb_l2l_matrix::helmholtz_2d_wb_l2l_matrix(helmholtz_2d_wb_level_data const &level_to,
    helmholtz_2d_wb_level_data const &level_from,
    cvector_t diag_coeffs)
    : m_level_data_to(&level_to)
    , m_level_data_from(&level_from)
    , m_cm(m_level_data_to->get_expansion_length(),
        m_level_data_from->get_expansion_length(),
        diag_coeffs)
{
    if (m_level_data_from->is_high_freq() && m_level_data_to->is_high_freq())
        m_transfer = m_level_data_from->dft(diag_coeffs);
}
 
helmholtz_2d_wb_l2l_matrix::cvector_t
helmholtz_2d_wb_l2l_matrix::operator*(cvector_t const &rhs) const
{
    if (!m_level_data_from->is_high_freq() && !m_level_data_to->is_high_freq())
        return m_cm * rhs;
 
    // high to low
    if (m_level_data_from->is_high_freq() && !m_level_data_to->is_high_freq())
        return m_cm * m_level_data_from->idft(rhs);
 
    // L2L high to high
    return m_level_data_to->interp_dn(m_transfer.array() * rhs.array());
}
 
 
helmholtz_2d_wb_m2l_matrix::helmholtz_2d_wb_m2l_matrix()
    : m_level_data(nullptr)
{
}
 
 
helmholtz_2d_wb_m2l_matrix::helmholtz_2d_wb_m2l_matrix(helmholtz_2d_wb_level_data const &level,
    cvector_t diag_coeffs)
    : m_level_data(&level)
    , m_cm(m_level_data->get_expansion_length(), m_level_data->get_expansion_length(), diag_coeffs)
{
    if (m_level_data->is_high_freq())
        m_transfer = m_level_data->dft(diag_coeffs);
}
 
helmholtz_2d_wb_m2l_matrix::cvector_t
helmholtz_2d_wb_m2l_matrix::operator*(cvector_t const &rhs) const
{
    if (!m_level_data->is_high_freq())
        return m_cm * rhs;
 
    return rhs.array() * m_transfer.array();
}
 
} // end of namespace fmm
} // end of namespace NiHu