Merge commit '737d35e33408263c04d7730f5487eed0d04938ba'
[ffmpeg.git] / libavcodec / vdpau_internal.h
1 /*
2  * Video Decode and Presentation API for UNIX (VDPAU) is used for
3  * HW decode acceleration for MPEG-1/2, H.264 and VC-1.
4  *
5  * Copyright (C) 2008 NVIDIA
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_VDPAU_INTERNAL_H
25 #define AVCODEC_VDPAU_INTERNAL_H
26
27 #include "config.h"
28 #include <stdint.h>
29 #if CONFIG_VDPAU
30 #include <vdpau/vdpau.h>
31 #endif
32 #include "h264.h"
33
34 #include "avcodec.h"
35 #include "mpeg4video.h"
36 #include "mpegvideo.h"
37 #include "version.h"
38
39 /** Extract VdpVideoSurface from an AVFrame */
40 static inline uintptr_t ff_vdpau_get_surface_id(AVFrame *pic)
41 {
42     return (uintptr_t)pic->data[3];
43 }
44
45 struct vdpau_picture_context;
46 #if CONFIG_VDPAU
47 union VDPAUPictureInfo {
48     VdpPictureInfoH264        h264;
49     VdpPictureInfoMPEG1Or2    mpeg;
50     VdpPictureInfoVC1          vc1;
51     VdpPictureInfoMPEG4Part2 mpeg4;
52 #ifdef VDP_DECODER_PROFILE_H264_HIGH_444_PREDICTIVE
53     VdpPictureInfoH264Predictive h264_predictive;
54 #endif
55 };
56
57 #include "vdpau.h"
58
59 typedef struct VDPAUHWContext {
60     AVVDPAUContext context;
61     VdpDevice device;
62     VdpGetProcAddress *get_proc_address;
63     char reset;
64     unsigned char flags;
65 } VDPAUHWContext;
66
67 typedef struct VDPAUContext {
68     /**
69      * VDPAU device handle
70      */
71     VdpDevice device;
72
73     /**
74      * VDPAU decoder handle
75      */
76     VdpDecoder decoder;
77
78     /**
79      * VDPAU device driver
80      */
81     VdpGetProcAddress *get_proc_address;
82
83     /**
84      * VDPAU decoder render callback
85      */
86     VdpDecoderRender *render;
87
88     uint32_t width;
89     uint32_t height;
90 } VDPAUContext;
91
92 struct vdpau_picture_context {
93     /**
94      * VDPAU picture information.
95      */
96     union VDPAUPictureInfo info;
97
98     /**
99      * Allocated size of the bitstream_buffers table.
100      */
101     int bitstream_buffers_allocated;
102
103     /**
104      * Useful bitstream buffers in the bitstream buffers table.
105      */
106     int bitstream_buffers_used;
107
108    /**
109      * Table of bitstream buffers.
110      */
111     VdpBitstreamBuffer *bitstream_buffers;
112 };
113
114 int ff_vdpau_common_init(AVCodecContext *avctx, VdpDecoderProfile profile,
115                          int level);
116 #endif //CONFIG_VDPAU
117
118 int ff_vdpau_common_uninit(AVCodecContext *avctx);
119
120 int ff_vdpau_common_start_frame(struct vdpau_picture_context *pic,
121                                 const uint8_t *buffer, uint32_t size);
122 int ff_vdpau_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
123                               struct vdpau_picture_context *pic);
124 int ff_vdpau_mpeg_end_frame(AVCodecContext *avctx);
125 int ff_vdpau_add_buffer(struct vdpau_picture_context *pic, const uint8_t *buf,
126                         uint32_t buf_size);
127
128
129 void ff_vdpau_add_data_chunk(uint8_t *data, const uint8_t *buf,
130                              int buf_size);
131
132 void ff_vdpau_mpeg_picture_complete(MpegEncContext *s, const uint8_t *buf,
133                                     int buf_size, int slice_count);
134
135 void ff_vdpau_h264_picture_start(H264Context *h);
136 void ff_vdpau_h264_set_reference_frames(H264Context *h);
137 void ff_vdpau_h264_picture_complete(H264Context *h);
138
139 void ff_vdpau_vc1_decode_picture(MpegEncContext *s, const uint8_t *buf,
140                                  int buf_size);
141
142 void ff_vdpau_mpeg4_decode_picture(Mpeg4DecContext *s, const uint8_t *buf,
143                                    int buf_size);
144
145 #endif /* AVCODEC_VDPAU_INTERNAL_H */