helmholtz_2d_wb_level_data.h

Go to the documentation of this file.

 
#ifndef FMM_HELMHOLTZ_2D_WB_LEVEL_DATA_HPP_INCLUDED
#define FMM_HELMHOLTZ_2D_WB_LEVEL_DATA_HPP_INCLUDED
 
#include "spectral_interpolate.h"
 
#include <fftw3.h>
#include <Eigen/Dense>
 
#include <complex>
#include <vector>
 
namespace NiHu
{
namespace fmm
{
 
class helmholtz_2d_wb_level_data
{
public:
    typedef Eigen::Matrix<std::complex<double>, Eigen::Dynamic, 1> cvector_t;
 
    helmholtz_2d_wb_level_data();
    ~helmholtz_2d_wb_level_data();
    helmholtz_2d_wb_level_data(helmholtz_2d_wb_level_data const &);
    helmholtz_2d_wb_level_data &operator=(helmholtz_2d_wb_level_data const &) = delete;
 
    void init(double drel);
 
    void set_expansion_length(size_t expansion_length);
    
    size_t get_expansion_length() const;
    
    size_t get_data_size() const;
 
    void set_high_freq(bool hf);
    
    bool is_high_freq() const;
 
    void set_interp_up(size_t  Nfrom);
    
    void set_interp_dn(size_t Nfrom);
 
    cvector_t const &interp_up(cvector_t const &multi) const;
 
    cvector_t const &interp_dn(cvector_t const &local) const;
 
    cvector_t const &dft(cvector_t const &multi) const;
    
    cvector_t const &idft(cvector_t const &local) const;
 
    void set_num_threads(size_t n);
 
private:
    size_t m_expansion_length;
    bool m_high_freq;
 
    std::vector<spectral_interpolate> m_interp_ups; // should be thread private
    std::vector<spectral_interpolate> m_interp_dns; // should be thread private
 
    fftw_plan m_dft_plan;
    mutable std::vector<cvector_t> m_multi_paddeds; // should be thread private
    mutable std::vector<cvector_t> m_multi_spectrums;   // should be thread private
 
    fftw_plan m_idft_plan;
    mutable std::vector <cvector_t> m_local_paddeds;    // should be thread private
    mutable std::vector <cvector_t> m_locals;           // should be thread private
};
 
} // end of namespace fmm
} // end of namespace NiHu
 
#endif /* FMM_HELMHOLTZ_2D_WB_LEVEL_DATA_HPP_INCLUDED */