blob: 19fa791c10cd98480700a9f963a890af4eabda0b [file] [log] [blame]
Anton Khirnov4e08c822015-02-10 09:40:591/*
2 * Intel MediaSDK QSV codec-independent code
3 *
4 * copyright (c) 2013 Luca Barbato
5 * copyright (c) 2015 Anton Khirnov <[email protected]>
6 *
Michael Niedermayer841e9f42015-02-19 19:52:297 * This file is part of FFmpeg.
Anton Khirnov4e08c822015-02-10 09:40:598 *
Michael Niedermayer841e9f42015-02-19 19:52:299 * FFmpeg is free software; you can redistribute it and/or
Anton Khirnov4e08c822015-02-10 09:40:5910 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
Michael Niedermayer841e9f42015-02-19 19:52:2914 * FFmpeg is distributed in the hope that it will be useful,
Anton Khirnov4e08c822015-02-10 09:40:5915 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
Michael Niedermayer841e9f42015-02-19 19:52:2920 * License along with FFmpeg; if not, write to the Free Software
Anton Khirnov4e08c822015-02-10 09:40:5921 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 */
23
Martin Storsjöa78f1362022-02-23 12:56:4924#include "config_components.h"
25
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4126#include <stdint.h>
Anton Khirnov4e08c822015-02-10 09:40:5927#include <string.h>
28#include <sys/types.h>
29
Haihao Xiang3e61b7d2020-09-08 03:17:2730#include <mfxvideo.h>
Anton Khirnov4e08c822015-02-10 09:40:5931
32#include "libavutil/common.h"
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4133#include "libavutil/fifo.h"
34#include "libavutil/frame.h"
Mark Thompson1f26a232016-10-21 17:57:1235#include "libavutil/hwcontext.h"
36#include "libavutil/hwcontext_qsv.h"
Anton Khirnov4e08c822015-02-10 09:40:5937#include "libavutil/mem.h"
38#include "libavutil/log.h"
Xu Guangxin399c1f92021-01-05 02:43:4039#include "libavutil/opt.h"
Anton Khirnov4e08c822015-02-10 09:40:5940#include "libavutil/pixfmt.h"
41#include "libavutil/time.h"
Linjie Fu53459652019-10-08 13:41:0242#include "libavutil/imgutils.h"
Haihao Xiang8ca06a82022-01-24 08:24:5543#include "libavutil/film_grain_params.h"
Haihao Xiang71cded62022-11-21 02:31:2444#include "libavutil/mastering_display_metadata.h"
Haihao Xianga00cfc62024-05-08 06:03:1445#include "libavutil/avassert.h"
Anton Khirnov4e08c822015-02-10 09:40:5946
47#include "avcodec.h"
Andreas Rheinhardt20f97272022-03-16 20:09:5448#include "codec_internal.h"
Anton Khirnov4e08c822015-02-10 09:40:5949#include "internal.h"
Linjie Fu53459652019-10-08 13:41:0250#include "decode.h"
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4151#include "hwconfig.h"
Ivan Uskov6e127992015-07-14 11:07:0452#include "qsv.h"
Anton Khirnovd0a63d82015-03-13 07:13:0053#include "qsv_internal.h"
Anton Khirnov56ba57b2024-10-15 20:37:5654#include "libavutil/refstruct.h"
Anton Khirnov4e08c822015-02-10 09:40:5955
Haihao Xiang6900fee2021-01-04 02:46:1456#if QSV_ONEVPL
57#include <mfxdispatcher.h>
58#else
59#define MFXUnload(a) do { } while(0)
60#endif
61
Haihao Xiang971b4ac2021-06-11 02:19:4562static const AVRational mfx_tb = { 1, 90000 };
63
64#define PTS_TO_MFX_PTS(pts, pts_tb) ((pts) == AV_NOPTS_VALUE ? \
65 MFX_TIMESTAMP_UNKNOWN : pts_tb.num ? \
66 av_rescale_q(pts, pts_tb, mfx_tb) : pts)
67
68#define MFX_PTS_TO_PTS(mfx_pts, pts_tb) ((mfx_pts) == MFX_TIMESTAMP_UNKNOWN ? \
69 AV_NOPTS_VALUE : pts_tb.num ? \
70 av_rescale_q(mfx_pts, mfx_tb, pts_tb) : mfx_pts)
71
Haihao Xianga00cfc62024-05-08 06:03:1472#define MFX_IMPL_VIA_MASK(impl) (0x0f00 & (impl))
73
Anton Khirnov08181622022-01-06 16:16:4774typedef struct QSVAsyncFrame {
75 mfxSyncPoint *sync;
76 QSVFrame *frame;
77} QSVAsyncFrame;
78
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4179typedef struct QSVContext {
80 // the session used for decoding
81 mfxSession session;
Haihao Xiang7c24a762022-01-24 08:24:5482 mfxVersion ver;
Haihao Xianga00cfc62024-05-08 06:03:1483 mfxHandleType handle_type;
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4184
85 // the session we allocated internally, in case the caller did not provide
86 // one
87 QSVSession internal_qs;
88
89 QSVFramesContext frames_ctx;
90
91 /**
92 * a linked list of frames currently being used by QSV
93 */
94 QSVFrame *work_frames;
95
Anton Khirnov08181622022-01-06 16:16:4796 AVFifo *async_fifo;
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4197 int zero_consume_run;
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:4198 int reinit_flag;
99
100 enum AVPixelFormat orig_pix_fmt;
101 uint32_t fourcc;
102 mfxFrameInfo frame_info;
103 AVBufferPool *pool;
Wenbin Chen0a0847d2022-03-18 06:25:11104 int suggest_pool_size;
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41105 int initialized;
106
107 // options set by the caller
108 int async_depth;
109 int iopattern;
110 int gpu_copy;
111
112 char *load_plugins;
113
114 mfxExtBuffer **ext_buffers;
115 int nb_ext_buffers;
116} QSVContext;
117
118static const AVCodecHWConfigInternal *const qsv_hw_configs[] = {
Mark Thompson758fbc52017-10-25 23:18:40119 &(const AVCodecHWConfigInternal) {
120 .public = {
121 .pix_fmt = AV_PIX_FMT_QSV,
122 .methods = AV_CODEC_HW_CONFIG_METHOD_HW_FRAMES_CTX |
Haihao Xiangecee3b02021-08-11 06:01:54123 AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX,
Mark Thompson758fbc52017-10-25 23:18:40124 .device_type = AV_HWDEVICE_TYPE_QSV,
125 },
126 .hwaccel = NULL,
127 },
128 NULL
129};
130
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41131static int qsv_get_continuous_buffer(AVCodecContext *avctx, AVFrame *frame,
132 AVBufferPool *pool)
Linjie Fu53459652019-10-08 13:41:02133{
134 int ret = 0;
135
Anton Khirnov5bf8f292023-01-05 08:50:57136 ret = ff_decode_frame_props(avctx, frame);
137 if (ret < 0)
138 return ret;
Linjie Fu53459652019-10-08 13:41:02139
Fei Wang01c7f682024-05-20 02:05:53140 frame->width = avctx->coded_width;
141 frame->height = avctx->coded_height;
Linjie Fu53459652019-10-08 13:41:02142
143 switch (avctx->pix_fmt) {
144 case AV_PIX_FMT_NV12:
Fei Wang01c7f682024-05-20 02:05:53145 frame->linesize[0] = FFALIGN(avctx->coded_width, 128);
Linjie Fu53459652019-10-08 13:41:02146 break;
147 case AV_PIX_FMT_P010:
Fei Wang56a52af2022-10-06 07:35:38148 case AV_PIX_FMT_P012:
Wenbin Chene0ae8102022-04-06 08:48:03149 case AV_PIX_FMT_YUYV422:
Fei Wang01c7f682024-05-20 02:05:53150 frame->linesize[0] = 2 * FFALIGN(avctx->coded_width, 128);
Linjie Fu53459652019-10-08 13:41:02151 break;
Wenbin Chene0ae8102022-04-06 08:48:03152 case AV_PIX_FMT_Y210:
Haihao Xiangdb85e012022-09-06 04:53:38153 case AV_PIX_FMT_VUYX:
Haihao Xiang1898dbd2022-10-06 07:35:36154 case AV_PIX_FMT_XV30:
Fei Wang56a52af2022-10-06 07:35:38155 case AV_PIX_FMT_Y212:
Fei Wang01c7f682024-05-20 02:05:53156 frame->linesize[0] = 4 * FFALIGN(avctx->coded_width, 128);
Wenbin Chene0ae8102022-04-06 08:48:03157 break;
Fei Wang56a52af2022-10-06 07:35:38158 case AV_PIX_FMT_XV36:
Fei Wang01c7f682024-05-20 02:05:53159 frame->linesize[0] = 8 * FFALIGN(avctx->coded_width, 128);
Fei Wang56a52af2022-10-06 07:35:38160 break;
Linjie Fu53459652019-10-08 13:41:02161 default:
162 av_log(avctx, AV_LOG_ERROR, "Unsupported pixel format.\n");
Zhong Li9fff5c42019-12-28 14:22:07163 return AVERROR(EINVAL);
Linjie Fu53459652019-10-08 13:41:02164 }
165
Linjie Fu53459652019-10-08 13:41:02166 frame->buf[0] = av_buffer_pool_get(pool);
167 if (!frame->buf[0])
168 return AVERROR(ENOMEM);
169
170 frame->data[0] = frame->buf[0]->data;
Wenbin Chene0ae8102022-04-06 08:48:03171 if (avctx->pix_fmt == AV_PIX_FMT_NV12 ||
Fei Wang56a52af2022-10-06 07:35:38172 avctx->pix_fmt == AV_PIX_FMT_P010 ||
173 avctx->pix_fmt == AV_PIX_FMT_P012) {
Wenbin Chene0ae8102022-04-06 08:48:03174 frame->linesize[1] = frame->linesize[0];
175 frame->data[1] = frame->data[0] +
Fei Wang01c7f682024-05-20 02:05:53176 frame->linesize[0] * FFALIGN(avctx->coded_height, 64);
Wenbin Chene0ae8102022-04-06 08:48:03177 }
Linjie Fu53459652019-10-08 13:41:02178
179 ret = ff_attach_decode_data(frame);
180 if (ret < 0)
181 return ret;
182
183 return 0;
184}
185
Mark Thompson1f26a232016-10-21 17:57:12186static int qsv_init_session(AVCodecContext *avctx, QSVContext *q, mfxSession session,
Mark Thompson8aa3c2d2017-03-04 23:57:36187 AVBufferRef *hw_frames_ref, AVBufferRef *hw_device_ref)
Anton Khirnov4e08c822015-02-10 09:40:59188{
Mark Thompson1f26a232016-10-21 17:57:12189 int ret;
Haihao Xianga00cfc62024-05-08 06:03:14190 mfxIMPL impl;
Mark Thompson1f26a232016-10-21 17:57:12191
Linjie Fu53459652019-10-08 13:41:02192 if (q->gpu_copy == MFX_GPUCOPY_ON &&
Zhong Li779951f2019-12-28 14:28:45193 !(q->iopattern & MFX_IOPATTERN_OUT_SYSTEM_MEMORY)) {
Linjie Fu53459652019-10-08 13:41:02194 av_log(avctx, AV_LOG_WARNING, "GPU-accelerated memory copy "
Zhong Li779951f2019-12-28 14:28:45195 "only works in system memory mode.\n");
196 q->gpu_copy = MFX_GPUCOPY_OFF;
197 }
Mark Thompson1f26a232016-10-21 17:57:12198 if (session) {
199 q->session = session;
200 } else if (hw_frames_ref) {
Zhong Li74007dd2019-09-19 20:45:26201 if (q->internal_qs.session) {
202 MFXClose(q->internal_qs.session);
203 q->internal_qs.session = NULL;
Mark Thompson1f26a232016-10-21 17:57:12204 }
205 av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
206
207 q->frames_ctx.hw_frames_ctx = av_buffer_ref(hw_frames_ref);
208 if (!q->frames_ctx.hw_frames_ctx)
209 return AVERROR(ENOMEM);
210
Zhong Li74007dd2019-09-19 20:45:26211 ret = ff_qsv_init_session_frames(avctx, &q->internal_qs.session,
Mark Thompson91c3b502017-03-04 23:57:35212 &q->frames_ctx, q->load_plugins,
Haihao Xiangc77149b2020-08-19 01:43:12213#if QSV_HAVE_OPAQUE
Linjie Fu53459652019-10-08 13:41:02214 q->iopattern == MFX_IOPATTERN_OUT_OPAQUE_MEMORY,
Haihao Xiangc77149b2020-08-19 01:43:12215#else
216 0,
217#endif
Linjie Fu53459652019-10-08 13:41:02218 q->gpu_copy);
Mark Thompson1f26a232016-10-21 17:57:12219 if (ret < 0) {
220 av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
221 return ret;
222 }
223
Zhong Li74007dd2019-09-19 20:45:26224 q->session = q->internal_qs.session;
Mark Thompson8aa3c2d2017-03-04 23:57:36225 } else if (hw_device_ref) {
Zhong Li74007dd2019-09-19 20:45:26226 if (q->internal_qs.session) {
227 MFXClose(q->internal_qs.session);
228 q->internal_qs.session = NULL;
Mark Thompson8aa3c2d2017-03-04 23:57:36229 }
230
Zhong Li74007dd2019-09-19 20:45:26231 ret = ff_qsv_init_session_device(avctx, &q->internal_qs.session,
Linjie Fu53459652019-10-08 13:41:02232 hw_device_ref, q->load_plugins, q->gpu_copy);
Mark Thompson8aa3c2d2017-03-04 23:57:36233 if (ret < 0)
234 return ret;
235
Zhong Li74007dd2019-09-19 20:45:26236 q->session = q->internal_qs.session;
Mark Thompson1f26a232016-10-21 17:57:12237 } else {
Zhong Li74007dd2019-09-19 20:45:26238 if (!q->internal_qs.session) {
239 ret = ff_qsv_init_internal_session(avctx, &q->internal_qs,
Linjie Fu53459652019-10-08 13:41:02240 q->load_plugins, q->gpu_copy);
Anton Khirnovd0a63d82015-03-13 07:13:00241 if (ret < 0)
242 return ret;
Anton Khirnov4e08c822015-02-10 09:40:59243 }
244
Zhong Li74007dd2019-09-19 20:45:26245 q->session = q->internal_qs.session;
Anton Khirnov4e08c822015-02-10 09:40:59246 }
247
Haihao Xianga00cfc62024-05-08 06:03:14248 if (MFXQueryIMPL(q->session, &impl) == MFX_ERR_NONE) {
249 switch (MFX_IMPL_VIA_MASK(impl)) {
250 case MFX_IMPL_VIA_VAAPI:
251 q->handle_type = MFX_HANDLE_VA_DISPLAY;
252 break;
253
254 case MFX_IMPL_VIA_D3D11:
255 q->handle_type = MFX_HANDLE_D3D11_DEVICE;
256 break;
257
258 case MFX_IMPL_VIA_D3D9:
259 q->handle_type = MFX_HANDLE_D3D9_DEVICE_MANAGER;
260 break;
261
262 default:
263 av_assert0(!"should not reach here");
264 }
265 } else {
266 av_log(avctx, AV_LOG_ERROR, "Error querying the implementation. \n");
267 goto fail;
268 }
269
Haihao Xiang7c24a762022-01-24 08:24:54270 if (MFXQueryVersion(q->session, &q->ver) != MFX_ERR_NONE) {
271 av_log(avctx, AV_LOG_ERROR, "Error querying the session version. \n");
Haihao Xianga00cfc62024-05-08 06:03:14272 goto fail;
Haihao Xiang7c24a762022-01-24 08:24:54273 }
274
Mark Thompson1f26a232016-10-21 17:57:12275 /* make sure the decoder is uninitialized */
276 MFXVideoDECODE_Close(q->session);
277
278 return 0;
Haihao Xianga00cfc62024-05-08 06:03:14279
280fail:
281 q->session = NULL;
282
283 if (q->internal_qs.session) {
284 MFXClose(q->internal_qs.session);
285 q->internal_qs.session = NULL;
286 }
287
288 if (q->internal_qs.loader) {
289 MFXUnload(q->internal_qs.loader);
290 q->internal_qs.loader = NULL;
291 }
292
293 return AVERROR_EXTERNAL;
Anton Khirnov4e08c822015-02-10 09:40:59294}
295
Zhong Li00d0a4a2019-08-13 06:11:09296static int qsv_decode_preinit(AVCodecContext *avctx, QSVContext *q, enum AVPixelFormat pix_fmt, mfxVideoParam *param)
Anton Khirnov4e08c822015-02-10 09:40:59297{
Anton Khirnov6f19bbc2016-05-21 16:26:40298 mfxSession session = NULL;
Mark Thompson1f26a232016-10-21 17:57:12299 int iopattern = 0;
Anton Khirnov4e08c822015-02-10 09:40:59300 int ret;
Zhong Li00d0a4a2019-08-13 06:11:09301 enum AVPixelFormat pix_fmts[3] = {
302 AV_PIX_FMT_QSV, /* opaque format in case of video memory output */
303 pix_fmt, /* system memory format obtained from bitstream parser */
304 AV_PIX_FMT_NONE };
Anton Khirnov4e08c822015-02-10 09:40:59305
Zhong Li00d0a4a2019-08-13 06:11:09306 ret = ff_get_format(avctx, pix_fmts);
307 if (ret < 0) {
308 q->orig_pix_fmt = avctx->pix_fmt = AV_PIX_FMT_NONE;
309 return ret;
310 }
Anton Khirnov92736c72016-06-22 09:53:00311
Mark Thompson1f26a232016-10-21 17:57:12312 if (!q->async_fifo) {
Anton Khirnov08181622022-01-06 16:16:47313 q->async_fifo = av_fifo_alloc2(q->async_depth, sizeof(QSVAsyncFrame), 0);
Mark Thompson1f26a232016-10-21 17:57:12314 if (!q->async_fifo)
315 return AVERROR(ENOMEM);
Hendrik Leppkesb54d6452015-10-22 15:00:49316 }
Hendrik Leppkesb54d6452015-10-22 15:00:49317
Anton Khirnove3281782016-07-30 14:38:51318 if (avctx->pix_fmt == AV_PIX_FMT_QSV && avctx->hwaccel_context) {
Mark Thompson1f26a232016-10-21 17:57:12319 AVQSVContext *user_ctx = avctx->hwaccel_context;
320 session = user_ctx->session;
321 iopattern = user_ctx->iopattern;
322 q->ext_buffers = user_ctx->ext_buffers;
323 q->nb_ext_buffers = user_ctx->nb_ext_buffers;
324 }
325
Haihao Xiangecee3b02021-08-11 06:01:54326 if (avctx->hw_device_ctx && !avctx->hw_frames_ctx && ret == AV_PIX_FMT_QSV) {
327 AVHWFramesContext *hwframes_ctx;
328 AVQSVFramesContext *frames_hwctx;
329
330 avctx->hw_frames_ctx = av_hwframe_ctx_alloc(avctx->hw_device_ctx);
331
332 if (!avctx->hw_frames_ctx) {
333 av_log(avctx, AV_LOG_ERROR, "av_hwframe_ctx_alloc failed\n");
334 return AVERROR(ENOMEM);
335 }
336
337 hwframes_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
338 frames_hwctx = hwframes_ctx->hwctx;
339 hwframes_ctx->width = FFALIGN(avctx->coded_width, 32);
340 hwframes_ctx->height = FFALIGN(avctx->coded_height, 32);
341 hwframes_ctx->format = AV_PIX_FMT_QSV;
342 hwframes_ctx->sw_format = avctx->sw_pix_fmt;
Haihao Xianga00cfc62024-05-08 06:03:14343 if (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 2, 9) && q->handle_type != MFX_HANDLE_D3D9_DEVICE_MANAGER)
344 hwframes_ctx->initial_pool_size = 0;
345 else
346 hwframes_ctx->initial_pool_size = q->suggest_pool_size + 16 + avctx->extra_hw_frames;
Haihao Xiangecee3b02021-08-11 06:01:54347 frames_hwctx->frame_type = MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET;
348
349 ret = av_hwframe_ctx_init(avctx->hw_frames_ctx);
350
351 if (ret < 0) {
352 av_log(NULL, AV_LOG_ERROR, "Error initializing a QSV frame pool\n");
353 av_buffer_unref(&avctx->hw_frames_ctx);
354 return ret;
355 }
356 }
357
Mark Thompson1f26a232016-10-21 17:57:12358 if (avctx->hw_frames_ctx) {
359 AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
360 AVQSVFramesContext *frames_hwctx = frames_ctx->hwctx;
361
362 if (!iopattern) {
Haihao Xiangc77149b2020-08-19 01:43:12363#if QSV_HAVE_OPAQUE
Mark Thompson1f26a232016-10-21 17:57:12364 if (frames_hwctx->frame_type & MFX_MEMTYPE_OPAQUE_FRAME)
365 iopattern = MFX_IOPATTERN_OUT_OPAQUE_MEMORY;
366 else if (frames_hwctx->frame_type & MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET)
367 iopattern = MFX_IOPATTERN_OUT_VIDEO_MEMORY;
Haihao Xiangc77149b2020-08-19 01:43:12368#else
369 if (frames_hwctx->frame_type & MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET)
370 iopattern = MFX_IOPATTERN_OUT_VIDEO_MEMORY;
371#endif
Mark Thompson1f26a232016-10-21 17:57:12372 }
373 }
374
375 if (!iopattern)
376 iopattern = MFX_IOPATTERN_OUT_SYSTEM_MEMORY;
377 q->iopattern = iopattern;
378
Zhong Li525de952019-09-19 20:45:27379 ff_qsv_print_iopattern(avctx, q->iopattern, "Decoder");
380
Mark Thompson8aa3c2d2017-03-04 23:57:36381 ret = qsv_init_session(avctx, q, session, avctx->hw_frames_ctx, avctx->hw_device_ctx);
Anton Khirnov4e08c822015-02-10 09:40:59382 if (ret < 0) {
383 av_log(avctx, AV_LOG_ERROR, "Error initializing an MFX session\n");
384 return ret;
Ivan Uskov6e127992015-07-14 11:07:04385 }
Anton Khirnov4e08c822015-02-10 09:40:59386
Zhong Li00d0a4a2019-08-13 06:11:09387 param->IOPattern = q->iopattern;
388 param->AsyncDepth = q->async_depth;
389 param->ExtParam = q->ext_buffers;
390 param->NumExtParam = q->nb_ext_buffers;
Mark Thompson1f26a232016-10-21 17:57:12391
Zhong Li00d0a4a2019-08-13 06:11:09392 return 0;
393 }
Mark Thompson1f26a232016-10-21 17:57:12394
Xu Guangxin399c1f92021-01-05 02:43:40395static int qsv_decode_init_context(AVCodecContext *avctx, QSVContext *q, mfxVideoParam *param)
Zhong Li00d0a4a2019-08-13 06:11:09396{
397 int ret;
Mark Thompson1f26a232016-10-21 17:57:12398
Zhong Li00d0a4a2019-08-13 06:11:09399 avctx->width = param->mfx.FrameInfo.CropW;
400 avctx->height = param->mfx.FrameInfo.CropH;
401 avctx->coded_width = param->mfx.FrameInfo.Width;
402 avctx->coded_height = param->mfx.FrameInfo.Height;
403 avctx->level = param->mfx.CodecLevel;
404 avctx->profile = param->mfx.CodecProfile;
405 avctx->field_order = ff_qsv_map_picstruct(param->mfx.FrameInfo.PicStruct);
406 avctx->pix_fmt = ff_qsv_map_fourcc(param->mfx.FrameInfo.FourCC);
Anton Khirnov4e08c822015-02-10 09:40:59407
Zhong Li00d0a4a2019-08-13 06:11:09408 ret = MFXVideoDECODE_Init(q->session, param);
Anton Khirnov95414eb2016-06-25 19:38:10409 if (ret < 0)
410 return ff_qsv_print_error(avctx, ret,
411 "Error initializing the MFX video decoder");
Anton Khirnov4e08c822015-02-10 09:40:59412
Zhong Li00d0a4a2019-08-13 06:11:09413 q->frame_info = param->mfx.FrameInfo;
414
Michael Niedermayer8789c552024-05-13 00:05:56415 if (!avctx->hw_frames_ctx) {
Fei Wang01c7f682024-05-20 02:05:53416 ret = av_image_get_buffer_size(avctx->pix_fmt, FFALIGN(avctx->coded_width, 128), FFALIGN(avctx->coded_height, 64), 1);
Michael Niedermayer8789c552024-05-13 00:05:56417 if (ret < 0)
418 return ret;
419 q->pool = av_buffer_pool_init(ret, av_buffer_allocz);
420 }
Zhong Li00d0a4a2019-08-13 06:11:09421 return 0;
422}
423
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41424static int qsv_decode_header(AVCodecContext *avctx, QSVContext *q,
425 const AVPacket *avpkt, enum AVPixelFormat pix_fmt,
426 mfxVideoParam *param)
Zhong Li00d0a4a2019-08-13 06:11:09427{
428 int ret;
Haihao Xiangc8cfe672021-07-28 08:15:46429 mfxExtVideoSignalInfo video_signal_info = { 0 };
430 mfxExtBuffer *header_ext_params[1] = { (mfxExtBuffer *)&video_signal_info };
Zhong Li00d0a4a2019-08-13 06:11:09431 mfxBitstream bs = { 0 };
432
433 if (avpkt->size) {
434 bs.Data = avpkt->data;
435 bs.DataLength = avpkt->size;
436 bs.MaxLength = bs.DataLength;
Haihao Xiang971b4ac2021-06-11 02:19:45437 bs.TimeStamp = PTS_TO_MFX_PTS(avpkt->pts, avctx->pkt_timebase);
Zhong Li00d0a4a2019-08-13 06:11:09438 if (avctx->field_order == AV_FIELD_PROGRESSIVE)
439 bs.DataFlag |= MFX_BITSTREAM_COMPLETE_FRAME;
440 } else
441 return AVERROR_INVALIDDATA;
442
443
444 if(!q->session) {
445 ret = qsv_decode_preinit(avctx, q, pix_fmt, param);
446 if (ret < 0)
447 return ret;
448 }
449
450 ret = ff_qsv_codec_id_to_mfx(avctx->codec_id);
451 if (ret < 0)
452 return ret;
453
454 param->mfx.CodecId = ret;
Haihao Xiangc8cfe672021-07-28 08:15:46455 video_signal_info.Header.BufferId = MFX_EXTBUFF_VIDEO_SIGNAL_INFO;
456 video_signal_info.Header.BufferSz = sizeof(video_signal_info);
457 // The SDK doesn't support other ext buffers when calling MFXVideoDECODE_DecodeHeader,
458 // so do not append this buffer to the existent buffer array
459 param->ExtParam = header_ext_params;
460 param->NumExtParam = 1;
Zhong Li00d0a4a2019-08-13 06:11:09461 ret = MFXVideoDECODE_DecodeHeader(q->session, &bs, param);
462 if (MFX_ERR_MORE_DATA == ret) {
463 return AVERROR(EAGAIN);
464 }
465 if (ret < 0)
466 return ff_qsv_print_error(avctx, ret,
467 "Error decoding stream header");
Anton Khirnovce320cf2016-06-22 09:44:09468
Haihao Xiangc8cfe672021-07-28 08:15:46469 avctx->color_range = video_signal_info.VideoFullRange ? AVCOL_RANGE_JPEG : AVCOL_RANGE_MPEG;
470
471 if (video_signal_info.ColourDescriptionPresent) {
472 avctx->color_primaries = video_signal_info.ColourPrimaries;
473 avctx->color_trc = video_signal_info.TransferCharacteristics;
474 avctx->colorspace = video_signal_info.MatrixCoefficients;
475 }
476
477 param->ExtParam = q->ext_buffers;
478 param->NumExtParam = q->nb_ext_buffers;
479
Fei Wang67fc9b82024-04-18 08:15:09480 if (param->mfx.FrameInfo.FrameRateExtN == 0 || param->mfx.FrameInfo.FrameRateExtD == 0) {
481 param->mfx.FrameInfo.FrameRateExtN = 25;
482 param->mfx.FrameInfo.FrameRateExtD = 1;
483 }
484
Haihao Xiang8ca06a82022-01-24 08:24:55485#if QSV_VERSION_ATLEAST(1, 34)
486 if (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 1, 34) && avctx->codec_id == AV_CODEC_ID_AV1)
487 param->mfx.FilmGrain = (avctx->export_side_data & AV_CODEC_EXPORT_DATA_FILM_GRAIN) ? 0 : param->mfx.FilmGrain;
488#endif
489
Anton Khirnov4e08c822015-02-10 09:40:59490 return 0;
491}
492
Anton Khirnovce320cf2016-06-22 09:44:09493static int alloc_frame(AVCodecContext *avctx, QSVContext *q, QSVFrame *frame)
Anton Khirnov4e08c822015-02-10 09:40:59494{
495 int ret;
496
Linjie Fu53459652019-10-08 13:41:02497 if (q->pool)
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41498 ret = qsv_get_continuous_buffer(avctx, frame->frame, q->pool);
Linjie Fu53459652019-10-08 13:41:02499 else
500 ret = ff_get_buffer(avctx, frame->frame, AV_GET_BUFFER_FLAG_REF);
501
Anton Khirnov4e08c822015-02-10 09:40:59502 if (ret < 0)
503 return ret;
504
505 if (frame->frame->format == AV_PIX_FMT_QSV) {
Anton Khirnov404e5142016-08-09 10:05:49506 frame->surface = *(mfxFrameSurface1*)frame->frame->data[3];
Anton Khirnov4e08c822015-02-10 09:40:59507 } else {
Wenbin Chene0ae8102022-04-06 08:48:03508 ret = ff_qsv_map_frame_to_surface(frame->frame, &frame->surface);
509 if (ret < 0) {
510 av_log(avctx, AV_LOG_ERROR, "map frame to surface failed.\n");
511 return ret;
512 }
Anton Khirnov4e08c822015-02-10 09:40:59513 }
514
Wenbin Chenb1c26ce2022-02-11 02:37:36515 frame->surface.Info = q->frame_info;
516
Anton Khirnov00aeedd2016-08-10 06:29:23517 if (q->frames_ctx.mids) {
518 ret = ff_qsv_find_surface_idx(&q->frames_ctx, frame);
519 if (ret < 0)
520 return ret;
521
522 frame->surface.Data.MemId = &q->frames_ctx.mids[ret];
523 }
Haihao Xiang8dd507b2022-01-24 08:24:53524
525 frame->surface.Data.ExtParam = frame->ext_param;
526 frame->surface.Data.NumExtParam = 0;
527 frame->num_ext_params = 0;
Zhong Li52ed83f2018-04-04 09:51:29528 frame->dec_info.Header.BufferId = MFX_EXTBUFF_DECODED_FRAME_INFO;
529 frame->dec_info.Header.BufferSz = sizeof(frame->dec_info);
Haihao Xiang8dd507b2022-01-24 08:24:53530 ff_qsv_frame_add_ext_param(avctx, frame, (mfxExtBuffer *)&frame->dec_info);
Haihao Xiang8ca06a82022-01-24 08:24:55531#if QSV_VERSION_ATLEAST(1, 34)
532 if (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 1, 34) && avctx->codec_id == AV_CODEC_ID_AV1) {
533 frame->av1_film_grain_param.Header.BufferId = MFX_EXTBUFF_AV1_FILM_GRAIN_PARAM;
534 frame->av1_film_grain_param.Header.BufferSz = sizeof(frame->av1_film_grain_param);
535 frame->av1_film_grain_param.FilmGrainFlags = 0;
536 ff_qsv_frame_add_ext_param(avctx, frame, (mfxExtBuffer *)&frame->av1_film_grain_param);
537 }
538#endif
Anton Khirnov00aeedd2016-08-10 06:29:23539
Haihao Xiang71cded62022-11-21 02:31:24540#if QSV_VERSION_ATLEAST(1, 35)
Haihao Xiang473e84a2024-05-22 04:15:59541 if ((QSV_RUNTIME_VERSION_ATLEAST(q->ver, 1, 35) && avctx->codec_id == AV_CODEC_ID_HEVC) ||
542 (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 2, 9) && avctx->codec_id == AV_CODEC_ID_AV1)) {
Haihao Xiang71cded62022-11-21 02:31:24543 frame->mdcv.Header.BufferId = MFX_EXTBUFF_MASTERING_DISPLAY_COLOUR_VOLUME;
544 frame->mdcv.Header.BufferSz = sizeof(frame->mdcv);
545 // The data in mdcv is valid when this flag is 1
546 frame->mdcv.InsertPayloadToggle = 0;
547 ff_qsv_frame_add_ext_param(avctx, frame, (mfxExtBuffer *)&frame->mdcv);
548
549 frame->clli.Header.BufferId = MFX_EXTBUFF_CONTENT_LIGHT_LEVEL_INFO;
550 frame->clli.Header.BufferSz = sizeof(frame->clli);
551 // The data in clli is valid when this flag is 1
552 frame->clli.InsertPayloadToggle = 0;
553 ff_qsv_frame_add_ext_param(avctx, frame, (mfxExtBuffer *)&frame->clli);
554 }
555#endif
556
Anton Khirnov404e5142016-08-09 10:05:49557 frame->used = 1;
558
Anton Khirnov4e08c822015-02-10 09:40:59559 return 0;
560}
561
562static void qsv_clear_unused_frames(QSVContext *q)
563{
564 QSVFrame *cur = q->work_frames;
565 while (cur) {
Anton Khirnov404e5142016-08-09 10:05:49566 if (cur->used && !cur->surface.Data.Locked && !cur->queued) {
567 cur->used = 0;
Anton Khirnov4e08c822015-02-10 09:40:59568 av_frame_unref(cur->frame);
569 }
570 cur = cur->next;
571 }
572}
573
574static int get_surface(AVCodecContext *avctx, QSVContext *q, mfxFrameSurface1 **surf)
575{
576 QSVFrame *frame, **last;
577 int ret;
578
579 qsv_clear_unused_frames(q);
580
581 frame = q->work_frames;
582 last = &q->work_frames;
583 while (frame) {
Anton Khirnov404e5142016-08-09 10:05:49584 if (!frame->used) {
Anton Khirnovce320cf2016-06-22 09:44:09585 ret = alloc_frame(avctx, q, frame);
Anton Khirnov4e08c822015-02-10 09:40:59586 if (ret < 0)
587 return ret;
Anton Khirnov404e5142016-08-09 10:05:49588 *surf = &frame->surface;
Anton Khirnov4e08c822015-02-10 09:40:59589 return 0;
590 }
591
592 last = &frame->next;
593 frame = frame->next;
594 }
595
596 frame = av_mallocz(sizeof(*frame));
597 if (!frame)
598 return AVERROR(ENOMEM);
599 frame->frame = av_frame_alloc();
600 if (!frame->frame) {
601 av_freep(&frame);
602 return AVERROR(ENOMEM);
603 }
604 *last = frame;
605
Anton Khirnovce320cf2016-06-22 09:44:09606 ret = alloc_frame(avctx, q, frame);
Anton Khirnov4e08c822015-02-10 09:40:59607 if (ret < 0)
608 return ret;
609
Anton Khirnov404e5142016-08-09 10:05:49610 *surf = &frame->surface;
Anton Khirnov4e08c822015-02-10 09:40:59611
612 return 0;
613}
614
Anton Khirnovf5c4d382015-07-14 16:16:26615static QSVFrame *find_frame(QSVContext *q, mfxFrameSurface1 *surf)
Anton Khirnov4e08c822015-02-10 09:40:59616{
617 QSVFrame *cur = q->work_frames;
618 while (cur) {
Anton Khirnov404e5142016-08-09 10:05:49619 if (surf == &cur->surface)
Anton Khirnovf5c4d382015-07-14 16:16:26620 return cur;
Anton Khirnov4e08c822015-02-10 09:40:59621 cur = cur->next;
622 }
623 return NULL;
624}
625
Haihao Xiang8ca06a82022-01-24 08:24:55626#if QSV_VERSION_ATLEAST(1, 34)
627static int qsv_export_film_grain(AVCodecContext *avctx, mfxExtAV1FilmGrainParam *ext_param, AVFrame *frame)
628{
629 AVFilmGrainParams *fgp;
630 AVFilmGrainAOMParams *aom;
631 int i;
632
633 if (!(ext_param->FilmGrainFlags & MFX_FILM_GRAIN_APPLY))
634 return 0;
635
636 fgp = av_film_grain_params_create_side_data(frame);
637
638 if (!fgp)
639 return AVERROR(ENOMEM);
640
641 fgp->type = AV_FILM_GRAIN_PARAMS_AV1;
642 fgp->seed = ext_param->GrainSeed;
643 aom = &fgp->codec.aom;
644
645 aom->chroma_scaling_from_luma = !!(ext_param->FilmGrainFlags & MFX_FILM_GRAIN_CHROMA_SCALING_FROM_LUMA);
646 aom->scaling_shift = ext_param->GrainScalingMinus8 + 8;
647 aom->ar_coeff_lag = ext_param->ArCoeffLag;
648 aom->ar_coeff_shift = ext_param->ArCoeffShiftMinus6 + 6;
649 aom->grain_scale_shift = ext_param->GrainScaleShift;
650 aom->overlap_flag = !!(ext_param->FilmGrainFlags & MFX_FILM_GRAIN_OVERLAP);
651 aom->limit_output_range = !!(ext_param->FilmGrainFlags & MFX_FILM_GRAIN_CLIP_TO_RESTRICTED_RANGE);
652
653 aom->num_y_points = ext_param->NumYPoints;
654
655 for (i = 0; i < aom->num_y_points; i++) {
656 aom->y_points[i][0] = ext_param->PointY[i].Value;
657 aom->y_points[i][1] = ext_param->PointY[i].Scaling;
658 }
659
660 aom->num_uv_points[0] = ext_param->NumCbPoints;
661
662 for (i = 0; i < aom->num_uv_points[0]; i++) {
663 aom->uv_points[0][i][0] = ext_param->PointCb[i].Value;
664 aom->uv_points[0][i][1] = ext_param->PointCb[i].Scaling;
665 }
666
667 aom->num_uv_points[1] = ext_param->NumCrPoints;
668
669 for (i = 0; i < aom->num_uv_points[1]; i++) {
670 aom->uv_points[1][i][0] = ext_param->PointCr[i].Value;
671 aom->uv_points[1][i][1] = ext_param->PointCr[i].Scaling;
672 }
673
674 for (i = 0; i < 24; i++)
675 aom->ar_coeffs_y[i] = ext_param->ArCoeffsYPlus128[i] - 128;
676
677 for (i = 0; i < 25; i++) {
678 aom->ar_coeffs_uv[0][i] = ext_param->ArCoeffsCbPlus128[i] - 128;
679 aom->ar_coeffs_uv[1][i] = ext_param->ArCoeffsCrPlus128[i] - 128;
680 }
681
682 aom->uv_mult[0] = ext_param->CbMult;
683 aom->uv_mult[1] = ext_param->CrMult;
684 aom->uv_mult_luma[0] = ext_param->CbLumaMult;
685 aom->uv_mult_luma[1] = ext_param->CrLumaMult;
686 aom->uv_offset[0] = ext_param->CbOffset;
687 aom->uv_offset[1] = ext_param->CrOffset;
688
689 return 0;
690}
691#endif
692
Haihao Xiang71cded62022-11-21 02:31:24693#if QSV_VERSION_ATLEAST(1, 35)
694static int qsv_export_hdr_side_data(AVCodecContext *avctx, mfxExtMasteringDisplayColourVolume *mdcv,
695 mfxExtContentLightLevelInfo *clli, AVFrame *frame)
696{
Anton Khirnov6169bd02024-02-23 09:21:54697 int ret;
698
Haihao Xiang71cded62022-11-21 02:31:24699 // The SDK re-uses this flag for HDR SEI parsing
700 if (mdcv->InsertPayloadToggle) {
Anton Khirnov6169bd02024-02-23 09:21:54701 AVMasteringDisplayMetadata *mastering;
Haihao Xiang71cded62022-11-21 02:31:24702 const int mapping[3] = {2, 0, 1};
703 const int chroma_den = 50000;
704 const int luma_den = 10000;
705 int i;
706
Anton Khirnov6169bd02024-02-23 09:21:54707 ret = ff_decode_mastering_display_new(avctx, frame, &mastering);
708 if (ret < 0)
709 return ret;
Haihao Xiang71cded62022-11-21 02:31:24710
Anton Khirnov6169bd02024-02-23 09:21:54711 if (mastering) {
712 for (i = 0; i < 3; i++) {
713 const int j = mapping[i];
714 mastering->display_primaries[i][0] = av_make_q(mdcv->DisplayPrimariesX[j], chroma_den);
715 mastering->display_primaries[i][1] = av_make_q(mdcv->DisplayPrimariesY[j], chroma_den);
716 }
717
718 mastering->white_point[0] = av_make_q(mdcv->WhitePointX, chroma_den);
719 mastering->white_point[1] = av_make_q(mdcv->WhitePointY, chroma_den);
720
721 mastering->max_luminance = av_make_q(mdcv->MaxDisplayMasteringLuminance, luma_den);
722 mastering->min_luminance = av_make_q(mdcv->MinDisplayMasteringLuminance, luma_den);
723
724 mastering->has_luminance = 1;
725 mastering->has_primaries = 1;
Haihao Xiang71cded62022-11-21 02:31:24726 }
Haihao Xiang71cded62022-11-21 02:31:24727 }
728
729 // The SDK re-uses this flag for HDR SEI parsing
730 if (clli->InsertPayloadToggle) {
Anton Khirnov6169bd02024-02-23 09:21:54731 AVContentLightMetadata *light;
Haihao Xiang71cded62022-11-21 02:31:24732
Anton Khirnov6169bd02024-02-23 09:21:54733 ret = ff_decode_content_light_new(avctx, frame, &light);
734 if (ret < 0)
735 return ret;
Haihao Xiang71cded62022-11-21 02:31:24736
Anton Khirnov6169bd02024-02-23 09:21:54737 if (light) {
738 light->MaxCLL = clli->MaxContentLightLevel;
739 light->MaxFALL = clli->MaxPicAverageLightLevel;
740 }
Haihao Xiang71cded62022-11-21 02:31:24741 }
742
743 return 0;
744}
745
Haihao Xiang473e84a2024-05-22 04:15:59746static int qsv_export_hdr_side_data_av1(AVCodecContext *avctx, mfxExtMasteringDisplayColourVolume *mdcv,
747 mfxExtContentLightLevelInfo *clli, AVFrame *frame)
748{
749 if (mdcv->InsertPayloadToggle) {
750 AVMasteringDisplayMetadata *mastering = av_mastering_display_metadata_create_side_data(frame);
751 const int chroma_den = 1 << 16;
752 const int max_luma_den = 1 << 8;
753 const int min_luma_den = 1 << 14;
754
755 if (!mastering)
756 return AVERROR(ENOMEM);
757
758 for (int i = 0; i < 3; i++) {
759 mastering->display_primaries[i][0] = av_make_q(mdcv->DisplayPrimariesX[i], chroma_den);
760 mastering->display_primaries[i][1] = av_make_q(mdcv->DisplayPrimariesY[i], chroma_den);
761 }
762
763 mastering->white_point[0] = av_make_q(mdcv->WhitePointX, chroma_den);
764 mastering->white_point[1] = av_make_q(mdcv->WhitePointY, chroma_den);
765
766 mastering->max_luminance = av_make_q(mdcv->MaxDisplayMasteringLuminance, max_luma_den);
767 mastering->min_luminance = av_make_q(mdcv->MinDisplayMasteringLuminance, min_luma_den);
768
769 mastering->has_luminance = 1;
770 mastering->has_primaries = 1;
771 }
772
773 if (clli->InsertPayloadToggle) {
774 AVContentLightMetadata *light = av_content_light_metadata_create_side_data(frame);
775 if (!light)
776 return AVERROR(ENOMEM);
777
778 light->MaxCLL = clli->MaxContentLightLevel;
779 light->MaxFALL = clli->MaxPicAverageLightLevel;
780 }
781
782 return 0;
783}
784
Haihao Xiang71cded62022-11-21 02:31:24785#endif
786
Mark Thompson1f26a232016-10-21 17:57:12787static int qsv_decode(AVCodecContext *avctx, QSVContext *q,
788 AVFrame *frame, int *got_frame,
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41789 const AVPacket *avpkt)
Anton Khirnov4e08c822015-02-10 09:40:59790{
791 mfxFrameSurface1 *insurf;
792 mfxFrameSurface1 *outsurf;
Mark Thompson1f26a232016-10-21 17:57:12793 mfxSyncPoint *sync;
Anton Khirnov4e08c822015-02-10 09:40:59794 mfxBitstream bs = { { { 0 } } };
795 int ret;
796
Mark Thompson1f26a232016-10-21 17:57:12797 if (avpkt->size) {
798 bs.Data = avpkt->data;
799 bs.DataLength = avpkt->size;
Anton Khirnov4e08c822015-02-10 09:40:59800 bs.MaxLength = bs.DataLength;
Haihao Xiang971b4ac2021-06-11 02:19:45801 bs.TimeStamp = PTS_TO_MFX_PTS(avpkt->pts, avctx->pkt_timebase);
Zhong Li54307b32018-04-07 17:38:55802 if (avctx->field_order == AV_FIELD_PROGRESSIVE)
803 bs.DataFlag |= MFX_BITSTREAM_COMPLETE_FRAME;
Anton Khirnov4e08c822015-02-10 09:40:59804 }
805
Mark Thompson1f26a232016-10-21 17:57:12806 sync = av_mallocz(sizeof(*sync));
807 if (!sync) {
808 av_freep(&sync);
809 return AVERROR(ENOMEM);
810 }
811
812 do {
Anton Khirnov4e08c822015-02-10 09:40:59813 ret = get_surface(avctx, q, &insurf);
Timothy Gub6f80b12016-12-05 18:20:26814 if (ret < 0) {
815 av_freep(&sync);
Anton Khirnov4e08c822015-02-10 09:40:59816 return ret;
Timothy Gub6f80b12016-12-05 18:20:26817 }
Mark Thompson1f26a232016-10-21 17:57:12818
819 ret = MFXVideoDECODE_DecodeFrameAsync(q->session, avpkt->size ? &bs : NULL,
820 insurf, &outsurf, sync);
821 if (ret == MFX_WRN_DEVICE_BUSY)
Ivan Uskov9f543e02015-07-24 11:45:38822 av_usleep(500);
Anton Khirnov4e08c822015-02-10 09:40:59823
Mark Thompson1f26a232016-10-21 17:57:12824 } while (ret == MFX_WRN_DEVICE_BUSY || ret == MFX_ERR_MORE_SURFACE);
Anton Khirnov4e08c822015-02-10 09:40:59825
Wenbin Chen40f38652022-03-18 06:25:09826 if (ret == MFX_ERR_INCOMPATIBLE_VIDEO_PARAM) {
827 q->reinit_flag = 1;
828 av_log(avctx, AV_LOG_DEBUG, "Video parameter change\n");
829 av_freep(&sync);
830 return 0;
831 }
832
Mark Thompson1f26a232016-10-21 17:57:12833 if (ret != MFX_ERR_NONE &&
834 ret != MFX_ERR_MORE_DATA &&
835 ret != MFX_WRN_VIDEO_PARAM_CHANGED &&
836 ret != MFX_ERR_MORE_SURFACE) {
Mark Thompson1f26a232016-10-21 17:57:12837 av_freep(&sync);
Anton Khirnov95414eb2016-06-25 19:38:10838 return ff_qsv_print_error(avctx, ret,
839 "Error during QSV decoding.");
Anton Khirnovf5c4d382015-07-14 16:16:26840 }
841
Anton Khirnovaa9d15d2015-07-09 18:08:13842 /* make sure we do not enter an infinite loop if the SDK
843 * did not consume any data and did not return anything */
Mark Thompson1f26a232016-10-21 17:57:12844 if (!*sync && !bs.DataOffset) {
Anton Khirnovaa9d15d2015-07-09 18:08:13845 bs.DataOffset = avpkt->size;
Mark Thompson0940b742016-10-30 16:58:23846 ++q->zero_consume_run;
Haihao Xiang697251b2024-03-11 04:37:57847 if (q->zero_consume_run > 1 &&
848 (avpkt->size ||
849 ret != MFX_ERR_MORE_DATA))
Mark Thompson0940b742016-10-30 16:58:23850 ff_qsv_print_warning(avctx, ret, "A decode call did not consume any data");
851 } else {
852 q->zero_consume_run = 0;
Anton Khirnovaa9d15d2015-07-09 18:08:13853 }
854
Mark Thompson1f26a232016-10-21 17:57:12855 if (*sync) {
Anton Khirnov08181622022-01-06 16:16:47856 QSVAsyncFrame aframe;
Mark Thompson1f26a232016-10-21 17:57:12857 QSVFrame *out_frame = find_frame(q, outsurf);
858
859 if (!out_frame) {
860 av_log(avctx, AV_LOG_ERROR,
861 "The returned surface does not correspond to any frame\n");
862 av_freep(&sync);
863 return AVERROR_BUG;
864 }
865
Chen,Wenbine6b990e2021-03-12 02:44:06866 out_frame->queued += 1;
Anton Khirnov08181622022-01-06 16:16:47867
868 aframe = (QSVAsyncFrame){ sync, out_frame };
869 av_fifo_write(q->async_fifo, &aframe, 1);
Mark Thompson1f26a232016-10-21 17:57:12870 } else {
871 av_freep(&sync);
Ivan Uskovc90dbc62015-07-24 10:26:14872 }
873
Anton Khirnov08181622022-01-06 16:16:47874 if ((av_fifo_can_read(q->async_fifo) >= q->async_depth) ||
875 (!avpkt->size && av_fifo_can_read(q->async_fifo))) {
876 QSVAsyncFrame aframe;
Anton Khirnovf5c4d382015-07-14 16:16:26877 AVFrame *src_frame;
878
Anton Khirnov08181622022-01-06 16:16:47879 av_fifo_read(q->async_fifo, &aframe, 1);
880 aframe.frame->queued -= 1;
Anton Khirnovf5c4d382015-07-14 16:16:26881
Anton Khirnovb68e3532017-01-07 20:06:16882 if (avctx->pix_fmt != AV_PIX_FMT_QSV) {
883 do {
Anton Khirnov08181622022-01-06 16:16:47884 ret = MFXVideoCORE_SyncOperation(q->session, *aframe.sync, 1000);
Anton Khirnovb68e3532017-01-07 20:06:16885 } while (ret == MFX_WRN_IN_EXECUTION);
886 }
Anton Khirnovf5c4d382015-07-14 16:16:26887
Anton Khirnov08181622022-01-06 16:16:47888 av_freep(&aframe.sync);
Mark Thompson1f26a232016-10-21 17:57:12889
Anton Khirnov08181622022-01-06 16:16:47890 src_frame = aframe.frame->frame;
Anton Khirnovf5c4d382015-07-14 16:16:26891
Anton Khirnov4e08c822015-02-10 09:40:59892 ret = av_frame_ref(frame, src_frame);
893 if (ret < 0)
894 return ret;
895
Anton Khirnov08181622022-01-06 16:16:47896 outsurf = &aframe.frame->surface;
Anton Khirnovf5c4d382015-07-14 16:16:26897
Haihao Xiang971b4ac2021-06-11 02:19:45898 frame->pts = MFX_PTS_TO_PTS(outsurf->Data.TimeStamp, avctx->pkt_timebase);
Haihao Xiang8ca06a82022-01-24 08:24:55899#if QSV_VERSION_ATLEAST(1, 34)
900 if ((avctx->export_side_data & AV_CODEC_EXPORT_DATA_FILM_GRAIN) &&
901 QSV_RUNTIME_VERSION_ATLEAST(q->ver, 1, 34) &&
902 avctx->codec_id == AV_CODEC_ID_AV1) {
Anton Khirnov08181622022-01-06 16:16:47903 ret = qsv_export_film_grain(avctx, &aframe.frame->av1_film_grain_param, frame);
Haihao Xiang8ca06a82022-01-24 08:24:55904
905 if (ret < 0)
906 return ret;
907 }
908#endif
Anton Khirnov4e08c822015-02-10 09:40:59909
Haihao Xiang71cded62022-11-21 02:31:24910#if QSV_VERSION_ATLEAST(1, 35)
911 if (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 1, 35) && avctx->codec_id == AV_CODEC_ID_HEVC) {
912 ret = qsv_export_hdr_side_data(avctx, &aframe.frame->mdcv, &aframe.frame->clli, frame);
913
914 if (ret < 0)
915 return ret;
916 }
Haihao Xiang473e84a2024-05-22 04:15:59917
918 if (QSV_RUNTIME_VERSION_ATLEAST(q->ver, 2, 9) && avctx->codec_id == AV_CODEC_ID_AV1) {
919 ret = qsv_export_hdr_side_data_av1(avctx, &aframe.frame->mdcv, &aframe.frame->clli, frame);
920 if (ret < 0)
921 return ret;
922 }
Haihao Xiang71cded62022-11-21 02:31:24923#endif
924
Anton Khirnov4e08c822015-02-10 09:40:59925 frame->repeat_pict =
926 outsurf->Info.PicStruct & MFX_PICSTRUCT_FRAME_TRIPLING ? 4 :
927 outsurf->Info.PicStruct & MFX_PICSTRUCT_FRAME_DOUBLING ? 2 :
928 outsurf->Info.PicStruct & MFX_PICSTRUCT_FIELD_REPEATED ? 1 : 0;
James Almer2f561ba2023-04-11 18:02:14929 frame->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST *
930 !!(outsurf->Info.PicStruct & MFX_PICSTRUCT_FIELD_TFF);
931 frame->flags |= AV_FRAME_FLAG_INTERLACED *
Anton Khirnov4e08c822015-02-10 09:40:59932 !(outsurf->Info.PicStruct & MFX_PICSTRUCT_PROGRESSIVE);
Anton Khirnov08181622022-01-06 16:16:47933 frame->pict_type = ff_qsv_map_pictype(aframe.frame->dec_info.FrameType);
Haihao Xiangdbdd9cc2024-05-22 04:31:53934
935 if (avctx->codec_id == AV_CODEC_ID_H264 ||
Fei Wangd30a9fd2024-06-14 08:28:54936 avctx->codec_id == AV_CODEC_ID_HEVC ||
937 avctx->codec_id == AV_CODEC_ID_VVC) {
James Almerdc7bd7c2023-04-12 16:58:54938 if (aframe.frame->dec_info.FrameType & MFX_FRAMETYPE_IDR)
939 frame->flags |= AV_FRAME_FLAG_KEY;
940 else
941 frame->flags &= ~AV_FRAME_FLAG_KEY;
Haihao Xiangdbdd9cc2024-05-22 04:31:53942 } else {
943 if (aframe.frame->dec_info.FrameType & MFX_FRAMETYPE_I)
944 frame->flags |= AV_FRAME_FLAG_KEY;
945 else
946 frame->flags &= ~AV_FRAME_FLAG_KEY;
James Almerdc7bd7c2023-04-12 16:58:54947 }
Fei Wang1c562632024-05-20 02:05:52948 frame->crop_left = outsurf->Info.CropX;
949 frame->crop_top = outsurf->Info.CropY;
950 frame->crop_right = outsurf->Info.Width - (outsurf->Info.CropX + outsurf->Info.CropW);
951 frame->crop_bottom = outsurf->Info.Height - (outsurf->Info.CropY + outsurf->Info.CropH);
Anton Khirnov4e08c822015-02-10 09:40:59952
Anton Khirnov404e5142016-08-09 10:05:49953 /* update the surface properties */
954 if (avctx->pix_fmt == AV_PIX_FMT_QSV)
955 ((mfxFrameSurface1*)frame->data[3])->Info = outsurf->Info;
956
Anton Khirnov4e08c822015-02-10 09:40:59957 *got_frame = 1;
958 }
959
Mark Thompson1f26a232016-10-21 17:57:12960 return bs.DataOffset;
Ivan Uskov3f8e2e92015-08-06 16:10:24961}
Anton Khirnov4e08c822015-02-10 09:40:59962
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41963static void qsv_decode_close_qsvcontext(QSVContext *q)
Anton Khirnov4e08c822015-02-10 09:40:59964{
Mark Thompson1f26a232016-10-21 17:57:12965 QSVFrame *cur = q->work_frames;
Anton Khirnov4e08c822015-02-10 09:40:59966
Mark Thompson1f26a232016-10-21 17:57:12967 if (q->session)
968 MFXVideoDECODE_Close(q->session);
Ivan Uskovcc167f72015-08-04 10:40:06969
Anton Khirnov08181622022-01-06 16:16:47970 if (q->async_fifo) {
971 QSVAsyncFrame aframe;
972 while (av_fifo_read(q->async_fifo, &aframe, 1) >= 0)
973 av_freep(&aframe.sync);
974 av_fifo_freep2(&q->async_fifo);
Mark Thompson1f26a232016-10-21 17:57:12975 }
976
977 while (cur) {
978 q->work_frames = cur->next;
979 av_frame_free(&cur->frame);
980 av_freep(&cur);
981 cur = q->work_frames;
982 }
Anton Khirnov4e08c822015-02-10 09:40:59983
Zhong Li74007dd2019-09-19 20:45:26984 ff_qsv_close_internal_session(&q->internal_qs);
Mark Thompson1f26a232016-10-21 17:57:12985
986 av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
Anton Khirnov56ba57b2024-10-15 20:37:56987 av_refstruct_unref(&q->frames_ctx.mids);
Linjie Fu53459652019-10-08 13:41:02988 av_buffer_pool_uninit(&q->pool);
Anton Khirnov4e08c822015-02-10 09:40:59989}
Mark Thompson1f26a232016-10-21 17:57:12990
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41991static int qsv_process_data(AVCodecContext *avctx, QSVContext *q,
992 AVFrame *frame, int *got_frame, const AVPacket *pkt)
Mark Thompson1f26a232016-10-21 17:57:12993{
Mark Thompson1f26a232016-10-21 17:57:12994 int ret;
Zhong Li00d0a4a2019-08-13 06:11:09995 mfxVideoParam param = { 0 };
996 enum AVPixelFormat pix_fmt = AV_PIX_FMT_NV12;
Mark Thompson1f26a232016-10-21 17:57:12997
Mark Thompson1f26a232016-10-21 17:57:12998 if (!pkt->size)
999 return qsv_decode(avctx, q, frame, got_frame, pkt);
1000
Mark Thompson1f26a232016-10-21 17:57:121001 /* TODO: flush delayed frames on reinit */
Zhong Li00d0a4a2019-08-13 06:11:091002
1003 // sw_pix_fmt, coded_width/height should be set for ff_get_format(),
1004 // assume sw_pix_fmt is NV12 and coded_width/height to be 1280x720,
1005 // the assumption may be not corret but will be updated after header decoded if not true.
1006 if (q->orig_pix_fmt != AV_PIX_FMT_NONE)
1007 pix_fmt = q->orig_pix_fmt;
1008 if (!avctx->coded_width)
1009 avctx->coded_width = 1280;
1010 if (!avctx->coded_height)
1011 avctx->coded_height = 720;
1012
Wenbin Chen4a633eb2022-03-18 06:25:101013 /* decode zero-size pkt to flush the buffered pkt before reinit */
1014 if (q->reinit_flag) {
Linjie Fu87368882018-10-16 01:36:131015 AVPacket zero_pkt = {0};
Wenbin Chen4a633eb2022-03-18 06:25:101016 ret = qsv_decode(avctx, q, frame, got_frame, &zero_pkt);
1017 if (ret < 0 || *got_frame)
1018 return ret;
1019 }
Linjie Fu87368882018-10-16 01:36:131020
Wenbin Chen4a633eb2022-03-18 06:25:101021 if (q->reinit_flag || !q->session || !q->initialized) {
Wenbin Chen0a0847d2022-03-18 06:25:111022 mfxFrameAllocRequest request;
1023 memset(&request, 0, sizeof(request));
1024
Linjie Fu87368882018-10-16 01:36:131025 q->reinit_flag = 0;
Wenbin Chen4a633eb2022-03-18 06:25:101026 ret = qsv_decode_header(avctx, q, pkt, pix_fmt, &param);
1027 if (ret < 0) {
1028 if (ret == AVERROR(EAGAIN))
Haihao Xiangfc73b372023-11-27 06:01:511029 av_log(avctx, AV_LOG_VERBOSE, "More data is required to decode header\n");
Wenbin Chen4a633eb2022-03-18 06:25:101030 else
1031 av_log(avctx, AV_LOG_ERROR, "Error decoding header\n");
1032 goto reinit_fail;
1033 }
Wenbin Chen0a0847d2022-03-18 06:25:111034 param.IOPattern = q->iopattern;
Mark Thompson1f26a232016-10-21 17:57:121035
Zhong Li00d0a4a2019-08-13 06:11:091036 q->orig_pix_fmt = avctx->pix_fmt = pix_fmt = ff_qsv_map_fourcc(param.mfx.FrameInfo.FourCC);
Mark Thompson1f26a232016-10-21 17:57:121037
Zhong Li00d0a4a2019-08-13 06:11:091038 avctx->coded_width = param.mfx.FrameInfo.Width;
1039 avctx->coded_height = param.mfx.FrameInfo.Height;
Mark Thompson1f26a232016-10-21 17:57:121040
Wenbin Chen0a0847d2022-03-18 06:25:111041 ret = MFXVideoDECODE_QueryIOSurf(q->session, &param, &request);
1042 if (ret < 0)
1043 return ff_qsv_print_error(avctx, ret, "Error querying IO surface");
1044
1045 q->suggest_pool_size = request.NumFrameSuggested;
1046
Zhong Li00d0a4a2019-08-13 06:11:091047 ret = qsv_decode_preinit(avctx, q, pix_fmt, &param);
Mark Thompson1f26a232016-10-21 17:57:121048 if (ret < 0)
1049 goto reinit_fail;
Zhong Li00d0a4a2019-08-13 06:11:091050 q->initialized = 0;
1051 }
Mark Thompson1f26a232016-10-21 17:57:121052
Zhong Li00d0a4a2019-08-13 06:11:091053 if (!q->initialized) {
Xu Guangxin399c1f92021-01-05 02:43:401054 ret = qsv_decode_init_context(avctx, q, &param);
Mark Thompson1f26a232016-10-21 17:57:121055 if (ret < 0)
1056 goto reinit_fail;
Zhong Li00d0a4a2019-08-13 06:11:091057 q->initialized = 1;
Mark Thompson1f26a232016-10-21 17:57:121058 }
1059
1060 return qsv_decode(avctx, q, frame, got_frame, pkt);
1061
1062reinit_fail:
Zhong Li0dfcfc52019-08-13 06:11:101063 q->orig_pix_fmt = avctx->pix_fmt = AV_PIX_FMT_NONE;
Mark Thompson1f26a232016-10-21 17:57:121064 return ret;
1065}
1066
Xu Guangxin399c1f92021-01-05 02:43:401067enum LoadPlugin {
1068 LOAD_PLUGIN_NONE,
1069 LOAD_PLUGIN_HEVC_SW,
1070 LOAD_PLUGIN_HEVC_HW,
1071};
1072
1073typedef struct QSVDecContext {
1074 AVClass *class;
1075 QSVContext qsv;
1076
1077 int load_plugin;
1078
Anton Khirnov08181622022-01-06 16:16:471079 AVFifo *packet_fifo;
Xu Guangxin399c1f92021-01-05 02:43:401080
1081 AVPacket buffer_pkt;
1082} QSVDecContext;
1083
1084static void qsv_clear_buffers(QSVDecContext *s)
1085{
1086 AVPacket pkt;
Anton Khirnov08181622022-01-06 16:16:471087 while (av_fifo_read(s->packet_fifo, &pkt, 1) >= 0)
Xu Guangxin399c1f92021-01-05 02:43:401088 av_packet_unref(&pkt);
Xu Guangxin399c1f92021-01-05 02:43:401089
1090 av_packet_unref(&s->buffer_pkt);
1091}
1092
1093static av_cold int qsv_decode_close(AVCodecContext *avctx)
1094{
1095 QSVDecContext *s = avctx->priv_data;
1096
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411097 qsv_decode_close_qsvcontext(&s->qsv);
Xu Guangxin399c1f92021-01-05 02:43:401098
1099 qsv_clear_buffers(s);
1100
Anton Khirnov08181622022-01-06 16:16:471101 av_fifo_freep2(&s->packet_fifo);
Xu Guangxin399c1f92021-01-05 02:43:401102
1103 return 0;
1104}
1105
1106static av_cold int qsv_decode_init(AVCodecContext *avctx)
1107{
1108 QSVDecContext *s = avctx->priv_data;
1109 int ret;
Xu Guangxin7a5a5e62021-01-05 02:43:421110 const char *uid = NULL;
Xu Guangxin399c1f92021-01-05 02:43:401111
Xu Guangxind78ecf12021-01-05 02:43:411112 if (avctx->codec_id == AV_CODEC_ID_VP8) {
Xu Guangxin7a5a5e62021-01-05 02:43:421113 uid = "f622394d8d87452f878c51f2fc9b4131";
Xu Guangxind78ecf12021-01-05 02:43:411114 } else if (avctx->codec_id == AV_CODEC_ID_VP9) {
Xu Guangxin7a5a5e62021-01-05 02:43:421115 uid = "a922394d8d87452f878c51f2fc9b4131";
Xu Guangxind78ecf12021-01-05 02:43:411116 }
1117 else if (avctx->codec_id == AV_CODEC_ID_HEVC && s->load_plugin != LOAD_PLUGIN_NONE) {
Xu Guangxin399c1f92021-01-05 02:43:401118 static const char * const uid_hevcdec_sw = "15dd936825ad475ea34e35f3f54217a6";
1119 static const char * const uid_hevcdec_hw = "33a61c0b4c27454ca8d85dde757c6f8e";
1120
1121 if (s->qsv.load_plugins[0]) {
1122 av_log(avctx, AV_LOG_WARNING,
1123 "load_plugins is not empty, but load_plugin is not set to 'none'."
1124 "The load_plugin value will be ignored.\n");
1125 } else {
Xu Guangxin399c1f92021-01-05 02:43:401126 if (s->load_plugin == LOAD_PLUGIN_HEVC_SW)
Xu Guangxin7a5a5e62021-01-05 02:43:421127 uid = uid_hevcdec_sw;
Xu Guangxin399c1f92021-01-05 02:43:401128 else
Xu Guangxin7a5a5e62021-01-05 02:43:421129 uid = uid_hevcdec_hw;
Xu Guangxin399c1f92021-01-05 02:43:401130 }
1131 }
Xu Guangxin7a5a5e62021-01-05 02:43:421132 if (uid) {
1133 av_freep(&s->qsv.load_plugins);
1134 s->qsv.load_plugins = av_strdup(uid);
1135 if (!s->qsv.load_plugins)
1136 return AVERROR(ENOMEM);
1137 }
Xu Guangxin399c1f92021-01-05 02:43:401138
1139 s->qsv.orig_pix_fmt = AV_PIX_FMT_NV12;
Anton Khirnov08181622022-01-06 16:16:471140 s->packet_fifo = av_fifo_alloc2(1, sizeof(AVPacket),
1141 AV_FIFO_FLAG_AUTO_GROW);
Xu Guangxin399c1f92021-01-05 02:43:401142 if (!s->packet_fifo) {
1143 ret = AVERROR(ENOMEM);
1144 goto fail;
1145 }
1146
Haihao Xiang971b4ac2021-06-11 02:19:451147 if (!avctx->pkt_timebase.num)
1148 av_log(avctx, AV_LOG_WARNING, "Invalid pkt_timebase, passing timestamps as-is.\n");
1149
Xu Guangxin399c1f92021-01-05 02:43:401150 return 0;
1151fail:
1152 qsv_decode_close(avctx);
1153 return ret;
1154}
1155
Andreas Rheinhardtce7dbd02022-03-30 19:33:241156static int qsv_decode_frame(AVCodecContext *avctx, AVFrame *frame,
Xu Guangxin399c1f92021-01-05 02:43:401157 int *got_frame, AVPacket *avpkt)
1158{
1159 QSVDecContext *s = avctx->priv_data;
Xu Guangxin399c1f92021-01-05 02:43:401160 int ret;
1161
1162 /* buffer the input packet */
1163 if (avpkt->size) {
1164 AVPacket input_ref;
1165
Xu Guangxin399c1f92021-01-05 02:43:401166 ret = av_packet_ref(&input_ref, avpkt);
1167 if (ret < 0)
1168 return ret;
Anton Khirnov08181622022-01-06 16:16:471169 av_fifo_write(s->packet_fifo, &input_ref, 1);
Xu Guangxin399c1f92021-01-05 02:43:401170 }
1171
1172 /* process buffered data */
1173 while (!*got_frame) {
1174 /* prepare the input data */
1175 if (s->buffer_pkt.size <= 0) {
1176 /* no more data */
Anton Khirnov08181622022-01-06 16:16:471177 if (!av_fifo_can_read(s->packet_fifo))
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411178 return avpkt->size ? avpkt->size : qsv_process_data(avctx, &s->qsv, frame, got_frame, avpkt);
Xu Guangxin399c1f92021-01-05 02:43:401179 /* in progress of reinit, no read from fifo and keep the buffer_pkt */
1180 if (!s->qsv.reinit_flag) {
1181 av_packet_unref(&s->buffer_pkt);
Anton Khirnov08181622022-01-06 16:16:471182 av_fifo_read(s->packet_fifo, &s->buffer_pkt, 1);
Xu Guangxin399c1f92021-01-05 02:43:401183 }
1184 }
1185
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411186 ret = qsv_process_data(avctx, &s->qsv, frame, got_frame, &s->buffer_pkt);
Xu Guangxin399c1f92021-01-05 02:43:401187 if (ret < 0){
Haihao Xiange233f3e2023-11-27 06:01:501188 if (ret == AVERROR(EAGAIN))
1189 ret = 0;
1190
Xu Guangxin399c1f92021-01-05 02:43:401191 /* Drop buffer_pkt when failed to decode the packet. Otherwise,
1192 the decoder will keep decoding the failure packet. */
1193 av_packet_unref(&s->buffer_pkt);
1194 return ret;
1195 }
1196 if (s->qsv.reinit_flag)
1197 continue;
1198
1199 s->buffer_pkt.size -= ret;
1200 s->buffer_pkt.data += ret;
1201 }
1202
1203 return avpkt->size;
1204}
1205
1206static void qsv_decode_flush(AVCodecContext *avctx)
1207{
1208 QSVDecContext *s = avctx->priv_data;
1209
1210 qsv_clear_buffers(s);
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411211
1212 s->qsv.orig_pix_fmt = AV_PIX_FMT_NONE;
1213 s->qsv.initialized = 0;
Xu Guangxin399c1f92021-01-05 02:43:401214}
1215
1216#define OFFSET(x) offsetof(QSVDecContext, x)
1217#define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
1218
1219#define DEFINE_QSV_DECODER_WITH_OPTION(x, X, bsf_name, opt) \
1220static const AVClass x##_qsv_class = { \
1221 .class_name = #x "_qsv", \
Anton Khirnov08bebeb2024-01-19 12:33:281222 .item_name = av_default_item_name, \
Xu Guangxin399c1f92021-01-05 02:43:401223 .option = opt, \
1224 .version = LIBAVUTIL_VERSION_INT, \
1225}; \
Andreas Rheinhardt20f97272022-03-16 20:09:541226const FFCodec ff_##x##_qsv_decoder = { \
1227 .p.name = #x "_qsv", \
Andreas Rheinhardt48286d42022-08-29 11:38:021228 CODEC_LONG_NAME(#X " video (Intel Quick Sync Video acceleration)"), \
Xu Guangxin399c1f92021-01-05 02:43:401229 .priv_data_size = sizeof(QSVDecContext), \
Andreas Rheinhardt20f97272022-03-16 20:09:541230 .p.type = AVMEDIA_TYPE_VIDEO, \
1231 .p.id = AV_CODEC_ID_##X, \
Xu Guangxin399c1f92021-01-05 02:43:401232 .init = qsv_decode_init, \
Andreas Rheinhardt4243da42022-03-30 21:28:241233 FF_CODEC_DECODE_CB(qsv_decode_frame), \
Xu Guangxin399c1f92021-01-05 02:43:401234 .flush = qsv_decode_flush, \
1235 .close = qsv_decode_close, \
1236 .bsfs = bsf_name, \
Andreas Rheinhardt20f97272022-03-16 20:09:541237 .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_DR1 | AV_CODEC_CAP_AVOID_PROBING | AV_CODEC_CAP_HYBRID, \
1238 .p.priv_class = &x##_qsv_class, \
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411239 .hw_configs = qsv_hw_configs, \
Andreas Rheinhardt20f97272022-03-16 20:09:541240 .p.wrapper_name = "qsv", \
Fei Wang1c562632024-05-20 02:05:521241 .caps_internal = FF_CODEC_CAP_NOT_INIT_THREADSAFE | FF_CODEC_CAP_EXPORTS_CROPPING, \
Xu Guangxin399c1f92021-01-05 02:43:401242}; \
1243
1244#define DEFINE_QSV_DECODER(x, X, bsf_name) DEFINE_QSV_DECODER_WITH_OPTION(x, X, bsf_name, options)
1245
1246#if CONFIG_HEVC_QSV_DECODER
1247static const AVOption hevc_options[] = {
1248 { "async_depth", "Internal parallelization depth, the higher the value the higher the latency.", OFFSET(qsv.async_depth), AV_OPT_TYPE_INT, { .i64 = ASYNC_DEPTH_DEFAULT }, 1, INT_MAX, VD },
1249
Anton Khirnov1e7d2002024-02-11 14:41:051250 { "load_plugin", "A user plugin to load in an internal session", OFFSET(load_plugin), AV_OPT_TYPE_INT, { .i64 = LOAD_PLUGIN_HEVC_HW }, LOAD_PLUGIN_NONE, LOAD_PLUGIN_HEVC_HW, VD, .unit = "load_plugin" },
1251 { "none", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = LOAD_PLUGIN_NONE }, 0, 0, VD, .unit = "load_plugin" },
1252 { "hevc_sw", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = LOAD_PLUGIN_HEVC_SW }, 0, 0, VD, .unit = "load_plugin" },
1253 { "hevc_hw", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = LOAD_PLUGIN_HEVC_HW }, 0, 0, VD, .unit = "load_plugin" },
Xu Guangxin399c1f92021-01-05 02:43:401254
1255 { "load_plugins", "A :-separate list of hexadecimal plugin UIDs to load in an internal session",
1256 OFFSET(qsv.load_plugins), AV_OPT_TYPE_STRING, { .str = "" }, 0, 0, VD },
1257
Anton Khirnov1e7d2002024-02-11 14:41:051258 { "gpu_copy", "A GPU-accelerated copy between video and system memory", OFFSET(qsv.gpu_copy), AV_OPT_TYPE_INT, { .i64 = MFX_GPUCOPY_DEFAULT }, MFX_GPUCOPY_DEFAULT, MFX_GPUCOPY_OFF, VD, .unit = "gpu_copy"},
1259 { "default", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_DEFAULT }, 0, 0, VD, .unit = "gpu_copy"},
1260 { "on", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_ON }, 0, 0, VD, .unit = "gpu_copy"},
1261 { "off", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_OFF }, 0, 0, VD, .unit = "gpu_copy"},
Xu Guangxin399c1f92021-01-05 02:43:401262 { NULL },
1263};
1264DEFINE_QSV_DECODER_WITH_OPTION(hevc, HEVC, "hevc_mp4toannexb", hevc_options)
1265#endif
1266
1267static const AVOption options[] = {
1268 { "async_depth", "Internal parallelization depth, the higher the value the higher the latency.", OFFSET(qsv.async_depth), AV_OPT_TYPE_INT, { .i64 = ASYNC_DEPTH_DEFAULT }, 1, INT_MAX, VD },
1269
Anton Khirnov1e7d2002024-02-11 14:41:051270 { "gpu_copy", "A GPU-accelerated copy between video and system memory", OFFSET(qsv.gpu_copy), AV_OPT_TYPE_INT, { .i64 = MFX_GPUCOPY_DEFAULT }, MFX_GPUCOPY_DEFAULT, MFX_GPUCOPY_OFF, VD, .unit = "gpu_copy"},
1271 { "default", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_DEFAULT }, 0, 0, VD, .unit = "gpu_copy"},
1272 { "on", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_ON }, 0, 0, VD, .unit = "gpu_copy"},
1273 { "off", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_OFF }, 0, 0, VD, .unit = "gpu_copy"},
Xu Guangxin399c1f92021-01-05 02:43:401274 { NULL },
1275};
1276
1277#if CONFIG_H264_QSV_DECODER
1278DEFINE_QSV_DECODER(h264, H264, "h264_mp4toannexb")
1279#endif
Xu Guangxind78ecf12021-01-05 02:43:411280
1281#if CONFIG_MPEG2_QSV_DECODER
1282DEFINE_QSV_DECODER(mpeg2, MPEG2VIDEO, NULL)
1283#endif
1284
1285#if CONFIG_VC1_QSV_DECODER
1286DEFINE_QSV_DECODER(vc1, VC1, NULL)
1287#endif
1288
1289#if CONFIG_MJPEG_QSV_DECODER
1290DEFINE_QSV_DECODER(mjpeg, MJPEG, NULL)
1291#endif
1292
1293#if CONFIG_VP8_QSV_DECODER
1294DEFINE_QSV_DECODER(vp8, VP8, NULL)
1295#endif
1296
1297#if CONFIG_VP9_QSV_DECODER
1298DEFINE_QSV_DECODER(vp9, VP9, NULL)
1299#endif
1300
1301#if CONFIG_AV1_QSV_DECODER
1302DEFINE_QSV_DECODER(av1, AV1, NULL)
1303#endif
Fei Wangd30a9fd2024-06-14 08:28:541304
1305#if CONFIG_VVC_QSV_DECODER
Fei Wangbe7ab632024-08-15 02:19:501306DEFINE_QSV_DECODER(vvc, VVC, "vvc_mp4toannexb")
Fei Wangd30a9fd2024-06-14 08:28:541307#endif