Add render worker IPC loop

This commit is contained in:
2026-06-04 13:57:36 +08:00
parent 8c1c7658d9
commit b23e01b916
32 changed files with 2346 additions and 5651 deletions
+1
View File
@@ -12,6 +12,7 @@ add_executable(olive-gtest
render_audioparams_branch_test.cpp
render_sampleformat_test.cpp
render_pixelformat_test.cpp
render_ipc_test.cpp
project_serializer_test.cpp
timeline_marker_test.cpp
undo_stack_test.cpp
+4
View File
@@ -50,6 +50,10 @@ TEST(ProjectSerializer, SaveLoadProjectRoundTrip)
olive::ProjectSerializer::kProject);
EXPECT_EQ(result.code(), olive::ProjectSerializer::kSuccess);
EXPECT_FALSE(loaded_project.nodes().isEmpty());
ASSERT_TRUE(result.GetLoadData().node_ptrs.contains(
reinterpret_cast<quintptr>(node)));
EXPECT_TRUE(loaded_project.nodes().contains(
result.GetLoadData().node_ptrs.value(reinterpret_cast<quintptr>(node))));
olive::ProjectSerializer::Destroy();
if (created_disk_manager) {
+381
View File
@@ -0,0 +1,381 @@
/*
* Oak Video Editor - Render IPC Primitive Tests
* Copyright (C) 2026 Oak Team
*
* Unit tests for the lock-free cross-process render IPC primitives:
* - SpscRingBuffer (single-producer/single-consumer lock-free index queue)
* - FrameSlotPool (shared-memory frame slot hand-off via two SPSC rings)
* - NDJSON control message encode/decode and framing
*
* The threaded tests stress the lock-free invariants (no loss, no duplication, FIFO order) and are
* intended to be run under ThreadSanitizer in CI as well.
*/
#include <gtest/gtest.h>
#include <atomic>
#include <cstring>
#include <thread>
#include <vector>
#include <QBuffer>
#include <QJsonDocument>
#include <QJsonObject>
#include "render/ipc/frameslotpool.h"
#include "render/ipc/ipcmessage.h"
#include "render/ipc/spscringbuffer.h"
using namespace olive::ipc;
// ============================================================================
// SpscRingBuffer
// ============================================================================
TEST(SpscRingBuffer, BasicPushPopAndCapacity)
{
std::vector<uint8_t> mem(SpscRingBuffer::BytesNeeded(4));
SpscRingBuffer *ring = SpscRingBuffer::Create(mem.data(), 4);
EXPECT_TRUE(ring->IsEmptyApprox());
uint32_t v = 0;
EXPECT_FALSE(ring->Pop(&v)); // empty
// Capacity 4 holds at most 3 entries (one slot reserved to disambiguate full/empty).
EXPECT_TRUE(ring->Push(10));
EXPECT_TRUE(ring->Push(20));
EXPECT_TRUE(ring->Push(30));
EXPECT_FALSE(ring->Push(40)); // full
EXPECT_TRUE(ring->Pop(&v));
EXPECT_EQ(v, 10u);
EXPECT_TRUE(ring->Pop(&v));
EXPECT_EQ(v, 20u);
EXPECT_TRUE(ring->Pop(&v));
EXPECT_EQ(v, 30u);
EXPECT_FALSE(ring->Pop(&v)); // empty again
}
TEST(SpscRingBuffer, WrapAround)
{
std::vector<uint8_t> mem(SpscRingBuffer::BytesNeeded(4));
SpscRingBuffer *ring = SpscRingBuffer::Create(mem.data(), 4);
// Repeatedly pushing then popping single values forces the cursors past the backing array end.
for (uint32_t i = 0; i < 100; i++) {
ASSERT_TRUE(ring->Push(i));
uint32_t got = 0;
ASSERT_TRUE(ring->Pop(&got));
EXPECT_EQ(got, i);
}
EXPECT_TRUE(ring->IsEmptyApprox());
}
TEST(SpscRingBuffer, ConcurrentProducerConsumer)
{
constexpr uint32_t kCapacity = 1024;
constexpr uint32_t kCount = 2'000'000; // values 0..kCount-1 streamed through the ring
std::vector<uint8_t> mem(SpscRingBuffer::BytesNeeded(kCapacity));
SpscRingBuffer *ring = SpscRingBuffer::Create(mem.data(), kCapacity);
std::atomic<bool> order_ok{true};
std::thread producer([&] {
for (uint32_t i = 0; i < kCount; i++) {
while (!ring->Push(i)) {
std::this_thread::yield(); // buffer full, spin until consumer drains
}
}
});
std::thread consumer([&] {
// Every value must arrive exactly once and strictly in order (FIFO).
uint32_t expected = 0;
while (expected < kCount) {
uint32_t got = 0;
if (ring->Pop(&got)) {
if (got != expected) {
order_ok.store(false);
return;
}
expected++;
} else {
std::this_thread::yield();
}
}
});
producer.join();
consumer.join();
EXPECT_TRUE(order_ok.load());
EXPECT_TRUE(ring->IsEmptyApprox());
}
// ============================================================================
// FrameSlotPool
// ============================================================================
TEST(FrameSlotPool, SingleThreadedHandoff)
{
constexpr uint32_t kSlots = 3;
constexpr size_t kSlotBytes = 256;
std::vector<uint8_t> mem(FrameSlotPool::BytesNeeded(kSlots, kSlotBytes));
FrameSlotPool filler = FrameSlotPool::Create(mem.data(), kSlots, kSlotBytes);
FrameSlotPool drainer = FrameSlotPool::Attach(mem.data());
ASSERT_TRUE(filler.IsValid());
ASSERT_TRUE(drainer.IsValid());
EXPECT_EQ(drainer.slot_count(), kSlots);
EXPECT_EQ(drainer.slot_data_bytes(), kSlotBytes);
// Fill one slot with a recognizable pattern + metadata, publish, then drain and verify.
uint32_t idx = 0;
ASSERT_TRUE(filler.Acquire(&idx));
auto *data = static_cast<uint8_t *>(filler.SlotData(idx));
for (size_t i = 0; i < kSlotBytes; i++) {
data[i] = uint8_t(i & 0xFF);
}
FrameSlotMeta *meta = filler.Meta(idx);
meta->id = 4242;
meta->width = 16;
meta->height = 8;
meta->data_size = int32_t(kSlotBytes);
ASSERT_TRUE(filler.Publish(idx));
uint32_t got_idx = 0;
ASSERT_TRUE(drainer.Consume(&got_idx));
EXPECT_EQ(got_idx, idx);
const FrameSlotMeta *got_meta = drainer.Meta(got_idx);
EXPECT_EQ(got_meta->id, 4242);
EXPECT_EQ(got_meta->width, 16);
const auto *got_data = static_cast<const uint8_t *>(drainer.SlotData(got_idx));
for (size_t i = 0; i < kSlotBytes; i++) {
ASSERT_EQ(got_data[i], uint8_t(i & 0xFF));
}
EXPECT_TRUE(drainer.Release(got_idx));
}
TEST(FrameSlotPool, ExhaustionAndRefill)
{
constexpr uint32_t kSlots = 3;
constexpr size_t kSlotBytes = 64;
std::vector<uint8_t> mem(FrameSlotPool::BytesNeeded(kSlots, kSlotBytes));
FrameSlotPool pool = FrameSlotPool::Create(mem.data(), kSlots, kSlotBytes);
// Acquire every slot, then confirm the pool reports empty.
std::vector<uint32_t> held;
for (uint32_t i = 0; i < kSlots; i++) {
uint32_t a = 0;
ASSERT_TRUE(pool.Acquire(&a));
held.push_back(a);
}
uint32_t overflow = 0;
EXPECT_FALSE(pool.Acquire(&overflow)); // pool exhausted
// Publishing then consuming + releasing returns the slots to the free pool.
for (uint32_t idx : held) {
ASSERT_TRUE(pool.Publish(idx));
}
for (uint32_t i = 0; i < kSlots; i++) {
uint32_t c = 0;
ASSERT_TRUE(pool.Consume(&c));
ASSERT_TRUE(pool.Release(c));
}
uint32_t again = 0;
EXPECT_TRUE(pool.Acquire(&again)); // free again
}
TEST(FrameSlotPool, ConcurrentFillDrainIntegrity)
{
constexpr uint32_t kSlots = 8;
constexpr size_t kSlotBytes = 4096;
constexpr int64_t kFrames = 200'000;
std::vector<uint8_t> mem(FrameSlotPool::BytesNeeded(kSlots, kSlotBytes));
FrameSlotPool filler = FrameSlotPool::Create(mem.data(), kSlots, kSlotBytes);
FrameSlotPool drainer = FrameSlotPool::Attach(mem.data());
std::atomic<bool> integrity_ok{true};
// Filler: for each frame id, acquire a slot, stamp the id into meta and a pattern into the data,
// publish. Spins when no slot is free (this is the natural backpressure path).
std::thread fill_thread([&] {
for (int64_t id = 0; id < kFrames; id++) {
uint32_t idx = 0;
while (!filler.Acquire(&idx)) {
std::this_thread::yield();
}
filler.Meta(idx)->id = id;
auto *d = static_cast<uint8_t *>(filler.SlotData(idx));
const uint8_t pat = uint8_t(id & 0xFF);
memset(d, pat, kSlotBytes);
while (!filler.Publish(idx)) {
std::this_thread::yield(); // ready ring transiently full
}
}
});
// Drainer: consume in order, verify the id is monotonic and the data matches the id pattern,
// then release the slot back to the filler.
std::thread drain_thread([&] {
int64_t expected = 0;
while (expected < kFrames) {
uint32_t idx = 0;
if (!drainer.Consume(&idx)) {
std::this_thread::yield();
continue;
}
const FrameSlotMeta *m = drainer.Meta(idx);
if (m->id != expected) {
integrity_ok.store(false);
return;
}
const auto *d = static_cast<const uint8_t *>(drainer.SlotData(idx));
const uint8_t pat = uint8_t(expected & 0xFF);
if (d[0] != pat || d[kSlotBytes - 1] != pat) {
integrity_ok.store(false);
return;
}
while (!drainer.Release(idx)) {
std::this_thread::yield();
}
expected++;
}
});
fill_thread.join();
drain_thread.join();
EXPECT_TRUE(integrity_ok.load());
}
// ============================================================================
// NDJSON control messages
// ============================================================================
TEST(IpcMessage, TypedRoundTrip)
{
// Write several typed messages into a buffer, then drain and parse them back the way a pipe
// reader would.
QByteArray storage;
QBuffer dev(&storage);
ASSERT_TRUE(dev.open(QIODevice::WriteOnly));
HandshakeMsg hs;
hs.protocol_version = 1;
hs.shm_key = QStringLiteral("olive-rw-1234-0");
hs.input_slots = 4;
hs.output_slots = 6;
hs.slot_data_bytes = 256ll * 1024 * 1024;
ASSERT_TRUE(WriteMessage(&dev, hs.ToJson()));
RenderFrameMsg rf;
rf.ticket_id = 99;
rf.node_uuid = QStringLiteral("{abcd-1234}");
rf.time_num = 1001;
rf.time_den = 30000;
rf.width = 1920;
rf.height = 1080;
rf.format = 3;
rf.channel_count = 4;
rf.mode = 1;
ASSERT_TRUE(WriteMessage(&dev, rf.ToJson()));
FrameReadyMsg fr;
fr.ticket_id = 99;
fr.output_slot = 2;
ASSERT_TRUE(WriteMessage(&dev, fr.ToJson()));
dev.close();
QByteArray reader = storage;
QJsonObject obj;
bool ok = false;
ASSERT_TRUE(ReadMessage(&reader, &obj, &ok));
ASSERT_TRUE(ok);
HandshakeMsg hs2;
ASSERT_TRUE(HandshakeMsg::FromJson(obj, &hs2));
EXPECT_EQ(hs2.protocol_version, 1);
EXPECT_EQ(hs2.shm_key, hs.shm_key);
EXPECT_EQ(hs2.input_slots, 4);
EXPECT_EQ(hs2.output_slots, 6);
EXPECT_EQ(hs2.slot_data_bytes, hs.slot_data_bytes);
ASSERT_TRUE(ReadMessage(&reader, &obj, &ok));
ASSERT_TRUE(ok);
RenderFrameMsg rf2;
ASSERT_TRUE(RenderFrameMsg::FromJson(obj, &rf2));
EXPECT_EQ(rf2.ticket_id, 99);
EXPECT_EQ(rf2.node_uuid, rf.node_uuid);
EXPECT_EQ(rf2.time_num, 1001);
EXPECT_EQ(rf2.time_den, 30000);
EXPECT_EQ(rf2.width, 1920);
EXPECT_EQ(rf2.format, 3);
ASSERT_TRUE(ReadMessage(&reader, &obj, &ok));
ASSERT_TRUE(ok);
FrameReadyMsg fr2;
ASSERT_TRUE(FrameReadyMsg::FromJson(obj, &fr2));
EXPECT_EQ(fr2.ticket_id, 99);
EXPECT_EQ(fr2.output_slot, 2);
// No more complete lines remain.
EXPECT_FALSE(ReadMessage(&reader, &obj, &ok));
}
TEST(IpcMessage, PartialFrameByteByByte)
{
CancelMsg c;
c.ticket_id = 7;
const QByteArray full =
QByteArray(QJsonDocument(c.ToJson()).toJson(QJsonDocument::Compact)) + '\n';
// Feed the bytes one at a time; ReadMessage must return false until the terminating '\n'.
QByteArray reader;
QJsonObject obj;
bool ok = false;
for (int i = 0; i < full.size() - 1; i++) {
reader.append(full.at(i));
ASSERT_FALSE(ReadMessage(&reader, &obj, &ok)); // no complete line yet
}
reader.append(full.at(full.size() - 1)); // the trailing newline
ASSERT_TRUE(ReadMessage(&reader, &obj, &ok));
ASSERT_TRUE(ok);
CancelMsg c2;
ASSERT_TRUE(CancelMsg::FromJson(obj, &c2));
EXPECT_EQ(c2.ticket_id, 7);
}
TEST(IpcMessage, MalformedLineIsSkipped)
{
QByteArray reader = QByteArray("this is not json\n");
QJsonObject obj;
bool ok = true;
// A complete but malformed line is consumed and reported as not-ok, leaving the buffer drained.
EXPECT_FALSE(ReadMessage(&reader, &obj, &ok));
EXPECT_FALSE(ok);
EXPECT_TRUE(reader.isEmpty());
}
TEST(IpcMessage, WrongTypeRejected)
{
// FromJson must reject an object whose "type" does not match the target struct.
HandshakeMsg hs;
hs.protocol_version = 1;
const QJsonObject obj = hs.ToJson();
RenderFrameMsg rf;
EXPECT_FALSE(RenderFrameMsg::FromJson(obj, &rf));
}