blob: 768968fd7bca0387de52011d26971ccf6bd7633d [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"
Andreas Rheinhardtd692c422023-09-24 10:49:5854#include "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);
Zhong Li52ed83f2018-04-04 09:51:29934 //Key frame is IDR frame is only suitable for H264. For HEVC, IRAPs are key frames.
James Almerdc7bd7c2023-04-12 16:58:54935 if (avctx->codec_id == AV_CODEC_ID_H264) {
936 if (aframe.frame->dec_info.FrameType & MFX_FRAMETYPE_IDR)
937 frame->flags |= AV_FRAME_FLAG_KEY;
938 else
939 frame->flags &= ~AV_FRAME_FLAG_KEY;
940 }
Fei Wang1c562632024-05-20 02:05:52941 frame->crop_left = outsurf->Info.CropX;
942 frame->crop_top = outsurf->Info.CropY;
943 frame->crop_right = outsurf->Info.Width - (outsurf->Info.CropX + outsurf->Info.CropW);
944 frame->crop_bottom = outsurf->Info.Height - (outsurf->Info.CropY + outsurf->Info.CropH);
Anton Khirnov4e08c822015-02-10 09:40:59945
Anton Khirnov404e5142016-08-09 10:05:49946 /* update the surface properties */
947 if (avctx->pix_fmt == AV_PIX_FMT_QSV)
948 ((mfxFrameSurface1*)frame->data[3])->Info = outsurf->Info;
949
Anton Khirnov4e08c822015-02-10 09:40:59950 *got_frame = 1;
951 }
952
Mark Thompson1f26a232016-10-21 17:57:12953 return bs.DataOffset;
Ivan Uskov3f8e2e92015-08-06 16:10:24954}
Anton Khirnov4e08c822015-02-10 09:40:59955
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41956static void qsv_decode_close_qsvcontext(QSVContext *q)
Anton Khirnov4e08c822015-02-10 09:40:59957{
Mark Thompson1f26a232016-10-21 17:57:12958 QSVFrame *cur = q->work_frames;
Anton Khirnov4e08c822015-02-10 09:40:59959
Mark Thompson1f26a232016-10-21 17:57:12960 if (q->session)
961 MFXVideoDECODE_Close(q->session);
Ivan Uskovcc167f72015-08-04 10:40:06962
Anton Khirnov08181622022-01-06 16:16:47963 if (q->async_fifo) {
964 QSVAsyncFrame aframe;
965 while (av_fifo_read(q->async_fifo, &aframe, 1) >= 0)
966 av_freep(&aframe.sync);
967 av_fifo_freep2(&q->async_fifo);
Mark Thompson1f26a232016-10-21 17:57:12968 }
969
970 while (cur) {
971 q->work_frames = cur->next;
972 av_frame_free(&cur->frame);
973 av_freep(&cur);
974 cur = q->work_frames;
975 }
Anton Khirnov4e08c822015-02-10 09:40:59976
Zhong Li74007dd2019-09-19 20:45:26977 ff_qsv_close_internal_session(&q->internal_qs);
Mark Thompson1f26a232016-10-21 17:57:12978
979 av_buffer_unref(&q->frames_ctx.hw_frames_ctx);
Andreas Rheinhardtd692c422023-09-24 10:49:58980 ff_refstruct_unref(&q->frames_ctx.mids);
Linjie Fu53459652019-10-08 13:41:02981 av_buffer_pool_uninit(&q->pool);
Anton Khirnov4e08c822015-02-10 09:40:59982}
Mark Thompson1f26a232016-10-21 17:57:12983
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:41984static int qsv_process_data(AVCodecContext *avctx, QSVContext *q,
985 AVFrame *frame, int *got_frame, const AVPacket *pkt)
Mark Thompson1f26a232016-10-21 17:57:12986{
Mark Thompson1f26a232016-10-21 17:57:12987 int ret;
Zhong Li00d0a4a2019-08-13 06:11:09988 mfxVideoParam param = { 0 };
989 enum AVPixelFormat pix_fmt = AV_PIX_FMT_NV12;
Mark Thompson1f26a232016-10-21 17:57:12990
Mark Thompson1f26a232016-10-21 17:57:12991 if (!pkt->size)
992 return qsv_decode(avctx, q, frame, got_frame, pkt);
993
Mark Thompson1f26a232016-10-21 17:57:12994 /* TODO: flush delayed frames on reinit */
Zhong Li00d0a4a2019-08-13 06:11:09995
996 // sw_pix_fmt, coded_width/height should be set for ff_get_format(),
997 // assume sw_pix_fmt is NV12 and coded_width/height to be 1280x720,
998 // the assumption may be not corret but will be updated after header decoded if not true.
999 if (q->orig_pix_fmt != AV_PIX_FMT_NONE)
1000 pix_fmt = q->orig_pix_fmt;
1001 if (!avctx->coded_width)
1002 avctx->coded_width = 1280;
1003 if (!avctx->coded_height)
1004 avctx->coded_height = 720;
1005
Wenbin Chen4a633eb2022-03-18 06:25:101006 /* decode zero-size pkt to flush the buffered pkt before reinit */
1007 if (q->reinit_flag) {
Linjie Fu87368882018-10-16 01:36:131008 AVPacket zero_pkt = {0};
Wenbin Chen4a633eb2022-03-18 06:25:101009 ret = qsv_decode(avctx, q, frame, got_frame, &zero_pkt);
1010 if (ret < 0 || *got_frame)
1011 return ret;
1012 }
Linjie Fu87368882018-10-16 01:36:131013
Wenbin Chen4a633eb2022-03-18 06:25:101014 if (q->reinit_flag || !q->session || !q->initialized) {
Wenbin Chen0a0847d2022-03-18 06:25:111015 mfxFrameAllocRequest request;
1016 memset(&request, 0, sizeof(request));
1017
Linjie Fu87368882018-10-16 01:36:131018 q->reinit_flag = 0;
Wenbin Chen4a633eb2022-03-18 06:25:101019 ret = qsv_decode_header(avctx, q, pkt, pix_fmt, &param);
1020 if (ret < 0) {
1021 if (ret == AVERROR(EAGAIN))
Haihao Xiangfc73b372023-11-27 06:01:511022 av_log(avctx, AV_LOG_VERBOSE, "More data is required to decode header\n");
Wenbin Chen4a633eb2022-03-18 06:25:101023 else
1024 av_log(avctx, AV_LOG_ERROR, "Error decoding header\n");
1025 goto reinit_fail;
1026 }
Wenbin Chen0a0847d2022-03-18 06:25:111027 param.IOPattern = q->iopattern;
Mark Thompson1f26a232016-10-21 17:57:121028
Zhong Li00d0a4a2019-08-13 06:11:091029 q->orig_pix_fmt = avctx->pix_fmt = pix_fmt = ff_qsv_map_fourcc(param.mfx.FrameInfo.FourCC);
Mark Thompson1f26a232016-10-21 17:57:121030
Zhong Li00d0a4a2019-08-13 06:11:091031 avctx->coded_width = param.mfx.FrameInfo.Width;
1032 avctx->coded_height = param.mfx.FrameInfo.Height;
Mark Thompson1f26a232016-10-21 17:57:121033
Wenbin Chen0a0847d2022-03-18 06:25:111034 ret = MFXVideoDECODE_QueryIOSurf(q->session, &param, &request);
1035 if (ret < 0)
1036 return ff_qsv_print_error(avctx, ret, "Error querying IO surface");
1037
1038 q->suggest_pool_size = request.NumFrameSuggested;
1039
Zhong Li00d0a4a2019-08-13 06:11:091040 ret = qsv_decode_preinit(avctx, q, pix_fmt, &param);
Mark Thompson1f26a232016-10-21 17:57:121041 if (ret < 0)
1042 goto reinit_fail;
Zhong Li00d0a4a2019-08-13 06:11:091043 q->initialized = 0;
1044 }
Mark Thompson1f26a232016-10-21 17:57:121045
Zhong Li00d0a4a2019-08-13 06:11:091046 if (!q->initialized) {
Xu Guangxin399c1f92021-01-05 02:43:401047 ret = qsv_decode_init_context(avctx, q, &param);
Mark Thompson1f26a232016-10-21 17:57:121048 if (ret < 0)
1049 goto reinit_fail;
Zhong Li00d0a4a2019-08-13 06:11:091050 q->initialized = 1;
Mark Thompson1f26a232016-10-21 17:57:121051 }
1052
1053 return qsv_decode(avctx, q, frame, got_frame, pkt);
1054
1055reinit_fail:
Zhong Li0dfcfc52019-08-13 06:11:101056 q->orig_pix_fmt = avctx->pix_fmt = AV_PIX_FMT_NONE;
Mark Thompson1f26a232016-10-21 17:57:121057 return ret;
1058}
1059
Xu Guangxin399c1f92021-01-05 02:43:401060enum LoadPlugin {
1061 LOAD_PLUGIN_NONE,
1062 LOAD_PLUGIN_HEVC_SW,
1063 LOAD_PLUGIN_HEVC_HW,
1064};
1065
1066typedef struct QSVDecContext {
1067 AVClass *class;
1068 QSVContext qsv;
1069
1070 int load_plugin;
1071
Anton Khirnov08181622022-01-06 16:16:471072 AVFifo *packet_fifo;
Xu Guangxin399c1f92021-01-05 02:43:401073
1074 AVPacket buffer_pkt;
1075} QSVDecContext;
1076
1077static void qsv_clear_buffers(QSVDecContext *s)
1078{
1079 AVPacket pkt;
Anton Khirnov08181622022-01-06 16:16:471080 while (av_fifo_read(s->packet_fifo, &pkt, 1) >= 0)
Xu Guangxin399c1f92021-01-05 02:43:401081 av_packet_unref(&pkt);
Xu Guangxin399c1f92021-01-05 02:43:401082
1083 av_packet_unref(&s->buffer_pkt);
1084}
1085
1086static av_cold int qsv_decode_close(AVCodecContext *avctx)
1087{
1088 QSVDecContext *s = avctx->priv_data;
1089
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411090 qsv_decode_close_qsvcontext(&s->qsv);
Xu Guangxin399c1f92021-01-05 02:43:401091
1092 qsv_clear_buffers(s);
1093
Anton Khirnov08181622022-01-06 16:16:471094 av_fifo_freep2(&s->packet_fifo);
Xu Guangxin399c1f92021-01-05 02:43:401095
1096 return 0;
1097}
1098
1099static av_cold int qsv_decode_init(AVCodecContext *avctx)
1100{
1101 QSVDecContext *s = avctx->priv_data;
1102 int ret;
Xu Guangxin7a5a5e62021-01-05 02:43:421103 const char *uid = NULL;
Xu Guangxin399c1f92021-01-05 02:43:401104
Xu Guangxind78ecf12021-01-05 02:43:411105 if (avctx->codec_id == AV_CODEC_ID_VP8) {
Xu Guangxin7a5a5e62021-01-05 02:43:421106 uid = "f622394d8d87452f878c51f2fc9b4131";
Xu Guangxind78ecf12021-01-05 02:43:411107 } else if (avctx->codec_id == AV_CODEC_ID_VP9) {
Xu Guangxin7a5a5e62021-01-05 02:43:421108 uid = "a922394d8d87452f878c51f2fc9b4131";
Xu Guangxind78ecf12021-01-05 02:43:411109 }
1110 else if (avctx->codec_id == AV_CODEC_ID_HEVC && s->load_plugin != LOAD_PLUGIN_NONE) {
Xu Guangxin399c1f92021-01-05 02:43:401111 static const char * const uid_hevcdec_sw = "15dd936825ad475ea34e35f3f54217a6";
1112 static const char * const uid_hevcdec_hw = "33a61c0b4c27454ca8d85dde757c6f8e";
1113
1114 if (s->qsv.load_plugins[0]) {
1115 av_log(avctx, AV_LOG_WARNING,
1116 "load_plugins is not empty, but load_plugin is not set to 'none'."
1117 "The load_plugin value will be ignored.\n");
1118 } else {
Xu Guangxin399c1f92021-01-05 02:43:401119 if (s->load_plugin == LOAD_PLUGIN_HEVC_SW)
Xu Guangxin7a5a5e62021-01-05 02:43:421120 uid = uid_hevcdec_sw;
Xu Guangxin399c1f92021-01-05 02:43:401121 else
Xu Guangxin7a5a5e62021-01-05 02:43:421122 uid = uid_hevcdec_hw;
Xu Guangxin399c1f92021-01-05 02:43:401123 }
1124 }
Xu Guangxin7a5a5e62021-01-05 02:43:421125 if (uid) {
1126 av_freep(&s->qsv.load_plugins);
1127 s->qsv.load_plugins = av_strdup(uid);
1128 if (!s->qsv.load_plugins)
1129 return AVERROR(ENOMEM);
1130 }
Xu Guangxin399c1f92021-01-05 02:43:401131
1132 s->qsv.orig_pix_fmt = AV_PIX_FMT_NV12;
Anton Khirnov08181622022-01-06 16:16:471133 s->packet_fifo = av_fifo_alloc2(1, sizeof(AVPacket),
1134 AV_FIFO_FLAG_AUTO_GROW);
Xu Guangxin399c1f92021-01-05 02:43:401135 if (!s->packet_fifo) {
1136 ret = AVERROR(ENOMEM);
1137 goto fail;
1138 }
1139
Haihao Xiang971b4ac2021-06-11 02:19:451140 if (!avctx->pkt_timebase.num)
1141 av_log(avctx, AV_LOG_WARNING, "Invalid pkt_timebase, passing timestamps as-is.\n");
1142
Xu Guangxin399c1f92021-01-05 02:43:401143 return 0;
1144fail:
1145 qsv_decode_close(avctx);
1146 return ret;
1147}
1148
Andreas Rheinhardtce7dbd02022-03-30 19:33:241149static int qsv_decode_frame(AVCodecContext *avctx, AVFrame *frame,
Xu Guangxin399c1f92021-01-05 02:43:401150 int *got_frame, AVPacket *avpkt)
1151{
1152 QSVDecContext *s = avctx->priv_data;
Xu Guangxin399c1f92021-01-05 02:43:401153 int ret;
1154
1155 /* buffer the input packet */
1156 if (avpkt->size) {
1157 AVPacket input_ref;
1158
Xu Guangxin399c1f92021-01-05 02:43:401159 ret = av_packet_ref(&input_ref, avpkt);
1160 if (ret < 0)
1161 return ret;
Anton Khirnov08181622022-01-06 16:16:471162 av_fifo_write(s->packet_fifo, &input_ref, 1);
Xu Guangxin399c1f92021-01-05 02:43:401163 }
1164
1165 /* process buffered data */
1166 while (!*got_frame) {
1167 /* prepare the input data */
1168 if (s->buffer_pkt.size <= 0) {
1169 /* no more data */
Anton Khirnov08181622022-01-06 16:16:471170 if (!av_fifo_can_read(s->packet_fifo))
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411171 return avpkt->size ? avpkt->size : qsv_process_data(avctx, &s->qsv, frame, got_frame, avpkt);
Xu Guangxin399c1f92021-01-05 02:43:401172 /* in progress of reinit, no read from fifo and keep the buffer_pkt */
1173 if (!s->qsv.reinit_flag) {
1174 av_packet_unref(&s->buffer_pkt);
Anton Khirnov08181622022-01-06 16:16:471175 av_fifo_read(s->packet_fifo, &s->buffer_pkt, 1);
Xu Guangxin399c1f92021-01-05 02:43:401176 }
1177 }
1178
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411179 ret = qsv_process_data(avctx, &s->qsv, frame, got_frame, &s->buffer_pkt);
Xu Guangxin399c1f92021-01-05 02:43:401180 if (ret < 0){
Haihao Xiange233f3e2023-11-27 06:01:501181 if (ret == AVERROR(EAGAIN))
1182 ret = 0;
1183
Xu Guangxin399c1f92021-01-05 02:43:401184 /* Drop buffer_pkt when failed to decode the packet. Otherwise,
1185 the decoder will keep decoding the failure packet. */
1186 av_packet_unref(&s->buffer_pkt);
1187 return ret;
1188 }
1189 if (s->qsv.reinit_flag)
1190 continue;
1191
1192 s->buffer_pkt.size -= ret;
1193 s->buffer_pkt.data += ret;
1194 }
1195
1196 return avpkt->size;
1197}
1198
1199static void qsv_decode_flush(AVCodecContext *avctx)
1200{
1201 QSVDecContext *s = avctx->priv_data;
1202
1203 qsv_clear_buffers(s);
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411204
1205 s->qsv.orig_pix_fmt = AV_PIX_FMT_NONE;
1206 s->qsv.initialized = 0;
Xu Guangxin399c1f92021-01-05 02:43:401207}
1208
1209#define OFFSET(x) offsetof(QSVDecContext, x)
1210#define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
1211
1212#define DEFINE_QSV_DECODER_WITH_OPTION(x, X, bsf_name, opt) \
1213static const AVClass x##_qsv_class = { \
1214 .class_name = #x "_qsv", \
Anton Khirnov08bebeb2024-01-19 12:33:281215 .item_name = av_default_item_name, \
Xu Guangxin399c1f92021-01-05 02:43:401216 .option = opt, \
1217 .version = LIBAVUTIL_VERSION_INT, \
1218}; \
Andreas Rheinhardt20f97272022-03-16 20:09:541219const FFCodec ff_##x##_qsv_decoder = { \
1220 .p.name = #x "_qsv", \
Andreas Rheinhardt48286d42022-08-29 11:38:021221 CODEC_LONG_NAME(#X " video (Intel Quick Sync Video acceleration)"), \
Xu Guangxin399c1f92021-01-05 02:43:401222 .priv_data_size = sizeof(QSVDecContext), \
Andreas Rheinhardt20f97272022-03-16 20:09:541223 .p.type = AVMEDIA_TYPE_VIDEO, \
1224 .p.id = AV_CODEC_ID_##X, \
Xu Guangxin399c1f92021-01-05 02:43:401225 .init = qsv_decode_init, \
Andreas Rheinhardt4243da42022-03-30 21:28:241226 FF_CODEC_DECODE_CB(qsv_decode_frame), \
Xu Guangxin399c1f92021-01-05 02:43:401227 .flush = qsv_decode_flush, \
1228 .close = qsv_decode_close, \
1229 .bsfs = bsf_name, \
Andreas Rheinhardt20f97272022-03-16 20:09:541230 .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_DR1 | AV_CODEC_CAP_AVOID_PROBING | AV_CODEC_CAP_HYBRID, \
1231 .p.priv_class = &x##_qsv_class, \
Andreas Rheinhardtf0dc8fa2021-03-04 15:08:411232 .hw_configs = qsv_hw_configs, \
Andreas Rheinhardt20f97272022-03-16 20:09:541233 .p.wrapper_name = "qsv", \
Fei Wang1c562632024-05-20 02:05:521234 .caps_internal = FF_CODEC_CAP_NOT_INIT_THREADSAFE | FF_CODEC_CAP_EXPORTS_CROPPING, \
Xu Guangxin399c1f92021-01-05 02:43:401235}; \
1236
1237#define DEFINE_QSV_DECODER(x, X, bsf_name) DEFINE_QSV_DECODER_WITH_OPTION(x, X, bsf_name, options)
1238
1239#if CONFIG_HEVC_QSV_DECODER
1240static const AVOption hevc_options[] = {
1241 { "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 },
1242
Anton Khirnov1e7d2002024-02-11 14:41:051243 { "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" },
1244 { "none", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = LOAD_PLUGIN_NONE }, 0, 0, VD, .unit = "load_plugin" },
1245 { "hevc_sw", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = LOAD_PLUGIN_HEVC_SW }, 0, 0, VD, .unit = "load_plugin" },
1246 { "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:401247
1248 { "load_plugins", "A :-separate list of hexadecimal plugin UIDs to load in an internal session",
1249 OFFSET(qsv.load_plugins), AV_OPT_TYPE_STRING, { .str = "" }, 0, 0, VD },
1250
Anton Khirnov1e7d2002024-02-11 14:41:051251 { "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"},
1252 { "default", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_DEFAULT }, 0, 0, VD, .unit = "gpu_copy"},
1253 { "on", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_ON }, 0, 0, VD, .unit = "gpu_copy"},
1254 { "off", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_OFF }, 0, 0, VD, .unit = "gpu_copy"},
Xu Guangxin399c1f92021-01-05 02:43:401255 { NULL },
1256};
1257DEFINE_QSV_DECODER_WITH_OPTION(hevc, HEVC, "hevc_mp4toannexb", hevc_options)
1258#endif
1259
1260static const AVOption options[] = {
1261 { "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 },
1262
Anton Khirnov1e7d2002024-02-11 14:41:051263 { "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"},
1264 { "default", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_DEFAULT }, 0, 0, VD, .unit = "gpu_copy"},
1265 { "on", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_ON }, 0, 0, VD, .unit = "gpu_copy"},
1266 { "off", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = MFX_GPUCOPY_OFF }, 0, 0, VD, .unit = "gpu_copy"},
Xu Guangxin399c1f92021-01-05 02:43:401267 { NULL },
1268};
1269
1270#if CONFIG_H264_QSV_DECODER
1271DEFINE_QSV_DECODER(h264, H264, "h264_mp4toannexb")
1272#endif
Xu Guangxind78ecf12021-01-05 02:43:411273
1274#if CONFIG_MPEG2_QSV_DECODER
1275DEFINE_QSV_DECODER(mpeg2, MPEG2VIDEO, NULL)
1276#endif
1277
1278#if CONFIG_VC1_QSV_DECODER
1279DEFINE_QSV_DECODER(vc1, VC1, NULL)
1280#endif
1281
1282#if CONFIG_MJPEG_QSV_DECODER
1283DEFINE_QSV_DECODER(mjpeg, MJPEG, NULL)
1284#endif
1285
1286#if CONFIG_VP8_QSV_DECODER
1287DEFINE_QSV_DECODER(vp8, VP8, NULL)
1288#endif
1289
1290#if CONFIG_VP9_QSV_DECODER
1291DEFINE_QSV_DECODER(vp9, VP9, NULL)
1292#endif
1293
1294#if CONFIG_AV1_QSV_DECODER
1295DEFINE_QSV_DECODER(av1, AV1, NULL)
1296#endif