testvideosink_p.h

Go to the documentation of this file.

// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
 
#ifndef TESTVIDEOSINK_H
#define TESTVIDEOSINK_H
 
//
//  W A R N I N G
//  -------------
//
// This file is not part of the Qt API. It exists purely as an
// implementation detail. This header file may change from version to
// version without notice, or even be removed.
//
// We mean it.
//
 
#include <qvideosink.h>
#include <qvideoframe.h>
#include <qelapsedtimer.h>
#include <qimage.h>
#include <chrono>
#include <vector>
 
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);
 
    QVideoFrame waitForFrame();
 
    void setStoreFrames(bool storeFrames = true) { m_storeFrames = storeFrames; }
 
    void setStoreImagesEnabled(bool storeImages = true);
    const std::vector<QImage> &images() const { return m_images; }
 
    std::chrono::milliseconds durationBetweenFrames(qsizetype frameCount);
 
private Q_SLOTS:
    void addVideoFrame(const QVideoFrame &frame);
    void storeImage(const QVideoFrame &frame);
 
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;
    using TimePoint = std::chrono::time_point<std::chrono::high_resolution_clock>;
    std::vector<TimePoint> m_frameTimes;
 
private:
    bool m_storeFrames;
    std::vector<QImage> m_images;
};
 
QT_END_NAMESPACE
 
#endif // TESTVIDEOSINK_H