1 files changed, 11 insertions, 63 deletions

diff --git a/plugin/src/Caelestia/beattracker.cpp b/plugin/src/Caelestia/beattracker.cpp
index aeca64b..a915c08 100644
--- a/plugin/src/Caelestia/beattracker.cpp
+++ b/plugin/src/Caelestia/beattracker.cpp
@@ -1,90 +1,38 @@
 #include "beattracker.hpp"
 
-#include "service.hpp"
-#include <QAudioSource>
-#include <QDebug>
-#include <QIODevice>
-#include <QMediaDevices>
+#include "audioprovider.hpp"
 #include <QObject>
 #include <aubio/aubio.h>
 
 namespace caelestia {
 
 BeatTracker::BeatTracker(uint_t sampleRate, uint_t hopSize, QObject* parent)
-    : Service(parent)
+    : AudioProvider(static_cast<int>(sampleRate), static_cast<int>(hopSize), parent)
     , m_tempo(new_aubio_tempo("default", 1024, hopSize, sampleRate))
     , m_in(new_fvec(hopSize))
     , m_out(new_fvec(2))
-    , m_hopSize(hopSize)
-    , m_bpm(120) {
-    QAudioFormat format;
-    format.setSampleRate(static_cast<int>(sampleRate));
-    format.setChannelCount(1);
-    format.setSampleFormat(QAudioFormat::Int16);
-
-    m_source = new QAudioSource(QMediaDevices::defaultAudioInput(), format, this);
-    connect(m_source, &QAudioSource::stateChanged, this, &BeatTracker::handleStateChanged);
-};
+    , m_bpm(120) {};
 
 BeatTracker::~BeatTracker() {
     del_aubio_tempo(m_tempo);
     del_fvec(m_in);
     del_fvec(m_out);
-
-    m_source->stop();
-    delete m_source;
 }
 
 smpl_t BeatTracker::bpm() const {
     return m_bpm;
 }
 
-void BeatTracker::start() {
-    m_device = m_source->start();
-    connect(m_device, &QIODevice::readyRead, this, &BeatTracker::process);
-}
-
-void BeatTracker::stop() {
-    m_source->stop();
-    m_device = nullptr;
-}
-
-void BeatTracker::process() {
-    const QByteArray data = m_device->readAll();
-    const int16_t* samples = reinterpret_cast<const int16_t*>(data.constData());
-    const size_t count = static_cast<size_t>(data.size()) / sizeof(int16_t);
-
-    for (size_t i = 0; i < count; ++i) {
-        m_in->data[i % m_hopSize] = samples[i] / 32768.0f;
-        if ((i + 1) % m_hopSize == 0) {
-            aubio_tempo_do(m_tempo, m_in, m_out);
-            if (m_out->data[0] != 0.0f) {
-                m_bpm = aubio_tempo_get_bpm(m_tempo);
-                emit bpmChanged();
-                emit beat(m_bpm);
-            }
-        }
-    }
+void BeatTracker::processData() {
+    process(m_in->data);
 }
 
-void BeatTracker::handleStateChanged(QtAudio::State state) const {
-    if (state == QtAudio::StoppedState && m_source->error() != QtAudio::NoError) {
-        switch (m_source->error()) {
-        case QtAudio::OpenError:
-            qWarning() << "BeatTracker: failed to open audio device";
-            break;
-        case QtAudio::IOError:
-            qWarning() << "BeatTracker: an error occurred during read/write of audio device";
-            break;
-        case QtAudio::UnderrunError:
-            qWarning() << "BeatTracker: audio data is not being fed to audio device fast enough";
-            break;
-        case QtAudio::FatalError:
-            qCritical() << "BeatTracker: fatal error in audio device";
-            break;
-        default:
-            break;
-        }
+void BeatTracker::consumeData() {
+    aubio_tempo_do(m_tempo, m_in, m_out);
+    if (m_out->data[0] != 0.0f) {
+        m_bpm = aubio_tempo_get_bpm(m_tempo);
+        emit bpmChanged();
+        emit beat(m_bpm);
     }
 }