]> git.sesse.net Git - ffmpeg/blob - libavcodec/vaapi_internal.h
avcodec/dvdsub: fix partial packet assembly
[ffmpeg] / libavcodec / vaapi_internal.h
1 /*
2  * Video Acceleration API (video decoding)
3  * HW decode acceleration for MPEG-2, MPEG-4, H.264 and VC-1
4  *
5  * Copyright (C) 2008-2009 Splitted-Desktop Systems
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * 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  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * 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
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 #ifndef AVCODEC_VAAPI_INTERNAL_H
25 #define AVCODEC_VAAPI_INTERNAL_H
26
27 #include <va/va.h>
28 #include "vaapi.h"
29 #include "avcodec.h"
30 #include "internal.h"
31
32 /**
33  * @addtogroup VAAPI_Decoding
34  *
35  * @{
36  */
37
38 typedef struct {
39     VADisplay display;                  ///< Windowing system dependent handle
40     VAConfigID config_id;               ///< Configuration ID
41     VAContextID context_id;             ///< Context ID (video decode pipeline)
42     VABufferID pic_param_buf_id;        ///< Picture parameter buffer
43     VABufferID iq_matrix_buf_id;        ///< Inverse quantiser matrix buffer
44     VABufferID bitplane_buf_id;         ///< Bitplane buffer (for VC-1 decoding)
45     VABufferID *slice_buf_ids;          ///< Slice parameter/data buffers
46     unsigned int n_slice_buf_ids;       ///< Number of effective slice buffers
47     unsigned int slice_buf_ids_alloc;   ///< Number of allocated slice buffers
48     void *slice_params;                 ///< Pointer to slice parameter buffers
49     unsigned int slice_param_size;      ///< Size of a slice parameter element
50     unsigned int slice_params_alloc;    ///< Number of allocated slice parameters
51     unsigned int slice_count;           ///< Number of slices currently filled in
52     const uint8_t *slice_data;          ///< Pointer to slice data buffer base
53     unsigned int slice_data_size;       ///< Current size of slice data
54 } FFVAContext;
55
56 /** Extract vaapi_context from an AVCodecContext */
57 static inline FFVAContext *ff_vaapi_get_context(AVCodecContext *avctx)
58 {
59     return avctx->internal->hwaccel_priv_data;
60 }
61
62 /** Extract VASurfaceID from an AVFrame */
63 static inline VASurfaceID ff_vaapi_get_surface_id(AVFrame *pic)
64 {
65     return (uintptr_t)pic->data[3];
66 }
67
68 /** Common AVHWAccel.init() implementation */
69 int ff_vaapi_context_init(AVCodecContext *avctx);
70
71 /** Common AVHWAccel.uninit() implementation */
72 int ff_vaapi_context_fini(AVCodecContext *avctx);
73
74 /** Common AVHWAccel.end_frame() implementation */
75 void ff_vaapi_common_end_frame(AVCodecContext *avctx);
76
77 /** Allocate a new picture parameter buffer */
78 void *ff_vaapi_alloc_pic_param(FFVAContext *vactx, unsigned int size);
79
80 /** Allocate a new IQ matrix buffer */
81 void *ff_vaapi_alloc_iq_matrix(FFVAContext *vactx, unsigned int size);
82
83 /** Allocate a new bit-plane buffer */
84 uint8_t *ff_vaapi_alloc_bitplane(FFVAContext *vactx, uint32_t size);
85
86 /**
87  * Allocate a new slice descriptor for the input slice.
88  *
89  * @param vactx the VA API context
90  * @param buffer the slice data buffer base
91  * @param size the size of the slice in bytes
92  * @return the newly allocated slice parameter
93  */
94 VASliceParameterBufferBase *ff_vaapi_alloc_slice(FFVAContext *vactx, const uint8_t *buffer, uint32_t size);
95
96 int ff_vaapi_mpeg_end_frame(AVCodecContext *avctx);
97 int ff_vaapi_commit_slices(FFVAContext *vactx);
98 int ff_vaapi_render_picture(FFVAContext *vactx, VASurfaceID surface);
99
100 /* @} */
101
102 #endif /* AVCODEC_VAAPI_INTERNAL_H */