75 lines
2.1 KiB
C++
75 lines
2.1 KiB
C++
#ifndef __FFTW_UNIFIED_CALLS_HPP
|
|
#define __FFTW_UNIFIED_CALLS_HPP
|
|
|
|
#include <fftw3.h>
|
|
|
|
namespace CosmoTool
|
|
{
|
|
|
|
static inline void init_fftw_wisdom()
|
|
{
|
|
fftw_import_system_wisdom();
|
|
fftw_import_wisdom_from_filename("fft_wisdom");
|
|
}
|
|
|
|
static inline void save_fftw_wisdom()
|
|
{
|
|
fftw_export_wisdom_to_filename("fft_wisdom");
|
|
}
|
|
|
|
template<typename T> class FFTW_Calls {};
|
|
|
|
|
|
#define FFTW_CALLS_BASE(rtype, prefix) \
|
|
template<> \
|
|
class FFTW_Calls<rtype> { \
|
|
public: \
|
|
typedef rtype real_type; \
|
|
typedef prefix ## _complex complex_type; \
|
|
typedef prefix ## _plan plan_type; \
|
|
\
|
|
static complex_type *alloc_complex(int N) { return prefix ## _alloc_complex(N); } \
|
|
static real_type *alloc_real(int N) { return prefix ## _alloc_real(N); } \
|
|
static void free(void *p) { fftw_free(p); } \
|
|
\
|
|
static void execute(plan_type p) { prefix ## _execute(p); } \
|
|
static plan_type plan_dft_r2c_2d(int Nx, int Ny, \
|
|
real_type *in, complex_type *out, \
|
|
unsigned flags) \
|
|
{ \
|
|
return prefix ## _plan_dft_r2c_2d(Nx, Ny, in, out, \
|
|
flags); \
|
|
} \
|
|
static plan_type plan_dft_c2r_2d(int Nx, int Ny, \
|
|
complex_type *in, real_type *out, \
|
|
unsigned flags) \
|
|
{ \
|
|
return prefix ## _plan_dft_c2r_2d(Nx, Ny, in, out, \
|
|
flags); \
|
|
} \
|
|
static plan_type plan_dft_r2c_3d(int Nx, int Ny, int Nz, \
|
|
real_type *in, complex_type *out, \
|
|
unsigned flags) \
|
|
{ \
|
|
return prefix ## _plan_dft_r2c_3d(Nx, Ny, Nz, in, out, flags); \
|
|
} \
|
|
static plan_type plan_dft_r2c(int rank, const int *n, real_type *in, \
|
|
complex_type *out, unsigned flags) \
|
|
{ \
|
|
return prefix ## _plan_dft_r2c(rank, n, in, out, flags); \
|
|
} \
|
|
static plan_type plan_dft_c2r(int rank, const int *n, complex_type *in, \
|
|
real_type *out, unsigned flags) \
|
|
{ \
|
|
return prefix ## _plan_dft_c2r(rank, n, in, out, flags); \
|
|
} \
|
|
static void destroy_plan(plan_type plan) { prefix ## _destroy_plan(plan); } \
|
|
}
|
|
|
|
|
|
FFTW_CALLS_BASE(double, fftw);
|
|
FFTW_CALLS_BASE(float, fftwf);
|
|
|
|
};
|
|
|
|
#endif
|