fft_dsp.h

Go to the documentation of this file.

#ifndef FFT_DSP_H
#define FFT_DSP_H
 
#include "fft_base.h"
#include <ctime>
#include <cstdlib>
#include <cmath>
 
#include "hthread_host.h"
#include "mtfft.h"
#include "fftw3.h"
 
namespace ModuleBase
{
    
template <typename FPTYPE>
class FFT_DSP : public FFT_BASE<FPTYPE>
{
    public:
        FFT_DSP(){};
        ~FFT_DSP(){}; 
        
        void setupFFT() override; 
 
        void clear() override;
 
        void cleanFFT() override;
        void resource_handler(const int flag) const override;
        virtual __attribute__((weak)) 
        void initfft(int nx_in, 
                     int ny_in, 
                     int nz_in) override;
        
        virtual __attribute__((weak)) 
        std::complex<FPTYPE>* get_auxr_3d_data() const override;
        
         virtual __attribute__((weak)) 
        void fft3D_forward(std::complex<FPTYPE>* in, 
                           std::complex<FPTYPE>* out) const override;
         virtual __attribute__((weak)) 
        void fft3D_backward(std::complex<FPTYPE>* in, 
                            std::complex<FPTYPE>* out) const override;
    public:
        int nxyz=0;
        INT cluster_id=0;
        mutable INT   b_id=0;
        mutable INT thread_id_for=0;
        PLAN* ptr_plan_forward=nullptr;
        PLAN* ptr_plan_backward=nullptr;
        mutable unsigned long args_for[2];
        mutable unsigned long args_back[2];
        E *   forward_in=nullptr;
        std::complex<float>*  c_auxr_3d = nullptr;  // fft space
        std::complex<double>* z_auxr_3d = nullptr; // fft space
 
};
} // namespace ModuleBase
#endif