init
This commit is contained in:
@@ -0,0 +1,216 @@
|
||||
#include "pipemini.h"
|
||||
#include <thread>
|
||||
#include <atomic>
|
||||
#include <mutex>
|
||||
#include <condition_variable>
|
||||
|
||||
#define DEFAULT_CHANNELS 2
|
||||
#define DEFAULT_FORMAT SPA_AUDIO_FORMAT_S16
|
||||
|
||||
struct PipeMiniData {
|
||||
PipeMini* self;
|
||||
std::thread loop_thread;
|
||||
std::atomic<bool> should_quit{false};
|
||||
std::mutex mtx;
|
||||
std::condition_variable cv;
|
||||
bool teardown_done = false;
|
||||
};
|
||||
|
||||
PipeMini::PipeMini() = default;
|
||||
|
||||
PipeMini::~PipeMini() {
|
||||
stop();
|
||||
}
|
||||
|
||||
void PipeMini::teardown() {
|
||||
if (stream_) {
|
||||
pw_stream_disconnect(stream_);
|
||||
pw_stream_destroy(stream_);
|
||||
stream_ = nullptr;
|
||||
}
|
||||
|
||||
spa_hook_remove(&stream_listener_);
|
||||
|
||||
if (core_) {
|
||||
pw_core_disconnect(core_);
|
||||
core_ = nullptr;
|
||||
}
|
||||
|
||||
if (context_) {
|
||||
pw_context_destroy(context_);
|
||||
context_ = nullptr;
|
||||
}
|
||||
|
||||
if (loop_) {
|
||||
pw_main_loop_quit(loop_);
|
||||
}
|
||||
}
|
||||
|
||||
static int on_teardown_event(spa_loop* loop, bool async, uint32_t id, const void* data, size_t size, void* userdata) {
|
||||
(void)loop;
|
||||
(void)async;
|
||||
(void)id;
|
||||
(void)data;
|
||||
(void)size;
|
||||
|
||||
auto* pmd = static_cast<PipeMiniData*>(userdata);
|
||||
PipeMini* self = pmd->self;
|
||||
|
||||
self->teardown();
|
||||
|
||||
{
|
||||
std::lock_guard<std::mutex> lock(pmd->mtx);
|
||||
pmd->teardown_done = true;
|
||||
}
|
||||
pmd->cv.notify_one();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static const pw_stream_events stream_events = {
|
||||
PW_VERSION_STREAM_EVENTS,
|
||||
.process = PipeMini::on_process,
|
||||
};
|
||||
|
||||
bool PipeMini::start(const Callback& cb, int sample_rate) {
|
||||
if (running_) return false;
|
||||
|
||||
callback_ = cb;
|
||||
sample_rate_ = sample_rate;
|
||||
|
||||
pw_init(nullptr, nullptr);
|
||||
|
||||
loop_ = pw_main_loop_new(nullptr);
|
||||
if (!loop_) return false;
|
||||
|
||||
context_ = pw_context_new(pw_main_loop_get_loop(loop_), nullptr, 0);
|
||||
if (!context_) {
|
||||
pw_main_loop_destroy(loop_);
|
||||
loop_ = nullptr;
|
||||
return false;
|
||||
}
|
||||
|
||||
core_ = pw_context_connect(context_, nullptr, 0);
|
||||
if (!core_) {
|
||||
pw_context_destroy(context_);
|
||||
context_ = nullptr;
|
||||
pw_main_loop_destroy(loop_);
|
||||
loop_ = nullptr;
|
||||
return false;
|
||||
}
|
||||
|
||||
stream_ = pw_stream_new(
|
||||
core_,
|
||||
"pipemini",
|
||||
pw_properties_new(
|
||||
PW_KEY_MEDIA_TYPE, "Audio",
|
||||
PW_KEY_MEDIA_CATEGORY, "Playback",
|
||||
PW_KEY_MEDIA_ROLE, "Music",
|
||||
nullptr));
|
||||
if (!stream_) {
|
||||
pw_main_loop_destroy(loop_);
|
||||
loop_ = nullptr;
|
||||
return false;
|
||||
}
|
||||
|
||||
pw_stream_add_listener(stream_, &stream_listener_, &stream_events, this);
|
||||
|
||||
uint8_t buffer[1024];
|
||||
spa_pod_builder b = SPA_POD_BUILDER_INIT(buffer, sizeof(buffer));
|
||||
|
||||
spa_audio_info_raw info = {};
|
||||
info.format = DEFAULT_FORMAT;
|
||||
info.channels = DEFAULT_CHANNELS;
|
||||
info.rate = (uint32_t)sample_rate_;
|
||||
info.position[0] = SPA_AUDIO_CHANNEL_FL;
|
||||
info.position[1] = SPA_AUDIO_CHANNEL_FR;
|
||||
|
||||
const spa_pod* params = spa_format_audio_raw_build(&b, SPA_PARAM_EnumFormat, &info);
|
||||
|
||||
if (pw_stream_connect(stream_,
|
||||
PW_DIRECTION_OUTPUT,
|
||||
PW_ID_ANY,
|
||||
(pw_stream_flags)(PW_STREAM_FLAG_AUTOCONNECT |
|
||||
PW_STREAM_FLAG_MAP_BUFFERS |
|
||||
PW_STREAM_FLAG_RT_PROCESS),
|
||||
¶ms, 1) < 0) {
|
||||
pw_stream_destroy(stream_);
|
||||
stream_ = nullptr;
|
||||
pw_core_disconnect(core_);
|
||||
core_ = nullptr;
|
||||
pw_context_destroy(context_);
|
||||
context_ = nullptr;
|
||||
pw_main_loop_destroy(loop_);
|
||||
loop_ = nullptr;
|
||||
return false;
|
||||
}
|
||||
|
||||
running_ = true;
|
||||
|
||||
// Run the PipeWire loop in a background thread.
|
||||
data_ = new PipeMiniData{this, {}, {}};
|
||||
data_->loop_thread = std::thread([this]() {
|
||||
pw_main_loop_run(loop_);
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void PipeMini::stop() {
|
||||
if (!running_) return;
|
||||
|
||||
if (data_ && loop_) {
|
||||
pw_loop* loop = pw_main_loop_get_loop(loop_);
|
||||
pw_loop_invoke(loop, on_teardown_event, 0, nullptr, 0, true, data_);
|
||||
|
||||
std::unique_lock<std::mutex> lock(data_->mtx);
|
||||
data_->cv.wait(lock, [this]() { return data_->teardown_done; });
|
||||
|
||||
if (data_->loop_thread.joinable()) {
|
||||
data_->loop_thread.join();
|
||||
}
|
||||
|
||||
delete data_;
|
||||
data_ = nullptr;
|
||||
}
|
||||
|
||||
pw_main_loop_destroy(loop_);
|
||||
loop_ = nullptr;
|
||||
|
||||
callback_ = nullptr;
|
||||
running_ = false;
|
||||
}
|
||||
|
||||
void PipeMini::on_process(void* userdata) {
|
||||
auto* self = static_cast<PipeMini*>(userdata);
|
||||
if (!self || !self->callback_) return;
|
||||
|
||||
pw_buffer* b = pw_stream_dequeue_buffer(self->stream_);
|
||||
if (!b) return;
|
||||
|
||||
spa_buffer* buf = b->buffer;
|
||||
int16_t* dst = static_cast<int16_t*>(buf->datas[0].data);
|
||||
if (!dst) {
|
||||
pw_stream_queue_buffer(self->stream_, b);
|
||||
return;
|
||||
}
|
||||
|
||||
uint32_t n_frames = buf->datas[0].maxsize / (DEFAULT_CHANNELS * sizeof(int16_t));
|
||||
if (b->requested) {
|
||||
n_frames = SPA_MIN(b->requested, n_frames);
|
||||
}
|
||||
|
||||
for (uint32_t i = 0; i < n_frames; ++i) {
|
||||
int16_t left = 0;
|
||||
int16_t right = 0;
|
||||
self->callback_(&left, &right);
|
||||
dst[i * 2 + 0] = left;
|
||||
dst[i * 2 + 1] = right;
|
||||
}
|
||||
|
||||
buf->datas[0].chunk->offset = 0;
|
||||
buf->datas[0].chunk->stride = DEFAULT_CHANNELS * sizeof(int16_t);
|
||||
buf->datas[0].chunk->size = n_frames * DEFAULT_CHANNELS * sizeof(int16_t);
|
||||
|
||||
pw_stream_queue_buffer(self->stream_, b);
|
||||
}
|
||||
@@ -0,0 +1,41 @@
|
||||
#ifndef PIPEMINI_H
|
||||
#define PIPEMINI_H
|
||||
|
||||
#include <pipewire/pipewire.h>
|
||||
#include <spa/param/audio/format-utils.h>
|
||||
#include <stdint.h>
|
||||
#include <functional>
|
||||
|
||||
// Simple PipeWire audio output interface.
|
||||
// Calls a user-provided callback once per sample at the configured sample rate.
|
||||
// The callback writes one 16-bit stereo sample (left, right).
|
||||
class PipeMini {
|
||||
public:
|
||||
using Callback = std::function<void(int16_t* left, int16_t* right)>;
|
||||
|
||||
PipeMini();
|
||||
~PipeMini();
|
||||
|
||||
// Start the stream with the given callback.
|
||||
// sample_rate is typically 48000.
|
||||
bool start(const Callback& cb, int sample_rate = 48000);
|
||||
void stop();
|
||||
|
||||
public:
|
||||
static void on_process(void* userdata);
|
||||
|
||||
void teardown();
|
||||
|
||||
private:
|
||||
pw_main_loop* loop_ = nullptr;
|
||||
pw_stream* stream_ = nullptr;
|
||||
pw_core* core_ = nullptr;
|
||||
pw_context* context_ = nullptr;
|
||||
spa_hook stream_listener_ = {};
|
||||
Callback callback_;
|
||||
int sample_rate_ = 48000;
|
||||
bool running_ = false;
|
||||
struct PipeMiniData* data_ = nullptr;
|
||||
};
|
||||
|
||||
#endif // PIPEMINI_H
|
||||
Reference in New Issue
Block a user