fftw_types.h

Go to the documentation of this file.

#ifndef FFTW_TYPES
#define FFTW_TYPES



//include fftw3 library
#include "fftw3.h"



#if defined (USE_FFTW_FLOAT)
//===========================
//use the float fftw library
//===========================

//complex & real definition
#define tFFTWComplex fftwf_complex
#define tFFTWReal float
//memory managment
#define FFTW_ALLOC(C,N) (C*) fftwf_malloc(sizeof(C) * (N+1))
#define FFTW_FREE  fftwf_free
#define FFTW_CLEANUP  fftwf_cleanup 

//DFT
#define FFTW_PLAN fftwf_plan
#define FFTW_PLAN_DFT_1D fftwf_plan_dft_1d
#define FFTW_DESTROY_PLAN fftwf_destroy_plan
#define FFTW_RUN fftwf_execute

//THREADS
#define FFTW_INIT_THREADS() fftwf_init_threads()
#define FFTW_PLAN_WITH_NTHREADS(A) fftwf_plan_with_nthreads(A)
#define FFTW_CLEANUP_THREADS() fftwf_cleanup_threads()

#elif defined(USE_FFTW_LONG)
//=================================
//use the long double fftw library
//==================================

//complex & real definition
#define tFFTWComplex fftwl_complex
#define tFFTWReal long double
//memory managment
#define FFTW_ALLOC(C,N) (C*) fftwl_malloc(sizeof(C) * (N+1))
#define FFTW_FREE  fftwl_free
#define FFTW_CLEANUP  fftwl_cleanup

//DFT
#define FFTW_PLAN fftwl_plan
#define FFTW_PLAN_DFT_1D fftwl_plan_dft_1d
#define FFTW_DESTROY_PLAN fftwl_destroy_plan
#define FFTW_RUN fftwl_execute

//THREADS
#define FFTW_INIT_THREADS() fftwl_init_threads()
#define FFTW_PLAN_WITH_NTHREADS(A) fftwl_plan_with_nthreads(A)
#define FFTW_CLEANUP_THREADS() fftwl_cleanup_threads()

#else
//=================================
//use the double fftw library
//==================================

//complex & real definition
#define tFFTWComplex fftw_complex
#define tFFTWReal double

//memory managment
#define FFTW_ALLOC(C,N) (C*) fftw_malloc(sizeof(C) * (N+1))
#define FFTW_FREE  fftw_free
#define FFTW_CLEANUP  fftw_cleanup

//DFT
#define FFTW_PLAN fftw_plan
#define FFTW_PLAN_DFT_1D fftw_plan_dft_1d
#define FFTW_DESTROY_PLAN fftw_destroy_plan
#define FFTW_RUN fftw_execute

//THREADS
#define FFTW_INIT_THREADS() fftw_init_threads()
#define FFTW_PLAN_WITH_NTHREADS(A) fftw_plan_with_nthreads(A)
#define FFTW_CLEANUP_THREADS() fftw_cleanup_threads()


#endif


//OPENMP
//=====
#ifdef _OPENMP

#define OMP_FFTW_INIT_THREADS() FFTW_INIT_THREADS()
#define OMP_FFTW_PLAN_WITH_NTHREADS(A) FFTW_PLAN_WITH_NTHREADS(A)
#define OMP_FFTW_CLEANUP_THREADS() FFTW_CLEANUP_THREADS()

#else

#define OMP_FFTW_INIT_THREADS()
#define OMP_FFTW_PLAN_WITH_NTHREADS(A)
#define OMP_FFTW_CLEANUP_THREADS();

#endif 

//intel compiler
//================
#if defined (__INTEL_COMPILER)
#undefine  FFTW_ALLOC(C,A)
#undefine  FFTW_FREE

#define FFTW_ALLOC(C,N) (C*) _mm_malloc(sizeof(C) * (N+1), 64)
#define FFTW_FREE _mm_free
#endif


//end FFTW types definition
#endif