fft_xyz_3dim.h

Go to the documentation of this file.

#ifndef FFT_XYZ_3DIM_INCLUDED
#define FFT_XYZ_3DIM_INCLUDED

#ifdef USE_FFTWLIB

#include "fft.h"


class FFT_xyz_3dim : public FFT
{
 public:
  static const std::string class_name;

 private:
  bool m_is_forward;

  Index_lex m_index;

  fftw_complex *m_in;
  fftw_complex *m_out;

  fftw_plan m_plan;

#ifdef USE_MPI
  ptrdiff_t m_Nz_p, m_z_start_p;
#endif


 public:
  FFT_xyz_3dim()
  {
    init();
  }

  ~FFT_xyz_3dim()
  {
    tidy_up();
  }

  void fft(Field& field);  // field is overwritten
  void fft(Field& field_out, const Field& field_in);
  void fft(Field& field_out, const Field& field_in, const Direction dir);

  void set_parameters(const Parameters&);
  void set_parameters(const std::string& str_fft_direction);

 private:
  void init();
  void tidy_up();

#ifdef USE_FACTORY
 private:
  static FFT *create_object()
  {
    return new FFT_xyz_3dim();
  }

 public:
  static bool register_factory()
  {
    return FFT::Factory::Register("FFT_xyz_3dim", create_object);
  }
#endif
};
//- #endif of #ifdef USE_FFTWLIB
#endif
#endif