virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-26 19:01:44 +02:00
Code Issues Releases Activity

lavc/qsvdec: update color properties in codec context

Browse Source 
User may get color properties from the SDK via VIDEO_SIGNAL_INFO extbuf

Reviewed-by: Xiang, Haihao <haihao.xiang@intel.com>
Reviewed-by: Soft Works <softworkz@hotmail.com>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Haihao Xiang 2021-07-28 16:15:46 +08:00 committed by James Almer
parent 7ab0207d4b
commit c8cfe67694
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
libavcodec/qsvdec.c
View File

@ -311,7 +311,8 @@ static int qsv_decode_header(AVCodecContext *avctx, QSVContext *q,
mfxVideoParam *param)
{
int ret;
mfxExtVideoSignalInfo video_signal_info = { 0 };
mfxExtBuffer *header_ext_params[1] = { (mfxExtBuffer *)&video_signal_info };
mfxBitstream bs = { 0 };
if (avpkt->size) {
@ -336,6 +337,12 @@ static int qsv_decode_header(AVCodecContext *avctx, QSVContext *q,
return ret;
param->mfx.CodecId = ret;
video_signal_info.Header.BufferId = MFX_EXTBUFF_VIDEO_SIGNAL_INFO;
video_signal_info.Header.BufferSz = sizeof(video_signal_info);
// The SDK doesn't support other ext buffers when calling MFXVideoDECODE_DecodeHeader,
// so do not append this buffer to the existent buffer array
param->ExtParam = header_ext_params;
param->NumExtParam = 1;
ret = MFXVideoDECODE_DecodeHeader(q->session, &bs, param);
if (MFX_ERR_MORE_DATA == ret) {
return AVERROR(EAGAIN);
@ -344,6 +351,17 @@ static int qsv_decode_header(AVCodecContext *avctx, QSVContext *q,
return ff_qsv_print_error(avctx, ret,
"Error decoding stream header");
avctx->color_range = video_signal_info.VideoFullRange ? AVCOL_RANGE_JPEG : AVCOL_RANGE_MPEG;
if (video_signal_info.ColourDescriptionPresent) {
avctx->color_primaries = video_signal_info.ColourPrimaries;
avctx->color_trc = video_signal_info.TransferCharacteristics;
avctx->colorspace = video_signal_info.MatrixCoefficients;
}
param->ExtParam = q->ext_buffers;
param->NumExtParam = q->nb_ext_buffers;
return 0;
}