FFT_wrapper.hpp

Go to the documentation of this file.

#ifndef FFT_WRAPPER_HPP
#define FFT_WRAPPER_HPP

#ifdef USE_FFTW
#include <fftw3.h>
#endif
#include "space_settings.hpp"
#include "complex.hpp"

class FFTWrapper{
#ifdef USE_FFTW
    fftw_plan plan;
#endif
    View<double*,CLayout,HostType> input;
    View<Complex*,CLayout,HostType> output;

    static int get_n_frequencies(int n_points){
        return (n_points/2 + 1);
    }

    public:

    FFTWrapper(){}

    FFTWrapper(int n_points);

    void execute(const View<double*,CLayout,HostType>& x, const View<Complex*,CLayout,HostType>& spec) const;

    void execute_no_openmp(const View<double*,CLayout,HostType>& x, const View<Complex*,CLayout,HostType>& spec) const;

    void destroy();
};

#endif