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 /*!
21 **************************************************************************
22 * \file ih264d_parse_headers.c
23 *
24 * \brief
25 * Contains High level syntax[above slice] parsing routines
26 *
27 * \date
28 * 19/12/2002
29 *
30 * \author AI
31 **************************************************************************
32 */
33 #include <string.h>
34
35 #include "ih264_typedefs.h"
36 #include "ih264_macros.h"
37 #include "ih264_platform_macros.h"
38 #include "ih264_defs.h"
39 #include "ih264d_bitstrm.h"
40 #include "ih264d_structs.h"
41 #include "ih264d_parse_cavlc.h"
42 #include "ih264d_defs.h"
43 #include "ih264d_defs.h"
44 #include "ih264d_defs.h"
45 #include "ih264d_parse_slice.h"
46 #include "ih264d_tables.h"
47 #include "ih264d_utils.h"
48 #include "ih264d_nal.h"
49 #include "ih264d_deblocking.h"
50
51 #include "ih264d_mem_request.h"
52 #include "ih264d_debug.h"
53 #include "ih264d_error_handler.h"
54 #include "ih264d_mb_utils.h"
55 #include "ih264d_sei.h"
56 #include "ih264d_vui.h"
57 #include "ih264d_thread_parse_decode.h"
58 #include "ih264d_thread_compute_bs.h"
59 #include "ih264d_quant_scaling.h"
60 #include "ih264d_defs.h"
61 #include "ivd.h"
62 #include "ih264d.h"
63
64 /*****************************************************************************/
65 /* */
66 /* Function Name : ih264d_get_pre_sei_params */
67 /* */
68 /* Description : Gets valid pre-sei params in decoder struct from parse */
69 /* struct. */
70 /* Inputs : u1_nal_unit_type slice type */
71 /* ps_dec Decoder parameters */
72 /* Globals : None */
73 /* Outputs : None */
74 /* Returns : None */
75 /* */
76 /* Issues : None */
77 /* */
78 /* Revision History: */
79 /* */
80 /* DD MM YYYY Author(s) Changes (Describe the changes made) */
81 /* Draft */
82 /* */
83 /*****************************************************************************/
84
ih264d_get_pre_sei_params(dec_struct_t * ps_dec,UWORD8 u1_nal_unit_type)85 void ih264d_get_pre_sei_params(dec_struct_t *ps_dec, UWORD8 u1_nal_unit_type)
86 {
87 if((NULL != ps_dec->ps_sei) &&
88 ((0 == ps_dec->ps_sei->s_sei_ccv_params.u1_ccv_cancel_flag) &&
89 (0 == ps_dec->ps_sei->s_sei_ccv_params.u1_ccv_persistence_flag)))
90 {
91 ps_dec->ps_sei->u1_sei_ccv_params_present_flag = 0;
92 memset(&ps_dec->ps_sei->s_sei_ccv_params, 0, sizeof(sei_ccv_params_t));
93 }
94
95 if((NULL != ps_dec->ps_cur_sps) &&
96 ((1 == ps_dec->ps_cur_sps->u1_vui_parameters_present_flag) &&
97 ((2 != ps_dec->ps_cur_sps->s_vui.u1_colour_primaries) &&
98 (2 != ps_dec->ps_cur_sps->s_vui.u1_matrix_coeffs) &&
99 (2 != ps_dec->ps_cur_sps->s_vui.u1_tfr_chars) &&
100 (4 != ps_dec->ps_cur_sps->s_vui.u1_tfr_chars) &&
101 (5 != ps_dec->ps_cur_sps->s_vui.u1_tfr_chars))))
102 {
103 if((1 == ps_dec->ps_sei_parse->u1_sei_ccv_params_present_flag) ||
104 (IDR_SLICE_NAL == u1_nal_unit_type))
105 {
106 ps_dec->ps_sei->u1_sei_ccv_params_present_flag =
107 ps_dec->ps_sei_parse->u1_sei_ccv_params_present_flag;
108 ps_dec->ps_sei->s_sei_ccv_params = ps_dec->ps_sei_parse->s_sei_ccv_params;
109 }
110 }
111 else
112 {
113 ps_dec->ps_sei->u1_sei_ccv_params_present_flag = 0;
114 memset(&ps_dec->ps_sei->s_sei_ccv_params, 0, sizeof(sei_ccv_params_t));
115 }
116
117 if(IDR_SLICE_NAL == u1_nal_unit_type)
118 {
119 ps_dec->ps_sei->u1_sei_mdcv_params_present_flag =
120 ps_dec->ps_sei_parse->u1_sei_mdcv_params_present_flag;
121 ps_dec->ps_sei->s_sei_mdcv_params = ps_dec->ps_sei_parse->s_sei_mdcv_params;
122 ps_dec->ps_sei->u1_sei_cll_params_present_flag =
123 ps_dec->ps_sei_parse->u1_sei_cll_params_present_flag;
124 ps_dec->ps_sei->s_sei_cll_params = ps_dec->ps_sei_parse->s_sei_cll_params;
125 ps_dec->ps_sei->u1_sei_ave_params_present_flag =
126 ps_dec->ps_sei_parse->u1_sei_ave_params_present_flag;
127 ps_dec->ps_sei->s_sei_ave_params = ps_dec->ps_sei_parse->s_sei_ave_params;
128 ps_dec->ps_sei->u1_sei_sii_params_present_flag =
129 ps_dec->ps_sei_parse->u1_sei_sii_params_present_flag;
130 ps_dec->ps_sei->s_sei_sii_params = ps_dec->ps_sei_parse->s_sei_sii_params;
131 }
132
133 if(NULL != ps_dec->ps_sei)
134 {
135 ps_dec->ps_sei->u1_sei_fgc_params_present_flag =
136 ps_dec->ps_sei_parse->u1_sei_fgc_params_present_flag;
137 ps_dec->ps_sei->s_sei_fgc_params = ps_dec->ps_sei_parse->s_sei_fgc_params;
138 }
139
140 ps_dec->ps_sei_parse->u1_sei_mdcv_params_present_flag = 0;
141 memset(&ps_dec->ps_sei_parse->s_sei_mdcv_params, 0, sizeof(sei_mdcv_params_t));
142 ps_dec->ps_sei_parse->u1_sei_cll_params_present_flag = 0;
143 memset(&ps_dec->ps_sei_parse->s_sei_cll_params, 0, sizeof(sei_cll_params_t));
144 ps_dec->ps_sei_parse->u1_sei_ave_params_present_flag = 0;
145 memset(&ps_dec->ps_sei_parse->s_sei_ave_params, 0, sizeof(sei_ave_params_t));
146 ps_dec->ps_sei_parse->u1_sei_ccv_params_present_flag = 0;
147 memset(&ps_dec->ps_sei_parse->s_sei_ccv_params, 0, sizeof(sei_ccv_params_t));
148 ps_dec->ps_sei_parse->u1_sei_sii_params_present_flag = 0;
149 memset(&ps_dec->ps_sei_parse->s_sei_sii_params, 0, sizeof(sei_sii_params_t));
150 }
151
152 /*****************************************************************************/
153 /* */
154 /* Function Name : ih264d_parse_slice_partition */
155 /* */
156 /* Description : This function is intended to parse and decode slice part */
157 /* itions. Currently it's not implemented. Decoder will */
158 /* print a message, skips this NAL and continues */
159 /* Inputs : ps_dec Decoder parameters */
160 /* ps_bitstrm Bitstream */
161 /* Globals : None */
162 /* Processing : This functionality needs to be implemented */
163 /* Outputs : None */
164 /* Returns : None */
165 /* */
166 /* Issues : Not implemented */
167 /* */
168 /* Revision History: */
169 /* */
170 /* DD MM YYYY Author(s) Changes (Describe the changes made) */
171 /* 06 05 2002 NS Draft */
172 /* */
173 /*****************************************************************************/
174
ih264d_parse_slice_partition(dec_struct_t * ps_dec,dec_bit_stream_t * ps_bitstrm)175 WORD32 ih264d_parse_slice_partition(dec_struct_t * ps_dec,
176 dec_bit_stream_t * ps_bitstrm)
177 {
178 H264_DEC_DEBUG_PRINT("\nSlice partition not supported");
179 UNUSED(ps_dec);
180 UNUSED(ps_bitstrm);
181 return (0);
182 }
183
184 /*****************************************************************************/
185 /* */
186 /* Function Name : ih264d_parse_sei */
187 /* */
188 /* Description : This function is intended to parse and decode SEI */
189 /* Currently it's not implemented. Decoder will print a */
190 /* message, skips this NAL and continues */
191 /* Inputs : ps_dec Decoder parameters */
192 /* ps_bitstrm Bitstream */
193 /* Globals : None */
194 /* Processing : This functionality needs to be implemented */
195 /* Outputs : None */
196 /* Returns : None */
197 /* */
198 /* Issues : Not implemented */
199 /* */
200 /* Revision History: */
201 /* */
202 /* DD MM YYYY Author(s) Changes (Describe the changes made) */
203 /* 06 05 2002 NS Draft */
204 /* */
205 /*****************************************************************************/
ih264d_parse_sei(dec_struct_t * ps_dec,dec_bit_stream_t * ps_bitstrm)206 WORD32 ih264d_parse_sei(dec_struct_t * ps_dec, dec_bit_stream_t * ps_bitstrm)
207 {
208 UNUSED(ps_dec);
209 UNUSED(ps_bitstrm);
210 return (0);
211 }
212
213 /*****************************************************************************/
214 /* */
215 /* Function Name : ih264d_parse_filler_data */
216 /* */
217 /* Description : This function is intended to parse and decode filler */
218 /* data NAL. Currently it's not implemented. Decoder will */
219 /* print a message, skips this NAL and continues */
220 /* Inputs : ps_dec Decoder parameters */
221 /* ps_bitstrm Bitstream */
222 /* Globals : None */
223 /* Processing : This functionality needs to be implemented */
224 /* Outputs : None */
225 /* Returns : None */
226 /* */
227 /* Issues : Not implemented */
228 /* */
229 /* Revision History: */
230 /* */
231 /* DD MM YYYY Author(s) Changes (Describe the changes made) */
232 /* 06 05 2002 NS Draft */
233 /* */
234 /*****************************************************************************/
ih264d_parse_filler_data(dec_struct_t * ps_dec,dec_bit_stream_t * ps_bitstrm)235 WORD32 ih264d_parse_filler_data(dec_struct_t * ps_dec,
236 dec_bit_stream_t * ps_bitstrm)
237 {
238 UNUSED(ps_dec);
239 UNUSED(ps_bitstrm);
240 return (0);
241 }
242
243 /*****************************************************************************/
244 /* */
245 /* Function Name : ih264d_parse_end_of_stream */
246 /* */
247 /* Description : This function is intended to parse and decode end of */
248 /* sequence. Currently it's not implemented. Decoder will */
249 /* print a message, skips this NAL and continues */
250 /* Inputs : ps_dec Decoder parameters */
251 /* Globals : None */
252 /* Processing : This functionality needs to be implemented */
253 /* Outputs : None */
254 /* Returns : None */
255 /* */
256 /* Issues : Not implemented */
257 /* */
258 /* Revision History: */
259 /* */
260 /* DD MM YYYY Author(s) Changes (Describe the changes made) */
261 /* 06 05 2002 NS Draft */
262 /* */
263 /*****************************************************************************/
ih264d_parse_end_of_stream(dec_struct_t * ps_dec)264 void ih264d_parse_end_of_stream(dec_struct_t * ps_dec)
265 {
266 UNUSED(ps_dec);
267 return;
268 }
269
270 /*!
271 **************************************************************************
272 * \if Function name : ih264d_parse_pps \endif
273 *
274 * \brief
275 * Decodes Picture Parameter set
276 *
277 * \return
278 * 0 on Success and Error code otherwise
279 **************************************************************************
280 */
ih264d_parse_pps(dec_struct_t * ps_dec,dec_bit_stream_t * ps_bitstrm)281 WORD32 ih264d_parse_pps(dec_struct_t * ps_dec, dec_bit_stream_t * ps_bitstrm)
282 {
283 UWORD8 uc_temp;
284 dec_seq_params_t * ps_sps = NULL;
285 dec_pic_params_t * ps_pps = NULL;
286 UWORD32 *pu4_bitstrm_buf = ps_dec->ps_bitstrm->pu4_buffer;
287 UWORD32 *pu4_bitstrm_ofst = &ps_dec->ps_bitstrm->u4_ofst;
288
289 /* Variables used for error resilience checks */
290 UWORD64 u8_temp;
291 UWORD32 u4_temp;
292 WORD32 i_temp;
293
294 /* For High profile related syntax elements */
295 UWORD8 u1_more_data_flag;
296 WORD32 i4_i;
297
298 if(!(ps_dec->i4_header_decoded & 1))
299 return ERROR_INV_SPS_PPS_T;
300
301 /*--------------------------------------------------------------------*/
302 /* Decode pic_parameter_set_id and find corresponding pic params */
303 /*--------------------------------------------------------------------*/
304 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
305 if(u4_temp & MASK_ERR_PIC_SET_ID)
306 return ERROR_INV_SPS_PPS_T;
307 ps_pps = ps_dec->pv_scratch_sps_pps;
308 *ps_pps = ps_dec->ps_pps[u4_temp];
309 ps_pps->u1_pic_parameter_set_id = (WORD8)u4_temp;
310 COPYTHECONTEXT("PPS: pic_parameter_set_id",ps_pps->u1_pic_parameter_set_id);
311
312 /************************************************/
313 /* initilization of High profile syntax element */
314 /************************************************/
315 ps_pps->i4_transform_8x8_mode_flag = 0;
316 ps_pps->i4_pic_scaling_matrix_present_flag = 0;
317
318 /*--------------------------------------------------------------------*/
319 /* Decode seq_parameter_set_id and map it to a seq_parameter_set */
320 /*--------------------------------------------------------------------*/
321 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
322 if(u4_temp & MASK_ERR_SEQ_SET_ID)
323 return ERROR_INV_SPS_PPS_T;
324 COPYTHECONTEXT("PPS: seq_parameter_set_id",u4_temp);
325 ps_sps = &ps_dec->ps_sps[u4_temp];
326
327 if(FALSE == ps_sps->u1_is_valid)
328 return ERROR_INV_SPS_PPS_T;
329 ps_pps->ps_sps = ps_sps;
330
331 /*--------------------------------------------------------------------*/
332 /* Decode entropy_coding_mode */
333 /*--------------------------------------------------------------------*/
334 ps_pps->u1_entropy_coding_mode = ih264d_get_bit_h264(ps_bitstrm);
335 COPYTHECONTEXT("PPS: entropy_coding_mode_flag",ps_pps->u1_entropy_coding_mode);
336
337 ps_pps->u1_pic_order_present_flag = ih264d_get_bit_h264(ps_bitstrm);
338 COPYTHECONTEXT("PPS: pic_order_present_flag",ps_pps->u1_pic_order_present_flag);
339
340 /*--------------------------------------------------------------------*/
341 /* Decode num_slice_groups_minus1 */
342 /*--------------------------------------------------------------------*/
343 u8_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf) + (UWORD64)1;
344 if(u8_temp != 1)
345 {
346 return ERROR_FEATURE_UNAVAIL;
347 }
348 ps_pps->u1_num_slice_groups = u8_temp;
349 COPYTHECONTEXT("PPS: num_slice_groups_minus1",ps_pps->u1_num_slice_groups -1);
350
351 /*--------------------------------------------------------------------*/
352 /* Other parameter set values */
353 /*--------------------------------------------------------------------*/
354 u8_temp = (UWORD64)1 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
355 if(u8_temp > H264_MAX_REF_IDX)
356 return ERROR_REF_IDX;
357 ps_pps->u1_num_ref_idx_lx_active[0] = u8_temp;
358 COPYTHECONTEXT("PPS: num_ref_idx_l0_active_minus1",
359 ps_pps->u1_num_ref_idx_lx_active[0] - 1);
360
361 u8_temp = (UWORD64)1 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
362 if(u8_temp > H264_MAX_REF_IDX)
363 return ERROR_REF_IDX;
364 ps_pps->u1_num_ref_idx_lx_active[1] = u8_temp;
365 COPYTHECONTEXT("PPS: num_ref_idx_l1_active_minus1",
366 ps_pps->u1_num_ref_idx_lx_active[1] - 1);
367
368 ps_pps->u1_wted_pred_flag = ih264d_get_bit_h264(ps_bitstrm);
369 COPYTHECONTEXT("PPS: weighted prediction u4_flag",ps_pps->u1_wted_pred_flag);
370 uc_temp = ih264d_get_bits_h264(ps_bitstrm, 2);
371 COPYTHECONTEXT("PPS: weighted_bipred_idc",uc_temp);
372 ps_pps->u1_wted_bipred_idc = uc_temp;
373
374 if(ps_pps->u1_wted_bipred_idc > MAX_WEIGHT_BIPRED_IDC)
375 return ERROR_INV_SPS_PPS_T;
376 {
377 WORD64 i8_temp;
378 i8_temp = (WORD64)26 + ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
379
380 if((i8_temp < MIN_H264_QP) || (i8_temp > MAX_H264_QP))
381 return ERROR_INV_RANGE_QP_T;
382
383 ps_pps->u1_pic_init_qp = i8_temp;
384 COPYTHECONTEXT("PPS: pic_init_qp_minus26",ps_pps->u1_pic_init_qp - 26);
385
386 i8_temp = (WORD64)26 + ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
387
388 if((i8_temp < MIN_H264_QP) || (i8_temp > MAX_H264_QP))
389 return ERROR_INV_RANGE_QP_T;
390
391 ps_pps->u1_pic_init_qs = i8_temp;
392 COPYTHECONTEXT("PPS: pic_init_qs_minus26",ps_pps->u1_pic_init_qs - 26);
393 }
394
395 i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
396 if((i_temp < -12) || (i_temp > 12))
397 return ERROR_INV_RANGE_QP_T;
398 ps_pps->i1_chroma_qp_index_offset = i_temp;
399 COPYTHECONTEXT("PPS: chroma_qp_index_offset",ps_pps->i1_chroma_qp_index_offset);
400
401 /***************************************************************************/
402 /* initialize second_chroma_qp_index_offset to i1_chroma_qp_index_offset if */
403 /* second_chroma_qp_index_offset is not present in bit-ps_bitstrm */
404 /***************************************************************************/
405 ps_pps->i1_second_chroma_qp_index_offset =
406 ps_pps->i1_chroma_qp_index_offset;
407
408 ps_pps->u1_deblocking_filter_parameters_present_flag = ih264d_get_bit_h264(
409 ps_bitstrm);
410 COPYTHECONTEXT("PPS: deblocking_filter_control_present_flag",
411 ps_pps->u1_deblocking_filter_parameters_present_flag);
412 ps_pps->u1_constrained_intra_pred_flag = ih264d_get_bit_h264(ps_bitstrm);
413 COPYTHECONTEXT("PPS: constrained_intra_pred_flag",
414 ps_pps->u1_constrained_intra_pred_flag);
415 ps_pps->u1_redundant_pic_cnt_present_flag = ih264d_get_bit_h264(ps_bitstrm);
416 COPYTHECONTEXT("PPS: redundant_pic_cnt_present_flag",
417 ps_pps->u1_redundant_pic_cnt_present_flag);
418
419 /* High profile related syntax elements */
420 u1_more_data_flag = MORE_RBSP_DATA(ps_bitstrm);
421 if(u1_more_data_flag && (ps_pps->ps_sps->u1_profile_idc == HIGH_PROFILE_IDC))
422 {
423 /* read transform_8x8_mode_flag */
424 ps_pps->i4_transform_8x8_mode_flag = (WORD32)ih264d_get_bit_h264(
425 ps_bitstrm);
426
427 /* read pic_scaling_matrix_present_flag */
428 ps_pps->i4_pic_scaling_matrix_present_flag =
429 (WORD32)ih264d_get_bit_h264(ps_bitstrm);
430
431 if(ps_pps->i4_pic_scaling_matrix_present_flag)
432 {
433 /* read the scaling matrices */
434 for(i4_i = 0; i4_i < (6 + (ps_pps->i4_transform_8x8_mode_flag << 1)); i4_i++)
435 {
436 ps_pps->u1_pic_scaling_list_present_flag[i4_i] =
437 ih264d_get_bit_h264(ps_bitstrm);
438
439 if(ps_pps->u1_pic_scaling_list_present_flag[i4_i])
440 {
441 WORD32 ret;
442 if(i4_i < 6)
443 {
444 ret = ih264d_scaling_list(
445 ps_pps->i2_pic_scalinglist4x4[i4_i],
446 16,
447 &ps_pps->u1_pic_use_default_scaling_matrix_flag[i4_i],
448 ps_bitstrm);
449 }
450 else
451 {
452 ret = ih264d_scaling_list(
453 ps_pps->i2_pic_scalinglist8x8[i4_i - 6],
454 64,
455 &ps_pps->u1_pic_use_default_scaling_matrix_flag[i4_i],
456 ps_bitstrm);
457 }
458
459 if(ret != OK)
460 {
461 return ret;
462 }
463 }
464 }
465 }
466
467 /* read second_chroma_qp_index_offset syntax element */
468 i_temp = ih264d_sev(
469 pu4_bitstrm_ofst, pu4_bitstrm_buf);
470
471 if((i_temp < -12) || (i_temp > 12))
472 return ERROR_INV_RANGE_QP_T;
473
474 ps_pps->i1_second_chroma_qp_index_offset = i_temp;
475 }
476
477 /* In case bitstream read has exceeded the filled size, then
478 return an error */
479 if(EXCEED_OFFSET(ps_bitstrm))
480 {
481 return ERROR_INV_SPS_PPS_T;
482 }
483 ps_pps->u1_is_valid = TRUE;
484 ps_dec->ps_pps[ps_pps->u1_pic_parameter_set_id] = *ps_pps;
485 return OK;
486 }
487
488 /*!
489 **************************************************************************
490 * \if Function name : ih264d_parse_sps \endif
491 *
492 * \brief
493 * Decodes Sequence parameter set from the bitstream
494 *
495 * \return
496 * 0 on Success and Error code otherwise
497 **************************************************************************
498 */
ih264d_correct_level_idc(UWORD32 u4_level_idc,UWORD32 u4_total_mbs)499 UWORD32 ih264d_correct_level_idc(UWORD32 u4_level_idc, UWORD32 u4_total_mbs)
500 {
501 UWORD32 u4_max_mbs_allowed;
502
503 switch(u4_level_idc)
504 {
505 case H264_LEVEL_1_0:
506 u4_max_mbs_allowed = MAX_MBS_LEVEL_10;
507 break;
508 case H264_LEVEL_1_1:
509 u4_max_mbs_allowed = MAX_MBS_LEVEL_11;
510 break;
511 case H264_LEVEL_1_2:
512 u4_max_mbs_allowed = MAX_MBS_LEVEL_12;
513 break;
514 case H264_LEVEL_1_3:
515 u4_max_mbs_allowed = MAX_MBS_LEVEL_13;
516 break;
517 case H264_LEVEL_2_0:
518 u4_max_mbs_allowed = MAX_MBS_LEVEL_20;
519 break;
520 case H264_LEVEL_2_1:
521 u4_max_mbs_allowed = MAX_MBS_LEVEL_21;
522 break;
523 case H264_LEVEL_2_2:
524 u4_max_mbs_allowed = MAX_MBS_LEVEL_22;
525 break;
526 case H264_LEVEL_3_0:
527 u4_max_mbs_allowed = MAX_MBS_LEVEL_30;
528 break;
529 case H264_LEVEL_3_1:
530 u4_max_mbs_allowed = MAX_MBS_LEVEL_31;
531 break;
532 case H264_LEVEL_3_2:
533 u4_max_mbs_allowed = MAX_MBS_LEVEL_32;
534 break;
535 case H264_LEVEL_4_0:
536 u4_max_mbs_allowed = MAX_MBS_LEVEL_40;
537 break;
538 case H264_LEVEL_4_1:
539 u4_max_mbs_allowed = MAX_MBS_LEVEL_41;
540 break;
541 case H264_LEVEL_4_2:
542 u4_max_mbs_allowed = MAX_MBS_LEVEL_42;
543 break;
544 case H264_LEVEL_5_0:
545 u4_max_mbs_allowed = MAX_MBS_LEVEL_50;
546 break;
547 case H264_LEVEL_5_1:
548 default:
549 u4_max_mbs_allowed = MAX_MBS_LEVEL_51;
550 break;
551
552 }
553
554 /*correct of the level is incorrect*/
555 if(u4_total_mbs > u4_max_mbs_allowed)
556 {
557 if(u4_total_mbs > MAX_MBS_LEVEL_50)
558 u4_level_idc = H264_LEVEL_5_1;
559 else if(u4_total_mbs > MAX_MBS_LEVEL_42)
560 u4_level_idc = H264_LEVEL_5_0;
561 else if(u4_total_mbs > MAX_MBS_LEVEL_41)
562 u4_level_idc = H264_LEVEL_4_2;
563 else if(u4_total_mbs > MAX_MBS_LEVEL_40)
564 u4_level_idc = H264_LEVEL_4_1;
565 else if(u4_total_mbs > MAX_MBS_LEVEL_32)
566 u4_level_idc = H264_LEVEL_4_0;
567 else if(u4_total_mbs > MAX_MBS_LEVEL_31)
568 u4_level_idc = H264_LEVEL_3_2;
569 else if(u4_total_mbs > MAX_MBS_LEVEL_30)
570 u4_level_idc = H264_LEVEL_3_1;
571 else if(u4_total_mbs > MAX_MBS_LEVEL_21)
572 u4_level_idc = H264_LEVEL_3_0;
573 else if(u4_total_mbs > MAX_MBS_LEVEL_20)
574 u4_level_idc = H264_LEVEL_2_1;
575 else if(u4_total_mbs > MAX_MBS_LEVEL_10)
576 u4_level_idc = H264_LEVEL_2_0;
577 }
578
579 return (u4_level_idc);
580
581 }
ih264d_parse_sps(dec_struct_t * ps_dec,dec_bit_stream_t * ps_bitstrm)582 WORD32 ih264d_parse_sps(dec_struct_t *ps_dec, dec_bit_stream_t *ps_bitstrm)
583 {
584 UWORD8 i;
585 dec_seq_params_t *ps_seq = NULL;
586 UWORD8 u1_profile_idc, u1_level_idc, u1_seq_parameter_set_id, u1_mb_aff_flag = 0;
587 UWORD32 u4_max_frm_num;
588 UWORD32 *pu4_bitstrm_buf = ps_bitstrm->pu4_buffer;
589 UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst;
590 UWORD8 u1_frm, uc_constraint_set0_flag, uc_constraint_set1_flag;
591 WORD32 i4_cropped_ht, i4_cropped_wd;
592 UWORD32 u4_temp;
593 UWORD64 u8_temp;
594 UWORD32 u4_pic_height_in_map_units, u4_pic_width_in_mbs;
595 UWORD32 u2_pic_wd = 0;
596 UWORD32 u2_pic_ht = 0;
597 UWORD32 u2_frm_wd_y = 0;
598 UWORD32 u2_frm_ht_y = 0;
599 UWORD32 u2_frm_wd_uv = 0;
600 UWORD32 u2_frm_ht_uv = 0;
601 UWORD32 u2_crop_offset_y = 0;
602 UWORD32 u2_crop_offset_uv = 0;
603 WORD32 ret;
604 WORD32 num_reorder_frames;
605 /* High profile related syntax element */
606 WORD32 i4_i;
607 /* G050 */
608 UWORD8 u1_frame_cropping_flag, u1_frame_cropping_rect_left_ofst,
609 u1_frame_cropping_rect_right_ofst,
610 u1_frame_cropping_rect_top_ofst,
611 u1_frame_cropping_rect_bottom_ofst;
612 /* G050 */
613 /*--------------------------------------------------------------------*/
614 /* Decode seq_parameter_set_id and profile and level values */
615 /*--------------------------------------------------------------------*/
616 SWITCHONTRACE;
617 u1_profile_idc = ih264d_get_bits_h264(ps_bitstrm, 8);
618 COPYTHECONTEXT("SPS: profile_idc",u1_profile_idc);
619
620 /* G050 */
621 uc_constraint_set0_flag = ih264d_get_bit_h264(ps_bitstrm);
622 uc_constraint_set1_flag = ih264d_get_bit_h264(ps_bitstrm);
623 ih264d_get_bit_h264(ps_bitstrm);
624
625 /*****************************************************/
626 /* Read 5 bits for uc_constraint_set3_flag (1 bit) */
627 /* and reserved_zero_4bits (4 bits) - Sushant */
628 /*****************************************************/
629 ih264d_get_bits_h264(ps_bitstrm, 5);
630 /* G050 */
631
632 /* Check whether particular profile is suported or not */
633 /* Check whether particular profile is suported or not */
634 if((u1_profile_idc != MAIN_PROFILE_IDC) &&
635
636 (u1_profile_idc != BASE_PROFILE_IDC) &&
637
638 (u1_profile_idc != HIGH_PROFILE_IDC)
639
640 )
641 {
642
643 /* Apart from Baseline, main and high profile,
644 * only extended profile is supported provided
645 * uc_constraint_set0_flag or uc_constraint_set1_flag are set to 1
646 */
647 if((u1_profile_idc != EXTENDED_PROFILE_IDC) ||
648 ((uc_constraint_set1_flag != 1) && (uc_constraint_set0_flag != 1)))
649 {
650 return (ERROR_FEATURE_UNAVAIL);
651 }
652 }
653
654 u1_level_idc = ih264d_get_bits_h264(ps_bitstrm, 8);
655
656
657
658 COPYTHECONTEXT("SPS: u4_level_idc",u1_level_idc);
659
660 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
661 if(u4_temp & MASK_ERR_SEQ_SET_ID)
662 return ERROR_INV_SPS_PPS_T;
663 u1_seq_parameter_set_id = u4_temp;
664 COPYTHECONTEXT("SPS: seq_parameter_set_id",
665 u1_seq_parameter_set_id);
666
667 /*--------------------------------------------------------------------*/
668 /* Find an seq param entry in seqparam array of decStruct */
669 /*--------------------------------------------------------------------*/
670
671 ps_seq = ps_dec->pv_scratch_sps_pps;
672 memset(ps_seq, 0, sizeof(dec_seq_params_t));
673
674 if(ps_dec->i4_header_decoded & 1)
675 {
676 *ps_seq = *ps_dec->ps_cur_sps;
677 }
678
679
680 if((ps_dec->i4_header_decoded & 1) && (ps_seq->u1_profile_idc != u1_profile_idc))
681 {
682 ps_dec->u1_res_changed = 1;
683 return IVD_RES_CHANGED;
684 }
685
686 if((ps_dec->i4_header_decoded & 1) && (ps_seq->u1_level_idc != u1_level_idc))
687 {
688 ps_dec->u1_res_changed = 1;
689 return IVD_RES_CHANGED;
690 }
691
692 ps_seq->u1_profile_idc = u1_profile_idc;
693 ps_seq->u1_level_idc = u1_level_idc;
694 ps_seq->u1_seq_parameter_set_id = u1_seq_parameter_set_id;
695
696 /*******************************************************************/
697 /* Initializations for high profile - Sushant */
698 /*******************************************************************/
699 ps_seq->i4_chroma_format_idc = 1;
700 ps_seq->i4_bit_depth_luma_minus8 = 0;
701 ps_seq->i4_bit_depth_chroma_minus8 = 0;
702 ps_seq->i4_qpprime_y_zero_transform_bypass_flag = 0;
703 ps_seq->i4_seq_scaling_matrix_present_flag = 0;
704 if(u1_profile_idc == HIGH_PROFILE_IDC)
705 {
706
707 /* reading chroma_format_idc */
708 ps_seq->i4_chroma_format_idc = ih264d_uev(pu4_bitstrm_ofst,
709 pu4_bitstrm_buf);
710
711 /* Monochrome is not supported */
712 if(ps_seq->i4_chroma_format_idc != 1)
713 {
714 return ERROR_FEATURE_UNAVAIL;
715 }
716
717 /* reading bit_depth_luma_minus8 */
718 ps_seq->i4_bit_depth_luma_minus8 = ih264d_uev(pu4_bitstrm_ofst,
719 pu4_bitstrm_buf);
720
721 if(ps_seq->i4_bit_depth_luma_minus8 != 0)
722 {
723 return ERROR_FEATURE_UNAVAIL;
724 }
725
726 /* reading bit_depth_chroma_minus8 */
727 ps_seq->i4_bit_depth_chroma_minus8 = ih264d_uev(pu4_bitstrm_ofst,
728 pu4_bitstrm_buf);
729
730 if(ps_seq->i4_bit_depth_chroma_minus8 != 0)
731 {
732 return ERROR_FEATURE_UNAVAIL;
733 }
734
735 /* reading qpprime_y_zero_transform_bypass_flag */
736 ps_seq->i4_qpprime_y_zero_transform_bypass_flag =
737 (WORD32)ih264d_get_bit_h264(ps_bitstrm);
738
739 if(ps_seq->i4_qpprime_y_zero_transform_bypass_flag != 0)
740 {
741 return ERROR_INV_SPS_PPS_T;
742 }
743
744 /* reading seq_scaling_matrix_present_flag */
745 ps_seq->i4_seq_scaling_matrix_present_flag =
746 (WORD32)ih264d_get_bit_h264(ps_bitstrm);
747
748 if(ps_seq->i4_seq_scaling_matrix_present_flag)
749 {
750 for(i4_i = 0; i4_i < 8; i4_i++)
751 {
752 ps_seq->u1_seq_scaling_list_present_flag[i4_i] =
753 ih264d_get_bit_h264(ps_bitstrm);
754
755 /* initialize u1_use_default_scaling_matrix_flag[i4_i] to zero */
756 /* before calling scaling list */
757 ps_seq->u1_use_default_scaling_matrix_flag[i4_i] = 0;
758
759 if(ps_seq->u1_seq_scaling_list_present_flag[i4_i])
760 {
761 if(i4_i < 6)
762 {
763 ret = ih264d_scaling_list(
764 ps_seq->i2_scalinglist4x4[i4_i],
765 16,
766 &ps_seq->u1_use_default_scaling_matrix_flag[i4_i],
767 ps_bitstrm);
768 }
769 else
770 {
771 ret = ih264d_scaling_list(
772 ps_seq->i2_scalinglist8x8[i4_i - 6],
773 64,
774 &ps_seq->u1_use_default_scaling_matrix_flag[i4_i],
775 ps_bitstrm);
776 }
777 if(ret != OK)
778 {
779 return ret;
780 }
781 }
782 }
783 }
784 }
785 /*--------------------------------------------------------------------*/
786 /* Decode MaxFrameNum */
787 /*--------------------------------------------------------------------*/
788 u8_temp = (UWORD64)4 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
789 if(u8_temp > MAX_BITS_IN_FRAME_NUM)
790 {
791 return ERROR_INV_SPS_PPS_T;
792 }
793 ps_seq->u1_bits_in_frm_num = u8_temp;
794 COPYTHECONTEXT("SPS: log2_max_frame_num_minus4",
795 (ps_seq->u1_bits_in_frm_num - 4));
796
797 u4_max_frm_num = (1 << (ps_seq->u1_bits_in_frm_num));
798 ps_seq->u2_u4_max_pic_num_minus1 = u4_max_frm_num - 1;
799 /*--------------------------------------------------------------------*/
800 /* Decode picture order count and related values */
801 /*--------------------------------------------------------------------*/
802 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
803
804 if(u4_temp > MAX_PIC_ORDER_CNT_TYPE)
805 {
806 return ERROR_INV_POC_TYPE_T;
807 }
808 ps_seq->u1_pic_order_cnt_type = u4_temp;
809 COPYTHECONTEXT("SPS: pic_order_cnt_type",ps_seq->u1_pic_order_cnt_type);
810
811 ps_seq->u1_num_ref_frames_in_pic_order_cnt_cycle = 1;
812 if(ps_seq->u1_pic_order_cnt_type == 0)
813 {
814 u8_temp = (UWORD64)4 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
815 if(u8_temp > MAX_BITS_IN_POC_LSB)
816 {
817 return ERROR_INV_SPS_PPS_T;
818 }
819 ps_seq->u1_log2_max_pic_order_cnt_lsb_minus = u8_temp;
820 ps_seq->i4_max_pic_order_cntLsb = (1 << u8_temp);
821 COPYTHECONTEXT("SPS: log2_max_pic_order_cnt_lsb_minus4",(u8_temp - 4));
822 }
823 else if(ps_seq->u1_pic_order_cnt_type == 1)
824 {
825 ps_seq->u1_delta_pic_order_always_zero_flag = ih264d_get_bit_h264(
826 ps_bitstrm);
827 COPYTHECONTEXT("SPS: delta_pic_order_always_zero_flag",
828 ps_seq->u1_delta_pic_order_always_zero_flag);
829
830 ps_seq->i4_ofst_for_non_ref_pic = ih264d_sev(pu4_bitstrm_ofst,
831 pu4_bitstrm_buf);
832 COPYTHECONTEXT("SPS: offset_for_non_ref_pic",
833 ps_seq->i4_ofst_for_non_ref_pic);
834
835 ps_seq->i4_ofst_for_top_to_bottom_field = ih264d_sev(
836 pu4_bitstrm_ofst, pu4_bitstrm_buf);
837 COPYTHECONTEXT("SPS: offset_for_top_to_bottom_field",
838 ps_seq->i4_ofst_for_top_to_bottom_field);
839
840 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
841 if(u4_temp > 255)
842 return ERROR_INV_SPS_PPS_T;
843 ps_seq->u1_num_ref_frames_in_pic_order_cnt_cycle = u4_temp;
844 COPYTHECONTEXT("SPS: num_ref_frames_in_pic_order_cnt_cycle",
845 ps_seq->u1_num_ref_frames_in_pic_order_cnt_cycle);
846
847 for(i = 0; i < ps_seq->u1_num_ref_frames_in_pic_order_cnt_cycle; i++)
848 {
849 ps_seq->i4_ofst_for_ref_frame[i] = ih264d_sev(
850 pu4_bitstrm_ofst, pu4_bitstrm_buf);
851 COPYTHECONTEXT("SPS: offset_for_ref_frame",
852 ps_seq->i4_ofst_for_ref_frame[i]);
853 }
854 }
855
856 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
857
858 if((u4_temp > H264_MAX_REF_PICS))
859 {
860 return ERROR_NUM_REF;
861 }
862
863 /* Compare with older num_ref_frames is header is already once */
864 if((ps_dec->i4_header_decoded & 1) && (ps_seq->u1_num_ref_frames != u4_temp))
865 {
866 ps_dec->u1_res_changed = 1;
867 return IVD_RES_CHANGED;
868 }
869
870 ps_seq->u1_num_ref_frames = u4_temp;
871 COPYTHECONTEXT("SPS: num_ref_frames",ps_seq->u1_num_ref_frames);
872
873 ps_seq->u1_gaps_in_frame_num_value_allowed_flag = ih264d_get_bit_h264(
874 ps_bitstrm);
875 COPYTHECONTEXT("SPS: gaps_in_frame_num_value_allowed_flag",
876 ps_seq->u1_gaps_in_frame_num_value_allowed_flag);
877
878 /*--------------------------------------------------------------------*/
879 /* Decode FrameWidth and FrameHeight and related values */
880 /*--------------------------------------------------------------------*/
881 u8_temp = (UWORD64)1 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
882 /* Check for unsupported resolutions*/
883 if(u8_temp > (H264_MAX_FRAME_WIDTH >> 4))
884 {
885 return IVD_STREAM_WIDTH_HEIGHT_NOT_SUPPORTED;
886 }
887 u4_pic_width_in_mbs = u8_temp;
888 COPYTHECONTEXT("SPS: pic_width_in_mbs_minus1",
889 u4_pic_width_in_mbs - 1);
890
891 u8_temp = (UWORD64)1 + ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
892 if (u8_temp > (H264_MAX_FRAME_HEIGHT >> 4))
893 {
894 return IVD_STREAM_WIDTH_HEIGHT_NOT_SUPPORTED;
895 }
896 u4_pic_height_in_map_units = u8_temp;
897
898 ps_seq->u2_frm_wd_in_mbs = u4_pic_width_in_mbs;
899 ps_seq->u2_frm_ht_in_mbs = u4_pic_height_in_map_units;
900
901 u2_pic_wd = (u4_pic_width_in_mbs << 4);
902 u2_pic_ht = (u4_pic_height_in_map_units << 4);
903 /*--------------------------------------------------------------------*/
904 /* Get the value of MaxMbAddress and Number of bits needed for it */
905 /*--------------------------------------------------------------------*/
906 ps_seq->u2_max_mb_addr = (ps_seq->u2_frm_wd_in_mbs
907 * ps_seq->u2_frm_ht_in_mbs) - 1;
908
909 ps_seq->u2_total_num_of_mbs = ps_seq->u2_max_mb_addr + 1;
910
911 ps_seq->u1_level_idc = ih264d_correct_level_idc(
912 u1_level_idc, ps_seq->u2_total_num_of_mbs);
913
914 u1_frm = ih264d_get_bit_h264(ps_bitstrm);
915 if((ps_dec->i4_header_decoded & 1) && (ps_seq->u1_frame_mbs_only_flag != u1_frm))
916 {
917 ps_dec->u1_res_changed = 1;
918 return IVD_RES_CHANGED;
919 }
920
921 ps_seq->u1_frame_mbs_only_flag = u1_frm;
922
923 COPYTHECONTEXT("SPS: frame_mbs_only_flag", u1_frm);
924
925 if(!u1_frm)
926 u1_mb_aff_flag = ih264d_get_bit_h264(ps_bitstrm);
927
928 if((ps_dec->i4_header_decoded & 1)
929 && (ps_seq->u1_mb_aff_flag != u1_mb_aff_flag))
930 {
931 ps_dec->u1_res_changed = 1;
932 return IVD_RES_CHANGED;
933 }
934
935 if(!u1_frm)
936 {
937 u2_pic_ht <<= 1;
938 ps_seq->u1_mb_aff_flag = u1_mb_aff_flag;
939 COPYTHECONTEXT("SPS: mb_adaptive_frame_field_flag",
940 ps_seq->u1_mb_aff_flag);
941
942 }
943 else
944 ps_seq->u1_mb_aff_flag = 0;
945
946 ps_seq->u1_direct_8x8_inference_flag = ih264d_get_bit_h264(ps_bitstrm);
947
948 COPYTHECONTEXT("SPS: direct_8x8_inference_flag",
949 ps_seq->u1_direct_8x8_inference_flag);
950
951 /* G050 */
952 u1_frame_cropping_flag = ih264d_get_bit_h264(ps_bitstrm);
953 COPYTHECONTEXT("SPS: frame_cropping_flag",u1_frame_cropping_flag);
954
955 if(u1_frame_cropping_flag)
956 {
957 u1_frame_cropping_rect_left_ofst = ih264d_uev(pu4_bitstrm_ofst,
958 pu4_bitstrm_buf);
959 COPYTHECONTEXT("SPS: frame_cropping_rect_left_offset",
960 u1_frame_cropping_rect_left_ofst);
961 u1_frame_cropping_rect_right_ofst = ih264d_uev(pu4_bitstrm_ofst,
962 pu4_bitstrm_buf);
963 COPYTHECONTEXT("SPS: frame_cropping_rect_right_offset",
964 u1_frame_cropping_rect_right_ofst);
965 u1_frame_cropping_rect_top_ofst = ih264d_uev(pu4_bitstrm_ofst,
966 pu4_bitstrm_buf);
967 COPYTHECONTEXT("SPS: frame_cropping_rect_top_offset",
968 u1_frame_cropping_rect_top_ofst);
969 u1_frame_cropping_rect_bottom_ofst = ih264d_uev(pu4_bitstrm_ofst,
970 pu4_bitstrm_buf);
971 COPYTHECONTEXT("SPS: frame_cropping_rect_bottom_offset",
972 u1_frame_cropping_rect_bottom_ofst);
973 }
974 /* G050 */
975
976 ps_seq->u1_vui_parameters_present_flag = ih264d_get_bit_h264(ps_bitstrm);
977 COPYTHECONTEXT("SPS: vui_parameters_present_flag",
978 ps_seq->u1_vui_parameters_present_flag);
979
980 u2_frm_wd_y = u2_pic_wd + (UWORD8)(PAD_LEN_Y_H << 1);
981 if(1 == ps_dec->u4_share_disp_buf)
982 {
983 if(ps_dec->u4_app_disp_width > u2_frm_wd_y)
984 u2_frm_wd_y = ps_dec->u4_app_disp_width;
985 }
986
987 u2_frm_ht_y = u2_pic_ht + (UWORD8)(PAD_LEN_Y_V << 2);
988 u2_frm_wd_uv = u2_pic_wd + (UWORD8)(PAD_LEN_UV_H << 2);
989 u2_frm_wd_uv = MAX(u2_frm_wd_uv, u2_frm_wd_y);
990
991 u2_frm_ht_uv = (u2_pic_ht >> 1) + (UWORD8)(PAD_LEN_UV_V << 2);
992 u2_frm_ht_uv = MAX(u2_frm_ht_uv, (u2_frm_ht_y >> 1));
993
994
995 /* Calculate display picture width, height and start u4_ofst from YUV420 */
996 /* pictute buffers as per cropping information parsed above */
997 {
998 UWORD16 u2_rgt_ofst = 0;
999 UWORD16 u2_lft_ofst = 0;
1000 UWORD16 u2_top_ofst = 0;
1001 UWORD16 u2_btm_ofst = 0;
1002 UWORD8 u1_frm_mbs_flag;
1003 UWORD8 u1_vert_mult_factor;
1004
1005 if(u1_frame_cropping_flag)
1006 {
1007 /* Calculate right and left u4_ofst for cropped picture */
1008 u2_rgt_ofst = u1_frame_cropping_rect_right_ofst << 1;
1009 u2_lft_ofst = u1_frame_cropping_rect_left_ofst << 1;
1010
1011 /* Know frame MBs only u4_flag */
1012 u1_frm_mbs_flag = (1 == ps_seq->u1_frame_mbs_only_flag);
1013
1014 /* Simplify the vertical u4_ofst calculation from field/frame */
1015 u1_vert_mult_factor = (2 - u1_frm_mbs_flag);
1016
1017 /* Calculate bottom and top u4_ofst for cropped picture */
1018 u2_btm_ofst = (u1_frame_cropping_rect_bottom_ofst
1019 << u1_vert_mult_factor);
1020 u2_top_ofst = (u1_frame_cropping_rect_top_ofst
1021 << u1_vert_mult_factor);
1022 }
1023
1024 /* Calculate u4_ofst from start of YUV 420 picture buffer to start of*/
1025 /* cropped picture buffer */
1026 u2_crop_offset_y = (u2_frm_wd_y * u2_top_ofst) + (u2_lft_ofst);
1027 u2_crop_offset_uv = (u2_frm_wd_uv * (u2_top_ofst >> 1))
1028 + (u2_lft_ofst >> 1) * YUV420SP_FACTOR;
1029 /* Calculate the display picture width and height based on crop */
1030 /* information */
1031 i4_cropped_ht = (WORD32)u2_pic_ht - (WORD32)(u2_btm_ofst + u2_top_ofst);
1032 i4_cropped_wd = (WORD32)u2_pic_wd - (WORD32)(u2_rgt_ofst + u2_lft_ofst);
1033
1034 if((i4_cropped_ht < MB_SIZE) || (i4_cropped_wd < MB_SIZE))
1035 {
1036 return ERROR_INV_SPS_PPS_T;
1037 }
1038
1039 if((ps_dec->i4_header_decoded & 1) && (ps_dec->u2_pic_wd != u2_pic_wd))
1040 {
1041 ps_dec->u1_res_changed = 1;
1042 return IVD_RES_CHANGED;
1043 }
1044
1045 if((ps_dec->i4_header_decoded & 1) && (ps_dec->u2_disp_width != i4_cropped_wd))
1046 {
1047 ps_dec->u1_res_changed = 1;
1048 return IVD_RES_CHANGED;
1049 }
1050
1051 if((ps_dec->i4_header_decoded & 1) && (ps_dec->u2_pic_ht != u2_pic_ht))
1052 {
1053 ps_dec->u1_res_changed = 1;
1054 return IVD_RES_CHANGED;
1055 }
1056
1057 if((ps_dec->i4_header_decoded & 1) && (ps_dec->u2_disp_height != i4_cropped_ht))
1058 {
1059 ps_dec->u1_res_changed = 1;
1060 return IVD_RES_CHANGED;
1061 }
1062
1063 /* Check again for unsupported resolutions with updated values*/
1064 if((u2_pic_wd > H264_MAX_FRAME_WIDTH) || (u2_pic_ht > H264_MAX_FRAME_HEIGHT)
1065 || (u2_pic_wd < H264_MIN_FRAME_WIDTH) || (u2_pic_ht < H264_MIN_FRAME_HEIGHT)
1066 || (u2_pic_wd * (UWORD32)u2_pic_ht > H264_MAX_FRAME_SIZE))
1067 {
1068 return IVD_STREAM_WIDTH_HEIGHT_NOT_SUPPORTED;
1069 }
1070
1071 /* If MBAff is enabled, decoder support is limited to streams with
1072 * width less than half of H264_MAX_FRAME_WIDTH.
1073 * In case of MBAff decoder processes two rows at a time
1074 */
1075 if((u2_pic_wd << ps_seq->u1_mb_aff_flag) > H264_MAX_FRAME_WIDTH)
1076 {
1077 return IVD_STREAM_WIDTH_HEIGHT_NOT_SUPPORTED;
1078 }
1079
1080 }
1081
1082 /* Backup num_reorder_frames if header is already decoded */
1083 if((ps_dec->i4_header_decoded & 1) &&
1084 (1 == ps_seq->u1_vui_parameters_present_flag) &&
1085 (1 == ps_seq->s_vui.u1_bitstream_restriction_flag))
1086 {
1087 num_reorder_frames = (WORD32)ps_seq->s_vui.u4_num_reorder_frames;
1088 }
1089 else
1090 {
1091 num_reorder_frames = -1;
1092 }
1093 if(1 == ps_seq->u1_vui_parameters_present_flag)
1094 {
1095 ret = ih264d_parse_vui_parametres(&ps_seq->s_vui, ps_bitstrm);
1096 if(ret != OK)
1097 return ret;
1098
1099 if (ps_dec->pu1_bits_buf_dynamic != NULL) {
1100 vui_t *ps_vui = &ps_seq->s_vui;
1101 dec_seq_params_t *ps_sps_old = ps_dec->ps_sps;
1102 vui_t *ps_vui_old = &ps_sps_old->s_vui;
1103
1104 if (ps_vui->u1_video_full_range_flag != ps_vui_old->u1_video_full_range_flag ||
1105 ps_vui->u1_colour_primaries != ps_vui_old->u1_colour_primaries ||
1106 ps_vui->u1_tfr_chars != ps_vui_old->u1_tfr_chars ||
1107 ps_vui->u1_matrix_coeffs != ps_vui_old->u1_matrix_coeffs) {
1108 ps_dec->u1_res_changed = 1;
1109 return IVD_RES_CHANGED;
1110 }
1111 }
1112 }
1113
1114 /* Compare older num_reorder_frames with the new one if header is already decoded */
1115 if((ps_dec->i4_header_decoded & 1) &&
1116 (-1 != num_reorder_frames) &&
1117 (1 == ps_seq->u1_vui_parameters_present_flag) &&
1118 (1 == ps_seq->s_vui.u1_bitstream_restriction_flag) &&
1119 ((WORD32)ps_seq->s_vui.u4_num_reorder_frames != num_reorder_frames))
1120 {
1121 ps_dec->u1_res_changed = 1;
1122 return IVD_RES_CHANGED;
1123 }
1124
1125 /* In case bitstream read has exceeded the filled size, then
1126 return an error */
1127 if (EXCEED_OFFSET(ps_bitstrm))
1128 {
1129 return ERROR_INV_SPS_PPS_T;
1130 }
1131
1132 /*--------------------------------------------------------------------*/
1133 /* All initializations to ps_dec are beyond this point */
1134 /*--------------------------------------------------------------------*/
1135 {
1136 WORD32 reorder_depth = ih264d_get_dpb_size(ps_seq);
1137 if((1 == ps_seq->u1_vui_parameters_present_flag) &&
1138 (1 == ps_seq->s_vui.u1_bitstream_restriction_flag))
1139 {
1140 reorder_depth = ps_seq->s_vui.u4_num_reorder_frames + 1;
1141 }
1142
1143 if (reorder_depth > H264_MAX_REF_PICS)
1144 {
1145 return ERROR_INV_SPS_PPS_T;
1146 }
1147
1148 if(ps_seq->u1_frame_mbs_only_flag != 1)
1149 reorder_depth *= 2;
1150 ps_dec->i4_reorder_depth = reorder_depth + DISPLAY_LATENCY;
1151 }
1152 ps_dec->u2_disp_height = i4_cropped_ht;
1153 ps_dec->u2_disp_width = i4_cropped_wd;
1154
1155 ps_dec->u2_pic_wd = u2_pic_wd;
1156 ps_dec->u2_pic_ht = u2_pic_ht;
1157 ps_dec->u4_total_mbs = ps_seq->u2_total_num_of_mbs << (1 - ps_seq->u1_frame_mbs_only_flag);
1158
1159 /* Determining the Width and Height of Frame from that of Picture */
1160 ps_dec->u2_frm_wd_y = u2_frm_wd_y;
1161 ps_dec->u2_frm_ht_y = u2_frm_ht_y;
1162
1163 ps_dec->u2_frm_wd_uv = u2_frm_wd_uv;
1164 ps_dec->u2_frm_ht_uv = u2_frm_ht_uv;
1165 ps_dec->s_pad_mgr.u1_pad_len_y_v = (UWORD8)(PAD_LEN_Y_V << (1 - u1_frm));
1166 ps_dec->s_pad_mgr.u1_pad_len_cr_v = (UWORD8)(PAD_LEN_UV_V << (1 - u1_frm));
1167
1168 ps_dec->u2_frm_wd_in_mbs = ps_seq->u2_frm_wd_in_mbs;
1169 ps_dec->u2_frm_ht_in_mbs = ps_seq->u2_frm_ht_in_mbs;
1170
1171 ps_dec->u2_crop_offset_y = u2_crop_offset_y;
1172 ps_dec->u2_crop_offset_uv = u2_crop_offset_uv;
1173
1174 ps_seq->u1_is_valid = TRUE;
1175 ps_dec->ps_sps[u1_seq_parameter_set_id] = *ps_seq;
1176 ps_dec->ps_cur_sps = &ps_dec->ps_sps[u1_seq_parameter_set_id];
1177
1178 return OK;
1179 }
1180
1181 /*!
1182 **************************************************************************
1183 * \if Function name : ih264d_parse_end_of_sequence \endif
1184 *
1185 * \brief
1186 * Decodes End of Sequence.
1187 *
1188 * \param ps_bitstrm : Pointer to bit ps_bitstrm containing the NAL unit
1189 *
1190 * \return
1191 * 0 on Success and error code otherwise
1192 **************************************************************************
1193 */
ih264d_parse_end_of_sequence(dec_struct_t * ps_dec)1194 WORD32 ih264d_parse_end_of_sequence(dec_struct_t * ps_dec)
1195 {
1196 WORD32 ret;
1197
1198 ret = ih264d_end_of_pic_processing(ps_dec);
1199 return ret;
1200 }
1201
1202 /*!
1203 **************************************************************************
1204 * \if Function name : AcessUnitDelimiterRbsp \endif
1205 *
1206 * \brief
1207 * Decodes AcessUnitDelimiterRbsp.
1208 *
1209 * \param ps_bitstrm : Pointer to bit ps_bitstrm containing the NAL unit
1210 *
1211 * \return
1212 * 0 on Success and error code otherwise
1213 **************************************************************************
1214 */
1215
ih264d_access_unit_delimiter_rbsp(dec_struct_t * ps_dec)1216 WORD32 ih264d_access_unit_delimiter_rbsp(dec_struct_t * ps_dec)
1217 {
1218 UWORD8 u1_primary_pic_type;
1219 u1_primary_pic_type = ih264d_get_bits_h264(ps_dec->ps_bitstrm, 3);
1220 switch(u1_primary_pic_type)
1221 {
1222 case I_PIC:
1223 case SI_PIC:
1224 case ISI_PIC:
1225 ps_dec->ps_dec_err_status->u1_pic_aud_i = PIC_TYPE_I;
1226 break;
1227 default:
1228 ps_dec->ps_dec_err_status->u1_pic_aud_i = PIC_TYPE_UNKNOWN;
1229 }
1230 return (0);
1231 }
1232 /*!
1233 **************************************************************************
1234 * \if Function name : ih264d_parse_nal_unit \endif
1235 *
1236 * \brief
1237 * Decodes NAL unit
1238 *
1239 * \return
1240 * 0 on Success and error code otherwise
1241 **************************************************************************
1242 */
1243
ih264d_parse_nal_unit(iv_obj_t * dec_hdl,ivd_video_decode_op_t * ps_dec_op,UWORD8 * pu1_buf,UWORD32 u4_length)1244 WORD32 ih264d_parse_nal_unit(iv_obj_t *dec_hdl,
1245 ivd_video_decode_op_t *ps_dec_op,
1246 UWORD8 *pu1_buf,
1247 UWORD32 u4_length)
1248 {
1249
1250 dec_bit_stream_t *ps_bitstrm;
1251
1252
1253 dec_struct_t *ps_dec = (dec_struct_t *)dec_hdl->pv_codec_handle;
1254 ivd_video_decode_ip_t *ps_dec_in =
1255 (ivd_video_decode_ip_t *)ps_dec->pv_dec_in;
1256 dec_slice_params_t * ps_cur_slice = ps_dec->ps_cur_slice;
1257 UWORD8 u1_first_byte, u1_nal_ref_idc;
1258 UWORD8 u1_nal_unit_type;
1259 WORD32 i_status = OK;
1260 ps_bitstrm = ps_dec->ps_bitstrm;
1261
1262 if(pu1_buf)
1263 {
1264 if(u4_length)
1265 {
1266 ps_dec_op->u4_frame_decoded_flag = 0;
1267 ih264d_process_nal_unit(ps_dec->ps_bitstrm, pu1_buf,
1268 u4_length);
1269
1270 SWITCHOFFTRACE;
1271 u1_first_byte = ih264d_get_bits_h264(ps_bitstrm, 8);
1272
1273 if(NAL_FORBIDDEN_BIT(u1_first_byte))
1274 {
1275 H264_DEC_DEBUG_PRINT("\nForbidden bit set in Nal Unit, Let's try\n");
1276 }
1277 u1_nal_unit_type = NAL_UNIT_TYPE(u1_first_byte);
1278 // if any other nal unit other than slice nal is encountered in between a
1279 // frame break out of loop without consuming header
1280 if ((ps_dec->u4_slice_start_code_found == 1)
1281 && (ps_dec->u1_pic_decode_done != 1)
1282 && (u1_nal_unit_type > IDR_SLICE_NAL))
1283 {
1284 return ERROR_INCOMPLETE_FRAME;
1285 }
1286 ps_dec->u1_nal_unit_type = u1_nal_unit_type;
1287 u1_nal_ref_idc = (UWORD8)(NAL_REF_IDC(u1_first_byte));
1288 //Skip all NALUs if SPS and PPS are not decoded
1289 switch(u1_nal_unit_type)
1290 {
1291 case SLICE_DATA_PARTITION_A_NAL:
1292 case SLICE_DATA_PARTITION_B_NAL:
1293 case SLICE_DATA_PARTITION_C_NAL:
1294 if(!ps_dec->i4_decode_header)
1295 ih264d_parse_slice_partition(ps_dec, ps_bitstrm);
1296
1297 break;
1298
1299 case IDR_SLICE_NAL:
1300 case SLICE_NAL:
1301
1302 /* ! */
1303 DEBUG_THREADS_PRINTF("Decoding a slice NAL\n");
1304 if(!ps_dec->i4_decode_header)
1305 {
1306 if(ps_dec->i4_header_decoded == 3)
1307 {
1308 ih264d_get_pre_sei_params(ps_dec, u1_nal_unit_type);
1309 /* ! */
1310 ps_dec->u4_slice_start_code_found = 1;
1311
1312 ih264d_rbsp_to_sodb(ps_dec->ps_bitstrm);
1313
1314 i_status = ih264d_parse_decode_slice(
1315 (UWORD8)(u1_nal_unit_type
1316 == IDR_SLICE_NAL),
1317 u1_nal_ref_idc, ps_dec);
1318
1319 if(i_status != OK)
1320 {
1321 return i_status;
1322 }
1323 }
1324 else
1325 {
1326 H264_DEC_DEBUG_PRINT(
1327 "\nSlice NAL Supplied but no header has been supplied\n");
1328 }
1329 }
1330 break;
1331
1332 case SEI_NAL:
1333 if(!ps_dec->i4_decode_header)
1334 {
1335 ih264d_rbsp_to_sodb(ps_dec->ps_bitstrm);
1336 i_status = ih264d_parse_sei_message(ps_dec, ps_bitstrm);
1337 if(i_status != OK)
1338 return i_status;
1339 ih264d_parse_sei(ps_dec, ps_bitstrm);
1340 }
1341 break;
1342 case SEQ_PARAM_NAL:
1343 /* ! */
1344 ih264d_rbsp_to_sodb(ps_dec->ps_bitstrm);
1345 i_status = ih264d_parse_sps(ps_dec, ps_bitstrm);
1346 ps_dec->u4_sps_cnt_in_process++;
1347 /*If a resolution change happens within a process call, due to multiple sps
1348 * we will not support it.
1349 */
1350 if((ps_dec->u4_sps_cnt_in_process > 1 ) &&
1351 (i_status == IVD_RES_CHANGED))
1352 {
1353 i_status = ERROR_INV_SPS_PPS_T;
1354 ps_dec->u1_res_changed = 0;
1355 }
1356 if(i_status == ERROR_INV_SPS_PPS_T)
1357 return i_status;
1358 if(!i_status)
1359 ps_dec->i4_header_decoded |= 0x1;
1360 break;
1361
1362 case PIC_PARAM_NAL:
1363 /* ! */
1364 ih264d_rbsp_to_sodb(ps_dec->ps_bitstrm);
1365 i_status = ih264d_parse_pps(ps_dec, ps_bitstrm);
1366 if(i_status == ERROR_INV_SPS_PPS_T)
1367 return i_status;
1368 if(!i_status)
1369 ps_dec->i4_header_decoded |= 0x2;
1370 break;
1371 case ACCESS_UNIT_DELIMITER_RBSP:
1372 if(!ps_dec->i4_decode_header)
1373 {
1374 ih264d_access_unit_delimiter_rbsp(ps_dec);
1375 }
1376 break;
1377 //Let us ignore the END_OF_SEQ_RBSP NAL and decode even after this NAL
1378 case END_OF_STREAM_RBSP:
1379 if(!ps_dec->i4_decode_header)
1380 {
1381 ih264d_parse_end_of_stream(ps_dec);
1382 }
1383 break;
1384 case FILLER_DATA_NAL:
1385 if(!ps_dec->i4_decode_header)
1386 {
1387 ih264d_parse_filler_data(ps_dec, ps_bitstrm);
1388 }
1389 break;
1390 default:
1391 H264_DEC_DEBUG_PRINT("\nUnknown NAL type %d\n", u1_nal_unit_type);
1392 break;
1393 }
1394
1395 }
1396
1397 }
1398
1399 return i_status;
1400
1401 }
1402