Revision control
Copy as Markdown
// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#include "testvideosink_p.h"
#include <qsignalspy.h>
#include <qtestsystem.h>
QT_BEGIN_NAMESPACE
TestVideoSink::TestVideoSink(bool storeFrames) : m_storeFrames(storeFrames)
{
connect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::addVideoFrame);
connect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::videoFrameChangedSync);
}
QVideoFrame TestVideoSink::waitForFrame()
{
QSignalSpy spy(this, &TestVideoSink::videoFrameChangedSync);
return spy.wait() ? spy.at(0).at(0).value<QVideoFrame>() : QVideoFrame{};
}
void TestVideoSink::addVideoFrame(const QVideoFrame &frame)
{
m_elapsedTimer.start();
if (m_storeFrames)
m_frameList.append(frame);
if (frame.isValid())
m_frameTimes.emplace_back(std::chrono::microseconds(frame.startTime()));
++m_totalFrames;
}
void TestVideoSink::setStoreImagesEnabled(bool storeImages)
{
if (storeImages)
connect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::storeImage,
Qt::UniqueConnection);
else
disconnect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::storeImage);
}
void TestVideoSink::storeImage(const QVideoFrame &frame)
{
QImage image = frame.toImage();
image.detach();
m_images.push_back(std::move(image));
}
std::chrono::milliseconds TestVideoSink::durationBetweenFrames(qsizetype frameCount)
{
Q_ASSERT(frameCount > 0);
QElapsedTimer timer;
qsizetype framesReceived = 0;
QObject context;
connect(this, &QVideoSink::videoFrameChanged, &context, [&] {
if (framesReceived++ == 0)
timer.start();
});
auto allFramesAreReceived = [&]() {
return framesReceived > frameCount;
};
using namespace std::chrono;
return QTest::qWaitFor(allFramesAreReceived, seconds(10))
? milliseconds(timer.elapsed() / frameCount)
: milliseconds(0);
}
QT_END_NAMESPACE
#include "moc_testvideosink_p.cpp"