ViPERFX_RE/src/cpp/viper/effects/ViPERBass.h

#pragma once

#include <cstdint>
#include "../utils/Biquad.h"
#include "../utils/Subwoofer.h"
#include "../utils/WaveBuffer.h"
#include "../utils/Polyphase.h"

class ViPERBass {
public:
    enum ProcessMode {
        NATURAL_BASS = 0,
        PURE_BASS_PLUS = 1,
        SUBWOOFER = 2,
    };

    ViPERBass();
    ~ViPERBass();

    void Process(float *samples, uint32_t size);
    void Reset();
    void SetBassFactor(float bassFactor);
    void SetProcessMode(ProcessMode processMode);
    void SetSamplingRate(uint32_t samplingRate);
    void SetSpeaker(uint32_t speaker);

private:
    Polyphase *polyphase;
    Biquad *biquad;
    Subwoofer *subwoofer;
    WaveBuffer *waveBuffer;
    ProcessMode processMode;
    uint32_t samplingRate;
    float invertedSamplingRate;
    float antiPop;
    uint32_t speaker;
    float bassFactor;
};