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 <string.h> 38 39 #include <OMX_Types.h> 40 #include <OMX_Component.h> 41 #include <OMX_Core.h> 42 43 #include <bellagio/st_static_component_loader.h> 44 #include <bellagio/omx_base_filter.h> 45 #include <bellagio/omx_base_video_port.h> 46 47 #include "pipe/p_video_state.h" 48 #include "os/os_thread.h" 49 #include "util/list.h" 50 51 #include "vl/vl_compositor.h" 52 53 #define OMX_VID_DEC_BASE_NAME "OMX.mesa.video_decoder" 54 55 #define OMX_VID_DEC_MPEG2_NAME "OMX.mesa.video_decoder.mpeg2" 56 #define OMX_VID_DEC_MPEG2_ROLE "video_decoder.mpeg2" 57 58 #define OMX_VID_DEC_AVC_NAME "OMX.mesa.video_decoder.avc" 59 #define OMX_VID_DEC_AVC_ROLE "video_decoder.avc" 60 61 #define OMX_VID_DEC_HEVC_NAME "OMX.mesa.video_decoder.hevc" 62 #define OMX_VID_DEC_HEVC_ROLE "video_decoder.hevc" 63 64 #define OMX_VID_DEC_TIMESTAMP_INVALID ((OMX_TICKS) -1) 65 66 struct vl_vlc; 67 68 DERIVEDCLASS(vid_dec_PrivateType, omx_base_filter_PrivateType) 69 #define vid_dec_PrivateType_FIELDS omx_base_filter_PrivateType_FIELDS \ 70 enum pipe_video_profile profile; \ 71 struct vl_screen *screen; \ 72 struct pipe_context *pipe; \ 73 struct pipe_video_codec *codec; \ 74 void (*Decode)(vid_dec_PrivateType *priv, struct vl_vlc *vlc, unsigned min_bits_left); \ 75 void (*EndFrame)(vid_dec_PrivateType *priv); \ 76 struct pipe_video_buffer *(*Flush)(vid_dec_PrivateType *priv, OMX_TICKS *timestamp); \ 77 struct pipe_video_buffer *target, *shadow; \ 78 union { \ 79 struct { \ 80 uint8_t intra_matrix[64]; \ 81 uint8_t non_intra_matrix[64]; \ 82 } mpeg12; \ 83 struct { \ 84 unsigned nal_ref_idc; \ 85 bool IdrPicFlag; \ 86 unsigned idr_pic_id; \ 87 unsigned pic_order_cnt_lsb; \ 88 unsigned pic_order_cnt_msb; \ 89 unsigned delta_pic_order_cnt_bottom; \ 90 unsigned delta_pic_order_cnt[2]; \ 91 unsigned prevFrameNumOffset; \ 92 struct pipe_h264_sps sps[32]; \ 93 struct pipe_h264_pps pps[256]; \ 94 struct list_head dpb_list; \ 95 unsigned dpb_num; \ 96 } h264; \ 97 struct { \ 98 unsigned temporal_id; \ 99 unsigned level_idc; \ 100 unsigned pic_width_in_luma_samples; \ 101 unsigned pic_height_in_luma_samples; \ 102 bool IdrPicFlag; \ 103 int slice_prev_poc; \ 104 void *ref_pic_set_list; \ 105 void *rps; \ 106 struct pipe_h265_sps sps[16]; \ 107 struct pipe_h265_pps pps[64]; \ 108 struct list_head dpb_list; \ 109 unsigned dpb_num; \ 110 } h265; \ 111 } codec_data; \ 112 union { \ 113 struct pipe_picture_desc base; \ 114 struct pipe_mpeg12_picture_desc mpeg12; \ 115 struct pipe_h264_picture_desc h264; \ 116 struct pipe_h265_picture_desc h265; \ 117 } picture; \ 118 unsigned num_in_buffers; \ 119 OMX_BUFFERHEADERTYPE *in_buffers[2]; \ 120 const void *inputs[2]; \ 121 unsigned sizes[2]; \ 122 OMX_TICKS timestamps[2]; \ 123 OMX_TICKS timestamp; \ 124 bool first_buf_in_frame; \ 125 bool frame_finished; \ 126 bool frame_started; \ 127 unsigned bytes_left; \ 128 const void *slice; \ 129 bool disable_tunnel; \ 130 struct vl_compositor compositor; \ 131 struct vl_compositor_state cstate; 132 ENDCLASS(vid_dec_PrivateType) 133 134 OMX_ERRORTYPE vid_dec_LoaderComponent(stLoaderComponentType *comp); 135 136 /* used by MPEG12 and H264 implementation */ 137 void vid_dec_NeedTarget(vid_dec_PrivateType *priv); 138 139 /* vid_dec_mpeg12.c */ 140 void vid_dec_mpeg12_Init(vid_dec_PrivateType *priv); 141 142 /* vid_dec_h264.c */ 143 void vid_dec_h264_Init(vid_dec_PrivateType *priv); 144 145 /* vid_dec_h265.c */ 146 void vid_dec_h265_Init(vid_dec_PrivateType *priv); 147 148 #endif 149