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