marvin_FFT.h

Go to the documentation of this file.

// ========================================================================================================
//  _______ _______ ______ ___ ___ _______ _______
// |   |   |   _   |   __ \   |   |_     _|    |  |
// |       |       |      <   |   |_|   |_|       |
// |__|_|__|___|___|___|__|\_____/|_______|__|____|
//
// This file is part of the Marvin open source library and is licensed under the terms of the MIT License.
//
// ========================================================================================================
 
#ifndef MARVIN_FFT_H
#define MARVIN_FFT_H
#include "marvin/library/marvin_Concepts.h"
#include "marvin/containers/marvin_StrideView.h"
#include <memory>
#include <span>
#include <complex>
#include <numbers>
#include <type_traits>
namespace marvin::dsp::spectral {
    enum class EngineType {
        Accelerate_FFT,
        Ipp_FFT,
        Fallback_FFT
    };

    template <typename T>
    struct getValueType {
        using ValueType = T;
    };

    template <typename T>
    requires ComplexFloatType<T>
    struct getValueType<T> {
        using ValueType = typename T::value_type;
    };

    template <RealOrComplexFloatType SampleType>
    class FFT final {
    public:
        using ValueType = typename getValueType<SampleType>::ValueType;

        explicit FFT(size_t order);

        ~FFT() noexcept;

        [[nodiscard]] EngineType getEngineType() const noexcept;

        [[nodiscard]] size_t getFFTSize() const noexcept;

        void forward(std::span<SampleType> source, std::span<std::complex<ValueType>> dest);
        std::span<std::complex<ValueType>> forward(std::span<SampleType> source);
        void inverse(std::span<std::complex<ValueType>> source, std::span<SampleType> dest);
        std::span<SampleType> inverse(std::span<std::complex<ValueType>> source);
 
    private:
        class Impl;
        std::unique_ptr<Impl> m_impl;
    };
} // namespace marvin::dsp::spectral
#endif