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_OPERATIONS_FFT_HPP_
#define _MGCPP_OPERATIONS_FFT_HPP_

#include <mgcpp/global/complex.hpp>
#include <mgcpp/matrix/dense_matrix.hpp>
#include <mgcpp/vector/dense_vector.hpp>
#include <mgcpp/vector/device_vector.hpp>

namespace mgcpp {
enum class fft_direction {
  forward,
  inverse
};

namespace strict {
template <typename DeviceVec, typename Type>
inline decltype(auto) rfft(dense_vector<DeviceVec, Type> const& vec);

template <typename DeviceVec, typename Type>
inline decltype(auto) irfft(dense_vector<DeviceVec, complex<Type>> const& vec,
                            int n = -1);

template <typename DeviceVec, typename Type>
inline decltype(auto) cfft(dense_vector<DeviceVec, complex<Type>> const& vec,
                           fft_direction direction);

template <typename DeviceMat, typename Type>
inline decltype(auto) rfft(dense_matrix<DeviceMat, Type> const& mat);

template <typename DeviceMat, typename Type>
inline decltype(auto) irfft(dense_matrix<DeviceMat, complex<Type>> const& mat,
                            int n = -1);

template <typename DeviceMat, typename Type>
inline decltype(auto) cfft(dense_matrix<DeviceMat, complex<Type>> const& mat,
                           fft_direction direction);
}  // namespace strict
}  // namespace mgcpp

#include <mgcpp/operations/fft.tpp>

#endif