fft_xyz_3dim.h

Go to the documentation of this file.

#ifndef FFT_XYZ_3DIM_INCLUDED
#define FFT_XYZ_3DIM_INCLUDED

#ifdef USE_FFTWLIB

#ifdef USE_MPI
#include "fftw3-mpi.h"
#ifdef USE_BGNET
#include "communicator_bgnet.h"
#else
#include "communicator_mpi.h"
#endif
#else
#include "fftw3.h"
#endif

#ifdef USE_OPENMP
#include "threadManager_OpenMP.h"
#endif

#include "field.h"
#include "index_lex.h"

#include "bridgeIO.h"
using Bridge::vout;


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

 protected:
  Bridge::VerboseLevel m_vl;

 public:
  FFT_xyz_3dim()
    : m_vl(CommonParameters::Vlevel())
  {
    init();
  }

  ~FFT_xyz_3dim()
  {
    tidy_up();
  }

 private:
  // non-copyable
  FFT_xyz_3dim(const FFT_xyz_3dim&);
  FFT_xyz_3dim& operator=(const FFT_xyz_3dim&);

 public:
  void FFT(Field& field, const bool is_forward);  // field is overwritten
  void FFT(Field& field_out, const Field& field_in, const bool is_forward);

 private:
  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

  void init();
  void tidy_up();
};
//- #endif of #ifdef USE_FFTWLIB
#endif
#endif