framegrabber_p.h

Go to the documentation of this file.

// Copyright (C) 2026 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
 
#ifndef FRAMEGRABBER_P_H
#define FRAMEGRABBER_P_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 <QtCore/qtconfigmacros.h>
#include <QtMultimedia/qvideoframe.h>
#include <QtMultimedia/qvideosink.h>
 
#include <chrono>
#include <optional>
#include <vector>
 
QT_BEGIN_NAMESPACE
 
namespace QtMultimediaTest {
 
// Generous timeout on CI, where machines are slow and heavily loaded. Short for
// blacklisted test cases, which are expected to fail.
[[nodiscard]] std::chrono::milliseconds surfaceCaptureTestTimeout();
 
struct VideoFrameInfo
{
    bool isValid = false;
    QSize size;
    QVideoFrameFormat::PixelFormat pixelFormat = QVideoFrameFormat::Format_Invalid;
    qint64 startTime = -1;
    qint64 endTime = -1;
};
 
/*!
    The FrameGrabber stores frames that arrive from a surface capture,
    and is used to inspect captured frames in the tests.
*/
class FrameGrabber : public QVideoSink
{
    Q_OBJECT
 
public:
    FrameGrabber();
 
    [[nodiscard]] const std::vector<VideoFrameInfo> &getFrameInfos() const;
 
    /*!
        Wait for at least \a minCount frames that are no older than noOlderThanTime.
 
        If \a expectedFrameRate is given, the timeout is extended by one frame interval per
        frame, rather than being a flat global timeout.
 
        Returns empty if not enough frames arrived, or if grabber was stopped before global timeout
        elapsed.
    */
    [[nodiscard]] std::vector<VideoFrameInfo> waitAndTakeFrameInfos(
        size_t minCount,
        std::optional<qreal> expectedFrameRate = std::nullopt,
        qint64 noOlderThanTime = 0);
 
    /*!
        Same as waitAndTakeFrameInfos(), but returns the video frames themselves.
        Video frames are only retained for the duration of this call.
    */
    [[nodiscard]] std::vector<QVideoFrame> waitAndTakeVideoFrames(
        size_t minCount,
        std::optional<qreal> expectedFrameRate = std::nullopt,
        qint64 noOlderThanTime = 0);
 
    /*!
        Waits for the first frame of the current stream, and returns a copy of its
        info. The capture becoming active marks the start of a new stream, so the
        next valid frame after it is the first frame of that stream. Calling this
        again within the same stream returns the same info.
 
        Returns std::nullopt if no first frame arrived before the global timeout,
        or if the grabber was stopped. Can never return an invalid frame.
    */
    [[nodiscard]] std::optional<VideoFrameInfo> tryWaitForFirstFrameInfo();
 
    /*!
        Same as tryWaitForFirstFrameInfo(), but returns the video frame itself, and
        consumes it. The first frame of a stream is only kept alive if
        setRetainFirstVideoFrame() was enabled before the stream started, since
        holding on to frames can stall backends that use a fixed-size frame pool.
    */
    [[nodiscard]] std::optional<QVideoFrame> consumeFirstVideoFrame();
 
    void setRetainFirstVideoFrame(bool retain);
 
    /*!
        Measures the average duration between \a frameCount consecutive frames.
 
        If \a expectedFrameRate is given, a frame that takes significantly longer than
        the frame interval to arrive ends the wait early. The first frame is still
        allowed the global timeout, since backend initialization can delay it.
 
        Returns std::nullopt if a frame stalled, or if not enough frames arrived.
    */
    [[nodiscard]] std::optional<std::chrono::milliseconds> durationBetweenFrames(
        qsizetype frameCount = 1,
        std::optional<qreal> expectedFrameRate = std::nullopt);
 
    [[nodiscard]] bool isStopped() const;
 
public slots:
    void stop();
 
    /*!
        Starts tracking a new stream when the capture becomes active, so that
        tryWaitForFirstFrameInfo() waits for the first frame of it.
    */
    void onCaptureActiveChanged(bool active);
 
private:
    void onFrameReceived(const QVideoFrame &frame);
    [[nodiscard]] bool waitForFirstFrame();
    void resetFirstFrame();
 
    template <typename Frame>
    [[nodiscard]] std::vector<Frame> waitAndTake(
        std::vector<Frame> &frames,
        size_t minCount,
        std::optional<qreal> expectedFrameRate,
        qint64 noOlderThanTime);
 
    std::vector<VideoFrameInfo> m_frameInfos;
 
    // Only set, and m_videoFrames only populated, during waitAndTakeVideoFrames().
    bool m_retainVideoFrames = false;
    std::vector<QVideoFrame> m_videoFrames;
 
    // The first frame of the current stream, until it is consumed.
    std::optional<VideoFrameInfo> m_firstFrameInfo;
    // Same frame as above. Only populated if m_retainFirstVideoFrame is set.
    bool m_retainFirstVideoFrame = false;
    std::optional<QVideoFrame> m_firstVideoFrame;
    // Whether we have received the first frame of the current stream.
    bool m_streamStarted = false;
 
    bool m_stopped = false;
};
 
} // namespace QtMultimediaTest
 
QT_END_NAMESPACE
 
#endif // FRAMEGRABBER_P_H