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 <stdint.h>
28#include <vdpau/vdpau.h>
29
30#include "libavutil/frame.h"
31#include "libavutil/hwcontext.h"
32#include "libavutil/hwcontext_vdpau.h"
33
34#include "avcodec.h"
35#include "vdpau.h"
36
37/** Extract VdpVideoSurface from an AVFrame */
38static inline uintptr_t ff_vdpau_get_surface_id(AVFrame *pic)
39{
40    return (uintptr_t)pic->data[3];
41}
42
43union VDPAUPictureInfo {
44    VdpPictureInfoH264        h264;
45    VdpPictureInfoMPEG1Or2    mpeg;
46    VdpPictureInfoVC1          vc1;
47    VdpPictureInfoMPEG4Part2 mpeg4;
48#ifdef VDP_DECODER_PROFILE_H264_HIGH_444_PREDICTIVE
49    VdpPictureInfoH264Predictive h264_predictive;
50#endif
51#ifdef VDP_DECODER_PROFILE_HEVC_MAIN
52    VdpPictureInfoHEVC        hevc;
53#endif
54#ifdef VDP_YCBCR_FORMAT_Y_U_V_444
55    VdpPictureInfoHEVC444     hevc_444;
56#endif
57#ifdef VDP_DECODER_PROFILE_VP9_PROFILE_0
58    VdpPictureInfoVP9        vp9;
59#endif
60#ifdef VDP_DECODER_PROFILE_AV1_MAIN
61    VdpPictureInfoAV1        av1;
62#endif
63};
64
65typedef struct VDPAUHWContext {
66    AVVDPAUContext context;
67    VdpDevice device;
68    VdpGetProcAddress *get_proc_address;
69    char reset;
70    unsigned char flags;
71} VDPAUHWContext;
72
73typedef struct VDPAUContext {
74    /**
75     * VDPAU device handle
76     */
77    VdpDevice device;
78
79    /**
80     * VDPAU decoder handle
81     */
82    VdpDecoder decoder;
83
84    /**
85     * VDPAU device driver
86     */
87    VdpGetProcAddress *get_proc_address;
88
89    /**
90     * VDPAU decoder render callback
91     */
92    VdpDecoderRender *render;
93
94    uint32_t width;
95    uint32_t height;
96} VDPAUContext;
97
98struct vdpau_picture_context {
99    /**
100     * VDPAU picture information.
101     */
102    union VDPAUPictureInfo info;
103
104    /**
105     * Allocated size of the bitstream_buffers table.
106     */
107    int bitstream_buffers_allocated;
108
109    /**
110     * Useful bitstream buffers in the bitstream buffers table.
111     */
112    int bitstream_buffers_used;
113
114   /**
115     * Table of bitstream buffers.
116     */
117    VdpBitstreamBuffer *bitstream_buffers;
118};
119
120int ff_vdpau_common_init(AVCodecContext *avctx, VdpDecoderProfile profile,
121                         int level);
122int ff_vdpau_common_uninit(AVCodecContext *avctx);
123
124int ff_vdpau_common_start_frame(struct vdpau_picture_context *pic,
125                                const uint8_t *buffer, uint32_t size);
126int ff_vdpau_common_end_frame(AVCodecContext *avctx, AVFrame *frame,
127                              struct vdpau_picture_context *pic);
128int ff_vdpau_mpeg_end_frame(AVCodecContext *avctx);
129int ff_vdpau_add_buffer(struct vdpau_picture_context *pic, const uint8_t *buf,
130                        uint32_t buf_size);
131int ff_vdpau_common_frame_params(AVCodecContext *avctx,
132                                 AVBufferRef *hw_frames_ctx);
133
134#endif /* AVCODEC_VDPAU_INTERNAL_H */
135