11 #include <QMutexLocker>
15 #include "libmythbase/mythconfig.h"
22 #include "libavcodec/avcodec.h"
23 #include "libavutil/imgutils.h"
24 #include "libavformat/avformat.h"
31 case FMT_YV12:
return AV_PIX_FMT_YUV420P;
37 case FMT_NV12:
return AV_PIX_FMT_NV12;
38 case FMT_P010:
return AV_PIX_FMT_P010;
39 case FMT_P016:
return AV_PIX_FMT_P016;
53 case FMT_BGRA:
return AV_PIX_FMT_BGRA;
57 case FMT_YUY2:
return AV_PIX_FMT_UYVY422;
59 case FMT_VTB:
return AV_PIX_FMT_VIDEOTOOLBOX;
63 case FMT_DXVA2:
return AV_PIX_FMT_DXVA2_VLD;
64 case FMT_MMAL:
return AV_PIX_FMT_MMAL;
68 return AV_PIX_FMT_NONE;
75 case AV_PIX_FMT_YUVJ420P:
76 case AV_PIX_FMT_YUV420P:
return FMT_YV12;
82 case AV_PIX_FMT_NV12:
return FMT_NV12;
83 case AV_PIX_FMT_P010:
return FMT_P010;
84 case AV_PIX_FMT_P016:
return FMT_P016;
85 case AV_PIX_FMT_YUVJ422P:
92 case AV_PIX_FMT_YUVJ444P:
99 case AV_PIX_FMT_UYVY422:
return FMT_YUY2;
103 case AV_PIX_FMT_BGRA:
return FMT_BGRA;
105 case AV_PIX_FMT_MMAL:
return FMT_MMAL;
107 case AV_PIX_FMT_VIDEOTOOLBOX:
return FMT_VTB;
109 case AV_PIX_FMT_DXVA2_VLD:
return FMT_DXVA2;
122 case AVCOL_TRC_BT2020_10:
168 if (Fmt == AV_PIX_FMT_NONE)
194 m_size = av_image_get_buffer_size(Fmt, Width, Height, IMAGE_ALIGN);
202 int Width,
int Height)
204 int newwidth = Width;
205 #ifdef Q_PROCESSOR_ARM
216 if (FromFmt == AV_PIX_FMT_YUV420P && ToFmt == AV_PIX_FMT_BGRA)
217 newwidth = Width - 1;
220 newwidth, Height, ToFmt, SWS_FAST_BILINEAR,
221 nullptr,
nullptr,
nullptr);
224 sws_scale(
m_swsctx, From->data, From->linesize, 0, Height, To->data, To->linesize);
225 return SizeData(Width, Height, ToFmt);
235 unsigned char*
Buffer, AVPixelFormat Fmt)
242 av_image_fill_arrays(To->data, To->linesize,
Buffer, Fmt, From->
m_width, From->
m_height, IMAGE_ALIGN);
256 const AVCodec* Codec,
260 AVCodecContext* avctx =
m_streamMap.value(Stream,
nullptr);
263 if (Stream ==
nullptr || Stream->codecpar ==
nullptr)
273 Codec = avcodec_find_decoder(Stream->codecpar->codec_id);
277 LOG(VB_GENERAL, LOG_WARNING, QString(
"avcodec_find_decoder fail for %1")
278 .arg(Stream->codecpar->codec_id));
282 avctx = avcodec_alloc_context3(Codec);
283 if (avcodec_parameters_to_context(avctx, Stream->codecpar) < 0)
284 avcodec_free_context(&avctx);
288 avctx->pkt_timebase = Stream->time_base;
310 avcodec_flush_buffers(avctx);
311 avcodec_free_context(&avctx);
318 QMap<const AVStream*, AVCodecContext*>::iterator i =
m_streamMap.begin();
321 const AVStream *stream = i.key();
329 const int probeBufferSize = 8 * 1024;
330 AVInputFormat *fmt =
nullptr;
332 memset(&probe, 0,
sizeof(AVProbeData));
334 probe.buf =
new unsigned char[probeBufferSize + AVPROBE_PADDING_SIZE];
335 probe.buf_size = probeBufferSize;
336 memset(probe.buf, 0, probeBufferSize + AVPROBE_PADDING_SIZE);
337 av_log_set_level(AV_LOG_FATAL);
342 if (
m_errorCode == 0 && !infile.open(QIODevice::ReadOnly))
345 int64_t leng = infile.read(
reinterpret_cast<char*
>(probe.buf), probeBufferSize);
346 probe.buf_size =
static_cast<int>(leng);
348 fmt = av_probe_input_format(&probe,
static_cast<int>(
true));
352 AVFormatContext *ctx =
nullptr;
355 ctx = avformat_alloc_context();
359 m_errorCode = avformat_find_stream_info(ctx,
nullptr);
363 for (
uint ix = 0; ix < ctx->nb_streams; ix++)
365 AVStream *stream = ctx->streams[ix];
366 if (stream ==
nullptr)
368 AVCodecParameters *codecpar = stream->codecpar;
369 const AVCodecDescriptor* desc =
nullptr;
372 if (codecpar !=
nullptr)
374 desc = avcodec_descriptor_get(codecpar->codec_id);
375 switch (codecpar->codec_type)
377 case AVMEDIA_TYPE_VIDEO:
380 case AVMEDIA_TYPE_AUDIO:
383 case AVMEDIA_TYPE_SUBTITLE:
392 info.
m_duration = stream->duration * stream->time_base.num / stream->time_base.den;
395 if (codecpar !=
nullptr)
397 info.
m_width = codecpar->width;
400 /
static_cast<float>(codecpar->sample_aspect_ratio.den);
401 switch (codecpar->field_order)
403 case AV_FIELD_PROGRESSIVE:
422 info.
m_frameRate =
static_cast<float>(stream->r_frame_rate.num)
423 /
static_cast<float>(stream->r_frame_rate.den);
424 info.
m_avgFrameRate =
static_cast<float>(stream->avg_frame_rate.num)
425 /
static_cast<float>(stream->avg_frame_rate.den);
439 m_errorMsg =
"av_probe_input_format returned no result";
451 LOG(VB_GENERAL, LOG_ERR,
452 QString(
"MythStreamInfoList failed for %1. Error code:%2 Message:%3")
459 avformat_close_input(&ctx);
460 avformat_free_context(ctx);