qffmpegvideoencoder_p.h

Enable keyboard shortcuts

// 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

#ifndef QFFMPEGVIDEOENCODER_P_H

#define QFFMPEGVIDEOENCODER_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 <QtFFmpegMediaPluginImpl/private/qffmpegencoderthread_p.h>

#include <QtFFmpegMediaPluginImpl/private/qffmpeg_p.h>

#include <QtFFmpegMediaPluginImpl/private/qffmpegvideoframeencoder_p.h>

#include <QtFFmpegMediaPluginImpl/private/qffmpegframerateadapter_p.h>

#include <qvideoframe.h>

#include <queue>

QT_BEGIN_NAMESPACE

class QVideoFrameFormat;

class QMediaEncoderSettings;

namespace QFFmpeg {

class VideoFrameEncoder;

class VideoEncoder : public EncoderThread

public:

    VideoEncoder(RecordingEngine &recordingEngine, const QMediaEncoderSettings &settings,

                 const QVideoFrameFormat &format, std::optional<AVPixelFormat> hwFormat);

    ~VideoEncoder() override;

    void addFrame(const QVideoFrame &frame);

protected:

    bool checkIfCanPushFrame() const override;

private:

    FrameInfo dequeueAdaptedFrame();

    void retrievePackets();

    [[nodiscard]] bool tryPushToInputQueue(const QVideoFrame &frame);

    void saturateEncoderQueue();

    void flushAdapter();

    bool init() override;

    void cleanup() override;

    bool hasData() const override;

    void processOne() override;

    void handlePausedChanged(bool paused) override;

    std::pair<qint64, qint64> frameTimeStamps(const QVideoFrame &frame) const;

    QMediaEncoderSettings m_settings;

    VideoFrameEncoder::SourceParams m_sourceParams;

    struct InputFrame

        QVideoFrame frame;

        bool restartTimeBase = false;

};

    std::queue<InputFrame> m_inputQueue;

    // Max queue size arbitrarily chosen to limit memory usage (332 MB @ 4K when including one

    // adapted frame in encoder queue)

    const size_t m_maxQueueSize = 9;

    std::queue<FrameInfo> m_encoderQueue;

    VideoFrameEncoderUPtr m_frameEncoder;

    qint64 m_baseTime = 0;

    bool m_shouldAdjustTimeBaseForNextFrame = true;

    qint64 m_lastFrameTime = 0;

    std::optional<qreal> m_fixedSourceFrameRate; // nullopt = variable-rate source

    FrameRateAdapter m_frameRateAdapter;

};

} // namespace QFFmpeg

QT_END_NAMESPACE

#endif