1 /**************************************************************************
2  *
3  * Copyright 2013 Advanced Micro Devices, Inc.
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the
8  * "Software"), to deal in the Software without restriction, including
9  * without limitation the rights to use, copy, modify, merge, publish,
10  * distribute, sub license, and/or sell copies of the Software, and to
11  * permit persons to whom the Software is furnished to do so, subject to
12  * the following conditions:
13  *
14  * The above copyright notice and this permission notice (including the
15  * next paragraph) shall be included in all copies or substantial portions
16  * of the Software.
17  *
18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21  * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25  *
26  **************************************************************************/
27 
28 /*
29  * Authors:
30  *      Christian König <christian.koenig@amd.com>
31  *
32  */
33 
34 #ifndef OMX_VID_DEC_H
35 #define OMX_VID_DEC_H
36 
37 #include <X11/Xlib.h>
38 
39 #include <string.h>
40 
41 #include <OMX_Types.h>
42 #include <OMX_Component.h>
43 #include <OMX_Core.h>
44 
45 #include <bellagio/st_static_component_loader.h>
46 #include <bellagio/omx_base_filter.h>
47 #include <bellagio/omx_base_video_port.h>
48 
49 #include "pipe/p_video_state.h"
50 #include "state_tracker/drm_driver.h"
51 #include "os/os_thread.h"
52 #include "util/list.h"
53 
54 #include "vl/vl_compositor.h"
55 
56 #define OMX_VID_DEC_BASE_NAME "OMX.mesa.video_decoder"
57 
58 #define OMX_VID_DEC_MPEG2_NAME "OMX.mesa.video_decoder.mpeg2"
59 #define OMX_VID_DEC_MPEG2_ROLE "video_decoder.mpeg2"
60 
61 #define OMX_VID_DEC_AVC_NAME "OMX.mesa.video_decoder.avc"
62 #define OMX_VID_DEC_AVC_ROLE "video_decoder.avc"
63 
64 #define OMX_VID_DEC_HEVC_NAME "OMX.mesa.video_decoder.hevc"
65 #define OMX_VID_DEC_HEVC_ROLE "video_decoder.hevc"
66 
67 #define OMX_VID_DEC_TIMESTAMP_INVALID ((OMX_TICKS) -1)
68 
69 struct vl_vlc;
70 
71 DERIVEDCLASS(vid_dec_PrivateType, omx_base_filter_PrivateType)
72 #define vid_dec_PrivateType_FIELDS omx_base_filter_PrivateType_FIELDS \
73    enum pipe_video_profile profile; \
74    struct vl_screen *screen; \
75    struct pipe_context *pipe; \
76    struct pipe_video_codec *codec; \
77    void (*Decode)(vid_dec_PrivateType *priv, struct vl_vlc *vlc, unsigned min_bits_left); \
78    void (*EndFrame)(vid_dec_PrivateType *priv); \
79    struct pipe_video_buffer *(*Flush)(vid_dec_PrivateType *priv, OMX_TICKS *timestamp); \
80    struct pipe_video_buffer *target, *shadow; \
81    union { \
82       struct { \
83          uint8_t intra_matrix[64]; \
84          uint8_t non_intra_matrix[64]; \
85       } mpeg12; \
86       struct { \
87          unsigned nal_ref_idc; \
88          bool IdrPicFlag; \
89          unsigned idr_pic_id; \
90          unsigned pic_order_cnt_lsb; \
91          unsigned pic_order_cnt_msb; \
92          unsigned delta_pic_order_cnt_bottom; \
93          unsigned delta_pic_order_cnt[2]; \
94          unsigned prevFrameNumOffset; \
95          struct pipe_h264_sps sps[32]; \
96          struct pipe_h264_pps pps[256]; \
97          struct list_head dpb_list; \
98          unsigned dpb_num; \
99       } h264; \
100       struct { \
101          unsigned temporal_id; \
102          unsigned level_idc; \
103          unsigned pic_width_in_luma_samples; \
104          unsigned pic_height_in_luma_samples; \
105          bool IdrPicFlag; \
106          int slice_prev_poc; \
107          void *ref_pic_set_list; \
108          void *rps; \
109          struct pipe_h265_sps sps[16]; \
110          struct pipe_h265_pps pps[64]; \
111          struct list_head dpb_list; \
112          unsigned dpb_num; \
113       } h265; \
114    } codec_data; \
115    union { \
116       struct pipe_picture_desc base; \
117       struct pipe_mpeg12_picture_desc mpeg12; \
118       struct pipe_h264_picture_desc h264; \
119       struct pipe_h265_picture_desc h265; \
120    } picture; \
121    unsigned num_in_buffers; \
122    OMX_BUFFERHEADERTYPE *in_buffers[2]; \
123    const void *inputs[2]; \
124    unsigned sizes[2]; \
125    OMX_TICKS timestamps[2]; \
126    OMX_TICKS timestamp; \
127    bool first_buf_in_frame; \
128    bool frame_finished; \
129    bool frame_started; \
130    unsigned bytes_left; \
131    const void *slice; \
132    bool disable_tunnel; \
133    struct vl_compositor compositor; \
134    struct vl_compositor_state cstate;
135 ENDCLASS(vid_dec_PrivateType)
136 
137 OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp);
138 
139 /* used by MPEG12 and H264 implementation */
140 void vid_dec_NeedTarget(vid_dec_PrivateType *priv);
141 
142 /* vid_dec_mpeg12.c */
143 void vid_dec_mpeg12_Init(vid_dec_PrivateType *priv);
144 
145 /* vid_dec_h264.c */
146 void vid_dec_h264_Init(vid_dec_PrivateType *priv);
147 
148 /* vid_dec_h265.c */
149 void vid_dec_h265_Init(vid_dec_PrivateType *priv);
150 
151 #endif
152