Merge commit 'a1a143adb0fd11c474221431417cff25db7d920f'
[ffmpeg.git] / libavcodec / qsv_internal.h
1 /*
2  * Intel MediaSDK QSV encoder/decoder shared code
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 #ifndef AVCODEC_QSV_INTERNAL_H
22 #define AVCODEC_QSV_INTERNAL_H
23
24 #include <mfx/mfxvideo.h>
25
26 #include "libavutil/frame.h"
27
28 #include "avcodec.h"
29
30 #define QSV_VERSION_MAJOR 1
31 #define QSV_VERSION_MINOR 1
32
33 #define ASYNC_DEPTH_DEFAULT 4       // internal parallelism
34
35 #define QSV_MAX_ENC_PAYLOAD 2       // # of mfxEncodeCtrl payloads supported
36
37 #define QSV_VERSION_ATLEAST(MAJOR, MINOR)   \
38     (MFX_VERSION_MAJOR > (MAJOR) ||         \
39      MFX_VERSION_MAJOR == (MAJOR) && MFX_VERSION_MINOR >= (MINOR))
40
41 typedef struct QSVMid {
42     AVBufferRef *hw_frames_ref;
43     mfxHDL handle;
44
45     AVFrame *locked_frame;
46     AVFrame *hw_frame;
47     mfxFrameSurface1 surf;
48 } QSVMid;
49
50 typedef struct QSVFrame {
51     AVFrame *frame;
52     mfxFrameSurface1 surface;
53     mfxEncodeCtrl enc_ctrl;
54
55     int queued;
56     int used;
57
58     struct QSVFrame *next;
59 } QSVFrame;
60
61 typedef struct QSVFramesContext {
62     AVBufferRef *hw_frames_ctx;
63     void *logctx;
64
65     /* The memory ids for the external frames.
66      * Refcounted, since we need one reference owned by the QSVFramesContext
67      * (i.e. by the encoder/decoder) and another one given to the MFX session
68      * from the frame allocator. */
69     AVBufferRef *mids_buf;
70     QSVMid *mids;
71     int  nb_mids;
72 } QSVFramesContext;
73
74 /**
75  * Convert a libmfx error code into an ffmpeg error code.
76  */
77 int ff_qsv_map_error(mfxStatus mfx_err, const char **desc);
78
79 int ff_qsv_print_error(void *log_ctx, mfxStatus err,
80                        const char *error_string);
81
82 int ff_qsv_print_warning(void *log_ctx, mfxStatus err,
83                          const char *warning_string);
84
85 int ff_qsv_codec_id_to_mfx(enum AVCodecID codec_id);
86 int ff_qsv_profile_to_mfx(enum AVCodecID codec_id, int profile);
87
88 int ff_qsv_map_pixfmt(enum AVPixelFormat format, uint32_t *fourcc);
89
90 int ff_qsv_init_internal_session(AVCodecContext *avctx, mfxSession *session,
91                                  const char *load_plugins);
92
93 int ff_qsv_init_session_device(AVCodecContext *avctx, mfxSession *psession,
94                                AVBufferRef *device_ref, const char *load_plugins);
95
96 int ff_qsv_init_session_frames(AVCodecContext *avctx, mfxSession *session,
97                                QSVFramesContext *qsv_frames_ctx,
98                                const char *load_plugins, int opaque);
99
100 int ff_qsv_find_surface_idx(QSVFramesContext *ctx, QSVFrame *frame);
101
102 #endif /* AVCODEC_QSV_INTERNAL_H */