/****************************************************************************** * * Copyright (C) 2015 The Android Open Source Project * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ***************************************************************************** * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore */ /*! ************************************************************************** * \file ih264d_process_pslice.c * * \brief * Contains routines that decode a I slice type * * Detailed_description * * \date * 21/12/2002 * * \author NS ************************************************************************** */ #include "ih264_typedefs.h" #include "ih264_macros.h" #include "ih264_platform_macros.h" #include #include "ih264d_bitstrm.h" #include "ih264d_defs.h" #include "ih264d_debug.h" #include "ih264d_structs.h" #include "ih264d_defs.h" #include "ih264d_parse_cavlc.h" #include "ih264d_mb_utils.h" #include "ih264d_deblocking.h" #include "ih264d_dpb_manager.h" #include "ih264d_mvpred.h" #include "ih264d_inter_pred.h" #include "ih264d_process_pslice.h" #include "ih264d_error_handler.h" #include "ih264d_cabac.h" #include "ih264d_debug.h" #include "ih264d_tables.h" #include "ih264d_parse_slice.h" #include "ih264d_utils.h" #include "ih264d_parse_islice.h" #include "ih264d_process_bslice.h" #include "ih264d_process_intra_mb.h" void ih264d_init_cabac_contexts(UWORD8 u1_slice_type, dec_struct_t * ps_dec); void ih264d_insert_pic_in_ref_pic_listx(struct pic_buffer_t *ps_ref_pic_buf_lx, struct pic_buffer_t *ps_pic) { *ps_ref_pic_buf_lx = *ps_pic; } WORD32 ih264d_mv_pred_ref_tfr_nby2_pmb(dec_struct_t * ps_dec, UWORD8 u1_mb_idx, UWORD8 u1_num_mbs) { parse_pmbarams_t * ps_mb_part_info; parse_part_params_t * ps_part; mv_pred_t *ps_mv_nmb, *ps_mv_nmb_start, *ps_mv_ntop, *ps_mv_ntop_start; UWORD32 i, j; const UWORD32 u1_mbaff = ps_dec->ps_cur_slice->u1_mbaff_frame_flag; dec_mb_info_t * ps_cur_mb_info; WORD32 i2_mv_x, i2_mv_y; WORD32 ret; ps_dec->i4_submb_ofst -= (u1_num_mbs - u1_mb_idx) << 4; ps_mb_part_info = ps_dec->ps_parse_mb_data; // + u1_mb_idx; ps_part = ps_dec->ps_parse_part_params; // + u1_mb_idx; /* N/2 Mb MvPred and Transfer Setup Loop */ for(i = u1_mb_idx; i < u1_num_mbs; i++, ps_mb_part_info++) { UWORD32 u1_colz; UWORD32 u1_field; mv_pred_t s_mvPred; mv_pred_t *ps_mv_pred = &s_mvPred; *ps_mv_pred = ps_dec->s_default_mv_pred; ps_dec->i4_submb_ofst += SUB_BLK_SIZE; /* Restore the slice scratch MbX and MbY context */ ps_cur_mb_info = ps_dec->ps_nmb_info + i; u1_field = ps_cur_mb_info->u1_mb_field_decodingflag; ps_mv_nmb_start = ps_dec->ps_mv_cur + (i << 4); ps_dec->u2_mbx = ps_cur_mb_info->u2_mbx; ps_dec->u2_mby = ps_cur_mb_info->u2_mby; ps_dec->u2_mv_2mb[i & 0x1] = 0; /* Look for MV Prediction and Reference Transfer in Non-I Mbs */ if(!ps_mb_part_info->u1_isI_mb) { UWORD32 u1_blk_no; WORD32 i1_ref_idx, i1_ref_idx1; UWORD32 u1_sub_mb_x, u1_sub_mb_y, u1_sub_mb_num; UWORD32 u1_num_part, u1_num_ref, u1_wd, u1_ht; UWORD32 *pu4_wt_offst, **ppu4_wt_ofst; UWORD32 u1_scale_ref, u4_bot_mb; WORD8 *pi1_ref_idx = ps_mb_part_info->i1_ref_idx[0]; pic_buffer_t *ps_ref_frame, **pps_ref_frame; deblk_mb_t * ps_cur_deblk_mb = ps_dec->ps_deblk_mbn + i; /* MB Level initialisations */ ps_dec->u4_num_pmbair = i >> u1_mbaff; ps_dec->u1_mb_idx_mv = i; ppu4_wt_ofst = ps_mb_part_info->pu4_wt_offst; pps_ref_frame = ps_dec->ps_ref_pic_buf_lx[0]; /* CHANGED CODE */ ps_mv_ntop_start = ps_mv_nmb_start - (ps_dec->u2_frm_wd_in_mbs << (4 + u1_mbaff)) + 12; u1_num_part = ps_mb_part_info->u1_num_part; ps_cur_deblk_mb->u1_mb_type |= (u1_num_part > 1) << 1; ps_cur_mb_info->u4_pred_info_pkd_idx = ps_dec->u4_pred_info_pkd_idx; ps_cur_mb_info->u1_num_pred_parts = 0; /****************************************************/ /* weighted u4_ofst pointer calculations, this loop */ /* runs maximum 4 times, even in direct cases */ /****************************************************/ u1_scale_ref = u1_mbaff & u1_field; u4_bot_mb = 1 - ps_cur_mb_info->u1_topmb; if(ps_dec->ps_cur_pps->u1_wted_pred_flag) { u1_num_ref = MIN(u1_num_part, 4); for(u1_blk_no = 0; u1_blk_no < u1_num_ref; u1_blk_no++) { i1_ref_idx = pi1_ref_idx[u1_blk_no]; if(u1_scale_ref) i1_ref_idx >>= 1; pu4_wt_offst = (UWORD32*)&ps_dec->pu4_wt_ofsts[2 * X3(i1_ref_idx)]; ppu4_wt_ofst[u1_blk_no] = pu4_wt_offst; } } else { ppu4_wt_ofst[0] = NULL; ppu4_wt_ofst[1] = NULL; ppu4_wt_ofst[2] = NULL; ppu4_wt_ofst[3] = NULL; } /**************************************************/ /* Loop on Partitions */ /**************************************************/ for(j = 0; j < u1_num_part; j++, ps_part++) { u1_sub_mb_num = ps_part->u1_sub_mb_num; ps_dec->u1_sub_mb_num = u1_sub_mb_num; if(PART_NOT_DIRECT != ps_part->u1_is_direct) { /* Mb Skip Mode */ /* Setting the default and other members of MvPred Structure */ s_mvPred.i2_mv[2] = -1; s_mvPred.i2_mv[3] = -1; s_mvPred.i1_ref_frame[0] = 0; i1_ref_idx = (u1_scale_ref && u4_bot_mb) ? MAX_REF_BUFS : 0; ps_ref_frame = pps_ref_frame[i1_ref_idx]; s_mvPred.u1_col_ref_pic_idx = ps_ref_frame->u1_mv_buf_id; s_mvPred.u1_pic_type = ps_ref_frame->u1_pic_type; pu4_wt_offst = (UWORD32*)&ps_dec->pu4_wt_ofsts[0]; ps_dec->pf_mvpred(ps_dec, ps_cur_mb_info, ps_mv_nmb_start, ps_mv_ntop_start, &s_mvPred, 0, 4, 0, 1, MB_SKIP); { pred_info_pkd_t *ps_pred_pkd; ps_pred_pkd = ps_dec->ps_pred_pkd + ps_dec->u4_pred_info_pkd_idx; ih264d_fill_pred_info (s_mvPred.i2_mv,4,4,0,PRED_L0,ps_pred_pkd,ps_ref_frame->u1_pic_buf_id, (i1_ref_idx >> u1_scale_ref),pu4_wt_offst, ps_ref_frame->u1_pic_type); ps_dec->u4_pred_info_pkd_idx++; ps_cur_mb_info->u1_num_pred_parts++; } /* Storing colocated zero information */ u1_colz = ((ABS(s_mvPred.i2_mv[0]) <= 1) && (ABS(s_mvPred.i2_mv[1]) <= 1)) + (u1_field << 1); ih264d_rep_mv_colz(ps_dec, &s_mvPred, ps_mv_nmb_start, 0, u1_colz, 4, 4); } else { u1_sub_mb_x = u1_sub_mb_num & 0x03; u1_sub_mb_y = u1_sub_mb_num >> 2; u1_blk_no = (u1_num_part < 4) ? j : (((u1_sub_mb_y >> 1) << 1) + (u1_sub_mb_x >> 1)); ps_mv_ntop = ps_mv_ntop_start + u1_sub_mb_x; ps_mv_nmb = ps_mv_nmb_start + u1_sub_mb_num; u1_wd = ps_part->u1_partwidth; u1_ht = ps_part->u1_partheight; /* Populate the colpic info and reference frames */ i1_ref_idx = pi1_ref_idx[u1_blk_no]; s_mvPred.i1_ref_frame[0] = i1_ref_idx; /********************************************************/ /* Predict Mv */ /* Add Mv Residuals and store back */ /********************************************************/ ps_dec->pf_mvpred(ps_dec, ps_cur_mb_info, ps_mv_nmb, ps_mv_ntop, &s_mvPred, u1_sub_mb_num, u1_wd, 0, 1, ps_cur_mb_info->u1_mb_mc_mode); i2_mv_x = ps_mv_nmb->i2_mv[0]; i2_mv_y = ps_mv_nmb->i2_mv[1]; i2_mv_x += s_mvPred.i2_mv[0]; i2_mv_y += s_mvPred.i2_mv[1]; s_mvPred.i2_mv[0] = i2_mv_x; s_mvPred.i2_mv[1] = i2_mv_y; /********************************************************/ /* Transfer setup call */ /* convert RefIdx if it is MbAff */ /* Pass Weight Offset and refFrame */ /********************************************************/ i1_ref_idx1 = i1_ref_idx >> u1_scale_ref; if(u1_scale_ref && ((i1_ref_idx & 0x01) != u4_bot_mb)) i1_ref_idx1 += MAX_REF_BUFS; ps_ref_frame = pps_ref_frame[i1_ref_idx1]; pu4_wt_offst = ppu4_wt_ofst[u1_blk_no]; { pred_info_pkd_t *ps_pred_pkd; ps_pred_pkd = ps_dec->ps_pred_pkd + ps_dec->u4_pred_info_pkd_idx; ih264d_fill_pred_info (s_mvPred.i2_mv,u1_wd,u1_ht,u1_sub_mb_num,PRED_L0,ps_pred_pkd, ps_ref_frame->u1_pic_buf_id,(i1_ref_idx >> u1_scale_ref),pu4_wt_offst, ps_ref_frame->u1_pic_type); ps_dec->u4_pred_info_pkd_idx++; ps_cur_mb_info->u1_num_pred_parts++; } /* Fill colocated info in MvPred structure */ s_mvPred.u1_col_ref_pic_idx = ps_ref_frame->u1_mv_buf_id; s_mvPred.u1_pic_type = ps_ref_frame->u1_pic_type; /* Calculating colocated zero information */ u1_colz = (u1_field << 1) | ((i1_ref_idx == 0) && (ABS(i2_mv_x) <= 1) && (ABS(i2_mv_y) <= 1)); u1_colz |= ps_mb_part_info->u1_col_info[u1_blk_no]; /* Replicate the motion vectors and colzero u4_flag */ /* for all sub-partitions */ ih264d_rep_mv_colz(ps_dec, &s_mvPred, ps_mv_nmb, u1_sub_mb_num, u1_colz, u1_ht, u1_wd); } } } else { /* Storing colocated zero information */ ih264d_rep_mv_colz(ps_dec, &s_mvPred, ps_mv_nmb_start, 0, (UWORD8)(u1_field << 1), 4, 4); } /*if num _cores is set to 3,compute bs will be done in another thread*/ if(ps_dec->u4_num_cores < 3) { if(ps_dec->u4_app_disable_deblk_frm == 0) ps_dec->pf_compute_bs(ps_dec, ps_cur_mb_info, (UWORD16)(i >> u1_mbaff)); } } return OK; } WORD32 ih264d_decode_recon_tfr_nmb(dec_struct_t * ps_dec, UWORD8 u1_mb_idx, UWORD8 u1_num_mbs, UWORD8 u1_num_mbs_next, UWORD8 u1_tfr_n_mb, UWORD8 u1_end_of_row) { WORD32 i,j; UWORD32 u1_end_of_row_next; dec_mb_info_t * ps_cur_mb_info; UWORD32 u4_update_mbaff = 0; WORD32 ret; const UWORD32 u1_mbaff = ps_dec->ps_cur_slice->u1_mbaff_frame_flag; const UWORD32 u1_slice_type = ps_dec->ps_cur_slice->u1_slice_type; const WORD32 u1_skip_th = ( (u1_slice_type != I_SLICE) ? (ps_dec->u1_B ? B_8x8 : PRED_8x8R0) : -1); const UWORD32 u1_ipcm_th = ( (u1_slice_type != I_SLICE) ? (ps_dec->u1_B ? 23 : 5) : 0); /* N Mb MC Loop */ for(i = u1_mb_idx; i < u1_num_mbs; i++) { ps_cur_mb_info = ps_dec->ps_nmb_info + i; ps_dec->u4_dma_buf_idx = 0; ps_dec->u4_pred_info_idx = 0; if(ps_cur_mb_info->u1_mb_type <= u1_skip_th) { { WORD32 pred_cnt = 0; pred_info_pkd_t *ps_pred_pkd; UWORD32 u4_pred_info_pkd_idx; WORD8 i1_pred; u4_pred_info_pkd_idx = ps_cur_mb_info->u4_pred_info_pkd_idx; while(pred_cnt < ps_cur_mb_info->u1_num_pred_parts) { ps_pred_pkd = ps_dec->ps_pred_pkd + u4_pred_info_pkd_idx; ps_dec->p_form_mb_part_info(ps_pred_pkd,ps_dec, ps_cur_mb_info->u2_mbx,ps_cur_mb_info->u2_mby,(i >> u1_mbaff), ps_cur_mb_info); u4_pred_info_pkd_idx++; pred_cnt++; } } ps_dec->p_motion_compensate(ps_dec, ps_cur_mb_info); } else if(ps_cur_mb_info->u1_mb_type == MB_SKIP) { { WORD32 pred_cnt = 0; pred_info_pkd_t *ps_pred_pkd; UWORD32 u4_pred_info_pkd_idx; WORD8 i1_pred; u4_pred_info_pkd_idx = ps_cur_mb_info->u4_pred_info_pkd_idx; while(pred_cnt < ps_cur_mb_info->u1_num_pred_parts) { ps_pred_pkd = ps_dec->ps_pred_pkd + u4_pred_info_pkd_idx; ps_dec->p_form_mb_part_info(ps_pred_pkd,ps_dec, ps_cur_mb_info->u2_mbx,ps_cur_mb_info->u2_mby,(i >> u1_mbaff), ps_cur_mb_info); u4_pred_info_pkd_idx++; pred_cnt++; } } /* Decode MB skip */ ps_dec->p_motion_compensate(ps_dec, ps_cur_mb_info); } } /* N Mb IQ IT RECON Loop */ for(j = u1_mb_idx; j < i; j++) { ps_cur_mb_info = ps_dec->ps_nmb_info + j; if(ps_cur_mb_info->u1_mb_type <= u1_skip_th) { ih264d_process_inter_mb(ps_dec, ps_cur_mb_info, j); } else if(ps_cur_mb_info->u1_mb_type != MB_SKIP) { if((u1_ipcm_th + 25) != ps_cur_mb_info->u1_mb_type) { ps_cur_mb_info->u1_mb_type -= (u1_skip_th + 1); ih264d_process_intra_mb(ps_dec, ps_cur_mb_info, j); } } if(ps_dec->u4_use_intrapred_line_copy) { ih264d_copy_intra_pred_line(ps_dec, ps_cur_mb_info, j); } } /*N MB deblocking*/ if(ps_dec->u4_nmb_deblk == 1) { UWORD32 u4_cur_mb, u4_right_mb; UWORD32 u4_mb_x, u4_mb_y; UWORD32 u4_wd_y, u4_wd_uv; tfr_ctxt_t *ps_tfr_cxt = &(ps_dec->s_tran_addrecon); UWORD8 u1_field_pic_flag = ps_dec->ps_cur_slice->u1_field_pic_flag; const WORD32 i4_cb_qp_idx_ofst = ps_dec->ps_cur_pps->i1_chroma_qp_index_offset; const WORD32 i4_cr_qp_idx_ofst = ps_dec->ps_cur_pps->i1_second_chroma_qp_index_offset; u4_wd_y = ps_dec->u2_frm_wd_y << u1_field_pic_flag; u4_wd_uv = ps_dec->u2_frm_wd_uv << u1_field_pic_flag; ps_cur_mb_info = ps_dec->ps_nmb_info + u1_mb_idx; ps_dec->u4_deblk_mb_x = ps_cur_mb_info->u2_mbx; ps_dec->u4_deblk_mb_y = ps_cur_mb_info->u2_mby; for(j = u1_mb_idx; j < i; j++) { ih264d_deblock_mb_nonmbaff(ps_dec, ps_tfr_cxt, i4_cb_qp_idx_ofst, i4_cr_qp_idx_ofst, u4_wd_y, u4_wd_uv); } } if(u1_tfr_n_mb) { /****************************************************************/ /* Check for End Of Row in Next iteration */ /****************************************************************/ u1_end_of_row_next = u1_num_mbs_next && (u1_num_mbs_next <= (ps_dec->u1_recon_mb_grp >> u1_mbaff)); /****************************************************************/ /* Transfer the Following things */ /* N-Mb DeblkParams Data ( To Ext DeblkParams Buffer ) */ /* N-Mb Recon Data ( To Ext Frame Buffer ) */ /* N-Mb Intrapredline Data ( Updated Internally) */ /* N-Mb MV Data ( To Ext MV Buffer ) */ /* N-Mb MVTop/TopRight Data ( To Int MV Top Scratch Buffers) */ /****************************************************************/ ih264d_transfer_mb_group_data(ps_dec, u1_num_mbs, u1_end_of_row, u1_end_of_row_next); ps_dec->u4_num_mbs_prev_nmb = u1_num_mbs; ps_dec->u4_pred_info_idx = 0; ps_dec->u4_dma_buf_idx = 0; } return OK; } /*! ************************************************************************** * \if Function name : ih264d_process_inter_mb \endif * * \brief * This function decodes an Inter MB. * * * \return * 0 on Success and Error code otherwise ************************************************************************** */ WORD32 ih264d_process_inter_mb(dec_struct_t * ps_dec, dec_mb_info_t * ps_cur_mb_info, UWORD8 u1_mb_num) { /* CHANGED CODE */ UWORD8 *pu1_rec_y, *pu1_rec_u, *pu1_rec_v; /*CHANGED CODE */ UWORD32 ui_rec_width, u4_recwidth_cr; WORD16 *pi2_y_coeff; UWORD32 u1_mb_field_decoding_flag; const UWORD8 u1_mbaff = ps_dec->ps_cur_slice->u1_mbaff_frame_flag; UWORD32 uc_botMb; UWORD32 u4_num_pmbair; /* CHANGED CODE */ tfr_ctxt_t *ps_frame_buf = ps_dec->ps_frame_buf_ip_recon; UWORD32 u4_luma_dc_only_csbp = 0; UWORD32 u4_luma_dc_only_cbp = 0; /* CHANGED CODE */ uc_botMb = 1 - ps_cur_mb_info->u1_topmb; u4_num_pmbair = (u1_mb_num >> u1_mbaff); u1_mb_field_decoding_flag = ps_cur_mb_info->u1_mb_field_decodingflag; /* CHANGED CODE */ pu1_rec_y = ps_frame_buf->pu1_dest_y + (u4_num_pmbair << 4); pu1_rec_u = ps_frame_buf->pu1_dest_u + (u4_num_pmbair << 3) * YUV420SP_FACTOR; pu1_rec_v = ps_frame_buf->pu1_dest_v + (u4_num_pmbair << 3); ui_rec_width = ps_dec->u2_frm_wd_y << u1_mb_field_decoding_flag; u4_recwidth_cr = ps_dec->u2_frm_wd_uv << u1_mb_field_decoding_flag; /* CHANGED CODE */ if(u1_mbaff) { if(uc_botMb) { pu1_rec_y += (u1_mb_field_decoding_flag ? (ui_rec_width >> 1) : (ui_rec_width << 4)); pu1_rec_u += (u1_mb_field_decoding_flag ? (u4_recwidth_cr >> 1) : (u4_recwidth_cr << 3)); pu1_rec_v += (u1_mb_field_decoding_flag ? (u4_recwidth_cr >> 1) : (u4_recwidth_cr << 3)); } } if(!ps_cur_mb_info->u1_tran_form8x8) { u4_luma_dc_only_csbp = ih264d_unpack_luma_coeff4x4_mb(ps_dec, ps_cur_mb_info, 0); } else { if(!ps_dec->ps_cur_pps->u1_entropy_coding_mode) { u4_luma_dc_only_cbp = ih264d_unpack_luma_coeff4x4_mb(ps_dec, ps_cur_mb_info, 0); } else { u4_luma_dc_only_cbp = ih264d_unpack_luma_coeff8x8_mb(ps_dec, ps_cur_mb_info); } } pi2_y_coeff = ps_dec->pi2_coeff_data; /* Inverse Transform and Reconstruction */ if(ps_cur_mb_info->u1_cbp & 0x0f) { /* CHANGED CODE */ if(!ps_cur_mb_info->u1_tran_form8x8) { UWORD32 i; WORD16 ai2_tmp[16]; for(i = 0; i < 16; i++) { if(CHECKBIT(ps_cur_mb_info->u2_luma_csbp, i)) { WORD16 *pi2_level = pi2_y_coeff + (i << 4); UWORD8 *pu1_pred_sblk = pu1_rec_y + ((i & 0x3) * BLK_SIZE) + (i >> 2) * (ui_rec_width << 2); PROFILE_DISABLE_IQ_IT_RECON() { if(CHECKBIT(u4_luma_dc_only_csbp, i)) { ps_dec->pf_iquant_itrans_recon_luma_4x4_dc( pi2_level, pu1_pred_sblk, pu1_pred_sblk, ui_rec_width, ui_rec_width, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qp_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[3], ps_cur_mb_info->u1_qp_div6, ai2_tmp, 0, NULL); } else { ps_dec->pf_iquant_itrans_recon_luma_4x4( pi2_level, pu1_pred_sblk, pu1_pred_sblk, ui_rec_width, ui_rec_width, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qp_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[3], ps_cur_mb_info->u1_qp_div6, ai2_tmp, 0, NULL); } } } } } else { WORD16 *pi2_scale_matrix_ptr; WORD32 i; pi2_scale_matrix_ptr = ps_dec->s_high_profile.i2_scalinglist8x8[1]; for(i = 0; i < 4; i++) { WORD16 ai2_tmp[64]; WORD16 *pi16_levelBlock = pi2_y_coeff + (i << 6); /* move to the next 8x8 adding 64 */ UWORD8 *pu1_pred_sblk = pu1_rec_y + ((i & 0x1) * BLK8x8SIZE) + (i >> 1) * (ui_rec_width << 3); if(CHECKBIT(ps_cur_mb_info->u1_cbp, i)) { PROFILE_DISABLE_IQ_IT_RECON() { if(CHECKBIT(u4_luma_dc_only_cbp, i)) { ps_dec->pf_iquant_itrans_recon_luma_8x8_dc( pi16_levelBlock, pu1_pred_sblk, pu1_pred_sblk, ui_rec_width, ui_rec_width, gau1_ih264d_dequant8x8_cavlc[ps_cur_mb_info->u1_qp_rem6], (UWORD16 *)pi2_scale_matrix_ptr, ps_cur_mb_info->u1_qp_div6, ai2_tmp, 0, NULL); } else { ps_dec->pf_iquant_itrans_recon_luma_8x8( pi16_levelBlock, pu1_pred_sblk, pu1_pred_sblk, ui_rec_width, ui_rec_width, gau1_ih264d_dequant8x8_cavlc[ps_cur_mb_info->u1_qp_rem6], (UWORD16 *)pi2_scale_matrix_ptr, ps_cur_mb_info->u1_qp_div6, ai2_tmp, 0, NULL); } } } } } } /* Decode Chroma Block */ ih264d_unpack_chroma_coeff4x4_mb(ps_dec, ps_cur_mb_info); /*--------------------------------------------------------------------*/ /* Chroma Blocks decoding */ /*--------------------------------------------------------------------*/ { UWORD8 u1_chroma_cbp = (UWORD8)(ps_cur_mb_info->u1_cbp >> 4); if(u1_chroma_cbp != CBPC_ALLZERO) { UWORD32 u4_scale_u = ps_cur_mb_info->u1_qpc_div6; UWORD32 u4_scale_v = ps_cur_mb_info->u1_qpcr_div6; UWORD16 u2_chroma_csbp = ps_cur_mb_info->u2_chroma_csbp; pi2_y_coeff = ps_dec->pi2_coeff_data; { UWORD32 i; WORD16 ai2_tmp[16]; for(i = 0; i < 4; i++) { WORD16 *pi2_level = pi2_y_coeff + (i << 4); UWORD8 *pu1_pred_sblk = pu1_rec_u + ((i & 0x1) * BLK_SIZE * YUV420SP_FACTOR) + (i >> 1) * (u4_recwidth_cr << 2); PROFILE_DISABLE_IQ_IT_RECON() { if(CHECKBIT(u2_chroma_csbp, i)) { ps_dec->pf_iquant_itrans_recon_chroma_4x4( pi2_level, pu1_pred_sblk, pu1_pred_sblk, u4_recwidth_cr, u4_recwidth_cr, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qpc_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[4], u4_scale_u, ai2_tmp, pi2_level); } else if(pi2_level[0] != 0) { ps_dec->pf_iquant_itrans_recon_chroma_4x4_dc( pi2_level, pu1_pred_sblk, pu1_pred_sblk, u4_recwidth_cr, u4_recwidth_cr, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qpc_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[4], u4_scale_u, ai2_tmp, pi2_level); } } } } pi2_y_coeff += MB_CHROM_SIZE; u2_chroma_csbp >>= 4; { UWORD32 i; WORD16 ai2_tmp[16]; for(i = 0; i < 4; i++) { WORD16 *pi2_level = pi2_y_coeff + (i << 4); UWORD8 *pu1_pred_sblk = pu1_rec_u + 1 + ((i & 0x1) * BLK_SIZE * YUV420SP_FACTOR) + (i >> 1) * (u4_recwidth_cr << 2); PROFILE_DISABLE_IQ_IT_RECON() { if(CHECKBIT(u2_chroma_csbp, i)) { ps_dec->pf_iquant_itrans_recon_chroma_4x4( pi2_level, pu1_pred_sblk, pu1_pred_sblk, u4_recwidth_cr, u4_recwidth_cr, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qpcr_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[5], u4_scale_v, ai2_tmp, pi2_level); } else if(pi2_level[0] != 0) { ps_dec->pf_iquant_itrans_recon_chroma_4x4_dc( pi2_level, pu1_pred_sblk, pu1_pred_sblk, u4_recwidth_cr, u4_recwidth_cr, gau2_ih264_iquant_scale_4x4[ps_cur_mb_info->u1_qpcr_rem6], (UWORD16 *)ps_dec->s_high_profile.i2_scalinglist4x4[5], u4_scale_v, ai2_tmp, pi2_level); } } } } } } return (0); } /*! ************************************************************************** * \if Function name : ih264d_parse_pred_weight_table \endif * * \brief * Implements pred_weight_table() of 7.3.3.2. * * \return * None * ************************************************************************** */ WORD32 ih264d_parse_pred_weight_table(dec_slice_params_t * ps_cur_slice, dec_bit_stream_t * ps_bitstrm) { UWORD32 *pu4_bitstrm_buf = ps_bitstrm->pu4_buffer; UWORD32 *pu4_bitstrm_ofst = &ps_bitstrm->u4_ofst; WORD8 i, cont, lx; UWORD8 uc_weight_flag; UWORD32 *pui32_weight_offset_lx; WORD16 c_weight, c_offset; UWORD32 ui32_y_def_weight_ofst, ui32_cr_def_weight_ofst; UWORD32 ui32_temp; UWORD8 uc_luma_log2_weight_denom; UWORD8 uc_chroma_log2_weight_denom; /* Variables for error resilience checks */ UWORD32 u4_temp; WORD32 i_temp; u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if(u4_temp > MAX_LOG2_WEIGHT_DENOM) { return ERROR_PRED_WEIGHT_TABLE_T; } uc_luma_log2_weight_denom = u4_temp; COPYTHECONTEXT("SH: luma_log2_weight_denom",uc_luma_log2_weight_denom); ui32_y_def_weight_ofst = (1 << uc_luma_log2_weight_denom); u4_temp = ih264d_uev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if(u4_temp > MAX_LOG2_WEIGHT_DENOM) { return ERROR_PRED_WEIGHT_TABLE_T; } uc_chroma_log2_weight_denom = u4_temp; COPYTHECONTEXT("SH: chroma_log2_weight_denom",uc_chroma_log2_weight_denom); ui32_cr_def_weight_ofst = (1 << uc_chroma_log2_weight_denom); ps_cur_slice->u2_log2Y_crwd = uc_luma_log2_weight_denom | (uc_chroma_log2_weight_denom << 8); cont = (ps_cur_slice->u1_slice_type == B_SLICE); lx = 0; do { for(i = 0; i < ps_cur_slice->u1_num_ref_idx_lx_active[lx]; i++) { pui32_weight_offset_lx = ps_cur_slice->u4_wt_ofst_lx[lx][i]; uc_weight_flag = ih264d_get_bit_h264(ps_bitstrm); pu4_bitstrm_buf = ps_bitstrm->pu4_buffer; COPYTHECONTEXT("SH: luma_weight_l0_flag",uc_weight_flag); if(uc_weight_flag) { i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_weight = i_temp; COPYTHECONTEXT("SH: luma_weight_l0",c_weight); i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_offset = i_temp; COPYTHECONTEXT("SH: luma_offset_l0",c_offset); ui32_temp = (c_offset << 16) | (c_weight & 0xFFFF); pui32_weight_offset_lx[0] = ui32_temp; } else { pui32_weight_offset_lx[0] = ui32_y_def_weight_ofst; } { WORD8 c_weightCb, c_weightCr, c_offsetCb, c_offsetCr; uc_weight_flag = ih264d_get_bit_h264(ps_bitstrm); pu4_bitstrm_buf = ps_bitstrm->pu4_buffer; COPYTHECONTEXT("SH: chroma_weight_l0_flag",uc_weight_flag); if(uc_weight_flag) { i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_weightCb = i_temp; COPYTHECONTEXT("SH: chroma_weight_l0",c_weightCb); i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_offsetCb = i_temp; COPYTHECONTEXT("SH: chroma_weight_l0",c_offsetCb); ui32_temp = (c_offsetCb << 16) | (c_weightCb & 0xFFFF); pui32_weight_offset_lx[1] = ui32_temp; i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_weightCr = i_temp; COPYTHECONTEXT("SH: chroma_weight_l0",c_weightCr); i_temp = ih264d_sev(pu4_bitstrm_ofst, pu4_bitstrm_buf); if((i_temp < PRED_WEIGHT_MIN) || (i_temp > PRED_WEIGHT_MAX)) return ERROR_PRED_WEIGHT_TABLE_T; c_offsetCr = i_temp; COPYTHECONTEXT("SH: chroma_weight_l0",c_offsetCr); ui32_temp = (c_offsetCr << 16) | (c_weightCr & 0xFFFF); pui32_weight_offset_lx[2] = ui32_temp; } else { pui32_weight_offset_lx[1] = ui32_cr_def_weight_ofst; pui32_weight_offset_lx[2] = ui32_cr_def_weight_ofst; } } } lx++; } while(cont--); return OK; } static int pic_num_compare(const void *pv_pic1, const void *pv_pic2) { struct pic_buffer_t *ps_pic1 = *(struct pic_buffer_t **) pv_pic1; struct pic_buffer_t *ps_pic2 = *(struct pic_buffer_t **) pv_pic2; if (ps_pic1->i4_pic_num < ps_pic2->i4_pic_num) { return -1; } else if (ps_pic1->i4_pic_num > ps_pic2->i4_pic_num) { return 1; } else { return 0; } } /*****************************************************************************/ /* */ /* Function Name : ih264d_init_ref_idx_lx_p */ /* */ /* Description : This function initializes the reference picture L0 list */ /* for P slices as per section 8.2.4.2.1 and 8.2.4.2.2. */ /* */ /* Inputs : pointer to ps_dec struture */ /* Globals : NO */ /* Processing : arranges all the short term pictures according to */ /* pic_num in descending order starting from curr pic_num. */ /* and inserts it in L0 list followed by all Long term */ /* pictures in ascending order. */ /* */ /* Returns : void */ /* */ /* Issues : */ /* */ /* Revision History: */ /* */ /* DD MM YYYY Author(s) Changes (Describe the changes made) */ /* 13 07 2002 Jay Draft */ /* */ /*****************************************************************************/ void ih264d_init_ref_idx_lx_p(dec_struct_t *ps_dec) { struct pic_buffer_t *ps_ref_pic_buf_lx; dpb_manager_t *ps_dpb_mgr; struct dpb_info_t *ps_next_dpb; WORD8 i, j; UWORD8 u1_max_lt_index, u1_min_lt_index; UWORD32 u4_lt_index; UWORD8 u1_field_pic_flag; dec_slice_params_t *ps_cur_slice; UWORD8 u1_L0; WORD32 i4_cur_pic_num, i4_min_st_pic_num; WORD32 i4_temp_pic_num, i4_ref_pic_num; UWORD8 u1_num_short_term_bufs; UWORD8 u1_max_ref_idx_l0; struct pic_buffer_t *aps_st_pic_bufs[2 * MAX_REF_BUFS] = {NULL}; ps_cur_slice = ps_dec->ps_cur_slice; u1_field_pic_flag = ps_cur_slice->u1_field_pic_flag; u1_max_ref_idx_l0 = ps_cur_slice->u1_num_ref_idx_lx_active[0] << u1_field_pic_flag; ps_dpb_mgr = ps_dec->ps_dpb_mgr; /* Get the current frame number */ i4_cur_pic_num = ps_dec->ps_cur_pic->i4_pic_num; /* Get Min pic_num,MinLt */ i4_min_st_pic_num = i4_cur_pic_num; u1_max_lt_index = MAX_REF_BUFS + 1; u1_min_lt_index = MAX_REF_BUFS + 1; /* Start from ST head */ ps_next_dpb = ps_dpb_mgr->ps_dpb_st_head; for(i = 0; i < ps_dpb_mgr->u1_num_st_ref_bufs; i++) { i4_ref_pic_num = ps_next_dpb->ps_pic_buf->i4_pic_num; if(i4_ref_pic_num < i4_cur_pic_num) { /* RefPic Buf pic_num is before Current pic_num in decode order */ i4_min_st_pic_num = MIN(i4_min_st_pic_num, i4_ref_pic_num); } /* Chase the next link */ ps_next_dpb = ps_next_dpb->ps_prev_short; } /* Sort ST ref pocs in ascending order */ ps_next_dpb = ps_dpb_mgr->ps_dpb_st_head; for (j = 0; j < ps_dpb_mgr->u1_num_st_ref_bufs; j++) { aps_st_pic_bufs[j] = ps_next_dpb->ps_pic_buf; ps_next_dpb = ps_next_dpb->ps_prev_short; } qsort(aps_st_pic_bufs, ps_dpb_mgr->u1_num_st_ref_bufs, sizeof(aps_st_pic_bufs[0]), pic_num_compare); /* Start from LT head */ ps_next_dpb = ps_dpb_mgr->ps_dpb_ht_head; if(ps_next_dpb) { u1_max_lt_index = ps_next_dpb->u1_lt_idx; u1_min_lt_index = ps_next_dpb->u1_lt_idx; for(i = 0; i < ps_dpb_mgr->u1_num_lt_ref_bufs; i++) { u4_lt_index = ps_next_dpb->u1_lt_idx; u1_max_lt_index = (UWORD8)(MAX(u1_max_lt_index, u4_lt_index)); u1_min_lt_index = (UWORD8)(MIN(u1_min_lt_index, u4_lt_index)); /* Chase the next link */ ps_next_dpb = ps_next_dpb->ps_prev_long; } } /* 1. Initialize refIdxL0 */ u1_L0 = 0; if(u1_field_pic_flag) { ps_ref_pic_buf_lx = ps_dpb_mgr->ps_init_dpb[0][0]; ps_ref_pic_buf_lx += MAX_REF_BUFS; i4_temp_pic_num = i4_cur_pic_num; } else { ps_ref_pic_buf_lx = ps_dpb_mgr->ps_init_dpb[0][0]; i4_temp_pic_num = i4_cur_pic_num; } /* Arrange all short term buffers in output order as given by pic_num */ /* Arrange pic_num's less than Curr pic_num in the descending pic_num */ /* order starting from (Curr pic_num - 1) */ for(j = ps_dpb_mgr->u1_num_st_ref_bufs - 1; j >= 0; j--) { if(aps_st_pic_bufs[j]) { /* Copy info in pic buffer */ ih264d_insert_pic_in_ref_pic_listx(ps_ref_pic_buf_lx, aps_st_pic_bufs[j]); ps_ref_pic_buf_lx++; u1_L0++; } } /* Arrange all Long term buffers in ascending order, in LongtermIndex */ /* Start from LT head */ u1_num_short_term_bufs = u1_L0; for(u4_lt_index = u1_min_lt_index; u4_lt_index <= u1_max_lt_index; u4_lt_index++) { ps_next_dpb = ps_dpb_mgr->ps_dpb_ht_head; for(i = 0; i < ps_dpb_mgr->u1_num_lt_ref_bufs; i++) { if(ps_next_dpb->u1_lt_idx == u4_lt_index) { ih264d_insert_pic_in_ref_pic_listx(ps_ref_pic_buf_lx, ps_next_dpb->ps_pic_buf); ps_ref_pic_buf_lx->u1_long_term_pic_num = ps_ref_pic_buf_lx->u1_long_term_frm_idx; ps_ref_pic_buf_lx++; u1_L0++; break; } ps_next_dpb = ps_next_dpb->ps_prev_long; } } if(u1_field_pic_flag) { /* Initialize the rest of the entries in the */ /* reference list to handle of errors */ { UWORD8 u1_i; pic_buffer_t ref_pic; ref_pic = *(ps_dpb_mgr->ps_init_dpb[0][0] + MAX_REF_BUFS); if(NULL == ref_pic.pu1_buf1) { ref_pic = *ps_dec->ps_cur_pic; } for(u1_i = u1_L0; u1_i < u1_max_ref_idx_l0; u1_i++) { *ps_ref_pic_buf_lx = ref_pic; ps_ref_pic_buf_lx++; } } ih264d_convert_frm_to_fld_list( ps_dpb_mgr->ps_init_dpb[0][0] + MAX_REF_BUFS, &u1_L0, ps_dec, u1_num_short_term_bufs); ps_ref_pic_buf_lx = ps_dpb_mgr->ps_init_dpb[0][0] + u1_L0; } /* Initialize the rest of the entries in the */ /* reference list to handle of errors */ { UWORD8 u1_i; pic_buffer_t ref_pic; ref_pic = *(ps_dpb_mgr->ps_init_dpb[0][0]); if(NULL == ref_pic.pu1_buf1) { ref_pic = *ps_dec->ps_cur_pic; } for(u1_i = u1_L0; u1_i < u1_max_ref_idx_l0; u1_i++) { *ps_ref_pic_buf_lx = ref_pic; ps_ref_pic_buf_lx++; } } ps_dec->ps_cur_slice->u1_initial_list_size[0] = u1_L0; }