fft_3d_local.h

Go to the documentation of this file.

 
#ifndef FFT_3D_LOCAL_INCLUDED
#define FFT_3D_LOCAL_INCLUDED
 
#ifdef USE_FFTWLIB
 
#include "fft.h"
 
#ifdef USE_MPI
#include <fftw3-mpi.h>
#else
#include <fftw3.h>
#endif
 
class FFT_3d_local : public FFT
{
 public:
  static const std::string class_name;
 
 public:
  FFT_3d_local();
 
  FFT_3d_local(const Parameters& params);
 
  virtual ~FFT_3d_local();
 
  void fft(Field& dst, const Field& src, enum Direction dir);
  void fft(Field& dst, const Field& src);
  void fft(Field& field);
 
  void set_parameters(const Parameters& params);
  void set_parameters(const std::string& direction);
 
  void get_parameters(Parameters& params) const;
 
 private:
 
  void initialize();
  void finalize();
 
  bool check_ok();
 
  void initialize_plan(const Field& src);
  void clear_plan();
 
 
  Bridge::VerboseLevel m_vl;
 
  // size info
  int m_ndim;
  int m_nsize[4]; // assume atmost 4dimensional
  int m_vol;
  int m_nv;       // number of complex elements
 
  fftw_complex *m_buf_in;
  fftw_complex *m_buf_out;
  fftw_plan m_plan_fw;
  fftw_plan m_plan_bw;
 
  Direction m_direction;
 
 
#ifdef USE_FACTORY
 private:
  static FFT *create_object()
  {
    return new FFT_3d_local();
  }
 
  static FFT *create_object_with_params(const Parameters& params)
  {
    return new FFT_3d_local(params);
  }
 
 public:
  static bool register_factory()
  {
    bool init = true;
    init &= FFT::Factory::Register("FFT_3d_local", create_object);
    init &= FFT::Factory_params::Register("FFT_3d_local", create_object_with_params);
    return init;
  }
#endif
};
 
#endif /* USE_FFTWLIB */
 
#endif /* FFT_3D_LOCAL_INCLUDED */