lavc/qsvdec: expose frame pic_type and key_frame
[libav.git] / libavcodec / qsv_internal.h
CommitLineData
d0a63d8b
AK
1/*
2 * Intel MediaSDK QSV encoder/decoder shared code
3 *
4 * This file is part of Libav.
5 *
6 * Libav 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 * Libav 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 Libav; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21#ifndef AVCODEC_QSV_INTERNAL_H
22#define AVCODEC_QSV_INTERNAL_H
23
72b7441a
AK
24#include <mfx/mfxvideo.h>
25
26#include "libavutil/frame.h"
27
9c0bc1e9
AK
28#include "avcodec.h"
29
d0a63d8b
AK
30#define QSV_VERSION_MAJOR 1
31#define QSV_VERSION_MINOR 1
32
33#define ASYNC_DEPTH_DEFAULT 4 // internal parallelism
34
66acb76b
AK
35#define QSV_VERSION_ATLEAST(MAJOR, MINOR) \
36 (MFX_VERSION_MAJOR > (MAJOR) || \
37 MFX_VERSION_MAJOR == (MAJOR) && MFX_VERSION_MINOR >= (MINOR))
38
00aeedd8 39typedef struct QSVMid {
4ab61cd9 40 AVBufferRef *hw_frames_ref;
00aeedd8 41 mfxHDL handle;
4ab61cd9
AK
42
43 AVFrame *locked_frame;
44 AVFrame *hw_frame;
45 mfxFrameSurface1 surf;
00aeedd8
AK
46} QSVMid;
47
72b7441a
AK
48typedef struct QSVFrame {
49 AVFrame *frame;
404e5147 50 mfxFrameSurface1 surface;
52ed83fa
ZL
51 mfxExtDecodedFrameInfo dec_info;
52 mfxExtBuffer *ext_param;
72b7441a 53
f5c4d38c 54 int queued;
404e5147 55 int used;
f5c4d38c 56
72b7441a
AK
57 struct QSVFrame *next;
58} QSVFrame;
59
a0524d9b
AK
60typedef struct QSVFramesContext {
61 AVBufferRef *hw_frames_ctx;
4ab61cd9
AK
62 void *logctx;
63
64 /* The memory ids for the external frames.
65 * Refcounted, since we need one reference owned by the QSVFramesContext
66 * (i.e. by the encoder/decoder) and another one given to the MFX session
67 * from the frame allocator. */
68 AVBufferRef *mids_buf;
00aeedd8
AK
69 QSVMid *mids;
70 int nb_mids;
a0524d9b
AK
71} QSVFramesContext;
72
d0a63d8b
AK
73/**
74 * Convert a libmfx error code into a libav error code.
75 */
95414eb2
AK
76int ff_qsv_map_error(mfxStatus mfx_err, const char **desc);
77
78int ff_qsv_print_error(void *log_ctx, mfxStatus err,
79 const char *error_string);
d0a63d8b 80
8e07c22e
AK
81int ff_qsv_print_warning(void *log_ctx, mfxStatus err,
82 const char *warning_string);
83
d0a63d8b 84int ff_qsv_codec_id_to_mfx(enum AVCodecID codec_id);
cd1047f3 85int ff_qsv_profile_to_mfx(enum AVCodecID codec_id, int profile);
d0a63d8b 86
536bb17e 87int ff_qsv_map_pixfmt(enum AVPixelFormat format, uint32_t *fourcc);
52ed83fa 88enum AVPictureType ff_qsv_map_pictype(int mfx_pic_type);
536bb17e 89
66acb76b
AK
90int ff_qsv_init_internal_session(AVCodecContext *avctx, mfxSession *session,
91 const char *load_plugins);
d0a63d8b 92
4936a48b
MT
93int ff_qsv_init_session_device(AVCodecContext *avctx, mfxSession *psession,
94 AVBufferRef *device_ref, const char *load_plugins);
95
96int ff_qsv_init_session_frames(AVCodecContext *avctx, mfxSession *session,
97 QSVFramesContext *qsv_frames_ctx,
98 const char *load_plugins, int opaque);
a0524d9b 99
00aeedd8
AK
100int ff_qsv_find_surface_idx(QSVFramesContext *ctx, QSVFrame *frame);
101
d0a63d8b 102#endif /* AVCODEC_QSV_INTERNAL_H */