• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /******************************************************************************
2  *
3  * Copyright (C) 2015 The Android Open Source Project
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *****************************************************************************
18  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19 */
20 #ifndef __IMPEG2D_STRUCTS_H__
21 #define __IMPEG2D_STRUCTS_H__
22 
23 /* Decoder needs at least 4 reference buffers in order to support format conversion in a thread and
24 to support B pictures. Because of format conversion in a thread, codec delay is now 2 frames instead of 1.
25 To reduce this delay, format conversion has to wait for MB status before converting for B pictures.
26 To avoid this check the delay is increased to 2 and hence number of reference frames minimum is 4.
27 Because of temporal dependency in deinterlacer one additional buffer is also needed */
28 #define NUM_INT_FRAME_BUFFERS                     5
29 
30 
31 #define MAX_WIDTH               4096
32 #define MAX_HEIGHT              2160
33 
34 #define MIN_WIDTH               16
35 #define MIN_HEIGHT              16
36 
37 
38 #define MAX_FRM_SIZE            (MAX_WIDTH * MAX_HEIGHT * 2)  /* Supports only 420P and 422ILE */
39 
40 #define DEC_ORDER               0
41 
42 #define MAX_BITSTREAM_BUFFER_SIZE       2000 * 1024
43 #define MIN_BUFFER_BYTES_AT_EOS 8
44 
45 /* Flag to signal that buffer is held by deinterlacing */
46 #define MPEG2_BUF_MGR_DEINT (BUF_MGR_DISP << 1)
47 
48 typedef enum
49 {
50     CMD_PROCESS,
51     CMD_FMTCONV,
52 }e_jobq_cmd_t;
53 
54 /**
55  * Structure to represent a processing job entry
56  */
57 typedef struct
58 {
59     /**
60      * Command
61      * Currently: PROCESS, FMTCONV are the only two jobs
62      */
63     WORD32 i4_cmd;
64 
65     /**
66      * MB y of the starting MB
67      */
68     WORD16 i2_start_mb_y;
69 
70     /**
71      * MB y of the last MB
72      */
73 
74     WORD16 i2_end_mb_y;
75 
76     /**
77      * Bitstream offset for the current job
78      */
79     WORD32 i4_bistream_ofst;
80 
81 }job_t;
82 
83 typedef struct
84 {
85     /* Params of the reference buffer used as input to MC */
86     UWORD32 u4_src_wd;
87     UWORD32 u4_src_offset;
88 
89     /* Params of the buffer where MC output will be written */
90     UWORD32 u4_dst_wd_res_buf;
91     UWORD32 u4_dst_wd_cur_frm;
92     UWORD32 u4_dst_offset_res_buf;
93     UWORD32 u4_dst_offset_cur_frm;
94 
95     /* Operation Parameters */
96     UWORD32 u4_rows;
97     UWORD32 u4_cols;
98     UWORD32 u4_mode;
99 }comp_mc_params_t;
100 
101 typedef struct
102 {
103     yuv_buf_t        s_ref;
104     comp_mc_params_t s_luma;
105     comp_mc_params_t s_chroma;
106 }mb_mc_params_t;
107 
108 struct _dec_mb_params_t;
109 
110 typedef WORD32 pf_inv_quant_t (WORD16 *blk,
111                                 UWORD8 *weighting_matrix,
112                                 UWORD8 quant_scale,
113                                 WORD32 intra_flag,
114                                 WORD32 i4_num_coeffs,
115                                 WORD16 *pi2_coeffs,
116                                 UWORD8 *pu1_pos,
117                                 const UWORD8   *scan,
118                                 UWORD16 *u2_def_dc_pred,
119                                 UWORD16 u2_intra_dc_precision);
120 
121 typedef IMPEG2D_ERROR_CODES_T  pf_vld_inv_quant_t  (void  *dec,
122                              WORD16       *out_addr,
123                              const UWORD8 *scan,
124                              UWORD16      intra_flag,
125                              UWORD16      colr_comp,
126                              UWORD16      d_picture);
127 
128 typedef void  pf_mc_t(void *, UWORD8 *, UWORD32 , UWORD8 *, UWORD32 ,
129                  UWORD32 , UWORD32  );
130 
131 typedef struct dec_state_struct_t
132 {
133     WORD16          ai2_vld_buf[NUM_PELS_IN_BLOCK];
134     WORD16          ai2_idct_stg1[NUM_PELS_IN_BLOCK];
135 
136 
137     UWORD8          au1_intra_quant_matrix[NUM_PELS_IN_BLOCK];
138     UWORD8          au1_inter_quant_matrix[NUM_PELS_IN_BLOCK];
139 
140     IMPEG2D_ERROR_CODES_T (*pf_decode_slice)(struct dec_state_struct_t *);
141 
142     pf_vld_inv_quant_t *pf_vld_inv_quant;
143 
144     pf_idct_recon_t *pf_idct_recon[4];
145 
146     pf_mc_t         *pf_mc[4];
147     pf_interpred_t  *pf_fullx_halfy_8x8;
148     pf_interpred_t  *pf_halfx_fully_8x8;
149     pf_interpred_t  *pf_halfx_halfy_8x8;
150     pf_interpred_t  *pf_fullx_fully_8x8;
151 
152 
153     pf_interpolate_t *pf_interpolate;
154     pf_copy_mb_t     *pf_copy_mb;
155 
156     pf_memset0_one_16bit_buf_t *pf_memset_16bit_8x8_linear_block;
157     pf_memset_8bit_t    *pf_memset_8bit_8x8_block;
158     pf_copy_yuv420p_buf_t *pf_copy_yuv420p_buf;
159     pf_fmt_conv_yuv420p_to_yuv422ile_t *pf_fmt_conv_yuv420p_to_yuv422ile;
160     pf_fmt_conv_yuv420p_to_yuv420sp_t  *pf_fmt_conv_yuv420p_to_yuv420sp_uv;
161     pf_fmt_conv_yuv420p_to_yuv420sp_t  *pf_fmt_conv_yuv420p_to_yuv420sp_vu;
162 
163     stream_t         s_bit_stream;
164 /* @ */
165 
166     UWORD16         u2_is_mpeg2; /* 0 if stream is MPEG1 1 otherwise */
167     UWORD16         u2_frame_width;  /* Width of the frame */
168     UWORD16         u2_frame_height; /* Height of the frame */
169     UWORD16         u2_picture_width;
170     UWORD16         u2_horizontal_size;
171     UWORD16         u2_vertical_size;
172     UWORD16         u2_create_max_width;
173     UWORD16         u2_create_max_height;
174     UWORD16         u2_reinit_max_width;
175     UWORD16         u2_reinit_max_height;
176     UWORD16         u2_header_done;
177     UWORD16         u2_decode_header;
178 
179     UWORD16         u2_mb_x;
180     UWORD16         u2_mb_y;
181     UWORD16         u2_num_horiz_mb;
182     UWORD16         u2_num_vert_mb;
183     UWORD16         u2_num_flds_decoded;
184     void            *pv_pic_buf_mg;
185 
186     UWORD32         u4_frm_buf_stride; /* for display Buffer */
187 
188     UWORD16         u2_field_dct;
189     UWORD16         u2_read_dct_type;
190 
191     UWORD16         u2_read_motion_type;
192     UWORD16         u2_motion_type;
193 
194     const UWORD16   *pu2_mb_type;
195     UWORD16         u2_fld_pic;
196     UWORD16         u2_frm_pic;
197 
198     yuv_buf_t       s_cur_frm_buf;
199 
200     UWORD16         u2_fld_parity;
201     UWORD16         u2_def_dc_pred[MAX_COLR_COMPS];
202 
203     /* Variables related to Motion Vector predictors */
204 
205     WORD16          ai2_pred_mv[2][2][2];
206     e_pred_direction_t   e_mb_pred;
207     UWORD16         au2_fcode_data[2];
208 
209     /* Variables related to reference pictures */
210     yuv_buf_t       as_recent_fld[2][2];
211 
212     UWORD8          u1_quant_scale;
213     UWORD16         u2_num_mbs_left;
214     UWORD16         u2_first_mb;
215     UWORD16         u2_num_skipped_mbs;
216 
217     UWORD8          *pu1_inv_scan_matrix;
218 
219     UWORD16         u2_progressive_sequence;
220     e_pic_type_t         e_pic_type;
221 
222     UWORD16         u2_full_pel_forw_vector;
223     UWORD16         u2_forw_f_code;
224     UWORD16         u2_full_pel_back_vector;
225     UWORD16         u2_back_f_code;
226 
227     WORD16          ai2_mv[2][2][2]; /* Motion vectors */
228 
229     /* Bitstream code present in Picture coding extension */
230     UWORD16         au2_f_code[2][2];
231     UWORD16         u2_intra_dc_precision;
232     UWORD16         u2_picture_structure;
233     UWORD16         u2_top_field_first;
234     UWORD16         u2_frame_pred_frame_dct;
235     UWORD16         u2_concealment_motion_vectors;
236     UWORD16         u2_q_scale_type;
237     UWORD16         u2_intra_vlc_format;
238     UWORD16         u2_alternate_scan;
239     UWORD16         u2_repeat_first_field;
240     UWORD16         u2_progressive_frame;
241 
242 
243     /* Bitstream code related to frame rate of the bitstream */
244     UWORD16         u2_frame_rate_code;
245     UWORD16         u2_frame_rate_extension_n;
246     UWORD16         u2_frame_rate_extension_d;
247     UWORD16         u2_framePeriod;   /* Frame period in milli seconds */
248 
249     /* Members related to display dimensions of bitstream */
250     /* The size values may not be returned right now. But they are read */
251     /* and can be returned if there is a requirement.                   */
252     UWORD8          u1_video_format;
253     UWORD8          u1_colour_description;
254     UWORD8          u1_colour_primaries;
255     UWORD8          u1_transfer_characteristics;
256     UWORD8          u1_matrix_coefficients;
257     UWORD16         u2_display_horizontal_size;
258     UWORD16         u2_display_vertical_size;
259     UWORD16         u2_aspect_ratio_info;
260 
261     /* Members related to motion compensation */
262     yuv_buf_t       s_mc_fw_buf;
263     yuv_buf_t       s_mc_bk_buf;
264     yuv_buf_t       s_mc_buf;
265     mb_mc_params_t  as_mb_mc_params[2][2];
266     yuv_buf_t       as_ref_buf[2][2];
267     e_mb_type_t       s_mb_type;
268 
269     yuv_buf_t       s_dest_buf;
270 
271     /* Variable to handle intra MB */
272     UWORD16         u2_prev_intra_mb;
273     UWORD16         u2_coded_mb;
274 
275     /* Bidirect function pointers */
276     const struct _dec_mb_params_t *ps_func_bi_direct;
277 
278     /* Forw or Back function pointers */
279     const struct _dec_mb_params_t *ps_func_forw_or_back;
280 
281 
282     /* CBP of the current MB        */
283     UWORD16         u2_cbp;
284     void            *pv_video_scratch;
285 
286 
287     /* For global error handling */
288     void            *pv_stack_cntxt;
289 
290 /* @ */
291     WORD32          i4_chromaFormat;
292     UWORD32         u4_xdmBufID;
293     UWORD32         u4_num_mem_records;
294     /* For holding memRecords */
295     void            *pv_memTab;
296 
297     UWORD8          u1_flushfrm;
298     UWORD8          u1_flushcnt;
299     iv_yuv_buf_t    as_frame_buf[MAX_FRAME_BUFFER];
300     iv_yuv_buf_t    ps_yuv_buf;
301 
302     ivd_get_display_frame_op_t  s_disp_op;
303 
304 
305     UWORD32         u4_non_zero_cols;
306     UWORD32         u4_non_zero_rows;
307 
308     UWORD32         u4_num_frames_decoded;
309 
310     /* Adding error code variable to signal benign errors. */
311     UWORD32         u4_error_code;
312 
313     WORD32          i4_num_cores;
314 
315     UWORD8          u1_first_frame_done;
316 
317     void            *pv_codec_thread_handle;
318     void            *ps_dec_state_multi_core;
319     UWORD32         u4_inp_ts;
320     pic_buf_t       *ps_cur_pic;
321     pic_buf_t       *ps_disp_pic;
322     pic_buf_t       *aps_ref_pics[2];
323 
324     WORD32          i4_disp_buf_id;
325     WORD32          i4_cur_buf_id;
326     iv_yuv_buf_t    *ps_disp_frm_buf;
327 
328     UWORD32         u4_share_disp_buf;
329     void            *pv_pic_buf_base;
330 
331     disp_mgr_t      s_disp_mgr;
332     UWORD8          *pu1_chroma_ref_buf[BUF_MGR_MAX_CNT];
333     ivd_out_bufdesc_t as_disp_buffers[BUF_MGR_MAX_CNT];
334 
335     /* Count the number of pictures decoded after init/reset */
336     WORD32          i4_pic_count;
337 
338     /* Flag to signal last coeff in a 8x8 block is one
339     after mismatch contol */
340     WORD32          i4_last_value_one;
341 
342     WORD32          i4_start_mb_y;
343     WORD32          i4_end_mb_y;
344 
345     /* Buffer to store the input stream */
346     UWORD8 *pu1_input_buffer;
347 
348     /**
349      * Job queue buffer base
350      */
351     void            *pv_jobq_buf;
352 
353     /**
354      * Job Queue mem tab size
355      */
356     WORD32          i4_jobq_buf_size;
357 
358     /**
359      * Job Queue context
360      */
361     void            *pv_jobq;
362 
363     /* Pointer to input bitstream */
364     UWORD8          *pu1_inp_bits_buf;
365 
366     /* Number of bytes in the input bitstream */
367     UWORD32         u4_num_inp_bytes;
368 
369     ivd_out_bufdesc_t *ps_out_buf;
370 
371     /* Bytes consumed */
372     WORD32          i4_bytes_consumed;
373 
374     IVD_ARCH_T      e_processor_arch;
375 
376     IVD_SOC_T       e_processor_soc;
377 
378     WORD32          i4_frame_decoded;
379 
380     /** Flag to enable deinterlace */
381     UWORD32          u4_deinterlace;
382 
383     /** Deinterlacer context */
384     void            *pv_deinterlacer_ctxt;
385 
386     /** Picture buffer held by deinterlacer */
387     pic_buf_t       *ps_deint_pic;
388 
389     /** Buffer used after deinterlacer for format conversion */
390     UWORD8          *pu1_deint_fmt_buf;
391 
392     /** Flag to indicate if Seq Display Extn is present */
393     UWORD8          u1_seq_disp_extn_present;
394 }dec_state_t;
395 
396 
397 
398 
399 typedef void (*func_decmb_params)(dec_state_t *);
400 typedef void  (*mc_funcs)(dec_state_t *);
401 typedef struct _dec_mb_params_t
402 {
403     func_decmb_params    pf_func_mb_params;
404     e_mb_type_t            s_mb_type;
405     mc_funcs             pf_mc;
406 }dec_mb_params_t;
407 
408 
409 
410 #define MAX_THREADS     4
411 
412 
413 #define MAX_MB_ROWS     (MAX_HEIGHT / 16) // number of rows for 1080p
414 
415 typedef struct _dec_state_multi_core
416 {
417     // contains the decoder state of decoder for each thread
418     dec_state_t *ps_dec_state[MAX_THREADS];
419     UWORD32     au4_thread_launched[MAX_THREADS];
420     // number of rows: first thread will populate the row offsets and update
421     // row_offset_cnt. Other threads should pick up offset from this thread
422     // and start decoding
423     UWORD32     au4_row_offset[MAX_MB_ROWS];
424     volatile    UWORD32 u4_row_offset_cnt;
425 }dec_state_multi_core_t;
426 
427 
428 
429 #endif /* #ifndef __IMPEG2D_STRUCTS_H__ */
430