qvideoframeinput.cpp

Go to the documentation of this file.

// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
 
#include "qvideoframeinput.h"
#include "qmediaframeinput_p.h"
#include "qmediainputencoderinterface_p.h"
#include "qplatformvideoframeinput_p.h"
 
QT_BEGIN_NAMESPACE
 
class QVideoFrameInputPrivate : public QMediaFrameInputPrivate
{
public:
    QVideoFrameInputPrivate(QVideoFrameInput *q) : q(q) { }
 
    bool sendVideoFrame(const QVideoFrame &frame)
    {
        return sendMediaFrame([&]() { emit m_platfromVideoFrameInput->newVideoFrame(frame); });
    }
 
    void initialize(QVideoFrameFormat format = {})
    {
        m_platfromVideoFrameInput = std::make_unique<QPlatformVideoFrameInput>(std::move(format));
        addUpdateSignal(m_platfromVideoFrameInput.get(), &QPlatformVideoFrameInput::encoderUpdated);
    }
 
    void uninitialize()
    {
        m_platfromVideoFrameInput.reset();
 
        if (captureSession())
            captureSession()->setVideoFrameInput(nullptr);
    }
 
    QPlatformVideoFrameInput *platfromVideoFrameInput() const
    {
        return m_platfromVideoFrameInput.get();
    }
 
protected:
    void updateCaptureSessionConnections(QMediaCaptureSession *prevSession,
                                         QMediaCaptureSession *newSession) override
    {
        if (prevSession)
            removeUpdateSignal(prevSession, &QMediaCaptureSession::videoOutputChanged);
 
        if (newSession)
            addUpdateSignal(newSession, &QMediaCaptureSession::videoOutputChanged);
    }
 
    bool checkIfCanSendMediaFrame() const override
    {
        if (auto encoderInterface = m_platfromVideoFrameInput->encoderInterface())
            return encoderInterface->canPushFrame();
 
        return captureSession()->videoOutput() || captureSession()->videoSink();
    }
 
    void emitReadyToSendMediaFrame() override { emit q->readyToSendVideoFrame(); }
 
private:
    QVideoFrameInput *q = nullptr;
    std::unique_ptr<QPlatformVideoFrameInput> m_platfromVideoFrameInput;
};
 
QVideoFrameInput::QVideoFrameInput(QObject *parent) : QVideoFrameInput({}, parent) { }
 
QVideoFrameInput::QVideoFrameInput(const QVideoFrameFormat &format, QObject *parent)
    : QObject(*new QVideoFrameInputPrivate(this), parent)
{
    Q_D(QVideoFrameInput);
    d->initialize(format);
}
 
QVideoFrameInput::~QVideoFrameInput()
{
    Q_D(QVideoFrameInput);
    d->uninitialize();
}
 
bool QVideoFrameInput::sendVideoFrame(const QVideoFrame &frame)
{
    Q_D(QVideoFrameInput);
    return d->sendVideoFrame(frame);
}
 
QVideoFrameFormat QVideoFrameInput::format() const
{
    Q_D(const QVideoFrameInput);
    return d->platfromVideoFrameInput()->frameFormat();
}
 
QMediaCaptureSession *QVideoFrameInput::captureSession() const
{
    Q_D(const QVideoFrameInput);
    return d->captureSession();
}
 
void QVideoFrameInput::setCaptureSession(QMediaCaptureSession *captureSession)
{
    Q_D(QVideoFrameInput);
    d->setCaptureSession(captureSession);
}
 
QPlatformVideoFrameInput *QVideoFrameInput::platformVideoFrameInput() const
{
    Q_D(const QVideoFrameInput);
    return d->platfromVideoFrameInput();
}
 
QT_END_NAMESPACE