2 * Intel MediaSDK QSV based MPEG-2 and VC-1 decoders
4 * copyright (c) 2015 Anton Khirnov
6 * This file is part of Libav.
8 * Libav is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * Libav is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with Libav; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27 #include <mfx/mfxvideo.h>
29 #include "libavutil/common.h"
30 #include "libavutil/fifo.h"
31 #include "libavutil/opt.h"
35 #include "qsv_internal.h"
39 typedef struct QSVOtherContext
{
43 AVFifoBuffer
*packet_fifo
;
48 static void qsv_clear_buffers(QSVOtherContext
*s
)
51 while (av_fifo_size(s
->packet_fifo
) >= sizeof(pkt
)) {
52 av_fifo_generic_read(s
->packet_fifo
, &pkt
, sizeof(pkt
), NULL
);
53 av_packet_unref(&pkt
);
56 av_packet_unref(&s
->input_ref
);
59 static av_cold
int qsv_decode_close(AVCodecContext
*avctx
)
61 QSVOtherContext
*s
= avctx
->priv_data
;
63 ff_qsv_decode_close(&s
->qsv
);
67 av_fifo_free(s
->packet_fifo
);
72 static av_cold
int qsv_decode_init(AVCodecContext
*avctx
)
74 QSVOtherContext
*s
= avctx
->priv_data
;
77 s
->packet_fifo
= av_fifo_alloc(sizeof(AVPacket
));
78 if (!s
->packet_fifo
) {
79 ret
= AVERROR(ENOMEM
);
85 qsv_decode_close(avctx
);
89 static int qsv_decode_frame(AVCodecContext
*avctx
, void *data
,
90 int *got_frame
, AVPacket
*avpkt
)
92 QSVOtherContext
*s
= avctx
->priv_data
;
93 AVFrame
*frame
= data
;
96 /* buffer the input packet */
98 AVPacket input_ref
= { 0 };
100 if (av_fifo_space(s
->packet_fifo
) < sizeof(input_ref
)) {
101 ret
= av_fifo_realloc2(s
->packet_fifo
,
102 av_fifo_size(s
->packet_fifo
) + sizeof(input_ref
));
107 ret
= av_packet_ref(&input_ref
, avpkt
);
110 av_fifo_generic_write(s
->packet_fifo
, &input_ref
, sizeof(input_ref
), NULL
);
113 /* process buffered data */
114 while (!*got_frame
) {
115 if (s
->input_ref
.size
<= 0) {
117 if (av_fifo_size(s
->packet_fifo
) < sizeof(AVPacket
))
118 return avpkt
->size ? avpkt
->size
: ff_qsv_process_data(avctx
, &s
->qsv
, frame
, got_frame
, avpkt
);
120 av_packet_unref(&s
->input_ref
);
121 av_fifo_generic_read(s
->packet_fifo
, &s
->input_ref
, sizeof(s
->input_ref
), NULL
);
124 ret
= ff_qsv_process_data(avctx
, &s
->qsv
, frame
, got_frame
, &s
->input_ref
);
128 s
->input_ref
.size
-= ret
;
129 s
->input_ref
.data
+= ret
;
135 static void qsv_decode_flush(AVCodecContext
*avctx
)
137 QSVOtherContext
*s
= avctx
->priv_data
;
139 qsv_clear_buffers(s
);
140 ff_qsv_decode_flush(avctx
, &s
->qsv
);
143 #define OFFSET(x) offsetof(QSVOtherContext, x)
144 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
145 static const AVOption options
[] = {
146 { "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
}, 0, INT_MAX
, VD
},
150 #if CONFIG_MPEG2_QSV_HWACCEL
151 AVHWAccel ff_mpeg2_qsv_hwaccel
= {
153 .type
= AVMEDIA_TYPE_VIDEO
,
154 .id
= AV_CODEC_ID_MPEG2VIDEO
,
155 .pix_fmt
= AV_PIX_FMT_QSV
,
159 #if CONFIG_MPEG2_QSV_DECODER
160 static const AVClass mpeg2_qsv_class
= {
161 .class_name
= "mpeg2_qsv",
162 .item_name
= av_default_item_name
,
164 .version
= LIBAVUTIL_VERSION_INT
,
167 AVCodec ff_mpeg2_qsv_decoder
= {
169 .long_name
= NULL_IF_CONFIG_SMALL("MPEG-2 video (Intel Quick Sync Video acceleration)"),
170 .priv_data_size
= sizeof(QSVOtherContext
),
171 .type
= AVMEDIA_TYPE_VIDEO
,
172 .id
= AV_CODEC_ID_MPEG2VIDEO
,
173 .init
= qsv_decode_init
,
174 .decode
= qsv_decode_frame
,
175 .flush
= qsv_decode_flush
,
176 .close
= qsv_decode_close
,
177 .capabilities
= AV_CODEC_CAP_DELAY
| AV_CODEC_CAP_DR1
,
178 .priv_class
= &mpeg2_qsv_class
,
179 .pix_fmts
= (const enum AVPixelFormat
[]){ AV_PIX_FMT_NV12
,
185 #if CONFIG_VC1_QSV_HWACCEL
186 AVHWAccel ff_vc1_qsv_hwaccel
= {
188 .type
= AVMEDIA_TYPE_VIDEO
,
189 .id
= AV_CODEC_ID_VC1
,
190 .pix_fmt
= AV_PIX_FMT_QSV
,
194 #if CONFIG_VC1_QSV_DECODER
195 static const AVClass vc1_qsv_class
= {
196 .class_name
= "vc1_qsv",
197 .item_name
= av_default_item_name
,
199 .version
= LIBAVUTIL_VERSION_INT
,
202 AVCodec ff_vc1_qsv_decoder
= {
204 .long_name
= NULL_IF_CONFIG_SMALL("VC-1 video (Intel Quick Sync Video acceleration)"),
205 .priv_data_size
= sizeof(QSVOtherContext
),
206 .type
= AVMEDIA_TYPE_VIDEO
,
207 .id
= AV_CODEC_ID_VC1
,
208 .init
= qsv_decode_init
,
209 .decode
= qsv_decode_frame
,
210 .flush
= qsv_decode_flush
,
211 .close
= qsv_decode_close
,
212 .capabilities
= AV_CODEC_CAP_DELAY
| AV_CODEC_CAP_DR1
,
213 .priv_class
= &vc1_qsv_class
,
214 .pix_fmts
= (const enum AVPixelFormat
[]){ AV_PIX_FMT_NV12
,