summaryrefslogtreecommitdiff
path: root/plugin/src/Caelestia/audioprovider.hpp
blob: 44c68900c7ba58c44ab7820566b4ddf4ca368242 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#pragma once

#include "service.hpp"
#include <QAudioSource>
#include <QIODevice>
#include <QMutex>
#include <QObject>
#include <QQueue>
#include <QThread>
#include <QTimer>
#include <QVector>
#include <qqmlintegration.h>

namespace caelestia {

class AudioProvider;

class AudioCollector : public QObject {
    Q_OBJECT

public:
    explicit AudioCollector(AudioProvider* provider, QObject* parent = nullptr);
    ~AudioCollector();

    void init();

private:
    QAudioSource* m_source;
    QIODevice* m_device;

    AudioProvider* m_provider;
    int m_sampleRate;
    int m_chunkSize;

    QVector<double> m_chunk;
    int m_chunkOffset;

    Q_INVOKABLE void start();
    Q_INVOKABLE void stop();

    void handleStateChanged(QtAudio::State state) const;
    void loadChunk();
};

class AudioProcessor : public QObject {
    Q_OBJECT

public:
    explicit AudioProcessor(AudioProvider* provider, QObject* parent = nullptr);
    ~AudioProcessor();

    void init();

protected:
    int m_sampleRate;
    int m_chunkSize;

private:
    AudioProvider* m_provider;
    QTimer* m_timer;

    Q_INVOKABLE void start();
    Q_INVOKABLE void stop();

    void handleTimeout();
    virtual void processChunk(const QVector<double>& chunk) = 0;
};

class AudioProvider : public Service {
    Q_OBJECT

public:
    explicit AudioProvider(int sampleRate = 44100, int chunkSize = 512, QObject* parent = nullptr);
    ~AudioProvider();

    [[nodiscard]] int sampleRate() const;
    [[nodiscard]] int chunkSize() const;

    [[nodiscard]] QVector<double> nextChunk();
    void loadChunk(const QVector<double>& chunk);

protected:
    int m_sampleRate;
    int m_chunkSize;

    QMutex m_mutex;
    QQueue<QVector<double>> m_chunks;

    AudioCollector* m_collector;
    AudioProcessor* m_processor;

    void init();

private:
    QThread* m_collectorThread;
    QThread* m_processorThread;

    void start() override;
    void stop() override;
};

} // namespace caelestia