FFT.h

Go to the documentation of this file.

 
/*
 * @file SEFramework/FFT/FFT.h
 * @date 11/09/18
 * @author Alejandro Alvarez Ayllon
 */
 
#ifndef _SEFRAMEWORK_FFT_FFT_H
#define _SEFRAMEWORK_FFT_FFT_H
 
#include <complex>
#include <fftw3.h>
#include <memory>
#include <vector>
 
namespace SourceXtractor {
 
template <typename T>
struct FFTTraits {};
 
template <>
struct FFTTraits<float> {
  typedef fftwf_plan_s                    plan_t;
  typedef fftwf_complex                   complex_t;
  typedef decltype(fftwf_plan_dft_r2c_2d) func_plan_fwd_t;
  typedef decltype(fftwf_plan_dft_c2r_2d) func_plan_inv_t;
  typedef decltype(fftwf_destroy_plan)    func_destroy_plan_t;
  typedef decltype(fftwf_execute_dft_r2c) func_execute_fwd_t;
  typedef decltype(fftwf_execute_dft_c2r) func_execute_inv_t;
 
  static func_plan_fwd_t*     func_plan_fwd;
  static func_plan_inv_t*     func_plan_inv;
  static func_destroy_plan_t* func_destroy_plan;
  static func_execute_fwd_t*  func_execute_fwd;
  static func_execute_inv_t*  func_execute_inv;
};
 
template <>
struct FFTTraits<double> {
  typedef fftw_plan_s                    plan_t;
  typedef fftw_complex                   complex_t;
  typedef decltype(fftw_plan_dft_r2c_2d) func_plan_fwd_t;
  typedef decltype(fftw_plan_dft_c2r_2d) func_plan_inv_t;
  typedef decltype(fftw_destroy_plan)    func_destroy_plan_t;
  typedef decltype(fftw_execute_dft_r2c) func_execute_fwd_t;
  typedef decltype(fftw_execute_dft_c2r) func_execute_inv_t;
 
  static func_plan_fwd_t*     func_plan_fwd;
  static func_plan_inv_t*     func_plan_inv;
  static func_destroy_plan_t* func_destroy_plan;
  static func_execute_fwd_t*  func_execute_fwd;
  static func_execute_inv_t*  func_execute_inv;
};
 
template <typename T>
struct FFT {
  static_assert(std::is_floating_point<T>::value, "FFTTraits only supported for floating point types");
 
  typedef FFTTraits<T> fftw_traits;
 
  typedef std::shared_ptr<typename fftw_traits::plan_t> plan_ptr_t;
  typedef typename fftw_traits::complex_t               complex_t;
 
  static plan_ptr_t createForwardPlan(int width, int height, std::vector<T>& inout);
 
  static plan_ptr_t createInversePlan(int width, int height, std::vector<T>& inout);
 
  static void executeForward(plan_ptr_t& plan, std::vector<T>& inout);
 
  static void executeInverse(plan_ptr_t& plan, std::vector<T>& inout);
};
 
int fftRoundDimension(int size);
 
extern template struct FFT<float>;
extern template struct FFT<double>;
 
}  // namespace SourceXtractor
 
#endif  // _SEFRAMEWORK_FFT_FFT_H