Files
oak-editor/tests/gtest/viewer_display_repro_test.cpp
T
Mike-Solar 5b68bbeec8 cli/worker/tests: facade-only linkage and updated test suites
oak-cli and oak-render-worker use the C ABI exclusively (0 olive::
symbols). Google Test suites updated for the migrated APIs, new engine
facade test coverage, CLI transcode verification.
2026-07-26 22:43:32 +08:00

405 lines
12 KiB
C++

/*
* Regression test for the direct-connection black screen bug. Drives the real
* ViewerWidget display path offscreen and checks the pixels the display
* widget actually paints, for both the direct (footage -> output) and
* indirect (footage -> effect -> output) wiring, plus runtime rewiring
* between the two.
*/
#include <gtest/gtest.h>
#include <QCoreApplication>
#include <QDir>
#include <QElapsedTimer>
#include <QFileInfo>
#include <QImage>
#include <QPixmap>
#include <QThread>
#include "audio/audiomanager.h"
#include "codec/conformmanager.h"
#include "codec/proxymanager.h"
#include "config/config.h"
#include "core.h"
#include "node/color/colormanager/colormanager.h"
#include "node/effect/opacity/opacityeffect.h"
#include "node/factory.h"
#include "node/output/viewer/viewer.h"
#include "node/project.h"
#include "node/project/footage/footage.h"
#include "node/project/sequence/sequence.h"
#include "node/project/serializer/serializer.h"
#include "render/diskmanager.h"
#include "render/framemanager.h"
#include "render/previewautocacher.h"
#include "render/rendermanager.h"
#include "task/taskmanager.h"
#include "widget/viewer/footageviewer.h"
#include "widget/viewer/viewer.h"
#include "widget/viewer/viewerdisplay.h"
#ifdef OAK_ENABLE_DYNAMIC_RENDER_BACKEND
#include "render/backend/dynamicrenderer.h"
#include "render/backend/renderbackend_c.h"
#endif
using namespace olive;
namespace
{
QString demo_video_path_t()
{
return QDir(QStringLiteral(OAK_TEST_SOURCE_DIR))
.filePath(QStringLiteral("tests/demo.mp4"));
}
QString worker_binary_path_t()
{
QDir dir(QCoreApplication::applicationDirPath());
dir.cdUp();
dir.cdUp();
dir.cd(QStringLiteral("worker"));
#if defined(_WIN32)
return dir.filePath(QStringLiteral("oak-render-worker.exe"));
#else
return dir.filePath(QStringLiteral("oak-render-worker"));
#endif
}
// Loads and initializes the requested render backend the same way the
// application does, verifying that the loaded backend is actually of the
// requested kind (a Vulkan request that fell back to OpenGL does not count).
bool is_render_backend_available(const QString &backend)
{
#ifdef OAK_ENABLE_DYNAMIC_RENDER_BACKEND
olive::DynamicRenderer renderer(backend);
if (!renderer.load()) {
return false;
}
OakRenderBackendInfo info = {};
if (!renderer.get_backend_info(&info)) {
return false;
}
if (backend == QStringLiteral("vulkan") &&
info.kind != oak_render_backend_vulkan) {
return false;
}
if (backend == QStringLiteral("opengl") &&
info.kind != oak_render_backend_opengl) {
return false;
}
return renderer.init();
#else
Q_UNUSED(backend)
return false;
#endif
}
double brightness_of_widget(QWidget *w)
{
QPixmap pm = w->grab();
QImage img = pm.toImage().convertToFormat(QImage::Format_RGB32);
if (img.isNull()) {
return -1.0;
}
double sum = 0;
int n = 0;
for (int y = img.height() / 4; y < img.height() * 3 / 4; y += 8) {
for (int x = img.width() / 4; x < img.width() * 3 / 4; x += 8) {
QRgb px = img.pixel(x, y);
sum += qRed(px) + qGreen(px) + qBlue(px);
n += 3;
}
}
return n ? sum / n / 255.0 : -1.0;
}
class TestViewerWidget : public ViewerWidget {
public:
using ViewerWidget::display_widget;
using ViewerWidget::ViewerWidget;
};
class TestFootageViewerWidget : public FootageViewerWidget {
public:
using FootageViewerWidget::display_widget;
using FootageViewerWidget::FootageViewerWidget;
};
} // namespace
class ViewerDisplayReproTest : public ::testing::TestWithParam<QString> {
protected:
static void SetUpTestSuite()
{
NodeFactory::initialize();
ColorManager::set_up_default_config();
TaskManager::create_instance();
ConformManager::create_instance();
ProxyManager::create_instance();
FrameManager::create_instance();
ProjectSerializer::initialize();
DiskManager::create_instance();
const QString worker = worker_binary_path_t();
if (QFileInfo::exists(worker)) {
qputenv("OAK_RENDER_WORKER", QFile::encodeName(worker));
}
if (!Core::instance()) {
new Core();
}
AudioManager::create_instance();
}
static void TearDownTestSuite()
{
AudioManager::destroy_instance();
DiskManager::destroy_instance();
ProjectSerializer::destroy();
FrameManager::destroy_instance();
ProxyManager::destroy_instance();
ConformManager::destroy_instance();
TaskManager::destroy_instance();
NodeFactory::destroy();
}
void SetUp() override
{
backend_ = GetParam();
if (backend_ != QStringLiteral("vulkan")) {
GTEST_SKIP() << "offscreen QOpenGLWidget cannot paint; Vulkan only";
}
if (!is_render_backend_available(backend_)) {
GTEST_SKIP() << "Render backend is not available: "
<< backend_.toStdString();
}
Config::current()[QStringLiteral("GraphicsBackend")] = backend_;
demo_path_ = demo_video_path_t();
ASSERT_TRUE(QFileInfo::exists(demo_path_));
project_ = std::make_unique<Project>();
project_->initialize();
footage_ = new Footage(demo_path_);
footage_->setParent(project_.get());
ASSERT_TRUE(footage_->is_valid());
RenderManager::create_instance();
RenderManager::instance()->get_cacher()->set_project(project_.get());
}
void TearDown() override
{
// May be null when SetUp() skipped before creating the instance.
if (RenderManager::instance()) {
RenderManager::instance()->get_cacher()->set_project(nullptr);
RenderManager::destroy_instance();
}
project_.reset();
}
// Pumps the event loop until the display widget has a texture or timeout.
bool wait_for_texture(ViewerDisplayWidget *display, int timeout_ms = 30000)
{
QElapsedTimer timer;
timer.start();
while (!timer.hasExpired(timeout_ms)) {
QCoreApplication::processEvents(QEventLoop::AllEvents, 50);
if (display->get_current_texture()) {
// Let a few more paints happen
for (int i = 0; i < 5; i++) {
QCoreApplication::processEvents(QEventLoop::AllEvents, 50);
QThread::msleep(20);
}
return true;
}
QThread::msleep(20);
}
return false;
}
QString backend_;
QString demo_path_;
std::unique_ptr<Project> project_;
Footage *footage_ = nullptr;
};
// Footage previewed in the footage viewer (the "direct" case: the footage
// node itself is the viewer output, nothing in between).
TEST_P(ViewerDisplayReproTest, FootageViewerNotBlack)
{
TestFootageViewerWidget *viewer = new TestFootageViewerWidget();
viewer->resize(800, 600);
viewer->show();
viewer->connect_viewer_node(footage_);
ASSERT_TRUE(wait_for_texture(viewer->display_widget()))
<< "Display widget never received a texture (backend="
<< backend_.toStdString() << ")";
double brightness = brightness_of_widget(viewer->display_widget());
EXPECT_GT(brightness, 0.01)
<< "Footage viewer paints BLACK (brightness=" << brightness
<< ", backend=" << backend_.toStdString() << ")";
delete viewer;
}
// Sequence with footage -> opacity -> sequence output (the "indirect" case).
TEST_P(ViewerDisplayReproTest, SequenceViewerIndirectNotBlack)
{
Sequence *sequence = new Sequence();
sequence->setParent(project_.get());
OpacityEffect *opacity = new OpacityEffect();
opacity->setParent(project_.get());
Node::connect_edge(footage_, NodeInput(opacity, OpacityEffect::k_texture_input));
Node::connect_edge(opacity, NodeInput(sequence, ViewerOutput::k_texture_input));
TestViewerWidget *viewer = new TestViewerWidget();
viewer->resize(800, 600);
viewer->show();
viewer->connect_viewer_node(sequence);
ASSERT_TRUE(wait_for_texture(viewer->display_widget()))
<< "Display widget never received a texture (backend="
<< backend_.toStdString() << ")";
double brightness = brightness_of_widget(viewer->display_widget());
EXPECT_GT(brightness, 0.01)
<< "Sequence viewer (indirect) paints BLACK (brightness=" << brightness
<< ", backend=" << backend_.toStdString() << ")";
delete viewer;
}
// Sequence with footage wired directly to the output (node-editor "direct"
// case).
TEST_P(ViewerDisplayReproTest, SequenceViewerDirectNotBlack)
{
Sequence *sequence = new Sequence();
sequence->setParent(project_.get());
Node::connect_edge(footage_,
NodeInput(sequence, ViewerOutput::k_texture_input));
TestViewerWidget *viewer = new TestViewerWidget();
viewer->resize(800, 600);
viewer->show();
viewer->connect_viewer_node(sequence);
ASSERT_TRUE(wait_for_texture(viewer->display_widget()))
<< "Display widget never received a texture (backend="
<< backend_.toStdString() << ")";
double brightness = brightness_of_widget(viewer->display_widget());
EXPECT_GT(brightness, 0.01)
<< "Sequence viewer (direct) paints BLACK (brightness=" << brightness
<< ", backend=" << backend_.toStdString() << ")";
delete viewer;
}
INSTANTIATE_TEST_SUITE_P(Backends, ViewerDisplayReproTest,
::testing::Values(QStringLiteral("opengl"),
QStringLiteral("vulkan")));
// Simulates the user's actual workflow: a sequence is playing through its
// normal chain, then at RUNTIME the input is rewired to connect the footage
// directly to the output. Vulkan only (offscreen QOpenGLWidget cannot paint).
class ViewerRuntimeRewireTest : public ViewerDisplayReproTest {
protected:
double pump_and_measure(TestViewerWidget *viewer, int timeout_ms = 30000)
{
if (!wait_for_texture(viewer->display_widget(), timeout_ms)) {
return -1.0;
}
return brightness_of_widget(viewer->display_widget());
}
};
TEST_P(ViewerRuntimeRewireTest, RewireToDirectConnectionNotBlack)
{
Sequence *sequence = new Sequence();
sequence->setParent(project_.get());
// Normal chain with a node in between: footage -> opacity -> sequence
OpacityEffect *opacity = new OpacityEffect();
opacity->setParent(project_.get());
Node::connect_edge(footage_,
NodeInput(opacity, OpacityEffect::k_texture_input));
Node::connect_edge(opacity, NodeInput(sequence, ViewerOutput::k_texture_input));
TestViewerWidget *viewer = new TestViewerWidget();
viewer->resize(800, 600);
viewer->show();
viewer->connect_viewer_node(sequence);
double brightness = pump_and_measure(viewer);
ASSERT_GT(brightness, 0.01)
<< "Precondition failed: indirect chain is already black";
// Now rewire at runtime: footage directly to the sequence output.
Node::disconnect_edge(opacity,
NodeInput(sequence, ViewerOutput::k_texture_input));
Node::connect_edge(footage_,
NodeInput(sequence, ViewerOutput::k_texture_input));
brightness = pump_and_measure(viewer);
EXPECT_GT(brightness, 0.01)
<< "Viewer paints BLACK after rewiring to a direct connection "
<< "(brightness=" << brightness << ")";
delete viewer;
}
TEST_P(ViewerRuntimeRewireTest, RewireToIndirectConnectionNotBlack)
{
Sequence *sequence = new Sequence();
sequence->setParent(project_.get());
// Start direct: footage -> sequence
Node::connect_edge(footage_,
NodeInput(sequence, ViewerOutput::k_texture_input));
TestViewerWidget *viewer = new TestViewerWidget();
viewer->resize(800, 600);
viewer->show();
viewer->connect_viewer_node(sequence);
double brightness = pump_and_measure(viewer);
ASSERT_GT(brightness, 0.01)
<< "Precondition failed: direct chain is already black";
// Insert a node at runtime: footage -> opacity -> sequence
OpacityEffect *opacity = new OpacityEffect();
opacity->setParent(project_.get());
Node::disconnect_edge(footage_,
NodeInput(sequence, ViewerOutput::k_texture_input));
Node::connect_edge(footage_,
NodeInput(opacity, OpacityEffect::k_texture_input));
Node::connect_edge(opacity, NodeInput(sequence, ViewerOutput::k_texture_input));
brightness = pump_and_measure(viewer);
EXPECT_GT(brightness, 0.01)
<< "Viewer paints BLACK after rewiring to an indirect connection "
<< "(brightness=" << brightness << ")";
delete viewer;
}
INSTANTIATE_TEST_SUITE_P(Backends, ViewerRuntimeRewireTest,
::testing::Values(QStringLiteral("opengl"),
QStringLiteral("vulkan")));