From d0fdbefbfb60994ad8d6cf3b7129dcdd556c1924 Mon Sep 17 00:00:00 2001 From: 2 * r + 2 * t <61896496+soramanew@users.noreply.github.com> Date: Mon, 8 Sep 2025 21:10:30 +1000 Subject: plugin/ap: fix collector Actually read from speakers not mic --- plugin/src/Caelestia/audiocollector.hpp | 76 +++++++++++++++++++++++++++++++++ 1 file changed, 76 insertions(+) create mode 100644 plugin/src/Caelestia/audiocollector.hpp (limited to 'plugin/src/Caelestia/audiocollector.hpp') diff --git a/plugin/src/Caelestia/audiocollector.hpp b/plugin/src/Caelestia/audiocollector.hpp new file mode 100644 index 0000000..3db294f --- /dev/null +++ b/plugin/src/Caelestia/audiocollector.hpp @@ -0,0 +1,76 @@ +#pragma once + +#include "service.hpp" +#include +#include +#include +#include +#include +#include +#include +#include + +namespace caelestia { + +class AudioCollector; + +class PipeWireWorker { +public: + explicit PipeWireWorker(std::stop_token token, AudioCollector* collector); + + void run(); + +private: + pw_main_loop* m_loop; + pw_stream* m_stream; + spa_source* m_timer; + + std::stop_token m_token; + AudioCollector* m_collector; + + void cleanup(); + + static void handleTimeout(void* data, uint64_t expirations); + void processStream(); + void processSamples(const int16_t* samples, uint32_t count); + + [[nodiscard]] unsigned int nextPowerOf2(unsigned int n); +}; + +class AudioCollector : public Service { + Q_OBJECT + +public: + explicit AudioCollector( + uint32_t sampleRate = 44100, uint32_t chunkSize = 512, uint32_t bufferSize = 1024, QObject* parent = nullptr); + ~AudioCollector(); + + static AudioCollector* instance(); + + [[nodiscard]] uint32_t sampleRate() const; + [[nodiscard]] uint32_t chunkSize() const; + [[nodiscard]] uint32_t bufferSize() const; + + void loadChunk(const int16_t* samples, uint32_t count); + uint32_t readChunk(float* out, uint32_t count = 0); + uint32_t readChunk(double* out, uint32_t count = 0); + +private: + inline static AudioCollector* s_instance = nullptr; + inline static std::mutex s_mutex; + + std::jthread m_thread; + std::vector m_buffer; + uint32_t m_bufferIndex; + bool m_bufferFull; + std::mutex m_bufferMutex; + + const uint32_t m_sampleRate; + const uint32_t m_chunkSize; + const uint32_t m_bufferSize; + + void start() override; + void stop() override; +}; + +} // namespace caelestia -- cgit v1.2.3-freya