10#include <QtCore/qloggingcategory.h>
11#include <QtCore/qoperatingsystemversion.h>
12#include <QtCore/private/qexpected_p.h>
15#include "libavutil/display.h"
16#include "libavutil/pixdesc.h"
21namespace ranges = QtMultimediaPrivate::ranges;
29AVCodecID avCodecID(
const QMediaEncoderSettings &settings)
31 const QMediaFormat::VideoCodec qVideoCodec = settings.videoCodec();
32 return QFFmpegMediaFormatInfo::codecIdForVideoCodec(qVideoCodec);
35[[maybe_unused]]
bool is420(AVPixelFormat fmt)
37 const auto desc = av_pix_fmt_desc_get(fmt);
41 return desc->log2_chroma_w == 1 && desc->log2_chroma_h == 1;
44constexpr bool isAndroid =
45 QOperatingSystemVersion::currentType() == QOperatingSystemVersion::Android;
49VideoFrameEncoderUPtr
VideoFrameEncoder::create(
const QMediaEncoderSettings &encoderSettings,
51 AVFormatContext *formatContext)
53 Q_ASSERT(isSwPixelFormat(sourceParams.swFormat));
54 Q_ASSERT(isHwPixelFormat(sourceParams.format) || sourceParams.swFormat == sourceParams.format);
56 AVStream *stream = createStream(sourceParams, formatContext);
61 auto findAndOpenAVEncoder = [&](
const auto &scoresGetter,
62 const auto &creator) -> VideoFrameEncoderUPtr {
63 VideoFrameEncoderUPtr encoder;
65 auto createWithTargetFormatFallback = [&](
const Codec &codec) {
66 auto result = creator(codec, AVPixelFormatSet{});
67 if constexpr (isAndroid) {
70 if (!result.encoder) {
71 if (is420(result.targetFormat) && result.targetFormat != AV_PIX_FMT_NV12)
72 result = creator(codec, AVPixelFormatSet{ result.targetFormat });
76 encoder = std::move(result.encoder);
79 QFFmpeg::findAndOpenAVEncoder(avCodecID(encoderSettings), scoresGetter,
80 createWithTargetFormatFallback);
85 const QSpan deviceTypes = HWAccel::encodingDeviceTypes();
87 auto findHwDeviceType = [&](
const Codec &codec) {
88 std::optional<AVPixelFormat> pixelFormat = findAVPixelFormat(codec, &isHwPixelFormat);
90 return deviceTypes.end();
92 const AVCodecHWConfig *cfg = codec.hwConfigForPixelFormat(*pixelFormat);
94 return deviceTypes.end();
96 bool supportsHwDeviceContext =
97 (cfg->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX) != 0;
98 bool supportsHwFramesContext =
99 (cfg->methods & AV_CODEC_HW_CONFIG_METHOD_HW_FRAMES_CTX) != 0;
100 if (!supportsHwDeviceContext && !supportsHwFramesContext)
101 return deviceTypes.end();
103 return ranges::find_if(deviceTypes, [&](AVHWDeviceType deviceType) {
104 return pixelFormatForHwDevice(deviceType) == pixelFormat;
109 VideoFrameEncoderUPtr encoder = findAndOpenAVEncoder([&](
const Codec &codec) {
110 const auto found = findHwDeviceType(codec);
111 if (found == deviceTypes.end())
112 return NotSuitableAVScore;
114 return DefaultAVScore -
static_cast<AVScore>(found - deviceTypes.begin());
115 }, [&](
const Codec &codec,
const AVPixelFormatSet &prohibitedTargetFormats) {
116 HWAccelUPtr hwAccel = HWAccel::create(*findHwDeviceType(codec));
118 return CreationResult{};
119 if (!hwAccel->matchesSizeContraints(encoderSettings.videoResolution()))
120 return CreationResult{};
121 return create(stream, codec, std::move(hwAccel), sourceParams, encoderSettings,
122 prohibitedTargetFormats);
127 encoder = findAndOpenAVEncoder([&](
const Codec &codec) {
128 return findSWFormatScores(codec, sourceParams.swFormat);
129 }, [&](
const Codec &codec,
const AVPixelFormatSet &prohibitedTargetFormats) {
130 return create(stream, codec,
nullptr, sourceParams, encoderSettings,
131 prohibitedTargetFormats);
136 qCDebug(qLcVideoFrameEncoder)
137 <<
"found" << (encoder->m_accel ?
"hw" :
"sw") <<
"encoder"
138 << encoder->m_codec.name() <<
"for id" << encoder->m_codec.id();
140 qCWarning(qLcVideoFrameEncoder) <<
"No valid video codecs found";
147 const QMediaEncoderSettings &encoderSettings)
148 : m_settings(encoderSettings),
151 m_accel(std::move(hwAccel)),
152 m_sourceSize(sourceParams.size),
153 m_sourceFormat(sourceParams.format),
154 m_sourceSWFormat(sourceParams.swFormat)
159 AVFormatContext *formatContext)
161 AVStream *stream = avformat_new_stream(formatContext,
nullptr);
166 stream->id = formatContext->nb_streams - 1;
167 stream->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
169 stream->codecpar->color_trc = sourceParams.colorTransfer;
170 stream->codecpar->color_space = sourceParams.colorSpace;
171 stream->codecpar->color_range = sourceParams.colorRange;
173 if (sourceParams.transform.rotation != QtVideo::Rotation::None || sourceParams.transform.mirroredHorizontallyAfterRotation) {
174 constexpr auto displayMatrixSize =
sizeof(int32_t) * 9;
175 AVPacketSideData sideData = {
reinterpret_cast<uint8_t *>(av_malloc(displayMatrixSize)),
176 displayMatrixSize, AV_PKT_DATA_DISPLAYMATRIX };
177 int32_t *matrix =
reinterpret_cast<int32_t *>(sideData.data);
178 av_display_rotation_set(matrix,
static_cast<
double>(sourceParams.transform.rotation));
179 if (sourceParams.transform.mirroredHorizontallyAfterRotation)
180 av_display_matrix_flip(matrix, sourceParams.transform.mirroredHorizontallyAfterRotation,
false);
182 addStreamSideData(stream, sideData);
191 const QMediaEncoderSettings &encoderSettings,
192 const AVPixelFormatSet &prohibitedTargetFormats)
195 sourceParams, encoderSettings));
196 frameEncoder->initTargetSize();
198 frameEncoder->initCodecFrameRate();
200 if (!frameEncoder->initTargetFormats(prohibitedTargetFormats))
203 frameEncoder->initStream();
205 const AVPixelFormat targetFormat = frameEncoder->m_targetFormat;
207 if (!frameEncoder->initCodecContext())
208 return {
nullptr, targetFormat };
210 if (!frameEncoder->open())
211 return {
nullptr, targetFormat };
213 frameEncoder->updateConversions();
215 return {
std::move(frameEncoder), targetFormat };
220 m_targetSize = adjustVideoResolution(m_codec, m_settings.videoResolution());
224 if (m_codec.name() == u"h264_mf") {
225 auto makeEven = [](
int size) {
return size & ~1; };
226 const QSize fixedSize(makeEven(m_targetSize.width()), makeEven(m_targetSize.height()));
227 if (fixedSize != m_targetSize) {
228 qCDebug(qLcVideoFrameEncoder) <<
"Fix odd video resolution for codec" << m_codec.name()
229 <<
":" << m_targetSize <<
"->" << fixedSize;
230 m_targetSize = fixedSize;
238 const auto frameRates = m_codec.frameRates();
239 if (qLcVideoFrameEncoder().isEnabled(QtDebugMsg))
240 for (AVRational rate : frameRates)
241 qCDebug(qLcVideoFrameEncoder) <<
"supported frame rate:" << rate;
243 m_codecFrameRate = adjustFrameRate(frameRates, m_settings.videoFrameRate());
244 qCDebug(qLcVideoFrameEncoder) <<
"Adjusted frame rate:" << m_codecFrameRate;
247bool VideoFrameEncoder::initTargetFormats(
const AVPixelFormatSet &prohibitedTargetFormats)
249 const std::optional format =
250 findTargetFormat(m_sourceSWFormat, m_codec, m_accel.get(), prohibitedTargetFormats);
253 qWarning() <<
"Could not find target format for codecId" << m_codec.id();
257 m_targetFormat = *format;
259 if (isHwPixelFormat(m_targetFormat)) {
264 const auto swFormat = findTargetSWFormat(m_sourceSWFormat, m_codec, *m_accel);
266 qWarning() <<
"Cannot find software target format. sourceSWFormat:" << m_sourceSWFormat
267 <<
"targetFormat:" << m_targetFormat;
271 m_targetSWFormat = *swFormat;
273 m_accel->createFramesContext(m_targetSWFormat, m_targetSize);
274 if (!m_accel->hwFramesContextAsBuffer())
277 m_targetSWFormat = m_targetFormat;
287 m_stream->codecpar->codec_id = m_codec.id();
291 if (m_codec.id() == AV_CODEC_ID_HEVC)
292 m_stream->codecpar->codec_tag = MKTAG(
'h',
'v',
'c',
'1');
294 m_stream->codecpar->codec_tag = 0;
297 m_stream->codecpar->format = m_targetFormat;
298 m_stream->codecpar->width = m_targetSize.width();
299 m_stream->codecpar->height = m_targetSize.height();
300 m_stream->codecpar->sample_aspect_ratio = AVRational{ 1, 1 };
301#if QT_CODEC_PARAMETERS_HAVE_FRAMERATE
302 m_stream->codecpar->framerate = m_codecFrameRate;
305 const auto frameRates = m_codec.frameRates();
306 m_stream->time_base = adjustFrameTimeBase(frameRates, m_codecFrameRate);
311 namespace views = QtMultimediaPrivate::views;
313 Q_ASSERT(m_stream->codecpar->codec_id);
315 m_codecContext.reset(avcodec_alloc_context3(m_codec.get()));
316 if (!m_codecContext) {
317 qWarning() <<
"Could not allocate codec context";
322 avcodec_parameters_to_context(m_codecContext.get(), m_stream->codecpar);
323#if !QT_CODEC_PARAMETERS_HAVE_FRAMERATE
324 m_codecContext->framerate = m_codecFrameRate;
326 m_codecContext->time_base = m_stream->time_base;
327 qCDebug(qLcVideoFrameEncoder) <<
"codecContext time base" << m_codecContext->time_base.num
328 << m_codecContext->time_base.den;
331 const AVHWDeviceType deviceType = m_accel->deviceType();
332 auto hwConfigs = m_codec.hwConfigs();
333 auto matchingConfigs = hwConfigs | views::filter([&](
const AVCodecHWConfig *cfg) {
334 return cfg->device_type == deviceType;
337 bool supportsDeviceCtx =
false;
338 bool supportsFramesCtx =
false;
339 for (
const AVCodecHWConfig *cfg : matchingConfigs) {
340 if (cfg->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX)
341 supportsDeviceCtx =
true;
342 if (cfg->methods & AV_CODEC_HW_CONFIG_METHOD_HW_FRAMES_CTX)
343 supportsFramesCtx =
true;
346 if (supportsDeviceCtx) {
347 auto deviceContext = m_accel->hwDeviceContextAsBuffer();
348 Q_ASSERT(deviceContext);
349 m_codecContext->hw_device_ctx = av_buffer_ref(deviceContext);
352 if (supportsFramesCtx)
353 if (
auto framesContext = m_accel->hwFramesContextAsBuffer())
354 m_codecContext->hw_frames_ctx = av_buffer_ref(framesContext);
357 avcodec_parameters_from_context(m_stream->codecpar, m_codecContext.get());
364 Q_ASSERT(m_codecContext);
366 AVDictionaryHolder opts;
367 applyVideoEncoderOptions(m_settings, m_codec.name(), m_codecContext.get(), opts);
368 applyExperimentalCodecOptions(m_codec, opts);
370 qCDebug(qLcVideoFrameEncoder) <<
"Opening encoder" << m_codec.name() <<
"with" << opts;
372 const int res = avcodec_open2(m_codecContext.get(), m_codec.get(), opts);
374 qCWarning(qLcVideoFrameEncoder)
375 <<
"Couldn't open video encoder" << m_codec.name() <<
"; result:" << AVError(res);
378 qCDebug(qLcVideoFrameEncoder) <<
"video codec opened" << res <<
"time base"
379 << m_codecContext->time_base;
385 return m_codecFrameRate.den ? qreal(m_codecFrameRate.num) / m_codecFrameRate.den : 0.;
390 qint64 div = 1'000'000 * m_stream->time_base.num;
391 return div != 0 ? (us * m_stream->time_base.den + div / 2) / div : 0;
396 return m_stream->time_base;
402 FrameConverter(AVFrameUPtr inputFrame) : m_inputFrame{ std::move(inputFrame) } { }
406 AVFrameUPtr cpuFrame = makeAVFrame();
408 int err = av_hwframe_transfer_data(cpuFrame.get(), currentFrame(), 0);
410 qCDebug(qLcVideoFrameEncoder)
411 <<
"Error transferring frame data to surface." << AVError(err);
415 setFrame(std::move(cpuFrame));
419 void convert(SwsContext *scaleContext, AVPixelFormat format,
const QSize &size)
421 AVFrameUPtr scaledFrame = makeAVFrame();
423 scaledFrame->format = format;
424 scaledFrame->width = size.width();
425 scaledFrame->height = size.height();
427 av_frame_get_buffer(scaledFrame.get(), 0);
429 const AVFrame *srcFrame = currentFrame();
431 const auto scaledHeight =
432 sws_scale(scaleContext, srcFrame->data, srcFrame->linesize, 0, srcFrame->height,
433 scaledFrame->data, scaledFrame->linesize);
435 if (scaledHeight != scaledFrame->height)
436 qCWarning(qLcVideoFrameEncoder)
437 <<
"Scaled height" << scaledHeight <<
"!=" << scaledFrame->height;
439 setFrame(std::move(scaledFrame));
442 int uploadToHw(HWAccel *accel)
444 auto *hwFramesContext = accel->hwFramesContextAsBuffer();
445 Q_ASSERT(hwFramesContext);
446 AVFrameUPtr hwFrame = makeAVFrame();
448 return AVERROR(ENOMEM);
450 int err = av_hwframe_get_buffer(hwFramesContext, hwFrame.get(), 0);
452 qCDebug(qLcVideoFrameEncoder) <<
"Error getting HW buffer" << AVError(err);
455 qCDebug(qLcVideoFrameEncoder) <<
"got HW buffer";
457 if (!hwFrame->hw_frames_ctx) {
458 qCDebug(qLcVideoFrameEncoder) <<
"no hw frames context";
459 return AVERROR(ENOMEM);
461 err = av_hwframe_transfer_data(hwFrame.get(), currentFrame(), 0);
463 qCDebug(qLcVideoFrameEncoder)
464 <<
"Error transferring frame data to surface." << AVError(err);
468 setFrame(std::move(hwFrame));
473 q23::expected<AVFrameUPtr,
int> takeResultFrame()
476 AVFrameUPtr converted =
std::move(m_convertedFrame);
477 AVFrameUPtr input =
std::move(m_inputFrame);
483 const int status = av_frame_copy_props(converted.get(), input.get());
485 return q23::unexpected{ status };
491 void setFrame(AVFrameUPtr frame) { m_convertedFrame = std::move(frame); }
493 AVFrame *currentFrame()
const
495 if (m_convertedFrame)
496 return m_convertedFrame.get();
497 return m_inputFrame.get();
500 AVFrameUPtr m_inputFrame;
501 AVFrameUPtr m_convertedFrame;
507 if (!m_codecContext) {
508 qWarning() <<
"codec context is not initialized!";
509 return AVERROR(EINVAL);
513 return avcodec_send_frame(m_codecContext.get(),
nullptr);
515 if (!updateSourceFormatAndSize(inputFrame.get()))
516 return AVERROR(EINVAL);
519 inputFrame->quality = m_codecContext->global_quality;
521 FrameConverter converter{
std::move(inputFrame) };
523 if (m_downloadFromHW) {
524 const int status = converter.downloadFromHw();
530 converter.convert(m_scaleContext.get(), m_targetSWFormat, m_targetSize);
533 const int status = converter.uploadToHw(m_accel.get());
538 const q23::expected<AVFrameUPtr,
int> resultFrame = converter.takeResultFrame();
540 return resultFrame.error();
542 AVRational timeBase{};
544 getAVFrameTime(*resultFrame.value(), pts, timeBase);
545 qCDebug(qLcVideoFrameEncoder) <<
"sending frame" << pts <<
"*" << timeBase;
547 return avcodec_send_frame(m_codecContext.get(), resultFrame.value().get());
550qint64
VideoFrameEncoder::estimateDuration(
const AVPacket &packet,
bool isFirstPacket)
558 const AVRational frameDuration = av_inv_q(m_codecContext->framerate);
559 duration = av_rescale_q(1, frameDuration, m_stream->time_base);
562 duration = packet.pts - m_lastPacketTime;
573 auto getPacket = [&]() {
574 AVPacketUPtr packet(av_packet_alloc());
575 const int ret = avcodec_receive_packet(m_codecContext.get(), packet.get());
577 if (ret != AVERROR(EOF) && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
578 qCDebug(qLcVideoFrameEncoder) <<
"Error receiving packet" << ret << AVError(ret);
579 return AVPacketUPtr{};
581 auto ts = timeStampMs(packet->pts, m_stream->time_base);
583 qCDebug(qLcVideoFrameEncoder)
584 <<
"got a packet" << packet->pts << packet->dts << (ts ? *ts : 0);
586 packet->stream_index = m_stream->id;
588 if (packet->duration == 0) {
589 const bool firstFrame = m_lastPacketTime == AV_NOPTS_VALUE;
590 packet->duration = estimateDuration(*packet, firstFrame);
593 m_lastPacketTime = packet->pts;
598 auto fixPacketDts = [&](AVPacket &packet) {
602 if (packet.dts == AV_NOPTS_VALUE)
605 packet.dts -= m_packetDtsOffset;
607 if (packet.pts != AV_NOPTS_VALUE && packet.pts < packet.dts) {
608 m_packetDtsOffset += packet.dts - packet.pts;
609 packet.dts = packet.pts;
611 if (m_prevPacketDts != AV_NOPTS_VALUE && packet.dts < m_prevPacketDts) {
612 qCWarning(qLcVideoFrameEncoder)
613 <<
"Skip packet; failed to fix dts:" << packet.dts << m_prevPacketDts;
618 m_prevPacketDts = packet.dts;
623 while (
auto packet = getPacket()) {
624 if (fixPacketDts(*packet))
635 const QSize frameSize(frame->width, frame->height);
636 const AVPixelFormat frameFormat =
static_cast<AVPixelFormat>(frame->format);
638 if (frameSize == m_sourceSize && frameFormat == m_sourceFormat)
641 auto applySourceFormatAndSize = [&](AVPixelFormat swFormat) {
642 m_sourceSize = frameSize;
643 m_sourceFormat = frameFormat;
644 m_sourceSWFormat = swFormat;
649 if (frameFormat == m_sourceFormat)
650 return applySourceFormatAndSize(m_sourceSWFormat);
652 if (frameFormat == AV_PIX_FMT_NONE) {
653 qWarning() <<
"Got a frame with invalid pixel format";
657 if (isSwPixelFormat(frameFormat))
658 return applySourceFormatAndSize(frameFormat);
660 auto framesCtx =
reinterpret_cast<
const AVHWFramesContext *>(frame->hw_frames_ctx->data);
661 if (!framesCtx || framesCtx->sw_format == AV_PIX_FMT_NONE) {
662 qWarning() <<
"Cannot update conversions as hw frame has invalid framesCtx" << framesCtx;
666 return applySourceFormatAndSize(framesCtx->sw_format);
671 const bool needToScale = m_sourceSize != m_targetSize;
672 const bool zeroCopy = m_sourceFormat == m_targetFormat && !needToScale;
674 m_scaleContext.reset();
677 m_downloadFromHW =
false;
678 m_uploadToHW =
false;
680 qCDebug(qLcVideoFrameEncoder) <<
"zero copy encoding, format" << m_targetFormat;
685 m_downloadFromHW = m_sourceFormat != m_sourceSWFormat;
686 m_uploadToHW = m_targetFormat != m_targetSWFormat;
688 if (m_sourceSWFormat != m_targetSWFormat || needToScale) {
689 qCDebug(qLcVideoFrameEncoder)
690 <<
"video source and encoder use different formats:" << m_sourceSWFormat
691 << m_targetSWFormat <<
"or sizes:" << m_sourceSize << m_targetSize;
693 const SwsFlags conversionType = getScaleConversionType(m_sourceSize, m_targetSize);
695 m_scaleContext = createSwsContext(m_sourceSize, m_sourceSWFormat, m_targetSize,
696 m_targetSWFormat, conversionType);
699 qCDebug(qLcVideoFrameEncoder) <<
"VideoFrameEncoder conversions initialized:"
700 <<
"sourceFormat:" << m_sourceFormat
701 << (isHwPixelFormat(m_sourceFormat) ?
"(hw)" :
"(sw)")
702 <<
"targetFormat:" << m_targetFormat
703 << (isHwPixelFormat(m_targetFormat) ?
"(hw)" :
"(sw)")
704 <<
"sourceSWFormat:" << m_sourceSWFormat
705 <<
"targetSWFormat:" << m_targetSWFormat
706 <<
"scaleContext:" << m_scaleContext.get();
const AVRational & getTimeBase() const
qint64 getPts(qint64 ms) const
int sendFrame(AVFrameUPtr inputFrame)
AVPacketUPtr retrievePacket()
qreal codecFrameRate() const
QT_MANGLE_NAMESPACE(QMacScreenCaptureStreamDelegate) QMacScreenCaptureStreamDelegate
QT_BEGIN_NAMESPACE Q_STATIC_LOGGING_CATEGORY(lcSynthesizedIterableAccess, "qt.iterable.synthesized", QtWarningMsg)