qffmpegvideoframeencoder_p.h

Enable keyboard shortcuts

// Copyright (C) 2022 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 QFFMPEGVIDEOFRAMEENCODER_P_H

#define QFFMPEGVIDEOFRAMEENCODER_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/qffmpeghwaccel_p.h>

#include <QtFFmpegMediaPluginImpl/private/qffmpegrecordingengineutils_p.h>

#include <QtMultimedia/private/qplatformmediarecorder_p.h>

#include <QtMultimedia/private/qmultimediautils_p.h>

QT_BEGIN_NAMESPACE

class QMediaEncoderSettings;

namespace QFFmpeg {

class VideoFrameEncoder;

using VideoFrameEncoderUPtr = std::unique_ptr<VideoFrameEncoder>;

struct ScoredPixelFormat;

class VideoFrameEncoder

public:

    struct SourceParams

        QSize size;

        AVPixelFormat format = AV_PIX_FMT_NONE;

        AVPixelFormat swFormat = AV_PIX_FMT_NONE;

        VideoTransformation transform;

        qreal frameRate = 0.;

        AVColorTransferCharacteristic colorTransfer = AVCOL_TRC_UNSPECIFIED;

        AVColorSpace colorSpace = AVCOL_SPC_UNSPECIFIED;

        AVColorRange colorRange = AVCOL_RANGE_UNSPECIFIED;

};

    static VideoFrameEncoderUPtr create(const QMediaEncoderSettings &encoderSettings,

                                        const SourceParams &sourceParams,

                                        AVFormatContext *formatContext);

    ~VideoFrameEncoder();

    AVPixelFormat sourceFormat() const { return m_sourceFormat; }

    AVPixelFormat targetFormat() const { return m_targetFormat; }

    qreal codecFrameRate() const;

    qint64 getPts(qint64 ms) const;

    const AVRational &getTimeBase() const;

    int sendFrame(AVFrameUPtr inputFrame);

    AVPacketUPtr retrievePacket();

private:

    VideoFrameEncoder(AVStream *stream, const Codec &codec, HWAccelUPtr hwAccel,

                      const SourceParams &sourceParams,

                      const QMediaEncoderSettings &encoderSettings, bool needsGlobalHeader);

    static AVStream *createStream(const SourceParams &sourceParams, AVFormatContext *formatContext);

    bool updateSourceFormatAndSize(const AVFrame *frame);

    void updateConversions();

    static VideoFrameEncoderUPtr create(AVStream *stream, const Codec &codec, HWAccelUPtr hwAccel,

                                        const SourceParams &sourceParams,

                                        const QMediaEncoderSettings &encoderSettings,

                                        bool needsGlobalHeader);

    void initCodecFrameRate();

    void initStream();

    std::vector<ScoredPixelFormat> enumerateTargetSwFormats() const; // sorted by score

    bool tryInitTargetSwFormat(AVPixelFormat targetSwFormat);

    bool initCodecContext();

    bool open();

    qint64 estimateDuration(const AVPacket &packet, bool isFirstPacket);

private:

    const QMediaEncoderSettings m_settings;

    AVStream *const m_stream = nullptr;

    const Codec m_codec;

    const HWAccelUPtr m_accel;

    const bool m_needsGlobalHeader = false;

    QSize m_sourceSize;

    QSize m_targetSize;

    qint64 m_lastPacketTime = AV_NOPTS_VALUE;

    AVCodecContextUPtr m_codecContext;

    SwsContextUPtr m_scaleContext;

    AVPixelFormat m_sourceFormat = AV_PIX_FMT_NONE;

    AVPixelFormat m_sourceSWFormat = AV_PIX_FMT_NONE;

    AVPixelFormat m_targetFormat;

    AVPixelFormat m_targetSWFormat = AV_PIX_FMT_NONE;

    bool m_downloadFromHW = false;

    bool m_uploadToHW = false;

    qreal m_sourceFrameRate = 0.f; // 0 = variable/unknown

    AVRational m_codecFrameRate = { 0, 1 };

    // m_codecFrameRate as fed to AVCodecContext::framerate and used to derive the

    // stream time base. Unlike m_codecFrameRate (which stays {0, 1} for a genuinely

    // variable-rate stream), this is never zero

    AVRational effectiveCodecFrameRate() const

        return m_codecFrameRate.den > 0 && m_codecFrameRate.num > 0

                ? m_codecFrameRate

                : AVRational{ DefaultVideoFrameRate, 1 };

    int64_t m_prevPacketDts = AV_NOPTS_VALUE;

    int64_t m_packetDtsOffset = 0;

};

} // namespace QFFmpeg

QT_END_NAMESPACE

#endif