blob: f922cfed1d28055ae90b63e6d99d75fa89929c08 [file] [log] [blame]
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301/******************************************************************************
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_slice.c
23 *
24 * \brief
25 * Contains routines that decodes a slice NAL unit
26 *
27 * \date
28 * 19/12/2002
29 *
30 * \author AI
31 **************************************************************************
32 */
33#include <string.h>
34#include "ih264_typedefs.h"
35#include "ih264_macros.h"
36#include "ih264_platform_macros.h"
37#include "ithread.h"
38#include "ih264d_structs.h"
39#include "ih264d_debug.h"
40#include "ih264d_bitstrm.h"
41#include "ih264d_parse_mb_header.h"
42#include "ih264d_process_bslice.h"
43#include "ih264d_process_pslice.h"
44#include "ih264d_parse_cavlc.h"
45#include "ih264d_utils.h"
46#include "ih264d_deblocking.h"
47#include "ih264d_defs.h"
48#include "ih264d_error_handler.h"
49#include "ih264d_tables.h"
50#include "ih264d_defs.h"
51#include "ih264d_mem_request.h"
52#include "ih264d_parse_islice.h"
53#include "ih264d_parse_slice.h"
54#include "ih264d_mvpred.h"
55#include "ih264d_mb_utils.h"
56
57#include "ih264d_defs.h"
58#include "ih264d_quant_scaling.h"
59
60#include "ih264d_inter_pred.h"
61
62#include "ih264d_sei.h"
63#include "ih264d.h"
64#include "ih264_error.h"
65#include "ih264_disp_mgr.h"
66#include "ih264_buf_mgr.h"
67
68#include "ih264d_thread_parse_decode.h"
69#include "ih264d_thread_compute_bs.h"
70#include "ih264d_dpb_manager.h"
71#include <assert.h>
72#include "ih264d_parse_islice.h"
73#define RET_LAST_SKIP 0x80000000
74
Hamsalekha S3f6c9412017-06-01 11:44:39 +053075WORD32 check_app_out_buf_size(dec_struct_t *ps_dec);
Hamsalekha S8d3d3032015-03-13 21:24:58 +053076/*!
77 **************************************************************************
78 * \if Function name : ih264d_form_pred_weight_matrix \endif
79 *
80 * \brief
81 * Forms pred weight matrix.
82 *
83 * \return
84 * None
85 *
86 **************************************************************************
87 */
88
89void ih264d_form_pred_weight_matrix(dec_struct_t *ps_dec)
90{
91 dec_slice_params_t *ps_cur_slice;
92 UWORD8 uc_num_ref_idx_l0_active, uc_num_ref_idx_l1_active;
93 UWORD8 i, j;
94 UWORD32 *pu4_mat_iwt_ofst;
95 UWORD16 i2_idx;
96 UWORD32 *pui32_weight_offset_l0, *pui32_weight_offset_l1;
97 UWORD32 u4_temp;
98
99 ps_cur_slice = ps_dec->ps_cur_slice;
100 uc_num_ref_idx_l0_active = ps_cur_slice->u1_num_ref_idx_lx_active[0];
101 uc_num_ref_idx_l1_active = ps_cur_slice->u1_num_ref_idx_lx_active[1];
102
103 pu4_mat_iwt_ofst = ps_dec->pu4_wts_ofsts_mat;
104
105 if(ps_cur_slice->u1_slice_type == B_SLICE)
106 {
107 for(i = 0; i < uc_num_ref_idx_l0_active; i++)
108 {
109 pui32_weight_offset_l0 = ps_cur_slice->u4_wt_ofst_lx[0][i];
110 for(j = 0; j < uc_num_ref_idx_l1_active; j++)
111 {
112 pui32_weight_offset_l1 = ps_cur_slice->u4_wt_ofst_lx[1][j];
113 i2_idx = i * uc_num_ref_idx_l0_active + j;
114 i2_idx = X3(i2_idx);
115 /* u4_temp = (pui32_weight_offset_l0[0] | (pui32_weight_offset_l1[0] << 16));
116 pu4_mat_iwt_ofst[0] = u4_temp;
117 u4_temp = (pui32_weight_offset_l0[1] | (pui32_weight_offset_l1[1] << 16));
118 pu4_mat_iwt_ofst[1] = u4_temp;
119 u4_temp = (pui32_weight_offset_l0[2] | (pui32_weight_offset_l1[2] << 16));
120 pu4_mat_iwt_ofst[2] = u4_temp;
121 pu4_mat_iwt_ofst += 3;*/
122 pu4_mat_iwt_ofst[0] = pui32_weight_offset_l0[0];
123 pu4_mat_iwt_ofst[1] = pui32_weight_offset_l1[0];
124 pu4_mat_iwt_ofst[2] = pui32_weight_offset_l0[1];
125 pu4_mat_iwt_ofst[3] = pui32_weight_offset_l1[1];
126 pu4_mat_iwt_ofst[4] = pui32_weight_offset_l0[2];
127 pu4_mat_iwt_ofst[5] = pui32_weight_offset_l1[2];
128 pu4_mat_iwt_ofst += 6;
129 }
130 }
131 }
132 else
133 {
134 for(i = 0; i < uc_num_ref_idx_l0_active; i++)
135 {
136 pui32_weight_offset_l0 = ps_cur_slice->u4_wt_ofst_lx[0][i];
137 i2_idx = X3(i);
138 u4_temp = (UWORD32)pui32_weight_offset_l0[0];
139 pu4_mat_iwt_ofst[0] = u4_temp;
140 u4_temp = (UWORD32)pui32_weight_offset_l0[1];
141 pu4_mat_iwt_ofst[2] = u4_temp;
142 u4_temp = (UWORD32)pui32_weight_offset_l0[2];
143 pu4_mat_iwt_ofst[4] = u4_temp;
144 pu4_mat_iwt_ofst += 6;
145 }
146 }
147}
148
149
150/*!
151 **************************************************************************
152 * \if Function name : init_firstSliceParam \endif
153 *
154 * \brief
155 * Initialize the Parameter required for all the slices for a picture
156 *
157 * \return : Nothing
158 *
159 **************************************************************************
160 */
161
162WORD32 ih264d_start_of_pic(dec_struct_t *ps_dec,
163 WORD32 i4_poc,
164 pocstruct_t *ps_temp_poc,
165 UWORD16 u2_frame_num,
166 dec_pic_params_t *ps_pps)
167{
168 pocstruct_t *ps_prev_poc = &ps_dec->s_cur_pic_poc;
169 pocstruct_t *ps_cur_poc = ps_temp_poc;
170
171 pic_buffer_t *pic_buf;
172
173 ivd_video_decode_op_t * ps_dec_output =
174 (ivd_video_decode_op_t *)ps_dec->pv_dec_out;
175 dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;
176 dec_seq_params_t *ps_seq = ps_pps->ps_sps;
177 UWORD8 u1_bottom_field_flag = ps_cur_slice->u1_bottom_field_flag;
178 UWORD8 u1_field_pic_flag = ps_cur_slice->u1_field_pic_flag;
179 /* high profile related declarations */
180 high_profile_tools_t s_high_profile;
181 WORD32 ret;
182
183 H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);
184
Hamsalekha S3f6c9412017-06-01 11:44:39 +0530185 /* check output buffer size given by the application */
186 if(check_app_out_buf_size(ps_dec) != IV_SUCCESS)
187 return IVD_DISP_FRM_ZERO_OP_BUF_SIZE;
188
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530189 ps_prev_poc->i4_pic_order_cnt_lsb = ps_cur_poc->i4_pic_order_cnt_lsb;
190 ps_prev_poc->i4_pic_order_cnt_msb = ps_cur_poc->i4_pic_order_cnt_msb;
191 ps_prev_poc->i4_delta_pic_order_cnt_bottom =
192 ps_cur_poc->i4_delta_pic_order_cnt_bottom;
193 ps_prev_poc->i4_delta_pic_order_cnt[0] =
194 ps_cur_poc->i4_delta_pic_order_cnt[0];
195 ps_prev_poc->i4_delta_pic_order_cnt[1] =
196 ps_cur_poc->i4_delta_pic_order_cnt[1];
197 ps_prev_poc->u1_bot_field = ps_dec->ps_cur_slice->u1_bottom_field_flag;
198 ps_prev_poc->i4_prev_frame_num_ofst = ps_cur_poc->i4_prev_frame_num_ofst;
199 ps_prev_poc->u2_frame_num = u2_frame_num;
200 ps_dec->i1_prev_mb_qp_delta = 0;
201 ps_dec->i1_next_ctxt_idx = 0;
202
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530203
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700204 ps_dec->u4_nmb_deblk = 0;
205 if(ps_dec->u4_num_cores == 1)
206 ps_dec->u4_nmb_deblk = 1;
207
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530208
209
210 if(ps_seq->u1_mb_aff_flag == 1)
211 {
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700212 ps_dec->u4_nmb_deblk = 0;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530213 if(ps_dec->u4_num_cores > 2)
214 ps_dec->u4_num_cores = 2;
215 }
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700216
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530217 ps_dec->u4_use_intrapred_line_copy = 0;
218
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700219
220
221 if (ps_seq->u1_mb_aff_flag == 0)
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530222 {
223 ps_dec->u4_use_intrapred_line_copy = 1;
224 }
225
226 ps_dec->u4_app_disable_deblk_frm = 0;
227 /* If degrade is enabled, set the degrade flags appropriately */
228 if(ps_dec->i4_degrade_type && ps_dec->i4_degrade_pics)
229 {
230 WORD32 degrade_pic;
231 ps_dec->i4_degrade_pic_cnt++;
232 degrade_pic = 0;
233
234 /* If degrade is to be done in all frames, then do not check further */
235 switch(ps_dec->i4_degrade_pics)
236 {
237 case 4:
238 {
239 degrade_pic = 1;
240 break;
241 }
242 case 3:
243 {
244 if(ps_cur_slice->u1_slice_type != I_SLICE)
245 degrade_pic = 1;
246
247 break;
248 }
249 case 2:
250 {
251
252 /* If pic count hits non-degrade interval or it is an islice, then do not degrade */
253 if((ps_cur_slice->u1_slice_type != I_SLICE)
254 && (ps_dec->i4_degrade_pic_cnt
255 != ps_dec->i4_nondegrade_interval))
256 degrade_pic = 1;
257
258 break;
259 }
260 case 1:
261 {
262 /* Check if the current picture is non-ref */
263 if(0 == ps_cur_slice->u1_nal_ref_idc)
264 {
265 degrade_pic = 1;
266 }
267 break;
268 }
269
270 }
271 if(degrade_pic)
272 {
273 if(ps_dec->i4_degrade_type & 0x2)
274 ps_dec->u4_app_disable_deblk_frm = 1;
275
276 /* MC degrading is done only for non-ref pictures */
277 if(0 == ps_cur_slice->u1_nal_ref_idc)
278 {
279 if(ps_dec->i4_degrade_type & 0x4)
280 ps_dec->i4_mv_frac_mask = 0;
281
282 if(ps_dec->i4_degrade_type & 0x8)
283 ps_dec->i4_mv_frac_mask = 0;
284 }
285 }
286 else
287 ps_dec->i4_degrade_pic_cnt = 0;
288 }
289
290 {
291 dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;
Hamsalekha Sf9d3f9a2017-05-23 11:53:52 +0530292 if((ps_cur_slice->u1_slice_type == I_SLICE)
293 || (ps_cur_slice->u1_slice_type == SI_SLICE))
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530294 ps_err->u1_cur_pic_type = PIC_TYPE_I;
295 else
296 ps_err->u1_cur_pic_type = PIC_TYPE_UNKNOWN;
297
298 if(ps_err->u1_pic_aud_i == PIC_TYPE_I)
299 {
300 ps_err->u1_cur_pic_type = PIC_TYPE_I;
301 ps_err->u1_pic_aud_i = PIC_TYPE_UNKNOWN;
302 }
303
304 if(ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL)
305 {
306 if(ps_err->u1_err_flag)
307 ih264d_reset_ref_bufs(ps_dec->ps_dpb_mgr);
308 ps_err->u1_err_flag = ACCEPT_ALL_PICS;
309 }
310 }
311
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530312 if(ps_dec->u1_init_dec_flag && ps_dec->s_prev_seq_params.u1_eoseq_pending)
313 {
314 /* Reset the decoder picture buffers */
315 WORD32 j;
316 for(j = 0; j < MAX_DISP_BUFS_NEW; j++)
317 {
318
319 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
320 j,
321 BUF_MGR_REF);
322 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
323 ps_dec->au1_pic_buf_id_mv_buf_id_map[j],
324 BUF_MGR_REF);
325 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
326 j,
327 BUF_MGR_IO);
328 }
329
330 /* reset the decoder structure parameters related to buffer handling */
331 ps_dec->u1_second_field = 0;
332 ps_dec->i4_cur_display_seq = 0;
333
334 /********************************************************************/
335 /* indicate in the decoder output i4_status that some frames are being */
336 /* dropped, so that it resets timestamp and wait for a new sequence */
337 /********************************************************************/
338
339 ps_dec->s_prev_seq_params.u1_eoseq_pending = 0;
340 }
341 ret = ih264d_init_pic(ps_dec, u2_frame_num, i4_poc, ps_pps);
342 if(ret != OK)
343 return ret;
344
345 ps_dec->pv_parse_tu_coeff_data = ps_dec->pv_pic_tu_coeff_data;
346 ps_dec->pv_proc_tu_coeff_data = ps_dec->pv_pic_tu_coeff_data;
347 ps_dec->ps_nmb_info = ps_dec->ps_frm_mb_info;
348 if(ps_dec->u1_separate_parse)
349 {
Harish Mahendrakar251b0072015-08-04 09:55:15 +0530350 UWORD16 pic_wd;
351 UWORD16 pic_ht;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530352 UWORD32 num_mbs;
353
Harish Mahendrakar251b0072015-08-04 09:55:15 +0530354 pic_wd = ps_dec->u2_pic_wd;
355 pic_ht = ps_dec->u2_pic_ht;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530356 num_mbs = (pic_wd * pic_ht) >> 8;
357
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530358 if(ps_dec->pu1_dec_mb_map)
359 {
360 memset((void *)ps_dec->pu1_dec_mb_map, 0, num_mbs);
361 }
362
363 if(ps_dec->pu1_recon_mb_map)
364 {
365
366 memset((void *)ps_dec->pu1_recon_mb_map, 0, num_mbs);
367 }
368
369 if(ps_dec->pu2_slice_num_map)
370 {
371 memset((void *)ps_dec->pu2_slice_num_map, 0,
372 (num_mbs * sizeof(UWORD16)));
373 }
374
375 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530376
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700377 ps_dec->ps_parse_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);
378 ps_dec->ps_decode_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);
379 ps_dec->ps_computebs_cur_slice = &(ps_dec->ps_dec_slice_buf[0]);
Harish Mahendrakar5e4f64c2016-06-01 13:31:45 +0530380 ps_dec->u2_cur_slice_num = 0;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530381
382 /* Initialize all the HP toolsets to zero */
383 ps_dec->s_high_profile.u1_scaling_present = 0;
384 ps_dec->s_high_profile.u1_transform8x8_present = 0;
385
386 /* Get Next Free Picture */
387 if(1 == ps_dec->u4_share_disp_buf)
388 {
389 UWORD32 i;
390 /* Free any buffer that is in the queue to be freed */
391 for(i = 0; i < MAX_DISP_BUFS_NEW; i++)
392 {
393 if(0 == ps_dec->u4_disp_buf_to_be_freed[i])
394 continue;
395 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr, i,
396 BUF_MGR_IO);
397 ps_dec->u4_disp_buf_to_be_freed[i] = 0;
398 ps_dec->u4_disp_buf_mapping[i] = 0;
399
400 }
401 }
402 if(!(u1_field_pic_flag && 0 != ps_dec->u1_top_bottom_decoded)) //ps_dec->u1_second_field))
403 {
404 pic_buffer_t *ps_cur_pic;
405 WORD32 cur_pic_buf_id, cur_mv_buf_id;
406 col_mv_buf_t *ps_col_mv;
407 while(1)
408 {
409 ps_cur_pic = (pic_buffer_t *)ih264_buf_mgr_get_next_free(
410 (buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
411 &cur_pic_buf_id);
412 if(ps_cur_pic == NULL)
413 {
414 ps_dec->i4_error_code = ERROR_UNAVAIL_PICBUF_T;
415 return ERROR_UNAVAIL_PICBUF_T;
416 }
417 if(0 == ps_dec->u4_disp_buf_mapping[cur_pic_buf_id])
418 {
419 break;
420 }
421
422 }
423 ps_col_mv = (col_mv_buf_t *)ih264_buf_mgr_get_next_free((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
424 &cur_mv_buf_id);
425 if(ps_col_mv == NULL)
426 {
427 ps_dec->i4_error_code = ERROR_UNAVAIL_MVBUF_T;
428 return ERROR_UNAVAIL_MVBUF_T;
429 }
430
431 ps_dec->ps_cur_pic = ps_cur_pic;
432 ps_dec->u1_pic_buf_id = cur_pic_buf_id;
433 ps_cur_pic->u4_ts = ps_dec->u4_ts;
Chamarthi Kishoread2eaf82019-10-01 11:59:43 +0530434 memcpy(&ps_cur_pic->s_sei_pic, ps_dec->ps_sei, sizeof(sei));
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530435
436 ps_cur_pic->u1_mv_buf_id = cur_mv_buf_id;
437 ps_dec->au1_pic_buf_id_mv_buf_id_map[cur_pic_buf_id] = cur_mv_buf_id;
438
439 ps_cur_pic->pu1_col_zero_flag = (UWORD8 *)ps_col_mv->pv_col_zero_flag;
440 ps_cur_pic->ps_mv = (mv_pred_t *)ps_col_mv->pv_mv;
441 ps_dec->au1_pic_buf_ref_flag[cur_pic_buf_id] = 0;
Harish Mahendrakarf6344812016-12-15 18:04:53 +0530442
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700443 {
Hamsalekha Sb8d36252017-06-06 10:26:16 +0530444 /*make first entry of list0 and list1 point to cur pic,
445 *so that if first slice is in error, ref pic struct will have valid entries*/
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700446 ps_dec->ps_ref_pic_buf_lx[0] = ps_dec->ps_dpb_mgr->ps_init_dpb[0];
Hamsalekha Sb8d36252017-06-06 10:26:16 +0530447 ps_dec->ps_ref_pic_buf_lx[1] = ps_dec->ps_dpb_mgr->ps_init_dpb[1];
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700448 *(ps_dec->ps_dpb_mgr->ps_init_dpb[0][0]) = *ps_cur_pic;
Harish Mahendrakarbc71ad22016-09-17 17:00:37 +0530449 /* Initialize for field reference as well */
450 *(ps_dec->ps_dpb_mgr->ps_init_dpb[0][MAX_REF_BUFS]) = *ps_cur_pic;
Hamsalekha Sb8d36252017-06-06 10:26:16 +0530451
452 *(ps_dec->ps_dpb_mgr->ps_mod_dpb[0][0]) = *ps_cur_pic;
453 /* Initialize for field reference as well */
454 *(ps_dec->ps_dpb_mgr->ps_mod_dpb[0][MAX_REF_BUFS]) = *ps_cur_pic;
455 *(ps_dec->ps_dpb_mgr->ps_init_dpb[1][0]) = *ps_cur_pic;
456 /* Initialize for field reference as well */
457 *(ps_dec->ps_dpb_mgr->ps_init_dpb[1][MAX_REF_BUFS]) = *ps_cur_pic;
458 *(ps_dec->ps_dpb_mgr->ps_mod_dpb[1][0]) = *ps_cur_pic;
459 /* Initialize for field reference as well */
460 *(ps_dec->ps_dpb_mgr->ps_mod_dpb[1][MAX_REF_BUFS]) = *ps_cur_pic;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700461 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530462
463 if(!ps_dec->ps_cur_pic)
464 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530465 WORD32 j;
Martin Storsjo086dd8e2015-06-13 00:35:01 +0300466 H264_DEC_DEBUG_PRINT("------- Display Buffers Reset --------\n");
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530467 for(j = 0; j < MAX_DISP_BUFS_NEW; j++)
468 {
469
470 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
471 j,
472 BUF_MGR_REF);
473 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
474 ps_dec->au1_pic_buf_id_mv_buf_id_map[j],
475 BUF_MGR_REF);
476 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
477 j,
478 BUF_MGR_IO);
479 }
480
481 ps_dec->i4_cur_display_seq = 0;
482 ps_dec->i4_prev_max_display_seq = 0;
483 ps_dec->i4_max_poc = 0;
484
485 ps_cur_pic = (pic_buffer_t *)ih264_buf_mgr_get_next_free(
486 (buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
487 &cur_pic_buf_id);
488 if(ps_cur_pic == NULL)
489 {
490 ps_dec->i4_error_code = ERROR_UNAVAIL_PICBUF_T;
491 return ERROR_UNAVAIL_PICBUF_T;
492 }
493
494 ps_col_mv = (col_mv_buf_t *)ih264_buf_mgr_get_next_free((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
495 &cur_mv_buf_id);
496 if(ps_col_mv == NULL)
497 {
498 ps_dec->i4_error_code = ERROR_UNAVAIL_MVBUF_T;
499 return ERROR_UNAVAIL_MVBUF_T;
500 }
501
502 ps_dec->ps_cur_pic = ps_cur_pic;
503 ps_dec->u1_pic_buf_id = cur_pic_buf_id;
504 ps_cur_pic->u4_ts = ps_dec->u4_ts;
505 ps_dec->apv_buf_id_pic_buf_map[cur_pic_buf_id] = (void *)ps_cur_pic;
506
507 ps_cur_pic->u1_mv_buf_id = cur_mv_buf_id;
508 ps_dec->au1_pic_buf_id_mv_buf_id_map[cur_pic_buf_id] = cur_mv_buf_id;
509
510 ps_cur_pic->pu1_col_zero_flag = (UWORD8 *)ps_col_mv->pv_col_zero_flag;
511 ps_cur_pic->ps_mv = (mv_pred_t *)ps_col_mv->pv_mv;
512 ps_dec->au1_pic_buf_ref_flag[cur_pic_buf_id] = 0;
513
514 }
515
516 ps_dec->ps_cur_pic->u1_picturetype = u1_field_pic_flag;
517 ps_dec->ps_cur_pic->u4_pack_slc_typ = SKIP_NONE;
518 H264_DEC_DEBUG_PRINT("got a buffer\n");
519 }
520 else
521 {
522 H264_DEC_DEBUG_PRINT("did not get a buffer\n");
523 }
524
525 ps_dec->u4_pic_buf_got = 1;
526
527 ps_dec->ps_cur_pic->i4_poc = i4_poc;
528 ps_dec->ps_cur_pic->i4_frame_num = u2_frame_num;
529 ps_dec->ps_cur_pic->i4_pic_num = u2_frame_num;
530 ps_dec->ps_cur_pic->i4_top_field_order_cnt = ps_pps->i4_top_field_order_cnt;
531 ps_dec->ps_cur_pic->i4_bottom_field_order_cnt =
532 ps_pps->i4_bottom_field_order_cnt;
533 ps_dec->ps_cur_pic->i4_avg_poc = ps_pps->i4_avg_poc;
534 ps_dec->ps_cur_pic->u4_time_stamp = ps_dec->u4_pts;
535
536 ps_dec->s_cur_pic = *(ps_dec->ps_cur_pic);
537 if(u1_field_pic_flag && u1_bottom_field_flag)
538 {
539 WORD32 i4_temp_poc;
540 WORD32 i4_top_field_order_poc, i4_bot_field_order_poc;
541 /* Point to odd lines, since it's bottom field */
542 ps_dec->s_cur_pic.pu1_buf1 += ps_dec->s_cur_pic.u2_frm_wd_y;
543 ps_dec->s_cur_pic.pu1_buf2 += ps_dec->s_cur_pic.u2_frm_wd_uv;
544 ps_dec->s_cur_pic.pu1_buf3 += ps_dec->s_cur_pic.u2_frm_wd_uv;
545 ps_dec->s_cur_pic.ps_mv +=
546 ((ps_dec->u2_pic_ht * ps_dec->u2_pic_wd) >> 5);
547 ps_dec->s_cur_pic.pu1_col_zero_flag += ((ps_dec->u2_pic_ht
548 * ps_dec->u2_pic_wd) >> 5);
549 ps_dec->ps_cur_pic->u1_picturetype |= BOT_FLD;
550 i4_top_field_order_poc = ps_dec->ps_cur_pic->i4_top_field_order_cnt;
551 i4_bot_field_order_poc = ps_dec->ps_cur_pic->i4_bottom_field_order_cnt;
552 i4_temp_poc = MIN(i4_top_field_order_poc,
553 i4_bot_field_order_poc);
554 ps_dec->ps_cur_pic->i4_avg_poc = i4_temp_poc;
555 }
556
557 ps_cur_slice->u1_mbaff_frame_flag = ps_seq->u1_mb_aff_flag
558 && (!u1_field_pic_flag);
559
560 ps_dec->ps_cur_pic->u1_picturetype |= (ps_cur_slice->u1_mbaff_frame_flag
561 << 2);
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530562
563 ps_dec->ps_cur_mb_row = ps_dec->ps_nbr_mb_row; //[0];
Harish Mahendrakar251b0072015-08-04 09:55:15 +0530564 //Increment by 2 ,so that left mb (mbaff decrements by 2) will always be valid
565 ps_dec->ps_cur_mb_row += 2;
566 ps_dec->ps_top_mb_row = ps_dec->ps_nbr_mb_row;
567 ps_dec->ps_top_mb_row += ((ps_dec->u2_frm_wd_in_mbs + 2) << (1 - ps_dec->ps_cur_sps->u1_frame_mbs_only_flag));
568 //Increment by 2 ,so that left mb (mbaff decrements by 2) will always be valid
569 ps_dec->ps_top_mb_row += 2;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530570
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530571 /* CHANGED CODE */
572 ps_dec->ps_mv_cur = ps_dec->s_cur_pic.ps_mv;
573 ps_dec->ps_mv_top = ps_dec->ps_mv_top_p[0];
574 /* CHANGED CODE */
575 ps_dec->u1_mv_top_p = 0;
576 ps_dec->u1_mb_idx = 0;
577 /* CHANGED CODE */
578 ps_dec->ps_mv_left = ps_dec->s_cur_pic.ps_mv;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530579 ps_dec->u2_total_mbs_coded = 0;
580 ps_dec->i4_submb_ofst = -(SUB_BLK_SIZE);
581 ps_dec->u4_pred_info_idx = 0;
582 ps_dec->u4_pred_info_pkd_idx = 0;
583 ps_dec->u4_dma_buf_idx = 0;
584 ps_dec->ps_mv = ps_dec->s_cur_pic.ps_mv;
585 ps_dec->ps_mv_bank_cur = ps_dec->s_cur_pic.ps_mv;
586 ps_dec->pu1_col_zero_flag = ps_dec->s_cur_pic.pu1_col_zero_flag;
587 ps_dec->ps_part = ps_dec->ps_parse_part_params;
588 ps_dec->i2_prev_slice_mbx = -1;
589 ps_dec->i2_prev_slice_mby = 0;
590 ps_dec->u2_mv_2mb[0] = 0;
591 ps_dec->u2_mv_2mb[1] = 0;
592 ps_dec->u1_last_pic_not_decoded = 0;
593
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530594 ps_dec->u2_cur_slice_num_dec_thread = 0;
595 ps_dec->u2_cur_slice_num_bs = 0;
596 ps_dec->u4_intra_pred_line_ofst = 0;
597 ps_dec->pu1_cur_y_intra_pred_line = ps_dec->pu1_y_intra_pred_line;
598 ps_dec->pu1_cur_u_intra_pred_line = ps_dec->pu1_u_intra_pred_line;
599 ps_dec->pu1_cur_v_intra_pred_line = ps_dec->pu1_v_intra_pred_line;
600
601 ps_dec->pu1_cur_y_intra_pred_line_base = ps_dec->pu1_y_intra_pred_line;
602 ps_dec->pu1_cur_u_intra_pred_line_base = ps_dec->pu1_u_intra_pred_line;
603 ps_dec->pu1_cur_v_intra_pred_line_base = ps_dec->pu1_v_intra_pred_line;
604
605
606
607
608
609 ps_dec->pu1_prev_y_intra_pred_line = ps_dec->pu1_y_intra_pred_line
610 + (ps_dec->u2_frm_wd_in_mbs * MB_SIZE);
611
612 ps_dec->pu1_prev_u_intra_pred_line = ps_dec->pu1_u_intra_pred_line
613 + ps_dec->u2_frm_wd_in_mbs * BLK8x8SIZE * YUV420SP_FACTOR;
614 ps_dec->pu1_prev_v_intra_pred_line = ps_dec->pu1_v_intra_pred_line
615 + ps_dec->u2_frm_wd_in_mbs * BLK8x8SIZE;
616
617 ps_dec->ps_deblk_mbn = ps_dec->ps_deblk_pic;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530618 /* Initialize The Function Pointer Depending Upon the Entropy and MbAff Flag */
619 {
620 if(ps_cur_slice->u1_mbaff_frame_flag)
621 {
622 ps_dec->pf_compute_bs = ih264d_compute_bs_mbaff;
623 ps_dec->pf_mvpred = ih264d_mvpred_mbaff;
624 }
625 else
626 {
627 ps_dec->pf_compute_bs = ih264d_compute_bs_non_mbaff;
628 ps_dec->u1_cur_mb_fld_dec_flag = ps_cur_slice->u1_field_pic_flag;
629 }
630 }
631 /* Set up the Parameter for DMA transfer */
632 {
633 UWORD8 u1_field_pic_flag = ps_dec->ps_cur_slice->u1_field_pic_flag;
634
635 UWORD8 u1_mbaff = ps_cur_slice->u1_mbaff_frame_flag;
636
637 UWORD8 uc_lastmbs = (((ps_dec->u2_pic_wd) >> 4)
638 % (ps_dec->u1_recon_mb_grp >> u1_mbaff));
639 UWORD16 ui16_lastmbs_widthY =
640 (uc_lastmbs ? (uc_lastmbs << 4) : ((ps_dec->u1_recon_mb_grp
641 >> u1_mbaff) << 4));
642 UWORD16 ui16_lastmbs_widthUV =
643 uc_lastmbs ? (uc_lastmbs << 3) : ((ps_dec->u1_recon_mb_grp
644 >> u1_mbaff) << 3);
645
646 ps_dec->s_tran_addrecon.pu1_dest_y = ps_dec->s_cur_pic.pu1_buf1;
647 ps_dec->s_tran_addrecon.pu1_dest_u = ps_dec->s_cur_pic.pu1_buf2;
648 ps_dec->s_tran_addrecon.pu1_dest_v = ps_dec->s_cur_pic.pu1_buf3;
649
650 ps_dec->s_tran_addrecon.u2_frm_wd_y = ps_dec->u2_frm_wd_y
651 << u1_field_pic_flag;
652 ps_dec->s_tran_addrecon.u2_frm_wd_uv = ps_dec->u2_frm_wd_uv
653 << u1_field_pic_flag;
654
655 if(u1_field_pic_flag)
656 {
657 ui16_lastmbs_widthY += ps_dec->u2_frm_wd_y;
658 ui16_lastmbs_widthUV += ps_dec->u2_frm_wd_uv;
659 }
660
661 /* Normal Increment of Pointer */
662 ps_dec->s_tran_addrecon.u4_inc_y[0] = ((ps_dec->u1_recon_mb_grp << 4)
663 >> u1_mbaff);
664 ps_dec->s_tran_addrecon.u4_inc_uv[0] = ((ps_dec->u1_recon_mb_grp << 4)
665 >> u1_mbaff);
666
667 /* End of Row Increment */
668 ps_dec->s_tran_addrecon.u4_inc_y[1] = (ui16_lastmbs_widthY
669 + (PAD_LEN_Y_H << 1)
670 + ps_dec->s_tran_addrecon.u2_frm_wd_y
671 * ((15 << u1_mbaff) + u1_mbaff));
672 ps_dec->s_tran_addrecon.u4_inc_uv[1] = (ui16_lastmbs_widthUV
673 + (PAD_LEN_UV_H << 2)
674 + ps_dec->s_tran_addrecon.u2_frm_wd_uv
675 * ((15 << u1_mbaff) + u1_mbaff));
676
677 /* Assign picture numbers to each frame/field */
678 /* only once per picture. */
679 ih264d_assign_pic_num(ps_dec);
680 ps_dec->s_tran_addrecon.u2_mv_top_left_inc = (ps_dec->u1_recon_mb_grp
681 << 2) - 1 - (u1_mbaff << 2);
682 ps_dec->s_tran_addrecon.u2_mv_left_inc = ((ps_dec->u1_recon_mb_grp
683 >> u1_mbaff) - 1) << (4 + u1_mbaff);
684 }
685 /**********************************************************************/
686 /* High profile related initialization at pictrue level */
687 /**********************************************************************/
688 if(ps_seq->u1_profile_idc == HIGH_PROFILE_IDC)
689 {
690 if((ps_seq->i4_seq_scaling_matrix_present_flag)
691 || (ps_pps->i4_pic_scaling_matrix_present_flag))
692 {
S Hamsalekhaf28de032018-12-27 13:57:55 +0530693 ret = ih264d_form_scaling_matrix_picture(ps_seq, ps_pps, ps_dec);
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530694 ps_dec->s_high_profile.u1_scaling_present = 1;
695 }
696 else
697 {
S Hamsalekhaf28de032018-12-27 13:57:55 +0530698 ret = ih264d_form_default_scaling_matrix(ps_dec);
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530699 }
700
701 if(ps_pps->i4_transform_8x8_mode_flag)
702 {
703 ps_dec->s_high_profile.u1_transform8x8_present = 1;
704 }
705 }
706 else
707 {
S Hamsalekhaf28de032018-12-27 13:57:55 +0530708 ret = ih264d_form_default_scaling_matrix(ps_dec);
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530709 }
S Hamsalekhaf28de032018-12-27 13:57:55 +0530710
711 if(ret != OK)
712 return ret;
713
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530714 /* required while reading the transform_size_8x8 u4_flag */
715 ps_dec->s_high_profile.u1_direct_8x8_inference_flag =
716 ps_seq->u1_direct_8x8_inference_flag;
717 ps_dec->s_high_profile.s_cavlc_ctxt = ps_dec->s_cavlc_ctxt;
718
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700719 ps_dec->i1_recon_in_thread3_flag = 1;
720 ps_dec->ps_frame_buf_ip_recon = &ps_dec->s_tran_addrecon;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530721 if(ps_dec->u1_separate_parse)
722 {
723 memcpy(&ps_dec->s_tran_addrecon_parse, &ps_dec->s_tran_addrecon,
724 sizeof(tfr_ctxt_t));
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700725 if(ps_dec->u4_num_cores >= 3 && ps_dec->i1_recon_in_thread3_flag)
726 {
727 memcpy(&ps_dec->s_tran_iprecon, &ps_dec->s_tran_addrecon,
728 sizeof(tfr_ctxt_t));
729 ps_dec->ps_frame_buf_ip_recon = &ps_dec->s_tran_iprecon;
730 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530731 }
732
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700733
734 ih264d_init_deblk_tfr_ctxt(ps_dec,&(ps_dec->s_pad_mgr), &(ps_dec->s_tran_addrecon),
735 ps_dec->u2_frm_wd_in_mbs, 0);
736
737 ps_dec->ps_cur_deblk_mb = ps_dec->ps_deblk_pic;
738 ps_dec->u4_cur_deblk_mb_num = 0;
739
740 ps_dec->u4_deblk_mb_x = 0;
741 ps_dec->u4_deblk_mb_y = 0;
Harish Mahendrakar33db7a02016-04-20 16:13:52 +0530742 ps_dec->pu4_wt_ofsts = ps_dec->pu4_wts_ofsts_mat;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -0700743
Harish Mahendrakar0b23c812017-01-16 14:43:42 +0530744 ps_dec->u4_first_slice_in_pic = 0;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530745 H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);
746 return OK;
747}
748
749/*!
750 **************************************************************************
751 * \if Function name : ih264d_deblock_display \endif
752 *
753 * \brief : The function callls the deblocking routine and manages
754 : the Recon buffers and displays .
755 * \return : Nothing
756 *
757 **************************************************************************
758 */
759WORD32 ih264d_end_of_pic_dispbuf_mgr(dec_struct_t * ps_dec)
760{
761 dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;
762 UWORD8 u1_num_of_users = 0;
763 WORD32 ret;
764
765 H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);
766 if(1)
767 {
768
769 {
770 ih264d_delete_nonref_nondisplay_pics(ps_dec->ps_dpb_mgr);
771 if(ps_cur_slice->u1_mmco_equalto5
772 || (ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL))
773 {
774 ps_dec->ps_cur_pic->i4_poc = 0;
775 if(ps_dec->u2_total_mbs_coded
776 == (ps_dec->ps_cur_sps->u2_max_mb_addr + 1))
777 ih264d_reset_ref_bufs(ps_dec->ps_dpb_mgr);
778 ih264d_release_display_bufs(ps_dec);
779 }
Harish Mahendrakar251b0072015-08-04 09:55:15 +0530780 if(IVD_DECODE_FRAME_OUT != ps_dec->e_frm_out_mode)
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530781 {
782 ret = ih264d_assign_display_seq(ps_dec);
783 if(ret != OK)
784 return ret;
785 }
786 }
787
788 if(ps_cur_slice->u1_nal_ref_idc)
789 {
790 /* Mark pic buf as needed for reference */
791 ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
792 ps_dec->u1_pic_buf_id,
793 BUF_MGR_REF);
794 /* Mark mv buf as needed for reference */
795 ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
796 ps_dec->au1_pic_buf_id_mv_buf_id_map[ps_dec->u1_pic_buf_id],
797 BUF_MGR_REF);
798 ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] = 1;
799 }
800
801 /* 420 consumer */
802 /* Increment the number of users by 1 for display based upon */
803 /*the SEEK KEY FRAME control sent to decoder */
804 if(((0 == ps_dec->u1_last_pic_not_decoded)
805 && (0
806 == (ps_dec->ps_cur_pic->u4_pack_slc_typ
807 & ps_dec->u4_skip_frm_mask)))
808 || (ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL))
809 {
810 /* Mark pic buf as needed for display */
811 ih264_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
812 ps_dec->u1_pic_buf_id,
813 BUF_MGR_IO);
814
815 }
816
817 if(!ps_cur_slice->u1_field_pic_flag
818 || ((TOP_FIELD_ONLY | BOT_FIELD_ONLY)
819 != ps_dec->u1_top_bottom_decoded))
820 {
821 pic_buffer_t *ps_cur_pic = ps_dec->ps_cur_pic;
822 ps_cur_pic->u2_disp_width = ps_dec->u2_disp_width;
823 ps_cur_pic->u2_disp_height = ps_dec->u2_disp_height >> 1;
824
825 ps_cur_pic->u2_crop_offset_y = ps_dec->u2_crop_offset_y;
826 ps_cur_pic->u2_crop_offset_uv = ps_dec->u2_crop_offset_uv;
827 ps_cur_pic->u1_pic_type = 0;
Harish Mahendrakar1d672d22019-07-03 10:12:53 -0700828 {
829 UWORD64 i8_display_poc;
830 i8_display_poc = (UWORD64)ps_dec->i4_prev_max_display_seq +
831 ps_dec->ps_cur_pic->i4_poc;
832 if(IS_OUT_OF_RANGE_S32(i8_display_poc))
833 {
834 ps_dec->i4_prev_max_display_seq = 0;
835 }
836 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530837 ret = ih264d_insert_pic_in_display_list(
838 ps_dec->ps_dpb_mgr,
839 ps_dec->u1_pic_buf_id,
840 ps_dec->i4_prev_max_display_seq
841 + ps_dec->ps_cur_pic->i4_poc,
842 ps_dec->ps_cur_pic->i4_frame_num);
843 if(ret != OK)
844 return ret;
845
Harish Mahendrakare0ad0012020-04-02 05:17:42 +0530846 {
847 ivd_video_decode_op_t * ps_dec_output =
848 (ivd_video_decode_op_t *)ps_dec->pv_dec_out;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530849
Harish Mahendrakare0ad0012020-04-02 05:17:42 +0530850 ps_dec_output->u4_frame_decoded_flag = 1;
851 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530852 if(ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] == 0)
853 {
854 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
855 ps_dec->au1_pic_buf_id_mv_buf_id_map[ps_dec->u1_pic_buf_id],
856 BUF_MGR_REF);
857 ps_dec->au1_pic_buf_ref_flag[ps_dec->u1_pic_buf_id] = 0;
858
859 }
860 }
861 else
862 {
863 H264_DEC_DEBUG_PRINT("pic not inserted display %d %d\n",
864 ps_cur_slice->u1_field_pic_flag,
865 ps_dec->u1_second_field);
866 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530867
868 if(!ps_cur_slice->u1_field_pic_flag
869 || ((TOP_FIELD_ONLY | BOT_FIELD_ONLY)
870 == ps_dec->u1_top_bottom_decoded))
871 {
Harish Mahendrakar251b0072015-08-04 09:55:15 +0530872 if(IVD_DECODE_FRAME_OUT == ps_dec->e_frm_out_mode)
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530873 {
874 ret = ih264d_assign_display_seq(ps_dec);
875 if(ret != OK)
876 return ret;
877 }
878 }
879 }
880
881 H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);
882
883 return OK;
884}
885
886void ih264d_err_pic_dispbuf_mgr(dec_struct_t *ps_dec)
887{
888 dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;
889 ivd_video_decode_op_t * ps_dec_output =
890 (ivd_video_decode_op_t *)ps_dec->pv_dec_out;
891
892 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
893 ps_dec->u1_pic_buf_id,
894 BUF_MGR_REF);
895 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_mv_buf_mgr,
896 ps_dec->au1_pic_buf_id_mv_buf_id_map[ps_dec->u1_pic_buf_id],
897 BUF_MGR_REF);
898 ih264_buf_mgr_release((buf_mgr_t *)ps_dec->pv_pic_buf_mgr,
899 ps_dec->u1_pic_buf_id,
900 BUF_MGR_IO);
901}
902
903void ih264d_deblock_picture(void *ptr)
904{
905 dec_struct_t *ps_dec = (dec_struct_t *)ptr;
906
907 {
908 /*Deblock picture only if all the mb's in the frame have been decoded*/
909 if(ps_dec->u1_pic_decode_done == 1)
910 {
911 if(ps_dec->ps_cur_slice->u1_mbaff_frame_flag
912 || ps_dec->ps_cur_slice->u1_field_pic_flag)
913 {
914 ps_dec->p_DeblockPicture[ps_dec->ps_cur_slice->u1_mbaff_frame_flag](
915 ps_dec);
916 }
917 else
918
919 {
920
921 ih264d_deblock_picture_progressive(ps_dec);
922 }
923
924 }
925 }
926
927}
928
929/*!
930 **************************************************************************
931 * \if Function name : ih264d_deblock_display \endif
932 *
933 * \brief : The function callls the deblocking routine and manages
934 : the Recon buffers and displays .
935 * \return : Nothing
936 *
937 **************************************************************************
938 */
939WORD32 ih264d_deblock_display(dec_struct_t *ps_dec)
940{
941 WORD32 ret;
942 /* Call deblocking */
943 ih264d_deblock_picture(ps_dec);
944
945 ret = ih264d_end_of_pic_dispbuf_mgr(ps_dec);
946 if(ret != OK)
947 return ret;
948
949 return OK;
950}
951
952/*
953 *!
954 **************************************************************************
955 * \if Function name : EndofPoc \endif
956 *
957 * \brief
958 * EndofPoc Processing
959 *
960 * \return
961 * 0 on Success and Error code otherwise
962 **************************************************************************
963 */
964
Harish Mahendrakar49456122017-01-05 12:10:47 +0530965WORD32 ih264d_end_of_pic(dec_struct_t *ps_dec)
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530966{
967 dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;
968 WORD32 ret;
969
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530970 {
971 dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;
972 if(ps_err->u1_err_flag & REJECT_CUR_PIC)
973 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530974 ih264d_err_pic_dispbuf_mgr(ps_dec);
Hamsalekha S9f0dcba2015-06-09 15:54:31 +0530975 return ERROR_NEW_FRAME_EXPECTED;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530976 }
977 }
978
979 H264_MUTEX_LOCK(&ps_dec->process_disp_mutex);
980 ret = ih264d_end_of_pic_processing(ps_dec);
981 if(ret != OK)
982 return ret;
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530983 /*--------------------------------------------------------------------*/
984 /* ih264d_decode_pic_order_cnt - calculate the Pic Order Cnt */
985 /* Needed to detect end of picture */
986 /*--------------------------------------------------------------------*/
Hamsalekha S9f0dcba2015-06-09 15:54:31 +0530987
988 H264_MUTEX_UNLOCK(&ps_dec->process_disp_mutex);
Hamsalekha S8d3d3032015-03-13 21:24:58 +0530989
990 return OK;
991}
992
Rakesh Kumarf312a1d2018-07-09 21:58:44 +0530993
994/*!
995 **************************************************************************
996 * \if Function name : ih264d_fix_error_in_dpb \endif
997 *
998 * \brief
999 * fix error in DPB
1000 *
1001 * \return
1002 * Number of node(s) deleted
1003 **************************************************************************
1004 */
1005
1006WORD32 ih264d_fix_error_in_dpb(dec_struct_t *ps_dec)
1007{
1008 /*--------------------------------------------------------------------*/
1009 /* If there is common node in lt_list and st_list then delete it from */
1010 /* st_list */
1011 /*--------------------------------------------------------------------*/
1012 UWORD8 no_of_nodes_deleted = 0;
1013 UWORD8 lt_ref_num = ps_dec->ps_dpb_mgr->u1_num_lt_ref_bufs;
1014 struct dpb_info_t *ps_lt_curr_dpb = ps_dec->ps_dpb_mgr->ps_dpb_ht_head;
1015 while(lt_ref_num && ps_lt_curr_dpb)
1016 {
1017 if(ps_dec->ps_dpb_mgr->ps_dpb_st_head
1018 && ((ps_lt_curr_dpb->s_bot_field.u1_reference_info
1019 | ps_lt_curr_dpb->s_top_field.u1_reference_info)
1020 == (IS_SHORT_TERM | IS_LONG_TERM)))
1021 {
1022 struct dpb_info_t *ps_st_next_dpb = ps_dec->ps_dpb_mgr->ps_dpb_st_head;
1023 struct dpb_info_t *ps_st_curr_dpb = ps_dec->ps_dpb_mgr->ps_dpb_st_head;
1024 UWORD8 st_ref_num = ps_dec->ps_dpb_mgr->u1_num_st_ref_bufs;
1025 while(st_ref_num && ps_st_curr_dpb)
1026 {
1027 if(ps_st_curr_dpb == ps_lt_curr_dpb)
1028 {
1029 if(st_ref_num == ps_dec->ps_dpb_mgr->u1_num_st_ref_bufs)
1030 {
1031 ps_dec->ps_dpb_mgr->ps_dpb_st_head =
1032 ps_dec->ps_dpb_mgr->ps_dpb_st_head->ps_prev_short;
1033 ps_st_curr_dpb = ps_dec->ps_dpb_mgr->ps_dpb_st_head;
1034 }
1035 else
1036 {
1037 ps_st_next_dpb->ps_prev_short = ps_st_curr_dpb->ps_prev_short;
1038 }
1039 ps_dec->ps_dpb_mgr->u1_num_st_ref_bufs--;
Rakesh Kumarf312a1d2018-07-09 21:58:44 +05301040 no_of_nodes_deleted++;
1041 break;
1042 }
1043 ps_st_next_dpb = ps_st_curr_dpb;
1044 ps_st_curr_dpb = ps_st_curr_dpb->ps_prev_short;
1045 st_ref_num--;
1046 }
1047 }
1048 ps_lt_curr_dpb = ps_lt_curr_dpb->ps_prev_long;
1049 lt_ref_num--;
1050 }
1051 return no_of_nodes_deleted;
1052}
1053
1054
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301055/*!
1056 **************************************************************************
1057 * \if Function name : DecodeSlice \endif
1058 *
1059 * \brief
1060 * Parses a slice
1061 *
1062 * \return
1063 * 0 on Success and Error code otherwise
1064 **************************************************************************
1065 */
1066
1067WORD32 ih264d_parse_decode_slice(UWORD8 u1_is_idr_slice,
1068 UWORD8 u1_nal_ref_idc,
1069 dec_struct_t *ps_dec /* Decoder parameters */
1070 )
1071{
1072 dec_bit_stream_t * ps_bitstrm = ps_dec->ps_bitstrm;
1073 dec_pic_params_t *ps_pps;
1074 dec_seq_params_t *ps_seq;
1075 dec_slice_params_t *ps_cur_slice = ps_dec->ps_cur_slice;
Thomala Srinivas2286dbb2018-10-09 18:29:54 +05301076 pocstruct_t s_tmp_poc = {0};
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301077 WORD32 i_delta_poc[2];
1078 WORD32 i4_poc = 0;
1079 UWORD16 u2_first_mb_in_slice, u2_frame_num;
1080 UWORD8 u1_field_pic_flag, u1_redundant_pic_cnt = 0, u1_slice_type;
1081 UWORD32 u4_idr_pic_id = 0;
1082 UWORD8 u1_bottom_field_flag, u1_pic_order_cnt_type;
1083
1084 UWORD8 u1_nal_unit_type;
1085 UWORD32 *pu4_bitstrm_buf = ps_bitstrm->pu4_buffer;
1086 UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst;
1087 WORD8 i1_is_end_of_poc;
1088
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001089 WORD32 ret, end_of_frame;
1090 WORD32 prev_slice_err, num_mb_skipped;
1091 UWORD8 u1_mbaff;
1092 pocstruct_t *ps_cur_poc;
1093
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301094 UWORD32 u4_temp;
1095 WORD32 i_temp;
1096 UWORD32 u4_call_end_of_pic = 0;
1097
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301098 /* read FirstMbInSlice and slice type*/
1099 ps_dec->ps_dpb_cmds->u1_dpb_commands_read_slc = 0;
1100 u2_first_mb_in_slice = ih264d_uev(pu4_bitstrm_ofst,
1101 pu4_bitstrm_buf);
1102 if(u2_first_mb_in_slice
Harish Mahendrakar7e069402020-12-02 11:54:47 -08001103 >= (ps_dec->u2_frm_ht_in_mbs * ps_dec->u2_frm_wd_in_mbs))
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301104 {
1105
1106 return ERROR_CORRUPTED_SLICE;
1107 }
1108
1109 /*we currently don not support ASO*/
1110 if(((u2_first_mb_in_slice << ps_cur_slice->u1_mbaff_frame_flag)
Harish Mahendrakaref274332017-01-16 17:10:07 +05301111 <= ps_dec->u2_cur_mb_addr) && (ps_dec->u4_first_slice_in_pic == 0))
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301112 {
1113 return ERROR_CORRUPTED_SLICE;
1114 }
1115
1116 COPYTHECONTEXT("SH: first_mb_in_slice",u2_first_mb_in_slice);
1117
1118 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
1119
1120 if(u4_temp > 9)
1121 return ERROR_INV_SLC_TYPE_T;
1122
1123 u1_slice_type = u4_temp;
1124 COPYTHECONTEXT("SH: slice_type",(u1_slice_type));
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301125 /* Find Out the Slice Type is 5 to 9 or not then Set the Flag */
1126 /* u1_sl_typ_5_9 = 1 .Which tells that all the slices in the Pic*/
1127 /* will be of same type of current */
1128 if(u1_slice_type > 4)
1129 {
1130 u1_slice_type -= 5;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301131 }
1132
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301133 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
1134 if(u4_temp & MASK_ERR_PIC_SET_ID)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301135 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301136 /* discard slice if pic param is invalid */
1137 COPYTHECONTEXT("SH: pic_parameter_set_id", u4_temp);
1138 ps_pps = &ps_dec->ps_pps[u4_temp];
1139 if(FALSE == ps_pps->u1_is_valid)
1140 {
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301141 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301142 }
1143 ps_seq = ps_pps->ps_sps;
1144 if(!ps_seq)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301145 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301146 if(FALSE == ps_seq->u1_is_valid)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301147 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301148
1149 /* Get the frame num */
1150 u2_frame_num = ih264d_get_bits_h264(ps_bitstrm,
1151 ps_seq->u1_bits_in_frm_num);
1152// H264_DEC_DEBUG_PRINT("FRAME %d First MB in slice: %d\n", u2_frame_num, u2_first_mb_in_slice);
1153
1154 COPYTHECONTEXT("SH: frame_num", u2_frame_num);
1155// H264_DEC_DEBUG_PRINT("Second field: %d frame num: %d prv_frame_num: %d \n", ps_dec->u1_second_field, u2_frame_num, ps_dec->u2_prv_frame_num);
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301156 if(!ps_dec->u1_first_slice_in_stream && ps_dec->u4_first_slice_in_pic)
Harish Mahendrakar49456122017-01-05 12:10:47 +05301157 {
1158 pocstruct_t *ps_prev_poc = &ps_dec->s_prev_pic_poc;
1159 pocstruct_t *ps_cur_poc = &ps_dec->s_cur_pic_poc;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301160
Harish Mahendrakar49456122017-01-05 12:10:47 +05301161 ps_dec->u2_mbx = 0xffff;
1162 ps_dec->u2_mby = 0;
1163
1164 if((0 == u1_is_idr_slice) && ps_cur_slice->u1_nal_ref_idc)
1165 ps_dec->u2_prev_ref_frame_num = ps_cur_slice->u2_frame_num;
1166
1167 if(u1_is_idr_slice || ps_cur_slice->u1_mmco_equalto5)
1168 ps_dec->u2_prev_ref_frame_num = 0;
1169
1170 if(ps_dec->ps_cur_sps->u1_gaps_in_frame_num_value_allowed_flag)
1171 {
1172 ih264d_decode_gaps_in_frame_num(ps_dec, u2_frame_num);
1173 }
1174
1175 ps_prev_poc->i4_prev_frame_num_ofst = ps_cur_poc->i4_prev_frame_num_ofst;
1176 ps_prev_poc->u2_frame_num = ps_cur_poc->u2_frame_num;
1177 ps_prev_poc->u1_mmco_equalto5 = ps_cur_slice->u1_mmco_equalto5;
1178 if(ps_cur_slice->u1_nal_ref_idc)
1179 {
1180 ps_prev_poc->i4_pic_order_cnt_lsb = ps_cur_poc->i4_pic_order_cnt_lsb;
1181 ps_prev_poc->i4_pic_order_cnt_msb = ps_cur_poc->i4_pic_order_cnt_msb;
1182 ps_prev_poc->i4_delta_pic_order_cnt_bottom =
1183 ps_cur_poc->i4_delta_pic_order_cnt_bottom;
1184 ps_prev_poc->i4_delta_pic_order_cnt[0] =
1185 ps_cur_poc->i4_delta_pic_order_cnt[0];
1186 ps_prev_poc->i4_delta_pic_order_cnt[1] =
1187 ps_cur_poc->i4_delta_pic_order_cnt[1];
1188 ps_prev_poc->u1_bot_field = ps_cur_poc->u1_bot_field;
1189 }
1190
1191 ps_dec->u2_total_mbs_coded = 0;
1192 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301193 /* Get the field related flags */
1194 if(!ps_seq->u1_frame_mbs_only_flag)
1195 {
1196
1197 u1_field_pic_flag = ih264d_get_bit_h264(ps_bitstrm);
1198 COPYTHECONTEXT("SH: field_pic_flag", u1_field_pic_flag);
1199 u1_bottom_field_flag = 0;
1200
1201 if(u1_field_pic_flag)
1202 {
1203 ps_dec->pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan_fld;
1204 u1_bottom_field_flag = ih264d_get_bit_h264(ps_bitstrm);
1205 COPYTHECONTEXT("SH: bottom_field_flag", u1_bottom_field_flag);
1206
1207 }
1208 else
1209 {
1210 ps_dec->pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan;
1211 }
1212 }
1213 else
1214 {
1215 u1_field_pic_flag = 0;
1216 u1_bottom_field_flag = 0;
1217
1218 ps_dec->pu1_inv_scan = (UWORD8 *)gau1_ih264d_inv_scan;
1219 }
1220
1221 u1_nal_unit_type = SLICE_NAL;
1222 if(u1_is_idr_slice)
1223 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301224 u1_nal_unit_type = IDR_SLICE_NAL;
1225 u4_idr_pic_id = ih264d_uev(pu4_bitstrm_ofst,
1226 pu4_bitstrm_buf);
1227 if(u4_idr_pic_id > 65535)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301228 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301229 COPYTHECONTEXT("SH: ", u4_idr_pic_id);
1230 }
1231
1232 /* read delta pic order count information*/
1233 i_delta_poc[0] = i_delta_poc[1] = 0;
1234 s_tmp_poc.i4_pic_order_cnt_lsb = 0;
1235 s_tmp_poc.i4_delta_pic_order_cnt_bottom = 0;
1236 u1_pic_order_cnt_type = ps_seq->u1_pic_order_cnt_type;
1237 if(u1_pic_order_cnt_type == 0)
1238 {
1239 i_temp = ih264d_get_bits_h264(
1240 ps_bitstrm,
1241 ps_seq->u1_log2_max_pic_order_cnt_lsb_minus);
1242 if(i_temp < 0 || i_temp >= ps_seq->i4_max_pic_order_cntLsb)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301243 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301244 s_tmp_poc.i4_pic_order_cnt_lsb = i_temp;
1245 COPYTHECONTEXT("SH: pic_order_cnt_lsb", s_tmp_poc.i4_pic_order_cnt_lsb);
1246
1247 if((ps_pps->u1_pic_order_present_flag == 1) && (!u1_field_pic_flag))
1248 {
1249 s_tmp_poc.i4_delta_pic_order_cnt_bottom = ih264d_sev(
1250 pu4_bitstrm_ofst, pu4_bitstrm_buf);
1251 //if(s_tmp_poc.i4_delta_pic_order_cnt_bottom > ps_seq->i4_max_pic_order_cntLsb)
1252 COPYTHECONTEXT("SH: delta_pic_order_cnt_bottom",
1253 s_tmp_poc.i4_delta_pic_order_cnt_bottom);
1254 }
1255 }
1256
1257 s_tmp_poc.i4_delta_pic_order_cnt[0] = 0;
1258 s_tmp_poc.i4_delta_pic_order_cnt[1] = 0;
1259 if(u1_pic_order_cnt_type == 1
1260 && (!ps_seq->u1_delta_pic_order_always_zero_flag))
1261 {
1262 s_tmp_poc.i4_delta_pic_order_cnt[0] = ih264d_sev(pu4_bitstrm_ofst,
1263 pu4_bitstrm_buf);
1264 COPYTHECONTEXT("SH: delta_pic_order_cnt[0]",
1265 s_tmp_poc.i4_delta_pic_order_cnt[0]);
1266
1267 if(ps_pps->u1_pic_order_present_flag && !u1_field_pic_flag)
1268 {
1269 s_tmp_poc.i4_delta_pic_order_cnt[1] = ih264d_sev(
1270 pu4_bitstrm_ofst, pu4_bitstrm_buf);
1271 COPYTHECONTEXT("SH: delta_pic_order_cnt[1]",
1272 s_tmp_poc.i4_delta_pic_order_cnt[1]);
1273 }
1274 }
1275
1276 if(ps_pps->u1_redundant_pic_cnt_present_flag)
1277 {
1278 u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf);
1279 if(u4_temp > MAX_REDUNDANT_PIC_CNT)
Harish Mahendrakar9a00f562017-01-05 13:48:55 +05301280 return ERROR_INV_SLICE_HDR_T;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301281 u1_redundant_pic_cnt = u4_temp;
1282 COPYTHECONTEXT("SH: redundant_pic_cnt", u1_redundant_pic_cnt);
1283 }
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001284
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301285 /*--------------------------------------------------------------------*/
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001286 /* Check if the slice is part of new picture */
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301287 /*--------------------------------------------------------------------*/
Harish Mahendrakar8b5fd8f2016-04-26 16:36:03 +05301288 /* First slice of a picture is always considered as part of new picture */
1289 i1_is_end_of_poc = 1;
1290 ps_dec->ps_dec_err_status->u1_err_flag &= MASK_REJECT_CUR_PIC;
1291
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301292 if(ps_dec->u4_first_slice_in_pic == 0)
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301293 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301294 i1_is_end_of_poc = ih264d_is_end_of_pic(u2_frame_num, u1_nal_ref_idc,
1295 &s_tmp_poc, &ps_dec->s_cur_pic_poc,
1296 ps_cur_slice, u1_pic_order_cnt_type,
1297 u1_nal_unit_type, u4_idr_pic_id,
1298 u1_field_pic_flag,
1299 u1_bottom_field_flag);
Harish Mahendrakar49456122017-01-05 12:10:47 +05301300 if(i1_is_end_of_poc)
1301 {
1302 ps_dec->u1_first_slice_in_stream = 0;
1303 return ERROR_INCOMPLETE_FRAME;
1304 }
1305
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001306 }
1307
1308 /*--------------------------------------------------------------------*/
1309 /* Check for error in slice and parse the missing/corrupted MB's */
1310 /* as skip-MB's in an inserted P-slice */
1311 /*--------------------------------------------------------------------*/
1312 u1_mbaff = ps_seq->u1_mb_aff_flag && (!u1_field_pic_flag);
1313 prev_slice_err = 0;
1314
1315 if(i1_is_end_of_poc || ps_dec->u1_first_slice_in_stream)
1316 {
Harish Mahendrakar6dd54b92019-12-30 16:56:12 -08001317 /* If the current slice is not a field or frame number of the current
1318 * slice doesn't match with previous slice, and decoder is expecting
1319 * to decode a field i.e. ps_dec->u1_top_bottom_decoded is not 0 and
1320 * is not (TOP_FIELD_ONLY | BOT_FIELD_ONLY), treat it as a dangling
1321 * field */
1322 if((u1_field_pic_flag == 0 || u2_frame_num != ps_dec->u2_prv_frame_num)
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001323 && ps_dec->u1_top_bottom_decoded != 0
1324 && ps_dec->u1_top_bottom_decoded
1325 != (TOP_FIELD_ONLY | BOT_FIELD_ONLY))
1326 {
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301327 ps_dec->u1_dangling_field = 1;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001328 if(ps_dec->u4_first_slice_in_pic)
1329 {
1330 // first slice - dangling field
1331 prev_slice_err = 1;
1332 }
1333 else
1334 {
1335 // last slice - dangling field
1336 prev_slice_err = 2;
1337 }
1338
1339 if(ps_dec->u1_top_bottom_decoded ==TOP_FIELD_ONLY)
1340 ps_cur_slice->u1_bottom_field_flag = 1;
1341 else
1342 ps_cur_slice->u1_bottom_field_flag = 0;
1343
1344 num_mb_skipped = (ps_dec->u2_frm_ht_in_mbs * ps_dec->u2_frm_wd_in_mbs)
1345 - ps_dec->u2_total_mbs_coded;
1346 ps_cur_poc = &ps_dec->s_cur_pic_poc;
1347
1348 u1_is_idr_slice = ps_cur_slice->u1_nal_unit_type == IDR_SLICE_NAL;
1349 }
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301350 else if(ps_dec->u4_first_slice_in_pic)
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001351 {
1352 if(u2_first_mb_in_slice > 0)
1353 {
1354 // first slice - missing/header corruption
1355 prev_slice_err = 1;
1356 num_mb_skipped = u2_first_mb_in_slice << u1_mbaff;
1357 ps_cur_poc = &s_tmp_poc;
1358
1359 // initializing slice parameters
1360 ps_cur_slice->u4_idr_pic_id = u4_idr_pic_id;
1361 ps_cur_slice->u1_field_pic_flag = u1_field_pic_flag;
1362 ps_cur_slice->u1_bottom_field_flag = u1_bottom_field_flag;
1363 ps_cur_slice->i4_pic_order_cnt_lsb =
1364 s_tmp_poc.i4_pic_order_cnt_lsb;
1365 ps_cur_slice->u1_nal_unit_type = u1_nal_unit_type;
1366 ps_cur_slice->u1_redundant_pic_cnt = u1_redundant_pic_cnt;
1367 ps_cur_slice->u1_nal_ref_idc = u1_nal_ref_idc;
1368 ps_cur_slice->u1_pic_order_cnt_type = u1_pic_order_cnt_type;
Harish Mahendrakar1d5640f2016-12-15 18:04:53 +05301369 ps_cur_slice->u1_mbaff_frame_flag = ps_seq->u1_mb_aff_flag
1370 && (!u1_field_pic_flag);
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001371 }
1372 }
1373 else
1374 {
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301375 /* since i1_is_end_of_poc is set ,means new frame num is encountered. so conceal the current frame
1376 * completely */
1377 prev_slice_err = 2;
1378 num_mb_skipped = (ps_dec->u2_frm_ht_in_mbs
1379 * ps_dec->u2_frm_wd_in_mbs)
1380 - ps_dec->u2_total_mbs_coded;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001381 ps_cur_poc = &s_tmp_poc;
1382 }
1383 }
1384 else
1385 {
1386 if((u2_first_mb_in_slice << u1_mbaff) > ps_dec->u2_total_mbs_coded)
1387 {
1388 // previous slice - missing/corruption
1389 prev_slice_err = 2;
1390 num_mb_skipped = (u2_first_mb_in_slice << u1_mbaff)
1391 - ps_dec->u2_total_mbs_coded;
1392 ps_cur_poc = &s_tmp_poc;
1393 }
1394 else if((u2_first_mb_in_slice << u1_mbaff) < ps_dec->u2_total_mbs_coded)
1395 {
1396 return ERROR_CORRUPTED_SLICE;
1397 }
1398 }
1399
1400 if(prev_slice_err)
1401 {
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301402 ret = ih264d_mark_err_slice_skip(ps_dec, num_mb_skipped, u1_is_idr_slice, u2_frame_num, ps_cur_poc, prev_slice_err);
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001403
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301404 if(ps_dec->u1_dangling_field == 1)
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001405 {
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301406 ps_dec->u1_second_field = 1 - ps_dec->u1_second_field;
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301407 ps_dec->u1_first_slice_in_stream = 0;
Hamsalekha S252628c2017-08-01 14:51:55 +05301408 ps_dec->u1_top_bottom_decoded = TOP_FIELD_ONLY | BOT_FIELD_ONLY;
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301409 return ERROR_DANGLING_FIELD_IN_PIC;
1410 }
1411
1412 if(prev_slice_err == 2)
1413 {
1414 ps_dec->u1_first_slice_in_stream = 0;
1415 return ERROR_INCOMPLETE_FRAME;
1416 }
1417
1418 if(ps_dec->u2_total_mbs_coded
1419 >= ps_dec->u2_frm_ht_in_mbs * ps_dec->u2_frm_wd_in_mbs)
1420 {
1421 /* return if all MBs in frame are parsed*/
1422 ps_dec->u1_first_slice_in_stream = 0;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001423 return ERROR_IN_LAST_SLICE_OF_PIC;
1424 }
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301425
1426 if(ps_dec->ps_dec_err_status->u1_err_flag & REJECT_CUR_PIC)
1427 {
1428 ih264d_err_pic_dispbuf_mgr(ps_dec);
1429 return ERROR_NEW_FRAME_EXPECTED;
1430 }
1431
1432 if(ret != OK)
1433 return ret;
1434
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001435 i1_is_end_of_poc = 0;
1436 }
1437
Rakesh Kumar620e0292021-04-28 23:44:50 +05301438 /* Increment only if the current slice has atleast 1 more MB */
1439 if (ps_dec->u4_first_slice_in_pic == 0 &&
1440 (ps_dec->ps_parse_cur_slice->u4_first_mb_in_slice <
1441 (UWORD32)(ps_dec->u2_total_mbs_coded >> ps_dec->ps_cur_slice->u1_mbaff_frame_flag)))
Harish Mahendrakar5e4f64c2016-06-01 13:31:45 +05301442 {
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001443 ps_dec->ps_parse_cur_slice++;
Harish Mahendrakar5e4f64c2016-06-01 13:31:45 +05301444 ps_dec->u2_cur_slice_num++;
Rakesh Kumar620e0292021-04-28 23:44:50 +05301445 // in the case of single core increment ps_decode_cur_slice
1446 if(ps_dec->u1_separate_parse == 0)
1447 {
1448 ps_dec->ps_decode_cur_slice++;
1449 }
Harish Mahendrakar5e4f64c2016-06-01 13:31:45 +05301450 }
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001451
1452 ps_dec->u1_slice_header_done = 0;
1453
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001454
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301455 if(u1_field_pic_flag)
1456 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301457 ps_dec->u2_prv_frame_num = u2_frame_num;
1458 }
1459
1460 if(ps_cur_slice->u1_mmco_equalto5)
1461 {
1462 WORD32 i4_temp_poc;
1463 WORD32 i4_top_field_order_poc, i4_bot_field_order_poc;
1464
1465 if(!ps_cur_slice->u1_field_pic_flag) // or a complementary field pair
1466 {
1467 i4_top_field_order_poc = ps_dec->ps_cur_pic->i4_top_field_order_cnt;
1468 i4_bot_field_order_poc =
1469 ps_dec->ps_cur_pic->i4_bottom_field_order_cnt;
1470 i4_temp_poc = MIN(i4_top_field_order_poc,
1471 i4_bot_field_order_poc);
1472 }
1473 else if(!ps_cur_slice->u1_bottom_field_flag)
1474 i4_temp_poc = ps_dec->ps_cur_pic->i4_top_field_order_cnt;
1475 else
1476 i4_temp_poc = ps_dec->ps_cur_pic->i4_bottom_field_order_cnt;
1477
1478 ps_dec->ps_cur_pic->i4_top_field_order_cnt = i4_temp_poc
1479 - ps_dec->ps_cur_pic->i4_top_field_order_cnt;
1480 ps_dec->ps_cur_pic->i4_bottom_field_order_cnt = i4_temp_poc
1481 - ps_dec->ps_cur_pic->i4_bottom_field_order_cnt;
1482 ps_dec->ps_cur_pic->i4_poc = i4_temp_poc;
1483 ps_dec->ps_cur_pic->i4_avg_poc = i4_temp_poc;
1484 }
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301485 if(ps_dec->u4_first_slice_in_pic)
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301486 {
1487 ret = ih264d_decode_pic_order_cnt(u1_is_idr_slice, u2_frame_num,
1488 &ps_dec->s_prev_pic_poc,
1489 &s_tmp_poc, ps_cur_slice, ps_pps,
1490 u1_nal_ref_idc,
1491 u1_bottom_field_flag,
1492 u1_field_pic_flag, &i4_poc);
1493 if(ret != OK)
1494 return ret;
1495 /* Display seq no calculations */
1496 if(i4_poc >= ps_dec->i4_max_poc)
1497 ps_dec->i4_max_poc = i4_poc;
1498 /* IDR Picture or POC wrap around */
1499 if(i4_poc == 0)
1500 {
Ritu Baldwa49afc8f2018-04-10 18:20:17 +05301501 UWORD64 u8_temp;
1502 u8_temp = (UWORD64)ps_dec->i4_prev_max_display_seq
1503 + ps_dec->i4_max_poc
1504 + ps_dec->u1_max_dec_frame_buffering + 1;
1505 /*If i4_prev_max_display_seq overflows integer range, reset it */
1506 ps_dec->i4_prev_max_display_seq = (u8_temp > 0x7fffffff)?
1507 0 : u8_temp;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301508 ps_dec->i4_max_poc = 0;
1509 }
1510 }
1511
1512 /*--------------------------------------------------------------------*/
1513 /* Copy the values read from the bitstream to the slice header and then*/
1514 /* If the slice is first slice in picture, then do Start of Picture */
1515 /* processing. */
1516 /*--------------------------------------------------------------------*/
1517 ps_cur_slice->i4_delta_pic_order_cnt[0] = i_delta_poc[0];
1518 ps_cur_slice->i4_delta_pic_order_cnt[1] = i_delta_poc[1];
1519 ps_cur_slice->u4_idr_pic_id = u4_idr_pic_id;
1520 ps_cur_slice->u2_first_mb_in_slice = u2_first_mb_in_slice;
1521 ps_cur_slice->u1_field_pic_flag = u1_field_pic_flag;
1522 ps_cur_slice->u1_bottom_field_flag = u1_bottom_field_flag;
1523 ps_cur_slice->u1_slice_type = u1_slice_type;
1524 ps_cur_slice->i4_pic_order_cnt_lsb = s_tmp_poc.i4_pic_order_cnt_lsb;
1525
1526 ps_cur_slice->u1_nal_unit_type = u1_nal_unit_type;
1527 ps_cur_slice->u1_redundant_pic_cnt = u1_redundant_pic_cnt;
1528 ps_cur_slice->u1_nal_ref_idc = u1_nal_ref_idc;
1529 ps_cur_slice->u1_pic_order_cnt_type = u1_pic_order_cnt_type;
1530
1531 if(ps_seq->u1_frame_mbs_only_flag)
1532 ps_cur_slice->u1_direct_8x8_inference_flag =
1533 ps_seq->u1_direct_8x8_inference_flag;
1534 else
1535 ps_cur_slice->u1_direct_8x8_inference_flag = 1;
1536
1537 if(u1_slice_type == B_SLICE)
1538 {
1539 ps_cur_slice->u1_direct_spatial_mv_pred_flag = ih264d_get_bit_h264(
1540 ps_bitstrm);
1541 COPYTHECONTEXT("SH: direct_spatial_mv_pred_flag",
1542 ps_cur_slice->u1_direct_spatial_mv_pred_flag);
1543
1544 if(ps_cur_slice->u1_direct_spatial_mv_pred_flag)
1545 ps_cur_slice->pf_decodeDirect = ih264d_decode_spatial_direct;
1546 else
1547 ps_cur_slice->pf_decodeDirect = ih264d_decode_temporal_direct;
1548 if(!((ps_pps->ps_sps->u1_mb_aff_flag) && (!u1_field_pic_flag)))
1549 ps_dec->pf_mvpred = ih264d_mvpred_nonmbaffB;
1550 }
1551 else
1552 {
1553 if(!((ps_pps->ps_sps->u1_mb_aff_flag) && (!u1_field_pic_flag)))
1554 ps_dec->pf_mvpred = ih264d_mvpred_nonmbaff;
1555 }
1556
Harish Mahendrakar0b23c812017-01-16 14:43:42 +05301557 if(ps_dec->u4_first_slice_in_pic)
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301558 {
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301559 if(u2_first_mb_in_slice == 0)
1560 {
1561 ret = ih264d_start_of_pic(ps_dec, i4_poc, &s_tmp_poc, u2_frame_num, ps_pps);
1562 if(ret != OK)
1563 return ret;
1564 }
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301565
1566 ps_dec->u4_output_present = 0;
1567
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301568 {
1569 ih264d_get_next_display_field(ps_dec,
1570 ps_dec->ps_out_buffer,
1571 &(ps_dec->s_disp_op));
1572 /* If error code is non-zero then there is no buffer available for display,
1573 hence avoid format conversion */
1574
1575 if(0 != ps_dec->s_disp_op.u4_error_code)
1576 {
1577 ps_dec->u4_fmt_conv_cur_row = ps_dec->s_disp_frame_info.u4_y_ht;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301578 }
1579 else
1580 ps_dec->u4_output_present = 1;
1581 }
1582 if(ps_dec->u1_separate_parse == 1)
1583 {
1584 if(ps_dec->u4_dec_thread_created == 0)
1585 {
1586 ithread_create(ps_dec->pv_dec_thread_handle, NULL,
1587 (void *)ih264d_decode_picture_thread,
1588 (void *)ps_dec);
1589
1590 ps_dec->u4_dec_thread_created = 1;
1591 }
1592
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001593 if((ps_dec->u4_num_cores == 3) &&
1594 ((ps_dec->u4_app_disable_deblk_frm == 0) || ps_dec->i1_recon_in_thread3_flag)
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301595 && (ps_dec->u4_bs_deblk_thread_created == 0))
1596 {
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001597 ps_dec->u4_start_recon_deblk = 0;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301598 ithread_create(ps_dec->pv_bs_deblk_thread_handle, NULL,
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001599 (void *)ih264d_recon_deblk_thread,
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301600 (void *)ps_dec);
1601 ps_dec->u4_bs_deblk_thread_created = 1;
1602 }
1603 }
1604
1605 }
1606
1607 /* INITIALIZATION of fn ptrs for MC and formMbPartInfo functions */
1608 {
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001609 UWORD8 uc_nofield_nombaff;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301610
1611
1612
1613 uc_nofield_nombaff = ((ps_dec->ps_cur_slice->u1_field_pic_flag == 0)
1614 && (ps_dec->ps_cur_slice->u1_mbaff_frame_flag == 0)
1615 && (u1_slice_type != B_SLICE)
1616 && (ps_dec->ps_cur_pps->u1_wted_pred_flag == 0));
1617
1618 /* Initialise MC and formMbPartInfo fn ptrs one time based on profile_idc */
1619
1620 if(uc_nofield_nombaff)
1621 {
1622 ps_dec->p_form_mb_part_info = ih264d_form_mb_part_info_bp;
1623 ps_dec->p_motion_compensate = ih264d_motion_compensate_bp;
1624 }
1625 else
1626 {
1627 ps_dec->p_form_mb_part_info = ih264d_form_mb_part_info_mp;
1628 ps_dec->p_motion_compensate = ih264d_motion_compensate_mp;
1629 }
1630
1631
1632 }
1633
1634 /*
1635 * Decide whether to decode the current picture or not
1636 */
1637 {
1638 dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;
1639 if(ps_err->u4_frm_sei_sync == u2_frame_num)
1640 {
1641 ps_err->u1_err_flag = ACCEPT_ALL_PICS;
1642 ps_err->u4_frm_sei_sync = SYNC_FRM_DEFAULT;
1643 }
1644 ps_err->u4_cur_frm = u2_frame_num;
1645 }
1646
1647 /* Decision for decoding if the picture is to be skipped */
1648 {
1649 WORD32 i4_skip_b_pic, i4_skip_p_pic;
1650
1651 i4_skip_b_pic = (ps_dec->u4_skip_frm_mask & B_SLC_BIT)
1652 && (B_SLICE == u1_slice_type) && (0 == u1_nal_ref_idc);
1653
1654 i4_skip_p_pic = (ps_dec->u4_skip_frm_mask & P_SLC_BIT)
1655 && (P_SLICE == u1_slice_type) && (0 == u1_nal_ref_idc);
1656
1657 /**************************************************************/
1658 /* Skip the B picture if skip mask is set for B picture and */
1659 /* Current B picture is a non reference B picture or there is */
1660 /* no user for reference B picture */
1661 /**************************************************************/
1662 if(i4_skip_b_pic)
1663 {
1664 ps_dec->ps_cur_pic->u4_pack_slc_typ |= B_SLC_BIT;
1665 /* Don't decode the picture in SKIP-B mode if that picture is B */
1666 /* and also it is not to be used as a reference picture */
1667 ps_dec->u1_last_pic_not_decoded = 1;
1668
1669 return OK;
1670 }
1671 /**************************************************************/
1672 /* Skip the P picture if skip mask is set for P picture and */
1673 /* Current P picture is a non reference P picture or there is */
1674 /* no user for reference P picture */
1675 /**************************************************************/
1676 if(i4_skip_p_pic)
1677 {
1678 ps_dec->ps_cur_pic->u4_pack_slc_typ |= P_SLC_BIT;
1679 /* Don't decode the picture in SKIP-P mode if that picture is P */
1680 /* and also it is not to be used as a reference picture */
1681 ps_dec->u1_last_pic_not_decoded = 1;
1682
1683 return OK;
1684 }
1685 }
1686
1687 {
1688 UWORD16 u2_mb_x, u2_mb_y;
1689
1690 ps_dec->i4_submb_ofst = ((u2_first_mb_in_slice
1691 << ps_cur_slice->u1_mbaff_frame_flag) * SUB_BLK_SIZE)
1692 - SUB_BLK_SIZE;
1693 if(u2_first_mb_in_slice)
1694 {
1695 UWORD8 u1_mb_aff;
1696 UWORD8 u1_field_pic;
1697 UWORD16 u2_frm_wd_in_mbs;
1698 u2_frm_wd_in_mbs = ps_seq->u2_frm_wd_in_mbs;
1699 u1_mb_aff = ps_cur_slice->u1_mbaff_frame_flag;
1700 u1_field_pic = ps_cur_slice->u1_field_pic_flag;
1701
1702 {
1703 UWORD32 x_offset;
1704 UWORD32 y_offset;
1705 UWORD32 u4_frame_stride;
1706 tfr_ctxt_t *ps_trns_addr; // = &ps_dec->s_tran_addrecon_parse;
1707
1708 if(ps_dec->u1_separate_parse)
1709 {
1710 ps_trns_addr = &ps_dec->s_tran_addrecon_parse;
1711 }
1712 else
1713 {
1714 ps_trns_addr = &ps_dec->s_tran_addrecon;
1715 }
1716 u2_mb_x = MOD(u2_first_mb_in_slice, u2_frm_wd_in_mbs);
1717 u2_mb_y = DIV(u2_first_mb_in_slice, u2_frm_wd_in_mbs);
1718
1719 u2_mb_y <<= u1_mb_aff;
1720
1721 if((u2_mb_x > u2_frm_wd_in_mbs - 1)
1722 || (u2_mb_y > ps_dec->u2_frm_ht_in_mbs - 1))
1723 {
1724 return ERROR_CORRUPTED_SLICE;
1725 }
1726
1727 u4_frame_stride = ps_dec->u2_frm_wd_y << u1_field_pic;
1728 x_offset = u2_mb_x << 4;
1729 y_offset = (u2_mb_y * u4_frame_stride) << 4;
1730
1731 ps_trns_addr->pu1_dest_y = ps_dec->s_cur_pic.pu1_buf1 + x_offset
1732 + y_offset;
1733
1734 u4_frame_stride = ps_dec->u2_frm_wd_uv << u1_field_pic;
1735 x_offset >>= 1;
1736 y_offset = (u2_mb_y * u4_frame_stride) << 3;
1737
1738 x_offset *= YUV420SP_FACTOR;
1739
1740 ps_trns_addr->pu1_dest_u = ps_dec->s_cur_pic.pu1_buf2 + x_offset
1741 + y_offset;
1742 ps_trns_addr->pu1_dest_v = ps_dec->s_cur_pic.pu1_buf3 + x_offset
1743 + y_offset;
1744
1745 ps_trns_addr->pu1_mb_y = ps_trns_addr->pu1_dest_y;
1746 ps_trns_addr->pu1_mb_u = ps_trns_addr->pu1_dest_u;
1747 ps_trns_addr->pu1_mb_v = ps_trns_addr->pu1_dest_v;
1748
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301749
1750 // assign the deblock structure pointers to start of slice
1751 if(ps_dec->u1_separate_parse == 1)
1752 {
1753 ps_dec->ps_deblk_mbn = ps_dec->ps_deblk_pic
1754 + (u2_first_mb_in_slice << u1_mb_aff);
1755 }
1756 else
1757 {
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301758 ps_dec->ps_deblk_mbn = ps_dec->ps_deblk_pic
1759 + (u2_first_mb_in_slice << u1_mb_aff);
1760 }
1761
1762 ps_dec->u2_cur_mb_addr = (u2_first_mb_in_slice << u1_mb_aff);
1763
1764 ps_dec->ps_mv_cur = ps_dec->s_cur_pic.ps_mv
1765 + ((u2_first_mb_in_slice << u1_mb_aff) << 4);
1766 }
1767 }
1768 else
1769 {
1770 tfr_ctxt_t *ps_trns_addr;
1771
1772 if(ps_dec->u1_separate_parse)
1773 {
1774 ps_trns_addr = &ps_dec->s_tran_addrecon_parse;
1775 }
1776 else
1777 {
1778 ps_trns_addr = &ps_dec->s_tran_addrecon;
1779 }
1780
1781 u2_mb_x = 0xffff;
1782 u2_mb_y = 0;
1783 // assign the deblock structure pointers to start of slice
1784 ps_dec->u2_cur_mb_addr = 0;
1785 ps_dec->ps_deblk_mbn = ps_dec->ps_deblk_pic;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301786 ps_dec->ps_mv_cur = ps_dec->s_cur_pic.ps_mv;
1787 ps_trns_addr->pu1_dest_y = ps_dec->s_cur_pic.pu1_buf1;
1788 ps_trns_addr->pu1_dest_u = ps_dec->s_cur_pic.pu1_buf2;
1789 ps_trns_addr->pu1_dest_v = ps_dec->s_cur_pic.pu1_buf3;
1790
1791 ps_trns_addr->pu1_mb_y = ps_trns_addr->pu1_dest_y;
1792 ps_trns_addr->pu1_mb_u = ps_trns_addr->pu1_dest_u;
1793 ps_trns_addr->pu1_mb_v = ps_trns_addr->pu1_dest_v;
1794
1795 }
1796
1797 ps_dec->ps_part = ps_dec->ps_parse_part_params;
1798
1799 ps_dec->u2_mbx =
1800 (MOD(u2_first_mb_in_slice - 1, ps_seq->u2_frm_wd_in_mbs));
1801 ps_dec->u2_mby =
1802 (DIV(u2_first_mb_in_slice - 1, ps_seq->u2_frm_wd_in_mbs));
1803 ps_dec->u2_mby <<= ps_cur_slice->u1_mbaff_frame_flag;
1804 ps_dec->i2_prev_slice_mbx = ps_dec->u2_mbx;
1805 ps_dec->i2_prev_slice_mby = ps_dec->u2_mby;
1806 }
1807
1808 /* RBSP stop bit is used for CABAC decoding*/
1809 ps_bitstrm->u4_max_ofst += ps_dec->ps_cur_pps->u1_entropy_coding_mode;
1810
1811 ps_dec->u1_B = (u1_slice_type == B_SLICE);
1812 ps_dec->u4_next_mb_skip = 0;
1813
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301814 ps_dec->ps_parse_cur_slice->u4_first_mb_in_slice =
1815 ps_dec->ps_cur_slice->u2_first_mb_in_slice;
1816 ps_dec->ps_parse_cur_slice->slice_type =
1817 ps_dec->ps_cur_slice->u1_slice_type;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301818
1819
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001820 ps_dec->u4_start_recon_deblk = 1;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301821 {
1822 WORD32 num_entries;
1823 WORD32 size;
1824 UWORD8 *pu1_buf;
1825
Harish Mahendrakar251b0072015-08-04 09:55:15 +05301826 num_entries = MAX_FRAMES;
1827 if((1 >= ps_dec->ps_cur_sps->u1_num_ref_frames) &&
1828 (0 == ps_dec->i4_display_delay))
1829 {
1830 num_entries = 1;
1831 }
1832 num_entries = ((2 * num_entries) + 1);
Hamsalekha Sa925a6b2017-04-21 11:01:52 +05301833 num_entries *= 2;
1834
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301835
1836 size = num_entries * sizeof(void *);
1837 size += PAD_MAP_IDX_POC * sizeof(void *);
1838
1839 pu1_buf = (UWORD8 *)ps_dec->pv_map_ref_idx_to_poc_buf;
1840 pu1_buf += size * ps_dec->u2_cur_slice_num;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001841 ps_dec->ps_parse_cur_slice->ppv_map_ref_idx_to_poc = ( void *)pu1_buf;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301842 }
1843
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001844 if(ps_dec->u1_separate_parse)
1845 {
1846 ps_dec->ps_parse_cur_slice->pv_tu_coeff_data_start = ps_dec->pv_parse_tu_coeff_data;
1847 }
1848 else
1849 {
1850 ps_dec->pv_proc_tu_coeff_data = ps_dec->pv_parse_tu_coeff_data;
1851 }
1852
Rakesh Kumarf312a1d2018-07-09 21:58:44 +05301853 ret = ih264d_fix_error_in_dpb(ps_dec);
1854 if(ret < 0)
1855 return ERROR_DBP_MANAGER_T;
1856
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301857 if(u1_slice_type == I_SLICE)
1858 {
1859 ps_dec->ps_cur_pic->u4_pack_slc_typ |= I_SLC_BIT;
1860
1861 ret = ih264d_parse_islice(ps_dec, u2_first_mb_in_slice);
Ritu Baldwaaecdfd12018-01-16 13:48:41 +05301862 ps_dec->u1_pr_sl_type = u1_slice_type;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301863 if(ps_dec->i4_pic_type != B_SLICE && ps_dec->i4_pic_type != P_SLICE)
1864 ps_dec->i4_pic_type = I_SLICE;
1865
1866 }
1867 else if(u1_slice_type == P_SLICE)
1868 {
1869 ps_dec->ps_cur_pic->u4_pack_slc_typ |= P_SLC_BIT;
1870 ret = ih264d_parse_pslice(ps_dec, u2_first_mb_in_slice);
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301871 ps_dec->u1_pr_sl_type = u1_slice_type;
1872 if(ps_dec->i4_pic_type != B_SLICE)
1873 ps_dec->i4_pic_type = P_SLICE;
1874 }
1875 else if(u1_slice_type == B_SLICE)
1876 {
1877 ps_dec->ps_cur_pic->u4_pack_slc_typ |= B_SLC_BIT;
1878 ret = ih264d_parse_bslice(ps_dec, u2_first_mb_in_slice);
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301879 ps_dec->u1_pr_sl_type = u1_slice_type;
1880 ps_dec->i4_pic_type = B_SLICE;
1881 }
1882 else
1883 return ERROR_INV_SLC_TYPE_T;
1884
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001885 if(ps_dec->u1_slice_header_done)
1886 {
Hamsalekha S9f0dcba2015-06-09 15:54:31 +05301887 /* set to zero to indicate a valid slice has been decoded */
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001888 ps_dec->u1_first_slice_in_stream = 0;
1889 }
1890
1891 if(ret != OK)
1892 return ret;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301893
Ritu Baldwa3c70b9a2017-10-09 13:52:45 +05301894 if(u1_nal_ref_idc != 0)
1895 {
1896 if(!ps_dec->ps_dpb_cmds->u1_dpb_commands_read)
1897 {
1898 memcpy((void *)ps_dec->ps_dpb_cmds, (void *)(&(ps_dec->s_dpb_cmds_scratch)),
1899 sizeof(dpb_commands_t));
1900 }
1901 }
1902
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301903 /* storing last Mb X and MbY of the slice */
1904 ps_dec->i2_prev_slice_mbx = ps_dec->u2_mbx;
1905 ps_dec->i2_prev_slice_mby = ps_dec->u2_mby;
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001906
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301907 /* End of Picture detection */
1908
1909 if(ps_dec->u2_total_mbs_coded >= (ps_seq->u2_max_mb_addr + 1))
1910 {
1911 ps_dec->u1_pic_decode_done = 1;
1912
1913 }
1914
1915 {
1916 dec_err_status_t * ps_err = ps_dec->ps_dec_err_status;
1917 if((ps_err->u1_err_flag & REJECT_PB_PICS)
1918 && (ps_err->u1_cur_pic_type == PIC_TYPE_I))
1919 {
1920 ps_err->u1_err_flag = ACCEPT_ALL_PICS;
1921 }
1922 }
1923
1924 PRINT_BIN_BIT_RATIO(ps_dec)
1925
Marco Nelissen8ef4c3f2015-06-03 07:26:32 -07001926 return ret;
Hamsalekha S8d3d3032015-03-13 21:24:58 +05301927}
1928