cufft_fft.hpp

Go to the documentation of this file.

//          Copyright RedPortal, mujjingun 2017 - 2018.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

#ifndef _MGCPP_CUDA_LIBS_FFT_HPP_
#define _MGCPP_CUDA_LIBS_FFT_HPP_

#include <mgcpp/global/complex.hpp>

#include <mgcpp/system/outcome.hpp>

#include <cufft.h>

namespace mgcpp {
namespace cufft {
outcome::result<void> rfft(size_t n, float const* x, cuComplex* result);

outcome::result<void> rfft(size_t n, double const* x, cuDoubleComplex* result);

outcome::result<void> irfft(size_t n, cuComplex const* x, float* result);

outcome::result<void> irfft(size_t n, cuDoubleComplex const* x, double* result);

enum class fft_direction {
  forward = CUFFT_FORWARD,

  inverse = CUFFT_INVERSE
};

outcome::result<void> cfft(size_t n,
                           cuComplex const* x,
                           cuComplex* result,
                           fft_direction direction);

outcome::result<void> cfft(size_t n,
                           cuDoubleComplex const* x,
                           cuDoubleComplex* result,
                           fft_direction direction);

outcome::result<void> rfft2(size_t n,
                            size_t m,
                            float const* x,
                            cuComplex* result);

outcome::result<void> rfft2(size_t n,
                            size_t m,
                            double const* x,
                            cuDoubleComplex* result);

outcome::result<void> irfft2(size_t n,
                             size_t m,
                             cuComplex const* x,
                             float* result);

outcome::result<void> irfft2(size_t n,
                             size_t m,
                             cuDoubleComplex const* x,
                             double* result);

outcome::result<void> cfft2(size_t n,
                            size_t m,
                            cuComplex const* x,
                            cuComplex* result,
                            fft_direction direction);

outcome::result<void> cfft2(size_t n,
                            size_t m,
                            cuDoubleComplex const* x,
                            cuDoubleComplex* result,
                            fft_direction direction);
}  // namespace cufft
}  // namespace mgcpp

#endif