blob: 8c3f9a5bd49eb08b7785ffb0bb838fbb31eff6e1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#ifndef TESTVIDEOSINK_H
#define TESTVIDEOSINK_H
#include <qvideosink.h>
#include <qvideoframe.h>
#include <qelapsedtimer.h>
#include <qsignalspy.h>
QT_BEGIN_NAMESPACE
/*
This is a simple video surface which records all presented frames.
*/
class TestVideoSink : public QVideoSink
{
Q_OBJECT
public:
explicit TestVideoSink(bool storeFrames = false) : m_storeFrames(storeFrames)
{
connect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::addVideoFrame);
connect(this, &QVideoSink::videoFrameChanged, this, &TestVideoSink::videoFrameChangedSync);
}
QVideoFrame waitForFrame()
{
QSignalSpy spy(this, &TestVideoSink::videoFrameChangedSync);
return spy.wait() ? spy.at(0).at(0).value<QVideoFrame>() : QVideoFrame{};
}
void setStoreFrames(bool storeFrames = true) { m_storeFrames = storeFrames; }
private Q_SLOTS:
void addVideoFrame(const QVideoFrame &frame)
{
if (!m_elapsedTimer.isValid())
m_elapsedTimer.start();
else
m_elapsedTimer.restart();
if (m_storeFrames)
m_frameList.append(frame);
++m_totalFrames;
}
signals:
void videoFrameChangedSync(const QVideoFrame &frame);
public:
QList<QVideoFrame> m_frameList;
int m_totalFrames = 0; // used instead of the list when frames are not stored
QElapsedTimer m_elapsedTimer;
private:
bool m_storeFrames;
};
QT_END_NAMESPACE
#endif // TESTVIDEOSINK_H
|