gpu_stft.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <memory>
#include <vector>
 
#include <pulp/render/gpu_compute.hpp>
#include <pulp/signal/windowing.hpp>
 
namespace pulp::gpu_audio {

class GpuStft {
  public:
    bool prepare(uint32_t fft_size,
                 signal::WindowFunction::Type window = signal::WindowFunction::Type::hann,
                 float window_param = 0.0f,
                 render::GpuCompute* shared_device = nullptr);
 
    bool gpu_available() const { return gpu_ != nullptr; }

    bool device_usable() const;
    uint32_t fft_size() const { return fft_size_; }
    const std::vector<float>& window() const { return window_; }

    render::GpuCompute* compute() { return gpu_; }

    bool analyze(const float* frame_in, float* spectrum_out);

    bool synthesize(const float* spectrum_in, float* frame_out);
 
private:
    uint32_t fft_size_ = 0;
    std::vector<float> window_;
    render::GpuCompute* gpu_ = nullptr;             // borrowed or == owned_.get()
    std::unique_ptr<render::GpuCompute> owned_;     // when not sharing a device
    std::vector<float> scratch_;  // 2*fft_size interleaved-complex scratch
    std::uint64_t device_generation_ = 0; // gpu_->device_generation() at prepare
};
 
} // namespace pulp::gpu_audio