#pragma once

#include "service.hpp"
#include <QAudioSource>
#include <QIODevice>
#include <QObject>
#include <aubio/aubio.h>
#include <qqmlintegration.h>

namespace caelestia {

class BeatTracker : public Service {
    Q_OBJECT
    QML_ELEMENT
    QML_SINGLETON

    Q_PROPERTY(smpl_t bpm READ bpm NOTIFY bpmChanged)

public:
    explicit BeatTracker(uint_t sampleRate = 44100, uint_t hopSize = 512, QObject* parent = nullptr);
    ~BeatTracker();

    [[nodiscard]] smpl_t bpm() const;

signals:
    void bpmChanged();
    void beat(smpl_t bpm);

private:
    QAudioSource* m_source;
    QIODevice* m_device;

    aubio_tempo_t* m_tempo;
    fvec_t* m_in;
    fvec_t* m_out;
    uint_t m_hopSize;

    smpl_t m_bpm;

    void start() override;
    void stop() override;
    void process();
    void handleStateChanged(QtAudio::State state) const;
};

} // namespace caelestia