Merge commit 'a6a750c7ef240b72ce01e9653343a0ddf247d196'
[ffmpeg.git] / libavcodec / qsvdec_mpeg2.c
1 /*
2  * Intel MediaSDK QSV based MPEG-2 video decoder
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <stdint.h>
22 #include <string.h>
23
24 #include "libavutil/common.h"
25 #include "libavutil/opt.h"
26
27 #include "avcodec.h"
28 #include "qsvdec.h"
29
30 typedef struct QSVMPEG2Context {
31     AVClass *class;
32     QSVContext qsv;
33 } QSVMPEG2Context;
34
35 static av_cold int qsv_decode_close(AVCodecContext *avctx)
36 {
37     QSVMPEG2Context *s = avctx->priv_data;
38
39     ff_qsv_decode_close(&s->qsv);
40
41     return 0;
42 }
43
44 static av_cold int qsv_decode_init(AVCodecContext *avctx)
45 {
46     return 0;
47 }
48
49 static int qsv_decode_frame(AVCodecContext *avctx, void *data,
50                             int *got_frame, AVPacket *avpkt)
51 {
52     QSVMPEG2Context *s = avctx->priv_data;
53     AVFrame *frame    = data;
54
55     return ff_qsv_decode(avctx, &s->qsv, frame, got_frame, avpkt);
56 }
57
58 static void qsv_decode_flush(AVCodecContext *avctx)
59 {
60     QSVMPEG2Context *s = avctx->priv_data;
61     ff_qsv_decode_reset(avctx, &s->qsv);
62 }
63
64 AVHWAccel ff_mpeg2_qsv_hwaccel = {
65     .name           = "mpeg2_qsv",
66     .type           = AVMEDIA_TYPE_VIDEO,
67     .id             = AV_CODEC_ID_MPEG2VIDEO,
68     .pix_fmt        = AV_PIX_FMT_QSV,
69 };
70
71 #define OFFSET(x) offsetof(QSVMPEG2Context, x)
72 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
73 static const AVOption options[] = {
74     { "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 },
75     { NULL },
76 };
77
78 static const AVClass class = {
79     .class_name = "mpeg2_qsv",
80     .item_name  = av_default_item_name,
81     .option     = options,
82     .version    = LIBAVUTIL_VERSION_INT,
83 };
84
85 AVCodec ff_mpeg2_qsv_decoder = {
86     .name           = "mpeg2_qsv",
87     .long_name      = NULL_IF_CONFIG_SMALL("MPEG-2 video (Intel Quick Sync Video acceleration)"),
88     .priv_data_size = sizeof(QSVMPEG2Context),
89     .type           = AVMEDIA_TYPE_VIDEO,
90     .id             = AV_CODEC_ID_MPEG2VIDEO,
91     .init           = qsv_decode_init,
92     .decode         = qsv_decode_frame,
93     .flush          = qsv_decode_flush,
94     .close          = qsv_decode_close,
95     .capabilities   = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_DR1,
96     .priv_class     = &class,
97     .pix_fmts       = (const enum AVPixelFormat[]){ AV_PIX_FMT_NV12,
98                                                     AV_PIX_FMT_QSV,
99                                                     AV_PIX_FMT_NONE },
100 };