fft_3d.h

Go to the documentation of this file.

#ifndef FFT_3D_INCLUDED
#define FFT_3D_INCLUDED

#ifdef USE_FFTWLIB

// implementations
#include "fft.h"
#include "fft_3d_local.h"
#ifdef USE_MPI
#include "fft_3d_parallel1d.h"
#include "fft_3d_parallel3d.h"
#endif

#include "Communicator/communicator.h"

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

#ifdef USE_FACTORY
#include "factory.h"
#endif



class FFT_3d
{
#ifdef USE_FACTORY
 private:
  static FFT *create_object()
  {
    // auto-select

    int npe_x = Communicator::npe(0);
    int npe_y = Communicator::npe(1);
    int npe_z = Communicator::npe(2);
    int npe_t = Communicator::npe(3);

    if ((npe_x == 1) && (npe_y == 1) && (npe_z == 1)) {
      // no parallelization in xyz-directions
      return new FFT_3d_local;

#ifdef USE_MPI
    } else if ((npe_x == 1) && (npe_y == 1)) {
      // parallelization only in z-direction
      return new FFT_3d_parallel1d;
    } else {
      // most general case
      return new FFT_3d_parallel3d;
#endif
    }

    // default
    return NULL;
  }

 public:
  static bool register_factory()
  {
    return FFT::Factory::Register("auto", create_object);
  }
#endif
};

//- #endif of #ifdef USE_FFTWLIB
#endif
#endif