gpu_spectral_freeze.hpp

Go to the documentation of this file.

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

class GpuSpectralFreeze {
  public:
    bool prepare(uint32_t fft_size, uint32_t hop,
                 signal::WindowFunction::Type window = signal::WindowFunction::Type::hann,
                 render::GpuCompute* shared_device = nullptr);
 
    bool gpu_available() const { return stft_.gpu_available(); }
    bool device_usable() const {
        return stft_.device_usable();
    }
    uint32_t fft_size() const { return stft_.fft_size(); }
    uint32_t hop() const { return hop_; }
    bool is_captured() const { return captured_; }

    bool capture(const float* frame_in);

    bool render(float* frame_out, float phase_jitter = 0.0f);
 
private:
    GpuStft stft_;
    uint32_t hop_ = 0;
    std::vector<float> mag_;     // per-bin magnitude (fft_size)
    std::vector<float> phase_;   // per-bin running phase (fft_size)
    std::vector<float> scratch_; // 2*fft_size interleaved-complex rebuild buffer
    std::uint32_t seed_ = 0x9E3779B9u;  // jitter seed, advanced each render
    bool captured_ = false;
};
 
} // namespace pulp::gpu_audio