#pragma once
#include <cstddef>
#include <cstdint>
#include <complex>
#include <vector>

class Detector {
public:
    Detector(size_t nfft, float sample_rate);
    void setParams(float sharpness, float selectivity, float depth);
    void processFrame(const std::complex<double>* spectrum, float* mask);

private:
    size_t nfft_;
    float sample_rate_;
    float sharpness_;
    float selectivity_;
    float depth_;

    std::vector<float> envelope_;
    std::vector<float> prev_mask_;
    std::vector<float> smooth_buf_;
};