gpu_spectral_morph.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 GpuSpectralMorph {
  public:
    bool prepare(uint32_t fft_size,
                 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(); }
    bool ready() const { return has_a_ && has_b_; }
 
    bool capture_a(const float* frame_in);
    bool capture_b(const float* frame_in);

    bool render(float t, float* frame_out);
 
private:
    GpuStft stft_;
    std::vector<float> a_, b_, mix_;  // 2*fft_size interleaved-complex spectra
    bool has_a_ = false;
    bool has_b_ = false;
};
 
} // namespace pulp::gpu_audio