fft_bundle.h

Go to the documentation of this file.

#ifndef FFT_TEMP_H
#define FFT_TEMP_H
 
#include "fft_base.h"
#include "fft_cpu.h"
 
#include <memory>
namespace ModuleBase
{
class FFT_Bundle
{
  public:
    FFT_Bundle() {};
    ~FFT_Bundle();
    FFT_Bundle(std::string device_in, std::string precision_in) : device(device_in), precision(precision_in) {};
 
    void setfft(std::string device_in, std::string precision_in);
 
    void set_dsp_cluster_id(int id) { this->dsp_cluster_id_ = id; }
 
    void initfft(int nx_in,
                 int ny_in,
                 int nz_in,
                 int lixy_in,
                 int rixy_in,
                 int ns_in,
                 int nplane_in,
                 int nproc_in,
                 bool gamma_only_in,
                 bool xprime_in = true,
                 bool mpifft_in = false);
 
    void initfftmode(int fft_mode_in)
    {
        this->fft_mode = fft_mode_in;
    }
 
    void setupFFT();
 
    void clearFFT();
 
    void clear();
 
    void resource_handler(const int flag) const;
    template <typename FPTYPE>
    FPTYPE* get_rspace_data() const;
    template <typename FPTYPE>
    std::complex<FPTYPE>* get_auxr_data() const;
    template <typename FPTYPE>
    std::complex<FPTYPE>* get_auxg_data() const;
    template <typename FPTYPE>
    std::complex<FPTYPE>* get_auxr_3d_data() const;
 
    template <typename FPTYPE>
    void fftzfor(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
    template <typename FPTYPE>
    void fftxyfor(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
    template <typename FPTYPE>
    void fftzbac(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
    template <typename FPTYPE>
    void fftxybac(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
 
    template <typename FPTYPE>
    void fftxyr2c(FPTYPE* in, std::complex<FPTYPE>* out) const;
    template <typename FPTYPE>
    void fftxyc2r(std::complex<FPTYPE>* in, FPTYPE* out) const;
 
    template <typename FPTYPE>
    void fft3D_forward(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
    template <typename FPTYPE>
    void fft3D_backward(std::complex<FPTYPE>* in, std::complex<FPTYPE>* out) const;
 
  private:
    int fft_mode = 0;
    bool float_flag = false;
    bool double_flag = false;
    std::shared_ptr<FFT_BASE<float>> fft_float = nullptr;
    std::shared_ptr<FFT_BASE<double>> fft_double = nullptr;
 
    std::string device = "cpu";
    std::string precision = "double";
    int dsp_cluster_id_ = 0;
};
// Use RAII (Resource Acquisition Is Initialization) to 
// control the resources used by hthread when setting the DSP
struct FFT_Guard
  {
      const FFT_Bundle& fft_;
      FFT_Guard(const FFT_Bundle& fft) : fft_(fft) 
        {fft_.resource_handler(1);}
      ~FFT_Guard()
      {
        fft_.resource_handler(0);
      }
  };
 
} // namespace ModuleBase
#endif // FFT_H